ATLAS Offline Software
ActsSeedingConfig.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 ActsConfig.ActsConfigFlags import SeedingStrategy
6 from ActsConfig.ActsUtilities import extractChildKwargs
7 from ActsInterop import UnitConstants
8 from AthenaCommon.Utils.unixtools import find_datafile
9 
10 # ACTS tools
11 def ActsPixelSeedingToolCfg(flags,
12  name: str = "ActsPixelSeedingTool",
13  **kwargs) -> ComponentAccumulator:
14  acc = ComponentAccumulator()
15 
16 
17  kwargs.setdefault("numSeedIncrement" , float("inf"))
18  kwargs.setdefault("deltaZMax" , float("inf"))
19  kwargs.setdefault("maxPtScattering", float("inf"))
20  kwargs.setdefault("useVariableMiddleSPRange", False)
21  kwargs.setdefault("rMax", 320. * UnitConstants.mm)
22  kwargs.setdefault("rBinEdges", [0, kwargs['rMax']])
23  kwargs.setdefault("rRangeMiddleSP", [
24  [0,0],
25  [140, 260],
26  [40, 260],
27  [40, 260],
28  [40, 260],
29  [40, 260],
30  [70, 260],
31  [40, 260],
32  [40, 260],
33  [40, 260],
34  [40, 260],
35  [140, 260],
36  [0, 0]])
37  acc.setPrivateTools(CompFactory.ActsTrk.SeedingTool(name, **kwargs))
38  return acc
39 
40 def ActsFastPixelSeedingToolCfg(flags,
41  name: str = "ActsFastPixelSeedingTool",
42  **kwargs) -> ComponentAccumulator:
43 
44  kwargs.setdefault("minPt", 1000 * UnitConstants.MeV)
45  kwargs.setdefault("collisionRegionMin", -150 * UnitConstants.mm)
46  kwargs.setdefault("collisionRegionMax", 150 * UnitConstants.mm)
47  kwargs.setdefault("maxPhiBins", 200)
48  kwargs.setdefault("gridRMax", 250 * UnitConstants.mm)
49  kwargs.setdefault("deltaRMax", 200 * UnitConstants.mm)
50  kwargs.setdefault("zBinsCustomLooping" , [3, 11, 4, 10, 7, 5, 9, 6, 8])
51  kwargs.setdefault("rRangeMiddleSP", [
52  [40.0, 80.0],
53  [40.0, 80.0],
54  [40.0, 200.0],
55  [70.0, 200.0],
56  [70.0, 200.0],
57  [70.0, 250.0],
58  [70.0, 250.0],
59  [70.0, 250.0],
60  [70.0, 200.0],
61  [70.0, 200.0],
62  [40.0, 200.0],
63  [40.0, 80.0],
64  [40.0, 80.0]])
65  kwargs.setdefault("useVariableMiddleSPRange", False)
66  kwargs.setdefault("useExperimentCuts", True)
67  kwargs.setdefault("rMax", 320 * UnitConstants.mm)
68  kwargs.setdefault("rBinEdges", [0, kwargs['rMax']])
69 
70  return ActsPixelSeedingToolCfg(flags, name, **kwargs)
71 
72 def ActsStripSeedingToolCfg(flags,
73  name: str = "ActsStripSeedingTool",
74  **kwargs) -> ComponentAccumulator:
75  acc = ComponentAccumulator()
76 
77 
78  kwargs.setdefault("doSeedQualitySelection", False)
79  # For SpacePointGridConfig
80  kwargs.setdefault("gridRMax" , 1000. * UnitConstants.mm)
81  kwargs.setdefault("deltaRMax" , 600. * UnitConstants.mm)
82  kwargs.setdefault("impactMax" , 20. * UnitConstants.mm)
83  # For SeedfinderConfig
84  kwargs.setdefault("rMax" , 1200. * UnitConstants.mm)
85  kwargs.setdefault("deltaRMinTopSP" , 20. * UnitConstants.mm)
86  kwargs.setdefault("deltaRMaxTopSP" , 300. * UnitConstants.mm)
87  kwargs.setdefault("deltaRMinBottomSP" , 20. * UnitConstants.mm)
88  kwargs.setdefault("deltaRMaxBottomSP" , 300. * UnitConstants.mm)
89  kwargs.setdefault("deltaZMax" , 900. * UnitConstants.mm)
90  kwargs.setdefault("interactionPointCut" , False)
91  kwargs.setdefault("zBinsCustomLooping" , [7, 8, 6, 9, 5, 10, 4, 11, 3, 12, 2])
92  kwargs.setdefault("deltaRMiddleMinSPRange" , 30 * UnitConstants.mm)
93  kwargs.setdefault("deltaRMiddleMaxSPRange" , 150 * UnitConstants.mm)
94  kwargs.setdefault("useDetailedDoubleMeasurementInfo" , True)
95  kwargs.setdefault("maxPtScattering", float("inf"))
96  # For SeedFilterConfig
97  kwargs.setdefault("useDeltaRorTopRadius" , False)
98  kwargs.setdefault("seedConfirmationInFilter" , False)
99  kwargs.setdefault("impactWeightFactor" , 1.)
100  kwargs.setdefault("compatSeedLimit" , 4)
101  kwargs.setdefault("numSeedIncrement" , 1.)
102  kwargs.setdefault("seedWeightIncrement" , 10100.)
103  kwargs.setdefault("maxSeedsPerSpMConf" , 100)
104  kwargs.setdefault("maxQualitySeedsPerSpMConf" , 100)
105  # For seeding algorithm
106  kwargs.setdefault("zBinNeighborsBottom" , [(0,0),(0,1),(0,1),(0,1),(0,2),(0,1),(0,0),(-1,0),(-2,0),(-1,0),(-1,0),(-1,0),(0,0)])
107  # Any other
108  kwargs.setdefault("rBinEdges", [0, kwargs['rMax']])
109 
110  acc.setPrivateTools(CompFactory.ActsTrk.SeedingTool(name, **kwargs))
111  return acc
112 
113 def ActsPixelOrthogonalSeedingToolCfg(flags,
114  name: str = "ActsPixelOrthogonalSeedingTool",
115  **kwargs) -> ComponentAccumulator:
116  acc = ComponentAccumulator()
117 
118  acc.setPrivateTools(CompFactory.ActsTrk.OrthogonalSeedingTool(name, **kwargs))
119  return acc
120 
121 def ActsFastPixelOrthogonalSeedingToolCfg(flags,
122  name: str = "ActsFastPixelOrthogonalSeedingTool",
123  **kwargs) -> ComponentAccumulator:
124  acc = ComponentAccumulator()
125 
126 
127 
128  kwargs.setdefault("minPt", 1000 * UnitConstants.MeV)
129  kwargs.setdefault("collisionRegionMin", -150 * UnitConstants.mm)
130  kwargs.setdefault("collisionRegionMax", 150 * UnitConstants.mm)
131  kwargs.setdefault("useExperimentCuts", True)
132 
133  acc.setPrivateTools(CompFactory.ActsTrk.OrthogonalSeedingTool(name, **kwargs))
134  return acc
135 
136 def ActsStripOrthogonalSeedingToolCfg(flags,
137  name: str = "ActsStripOrthogonalSeedingTool",
138  **kwargs) -> ComponentAccumulator:
139  acc = ComponentAccumulator()
140 
141 
142  kwargs.setdefault("impactMax" , 20. * UnitConstants.mm)
143  kwargs.setdefault('rMax', 1200. * UnitConstants.mm)
144  kwargs.setdefault("deltaRMinTopSP" , 20. * UnitConstants.mm)
145  kwargs.setdefault("deltaRMaxTopSP" , 300. * UnitConstants.mm)
146  kwargs.setdefault("deltaRMinBottomSP" , 20. * UnitConstants.mm)
147  kwargs.setdefault("deltaRMaxBottomSP" , 300. * UnitConstants.mm)
148  kwargs.setdefault("deltaZMax" , 900. * UnitConstants.mm)
149  kwargs.setdefault("interactionPointCut" , False)
150  kwargs.setdefault("impactWeightFactor" , 1.)
151  kwargs.setdefault("compatSeedLimit" , 4)
152  kwargs.setdefault("seedWeightIncrement" , 10100.)
153  kwargs.setdefault("numSeedIncrement" , 1.)
154  kwargs.setdefault("seedConfirmationInFilter" , False)
155  kwargs.setdefault("maxSeedsPerSpMConf" , 100)
156  kwargs.setdefault("maxQualitySeedsPerSpMConf" , 100)
157  kwargs.setdefault("useDeltaRorTopRadius" , False)
158  kwargs.setdefault("rMinMiddle", 33. * UnitConstants.mm)
159  kwargs.setdefault("rMaxMiddle", 1200. * UnitConstants.mm)
160 
161  acc.setPrivateTools(CompFactory.ActsTrk.OrthogonalSeedingTool(name, **kwargs))
162  return acc
163 
164 def ActsPixelGbtsSeedingToolCfg(flags,
165  name: str = "ActsPixelGbtsSeedingTool",
166  **kwargs) -> ComponentAccumulator:
167  acc = ComponentAccumulator()
168 
170  kwargs.setdefault("connector_input_file" , find_datafile("ActsPatternRecognition/GBTS_EdgeProbabilites_ITKPixels.txt"))
171  acc.setPrivateTools(CompFactory.ActsTrk.GbtsSeedingTool(name = name, **kwargs))
172  return acc
173 
174 def ActsSiSpacePointsSeedMakerToolCfg(flags,
175  name: str = 'ActsSiSpacePointsSeedMakerTool',
176  **kwargs) -> ComponentAccumulator:
177  assert isinstance(name, str)
178 
179  acc = ComponentAccumulator()
180 
181  if flags.Tracking.ActiveConfig.extension == "ActsValidateConversionSeeds":
182  kwargs.setdefault('useOverlapSpCollection', False)
183 
184  # Main properties
185  kwargs.setdefault('usePixel',
186  flags.Tracking.ActiveConfig.useITkPixel and
187  flags.Tracking.ActiveConfig.useITkPixelSeeding)
188  kwargs.setdefault('useStrip',
189  flags.Tracking.ActiveConfig.useITkStrip and
190  flags.Tracking.ActiveConfig.useITkStripSeeding)
191  kwargs.setdefault('useOverlapSpCollection',
192  flags.Tracking.ActiveConfig.useITkStrip and
193  flags.Tracking.ActiveConfig.useITkStripSeeding)
194  kwargs.setdefault('ActsSpacePointsPixelName' , "ITkPixelSpacePoints")
195  kwargs.setdefault('ActsSpacePointsStripName' , "ITkStripSpacePoints")
196  kwargs.setdefault('ActsSpacePointsOverlapName' , "ITkStripOverlapSpacePoints")
197 
198 
199  # The code will need to use Trk::SpacePoint object for downstream Athena tracking
200  # If we run this tool we have two options to retrieve this:
201  # (1) Have the Athena->Acts Space Point Converter scheduled beforehand
202  # (2) Have the Athena->Acts Cluster Converter scheduled beforehand
203  # In case (1) the link xAOD -> Trk Space Point will be used to retrieve the Trk::SpacePoints
204  # In case (2) the link xAOD -> InDet Cluster will be used to create the Trk::SpacePoints
205  # If none of the above conditions are met, it means there is a misconfiguration of the algorithms
206  useClusters = flags.Tracking.ActiveConfig.doAthenaToActsCluster and not flags.Tracking.ActiveConfig.doAthenaToActsSpacePoint
207  kwargs.setdefault('useClustersForSeedConversion', useClusters)
208 
209  if flags.Tracking.ActiveConfig.usePrdAssociationTool:
210  # not all classes have that property !!!
211  kwargs.setdefault('PRDtoTrackMap', (
212  'ITkPRDtoTrackMap' + flags.Tracking.ActiveConfig.extension))
213 
214  # Acts Seed Tools
215  # Do not overwrite if already present in `kwargs`
216  seedTool_pixel = None
217  if 'SeedToolPixel' not in kwargs:
218  if flags.Acts.SeedingStrategy is SeedingStrategy.Orthogonal:
219  if flags.Tracking.doITkFastTracking:
220  seedTool_pixel = acc.popToolsAndMerge(ActsFastPixelOrthogonalSeedingToolCfg(flags))
221  else:
222  seedTool_pixel = acc.popToolsAndMerge(ActsPixelOrthogonalSeedingToolCfg(flags))
223  elif flags.Acts.SeedingStrategy is SeedingStrategy.Gbts:
224  seedTool_pixel = acc.popToolsAndMerge(ActsPixelGbtsSeedingToolCfg(flags))
225  else:
226  if flags.Tracking.doITkFastTracking:
227  kwargs.setdefault("useFastTracking", True)
228  seedTool_pixel = acc.popToolsAndMerge(ActsFastPixelSeedingToolCfg(flags))
229  else:
230  seedTool_pixel = acc.popToolsAndMerge(ActsPixelSeedingToolCfg(flags))
231 
232  seedTool_strip = None
233  if 'SeedToolStrip' not in kwargs:
234  if flags.Acts.SeedingStrategy is SeedingStrategy.Orthogonal:
235  seedTool_strip = acc.popToolsAndMerge(ActsStripOrthogonalSeedingToolCfg(flags))
236  else:
237  seedTool_strip = acc.popToolsAndMerge(ActsStripSeedingToolCfg(flags,
238  rMax=flags.Tracking.ActiveConfig.radMax))
239 
240  kwargs.setdefault('SeedToolPixel', seedTool_pixel)
241  kwargs.setdefault('SeedToolStrip', seedTool_strip)
242 
243  # Validation
244  if flags.Tracking.writeSeedValNtuple:
245  kwargs.setdefault('WriteNtuple', True)
246  HistService = CompFactory.THistSvc(Output = ["valNtuples DATAFILE='SeedMakerValidation.root' OPT='RECREATE'"])
247  acc.addService(HistService)
248 
249  acc.setPrivateTools(CompFactory.ActsTrk.SiSpacePointsSeedMaker(name, **kwargs))
250  return acc
251 
252 
253 # ACTS algorithm using Athena objects upstream
254 def ActsPixelSeedingAlgCfg(flags,
255  name: str = 'ActsPixelSeedingAlg',
256  **kwargs) -> ComponentAccumulator:
257  acc = ComponentAccumulator()
258 
259  # Beam Spot Cond is a requirement
260  from BeamSpotConditions.BeamSpotConditionsConfig import BeamSpotCondAlgCfg
261  acc.merge(BeamSpotCondAlgCfg(flags))
262 
263  from MagFieldServices.MagFieldServicesConfig import AtlasFieldCacheCondAlgCfg
264  acc.merge(AtlasFieldCacheCondAlgCfg(flags))
265 
266  from PixelGeoModelXml.ITkPixelGeoModelConfig import ITkPixelReadoutGeometryCfg
267  acc.merge(ITkPixelReadoutGeometryCfg(flags))
268 
269  # Need To add additional tool(s)
270  # Tracking Geometry Tool
271  if 'TrackingGeometryTool' not in kwargs:
272  from ActsConfig.ActsGeometryConfig import ActsTrackingGeometryToolCfg
273  geoTool = acc.popToolsAndMerge(ActsTrackingGeometryToolCfg(flags))
274  acc.addPublicTool(geoTool)
275  kwargs.setdefault('TrackingGeometryTool', acc.getPublicTool(geoTool.name))
276 
277  # ATLAS Converter Tool
278  if 'ATLASConverterTool' not in kwargs:
279  from ActsConfig.ActsEventCnvConfig import ActsToTrkConverterToolCfg
280  kwargs.setdefault('ATLASConverterTool', acc.popToolsAndMerge(ActsToTrkConverterToolCfg(flags)))
281 
282  # Track Param Estimation Tool
283  if 'TrackParamsEstimationTool' not in kwargs:
284  from ActsConfig.ActsTrackParamsEstimationConfig import ActsTrackParamsEstimationToolCfg
285  kwargs.setdefault('TrackParamsEstimationTool', acc.popToolsAndMerge(ActsTrackParamsEstimationToolCfg(flags)))
286 
287  useFastTracking = kwargs.get("useFastTracking", flags.Tracking.doITkFastTracking)
288 
289  if "SeedTool" not in kwargs:
290  if flags.Acts.SeedingStrategy is SeedingStrategy.Orthogonal:
291  if useFastTracking:
292  kwargs.setdefault('SeedTool', acc.popToolsAndMerge(ActsFastPixelOrthogonalSeedingToolCfg(flags)))
293  else:
294  kwargs.setdefault('SeedTool', acc.popToolsAndMerge(ActsPixelOrthogonalSeedingToolCfg(flags)))
295  elif flags.Acts.SeedingStrategy is SeedingStrategy.Gbts:
296  kwargs.setdefault('SeedTool', acc.popToolsAndMerge(ActsPixelGbtsSeedingToolCfg(flags)))
297  else:
298  if useFastTracking:
299  kwargs.setdefault('SeedTool', acc.popToolsAndMerge(ActsFastPixelSeedingToolCfg(flags)))
300  else:
301  kwargs.setdefault('SeedTool', acc.popToolsAndMerge(ActsPixelSeedingToolCfg(flags)))
302 
303  kwargs.setdefault("useFastTracking", useFastTracking)
304  kwargs.setdefault('InputSpacePoints', ['ITkPixelSpacePoints_Cached'] if flags.Acts.useCache else ['ITkPixelSpacePoints'])
305  kwargs.setdefault('OutputSeeds', 'ActsPixelSeeds')
306  kwargs.setdefault('OutputEstimatedTrackParameters', 'ActsPixelEstimatedTrackParams')
307  kwargs.setdefault('DetectorElements', 'ITkPixelDetectorElementCollection')
308  kwargs.setdefault('UsePixel', True)
309 
310  if flags.Acts.doMonitoring and 'MonTool' not in kwargs:
311  from ActsConfig.ActsMonitoringConfig import ActsITkPixelSeedingMonitoringToolCfg
312  kwargs.setdefault('MonTool', acc.popToolsAndMerge(ActsITkPixelSeedingMonitoringToolCfg(flags)))
313 
314  acc.addEventAlgo(CompFactory.ActsTrk.SeedingAlg(name, **kwargs))
315  return acc
316 
317 
318 def ActsStripSeedingAlgCfg(flags,
319  name: str = 'ActsStripSeedingAlg',
320  **kwargs) -> ComponentAccumulator:
321  acc = ComponentAccumulator()
322 
323  # Beam Spot Cond is a requirement
324  from BeamSpotConditions.BeamSpotConditionsConfig import BeamSpotCondAlgCfg
325  acc.merge(BeamSpotCondAlgCfg(flags))
326 
327  from MagFieldServices.MagFieldServicesConfig import AtlasFieldCacheCondAlgCfg
328  acc.merge(AtlasFieldCacheCondAlgCfg(flags))
329 
330  from StripGeoModelXml.ITkStripGeoModelConfig import ITkStripReadoutGeometryCfg
331  acc.merge(ITkStripReadoutGeometryCfg(flags))
332 
333  # Need To add additional tool(s)
334  # Tracking Geometry Tool
335  if 'TrackingGeometryTool' not in kwargs:
336  from ActsConfig.ActsGeometryConfig import ActsTrackingGeometryToolCfg
337  geoTool = acc.popToolsAndMerge(ActsTrackingGeometryToolCfg(flags))
338  acc.addPublicTool(geoTool)
339  kwargs.setdefault('TrackingGeometryTool', acc.getPublicTool(geoTool.name))
340 
341  # ATLAS Converter Tool
342  if 'ATLASConverterTool' not in kwargs:
343  from ActsConfig.ActsEventCnvConfig import ActsToTrkConverterToolCfg
344  kwargs.setdefault('ATLASConverterTool', acc.popToolsAndMerge(ActsToTrkConverterToolCfg(flags)))
345 
346  # Track Param Estimation Tool
347  if 'TrackParamsEstimationTool' not in kwargs:
348  from ActsConfig.ActsTrackParamsEstimationConfig import ActsTrackParamsEstimationToolCfg
349  kwargs.setdefault('TrackParamsEstimationTool', acc.popToolsAndMerge(ActsTrackParamsEstimationToolCfg(flags, useTopSp=flags.Acts.reverseTrackFindingForStrips)))
350 
351  if "SeedTool" not in kwargs:
352  if flags.Acts.SeedingStrategy is SeedingStrategy.Orthogonal:
353  kwargs.setdefault('SeedTool', acc.popToolsAndMerge(ActsStripOrthogonalSeedingToolCfg(flags)))
354  else:
355  kwargs.setdefault('SeedTool', acc.popToolsAndMerge(ActsStripSeedingToolCfg(flags)))
356 
357  kwargs.setdefault('InputSpacePoints', ['ITkStripSpacePoints_Cached', 'ITkStripOverlapSpacePoints_Cached'] if flags.Acts.useCache else ['ITkStripSpacePoints', 'ITkStripOverlapSpacePoints'])
358  kwargs.setdefault('OutputSeeds', 'ActsStripSeeds')
359  kwargs.setdefault('OutputEstimatedTrackParameters', 'ActsStripEstimatedTrackParams')
360  kwargs.setdefault('DetectorElements', 'ITkStripDetectorElementCollection')
361  kwargs.setdefault('UsePixel', False)
362 
363  if flags.Acts.doMonitoring and 'MonTool' not in kwargs:
364  from ActsConfig.ActsMonitoringConfig import ActsITkStripSeedingMonitoringToolCfg
365  kwargs.setdefault('MonTool', acc.popToolsAndMerge(ActsITkStripSeedingMonitoringToolCfg(flags)))
366 
367  kwargs.setdefault('useTopSp', flags.Acts.reverseTrackFindingForStrips)
368 
369  acc.addEventAlgo(CompFactory.ActsTrk.SeedingAlg(name, **kwargs))
370  return acc
371 
372 
373 def ActsMainSeedingCfg(flags,
374  **kwargs) -> ComponentAccumulator:
375  acc = ComponentAccumulator()
376 
377  kwargs.setdefault('processPixels', flags.Detector.EnableITkPixel)
378  kwargs.setdefault('processStrips', flags.Detector.EnableITkStrip)
379 
380  if kwargs['processPixels']:
381  acc.merge(ActsPixelSeedingAlgCfg(flags, **extractChildKwargs(prefix='PixelSeedingAlg.', **kwargs)))
382  if kwargs['processStrips']:
383  acc.merge(ActsStripSeedingAlgCfg(flags, **extractChildKwargs(prefix='StripSeedingAlg.', **kwargs)))
384 
385  # Analysis extensions
386  if flags.Acts.doAnalysis:
387  if kwargs['processPixels']:
388  from ActsConfig.ActsAnalysisConfig import ActsPixelSeedAnalysisAlgCfg, ActsPixelEstimatedTrackParamsAnalysisAlgCfg
389  acc.merge(ActsPixelSeedAnalysisAlgCfg(flags, **extractChildKwargs(prefix='PixelSeedAnalysisAlg.', **kwargs)))
390  acc.merge(ActsPixelEstimatedTrackParamsAnalysisAlgCfg(flags, **extractChildKwargs(prefix='PixelEstimatedTrackParamsAnalysisAlg.', **kwargs)))
391 
392  if kwargs['processStrips']:
393  from ActsConfig.ActsAnalysisConfig import ActsStripSeedAnalysisAlgCfg, ActsStripEstimatedTrackParamsAnalysisAlgCfg
394  acc.merge(ActsStripSeedAnalysisAlgCfg(flags, **extractChildKwargs(prefix='StripSeedAnalysisAlg.', **kwargs)))
395  acc.merge(ActsStripEstimatedTrackParamsAnalysisAlgCfg(flags, **extractChildKwargs(prefix='StripEstimatedTrackParamsAnalysisAlg.', **kwargs)))
396 
397  return acc
398 
399 def ActsSeedingCfg(flags) -> ComponentAccumulator:
400  acc = ComponentAccumulator()
401  processPixels = flags.Detector.EnableITkPixel
402  processStrips = flags.Detector.EnableITkStrip
403 
404  # For conversion pass we do not process pixels
405  if flags.Tracking.ActiveConfig.extension in ["ActsConversion", "ActsLargeRadius"]:
406  processPixels = False
407  # For main pass disable strips if fast tracking configuration
408  elif flags.Tracking.doITkFastTracking:
409  processStrips = False
410 
411  kwargs = dict()
412  kwargs.setdefault('processPixels', processPixels)
413  kwargs.setdefault('processStrips', processStrips)
414 
415  # TO-DO: refactor this seeding tool configuration
416  if flags.Tracking.ActiveConfig.extension == "ActsHeavyIon" and processPixels:
417  kwargs.setdefault('PixelSeedingAlg.SeedTool', acc.popToolsAndMerge(ActsPixelSeedingToolCfg(flags,
418  name=f'{flags.Tracking.ActiveConfig.extension}PixelSeedingTool',
419  minPt=flags.Tracking.ActiveConfig.minPTSeed)))
420  if processStrips and flags.Acts.SeedingStrategy is SeedingStrategy.Default:
421  kwargs.setdefault('StripSeedingAlg.SeedTool', acc.popToolsAndMerge(ActsStripSeedingToolCfg(flags,
422  name=f'{flags.Tracking.ActiveConfig.extension}StripSeedingTool',
423  rMax=flags.Tracking.ActiveConfig.radMax)))
424 
425  if processPixels:
426  # Seeding algo
427  kwargs.setdefault('PixelSeedingAlg.name', f'{flags.Tracking.ActiveConfig.extension}PixelSeedingAlg')
428  kwargs.setdefault('PixelSeedingAlg.useFastTracking', flags.Tracking.doITkFastTracking)
429  kwargs.setdefault('PixelSeedingAlg.OutputSeeds', f'{flags.Tracking.ActiveConfig.extension}PixelSeeds')
430  kwargs.setdefault('PixelSeedingAlg.OutputEstimatedTrackParameters', f'{flags.Tracking.ActiveConfig.extension}PixelEstimatedTrackParams')
431 
432  pixelSpacePoints = ['ITkPixelSpacePoints_Cached'] if flags.Acts.useCache else ['ITkPixelSpacePoints']
433  if flags.Tracking.ActiveConfig.isSecondaryPass:
434  pixelSpacePoints = [f'ITk{flags.Tracking.ActiveConfig.extension.replace("Acts", "")}PixelSpacePoints_Cached'] if flags.Acts.useCache else [f'ITk{flags.Tracking.ActiveConfig.extension.replace("Acts", "")}PixelSpacePoints']
435  kwargs.setdefault('PixelSeedingAlg.InputSpacePoints', pixelSpacePoints)
436 
437  # Analysis algo(s)
438  if flags.Acts.doAnalysis:
439  kwargs.setdefault('PixelSeedAnalysisAlg.name', f'{flags.Tracking.ActiveConfig.extension}PixelSeedAnalysisAlg')
440  kwargs.setdefault('PixelSeedAnalysisAlg.extension', flags.Tracking.ActiveConfig.extension)
441  kwargs.setdefault('PixelSeedAnalysisAlg.InputSeedCollection', kwargs['PixelSeedingAlg.OutputSeeds'])
442 
443  kwargs.setdefault('PixelEstimatedTrackParamsAnalysisAlg.name', f'{flags.Tracking.ActiveConfig.extension}PixelEstimatedTrackParamsAnalysisAlg')
444  kwargs.setdefault('PixelEstimatedTrackParamsAnalysisAlg.extension', flags.Tracking.ActiveConfig.extension)
445  kwargs.setdefault('PixelEstimatedTrackParamsAnalysisAlg.InputTrackParamsCollection', kwargs['PixelSeedingAlg.OutputEstimatedTrackParameters'])
446 
447  if processStrips:
448  # Seeding algo
449  kwargs.setdefault('StripSeedingAlg.name', f'{flags.Tracking.ActiveConfig.extension}StripSeedingAlg')
450  kwargs.setdefault('StripSeedingAlg.OutputSeeds', f'{flags.Tracking.ActiveConfig.extension}StripSeeds')
451  kwargs.setdefault('StripSeedingAlg.OutputEstimatedTrackParameters', f'{flags.Tracking.ActiveConfig.extension}StripEstimatedTrackParams')
452  # Conversion pass does not use overlap space points
453  # Space Point naming is not yet fully connected to tracking passes - this will change
454  if flags.Tracking.ActiveConfig.extension == 'ActsConversion':
455  kwargs.setdefault('StripSeedingAlg.InputSpacePoints', ['ITkConversionStripSpacePoints_Cached'] if flags.Acts.useCache else ['ITkConversionStripSpacePoints'])
456  elif flags.Tracking.ActiveConfig.extension == 'ActsLargeRadius':
457  kwargs.setdefault('StripSeedingAlg.InputSpacePoints', ['ITkLargeRadiusStripSpacePoints_Cached',
458  'ITkLargeRadiusStripOverlapSpacePoints_Cached'] if flags.Acts.useCache else ['ITkLargeRadiusStripSpacePoints',
459  'ITkLargeRadiusStripOverlapSpacePoints'])
460  elif flags.Tracking.ActiveConfig.extension == 'ActsLowPt':
461  kwargs.setdefault('StripSeedingAlg.InputSpacePoints', ['ITkLowPtStripSpacePoints_Cached',
462  'ITkLowPtStripOverlapSpacePoints_Cached'] if flags.Acts.useCache else ['ITkLowPtStripSpacePoints',
463  'ITkLowPtStripOverlapSpacePoints'])
464  else:
465  kwargs.setdefault('StripSeedingAlg.InputSpacePoints', ['ITkStripSpacePoints_Cached',
466  'ITkStripOverlapSpacePoints_Cached'] if flags.Acts.useCache else ['ITkStripSpacePoints',
467  'ITkStripOverlapSpacePoints'])
468 
469  # Analysis algo(s)
470  if flags.Acts.doAnalysis:
471  kwargs.setdefault('StripSeedAnalysisAlg.name', f'{flags.Tracking.ActiveConfig.extension}StripSeedAnalysisAlg')
472  kwargs.setdefault('StripSeedAnalysisAlg.extension', flags.Tracking.ActiveConfig.extension)
473  kwargs.setdefault('StripSeedAnalysisAlg.InputSeedCollection', kwargs['StripSeedingAlg.OutputSeeds'])
474 
475  kwargs.setdefault('StripEstimatedTrackParamsAnalysisAlg.name', f'{flags.Tracking.ActiveConfig.extension}StripEstimatedTrackParamsAnalysisAlg')
476  kwargs.setdefault('StripEstimatedTrackParamsAnalysisAlg.extension', flags.Tracking.ActiveConfig.extension)
477  kwargs.setdefault('StripEstimatedTrackParamsAnalysisAlg.InputTrackParamsCollection', kwargs['StripSeedingAlg.OutputEstimatedTrackParameters'])
478 
479  acc.merge(ActsMainSeedingCfg(flags, **kwargs))
480 
481  if flags.Tracking.ActiveConfig.storeTrackSeeds:
482  # For the time being this only saves Pixel Seeds. Will add the Strip Seed case later
483  seedKey = f'{flags.Tracking.ActiveConfig.extension}PixelSeeds'
484  trackKey = f'SiSPTracksSeedSegments{flags.Tracking.ActiveConfig.extension}Tracks'
485  particleKey = f'SiSPSeedSegments{flags.Tracking.ActiveConfig.extension}'
486  acc.merge(ActsSeedToTrackCnvAlgCfg(flags,
487  name=f"{flags.Tracking.ActiveConfig.extension}SeedToTrackCnvAlg"),
488  SeedContainerKey=seedKey,
489  ACTSTracksLocation=trackKey)
490  from ActsConfig.ActsTrackFindingConfig import ActsTrackToTrackParticleCnvAlgCfg
491  acc.merge(ActsTrackToTrackParticleCnvAlgCfg(flags,
492  name=f"{flags.Tracking.ActiveConfig.extension}TracksSeedToTrackParticleCnvAlg",
493  TrackParticlesOutKey=particleKey,
494  ACTSTracksLocation=[trackKey]))
495  return acc
496 
497 
498 def ActsSeedToTrackCnvAlgCfg(flags,
499  name: str = 'ActsSeedToTrackCnvAlg',
500  **kwargs: dict) -> ComponentAccumulator:
501  acc = ComponentAccumulator()
502  from ActsConfig.ActsGeometryConfig import ActsDetectorElementToActsGeometryIdMappingAlgCfg
503  acc.merge( ActsDetectorElementToActsGeometryIdMappingAlgCfg(flags) )
504  kwargs.setdefault('DetectorElementToActsGeometryIdMapKey', 'DetectorElementToActsGeometryIdMap')
505 
506  kwargs.setdefault('SeedContainerKey', f'{flags.Tracking.ActiveConfig.extension}PixelSeeds')
507  kwargs.setdefault('EstimatedTrackParametersKey',f'{flags.Tracking.ActiveConfig.extension}PixelEstimatedTrackParams')
508 
509  kwargs.setdefault('ACTSTracksLocation', f'SiSPTracksSeedSegments{flags.Tracking.ActiveConfig.extension}Tracks') # This uses the same naming convention than the legacy code
510 
511  if 'TrackingGeometryTool' not in kwargs:
512  from ActsConfig.ActsGeometryConfig import ActsTrackingGeometryToolCfg
513  kwargs.setdefault('TrackingGeometryTool', acc.popToolsAndMerge(ActsTrackingGeometryToolCfg(flags)))
514 
515  acc.addEventAlgo(CompFactory.ActsTrk.SeedToTrackCnvAlg(name, **kwargs), primary=True)
516  return acc
517 
ActsSeedingConfig.ActsPixelOrthogonalSeedingToolCfg
ComponentAccumulator ActsPixelOrthogonalSeedingToolCfg(flags, str name="ActsPixelOrthogonalSeedingTool", **kwargs)
Definition: ActsSeedingConfig.py:113
ActsSeedingConfig.ActsPixelGbtsSeedingToolCfg
ComponentAccumulator ActsPixelGbtsSeedingToolCfg(flags, str name="ActsPixelGbtsSeedingTool", **kwargs)
Definition: ActsSeedingConfig.py:164
python.JetAnalysisCommon.ComponentAccumulator
ComponentAccumulator
Definition: JetAnalysisCommon.py:302
ActsUtilities.extractChildKwargs
dict extractChildKwargs(*str prefix, **dict kwargs)
Definition: ActsUtilities.py:3
ActsSeedingConfig.ActsPixelSeedingAlgCfg
ComponentAccumulator ActsPixelSeedingAlgCfg(flags, str name='ActsPixelSeedingAlg', **kwargs)
Definition: ActsSeedingConfig.py:254
python.BeamSpotConditionsConfig.BeamSpotCondAlgCfg
def BeamSpotCondAlgCfg(flags, name="BeamSpotCondAlg", **kwargs)
Definition: BeamSpotConditionsConfig.py:7
ActsAnalysisConfig.ActsPixelEstimatedTrackParamsAnalysisAlgCfg
ComponentAccumulator ActsPixelEstimatedTrackParamsAnalysisAlgCfg(flags, str name='ActsPixelEstimatedTrackParamsAnalysisAlg', str extension="Acts", **kwargs)
Definition: ActsAnalysisConfig.py:568
ActsAnalysisConfig.ActsStripSeedAnalysisAlgCfg
ComponentAccumulator ActsStripSeedAnalysisAlgCfg(flags, str name="ActsStripSeedAnalysisAlg", str extension="Acts", **kwargs)
Definition: ActsAnalysisConfig.py:459
ActsEventCnvConfig.ActsToTrkConverterToolCfg
ComponentAccumulator ActsToTrkConverterToolCfg(flags, str name="ActsToTrkConverterTool", **kwargs)
Definition: ActsEventCnvConfig.py:6
ActsSeedingConfig.ActsPixelSeedingToolCfg
ComponentAccumulator ActsPixelSeedingToolCfg(flags, str name="ActsPixelSeedingTool", **kwargs)
Definition: ActsSeedingConfig.py:11
ActsAnalysisConfig.ActsPixelSeedAnalysisAlgCfg
ComponentAccumulator ActsPixelSeedAnalysisAlgCfg(flags, str name="ActsPixelSeedAnalysisAlg", str extension="Acts", **kwargs)
Definition: ActsAnalysisConfig.py:441
ActsMonitoringConfig.ActsITkPixelSeedingMonitoringToolCfg
ComponentAccumulator ActsITkPixelSeedingMonitoringToolCfg(flags, str name="ActsITkPixelSeedingMonitoringTool", **kwargs)
Definition: ActsMonitoringConfig.py:120
ActsSeedingConfig.ActsFastPixelOrthogonalSeedingToolCfg
ComponentAccumulator ActsFastPixelOrthogonalSeedingToolCfg(flags, str name="ActsFastPixelOrthogonalSeedingTool", **kwargs)
Definition: ActsSeedingConfig.py:121
ActsSeedingConfig.ActsStripSeedingAlgCfg
ComponentAccumulator ActsStripSeedingAlgCfg(flags, str name='ActsStripSeedingAlg', **kwargs)
Definition: ActsSeedingConfig.py:318
ActsTrackParamsEstimationConfig.ActsTrackParamsEstimationToolCfg
ComponentAccumulator ActsTrackParamsEstimationToolCfg(ConfigFlags, str name="ActsTrackParamsEstimationTool", **kwargs)
Definition: ActsTrackParamsEstimationConfig.py:6
ActsGeometryConfig.ActsTrackingGeometryToolCfg
ComponentAccumulator ActsTrackingGeometryToolCfg(flags, str name="ActsTrackingGeometryTool")
Definition: ActsGeometryConfig.py:100
ActsAnalysisConfig.ActsStripEstimatedTrackParamsAnalysisAlgCfg
ComponentAccumulator ActsStripEstimatedTrackParamsAnalysisAlgCfg(flags, str name='ActsStripEstimatedTrackParamsAnalysisAlg', str extension="Acts", **kwargs)
Definition: ActsAnalysisConfig.py:581
ActsSeedingConfig.ActsSiSpacePointsSeedMakerToolCfg
ComponentAccumulator ActsSiSpacePointsSeedMakerToolCfg(flags, str name='ActsSiSpacePointsSeedMakerTool', **kwargs)
Definition: ActsSeedingConfig.py:174
ITkPixelGeoModelConfig.ITkPixelReadoutGeometryCfg
def ITkPixelReadoutGeometryCfg(flags, setGeometryAlignable=False, setAlignmentFolderName="/Indet/Align")
Definition: ITkPixelGeoModelConfig.py:39
ActsSeedingConfig.ActsSeedingCfg
ComponentAccumulator ActsSeedingCfg(flags)
Definition: ActsSeedingConfig.py:399
ActsGeometryConfig.ActsDetectorElementToActsGeometryIdMappingAlgCfg
ComponentAccumulator ActsDetectorElementToActsGeometryIdMappingAlgCfg(flags, str name="ActsDetectorElementToActsGeometryIdMappingAlg", **kwargs)
Definition: ActsGeometryConfig.py:257
ActsSeedingConfig.ActsFastPixelSeedingToolCfg
ComponentAccumulator ActsFastPixelSeedingToolCfg(flags, str name="ActsFastPixelSeedingTool", **kwargs)
Definition: ActsSeedingConfig.py:40
ITkStripGeoModelConfig.ITkStripReadoutGeometryCfg
def ITkStripReadoutGeometryCfg(flags, setGeometryAlignable=False, setAlignmentFolderName="/Indet/Align")
Definition: ITkStripGeoModelConfig.py:39
ActsMonitoringConfig.ActsITkStripSeedingMonitoringToolCfg
ComponentAccumulator ActsITkStripSeedingMonitoringToolCfg(flags, str name="ActsITkStripSeedingMonitoringTool", **kwargs)
Definition: ActsMonitoringConfig.py:141
python.MagFieldServicesConfig.AtlasFieldCacheCondAlgCfg
def AtlasFieldCacheCondAlgCfg(flags, **kwargs)
Definition: MagFieldServicesConfig.py:8
python.Utils.unixtools.find_datafile
def find_datafile(fname, pathlist=None, access=os.R_OK)
pathresolver-like helper function --------------------------------------—
Definition: unixtools.py:67
ActsSeedingConfig.ActsSeedToTrackCnvAlgCfg
ComponentAccumulator ActsSeedToTrackCnvAlgCfg(flags, str name='ActsSeedToTrackCnvAlg', **dict kwargs)
Definition: ActsSeedingConfig.py:498
readCCLHist.float
float
Definition: readCCLHist.py:83
ActsSeedingConfig.ActsMainSeedingCfg
ComponentAccumulator ActsMainSeedingCfg(flags, **kwargs)
Definition: ActsSeedingConfig.py:373
ActsSeedingConfig.ActsStripSeedingToolCfg
ComponentAccumulator ActsStripSeedingToolCfg(flags, str name="ActsStripSeedingTool", **kwargs)
Definition: ActsSeedingConfig.py:72
ActsTrackFindingConfig.ActsTrackToTrackParticleCnvAlgCfg
ComponentAccumulator ActsTrackToTrackParticleCnvAlgCfg(flags, str name="ActsTrackToTrackParticleCnvAlg", **kwargs)
Definition: ActsTrackFindingConfig.py:322
ActsSeedingConfig.ActsStripOrthogonalSeedingToolCfg
ComponentAccumulator ActsStripOrthogonalSeedingToolCfg(flags, str name="ActsStripOrthogonalSeedingTool", **kwargs)
Definition: ActsSeedingConfig.py:136
Generated on Thu Nov 7 2024 21:09:30 for ATLAS Offline Software by   doxygen 1.8.18