PFTrack.h

Go to the documentation of this file.

#ifndef DataFormats_ParticleFlowReco_PFTrack_h
#define DataFormats_ParticleFlowReco_PFTrack_h
 
#include "DataFormats/ParticleFlowReco/interface/PFTrajectoryPoint.h"
 
#include <iostream>
#include <vector>
 
namespace reco {
 
  class PFTrack {
  public:
    PFTrack();
 
    PFTrack(double charge);
 
    PFTrack(const PFTrack& other);
 
    void addPoint(const reco::PFTrajectoryPoint& trajPt);
 
    void setPoint(unsigned int index, const reco::PFTrajectoryPoint& measurement) {
      trajectoryPoints_[index] = measurement;
    }
 
    void calculatePositionREP();
 
    double charge() const { return charge_; }
 
    unsigned int nTrajectoryPoints() const { return trajectoryPoints_.size(); }
 
    unsigned int nTrajectoryMeasurements() const {
      return (indexOutermost_ ? indexOutermost_ - indexInnermost_ + 1 : 0);
    }
 
    const std::vector<reco::PFTrajectoryPoint>& trajectoryPoints() const { return trajectoryPoints_; }
 
    const reco::PFTrajectoryPoint& trajectoryPoint(unsigned index) const { return trajectoryPoints_[index]; }
 
    const reco::PFTrajectoryPoint& extrapolatedPoint(unsigned layerid) const;
 
    std::vector<reco::PFTrajectoryPoint>::const_iterator innermostMeasurement() const {
      return trajectoryPoints_.begin() + indexInnermost_;
    }
 
    std::vector<reco::PFTrajectoryPoint>::const_iterator outermostMeasurement() const {
      return trajectoryPoints_.begin() + indexOutermost_;
    }
 
    void setColor(int color) { color_ = color; }
 
    int color() const { return color_; }
 
  protected:
    static const unsigned int nMaxTrackingLayers_;
 
    double charge_;
 
    std::vector<reco::PFTrajectoryPoint> trajectoryPoints_;
 
    unsigned int indexInnermost_;
 
    unsigned int indexOutermost_;
 
    int color_;
  };
  std::ostream& operator<<(std::ostream& out, const PFTrack& track);
 
}  // namespace reco
 
#endif