FPGATrackSimGNNGraphHitSelectorTool.h

Go to the documentation of this file.

// Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
 
#ifndef FPGATRACKSIMGNNGRAPHHITSELECTORTOOL_H
#define FPGATRACKSIMGNNGRAPHHITSELECTORTOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
 
#include "FPGATrackSimObjects/FPGATrackSimHit.h"
#include "FPGATrackSimObjects/FPGATrackSimGNNHit.h"
 
class FPGATrackSimGNNGraphHitSelectorTool : public AthAlgTool
{
    public:
 
        // AthAlgTool
 
        FPGATrackSimGNNGraphHitSelectorTool(const std::string&, const std::string&, const IInterface*);
 
        // Functions
 
        virtual StatusCode selectHits(const std::vector<std::shared_ptr<const FPGATrackSimHit>> & hits, std::vector<std::shared_ptr<FPGATrackSimGNNHit>> & graph_hits);
 
    private:
 
        // Helpers
 
        float getEta(const std::shared_ptr<const FPGATrackSimHit> & hit);
 
};
 
 
#endif // FPGATRACKSIMGNNGRAPHHITSELECTORTOOL_H