ATLAS Offline Software
ActsClusterizationConfig.py
Go to the documentation of this file.
1 # Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
2 
3 from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
4 from AthenaConfiguration.ComponentFactory import CompFactory
5 from AthenaConfiguration.Enums import BeamType
6 from ActsConfig.ActsUtilities import extractChildKwargs
7 
8 def ActsHgtdClusteringToolCfg(flags,
9  name: str = "ActsHgtdClusteringTool",
10  **kwargs) -> ComponentAccumulator:
11  acc = ComponentAccumulator()
12  acc.setPrivateTools(CompFactory.ActsTrk.HgtdClusteringTool(name, **kwargs))
13  return acc
14 
15 def ActsHgtdTimedClusteringToolCfg(flags,
16  name: str = "ActsHgtdTimedClusteringTool",
17  **kwargs) -> ComponentAccumulator:
18  acc = ComponentAccumulator()
19  acc.setPrivateTools(CompFactory.ActsTrk.HgtdTimedClusteringTool(name, **kwargs))
20  return acc
21 
22 def ActsHgtdClusterizationAlgCfg(flags,
23  name: str = "ActsHgtdClusterizationAlg",
24  **kwargs) -> ComponentAccumulator:
25  acc = ComponentAccumulator()
26 
27  if flags.HGTD.Geometry.useGeoModelXml:
28  from HGTD_GeoModelXml.HGTD_GeoModelConfig import HGTD_ReadoutGeometryCfg
29  else:
30  from HGTD_GeoModel.HGTD_GeoModelConfig import HGTD_ReadoutGeometryCfg
31  acc.merge(HGTD_ReadoutGeometryCfg(flags))
32 
33  kwargs.setdefault('RDOContainerName', 'HGTD_RDOs')
34  kwargs.setdefault('ClusterContainerName', 'HGTD_Clusters')
35 
36  if 'ClusteringTool' not in kwargs:
37  from HGTD_Config.HGTD_ConfigFlags import ClusteringStrategy
38  if flags.HGTD.Acts.ClusteringStrategy is ClusteringStrategy.SinglePad:
39  kwargs.setdefault('ClusteringTool', acc.popToolsAndMerge(ActsHgtdClusteringToolCfg(flags)))
40  else:
41  kwargs.setdefault('ClusteringTool', acc.popToolsAndMerge(ActsHgtdTimedClusteringToolCfg(flags)))
42 
43  if flags.Acts.doMonitoring and 'MonTool' not in kwargs:
44  from ActsConfig.ActsMonitoringConfig import ActsHgtdClusterizationMonitoringToolCfg
45  kwargs.setdefault('MonTool', acc.popToolsAndMerge(ActsHgtdClusterizationMonitoringToolCfg(flags)))
46 
47  acc.addEventAlgo(CompFactory.ActsTrk.HgtdClusterizationAlg(name, **kwargs))
48  return acc
49 
50 
51 
52 def ActsPixelClusteringToolCfg(flags,
53  name: str = "ActsPixelClusteringTool",
54  **kwargs) -> ComponentAccumulator:
55  acc = ComponentAccumulator()
56 
57  if flags.Acts.Clusters.RetrieveChargeInformation:
58  from PixelConditionsAlgorithms.ITkPixelConditionsConfig import ITkPixelChargeCalibCondAlgCfg, ITkPixelOfflineCalibCondAlgCfg
59  acc.merge(ITkPixelChargeCalibCondAlgCfg(flags))
60  acc.merge(ITkPixelOfflineCalibCondAlgCfg(flags))
61  kwargs.setdefault('PixelChargeCalibCondData', 'ITkPixelChargeCalibCondData')
62 
63  from PixelReadoutGeometry.PixelReadoutGeometryConfig import ITkPixelReadoutManagerCfg
64  acc.merge(ITkPixelReadoutManagerCfg(flags))
65 
66  if "PixelLorentzAngleTool" not in kwargs:
67  from SiLorentzAngleTool.ITkPixelLorentzAngleConfig import ITkPixelLorentzAngleToolCfg
68  kwargs.setdefault("PixelLorentzAngleTool", acc.popToolsAndMerge( ITkPixelLorentzAngleToolCfg(flags) ))
69 
70  kwargs.setdefault('UseWeightedPosition', flags.Acts.Clusters.UseWeightedPosition)
71  kwargs.setdefault('UseBroadErrors', flags.Beam.Type is BeamType.Cosmics)
72 
73  acc.setPrivateTools(CompFactory.ActsTrk.PixelClusteringTool(name, **kwargs))
74  return acc
75 
76 
77 def ActsStripClusteringToolCfg(flags,
78  name: str = "ActsStripClusteringTool",
79  **kwargs) -> ComponentAccumulator:
80  acc = ComponentAccumulator()
81 
82  if 'LorentzAngleTool' not in kwargs:
83  from SiLorentzAngleTool.ITkStripLorentzAngleConfig import ITkStripLorentzAngleToolCfg
84  kwargs.setdefault("LorentzAngleTool", acc.popToolsAndMerge(ITkStripLorentzAngleToolCfg(flags)))
85 
86  kwargs.setdefault("conditionsTool",None)
87  if "StripDetElStatus" not in kwargs :
88  from SCT_ConditionsAlgorithms.ITkStripConditionsAlgorithmsConfig import (
89  ITkStripDetectorElementStatusAlgCfg)
90  acc.merge(ITkStripDetectorElementStatusAlgCfg(flags))
91  kwargs.setdefault("StripDetElStatus", "ITkStripDetectorElementStatus")
92 
93  # Disable noisy modules suppression
94  kwargs.setdefault("maxFiredStrips", 0)
95 
96  if flags.ITk.selectStripIntimeHits and 'timeBins' not in kwargs:
97  coll_25ns = flags.Beam.BunchSpacing<=25 and flags.Beam.Type is BeamType.Collisions
98  kwargs.setdefault("timeBins", "01X" if coll_25ns else "X1X")
99 
100  acc.setPrivateTools(CompFactory.ActsTrk.StripClusteringTool(name, **kwargs))
101  return acc
102 
103 def ActsPixelClusterizationAlgCfg(flags,
104  name: str = 'ActsPixelClusterizationAlg',
105  *,
106  useCache: bool = False,
107  **kwargs) -> ComponentAccumulator:
108  acc = ComponentAccumulator()
109 
110  kwargs.setdefault("IDHelper", "PixelID")
111  kwargs.setdefault("RDOContainerKey", "ITkPixelRDOs")
112  kwargs.setdefault("ClustersKey", "ITkPixelClusters")
113  kwargs.setdefault("DetEleCollKey", "ITkPixelDetectorElementCollection")
114  # Regional selection
115  kwargs.setdefault('RoIs', 'ActsRegionOfInterest')
116 
117  kwargs.setdefault('ClusterCacheBackend', 'ActsPixelClusterCache_Back')
118  kwargs.setdefault('ClusterCache', 'ActsPixelClustersCache')
119 
120  if 'RegSelTool' not in kwargs:
121  from RegionSelector.RegSelToolConfig import regSelTool_ITkPixel_Cfg
122  kwargs.setdefault('RegSelTool', acc.popToolsAndMerge(regSelTool_ITkPixel_Cfg(flags)))
123 
124  if 'ClusteringTool' not in kwargs:
125  kwargs.setdefault("ClusteringTool", acc.popToolsAndMerge(ActsPixelClusteringToolCfg(flags)))
126 
127  if 'DetElStatus' not in kwargs:
128  from PixelConditionsAlgorithms.ITkPixelConditionsConfig import ITkPixelDetectorElementStatusAlgCfg
129  acc.merge(ITkPixelDetectorElementStatusAlgCfg(flags))
130  kwargs.setdefault('DetElStatus', 'ITkPixelDetectorElementStatus')
131 
132  if flags.Acts.doMonitoring and 'MonTool' not in kwargs:
133  from ActsConfig.ActsMonitoringConfig import ActsITkPixelClusterizationMonitoringToolCfg
134  kwargs.setdefault('MonTool', acc.popToolsAndMerge(ActsITkPixelClusterizationMonitoringToolCfg(flags)))
135 
136  if not useCache:
137  acc.addEventAlgo(CompFactory.ActsTrk.PixelClusterizationAlg(name, **kwargs))
138  else:
139  acc.addEventAlgo(CompFactory.ActsTrk.PixelCacheClusterizationAlg(name, **kwargs))
140  return acc
141 
142 def ActsStripClusterizationAlgCfg(flags,
143  name: str = 'ActsStripClusterizationAlg',
144  useCache: bool = False,
145  **kwargs) -> ComponentAccumulator:
146  acc = ComponentAccumulator()
147 
148  kwargs.setdefault("RDOContainerKey", "ITkStripRDOs")
149  kwargs.setdefault("ClustersKey", "ITkStripClusters")
150  kwargs.setdefault("IDHelper", "SCT_ID")
151  kwargs.setdefault("DetEleCollKey", "ITkStripDetectorElementCollection")
152  # Regional selection
153  kwargs.setdefault('RoIs', 'ActsRegionOfInterest')
154 
155  kwargs.setdefault('ClusterCacheBackend', 'ActsStripClusterCache_Back')
156  kwargs.setdefault('ClusterCache', 'ActsStripClustersCache')
157 
158  if 'RegSelTool' not in kwargs:
159  from RegionSelector.RegSelToolConfig import regSelTool_ITkStrip_Cfg
160  kwargs.setdefault('RegSelTool', acc.popToolsAndMerge(regSelTool_ITkStrip_Cfg(flags)))
161 
162  if 'DetElStatus' not in kwargs :
163  from SCT_ConditionsAlgorithms.ITkStripConditionsAlgorithmsConfig import ITkStripDetectorElementStatusAlgCfg
164  acc.merge(ITkStripDetectorElementStatusAlgCfg(flags))
165  kwargs.setdefault("DetElStatus", "ITkStripDetectorElementStatus")
166 
167  if 'ClusteringTool' not in kwargs:
168  kwargs.setdefault("ClusteringTool", acc.popToolsAndMerge(ActsStripClusteringToolCfg(flags)))
169 
170  if flags.Acts.doMonitoring and 'MonTool' not in kwargs:
171  from ActsConfig.ActsMonitoringConfig import ActsITkStripClusterizationMonitoringToolCfg
172  kwargs.setdefault('MonTool', acc.popToolsAndMerge(ActsITkStripClusterizationMonitoringToolCfg(flags)))
173 
174  if not useCache:
175  acc.addEventAlgo(CompFactory.ActsTrk.StripClusterizationAlg(name, **kwargs))
176  else:
177  acc.addEventAlgo(CompFactory.ActsTrk.StripCacheClusterizationAlg(name, **kwargs))
178  return acc
179 
180 def ActsClusterCacheCreatorAlgCfg(flags,
181  name: str = "ActsClusterCacheCreatorAlg",
182  **kwargs) -> ComponentAccumulator:
183  acc = ComponentAccumulator()
184  kwargs.setdefault("PixelClustersCacheKey", "ActsPixelClusterCache_Back")
185  kwargs.setdefault("StripClustersCacheKey", "ActsStripClusterCache_Back")
186  acc.addEventAlgo(CompFactory.ActsTrk.Cache.CreatorAlg(name, **kwargs))
187  return acc
188 
189 def ActsPixelClusterPreparationAlgCfg(flags,
190  name: str = "ActsPixelClusterPreparationAlg",
191  useCache: bool = False,
192  **kwargs) -> ComponentAccumulator:
193  acc = ComponentAccumulator()
194 
195  kwargs.setdefault('InputCollection', 'ITkPixelClusters')
196  kwargs.setdefault('DetectorElements', 'ITkPixelDetectorElementCollection')
197 
198  if 'RegSelTool' not in kwargs:
199  from RegionSelector.RegSelToolConfig import regSelTool_ITkPixel_Cfg
200  kwargs.setdefault('RegSelTool', acc.popToolsAndMerge(regSelTool_ITkPixel_Cfg(flags)))
201 
202  if flags.Acts.doMonitoring and 'MonTool' not in kwargs:
203  from ActsConfig.ActsMonitoringConfig import ActsDataPreparationMonitoringToolCfg
204  kwargs.setdefault('MonTool', acc.popToolsAndMerge(ActsDataPreparationMonitoringToolCfg(flags,
205  name = "ActsPixelClusterPreparationMonitoringTool")))
206 
207  if not useCache:
208  acc.addEventAlgo(CompFactory.ActsTrk.PixelClusterDataPreparationAlg(name, **kwargs))
209  else:
210  acc.addEventAlgo(CompFactory.ActsTrk.PixelClusterCacheDataPreparationAlg(name, **kwargs))
211  return acc
212 
213 def ActsStripClusterPreparationAlgCfg(flags,
214  name: str = "ActsStripClusterPreparationAlg",
215  useCache: bool = False,
216  **kwargs) -> ComponentAccumulator:
217  acc = ComponentAccumulator()
218 
219  kwargs.setdefault('InputCollection', 'ITkStripClusters')
220  kwargs.setdefault('DetectorElements', 'ITkStripDetectorElementCollection')
221 
222  if 'RegSelTool' not in kwargs:
223  from RegionSelector.RegSelToolConfig import regSelTool_ITkStrip_Cfg
224  kwargs.setdefault('RegSelTool', acc.popToolsAndMerge(regSelTool_ITkStrip_Cfg(flags)))
225 
226  if flags.Acts.doMonitoring and 'MonTool' not in kwargs:
227  from ActsConfig.ActsMonitoringConfig import ActsDataPreparationMonitoringToolCfg
228  kwargs.setdefault('MonTool', acc.popToolsAndMerge(ActsDataPreparationMonitoringToolCfg(flags,
229  name = "ActsStripClusterPreparationMonitoringTool")))
230 
231  if not useCache:
232  acc.addEventAlgo(CompFactory.ActsTrk.StripClusterDataPreparationAlg(name, **kwargs))
233  else:
234  acc.addEventAlgo(CompFactory.ActsTrk.StripClusterCacheDataPreparationAlg(name, **kwargs))
235  return acc
236 
237 
238 def ActsHgtdClusterPreparationAlgCfg(flags,
239  name: str = 'ActsHgtdClusterPreparationAlg',
240  *,
241  useCache: bool = False,
242  **kwargs) -> ComponentAccumulator:
243  acc = ComponentAccumulator()
244  kwargs.setdefault('InputCollection', 'HGTDClusters')
245  kwargs.setdefault('DetectorElements', 'HGTD_DetectorElementCollection')
246 
247  # For the time being the HGTD Regional Selector tool does not exist
248 
249  if flags.Acts.doMonitoring and 'MonTool' not in kwargs:
250  from ActsConfig.ActsMonitoringConfig import ActsDataPreparationMonitoringToolCfg
251  kwargs.setdefault('MonTool', acc.popToolsAndMerge(ActsDataPreparationMonitoringToolCfg(flags,
252  name = "ActsHgtdClusterPreparationMonitoringTool")))
253 
254  acc.addEventAlgo(CompFactory.ActsTrk.HgtdClusterDataPreparationAlg(name, **kwargs))
255  return acc
256 
257 
258 def ActsMainClusterizationCfg(flags,
259  *,
260  RoIs: str = "ActsRegionOfInterest",
261  **kwargs: dict) -> ComponentAccumulator:
262  assert isinstance(RoIs, str)
263  assert isinstance(kwargs, dict)
264 
265  acc = ComponentAccumulator()
266 
267  # Clusterization is a three step process at maximum:
268  # (1) Cache Creation
269  # (2) Clusterization algorithm (reconstruction of clusters)
270  # (3) Preparation of collection for downstream algorithms
271  # What step is scheduled depends on the tracking pass and the activation
272  # or de-activation of caching mechanism
273 
274  kwargs.setdefault('processPixels', flags.Detector.EnableITkPixel)
275  kwargs.setdefault('processStrips', flags.Detector.EnableITkStrip)
276  kwargs.setdefault('processHGTD', flags.Acts.useHGTDClusterInTrackFinding and flags.Detector.EnableHGTD)
277  kwargs.setdefault('runCacheCreation', flags.Acts.useCache)
278  kwargs.setdefault('runReconstruction', True)
279  kwargs.setdefault('runPreparation', flags.Acts.useCache)
280 
281  # Step (1)
282  if kwargs['runCacheCreation']:
283  acc.merge(ActsClusterCacheCreatorAlgCfg(flags,
284  **extractChildKwargs(prefix='ClusterCacheCreatorAlg.', **kwargs)))
285 
286  # Step (2)
287  if kwargs['runReconstruction']:
288  if kwargs['processPixels']:
289  acc.merge(ActsPixelClusterizationAlgCfg(flags,
290  RoIs=RoIs,
291  **extractChildKwargs(prefix='PixelClusterizationAlg.', **kwargs)))
292  if kwargs['processStrips']:
293  acc.merge(ActsStripClusterizationAlgCfg(flags,
294  RoIs=RoIs,
295  **extractChildKwargs(prefix='StripClusterizationAlg.', **kwargs)))
296  if kwargs['processHGTD']:
297  acc.merge(ActsHgtdClusterizationAlgCfg(flags,
298  **extractChildKwargs(prefix='HgtdClusterizationAlg.', **kwargs)))
299  # Step (3)
300  if kwargs['runPreparation']:
301  if kwargs['processPixels']:
302  acc.merge(ActsPixelClusterPreparationAlgCfg(flags,
303  RoIs=RoIs,
304  **extractChildKwargs(prefix='PixelClusterPreparationAlg.', **kwargs)))
305 
306  if kwargs['processStrips']:
307  acc.merge(ActsStripClusterPreparationAlgCfg(flags,
308  RoIs=RoIs,
309  **extractChildKwargs(prefix='StripClusterPreparationAlg.', **kwargs)))
310 
311  if kwargs['processHGTD']:
312  acc.merge(ActsHgtdClusterPreparationAlgCfg(flags,
313  RoIs=RoIs,
314  **extractChildKwargs(prefix='HgtdClusterPreparationAlg.', **kwargs)))
315 
316  # Analysis extensions
317  if flags.Acts.Clusters.doAnalysis:
318  if kwargs['processPixels']:
319  from ActsConfig.ActsAnalysisConfig import ActsPixelClusterAnalysisAlgCfg
320  acc.merge(ActsPixelClusterAnalysisAlgCfg(flags, **extractChildKwargs(prefix='PixelClusterAnalysisAlg.', **kwargs)))
321 
322  if kwargs['processStrips']:
323  from ActsConfig.ActsAnalysisConfig import ActsStripClusterAnalysisAlgCfg
324  acc.merge(ActsStripClusterAnalysisAlgCfg(flags, **extractChildKwargs(prefix='StripClusterAnalysisAlg.', **kwargs)))
325 
326  return acc
327 
328 def ActsClusterizationCfg(flags,
329  *,
330  previousActsExtension: str = None) -> ComponentAccumulator:
331  assert previousActsExtension is None or isinstance(previousActsExtension, str)
332 
333  acc = ComponentAccumulator()
334 
335  processPixels = flags.Detector.EnableITkPixel
336  processStrips = flags.Detector.EnableITkStrip
337  processHGTD = flags.Acts.useHGTDClusterInTrackFinding and flags.Detector.EnableHGTD
338 
339  kwargs = dict()
340  kwargs.setdefault('processPixels', processPixels)
341  kwargs.setdefault('processStrips', processStrips)
342  kwargs.setdefault('processHGTD', processHGTD)
343 
344  # Clusterization is a three step process at maximum:
345  # (1) Cache Creation
346  # (2) Clusterization algorithm (reconstruction of clusters)
347  # (3) Preparation of collection for downstream algorithms
348  # What step is scheduled depends on the tracking pass and the activation
349  # or de-activation of caching mechanism.
350 
351  # Secondary passes do not need cache creation, that has to be performed
352  # on the primary pass, and only if the caching is enabled.
353  # Reconstruction can run on secondary passes only if the caching is enabled,
354  # this is because we may need to process detector elements not processed
355  # on the primary pass.
356  # Preparation has to be performed on secondary passes always, and on primary
357  # pass only if cache is enabled. In the latter case it is useed to collect all
358  # the clusters from all views before passing them to the downstream algorithms
359 
360  from InDetConfig.ITkActsHelpers import isPrimaryPass, isValidationPass
361  if isPrimaryPass(flags) or isValidationPass(flags):
362  # Primary pass
363  # Validation passes count as primary passes
364  kwargs.setdefault('runCacheCreation', flags.Acts.useCache)
365  kwargs.setdefault('runReconstruction', True)
366  kwargs.setdefault('runPreparation', flags.Acts.useCache)
367  else:
368  # Secondary passes
369  kwargs.setdefault('runCacheCreation', False)
370  kwargs.setdefault('runReconstruction', flags.Acts.useCache)
371  kwargs.setdefault('runPreparation', True)
372 
373  # Name of the RoI to be used
374  roisName = f'{flags.Tracking.ActiveConfig.extension}RegionOfInterest'
375  # Large Radius Tracking uses full scan RoI created in the primary pass
376  if flags.Tracking.ActiveConfig.extension == 'ActsLargeRadius':
377  from InDetConfig.ITkActsHelpers import primaryPassExtension
378  roisName = f'{primaryPassExtension(flags)}RegionOfInterest'
379 
380  # Name of the Cluster container -> ITk + extension without "Acts" + Pixel or Strip + Clusters
381  # We also define the same collection from the main ACTS pass (primary)
382  primaryPixelClustersName = 'ITkPixelClusters'
383  primaryStripClustersName = 'ITkStripClusters'
384  primaryHgtdClustersName = 'HGTD_Clusters'
385  pixelClustersName = primaryPixelClustersName
386  stripClustersName = primaryStripClustersName
387  hgtdClustersName = primaryHgtdClustersName
388 
389  # If the workflow is not a primary pass, then change the name of the cluster collections adding that information
390  if flags.Tracking.ActiveConfig.isSecondaryPass:
391  keyPrefix = flags.Tracking.ActiveConfig.extension.replace("Acts", "")
392  pixelClustersName = f'ITk{keyPrefix}PixelClusters'
393  stripClustersName = f'ITk{keyPrefix}StripClusters'
394  hgtdClustersName = f'{keyPrefix}HGTD_Clusters'
395 
396  # Configuration for (1)
397  if kwargs['runCacheCreation']:
398  kwargs.setdefault('ClusterCacheCreatorAlg.name', f'{flags.Tracking.ActiveConfig.extension}ClusterCacheCreatorAlg')
399 
400  # Configuration for (2)
401  if kwargs['runReconstruction']:
402  if kwargs['processPixels']:
403  kwargs.setdefault('PixelClusterizationAlg.name', f'{flags.Tracking.ActiveConfig.extension}PixelClusterizationAlg')
404  kwargs.setdefault('PixelClusterizationAlg.useCache', flags.Acts.useCache)
405  kwargs.setdefault('PixelClusterizationAlg.ClustersKey', pixelClustersName)
406  kwargs.setdefault('PixelClusterizationAlg.ClusterCache', f'{flags.Tracking.ActiveConfig.extension}PixelClustersCache')
407  if flags.Tracking.ActiveConfig.isSecondaryPass and previousActsExtension is not None:
408  kwargs.setdefault('PixelClusterizationAlg.ExtraInputs', {('ActsTrk::PrepRawDataAssociation', f'StoreGateSvc+{previousActsExtension}PrdMap')})
409 
410  if kwargs['processStrips']:
411  kwargs.setdefault('StripClusterizationAlg.name', f'{flags.Tracking.ActiveConfig.extension}StripClusterizationAlg')
412  kwargs.setdefault('StripClusterizationAlg.useCache', flags.Acts.useCache)
413  kwargs.setdefault('StripClusterizationAlg.ClustersKey', stripClustersName)
414  kwargs.setdefault('StripClusterizationAlg.ClusterCache', f'{flags.Tracking.ActiveConfig.extension}StripClustersCache')
415  if flags.Tracking.ActiveConfig.isSecondaryPass and previousActsExtension is not None:
416  kwargs.setdefault('StripClusterizationAlg.ExtraInputs', {('ActsTrk::PrepRawDataAssociation', f'StoreGateSvc+{previousActsExtension}PrdMap')})
417 
418  if kwargs['processHGTD']:
419  kwargs.setdefault('HgtdClusterizationAlg.name', f'{flags.Tracking.ActiveConfig.extension}HgtdClusterizationAlg')
420  # HGTD algo has different convention ... need to use the same
421  kwargs.setdefault('HgtdClusterizationAlg.ClusterContainerName', hgtdClustersName)
422  if flags.Tracking.ActiveConfig.isSecondaryPass and previousActsExtension is not None:
423  kwargs.setdefault('HgtdClusterizationAlg.ExtraInputs', {('ActsTrk::PrepRawDataAssociation', f'StoreGateSvc+{previousActsExtension}PrdMap')})
424 
425 
426  # Configuration for (3)
427  if kwargs['runPreparation']:
428  if kwargs['processPixels']:
429  kwargs.setdefault('PixelClusterPreparationAlg.name', f'{flags.Tracking.ActiveConfig.extension}PixelClusterPreparationAlg')
430  kwargs.setdefault('PixelClusterPreparationAlg.useCache', flags.Acts.useCache)
431  kwargs.setdefault('PixelClusterPreparationAlg.OutputCollection', f'{pixelClustersName}_Cached' if kwargs['runReconstruction'] else pixelClustersName)
432  # The input is one between the collection (w/o cache) and the IDC (w/ cache)
433  if not flags.Acts.useCache:
434  # Take the collection from the reconstruction step. If not available take the collection from the primary pass
435  kwargs.setdefault('PixelClusterPreparationAlg.InputCollection', pixelClustersName if kwargs['runReconstruction'] else primaryPixelClustersName)
436  kwargs.setdefault('PixelClusterPreparationAlg.InputIDC', '')
437  else:
438  kwargs.setdefault('PixelClusterPreparationAlg.InputCollection', '')
439  kwargs.setdefault('PixelClusterPreparationAlg.InputIDC', f'{flags.Tracking.ActiveConfig.extension}PixelClustersCache')
440  # Prd Map for removing previously used measurements
441  if flags.Tracking.ActiveConfig.isSecondaryPass and previousActsExtension is not None:
442  kwargs.setdefault('PixelClusterPreparationAlg.InputPrdMap', f'{previousActsExtension}PrdMap')
443 
444  if kwargs['processStrips']:
445  kwargs.setdefault('StripClusterPreparationAlg.name', f'{flags.Tracking.ActiveConfig.extension}StripClusterPreparationAlg')
446  kwargs.setdefault('StripClusterPreparationAlg.useCache', flags.Acts.useCache)
447  kwargs.setdefault('StripClusterPreparationAlg.OutputCollection', f'{stripClustersName}_Cached' if kwargs['runReconstruction'] else stripClustersName)
448  if not flags.Acts.useCache:
449  kwargs.setdefault('StripClusterPreparationAlg.InputCollection', stripClustersName if kwargs['runReconstruction'] else primaryStripClustersName)
450  kwargs.setdefault('StripClusterPreparationAlg.InputIDC', '')
451  else:
452  kwargs.setdefault('StripClusterPreparationAlg.InputCollection', '')
453  kwargs.setdefault('StripClusterPreparationAlg.InputIDC', f'{flags.Tracking.ActiveConfig.extension}StripClustersCache')
454  # Prd Map for removing previously used measurements
455  if flags.Tracking.ActiveConfig.isSecondaryPass and previousActsExtension is not None:
456  kwargs.setdefault('StripClusterPreparationAlg.InputPrdMap', f'{previousActsExtension}PrdMap')
457 
458  if kwargs['processHGTD']:
459  kwargs.setdefault('HgtdClusterPreparationAlg.name', f'{flags.Tracking.ActiveConfig.extension}HgtdClusterPreparationAlg')
460  kwargs.setdefault('HgtdClusterPreparationAlg.OutputCollection', f'{hgtdClustersName}_Cached' if kwargs['runReconstruction'] else hgtdClustersName)
461  kwargs.setdefault('HgtdClusterPreparationAlg.InputCollection', hgtdClustersName if kwargs['runReconstruction'] else primaryHgtdClustersName)
462  kwargs.setdefault('HgtdClusterPreparationAlg.InputIDC', '')
463  if flags.Tracking.ActiveConfig.isSecondaryPass and previousActsExtension is not None:
464  kwargs.setdefault('HgtdClusterPreparationAlg.InputPrdMap', f'{previousActsExtension}PrdMap')
465 
466  # Analysis algo(s)
467  if flags.Acts.Clusters.doAnalysis:
468  # Run analysis code on the resulting cluster collection produced by this tracking pass
469  # This collection is the result of (3) if it ran, else the result of (2). We are sure at least one of them run
470  if kwargs['processPixels']:
471  kwargs.setdefault('PixelClusterAnalysisAlg.name', f'{flags.Tracking.ActiveConfig.extension}PixelClusterAnalysisAlg')
472  kwargs.setdefault('PixelClusterAnalysisAlg.extension', flags.Tracking.ActiveConfig.extension)
473  kwargs.setdefault('PixelClusterAnalysisAlg.MonGroupName', f'{flags.Tracking.ActiveConfig.extension}ClusterAnalysisAlg')
474  kwargs.setdefault('PixelClusterAnalysisAlg.ClusterContainerKey', kwargs['PixelClusterPreparationAlg.OutputCollection'] if kwargs['runPreparation'] else kwargs['PixelClusterizationAlg.ClustersKey'])
475 
476  if kwargs['processStrips']:
477  kwargs.setdefault('StripClusterAnalysisAlg.name', f'{flags.Tracking.ActiveConfig.extension}StripClusterAnalysisAlg')
478  kwargs.setdefault('StripClusterAnalysisAlg.extension', flags.Tracking.ActiveConfig.extension)
479  kwargs.setdefault('StripClusterAnalysisAlg.MonGroupName', f'{flags.Tracking.ActiveConfig.extension}ClusterAnalysisAlg')
480  kwargs.setdefault('StripClusterAnalysisAlg.ClusterContainerKey', kwargs['StripClusterPreparationAlg.OutputCollection'] if kwargs['runPreparation'] else kwargs['StripClusterizationAlg.ClustersKey'])
481 
482  if kwargs['processHGTD']:
483  kwargs.setdefault('HgtdClusterAnalysisAlg.name', f'{flags.Tracking.ActiveConfig.extension}HgtdClusterAnalysisAlg')
484  kwargs.setdefault('HgtdClusterAnalysisAlg.extension', flags.Tracking.ActiveConfig.extension)
485  kwargs.setdefault('HgtdClusterAnalysisAlg.MonGroupName', f'{flags.Tracking.ActiveConfig.extension}ClusterAnalysisAlg')
486 
487  acc.merge(ActsMainClusterizationCfg(flags, RoIs=roisName, **kwargs))
488 
489  # Persistification
490  if flags.Acts.EDM.PersistifyClusters and kwargs['runReconstruction']:
491  toAOD = []
492  if kwargs['processPixels']:
493  pixel_cluster_shortlist = ['-validationMeasurementLink']
494  pixel_cluster_variables = '.'.join(pixel_cluster_shortlist)
495 
496  pixelClusterCollection = kwargs['PixelClusterizationAlg.ClustersKey']
497  toAOD += [f'xAOD::PixelClusterContainer#{pixelClusterCollection}',
498  f'xAOD::PixelClusterAuxContainer#{pixelClusterCollection}Aux.{pixel_cluster_variables}']
499 
500  if kwargs['processStrips']:
501  strip_cluster_shortlist = ['-validationMeasurementLink']
502  strip_cluster_variables = '.'.join(strip_cluster_shortlist)
503 
504  stripClusterCollection = kwargs['StripClusterizationAlg.ClustersKey']
505  toAOD += [f"xAOD::StripClusterContainer#{stripClusterCollection}",
506  f"xAOD::StripClusterAuxContainer#{stripClusterCollection}Aux.{strip_cluster_variables}"]
507 
508  if kwargs['processHGTD']:
509  hgtdClusterCollection = kwargs['HgtdClusterizationAlg.ClusterContainerName']
510  toAOD += [f"xAOD::HGTDClusterContainer#{hgtdClusterCollection}",
511  f"xAOD::HGTDClusterAuxContainer#{hgtdClusterCollection}Aux."]
512 
513  from OutputStreamAthenaPool.OutputStreamConfig import addToAOD
514  acc.merge(addToAOD(flags, toAOD))
515 
516  return acc
517 
ActsClusterizationConfig.ActsPixelClusterizationAlgCfg
ComponentAccumulator ActsPixelClusterizationAlgCfg(flags, str name='ActsPixelClusterizationAlg', *bool useCache=False, **kwargs)
Definition: ActsClusterizationConfig.py:103
ActsClusterizationConfig.ActsHgtdClusteringToolCfg
ComponentAccumulator ActsHgtdClusteringToolCfg(flags, str name="ActsHgtdClusteringTool", **kwargs)
Definition: ActsClusterizationConfig.py:8
python.ITkStripLorentzAngleConfig.ITkStripLorentzAngleToolCfg
def ITkStripLorentzAngleToolCfg(flags, name="ITkStripLorentzAngleTool", **kwargs)
Definition: ITkStripLorentzAngleConfig.py:10
python.ITkActsHelpers.isPrimaryPass
bool isPrimaryPass(flags)
Definition: ITkActsHelpers.py:3
python.JetAnalysisCommon.ComponentAccumulator
ComponentAccumulator
Definition: JetAnalysisCommon.py:302
ActsUtilities.extractChildKwargs
dict extractChildKwargs(*str prefix, **dict kwargs)
Definition: ActsUtilities.py:3
ITkPixelConditionsConfig.ITkPixelDetectorElementStatusAlgCfg
def ITkPixelDetectorElementStatusAlgCfg(flags, name="ITkPixelDetectorElementStatusAlg", **kwargs)
Definition: ITkPixelConditionsConfig.py:178
ActsMonitoringConfig.ActsHgtdClusterizationMonitoringToolCfg
ComponentAccumulator ActsHgtdClusterizationMonitoringToolCfg(flags, str name="ActsHgtdClusterizationMonitoringTool", **kwargs)
Definition: ActsMonitoringConfig.py:64
ActsMonitoringConfig.ActsITkStripClusterizationMonitoringToolCfg
ComponentAccumulator ActsITkStripClusterizationMonitoringToolCfg(flags, str name="ActsITkStripClusterizationMonitoringTool", **kwargs)
Definition: ActsMonitoringConfig.py:31
ActsClusterizationConfig.ActsHgtdClusterizationAlgCfg
ComponentAccumulator ActsHgtdClusterizationAlgCfg(flags, str name="ActsHgtdClusterizationAlg", **kwargs)
Definition: ActsClusterizationConfig.py:22
ActsAnalysisConfig.ActsStripClusterAnalysisAlgCfg
ComponentAccumulator ActsStripClusterAnalysisAlgCfg(flags, str name="ActsStripClusterAnalysisAlg", str extension="Acts", **kwargs)
Definition: ActsAnalysisConfig.py:385
RegSelToolConfig.regSelTool_ITkPixel_Cfg
def regSelTool_ITkPixel_Cfg(flags)
Definition: RegSelToolConfig.py:120
python.ITkStripConditionsAlgorithmsConfig.ITkStripDetectorElementStatusAlgCfg
def ITkStripDetectorElementStatusAlgCfg(flags, name="ITkStripDetectorElementStatusAlg", **kwargs)
Definition: ITkStripConditionsAlgorithmsConfig.py:91
ActsClusterizationConfig.ActsStripClusterizationAlgCfg
ComponentAccumulator ActsStripClusterizationAlgCfg(flags, str name='ActsStripClusterizationAlg', bool useCache=False, **kwargs)
Definition: ActsClusterizationConfig.py:142
ActsClusterizationConfig.ActsHgtdClusterPreparationAlgCfg
ComponentAccumulator ActsHgtdClusterPreparationAlgCfg(flags, str name='ActsHgtdClusterPreparationAlg', *bool useCache=False, **kwargs)
Definition: ActsClusterizationConfig.py:238
python.ITkPixelLorentzAngleConfig.ITkPixelLorentzAngleToolCfg
def ITkPixelLorentzAngleToolCfg(flags, name="ITkPixelLorentzAngleTool", **kwargs)
Definition: ITkPixelLorentzAngleConfig.py:14
RegSelToolConfig.regSelTool_ITkStrip_Cfg
def regSelTool_ITkStrip_Cfg(flags)
Definition: RegSelToolConfig.py:126
ActsMonitoringConfig.ActsDataPreparationMonitoringToolCfg
ComponentAccumulator ActsDataPreparationMonitoringToolCfg(flags, str name="ActsDataPreparationMonitoringTool", **kwargs)
Definition: ActsMonitoringConfig.py:49
python.PixelReadoutGeometryConfig.ITkPixelReadoutManagerCfg
def ITkPixelReadoutManagerCfg(flags, name="ITkPixelReadoutManager", **kwargs)
Definition: PixelReadoutGeometryConfig.py:13
ActsClusterizationConfig.ActsHgtdTimedClusteringToolCfg
ComponentAccumulator ActsHgtdTimedClusteringToolCfg(flags, str name="ActsHgtdTimedClusteringTool", **kwargs)
Definition: ActsClusterizationConfig.py:15
HGTD_GeoModelConfig.HGTD_ReadoutGeometryCfg
def HGTD_ReadoutGeometryCfg(flags)
Definition: python/HGTD_GeoModelConfig.py:23
ActsClusterizationConfig.ActsClusterizationCfg
ComponentAccumulator ActsClusterizationCfg(flags, *str previousActsExtension=None)
Definition: ActsClusterizationConfig.py:328
TCS::join
std::string join(const std::vector< std::string > &v, const char c=',')
Definition: Trigger/TrigT1/L1Topo/L1TopoCommon/Root/StringUtils.cxx:10
ActsClusterizationConfig.ActsStripClusterPreparationAlgCfg
ComponentAccumulator ActsStripClusterPreparationAlgCfg(flags, str name="ActsStripClusterPreparationAlg", bool useCache=False, **kwargs)
Definition: ActsClusterizationConfig.py:213
python.ITkActsHelpers.isValidationPass
bool isValidationPass(flags)
Definition: ITkActsHelpers.py:11
ActsMonitoringConfig.ActsITkPixelClusterizationMonitoringToolCfg
ComponentAccumulator ActsITkPixelClusterizationMonitoringToolCfg(flags, str name="ActsITkPixelClusterizationMonitoringTool", **kwargs)
Definition: ActsMonitoringConfig.py:13
python.OutputStreamConfig.addToAOD
def addToAOD(flags, itemOrList, **kwargs)
Definition: OutputStreamConfig.py:188
ITkPixelConditionsConfig.ITkPixelChargeCalibCondAlgCfg
def ITkPixelChargeCalibCondAlgCfg(flags, name="ITkPixelChargeCalibCondAlg", **kwargs)
Definition: ITkPixelConditionsConfig.py:63
ActsClusterizationConfig.ActsPixelClusterPreparationAlgCfg
ComponentAccumulator ActsPixelClusterPreparationAlgCfg(flags, str name="ActsPixelClusterPreparationAlg", bool useCache=False, **kwargs)
Definition: ActsClusterizationConfig.py:189
ActsClusterizationConfig.ActsMainClusterizationCfg
ComponentAccumulator ActsMainClusterizationCfg(flags, *str RoIs="ActsRegionOfInterest", **dict kwargs)
Definition: ActsClusterizationConfig.py:258
ITkPixelConditionsConfig.ITkPixelOfflineCalibCondAlgCfg
def ITkPixelOfflineCalibCondAlgCfg(flags, name="ITkPixelOfflineCalibCondAlg", **kwargs)
Definition: ITkPixelConditionsConfig.py:141
ActsClusterizationConfig.ActsClusterCacheCreatorAlgCfg
ComponentAccumulator ActsClusterCacheCreatorAlgCfg(flags, str name="ActsClusterCacheCreatorAlg", **kwargs)
Definition: ActsClusterizationConfig.py:180
ActsClusterizationConfig.ActsStripClusteringToolCfg
ComponentAccumulator ActsStripClusteringToolCfg(flags, str name="ActsStripClusteringTool", **kwargs)
Definition: ActsClusterizationConfig.py:77
ActsClusterizationConfig.ActsPixelClusteringToolCfg
ComponentAccumulator ActsPixelClusteringToolCfg(flags, str name="ActsPixelClusteringTool", **kwargs)
Definition: ActsClusterizationConfig.py:52
ActsAnalysisConfig.ActsPixelClusterAnalysisAlgCfg
ComponentAccumulator ActsPixelClusterAnalysisAlgCfg(flags, str name="ActsPixelClusterAnalysisAlg", str extension="Acts", **kwargs)
Definition: ActsAnalysisConfig.py:350
Generated on Mon Sep 29 2025 21:06:06 for ATLAS Offline Software by   doxygen 1.8.18