File.cxx

 
#include "Packing/RawDataFile/File.h"
 
#include "DetDescr/DetectorID.h"
#include "Packing/Utility/CRC.h"
#include "Packing/Utility/Mask.h"
 
namespace packing {
namespace rawdatafile {
 
File::File(const framework::config::Parameters &ps) {
  is_output_ = ps.get<bool>("is_output");
  skip_unavailable_ = ps.get<bool>("skip_unavailable");
 
  std::string fn = ps.get<std::string>("filename");
 
  ecal_object_name_ = ps.get<std::string>("ecal_object_name");
  hcal_object_name_ = ps.get<std::string>("hcal_object_name");
  tracker_object_name_ = ps.get<std::string>("tracker_object_name");
  triggerpad_object_name_ = ps.get<std::string>("triggerpad_object_name");
  pass_name_ = ps.get<std::string>("pass_name");
 
  std::cerr << "creating file" << std::endl;
  if (is_output_) {
    writer_.open(fn);
    // leave entry count undefined
    entries_ = 0;
    i_entry_ = 0;
  } else {
    reader_.open(fn);
    // get entry count from file
    // get run id number from file
    uint32_t word;
    reader_ >> word;
 
    uint8_t version = word & utility::MASK<4>;
    if (version != 0) {
      EXCEPTION_RAISE("RawFileVers", "Unable to handle raw file version " +
                                         std::to_string(version));
    }
 
    run_ = ((word >> 4) & utility::MASK<28>);
 
    reader_.seek<uint32_t>(-2, std::ios::end);
    // save EOF in number of 32-bit-width words
    auto eof{reader_.tell<uint32_t>()};
    uint32_t crc_read_in;
    reader_ >> entries_ >> crc_read_in;
    i_entry_ = 0;
 
    reader_.seek<uint32_t>(1, std::ios::beg);
 
    if (ps.get<bool>("verify_checksum")) {
      utility::CRC crc;
      for (auto ifile{reader_.tell<uint32_t>()}; ifile < eof; ifile++) {
        reader_ >> word;
        crc << word;
      }
 
      if (crc.get() != crc_read_in) {
        EXCEPTION_RAISE("CRCNotOk",
                        "Failure to verify CRC checksum of entire input file.");
      }
 
      reader_.seek<uint32_t>(1, std::ios::beg);
    }  // verify checksum of input file
  }  // input or output file
}
 
bool File::connect(framework::Event &event) {
  event_ = &event;
  return true;
}
 
bool File::nextEvent() {
  static std::map<uint16_t, std::string> eid_to_name = {
      {ldmx::SubdetectorIDType::EID_TRACKER, tracker_object_name_},
      {ldmx::SubdetectorIDType::EID_TRIGGER_SCINT, triggerpad_object_name_},
      {ldmx::SubdetectorIDType::EID_ECAL, ecal_object_name_},
      {ldmx::SubdetectorIDType::EID_HCAL, hcal_object_name_},
  };
 
  if (is_output_) {
    // dump buffers into event packet and write out
    std::map<uint16_t, std::vector<uint32_t>> the_subsys_data;
    for (auto const &[id, name] : eid_to_name) {
      if (skip_unavailable_ and not event_->exists(name, pass_name_)) continue;
      the_subsys_data[id] = event_->getCollection<uint32_t>(name, pass_name_);
    }
 
    EventPacket write_event(event_->getEventNumber(), the_subsys_data);
    writer_ << write_event;
    crc_ << write_event;
    if (!writer_) return false;
 
    entries_++;
    i_entry_++;
  } else {
    // check for EoF
    if (i_entry_ + 1 > entries_) return false;
 
    i_entry_++;
 
    // read buffers from event packet and add to event bus
    static EventPacket read_event;
    reader_ >> read_event;
    if (!reader_) {
      // ERROR
      return false;
    }
 
    event_->getEventHeader().setEventNumber(read_event.id());
 
    for (auto &subsys : read_event.data()) {
      // construct name if not provided by default EID mappings
      if (eid_to_name.find(subsys.id()) == eid_to_name.end()) {
        std::cerr << subsys.id() << " unrecognized electronics ID."
                  << std::endl;
        eid_to_name[subsys.id()] = "EID" + std::to_string(subsys.id());
      }
 
      event_->add(eid_to_name.at(subsys.id()), subsys.data());
    }  // loop over subsystems
  }  // input or output
  return true;
}
 
void File::writeRunHeader(ldmx::RunHeader &header) {
  if (is_output_) {
    // use passed run number
    run_ = header.getRunNumber();
    // Why cant we just take the header from above?
    uint32_t temp_header =
        (0 & utility::MASK<4>)+((run_ & utility::MASK<28>) << 4);
    writer_ << temp_header;
    crc_ << temp_header;
  } else {
    // put our read-in run number here
    header.setIntParameter("raw_run", run_);
  }
}
 
void File::close() {
  event_ = nullptr;
  if (is_output_) {
    writer_ << entries_;
    crc_ << entries_;
    writer_ << crc_.get();
  }
}
 
}  // namespace rawdatafile
}  // namespace packing