#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
	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

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 39 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::MVAMEtJetInfoCollection>();
   produces<reco::MVAMEtPFCandInfoCollection>();
 }

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
   typedef edm::View<reco::PFCandidate> PFCandidateView;
   edm::Handle<PFCandidateView> 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);
 
   reversedPFCandidateToVertexAssociationMap pfCandToVertexAssociations_reversed = reversePFCandToVertexAssociation(*pfCandToVertexAssociations);
  
   edm::Handle<reco::VertexCollection> hardScatterVertex;
   evt.getByToken(srcHardScatterVertex_, hardScatterVertex);
   // if ( verbosity_ && hardScatterVertex->size() >= 1 ) {
   //   reco::Vertex::Point hardScatterVertexPos = hardScatterVertex->front().position();
   //   std::cout << "hard-scatter Vertex: x = " << hardScatterVertexPos.x() << ", y = " << hardScatterVertexPos.y() << ", z = " << hardScatterVertexPos.z() << std::endl;    
   //   for ( PFCandToVertexAssMap::const_iterator pfCandToVertexAssociation = pfCandToVertexAssociations->begin();
   //      pfCandToVertexAssociation != pfCandToVertexAssociations->end(); ++pfCandToVertexAssociation ) {
   //     reco::VertexRef vertex = pfCandToVertexAssociation->key;      
   //     const PFCandQualityPairVector& pfCandidates_vertex = pfCandToVertexAssociation->val;
   //     for ( PFCandQualityPairVector::const_iterator pfCandidate_vertex = pfCandidates_vertex.begin();
   //        pfCandidate_vertex != pfCandidates_vertex.end(); ++pfCandidate_vertex ) {
   //    const reco::PFCandidate& pfCandidate = (*pfCandidate_vertex->first);
   //    int pfCandToVertexAssocQuality = pfCandidate_vertex->second;
   //    if ( pfCandidate.pt() > 2. && pfCandidate.charge() != 0 ) {
   //      double dZ = -1.;
   //      if      ( pfCandidate.trackRef().isNonnull()    ) dZ = std::abs(pfCandidate.trackRef()->dz(hardScatterVertexPos));
   //      else if ( pfCandidate.gsfTrackRef().isNonnull() ) dZ = std::abs(pfCandidate.gsfTrackRef()->dz(hardScatterVertexPos));
   //      std::cout << "pfCand #" << pfCandidate_vertex->first.key() << ": Pt = " << pfCandidate.pt() << ", eta = " << pfCandidate.eta() << ", phi = " << pfCandidate.phi() 
   //            << " (charge = " << pfCandidate.charge() << ", dZ = " << dZ << ")" << std::endl;
   //      reco::Vertex::Point vertexPos = vertex->position();
   //      std::cout << " associated vertex (quality = " << pfCandToVertexAssocQuality << "):" 
   //            << " x = " << vertexPos.x() << ", y = " << vertexPos.y() << ", z = " << vertexPos.z() << std::endl;
   //    }
   //     }
   //   }
   // }
 
   std::auto_ptr<reco::MVAMEtJetInfoCollection> jetInfos(new reco::MVAMEtJetInfoCollection());
   std::auto_ptr<reco::MVAMEtPFCandInfoCollection> pfCandInfos(new reco::MVAMEtPFCandInfoCollection());
  
   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::MVAMEtJetInfo jetInfo;
     jetInfo.p4_ = jet->p4();
     int jetId = (*jetIds)[jet];
     bool jetIdSelection_passed = PileupJetIdentifier::passJetId(jetId, jetIdSelection_);
     jetInfo.type_ = ( jetIdSelection_passed ) ?
       reco::MVAMEtJetInfo::kNoPileUp : reco::MVAMEtJetInfo::kPileUp;
     jetInfo.passesLooseJetId_ = 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.neutralEnFrac_ = jetNeutralEnFrac;
     jetInfo.offsetEnCorr_ = ( jetEnOffsetCorrector ) ?
       rawJet.energy()*(1. - jetEnOffsetCorrector->correction(rawJet, evt, es)) : 0.;
     jetInfo.pfMEtSignObj_ = pfMEtSignInterface_->compResolution(&(*jet));
     // if ( verbosity_ ) {
     //   std::cout << " jet: Pt = " << jet->pt() << ", eta = " << jet->eta() << ", phi = " << jet->phi() << ":" 
     //          << " sigma(En)/En = " << (jetInfo.pfMEtSignObj_.get_sigma_e()/jet->energy()) << std::endl;
     //   std::cout << "(offsetEnCorr = " << jetInfo.offsetEnCorr_ << ")" << std::endl;
     //   std::cout << "  id. flags: PU = " << jetIdSelection_passed << ", anti-noise = " << passesLooseJetId << std::endl;
      
     //   size_t numPFJetConstituents = jet->getPFConstituents().size();
     //   std::cout << "numConstituents = " << numPFJetConstituents << std::endl;
     //   for ( size_t iPFJetConstituent = 0; iPFJetConstituent < numPFJetConstituents; ++iPFJetConstituent ) {
     //  const reco::PFCandidatePtr& pfJetConstituent = jet->getPFConstituents()[iPFJetConstituent];
     //  std::cout << " constituent #" << iPFJetConstituent << ": Pt = " << pfJetConstituent->pt() << ", eta = " << pfJetConstituent->eta() << ", phi = " << pfJetConstituent->phi() << std::endl;
     //   }
     // }
     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::MVAMEtPFCandInfo pfCandInfo;
     pfCandInfo.p4_ = pfCandidatePtr->p4();
     pfCandInfo.charge_ = pfCandidatePtr->charge();
     pfCandInfo.type_ = -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 = isVertexAssociated_fast(pfCandidateRef, pfCandToVertexAssociations_reversed, *hardScatterVertex, dZcut_, numWarnings_, maxWarnings_);
     bool isHardScatterVertex_associated = (vtxAssociationType == noPuUtils::kChHSAssoc);
     if      ( pfCandidatePtr->charge() != 0 )   pfCandInfo.type_ = reco::MVAMEtPFCandInfo::kNeutral;
     else if ( isHardScatterVertex_associated       ) pfCandInfo.type_ = reco::MVAMEtPFCandInfo::kNoPileUpCharged;
     else                                             pfCandInfo.type_ = reco::MVAMEtPFCandInfo::kPileUpCharged;
     pfCandInfo.isWithinJet_ = (pfCandidateFlags[idx] & flag_isWithinSelectedJet);
     if ( pfCandInfo.isWithinJet_ ) pfCandInfo.passesLooseJetId_ = (pfCandidateFlags[idx] & flag_isWithinFakeJet);
     else pfCandInfo.passesLooseJetId_ = 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);
       //std::cout << " jet (matched): Pt = " << jet_matched->pt() << ", eta = " << jet_matched->eta() << ", phi = " << jet_matched->phi() << ":" 
       //      << " sigma(En)/En = " << (jetResolution.get_sigma_e()/jet_matched->energy()) << std::endl;
       //std::cout << "(pfCand: type = " << getPFCandidateType(*pfCandidate) << "," 
       //      << " Pt = " << pfCandidatePtr->pt() << ", eta = " << pfCandidatePtr->eta() << ", phi = " << pfCandidatePtr->phi() << ":"
       //      << " sigma(En)/En = " << (pfCandResolution.get_sigma_e()/pfCandidatePtr->energy()) << ")" << std::endl;
       pfCandInfo.pfMEtSignObj_.set(pfCandResolution.get_type(),
                    pfCandResolution.get_energy(),
                    pfCandResolution.get_phi(),
                    jetResolution.get_sigma_e()*(pfCandidatePtr->energy()/jet_matched->energy()),
                    jetResolution.get_sigma_tan());
     } else {
       pfCandInfo.pfMEtSignObj_ = pfMEtSignInterface_->compResolution(pfCandidatePtr.get());
       //std::cout << " pfCand: type = " << getPFCandidateType(*pfCandidate) << "," 
       //      << " Pt = " << pfCandidatePtr->pt() << ", eta = " << pfCandidatePtr->eta() << ", phi = " << pfCandidatePtr->phi() << ":"
       //      << " sigma(En)/En = " << (pfCandInfo.pfMEtSignObj_.get_sigma_e()/pfCandidatePtr->energy()) << std::endl;
     }
     
     pfCandInfos->push_back(pfCandInfo);
   }
   
   LogDebug ("produce") << "#pfCandInfos = " << pfCandInfos->size() << std::endl;
     
   evt.put(jetInfos);
   evt.put(pfCandInfos);
 }