MCVerticesWeight.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    PileUp
// Class:      MCVerticesWeight
//
//
// Original Author:  Andrea Venturi
//         Created:  Tue Oct 21 20:55:22 CEST 2008
//
//
 
// system include files
#include <memory>
#include <string>
 
// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/global/EDFilter.h"
 
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/ESWatcher.h"
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
 
#include "FWCore/MessageLogger/interface/MessageLogger.h"
 
#include "FWCore/Utilities/interface/InputTag.h"
 
#include "SimDataFormats/PileupSummaryInfo/interface/PileupSummaryInfo.h"
#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
#include "SimDataFormats/GeneratorProducts/interface/GenEventInfoProduct.h"
 
#include "Validation/RecoVertex/interface/VertexWeighter.h"
 
//
// class declaration
//
 
class MCVerticesWeight : public edm::global::EDFilter<> {
public:
  explicit MCVerticesWeight(const edm::ParameterSet&);
  ~MCVerticesWeight() override;
 
private:
  bool filter(edm::StreamID, edm::Event&, const edm::EventSetup&) const override;
 
  // ----------member data ---------------------------
 
  edm::EDGetTokenT<std::vector<PileupSummaryInfo> > m_vecPileupSummaryInfoToken;
  edm::EDGetTokenT<edm::HepMCProduct> m_hepMCProductToken;
  const VertexWeighter m_weighter;
};
 
//
// constructors and destructor
//
MCVerticesWeight::MCVerticesWeight(const edm::ParameterSet& iConfig)
    : m_vecPileupSummaryInfoToken(
          consumes<std::vector<PileupSummaryInfo> >(iConfig.getParameter<edm::InputTag>("pileupSummaryCollection"))),
      m_hepMCProductToken(consumes<edm::HepMCProduct>(iConfig.getParameter<edm::InputTag>("mcTruthCollection"))),
      m_weighter(iConfig.getParameter<edm::ParameterSet>("weighterConfig")) {
  produces<double>();
}
 
MCVerticesWeight::~MCVerticesWeight() {}
 
//
// member functions
//
 
// ------------ method called on each new Event  ------------
bool MCVerticesWeight::filter(edm::StreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const {
  bool selected = true;
 
  double computed_weight(1);
 
  edm::Handle<std::vector<PileupSummaryInfo> > pileupinfos;
  iEvent.getByToken(m_vecPileupSummaryInfoToken, pileupinfos);
 
  // look for the intime PileupSummaryInfo
 
  std::vector<PileupSummaryInfo>::const_iterator pileupinfo;
  for (pileupinfo = pileupinfos->begin(); pileupinfo != pileupinfos->end(); ++pileupinfo) {
    if (pileupinfo->getBunchCrossing() == 0)
      break;
  }
 
  //
  if (pileupinfo->getBunchCrossing() != 0) {
    edm::LogError("NoInTimePileUpInfo") << "Cannot find the in-time pileup info " << pileupinfo->getBunchCrossing();
  } else {
    //     pileupinfo->getPU_NumInteractions();
 
    const std::vector<float>& zpositions = pileupinfo->getPU_zpositions();
 
    //     for(std::vector<float>::const_iterator zpos = zpositions.begin() ; zpos != zpositions.end() ; ++zpos) {
 
    //     }
 
    // main interaction part
 
    edm::Handle<edm::HepMCProduct> EvtHandle;
    iEvent.getByToken(m_hepMCProductToken, EvtHandle);
 
    const HepMC::GenEvent* Evt = EvtHandle->GetEvent();
 
    // get the first vertex
 
    double zmain = 0.0;
    if (Evt->vertices_begin() != Evt->vertices_end()) {
      zmain = (*Evt->vertices_begin())->point3d().z() / 10.;
    }
 
    //
 
    computed_weight = m_weighter.weight(zpositions, zmain);
  }
 
  std::unique_ptr<double> weight(new double(computed_weight));
 
  iEvent.put(std::move(weight));
 
  //
 
  return selected;
}
 
//define this as a plug-in
DEFINE_FWK_MODULE(MCVerticesWeight);