wib2.cpp

Go to the documentation of this file.

 
#include "fddetdataformats/WIB2Frame.hpp"
 
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
 
namespace py = pybind11;
 
namespace dunedaq::fddetdataformats::python {
 
void
register_wib2(py::module& m)
{
 
  py::class_<WIB2Frame>(m, "WIB2Frame", py::buffer_protocol())
    .def(py::init())
    .def(py::init([](py::capsule capsule) {
      auto wfp = *static_cast<WIB2Frame*>(capsule.get_pointer());
      return wfp;
    }))
    .def(py::init([](py::bytes bytes) {
      py::buffer_info info(py::buffer(bytes).request());
      auto wfp = *static_cast<WIB2Frame*>(info.ptr);
      return wfp;
    }))
    .def("get_adc", static_cast<uint16_t (WIB2Frame::*)(const int) const>(&WIB2Frame::get_adc))
    .def("set_adc", static_cast<void (WIB2Frame::*)(int, uint16_t)>(&WIB2Frame::set_adc))
    .def("get_timestamp", &WIB2Frame::get_timestamp)
    .def("get_header", [](WIB2Frame& self) -> const WIB2Frame::Header& { return self.header; })
    .def("get_trailer", [](WIB2Frame& self) -> const WIB2Frame::Trailer& { return self.trailer; })
    .def_static("sizeof", []() { return sizeof(WIB2Frame); })
    .def("get_bytes",
         [](WIB2Frame* fr) -> py::bytes { return py::bytes(reinterpret_cast<char*>(fr), sizeof(WIB2Frame)); });
 
  py::class_<WIB2Frame::Header>(m, "WIB2Header")
    .def_property_readonly("version", [](WIB2Frame::Header& self) -> uint32_t { return self.version; })
    .def_property_readonly("detector_id", [](WIB2Frame::Header& self) -> uint32_t { return self.detector_id; })
    .def_property_readonly("crate", [](WIB2Frame::Header& self) -> uint32_t { return self.crate; })
    .def_property_readonly("slot", [](WIB2Frame::Header& self) -> uint32_t { return self.slot; })
    .def_property_readonly("link", [](WIB2Frame::Header& self) -> uint32_t { return self.link; })
    .def_property_readonly("timestamp_1", [](WIB2Frame::Header& self) -> uint32_t { return self.timestamp_1; })
    .def_property_readonly("timestamp_2", [](WIB2Frame::Header& self) -> uint32_t { return self.timestamp_2; })
    .def_property_readonly("colddata_timestamp_id",
                           [](WIB2Frame::Header& self) -> uint32_t { return self.colddata_timestamp_id; })
    .def_property_readonly("femb_valid", [](WIB2Frame::Header& self) -> uint32_t { return self.femb_valid; })
    .def_property_readonly("link_mask", [](WIB2Frame::Header& self) -> uint32_t { return self.link_mask; })
    .def_property_readonly("lock_output_status",
                           [](WIB2Frame::Header& self) -> uint32_t { return self.lock_output_status; })
    .def_property_readonly("femb_pulser_frame_bits",
                           [](WIB2Frame::Header& self) -> uint32_t { return self.femb_pulser_frame_bits; })
    .def_property_readonly("femb_sync_flags", [](WIB2Frame::Header& self) -> uint32_t { return self.femb_sync_flags; })
    .def_property_readonly("colddata_timestamp",
                           [](WIB2Frame::Header& self) -> uint32_t { return self.colddata_timestamp; });
 
  py::class_<WIB2Frame::Trailer>(m, "WIB2Trailer")
    .def_property_readonly("flex_bits", [](WIB2Frame::Trailer& self) -> uint32_t { return self.flex_bits; })
    .def_property_readonly("ws", [](WIB2Frame::Trailer& self) -> uint32_t { return self.ws; })
    .def_property_readonly("psr_cal", [](WIB2Frame::Trailer& self) -> uint32_t { return self.psr_cal; })
    .def_property_readonly("ready", [](WIB2Frame::Trailer& self) -> uint32_t { return self.ready; })
    .def_property_readonly("context_code", [](WIB2Frame::Trailer& self) -> uint32_t { return self.context_code; });
}
 
} // namespace dunedaq::fddetdataformats::python