Line data Source code
1 : /**
2 : * @file daphneethstream.cpp Python bindings for the DAPHNEEthStreamFrame format
3 : *
4 : * This is part of the DUNE DAQ Software Suite, copyright 2020.
5 : * Licensing/copyright details are in the COPYING file that you should have
6 : * received with this code.
7 : */
8 :
9 : #include "fddetdataformats/DAPHNEEthStreamFrame.hpp"
10 :
11 : #include <pybind11/pybind11.h>
12 : #include <pybind11/stl.h>
13 :
14 : namespace py = pybind11;
15 :
16 : namespace dunedaq::fddetdataformats::python {
17 :
18 : // NOLINTBEGIN(build/unsigned)
19 :
20 : void
21 0 : register_daphneethstream(py::module& m)
22 : {
23 :
24 0 : py::class_<DAPHNEEthStreamFrame::ChannelWord>(m, "DAPHNEEthStreamChannelWord")
25 0 : .def_property(
26 : "tbd",
27 0 : [](DAPHNEEthStreamFrame::ChannelWord& self) -> uint64_t { return self.tbd; },
28 0 : [](DAPHNEEthStreamFrame::ChannelWord& self, uint64_t tbd) { self.tbd = tbd; })
29 0 : .def_property(
30 : "version",
31 0 : [](DAPHNEEthStreamFrame::ChannelWord& self) -> uint64_t { return self.version; },
32 0 : [](DAPHNEEthStreamFrame::ChannelWord& self, uint64_t version) { self.version = version; })
33 0 : .def_property(
34 : "channel",
35 0 : [](DAPHNEEthStreamFrame::ChannelWord& self) -> uint64_t { return self.channel; },
36 0 : [](DAPHNEEthStreamFrame::ChannelWord& self, uint64_t channel) { self.channel = channel; })
37 0 : .def_property_readonly_static("s_expected_bytes",
38 0 : [](py::object /*self*/) { return sizeof(DAPHNEEthStreamFrame::ChannelWord); });
39 :
40 0 : py::class_<DAPHNEEthStreamFrame::Header>(m, "DAPHNEEthStreamHeader")
41 0 : .def_property(
42 : "channel_words",
43 0 : [](DAPHNEEthStreamFrame::Header& self) -> py::list {
44 0 : py::list result;
45 0 : for (int i = 0; i < 4; i++) {
46 0 : result.append(self.channel_words[i]);
47 : }
48 0 : return result;
49 0 : },
50 0 : [](DAPHNEEthStreamFrame::Header& self, py::list channel_words) {
51 0 : for (int i = 0; i < 4 && i < static_cast<int>(len(channel_words)); i++) {
52 0 : self.channel_words[i] = channel_words[i].cast<DAPHNEEthStreamFrame::ChannelWord>();
53 : }
54 0 : })
55 0 : .def_property_readonly_static("s_expected_bytes",
56 0 : [](py::object /*self*/) { return sizeof(DAPHNEEthStreamFrame::Header); });
57 :
58 0 : py::class_<DAPHNEEthStreamFrame>(m, "DAPHNEEthStreamFrame", py::buffer_protocol())
59 0 : .def(py::init())
60 0 : .def(py::init([](py::capsule capsule) {
61 0 : auto wfp = *static_cast<DAPHNEEthStreamFrame*>(capsule.get_pointer());
62 0 : return wfp;
63 : }))
64 0 : .def(py::init([](py::bytes bytes) {
65 0 : py::buffer_info info(py::buffer(bytes).request());
66 0 : auto wfp = *static_cast<DAPHNEEthStreamFrame*>(info.ptr);
67 0 : return wfp;
68 0 : }))
69 0 : .def(
70 : "get_daqheader",
71 0 : [](DAPHNEEthStreamFrame& self) -> const detdataformats::DAQEthHeader& { return self.daq_header; },
72 0 : py::return_value_policy::reference_internal)
73 0 : .def(
74 : "get_daphneheader",
75 0 : [](DAPHNEEthStreamFrame& self) -> const DAPHNEEthStreamFrame::Header& { return self.header; },
76 0 : py::return_value_policy::reference_internal)
77 0 : .def(
78 : "get_header",
79 0 : [](DAPHNEEthStreamFrame& self) -> const DAPHNEEthStreamFrame::Header& { return self.header; },
80 0 : py::return_value_policy::reference_internal)
81 0 : .def_property_readonly(
82 : "daq_header",
83 0 : [](DAPHNEEthStreamFrame& self) -> detdataformats::DAQEthHeader& { return self.daq_header; },
84 0 : py::return_value_policy::reference_internal)
85 0 : .def_property_readonly(
86 : "header",
87 0 : [](DAPHNEEthStreamFrame& self) -> DAPHNEEthStreamFrame::Header& { return self.header; },
88 0 : py::return_value_policy::reference_internal)
89 0 : .def("get_adc", &DAPHNEEthStreamFrame::get_adc)
90 0 : .def("set_adc", &DAPHNEEthStreamFrame::set_adc)
91 0 : .def("get_timestamp", &DAPHNEEthStreamFrame::get_timestamp)
92 0 : .def("set_timestamp", &DAPHNEEthStreamFrame::set_timestamp)
93 0 : .def("__lt__", [](const DAPHNEEthStreamFrame& lhs, const DAPHNEEthStreamFrame& rhs) { return lhs < rhs; })
94 0 : .def("get_channel", &DAPHNEEthStreamFrame::get_channel)
95 0 : .def("set_channel", &DAPHNEEthStreamFrame::set_channel)
96 0 : .def_property_readonly_static("version", [](py::object /*self*/) { return DAPHNEEthStreamFrame::version; })
97 0 : .def_property_readonly_static("s_bits_per_adc",
98 0 : [](py::object /*self*/) { return DAPHNEEthStreamFrame::s_bits_per_adc; })
99 0 : .def_property_readonly_static("s_bits_per_word",
100 0 : [](py::object /*self*/) { return DAPHNEEthStreamFrame::s_bits_per_word; })
101 0 : .def_property_readonly_static("s_adcs_per_channel",
102 0 : [](py::object /*self*/) { return DAPHNEEthStreamFrame::s_adcs_per_channel; })
103 0 : .def_property_readonly_static("s_num_channels",
104 0 : [](py::object /*self*/) { return DAPHNEEthStreamFrame::s_num_channels; })
105 0 : .def_property_readonly_static("s_num_adc_words",
106 0 : [](py::object /*self*/) { return DAPHNEEthStreamFrame::s_num_adc_words; })
107 0 : .def_property_readonly_static("s_expected_bytes",
108 0 : [](py::object /*self*/) { return DAPHNEEthStreamFrame::s_expected_bytes; })
109 0 : .def("get_channel0", &DAPHNEEthStreamFrame::get_channel0)
110 0 : .def("get_channel1", &DAPHNEEthStreamFrame::get_channel1)
111 0 : .def("get_channel2", &DAPHNEEthStreamFrame::get_channel2)
112 0 : .def("get_channel3", &DAPHNEEthStreamFrame::get_channel3)
113 0 : .def_static("sizeof", []() { return sizeof(DAPHNEEthStreamFrame); })
114 0 : .def("get_bytes", [](DAPHNEEthStreamFrame* fr) -> py::bytes {
115 0 : return py::bytes(reinterpret_cast<char*>(fr), sizeof(DAPHNEEthStreamFrame)); // NOLINT reinterpret_cast
116 : });
117 0 : } // NOLINT (defensible use of "overly long" function)
118 :
119 : // NOLINTEND(build/unsigned)
120 :
121 : } // namespace dunedaq::fddetdataformats::python
|
