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);

        // Properties
 
        Gaudi::Property<bool> m_doPixelHits { this, "doPixelHits", true, "Flag for selecting pixel hits" };
        Gaudi::Property<bool> m_doStripHits { this, "doStripHits", true, "Flag for selecting strip hits" };
 
};
 
 
#endif // FPGATRACKSIMGNNGRAPHHITSELECTORTOOL_H