ATLAS Offline Software
Loading...
Searching...
No Matches
MuonML Namespace Reference

Namespaces

namespace  BucketGraphUtils
namespace  Factory

Classes

class  BucketInferenceToolBase
class  GraphBucketFilterTool
 Runs GNN-based bucket-level inference and optionally writes filtered space-point buckets to the event store. More...
class  GraphSPFilterTool
class  InferenceAlg
struct  BucketSegmentFeatures
 Segment features derived from or stored in bucket metadata. More...
class  SegmentEdgeClassifierTool
 Runs a segment-level GNN on reconstructed muon segments to classify segment-pair edges as "good" or "background". More...
class  SegmentEdgeInferenceAlg
class  SegmentTrackCandidateBuilderTool
class  SPInferenceToolBase
 Baseline tool to handle the. More...
struct  InferenceGraph
 Helper struct containing all the information needed to process. More...
struct  GraphRawData
 Helper struct to ship the Graph from the space point buckets to ONNX. More...
class  IGraphInferenceTool
class  ISegmentEdgeClassifierTool
 Interface for segment-edge GNN classification. More...
class  ISegmentTrackCandidateBuilderTool
 Interface for building track candidates from classified segment edges. More...
class  LayerSpBucket
 The LayerSpBucket is a space pointbucket where the points are internally sorted by their layer number as defined in the SpacePointLayerSorter. More...
class  NodeConnector
 The NodeConnector is indicating whether two space points inside a bucket, the graph nodes, shall have a connection in their graph neural net representation. More...
class  NodeFeature
 The NodeFeature is the gluing instance to extract the information from the space point inside a MuonBucket and then to parse it to the ML inference framework. More...
class  NodeFeatureList
struct  SegmentEdgeGraph
struct  SegmentEdgeScore

Typedefs

using HitVec = MuonR4::SpacePointPerLayerSplitter::HitVec
using Feature_t = NodeFeatureList::Feature_t
using Connector_t = NodeFeatureList::Connector_t
using Bucket_t = NodeFeature::Bucket_t

Enumerations

enum class  SegmentNodeFeatureId : uint8_t {
  SegmentPositionX , SegmentPositionY , SegmentPositionZ , SegmentDirectionX ,
  SegmentDirectionY , SegmentDirectionZ , BucketChamberIndex , BucketLayers ,
  BucketSector , BucketSegments
}
 Identifier for each node feature in segment-based GNNs. More...

Functions

bool operator< (const std::string &a, const Feature_t &b)
bool operator< (const Feature_t &a, const std::string &b)
bool operator< (const Feature_t &a, const Feature_t &b)
bool operator< (const std::string &a, const Connector_t &b)
bool operator< (const Connector_t &a, const std::string &b)
bool operator< (const Connector_t &a, const Connector_t &b)

Typedef Documentation

◆ Bucket_t

Definition at line 15 of file NodeFeatureFactory.cxx.

◆ Connector_t

◆ Feature_t

◆ HitVec

Enumeration Type Documentation

◆ SegmentNodeFeatureId

enum class MuonML::SegmentNodeFeatureId : uint8_t
strong

Identifier for each node feature in segment-based GNNs.

The order of these enum values matches the training data export order. They are used to look up feature indices by name from model metadata and to compute feature vectors during inference.

Enumerator
SegmentPositionX 

Segment position (x-coordinate, in meters).

SegmentPositionY 

Segment position (y-coordinate, in meters).

SegmentPositionZ 

Segment position (z-coordinate, in meters).

SegmentDirectionX 

Segment direction (x-component, unit vector).

SegmentDirectionY 

Segment direction (y-component, unit vector).

SegmentDirectionZ 

Segment direction (z-component, unit vector).

BucketChamberIndex 

Muon chamber index (integer ID).

BucketLayers 

Number of precision/phi/trigger layers in segment.

BucketSector 

Sector number (0 to SectorModulo-1, typically 16).

BucketSegments 

Count of segments in the same chamber/layer group.

Definition at line 28 of file MuonMLEvent.h.

28 : uint8_t {
39};
@ SegmentDirectionX
Segment direction (x-component, unit vector).
Definition MuonMLEvent.h:32
@ BucketChamberIndex
Muon chamber index (integer ID).
Definition MuonMLEvent.h:35
@ BucketSegments
Count of segments in the same chamber/layer group.
Definition MuonMLEvent.h:38
@ BucketSector
Sector number (0 to SectorModulo-1, typically 16).
Definition MuonMLEvent.h:37
@ SegmentPositionY
Segment position (y-coordinate, in meters).
Definition MuonMLEvent.h:30
@ SegmentDirectionZ
Segment direction (z-component, unit vector).
Definition MuonMLEvent.h:34
@ SegmentPositionX
Segment position (x-coordinate, in meters).
Definition MuonMLEvent.h:29
@ SegmentPositionZ
Segment position (z-coordinate, in meters).
Definition MuonMLEvent.h:31
@ BucketLayers
Number of precision/phi/trigger layers in segment.
Definition MuonMLEvent.h:36
@ SegmentDirectionY
Segment direction (y-component, unit vector).
Definition MuonMLEvent.h:33

Function Documentation

◆ operator<() [1/6]

bool MuonML::operator< ( const Connector_t & a,
const Connector_t & b )

Definition at line 33 of file NodeFeatureFactory.cxx.

33 {
34 return a->name() < b->name();
35 }
static Double_t a

◆ operator<() [2/6]

bool MuonML::operator< ( const Connector_t & a,
const std::string & b )

Definition at line 30 of file NodeFeatureFactory.cxx.

30 {
31 return a->name() < b;
32 }

◆ operator<() [3/6]

bool MuonML::operator< ( const Feature_t & a,
const Feature_t & b )

Definition at line 23 of file NodeFeatureFactory.cxx.

23 {
24 return a->name() < b->name();
25 }

◆ operator<() [4/6]

bool MuonML::operator< ( const Feature_t & a,
const std::string & b )

Definition at line 20 of file NodeFeatureFactory.cxx.

20 {
21 return a->name() < b;
22 }

◆ operator<() [5/6]

bool MuonML::operator< ( const std::string & a,
const Connector_t & b )

Definition at line 27 of file NodeFeatureFactory.cxx.

27 {
28 return a < b->name();
29 }

◆ operator<() [6/6]

bool MuonML::operator< ( const std::string & a,
const Feature_t & b )

Definition at line 17 of file NodeFeatureFactory.cxx.

17 {
18 return a < b->name();
19 }