MsTrackSeeder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONTRACKFINDINGTOOLS_MSTRACKSEEDER_H
#define MUONTRACKFINDINGTOOLS_MSTRACKSEEDER_H
 
#include "AthenaBaseComps/AthMessaging.h"
 
#include "GaudiKernel/SystemOfUnits.h"
#include "Acts/Utilities/KDTree.hpp"
#include "xAODMuon/MuonSegmentContainer.h"
#include "MuonTrackEvent/MsTrackSeed.h"
#include "MuonRecToolInterfacesR4/ISegmentSelectionTool.h"
 
 
namespace MuonR4{
    class MsTrackSeeder: public AthMessaging {
        public:
 
            struct Config{
                double barrelRadius{7.*Gaudi::Units::m};
                double barrelLength{25.*Gaudi::Units::m};
                double endcapDiscZ{15.*Gaudi::Units::m};
                double endcapDiscRadius{12.*Gaudi::Units::m};
                double seedHalfLength{25.*Gaudi::Units::cm};
                const ISegmentSelectionTool* selector{nullptr};
            };
            using SearchTree_t = Acts::KDTree<3, const xAOD::MuonSegment*, double, std::array, 6>;
            using Location = MsTrackSeed::Location;
            enum SeedCoords{
                eDetSection,
                eSector,
                ePosOnCylinder
            };
 
            MsTrackSeeder(const std::string& msgName, Config&& cfg);   
            SearchTree_t constructTree(const xAOD::MuonSegmentContainer& segments) const;
            Amg::Vector2D expressOnCylinder(const xAOD::MuonSegment& segment,
                                            const Location loc) const;
            
            bool withinBounds(const Amg::Vector2D& projPos,
                              const Location loc) const;
            std::unique_ptr<MsTrackSeedContainer> findTrackSeeds(const EventContext& ctx,
                                                                 const xAOD::MuonSegmentContainer& segments) const;
        private:
            using TreeRawVec_t = SearchTree_t::vector_t;
            void appendSegment(const xAOD::MuonSegment* segment,
                               const Location loc,
                               TreeRawVec_t& outContainer) const;
            std::unique_ptr<MsTrackSeedContainer> resolveOverlaps(MsTrackSeedContainer&& unresolved) const;
 
            Config m_cfg{};
    };
}
 
 
#endif