MyPyModule.cc

Go to the documentation of this file.

#include "MyPyModule.h"

#include <fwk/CentralConfig.h>
#include <utl/ErrorLogger.h>
#include <utl/Reader.h>
#include <evt/Event.h>

// everything needed for embedding
#include <Eigen/Core>
#include <pybind11/embed.h> // everything needed for embedding
#include <pybind11/eigen.h> // required for auto-conversion of eigen matrix

#include <iostream>

using namespace std;
using namespace utl;
using namespace fwk;

using Mat = Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic>;


namespace MyPyModuleNS {

  VModule::ResultFlag
  MyPyModule::Init()
  {
    INFO("Loading XML-config");
    CentralConfig& cc = *CentralConfig::GetInstance();

    Branch topB = cc.GetTopBranch("MyPyModule");

    topB.GetChild("infoLevel").GetData(fInfoLevel);
    topB.GetChild("filename").GetData(fFilename);

    INFO("Initialize the python interpreter.");
    pybind11::initialize_interpreter();

    INFO("Initialize main module and main class.");
    fModule = pybind11::module_::import(fFilename.c_str());
    fMainClass = fModule.attr("Dog")();

    return eSuccess;
  }


  VModule::ResultFlag
  MyPyModule::Run(evt::Event& event)
  {
    // (0) prepare data
    typedef Eigen::Matrix<float, 10, 10> M;
    auto m1 = M::Zero();
    auto m2 = M::Zero();

    // (1) prepare data
    fMainClass.attr("bark")("It works!");
    fMainClass.attr("woof")();

    pybind11::tuple args = pybind11::make_tuple(m1, m2);
    pybind11::object result_py = fMainClass.attr("fetch_neural_network_result")(args);

    const auto result = result_py.cast<Mat>();

    cout << "result has the values " << result << "\n"
            "result is of size     " << result.rows() << 'x' << result.cols() << std::endl;

    return eSuccess;
  }


  VModule::ResultFlag
  MyPyModule::Finish()
  {
    // Put any termination or cleanup code here.
    // This method is called once at the end of the run.

    INFO("Finalize the python interpreter.");

    /* We have to do this here to not run into memory leaks due to
       still living (py-)objects before the interpreter shuts down. */

    fMainClass.dec_ref();
    fModule.dec_ref();

    pybind11::finalize_interpreter();

    return eSuccess;
  }

}