Inheritance diagram for PhotonIDValueMapProducer:

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

	~PhotonIDValueMapProducer () override

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

bool	hasAbilityToProduceInLumis () const final

bool	hasAbilityToProduceInRuns () const final

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Private Member Functions
reco::PFCandidate::ParticleType	candidatePdgId (const edm::Ptr< reco::Candidate > candidate)

template<class T , class U >
float	computeWorstPFChargedIsolation (const T &photon, const U &pfCands, const edm::Handle< reco::VertexCollection > vertices, const reco::Vertex &pv, unsigned char options)

void	getImpactParameters (const edm::Ptr< reco::Candidate > &candidate, const reco::Vertex &pv, float &dxy, float &dz)

const reco::Track *	getTrackPointer (const edm::Ptr< reco::Candidate > candidate)

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

Private Attributes
MultiTokenT< EcalRecHitCollection >	ebRecHits_

MultiTokenT< EcalRecHitCollection >	eeRecHits_

MultiTokenT< EcalRecHitCollection >	esRecHits_

bool	isAOD_

edm::EDGetToken	particleBasedIsolationToken_

MultiTokenT< edm::View< reco::Candidate > >	pfCandsToken_

MultiTokenT< edm::View< reco::Photon > >	src_

const bool	usesES_

MultiTokenT< reco::VertexCollection >	vtxToken_

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

Definition at line 32 of file PhotonIDValueMapProducer.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 120 of file PhotonIDValueMapProducer.cc.

References mps_fire::i, names, and nVars_.

     : usesES_(!cfg.getParameter<edm::InputTag>("esReducedRecHitCollection").label().empty()
           || !cfg.getParameter<edm::InputTag>("esReducedRecHitCollectionMiniAOD").label().empty())
     , src_(               consumesCollector(), cfg, "src", "srcMiniAOD")
     , ebRecHits_   (src_, consumesCollector(), cfg, "ebReducedRecHitCollection", "ebReducedRecHitCollectionMiniAOD")
     , eeRecHits_   (src_, consumesCollector(), cfg, "eeReducedRecHitCollection", "eeReducedRecHitCollectionMiniAOD")
     , esRecHits_   (src_, consumesCollector(), cfg, "esReducedRecHitCollection", "esReducedRecHitCollectionMiniAOD")
     , vtxToken_    (src_, consumesCollector(), cfg, "vertices", "verticesMiniAOD")
     , pfCandsToken_(src_, consumesCollector(), cfg, "pfCandidates", "pfCandidatesMiniAOD")
     , particleBasedIsolationToken_(mayConsume<edm::ValueMap<std::vector<reco::PFCandidateRef>>>(
           cfg.getParameter<edm::InputTag>("particleBasedIsolation")) /* ...only for AOD... */ )
 {
 
     // Declare producibles
     for (int i = 0; i < nVars_; ++i)
         produces<edm::ValueMap<float>>(names[i]);
 }

PhotonIDValueMapProducer::~PhotonIDValueMapProducer ( )

override

Definition at line 138 of file PhotonIDValueMapProducer.cc.

138 {}

Member Function Documentation

reco::PFCandidate::ParticleType PhotonIDValueMapProducer::candidatePdgId ( const edm::Ptr< reco::Candidate > candidate )

private

Definition at line 389 of file PhotonIDValueMapProducer.cc.

References funct::abs(), reco::PFCandidate::gamma, reco::PFCandidate::h, reco::PFCandidate::h0, isAOD_, common_cff::pdgId, and reco::PFCandidate::X.

Referenced by computeWorstPFChargedIsolation(), and produce().

 {
     if (isAOD_)
         return ((const edm::Ptr<reco::PFCandidate>)candidate)->particleId();
 
     // the neutral hadrons and charged hadrons can be of pdgId types
     // only 130 (K0L) and +-211 (pi+-) in packed candidates
     const int pdgId = ((const edm::Ptr<pat::PackedCandidate>)candidate)->pdgId();
     if (pdgId == 22)
         return reco::PFCandidate::gamma;
     else if (abs(pdgId) == 130) // PDG ID for K0L
         return reco::PFCandidate::h0;
     else if (abs(pdgId) == 211) // PDG ID for pi+-
         return reco::PFCandidate::h;
     else
         return reco::PFCandidate::X;
 }

template<class T , class U >

float PhotonIDValueMapProducer::computeWorstPFChargedIsolation	(	const T &	photon,
		const U &	pfCands,
		const edm::Handle< reco::VertexCollection >	vertices,
		const reco::Vertex &	pv,
		unsigned char	options
	)

private

Definition at line 336 of file PhotonIDValueMapProducer.cc.

References candidatePdgId(), coneSizeDR, reco::deltaR2(), DR_VETO, dRvetoBarrel, dRvetoEndcap, PVValHelper::dxy, dxyMax, PVValHelper::dz, dzMax, getImpactParameters(), reco::PFCandidate::h, mps_fire::i, SiStripPI::max, PT_MIN_THRESH, ptMin, PV_CONSTRAINT, and badGlobalMuonTaggersAOD_cff::vtx.

Referenced by produce().

 {
     float worstIsolation = 0.0;
 
     const float dRveto = photon->isEB() ? dRvetoBarrel : dRvetoEndcap;
 
     // Calculate isolation sum separately for each vertex
     for (unsigned int ivtx = 0; ivtx < vertices->size(); ++ivtx) {
 
         // Shift the photon according to the vertex
         reco::VertexRef vtx(vertices, ivtx);
         math::XYZVector phoWrtVtx(photon->superCluster()->x() - vtx->x(),
             photon->superCluster()->y() - vtx->y(), photon->superCluster()->z() - vtx->z());
 
         float sum = 0;
         // Loop over the PFCandidates
         for (unsigned int i = 0; i < pfCands->size(); i++) {
 
             const auto& iCand = pfCands->ptrAt(i);
 
             // require that PFCandidate is a charged hadron
             reco::PFCandidate::ParticleType thisCandidateType = candidatePdgId(iCand);
             if (thisCandidateType != reco::PFCandidate::h)
                 continue;
 
             if ((options & PT_MIN_THRESH) && iCand->pt() < ptMin)
                 continue;
 
             float dxy = -999;
             float dz = -999;
             if (options & PV_CONSTRAINT)
                 getImpactParameters(iCand, pv, dxy, dz);
             else
                 getImpactParameters(iCand, *vtx, dxy, dz);
 
             if (fabs(dxy) > dxyMax || fabs(dz) > dzMax)
                 continue;
 
             float dR2 = deltaR2(phoWrtVtx.Eta(), phoWrtVtx.Phi(), iCand->eta(), iCand->phi());
             if (dR2 > coneSizeDR * coneSizeDR ||
                     (options & DR_VETO && dR2 < dRveto * dRveto))
                 continue;
 
             sum += iCand->pt();
         }
 
         worstIsolation = std::max(sum, worstIsolation);
     }
 
     return worstIsolation;
 }

void PhotonIDValueMapProducer::fillDescriptions ( edm::ConfigurationDescriptions & descriptions )

static

Definition at line 313 of file PhotonIDValueMapProducer.cc.

References edm::ConfigurationDescriptions::add(), and edm::ParameterSetDescription::add().

                                                                                           {
   // photonIDValueMapProducer
   edm::ParameterSetDescription desc;
   desc.add<edm::InputTag>("particleBasedIsolation",           edm::InputTag("particleBasedIsolation","gedPhotons"));
   desc.add<edm::InputTag>("src",                              edm::InputTag("gedPhotons"));
   desc.add<edm::InputTag>("srcMiniAOD",                       edm::InputTag("slimmedPhotons","","@skipCurrentProcess"));
   desc.add<edm::InputTag>("esReducedRecHitCollectionMiniAOD", edm::InputTag("reducedEgamma","reducedESRecHits"));
   desc.add<edm::InputTag>("eeReducedRecHitCollection",        edm::InputTag("reducedEcalRecHitsEE"));
   desc.add<edm::InputTag>("pfCandidates",                     edm::InputTag("particleFlow"));
   desc.add<edm::InputTag>("vertices",                         edm::InputTag("offlinePrimaryVertices"));
   desc.add<edm::InputTag>("ebReducedRecHitCollectionMiniAOD", edm::InputTag("reducedEgamma","reducedEBRecHits"));
   desc.add<edm::InputTag>("eeReducedRecHitCollectionMiniAOD", edm::InputTag("reducedEgamma","reducedEERecHits"));
   desc.add<edm::InputTag>("esReducedRecHitCollection",        edm::InputTag("reducedEcalRecHitsES"));
   desc.add<edm::InputTag>("pfCandidatesMiniAOD",              edm::InputTag("packedPFCandidates"));
   desc.add<edm::InputTag>("verticesMiniAOD",                  edm::InputTag("offlineSlimmedPrimaryVertices"));
   desc.add<edm::InputTag>("ebReducedRecHitCollection",        edm::InputTag("reducedEcalRecHitsEB"));
   descriptions.add("photonIDValueMapProducer", desc);
 }

void PhotonIDValueMapProducer::getImpactParameters	(	const edm::Ptr< reco::Candidate > &	candidate,
		const reco::Vertex &	pv,
		float &	dxy,
		float &	dz
	)

private

Definition at line 415 of file PhotonIDValueMapProducer.cc.

References DEFINE_FWK_MODULE, isAOD_, and reco::Vertex::position().

Referenced by computeWorstPFChargedIsolation(), and produce().

 {
     if (isAOD_) {
         auto const& theTrack = *static_cast<const edm::Ptr<reco::PFCandidate>>(candidate)->trackRef();
         dxy = theTrack.dxy(pv.position());
         dz = theTrack.dz(pv.position());
     } else {
         auto const& aCand = *static_cast<const edm::Ptr<pat::PackedCandidate>>(candidate);
         dxy = aCand.dxy(pv.position());
         dz = aCand.dz(pv.position());
     }
 }

const reco::Track * PhotonIDValueMapProducer::getTrackPointer ( const edm::Ptr< reco::Candidate > candidate )

private

Definition at line 408 of file PhotonIDValueMapProducer.cc.

References isAOD_.

 {
     return isAOD_ ?
         &*(((const edm::Ptr<reco::PFCandidate>)candidate)->trackRef()) :
         &(((const edm::Ptr<pat::PackedCandidate>)candidate)->pseudoTrack());
 }

void PhotonIDValueMapProducer::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivate

Definition at line 140 of file PhotonIDValueMapProducer.cc.

References candidatePdgId(), computeWorstPFChargedIsolation(), coneSizeDR, reco::deltaR2(), DR_VETO, PVValHelper::dxy, dxyMax, PVValHelper::dz, dzMax, ebRecHits_, eeRecHits_, esRecHits_, Exception, f, reco::PFCandidate::gamma, edm::Event::getByToken(), MultiTokenT< T >::getGoodTokenIndex(), getImpactParameters(), MultiTokenT< T >::getValidHandle(), reco::PFCandidate::h, reco::PFCandidate::h0, mps_fire::i, iEvent, isAOD_, isInFootprint(), edm::detail::isnan(), names, nVars_, particleBasedIsolationToken_, pfCandsToken_, PT_MIN_THRESH, MetAnalyzer::pv(), PV_CONSTRAINT, mathSSE::sqrt(), TrackRefitter_38T_cff::src, src_, PFTauMVAInputDiscriminatorTranslator_cfi::test, usesES_, electrons_cff::vertices, vtxToken_, writeValueMap(), reco::Vertex::x(), reco::Vertex::y(), and reco::Vertex::z().

 {
     // Get the handles
     auto src           = src_.getValidHandle(iEvent);
     auto vertices      = vtxToken_.getValidHandle(iEvent);
     auto pfCandsHandle = pfCandsToken_.getValidHandle(iEvent);
 
     isAOD_ = src_.getGoodTokenIndex() == 0;
     edm::Handle<edm::ValueMap<std::vector<reco::PFCandidateRef>>> particleBasedIsolationMap;
     if (isAOD_) { // this exists only in AOD
         iEvent.getByToken(particleBasedIsolationToken_, particleBasedIsolationMap);
     }
     if (!isAOD_ && !src->empty()) {
         edm::Ptr<pat::Photon> test(src->ptrAt(0));
         if (test.isNull() || !test.isAvailable()) {
             throw cms::Exception("InvalidConfiguration")
                 << "DataFormat is detected as miniAOD but cannot cast to pat::Photon!";
         }
     }
 
     // Configure Lazy Tools, which will compute 5x5 quantities
     std::unique_ptr<noZS::EcalClusterLazyTools> lazyToolnoZS;
 
     if (usesES_) {
         lazyToolnoZS = std::make_unique<noZS::EcalClusterLazyTools>(
             iEvent, iSetup, ebRecHits_.get(iEvent), eeRecHits_.get(iEvent), esRecHits_.get(iEvent));
     } else {
         lazyToolnoZS = std::make_unique<noZS::EcalClusterLazyTools>(
             iEvent, iSetup, ebRecHits_.get(iEvent), eeRecHits_.get(iEvent));
     }
 
     // Get PV
     if (vertices->empty())
         return; // skip the event if no PV found
     const reco::Vertex& pv = vertices->front();
 
     std::vector<float> vars[nVars_];
 
     // reco::Photon::superCluster() is virtual so we can exploit polymorphism
     for (unsigned i = 0; i < src->size(); ++i) {
         const auto& iPho = src->ptrAt(i);
 
         //
         // Compute full 5x5 quantities
         //
         const auto& theseed = *(iPho->superCluster()->seed());
 
         // For full5x5_sigmaIetaIeta, for 720 we use: lazy tools for AOD,
         // and userFloats or lazy tools for miniAOD. From some point in 72X and on, one can
         // retrieve the full5x5 directly from the object with ->full5x5_sigmaIetaIeta()
         // for both formats.
         std::vector<float> vCov = lazyToolnoZS->localCovariances(theseed);
         vars[0].push_back(isnan(vCov[0]) ? 0. : sqrt(vCov[0]));
         vars[1].push_back(vCov[1]);
         vars[2].push_back(lazyToolnoZS->e1x3(theseed));
         vars[3].push_back(lazyToolnoZS->e2x2(theseed));
         vars[4].push_back(lazyToolnoZS->e2x5Max(theseed));
         vars[5].push_back(lazyToolnoZS->e5x5(theseed));
         vars[6].push_back(lazyToolnoZS->eseffsirir(*(iPho->superCluster())));
         vars[7].push_back(vars[2][i] / vars[5][i]);
         vars[8].push_back(vars[3][i] / vars[5][i]);
         vars[9].push_back(vars[4][i] / vars[5][i]);
 
         //
         // Compute absolute uncorrected isolations with footprint removal
         //
 
         // First, find photon direction with respect to the good PV
         math::XYZVector phoWrtVtx(
             iPho->superCluster()->x() - pv.x(), iPho->superCluster()->y() - pv.y(), iPho->superCluster()->z() - pv.z());
 
         // isolation sums
         float chargedIsoSum = 0.;
         float neutralHadronIsoSum = 0.;
         float photonIsoSum = 0.;
 
         // Loop over all PF candidates
         for (unsigned int idxcand = 0; idxcand < pfCandsHandle->size(); ++idxcand) {
 
             // Here, the type will be a simple reco::Candidate. We cast it
             // for full PFCandidate or PackedCandidate below as necessary
             const auto& iCand = pfCandsHandle->ptrAt(idxcand);
 
             // One would think that we should check that this iCand from the
             // generic PF collection is not identical to the iPho photon for
             // which we are computing the isolations. However, it turns out to
             // be unnecessary. Below, in the function isInFootprint(), we drop
             // this iCand if it is in the footprint, and this always removes
             // the iCand if it matches the iPho. The explicit check at this
             // point is not totally trivial because of non-triviality of
             // implementation of this check for miniAOD (PackedCandidates of
             // the PF collection do not contain the supercluser link, so can't
             // use that).
             //
             // if( isAOD_ ) {
             //     if( ((const edm::Ptr<reco::PFCandidate>)iCand)->superClusterRef() == iPho->superCluster() )
             //     continue;
             // }
 
             // Check if this candidate is within the isolation cone
             float dR2 = deltaR2(phoWrtVtx.Eta(), phoWrtVtx.Phi(), iCand->eta(), iCand->phi());
             if (dR2 > coneSizeDR * coneSizeDR)
                 continue;
 
             // Check if this candidate is not in the footprint
             if (isAOD_) {
                 if(isInFootprint((*particleBasedIsolationMap)[iPho], iCand))
                     continue;
             } else {
                 edm::Ptr<pat::Photon> patPhotonPtr(src->ptrAt(i));
                 if(isInFootprint(patPhotonPtr->associatedPackedPFCandidates(), iCand))
                     continue;
             }
 
             // Find candidate type
             reco::PFCandidate::ParticleType thisCandidateType = candidatePdgId(iCand);
 
             // Increment the appropriate isolation sum
             if (thisCandidateType == reco::PFCandidate::h) {
                 // for charged hadrons, additionally check consistency
                 // with the PV
                 float dxy = -999;
                 float dz = -999;
                 getImpactParameters(iCand, pv, dxy, dz);
 
                 if (fabs(dxy) > dxyMax || fabs(dz) > dzMax)
                     continue;
 
                 // The candidate is eligible, increment the isolaiton
                 chargedIsoSum += iCand->pt();
             }
 
             if (thisCandidateType == reco::PFCandidate::h0)
                 neutralHadronIsoSum += iCand->pt();
 
             if (thisCandidateType == reco::PFCandidate::gamma)
                 photonIsoSum += iCand->pt();
         }
 
         vars[10].push_back(chargedIsoSum);
         vars[11].push_back(neutralHadronIsoSum);
         vars[12].push_back(photonIsoSum);
 
         // Worst isolation computed with no vetos or ptMin cut, as in Run 1 Hgg code.
         unsigned char options = 0;
         vars[13].push_back(computeWorstPFChargedIsolation(iPho, pfCandsHandle, vertices, pv, options));
 
         // Worst isolation computed with cone vetos and a ptMin cut, as in Run 2 Hgg code.
         options |= PT_MIN_THRESH | DR_VETO;
         vars[14].push_back(computeWorstPFChargedIsolation(iPho, pfCandsHandle, vertices, pv, options));
 
         // Like before, but adding primary vertex constraint
         options |= PV_CONSTRAINT;
         vars[15].push_back(computeWorstPFChargedIsolation(iPho, pfCandsHandle, vertices, pv, options));
 
         // PFCluster Isolations
         vars[16].push_back(iPho->trkSumPtSolidConeDR04());
         if (isAOD_) {
             vars[17].push_back(0.f);
             vars[18].push_back(0.f);
         } else {
             edm::Ptr<pat::Photon> patPhotonPtr{ src->ptrAt(i) };
             vars[17].push_back(patPhotonPtr->hcalPFClusterIso());
             vars[18].push_back(patPhotonPtr->ecalPFClusterIso());
         }
     }
 
     // write the value maps
     for (int i = 0; i < nVars_; ++i) {
         writeValueMap(iEvent, src, vars[i], names[i]);
     }
 }