#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	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () 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 ( const edm::ParameterSet & cfg )

Definition at line 16 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 58 of file NoPileUpPFMEtDataProducer.cc.

References reco::deltaR2(), dR2Min, edm::View< T >::empty(), flags, reco::PFJet::getPFConstituents(), edm::Ptr< T >::id(), training_settings::idx, createfilelist::int, edm::Ptr< T >::key(), looseJetIdAlgo_, reco::LeafCandidate::p4(), pfMEtSignInterface_, edm::View< T >::ptrAt(), edm::View< T >::size(), and relativeConstraints::value.

 {
   delete looseJetIdAlgo_;
   delete pfMEtSignInterface_;
 }

Member Function Documentation

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

overrideprivate

Definition at line 125 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_ != "" ) {
     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");
 }