#include <NoPileUpPFMEtDataProducer.h>

Inheritance diagram for NoPileUpPFMEtDataProducer:

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

	~NoPileUpPFMEtDataProducer ()

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

Public Member Functions inherited from edm::stream::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducerBase ()

Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

virtual	~ProducerBase ()

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

	EDConsumerBase ()

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

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

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

Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Static Public Member Functions inherited from edm::stream::EDProducerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

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_, edm::hlt::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 ( )

Definition at line 58 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
	)

privatevirtual

Implements edm::stream::EDProducerBase.

Definition at line 121 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);
   
   std::auto_ptr<reco::PUSubMETCandInfoCollection> jetInfos(new reco::PUSubMETCandInfoCollection());
   std::auto_ptr<reco::PUSubMETCandInfoCollection> pfCandInfos(new 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(jetInfos,"jetInfos");
   evt.put(pfCandInfos,"pfCandInfos");
 }