1 #ifndef mu2e_artdaq_Overlays_DetectorFragmentWriter_hh
2 #define mu2e_artdaq_Overlays_DetectorFragmentWriter_hh
17 #include "artdaq-core/Data/Fragment.hh"
18 #include "mu2e-artdaq-core/Overlays/DetectorFragment.hh"
23 class DetectorFragmentWriter;
42 assert(artdaq_Fragment_.dataSize() >= words_to_frag_words_(Header::size_words));
43 return reinterpret_cast<Header *
>(&*artdaq_Fragment_.dataBegin());
46 void set_hdr_run_number(Header::run_number_t run_number) { header_()->run_number = run_number; }
48 void resize(
size_t nAdcs);
50 virtual void printAll(){};
52 void generateOffsetTable(
const std::vector<size_t> dataBlockVec);
55 size_t calc_event_size_words_(
size_t nAdcs);
57 static size_t adcs_to_words_(
size_t nAdcs);
58 static size_t words_to_frag_words_(
size_t nWords);
61 artdaq::Fragment &artdaq_Fragment_;
68 mu2e::DetectorFragmentWriter::DetectorFragmentWriter(artdaq::Fragment &f) :
DetectorFragment(f), artdaq_Fragment_(f) {
74 assert(
sizeof(Metadata::data_t) ==
sizeof(Header::data_t));
76 if (artdaq_Fragment_.size() !=
77 artdaq::detail::RawFragmentHeader::num_words() + words_to_frag_words_(Metadata::size_words)) {
78 std::cerr <<
"artdaq_Fragment size: " << artdaq_Fragment_.size() << std::endl;
79 std::cerr <<
"Expected size: "
80 << artdaq::detail::RawFragmentHeader::num_words() + words_to_frag_words_(Metadata::size_words)
84 "DetectorFragmentWriter: Raw artdaq::Fragment object size suggests it does not consist of its own header + the "
85 "DetectorFragment::Metadata object");
89 artdaq_Fragment_.resize(words_to_frag_words_(Header::size_words));
92 inline mu2e::DetectorFragment::adc_t *mu2e::DetectorFragmentWriter::dataBegin() {
93 assert(artdaq_Fragment_.dataSize() > words_to_frag_words_(Header::size_words));
95 return (reinterpret_cast<adc_t *>(header_() + 1)) + current_offset_;
98 inline mu2e::DetectorFragment::adc_t *mu2e::DetectorFragmentWriter::dataEnd() {
99 return dataBegin() + total_adc_values_in_data_block();
103 inline void mu2e::DetectorFragmentWriter::resize(
size_t nAdcs) {
104 auto es(calc_event_size_words_(nAdcs));
105 artdaq_Fragment_.resize(words_to_frag_words_(es));
106 header_()->event_size = es;
109 inline size_t mu2e::DetectorFragmentWriter::calc_event_size_words_(
size_t nAdcs) {
110 return adcs_to_words_(nAdcs) + hdr_size_words();
113 inline size_t mu2e::DetectorFragmentWriter::adcs_to_words_(
size_t nAdcs) {
114 auto mod(nAdcs % adcs_per_word_());
115 return (mod == 0) ? nAdcs / adcs_per_word_() : nAdcs / adcs_per_word_() + 1;
118 inline size_t mu2e::DetectorFragmentWriter::words_to_frag_words_(
size_t nWords) {
119 size_t mod = nWords % words_per_frag_word_();
120 return mod ? nWords / words_per_frag_word_() + 1 : nWords / words_per_frag_word_();
123 inline void mu2e::DetectorFragmentWriter::generateOffsetTable(
const std::vector<size_t> dataBlockVec) {
129 *(dataBegin()) = (adc_t)dataBlockVec.size();
130 for (
size_t cur_index = 0, cur_offset = dataBlockVec.size() + 1; cur_index < dataBlockVec.size(); cur_index++) {
133 cur_offset += 8 * dataBlockVec[cur_index];
134 *(dataBegin() + 1 + cur_index) = (adc_t)cur_offset;
136 setDataBlockIndex(0);