ATLAS Offline Software
Functions
python.OverlayCopyAlgsConfig Namespace Reference

Functions

def CopyCaloCalibrationHitContainerAlgCfg (flags, collectionName, name="CopyCaloCalibrationHitContainer", **kwargs)
 
def CopyJetTruthInfoAlgCfg (flags, name, **kwargs)
 
def CopyInTimeAntiKt4JetTruthInfoCfg (flags, name="CopyInTimeAntiKt4JetTruthInfo", **kwargs)
 
def CopyOutOfTimeAntiKt4JetTruthInfoCfg (flags, name="CopyOutOfTimeAntiKt4JetTruthInfo", **kwargs)
 
def CopyInTimeAntiKt6JetTruthInfoCfg (flags, name="CopyInTimeAntiKt6JetTruthInfo", **kwargs)
 
def CopyOutOfTimeAntiKt6JetTruthInfoCfg (flags, name="CopyOutOfTimeAntiKt6JetTruthInfo", **kwargs)
 
def CopyJetTruthInfoCfg (flags, **kwargs)
 
def CopyPileupParticleTruthInfoCfg (flags, name="CopyPileupParticleTruthInfo", **kwargs)
 
def CopyMcEventCollectionCfg (flags, name="CopyMcEventCollection", **kwargs)
 
def CopyPixelClusterContainerAlgCfg (flags, **kwargs)
 
def CopyITkPixelClusterContainerAlgCfg (flags, **kwargs)
 
def CopySCT_ClusterContainerAlgCfg (flags, **kwargs)
 
def CopyITkStripClusterContainerAlgCfg (flags, **kwargs)
 
def CopyTRT_DriftCircleContainerAlgCfg (flags, **kwargs)
 
def CopyTrackCollectionAlgCfg (flags, collectionName, **kwargs)
 
def CopyDetailedTrackTruthCollectionAlgCfg (flags, collectionName, **kwargs)
 
def CopyPRD_MultiTruthCollectionAlgCfg (flags, collectionName, **kwargs)
 
def CopyTrackRecordCollectionAlgCfg (flags, collectionName, name="CopyTrackRecordCollection", **kwargs)
 
def CopyCaloCalibrationHitContainersCfg (flags, **kwargs)
 
def CopyPixelClusterContainerCfg (flags, **kwargs)
 
def CopyITkPixelClusterContainerCfg (flags, **kwargs)
 
def CopySCT_ClusterContainerCfg (flags, **kwargs)
 
def CopyITkStripClusterContainerCfg (flags, **kwargs)
 
def CopyTRT_DriftCircleContainerCfg (flags, **kwargs)
 
def CopyTrackRecordCollectionsCfg (flags, **kwargs)
 
def CopyTrackCollectionsCfg (flags, **kwargs)
 
def CopyITkTrackCollectionsCfg (flags, **kwargs)
 
def CopyDetailedTrackTruthCollectionsCfg (flags, **kwargs)
 
def CopyPRD_MultiTruthCollectionsCfg (flags, **kwargs)
 

Detailed Description

Define methods to construct configured overlay copy algorithms

Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration

Function Documentation

◆ CopyCaloCalibrationHitContainerAlgCfg()

def python.OverlayCopyAlgsConfig.CopyCaloCalibrationHitContainerAlgCfg (   flags,
  collectionName,
  name = "CopyCaloCalibrationHitContainer",
**  kwargs 
)
Return a ComponentAccumulator for the CaloCalibrationHitContainer copying

Definition at line 11 of file OverlayCopyAlgsConfig.py.

11 def CopyCaloCalibrationHitContainerAlgCfg(flags, collectionName, name="CopyCaloCalibrationHitContainer", **kwargs):
12  """Return a ComponentAccumulator for the CaloCalibrationHitContainer copying"""
13  acc = ComponentAccumulator()
14 
15  kwargs.setdefault("collectionName", collectionName)
16  kwargs.setdefault("SignalInputKey", flags.Overlay.SigPrefix + collectionName)
17  kwargs.setdefault("OutputKey", collectionName)
18 
19  from SGComps.SGInputLoaderConfig import SGInputLoaderCfg
20  acc.merge(SGInputLoaderCfg(flags, [f'CaloCalibrationHitContainer#{kwargs["SignalInputKey"]}']))
21 
22  # Copy CaloCalibrationHitContainer
23  acc.addEventAlgo(CompFactory.CopyCaloCalibrationHitContainer(name + collectionName, **kwargs))
24 
25  # Re-map signal address
26  from SGComps.AddressRemappingConfig import AddressRemappingCfg
27  acc.merge(AddressRemappingCfg([
28  f"CaloCalibrationHitContainer#{collectionName}->{flags.Overlay.SigPrefix}{collectionName}"
29  ]))
30 
31  # Output
32  if flags.Output.doWriteRDO:
33  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
34  acc.merge(OutputStreamCfg(flags, "RDO", ItemList=[
35  f"CaloCalibrationHitContainer#{collectionName}"
36  ]))
37 
38  # Add signal output
39  if flags.Output.doWriteRDO_SGNL:
40  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
41  acc.merge(OutputStreamCfg(flags, "RDO_SGNL", ItemList=[
42  f"CaloCalibrationHitContainer#{flags.Overlay.SigPrefix}{collectionName}"
43  ]))
44 
45  return acc
46 
47 

◆ CopyCaloCalibrationHitContainersCfg()

def python.OverlayCopyAlgsConfig.CopyCaloCalibrationHitContainersCfg (   flags,
**  kwargs 
)
Return overlay configuration for the CopyCalibrationHitContainer algorithms

Definition at line 385 of file OverlayCopyAlgsConfig.py.

385 def CopyCaloCalibrationHitContainersCfg(flags, **kwargs):
386  """Return overlay configuration for the CopyCalibrationHitContainer algorithms"""
387 
388  acc = ComponentAccumulator()
389  kwargs.setdefault("ExtraInputs", flags.Overlay.ExtraInputs)
390 
391  allowedContainers = [
392  "LArCalibrationHitActive", "LArCalibrationHitDeadMaterial", "LArCalibrationHitInactive"
393  "TileCalibrationCellHitCnt", "TileCalibrationDMHitCnt",
394  "TileCalibHitActiveCell", "TileCalibHitInactiveCell", "TileCalibHitDeadMaterial",
395  ]
396  availableContainers = []
397 
398  # Detect the list of calibration hit containers
399  for container in allowedContainers:
400  if (flags.Overlay.ByteStream and container in flags.Input.Collections) \
401  or (not flags.Overlay.ByteStream and container in flags.Input.SecondaryCollections):
402  availableContainers.append(container)
403 
404  for container in availableContainers:
405  acc.merge(CopyCaloCalibrationHitContainerAlgCfg(flags, container, **kwargs))
406 
407  return acc
408 
409 

◆ CopyDetailedTrackTruthCollectionAlgCfg()

def python.OverlayCopyAlgsConfig.CopyDetailedTrackTruthCollectionAlgCfg (   flags,
  collectionName,
**  kwargs 
)
Return a ComponentAccumulator for the DetailedTrackTruthCollection copying

Definition at line 308 of file OverlayCopyAlgsConfig.py.

308 def CopyDetailedTrackTruthCollectionAlgCfg(flags, collectionName, **kwargs):
309  """Return a ComponentAccumulator for the DetailedTrackTruthCollection copying"""
310  acc = ComponentAccumulator()
311 
312  kwargs.setdefault("OutputKey", flags.Overlay.BkgPrefix + collectionName)
313  kwargs.setdefault("InputKey", collectionName)
314 
315  alg = CompFactory.CopyDetailedTrackTruthCollection("CopyDetailedTrackTruthCollection"+collectionName)
316  acc.addEventAlgo(alg)
317 
318  if flags.Output.doWriteRDO:
319  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
320  acc.merge(OutputStreamCfg(flags, "RDO", ItemList=[
321  "DetailedTrackTruthCollection#" + flags.Overlay.BkgPrefix+collectionName
322  ]))
323 
324  return acc
325 
326 

◆ CopyDetailedTrackTruthCollectionsCfg()

def python.OverlayCopyAlgsConfig.CopyDetailedTrackTruthCollectionsCfg (   flags,
**  kwargs 
)
Return overlay configuration for copying detailed track truth

Definition at line 545 of file OverlayCopyAlgsConfig.py.

545 def CopyDetailedTrackTruthCollectionsCfg(flags, **kwargs):
546  """ Return overlay configuration for copying detailed track truth"""
547 
548  acc = ComponentAccumulator()
549 
550  allowedContainers = [
551  "DisappearingTracksDetailedTruth",
552  "ResolvedForwardTracksDetailedTruth",
553  "CombinedInDetTracksDetailedTruth",
554  "ExtendedLargeD0TracksDetailedTruth"
555  ]
556 
557  availableContainers = []
558 
559  # Detect the list of detailed track truth collections
560  if flags.Common.ProductionStep == ProductionStep.FastChain:
561  availableContainers = allowedContainers
562  else:
563  hardScatterInputCollections = flags.Input.Collections
564  for container in allowedContainers:
565  if container in hardScatterInputCollections:
566  availableContainers.append(container)
567 
568  for container in availableContainers:
569  acc.merge(CopyDetailedTrackTruthCollectionAlgCfg(flags, container, **kwargs))
570 
571  return acc
572 
573 

◆ CopyInTimeAntiKt4JetTruthInfoCfg()

def python.OverlayCopyAlgsConfig.CopyInTimeAntiKt4JetTruthInfoCfg (   flags,
  name = "CopyInTimeAntiKt4JetTruthInfo",
**  kwargs 
)
Return a ComponentAccumulator for the in-time pile-up jets copying

Definition at line 72 of file OverlayCopyAlgsConfig.py.

72 def CopyInTimeAntiKt4JetTruthInfoCfg(flags, name="CopyInTimeAntiKt4JetTruthInfo", **kwargs):
73  """Return a ComponentAccumulator for the in-time pile-up jets copying"""
74  kwargs.setdefault("BkgInputKey", f"{flags.Overlay.BkgPrefix}InTimeAntiKt4TruthJets")
75  kwargs.setdefault("OutputKey", "InTimeAntiKt4TruthJets")
76 
77  return CopyJetTruthInfoAlgCfg(flags, name, **kwargs)
78 
79 

◆ CopyInTimeAntiKt6JetTruthInfoCfg()

def python.OverlayCopyAlgsConfig.CopyInTimeAntiKt6JetTruthInfoCfg (   flags,
  name = "CopyInTimeAntiKt6JetTruthInfo",
**  kwargs 
)
Return a ComponentAccumulator for the in-time pile-up jets copying

Definition at line 88 of file OverlayCopyAlgsConfig.py.

88 def CopyInTimeAntiKt6JetTruthInfoCfg(flags, name="CopyInTimeAntiKt6JetTruthInfo", **kwargs):
89  """Return a ComponentAccumulator for the in-time pile-up jets copying"""
90  kwargs.setdefault("BkgInputKey", f"{flags.Overlay.BkgPrefix}InTimeAntiKt6TruthJets")
91  kwargs.setdefault("OutputKey", "InTimeAntiKt6TruthJets")
92 
93  return CopyJetTruthInfoAlgCfg(flags, name, **kwargs)
94 
95 

◆ CopyITkPixelClusterContainerAlgCfg()

def python.OverlayCopyAlgsConfig.CopyITkPixelClusterContainerAlgCfg (   flags,
**  kwargs 
)
Return a ComponentAccumulator for the CopyPixelClusterContainer algorithm

Definition at line 224 of file OverlayCopyAlgsConfig.py.

224 def CopyITkPixelClusterContainerAlgCfg(flags, **kwargs):
225  """Return a ComponentAccumulator for the CopyPixelClusterContainer algorithm"""
226  acc = ComponentAccumulator()
227 
228  kwargs.setdefault("InputKey", "ITkPixelClusters")
229  kwargs.setdefault("OutputKey", flags.Overlay.BkgPrefix+"ITkPixelClusters")
230  kwargs.setdefault("ExtraInputs", [('InDetDD::SiDetectorElementCollection', 'ConditionStore+ITkPixelDetectorElementCollection')])
231 
232  alg = CompFactory.CopyPixelClusterContainer("CopyPixelClusterContainer", **kwargs)
233  acc.addEventAlgo(alg)
234 
235  return acc
236 

◆ CopyITkPixelClusterContainerCfg()

def python.OverlayCopyAlgsConfig.CopyITkPixelClusterContainerCfg (   flags,
**  kwargs 
)
Return overlay configuration for the CopyITkPixelClusterContainer algorithm

Definition at line 422 of file OverlayCopyAlgsConfig.py.

422 def CopyITkPixelClusterContainerCfg(flags, **kwargs):
423  """Return overlay configuration for the CopyITkPixelClusterContainer algorithm"""
424  acc = CopyITkPixelClusterContainerAlgCfg(flags, **kwargs)
425  if flags.Output.doWriteRDO:
426  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
427  acc.merge(OutputStreamCfg(flags, "RDO", ItemList=[
428  "InDet::PixelClusterContainer#Bkg_ITkPixelClusters"
429  ]))
430 
431  return acc
432 

◆ CopyITkStripClusterContainerAlgCfg()

def python.OverlayCopyAlgsConfig.CopyITkStripClusterContainerAlgCfg (   flags,
**  kwargs 
)
Return a ComponentAccumulator for the CopySCT_ClusterContainer algorithm

Definition at line 250 of file OverlayCopyAlgsConfig.py.

250 def CopyITkStripClusterContainerAlgCfg(flags, **kwargs):
251  """Return a ComponentAccumulator for the CopySCT_ClusterContainer algorithm"""
252  acc = ComponentAccumulator()
253 
254  kwargs.setdefault("InputKey", "ITkStripClusters")
255  kwargs.setdefault("OutputKey", flags.Overlay.BkgPrefix+"ITkStripClusters")
256  kwargs.setdefault("ExtraInputs", [('InDetDD::SiDetectorElementCollection', 'ConditionStore+ITkStripDetectorElementCollection')])
257 
258  alg = CompFactory.CopySCT_ClusterContainer("CopySCT_ClusterContainer", **kwargs)
259  acc.addEventAlgo(alg)
260 
261  return acc
262 

◆ CopyITkStripClusterContainerCfg()

def python.OverlayCopyAlgsConfig.CopyITkStripClusterContainerCfg (   flags,
**  kwargs 
)
Return overlay configuration for the CopySCT_ClusterContainer algorithm

Definition at line 445 of file OverlayCopyAlgsConfig.py.

445 def CopyITkStripClusterContainerCfg(flags, **kwargs):
446  """Return overlay configuration for the CopySCT_ClusterContainer algorithm"""
447 
448  acc = CopyITkStripClusterContainerAlgCfg(flags, **kwargs)
449  if flags.Output.doWriteRDO:
450  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
451  acc.merge(OutputStreamCfg(flags, "RDO", ItemList=[
452  "InDet::SCT_ClusterContainer#Bkg_ITkStripClusters"
453  ]))
454 
455  return acc
456 

◆ CopyITkTrackCollectionsCfg()

def python.OverlayCopyAlgsConfig.CopyITkTrackCollectionsCfg (   flags,
**  kwargs 
)
Return overlay configuration for copying tracks

Definition at line 522 of file OverlayCopyAlgsConfig.py.

522 def CopyITkTrackCollectionsCfg(flags, **kwargs):
523  """ Return overlay configuration for copying tracks"""
524 
525  acc = ComponentAccumulator()
526 
527  allowedContainers = [
528  "CombinedITkTracks",
529  "ResolvedConversionTracks"
530  ]
531  availableContainers = []
532  # Detect the list of track collections
533  if flags.Common.ProductionStep == ProductionStep.FastChain:
534  availableContainers = allowedContainers
535  else:
536  hardScatterInputCollections = flags.Input.Collections
537  for container in allowedContainers:
538  if container in hardScatterInputCollections:
539  availableContainers.append(container)
540  for container in availableContainers:
541  acc.merge(CopyTrackCollectionAlgCfg(flags, container, **kwargs))
542 
543  return acc
544 

◆ CopyJetTruthInfoAlgCfg()

def python.OverlayCopyAlgsConfig.CopyJetTruthInfoAlgCfg (   flags,
  name,
**  kwargs 
)
Return a ComponentAccumulator for the pile-up jets copying

Definition at line 48 of file OverlayCopyAlgsConfig.py.

48 def CopyJetTruthInfoAlgCfg(flags, name, **kwargs):
49  """Return a ComponentAccumulator for the pile-up jets copying"""
50  acc = ComponentAccumulator()
51 
52  if "OutputKey" not in kwargs or "BkgInputKey" not in kwargs:
53  raise RuntimeError("'OutputKey' and 'BkgInputKey' need to be defined")
54 
55  from SGComps.SGInputLoaderConfig import SGInputLoaderCfg
56  acc.merge(SGInputLoaderCfg(flags, [f'xAOD::JetContainer#{kwargs["BkgInputKey"]}']))
57 
58  # Copy jets
59  acc.addEventAlgo(CompFactory.CopyJetTruthInfo(name, **kwargs))
60 
61  # Output
62  if flags.Output.doWriteRDO:
63  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
64  acc.merge(OutputStreamCfg(flags, "RDO", ItemList=[
65  f'xAOD::JetContainer#{kwargs["OutputKey"]}',
66  f'xAOD::AuxContainerBase!#{kwargs["OutputKey"]}Aux.-constituentLinks.-constituentWeights',
67  ]))
68 
69  return acc
70 
71 

◆ CopyJetTruthInfoCfg()

def python.OverlayCopyAlgsConfig.CopyJetTruthInfoCfg (   flags,
**  kwargs 
)
Return a ComponentAccumulator for the CopyJetTruthInfo algorithms

Definition at line 104 of file OverlayCopyAlgsConfig.py.

104 def CopyJetTruthInfoCfg(flags, **kwargs):
105  """Return a ComponentAccumulator for the CopyJetTruthInfo algorithms"""
106  acc = ComponentAccumulator()
107  if flags.Overlay.DataOverlay:
108  return acc
109 
110  allowedContainers = [
111  f"{flags.Overlay.BkgPrefix}InTimeAntiKt4TruthJets",
112  f"{flags.Overlay.BkgPrefix}OutOfTimeAntiKt4TruthJets",
113  f"{flags.Overlay.BkgPrefix}InTimeAntiKt6TruthJets",
114  f"{flags.Overlay.BkgPrefix}OutOfTimeAntiKt6TruthJets"
115  ]
116  availableContainers = []
117 
118  # Detect the list of track record collections
119  for container in allowedContainers:
120  if container in flags.Input.Collections: # SecondaryCollections
121  availableContainers.append(container)
122  if allowedContainers[0] in availableContainers:
123  acc.merge(CopyInTimeAntiKt4JetTruthInfoCfg(flags, **kwargs))
124  if allowedContainers[1] in availableContainers:
125  acc.merge(CopyOutOfTimeAntiKt4JetTruthInfoCfg(flags, **kwargs))
126  if allowedContainers[2] in availableContainers:
127  acc.merge(CopyInTimeAntiKt6JetTruthInfoCfg(flags, **kwargs))
128  if allowedContainers[3] in availableContainers:
129  acc.merge(CopyOutOfTimeAntiKt6JetTruthInfoCfg(flags, **kwargs))
130  return acc
131 
132 

◆ CopyMcEventCollectionCfg()

def python.OverlayCopyAlgsConfig.CopyMcEventCollectionCfg (   flags,
  name = "CopyMcEventCollection",
**  kwargs 
)
Return a ComponentAccumulator for the CopyMcEventCollection algorithm

Definition at line 161 of file OverlayCopyAlgsConfig.py.

161 def CopyMcEventCollectionCfg(flags, name="CopyMcEventCollection", **kwargs):
162  """Return a ComponentAccumulator for the CopyMcEventCollection algorithm"""
163  acc = ComponentAccumulator()
164 
165  kwargs.setdefault("RemoveBkgHardScatterTruth", True)
166 
167  inputs = []
168  if flags.Overlay.DataOverlay:
169  # Disable background for data overlay
170  kwargs.setdefault("BkgInputKey", "")
171  else:
172  kwargs.setdefault("BkgInputKey", f"{flags.Overlay.BkgPrefix}TruthEvent")
173  inputs.append(f'McEventCollection#{kwargs["BkgInputKey"]}')
174  kwargs.setdefault("SignalInputKey", f"{flags.Overlay.SigPrefix}TruthEvent")
175  inputs.append(f'McEventCollection#{kwargs["SignalInputKey"]}')
176 
177  if flags.Sim.DoFullChain and flags.Common.isOverlay:
178  kwargs.setdefault("OutputKey", "BeamTruthEvent")
179  else:
180  kwargs.setdefault("OutputKey", "TruthEvent")
181 
182  from SGComps.SGInputLoaderConfig import SGInputLoaderCfg
183  acc.merge(SGInputLoaderCfg(flags, inputs))
184 
185  # Merge and copy McEventCollection
186  acc.addEventAlgo(CompFactory.CopyMcEventCollection(name, **kwargs))
187 
188  # Re-map signal address
189  from SGComps.AddressRemappingConfig import AddressRemappingCfg
190  acc.merge(AddressRemappingCfg([
191  f"McEventCollection#TruthEvent->{flags.Overlay.SigPrefix}TruthEvent"
192  ]))
193 
194  # Output
195  if flags.Output.doWriteRDO:
196  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
197  acc.merge(OutputStreamCfg(flags, "RDO", ItemList=[
198  "McEventCollection#TruthEvent"
199  ]))
200 
201  # Add signal output
202  if flags.Output.doWriteRDO_SGNL:
203  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
204  acc.merge(OutputStreamCfg(flags, "RDO_SGNL", ItemList=[
205  f"McEventCollection#{flags.Overlay.SigPrefix}TruthEvent"
206  ]))
207 
208  return acc
209 
210 

◆ CopyOutOfTimeAntiKt4JetTruthInfoCfg()

def python.OverlayCopyAlgsConfig.CopyOutOfTimeAntiKt4JetTruthInfoCfg (   flags,
  name = "CopyOutOfTimeAntiKt4JetTruthInfo",
**  kwargs 
)
Return a ComponentAccumulator for the out-of-time pile-up jets copying

Definition at line 80 of file OverlayCopyAlgsConfig.py.

80 def CopyOutOfTimeAntiKt4JetTruthInfoCfg(flags, name="CopyOutOfTimeAntiKt4JetTruthInfo", **kwargs):
81  """Return a ComponentAccumulator for the out-of-time pile-up jets copying"""
82  kwargs.setdefault("BkgInputKey", f"{flags.Overlay.BkgPrefix}OutOfTimeAntiKt4TruthJets")
83  kwargs.setdefault("OutputKey", "OutOfTimeAntiKt4TruthJets")
84 
85  return CopyJetTruthInfoAlgCfg(flags, name, **kwargs)
86 
87 

◆ CopyOutOfTimeAntiKt6JetTruthInfoCfg()

def python.OverlayCopyAlgsConfig.CopyOutOfTimeAntiKt6JetTruthInfoCfg (   flags,
  name = "CopyOutOfTimeAntiKt6JetTruthInfo",
**  kwargs 
)
Return a ComponentAccumulator for the out-of-time pile-up jets copying

Definition at line 96 of file OverlayCopyAlgsConfig.py.

96 def CopyOutOfTimeAntiKt6JetTruthInfoCfg(flags, name="CopyOutOfTimeAntiKt6JetTruthInfo", **kwargs):
97  """Return a ComponentAccumulator for the out-of-time pile-up jets copying"""
98  kwargs.setdefault("BkgInputKey", f"{flags.Overlay.BkgPrefix}OutOfTimeAntiKt6TruthJets")
99  kwargs.setdefault("OutputKey", "OutOfTimeAntiKt6TruthJets")
100 
101  return CopyJetTruthInfoAlgCfg(flags, name, **kwargs)
102 
103 

◆ CopyPileupParticleTruthInfoCfg()

def python.OverlayCopyAlgsConfig.CopyPileupParticleTruthInfoCfg (   flags,
  name = "CopyPileupParticleTruthInfo",
**  kwargs 
)
Return a ComponentAccumulator for the in-time pile-up jets copying

Definition at line 133 of file OverlayCopyAlgsConfig.py.

133 def CopyPileupParticleTruthInfoCfg(flags, name="CopyPileupParticleTruthInfo", **kwargs):
134  """Return a ComponentAccumulator for the in-time pile-up jets copying"""
135  acc = ComponentAccumulator()
136  if flags.Overlay.DataOverlay:
137  return acc
138 
139  requiredContainer = f"{flags.Overlay.BkgPrefix}TruthPileupParticles"
140 
141  # Detect the list of track record collections
142  if requiredContainer in flags.Input.Collections: # SecondaryCollections
143  kwargs.setdefault("BkgInputKey", requiredContainer)
144  kwargs.setdefault("OutputKey", "TruthPileupParticles")
145 
146  from SGComps.SGInputLoaderConfig import SGInputLoaderCfg
147  acc.merge(SGInputLoaderCfg(flags, [f'xAOD::TruthParticleContainer#{kwargs["BkgInputKey"]}']))
148 
149  # Copy jets
150  acc.addEventAlgo(CompFactory.CopyPileupParticleTruthInfo(name, **kwargs))
151 
152  if flags.Output.doWriteRDO:
153  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
154  acc.merge(OutputStreamCfg(flags, "RDO", ItemList=[
155  "xAOD::TruthParticleContainer#TruthPileupParticles",
156  "xAOD::TruthParticleAuxContainer#TruthPileupParticlesAux."
157  ]))
158  return acc
159 
160 

◆ CopyPixelClusterContainerAlgCfg()

def python.OverlayCopyAlgsConfig.CopyPixelClusterContainerAlgCfg (   flags,
**  kwargs 
)
Return a ComponentAccumulator for the CopyPixelClusterContainer algorithm

Definition at line 211 of file OverlayCopyAlgsConfig.py.

211 def CopyPixelClusterContainerAlgCfg(flags, **kwargs):
212  """Return a ComponentAccumulator for the CopyPixelClusterContainer algorithm"""
213  acc = ComponentAccumulator()
214 
215  kwargs.setdefault("InputKey", "PixelClusters")
216  kwargs.setdefault("OutputKey", flags.Overlay.BkgPrefix+"PixelClusters")
217  kwargs.setdefault("ExtraInputs", {('InDetDD::SiDetectorElementCollection', 'ConditionStore+PixelDetectorElementCollection')})
218 
219  alg = CompFactory.CopyPixelClusterContainer("CopyPixelClusterContainer", **kwargs)
220  acc.addEventAlgo(alg)
221 
222  return acc
223 

◆ CopyPixelClusterContainerCfg()

def python.OverlayCopyAlgsConfig.CopyPixelClusterContainerCfg (   flags,
**  kwargs 
)
Return overlay configuration for the CopyPixelClusterContainer algorithm

Definition at line 410 of file OverlayCopyAlgsConfig.py.

410 def CopyPixelClusterContainerCfg(flags, **kwargs):
411  """Return overlay configuration for the CopyPixelClusterContainer algorithm"""
412 
413  acc = CopyPixelClusterContainerAlgCfg(flags, **kwargs)
414  if flags.Output.doWriteRDO:
415  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
416  acc.merge(OutputStreamCfg(flags, "RDO", ItemList=[
417  "InDet::PixelClusterContainer#Bkg_PixelClusters"
418  ]))
419 
420  return acc
421 

◆ CopyPRD_MultiTruthCollectionAlgCfg()

def python.OverlayCopyAlgsConfig.CopyPRD_MultiTruthCollectionAlgCfg (   flags,
  collectionName,
**  kwargs 
)
Return a ComponentAccumulator for the PRD_MultiTruthCollection copying

Definition at line 327 of file OverlayCopyAlgsConfig.py.

327 def CopyPRD_MultiTruthCollectionAlgCfg(flags, collectionName, **kwargs):
328  """Return a ComponentAccumulator for the PRD_MultiTruthCollection copying"""
329  acc = ComponentAccumulator()
330 
331  kwargs.setdefault("OutputKey", flags.Overlay.BkgPrefix + collectionName)
332  kwargs.setdefault("InputKey", collectionName)
333 
334  alg = CompFactory.CopyPRD_MultiTruthCollection("CopyPRD_MultiTruthCollection"+collectionName)
335  acc.addEventAlgo(alg)
336 
337  if flags.Output.doWriteRDO:
338  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
339  acc.merge(OutputStreamCfg(flags, "RDO", ItemList=[
340  "PRD_MultiTruthCollection#" + flags.Overlay.BkgPrefix+collectionName
341  ]))
342 
343  return acc
344 
345 

◆ CopyPRD_MultiTruthCollectionsCfg()

def python.OverlayCopyAlgsConfig.CopyPRD_MultiTruthCollectionsCfg (   flags,
**  kwargs 
)
Return overlay configuration for copying detailed track truth

Definition at line 574 of file OverlayCopyAlgsConfig.py.

574 def CopyPRD_MultiTruthCollectionsCfg(flags, **kwargs):
575  """ Return overlay configuration for copying detailed track truth"""
576 
577  acc = ComponentAccumulator()
578 
579  allowedContainers = [
580  "PRD_MultiTruthTRT",
581  "PRD_MultiTruthPixel",
582  "PRD_MultiTruthSCT"
583  ]
584 
585  availableContainers = []
586 
587  # Detect the list of detailed track truth collections
588  if flags.Common.ProductionStep == ProductionStep.FastChain:
589  availableContainers = allowedContainers
590  else:
591  hardScatterInputCollections = flags.Input.Collections
592  for container in allowedContainers:
593  if container in hardScatterInputCollections:
594  availableContainers.append(container)
595 
596  for container in availableContainers:
597  acc.merge(CopyPRD_MultiTruthCollectionAlgCfg(flags, container, **kwargs))
598 
599  return acc

◆ CopySCT_ClusterContainerAlgCfg()

def python.OverlayCopyAlgsConfig.CopySCT_ClusterContainerAlgCfg (   flags,
**  kwargs 
)
Return a ComponentAccumulator for the CopySCT_ClusterContainer algorithm

Definition at line 237 of file OverlayCopyAlgsConfig.py.

237 def CopySCT_ClusterContainerAlgCfg(flags, **kwargs):
238  """Return a ComponentAccumulator for the CopySCT_ClusterContainer algorithm"""
239  acc = ComponentAccumulator()
240 
241  kwargs.setdefault("InputKey", "SCT_Clusters")
242  kwargs.setdefault("OutputKey", flags.Overlay.BkgPrefix+"SCT_Clusters")
243  kwargs.setdefault("ExtraInputs", {('InDetDD::SiDetectorElementCollection', 'ConditionStore+SCT_DetectorElementCollection')})
244 
245  alg = CompFactory.CopySCT_ClusterContainer("CopySCT_ClusterContainer", **kwargs)
246  acc.addEventAlgo(alg)
247 
248  return acc
249 

◆ CopySCT_ClusterContainerCfg()

def python.OverlayCopyAlgsConfig.CopySCT_ClusterContainerCfg (   flags,
**  kwargs 
)
Return overlay configuration for the CopySCT_ClusterContainer algorithm

Definition at line 433 of file OverlayCopyAlgsConfig.py.

433 def CopySCT_ClusterContainerCfg(flags, **kwargs):
434  """Return overlay configuration for the CopySCT_ClusterContainer algorithm"""
435 
436  acc = CopySCT_ClusterContainerAlgCfg(flags, **kwargs)
437  if flags.Output.doWriteRDO:
438  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
439  acc.merge(OutputStreamCfg(flags, "RDO", ItemList=[
440  "InDet::SCT_ClusterContainer#Bkg_SCT_Clusters"
441  ]))
442 
443  return acc
444 

◆ CopyTrackCollectionAlgCfg()

def python.OverlayCopyAlgsConfig.CopyTrackCollectionAlgCfg (   flags,
  collectionName,
**  kwargs 
)
Return a ComponentAccumulator for the TrackCollection copying

Definition at line 277 of file OverlayCopyAlgsConfig.py.

277 def CopyTrackCollectionAlgCfg(flags, collectionName, **kwargs):
278  """Return a ComponentAccumulator for the TrackCollection copying"""
279  acc = ComponentAccumulator()
280 
281  kwargs.setdefault("OutputKey", flags.Overlay.BkgPrefix + collectionName)
282  kwargs.setdefault("InputKey", collectionName)
283  extra_inputs = set()
284  if flags.Detector.GeometryPixel:
285  extra_inputs.add(('InDetDD::SiDetectorElementCollection', 'ConditionStore+PixelDetectorElementCollection'))
286  if flags.Detector.GeometrySCT:
287  extra_inputs.add(('InDetDD::SiDetectorElementCollection', 'ConditionStore+SCT_DetectorElementCollection'))
288  if flags.Detector.GeometryTRT:
289  extra_inputs.add(( 'InDetDD::TRT_DetElementContainer' , 'ConditionStore+TRT_DetElementContainer' ))
290  if flags.Detector.GeometryITkPixel:
291  extra_inputs.add(( 'InDetDD::SiDetectorElementCollection' , 'ConditionStore+ITkPixelDetectorElementCollection' ))
292  if flags.Detector.GeometryITkStrip:
293  extra_inputs.add(( 'InDetDD::SiDetectorElementCollection' , 'ConditionStore+ITkStripDetectorElementCollection' ))
294 
295  kwargs.setdefault("ExtraInputs", extra_inputs)
296 
297  alg = CompFactory.CopyTrackCollection("CopyTrackCollection"+collectionName, **kwargs)
298  acc.addEventAlgo(alg)
299 
300  if flags.Output.doWriteRDO:
301  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
302  acc.merge(OutputStreamCfg(flags, "RDO", ItemList=[
303  "TrackCollection#" + flags.Overlay.BkgPrefix+collectionName]))
304 
305  return acc
306 
307 

◆ CopyTrackCollectionsCfg()

def python.OverlayCopyAlgsConfig.CopyTrackCollectionsCfg (   flags,
**  kwargs 
)
Return overlay configuration for copying tracks

Definition at line 496 of file OverlayCopyAlgsConfig.py.

496 def CopyTrackCollectionsCfg(flags, **kwargs):
497  """ Return overlay configuration for copying tracks"""
498 
499  acc = ComponentAccumulator()
500 
501  allowedContainers = [
502  "CombinedInDetTracks",
503  "DisappearingTracks",
504  "ResolvedForwardTracks",
505  "ExtendedLargeD0Tracks"
506  ]
507 
508  availableContainers = []
509  # Detect the list of track collections
510  if flags.Common.ProductionStep == ProductionStep.FastChain:
511  availableContainers = allowedContainers
512  else:
513  hardScatterInputCollections = flags.Input.Collections
514  for container in allowedContainers:
515  if container in hardScatterInputCollections:
516  availableContainers.append(container)
517  for container in availableContainers:
518  acc.merge(CopyTrackCollectionAlgCfg(flags, container, **kwargs))
519 
520  return acc
521 

◆ CopyTrackRecordCollectionAlgCfg()

def python.OverlayCopyAlgsConfig.CopyTrackRecordCollectionAlgCfg (   flags,
  collectionName,
  name = "CopyTrackRecordCollection",
**  kwargs 
)
Return a ComponentAccumulator for the TrackRecordCollection copying

Definition at line 346 of file OverlayCopyAlgsConfig.py.

346 def CopyTrackRecordCollectionAlgCfg(flags, collectionName, name="CopyTrackRecordCollection", **kwargs):
347  """Return a ComponentAccumulator for the TrackRecordCollection copying"""
348  acc = ComponentAccumulator()
349 
350  kwargs.setdefault("collectionName", collectionName)
351  kwargs.setdefault("SignalInputKey", flags.Overlay.SigPrefix + collectionName)
352  kwargs.setdefault("OutputKey", collectionName)
353  kwargs.setdefault("ExtraInputs", flags.Overlay.ExtraInputs)
354 
355  if flags.Common.ProductionStep is not ProductionStep.FastChain:
356  from SGComps.SGInputLoaderConfig import SGInputLoaderCfg
357  acc.merge(SGInputLoaderCfg(flags, [f'TrackRecordCollection#{kwargs["SignalInputKey"]}']))
358 
359  # Re-map signal address
360  from SGComps.AddressRemappingConfig import AddressRemappingCfg
361  acc.merge(AddressRemappingCfg([
362  f"TrackRecordCollection#{collectionName}->{flags.Overlay.SigPrefix}{collectionName}"
363  ]))
364 
365  # Copy TrackRecordCollection
366  acc.addEventAlgo(CompFactory.CopyTrackRecordCollection(name + collectionName, **kwargs))
367 
368  # Output
369  if flags.Output.doWriteRDO:
370  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
371  acc.merge(OutputStreamCfg(flags, "RDO", ItemList=[
372  f"TrackRecordCollection#{collectionName}"
373  ]))
374 
375  # Add signal output
376  if flags.Output.doWriteRDO_SGNL:
377  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
378  acc.merge(OutputStreamCfg(flags, "RDO_SGNL", ItemList=[
379  f"TrackRecordCollection#{flags.Overlay.SigPrefix}{collectionName}"
380  ]))
381 
382  return acc
383 
384 

◆ CopyTrackRecordCollectionsCfg()

def python.OverlayCopyAlgsConfig.CopyTrackRecordCollectionsCfg (   flags,
**  kwargs 
)
Return overlay configuration for the TrackRecordCollection algorithms

Definition at line 469 of file OverlayCopyAlgsConfig.py.

469 def CopyTrackRecordCollectionsCfg(flags, **kwargs):
470  """Return overlay configuration for the TrackRecordCollection algorithms"""
471 
472  acc = ComponentAccumulator()
473 
474  allowedContainers = [
475  "MuonExitLayer",
476  "MuonEntryLayer",
477  "CaloEntryLayer"
478  ]
479  availableContainers = []
480 
481  # Detect the list of track record collections
482  if flags.Common.ProductionStep == ProductionStep.FastChain:
483  availableContainers = allowedContainers
484  else:
485  hardScatterInputCollections = flags.Input.Collections if flags.Overlay.ByteStream else flags.Input.SecondaryCollections
486  for container in allowedContainers:
487  if container in hardScatterInputCollections:
488  availableContainers.append(container)
489 
490  for container in availableContainers:
491  acc.merge(CopyTrackRecordCollectionAlgCfg(flags, container, **kwargs))
492 
493  return acc
494 
495 

◆ CopyTRT_DriftCircleContainerAlgCfg()

def python.OverlayCopyAlgsConfig.CopyTRT_DriftCircleContainerAlgCfg (   flags,
**  kwargs 
)
Return a ComponentAccumulator for the CopyTRT_DriftCircleContainer algorithm

Definition at line 263 of file OverlayCopyAlgsConfig.py.

263 def CopyTRT_DriftCircleContainerAlgCfg(flags, **kwargs):
264  """Return a ComponentAccumulator for the CopyTRT_DriftCircleContainer algorithm"""
265  acc = ComponentAccumulator()
266 
267  kwargs.setdefault("InputKey", "TRT_DriftCircles")
268  kwargs.setdefault("OutputKey", flags.Overlay.BkgPrefix+"TRT_DriftCircles")
269  kwargs.setdefault("ExtraInputs", {('InDetDD::TRT_DetElementContainer' , 'ConditionStore+TRT_DetElementContainer')})
270 
271  alg = CompFactory.CopyTRT_DriftCircleContainer("CopyTRT_DriftCircleContainer", **kwargs)
272  acc.addEventAlgo(alg)
273 
274  return acc
275 
276 

◆ CopyTRT_DriftCircleContainerCfg()

def python.OverlayCopyAlgsConfig.CopyTRT_DriftCircleContainerCfg (   flags,
**  kwargs 
)
Return overlay configuration for the CopyTRT_DriftCircleContainer algorithm

Definition at line 457 of file OverlayCopyAlgsConfig.py.

457 def CopyTRT_DriftCircleContainerCfg(flags, **kwargs):
458  """Return overlay configuration for the CopyTRT_DriftCircleContainer algorithm"""
459 
460  acc = CopyTRT_DriftCircleContainerAlgCfg(flags, **kwargs)
461  if flags.Output.doWriteRDO:
462  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
463  acc.merge(OutputStreamCfg(flags, "RDO", ItemList=[
464  "InDet::TRT_DriftCircleContainer#Bkg_TRT_DriftCircles"]))
465 
466  return acc
467 
468 
AddressRemappingConfig.AddressRemappingCfg
def AddressRemappingCfg(renameMaps=[], overwriteMaps=[])
Definition: AddressRemappingConfig.py:10
python.JetAnalysisCommon.ComponentAccumulator
ComponentAccumulator
Definition: JetAnalysisCommon.py:302
python.OutputStreamConfig.OutputStreamCfg
def OutputStreamCfg(flags, streamName, ItemList=[], MetadataItemList=[], disableEventTag=False, trigNavThinningSvc=None, takeItemsFromInput=False, extendProvenanceRecord=True, AcceptAlgs=[], HelperTools=[])
Definition: OutputStreamConfig.py:12
python.OverlayCopyAlgsConfig.CopyMcEventCollectionCfg
def CopyMcEventCollectionCfg(flags, name="CopyMcEventCollection", **kwargs)
Definition: OverlayCopyAlgsConfig.py:161
python.OverlayCopyAlgsConfig.CopyPixelClusterContainerCfg
def CopyPixelClusterContainerCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:410
python.OverlayCopyAlgsConfig.CopyITkPixelClusterContainerCfg
def CopyITkPixelClusterContainerCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:422
python.OverlayCopyAlgsConfig.CopyTrackCollectionAlgCfg
def CopyTrackCollectionAlgCfg(flags, collectionName, **kwargs)
Definition: OverlayCopyAlgsConfig.py:277
python.OverlayCopyAlgsConfig.CopyDetailedTrackTruthCollectionAlgCfg
def CopyDetailedTrackTruthCollectionAlgCfg(flags, collectionName, **kwargs)
Definition: OverlayCopyAlgsConfig.py:308
python.OverlayCopyAlgsConfig.CopyCaloCalibrationHitContainersCfg
def CopyCaloCalibrationHitContainersCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:385
python.OverlayCopyAlgsConfig.CopyOutOfTimeAntiKt4JetTruthInfoCfg
def CopyOutOfTimeAntiKt4JetTruthInfoCfg(flags, name="CopyOutOfTimeAntiKt4JetTruthInfo", **kwargs)
Definition: OverlayCopyAlgsConfig.py:80
python.OverlayCopyAlgsConfig.CopyPileupParticleTruthInfoCfg
def CopyPileupParticleTruthInfoCfg(flags, name="CopyPileupParticleTruthInfo", **kwargs)
Definition: OverlayCopyAlgsConfig.py:133
python.OverlayCopyAlgsConfig.CopyPRD_MultiTruthCollectionsCfg
def CopyPRD_MultiTruthCollectionsCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:574
python.OverlayCopyAlgsConfig.CopySCT_ClusterContainerCfg
def CopySCT_ClusterContainerCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:433
python.OverlayCopyAlgsConfig.CopyITkTrackCollectionsCfg
def CopyITkTrackCollectionsCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:522
python.OverlayCopyAlgsConfig.CopyInTimeAntiKt6JetTruthInfoCfg
def CopyInTimeAntiKt6JetTruthInfoCfg(flags, name="CopyInTimeAntiKt6JetTruthInfo", **kwargs)
Definition: OverlayCopyAlgsConfig.py:88
python.OverlayCopyAlgsConfig.CopyDetailedTrackTruthCollectionsCfg
def CopyDetailedTrackTruthCollectionsCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:545
python.OverlayCopyAlgsConfig.CopyInTimeAntiKt4JetTruthInfoCfg
def CopyInTimeAntiKt4JetTruthInfoCfg(flags, name="CopyInTimeAntiKt4JetTruthInfo", **kwargs)
Definition: OverlayCopyAlgsConfig.py:72
python.OverlayCopyAlgsConfig.CopyTrackCollectionsCfg
def CopyTrackCollectionsCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:496
python.OverlayCopyAlgsConfig.CopyTRT_DriftCircleContainerCfg
def CopyTRT_DriftCircleContainerCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:457
python.OverlayCopyAlgsConfig.CopyTrackRecordCollectionsCfg
def CopyTrackRecordCollectionsCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:469
SGInputLoaderConfig.SGInputLoaderCfg
def SGInputLoaderCfg(flags, Load=None, **kwargs)
Definition: SGInputLoaderConfig.py:7
python.OverlayCopyAlgsConfig.CopyITkStripClusterContainerCfg
def CopyITkStripClusterContainerCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:445
python.OverlayCopyAlgsConfig.CopyPixelClusterContainerAlgCfg
def CopyPixelClusterContainerAlgCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:211
CxxUtils::set
constexpr std::enable_if_t< is_bitmask_v< E >, E & > set(E &lhs, E rhs)
Convenience function to set bits in a class enum bitmask.
Definition: bitmask.h:232
python.OverlayCopyAlgsConfig.CopyTRT_DriftCircleContainerAlgCfg
def CopyTRT_DriftCircleContainerAlgCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:263
python.OverlayCopyAlgsConfig.CopyITkStripClusterContainerAlgCfg
def CopyITkStripClusterContainerAlgCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:250
python.OverlayCopyAlgsConfig.CopyCaloCalibrationHitContainerAlgCfg
def CopyCaloCalibrationHitContainerAlgCfg(flags, collectionName, name="CopyCaloCalibrationHitContainer", **kwargs)
Definition: OverlayCopyAlgsConfig.py:11
python.OverlayCopyAlgsConfig.CopyPRD_MultiTruthCollectionAlgCfg
def CopyPRD_MultiTruthCollectionAlgCfg(flags, collectionName, **kwargs)
Definition: OverlayCopyAlgsConfig.py:327
python.OverlayCopyAlgsConfig.CopyITkPixelClusterContainerAlgCfg
def CopyITkPixelClusterContainerAlgCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:224
python.OverlayCopyAlgsConfig.CopyTrackRecordCollectionAlgCfg
def CopyTrackRecordCollectionAlgCfg(flags, collectionName, name="CopyTrackRecordCollection", **kwargs)
Definition: OverlayCopyAlgsConfig.py:346
python.OverlayCopyAlgsConfig.CopyJetTruthInfoCfg
def CopyJetTruthInfoCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:104
python.OverlayCopyAlgsConfig.CopyJetTruthInfoAlgCfg
def CopyJetTruthInfoAlgCfg(flags, name, **kwargs)
Definition: OverlayCopyAlgsConfig.py:48
python.OverlayCopyAlgsConfig.CopySCT_ClusterContainerAlgCfg
def CopySCT_ClusterContainerAlgCfg(flags, **kwargs)
Definition: OverlayCopyAlgsConfig.py:237
python.OverlayCopyAlgsConfig.CopyOutOfTimeAntiKt6JetTruthInfoCfg
def CopyOutOfTimeAntiKt6JetTruthInfoCfg(flags, name="CopyOutOfTimeAntiKt6JetTruthInfo", **kwargs)
Definition: OverlayCopyAlgsConfig.py:96