TDEUnpacker.cpp

Go to the documentation of this file.

 
#include "daqdataformats/Fragment.hpp"
#include "fddetdataformats/TDEEthFrame.hpp"
 
#include <cstdint>
#include <pybind11/numpy.h>
 
namespace py = pybind11;
namespace dunedaq::rawdatautils::tde {

uint32_t
get_n_frames(daqdataformats::Fragment const& frag)
{
  return (frag.get_size() - sizeof(daqdataformats::FragmentHeader)) / sizeof(fddetdataformats::TDEEthFrame);
}

py::array_t<uint16_t>
np_array_adc_data(void* data, uint32_t n_frames)
{
 
  uint32_t n_ch = fddetdataformats::TDEEthFrame::s_num_channels;
  uint32_t n_smpl = fddetdataformats::TDEEthFrame::s_time_samples_per_frame;
 
  py::array_t<uint16_t> result(n_ch * n_smpl * n_frames);
 
  py::buffer_info buf_res = result.request();
 
  auto ptr_res = static_cast<uint16_t*>(buf_res.ptr);
 
  for (size_t i = 0; i < n_frames; ++i) {
 
    auto fr = reinterpret_cast<fddetdataformats::TDEEthFrame*>(static_cast<char*>(data) +
                                                               i * sizeof(fddetdataformats::TDEEthFrame));
 
    for (size_t j = 0; j < n_smpl; ++j) {
      for (size_t k = 0; k < n_ch; ++k) {
        ptr_res[(n_smpl * n_ch) * i + n_ch * j + k] = fr->get_adc(k, j);
      }
    }
  }
  result.resize({ n_frames * n_smpl, n_ch });
 
  return result;
}

py::array_t<long double>
np_array_timestamp_data(void* data, uint32_t n_frames)
{
 
  uint32_t n_smpl = fddetdataformats::TDEEthFrame::s_time_samples_per_frame;
 
  py::array_t<long double> result(n_smpl * n_frames);
 
  auto ptr = static_cast<long double*>(result.request().ptr);
 
  for (size_t i = 0; i < n_frames; ++i) {
    auto fr = reinterpret_cast<fddetdataformats::TDEEthFrame*>(static_cast<char*>(data) +
                                                               i * sizeof(fddetdataformats::TDEEthFrame));
    long double ts_0 = fr->get_timestamp();
    for (size_t j = 0; j < n_smpl; ++j)
      ptr[i * n_smpl + j] = ts_0 + 31.25 * j;
  }
 
  return result;
}

py::array_t<uint16_t>
np_array_adc(daqdataformats::Fragment const& frag)
{
  return np_array_adc_data(frag.get_data(), get_n_frames(frag));
}

py::array_t<long double>
np_array_timestamp(daqdataformats::Fragment const& frag)
{
  return np_array_timestamp_data(frag.get_data(), get_n_frames(frag));
}
 
} // namespace dunedaq::rawdatautils::tde // NOLINT