FPGATrackSimGNNPatternRecoTool.h

Go to the documentation of this file.

// Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
 
#ifndef FPGATRACKSIMGNNPATTERNRECOTOOL_H
#define FPGATRACKSIMGNNPATTERNRECOTOOL_H

 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
 
#include "FPGATrackSimHough/IFPGATrackSimRoadFinderTool.h"
 
#include "FPGATrackSimGNNGraphHitSelectorTool.h"
#include "FPGATrackSimGNNGraphConstructionTool.h"
#include "FPGATrackSimGNNEdgeClassifierTool.h"
#include "FPGATrackSimGNNRoadMakerTool.h"
#include "FPGATrackSimGNNRootOutputTool.h"
 
class FPGATrackSimGNNPatternRecoTool : public extends <AthAlgTool, IFPGATrackSimRoadFinderTool>
{
    public:
        using base_class::base_class;
 
        virtual StatusCode initialize() override;

        // FPGATrackSimRoadFinderToolI
 
        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{ return 0; }
 
    private: 

        // Properties
 
        Gaudi::Property<bool> m_doGNNRootOutput { this, "doGNNRootOutput", false, "Flag for GNN Root Output Tool" };
        Gaudi::Property<int> m_regionNum { this, "regionNum", -1, "region number"};
        
        // Handles
        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"};
};
 
 
#endif // FPGATRACKSIMGNNPATTERNRECOTOOL_H