SegmentEdgeData.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef MUONINFERENCEINTERFACES_SEGMENTEDGEDATA_H
#define MUONINFERENCEINTERFACES_SEGMENTEDGEDATA_H
 
#include "xAODMuon/MuonSegment.h"
#include <array>
#include <cstddef>
#include <cstdint>
#include <vector>
 
namespace MuonML {

  enum class PairGate : std::uint8_t {
    Scored = 0,     
    BucketCap,      
    SectorWindow,   
    SameChamber,    
    AngleWindow,    
    EdgeCaps        
  };

  struct PairFate {
    std::uint32_t first{0};
    std::uint32_t second{0};
    PairGate gate{PairGate::Scored};
    std::uint8_t sectorDelta{0};
  };

  inline constexpr std::int32_t kDroppedByBucketCap = -1;
  inline constexpr std::int32_t kDroppedAsIsolated = -2;
 
  struct SegmentEdgeGraph {
    std::vector<const xAOD::MuonSegment_v1*> segments{};
    std::vector<float> nodeFeatures{};      
    std::vector<int64_t> edgeIndex{};       
    std::vector<float> edgeFeatures{};      
    std::size_t nNodes{0};
    std::size_t nEdges{0};
 
    // Truth diagnostics: filled by the tool only when its EnableTruthDiagnostics
    // is set and DEBUG output is on; empty (never allocated) otherwise.
    std::vector<std::int32_t> inputNodeIndex{};  
    std::vector<PairFate> truthPairFates{};      
  };
 
  struct SegmentEdgeScore {
    std::size_t src{0};
    std::size_t dst{0};
    float logit{0.f};
    float probability{0.f};
  };
 
}
#endif