TDEUnpacker.cpp

Go to the documentation of this file.

 
#include "fddetdataformats/TDEEthFrame.hpp"
#include "daqdataformats/Fragment.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