GlobalChiSquareFitterTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ACTSGEOMETRY_GLOBALCHISQUAREFITTERTOOL_H
#define ACTSGEOMETRY_GLOBALCHISQUAREFITTERTOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "ActsToolInterfaces/IFitterTool.h"
#include "TrkToolInterfaces/IRIO_OnTrackCreator.h"
#include "src/detail/FitterHelperFunctions.h"
#include "src/detail/OnTrackCalibrator.h"
 
#include "MuonRecToolInterfacesR4/ISpacePointCalibrator.h"
// ACTS
#include "Acts/EventData/BoundTrackParameters.hpp"
#include "Acts/EventData/VectorTrackContainer.hpp"
#include "Acts/Geometry/GeometryIdentifier.hpp"
#include "Acts/Propagator/Navigator.hpp"
#include "Acts/Propagator/Propagator.hpp"
#include "Acts/Propagator/StraightLineStepper.hpp"
#include "Acts/Propagator/EigenStepper.hpp"
#include "Acts/TrackFitting/GlobalChiSquareFitter.hpp"
 
// PACKAGE
#include "ActsEvent/TrackContainer.h"
#include "ActsGeometryInterfaces/IExtrapolationTool.h"
#include "ActsGeometryInterfaces/ITrackingGeometryTool.h"
#include "ActsGeometryInterfaces/IGeometryRealmConvTool.h"
 
#include "ActsCalibrators/TrkMeasurementCalibrator.h"
#include "ActsCalibrators/TrkPrepRawDataCalibrator.h"
#include "ActsCalibrators/TrkMeasSurfaceAccessor.h"
#include "ActsCalibrators/TrkPrepRawDataSurfaceAcc.h"
#include "ActsCalibrators/xAODUncalibMeasSurfAcc.h"
#include "ActsCalibrators/xAODUncalibMeasCalibrator.h"
 
namespace Trk {
class PrepRawData;
}  // namespace Trk
 
namespace ActsTrk {
 
 
 
class GlobalChiSquareFitterTool
    : public extends<AthAlgTool, IFitterTool> {
 public:
 
  using base_class::base_class;
  virtual ~GlobalChiSquareFitterTool() = default;
 
  // standard Athena methods
  virtual StatusCode initialize() override;

  virtual std::unique_ptr<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<MutableTrackContainer> fit(
      const 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 TrackContainer::ConstTrackProxy& track,          
    MutableTrackContainer& trackContainer,
    const Acts::PerigeeSurface& pSurface) const override;

  virtual std::unique_ptr<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  // optional target surface - defaults to perigee in global
                   // origin
  ) const override;
 

    using StraightPropagator_t = Acts::Propagator<Acts::StraightLineStepper, Acts::Navigator>;
    using CurvedPropagator_t   = Acts::Propagator<Acts::EigenStepper<>, Acts::Navigator>;
 
    using StraightFitter_t = Acts::Experimental::Gx2Fitter<StraightPropagator_t, MutableTrackStateBackend>;
    using CurvedFitter_t   = Acts::Experimental::Gx2Fitter<CurvedPropagator_t, MutableTrackStateBackend>;

    using Gx2FitterOptions_t = Acts::Experimental::Gx2FitterOptions<MutableTrackStateBackend>;
    using Gx2FitterExtension_t = Acts::Experimental::Gx2FitterExtensions<MutableTrackStateBackend>;
   
  private:
    using TrackState_t = MutableTrackStateBackend::TrackStateProxy;

    Gx2FitterOptions_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<MuonR4::ISpacePointCalibrator> m_muonCalibrator{this, "MuonCalibrationTool", ""};
 
    ToolHandle<Trk::IRIO_OnTrackCreator> m_ROTcreator{this, "RotCreatorTool", ""};
    
    Gaudi::Property<double> m_option_outlierChi2Cut{this, "OutlierChi2Cut", 12.5};
    Gaudi::Property<unsigned> m_option_maxPropagationStep{this, "MaxPropagationStep", 5000};
    Gaudi::Property<unsigned> m_option_maxNavSurfaces{this, "MaxSurfacesPerNavStep", 100};
    Gaudi::Property<bool> m_option_includeELoss{this, "IncludeELoss", true};
    Gaudi::Property<bool> m_option_includeScat{this, "IncludeScattering", true};
    Gaudi::Property<bool> m_doStraightLine{this, "DoStraightLine" , false};
    Gaudi::Property<bool> m_doJacobianCorr{this, "DoFreeToBoundCorrection", false };
    Gaudi::Property<unsigned> m_nIterMax{this, "MaxIterations", 5};
    detail::TrkMeasurementCalibrator m_trkMeasCalibrator{};
    detail::TrkPrepRawDataCalibrator m_prdCalibrator{};
    detail::TrkMeasSurfaceAccessor m_trkMeasSurfAcc{};
    detail::TrkPrepRawDataSurfaceAcc m_prdSurfaceAcc{};
    detail::xAODUncalibMeasSurfAcc m_unalibMeasSurfAcc{};
    detail::xAODUncalibMeasCalibrator m_uncalibMeasCalibrator{};
    using xAODItkCalibrator_t = detail::OnTrackCalibrator<ActsTrk::MutableTrackStateBackend> ;
    xAODItkCalibrator_t m_idCalibrator{};

    static constexpr unsigned s_nExtensions = static_cast<unsigned>(detail::SourceLinkType::nTypes);
    std::array<Gx2FitterExtension_t, s_nExtensions>  m_gx2fExtensions{};
    std::unique_ptr<CurvedFitter_t> m_fitter{nullptr};
    std::unique_ptr<StraightFitter_t> m_slFitter{nullptr};
 
    detail::FitterHelperFunctions::ATLASOutlierFinder m_outlierFinder{0};

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

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