unpack.cpp

Go to the documentation of this file.

 
#include "daqdataformats/Fragment.hpp"
#include "fddetdataformats/DAPHNEEthFrame.hpp"
#include "fddetdataformats/DAPHNEEthStreamFrame.hpp"
#include "fddetdataformats/DAPHNEFrame.hpp"
#include "fddetdataformats/TDEEthFrame.hpp"
#include "fddetdataformats/WIBEthFrame.hpp"
 
#include <pybind11/numpy.h>
#include <pybind11/pybind11.h>
#include <pybind11/stl.h>
 
#include <fmt/core.h>
 
namespace py = pybind11;
 
namespace dunedaq {
namespace rawdatautils {
 
void
print_hex_fragment(daqdataformats::Fragment const& frag)
{
  uint64_t* data = static_cast<uint64_t*>(frag.get_data());
  size_t data_size = (frag.get_size() - sizeof(daqdataformats::FragmentHeader)) / 8;
 
  for (size_t i(0); i < data_size; ++i) {
    fmt::print("{:06d} 0x{:016x}\n", i, data[i]);
  }
}
 
namespace wibeth {
extern uint32_t
get_n_frames(daqdataformats::Fragment const& frag);
extern py::array_t<uint16_t>
np_array_adc(daqdataformats::Fragment const& frag);
extern py::array_t<uint16_t>
np_array_adc_data(void* data, uint32_t n_frames);
extern py::array_t<uint64_t>
np_array_timestamp(daqdataformats::Fragment const& frag);
extern py::array_t<uint64_t>
np_array_timestamp_data(void* data, uint32_t n_frames);
}
 
namespace daphne {
extern uint32_t
get_n_frames(daqdataformats::Fragment const& frag);
extern py::array_t<uint16_t>
np_array_adc(daqdataformats::Fragment& frag);
extern py::array_t<uint8_t>
np_array_channels(daqdataformats::Fragment& frag);
extern py::array_t<uint16_t>
np_array_adc_data(void* data, int nframes);
extern py::array_t<uint64_t>
np_array_timestamp(daqdataformats::Fragment& frag);
extern py::array_t<uint64_t>
np_array_timestamp_data(void* data, int nframes);
extern py::array_t<uint8_t>
np_array_channels_data(void* data, int nframes);
 
extern uint32_t
get_n_frames_stream(daqdataformats::Fragment const& frag);
extern py::array_t<uint16_t>
np_array_adc_stream(daqdataformats::Fragment& frag);
extern py::array_t<uint8_t>
np_array_channels_stream(daqdataformats::Fragment& frag);
extern py::array_t<uint16_t>
np_array_adc_stream_data(void* data, int nframes);
extern py::array_t<uint64_t>
np_array_timestamp_stream(daqdataformats::Fragment& frag);
extern py::array_t<uint64_t>
np_array_timestamp_stream_data(void* data, int nframes);
extern py::array_t<uint8_t>
np_array_channels_stream_data(void* data, int nframes);
 
}
 
namespace daphneeth {
extern uint32_t
get_n_frames(daqdataformats::Fragment const& frag);
extern py::array_t<uint16_t>
np_array_adc(daqdataformats::Fragment& frag);
extern py::array_t<uint8_t>
np_array_channels(daqdataformats::Fragment& frag);
extern py::array_t<uint16_t>
np_array_adc_data(void* data, int nframes);
extern py::array_t<uint64_t>
np_array_timestamp(daqdataformats::Fragment& frag);
extern py::array_t<uint64_t>
np_array_timestamp_data(void* data, int nframes);
extern py::array_t<uint8_t>
np_array_channels_data(void* data, int nframes);
 
extern uint32_t
get_n_frames_stream(daqdataformats::Fragment const& frag);
extern py::array_t<uint16_t>
np_array_adc_stream(daqdataformats::Fragment& frag);
extern py::array_t<uint8_t>
np_array_channels_stream(daqdataformats::Fragment& frag);
extern py::array_t<uint16_t>
np_array_adc_stream_data(void* data, int nframes);
extern py::array_t<uint64_t>
np_array_timestamp_stream(daqdataformats::Fragment& frag);
extern py::array_t<uint64_t>
np_array_timestamp_stream_data(void* data, int nframes);
extern py::array_t<uint8_t>
np_array_channels_stream_data(void* data, int nframes);
 
}
 
namespace tde {
extern uint32_t
get_n_frames(daqdataformats::Fragment const& frag);
extern py::array_t<uint16_t>
np_array_adc(daqdataformats::Fragment const& frag);
extern py::array_t<uint16_t>
np_array_adc_data(void* data, uint32_t n_frames);
extern py::array_t<long double>
np_array_timestamp(daqdataformats::Fragment const& frag);
extern py::array_t<long double>
np_array_timestamp_data(void* data, uint32_t n_frames);
 
}
 
namespace unpack {
namespace python {
 
void
register_unpack(py::module& m)
{
 
  m.def("print_hex_fragment", &print_hex_fragment);
 
  py::module_ wibeth_module = m.def_submodule("wibeth");
  wibeth_module.def("get_n_frames", &wibeth::get_n_frames);
  wibeth_module.def("np_array_adc", &wibeth::np_array_adc);
  wibeth_module.def("np_array_timestamp", &wibeth::np_array_timestamp);
  wibeth_module.def("np_array_adc_data", &wibeth::np_array_adc_data);
  wibeth_module.def("np_array_timestamp_data", &wibeth::np_array_timestamp_data);
 
  py::module_ daphne_module = m.def_submodule("daphne");
  daphne_module.def("get_n_frames", &daphne::get_n_frames);
  daphne_module.def("np_array_adc", &daphne::np_array_adc);
  daphne_module.def("np_array_timestamp", &daphne::np_array_timestamp);
  daphne_module.def("np_array_adc_data", &daphne::np_array_adc_data);
  daphne_module.def("np_array_timestamp_data", &daphne::np_array_timestamp_data);
  daphne_module.def("np_array_channels_data", &daphne::np_array_channels_data);
  daphne_module.def("np_array_channels", &daphne::np_array_channels);
 
  daphne_module.def("get_n_frames_stream", &daphne::get_n_frames_stream);
  daphne_module.def("np_array_adc_stream", &daphne::np_array_adc_stream);
  daphne_module.def("np_array_timestamp_stream", &daphne::np_array_timestamp_stream);
  daphne_module.def("np_array_adc_stream_data", &daphne::np_array_adc_stream_data);
  daphne_module.def("np_array_timestamp_stream_data", &daphne::np_array_timestamp_stream_data);
  daphne_module.def("np_array_channels_stream_data", &daphne::np_array_channels_stream_data);
  daphne_module.def("np_array_channels_stream", &daphne::np_array_channels_stream);
 
  py::module_ daphneeth_module = m.def_submodule("daphneeth");
  daphneeth_module.def("get_n_frames", &daphneeth::get_n_frames);
  daphneeth_module.def("np_array_adc", &daphneeth::np_array_adc);
  daphneeth_module.def("np_array_timestamp", &daphneeth::np_array_timestamp);
  daphneeth_module.def("np_array_adc_data", &daphneeth::np_array_adc_data);
  daphneeth_module.def("np_array_timestamp_data", &daphneeth::np_array_timestamp_data);
  daphneeth_module.def("np_array_channels_data", &daphneeth::np_array_channels_data);
  daphneeth_module.def("np_array_channels", &daphneeth::np_array_channels);
 
  daphneeth_module.def("get_n_frames_stream", &daphneeth::get_n_frames_stream);
  daphneeth_module.def("np_array_adc_stream", &daphneeth::np_array_adc_stream);
  daphneeth_module.def("np_array_timestamp_stream", &daphneeth::np_array_timestamp_stream);
  daphneeth_module.def("np_array_adc_stream_data", &daphneeth::np_array_adc_stream_data);
  daphneeth_module.def("np_array_timestamp_stream_data", &daphneeth::np_array_timestamp_stream_data);
  daphneeth_module.def("np_array_channels_stream_data", &daphneeth::np_array_channels_stream_data);
  daphneeth_module.def("np_array_channels_stream", &daphneeth::np_array_channels_stream);
 
  py::module_ tde_module = m.def_submodule("tde");
  tde_module.def("get_n_frames", &tde::get_n_frames);
  tde_module.def("np_array_adc", &tde::np_array_adc);
  tde_module.def("np_array_timestamp", &tde::np_array_timestamp);
  tde_module.def("np_array_adc_data", &tde::np_array_adc_data);
  tde_module.def("np_array_timestamp_data", &tde::np_array_timestamp_data);
}
 
} // namespace python
} // namespace rawdatautils
} // namespace dunedaq
}