ATLAS Offline Software
Loading...
Searching...
No Matches
TrigTauRecToolsConfig.py
Go to the documentation of this file.
1# Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
2
3from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
4from AthenaConfiguration.ComponentFactory import CompFactory
5from AthenaCommon.SystemOfUnits import mm
6
7def trigTauVertexFinderCfg(flags, name=''):
8 '''Algorithm that overwrites numTrack() and charge() of tauJets in container'''
9 acc = ComponentAccumulator()
10
11 acc.setPrivateTools(CompFactory.TauVertexFinder(
12 name = name,
13 UseTJVA = False,
14 AssociatedTracks = 'GhostTrack',
15 InDetTrackSelectionToolForTJVA = '',
16 Key_vertexInputContainer = '',
17 TVATool = '',
18 ))
19
20 return acc
21
22def trigTauTrackFinderCfg(flags, name='', TrackParticlesContainer=''):
23 '''Tau track association'''
24 acc = ComponentAccumulator()
25
26 from TrkConfig.TrkVertexFitterUtilsConfig import AtlasTrackToVertexIPEstimatorCfg
27 AtlasTrackToVertexIPEstimator = acc.popToolsAndMerge(AtlasTrackToVertexIPEstimatorCfg(flags))
28
29 from TrackToVertex.TrackToVertexConfig import TrackToVertexCfg
30 TrackToVertexTool = acc.popToolsAndMerge(TrackToVertexCfg(flags))
31
32 from TrackToCalo.TrackToCaloConfig import ParticleCaloExtensionToolCfg
33 ParticleCaloExtensionTool = acc.popToolsAndMerge(ParticleCaloExtensionToolCfg(flags))
34
35 from InDetConfig.InDetTrackSelectorToolConfig import TrigTauInDetTrackSelectorToolCfg
36 TrigTauInDetTrackSelectorTool = acc.popToolsAndMerge(TrigTauInDetTrackSelectorToolCfg(flags))
37
38 acc.setPrivateTools(CompFactory.TauTrackFinder(
39 name = name,
40 MaxJetDrTau = 0.2,
41 MaxJetDrWide = 0.4,
42 TrackSelectorToolTau = TrigTauInDetTrackSelectorTool,
43 TrackToVertexTool = TrackToVertexTool,
44 Key_trackPartInputContainer = TrackParticlesContainer,
45 maxDeltaZ0wrtLeadTrk = 0.75*mm,
46 removeTracksOutsideZ0wrtLeadTrk = True,
47 ParticleCaloExtensionTool = ParticleCaloExtensionTool,
48 BypassExtrapolator = True,
49 tauParticleCache = "",
50 TrackToVertexIPEstimator = AtlasTrackToVertexIPEstimator,
51 ))
52
53 return acc
54
55def tauVertexVariablesCfg(flags, name=''):
56 '''Vertex variables calculation'''
57 acc = ComponentAccumulator()
58
59 from TrkConfig.TrkVertexFittersConfig import TauAdaptiveVertexFitterCfg
60 TauAdaptiveVertexFitter = acc.popToolsAndMerge(TauAdaptiveVertexFitterCfg(flags))
61
62 from TrkConfig.TrkVertexSeedFinderToolsConfig import CrossDistancesSeedFinderCfg
63 CrossDistancesSeedFinder = acc.popToolsAndMerge(CrossDistancesSeedFinderCfg(flags))
64
65 acc.setPrivateTools(CompFactory.TauVertexVariables(
66 name = name,
67 VertexFitter = TauAdaptiveVertexFitter,
68 SeedFinder = CrossDistancesSeedFinder,
69 ))
70
71 return acc
72
73def trigTauJetLVNNEvaluatorCfg(flags, tau_id='', use_taujet_rnnscore=True):
74 '''TauJet identification inference based on LVNN models, for RNNs and DeepSets'''
75 acc = ComponentAccumulator()
76
77 try: id_flags = getattr(flags.Trigger.Offline.Tau, tau_id)
78 except NameError: raise ValueError(f'Invalid LVNN (RNN, DeepSet) TauID configuration: {tau_id}')
79
80 # For legacy mediumRNN/tightRNN_tracktwoMVA/tracktwoLLP/trackLRT chains
81 output_variable = 'RNNJetScore' if use_taujet_rnnscore else f'{tau_id}_Score'
82
83 # This is to prevent clashes between the same ID running over the legacy chains, and the new decorator-based chains
84 sfx = '_RNNJetScore' if use_taujet_rnnscore else ''
85
86 acc.setPrivateTools(CompFactory.TauJetRNNEvaluator(
87 name = f'TrigTau_TauJetLVNNEvaluator_{tau_id}{sfx}',
88 useTRT = flags.Detector.EnableTRT,
89
90 # Network config:
91 NetworkFile0P = id_flags.NetworkConfig[0],
92 NetworkFile1P = id_flags.NetworkConfig[1],
93 NetworkFile3P = id_flags.NetworkConfig[2],
94 InputLayerScalar = 'scalar',
95 InputLayerTracks = 'tracks',
96 InputLayerClusters = 'clusters',
97 OutputLayer = 'rnnid_output',
98 OutputNode = 'sig_prob',
99
100 # Inputs:
101 MaxTracks = id_flags.MaxTracks,
102 MaxClusters = id_flags.MaxClusters,
103 MaxClusterDR = 1.0,
104 VertexCorrection = False,
105 TrackClassification = False,
106
107 # Decorated TauJet variable names:
108 OutputVarname = output_variable,
109 ))
110
111 return acc
112
113def trigTauJetONNXEvaluatorCfg(flags, tau_id: str, tau_container: str = '', hits_decoration_container: str = ''):
114 '''TauJet identification inference based on ONNX models, for GNNs, transformers, etc...'''
115 acc = ComponentAccumulator()
116
117 try: id_flags = getattr(flags.Trigger.Offline.Tau, tau_id)
118 except NameError: raise ValueError(f'Invalid ONNX TauID configuration: {tau_id}')
119
120 if isinstance(id_flags.ONNXConfig, str):
121 network_config = {'NetworkFileInclusive': id_flags.ONNXConfig}
122 elif isinstance(id_flags.ONNXConfig, (list, tuple)) and len(id_flags.ONNXConfig) == 3:
123 network_config = {
124 'NetworkFile0P': id_flags.ONNXConfig[0],
125 'NetworkFile1P': id_flags.ONNXConfig[1],
126 'NetworkFile3P': id_flags.ONNXConfig[2],
127 }
128 else:
129 raise ValueError(f'Invalid {tau_id} ONNX network config file')
130
131 name = f'TrigTau_TauJetONNXEvaluator_{tau_id}'
132 if tau_container: name += f'_{tau_container}'
133 if hits_decoration_container: name += f'_{hits_decoration_container}'
134
135 acc.setPrivateTools(CompFactory.TauGNNEvaluator(
136 name = name,
137 useTRT = flags.Detector.EnableTRT,
138
139 # Network config:
140 **network_config,
141 InputLayerScalar = 'tau_vars',
142 InputLayerTracks = 'track_vars',
143 InputLayerClusters = 'cluster_vars',
144 InputLayerHits = 'hit_vars',
145 NodeNameTau = id_flags.NodeNameTau if hasattr(id_flags, 'NodeNameTau') else 'pTau',
146 NodeNameJet = id_flags.NodeNameJet if hasattr(id_flags, 'NodeNameJet') else 'pJet',
147
148 # Inputs:
149 MaxTracks = id_flags.MaxTracks if hasattr(id_flags, 'MaxTracks') else 0,
150 MaxClusters = id_flags.MaxClusters if hasattr(id_flags, 'MaxClusters') else 0,
151 MaxClusterDR = 1.0,
152 MaxHits = id_flags.MaxHits if hasattr(id_flags, 'MaxHits') else 0,
153 TauContainerName = tau_container,
154 HitsHandleKey = hits_decoration_container,
155 VertexCorrection = False,
156 TrackClassification = False,
157
158 # Decorated TauJet variable names:
159 OutputVarname = f'{tau_id}_Score',
160 OutputDiscriminant = id_flags.OutputDiscriminant,
161 OutputPTau = id_flags.OutputPTau if hasattr(id_flags, 'OutputPTau') else f'{tau_id}_ProbTau',
162 OutputPJet = id_flags.OutputPJet if hasattr(id_flags, 'OutputPJet') else f'{tau_id}_ProbJet',
163
164 # Tau prongness selection
165 MinProngTrackPt = id_flags.MinProngTrackPt if hasattr(id_flags, 'MinProngTrackPt') else 0,
166 ))
167
168 return acc
169
170def trigTauWPDecoratorRNNCfg(flags, tau_id: str, precision_seq_name: str):
171 '''
172 TauJet signal transformed score and ID WPs decorator tool,
173 for the legacy mediumRNN/tightRNN_tracktwoMVA/tracktwoLLP/trackLRT chains ONLY!
174 '''
175
176 acc = ComponentAccumulator()
177
178 try: id_flags = getattr(flags.Trigger.Offline.Tau, tau_id)
179 except NameError: raise ValueError(f'Invalid TauID configuration: {tau_id}')
180
181 # In this version we store the WPs as flags accessable through
182 # tau->isTau(xAOD::TauJetParameters::IsTauFlags::JetRNNSig...)
183 import PyUtils.RootUtils as ru
184 ROOT = ru.import_root()
185 import cppyy
186 cppyy.load_library('libxAODTau_cDict')
187 WPEnumVals = [
188 ROOT.xAOD.TauJetParameters.IsTauFlag.JetRNNSigVeryLoose,
189 ROOT.xAOD.TauJetParameters.IsTauFlag.JetRNNSigLoose,
190 ROOT.xAOD.TauJetParameters.IsTauFlag.JetRNNSigMedium,
191 ROOT.xAOD.TauJetParameters.IsTauFlag.JetRNNSigTight,
192 ]
193
194 acc.setPrivateTools(CompFactory.TauWPDecorator(
195 name=f'TrigTau_TauWPDecoratorRNN_{precision_seq_name}_{tau_id}',
196 flatteningFile0Prong=id_flags.ScoreFlatteningConfig[0],
197 flatteningFile1Prong=id_flags.ScoreFlatteningConfig[1],
198 flatteningFile3Prong=id_flags.ScoreFlatteningConfig[2],
199 CutEnumVals=WPEnumVals,
200 SigEff0P=id_flags.TargetEff[0],
201 SigEff1P=id_flags.TargetEff[1],
202 SigEff3P=id_flags.TargetEff[2],
203 ScoreName='RNNJetScore',
204 NewScoreName='RNNJetScoreSigTrans',
205 DefineWPs=True,
206 ))
207
208 return acc
209
210def trigTauWPDecoratorCfg(flags, tau_id: str, precision_seq_name: str, tauContainerName: str):
211 '''TauJet signal transformed score and ID WPs decorator tool'''
212 acc = ComponentAccumulator()
213
214 try: id_flags = getattr(flags.Trigger.Offline.Tau, tau_id)
215 except NameError: raise ValueError(f'Invalid TauID configuration: {tau_id}')
216
217 if isinstance(id_flags.ScoreFlatteningConfig, str):
218 cfg = {
219 'flatteningFile0Prong': id_flags.ScoreFlatteningConfig,
220 'flatteningFile1Prong': id_flags.ScoreFlatteningConfig,
221 'flatteningFile3Prong': id_flags.ScoreFlatteningConfig,
222 }
223 elif isinstance(id_flags.ScoreFlatteningConfig, (list, tuple)) and len(id_flags.ScoreFlatteningConfig) == 3:
224 cfg = {
225 'flatteningFile0Prong': id_flags.ScoreFlatteningConfig[0],
226 'flatteningFile1Prong': id_flags.ScoreFlatteningConfig[1],
227 'flatteningFile3Prong': id_flags.ScoreFlatteningConfig[2],
228 }
229 else:
230 raise ValueError(f'Invalid {tau_id} WP decorator flattening config')
231
232
233 acc.setPrivateTools(CompFactory.TauWPDecorator(
234 name=f'TrigTau_TauWPDecorator_{precision_seq_name}_{tau_id}',
235 **cfg,
236 TauContainerName=tauContainerName,
237 DecorWPNames=[f'{tau_id}_{wp}' for wp in id_flags.TargetWPs],
238 DecorWPCutEffs0P=[eff[0] for eff in id_flags.TargetWPs.values()],
239 DecorWPCutEffs1P=[eff[1] for eff in id_flags.TargetWPs.values()],
240 DecorWPCutEffs3P=[eff[2] for eff in id_flags.TargetWPs.values()],
241 ScoreName=f'{tau_id}_Score',
242 NewScoreName=f'{tau_id}_ScoreSigTrans',
243 DefineWPs=True,
244 ))
245
246 return acc
TrigTauRecToolsConfig.trigTauVertexFinderCfg
trigTauVertexFinderCfg(flags, name='')
Definition TrigTauRecToolsConfig.py:7
Generated on for ATLAS Offline Software by doxygen 1.16.1