ATLAS Offline Software
Functions
python.InDetGNNTrackingConfigFlags Namespace Reference

Functions

def createGNNTrackingConfigFlags ()
 

Function Documentation

◆ createGNNTrackingConfigFlags()

def python.InDetGNNTrackingConfigFlags.createGNNTrackingConfigFlags ( )
Create flags for configuring the GNN tracking.

Definition at line 7 of file InDetGNNTrackingConfigFlags.py.

8  """Create flags for configuring the GNN tracking."""
9  icf = AthConfigFlags()
10  icf.addFlag("Tracking.GNN.useTrackFinder", False)
11  icf.addFlag("Tracking.GNN.useTrackReader", False)
12  icf.addFlag("Tracking.GNN.usePixelHitsOnly", False)
13 
14  # Dump objects
15  icf.addFlag("Tracking.GNN.DumpObjects.NtupleFileName", "/DumpObjects/")
16  icf.addFlag("Tracking.GNN.DumpObjects.NtupleTreeName", "GNN4ITk")
17 
18  # GNN Track finder tool
19  icf.addFlag("Tracking.GNN.TrackFinder.inputMLModelDir", "TrainedMLModels4ITk")
20  icf.addFlag("Tracking.GNN.TrackFinder.ORTExeProvider", OnnxRuntimeType.CPU)
21 
22  # GNN Track Reader Tool
23  icf.addFlag("Tracking.GNN.TrackReader.inputTracksDir", "gnntracks")
24  icf.addFlag("Tracking.GNN.TrackReader.csvPrefix", "track")
25 
26  icf.addFlag("Tracking.GNN.useClusterTracks", False)
27 
28  # the following cuts are applied to the tracks before the track fitting
29  icf.addFlag("Tracking.GNN.minPixelClusters", 1)
30  icf.addFlag("Tracking.GNN.minStripClusters", 0)
31  icf.addFlag("Tracking.GNN.minClusters", 6)
32 
33  # the following cuts are applied to the tracks after the track fitting
34  icf.addFlag("Tracking.GNN.etamax", 4.0)
35  import AthenaCommon.SystemOfUnits as Units
36  icf.addFlag("Tracking.GNN.pTmin", 400. * Units.MeV)
37 
38  # This option applies eta dependent track selection to the output tracks
39  icf.addFlag("Tracking.GNN.doRecoTrackCuts", True)
40 
41  # this option turns on the ambiguity resolution, False by default
42  icf.addFlag("Tracking.GNN.doAmbiResolution", False)
43 
44  return icf
45 
SystemOfUnits
python.InDetGNNTrackingConfigFlags.createGNNTrackingConfigFlags
def createGNNTrackingConfigFlags()
Definition: InDetGNNTrackingConfigFlags.py:7