ShallowEventDataProducer.cc

#include "CalibTracker/SiStripCommon/interface/ShallowEventDataProducer.h"
#include "DataFormats/L1GlobalTrigger/interface/L1GlobalTriggerReadoutRecord.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

// #include <bitset>

ShallowEventDataProducer::ShallowEventDataProducer(const edm::ParameterSet& iConfig) {
  produces<unsigned int>("run");
  produces<unsigned int>("event");
  produces<unsigned int>("bx");
  produces<unsigned int>("lumi");
  produces<float>("instLumi");
  produces<float>("PU");
#ifdef ExtendedCALIBTree
  produces<std::vector<bool>>("TrigTech");
  produces<std::vector<bool>>("TrigPh");
#endif

  trig_token_ = consumes<L1GlobalTriggerReadoutRecord>(iConfig.getParameter<edm::InputTag>("trigRecord"));
  scalerToken_ = consumes<LumiScalersCollection>(iConfig.getParameter<edm::InputTag>("lumiScalers"));
}

void ShallowEventDataProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  auto run = std::make_unique<unsigned int>(iEvent.id().run());
  auto event = std::make_unique<unsigned int>(iEvent.id().event());
  auto bx = std::make_unique<unsigned int>(iEvent.bunchCrossing());
  auto lumi = std::make_unique<unsigned int>(iEvent.luminosityBlock());

  edm::Handle<L1GlobalTriggerReadoutRecord> gtRecord;
  iEvent.getByToken(trig_token_, gtRecord);

#ifdef ExtendedCALIBTree
  std::vector<bool> TrigTech_(64, false);
  std::vector<bool> TrigPh_(128, false);
#endif

#ifdef ExtendedCALIBTree
  // Get dWord after masking disabled bits
  DecisionWord dWord = gtRecord->decisionWord();
  if (!dWord.empty()) {  // if board not there this is zero
    // loop over dec. bit to get total rate (no overlap)
    for (int i = 0; i < 64; ++i) {
      TrigPh_[i] = dWord[i];
    }
  }

  TechnicalTriggerWord tw = gtRecord->technicalTriggerWord();
  if (!tw.empty()) {
    // loop over dec. bit to get total rate (no overlap)
    for (int i = 0; i < 64; ++i) {
      TrigTech_[i] = tw[i];
    }
  }

  auto TrigTech = std::make_unique<std::vector<bool>>(TrigTech_);
  auto TrigPh = std::make_unique<std::vector<bool>>(TrigPh_);
#endif

  // Luminosity informations
  edm::Handle<LumiScalersCollection> lumiScalers;
  float instLumi_ = 0;
  float PU_ = 0;
  iEvent.getByToken(scalerToken_, lumiScalers);
  if (lumiScalers.isValid()) {
    if (lumiScalers->begin() != lumiScalers->end()) {
      instLumi_ = lumiScalers->begin()->instantLumi();
      PU_ = lumiScalers->begin()->pileup();
    }
  } else {
    edm::LogInfo("ShallowEventDataProducer")
        << "LumiScalers collection not found in the event; will write dummy values";
  }

  auto instLumi = std::make_unique<float>(instLumi_);
  auto PU = std::make_unique<float>(PU_);

  iEvent.put(std::move(run), "run");
  iEvent.put(std::move(event), "event");
  iEvent.put(std::move(bx), "bx");
  iEvent.put(std::move(lumi), "lumi");
#ifdef ExtendedCALIBTree
  iEvent.put(std::move(TrigTech), "TrigTech");
  iEvent.put(std::move(TrigPh), "TrigPh");
#endif
  iEvent.put(std::move(instLumi), "instLumi");
  iEvent.put(std::move(PU), "PU");
}