Line data Source code
1 : /**
2 : * @file trigger_record.cpp
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 "daqdataformats/TriggerRecord.hpp"
10 : #include "daqdataformats/TriggerRecordHeader.hpp"
11 :
12 : #include <pybind11/pybind11.h>
13 : #include <pybind11/stl.h>
14 :
15 : #include <memory>
16 : #include <vector>
17 :
18 : namespace py = pybind11;
19 :
20 : namespace dunedaq::daqdataformats::python {
21 :
22 : void
23 0 : register_trigger_record(py::module& m)
24 : {
25 0 : py::class_<TriggerRecordHeader>(m, "TriggerRecordHeader", pybind11::buffer_protocol())
26 0 : .def(py::init<std::vector<ComponentRequest> const&>())
27 0 : .def(py::init([](py::capsule capsule, bool copy_from_buffer) {
28 0 : return std::make_unique<TriggerRecordHeader>(capsule.get_pointer(), copy_from_buffer);
29 : }))
30 0 : .def(py::init<TriggerRecordHeader const&>())
31 0 : .def("get_header", &TriggerRecordHeader::get_header)
32 0 : .def("get_trigger_number", &TriggerRecordHeader::get_trigger_number)
33 : //.def("set_trigger_number", &TriggerRecordHeader::set_trigger_number)
34 0 : .def("get_trigger_timestamp", &TriggerRecordHeader::get_trigger_timestamp)
35 : //.def("set_trigger_timestamp", &TriggerRecordHeader::set_trigger_timestamp)
36 0 : .def("get_num_requested_components", &TriggerRecordHeader::get_num_requested_components)
37 0 : .def("get_run_number", &TriggerRecordHeader::get_run_number)
38 : //.def("set_run_number", &TriggerRecordHeader::set_run_number)
39 0 : .def("get_status_bits", &TriggerRecordHeader::get_status_bits)
40 : //.def("set_status_bits", &TriggerRecordHeader::set_status_bits)
41 0 : .def("get_status_bit", &TriggerRecordHeader::get_status_bit)
42 : //.def("set_status_bit", &TriggerRecordHeader::set_status_bit)
43 0 : .def("get_trigger_type", &TriggerRecordHeader::get_trigger_type)
44 : //.def("set_trigger_type", &TriggerRecordHeader::set_trigger_type)
45 0 : .def("get_sequence_number", &TriggerRecordHeader::get_sequence_number)
46 : //.def("set_sequence_number", &TriggerRecordHeader::set_sequence_number)
47 0 : .def("get_max_sequence_number", &TriggerRecordHeader::get_max_sequence_number)
48 : //.def("set_max_sequence_number", &TriggerRecordHeader::set_max_sequence_number)
49 0 : .def("get_total_size_bytes", &TriggerRecordHeader::get_total_size_bytes)
50 0 : .def(
51 0 : "get_storage_location", &TriggerRecordHeader::get_storage_location, py::return_value_policy::reference_internal)
52 0 : .def("at", &TriggerRecordHeader::at)
53 0 : .def("__getitem__", &TriggerRecordHeader::operator[], py::return_value_policy::reference_internal);
54 :
55 0 : py::class_<TriggerRecordHeaderData>(m, "TriggerRecordHeaderData")
56 0 : .def_property_readonly_static("s_trigger_record_header_magic",
57 0 : [](const TriggerRecordHeaderData& self) -> uint32_t { // NOLINT(build/unsigned)
58 : return self.s_trigger_record_header_magic;
59 : })
60 0 : .def_property_readonly_static("s_trigger_record_header_version",
61 0 : [](const TriggerRecordHeaderData& self) -> uint32_t { // NOLINT(build/unsigned)
62 : return self.s_trigger_record_header_version;
63 : })
64 0 : .def_property_readonly_static("s_invalid_number_components",
65 0 : [](const TriggerRecordHeaderData& self) -> uint64_t { // NOLINT(build/unsigned)
66 : return self.s_invalid_number_components;
67 : })
68 0 : .def_property_readonly_static("s_default_status_bits",
69 0 : [](const TriggerRecordHeaderData& self) -> uint32_t { // NOLINT(build/unsigned)
70 : return self.s_default_status_bits;
71 : })
72 0 : .def_property_readonly("trigger_record_header_marker",
73 0 : [](const TriggerRecordHeaderData& self) -> uint32_t { // NOLINT(build/unsigned)
74 0 : return self.trigger_record_header_marker;
75 : })
76 0 : .def_property_readonly(
77 0 : "version", [](const TriggerRecordHeaderData& self) -> uint32_t { return self.version; }) // NOLINT(build/unsigned)
78 0 : .def_property_readonly("trigger_number",
79 0 : [](const TriggerRecordHeaderData& self) -> trigger_number_t { return self.trigger_number; })
80 0 : .def_property_readonly("trigger_timestamp",
81 0 : [](const TriggerRecordHeaderData& self) -> timestamp_t { return self.trigger_timestamp; })
82 0 : .def_property_readonly("num_requested_components",
83 0 : [](const TriggerRecordHeaderData& self) -> uint64_t { // NOLINT(build/unsigned)
84 0 : return self.num_requested_components;
85 : })
86 0 : .def_property_readonly("run_number",
87 0 : [](const TriggerRecordHeaderData& self) -> run_number_t { return self.run_number; })
88 0 : .def_property_readonly(
89 : "status_bits",
90 0 : [](const TriggerRecordHeaderData& self) -> uint32_t { return self.status_bits; }) // NOLINT(build/unsigned)
91 0 : .def_property_readonly("trigger_type",
92 0 : [](const TriggerRecordHeaderData& self) -> trigger_type_t { return self.trigger_type; })
93 0 : .def_property_readonly(
94 0 : "sequence_number", [](const TriggerRecordHeaderData& self) -> sequence_number_t { return self.sequence_number; })
95 0 : .def_property_readonly("max_sequence_number", [](const TriggerRecordHeaderData& self) -> sequence_number_t {
96 0 : return self.max_sequence_number;
97 : });
98 :
99 0 : py::enum_<TriggerRecordStatusBits>(m, "TriggerRecordStatusBits")
100 0 : .value("kIncomplete", TriggerRecordStatusBits::kIncomplete)
101 0 : .value("kMismatch", TriggerRecordStatusBits::kMismatch)
102 0 : .value("kUnassigned2", TriggerRecordStatusBits::kUnassigned2)
103 0 : .value("kUnassigned3", TriggerRecordStatusBits::kUnassigned3)
104 0 : .value("kUnassigned4", TriggerRecordStatusBits::kUnassigned4)
105 0 : .value("kUnassigned5", TriggerRecordStatusBits::kUnassigned5)
106 0 : .value("kUnassigned6", TriggerRecordStatusBits::kUnassigned6)
107 0 : .value("kUnassigned7", TriggerRecordStatusBits::kUnassigned7)
108 0 : .value("kUnassigned8", TriggerRecordStatusBits::kUnassigned8)
109 0 : .value("kUnassigned9", TriggerRecordStatusBits::kUnassigned9)
110 0 : .value("kUnassigned10", TriggerRecordStatusBits::kUnassigned10)
111 0 : .value("kUnassigned11", TriggerRecordStatusBits::kUnassigned11)
112 0 : .value("kUnassigned12", TriggerRecordStatusBits::kUnassigned12)
113 0 : .value("kUnassigned13", TriggerRecordStatusBits::kUnassigned13)
114 0 : .value("kUnassigned14", TriggerRecordStatusBits::kUnassigned14)
115 0 : .value("kUnassigned15", TriggerRecordStatusBits::kUnassigned15)
116 0 : .value("kUnassigned16", TriggerRecordStatusBits::kUnassigned16)
117 0 : .value("kUnassigned17", TriggerRecordStatusBits::kUnassigned17)
118 0 : .value("kUnassigned18", TriggerRecordStatusBits::kUnassigned18)
119 0 : .value("kUnassigned19", TriggerRecordStatusBits::kUnassigned19)
120 0 : .value("kUnassigned20", TriggerRecordStatusBits::kUnassigned20)
121 0 : .value("kUnassigned21", TriggerRecordStatusBits::kUnassigned21)
122 0 : .value("kUnassigned22", TriggerRecordStatusBits::kUnassigned22)
123 0 : .value("kUnassigned23", TriggerRecordStatusBits::kUnassigned23)
124 0 : .value("kUnassigned24", TriggerRecordStatusBits::kUnassigned24)
125 0 : .value("kUnassigned25", TriggerRecordStatusBits::kUnassigned25)
126 0 : .value("kUnassigned26", TriggerRecordStatusBits::kUnassigned26)
127 0 : .value("kUnassigned27", TriggerRecordStatusBits::kUnassigned27)
128 0 : .value("kUnassigned28", TriggerRecordStatusBits::kUnassigned28)
129 0 : .value("kUnassigned29", TriggerRecordStatusBits::kUnassigned29)
130 0 : .value("kUnassigned30", TriggerRecordStatusBits::kUnassigned30)
131 0 : .value("kUnassigned31", TriggerRecordStatusBits::kUnassigned31)
132 0 : .value("kInvalid", TriggerRecordStatusBits::kInvalid);
133 :
134 0 : py::class_<TriggerRecord> py_trigger_record(m, "TriggerRecord", pybind11::buffer_protocol());
135 0 : py_trigger_record.def(py::init<TriggerRecordHeader const&>())
136 0 : .def(py::init<std::vector<ComponentRequest> const&>())
137 0 : .def(
138 : "get_header_ref",
139 0 : [](TriggerRecord& self) { return self.get_header_ref(); },
140 0 : py::return_value_policy::reference_internal)
141 : // .def("set_header", &TriggerRecord::set_header)
142 0 : .def("get_header_data", &TriggerRecord::get_header_data)
143 0 : .def(
144 : "get_fragments_ref",
145 0 : [](TriggerRecord& self) {
146 0 : auto fragments = py::list();
147 0 : for (auto& fragment : self.get_fragments_ref()) {
148 0 : auto py_fragment = py::cast(*fragment, py::return_value_policy::reference);
149 0 : fragments.append(py_fragment);
150 0 : }
151 0 : return fragments;
152 0 : },
153 0 : py::return_value_policy::reference_internal)
154 0 : .def("get_total_size_bytes", &TriggerRecord::get_total_size_bytes)
155 0 : .def("get_sum_of_fragment_payload_sizes", &TriggerRecord::get_sum_of_fragment_payload_sizes);
156 0 : } // NOLINT
157 :
158 : } // namespace dunedaq::daqdataformats::python
|
