TruthCalibrator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONSPACEPOINTCALIBRATOR_TRUTHCALIBRATOR_H
#define MUONSPACEPOINTCALIBRATOR_TRUTHCALIBRATOR_H
 
#include "MuonRecToolInterfacesR4/ISpacePointCalibrator.h"
 
#include "AthenaBaseComps/AthAlgTool.h"
 
#include "StoreGate/ReadDecorHandleKeyArray.h"
#include "xAODMeasurementBase/UncalibratedMeasurementContainer.h"
#include "ActsEvent/ContextUtility.h"
#include "ActsCalibBase/MeasurementCalibratorBase.h"
 
 
namespace MuonR4 {
    class TruthCalibrator : public extends<AthAlgTool, ISpacePointCalibrator>,
                            public ActsTrk::detail::MeasurementCalibratorBase {
        public:
            using base_class::base_class;
            virtual StatusCode initialize() override final;
            virtual CalibSpacePointPtr calibrate(const EventContext& ctx,
                                                 const SpacePoint* spacePoint,
                                                 const Amg::Vector3D& seedPosInChamb,
                                                 const Amg::Vector3D& seedDirInChamb,
                                                 const double timeDelay) const override final;
            virtual CalibSpacePointPtr calibrate(const EventContext& ctx,
                                                 const CalibratedSpacePoint& spacePoint,
                                                 const Amg::Vector3D& seedPosInChamb,
                                                 const Amg::Vector3D& seedDirInChamb,
                                                 const double timeDelay) const override final;
            virtual 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;
    
            virtual CalibSpacePointVec calibrate(const Acts::CalibrationContext& cctx,                                            
                                                  const Amg::Vector3D& seedPosInChamb,
                                                  const Amg::Vector3D& seedDirInChamb,
                                                  const double timeDelay,
                                                  const CalibSpacePointVec& spacePoints) const override final;

            virtual double driftVelocity(const Acts::CalibrationContext& cctx,
                                         const CalibratedSpacePoint& spacePoint) const override final;
            virtual void calibrateSourceLink(const Acts::GeometryContext& geoctx,
                                             const Acts::CalibrationContext& cctx,
                                             const Acts::SourceLink& link,
                                             ActsTrk::MutableTrackStateBackend::TrackStateProxy state) const override final;
            virtual void updateSigns(const Amg::Vector3D& trackPos,
                                     const Amg::Vector3D& trackDir,
                                     CalibSpacePointVec& hitsToCalib) const override final;
            virtual void stampSignsOnMeasurements(const xAOD::MuonSegment& segment) const override final;
            virtual double driftRadius(const Acts::CalibrationContext& cctx,
                                       const CalibratedSpacePoint& spacePoint, 
                                       const double timeDelay) const override final;
            virtual double driftVelocity(const Acts::CalibrationContext& cctx,
                                         const CalibratedSpacePoint& spacePoint, 
                                         const double timeDelay) const override final;
            virtual double driftAcceleration(const Acts::CalibrationContext& cctx,
                                             const CalibratedSpacePoint& spacePoint, 
                                             const double timeDelay) const override final;
            virtual double driftAcceleration(const Acts::CalibrationContext& cctx,
                                             const CalibratedSpacePoint& spacePoint) const override final;
        private:
            ActsTrk::ContextUtility m_ctxProvider{this};
            SG::ReadDecorHandleKeyArray<xAOD::UncalibratedMeasurementContainer> m_truthLinks{this, "truthLinks", {}};
            Gaudi::Property<std::vector<std::string>> m_prdContainers{this, "prdContainers", {}};
            Gaudi::Property<std::string> m_simLinkDecor{this, "simHitLinkDecor", "simHitLink"};
 
 
    };
}
 
#endif