ATLAS Offline Software
Loading...
Searching...
No Matches
InDetToolsConfig.py
Go to the documentation of this file.
1# Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
2
3# ==============================================================================
4# Provides configs for the tools used for building/thinning tracking related
5# object containers and decorations in the DAODs
6# ==============================================================================
7
8from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
9from AthenaConfiguration.ComponentFactory import CompFactory
10from AthenaConfiguration.Enums import BeamType
11from AthenaCommon.Constants import INFO
12from AthenaCommon.Logging import logging
13logIDTools = logging.getLogger('InDetToolsConfig')
14
15# Track collection merger
16def InDetLRTMergeCfg(flags, name="InDetLRTMerge", **kwargs):
17 acc = ComponentAccumulator()
18 alg = CompFactory.CP.TrackParticleMergerAlg(name, **kwargs)
19 kwargs.setdefault("InputTrackParticleLocations",
20 ["InDetTrackParticles", "InDetLargeD0TrackParticles"])
21 kwargs.setdefault("OutputTrackParticleLocation",
22 "InDetWithLRTTrackParticles")
23 kwargs.setdefault("CreateViewCollection", True)
24 acc.addEventAlgo(alg, primary=True)
25 return acc
26
27# Used in vertex fit track decorator
28
29
30def UsedInVertexFitTrackDecoratorCfg(
31 flags, name="UsedInVertexFitTrackDecorator", **kwargs):
32 """Configure the UsedInVertexFitTrackDecorator"""
33 acc = ComponentAccumulator()
34
35 if "UsedInFitDecoratorTool" not in kwargs:
36 from InDetConfig.InDetUsedInFitTrackDecoratorToolConfig import (
37 InDetUsedInFitTrackDecoratorToolCfg)
38 kwargs.setdefault("UsedInFitDecoratorTool", acc.popToolsAndMerge(
39 InDetUsedInFitTrackDecoratorToolCfg(flags)))
40
41 acc.addPublicTool(
42 CompFactory.DerivationFramework.UsedInVertexFitTrackDecorator(
43 name, **kwargs), primary=True)
44 return acc
45
46
47
48def HardScatterVertexDecoratorCfg(flags, name = "DFCommonHSDecorator", **kwargs):
49 """Configure the hard process vertex decorator"""
50 acc = ComponentAccumulator()
51 from InDetConfig.InDetHardScatterSelectionToolConfig import InDetHardScatterSelectionToolCfg
52 kwargs.setdefault("HardScatterSelectionTool", acc.getPrimaryAndMerge(InDetHardScatterSelectionToolCfg(flags, name = "HSSelectionTool",
53 ReturnDeco = False)))
54 kwargs.setdefault("VertexContainerName", "PrimaryVertices")
55 kwargs.setdefault("HardScatterDecoName", "hardScatterVertexLink")
56 the_tool = CompFactory.DerivationFramework.HardScatterVertexDecorator(name = "HardScatterDecorTool", **kwargs)
57 acc.addPublicTool(the_tool, primary=True)
58 the_alg = CompFactory.DerivationFramework.CommonAugmentation(name, AugmentationTools=[the_tool])
59 acc.addEventAlgo(the_alg)
60 return acc
61
62# TrackStateOnSurface decorator
63
64def TrackStateOnSurfaceDecoratorCfg(flags, name, **kwargs):
65 """Configure the TSOS decorator"""
66 acc = ComponentAccumulator()
67
68 # To produce SCT_DetectorElementCollection
69 from SCT_GeoModel.SCT_GeoModelConfig import SCT_ReadoutGeometryCfg
70 acc.merge(SCT_ReadoutGeometryCfg(flags))
71
72 kwargs.setdefault("DecorationPrefix", "notSet")
73
74 if "Updator" not in kwargs:
75 from TrkConfig.TrkMeasurementUpdatorConfig import KalmanUpdatorCfg
76 kwargs.setdefault("Updator", acc.popToolsAndMerge(
77 KalmanUpdatorCfg(flags)))
78
79 if "TrackExtrapolator" not in kwargs:
80 from TrkConfig.AtlasExtrapolatorConfig import AtlasExtrapolatorCfg
81 kwargs.setdefault("TrackExtrapolator", acc.popToolsAndMerge(
82 AtlasExtrapolatorCfg(flags)))
83
84 if "HoleSearch" not in kwargs:
85 from InDetConfig.InDetTrackHoleSearchConfig import (
86 InDetTrackHoleSearchToolCfg)
87 kwargs.setdefault("HoleSearch", acc.popToolsAndMerge(
88 InDetTrackHoleSearchToolCfg(flags)))
89
90 if "ResidualPullCalculator" not in kwargs:
91 from TrkConfig.TrkResidualPullCalculatorConfig import (
92 ResidualPullCalculatorCfg)
93 kwargs.setdefault("ResidualPullCalculator", acc.popToolsAndMerge(
94 ResidualPullCalculatorCfg(flags)))
95
96 kwargs.setdefault("StorePixel", flags.Tracking.writeExtendedSi_PRDInfo)
97 kwargs.setdefault("StoreSCT", flags.Tracking.writeExtendedSi_PRDInfo)
98 kwargs.setdefault("StoreTRT", flags.Tracking.writeExtendedTRT_PRDInfo)
99 kwargs.setdefault("AddExtraEventInfo", flags.Beam.Type is BeamType.Cosmics)
100
101 if kwargs["StoreTRT"]:
102 if "TRT_ToT_dEdx" not in kwargs:
103 from InDetConfig.TRT_ElectronPidToolsConfig import TRT_dEdxToolCfg
104 kwargs.setdefault("TRT_ToT_dEdx", acc.popToolsAndMerge(
105 TRT_dEdxToolCfg(flags)))
106
107 if "TRT_CalDbTool" not in kwargs:
108 from TRT_ConditionsServices.TRT_ConditionsServicesConfig import (
109 TRT_CalDbToolCfg)
110 kwargs.setdefault("TRT_CalDbTool", acc.popToolsAndMerge(
111 TRT_CalDbToolCfg(flags)))
112
113 kwargs.setdefault("PRDtoTrackMap", "PRDtoTrackMapMerge_CombinedInDetTracks")
114
115 acc.addPublicTool(
116 CompFactory.DerivationFramework.TrackStateOnSurfaceDecorator(
117 name, **kwargs), primary=True)
118 return acc
119
120def TSOS_CommonKernelCfg(flags, name="TSOS_CommonKernel",
121 listOfExtensions=[]):
122 acc = ComponentAccumulator()
123
124 listOfAugmTools = []
125 for extension in listOfExtensions:
126 # Fix inconsistency in naming convention for R3LargeD0
127 if extension == "R3LargeD0":
128 extension = "LargeD0"
129 TrackStateOnSurfaceDecorator = acc.getPrimaryAndMerge(
130 TrackStateOnSurfaceDecoratorCfg(
131 flags, name = f"{extension}TrackStateOnSurfaceDecorator",
132 ContainerName = f"InDet{extension}TrackParticles",
133 PixelMsosName = f"{extension}PixelMSOSs",
134 SctMsosName = f"{extension}SCT_MSOSs",
135 TrtMsosName = f"{extension}TRT_MSOSs"))
136 TrackStateOnSurfaceDecorator.DecorationPrefix = "Reco_"
137 listOfAugmTools.append(TrackStateOnSurfaceDecorator)
138
139 acc.addEventAlgo(CompFactory.DerivationFramework.CommonAugmentation(
140 name, AugmentationTools=listOfAugmTools))
141 return acc
142
143def DFTrackStateOnSurfaceDecoratorCfg(
144 flags, name="DFTrackStateOnSurfaceDecorator", **kwargs):
145 kwargs.setdefault("StorePixel", flags.InDet.DAODStorePixel)
146 kwargs.setdefault("StoreSCT", flags.InDet.DAODStoreSCT)
147 kwargs.setdefault("StoreTRT", flags.InDet.DAODStoreTRT)
148 # never decorate EventInfo with TRTPhase, doubt this is useful for IDTIDE
149 kwargs.setdefault("AddExtraEventInfo", flags.InDet.DAODStoreExtra and flags.Beam.Type is BeamType.Cosmics)
150 kwargs.setdefault("DecorationPrefix", "")
151 kwargs.setdefault("PRDtoTrackMap", "")
152 kwargs.setdefault("OutputLevel", INFO)
153 return TrackStateOnSurfaceDecoratorCfg(flags, name, **kwargs)
154
155def DFInDetTSOSKernelCfg(flags, name='DFInDetTSOSKernel'):
156 acc = ComponentAccumulator()
157
158 # ====================================================================
159 # AUGMENTATION TOOLS
160 # ====================================================================
161 tsos_augmentationTools = []
162
163 DFTSOS = acc.getPrimaryAndMerge(DFTrackStateOnSurfaceDecoratorCfg(flags))
164 tsos_augmentationTools.append(DFTSOS)
165
166 # shared between IDTIDE and IDTRKVALID
167 acc.addEventAlgo(CompFactory.DerivationFramework.DerivationKernel(
168 name,
169 AugmentationTools=tsos_augmentationTools,
170 ThinningTools=[],
171 doChronoStat=(flags.Concurrency.NumThreads <= 1)))
172
173 return acc
174
175def ObserverTrackStateOnSurfaceDecoratorCfg(
176 flags, name="ObserverTrackStateOnSurfaceDecorator", **kwargs):
177 kwargs.setdefault("ContainerName", "InDetObservedTrackParticles")
178 kwargs.setdefault("DecorationPrefix", "ObservedTrack_")
179 kwargs.setdefault("PixelMsosName", "ObservedTrack_Pixel_MSOSs")
180 kwargs.setdefault("SctMsosName", "ObservedTrack_SCT_MSOSs")
181 kwargs.setdefault("TrtMsosName", "ObservedTrack_TRT_MSOSs")
182 kwargs.setdefault("AddPRD", True)
183 kwargs.setdefault("StoreHoles", False)
184 return TrackStateOnSurfaceDecoratorCfg(flags, name, **kwargs)
185
186def ObserverTSOS_CommonKernelCfg(flags, name="ObserverTSOS_CommonKernel"):
187 acc = ComponentAccumulator()
188 ObserverTrackStateOnSurfaceDecorator = acc.getPrimaryAndMerge(
189 ObserverTrackStateOnSurfaceDecoratorCfg(flags))
190 acc.addEventAlgo(CompFactory.DerivationFramework.CommonAugmentation(
191 name, AugmentationTools=[ObserverTrackStateOnSurfaceDecorator]))
192 return acc
193
194def PseudoTrackStateOnSurfaceDecoratorCfg(
195 flags, name="PseudoTrackStateOnSurfaceDecorator", **kwargs):
196 kwargs.setdefault("ContainerName", "InDetPseudoTrackParticles")
197 kwargs.setdefault("DecorationPrefix", "Pseudo_")
198 kwargs.setdefault("PixelMsosName", "Pseudo_Pixel_MSOSs")
199 kwargs.setdefault("SctMsosName", "Pseudo_SCT_MSOSs")
200 kwargs.setdefault("TrtMsosName", "Pseudo_TRT_MSOSs")
201 kwargs.setdefault("AddPRD", True)
202 kwargs.setdefault("StoreHoles", False)
203 return TrackStateOnSurfaceDecoratorCfg(flags, name, **kwargs)
204
205def PseudoTSOS_CommonKernelCfg(flags, name="PseudoTSOS_CommonKernel"):
206 acc = ComponentAccumulator()
207 PseudoTrackStateOnSurfaceDecorator = acc.getPrimaryAndMerge(
208 PseudoTrackStateOnSurfaceDecoratorCfg(flags))
209 acc.addEventAlgo(CompFactory.DerivationFramework.CommonAugmentation(
210 name, AugmentationTools=[PseudoTrackStateOnSurfaceDecorator]))
211 return acc
212
213def SiSPSeedsTrackStateOnSurfaceDecoratorCfg(
214 flags, name="SiSPSeedsTrackStateOnSurfaceDecorator", **kwargs):
215 kwargs.setdefault("ContainerName", "SiSPSeedSegmentsTrackParticles")
216 kwargs.setdefault("DecorationPrefix", "Reco_")
217 kwargs.setdefault("PixelMsosName", "SiSPSeedSegments_PixelMSOSs")
218 kwargs.setdefault("SctMsosName", "SiSPSeedSegments_SCT_MSOSs")
219 kwargs.setdefault("TrtMsosName", "SiSPSeedSegments_TRT_MSOSs")
220 kwargs.setdefault("AddPRD", True)
221 kwargs.setdefault("StoreHoles", False)
222 return TrackStateOnSurfaceDecoratorCfg(flags, name, **kwargs)
223
224def SiSPSeedsTSOS_CommonKernelCfg(flags, name="SiSPSeedsTSOS_CommonKernel",
225 listOfExtensions=[]):
226 acc = ComponentAccumulator()
227
228 listOfAugmTools = []
229 for extension in listOfExtensions:
230 SiSPSeedsTrackStateOnSurfaceDecorator = acc.getPrimaryAndMerge(
231 SiSPSeedsTrackStateOnSurfaceDecoratorCfg(
232 flags, name = f"SiSPSeeds{extension}TrackStateOnSurfaceDecorator",
233 ContainerName = f"SiSPSeedSegments{extension}TrackParticles",
234 PixelMsosName = f"SiSPSeedSegments{extension}_PixelMSOSs",
235 SctMsosName = f"SiSPSeedSegments{extension}_SCT_MSOSs",
236 TrtMsosName = f"SiSPSeedSegments{extension}_TRT_MSOSs"))
237 listOfAugmTools.append(SiSPSeedsTrackStateOnSurfaceDecorator)
238
239 acc.addEventAlgo(CompFactory.DerivationFramework.CommonAugmentation(
240 name, AugmentationTools=listOfAugmTools))
241 return acc
242
243def SiSPTrackStateOnSurfaceDecoratorCfg(
244 flags, name="SiSPTrackStateOnSurfaceDecorator", **kwargs):
245 kwargs.setdefault("ContainerName", "SiSPSeededTracksTrackParticles")
246 kwargs.setdefault("DecorationPrefix", "Reco_")
247 kwargs.setdefault("PixelMsosName", "SiSP_PixelMSOSs")
248 kwargs.setdefault("SctMsosName", "SiSP_SCT_MSOSs")
249 kwargs.setdefault("TrtMsosName", "SiSP_TRT_MSOSs")
250 kwargs.setdefault("AddPRD", True)
251 kwargs.setdefault("StoreHoles", False)
252 return TrackStateOnSurfaceDecoratorCfg(flags, name, **kwargs)
253
254def SiSPTSOS_CommonKernelCfg(flags, name="SiSPTSOS_CommonKernel",
255 listOfExtensions=[]):
256 acc = ComponentAccumulator()
257
258 listOfAugmTools = []
259 for extension in listOfExtensions:
260 SiSPTrackStateOnSurfaceDecorator = acc.getPrimaryAndMerge(
261 SiSPTrackStateOnSurfaceDecoratorCfg(
262 flags, name = f"SiSP{extension}TrackStateOnSurfaceDecorator",
263 ContainerName = f"SiSPSeededTracks{extension}TrackParticles",
264 PixelMsosName = f"SiSP{extension}_PixelMSOSs",
265 SctMsosName = f"SiSP{extension}_SCT_MSOSs",
266 TrtMsosName = f"SiSP{extension}_TRT_MSOSs"))
267 listOfAugmTools.append(SiSPTrackStateOnSurfaceDecorator)
268
269 acc.addEventAlgo(CompFactory.DerivationFramework.CommonAugmentation(
270 name, AugmentationTools=listOfAugmTools))
271 return acc
272
273def GSFTrackStateOnSurfaceDecoratorCfg(
274 flags, name="GSFTrackStateOnSurfaceDecorator", **kwargs):
275 kwargs.setdefault("ContainerName", "GSFTrackParticles")
276 kwargs.setdefault("DecorationPrefix", "GSF_")
277 kwargs.setdefault("PixelMsosName", "GSF_Pixel_MSOSs")
278 kwargs.setdefault("SctMsosName", "GSF_SCT_MSOSs")
279 kwargs.setdefault("TrtMsosName", "GSF_TRT_MSOSs")
280 kwargs.setdefault("PRDtoTrackMap", "")
281 return TrackStateOnSurfaceDecoratorCfg(flags, name, **kwargs)
282
283def GSFTSOS_CommonKernelCfg(flags, name="GSFTSOS_CommonKernel"):
284 if flags.Detector.GeometryITk:
285 return ITkGSFTSOS_CommonKernelCfg(flags)
286 acc = ComponentAccumulator()
287 GSFTrackStateOnSurfaceDecorator = acc.getPrimaryAndMerge(
288 GSFTrackStateOnSurfaceDecoratorCfg(flags))
289 acc.addEventAlgo(CompFactory.DerivationFramework.CommonAugmentation(
290 name, AugmentationTools=[GSFTrackStateOnSurfaceDecorator]))
291 return acc
292
293def ITkTrackStateOnSurfaceDecoratorCfg(
294 flags, name="TrackStateOnSurfaceDecorator", **kwargs):
295 """Configure the TSOS decorator"""
296 # To produce ITkStripDetectorElementCollection
297 from StripGeoModelXml.ITkStripGeoModelConfig import (
298 ITkStripReadoutGeometryCfg)
299 acc = ITkStripReadoutGeometryCfg(flags)
300
301 kwargs.setdefault("DecorationPrefix", "notSet")
302
303 if "Updator" not in kwargs:
304 from TrkConfig.TrkMeasurementUpdatorConfig import KalmanUpdatorCfg
305 kwargs.setdefault("Updator", acc.popToolsAndMerge(
306 KalmanUpdatorCfg(flags)))
307
308 if "TrackExtrapolator" not in kwargs:
309 from TrkConfig.AtlasExtrapolatorConfig import AtlasExtrapolatorCfg
310 kwargs.setdefault("TrackExtrapolator", acc.popToolsAndMerge(
311 AtlasExtrapolatorCfg(flags)))
312
313 if "HoleSearch" not in kwargs:
314 from InDetConfig.InDetTrackHoleSearchConfig import (
315 ITkTrackHoleSearchToolCfg)
316 kwargs.setdefault("HoleSearch", acc.popToolsAndMerge(
317 ITkTrackHoleSearchToolCfg(flags)))
318
319 if "ResidualPullCalculator" not in kwargs:
320 from TrkConfig.TrkResidualPullCalculatorConfig import (
321 ResidualPullCalculatorCfg)
322 kwargs.setdefault("ResidualPullCalculator", acc.popToolsAndMerge(
323 ResidualPullCalculatorCfg(flags)))
324
325 kwargs.setdefault("AddExtraEventInfo", False) # Controls decoration of TrtPhaseTime
326 kwargs.setdefault("PixelMapName", "ITkPixelClustersOffsets")
327 kwargs.setdefault("SctMapName", "ITkStripClustersOffsets")
328 kwargs.setdefault("PixelClustersName", "ITkPixelMeasurements")
329 kwargs.setdefault("SctClustersName", "ITkStripMeasurements")
330 kwargs.setdefault("PRDtoTrackMap", "ITkPRDtoTrackMapMerge_CombinedITkTracks")
331 kwargs.setdefault("PixelMsosName", "ITkPixelMSOSs")
332 kwargs.setdefault("SctMsosName", "ITkStripMSOSs")
333 kwargs.setdefault("SCTDetEleCollKey", "ITkStripDetectorElementCollection")
334
335 acc.addPublicTool(
336 CompFactory.DerivationFramework.TrackStateOnSurfaceDecorator(
337 name, **kwargs), primary=True)
338 return acc
339
340def ITkTSOS_CommonKernelCfg(flags, name="ITkTSOS_CommonKernel",
341 listOfExtensions=[]):
342 acc = ComponentAccumulator()
343 listOfAugmTools = []
344 for extension in listOfExtensions:
345 TrackStateOnSurfaceDecorator = acc.getPrimaryAndMerge(
346 ITkTrackStateOnSurfaceDecoratorCfg(flags,
347 DecorationPrefix = "Reco_"))
348 listOfAugmTools.append(TrackStateOnSurfaceDecorator)
349
350 acc.addEventAlgo(CompFactory.DerivationFramework.CommonAugmentation(
351 name, AugmentationTools=listOfAugmTools))
352 return acc
353
354def DFITkTrackStateOnSurfaceDecoratorCfg(
355 flags, name="DFITkTrackStateOnSurfaceDecorator", **kwargs):
356 kwargs.setdefault("StorePixel", flags.ITk.DAODStorePixel)
357 kwargs.setdefault("StoreSCT", flags.ITk.DAODStoreStrip)
358 kwargs.setdefault("DecorationPrefix", "")
359 kwargs.setdefault("PRDtoTrackMap", "")
360 kwargs.setdefault("OutputLevel", INFO)
361 return ITkTrackStateOnSurfaceDecoratorCfg(flags, name, **kwargs)
362
363def DFITkTSOSKernelCfg(flags, name='DFITkTSOSKernel'):
364 acc = ComponentAccumulator()
365
366 # ====================================================================
367 # AUGMENTATION TOOLS
368 # ====================================================================
369 tsos_augmentationTools = []
370
371 from InDetConfig.ITkActsHelpers import primaryPassUsesActs
372 if primaryPassUsesActs(flags):
373 logIDTools.warning("No TrackStateOnSurfaceDecorator available yet for ACTS")
374 else:
375 DFTSOS = acc.getPrimaryAndMerge(DFITkTrackStateOnSurfaceDecoratorCfg(flags))
376 tsos_augmentationTools.append(DFTSOS)
377
378 # shared between IDTIDE and IDTRKVALID
379 acc.addEventAlgo(CompFactory.DerivationFramework.DerivationKernel(
380 name,
381 AugmentationTools=tsos_augmentationTools,
382 ThinningTools=[],
383 OutputLevel=INFO))
384 return acc
385
386def ITkSiSPSeedsTrackStateOnSurfaceDecoratorCfg(
387 flags, name="ITkSiSPSeedsTrackStateOnSurfaceDecorator", **kwargs):
388 kwargs.setdefault("ContainerName", "SiSPSeedSegmentsTrackParticles")
389 kwargs.setdefault("DecorationPrefix", "Reco_")
390 kwargs.setdefault("PixelMsosName", "SiSPSeedSegments_ITkPixelMSOSs")
391 kwargs.setdefault("SctMsosName", "SiSPSeedSegments_ITkStripMSOSs")
392 kwargs.setdefault("TrtMsosName", "")
393 kwargs.setdefault("AddPRD", True)
394 kwargs.setdefault("StoreHoles", False)
395 return ITkTrackStateOnSurfaceDecoratorCfg(flags, name, **kwargs)
396
397def ITkSiSPSeedsTSOS_CommonKernelCfg(flags, name="ITkSiSPSeedsTSOS_CommonKernel",
398 listOfExtensions=[]):
399 acc = ComponentAccumulator()
400
401 listOfAugmTools = []
402 for extension in listOfExtensions:
403 SiSPSeedsTrackStateOnSurfaceDecorator = acc.getPrimaryAndMerge(
404 ITkSiSPSeedsTrackStateOnSurfaceDecoratorCfg(
405 flags, name = f"SiSPSeeds{extension}TrackStateOnSurfaceDecorator",
406 ContainerName = f"SiSPSeedSegments{extension}TrackParticles",
407 PixelMsosName = f"SiSPSeedSegments{extension}_ITkPixelMSOSs",
408 SctMsosName = f"SiSPSeedSegments{extension}_ITkStripMSOSs"))
409 listOfAugmTools.append(SiSPSeedsTrackStateOnSurfaceDecorator)
410
411 acc.addEventAlgo(CompFactory.DerivationFramework.CommonAugmentation(
412 name, AugmentationTools=listOfAugmTools))
413 return acc
414
415def ITkSiSPTrackStateOnSurfaceDecoratorCfg(
416 flags, name="SiSPTrackStateOnSurfaceDecorator", **kwargs):
417 kwargs.setdefault("ContainerName", "SiSPSeededTracksTrackParticles")
418 kwargs.setdefault("DecorationPrefix", "SiSP_")
419 kwargs.setdefault("PixelMsosName", "SiSP_ITkPixel_MSOSs")
420 kwargs.setdefault("SctMsosName", "SiSP_ITkStrip_MSOSs")
421 kwargs.setdefault("AddPRD", True)
422 kwargs.setdefault("StoreHoles", False)
423 return ITkTrackStateOnSurfaceDecoratorCfg(flags, name, **kwargs)
424
425def ITkSiSPTSOS_CommonKernelCfg(flags, name="ITkSiSPTSOS_CommonKernel",
426 listOfExtensions=[]):
427 acc = ComponentAccumulator()
428 listOfAugmTools = []
429 for extension in listOfExtensions:
430 ITkSiSPTrackStateOnSurfaceDecorator = acc.getPrimaryAndMerge(
431 ITkSiSPTrackStateOnSurfaceDecoratorCfg(
432 flags, name = f"SiSP{extension}TrackStateOnSurfaceDecorator",
433 ContainerName = f"SiSPSeededTracks{extension}TrackParticles",
434 PixelMsosName = f"SiSP{extension}_ITkPixel_MSOSs",
435 SctMsosName = f"SiSP{extension}_ITkStrip_MSOSs"))
436 listOfAugmTools.append(ITkSiSPTrackStateOnSurfaceDecorator)
437
438 acc.addEventAlgo(CompFactory.DerivationFramework.CommonAugmentation(
439 name, AugmentationTools=listOfAugmTools))
440 return acc
441
442def ITkGSFTrackStateOnSurfaceDecoratorCfg(
443 flags, name="ITkGSFTrackStateOnSurfaceDecorator", **kwargs):
444 kwargs.setdefault("ContainerName", "GSFTrackParticles")
445 kwargs.setdefault("DecorationPrefix", "GSF_")
446 kwargs.setdefault("PixelMsosName", "GSF_ITkPixel_MSOSs")
447 kwargs.setdefault("SctMsosName", "GSF_ITkStrip_MSOSs")
448 kwargs.setdefault("TrtMsosName", "")
449 kwargs.setdefault("PRDtoTrackMap", "")
450 kwargs.setdefault("AddExtraEventInfo", flags.Beam.Type is BeamType.Cosmics)
451 return ITkTrackStateOnSurfaceDecoratorCfg(flags, name, **kwargs)
452
453def ITkGSFTSOS_CommonKernelCfg(flags, name="ITkGSFTSOS_CommonKernel"):
454 acc = ComponentAccumulator()
455 GSFTrackStateOnSurfaceDecorator = acc.getPrimaryAndMerge(
456 ITkGSFTrackStateOnSurfaceDecoratorCfg(flags))
457 acc.addEventAlgo(CompFactory.DerivationFramework.CommonAugmentation(
458 name, AugmentationTools=[GSFTrackStateOnSurfaceDecorator]))
459 return acc
460
461# Expression of Z0 at the primary vertex
462
463
464def TrackParametersAtPVCfg(flags, name, **kwargs):
465 """Configure the TrackParametersAtPV tool"""
466 acc = ComponentAccumulator()
467 acc.addPublicTool(CompFactory.DerivationFramework.TrackParametersAtPV(
468 name, **kwargs), primary=True)
469 return acc
470
471# Pseudotrack selector
472
473
474def PseudoTrackSelectorCfg(flags, name, **kwargs):
475 """Configure the pseudotrack selector"""
476 acc = ComponentAccumulator()
477
478 if "trackTruthOriginTool" not in kwargs:
479 from InDetTrackSystematicsTools.InDetTrackSystematicsToolsConfig import InDetTrackTruthOriginToolCfg
480 kwargs.setdefault("trackTruthOriginTool", acc.popToolsAndMerge(
481 InDetTrackTruthOriginToolCfg(flags)))
482
483 acc.addPublicTool(
484 CompFactory.DerivationFramework.PseudoTrackSelector(
485 name, **kwargs), primary=True)
486 return acc
487
488# Tool for decorating tracks with the outcome of the track selector tool
489
490
491def InDetTrackSelectionToolWrapperCfg(
492 flags, name, CutLevel="TightPrimary", **kwargs):
493 """Configure the InDetTrackSelectionToolWrapper"""
494 acc = ComponentAccumulator()
495
496 if "TrackSelectionTool" not in kwargs:
497 from InDetTrackSelectionTool.InDetTrackSelectionToolConfig import (
498 InDetTrackSelectionToolCfg)
499 kwargs.setdefault("TrackSelectionTool", acc.popToolsAndMerge(
500 InDetTrackSelectionToolCfg(
501 flags, name="InDetTrackSelectionTool_"+CutLevel,
502 CutLevel=CutLevel)))
503
504 acc.addPublicTool(
505 CompFactory.DerivationFramework.InDetTrackSelectionToolWrapper(
506 name, **kwargs), primary=True)
507 return acc
508
509# Tool for decorating the track and calo isolation
510
511
512def IsolationTrackDecoratorCfg(flags, name, **kwargs):
513 """Confiure the isolation track decorator tool"""
514 acc = ComponentAccumulator()
515 IsolationTrackDecorator = CompFactory.DerivationFramework.IsolationTrackDecorator
516 acc.addPublicTool(IsolationTrackDecorator(name, **kwargs),
517 primary = True)
518 return acc
519
520# Tool for thinning TrackParticle containers via string selection
521
522
523def TrackParticleThinningCfg(flags, name, **kwargs):
524 """Configure the TrackParticleThining tool"""
525 if flags.Detector.GeometryITk:
526 return ITkTrackParticleThinningCfg(flags, name, **kwargs)
527
528 # To produce SCT_DetectorElementCollection
529 from SCT_GeoModel.SCT_GeoModelConfig import SCT_ReadoutGeometryCfg
530 acc = SCT_ReadoutGeometryCfg(flags)
531
532 acc.addPublicTool(CompFactory.DerivationFramework.TrackParticleThinning(
533 name, **kwargs), primary=True)
534 return acc
535
536# Tool for thinning TrackParticle containers via string selection
537
538
539def TrackParticleThinningPHYSCfg(flags, name, **kwargs):
540 """Configure the TrackParticleThining tool"""
541 if flags.Detector.GeometryITk:
542 return ITkTrackParticleThinningPHYSCfg(flags, name, **kwargs)
543
544 # To produce SCT_DetectorElementCollection
545 from SCT_GeoModel.SCT_GeoModelConfig import SCT_ReadoutGeometryCfg
546 acc = SCT_ReadoutGeometryCfg(flags)
547 acc.addPublicTool(CompFactory.DerivationFramework.TrackParticleThinningPHYS(
548 name, **kwargs), primary=True)
549 return acc
550
551# Tool for thinning PixelClusters via string selection
552
553
554def PixelClusterThinningCfg(
555 flags, name="PixelClusterThinningTool", **kwargs):
556 pix_lumi_sel = "PixelClusters.bec == 0 && PixelClusters.layer >= 0 && PixelClusters.sizeZ > 1 && ( PixelClusters.nRDO < PixelClusters.sizeZ + 5 ) && PixelClusters.charge > 30000"
557 kwargs.setdefault("SelectionString", pix_lumi_sel)
558 acc = ComponentAccumulator()
559 acc.addPublicTool(
560 CompFactory.DerivationFramework.TrackMeasurementThinning(
561 name, **kwargs), primary=True)
562 return acc
563
564
565def IDTIDEThinningToolCfg(flags, name="IDTIDEThinningTool", **kwargs):
566 if not flags.Detector.EnablePixel:
567 kwargs.setdefault("InDetTrackStatesPixKey", "")
568 kwargs.setdefault("InDetTrackMeasurementsPixKey", "")
569 if not flags.Detector.EnableSCT:
570 kwargs.setdefault("InDetTrackStatesSctKey", "")
571 kwargs.setdefault("InDetTrackMeasurementsSctKey", "")
572 if not flags.Detector.EnableTRT:
573 kwargs.setdefault("InDetTrackStatesTrtKey", "")
574 kwargs.setdefault("InDetTrackMeasurementsTrtKey", "")
575
576 kwargs.setdefault("SelectionString", "abs(InDetTrackParticles.IDTIDEInDetTrackZ0AtPV) < 5.0")
577 # If true, Complains about missing PixelMSOSs
578 kwargs.setdefault("ThinHitsOnTrack", False)
579
580 return TrackParticleThinningCfg(flags, name, **kwargs)
581
582def ITkTrackParticleThinningCfg(flags, name, **kwargs):
583 """Configure the TrackParticleThining tool"""
584 # To produce ITkStripDetectorElementCollection
585 from StripGeoModelXml.ITkStripGeoModelConfig import (
586 ITkStripReadoutGeometryCfg)
587 acc = ITkStripReadoutGeometryCfg(flags)
588
589 kwargs.setdefault("InDetTrackStatesPixKey", "ITkPixelMSOSs")
590 kwargs.setdefault("InDetTrackMeasurementsPixKey", "ITkPixelMeasurements")
591 kwargs.setdefault("InDetTrackStatesSctKey", "ITkStripMSOSs")
592 kwargs.setdefault("InDetTrackMeasurementsSctKey", "ITkStripMeasurements")
593 kwargs.setdefault("InDetTrackStatesTrtKey", "")
594 kwargs.setdefault("InDetTrackMeasurementsTrtKey", "")
595 kwargs.setdefault("SCTDetEleCollKey", "ITkStripDetectorElementCollection")
596
597 acc.addPublicTool(CompFactory.DerivationFramework.TrackParticleThinning(
598 name, **kwargs), primary=True)
599 return acc
600
601def ITkTrackParticleThinningPHYSCfg(flags, name, **kwargs):
602 """Configure the TrackParticleThiningPHYS tool"""
603 # To produce ITkStripDetectorElementCollection
604 from StripGeoModelXml.ITkStripGeoModelConfig import (
605 ITkStripReadoutGeometryCfg)
606 acc = ITkStripReadoutGeometryCfg(flags)
607
608 kwargs.setdefault("InDetTrackStatesPixKey", "ITkPixelMSOSs")
609 kwargs.setdefault("InDetTrackMeasurementsPixKey", "ITkPixelMeasurements")
610 kwargs.setdefault("InDetTrackStatesSctKey", "ITkStripMSOSs")
611 kwargs.setdefault("InDetTrackMeasurementsSctKey", "ITkStripMeasurements")
612 kwargs.setdefault("InDetTrackStatesTrtKey", "")
613 kwargs.setdefault("InDetTrackMeasurementsTrtKey", "")
614 kwargs.setdefault("SCTDetEleCollKey", "ITkStripDetectorElementCollection")
615
616 acc.addPublicTool(CompFactory.DerivationFramework.TrackParticleThinningPHYS(
617 name, **kwargs), primary=True)
618 return acc
619
620def ITkTIDEThinningToolCfg(flags, name="ITkTIDEThinningTool", **kwargs):
621 if not flags.Detector.EnableITkPixel:
622 kwargs.setdefault("InDetTrackStatesPixKey", "")
623 kwargs.setdefault("InDetTrackMeasurementsPixKey", "")
624 if not flags.Detector.EnableITkStrip:
625 kwargs.setdefault("InDetTrackStatesSctKey", "")
626 kwargs.setdefault("InDetTrackMeasurementsSctKey", "")
627
628 kwargs.setdefault("SelectionString", "abs(InDetTrackParticles.IDTIDEInDetTrackZ0AtPV) < 5.0")
629 # If true, Complains about missing PixelMSOSs
630 kwargs.setdefault("ThinHitsOnTrack", False)
631
632 return ITkTrackParticleThinningCfg(flags, name, **kwargs)
633
634# Tool for thinning TrackParticles that aren't associated with muons
635
636
637def MuonTrackParticleThinningCfg(flags, name, **kwargs):
638 """Configure the MuonTrackParticleThinning tool"""
639 acc = ComponentAccumulator()
640 acc.addPublicTool(
641 CompFactory.DerivationFramework.MuonTrackParticleThinning(
642 name, **kwargs), primary=True)
643 return acc
644
645# Tool for thinning TrackParticles that aren't associated with taus
646
647
648def TauTrackParticleThinningCfg(flags, name, **kwargs):
649 """Configure the TauTrackParticleThinning tool"""
650 acc = ComponentAccumulator()
651 acc.addPublicTool(CompFactory.DerivationFramework.TauTrackParticleThinning(
652 name, **kwargs), primary=True)
653 return acc
654
655# Tool for thinning TrackParticles that aren't associated high-pt di-taus
656
657
658def DiTauTrackParticleThinningCfg(flags, name, **kwargs):
659 """Configure the DiTauTrackParticleThinning tool"""
660 acc = ComponentAccumulator()
661 acc.addPublicTool(
662 CompFactory.DerivationFramework.DiTauTrackParticleThinning(
663 name, **kwargs), primary=True)
664 return acc
665
666# Tool for thinning TrackParticles that are associated with jets
667
668
669def JetTrackParticleThinningCfg(flags, name, **kwargs):
670 """Configure the JetTrackParticleThinning tool"""
671 acc = ComponentAccumulator()
672 acc.addPublicTool(CompFactory.DerivationFramework.JetTrackParticleThinning(
673 name, **kwargs), primary=True)
674 return acc
675
676
677def UFOTrackParticleThinningCfg(flags, name, **kwargs):
678 """Configure the UFOTrackParticleThinning tool"""
679 acc = ComponentAccumulator()
680 acc.addPublicTool(CompFactory.DerivationFramework.UFOTrackParticleThinning(
681 name, **kwargs), primary=True)
682 return acc
683
684
685def JetConstituentThinningCfg(flags, name, **kwargs):
686 """Configure the JetConstituentThinning tool"""
687 acc = ComponentAccumulator()
688 acc.addPublicTool(CompFactory.DerivationFramework.JetConstituentThinning(
689 name, **kwargs), primary=True)
690 return acc
691
692
693def JetGhostThinningCfg(flags, name, **kwargs):
694 """Configure the JetGhostThinning tool"""
695 acc = ComponentAccumulator()
696 acc.addPublicTool(CompFactory.DerivationFramework.JetGhostThinning(
697 name, **kwargs), primary=True)
698 return acc
699
700
701def TauJetLepRMParticleThinningCfg(flags, name, **kwargs):
702 """Configure the DiTauTrackParticleThinning tool"""
703 acc = ComponentAccumulator()
704 acc.addPublicTool(
705 CompFactory.DerivationFramework.TauJets_LepRMParticleThinning(
706 name, **kwargs), primary=True)
707 return acc
708
709# Tool for thinning TrackParticles that aren't associated with egamma objects
710
711
712def EgammaTrackParticleThinningCfg(flags, name, **kwargs):
713 """Configure the EgammaTrackParticleThinning tool"""
714 acc = ComponentAccumulator()
715 acc.addPublicTool(
716 CompFactory.DerivationFramework.EgammaTrackParticleThinning(
717 name, **kwargs), primary=True)
718 return acc
719
720# Track to vertex wrapper
721
722
723def TrackToVertexWrapperCfg(flags, name, **kwargs):
724 """Configure the TrackToVertexWrapper tool"""
725 acc = ComponentAccumulator()
726
727 if "TrackToVertexIPEstimator" not in kwargs:
728 from TrkConfig.TrkVertexFitterUtilsConfig import (
729 TrackToVertexIPEstimatorCfg)
730 kwargs.setdefault("TrackToVertexIPEstimator", acc.popToolsAndMerge(
731 TrackToVertexIPEstimatorCfg(flags)))
732
733 kwargs.setdefault("ContainerName", "InDetTrackParticles")
734
735 acc.addPublicTool(CompFactory.DerivationFramework.TrackToVertexWrapper(
736 name, **kwargs), primary=True)
737 return acc
738
739
740def IDTIDETruthThinningToolCfg(flags, name="IDTIDETruthThinningTool", **kwargs):
741 kwargs.setdefault("WritePartons", True)
742 kwargs.setdefault("WriteHadrons", True)
743 kwargs.setdefault("WriteBHadrons", True)
744 kwargs.setdefault("WriteGeant", True)
745 kwargs.setdefault("GeantPhotonPtThresh", 20000)
746 kwargs.setdefault("WriteTauHad", True)
747 kwargs.setdefault("PartonPtThresh", -1.0)
748 kwargs.setdefault("WriteBSM", True)
749 kwargs.setdefault("WriteBosons", True)
750 kwargs.setdefault("WriteBosonProducts", True)
751 kwargs.setdefault("WriteBSMProducts", True)
752 kwargs.setdefault("WriteTopAndDecays", True)
753 kwargs.setdefault("WriteEverything", True)
754 kwargs.setdefault("WriteAllLeptons", True)
755 kwargs.setdefault("WriteLeptonsNotFromHadrons", True)
756 kwargs.setdefault("WriteNotPhysical", True)
757 kwargs.setdefault("WriteFirstN", -1)
758 kwargs.setdefault("PreserveAncestors", True)
759 kwargs.setdefault("PreserveGeneratorDescendants", True)
760
761 from DerivationFrameworkMCTruth.TruthDerivationToolsConfig import (
762 MenuTruthThinningCfg)
763 return MenuTruthThinningCfg(flags, name, **kwargs)
764
765def IDTRKVALIDTruthThinningToolCfg(
766 flags, name="IDTRKVALIDTruthThinningTool", **kwargs):
767
768 kwargs.setdefault("WriteEverything", True)
769 kwargs.setdefault("WriteFirstN", -1)
770 kwargs.setdefault("PreserveAncestors", True)
771 kwargs.setdefault("PreserveGeneratorDescendants", True)
772
773 from DerivationFrameworkMCTruth.TruthDerivationToolsConfig import (
774 MenuTruthThinningCfg)
775 return MenuTruthThinningCfg(flags, name, **kwargs)
776
777def EventInfoBSErrDecoratorCfg(
778 flags, name="EventInfoBSErrDecorator", **kwargs):
779 acc = ComponentAccumulator()
780 acc.addPublicTool(CompFactory.DerivationFramework.EventInfoBSErrDecorator(
781 name, **kwargs), primary=True)
782 return acc
783
784def UnassociatedHitsGetterToolCfg(
785 flags, name="UnassociatedHitsGetterTool", **kwargs):
786 acc = ComponentAccumulator()
787
788 if "AssociationTool" not in kwargs:
789 from InDetConfig.InDetAssociationToolsConfig import (
790 InDetPRDtoTrackMapToolGangedPixelsCfg)
791 kwargs.setdefault("AssociationTool", acc.popToolsAndMerge(
792 InDetPRDtoTrackMapToolGangedPixelsCfg(flags)))
793
794 acc.setPrivateTools(
795 CompFactory.DerivationFramework.UnassociatedHitsGetterTool(name, **kwargs))
796 return acc
797
798def UnassociatedHitsDecoratorCfg(
799 flags, name="UnassociatedHitsDecorator", **kwargs):
800 acc = ComponentAccumulator()
801
802 if "UnassociatedHitsGetter" not in kwargs:
803 kwargs.setdefault("UnassociatedHitsGetter", acc.popToolsAndMerge(
804 UnassociatedHitsGetterToolCfg(flags)))
805
806 acc.addPublicTool(CompFactory.DerivationFramework.UnassociatedHitsDecorator(
807 name, **kwargs), primary=True)
808 return acc
python.InDetToolsConfig.InDetLRTMergeCfg
InDetLRTMergeCfg(flags, name="InDetLRTMerge", **kwargs)
Definition InDetToolsConfig.py:16
python.InDetToolsConfig.UsedInVertexFitTrackDecoratorCfg
UsedInVertexFitTrackDecoratorCfg(flags, name="UsedInVertexFitTrackDecorator", **kwargs)
Definition InDetToolsConfig.py:31
Generated on for ATLAS Offline Software by doxygen 1.17.0