FPGATrackSimGNNPatternRecoTool Class Reference

#include <FPGATrackSimGNNPatternRecoTool.h>

Inheritance diagram for FPGATrackSimGNNPatternRecoTool:

Collaboration diagram for FPGATrackSimGNNPatternRecoTool:

Public Member Functions
virtual StatusCode	initialize () override
virtual StatusCode	getRoads (const std::vector< std::shared_ptr< const FPGATrackSimHit > > &hits, std::vector< FPGATrackSimRoad > &roads) override
virtual int	getSubRegion () const override

Private Attributes
Gaudi::Property< bool >	m_doGNNRootOutput { this, "doGNNRootOutput", false, "Flag for GNN Root Output Tool" }
Gaudi::Property< int >	m_regionNum { this, "regionNum", -1, "region number"}
ToolHandle< FPGATrackSimGNNGraphHitSelectorTool >	m_GNNGraphHitSelectorTool {this, "GNNGraphHitSelector", "FPGATrackSimGNNGraphHitSelectorTool", "Graph HitSelector Tool"}
ToolHandle< FPGATrackSimGNNGraphConstructionTool >	m_GNNGraphConstructionTool {this, "GNNGraphConstruction", "FPGATrackSimGNNGraphConstructionTool", "Graph Construction Tool"}
ToolHandleArray< FPGATrackSimGNNEdgeClassifierTool >	m_GNNEdgeClassifierTools {this, "GNNEdgeClassifiers", {}, "One GNNEdgeClassifierTool per region"}
ToolHandle< FPGATrackSimGNNEdgeClassifierTool >	m_selectedGNNEdgeClassifierTool
ToolHandle< FPGATrackSimGNNRoadMakerTool >	m_GNNRoadMakerTool {this, "GNNRoadMaker", "FPGATrackSimGNNRoadMakerTool", "Road Maker Tool"}
ToolHandle< FPGATrackSimGNNRootOutputTool >	m_GNNRootOutputTool {this, "GNNRootOutput", "FPGATrackSimGNNRootOutputTool", "GNN ROOT Output Tool"}

Detailed Description

Definition at line 38 of file FPGATrackSimGNNPatternRecoTool.h.

Member Function Documentation

getRoads()

StatusCode FPGATrackSimGNNPatternRecoTool::getRoads ( const std::vector< std::shared_ptr< const FPGATrackSimHit > > & hits, std::vector< FPGATrackSimRoad > & roads )

overridevirtual

Definition at line 29 of file FPGATrackSimGNNPatternRecoTool.cxx.

{
    std::vector<std::shared_ptr<FPGATrackSimGNNHit>> graph_hits;
    std::vector<std::shared_ptr<FPGATrackSimGNNEdge>> graph_edges;
    
    ATH_CHECK(m_GNNGraphHitSelectorTool->selectHits(hits, graph_hits)); // Go from FPGATrackSimHits to FPGATrackSimGNNHit -> get information needed for GNNPipeline
    ATH_CHECK(m_GNNGraphConstructionTool->getEdges(graph_hits, graph_edges)); // Build edges using module map (or metric learning)
    if(graph_edges.size() > 0){
        ATH_CHECK(m_selectedGNNEdgeClassifierTool->scoreEdges(graph_hits, graph_edges)); // Score edges using IN GNN
        ATH_CHECK(m_GNNRoadMakerTool->makeRoads(hits, graph_hits, graph_edges, roads)); // Build road candidates using connected components (need to make a C++ version of it)
    }
    if(m_doGNNRootOutput) ATH_CHECK(m_GNNRootOutputTool->fillTree(hits, graph_hits, graph_edges, roads)); // Output the hit/edge/road information into a ROOT file
 
    graph_hits.clear();
    graph_edges.clear();
    return StatusCode::SUCCESS;
}

getSubRegion()

virtual int FPGATrackSimGNNPatternRecoTool::getSubRegion ( ) const

inlineoverridevirtual

Definition at line 50 of file FPGATrackSimGNNPatternRecoTool.h.

{ return 0; }

initialize()

StatusCode FPGATrackSimGNNPatternRecoTool::initialize ( )

overridevirtual

Definition at line 6 of file FPGATrackSimGNNPatternRecoTool.cxx.

{
    ATH_CHECK(m_GNNGraphHitSelectorTool.retrieve());
    ATH_CHECK(m_GNNGraphConstructionTool.retrieve());
    ATH_CHECK(m_GNNEdgeClassifierTools.retrieve());
    ATH_CHECK(m_GNNRoadMakerTool.retrieve());
    ATH_CHECK(m_GNNRootOutputTool.retrieve());
 
    // Need to selected the right tool based on region
    for (auto& tool : m_GNNEdgeClassifierTools) {
        if (tool->regionNum() == m_regionNum) {
            m_selectedGNNEdgeClassifierTool = tool;
            break;
        }
    }
    if (!m_selectedGNNEdgeClassifierTool.isValid()) {
        ATH_MSG_ERROR("No GNNEdgeClassifierTool found for region " << m_regionNum);
        return StatusCode::FAILURE;
    }
 
    return StatusCode::SUCCESS;
}

Member Data Documentation

m_doGNNRootOutput

Gaudi::Property<bool> FPGATrackSimGNNPatternRecoTool::m_doGNNRootOutput { this, "doGNNRootOutput", false, "Flag for GNN Root Output Tool" }

private

Definition at line 57 of file FPGATrackSimGNNPatternRecoTool.h.

{ this, "doGNNRootOutput", false, "Flag for GNN Root Output Tool" };

m_GNNEdgeClassifierTools

ToolHandleArray<FPGATrackSimGNNEdgeClassifierTool> FPGATrackSimGNNPatternRecoTool::m_GNNEdgeClassifierTools {this, "GNNEdgeClassifiers", {}, "One GNNEdgeClassifierTool per region"}

private

Definition at line 64 of file FPGATrackSimGNNPatternRecoTool.h.

{this, "GNNEdgeClassifiers", {}, "One GNNEdgeClassifierTool per region"};

m_GNNGraphConstructionTool

ToolHandle<FPGATrackSimGNNGraphConstructionTool> FPGATrackSimGNNPatternRecoTool::m_GNNGraphConstructionTool {this, "GNNGraphConstruction", "FPGATrackSimGNNGraphConstructionTool", "Graph Construction Tool"}

private

Definition at line 63 of file FPGATrackSimGNNPatternRecoTool.h.

{this, "GNNGraphConstruction", "FPGATrackSimGNNGraphConstructionTool", "Graph Construction Tool"};

m_GNNGraphHitSelectorTool

ToolHandle<FPGATrackSimGNNGraphHitSelectorTool> FPGATrackSimGNNPatternRecoTool::m_GNNGraphHitSelectorTool {this, "GNNGraphHitSelector", "FPGATrackSimGNNGraphHitSelectorTool", "Graph HitSelector Tool"}

private

Definition at line 62 of file FPGATrackSimGNNPatternRecoTool.h.

{this, "GNNGraphHitSelector", "FPGATrackSimGNNGraphHitSelectorTool", "Graph HitSelector Tool"};

m_GNNRoadMakerTool

ToolHandle<FPGATrackSimGNNRoadMakerTool> FPGATrackSimGNNPatternRecoTool::m_GNNRoadMakerTool {this, "GNNRoadMaker", "FPGATrackSimGNNRoadMakerTool", "Road Maker Tool"}

private

Definition at line 66 of file FPGATrackSimGNNPatternRecoTool.h.

{this, "GNNRoadMaker", "FPGATrackSimGNNRoadMakerTool", "Road Maker Tool"};

m_GNNRootOutputTool

ToolHandle<FPGATrackSimGNNRootOutputTool> FPGATrackSimGNNPatternRecoTool::m_GNNRootOutputTool {this, "GNNRootOutput", "FPGATrackSimGNNRootOutputTool", "GNN ROOT Output Tool"}

private

Definition at line 67 of file FPGATrackSimGNNPatternRecoTool.h.

{this, "GNNRootOutput", "FPGATrackSimGNNRootOutputTool", "GNN ROOT Output Tool"};

m_regionNum

Gaudi::Property<int> FPGATrackSimGNNPatternRecoTool::m_regionNum { this, "regionNum", -1, "region number"}

private

Definition at line 58 of file FPGATrackSimGNNPatternRecoTool.h.

{ this, "regionNum", -1, "region number"};

m_selectedGNNEdgeClassifierTool

ToolHandle<FPGATrackSimGNNEdgeClassifierTool> FPGATrackSimGNNPatternRecoTool::m_selectedGNNEdgeClassifierTool

private

Definition at line 65 of file FPGATrackSimGNNPatternRecoTool.h.

---

The documentation for this class was generated from the following files:

• FPGATrackSimGNNPatternRecoTool.h
• FPGATrackSimGNNPatternRecoTool.cxx