MuonSeededSegmentFinder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef MUON_MUONSEEDEDSEGMENTFINDER_H
#define MUON_MUONSEEDEDSEGMENTFINDER_H
 
#include <set>
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "MuonIdHelpers/IMuonIdHelperSvc.h"
#include "MuonPrepRawData/MdtPrepDataCollection.h"
#include "MuonPrepRawData/MuonPrepDataContainer.h"
#include "MuonReadoutGeometry/MuonDetectorManager.h"
#include "MuonRecHelperTools/MuonEDMPrinterTool.h"
#include "MuonRecToolInterfaces/IMdtDriftCircleOnTrackCreator.h"
#include "MuonRecToolInterfaces/IMuonSeededSegmentFinder.h"
#include "MuonRecToolInterfaces/IMuonSegmentMaker.h"
#include "TrkExInterfaces/IPropagator.h"
#include "TrkGeometry/MagneticFieldMode.h"
#include "TrkGeometry/MagneticFieldProperties.h"
#include "TrkParameters/TrackParameters.h"
 
namespace Muon {

    class MuonSeededSegmentFinder : public extends<AthAlgTool,  IMuonSeededSegmentFinder> {
    public:
        MuonSeededSegmentFinder(const std::string&, const std::string&, const IInterface*);

        ~MuonSeededSegmentFinder() = default;

        StatusCode initialize();

        std::unique_ptr<Trk::SegmentCollection> find(const EventContext& ctx, const Trk::TrackParameters& pars,
                                                     const std::set<Identifier>& chIds) const;

        std::unique_ptr<Trk::SegmentCollection> find(const EventContext& ctx, const Trk::TrackParameters& pars,
                                                     const std::set<IdentifierHash>& chIdHs) const;

        std::unique_ptr<Trk::SegmentCollection> find(const EventContext& ctx, const Trk::TrackParameters& pars,
                                                     const std::vector<const MdtPrepData*>& mdtPrds) const;

        void extractMdtPrdCols(const EventContext& ctx, const std::set<IdentifierHash>& chIdHs,
                               std::vector<const MdtPrepDataCollection*>& target) const;

        void extractRpcPrdCols(const EventContext& ctx, const std::set<IdentifierHash>& chIdHs,
                               std::vector<const RpcPrepDataCollection*>& target) const;

        void extractTgcPrdCols(const EventContext& ctx, const std::set<IdentifierHash>& chIdHs,
                               std::vector<const TgcPrepDataCollection*>& target) const;

        void extractCscPrdCols(const std::set<IdentifierHash>& chIdHs, std::vector<const CscPrepDataCollection*>& target) const;
 
        // New Small Wheel

        void extractsTgcPrdCols(const EventContext& ctx, const std::set<IdentifierHash>& chIdHs,
                                std::vector<const sTgcPrepDataCollection*>& target) const;

        void extractMMPrdCols(const std::set<IdentifierHash>& chIdHs, std::vector<const MMPrepDataCollection*>& target) const;
 
    private:
        std::vector<const MdtPrepData*> extractPrds(const EventContext& ctx, const std::set<Identifier>& chIds) const;

        std::vector<const MdtPrepData*> extractPrds(const EventContext& ctx, const std::set<IdentifierHash>& chIdHs) const;

        void selectAndCalibrate(const EventContext& ctx, const Trk::TrackParameters& pars,
                                const std::vector<const MdtPrepData*>& mdtPrdCols, std::vector<const MdtDriftCircleOnTrack*>& mdtROTs,
                                bool& doHoleSearch) const;

        const MdtDriftCircleOnTrack* handleMdtPrd(const EventContext& ctx, const Trk::TrackParameters& pars, const MdtPrepData& mdtPrd,
                                                  bool& doHoleSearch) const;
 
        SG::ReadCondHandleKey<MuonGM::MuonDetectorManager> m_DetectorManagerKey{this, "DetectorManagerKey", "MuonDetectorManager",
                                                                                "Key of input MuonDetectorManager condition data"};
 
        ToolHandle<Muon::IMuonSegmentMaker> m_segMaker{
            this, "SegmentMaker", "Muon::DCMathSegmentMaker/DCMathSegmentMaker"};  
        ToolHandle<Muon::IMuonSegmentMaker> m_segMakerNoHoles{
            this, "SegmentMakerNoHoles", "Muon::DCMathSegmentMaker/DCMathSegmentMaker"};  
        ToolHandle<Trk::IPropagator> m_propagator{this, "Propagator",
                                                  "Trk::RungeKuttaPropagator/AtlasRungeKuttaPropagator"};  
        ToolHandle<Muon::IMdtDriftCircleOnTrackCreator> m_mdtRotCreator{
            this, "MdtRotCreator", "Muon::MdtDriftCircleOnTrackCreator/MdtDriftCircleOnTrackCreator"};  
        Trk::MagneticFieldProperties m_magFieldProperties;                                              
        ServiceHandle<Muon::IMuonIdHelperSvc> m_idHelperSvc{this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};
        PublicToolHandle<Muon::MuonEDMPrinterTool> m_printer{this, "Printer",
                                                       "Muon::MuonEDMPrinterTool/MuonEDMPrinterTool"};  
 
        SG::ReadHandleKey<Muon::MdtPrepDataContainer> m_key_mdt{this, "MdtPrepDataContainer", "MDT_DriftCircles", "MDT PRDs"};
        SG::ReadHandleKey<Muon::CscPrepDataContainer> m_key_csc{this, "CscPrepDataContainer", "CSC_Clusters", "CSC PRDS"};
        SG::ReadHandleKey<Muon::TgcPrepDataContainer> m_key_tgc{this, "TgcPrepDataContainer", "TGC_Measurements", "TGC PRDs"};
        SG::ReadHandleKey<Muon::RpcPrepDataContainer> m_key_rpc{this, "RpcPrepDataContainer", "RPC_Measurements", "RPC PRDs"};
        SG::ReadHandleKey<Muon::sTgcPrepDataContainer> m_key_stgc{this, "sTgcPrepDataContainer", "STGC_Measurements", "sTGC PRDs"};
        SG::ReadHandleKey<Muon::MMPrepDataContainer> m_key_mm{this, "MMPrepDataContainer", "MM_Measurements", "MM PRDs"};
 
        Gaudi::Property<double> m_adcCut{this, "AdcCut", 50};
        Gaudi::Property<double> m_maxSigma{this, "MdtSigmaFromTrack", 3};
    };
 
}  // namespace Muon
 
#endif