Line data Source code
1 : /**
2 : * @file trigger_primitive.cpp Python bindings for the trigger primitive formats
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 "trgdataformats/TriggerObjectOverlay.hpp"
10 :
11 : #include <pybind11/pybind11.h>
12 : #include <pybind11/stl.h>
13 : #include <cstring>
14 :
15 : namespace py = pybind11;
16 :
17 : namespace dunedaq {
18 : namespace trgdataformats {
19 : namespace python {
20 :
21 : /*
22 : Doesn't work
23 : */
24 : struct TriggerCandidateHolder {
25 :
26 :
27 0 : TriggerCandidateHolder(void* ptr, size_t size) {
28 0 : m_size = size;
29 0 : m_data.reset(new uint8_t[m_size]);
30 0 : std::memcpy(m_data.get(), ptr, size);
31 0 : }
32 :
33 0 : TriggerCandidate* ptr() { return reinterpret_cast<TriggerCandidate*>(m_data.get()); }
34 :
35 : size_t m_size;
36 : std::unique_ptr<uint8_t[]> m_data;
37 :
38 : };
39 :
40 : void
41 0 : register_trigger_candidate(py::module& m)
42 : {
43 :
44 0 : m.def("string_to_trigger_candidate_type", &trgdataformats::string_to_trigger_candidate_type);
45 0 : m.def("trigger_candidate_type_to_string", &trgdataformats::trigger_candidate_type_to_string);
46 :
47 0 : py::class_<TriggerCandidateData> trigger_candidate_data(m, "TriggerCandidateData", py::buffer_protocol());
48 0 : trigger_candidate_data
49 0 : .def(py::init())
50 0 : .def(py::init([](py::capsule capsule) {
51 0 : auto tp = *static_cast<TriggerCandidateData*>(capsule.get_pointer());
52 0 : return tp;
53 : } ))
54 0 : .def(py::init([](py::bytes bytes){
55 0 : py::buffer_info info(py::buffer(bytes).request());
56 0 : auto tp = *static_cast<TriggerCandidateData*>(info.ptr);
57 0 : return tp;
58 0 : }))
59 0 : .def_property_readonly("version", [](TriggerCandidateData& self) -> uint16_t {return self.version;})
60 0 : .def_property_readonly("time_start", [](TriggerCandidateData& self) -> uint64_t {return self.time_start;})
61 0 : .def_property_readonly("time_end", [](TriggerCandidateData& self) -> uint64_t {return self.time_end;})
62 0 : .def_property_readonly("time_candidate", [](TriggerCandidateData& self) -> uint64_t {return self.time_candidate;})
63 0 : .def_property_readonly("detid", [](TriggerCandidateData& self) -> uint16_t {return self.detid;})
64 0 : .def_property_readonly("type", [](TriggerCandidateData& self) -> TriggerCandidateData::Type {return self.type;})
65 0 : .def_property_readonly("algorithm", [](TriggerCandidateData& self) -> TriggerCandidateData::Algorithm {return self.algorithm;})
66 0 : .def_static("sizeof", [](){ return sizeof(TriggerCandidateData); })
67 : ;
68 :
69 0 : py::enum_<TriggerCandidateData::Type>(trigger_candidate_data, "Type")
70 0 : .value("kUnknown", TriggerCandidateData::Type::kUnknown)
71 0 : .value("kTiming", TriggerCandidateData::Type::kTiming)
72 0 : .value("kTPCLowE", TriggerCandidateData::Type::kTPCLowE)
73 0 : .value("kSupernova", TriggerCandidateData::Type::kSupernova)
74 0 : .value("kRandom", TriggerCandidateData::Type::kRandom)
75 0 : .value("kPrescale", TriggerCandidateData::Type::kPrescale)
76 0 : .value("kADCSimpleWindow", TriggerCandidateData::Type::kADCSimpleWindow)
77 0 : .value("kHorizontalMuon", TriggerCandidateData::Type::kHorizontalMuon)
78 0 : .value("kMichelElectron", TriggerCandidateData::Type::kMichelElectron)
79 0 : .value("kPlaneCoincidence", TriggerCandidateData::Type::kPlaneCoincidence)
80 0 : .value("kBundle", TriggerCandidateData::Type::kBundle)
81 0 : .value("kChannelDistance", TriggerCandidateData::Type::kChannelDistance)
82 0 : .value("kDBSCAN", TriggerCandidateData::Type::kDBSCAN)
83 0 : .value("kCTBFakeTrigger", TriggerCandidateData::Type::kCTBFakeTrigger)
84 0 : .value("kCTBBeam", TriggerCandidateData::Type::kCTBBeam)
85 0 : .value("kCTBBeamChkvHL", TriggerCandidateData::Type::kCTBBeamChkvHL)
86 0 : .value("kCTBCustomD", TriggerCandidateData::Type::kCTBCustomD)
87 0 : .value("kCTBCustomE", TriggerCandidateData::Type::kCTBCustomE)
88 0 : .value("kCTBCustomF", TriggerCandidateData::Type::kCTBCustomF)
89 0 : .value("kCTBCustomG", TriggerCandidateData::Type::kCTBCustomG)
90 0 : .value("kCTBBeamChkvHLx", TriggerCandidateData::Type::kCTBBeamChkvHLx)
91 0 : .value("kCTBBeamChkvHxL", TriggerCandidateData::Type::kCTBBeamChkvHxL)
92 0 : .value("kCTBBeamChkvHxLx", TriggerCandidateData::Type::kCTBBeamChkvHxLx)
93 0 : .value("kNeutronSourceCalib", TriggerCandidateData::Type::kNeutronSourceCalib)
94 0 : .value("kChannelAdjacency", TriggerCandidateData::Type::kChannelAdjacency)
95 0 : .value("kCIBFakeTrigger", TriggerCandidateData::Type::kCIBFakeTrigger)
96 0 : .value("kCIBLaserTriggerP1", TriggerCandidateData::Type::kCIBLaserTriggerP1)
97 0 : .value("kCIBLaserTriggerP2", TriggerCandidateData::Type::kCIBLaserTriggerP2)
98 0 : .value("kCIBLaserTriggerP3", TriggerCandidateData::Type::kCIBLaserTriggerP3)
99 0 : .value("kCTBOffSpillSnapshot", TriggerCandidateData::Type::kCTBOffSpillSnapshot)
100 0 : .value("kCTBOffSpillCosmicJura", TriggerCandidateData::Type::kCTBOffSpillCosmicJura)
101 0 : .value("kCTBOffSpillCRTCosmic", TriggerCandidateData::Type::kCTBOffSpillCRTCosmic)
102 0 : .value("kCTBBeamSpillStart", TriggerCandidateData::Type::kCTBBeamSpillStart)
103 0 : .value("kCTBBeamSpillSnapshot", TriggerCandidateData::Type::kCTBBeamSpillSnapshot)
104 0 : .value("kCTBCustomC", TriggerCandidateData::Type::kCTBCustomC)
105 0 : .value("kCTBCustomPulseTrain", TriggerCandidateData::Type::kCTBCustomPulseTrain)
106 0 : .value("kDTSPulser", TriggerCandidateData::Type::kDTSPulser)
107 0 : .value("kDTSCosmic", TriggerCandidateData::Type::kDTSCosmic)
108 0 : .value("kSSPLEDCalibration", TriggerCandidateData::Type::kSSPLEDCalibration);
109 :
110 0 : py::enum_<TriggerCandidateData::Algorithm>(trigger_candidate_data, "Algorithm")
111 0 : .value("kUnknown", TriggerCandidateData::Algorithm::kUnknown)
112 0 : .value("kSupernova", TriggerCandidateData::Algorithm::kSupernova)
113 0 : .value("kHSIEventToTriggerCandidate", TriggerCandidateData::Algorithm::kHSIEventToTriggerCandidate)
114 0 : .value("kPrescale", TriggerCandidateData::Algorithm::kPrescale)
115 0 : .value("kADCSimpleWindow", TriggerCandidateData::Algorithm::kADCSimpleWindow)
116 0 : .value("kHorizontalMuon", TriggerCandidateData::Algorithm::kHorizontalMuon)
117 0 : .value("kMichelElectron", TriggerCandidateData::Algorithm::kMichelElectron)
118 0 : .value("kPlaneCoincidence", TriggerCandidateData::Algorithm::kPlaneCoincidence)
119 0 : .value("kCustom", TriggerCandidateData::Algorithm::kCustom)
120 0 : .value("kDBSCAN", TriggerCandidateData::Algorithm::kDBSCAN)
121 0 : .value("kChannelDistance", TriggerCandidateData::Algorithm::kChannelDistance)
122 0 : .value("kBundle", TriggerCandidateData::Algorithm::kBundle)
123 0 : .value("kChannelAdjacency", TriggerCandidateData::Algorithm::kChannelAdjacency);
124 :
125 0 : py::class_<TriggerCandidate>(m, "TriggerCandidateOverlay", py::buffer_protocol())
126 0 : .def(py::init([](py::capsule capsule) {
127 0 : auto tp = *static_cast<TriggerCandidate*>(capsule.get_pointer());
128 0 : return tp;
129 : } ))
130 0 : .def_property_readonly("data", [](TriggerCandidate& self) -> TriggerCandidateData& {return self.data;})
131 0 : .def("__len__", [](TriggerCandidate& self){ return self.n_inputs; })
132 0 : .def("sizeof", [](TriggerCandidate& self){ return sizeof(TriggerCandidate)+self.n_inputs*sizeof(TriggerActivityData); })
133 : ;
134 :
135 :
136 0 : py::class_<TriggerCandidateHolder>(m, "TriggerCandidate", py::buffer_protocol())
137 0 : .def(py::init([](py::capsule capsule) {
138 0 : auto tc_ptr = static_cast<TriggerCandidate*>(capsule.get_pointer());
139 0 : TriggerCandidateHolder tch(tc_ptr, sizeof(TriggerCandidate)+tc_ptr->n_inputs*sizeof(TriggerActivityData));
140 0 : return tch;
141 : }))
142 0 : .def(py::init([](py::bytes bytes){
143 0 : py::buffer_info info(py::buffer(bytes).request());
144 :
145 0 : TriggerCandidateHolder tch(info.ptr, info.size);
146 :
147 0 : return tch;
148 0 : }))
149 :
150 0 : .def("get_bytes",
151 0 : [](TriggerCandidateHolder& tch) -> py::bytes {
152 0 : return py::bytes(reinterpret_cast<char*>(tch.ptr()), tch.m_size);
153 0 : }, py::return_value_policy::reference_internal
154 : )
155 0 : .def_property_readonly("data", [](TriggerCandidateHolder& self) -> TriggerCandidateData& {return self.ptr()->data;})
156 0 : .def("n_inputs", [](TriggerCandidateHolder& self){ return self.ptr()->n_inputs; })
157 0 : .def("__len__", [](TriggerCandidateHolder& self){ return self.ptr()->n_inputs; })
158 0 : .def("__getitem__",
159 0 : [](TriggerCandidateHolder &self, size_t i) -> const TriggerActivityData& {
160 0 : if (i >= self.ptr()->n_inputs) {
161 0 : throw py::index_error();
162 : }
163 0 : return self.ptr()->inputs[i];
164 0 : }, py::return_value_policy::reference_internal)
165 0 : .def("sizeof", [](TriggerCandidateHolder& self){ return self.m_size; })
166 : ;
167 0 : }
168 :
169 : } // namespace python
170 : } // namespace trgdataformats
171 : } // namespace dunedaq
|
