MdtSegmentFitter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 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 <MuonSpacePoint/CalibratedSpacePoint.h>
#include <MuonPatternEvent/SegmentSeed.h>
#include <MuonPatternEvent/SegmentFitterEventData.h>
#include <GaudiKernel/SystemOfUnits.h>
 
 
namespace MuonR4{
    class ISpacePointCalibrator;
    class MdtSegmentFitter: public AthMessaging{
        public:
            using ParamDefs = SegmentFit::ParamDefs;
            using Parameters = SegmentFit::Parameters;
            using HitType = std::unique_ptr<CalibratedSpacePoint>;
            using HitVec = std::vector<HitType>;
 
            struct Config{
                unsigned int nMaxCalls{100};
                double tolerance{1.e-5};
                bool doTimeFit{true};
                unsigned int nParsOutOfBounds{1};
                const ISpacePointCalibrator* calibrator{nullptr};
                unsigned int noMoveIter{2};
                using RangeArray = std::array<std::array<double,2>, SegmentFit::toInt(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{};
 
            using LinePartialArray = std::array<Amg::Vector3D, toInt(ParamDefs::nPars)>;
            void updateLinePartials(const Parameters& fitPars, 
                                    LinePartialArray& linePartials) const;
 
            static Amg::Vector3D partialClosestApproach(const MuonR4::CalibratedSpacePoint& sp,
                                                        const Amg::Vector3D& segPos,
                                                        const Amg::Vector3D& segDir,
                                                        const LinePartialArray& linePartials,
                                                        const ParamDefs fitPar);
 
 
            static Amg::Vector3D partialPlaneIntersect(const Amg::Vector3D& normal,
                                                       const double offset, 
                                                       const Amg::Vector3D& segPos, 
                                                       const Amg::Vector3D& segDir,
                                                       const LinePartialArray& linePartials,
                                                       const ParamDefs fitPar);
            
            using MeasCov_t = CalibratedSpacePoint::Covariance_t;
            void updateDerivatives(const Amg::Vector3D& residual,
                                    const LinePartialArray& residualPartials,
                                    const MeasCov_t& measCovariance,
                                    AmgVector(5)& gradient,
                                    AmgSymMatrix(5)& hessian,
                                    double& chi2,
                                    int startPar) const;
            
 
            enum class UpdateStatus{
                allOkay = 0,
                outOfBounds = 1,
                noChange = 2,
            };
            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