TrackToVertexIPEstimator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRKVERTEXFITTERUTILS_TRACKTOVERTEXIPESTIMATOR_H
#define TRKVERTEXFITTERUTILS_TRACKTOVERTEXIPESTIMATOR_H
 
#include "TrkVertexFitterInterfaces/ITrackToVertexIPEstimator.h"
//
#include "AthenaBaseComps/AthAlgTool.h"
//
#include "GaudiKernel/ToolHandle.h"
#include "TrkExInterfaces/IExtrapolator.h"
#include "TrkVertexFitterInterfaces/IVertexLinearizedTrackFactory.h"
#include "TrkVertexFitterInterfaces/IVertexUpdator.h"
//
#include "xAODTracking/TrackParticleFwd.h"
#include "xAODTracking/VertexFwd.h"
//
#include "CLHEP/Vector/ThreeVector.h"
namespace Trk {
 
class Track;
class TrackParticleBase;

 
class TrackToVertexIPEstimator : public extends<AthAlgTool,
                                                ITrackToVertexIPEstimator> {
 
 public:
  virtual StatusCode initialize() override;

  TrackToVertexIPEstimator(const std::string& t, const std::string& n,
                           const IInterface* p);
 
  ~TrackToVertexIPEstimator();

  virtual std::unique_ptr<ImpactParametersAndSigma> estimate(
      const xAOD::TrackParticle* track, const xAOD::Vertex* vtx,
      bool doRemoval) const override;
  virtual std::unique_ptr<ImpactParametersAndSigma> estimate(
      const TrackParameters* track, const xAOD::Vertex* vtx,
      bool doRemoval) const override;
 
  virtual std::unique_ptr<ImpactParametersAndSigma> estimate(
      const xAOD::TrackParticle* track, const xAOD::TrackParticle* newtrack,
      const xAOD::Vertex* vtx, bool doRemoval) const override;
  virtual std::unique_ptr<ImpactParametersAndSigma> estimate(
      const TrackParameters* track, const TrackParameters* newtrack,
      const xAOD::Vertex* vtx, bool doRemoval) const override;
 
  virtual xAOD::Vertex* getUnbiasedVertex(
      const xAOD::TrackParticle* track, const xAOD::Vertex* vtx) const override;
  virtual xAOD::Vertex* getUnbiasedVertex(
      const TrackParameters* track, const xAOD::Vertex* vtx) const override;
 
  virtual std::unique_ptr<ImpactParametersAndSigma> estimate(
      const xAOD::TrackParticle* track, const xAOD::Vertex* vtx) const override;
  virtual std::unique_ptr<ImpactParametersAndSigma> estimate(
      const TrackParameters* track, const xAOD::Vertex* vtx) const override;
 
  /* Methods which provide the lifetime Sign for the Impact Parameter (2D, Z,
   * 3D)  */
  virtual double get3DLifetimeSignOfTrack(
      const TrackParameters& track, const Amg::Vector3D& jetDirection,
      const xAOD::Vertex& primaryVertex) const override;
  virtual double get3DLifetimeSignOfTrack(
      const TrackParameters& track, const CLHEP::Hep3Vector& jetDirection,
      const xAOD::Vertex& primaryVertex) const override;
 
  virtual double get2DLifetimeSignOfTrack(
      const TrackParameters& track, const Amg::Vector3D& jetDirection,
      const xAOD::Vertex& primaryVertex) const override;
  virtual double get2DLifetimeSignOfTrack(
      const TrackParameters& track, const CLHEP::Hep3Vector& jetDirection,
      const xAOD::Vertex& primaryVertex) const override;
 
  virtual double getZLifetimeSignOfTrack(
      const TrackParameters& track, const Amg::Vector3D& jetDirection,
      const xAOD::Vertex& primaryVertex) const override;
  virtual double getZLifetimeSignOfTrack(
      const TrackParameters& track, const CLHEP::Hep3Vector& jetDirection,
      const xAOD::Vertex& primaryVertex) const override;
 
 private:
  std::unique_ptr<ImpactParametersAndSigma> calculate(
      const TrackParameters* track, const xAOD::Vertex& vtx) const;
 
  ToolHandle<Trk::IExtrapolator> m_extrapolator{this, "Extrapolator",
                                                "Trk::Extrapolator"};
  ToolHandle<Trk::IVertexUpdator> m_Updator{this, "VertexUpdator",
                                            "Trk::KalmanVertexUpdator"};
  ToolHandle<Trk::IVertexLinearizedTrackFactory> m_linFactory{
      this, "LinearizedTrackFactory", "Trk::FullLinearizedTrackFactory"};
 
};  // end of class definitions
 
}  // namespace Trk
 
#endif