SpacePointCalibrator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONSPACEPOINTCALIBRATOR_SPACEPOINTCALIBRATOR_H
#define MUONSPACEPOINTCALIBRATOR_SPACEPOINTCALIBRATOR_H
 
#include "MuonRecToolInterfacesR4/ISpacePointCalibrator.h"
 
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "ActsCalibBase/MeasurementCalibratorBase.h"
#include "MuonSpacePoint/SpacePoint.h"
#include "MuonSpacePoint/CalibratedSpacePoint.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
 
#include "MdtCalibInterfaces/IMdtCalibrationTool.h"
#include "ActsGeometryInterfaces/GeometryContext.h"
#include "MuonReadoutGeometryR4/MuonDetectorManager.h"
 
#include "GaudiKernel/PhysicalConstants.h"
#include "NSWCalibTools/INSWCalibTool.h"
#include "MMClusterization/IMMClusterBuilderTool.h"
 
#include "xAODMuonPrepData/sTgcStripClusterFwd.h"
#include "xAODMuonPrepData/CombinedMuonStripFwd.h"
 
 
namespace MuonR4{
    /*** @brief Implementation of the space point calibrator interface */
    class SpacePointCalibrator : public extends<AthAlgTool, ISpacePointCalibrator>,
                                 public ActsTrk::detail::MeasurementCalibratorBase {
        public:
            using base_class::base_class;
 
            StatusCode initialize() override final;
 
            CalibSpacePointPtr calibrate(const EventContext& ctx,
                                         const SpacePoint* spacePoint,
                                         const Amg::Vector3D& seedPosInChamb,
                                         const Amg::Vector3D& seedDirInChamb,
                                         const double timeDelay) const override final;
            
            CalibSpacePointPtr calibrate(const EventContext& ctx,
                                         const CalibratedSpacePoint& spacePoint,
                                         const Amg::Vector3D& seedPosInChamb,
                                         const Amg::Vector3D& seedDirInChamb,
                                         const double timeDelay) const override final;
            
            
            CalibSpacePointVec calibrate(const EventContext& ctx,
                                         const std::vector<const SpacePoint*>& spacePoints,
                                         const Amg::Vector3D& seedPosInChamb,
                                         const Amg::Vector3D& seedDirInChamb,
                                         const double timeDelay) const override final;
 
            CalibSpacePointVec calibrate(const Acts::CalibrationContext& ctx,
                                         const Amg::Vector3D& seedPosInChamb,
                                         const Amg::Vector3D& seedDirInChamb,
                                         const double timeDelay,
                                         const CalibSpacePointVec& spacePoints) const override final;
 
            void updateSigns(const Amg::Vector3D& trackPos,
                             const Amg::Vector3D& trackDir,
                             CalibSpacePointVec& hitsToCalib) const override final;
 
            double driftVelocity(const Acts::CalibrationContext& ctx,
                                 const CalibratedSpacePoint& spacePoint) const override final;
 
            double driftAcceleration(const Acts::CalibrationContext& ctx,
                                     const CalibratedSpacePoint& spacePoint) const override final;
 
            void calibrateSourceLink(const Acts::GeometryContext& geoctx,
                                     const Acts::CalibrationContext& cctx,
                                     const Acts::SourceLink& link,
                                     ActsTrk::MutableTrackContainer::TrackStateProxy state) const override final;
 
            void stampSignsOnMeasurements(const xAOD::MuonSegment& segment) const override final;
        private:
            void calibrateCombinedPrd(const EventContext& ctx, 
                                      const ActsTrk::GeometryContext& gctx,
                                      const xAOD::CombinedMuonStrip* combinedPrd,
                                      ActsTrk::MutableTrackContainer::TrackStateProxy state) const;
            std::pair<double, double>  calibrateMM(const EventContext& ctx, 
                                                   const ActsTrk::GeometryContext& gctx, 
                                                   const xAOD::MMCluster& cluster,
                                                   const Amg::Vector3D& globalPos, 
                                                   const Amg::Vector3D& globalDir) const;
            std::pair<double, double>  calibratesTGC(const EventContext& ctx, 
                                                     const ActsTrk::GeometryContext& gctx, 
                                                     const xAOD::sTgcStripCluster& cluster,
                                                     std::optional<double> posAlongTheStrip, 
                                                     const Amg::Vector3D& globalPos, 
                                                     const Amg::Vector3D& globalDir) const;
 
            SG::ReadHandleKey<ActsTrk::GeometryContext> m_geoCtxKey{this, "AlignmentKey", "ActsAlignment", "cond handle key"}; 
 
            ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
 
            ToolHandle<IMdtCalibrationTool> m_mdtCalibrationTool{this, "MdtCalibrationTool", ""};
 
            ToolHandle<Muon::INSWCalibTool> m_nswCalibTool{this, "NSWCalibTool", ""};
 
            ToolHandle<Muon::IMMClusterBuilderTool> m_clusterBuilderToolMM{this, "MMClusterBuilder", ""};
 
            const MuonGMR4::MuonDetectorManager* m_detMgr{nullptr};
 
            Gaudi::Property<double> m_muonPropSpeed{this, "PropagationSpeed", 1./ Gaudi::Units::c_light };
 
            Gaudi::Property<double> m_rpcSignalVelocity {this, "rpcSignalVelocity", 0.5 * Gaudi::Units::c_light,
                                                         "Propagation speed of the signal inside the rpc strip"}; 
            /*** Resolution of the rpc time measurement  */
            Gaudi::Property<double> m_rpcTimeResolution{this, "rpcTimeResolution", 0.6 * Gaudi::Units::nanosecond,
                                                          "Estimated time resolution of the strip readout"};
            
            Gaudi::Property<bool> m_useRpcTime{this, "useRpcTime", false};
            Gaudi::Property<bool> m_useTgcTime{this, "useTgcTime", false,
                                               "Load the Tgc BC-ID on the track states for the fit"};
            Gaudi::Property<bool> m_usesTgcTime{this, "usesTgcTime", false,
                                               "Load the sTgc time on the track states for the fit"};
            Gaudi::Property<bool> m_MdtSignFromSegment{this, "useSegmentSigns", true,
                                    "Mdt drift signs are copied from the segment line instead from the track state"};
    };
 
}
#endif