wib.cpp

Go to the documentation of this file.

 
#include "fddetdataformats/WIBFrame.hpp"
 
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
 
namespace py = pybind11;
 
namespace dunedaq::fddetdataformats::python {
 
void
register_wib(py::module& m)
{
 
  py::class_<WIBFrame>(m, "WIBFrame", py::buffer_protocol())
    .def(py::init())
    .def(py::init([](py::capsule capsule) {
      auto wfp = *static_cast<WIBFrame*>(capsule.get_pointer());
      return wfp;
    }))
    .def(py::init([](py::bytes bytes) {
      py::buffer_info info(py::buffer(bytes).request());
      auto wfp = *static_cast<WIBFrame*>(info.ptr);
      return wfp;
    }))
    .def("get_wib_header",
         static_cast<const WIBHeader* (WIBFrame::*)() const>(&WIBFrame::get_wib_header),
         py::return_value_policy::reference_internal)
    .def("get_coldata_header", &WIBFrame::get_coldata_header, py::return_value_policy::reference_internal)
    .def("get_block", &WIBFrame::get_block, py::return_value_policy::reference_internal)
    .def("get_channel",
         static_cast<uint16_t (WIBFrame::*)(const uint8_t, const uint8_t, const uint8_t) const>(&WIBFrame::get_channel))
    .def("get_channel", static_cast<uint16_t (WIBFrame::*)(const uint8_t, const uint8_t) const>(&WIBFrame::get_channel))
    .def("get_channel", static_cast<uint16_t (WIBFrame::*)(const uint8_t) const>(&WIBFrame::get_channel))
    .def("set_channel", static_cast<void (WIBFrame::*)(const uint8_t, const uint16_t)>(&WIBFrame::set_channel))
    .def("get_timestamp", &WIBFrame::get_timestamp)
    .def_static("sizeof", []() { return sizeof(WIBFrame); })
    .def("get_bytes",
         [](WIBFrame* fr) -> py::bytes { return py::bytes(reinterpret_cast<char*>(fr), sizeof(WIBFrame)); });
 
  py::class_<WIBHeader>(m, "WIBHeader")
    .def_property_readonly("sof", [](WIBHeader& self) -> uint32_t { return self.sof; })
    .def_property_readonly("version", [](WIBHeader& self) -> uint32_t { return self.version; })
    .def_property_readonly("fiber_no", [](WIBHeader& self) -> uint32_t { return self.fiber_no; })
    .def_property_readonly("crate_no", [](WIBHeader& self) -> uint32_t { return self.crate_no; })
    .def_property_readonly("slot_no", [](WIBHeader& self) -> uint32_t { return self.slot_no; })
    .def_property_readonly("mm", [](WIBHeader& self) -> uint32_t { return self.mm; })
    .def_property_readonly("oos", [](WIBHeader& self) -> uint32_t { return self.oos; })
    .def_property_readonly("mm", [](WIBHeader& self) -> uint32_t { return self.mm; })
    .def_property_readonly("wib_errors", [](WIBHeader& self) -> uint32_t { return self.wib_errors; })
    .def("get_timestamp", &WIBHeader::get_timestamp)
    .def("get_wib_counter", &WIBHeader::get_wib_counter);
  py::class_<ColdataHeader>(m, "ColdataHeader")
    .def_property_readonly("s1_error", [](ColdataHeader& self) -> uint32_t { return self.s1_error; })
    .def_property_readonly("s2_error", [](ColdataHeader& self) -> uint32_t { return self.s2_error; })
    .def_property_readonly("coldata_convert_count",
                           [](ColdataHeader& self) -> uint32_t { return self.coldata_convert_count; })
    .def_property_readonly("error_register", [](ColdataHeader& self) -> uint32_t { return self.error_register; })
    .def_property_readonly("hdr_1", [](ColdataHeader& self) -> uint32_t { return self.hdr_1; })
    .def_property_readonly("hdr_3", [](ColdataHeader& self) -> uint32_t { return self.hdr_3; })
    .def_property_readonly("hdr_2", [](ColdataHeader& self) -> uint32_t { return self.hdr_2; })
    .def_property_readonly("hdr_4", [](ColdataHeader& self) -> uint32_t { return self.hdr_4; })
    .def_property_readonly("hdr_5", [](ColdataHeader& self) -> uint32_t { return self.hdr_5; })
    .def_property_readonly("hdr_7", [](ColdataHeader& self) -> uint32_t { return self.hdr_7; })
    .def_property_readonly("hdr_6", [](ColdataHeader& self) -> uint32_t { return self.hdr_6; })
    .def_property_readonly("hdr_8", [](ColdataHeader& self) -> uint32_t { return self.hdr_8; })
    .def("get_checksum_a", &ColdataHeader::get_checksum_a)
    .def("get_checksum_b", &ColdataHeader::get_checksum_b)
    .def("get_hdr", &ColdataHeader::get_hdr);
 
  py::class_<ColdataSegment>(m, "ColdataSegment").def("get_channel", &ColdataSegment::get_channel);
  py::class_<ColdataBlock>(m, "ColdataBlock").def("get_channel", &ColdataBlock::get_channel);
}
 
} // namespace dunedaq::fddetdataformats::python