PFRecTrack.h

Go to the documentation of this file.

#ifndef DataFormats_ParticleFlowReco_PFRecTrack_h
#define DataFormats_ParticleFlowReco_PFRecTrack_h

#include "DataFormats/ParticleFlowReco/interface/PFTrack.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"

namespace reco {

  class PFRecTrack : public PFTrack {
  public:
    enum AlgoType_t {
      Unknown = 0,
      KF = 1,  // Kalman filter
      GSF = 2,
      KF_ELCAND = 3  // Gaussian sum filter
    };

    PFRecTrack();
    ~PFRecTrack(){};
    PFRecTrack(double charge, AlgoType_t algoType, int trackId, const reco::TrackRef& trackref);

    PFRecTrack(double charge, AlgoType_t algoType);

    /*     PFRecTrack(const PFRecTrack& other); */

    unsigned int algoType() const { return algoType_; }

    int trackId() const { return trackId_; }

    const reco::TrackRef& trackRef() const { return trackRef_; }

    void setSTIP(float STIP) { STIP_ = STIP; }

    const float STIP() const { return STIP_; }

    inline auto eta() const { return trackRef_->eta(); }

    inline auto phi() const { return trackRef_->phi(); }

  private:
    AlgoType_t algoType_;

    int trackId_;

    reco::TrackRef trackRef_;
    float STIP_;
  };

  std::ostream& operator<<(std::ostream& out, const PFRecTrack& track);

}  // namespace reco

#endif