FastMuonSABuilder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONR4_FASTRECONSTRUCTIONALGS_FASTMUONSABUILDER__H
#define MUONR4_FASTRECONSTRUCTIONALGS_FASTMUONSABUILDER__H
 
#include <xAODMuon/Muon.h>
#include <xAODMuon/MuonContainer.h>
#include "xAODMuon/MuonAuxContainerR4.h"
#include "xAODMuonViews/FillContainer.h"
 
#include <MuonFastRecoEvent/GlobalPattern.h>
 
#include <MuonRecToolInterfacesR4/ISpacePointCalibrator.h>
#include <MuonPatternHelpers/SegmentLineFitter.h>
#include <MuonPatternHelpers/MdtSegmentSeedGenerator.h>
#include <MuonTrackFindingTools/MsTrackSeeder.h>
#include "MuonSpacePoint/SpacePointPerLayerSorter.h"
#include "MagFieldConditions/AtlasFieldCacheCondObj.h"
 
#include <MuonRecToolInterfacesR4/IPatternVisualizationTool.h>
#include <MuonIdHelpers/IMuonIdHelperSvc.h>
 
 
namespace MuonR4::FastReco{
    
 
    class FastMuonSABuilder : public AthMessaging {
        public:
            using StIndex = Muon::MuonStationIndex::StIndex;
            using Parameters = SegmentFit::Parameters;
            using LineFitter = MuonR4::SegmentFit::SegmentLineFitter;
            using MdtSegmentSeeder = MuonR4::SegmentFit::MdtSegmentSeedGenerator;
            using MuonCont_t = xAOD::FillContainer<xAOD::MuonContainer, xAOD::MuonAuxContainerR4>;           
            struct Config {
                bool recalibInFit{false};
                bool useHessianResidual{false};
                double seedHitChi2{5.};
                bool recalibSeed{false};
                double goodSegmentCut{2.};
                double outlierRemovalCut{5.};
                double recoveryPull{5.};
                unsigned precHitCut{3};
                bool useFastFitter{true};
                bool fastPreFitter{false};
                bool ignoreFailedPreFit{false};
                unsigned maxIter{50};
                unsigned nFieldSteps{30};
                const ISpacePointCalibrator* calibrator{nullptr};
                const MuonValR4::IPatternVisualizationTool* visionTool{nullptr};
                const Muon::IMuonIdHelperSvc* idHelperSvc{nullptr};
            };

            FastMuonSABuilder(const std::string& name,
                              Config&& config);

            xAOD::Muon* buildMuonCandidate(const EventContext& ctx,
                                           const ActsTrk::GeometryContext& gctx,
                                           const AtlasFieldCacheCondObj& magField,
                                           const GlobalPattern& pattern,
                                           MuonCont_t& outMuons) const;
        private:
            using Hit_t = const SpacePoint*;
            using HitVec_t = std::vector<Hit_t>;
            using Bucket_t = const SpacePointBucket*;
            using Segment_t = std::unique_ptr<Segment>;
            Segment_t fitSegment(const EventContext& ctx,
                                 const Amg::Transform3D& localToGlobal,
                                 Bucket_t parentBucket,
                                 std::vector<Hit_t>&& hits) const;

            HitVec_t estimateBendingPars(HitVec_t&& hits,
                                         Parameters& pars) const;
            Config m_cfg;
            SpacePointPerLayerSorter m_spSorter{};
            std::unique_ptr<LineFitter> m_fitter{};
            std::unique_ptr<LineFitter> m_nswFitter{};
            std::unique_ptr<MdtSegmentSeeder> m_mdtSeeder{};
            std::unique_ptr<MsTrackSeeder> m_trackSeeder{};
    };
}
 
 
#endif