SegmentActsRefitAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONPATTERNRECOGNITIONALGS_SEGMENTACTSREFITALG_H
#define MUONPATTERNRECOGNITIONALGS_SEGMENTACTSREFITALG_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "AthenaKernel/IAthRNGSvc.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
 
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
#include "StoreGate/WriteDecorHandleKey.h"
 
 
#include "ActsGeometryInterfaces/ITrackingGeometrySvc.h"
#include "ActsGeometryInterfaces/IExtrapolationTool.h"
#include "MuonRecToolInterfacesR4/ISegmentSelectionTool.h"
#include "MuonRecToolInterfacesR4/ISpacePointCalibrator.h"
 
#include "xAODMuon/MuonSegmentContainer.h"
#include "MuonPatternEvent/MuonPatternContainer.h"
 
#include "ActsEvent/TrackContainer.h"
#include "ActsEvent/AuxiliaryMeasurementHandler.h"
 
#include "ActsCalibrators/xAODUncalibMeasSurfAcc.h"
#include "ActsCalibrators/xAODUncalibMeasCalibrator.h"
 
 
#include "Acts/Propagator/Navigator.hpp"
#include "Acts/Propagator/Propagator.hpp"
#include "Acts/Propagator/StraightLineStepper.hpp"
#include "Acts/TrackFitting/GlobalChiSquareFitter.hpp"
 
#include "ActsEvent/ContextUtility.h"
namespace CLHEP{
    class HepRandomEngine;
}
namespace MuonR4{
    class SegmentActsRefitAlg: public AthReentrantAlgorithm {
        public:
          using AthReentrantAlgorithm::AthReentrantAlgorithm;
          virtual StatusCode initialize() override final;
          virtual StatusCode execute(const EventContext& ctx) const override final;
 

          using Propagator_t = Acts::Propagator<Acts::StraightLineStepper, Acts::Navigator>;
          using Fitter_t = Acts::Experimental::Gx2Fitter<Propagator_t, ActsTrk::MutableTrackStateBackend>;

          using Gx2FitterOptions_t = Acts::Experimental::Gx2FitterOptions<ActsTrk::MutableTrackStateBackend>;
          using Gx2FitterExtension_t = Acts::Experimental::Gx2FitterExtensions<ActsTrk::MutableTrackStateBackend>;
 
        private:
            const Acts::Surface* portalSurface(const xAOD::UncalibratedMeasurement* measurement,
                                               bool entrance) const;
            std::tuple<Amg::Vector3D, Amg::Vector3D> smearSegment(const Acts::GeometryContext& gctx,
                                                                  const MuonR4::Segment& segment,
                                                                  CLHEP::HepRandomEngine* engine) const;
            SG::ReadHandleKey<xAOD::MuonSegmentContainer> m_readKey{this, "SegmentContainer", "MuonSegmentsFromR4"};
            SG::WriteHandleKey<xAOD::MuonSegmentContainer> m_writeKey{this, "OutContainer", "ActsRefitSegments"};
            SG::WriteDecorHandleKey<xAOD::MuonSegmentContainer> m_linkKey{this, "LinkKey", m_writeKey, "prefitSegmentLink"};
            SG::WriteDecorHandleKey<xAOD::MuonSegmentContainer> m_localParsKey{this, "LocalParsKey", m_writeKey, "localSegPars"};
            SG::WriteDecorHandleKey<xAOD::MuonSegmentContainer> m_seedParsKey{this, "SeedParsKey", m_readKey, "seedSegPars"};
            
            ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "IdHelperSvc",  "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
           ServiceHandle<ActsTrk::ITrackingGeometrySvc> m_trackingGeometrySvc{this, "TrackingGeometrySvc", "ActsTrackingGeometrySvc"};
            ActsTrk::ContextUtility m_ctxProvider{this};
            ToolHandle<MuonR4::ISegmentSelectionTool> m_segSelector{this, "SegmentSelectionTool" , "" };
            ServiceHandle<IAthRNGSvc> m_rndmSvc{this, "RndmSvc", "AthRNGSvc", ""};
            Gaudi::Property<double> m_smearRange{this, "SmearRange", 1.};
            ToolHandle<ISpacePointCalibrator> m_calibTool{this, "Calibrator", "" };
            Gaudi::Property<bool> m_drawEvent{this , "drawEvent", false };
              
            ActsTrk::AuxiliaryMeasurementHandler m_auxMeasProv{this};
            const MuonGMR4::MuonDetectorManager* m_detMgr{nullptr};
 
            Gaudi::Property<bool> m_smearSegPars{this, "smearSegPars", true};
            Gaudi::Property<unsigned> m_maxPropSteps{this,"maxPropagationSteps", 100000};
            Gaudi::Property<unsigned> m_maxTargetSurfSkip{this, "maxTargetSurfSkip", 100000};
            Gaudi::Property<unsigned> m_maxIter{this, "maxIter", 50};
            Gaudi::Property<bool> m_doJacobianCorr{this,"freeToBoundJacobian", true};

            ActsTrk::detail::xAODUncalibMeasSurfAcc m_surfAccessor{};
            Gx2FitterExtension_t m_fitExtension{};
 
            std::unique_ptr<Fitter_t> m_fitter{};
 
 
 
    };
}
 
#endif