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(){};

    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);

}

#endif