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

    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