KalmanFitterTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ACTSTRACKRECONSTRUCTION_KALMANFITTERTOOL_H
#define ACTSTRACKRECONSTRUCTION_KALMANFITTERTOOL_H
 
#include "src/detail/FitterHelperFunctions.h"
#include "AthenaBaseComps/AthAlgTool.h"
 
// ACTS
#include "Acts/EventData/BoundTrackParameters.hpp"
#include "Acts/TrackFitting/KalmanFitter.hpp"
#include "Acts/Propagator/SympyStepper.hpp"
#include "Acts/Propagator/Propagator.hpp"
#include "Acts/Propagator/Navigator.hpp"
#include "Acts/EventData/VectorTrackContainer.hpp"
#include "Acts/Geometry/GeometryIdentifier.hpp"
 
// PACKAGE
#include "ActsEvent/TrackContainer.h"
#include "ActsGeometryInterfaces/IExtrapolationTool.h"
#include "ActsGeometryInterfaces/ITrackingGeometryTool.h"
#include "ActsGeometryInterfaces/IGeometryRealmConvTool.h"
 
#include "ActsCalibrators/TrkMeasSurfaceAccessor.h"
#include "ActsCalibrators/TrkPrepRawDataCalibrator.h"
#include "ActsCalibrators/xAODUncalibMeasSurfAcc.h"
#include "ActsCalibrators/TrkPrepRawDataSurfaceAcc.h"
#include "src/detail/OnTrackCalibrator.h"
 
#include "ActsToolInterfaces/IFitterTool.h"
 
class EventContext;
 
namespace Trk{
  class Track;
  class PrepRawData;
}
 
namespace ActsTrk {
 
class KalmanFitterTool
  : public extends<AthAlgTool, ActsTrk::IFitterTool> { 
public:
 
  using base_class::base_class;
  virtual ~KalmanFitterTool() = default;
 
  // standard Athena methods
  virtual StatusCode initialize() override;
  
  virtual  
      std::unique_ptr< ActsTrk::MutableTrackContainer >
      fit(const std::vector<const xAOD::UncalibratedMeasurement*> & clusterList,
      const Acts::BoundTrackParameters& initialParams,
      const Acts::GeometryContext& tgContext,
      const Acts::MagneticFieldContext& mfContext,
      const Acts::CalibrationContext& calContext,      
      const Acts::Surface* targetSurface = nullptr  // optional target surface - defaults to perigee in global origin
      ) const override;

  virtual
    std::unique_ptr< ActsTrk::MutableTrackContainer >
    fit(const ActsTrk::Seed &seed,
        const Acts::BoundTrackParameters& initialParams,
        const Acts::GeometryContext& tgContext,
        const Acts::MagneticFieldContext& mfContext,
        const Acts::CalibrationContext& calContext,
    const Acts::Surface& targetSurface) const override;
 
    virtual StatusCode fit(
        const EventContext& ctx,
    const ActsTrk::TrackContainer::ConstTrackProxy& track,          
        ActsTrk::MutableTrackContainer& trackContainer,
    const Acts::PerigeeSurface& pSurface) const override;

  virtual
    std::unique_ptr< ActsTrk::MutableTrackContainer >
    fit(const std::vector<Acts::SourceLink>& sourceLinks,
        const Acts::BoundTrackParameters& initialParams,
        const Acts::GeometryContext& tgContext,
        const Acts::MagneticFieldContext& mfContext,
        const Acts::CalibrationContext& calContext,
        const Acts::Surface* targetSurface = nullptr) const override;
  
  // Private methods:
private:
    using TrackState_t = MutableTrackStateBackend::TrackStateProxy;
    using FitterExtension_t = Acts::KalmanFitterExtensions<MutableTrackStateBackend>;
    using FitterOptions_t = Acts::KalmanFitterOptions<MutableTrackStateBackend>;
    FitterOptions_t configureFit(const Acts::GeometryContext& tgContext,
                                 const Acts::MagneticFieldContext& mfContext,
                                 const Acts::CalibrationContext& calContext,
                                 const Acts::Surface* surface,
                                 detail::SourceLinkType slType) const;
 
 
  ToolHandle<IExtrapolationTool> m_extrapolationTool{this, "ExtrapolationTool", ""};
  PublicToolHandle<ITrackingGeometryTool> m_trackingGeometryTool{this, "TrackingGeometryTool", ""};
  PublicToolHandle<IGeometryRealmConvTool> m_geometryConvTool{this, "GeometryRealmConvTool", ""};
 
  ToolHandle<Trk::IRIO_OnTrackCreator> m_ROTcreator {this, "RotCreatorTool", ""};
 
  // the settable job options
  Gaudi::Property< double > m_option_outlierChi2Cut {this, "OutlierChi2Cut", 12.5, 
      "Chi2 cut used by the outlier finder" };
  Gaudi::Property< double > m_option_ReverseFilteringPt {this, "ReverseFilteringPt", 1.0 * Acts::UnitConstants::GeV,
      "Pt cut used for the ReverseFiltering logic"};
  Gaudi::Property< int > m_option_maxPropagationStep {this, "MaxPropagationStep", 5000, 
      "Maximum number of steps for one propagate call"};
  Gaudi::Property<bool> m_useDirectNavigation{this, "UseDirectNavigation", true,
      "GSF with direct navigation when refitting measurements"};

  detail::TrkMeasurementCalibrator m_trkCalibrator{};
  detail::TrkMeasSurfaceAccessor m_trkSurfAcc{};
  detail::TrkPrepRawDataCalibrator m_prdCalibrator{};
  detail::TrkPrepRawDataSurfaceAcc m_prdSurfAcc{};
  detail::xAODUncalibMeasSurfAcc m_unalibMeasSurfAcc{};
  using xAODUnCalibrator_t = detail::OnTrackCalibrator<ActsTrk::MutableTrackStateBackend> ;
  xAODUnCalibrator_t m_uncalibMeasCalibrator{};
 

  using Fitter = Acts::KalmanFitter<Acts::Propagator<Acts::SympyStepper, Acts::Navigator>, ActsTrk::MutableTrackStateBackend>;
  std::unique_ptr<Fitter> m_fitter {nullptr};
 
  using DirectFitter = Acts::KalmanFitter<Acts::Propagator<Acts::SympyStepper, Acts::DirectNavigator>, ActsTrk::MutableTrackStateBackend>;
  std::unique_ptr<DirectFitter> m_directFitter {nullptr};

  static constexpr unsigned s_nExtensions = static_cast<unsigned>(detail::SourceLinkType::nTypes);
  std::array<FitterExtension_t, s_nExtensions>  m_kfExtensions{};
 
  ActsTrk::detail::FitterHelperFunctions::ATLASOutlierFinder m_outlierFinder{0};
  ActsTrk::detail::FitterHelperFunctions::ReverseFilteringLogic m_reverseFilteringLogic{0};

  const Acts::Logger& logger() const {
    return *m_logger;
  }

  std::unique_ptr<const Acts::Logger> m_logger;
 
}; // end of namespace
 
}
#endif