stateSpace.h

Go to the documentation of this file.

#ifndef MDP_STATE_SPACE_H
#define MDP_STATE_SPACE_H

#include <memory>
#include <vector>

#include "action_impl.h"
#include "state.h"
#include "stateSpaceDimension.h"


namespace Mdp
{

typedef std::vector<statePosition_t> StateInternal;

class DomainModel;
class PriorityState;

class StateSpace
{
    friend class StateSpaceBuilder;
public:
    StateSpace(const StateSpace&) = delete;
    StateSpace& operator=(const StateSpace&) = delete;

    ~StateSpace();
    virtual size_t size();
    virtual state_t getState();
    virtual void updateCurrentState();
    virtual double getReward();
    virtual size_t getNbOfDimensions();
    virtual std::vector<size_t> *factorize(state_t state);
    virtual std::vector<size_t> getDimensionSizes();
protected:
    StateSpace(){};
    void updateInternalState();
    int getPriorityStateInternal();
    /*state space has to be constructed with the builder*/
    StateSpace(size_t nbOfStates, size_t nbOfPriorityStates,
        std::vector<StateSpaceDimension *> dimensions,
        std::vector<PriorityState *> priorityStates);
    state_t convertState(StateInternal);

    size_t nbOfStates{0}; /*excluding priority states*/
    size_t nbOfPriorityStates{0};
    std::vector<StateSpaceDimension *> dimensions;
    std::vector<PriorityState *> priorityStates;
    StateInternal currentState;
    std::shared_ptr<DomainModel> domainModel{nullptr};
    state_t currentExternalState{0};

    StateInternal internalState;

};





}




#endif