#include <RecoTauIsolationMasking.h>

Classes
struct	IsoMaskResult

Public Member Functions
IsoMaskResult	mask (const reco::PFTau &) const
	Return a new isolation collections with masking applied. More...

	RecoTauIsolationMasking (const edm::ParameterSet &pset)

void	setMaxSigmas (double maxSigmas)

	~RecoTauIsolationMasking ()

Private Member Functions
bool	inCone (const reco::PFCandidate &track, const reco::Candidate &cand) const

double	resolution (const reco::Candidate &cand) const

Private Attributes
double	ecalCone_

double	finalHcalCone_

double	hcalCone_

double	maxSigmas_

std::unique_ptr< PFEnergyResolution >	resolutions_

Detailed Description

Definition at line 24 of file RecoTauIsolationMasking.h.

Constructor & Destructor Documentation

reco::tau::RecoTauIsolationMasking::RecoTauIsolationMasking ( const edm::ParameterSet & pset )

Definition at line 62 of file RecoTauIsolationMasking.cc.

References ecalCone_, finalHcalCone_, edm::ParameterSet::getParameter(), hcalCone_, and maxSigmas_.

                                 :resolutions_(new PFEnergyResolution) {
   ecalCone_ = pset.getParameter<double>("ecalCone");
   hcalCone_ = pset.getParameter<double>("hcalCone");
   finalHcalCone_ = pset.getParameter<double>("finalHcalCone");
   maxSigmas_ = pset.getParameter<double>("maxSigmas");
 }

reco::tau::RecoTauIsolationMasking::~RecoTauIsolationMasking ( )

Definition at line 71 of file RecoTauIsolationMasking.cc.

71 {}

Member Function Documentation

bool reco::tau::RecoTauIsolationMasking::inCone	(	const reco::PFCandidate &	track,
		const reco::Candidate &	cand
	)		const

private

Definition at line 161 of file RecoTauIsolationMasking.cc.

References funct::abs(), reco::deltaR(), ecalCone_, hcalCone_, reco::Candidate::p4(), reco::Candidate::pdgId(), and reco::PFCandidate::positionAtECALEntrance().

Referenced by mask(), and setMaxSigmas().

                                      {
   double openingDR = reco::deltaR(track.positionAtECALEntrance(), cand.p4());
   if (cand.pdgId() == 130) {
     return (openingDR < hcalCone_);
   } else if (cand.pdgId() == 22 ||
       abs(cand.pdgId()) == 11) {
     return openingDR < ecalCone_;
   } else {
     edm::LogWarning("IsoMask::inCone (bad pf id)")
       << "Unknown PF PDG ID: " << cand.pdgId();
   }
   return -1;
 }

RecoTauIsolationMasking::IsoMaskResult reco::tau::RecoTauIsolationMasking::mask ( const reco::PFTau & tau ) const

Return a new isolation collections with masking applied.

Definition at line 74 of file RecoTauIsolationMasking.cc.

References popcon2dropbox::copy(), reco::Candidate::energy(), finalHcalCone_, reco::tau::RecoTauIsolationMasking::IsoMaskResult::gammas, reco::tau::RecoTauIsolationMasking::IsoMaskResult::h0s, inCone(), reco::PFTau::isolationGammaCands(), reco::PFTau::isolationNeutrHadrCands(), maxSigmas_, convertSQLitetoXML_cfg::output, resolution(), resolutions_, reco::PFTau::signalChargedHadrCands(), reco::PFTau::signalPFChargedHadrCands(), MCScenario_CRAFT1_22X::sorter(), mathSSE::sqrt(), square(), and HiIsolationCommonParameters_cff::track.

                                                         {
   IsoMaskResult output;
 
   typedef std::list<reco::CandidatePtr> CandList;
   // Copy original iso collections.
   std::copy(tau.isolationGammaCands().begin(),
       tau.isolationGammaCands().end(), std::back_inserter(output.gammas));
   std::copy(tau.isolationNeutrHadrCands().begin(),
       tau.isolationNeutrHadrCands().end(),
       std::back_inserter(output.h0s));
 
   std::vector<CandList*> courses;
   courses.push_back(&(output.h0s));
   courses.push_back(&(output.gammas));
   // Mask using each one of the tracks
   for(auto const& track : tau.signalPFChargedHadrCands()) {
     double trackerEnergy = track->energy();
     double linkedEcalEnergy = track->ecalEnergy();
     double linkedHcalEnergy = track->hcalEnergy();
     math::XYZPointF posAtCalo = track->positionAtECALEntrance();
     // Get the linked calo energies & their errors
     double linkedSumErrSquared = 0;
     linkedSumErrSquared += square(
         resolutions_->getEnergyResolutionEm(linkedEcalEnergy, posAtCalo.eta()));
     linkedSumErrSquared += square(
         resolutions_->getEnergyResolutionHad(
           linkedHcalEnergy, posAtCalo.eta(), posAtCalo.phi()));
 
     // energyDelta is the difference between associated Calo and tracker energy
     double energyDelta = linkedEcalEnergy + linkedHcalEnergy - trackerEnergy;
 
     // Sort the neutral hadrons by DR to the track
     //DRSorter sorter(track->p4());
     PtSorter sorter;
 
     for(auto* course : courses) {
       // Sort by deltaR to the current track
       course->sort(sorter);
       CandList::iterator toEatIter = course->begin();
       // While there are still candidates to eat in this course and they are
       // within the cone.
       while (toEatIter != course->end()) {
         const reco::Candidate& toEat = **toEatIter;
         double toEatEnergy = toEat.energy();
         double toEatErrorSq = square(resolution(toEat));
         // Check if we can absorb this candidate into the track.
         if (inCone(*track, **toEatIter) &&
             (energyDelta + toEatEnergy)/std::sqrt(
               linkedSumErrSquared + toEatErrorSq) < maxSigmas_ ) {
           energyDelta += toEatEnergy;
           linkedSumErrSquared += toEatErrorSq;
           toEatIter = course->erase(toEatIter);
         } else {
           // otherwise skip to the next one
           ++toEatIter;
         }
       }
     }
   }
   // Filter out any final HCAL objects with in cones about the tracks.
   // This removes upward fluctuating HCAL objects
   if (finalHcalCone_ > 0) {
     MultiTrackDRFilter hcalFinalFilter(finalHcalCone_,
         tau.signalChargedHadrCands());
     std::remove_if(output.h0s.begin(), output.h0s.end(), hcalFinalFilter);
   }
   return output;
 }

double reco::tau::RecoTauIsolationMasking::resolution ( const reco::Candidate & cand ) const

private

Definition at line 143 of file RecoTauIsolationMasking.cc.

References funct::abs(), reco::Candidate::energy(), reco::Candidate::eta(), reco::Candidate::pdgId(), reco::Candidate::phi(), and resolutions_.

Referenced by mask(), and setMaxSigmas().

                                      {
   if (cand.pdgId() == 130) {
     // NB for HCAL it returns relative energy
     return cand.energy()*resolutions_->getEnergyResolutionHad(cand.energy(),
         cand.eta(), cand.phi());
   } else if (cand.pdgId() == 22) {
     return resolutions_->getEnergyResolutionEm(cand.energy(), cand.eta());
   } else if (std::abs(cand.pdgId()) == 11) {
     // FIXME what is the electron resolution??
     return 0.15;
   } else {
     edm::LogWarning("IsoMask::res (bad pf id)")
       << "Unknown PF PDG ID: " << cand.pdgId();
   }
   return -1;
 }