PFTrajectoryPoint.h

Go to the documentation of this file.

#ifndef DataFormats_ParticleFlowReco_PFTrajectoryPoint_h
#define DataFormats_ParticleFlowReco_PFTrajectoryPoint_h

#include <vector>
#include <map>
#include <iosfwd>

#include "DataFormats/Math/interface/Point3D.h"
#include "Rtypes.h" 
#include "DataFormats/Math/interface/LorentzVector.h"
#include "Math/GenVector/PositionVector3D.h"

namespace reco {

  class PFTrajectoryPoint {

  public:
    // Next typedef uses double in ROOT 6 rather than Double32_t due to a bug in ROOT 5,
    // which otherwise would make ROOT5 files unreadable in ROOT6.  This does not increase
    // the size on disk, because due to the bug, double was actually stored on disk in ROOT 5.
    typedef ROOT::Math::PositionVector3D<ROOT::Math::CylindricalEta3D<double> > REPPoint;

    enum LayerType {
      ClosestApproach = 0,
      BeamPipeOrEndVertex = 1,       
      PS1 = 2,             
      PS2 = 3,             
      ECALEntrance = 4,  
      ECALShowerMax = 5,   
      HCALEntrance = 6,
      HCALExit = 7,
      HOLayer = 8,

      NLayers = 9
    };

    PFTrajectoryPoint();

    PFTrajectoryPoint(int detId,
                      int layer,
                      const math::XYZPoint& posxyz, 
                      const math::XYZTLorentzVector& momentum); 

    PFTrajectoryPoint(const PFTrajectoryPoint& other);

    virtual ~PFTrajectoryPoint();


    int detId() const    { return detId_; }

    int layer() const    { return layer_; }

    bool     isValid() const {
      if( layer_ == -1 && detId_ == -1 ) return false;
      else return true;
    }

    bool isTrackerLayer() const {
      if(detId_ >= 0 ) return true; 
      else return false;
    }

    const math::XYZPoint& position() const { return posxyz_; }

    const REPPoint& positionREP() const { return posrep_; }

    void calculatePositionREP() {
      posrep_.SetCoordinates( posxyz_.Rho(), posxyz_.Eta(), posxyz_.Phi() );
    }

    const math::XYZTLorentzVector& momentum() const    { return momentum_; }

    bool   operator==(const reco::PFTrajectoryPoint& other) const;

    friend std::ostream& operator<<(std::ostream& out, const reco::PFTrajectoryPoint& trajPoint);

  private:

    bool isTrackerLayer_;

    int detId_;             

    int layer_;

    math::XYZPoint          posxyz_;

    REPPoint                posrep_;

    math::XYZTLorentzVector momentum_;

  };

  std::ostream& operator<<(std::ostream& out, const reco::PFTrajectoryPoint& trajPoint); 
}

#endif