evtPython.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 <evt/Header.h>
#include <evt/ShowerRecData.h>
#include <evt/ShowerSRecData.h>
#include <evt/ShowerFRecData.h>

// Event
const evt::ShowerRecData& (evt::Event::*get_recshower1)() const = &evt::Event::GetRecShower;
evt::ShowerRecData& (evt::Event::*get_recshower2)() = &evt::Event::GetRecShower;

// Header
const evt::Header& (evt::Event::*get_header1)() const = &evt::Event::GetHeader;
evt::Header& (evt::Event::*get_header2)() = &evt::Event::GetHeader;

// ShowerRecData
const evt::ShowerSRecData& (evt::ShowerRecData::*get_srecshower1)() const = &evt::ShowerRecData::GetSRecShower;
evt::ShowerSRecData& (evt::ShowerRecData::*get_srecshower2)() = &evt::ShowerRecData::GetSRecShower;

const evt::ShowerFRecData& (evt::ShowerRecData::*get_frecshower1)() const = &evt::ShowerRecData::GetFRecShower;
evt::ShowerFRecData& (evt::ShowerRecData::*get_frecshower2)() = &evt::ShowerRecData::GetFRecShower;

// ShowerSRecData
void (evt::ShowerSRecData::*set_energy)(const double, const double) = &evt::ShowerSRecData::SetEnergy;
void (evt::ShowerSRecData::*set_s1000)(const double, const double) = &evt::ShowerSRecData::SetS1000;

// ShowerFRecData
void (evt::ShowerFRecData::*set_totalenergy)(const double, const double,
                                             evt::ShowerFRecData::EUncertaintyType) = &evt::ShowerFRecData::SetTotalEnergy;

void evtPython()
{
    class_<evt::Event>("Event")
        .def("GetHeader", get_header1,
             return_internal_reference<>())
        .def("HasRawEvent",   &evt::Event::HasRawEvent)
        //.def("MakeRawEvent",  &evt::Event::MakeRawEvent)
        .def("HasSEvent",     &evt::Event::HasSEvent)
        //.def("MakeSEvent",    &evt::Event::MakeSEvent)
        .def("HasFEvent",     &evt::Event::HasFEvent)
        //.def("MakeFEvent",    &evt::Event::MakeFEvent)
        .def("HasRecShower",  &evt::Event::HasRecShower)
        //.def("MakeRecShower", &evt::Event::MakeRecShower)
        .def("HasSimShower",  &evt::Event::HasSimShower)
        //.def("MakeSimShower", &evt::Event::MakeSimShower)
        .def("GetRecShower", get_recshower1,
             return_internal_reference<>());

    class_<evt::Header>("Header")
        .def("GetId", &evt::Header::GetId)
        .def("GetTime", &evt::Header::GetTime,
             return_value_policy<return_by_value>());

    class_<evt::ShowerRecData, boost::noncopyable>("ShowerRecData", no_init)
        .def("GetSRecShower", get_srecshower1,
             return_internal_reference<>())
        .def("GetFRecShower", get_frecshower1,
             return_internal_reference<>());

    class_<evt::ShowerFRecData, boost::noncopyable>("ShowerFRecData", no_init)
        .def("GetTotalEnergy", &evt::ShowerFRecData::GetTotalEnergy)
        .def("SetTotalEnergy", set_totalenergy)
        .def("GetTotalEnergyError", &evt::ShowerFRecData::GetTotalEnergyError);

    enum_<evt::ShowerFRecData::EUncertaintyType>("EUncertaintyType")
        .value("eTotal",       evt::ShowerFRecData::eTotal)
        .value("eStatistical", evt::ShowerFRecData::eStatistical)
        .value("eAtmospheric", evt::ShowerFRecData::eAtmospheric);

    enum_<evt::ShowerSRecData::T4Flags>("T4Flags")
        .value("eT4_None", evt::ShowerSRecData::eT4_None)
        .value("eT4_FD",   evt::ShowerSRecData::eT4_FD)
        .value("eT4_3TOT", evt::ShowerSRecData::eT4_3TOT)
        .value("eT4_4C1",  evt::ShowerSRecData::eT4_4C1);

    enum_<evt::ShowerSRecData::T5Flags>("T5Flags")
        .value("eT5_None",      evt::ShowerSRecData::eT5_None)
        .value("eT5_Prior",     evt::ShowerSRecData::eT5_Prior)
        .value("eT5_Posterior", evt::ShowerSRecData::eT5_Posterior);

    class_<evt::ShowerSRecData, boost::noncopyable>("ShowerSRecData", no_init)
        .def("GetAngleChi2", &evt::ShowerSRecData::GetAngleChi2)
        .def("GetAngleNdof", &evt::ShowerSRecData::GetAngleNdof)
        //.def("GetAxis")
        .def("GetBadPeriodId", &evt::ShowerSRecData::GetBadPeriodId)
        //.def("GetBarycenter")
        .def("GetBeta", &evt::ShowerSRecData::GetBeta)
        .def("GetBetaError", &evt::ShowerSRecData::GetBetaError)
        .def("GetBetaSystematics", &evt::ShowerSRecData::GetBetaSystematics)
        //.def("GetCoreError")
        //.def("GetCorePosition")
        //.def("GetCoreTime")
        //.def("GetCoreTimeError")
        .def("GetCurvature", &evt::ShowerSRecData::GetCurvature)
        .def("GetCurvatureError", &evt::ShowerSRecData::GetCurvatureError)
        .def("GetEnergy",  &evt::ShowerSRecData::GetEnergy)
        .def("GetEnergyError", &evt::ShowerSRecData::GetEnergyError)
        //.def("GetEstimatedAxis")
        //.def("GetEstimatedCore")
        .def("GetGamma", &evt::ShowerSRecData::GetGamma)
        .def("GetGammaError", &evt::ShowerSRecData::GetGammaError)
        //.def("GetLDF")
        .def("GetLDFChi2", &evt::ShowerSRecData::GetLDFChi2)
        //.def("GetLDFLikelihood", &evt::ShowerSRecData::GetLDFLikelihood)
        .def("GetLDFNdof", &evt::ShowerSRecData::GetLDFNdof)
        .def("GetLDFOptimumDistance", &evt::ShowerSRecData::GetLDFOptimumDistance)
        .def("GetLDFRecStage", &evt::ShowerSRecData::GetLDFRecStage)
        //.def("GetLogProductionDistance")
        .def("GetNumberOfActiveStations", &evt::ShowerSRecData::GetNumberOfActiveStations)
        .def("GetPhiError", &evt::ShowerSRecData::GetPhiError)
        //.def("GetReconstructionSeed")
        .def("SetS1000", set_s1000)
        .def("GetS1000", &evt::ShowerSRecData::GetS1000)
        .def("GetS1000Error", &evt::ShowerSRecData::GetS1000Error)
        .def("GetS1000Systematics", &evt::ShowerSRecData::GetS1000Systematics)
        //.def("GetShowerCenter")
        .def("GetT4Trigger", &evt::ShowerSRecData::GetT4Trigger)
        .def("GetT5Trigger", &evt::ShowerSRecData::GetT5Trigger)
        .def("GetThetaError", &evt::ShowerSRecData::GetThetaError)
        .def("GetTimeResidualMean", &evt::ShowerSRecData::GetTimeResidualMean)
        .def("GetTimeResidualSpread", &evt::ShowerSRecData::GetTimeResidualSpread)
        .def("HasLDF", &evt::ShowerSRecData::HasLDF)
        .def("HasLogProductionDistance", &evt::ShowerSRecData::HasLogProductionDistance)
        .def("IsT4", &evt::ShowerSRecData::IsT4)
        .def("IsT5", &evt::ShowerSRecData::IsT5)
        .def("SetEnergy", set_energy)
        ;


}