StateRepresentation.hpp

 
#ifndef GYM_PCB_RL_STATEREPRESENTATION_H
#define GYM_PCB_RL_STATEREPRESENTATION_H
 
#include "Py.hpp"
 
class Action;
class PCBoard;
class Connection;


 
class StateRepresentation
{
public:
    static StateRepresentation *create(PyObject *);
public:
    virtual PyObject *getPy(PyObject *args) { return py::None(); }
 
    virtual void init(PCBoard &PCB) { mPCB = &PCB; }
    virtual ~StateRepresentation() { }
    virtual void reset() { }
    virtual const char *name() const { return ""; }

    void setFocus(const Connection *);
 
    virtual void setFocus(const std::vector<Connection *> *) { }
 
    virtual void updateView(const Bbox_2 * = 0) { }
 
    void setLockedView(bool v) { mLockedView = v; }

    virtual void logAction(const Action *) { }
protected:
    PCBoard *mPCB{0};
    const Connection *mFocus{0};
    bool mLockedView{false};
};
 
inline void StateRepresentation::setFocus(const Connection *X)
{
    mFocus = X;
}
 
#endif // GYM_PCB_RL_STATEREPRESENTATION_H