00001 #ifndef mu2e_artdaq_core_Overlays_DTCFragmentWriter_hh
00002 #define mu2e_artdaq_core_Overlays_DTCFragmentWriter_hh
00003
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00016
00017 #include "artdaq-core/Data/Fragment.hh"
00018 #include "mu2e-artdaq-core/Overlays/DTCFragment.hh"
00019
00020 #include <iostream>
00021
00022 namespace mu2e {
00023 class DTCFragmentWriter;
00024 }
00025
00026 class mu2e::DTCFragmentWriter : public mu2e::DTCFragment {
00027 public:
00028 DTCFragmentWriter(artdaq::Fragment &f);
00029
00030
00031
00032
00033 packet_t *dataBegin();
00034 packet_t *dataEnd();
00035
00036
00037
00038
00039 Header *header_() {
00040 assert(artdaq_Fragment_.dataSize() >= words_to_frag_words_(Header::size_words));
00041 return reinterpret_cast<Header *>(&*artdaq_Fragment_.dataBegin());
00042 }
00043
00044 void set_hdr_timestamp(Header::timestamp_t timestamp) { header_()->timestamp = timestamp; }
00045
00046 void resize(size_t nPackets);
00047
00048 size_t size() { return artdaq_Fragment_.size(); }
00049
00050 private:
00051 size_t calc_event_size_words_(size_t nPackets);
00052
00053 static size_t packets_to_words_(size_t nPackets);
00054 static size_t words_to_frag_words_(size_t nWords);
00055
00056
00057 artdaq::Fragment &artdaq_Fragment_;
00058 };
00059
00060
00061
00062
00063
00064 mu2e::DTCFragmentWriter::DTCFragmentWriter(artdaq::Fragment &f) : DTCFragment(f), artdaq_Fragment_(f) {
00065
00066
00067
00068
00069
00070 assert(sizeof(Metadata::data_t) == sizeof(Header::data_t));
00071
00072 if (artdaq_Fragment_.size() !=
00073 artdaq::detail::RawFragmentHeader::num_words() + words_to_frag_words_(Metadata::size_words)) {
00074 std::cerr << "artdaq_Fragment size: " << artdaq_Fragment_.size() << std::endl;
00075 std::cerr << "Expected size: "
00076 << artdaq::detail::RawFragmentHeader::num_words() + words_to_frag_words_(Metadata::size_words)
00077 << std::endl;
00078
00079 throw cet::exception(
00080 "DTCFragmentWriter: Raw artdaq::Fragment object size suggests it does not consist of its own header + the "
00081 "DTCFragment::Metadata object");
00082 }
00083
00084
00085 artdaq_Fragment_.resize(words_to_frag_words_(Header::size_words));
00086 }
00087
00088 inline mu2e::packet_t *mu2e::DTCFragmentWriter::dataBegin() {
00089 assert(artdaq_Fragment_.dataSize() > words_to_frag_words_(Header::size_words));
00090 return reinterpret_cast<packet_t *>(header_() + 1);
00091 }
00092
00093 inline mu2e::packet_t *mu2e::DTCFragmentWriter::dataEnd() { return dataBegin() + hdr_packet_count(); }
00094
00095 inline void mu2e::DTCFragmentWriter::resize(size_t nPackets) {
00096 auto es(calc_event_size_words_(nPackets));
00097 artdaq_Fragment_.resize(words_to_frag_words_(es));
00098 header_()->event_size = nPackets;
00099 }
00100
00101 inline size_t mu2e::DTCFragmentWriter::calc_event_size_words_(size_t nPackets) {
00102 return packets_to_words_(nPackets) + hdr_size_words();
00103 }
00104
00105 inline size_t mu2e::DTCFragmentWriter::packets_to_words_(size_t nPackets) { return nPackets * words_per_packet_(); }
00106
00107 inline size_t mu2e::DTCFragmentWriter::words_to_frag_words_(size_t nWords) {
00108 size_t mod = nWords % words_per_frag_word_();
00109 return mod ? nWords / words_per_frag_word_() + 1 : nWords / words_per_frag_word_();
00110 }
00111
00112 #endif