StateIndexArray.h

#ifndef STATEINDEXARRAY_H
#define STATEINDEXARRAY_H
 
#include <boost/pool/object_pool.hpp>
 
#include <vector>
#include <cstdint>
#include <memory>
 
namespace stamina {
    // Pool pointer is not within the util namespace
    /* template <typename T, typename P = boost::object_pool<T>>
    struct poolDeleter {
 
    private:
        P & pool;
    }; */
    namespace util {
        template <typename StateType, typename ProbabilityStateType>
        class StateIndexArray {
        public:
            StateIndexArray(uint8_t blockSizeExponent = 15); // 2 ^ 15
            ~StateIndexArray();
            void clear();
            void reserve(uint32_t numToReserve);
            ProbabilityStateType * get(StateType index);
            void put(StateType index, ProbabilityStateType * probabilityState);
            std::vector<StateType> getPerimeterStates();
            uint32_t getNumberTerminal();
        protected:
            uint32_t sizeToActualSize(uint32_t size);
        private:
            uint32_t numElements;
            uint32_t blockSize;
            std::vector<std::shared_ptr<ProbabilityStateType *>> stateArray;
        };
    }
}
 
#endif // STATEINDEXARRAY_H