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