fwkPython.cc

Go to the documentation of this file.

#include <boost/python.hpp>
#include <boost/preprocessor.hpp>
using namespace boost::python;

#include <evt/Event.h>

#include <fwk/RunController.h>
#include <fwk/CentralConfig.h>
#include <fwk/VModule.h>

#include <string>

void Config(const std::string& infile)
{
    fwk::CentralConfig::GetInstance(infile, false);
}

void WriteConfig(std::string filename)
{
    fwk::CentralConfig::GetInstance()->WriteConfig(filename);
}

std::string GetConfig()
{
    return fwk::CentralConfig::GetInstance()->GetConfig();
}

struct VModuleWrap : fwk::VModule, wrapper<fwk::VModule>
{
    fwk::VModule::ResultFlag Init()
        {
            return this->get_override("Init")();
        }

    fwk::VModule::ResultFlag Run(evt::Event& event)
        {
            return this->get_override("Run")(event);
        }

    fwk::VModule::ResultFlag Finish()
        {
            return this->get_override("Finish")();
        }
};

// CentralConfig
// NOTE: still can't get this to work; no use banging my head...
/*
static fwk::CentralConfig* (fwk::CentralConfig::*get_instance2)(const std::string&,
                                                                const bool) = &fwk::CentralConfig::GetInstance;

void (fwk::CentralConfig::*write_config1)(std::string) = &fwk::CentralConfig::WriteConfig;
*/

// RunController
fwk::VModule& (fwk::RunController::*get_module)(const std::string&) const = &fwk::RunController::GetModule;

// VModule
fwk::VModule::ResultFlag (fwk::VModule::*run)(evt::Event&) = &fwk::VModule::Run;

void fwkPython()
{
    class_<fwk::RunController, boost::noncopyable>("RunController", no_init)
        .def("GetInstance", &fwk::RunController::GetInstance,
             return_value_policy<reference_existing_object>())
        .staticmethod("GetInstance")
        .def("Init", &fwk::RunController::Init)
        .def("Run", &fwk::RunController::Run)
        .def("Finish", &fwk::RunController::Finish)
        .def("GetNumberOfRegisteredModules", &fwk::RunController::GetNumberOfRegisteredModules)
        .def("GetRegisteredModuleNames", &fwk::RunController::GetRegisteredModuleNames)
        .def("GetModule", get_module,
             return_internal_reference<>());

    def("Config", Config);
    def("WriteConfig", WriteConfig);
    def("GetConfig", GetConfig);

    /*
    class_<fwk::CentralConfig, boost::noncopyable>("CentralConfig", no_init)
        .def("GetInstance", &fwk::CentralConfig::GetInstance,
             return_value_policy<reference_existing_object>())
        .staticmethod("GetInstance")
        .def("GetInstance", get_instance2,
             return_value_policy<reference_existing_object>())
        .staticmethod("GetInstance")
        .def("WriteConfig", write_config1)
        .def("GetConfig", &fwk::CentralConfig::GetConfig);
    */

    enum_<fwk::VModule::ResultFlag>("ResultFlag")
        .value("eSuccess",      fwk::VModule::eSuccess)
        .value("eFailure",      fwk::VModule::eFailure)
        .value("eBreakLoop",    fwk::VModule::eBreakLoop)
        .value("eContinueLoop", fwk::VModule::eContinueLoop);

    class_<VModuleWrap, boost::noncopyable>("VModule")
        .def("Init",   pure_virtual(&fwk::VModule::Init))
        .def("Run",    pure_virtual(run))
        .def("Finish", pure_virtual(&fwk::VModule::Finish));
}