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,
      VFcalEntrance = 9,
      // Number of valid layers
      NLayers = 10,
      // Unknown
      Unknown = -1
    };
 
    static const std::array<std::string, NLayers> layerTypeNames;
    static LayerType layerTypeByName(const std::string& name);
 
    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);
}  // namespace reco
 
#endif