MuonStauRecoTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUON_MUONSTAURECOTOOL_H
#define MUON_MUONSTAURECOTOOL_H
 
#include <iostream>
#include <string>
#include <vector>
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/PhysicalConstants.h"
#include "MdtCalibData/MdtCalibDataContainer.h"
#include "MuidInterfaces/ICombinedMuonTrackBuilder.h"
#include "MuonCombinedEvent/MuGirlLowBetaTag.h"
#include "MuonCombinedToolInterfaces/IMuonCombinedInDetExtensionTool.h"
#include "MuonCombinedToolInterfaces/IMuonLayerSegmentMatchingTool.h"
#include "MuonDetDescrUtils/MuonSectorMapping.h"
#include "MuonHoughPatternTools/MuonLayerHoughTool.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "MuonInsideOutRecoTool.h"
#include "MuonLayerEvent/MuonCandidate.h"
#include "MuonLayerEvent/MuonLayerRecoData.h"
#include "MuonLayerEvent/MuonSystemExtension.h"
#include "MuonLayerHough/MuonLayerHough.h"
#include "MuonRIO_OnTrack/MuonClusterOnTrack.h"
#include "MuonRecHelperTools/IMuonEDMHelperSvc.h"
#include "MuonRecHelperTools/MuonEDMPrinterTool.h"
#include "MuonRecHelperTools/TimePointBetaFitter.h"
#include "MuonRecToolInterfaces/IMdtDriftCircleOnTrackCreator.h"
#include "MuonRecToolInterfaces/IMuonHitTimingTool.h"
#include "MuonRecToolInterfaces/IMuonPRDSelectionTool.h"
#include "MuonRecToolInterfaces/IMuonRecoValidationTool.h"
#include "MuonRecToolInterfaces/IMuonSegmentMaker.h"
#include "TrkToolInterfaces/ITrackAmbiguityProcessorTool.h"
#include "TrkToolInterfaces/IUpdator.h"
#include "TrkTrack/Track.h"
 
namespace Muon {
    class RpcClusterOnTrack;
    struct RpcClusterObj;
}
 
namespace MuonCombined {
    class MuonStauRecoTool : virtual public MuonCombined::IMuonCombinedInDetExtensionTool, public AthAlgTool {
    public:
        typedef std::vector<const MuonHough::MuonLayerHough::Maximum*> MaximumVec;
        typedef std::vector<const MuonHough::MuonPhiLayerHough::Maximum*> PhiMaximumVec;
 
        struct BetaSeed {
            BetaSeed(float beta_, float error_) : beta(beta_), error(error_) {}
            float beta{0.f};
            float error{0.f};
        };
 
        struct RpcTimeMeasurement {
            std::vector<std::shared_ptr<const Muon::RpcClusterOnTrack>> rpcClusters;
            float time{0.f};
            float error{0.f};
        };
        typedef std::vector<RpcTimeMeasurement> RpcTimeMeasurementVec;
 
        struct MaximumData {
            MaximumData(const Muon::MuonSystemExtension::Intersection& intersection_, 
                        const MuonHough::MuonLayerHough::Maximum* maximum_,
                        const std::vector<std::shared_ptr<const Muon::MuonClusterOnTrack>> & phiClusterOnTracks_) :
                intersection(intersection_), maximum(maximum_), phiClusterOnTracks(phiClusterOnTracks_) {}
            Muon::MuonSystemExtension::Intersection intersection;
            const MuonHough::MuonLayerHough::Maximum* maximum;
            std::vector<std::shared_ptr<const Muon::MuonClusterOnTrack>> phiClusterOnTracks;
 
            std::vector<std::shared_ptr<const Muon::MuonSegment>> t0fittedSegments;
            RpcTimeMeasurementVec rpcTimeMeasurements;
            std::vector<BetaSeed> betaSeeds;
        };
        using MaximumDataVec = std::vector<std::shared_ptr<MaximumData>>;
        struct LayerData {
            LayerData(const Muon::MuonSystemExtension::Intersection& intersection_) : intersection(intersection_) {}
            Muon::MuonSystemExtension::Intersection intersection;
            MaximumDataVec maximumDataVec;
        };
        typedef std::vector<LayerData> LayerDataVec;
 
        struct AssociatedData {
            LayerDataVec layerData;
            PhiMaximumVec phiMaxima;
        };
 
        struct Candidate {
            Candidate(const BetaSeed& betaSeed_) : betaSeed(betaSeed_) {}
 
            // inital parameters
            BetaSeed betaSeed;
 
            // information filled by createCandidates: associated layers with MaximumData, time measurements and final beta fit result
            LayerDataVec layerDataVec;
            Muon::TimePointBetaFitter::HitVec hits;
            Muon::TimePointBetaFitter::FitResult betaFitResult;
 
            // information filled by refineCandidates: segments found using the beta of the final fit
            std::vector<Muon::MuonLayerRecoData> allLayers;
 
            // information filled by the final tracking: MuonCandidate and the track
            std::unique_ptr<const Muon::MuonCandidate> muonCandidate;
            std::unique_ptr<Trk::Track> combinedTrack;
            MuGirlNS::StauHits stauHits;
            MuGirlNS::StauMDTHitExtras stauMDTHitExtras;
            Muon::TimePointBetaFitter::FitResult finalBetaFitResult;
        };
        typedef std::vector<std::shared_ptr<Candidate>> CandidateVec;
 
        struct TruthInfo {
            TruthInfo(int pdgId_, float mass_, float beta_) : pdgId(pdgId_), mass(mass_), beta(beta_) {}
            int pdgId{0};
            float mass{0.f};
            float beta{0.f};
            std::string toString() const {
                std::ostringstream sout;
                sout << " pdgId " << pdgId << " mass " << mass << " beta " << beta;
                return sout.str();
            }
        };
 
        MuonStauRecoTool(const std::string& type, const std::string& name, const IInterface* parent);
        virtual ~MuonStauRecoTool() = default;
        virtual StatusCode initialize() override;
 
        virtual void extend(const InDetCandidateCollection& inDetCandidates, InDetCandidateToTagMap* tagMap, TrackCollection* combTracks,
                            TrackCollection* meTracks, Trk::SegmentCollection* segments, const EventContext& ctx) const override;
 
        virtual void extendWithPRDs(const InDetCandidateCollection& inDetCandidates, InDetCandidateToTagMap* tagMap,
                                    IMuonCombinedInDetExtensionTool::MuonPrdData prdData, TrackCollection* combTracks,
                                    TrackCollection* meTracks, Trk::SegmentCollection* segments, const EventContext& ctx) const override;
 
    private:
        void handleCandidate(const EventContext& ctx, const InDetCandidate& inDetCandidate, InDetCandidateToTagMap* tagMap,
                             TrackCollection* combTracks, Trk::SegmentCollection* segments) const;
 
        void associateHoughMaxima(const EventContext& ctx, LayerData& layerData) const;
 
        void extractRpcTimingFromMaximum(const Muon::MuonSystemExtension::Intersection& intersection, MaximumData& maximumData) const;
 
        void createRpcTimeMeasurementsFromClusters(const Muon::MuonSystemExtension::Intersection& intersection,
                                                   const std::vector<Muon::RpcClusterObj>& clusterObjects,
                                                   RpcTimeMeasurementVec& rpcTimeMeasurements) const;
 
        void findSegments(const Muon::MuonSystemExtension::Intersection& intersection, MaximumData& maximumData,
                          std::vector<std::shared_ptr<const Muon::MuonSegment>>& t0fittedSegments,
                          const ToolHandle<Muon::IMuonPRDSelectionTool>& muonPRDSelectionTool,
                          const ToolHandle<Muon::IMuonSegmentMaker>& segmentMaker,
                          float beta = 1.) const;
 
        bool extractTimeMeasurements(const EventContext& ctx, const Muon::MuonSystemExtension& muonSystemExtension,
                                     AssociatedData& associatedData) const;
 
        bool createCandidates(const AssociatedData& associatedData, CandidateVec& candidates) const;
 
        void extendCandidates(CandidateVec& candidates, std::set<const MaximumData*>& usedMaximumData, LayerDataVec::const_iterator it,
                              LayerDataVec::const_iterator it_end) const;
 
        void getBetaSeeds(MaximumData& maximumData) const;
 
        bool extractTimeHits(const MaximumData& maximumData, Muon::TimePointBetaFitter::HitVec& hits, const BetaSeed* seed = 0) const;
 
        bool refineCandidates(const EventContext& ctx, CandidateVec& candidates) const;
 
        bool combineCandidates(const EventContext& ctx, const xAOD::TrackParticle& indetTrackParticle, CandidateVec& candidates) const;
 
        bool resolveAmbiguities(CandidateVec& candidates) const;
 
        void addTag(const InDetCandidate& inDetCandidate, Candidate& candidate, InDetCandidateToTagMap* tagMap, TrackCollection* combTracks,
                    Trk::SegmentCollection* segments) const;
 
        void extractTimeMeasurementsFromTrack(const EventContext& ctx, Candidate& candidate) const;
 
        std::unique_ptr<TruthInfo> getTruth(const xAOD::TrackParticle& indetTrackParticle) const;
 
        bool selectTruth(const TruthInfo* truthInfo) const {
            if (!m_useTruthMatching) return true;
            if (truthInfo && m_selectedPdgs.count(truthInfo->pdgId)) return true;
            return false;
        }
 
        void addCandidatesToNtuple(const xAOD::TrackParticle& indetTrackParticle, const CandidateVec& candidates, int stage) const;
 
        bool processMuonSystemExtension(const xAOD::TrackParticle& indetTrackParticle, const Muon::MuonSystemExtension& muonSystemExtension,
                                        CandidateVec& candidates);
 
        void mdtTimeCalibration(const Identifier& id, float& time, float& error) const;
        void rpcTimeCalibration(const Identifier& id, float& time, float& error) const;
        void segmentTimeCalibration(const Identifier& id, float& time, float& error) const;
 
        float calculateTof(const float beta, const float dist) const;
        float calculateBeta(const float time, const float dist) const;
 
        SG::ReadHandleKey<Muon::MuonLayerHoughTool::HoughDataPerSectorVec> m_houghDataPerSectorVecKey{
            this, "Key_MuonLayerHoughToolHoughDataPerSectorVec", "HoughDataPerSectorVec", "HoughDataPerSectorVec key"};
 
        ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
        ServiceHandle<Muon::IMuonEDMHelperSvc> m_edmHelperSvc{this, "edmHelper", "Muon::MuonEDMHelperSvc/MuonEDMHelperSvc",
                                                              "Handle to the service providing the IMuonEDMHelperSvc interface"};
 
        PublicToolHandle<Muon::MuonEDMPrinterTool> m_printer{this, "MuonEDMPrinterTool", ""};
        ToolHandle<Muon::IMuonSegmentMaker> m_segmentMaker{this, "MuonSegmentMaker", "Muon::DCMathSegmentMaker/DCMathSegmentMaker"};
        ToolHandle<Muon::IMuonSegmentMaker> m_segmentMakerT0Fit{this, "MuonSegmentMakerT0Fit",
                                                                "Muon::DCMathSegmentMaker/DCMathT0FitSegmentMaker"};
        ToolHandle<Muon::IMuonLayerSegmentMatchingTool> m_segmentMatchingTool{
            this, "MuonLayerSegmentMatchingTool", "Muon::MuonLayerSegmentMatchingTool/MuonLayerSegmentMatchingTool"};
 
        ToolHandle<Muon::IMuonRecoValidationTool> m_recoValidationTool{this, "MuonRecoValidationTool", ""};
        ToolHandle<Trk::ITrackAmbiguityProcessorTool> m_trackAmbibuityResolver{this, "TrackAmbiguityProcessor",
                                                                               "Trk::TrackSelectionProcessorTool/MuonAmbiProcessor"};
        ToolHandle<Muon::IMuonHitTimingTool> m_hitTimingTool{this, "MuonHitTimingTool", "Muon::MuonHitTimingTool/MuonHitTimingTool"};
        ToolHandle<Muon::IMuonPRDSelectionTool> m_muonPRDSelectionTool{this, "MuonPRDSelectionTool", ""};
        ToolHandle<Muon::IMuonPRDSelectionTool> m_muonPRDSelectionToolStau{this, "MuonPRDSelectionToolStau", ""};
        ToolHandle<Muon::IMdtDriftCircleOnTrackCreator> m_mdtCreator{this, "MdtDriftCircleOnTrackCreator",
                                                                     "Muon::MdtDriftCircleOnTrackCreator/MdtDriftCircleOnTrackCreator"};
        ToolHandle<Muon::IMdtDriftCircleOnTrackCreator> m_mdtCreatorStau{
            this, "MdtDriftCircleOnTrackCreatorStau", "Muon::MdtDriftCircleOnTrackCreator/MdtDriftCircleOnTrackCreatorStau"};
        ToolHandle<MuonCombined::MuonInsideOutRecoTool> m_insideOutRecoTool{this, "MuonInsideOutRecoTool",
                                                                            "MuonCombined::MuonInsideOutRecoTool/MuonInsideOutRecoTool"};
        ToolHandle<Trk::IUpdator> m_updator{this, "Updator", "Trk::KalmanUpdator/KalmanUpdator"};
        SG::ReadCondHandleKey<MuonCalib::MdtCalibDataContainer> m_calibDbKey{this, "CalibDataKey", "MdtCalibConstants",
                                                                       "Conditions object containing the calibrations"};
 
        Muon::MuonSectorMapping m_muonSectorMapping;
 
        Gaudi::Property<bool> m_doSummary{this, "DoSummary", false, "enable summary output"};
        Gaudi::Property<bool> m_useTruthMatching{this, "UseTruthMatching", false, "enable usage of truth info for reconstruction"};
        Gaudi::Property<bool> m_doTruth{this, "DoTruth", false, "enable truth matching"};
        Gaudi::Property<bool> m_segmentMDTT{this, "UseSegmentMDTT", true};
        Gaudi::Property<bool> m_ignoreSiAssocated{this, "IgnoreSiAssociatedCandidates", true};
        Gaudi::Property<std::vector<int>> m_pdgsToBeConsidered{this, "ConsideredPDGs", {}, "PDG IDs considered in truth matching"};
        Gaudi::Property<double> m_ptThreshold{this, "PtThreshold", 10000};
        Gaudi::Property<double> m_houghAssociationPullCut{this, "HoughAssociationPullCut", 5};
        Gaudi::Property<double> m_mdttBetaAssociationCut{this, "MDTTAssocationCut", 0.4};
        Gaudi::Property<double> m_rpcBetaAssociationCut{this, "RPCAssocationCut", 0.2};
        Gaudi::Property<double> m_segmentBetaAssociationCut{this, "SegmentAssocationCut", 0.2};
        Gaudi::Property<bool> m_addMDTExtrasMuGirlLowBeta{this, "AddMDTExtrasMuGirlLowBeta", false};
 
        std::set<int> m_selectedPdgs;  // set storing particle PDG's considered for matching
 
    };
 
}  // namespace MuonCombined
 
#endif