actionSpaceBuilder.h

Go to the documentation of this file.

#ifndef ACTIONSPACEBUILDER_H
#define ACTIONSPACEBUILDER_H

#include <memory>
#include <vector>

#include <utils/counter.h>

namespace Mdp
{

class Action;
class ActionSpace;
class ActionDimension;
class DomainModel;

class ActionSpaceBuilder
{
public:
    void setDomainModel(std::shared_ptr<DomainModel> model);
    void addAction(Action *action);
    std::shared_ptr<ActionSpace> getActionSpace();
    void addActionDimension(ActionDimension*);
private:
    std::vector<Action*> actions;
    std::shared_ptr<DomainModel> domainModel;
    std::vector<ActionDimension*> dimensions;
    Utils::Counter actionDimensionCounter;
};

}


#endif