ATLAS Offline Software
InnerDetector
InDetGNNTracking
python
InDetGNNTrackingConfigFlags.py
Go to the documentation of this file.
1
#
2
# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
3
#
4
from
AthenaConfiguration.AthConfigFlags
import
AthConfigFlags
5
from
AthOnnxComps.OnnxRuntimeFlags
import
OnnxRuntimeType
6
7
def
createGNNTrackingConfigFlags
():
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
Generated on Sun Dec 22 2024 21:11:37 for ATLAS Offline Software by
1.8.18