PythonExpr.h

Go to the documentation of this file.

#ifndef __PYTHON_EXPR_H__
#define __PYTHON_EXPR_H__

#include "boost/tuple/tuple.hpp"
#include "boost/variant/get.hpp"
#include "boost/variant/variant.hpp"

#include <boost/python.hpp>
#include <boost/python/suite/indexing/vector_indexing_suite.hpp>

#include "Util/Types.h"
#include "Util/OptPilotException.h"
#include "Expression/Expression.h"
#include "Expression/Parser/function.hpp"

using namespace boost::python;

struct PythonExpression {

    Expressions::Result_t operator()(client::function::arguments_t args) {
        std::string script = boost::get<std::string>(args[0]);
        std::vector<double> pargs;
        for(size_t i = 1; i < args.size(); i++) {
            pargs.push_back(boost::get<double>(args[i]));
        }

        try {
            Py_Initialize();
            object main_module    = import("__main__");
            object main_namespace = main_module.attr("__dict__");

            boost::python::class_<std::vector<double> >("PyVec")
                .def(boost::python::vector_indexing_suite<std::vector<double> >());
            main_namespace["arguments"] = pargs;

            object ignored = exec_file(script.c_str(), main_namespace);
            double res = extract<double>(main_namespace["result"]);

            Py_Finalize();
            return boost::make_tuple(res, true);

        } catch (error_already_set) {
            PyErr_Print();
            return boost::make_tuple(0.0, false);
        }
    }
};


#endif