Tau.h

Go to the documentation of this file.

//
// $Id: Tau.h,v 1.34 2012/04/25 07:32:20 cbern Exp $
//

#ifndef DataFormats_PatCandidates_Tau_h
#define DataFormats_PatCandidates_Tau_h

#include "DataFormats/TauReco/interface/BaseTau.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/PatCandidates/interface/Lepton.h"
#include "DataFormats/JetReco/interface/GenJetCollection.h"
#include "DataFormats/Candidate/interface/Candidate.h"

#include "DataFormats/PatCandidates/interface/TauPFSpecific.h"
#include "DataFormats/PatCandidates/interface/TauCaloSpecific.h"
#include "DataFormats/PatCandidates/interface/TauJetCorrFactors.h"

// Define typedefs for convenience
namespace pat {
  class Tau;
  typedef std::vector<Tau>              TauCollection; 
  typedef edm::Ref<TauCollection>       TauRef; 
  typedef edm::RefVector<TauCollection> TauRefVector; 
}

namespace reco {
  std::ostream& operator<<(std::ostream& out, const pat::Tau& obj);
}

// Class definition
namespace pat {

  class Tau : public Lepton<reco::BaseTau> {
    friend class PATTauProducer;

    public:

      typedef std::pair<std::string,float> IdPair;

      Tau();
      Tau(const reco::BaseTau & aTau);
      Tau(const edm::RefToBase<reco::BaseTau> & aTauRef);
      Tau(const edm::Ptr<reco::BaseTau> & aTauRef);
      virtual ~Tau();

      virtual Tau * clone() const { return new Tau(*this); }

      // ---- methods for content embedding ----
      const reco::TrackRefVector & isolationTracks() const;
      reco::TrackRef leadTrack() const;
      const reco::TrackRefVector & signalTracks() const;    
      void embedIsolationTracks();
      void embedLeadTrack();
      void embedSignalTracks();
      void embedLeadPFCand();
      void embedLeadPFChargedHadrCand();
      void embedLeadPFNeutralCand();
      void embedSignalPFCands();
      void embedSignalPFChargedHadrCands();
      void embedSignalPFNeutralHadrCands();
      void embedSignalPFGammaCands();
      void embedIsolationPFCands();
      void embedIsolationPFChargedHadrCands();
      void embedIsolationPFNeutralHadrCands();
      void embedIsolationPFGammaCands();
      
      // ---- matched GenJet methods ----
      const reco::GenJet * genJet() const;
      void setGenJet(const reco::GenJetRef & ref);

      // ---- CaloTau accessors (getters only) ----
      bool isCaloTau() const { return !caloSpecific_.empty(); }
      const pat::tau::TauCaloSpecific & caloSpecific() const ;
      reco::CaloTauTagInfoRef caloTauTagInfoRef() const { return caloSpecific().CaloTauTagInfoRef_; }
      float leadTracksignedSipt() const { return caloSpecific().leadTracksignedSipt_; }
      float leadTrackHCAL3x3hitsEtSum() const { return caloSpecific().leadTrackHCAL3x3hitsEtSum_; }
      float leadTrackHCAL3x3hottesthitDEta() const { return caloSpecific().leadTrackHCAL3x3hottesthitDEta_; }
      float signalTracksInvariantMass() const { return caloSpecific().signalTracksInvariantMass_; }
      float TracksInvariantMass() const { return caloSpecific().TracksInvariantMass_; } 
      float isolationTracksPtSum() const { return caloSpecific().isolationTracksPtSum_; }
      float isolationECALhitsEtSum() const { return caloSpecific().isolationECALhitsEtSum_; }
      float maximumHCALhitEt() const { return caloSpecific().maximumHCALhitEt_; }

      // ---- PFTau accessors (getters only) ----
      bool isPFTau() const { return !pfSpecific_.empty(); }
      const pat::tau::TauPFSpecific & pfSpecific() const ;
      const reco::PFJetRef & pfJetRef() const { return pfSpecific().pfJetRef_; }
      const reco::PFCandidateRef leadPFChargedHadrCand() const;
      float leadPFChargedHadrCandsignedSipt() const { return pfSpecific().leadPFChargedHadrCandsignedSipt_; }
      const reco::PFCandidateRef leadPFNeutralCand() const;
      const reco::PFCandidateRef leadPFCand() const;
      const reco::PFCandidateRefVector & signalPFCands() const;
      const reco::PFCandidateRefVector & signalPFChargedHadrCands() const;
      const reco::PFCandidateRefVector & signalPFNeutrHadrCands() const;
      const reco::PFCandidateRefVector & signalPFGammaCands() const;
      const std::vector<reco::RecoTauPiZero> & signalPiZeroCandidates() const;
      const reco::PFCandidateRefVector & isolationPFCands() const;
      const reco::PFCandidateRefVector & isolationPFChargedHadrCands() const;
      const reco::PFCandidateRefVector & isolationPFNeutrHadrCands() const;
      const reco::PFCandidateRefVector & isolationPFGammaCands() const;
      const std::vector<reco::RecoTauPiZero> & isolationPiZeroCandidates() const;
      float isolationPFChargedHadrCandsPtSum() const { return pfSpecific().isolationPFChargedHadrCandsPtSum_; }
      float isolationPFGammaCandsEtSum() const { return pfSpecific().isolationPFGammaCandsEtSum_; }
      float maximumHCALPFClusterEt() const { return pfSpecific().maximumHCALPFClusterEt_; }
      float emFraction() const { return pfSpecific().emFraction_; }
      float hcalTotOverPLead() const { return pfSpecific().hcalTotOverPLead_; }
      float hcalMaxOverPLead() const { return pfSpecific().hcalMaxOverPLead_; }
      float hcal3x3OverPLead() const { return pfSpecific().hcal3x3OverPLead_; }
      float ecalStripSumEOverPLead() const { return pfSpecific().ecalStripSumEOverPLead_; }
      float bremsRecoveryEOverPLead() const { return pfSpecific().bremsRecoveryEOverPLead_; }
      const reco::TrackRef & electronPreIDTrack() const { return pfSpecific().electronPreIDTrack_; }
      float electronPreIDOutput() const { return pfSpecific().electronPreIDOutput_; }
      bool electronPreIDDecision() const { return pfSpecific().electronPreIDDecision_; }
      float caloComp() const { return pfSpecific().caloComp_; }
      float segComp() const { return pfSpecific().segComp_; }
      bool muonDecision() const { return pfSpecific().muonDecision_; }

      const reco::Candidate::LorentzVector& p4Jet() const;
      float etaetaMoment() const;
      float phiphiMoment() const;
      float etaphiMoment() const;

      int decayMode() const { return pfSpecific().decayMode_; }
      void setDecayMode(int);

      // ---- methods for tau ID ----
      float tauID(const std::string & name) const;
      float tauID(const char* name ) const {return tauID( std::string(name) );}
      bool isTauIDAvailable(const std::string & name) const;
      const std::vector<IdPair> &  tauIDs() const { return tauIDs_; }
      void setTauIDs(const std::vector<IdPair> & ids) { tauIDs_ = ids; }

      friend std::ostream& reco::operator<<(std::ostream& out, const Tau& obj);

      const std::vector<std::string> availableJECSets() const;
      // returns the available JEC Levels for a given jecSet
      const std::vector<std::string> availableJECLevels(const int& set = 0) const;
      // returns the available JEC Levels for a given jecSet
      const std::vector<std::string> availableJECLevels(const std::string& set) const { return availableJECLevels(jecSet(set)); };
      bool jecSetsAvailable() const { return !jec_.empty(); }
      bool jecSetAvailable(const std::string& set) const {return (jecSet(set) >= 0); };
      bool jecSetAvailable(const unsigned int& set) const {return (set < jec_.size()); };
      std::string currentJECSet() const { 
    return currentJECSet_<jec_.size() ? jec_.at(currentJECSet_).jecSet() : std::string("ERROR"); 
      }
      std::string currentJECLevel() const { 
    return currentJECSet_<jec_.size() ? jec_.at(currentJECSet_).jecLevel(currentJECLevel_) : std::string("ERROR"); 
      }
      float jecFactor(const std::string& level, const std::string& set = "") const;
      float jecFactor(const unsigned int& level, const unsigned int& set = 0) const;
      Tau correctedTauJet(const std::string& level, const std::string& set = "") const;
      Tau correctedTauJet(const unsigned int& level, const unsigned int& set = 0) const;
      const LorentzVector& correctedP4(const std::string& level, const std::string& set = "") const { 
    return correctedTauJet(level, set).p4(); 
      }
      const LorentzVector& correctedP4(const unsigned int& level, const unsigned int& set = 0) const { 
    return correctedTauJet(level, set).p4(); 
      }

    protected:
   
      int jecSet(const std::string& label) const;
      void currentJECSet(const unsigned int& set) { currentJECSet_=set; };
      void currentJECLevel(const unsigned int& level) { currentJECLevel_=level; };
      void addJECFactors(const TauJetCorrFactors& jec) {jec_.push_back(jec); };
      void initializeJEC(unsigned int level, const unsigned int set = 0);

      // ---- for content embedding ----
      bool embeddedIsolationTracks_;
      std::vector<reco::Track> isolationTracks_;
      mutable reco::TrackRefVector isolationTracksTransientRefVector_;
      mutable bool       isolationTracksTransientRefVectorFixed_;
      bool embeddedLeadTrack_;
      std::vector<reco::Track> leadTrack_;
      bool embeddedSignalTracks_;
      std::vector<reco::Track> signalTracks_;
      mutable reco::TrackRefVector signalTracksTransientRefVector_;
      mutable bool       signalTracksTransientRefVectorFixed_;
      // specific for PFTau
      std::vector<reco::PFCandidate> leadPFCand_;
      bool embeddedLeadPFCand_;
      std::vector<reco::PFCandidate> leadPFChargedHadrCand_;
      bool embeddedLeadPFChargedHadrCand_;
      std::vector<reco::PFCandidate> leadPFNeutralCand_;
      bool embeddedLeadPFNeutralCand_;

      std::vector<reco::PFCandidate> signalPFCands_;
      bool embeddedSignalPFCands_;
      mutable reco::PFCandidateRefVector signalPFCandsTransientRefVector_;
      mutable bool signalPFCandsRefVectorFixed_;
      std::vector<reco::PFCandidate> signalPFChargedHadrCands_;
      bool embeddedSignalPFChargedHadrCands_;
      mutable reco::PFCandidateRefVector signalPFChargedHadrCandsTransientRefVector_;
      mutable bool signalPFChargedHadrCandsRefVectorFixed_;
      std::vector<reco::PFCandidate> signalPFNeutralHadrCands_;
      bool embeddedSignalPFNeutralHadrCands_;
      mutable reco::PFCandidateRefVector signalPFNeutralHadrCandsTransientRefVector_;
      mutable bool signalPFNeutralHadrCandsRefVectorFixed_;
      std::vector<reco::PFCandidate> signalPFGammaCands_;
      bool embeddedSignalPFGammaCands_;
      mutable reco::PFCandidateRefVector signalPFGammaCandsTransientRefVector_;
      mutable bool signalPFGammaCandsRefVectorFixed_;
      std::vector<reco::PFCandidate> isolationPFCands_;
      bool embeddedIsolationPFCands_;
      mutable reco::PFCandidateRefVector isolationPFCandsTransientRefVector_;
      mutable bool isolationPFCandsRefVectorFixed_;
      std::vector<reco::PFCandidate> isolationPFChargedHadrCands_;
      bool embeddedIsolationPFChargedHadrCands_;
      mutable reco::PFCandidateRefVector isolationPFChargedHadrCandsTransientRefVector_;
      mutable bool isolationPFChargedHadrCandsRefVectorFixed_;
      std::vector<reco::PFCandidate> isolationPFNeutralHadrCands_;
      bool embeddedIsolationPFNeutralHadrCands_;
      mutable reco::PFCandidateRefVector isolationPFNeutralHadrCandsTransientRefVector_;
      mutable bool isolationPFNeutralHadrCandsRefVectorFixed_;
      std::vector<reco::PFCandidate> isolationPFGammaCands_;
      bool embeddedIsolationPFGammaCands_;
      mutable reco::PFCandidateRefVector isolationPFGammaCandsTransientRefVector_;
      mutable bool isolationPFGammaCandsRefVectorFixed_;

      // ---- matched GenJet holder ----
      std::vector<reco::GenJet> genJet_;

      // ---- tau ID's holder ----
      std::vector<IdPair> tauIDs_;

      // ---- CaloTau specific variables  ----
      std::vector<pat::tau::TauCaloSpecific> caloSpecific_;

      // ---- PFTau specific variables  ----
      std::vector<pat::tau::TauPFSpecific> pfSpecific_;

      // ---- energy scale correction factors ----
      // energy scale correction factors; the string carries a potential label if
      // more then one set of correction factors is embedded. The label corresponds
      // to the label of the jetCorrFactors module that has been embedded.
      std::vector<pat::TauJetCorrFactors> jec_;
      // currently applied set of jet energy correction factors (i.e. the index in
      // jetEnergyCorrections_)
      unsigned int currentJECSet_;
      // currently applied jet energy correction level
      unsigned int currentJECLevel_;
  };
}

#endif