ATLAS Offline Software
Public Member Functions | Private Attributes | List of all members
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< std::shared_ptr< const FPGATrackSimRoad >> &roads) override
 
virtual int getSubRegion () const override
 

Private Attributes

Gaudi::Property< bool > m_doGNNRootOutput { this, "doGNNRootOutput", false, "Flag for GNN Root Output Tool" }
 
ToolHandle< FPGATrackSimGNNGraphHitSelectorToolm_GNNGraphHitSelectorTool {this, "GNNGraphHitSelector", "FPGATrackSimGNNGraphHitSelectorTool", "Graph HitSelector Tool"}
 
ToolHandle< FPGATrackSimGNNGraphConstructionToolm_GNNGraphConstructionTool {this, "GNNGraphConstruction", "FPGATrackSimGNNGraphConstructionTool", "Graph Construction Tool"}
 
ToolHandle< FPGATrackSimGNNEdgeClassifierToolm_GNNEdgeClassifierTool {this, "GNNEdgeClassifier", "FPGATrackSimGNNEdgeClassifierTool", "Edge Classifier Tool"}
 
ToolHandle< FPGATrackSimGNNRoadMakerToolm_GNNRoadMakerTool {this, "GNNRoadMaker", "FPGATrackSimGNNRoadMakerTool", "Road Maker Tool"}
 
ToolHandle< FPGATrackSimGNNRootOutputToolm_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< std::shared_ptr< const FPGATrackSimRoad >> &  roads 
)
overridevirtual

Definition at line 17 of file FPGATrackSimGNNPatternRecoTool.cxx.

18 {
19  std::vector<std::shared_ptr<FPGATrackSimGNNHit>> graph_hits;
20  std::vector<std::shared_ptr<FPGATrackSimGNNEdge>> graph_edges;
21 
22  ATH_CHECK(m_GNNGraphHitSelectorTool->selectHits(hits, graph_hits)); // Go from FPGATrackSimHits to FPGATrackSimGNNHit -> get information needed for GNNPipeline
23  ATH_CHECK(m_GNNGraphConstructionTool->getEdges(graph_hits, graph_edges)); // Build edges using module map (or metric learning)
24  ATH_CHECK(m_GNNEdgeClassifierTool->scoreEdges(graph_hits, graph_edges)); // Score edges using IN GNN
25  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)
26  if(m_doGNNRootOutput) ATH_CHECK(m_GNNRootOutputTool->fillTree(hits, graph_hits, graph_edges, roads)); // Output the hit/edge/road information into a ROOT file
27 
28  graph_hits.clear();
29  graph_edges.clear();
30  return StatusCode::SUCCESS;
31 }

◆ getSubRegion()

virtual int FPGATrackSimGNNPatternRecoTool::getSubRegion ( ) const
inlineoverridevirtual

Definition at line 50 of file FPGATrackSimGNNPatternRecoTool.h.

50 { return 0; }

◆ initialize()

StatusCode FPGATrackSimGNNPatternRecoTool::initialize ( )
overridevirtual

Definition at line 6 of file FPGATrackSimGNNPatternRecoTool.cxx.

7 {
11  ATH_CHECK(m_GNNRoadMakerTool.retrieve());
12  ATH_CHECK(m_GNNRootOutputTool.retrieve());
13 
14  return StatusCode::SUCCESS;
15 }

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.

◆ m_GNNEdgeClassifierTool

ToolHandle<FPGATrackSimGNNEdgeClassifierTool> FPGATrackSimGNNPatternRecoTool::m_GNNEdgeClassifierTool {this, "GNNEdgeClassifier", "FPGATrackSimGNNEdgeClassifierTool", "Edge Classifier Tool"}
private

Definition at line 63 of file FPGATrackSimGNNPatternRecoTool.h.

◆ m_GNNGraphConstructionTool

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

Definition at line 62 of file FPGATrackSimGNNPatternRecoTool.h.

◆ m_GNNGraphHitSelectorTool

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

Definition at line 61 of file FPGATrackSimGNNPatternRecoTool.h.

◆ m_GNNRoadMakerTool

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

Definition at line 64 of file FPGATrackSimGNNPatternRecoTool.h.

◆ m_GNNRootOutputTool

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

Definition at line 65 of file FPGATrackSimGNNPatternRecoTool.h.


The documentation for this class was generated from the following files:
FPGATrackSimGNNPatternRecoTool::m_doGNNRootOutput
Gaudi::Property< bool > m_doGNNRootOutput
Definition: FPGATrackSimGNNPatternRecoTool.h:57
TRTCalib_Extractor.hits
hits
Definition: TRTCalib_Extractor.py:35
FPGATrackSimGNNPatternRecoTool::m_GNNEdgeClassifierTool
ToolHandle< FPGATrackSimGNNEdgeClassifierTool > m_GNNEdgeClassifierTool
Definition: FPGATrackSimGNNPatternRecoTool.h:63
FPGATrackSimGNNPatternRecoTool::m_GNNGraphHitSelectorTool
ToolHandle< FPGATrackSimGNNGraphHitSelectorTool > m_GNNGraphHitSelectorTool
Definition: FPGATrackSimGNNPatternRecoTool.h:61
ATH_CHECK
#define ATH_CHECK
Definition: AthCheckMacros.h:40
FPGATrackSimGNNPatternRecoTool::m_GNNGraphConstructionTool
ToolHandle< FPGATrackSimGNNGraphConstructionTool > m_GNNGraphConstructionTool
Definition: FPGATrackSimGNNPatternRecoTool.h:62
FPGATrackSimGNNPatternRecoTool::m_GNNRoadMakerTool
ToolHandle< FPGATrackSimGNNRoadMakerTool > m_GNNRoadMakerTool
Definition: FPGATrackSimGNNPatternRecoTool.h:64
FPGATrackSimGNNPatternRecoTool::m_GNNRootOutputTool
ToolHandle< FPGATrackSimGNNRootOutputTool > m_GNNRootOutputTool
Definition: FPGATrackSimGNNPatternRecoTool.h:65