TrackToVertex.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
// TrackToVertex.h, (c) ATLAS Detector software 2005
 
 
#ifndef RECOTOOLS_TRACKTOVERTEX_H
#define RECOTOOLS_TRACKTOVERTEX_H
 
// Gaudi
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
// Trk
#include "ITrackToVertex/ITrackToVertex.h"
#include "TrkParameters/TrackParameters.h"
#include "TrkTrack/Track.h"
#include "GeoPrimitives/GeoPrimitives.h"
#include "BeamSpotConditionsData/BeamSpotData.h"
#include "TrkExInterfaces/IExtrapolator.h"
 
namespace Rec {
  class TrackParticle;
}
 
namespace Trk {
  class StraightLineSurface;
}
 
namespace Reco {
                        
  class TrackToVertex : public extends <AthAlgTool, ITrackToVertex>
  {
    public:
           
      TrackToVertex(const std::string&,const std::string&,const IInterface*);
 
      virtual ~TrackToVertex() = default;
 
      virtual StatusCode initialize() override final;
      virtual StatusCode finalize() override final;
 
      virtual std::unique_ptr<Trk::StraightLineSurface> GetBeamLine(
        const InDet::BeamSpotData*)
        const override final; // In C++17 make this [[nodiscard]]
 
      virtual std::unique_ptr<Trk::Perigee> perigeeAtVertex(
        const EventContext& ctx,
        const Rec::TrackParticle& tp,
        const Amg::Vector3D& gp) const override final;
 
      virtual std::unique_ptr<Trk::Perigee> perigeeAtVertex(
        const EventContext& ctx,
        const xAOD::TrackParticle& tp,
        const Amg::Vector3D& gp) const override final;
 
      virtual std::unique_ptr<Trk::Perigee> perigeeAtVertex(
        const EventContext& ctx,
        const Rec::TrackParticle& tp) const override final;
 
      virtual std::unique_ptr<Trk::Perigee> perigeeAtVertex(
        const EventContext& ctx,
        const xAOD::TrackParticle& tp) const override final;
 
      virtual std::unique_ptr<Trk::Perigee> perigeeAtVertex(
        const EventContext& ctx,
        const Trk::Track& trk,
        const Amg::Vector3D& gp) const override final;
 
      virtual std::unique_ptr<Trk::Perigee> perigeeAtBeamline(
        const EventContext& ctx,
        const Trk::Track& trk,
        const InDet::BeamSpotData*) const override final;
 
      virtual std::unique_ptr<Trk::TrackParameters> trackAtBeamline(
        const EventContext& ctx,
        const Rec::TrackParticle& tp) const override final;
 
      virtual std::unique_ptr<Trk::TrackParameters> trackAtBeamline(
        const EventContext& ctx,
        const xAOD::TrackParticle& tp,
        const InDet::BeamSpotData*) const override final;
 
      virtual std::unique_ptr<Trk::TrackParameters> trackAtBeamline(
        const EventContext& ctx,
        const Trk::Track& trk,
        const Trk::StraightLineSurface* beamline) const override final;
 
      virtual std::unique_ptr<Trk::TrackParameters> trackAtBeamline(
        const EventContext& ctx,
        const Trk::TrackParameters& tpars,
        const Trk::StraightLineSurface* beamline) const override final;
 
    private:
      inline bool startAtOriginalPerigee(const Trk::Track &track) const {
        return m_startTRTSAAtPerigee.value()
            && track.info().patternRecognition().test(Trk::TrackInfo::TRTStandalone)
            && track.perigeeParameters();
      }
 
      ToolHandle<Trk::IExtrapolator>
        m_extrapolator {this, "Extrapolator", "Trk::Extrapolator/AtlasExtrapolator"}; 
      Gaudi::Property<bool> m_startTRTSAAtPerigee
        {this, "StartTRTStandaloneTracksAtOriginalPerigee", false,
         "When extrapolating TRT standalone start at their original perigee which may have a more realistic covariance"};
 
      const static Amg::Vector3D s_origin; 
  };
 
} // end of namespace
 
 
#endif // RECOTOOLS_TRACKTOVERTEX_H