RecoTauPiZero.h

Go to the documentation of this file.

#ifndef DataFormats_TauReco_RecoTauPiZero_h
#define DataFormats_TauReco_RecoTauPiZero_h
 
#include "DataFormats/Candidate/interface/CompositePtrCandidate.h"
 
namespace reco {
  class RecoTauPiZero : public CompositePtrCandidate {
  public:
    enum PiZeroAlgorithm {
      // Algorithm where each photon becomes a pi zero
      kUndefined = 0,
      kTrivial = 1,
      kCombinatoric = 2,
      kStrips = 3
    };
 
    RecoTauPiZero() : CompositePtrCandidate(), algoName_(kUndefined), bendCorrEta_(0.), bendCorrPhi_(0.) {
      this->setPdgId(111);
    }
 
    RecoTauPiZero(PiZeroAlgorithm algoName)
        : CompositePtrCandidate(), algoName_(algoName), bendCorrEta_(0.), bendCorrPhi_(0.) {
      this->setPdgId(111);
    }
 
    RecoTauPiZero(Charge q,
                  const LorentzVector& p4,
                  const Point& vtx = Point(0, 0, 0),
                  int pdgId = 111,
                  int status = 0,
                  bool integerCharge = true,
                  PiZeroAlgorithm algoName = kUndefined)
        : CompositePtrCandidate(q, p4, vtx, pdgId, status, integerCharge),
          algoName_(algoName),
          bendCorrEta_(0.),
          bendCorrPhi_(0.) {}
 
    RecoTauPiZero(Charge q,
                  const PolarLorentzVector& p4,
                  const Point& vtx = Point(0, 0, 0),
                  int pdgId = 111,
                  int status = 0,
                  bool integerCharge = true,
                  PiZeroAlgorithm algoName = kUndefined)
        : CompositePtrCandidate(q, p4, vtx, pdgId, status, integerCharge),
          algoName_(algoName),
          bendCorrEta_(0.),
          bendCorrPhi_(0.) {}
 
    explicit RecoTauPiZero(const Candidate& p, PiZeroAlgorithm algoName = kUndefined)
        : CompositePtrCandidate(p), algoName_(algoName), bendCorrEta_(0.), bendCorrPhi_(0.) {
      this->setPdgId(111);
    }
 
    ~RecoTauPiZero() override{};
 
    size_t numberOfGammas() const;
 
    size_t numberOfElectrons() const;
 
    double maxDeltaPhi() const;
 
    double maxDeltaEta() const;
 
    PiZeroAlgorithm algo() const;
 
    bool algoIs(PiZeroAlgorithm algo) const;
 
    float bendCorrEta() const { return bendCorrEta_; }
    float bendCorrPhi() const { return bendCorrPhi_; }
    void setBendCorrEta(float bendCorrEta) { bendCorrEta_ = bendCorrEta; }
    void setBendCorrPhi(float bendCorrPhi) { bendCorrPhi_ = bendCorrPhi; }
 
    void print(std::ostream& out = std::cout) const;
 
  private:
    PiZeroAlgorithm algoName_;
 
    float bendCorrEta_;
    float bendCorrPhi_;
  };
 
  std::ostream& operator<<(std::ostream& out, const RecoTauPiZero& c);
 
}  // namespace reco
 
#endif