PUDumper.cc

Go to the documentation of this file.

#include "Calibration/EcalAlCaRecoProducers/plugins/PUDumper.h"

PUDumper::PUDumper(const edm::ParameterSet& iConfig) {
  pileupSummaryToken_ = consumes<std::vector<PileupSummaryInfo> >(iConfig.getParameter<edm::InputTag>("pileupSummary"));

  // create TTree
  edm::Service<TFileService> fs;
  PUTree_ = fs->make<TTree>("pileup", "pileup");

  PUTree_->Branch("runNumber", &runNumber, "runNumber/I");
  PUTree_->Branch("eventNumber", &eventNumber, "eventNumber/l");
  PUTree_->Branch("lumiBlock", &lumiBlock, "lumiBlock/I");

  PUTree_->Branch("nBX", &nBX, "nBX/I");
  PUTree_->Branch("BX", BX_, "BX[nBX]/I");
  PUTree_->Branch("nPUtrue", &nPUtrue_, "nPUtrue/I");
  PUTree_->Branch("nPUobs", nPUobs_, "nPUobs[nBX]/I");
}

// ----------------------------------------------------------------

PUDumper::~PUDumper() {}

// ----------------------------------------------------------------

void PUDumper::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) {
  // get the PU collection
  edm::Handle<std::vector<PileupSummaryInfo> > PupInfo;
  if (!iEvent.isRealData()) {
    iEvent.getByToken(pileupSummaryToken_, PupInfo);
  } else
    return;

  runNumber = iEvent.id().run();
  eventNumber = iEvent.id().event();
  if (iEvent.isRealData()) {
    lumiBlock = iEvent.luminosityBlock();
  } else {
    lumiBlock = -1;
  }

  // loop on BX
  nBX = 0;
  std::vector<PileupSummaryInfo>::const_iterator PVI;
  nPUtrue_ = PupInfo->begin()->getTrueNumInteractions();

  for (PVI = PupInfo->begin(); PVI != PupInfo->end(); ++PVI) {
    BX_[nBX] = PVI->getBunchCrossing();
    nPUobs_[nBX] = PVI->getPU_NumInteractions();
#ifdef DEBUG
    std::cout << "PUDumper::runNumber: " << runNumber_ << "   BX[1]: " << BX_[1] << "   nPUtrue: " << nPUtrue_
              << "   nPUobs[1]: " << nPUobs_[1] << std::endl;
#endif
    nBX++;
  }
  PUTree_->Fill();
}

DEFINE_FWK_MODULE(PUDumper);