detid.cpp

Go to the documentation of this file.

 
#include "detdataformats/DetID.hpp"
 
#include <pybind11/operators.h>
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
 
#include <sstream>
 
namespace py = pybind11;
 
namespace dunedaq::detdataformats::python {
 
void
register_detid(py::module& m)
{
  py::class_<DetID> py_detid(m, "DetID");
  py_detid.def(py::init())
    .def(py::init<const DetID::Subdetector&>())
    .def("__repr__",
         [](const DetID& gid) {
           std::ostringstream oss;
           oss << "<detdataformats::DetID " << DetID::subdetector_to_string(gid.subdetector) << ">";
           return oss.str();
         })
    .def("subdetector_to_string", &DetID::subdetector_to_string)
    .def("string_to_subdetector", &DetID::string_to_subdetector);
 
  py::enum_<DetID::Subdetector>(py_detid, "Subdetector")
    .value("kUnknown", DetID::Subdetector::kUnknown)
    .value("kDAQ", DetID::Subdetector::kDAQ)
    .value("kHD_PDS", DetID::Subdetector::kHD_PDS)
    .value("kHD_TPC", DetID::Subdetector::kHD_TPC)
    .value("kHD_CRT", DetID::Subdetector::kHD_CRT)
    .value("kVD_CathodePDS", DetID::Subdetector::kVD_CathodePDS)
    .value("kVD_MembranePDS", DetID::Subdetector::kVD_MembranePDS)
    .value("kVD_BottomTPC", DetID::Subdetector::kVD_BottomTPC)
    .value("kVD_TopTPC", DetID::Subdetector::kVD_TopTPC)
    .value("kVD_BernCRT", DetID::Subdetector::kVD_BernCRT)
    .value("kVD_GrenobleCRT", DetID::Subdetector::kVD_GrenobleCRT)
    .value("kNDLAr_TPC", DetID::Subdetector::kNDLAr_TPC)
    .value("kNDLAr_PDS", DetID::Subdetector::kNDLAr_PDS)
    .value("kND_GAr", DetID::Subdetector::kND_GAr)
    .export_values();
 
  py_detid.def_readwrite("subdetector", &DetID::subdetector);
}
 
} // namespace dunedaq::detdataformats::python