ISegmentEdgeClassifierTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONINFERENCEINTERFACES_ISEGMENTEDGECLASSIFIERTOOL_H
#define MUONINFERENCEINTERFACES_ISEGMENTEDGECLASSIFIERTOOL_H
 
#include "GaudiKernel/IAlgTool.h"
#include "GaudiKernel/EventContext.h"
#include "MuonInferenceInterfaces/SegmentEdgeData.h"
#include "xAODMuon/MuonSegmentContainer.h"
 
namespace MuonML {

  class ISegmentEdgeClassifierTool : virtual public IAlgTool {
  public:
    virtual ~ISegmentEdgeClassifierTool() = default;
    DeclareInterfaceID(ISegmentEdgeClassifierTool, 1, 0);

    virtual StatusCode buildGraph(const EventContext& ctx,
                                  const xAOD::MuonSegmentContainer& segments,
                                  SegmentEdgeGraph& graph) const = 0;

    virtual StatusCode classifyEdges(const EventContext& ctx,
                                     const SegmentEdgeGraph& graph,
                                     std::vector<SegmentEdgeScore>& scores) const = 0;
  };
}
#endif