WIBEthUnpacker.cpp

Go to the documentation of this file.

 
#include "fddetdataformats/WIBEthFrame.hpp"
#include "daqdataformats/Fragment.hpp"
 
#include <cstdint>
#include <pybind11/numpy.h>
// #include <fmt/core.h>
// #include <iostream>
 
namespace py = pybind11;
namespace dunedaq::rawdatautils::wibeth {

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

py::array_t<uint16_t> np_array_adc_data(void* data, uint32_t n_frames){
 
  uint32_t n_ch = fddetdataformats::WIBEthFrame::s_num_channels;
  uint32_t n_smpl = fddetdataformats::WIBEthFrame::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::WIBEthFrame*>(
      static_cast<char*>(data) + i * sizeof(fddetdataformats::WIBEthFrame)
    );
 
    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<uint64_t> np_array_timestamp_data(void* data, uint32_t n_frames){
 
  uint32_t n_smpl = fddetdataformats::WIBEthFrame::s_time_samples_per_frame;
 
  py::array_t<uint64_t> result(n_smpl*n_frames);
 
  auto ptr = static_cast<uint64_t*>(result.request().ptr);
  
  for (size_t i=0; i<n_frames; ++i) {
    auto fr = reinterpret_cast<fddetdataformats::WIBEthFrame*>(
      static_cast<char*>(data) + i * sizeof(fddetdataformats::WIBEthFrame)
    );
    uint64_t ts_0 = fr->get_timestamp();
    for(size_t j=0; j<n_smpl; ++j )
      ptr[i*n_smpl+j] = ts_0+32*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<uint64_t> np_array_timestamp(daqdataformats::Fragment const& frag){
  return np_array_timestamp_data(frag.get_data(), get_n_frames(frag));
}
 
 
} // namespace dunedaq::rawdatautils::wibeth // NOLINT