FPGATrackSimGNNRoadMakerTool.h

Go to the documentation of this file.

// Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
 
#ifndef FPGATrackSimGNNRoadMakerTool_H
#define FPGATrackSimGNNRoadMakerTool_H
 
#include "GaudiKernel/ServiceHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
 
#include "FPGATrackSimMaps/IFPGATrackSimMappingSvc.h"
#include "FPGATrackSimObjects/FPGATrackSimRoad.h"
#include "FPGATrackSimObjects/FPGATrackSimGNNHit.h"
#include "FPGATrackSimObjects/FPGATrackSimGNNEdge.h"
#include <boost/graph/adjacency_list.hpp>
#include <boost/graph/connected_components.hpp>
 
#include <memory>
#include <vector>
#include <map>
#include <set>
 
class FPGATrackSimGNNRoadMakerTool : public AthAlgTool
{
    public:
 
        // AthAlgTool
 
        FPGATrackSimGNNRoadMakerTool(const std::string&, const std::string&, const IInterface*);
 
        virtual StatusCode initialize() override;
 
        // Functions
 
        virtual StatusCode makeRoads(const std::vector<std::shared_ptr<const FPGATrackSimHit>> & hits, 
                                     const std::vector<std::shared_ptr<FPGATrackSimGNNHit>> & gnn_hits, 
                                     const std::vector<std::shared_ptr<FPGATrackSimGNNEdge>> & edges, 
                                     std::vector<std::shared_ptr<const FPGATrackSimRoad>> & roads);
 
    private:
        
        // Handles
        
        ServiceHandle<IFPGATrackSimMappingSvc> m_FPGATrackSimMapping {this, "FPGATrackSimMappingSvc", "FPGATrackSimMappingSvc"};
        
        // Properties
 
        Gaudi::Property<float> m_edgeScoreCut { this, "edgeScoreCut", 0.0, "Cut value for edge scores to pass for road making algorithm" };
        Gaudi::Property<std::string> m_roadMakerTool { this, "roadMakerTool", "", "Algorithm to perform graph segmentation into roads"};
 
        // Convenience
 
        int m_num_nodes = 0;
        std::vector<int> m_pass_edge_index_1{};
        std::vector<int> m_pass_edge_index_2{};
        unsigned m_nLayers = 0;
        std::set<int> m_unique_nodes{};
        std::map<int, int> m_node_index_map{};
        std::vector<int> m_unique_indices{};
        typedef boost::adjacency_list<boost::vecS, boost::vecS, boost::undirectedS> m_Graph;
        std::vector<int> m_component{};
        int m_num_components = 0;
        std::vector<int> m_labels{};
        std::vector<std::vector<int>> m_road_hit_list{};
 
        // Helpers
 
        void doScoreCut(const std::vector<std::shared_ptr<FPGATrackSimGNNEdge>> & edges);
        void doConnectedComponents();
        void addRoads(const std::vector<std::shared_ptr<const FPGATrackSimHit>> & hits, 
                      const std::vector<std::shared_ptr<FPGATrackSimGNNHit>> & gnn_hits, 
                      std::vector<std::shared_ptr<const FPGATrackSimRoad>> & roads);
        void addRoad(const std::vector<std::shared_ptr<const FPGATrackSimHit>> & hits, const std::vector<int>& road_hitIDs);
        void resetVectors();
 
        // Event Storage
 
        std::vector<FPGATrackSimRoad> m_roads{};
};
 
 
#endif // FPGATrackSimGNNRoadMakerTool_H