GenerativeModelPython.hpp

Go to the documentation of this file.

#ifndef AI_TOOLBOX_MDP_GENERATIVE_MODEL_PYTHON_HEADER_FILE
#define AI_TOOLBOX_MDP_GENERATIVE_MODEL_PYTHON_HEADER_FILE
 
#include <boost/python.hpp>
#include <boost/python/object.hpp>
 
namespace AIToolbox::MDP {
    class GenerativeModelPython {
        public:
            GenerativeModelPython(boost::python::object instance) :
                    instance_(instance) {}
 
            size_t getS() const { return boost::python::extract<size_t>(instance_.attr("getS")()); }
 
            size_t getA() const { return boost::python::extract<size_t>(instance_.attr("getA")()); }
 
            double getDiscount() const { return boost::python::extract<double>(instance_.attr("getDiscount")()); }
 
            bool isTerminal(const size_t s) const { return boost::python::extract<bool>(instance_.attr("isTerminal")(s)); }
 
            std::tuple<size_t, double> sampleSR(const size_t s, const size_t a) const {
                return boost::python::extract<std::tuple<size_t, double>>(instance_.attr("sampleSR")(s, a));
            }
 
        private:
            boost::python::object instance_;
    };
}
 
#endif