Stack.h

Go to the documentation of this file.

 
#ifndef PBAT_COMMON_STACK_H
#define PBAT_COMMON_STACK_H
 
#include "pbat/HostDevice.h"
 
namespace pbat {
namespace common {

template <class T, auto kCapacity = 64>
class Stack
{
  public:
    PBAT_HOST_DEVICE Stack() : stack{}, size{0} {}
    PBAT_HOST_DEVICE void Push(T value) { stack[size++] = value; }
    PBAT_HOST_DEVICE T Pop() { return stack[--size]; }
    PBAT_HOST_DEVICE T const& Top() const { return stack[size - 1]; }
    PBAT_HOST_DEVICE auto Size() const { return size; }
    PBAT_HOST_DEVICE bool IsEmpty() const { return size == 0; }
    PBAT_HOST_DEVICE bool IsFull() const { return size == kCapacity; }
    PBAT_HOST_DEVICE void Clear() { size = 0; }
    PBAT_HOST_DEVICE T& operator[](auto i) { return stack[i]; }
    PBAT_HOST_DEVICE T const& operator[](auto i) const { return stack[i]; }
    PBAT_HOST_DEVICE T* begin() { return stack; }
    PBAT_HOST_DEVICE T* end() { return stack + size; }
 
  private:
    T stack[kCapacity];
    int size; 
};
 
} // namespace common
} // namespace pbat
 
#endif // PBAT_COMMON_STACK_H