PFTau.h

Go to the documentation of this file.

#ifndef DataFormats_TauReco_PFTau_h
#define DataFormats_TauReco_PFTau_h

/* class PFTau
 * the object of this class is created by RecoTauTag/RecoTau PFRecoTauProducer EDProducer starting from the PFTauTagInfo object,
 *                          is a hadronic tau-jet candidate -built from a jet made employing a particle flow technique- that analysts manipulate;
 * authors: Simone Gennai (simone.gennai@cern.ch), Ludovic Houchu (Ludovic.Houchu@cern.ch), Evan Friis (evan.klose.friis@ucdavis.edu)
 * created: Jun 21 2007,
 * revised: Tue Aug 31 13:34:40 CEST 2010
 */
#include "DataFormats/Math/interface/LorentzVector.h"
#include "DataFormats/Common/interface/AtomicPtrCache.h"
#include "DataFormats/TauReco/interface/BaseTau.h"
#include "DataFormats/TauReco/interface/PFTauFwd.h"
#include "DataFormats/TauReco/interface/PFTauTagInfo.h"
#include "DataFormats/JetReco/interface/PFJetCollection.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/TauReco/interface/RecoTauPiZero.h"
#include "DataFormats/TauReco/interface/RecoTauPiZeroFwd.h"
#include "DataFormats/TauReco/interface/PFRecoTauChargedHadron.h"
#include "DataFormats/TauReco/interface/PFRecoTauChargedHadronFwd.h"

#include <iostream>
#include <limits>

namespace reco { namespace tau {
  class RecoTauConstructor;
  class PFRecoTauEnergyAlgorithmPlugin;
}} 

namespace reco {

class PFTau : public BaseTau {
  public:
    enum hadronicDecayMode {
      kNull = -1,
      kOneProng0PiZero,
      kOneProng1PiZero,
      kOneProng2PiZero,
      kOneProng3PiZero,
      kOneProngNPiZero,
      kTwoProng0PiZero,
      kTwoProng1PiZero,
      kTwoProng2PiZero,
      kTwoProng3PiZero,
      kTwoProngNPiZero,
      kThreeProng0PiZero,
      kThreeProng1PiZero,
      kThreeProng2PiZero,
      kThreeProng3PiZero,
      kThreeProngNPiZero,
      kRareDecayMode
    };

    PFTau();
    PFTau(Charge q,const LorentzVector &,const Point & = Point( 0, 0, 0 ) );
    ~PFTau() override {};
    PFTau* clone() const override;

    const PFJetRef& jetRef() const;
    void setjetRef(const PFJetRef&);

    // functions to access the PFTauTagInfoRef used by HLT
    const PFTauTagInfoRef& pfTauTagInfoRef() const;
    void setpfTauTagInfoRef(const PFTauTagInfoRef);

    PFRecoTauChargedHadronRef leadTauChargedHadronCandidate() const;
    const PFCandidatePtr& leadPFChargedHadrCand() const;
    const PFCandidatePtr& leadPFNeutralCand() const;
    //Can be either the charged or the neutral one
    const PFCandidatePtr& leadPFCand() const;

    void setleadPFChargedHadrCand(const PFCandidatePtr&);
    void setleadPFNeutralCand(const PFCandidatePtr&);
    void setleadPFCand(const PFCandidatePtr&);

    float leadPFChargedHadrCandsignedSipt() const;
    void setleadPFChargedHadrCandsignedSipt(const float&);

    const std::vector<reco::PFCandidatePtr>& signalPFCands() const;
    void setsignalPFCands(const std::vector<reco::PFCandidatePtr>&);

    const std::vector<reco::PFCandidatePtr>& signalPFChargedHadrCands() const;
    void setsignalPFChargedHadrCands(const std::vector<reco::PFCandidatePtr>&);

    const std::vector<reco::PFCandidatePtr>& signalPFNeutrHadrCands() const;
    void setsignalPFNeutrHadrCands(const std::vector<reco::PFCandidatePtr>&);

    const std::vector<reco::PFCandidatePtr>& signalPFGammaCands() const;
    void setsignalPFGammaCands(const std::vector<reco::PFCandidatePtr>&);

    const std::vector<reco::PFCandidatePtr>& isolationPFCands() const;
    void setisolationPFCands(const std::vector<reco::PFCandidatePtr>&);

    const std::vector<reco::PFCandidatePtr>& isolationPFChargedHadrCands() const;
    void setisolationPFChargedHadrCands(const std::vector<reco::PFCandidatePtr>&);

    const std::vector<reco::PFCandidatePtr>& isolationPFNeutrHadrCands() const;
    void setisolationPFNeutrHadrCands(const std::vector<reco::PFCandidatePtr>&);

    const std::vector<reco::PFCandidatePtr>& isolationPFGammaCands() const;
    void setisolationPFGammaCands(const std::vector<reco::PFCandidatePtr>&);

    float isolationPFChargedHadrCandsPtSum() const;
    void setisolationPFChargedHadrCandsPtSum(const float&);

    float isolationPFGammaCandsEtSum() const;
    void setisolationPFGammaCandsEtSum(const float&);

    float maximumHCALPFClusterEt() const;
    void setmaximumHCALPFClusterEt(const float&);

    const std::vector<RecoTauPiZero>& signalPiZeroCandidates() const;
    void setsignalPiZeroCandidates(std::vector<RecoTauPiZero>);
    void setSignalPiZeroCandidatesRefs(RecoTauPiZeroRefVector);

    const std::vector<RecoTauPiZero>& isolationPiZeroCandidates() const;
    void setisolationPiZeroCandidates(std::vector<RecoTauPiZero>);
    void setIsolationPiZeroCandidatesRefs(RecoTauPiZeroRefVector);

    const std::vector<PFRecoTauChargedHadron>& signalTauChargedHadronCandidates() const;
    void setSignalTauChargedHadronCandidates(std::vector<PFRecoTauChargedHadron>);
    void setSignalTauChargedHadronCandidatesRefs(PFRecoTauChargedHadronRefVector);

    const std::vector<PFRecoTauChargedHadron>& isolationTauChargedHadronCandidates() const;
    void setIsolationTauChargedHadronCandidates(std::vector<PFRecoTauChargedHadron>);
    void setIsolationTauChargedHadronCandidatesRefs(PFRecoTauChargedHadronRefVector);

    hadronicDecayMode decayMode() const;
    void setDecayMode(const hadronicDecayMode&);

    float bendCorrMass() const { return bendCorrMass_; }
    void setBendCorrMass(float bendCorrMass) { bendCorrMass_ = bendCorrMass; }

    double signalConeSize() const { return signalConeSize_; }
    void setSignalConeSize(double signalConeSize) { signalConeSize_ = signalConeSize; }  

    //Electron rejection
    float emFraction() const; // Ecal/Hcal Cluster Energy
    float hcalTotOverPLead() const; // total Hcal Cluster E / leadPFChargedHadron P
    float hcalMaxOverPLead() const; // max. Hcal Cluster E / leadPFChargedHadron P
    float hcal3x3OverPLead() const; // Hcal Cluster E in R<0.184 around Ecal impact point of leading track / leadPFChargedHadron P
    float ecalStripSumEOverPLead() const; // Simple BremsRecovery Sum E / leadPFChargedHadron P
    float bremsRecoveryEOverPLead() const; // BremsRecovery Sum E / leadPFChargedHadron P
    reco::TrackRef electronPreIDTrack() const; // Ref to KF track from Electron PreID
    float electronPreIDOutput() const; // BDT output from Electron PreID
    bool electronPreIDDecision() const; // Decision from Electron PreID

    void setemFraction(const float&);
    void sethcalTotOverPLead(const float&);
    void sethcalMaxOverPLead(const float&);
    void sethcal3x3OverPLead(const float&);
    void setecalStripSumEOverPLead(const float&);
    void setbremsRecoveryEOverPLead(const float&);
    void setelectronPreIDTrack(const reco::TrackRef&);
    void setelectronPreIDOutput(const float&);
    void setelectronPreIDDecision(const bool&);

    // For Muon Rejection
    bool hasMuonReference() const; // check if muon ref exists
    float caloComp() const;
    float segComp() const;
    bool muonDecision() const;
    void setCaloComp(const float&);
    void setSegComp(const float&);
    void setMuonDecision(const bool&);

    size_type numberOfSourceCandidatePtrs() const override {return 1;}

    CandidatePtr sourceCandidatePtr( size_type i ) const override;

    void dump(std::ostream& out = std::cout) const;

  private:
    friend class tau::RecoTauConstructor;
    friend class tau::PFRecoTauEnergyAlgorithmPlugin;

    //These are used by the friends
    std::vector<RecoTauPiZero>& signalPiZeroCandidatesRestricted();
    std::vector<RecoTauPiZero>& isolationPiZeroCandidatesRestricted();
    std::vector<PFRecoTauChargedHadron>& signalTauChargedHadronCandidatesRestricted();
    std::vector<PFRecoTauChargedHadron>& isolationTauChargedHadronCandidatesRestricted();

    // check overlap with another candidate
    bool overlap(const Candidate&) const override;

    bool muonDecision_;
    bool electronPreIDDecision_;

    // SIP
    float leadPFChargedHadrCandsignedSipt_;
    // Isolation variables
    float isolationPFChargedHadrCandsPtSum_;
    float isolationPFGammaCandsEtSum_;
    float maximumHCALPFClusterEt_;

    // Electron rejection variables
    float emFraction_;
    float hcalTotOverPLead_;
    float hcalMaxOverPLead_;
    float hcal3x3OverPLead_;
    float ecalStripSumEOverPLead_;
    float bremsRecoveryEOverPLead_;
    float electronPreIDOutput_;

    // Muon rejection variables
    float caloComp_;
    float segComp_;

    hadronicDecayMode decayMode_;

    float bendCorrMass_;

    float signalConeSize_; 

    reco::PFJetRef jetRef_;
    PFTauTagInfoRef PFTauTagInfoRef_;
    reco::PFCandidatePtr leadPFChargedHadrCand_;
    reco::PFCandidatePtr leadPFNeutralCand_;
    reco::PFCandidatePtr leadPFCand_;
    reco::TrackRef electronPreIDTrack_;

    // Signal candidates
    std::vector<reco::PFCandidatePtr> selectedSignalPFCands_;
    std::vector<reco::PFCandidatePtr> selectedSignalPFChargedHadrCands_;
    std::vector<reco::PFCandidatePtr> selectedSignalPFNeutrHadrCands_;
    std::vector<reco::PFCandidatePtr> selectedSignalPFGammaCands_;

    // Isolation candidates
    std::vector<reco::PFCandidatePtr> selectedIsolationPFCands_;
    std::vector<reco::PFCandidatePtr> selectedIsolationPFChargedHadrCands_;
    std::vector<reco::PFCandidatePtr> selectedIsolationPFNeutrHadrCands_;
    std::vector<reco::PFCandidatePtr> selectedIsolationPFGammaCands_;

    RecoTauPiZeroRefVector signalPiZeroCandidatesRefs_;
    RecoTauPiZeroRefVector isolationPiZeroCandidatesRefs_;

    PFRecoTauChargedHadronRefVector signalTauChargedHadronCandidatesRefs_;
    PFRecoTauChargedHadronRefVector isolationTauChargedHadronCandidatesRefs_;

    // Association of gamma candidates into PiZeros (transient)
    edm::AtomicPtrCache<std::vector<reco::RecoTauPiZero>> signalPiZeroCandidates_;
    edm::AtomicPtrCache<std::vector<reco::RecoTauPiZero>> isolationPiZeroCandidates_;

    // Association of PF candidates into PFRecoTauChargedHadrons (transient)
    edm::AtomicPtrCache<std::vector<reco::PFRecoTauChargedHadron>> signalTauChargedHadronCandidates_;
    edm::AtomicPtrCache<std::vector<reco::PFRecoTauChargedHadron>> isolationTauChargedHadronCandidates_;
};

std::ostream & operator<<(std::ostream& out, const PFTau& c);

} // end namespace reco

#endif