Inheritance diagram for reco::tau::RecoTauEnergyRecoveryPlugin:

Public Member Functions
virtual void	beginEvent ()

void	operator() (PFTau &) const

	RecoTauEnergyRecoveryPlugin (const edm::ParameterSet &)

virtual	~RecoTauEnergyRecoveryPlugin ()

Public Member Functions inherited from reco::tau::RecoTauModifierPlugin
	RecoTauModifierPlugin (const edm::ParameterSet &pset)

virtual	~RecoTauModifierPlugin ()

Public Member Functions inherited from reco::tau::RecoTauEventHolderPlugin
const edm::Event *	evt () const

const edm::EventSetup *	evtSetup () const

	RecoTauEventHolderPlugin (const edm::ParameterSet &pset)

void	setup (const edm::Event &, const edm::EventSetup &)

virtual	~RecoTauEventHolderPlugin ()

Public Member Functions inherited from reco::tau::RecoTauNamedPlugin
const std::string &	name () const

	RecoTauNamedPlugin (const edm::ParameterSet &pset)

virtual	~RecoTauNamedPlugin ()

Private Attributes
unsigned	corrLevel_

double	lev1EtaWindow_

double	lev1PhiWindow_

RecoTauQualityCuts *	qcuts_

edm::InputTag	srcVertices_

const reco::Vertex *	theEventVertex_

Detailed Description

Definition at line 31 of file RecoTauEnergyRecoveryPlugin.cc.

Constructor & Destructor Documentation

reco::tau::RecoTauEnergyRecoveryPlugin::RecoTauEnergyRecoveryPlugin ( const edm::ParameterSet & cfg )

explicit

Definition at line 53 of file RecoTauEnergyRecoveryPlugin.cc.

References edm::ParameterSet::getParameter(), qcuts_, and srcVertices_.

   : RecoTauModifierPlugin(cfg),
     theEventVertex_(0),
     corrLevel_(cfg.getParameter<unsigned>("corrLevel")),
     lev1PhiWindow_(cfg.getParameter<double>("lev1PhiWindow")),
     lev1EtaWindow_(cfg.getParameter<double>("lev1EtaWindow"))
 {
   edm::ParameterSet cfgQualityCuts = cfg.getParameter<edm::ParameterSet>("qualityCuts");
   srcVertices_ = cfgQualityCuts.getParameter<edm::InputTag>("primaryVertexSrc");
   qcuts_ = new RecoTauQualityCuts(cfgQualityCuts.getParameter<edm::ParameterSet>("signalQualityCuts"));
 }

reco::tau::RecoTauEnergyRecoveryPlugin::~RecoTauEnergyRecoveryPlugin ( )

virtual

Definition at line 65 of file RecoTauEnergyRecoveryPlugin.cc.

References qcuts_.

 {
   delete qcuts_;
 }

Member Function Documentation

void reco::tau::RecoTauEnergyRecoveryPlugin::beginEvent ( )

virtual

Reimplemented from reco::tau::RecoTauModifierPlugin.

Definition at line 70 of file RecoTauEnergyRecoveryPlugin.cc.

References reco::tau::RecoTauEventHolderPlugin::evt(), edm::Event::getByLabel(), qcuts_, reco::tau::RecoTauQualityCuts::setPV(), srcVertices_, and theEventVertex_.

 {
   edm::Handle<reco::VertexCollection> vertices;
   evt()->getByLabel(srcVertices_, vertices);
 
   if ( vertices->size() >= 1 ) {
     qcuts_->setPV(reco::VertexRef(vertices, 0));
     theEventVertex_ = &vertices->at(0);
   } else {
     theEventVertex_ = 0;
   }
 }

void reco::tau::RecoTauEnergyRecoveryPlugin::operator() ( PFTau & tau ) const

virtual

Implements reco::tau::RecoTauModifierPlugin.

Definition at line 108 of file RecoTauEnergyRecoveryPlugin.cc.

References corrLevel_, reco::TrackBase::dxy(), reco::TrackBase::dz(), reco::LeafCandidate::energy(), reco::LeafCandidate::eta(), reco::tau::RecoTauQualityCuts::filter(), reco::tau::getTrack(), reco::PFCandidate::h0, reco::TrackBase::hitPattern(), edm::RefToBase< T >::isNonnull(), reco::tau::isTauSignalPFCandidate(), reco::PFTau::jetRef(), lev1EtaWindow_, lev1PhiWindow_, max(), reco::TrackBase::normalizedChi2(), reco::HitPattern::numberOfValidPixelHits(), reco::LeafCandidate::p(), reco::TrackBase::p(), reco::LeafCandidate::phi(), reco::Vertex::position(), reco::TrackBase::pt(), reco::TrackBase::ptError(), reco::LeafCandidate::px(), reco::LeafCandidate::py(), reco::LeafCandidate::pz(), qcuts_, reco::BaseTau::setalternatLorentzVect(), mathSSE::sqrt(), reco::tau::square(), and theEventVertex_.

 {
   double tauEnergyCorr = 0.;
 
   if ( corrLevel_ & 1 && theEventVertex_ ) {
 
     bool needsCorrLevel1 = false;
 
     std::vector<reco::PFCandidatePtr> pfJetConstituents = tau.jetRef()->getPFConstituents();
     for ( std::vector<reco::PFCandidatePtr>::const_iterator pfJetConstituent = pfJetConstituents.begin();
       pfJetConstituent != pfJetConstituents.end(); ++pfJetConstituent ) {
       const reco::TrackBaseRef track = getTrack(**pfJetConstituent);
       if ( track.isNonnull() ) {
     double dIP = fabs(track->dxy(theEventVertex_->position()));
     double dZ = fabs(track->dz(theEventVertex_->position()));
       if ( track->pt() > 2.0 && fabs(tau.eta() - (*pfJetConstituent)->eta()) < lev1EtaWindow_ &&
        !isTauSignalPFCandidate(tau, *pfJetConstituent) && (dZ < 0.2 || dIP > 0.10) ) needsCorrLevel1 = true;
       }
     }
 
     if ( needsCorrLevel1 ) {
       std::vector<reco::PFCandidatePtr> pfJetConstituents = tau.jetRef()->getPFConstituents();
       for ( std::vector<reco::PFCandidatePtr>::const_iterator pfJetConstituent = pfJetConstituents.begin();
         pfJetConstituent != pfJetConstituents.end(); ++pfJetConstituent ) {
     if ( fabs(tau.eta() - (*pfJetConstituent)->eta()) < lev1EtaWindow_ &&
          fabs(tau.phi() - (*pfJetConstituent)->phi()) < lev1PhiWindow_ ) {
       if ( (*pfJetConstituent)->particleId() == reco::PFCandidate::h0 ) {
         tauEnergyCorr += (*pfJetConstituent)->energy();
       } else {
         if ( !isTauSignalPFCandidate(tau, *pfJetConstituent) ) {
           double caloEn = (*pfJetConstituent)->ecalEnergy() + (*pfJetConstituent)->hcalEnergy();
           tauEnergyCorr += caloEn;
         }
       }
     }
       }
     }
   }
 
   if ( corrLevel_ & 2 && theEventVertex_ ) {
 
     double leadTrackMom = 0.;
     double leadTrackMomErr = 0.;
     double jetCaloEn = 0.;
 
     std::vector<reco::PFCandidatePtr> pfJetConstituents = tau.jetRef()->getPFConstituents();
     for ( std::vector<reco::PFCandidatePtr>::const_iterator pfJetConstituent = pfJetConstituents.begin();
       pfJetConstituent != pfJetConstituents.end(); ++pfJetConstituent ) {
       const reco::TrackBaseRef track = getTrack(**pfJetConstituent);
       if ( track.isNonnull() ) {
     double trackPt = track->pt();
     double trackPtErr = track->ptError();
     if ( qcuts_->filter(**pfJetConstituent) &&
          trackPtErr < (0.20*trackPt) && track->normalizedChi2() < 5.0 && track->hitPattern().numberOfValidPixelHits() >= 1 &&
          (trackPt - 3.*trackPtErr) > (*pfJetConstituent)->pt() && trackPt < (3.*tau.jetRef()->pt()) ) {
       if ( track->p() > leadTrackMom ) {
         leadTrackMom = track->p();
         leadTrackMomErr = leadTrackMom*(trackPtErr/trackPt);
       }
     }
       }
 
       double caloEn = (*pfJetConstituent)->ecalEnergy() + (*pfJetConstituent)->hcalEnergy();
       jetCaloEn += caloEn;
     }
 
     if ( leadTrackMom > tau.p() ) {
       const double chargedPionMass = 0.13957; // GeV
       double leadTrackEn = sqrt(square(leadTrackMom) + square(chargedPionMass));
       double jetCaloEnErr = 1.00*sqrt(std::max(jetCaloEn, leadTrackEn));
       double combEn = ((1./square(jetCaloEnErr))*jetCaloEn + (1./square(leadTrackMomErr))*leadTrackEn)/
                       ((1./square(jetCaloEnErr)) + (1./square(leadTrackMomErr)));
       tauEnergyCorr = std::max(tauEnergyCorr, combEn - tau.energy());
     }
   }
 
   if ( tau.energy() > 0. ) {
     double tauEnergy_corrected = tau.energy() + tauEnergyCorr;
     double scaleFactor = tauEnergy_corrected/tau.energy();
     double tauPx_corrected = scaleFactor*tau.px();
     double tauPy_corrected = scaleFactor*tau.py();
     double tauPz_corrected = scaleFactor*tau.pz();
     tau.setalternatLorentzVect(reco::Candidate::LorentzVector(tauPx_corrected, tauPy_corrected, tauPz_corrected, tauEnergy_corrected));
   }
 }