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

}

#endif