PFRecoTauChargedHadron.h

Go to the documentation of this file.

#ifndef DataFormats_TauReco_PFRecoTauChargedHadron_h
#define DataFormats_TauReco_PFRecoTauChargedHadron_h

#include "DataFormats/Candidate/interface/CompositePtrCandidate.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/Common/interface/Ptr.h"
#include "DataFormats/Math/interface/Point3D.h"

namespace reco { namespace tau {
  class PFRecoTauChargedHadronFromPFCandidatePlugin;
  class PFRecoTauChargedHadronFromTrackPlugin;
  class RecoTauConstructor;
  class PFRecoTauEnergyAlgorithmPlugin;
}} 
class PFRecoTauChargedHadronProducer;

namespace reco {

class PFRecoTauChargedHadron : public CompositePtrCandidate
{
 public:
  typedef edm::Ptr<Track> TrackPtr;

  enum PFRecoTauChargedHadronAlgorithm {
    // Algorithm where each photon becomes a pi zero
    kUndefined = 0,
    kChargedPFCandidate = 1,
    kTrack = 2,
    kPFNeutralHadron = 3
  };

  PFRecoTauChargedHadron();
  PFRecoTauChargedHadron(PFRecoTauChargedHadronAlgorithm algo, Charge q);


  PFRecoTauChargedHadron(const Candidate & c, PFRecoTauChargedHadronAlgorithm algo = kUndefined);

  PFRecoTauChargedHadron(Charge q, const LorentzVector& p4,
             const Point& vtx = Point( 0, 0, 0 ),
             int status = 0, bool integerCharge = true,
             PFRecoTauChargedHadronAlgorithm algo = kUndefined);

  ~PFRecoTauChargedHadron() override;

  const PFCandidatePtr& getChargedPFCandidate() const;

  const TrackPtr& getTrack() const;

  const std::vector<PFCandidatePtr>& getNeutralPFCandidates() const;  

  const math::XYZPointF& positionAtECALEntrance() const;

  PFRecoTauChargedHadronAlgorithm algo() const;

  bool algoIs(PFRecoTauChargedHadronAlgorithm algo) const;

  void print(std::ostream& stream = std::cout) const;
    
 private:
  friend class tau::PFRecoTauChargedHadronFromPFCandidatePlugin;
  friend class tau::PFRecoTauChargedHadronFromTrackPlugin;
  friend class tau::RecoTauConstructor;
  friend class tau::PFRecoTauEnergyAlgorithmPlugin;
  friend class ::PFRecoTauChargedHadronProducer;

  PFRecoTauChargedHadronAlgorithm algo_;

  PFCandidatePtr chargedPFCandidate_;
  TrackPtr track_;
  std::vector<PFCandidatePtr> neutralPFCandidates_;

  math::XYZPointF positionAtECALEntrance_;
};

std::ostream& operator<<(std::ostream& stream, const PFRecoTauChargedHadron& c);

} // end namespace reco

#endif