FitterTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTRACKFINDINGTOOLS_FITTERTOOL_H
#define MUONTRACKFINDINGTOOLS_FITTERTOOL_H
 
 
#include "GeoPrimitives/GeoPrimitives.h"
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "ActsToolInterfaces/IFitterTool.h"
#include "ActsGeometryInterfaces/ITrackingGeometrySvc.h"
#include "MuonRecToolInterfacesR4/ISpacePointCalibrator.h"
 
#include "ActsCalibrators/xAODUncalibMeasSurfAcc.h"
#include "ActsCalibrators/xAODUncalibMeasCalibrator.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 "MuonTrackFindingTools/Chi2Fitter.h"
 
namespace MuonR4 {
    class FitterTool : public extends<AthAlgTool, ActsTrk::IFitterTool> {
        public:
            using base_class::base_class;
            virtual StatusCode initialize() override final;
            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 final;

            virtual std::unique_ptr<ActsTrk::MutableTrackContainer > 
                    fit(const std::vector<const xAOD::UncalibratedMeasurement* > & measList,
                        const Acts::BoundTrackParameters& initialParams,
                        const Acts::GeometryContext& tgContext,
                        const Acts::MagneticFieldContext& mfContext,
                        const Acts::CalibrationContext& calContext,
                        const Acts::Surface* targetSurface = nullptr) const override final;
            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 final;
            virtual StatusCode fit(const EventContext& ctx,
                                   const ActsTrk::TrackContainer::ConstTrackProxy& track,          
                                   ActsTrk::MutableTrackContainer& trackContainer,
                                   const Acts::PerigeeSurface& pSurface) const override final;
            
            using StraightPropagator_t = Acts::Propagator<Acts::StraightLineStepper, Acts::Navigator>;
            using CurvedPropagator_t   = Acts::Propagator<Acts::EigenStepper<>, Acts::Navigator>;
            using StraightFitter_t = MuonExperimental::Gx2Fitter<StraightPropagator_t, ActsTrk::MutableTrackStateBackend>;
            using CurvedFitter_t   = MuonExperimental::Gx2Fitter<CurvedPropagator_t, ActsTrk::MutableTrackStateBackend>;

            using Gx2FitterOptions_t = MuonExperimental::Gx2FitterOptions<ActsTrk::MutableTrackStateBackend>;
            using Gx2FitterExtension_t = MuonExperimental::Gx2FitterExtensions<ActsTrk::MutableTrackStateBackend>;
        private:
            Gaudi::Property<bool> m_doStraightLineFit{this, "doStraightLine", false};
            Gaudi::Property<unsigned> m_doMaxSteps{this, "maxSteps", 1000000 };
            Gaudi::Property<unsigned> m_maxSurfaces{this, "maxSurfaces", 10000};
            Gaudi::Property<unsigned> m_maxIter{this, "nIterMax" , 250};
            Gaudi::Property<bool> m_doEnergyLoss{this, "doEnergyLoss" , false};
            Gaudi::Property<bool> m_doScattering{this, "doScattering" , false};
            ServiceHandle<ActsTrk::ITrackingGeometrySvc> m_trackingGeometrySvc{this, "TrackingGeometrySvc", 
                                                                                "ActsTrackingGeometrySvc"};

            ToolHandle<MuonR4::ISpacePointCalibrator> m_muonCalibrator{this, "MuonCalibrationTool", ""};
            
            ActsTrk::detail::xAODUncalibMeasSurfAcc m_unalibMeasSurfAcc{};
            ActsTrk::detail::xAODUncalibMeasCalibrator m_uncalibMeasCalibrator{};
 
            Gx2FitterExtension_t m_extensions{};
            
            using Fitter_t = std::variant<std::monostate,
                                         StraightFitter_t, 
                                         CurvedFitter_t>;
            Fitter_t m_fitter{};
 
 
 
 
    };
}
#endif