PyModule.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <boost/python/call_method.hpp>
#include <framework/core/Module.h>
 
namespace Belle2 {
  class PyModule : public Module {
  public:
    explicit PyModule(PyObject* p):
      Module(),
      m_self(p)
    {
      setName(p->ob_type->tp_name);
      setType("PyModule");
    }
 
    PyModule(PyObject* p, const Module& m):
      Module(m),
      m_self(p)
    {
      setName(p->ob_type->tp_name);
      setType("PyModule");
    }
 
 
    /* reimplement all virtual functions of base class and call the corresponding method in the python class
    *  note that all methods must be registered in Module::exposePythonAPI()
    */
 
    virtual void initialize() override final
    {
      boost::python::call_method<void>(m_self, "initialize");
    }
 
    virtual void beginRun() override final
    {
      boost::python::call_method<void>(m_self, "beginRun");
    }
 
    virtual void event() override final
    {
      boost::python::call_method<void>(m_self, "event");
    }
 
    virtual void endRun() override final
    {
      boost::python::call_method<void>(m_self, "endRun");
    }
 
    virtual void terminate() override final
    {
      boost::python::call_method<void>(m_self, "terminate");
    }
 
  private:
 
    virtual void def_initialize() override final
    {
      // Call empty base implementation if initialize() is not overriden from Python.
      Module::initialize();
    }
 
    virtual void def_beginRun() override final
    {
      // Call empty base implementation if beginRun() is not overriden from Python.
      Module::beginRun();
    }
 
    virtual void def_event() override final
    {
      // Call empty base implementation if event() is not overriden from Python.
      Module::event();
    }
 
    virtual void def_endRun() override final
    {
      // Call empty base implementation if endRun() is not overriden from Python.
      Module::endRun();
    }
 
    virtual void def_terminate() override final
    {
      // Call empty base implementation if terminate() is not overriden from Python.
      Module::terminate();
    }
 
    PyObject* m_self; 
  };
}