#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

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndRuns () const final

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<>
typedef CacheContexts< T... >	CacheTypes

typedef CacheTypes::GlobalCache	GlobalCache

typedef AbilityChecker< T... >	HasAbility

typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache

typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext

typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache

typedef CacheTypes::RunCache	RunCache

typedef RunContextT< RunCache, GlobalCache >	RunContext

typedef CacheTypes::RunSummaryCache	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::NoPileUpPFMEtDataProducer ( const edm::ParameterSet & cfg )

Definition at line 15 of file NoPileUpPFMEtDataProducer.cc.

     : 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");
 }

References edm::ParameterSet::addParameter(), looper::cfg, dZcut_, Exception, 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_.

◆ ~NoPileUpPFMEtDataProducer()

NoPileUpPFMEtDataProducer::~NoPileUpPFMEtDataProducer ( )

override

Definition at line 59 of file NoPileUpPFMEtDataProducer.cc.

                                                       {
   delete looseJetIdAlgo_;
   delete pfMEtSignInterface_;
 }

References looseJetIdAlgo_, and pfMEtSignInterface_.

Member Function Documentation

◆ produce()

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");
 }