Stack.hpp 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. #ifndef CORE_STACK_HPP
  2. #define CORE_STACK_HPP
  3. #include "core/data/ArrayList.hpp"
  4. #include "core/data/List.hpp"
  5. #include "core/utils/Error.hpp"
  6. #include "core/utils/Meta.hpp"
  7. namespace Core {
  8. namespace Internal {
  9. template<typename T, typename S>
  10. class BaseStack final {
  11. S data{};
  12. public:
  13. template<typename... Args>
  14. check_return Error push(Args&&... args) {
  15. if constexpr(Core::IsSame<S, List<T>>) {
  16. data.add(Core::forward<Args>(args)...);
  17. return ErrorCode::NONE;
  18. } else {
  19. return data.add(Core::forward<Args>(args)...);
  20. }
  21. }
  22. void clear() {
  23. data.clear();
  24. }
  25. void pop() {
  26. assert(data.getLength() > 0);
  27. data.removeBySwap(data.getLength() - 1);
  28. }
  29. bool isEmpty() const {
  30. return data.getLength() == 0;
  31. }
  32. T& peek() {
  33. assert(data.getLength() > 0);
  34. return data[data.getLength() - 1];
  35. }
  36. const T& peek() const {
  37. assert(data.getLength() > 0);
  38. return data[data.getLength() - 1];
  39. }
  40. template<typename String>
  41. check_return Error toString(String& s) const {
  42. return s.append(data);
  43. }
  44. };
  45. }
  46. template<typename T>
  47. using ListStack = Internal::BaseStack<T, List<T>>;
  48. template<typename T, size_t N>
  49. using ArrayStack = Internal::BaseStack<T, ArrayList<T, N>>;
  50. }
  51. #endif