DQMStreamerReader.cc

Go to the documentation of this file.

#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Utilities/interface/Exception.h"
#include "FWCore/Utilities/interface/EDMException.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/ParameterSet/interface/Registry.h"
#include "FWCore/Sources/interface/EventSkipperByID.h"

#include "DataFormats/Provenance/interface/ProductRegistry.h"
#include "DataFormats/Provenance/interface/ProcessHistoryRegistry.h"
#include "FWCore/Utilities/interface/RegexMatch.h"
#include "DQMStreamerReader.h"

#include <fstream>
#include <queue>
#include <cstdlib>
#include <boost/regex.hpp>
#include <boost/format.hpp>
#include <boost/range.hpp>
#include <boost/filesystem.hpp>
#include <boost/algorithm/string.hpp>

#include <IOPool/Streamer/interface/DumpTools.h>

namespace dqmservices {

DQMStreamerReader::DQMStreamerReader(edm::ParameterSet const& pset,
                                     edm::InputSourceDescription const& desc)
    : StreamerInputSource(pset, desc),
      fiterator_(pset),
      streamReader_() {

  runNumber_ = pset.getUntrackedParameter<unsigned int>("runNumber");
  runInputDir_ = pset.getUntrackedParameter<std::string>("runInputDir");
  hltSel_ =
      pset.getUntrackedParameter<std::vector<std::string> >("SelectEvents");

  minEventsPerLs_ = pset.getUntrackedParameter<int>("minEventsPerLumi");
  flagSkipFirstLumis_ = pset.getUntrackedParameter<bool>("skipFirstLumis");
  flagEndOfRunKills_ = pset.getUntrackedParameter<bool>("endOfRunKills");
  flagDeleteDatFiles_ = pset.getUntrackedParameter<bool>("deleteDatFiles");

  triggerSel();

  reset_();
}

DQMStreamerReader::~DQMStreamerReader() { closeFile_(); }

void DQMStreamerReader::reset_() {
  // We have to load at least a single header,
  // so the ProductRegistry gets initialized.
  //
  // This must happen here (inside the constructor),
  // as ProductRegistry gets frozen after we initialize:
  // https://cmssdt.cern.ch/SDT/lxr/source/FWCore/Framework/src/Schedule.cc#441

  fiterator_.logFileAction(
      "Waiting for the first lumi in order to initialize.");

  fiterator_.update_state();

  // Fast-forward to the last open file.
  if (flagSkipFirstLumis_) {
    unsigned int l = fiterator_.lastLumiFound();
    if (l > 1) {
      fiterator_.advanceToLumi(l);
    }
  }

  for (;;) {
    bool next = prepareNextFile();

    // check for end of run
    if (!next) {
      fiterator_.logFileAction(
          "End of run reached before DQMStreamerReader was initialised.");
      return;
    }

    // check if we have a file openned
    if (streamReader_.get() != nullptr) {
      // we are now initialised
      break;
    }

    // wait
    fiterator_.delay();
  }

  fiterator_.logFileAction("DQMStreamerReader initialised.");
}

void DQMStreamerReader::openFile_(std::string newStreamerFile_) {
  processedEventPerLs_ = 0;
  edm::ParameterSet pset;

  streamReader_ = std::unique_ptr<edm::StreamerInputFile>(
      new edm::StreamerInputFile(newStreamerFile_));

  InitMsgView const* header = getHeaderMsg();
  deserializeAndMergeWithRegistry(*header, false);

  // dump the list of HLT trigger name from the header
  //  dumpInitHeader(header);

  // if specific trigger selection is requested, check if the requested triggers 
  // match with trigger paths in the header file 
  if (!acceptAllEvt_){
    Strings tnames;
    header->hltTriggerNames(tnames);
    
    pset.addParameter<Strings>("SelectEvents", hltSel_);
    eventSelector_.reset(new TriggerSelector(pset, tnames));

    // check if any trigger path name requested matches with trigger name in the header file
    matchTriggerSel(tnames);
  }

  // our initialization
  processedEventPerLs_ = 0;

  if (flagDeleteDatFiles_) {
    // unlink the file
    unlink(newStreamerFile_.c_str());
  }
}

void DQMStreamerReader::closeFile_() {
  if (streamReader_.get() != nullptr) {
    streamReader_->closeStreamerFile();
    streamReader_ = nullptr;
  }
}

bool DQMStreamerReader::openNextFile_() {
  closeFile_();

  currentLumi_  = fiterator_.open();
  std::string p = fiterator_.make_path_data(currentLumi_);

  if (boost::filesystem::exists(p)) {
    openFile_(p);
    return true;
  } else {
    /* dat file missing */
    fiterator_.logFileAction("Data file (specified in json) is missing:", p);
    fiterator_.logLumiState(currentLumi_, "error: data file missing");

    return false;
  }
}

InitMsgView const* DQMStreamerReader::getHeaderMsg() {
  InitMsgView const* header = streamReader_->startMessage();

  if (header->code() != Header::INIT) {  // INIT Msg
    throw edm::Exception(edm::errors::FileReadError, "DQMStreamerReader::readHeader")
        << "received wrong message type: expected INIT, got " << header->code()
        << "\n";
  }

  return header;
}

EventMsgView const* DQMStreamerReader::getEventMsg() {
  if (!streamReader_->next()) {
    return nullptr;
  }

  EventMsgView const* msg = streamReader_->currentRecord();

  //  if (msg != nullptr) dumpEventView(msg);
  return msg;
}

bool DQMStreamerReader::prepareNextFile() {
  typedef DQMFileIterator::State State;

  for (;;) {
    fiterator_.update_state();

    // check for end of run file and force quit
    if (flagEndOfRunKills_ && (fiterator_.state() != State::OPEN)) {
      closeFile_();
      return false;
    }

    // check for end of run and quit if everything has been processed.
    // this clean exit
    if ((streamReader_.get() == nullptr) && (!fiterator_.lumiReady()) &&
        (fiterator_.state() == State::EOR)) {

      closeFile_();
      return false;
    }

    // if this is end of run and no more files to process
    // close it
    if ((processedEventPerLs_ >= minEventsPerLs_) &&
        (!fiterator_.lumiReady()) && (fiterator_.state() == State::EOR)) {

      closeFile_();
      return false;
    }

    // skip to the next file if we have no files openned yet
    if (streamReader_.get() == nullptr) {
      if (fiterator_.lumiReady()) {
        openNextFile_();
        // we might need to open once more (if .dat is missing)
        continue;
      }
    }

    // or if there is a next file and enough eventshas been processed.
    if (fiterator_.lumiReady() && (processedEventPerLs_ >= minEventsPerLs_)) {
      openNextFile_();
      // we might need to open once more (if .dat is missing)
      continue;
    }

    return true;
  }
}

EventMsgView const* DQMStreamerReader::prepareNextEvent() {
  EventMsgView const* eview = nullptr;
  typedef DQMFileIterator::State State;

  // wait for the next event
  for (;;) {
    // edm::LogAbsolute("DQMStreamerReader")
    //     << "State loop.";
    bool next = prepareNextFile();
    if (!next) return nullptr;

    // sleep
    if (streamReader_.get() == nullptr) {
      // the reader does not exist
      fiterator_.delay();
    } else {
      // our reader exists, try to read out an event
      eview = getEventMsg();

      if (eview == nullptr) {
        // read unsuccessful
        // this means end of file, so close the file
        closeFile_();
      } else {
        if (!acceptEvent(eview)) {
          continue;
        } else {
          return eview;
        }
      }
    }
  }
  return eview;
}

bool DQMStreamerReader::checkNextEvent() {
  EventMsgView const* eview = prepareNextEvent();
  if (eview == nullptr) {
    return false;
  }

  // this is reachable only if eview is set
  // and the file is openned
  if (streamReader_->newHeader()) {
    // A new file has been opened and we must compare Headers here !!
    // Get header/init from reader
    InitMsgView const* header = getHeaderMsg();
    deserializeAndMergeWithRegistry(*header, true);
  }

  processedEventPerLs_ += 1;
  deserializeEvent(*eview);

  return true;
}

bool DQMStreamerReader::triggerSel() {
  acceptAllEvt_ = false;
  for (Strings::const_iterator i(hltSel_.begin()), end(hltSel_.end()); 
       i!=end; ++i){
    std::string hltPath(*i);
    boost::erase_all(hltPath, " \t"); 
    if (hltPath == "*") acceptAllEvt_ = true;
  }
  return acceptAllEvt_;
}

bool DQMStreamerReader::matchTriggerSel(Strings const& tnames) {
  matchTriggerSel_ = false;
  for (Strings::const_iterator i(hltSel_.begin()), end(hltSel_.end()); 
       i!=end; ++i){
    std::string hltPath(*i);
    boost::erase_all(hltPath, " \t");
    std::vector<Strings::const_iterator> matches = edm::regexMatch(tnames, hltPath);
    if (matches.empty()){
      edm::LogWarning("Trigger selection does not match any trigger path!!!") << std::endl; 
      matchTriggerSel_ = false;
    }else{
      matchTriggerSel_ = true;
    }
  }
  return matchTriggerSel_;
}

bool DQMStreamerReader::acceptEvent(const EventMsgView* evtmsg) {

  if (acceptAllEvt_) return true;
  if (!matchTriggerSel_) return false;

  std::vector<unsigned char> hltTriggerBits_;
  int hltTriggerCount_ = evtmsg->hltCount();
  if (hltTriggerCount_ > 0) {
    hltTriggerBits_.resize(1 + (hltTriggerCount_ - 1) / 4);
  }
  evtmsg->hltTriggerBits(&hltTriggerBits_[0]);

  if (eventSelector_->wantAll() ||
      eventSelector_->acceptEvent(&hltTriggerBits_[0], evtmsg->hltCount())) {
    return true;
  }else{
    return false;
  }
}

void DQMStreamerReader::skip(int toSkip) {
  for (int i = 0; i != toSkip; ++i) {
    EventMsgView const* evMsg = prepareNextEvent();

    if (evMsg == nullptr) {
      return;
    }
  }
}

void DQMStreamerReader::fillDescriptions(
    edm::ConfigurationDescriptions& descriptions) {

  edm::ParameterSetDescription desc;
  desc.setComment("Reads events from streamer files.");

  desc.addUntracked<std::vector<std::string> >("SelectEvents")
      ->setComment("HLT path to select events ");

  desc.addUntracked<int>("minEventsPerLumi", 1)->setComment(
      "Minimum number of events to process per lumisection, "
      "before switching to a new input file. If the next file "
      "does not yet exist, "
      "the number of processed events will be bigger.");

  desc.addUntracked<bool>("skipFirstLumis", false)->setComment(
      "Skip (and ignore the minEventsPerLumi parameter) for the files "
      "which have been available at the begining of the processing. "
      "If set to true, the reader will open last available file for "
      "processing.");

  desc.addUntracked<bool>("deleteDatFiles", false)->setComment(
      "Delete data files after they have been closed, in order to "
      "save disk space.");

  desc.addUntracked<bool>("endOfRunKills", false)->setComment(
      "Kill the processing as soon as the end-of-run file appears, even if "
      "there are/will be unprocessed lumisections.");

  // desc.addUntracked<unsigned int>("skipEvents", 0U)
  //    ->setComment("Skip the first 'skipEvents' events that otherwise would "
  //                 "have been processed.");

  // This next parameter is read in the base class, but its default value
  // depends on the derived class, so it is set here.
  desc.addUntracked<bool>("inputFileTransitionsEachEvent", false);

  DQMFileIterator::fillDescription(desc);
  edm::StreamerInputSource::fillDescription(desc);
  edm::EventSkipperByID::fillDescription(desc);

  descriptions.add("source", desc);
}

} // end of namespace

#include "FWCore/Framework/interface/InputSourceMacros.h"
#include "FWCore/Framework/interface/MakerMacros.h"

typedef dqmservices::DQMStreamerReader DQMStreamerReader;
DEFINE_FWK_INPUT_SOURCE(DQMStreamerReader);