MdtSegmentFitter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONPATTERNHELPERS_MDTSEGMENTFITTER_H
#define MUONPATTERNHELPERS_MDTSEGMENTFITTER_H
 
#include <GeoPrimitives/GeoPrimitives.h>
#include <AthenaBaseComps/AthMessaging.h>
 
#include <MuonPatternEvent/HoughEventData.h>
#include <MuonPatternEvent/SegmentSeed.h>
#include <MuonPatternEvent/SegmentFitterEventData.h>
 
namespace MuonR4{
    class ISpacePointCalibrator;
    class CalibratedSpacePoint;
}
 
namespace MuonR4::SegmentFit{
    class MdtSegmentFitter: public AthMessaging{
        public:
            using HitType = std::unique_ptr<CalibratedSpacePoint>;
            using HitVec = std::vector<HitType>;
 
            struct Config{
                unsigned int nMaxCalls{100};
                double tolerance{1.e-7};
                double angularCalibCutOff{1.*Gaudi::Units::mrad};
                double spatialCalibCuttOff{0.1*Gaudi::Units::mm};
                bool doTimeFit{true};
                bool reCalibrate{false};
                bool useSecOrderDeriv{false};
                bool useFastFit{true};
                unsigned int nParsOutOfBounds{1};
                const ISpacePointCalibrator* calibrator{nullptr};
                unsigned int noMoveIter{2};
                using RangeArray = std::array<std::array<double,2>, Acts::toUnderlying(ParamDefs::nPars)>;
                static RangeArray defaultRanges();
 
                RangeArray ranges{defaultRanges()};
            };
            MdtSegmentFitter(const std::string& name,
                             Config&& config);
            
            SegmentFitResult fitSegment(const EventContext& ctx,
                                        HitVec&& calibHits,
                                        const Parameters& startPars,
                                        const Amg::Transform3D& localToGlobal) const;
        private:
            Config m_cfg{};
            bool recalibrate(const EventContext& ctx,
                             const Line_t& segmentLine,
                             SegmentFitResult& fitResult) const;
            bool updateHitSummary(SegmentFitResult& fitResult) const;
            void centerAlongWire(SegmentFitResult& fitResult) const;
            void updateDriftSigns(const Line_t& segmentLine, SegmentFitResult& fitRes)const;
            enum class UpdateStatus{
                allOkay = 0,     
                smallStep = 1,   
                outOfBounds = 2, 
                noChange = 3,    
            };
            template <unsigned int nDim>
                UpdateStatus updateParameters(Parameters& currentPars,
                                              Parameters& previousPars,
                                              Parameters& currGrad,
                                              Parameters& prevGrad,
                                              const AmgSymMatrix(5)& hessian) const;
            
            template <unsigned int nDim>
                void blockCovariance(const AmgSymMatrix(5)& hessian,                                    
                                     SegmentFit::Covariance&  covariance) const;
 
 
    };
}
 
 
#endif