#include <NoPileUpPFMEtDataProducer.h>

Inheritance diagram for NoPileUpPFMEtDataProducer:

Public Member Functions
	NoPileUpPFMEtDataProducer (const edm::ParameterSet &)

	~NoPileUpPFMEtDataProducer () override

Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default

	EDProducer (const EDProducer &)=delete

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginProcessBlocks () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndProcessBlocks () const final

bool	hasAbilityToProduceInEndRuns () const final

const EDProducer &	operator= (const EDProducer &)=delete

Private Member Functions
void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
double	dZcut_

std::string	jetEnOffsetCorrLabel_

PileupJetIdentifier::Id	jetIdSelection_

PFJetIDSelectionFunctor *	looseJetIdAlgo_

int	maxWarnings_

double	minJetPt_

double	minJetPtForMEtCov_

std::string	moduleLabel_

int	numWarnings_

PFMEtSignInterfaceBase *	pfMEtSignInterface_

edm::EDGetTokenT < reco::VertexCollection >	srcHardScatterVertex_

edm::EDGetTokenT < edm::ValueMap< int > >	srcJetIds_

edm::EDGetTokenT < reco::PFJetCollection >	srcJets_

edm::EDGetTokenT < reco::PFJetCollection >	srcJetsForMEtCov_

edm::EDGetTokenT < reco::PFCandidateCollection >	srcPFCandidates_

edm::EDGetTokenT< edm::View < reco::PFCandidate > >	srcPFCandidatesView_

edm::EDGetTokenT < PFCandToVertexAssMap >	srcPFCandToVertexAssociations_

int	verbosity_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
using	CacheTypes = CacheContexts< T...>

using	GlobalCache = typename CacheTypes::GlobalCache

using	HasAbility = AbilityChecker< T...>

using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache

using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache

using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >

using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache

using	RunCache = typename CacheTypes::RunCache

using	RunContext = RunContextT< RunCache, GlobalCache >

using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Produce input objects used to compute MVA/No-PU MET

Authors: Phil Harris, CERN Christian Veelken, LLR

Definition at line 41 of file NoPileUpPFMEtDataProducer.h.

Constructor & Destructor Documentation

NoPileUpPFMEtDataProducer::NoPileUpPFMEtDataProducer ( const edm::ParameterSet & cfg )

Definition at line 15 of file NoPileUpPFMEtDataProducer.cc.

References edm::ParameterSet::addParameter(), dZcut_, Exception, edm::ParameterSet::exists(), edm::ParameterSet::getParameter(), jetEnOffsetCorrLabel_, jetIdSelection_, PileupJetIdentifier::kLoose, PileupJetIdentifier::kMedium, PileupJetIdentifier::kTight, looseJetIdAlgo_, maxWarnings_, minJetPt_, minJetPtForMEtCov_, numWarnings_, pfMEtSignInterface_, srcHardScatterVertex_, srcJetIds_, srcJets_, srcJetsForMEtCov_, srcPFCandidates_, srcPFCandidatesView_, srcPFCandToVertexAssociations_, AlCaHLTBitMon_QueryRunRegistry::string, and verbosity_.

     : moduleLabel_(cfg.getParameter<std::string>("@module_label")),
       looseJetIdAlgo_(nullptr),
       pfMEtSignInterface_(nullptr) {
   srcJets_ = consumes<reco::PFJetCollection>(cfg.getParameter<edm::InputTag>("srcJets"));
   srcJetIds_ = consumes<edm::ValueMap<int> >(cfg.getParameter<edm::InputTag>("srcJetIds"));
   minJetPt_ = cfg.getParameter<double>("minJetPt");
   std::string jetIdSelection_string = cfg.getParameter<std::string>("jetIdSelection");
   if (jetIdSelection_string == "loose")
     jetIdSelection_ = PileupJetIdentifier::kLoose;
   else if (jetIdSelection_string == "medium")
     jetIdSelection_ = PileupJetIdentifier::kMedium;
   else if (jetIdSelection_string == "tight")
     jetIdSelection_ = PileupJetIdentifier::kTight;
   else
     throw cms::Exception("NoPileUpPFMEtDataProducer")
         << "Invalid Configuration Parameter 'jetIdSelection' = " << jetIdSelection_string << " !!\n";
   jetEnOffsetCorrLabel_ = cfg.getParameter<std::string>("jetEnOffsetCorrLabel");
 
   srcPFCandidates_ = consumes<reco::PFCandidateCollection>(cfg.getParameter<edm::InputTag>("srcPFCandidates"));
   srcPFCandidatesView_ = consumes<edm::View<reco::PFCandidate> >(cfg.getParameter<edm::InputTag>("srcPFCandidates"));
   srcPFCandToVertexAssociations_ =
       consumes<PFCandToVertexAssMap>(cfg.getParameter<edm::InputTag>("srcPFCandToVertexAssociations"));
   srcJetsForMEtCov_ = mayConsume<reco::PFJetCollection>(cfg.getParameter<edm::InputTag>("srcJetsForMEtCov"));
   minJetPtForMEtCov_ = cfg.getParameter<double>("minJetPtForMEtCov");
   srcHardScatterVertex_ = consumes<reco::VertexCollection>(cfg.getParameter<edm::InputTag>("srcHardScatterVertex"));
   dZcut_ = cfg.getParameter<double>("dZcut");
 
   edm::ParameterSet cfgPFJetIdAlgo;
   cfgPFJetIdAlgo.addParameter<std::string>("version", "FIRSTDATA");
   cfgPFJetIdAlgo.addParameter<std::string>("quality", "LOOSE");
   looseJetIdAlgo_ = new PFJetIDSelectionFunctor(cfgPFJetIdAlgo);
 
   pfMEtSignInterface_ = new PFMEtSignInterfaceBase(cfg.getParameter<edm::ParameterSet>("resolution"));
 
   maxWarnings_ = (cfg.exists("maxWarnings")) ? cfg.getParameter<int>("maxWarnings") : 1;
   numWarnings_ = 0;
 
   verbosity_ = (cfg.exists("verbosity")) ? cfg.getParameter<int>("verbosity") : 0;
 
   produces<reco::PUSubMETCandInfoCollection>("jetInfos");
   produces<reco::PUSubMETCandInfoCollection>("pfCandInfos");
 }

NoPileUpPFMEtDataProducer::~NoPileUpPFMEtDataProducer ( )

override

Definition at line 59 of file NoPileUpPFMEtDataProducer.cc.

References looseJetIdAlgo_, and pfMEtSignInterface_.

                                                       {
   delete looseJetIdAlgo_;
   delete pfMEtSignInterface_;
 }

Member Function Documentation

void NoPileUpPFMEtDataProducer::produce	(	edm::Event &	evt,
		const edm::EventSetup &	es
	)

overrideprivate

Definition at line 130 of file NoPileUpPFMEtDataProducer.cc.

                                                                               {
   LogDebug("produce") << "<NoPileUpPFMEtDataProducer::produce>:\n"
                       << " moduleLabel = " << moduleLabel_ << std::endl;
 
   // get jets
   edm::Handle<reco::PFJetCollection> jets;
   evt.getByToken(srcJets_, jets);
 
   typedef edm::ValueMap<int> jetIdMap;
   edm::Handle<jetIdMap> jetIds;
   evt.getByToken(srcJetIds_, jetIds);
 
   // get jets for computing contributions to PFMEt significance matrix
   edm::Handle<reco::PFJetCollection> jetsForMEtCov;
   if (!srcJetsForMEtCov_.isUninitialized())
     evt.getByToken(srcJetsForMEtCov_, jetsForMEtCov);
 
   // get PFCandidates
   edm::Handle<edm::View<reco::PFCandidate> > pfCandidates;
   evt.getByToken(srcPFCandidatesView_, pfCandidates);
 
   std::vector<int> pfCandidateFlags(pfCandidates->size());
   std::vector<const reco::PFJet*> pfCandidateToJetAssociations(pfCandidates->size());
 
   edm::Handle<reco::PFCandidateCollection> pfCandidateHandle;
   evt.getByToken(srcPFCandidates_, pfCandidateHandle);
 
   // get PFCandidate-to-vertex associations and "the" hard-scatter vertex
   edm::Handle<PFCandToVertexAssMap> pfCandToVertexAssociations;
   evt.getByToken(srcPFCandToVertexAssociations_, pfCandToVertexAssociations);
 
   noPuUtils::reversedPFCandToVertexAssMap pfCandToVertexAssociations_reversed =
       noPuUtils::reversePFCandToVertexAssociation(*pfCandToVertexAssociations);
 
   edm::Handle<reco::VertexCollection> hardScatterVertex;
   evt.getByToken(srcHardScatterVertex_, hardScatterVertex);
 
   auto jetInfos = std::make_unique<reco::PUSubMETCandInfoCollection>();
   auto pfCandInfos = std::make_unique<reco::PUSubMETCandInfoCollection>();
 
   const JetCorrector* jetEnOffsetCorrector = nullptr;
   if (!jetEnOffsetCorrLabel_.empty()) {
     jetEnOffsetCorrector = JetCorrector::getJetCorrector(jetEnOffsetCorrLabel_, es);
     if (!jetEnOffsetCorrector)
       throw cms::Exception("NoPileUpPFMEtDataProducer::produce")
           << "Failed to access Jet corrections for = " << jetEnOffsetCorrLabel_ << " !!\n";
   }
 
   size_t numJets = jets->size();
   for (size_t iJet = 0; iJet < numJets; ++iJet) {
     reco::PFJetRef jet(jets, iJet);
     if (!(jet->pt() > minJetPt_))
       continue;
 
     bool passesLooseJetId = (*looseJetIdAlgo_)(*jet);
     if (!passesLooseJetId) {
       setPFCandidateFlag(*jet, *pfCandidates, pfCandidateFlags, flag_isWithinFakeJet, numWarnings_, maxWarnings_);
     }
     setPFCandidateFlag(*jet, *pfCandidates, pfCandidateFlags, flag_isWithinSelectedJet, numWarnings_, maxWarnings_);
 
     reco::PUSubMETCandInfo jetInfo;
     jetInfo.setP4(jet->p4());
     int jetId = (*jetIds)[jet];
     bool jetIdSelection_passed = PileupJetIdentifier::passJetId(jetId, jetIdSelection_);
     jetInfo.setType((jetIdSelection_passed) ? reco::PUSubMETCandInfo::kHS : reco::PUSubMETCandInfo::kPU);
     jetInfo.setPassesLooseJetId(passesLooseJetId);
     double jetEnergy_uncorrected = jet->chargedHadronEnergy() + jet->neutralHadronEnergy() + jet->photonEnergy() +
                                    jet->electronEnergy() + jet->muonEnergy() + jet->HFHadronEnergy() +
                                    jet->HFEMEnergy();
     double jetPx_uncorrected = cos(jet->phi()) * sin(jet->theta()) * jetEnergy_uncorrected;
     double jetPy_uncorrected = sin(jet->phi()) * sin(jet->theta()) * jetEnergy_uncorrected;
     double jetPz_uncorrected = cos(jet->theta()) * jetEnergy_uncorrected;
     reco::Candidate::LorentzVector rawJetP4(
         jetPx_uncorrected, jetPy_uncorrected, jetPz_uncorrected, jetEnergy_uncorrected);
     reco::PFJet rawJet(*jet);
     rawJet.setP4(rawJetP4);
     double jetNeutralEnFrac = (jetEnergy_uncorrected > 0.)
                                   ? (jet->neutralEmEnergy() + jet->neutralHadronEnergy()) / jetEnergy_uncorrected
                                   : -1.;
     jetInfo.setChargedEnFrac((1 - jetNeutralEnFrac));
     jetInfo.setOffsetEnCorr(
         (jetEnOffsetCorrector) ? rawJet.energy() * (1. - jetEnOffsetCorrector->correction(rawJet, evt, es)) : 0.);
     jetInfo.setMEtSignObj(pfMEtSignInterface_->compResolution(&(*jet)));
 
     jetInfos->push_back(jetInfo);
   }
   LogDebug("produce") << "#jetInfos = " << jetInfos->size() << std::endl;
 
   for (reco::PFJetCollection::const_iterator jet = jets->begin(); jet != jets->end(); ++jet) {
     if (jet->pt() > minJetPtForMEtCov_) {
       setPFCandidateFlag(*jet,
                          *pfCandidates,
                          pfCandidateFlags,
                          flag_isWithinJetForMEtCov,
                          numWarnings_,
                          maxWarnings_,
                          &pfCandidateToJetAssociations);
     }
   }
 
   size_t numPFCandidates = pfCandidates->size();
   for (size_t iPFCandidate = 0; iPFCandidate < numPFCandidates; ++iPFCandidate) {
     reco::PFCandidatePtr pfCandidatePtr = pfCandidates->ptrAt(iPFCandidate);
 
     int idx = pfCandidatePtr.key();
     reco::PUSubMETCandInfo pfCandInfo;
     pfCandInfo.setP4(pfCandidatePtr->p4());
     pfCandInfo.setCharge(pfCandidatePtr->charge());
     pfCandInfo.setType(-1);
     // CV: need to call isVertexAssociated_fast instead of isVertexAssociated function
     //    (makes run-time of MVAPFMEtDataProducer::produce decrease from ~1s per event to ~0.35s per event)
     //int vtxAssociationType = isVertexAssociated(*pfCandidatePtr, *pfCandToVertexAssociations, *hardScatterVertex, dZcut_);
     reco::PFCandidateRef pfCandidateRef(pfCandidateHandle, iPFCandidate);
     int vtxAssociationType = noPuUtils::isVertexAssociated_fast(
         pfCandidateRef, pfCandToVertexAssociations_reversed, *hardScatterVertex, dZcut_, numWarnings_, maxWarnings_);
     bool isHardScatterVertex_associated = (vtxAssociationType == noPuUtils::kChHSAssoc);
     if (pfCandidatePtr->charge() == 0)
       pfCandInfo.setType(reco::PUSubMETCandInfo::kNeutral);
     else if (isHardScatterVertex_associated)
       pfCandInfo.setType(reco::PUSubMETCandInfo::kChHS);
     else
       pfCandInfo.setType(reco::PUSubMETCandInfo::kChPU);
     pfCandInfo.setIsWithinJet((pfCandidateFlags[idx] & flag_isWithinSelectedJet));
     if (pfCandInfo.isWithinJet())
       pfCandInfo.setPassesLooseJetId((pfCandidateFlags[idx] & flag_isWithinFakeJet));
     else
       pfCandInfo.setPassesLooseJetId(true);
 
     // CV: for PFCandidates that are within PFJets (of Pt between 'minJetPtForMEtCov' and 'minJetPt'),
     //     take contribution to PFMEt significance matrix from associated PFJet.
     //    (energy uncertainty scaled by ratio of PFCandidate/PFJet energy)
     const reco::PFJet* jet_matched = pfCandidateToJetAssociations[idx];
     if (jet_matched) {
       metsig::SigInputObj pfCandResolution = pfMEtSignInterface_->compResolution(pfCandidatePtr.get());
       metsig::SigInputObj jetResolution = pfMEtSignInterface_->compResolution(jet_matched);
 
       metsig::SigInputObj metSign;
       metSign.set(pfCandResolution.get_type(),
                   pfCandResolution.get_energy(),
                   pfCandResolution.get_phi(),
                   jetResolution.get_sigma_e() * (pfCandidatePtr->energy() / jet_matched->energy()),
                   jetResolution.get_sigma_tan());
       pfCandInfo.setMEtSignObj(metSign);
     } else {
       pfCandInfo.setMEtSignObj(pfMEtSignInterface_->compResolution(pfCandidatePtr.get()));
     }
 
     pfCandInfos->push_back(pfCandInfo);
   }
 
   LogDebug("produce") << "#pfCandInfos = " << pfCandInfos->size() << std::endl;
 
   evt.put(std::move(jetInfos), "jetInfos");
   evt.put(std::move(pfCandInfos), "pfCandInfos");
 }