ATLAS Offline Software
Functions | Variables
BeamEffectsAlgConfig Namespace Reference

Functions

def ValidityCheckerCfg (flags, name="GenEventValidityChecker", **kwargs)
 
def GenEventRotatorCfg (flags, name="GenEventRotator", **kwargs)
 
def GenEventBeamEffectBoosterCfg (flags, name="GenEventBeamEffectBooster", **kwargs)
 
def GenEventVertexPositionerCfg (flags, name="GenEventVertexPositioner", **kwargs)
 
def VertexBeamCondPositionerCfg (flags, name="VertexBeamCondPositioner", **kwargs)
 
def VertexPositionFromFileCfg (flags, name="VertexPositionFromFile", **kwargs)
 
def CrabKissingVertexPositionerCfg (flags, name="CrabKissingVertexPositioner", **kwargs)
 
def LongBeamspotVertexPositionerCfg (flags, name="LongBeamspotVertexPositioner", **kwargs)
 
def BeamEffectsAlgCfg (flags, name="BeamEffectsAlg", **kwargs)
 
def BeamEffectsAlgOutputCfg (flags, **kwargs)
 
def BeamSpotFixerAlgCfg (flags, name="BeamSpotFixerAlg", **kwargs)
 
def ZeroLifetimePositionerCfg (flags, name="ZeroLifetimePositioner", **kwargs)
 
def BeamSpotReweightingAlgCfg (flags, name="BeamSpotReweightingAlg", **kwargs)
 

Variables

 flags = initConfigFlags()
 
 inputDir
 
 Files
 
 HITSFileName
 
 VertexSource
 
 isMC
 
 AtlasVersion
 
 GlobalTag
 
 RunNumbers
 
 cfg = MainServicesCfg(flags)
 
 Dump
 
 withDetails
 
 maxEvents
 

Function Documentation

◆ BeamEffectsAlgCfg()

def BeamEffectsAlgConfig.BeamEffectsAlgCfg (   flags,
  name = "BeamEffectsAlg",
**  kwargs 
)
Return an accumulator and algorithm for beam effects, wihout output

Definition at line 101 of file BeamEffectsAlgConfig.py.

101 def BeamEffectsAlgCfg(flags, name="BeamEffectsAlg", **kwargs):
102  """Return an accumulator and algorithm for beam effects, wihout output"""
103  acc = ComponentAccumulator()
104 
105  from SimulationConfig.SimEnums import LArParameterization
106  if flags.Sim.LArParameterization is LArParameterization.FastCaloSim:
107  kwargs.setdefault("ISFRun", True)
108  else:
109  kwargs.setdefault("ISFRun", flags.Sim.ISFRun)
110 
111  # Set default properties
112  if flags.Sim.DoFullChain and flags.Digitization.PileUp:
113  kwargs.setdefault("InputMcEventCollection", "OriginalEvent_SG+GEN_EVENT")
114  else:
115  kwargs.setdefault("InputMcEventCollection", "GEN_EVENT")
116 
117  if flags.Common.isOverlay and flags.Sim.DoFullChain:
118  kwargs.setdefault('OutputMcEventCollection', f"{flags.Overlay.SigPrefix}TruthEvent")
119  else:
120  kwargs.setdefault('OutputMcEventCollection', 'BeamTruthEvent')
121 
122  # Set (todo) the appropriate manipulator tools
123  manipulators = []
124  manipulators.append(acc.popToolsAndMerge(ValidityCheckerCfg(flags)))
125  from SimulationConfig.SimEnums import VertexSource
126  if not flags.Sim.VertexSource == VertexSource.AsGenerated:
127  # Vertex manipulation required
128  from SimulationConfig.SimEnums import CavernBackground
129  if flags.Beam.Type not in [BeamType.Cosmics, BeamType.TestBeam] and flags.Sim.CavernBackground is not CavernBackground.Read:
130  manipulators.append(acc.popToolsAndMerge(GenEventVertexPositionerCfg(flags)))
131  # manipulators.append(acc.popToolsAndMerge(GenEventBeamEffectBoosterCfg(flags))) # todo segmentation violation
132  # manipulators.append(acc.popToolsAndMerge(VertexPositionFromFileCfg(flags))) # todo
133  # manipulators.append(acc.popToolsAndMerge(CrabKissingVertexPositionerCfg(flags))) # todo Callback registration failed
134  # manipulators.append(acc.popToolsAndMerge(LongBeamspotVertexPositionerCfg(flags))) # todo Callback registration failed
135  kwargs.setdefault("GenEventManipulators", manipulators)
136 
137  acc.addEventAlgo(CompFactory.Simulation.BeamEffectsAlg(name, **kwargs), primary=True)
138  return acc
139 
140 

◆ BeamEffectsAlgOutputCfg()

def BeamEffectsAlgConfig.BeamEffectsAlgOutputCfg (   flags,
**  kwargs 
)
Return an accumulator and algorithm for beam effects, with output

Definition at line 141 of file BeamEffectsAlgConfig.py.

141 def BeamEffectsAlgOutputCfg(flags, **kwargs):
142  """Return an accumulator and algorithm for beam effects, with output"""
143  acc = BeamEffectsAlgCfg(flags, **kwargs)
144  # Set to write HITS pool file
145  alg = acc.getPrimary()
146  ItemList = ["McEventCollection#" + alg.OutputMcEventCollection]
147  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
148  acc.merge(OutputStreamCfg(flags, "HITS", ItemList=ItemList, disableEventTag=True))
149  return acc
150 
151 

◆ BeamSpotFixerAlgCfg()

def BeamEffectsAlgConfig.BeamSpotFixerAlgCfg (   flags,
  name = "BeamSpotFixerAlg",
**  kwargs 
)

Definition at line 152 of file BeamEffectsAlgConfig.py.

152 def BeamSpotFixerAlgCfg(flags, name="BeamSpotFixerAlg", **kwargs):
153  from BeamSpotConditions.BeamSpotConditionsConfig import BeamSpotCondAlgCfg
154  acc = BeamSpotCondAlgCfg(flags)
155 
156  kwargs.setdefault("InputKey", "Input_EventInfo")
157  if flags.Common.ProductionStep == ProductionStep.PileUpPresampling:
158  kwargs.setdefault("OutputKey", flags.Overlay.BkgPrefix + "EventInfo")
159  else:
160  kwargs.setdefault("OutputKey", "EventInfo")
161 
162  acc.addEventAlgo(CompFactory.Simulation.BeamSpotFixerAlg(name, **kwargs))
163  return acc
164 
165 

◆ BeamSpotReweightingAlgCfg()

def BeamEffectsAlgConfig.BeamSpotReweightingAlgCfg (   flags,
  name = "BeamSpotReweightingAlg",
**  kwargs 
)

Definition at line 174 of file BeamEffectsAlgConfig.py.

174 def BeamSpotReweightingAlgCfg(flags, name="BeamSpotReweightingAlg", **kwargs):
175  from BeamSpotConditions.BeamSpotConditionsConfig import BeamSpotCondAlgCfg
176  acc = BeamSpotCondAlgCfg(flags)
177 
178  kwargs.setdefault("Input_beam_sigma_z", flags.Digitization.InputBeamSigmaZ)
179 
180  acc.addEventAlgo(CompFactory.Simulation.BeamSpotReweightingAlg(name, **kwargs))
181 
182  # Ignore dependencies
183  from AthenaConfiguration.MainServicesConfig import OutputUsageIgnoreCfg
184  acc.merge(OutputUsageIgnoreCfg(flags, name))
185 
186  return acc
187 
188 

◆ CrabKissingVertexPositionerCfg()

def BeamEffectsAlgConfig.CrabKissingVertexPositionerCfg (   flags,
  name = "CrabKissingVertexPositioner",
**  kwargs 
)
Return a Crab-Kissing vertex positioner tool

Definition at line 84 of file BeamEffectsAlgConfig.py.

84 def CrabKissingVertexPositionerCfg(flags, name="CrabKissingVertexPositioner", **kwargs):
85  """Return a Crab-Kissing vertex positioner tool"""
86  # todo needs BunchLength, RandomSvc, BunchShape
87  acc = ComponentAccumulator()
88  acc.setPrivateTools(CompFactory.Simulation.CrabKissingVertexPositioner(name, **kwargs))
89  return acc
90 
91 

◆ GenEventBeamEffectBoosterCfg()

def BeamEffectsAlgConfig.GenEventBeamEffectBoosterCfg (   flags,
  name = "GenEventBeamEffectBooster",
**  kwargs 
)
Return a lorentz booster tool

Definition at line 31 of file BeamEffectsAlgConfig.py.

31 def GenEventBeamEffectBoosterCfg(flags, name="GenEventBeamEffectBooster", **kwargs):
32  """Return a lorentz booster tool"""
33  # todo needs random seed, more?
34  acc = ComponentAccumulator()
35  acc.setPrivateTools(CompFactory.Simulation.GenEventBeamEffectBooster(name, **kwargs))
36  return acc
37 
38 

◆ GenEventRotatorCfg()

def BeamEffectsAlgConfig.GenEventRotatorCfg (   flags,
  name = "GenEventRotator",
**  kwargs 
)
Return a event rotator tool

Definition at line 24 of file BeamEffectsAlgConfig.py.

24 def GenEventRotatorCfg(flags, name="GenEventRotator", **kwargs):
25  """Return a event rotator tool"""
26  acc = ComponentAccumulator()
27  acc.setPrivateTools(CompFactory.Simulation.GenEventRotator(name, **kwargs))
28  return acc
29 
30 

◆ GenEventVertexPositionerCfg()

def BeamEffectsAlgConfig.GenEventVertexPositionerCfg (   flags,
  name = "GenEventVertexPositioner",
**  kwargs 
)
Return a vertex positioner tool

Definition at line 39 of file BeamEffectsAlgConfig.py.

39 def GenEventVertexPositionerCfg(flags, name="GenEventVertexPositioner", **kwargs):
40  """Return a vertex positioner tool"""
41  # todo needs input file(s?)
42 
43  acc = ComponentAccumulator()
44 
45  from SimulationConfig.SimEnums import VertexSource
46  readVtxPosFromFile = flags.Sim.VertexSource in [VertexSource.VertexOverrideFile, VertexSource.VertexOverrideEventFile]
47  if readVtxPosFromFile:
48  kwargs.setdefault("VertexShifters", [acc.popToolsAndMerge(VertexPositionFromFileCfg(flags))])
49  elif flags.Sim.VertexSource is VertexSource.CondDB:
50  kwargs.setdefault("VertexShifters", [acc.popToolsAndMerge(VertexBeamCondPositionerCfg(flags))])
51  elif flags.Sim.VertexSource is VertexSource.LongBeamspotVertexPositioner:
52  kwargs.setdefault("VertexShifters", [acc.popToolsAndMerge(LongBeamspotVertexPositionerCfg(flags))])
53 
54  acc.setPrivateTools(CompFactory.Simulation.GenEventVertexPositioner(name, **kwargs))
55  return acc
56 
57 
58 # LorentzVectorGenerators

◆ LongBeamspotVertexPositionerCfg()

def BeamEffectsAlgConfig.LongBeamspotVertexPositionerCfg (   flags,
  name = "LongBeamspotVertexPositioner",
**  kwargs 
)
Return a long beamspot vertex positioner tool

Definition at line 92 of file BeamEffectsAlgConfig.py.

92 def LongBeamspotVertexPositionerCfg(flags, name="LongBeamspotVertexPositioner", **kwargs):
93  """Return a long beamspot vertex positioner tool"""
94  # todo needs LParameter and RandomSvc
95  acc = ComponentAccumulator()
96  kwargs.setdefault("SimpleTimeSmearing", flags.Sim.VertexTimeSmearing)
97  acc.setPrivateTools(CompFactory.Simulation.LongBeamspotVertexPositioner(name, **kwargs))
98  return acc
99 
100 

◆ ValidityCheckerCfg()

def BeamEffectsAlgConfig.ValidityCheckerCfg (   flags,
  name = "GenEventValidityChecker",
**  kwargs 
)
Return a validity checker tool

Definition at line 17 of file BeamEffectsAlgConfig.py.

17 def ValidityCheckerCfg(flags, name="GenEventValidityChecker", **kwargs):
18  """Return a validity checker tool"""
19  acc = ComponentAccumulator()
20  acc.setPrivateTools(CompFactory.Simulation.GenEventValidityChecker(name, **kwargs))
21  return acc
22 
23 

◆ VertexBeamCondPositionerCfg()

def BeamEffectsAlgConfig.VertexBeamCondPositionerCfg (   flags,
  name = "VertexBeamCondPositioner",
**  kwargs 
)
Return a conditional (? todo) vertex positioner tool

Definition at line 59 of file BeamEffectsAlgConfig.py.

59 def VertexBeamCondPositionerCfg(flags, name="VertexBeamCondPositioner", **kwargs):
60  """Return a conditional (? todo) vertex positioner tool"""
61  from RngComps.RngCompsConfig import AthRNGSvcCfg
62 
63  acc = ComponentAccumulator()
64 
65  kwargs.setdefault("RandomSvc", acc.getPrimaryAndMerge(AthRNGSvcCfg(flags)).name)
66  kwargs.setdefault("SimpleTimeSmearing", flags.Sim.VertexTimeSmearing)
67  kwargs.setdefault("TimeWidth", flags.Sim.VertexTimeWidth)
68 
69  from BeamSpotConditions.BeamSpotConditionsConfig import BeamSpotCondAlgCfg
70  acc.merge(BeamSpotCondAlgCfg(flags))
71 
72  acc.setPrivateTools(CompFactory.Simulation.VertexBeamCondPositioner(name, **kwargs))
73  return acc
74 
75 

◆ VertexPositionFromFileCfg()

def BeamEffectsAlgConfig.VertexPositionFromFileCfg (   flags,
  name = "VertexPositionFromFile",
**  kwargs 
)
Return a vertex positioner tool

Definition at line 76 of file BeamEffectsAlgConfig.py.

76 def VertexPositionFromFileCfg(flags, name="VertexPositionFromFile", **kwargs):
77  """Return a vertex positioner tool"""
78  # todo input file? look at cxx for details
79  acc = ComponentAccumulator()
80  acc.setPrivateTools(CompFactory.Simulation.VertexPositionFromFile(name, **kwargs))
81  return acc
82 
83 

◆ ZeroLifetimePositionerCfg()

def BeamEffectsAlgConfig.ZeroLifetimePositionerCfg (   flags,
  name = "ZeroLifetimePositioner",
**  kwargs 
)

Definition at line 166 of file BeamEffectsAlgConfig.py.

166 def ZeroLifetimePositionerCfg(flags, name="ZeroLifetimePositioner", **kwargs):
167  result = ComponentAccumulator()
168  kwargs.setdefault('ApplyPatch', True)
169  kwargs.setdefault('RemovePatch', True)
170  result.addService(CompFactory.Simulation.ZeroLifetimePositioner(name, **kwargs), primary = True)
171  return result
172 
173 

Variable Documentation

◆ AtlasVersion

BeamEffectsAlgConfig.AtlasVersion

Definition at line 222 of file BeamEffectsAlgConfig.py.

◆ cfg

BeamEffectsAlgConfig.cfg = MainServicesCfg(flags)

Definition at line 230 of file BeamEffectsAlgConfig.py.

◆ Dump

BeamEffectsAlgConfig.Dump

Definition at line 237 of file BeamEffectsAlgConfig.py.

◆ Files

BeamEffectsAlgConfig.Files

Definition at line 206 of file BeamEffectsAlgConfig.py.

◆ flags

BeamEffectsAlgConfig.flags = initConfigFlags()

Definition at line 197 of file BeamEffectsAlgConfig.py.

◆ GlobalTag

BeamEffectsAlgConfig.GlobalTag

Definition at line 223 of file BeamEffectsAlgConfig.py.

◆ HITSFileName

BeamEffectsAlgConfig.HITSFileName

Definition at line 212 of file BeamEffectsAlgConfig.py.

◆ inputDir

BeamEffectsAlgConfig.inputDir
Initial value:
1 = os.environ.get("ATLAS_REFERENCE_DATA",
2  "/cvmfs/atlas-nightlies.cern.ch/repo/data/data-art")

Definition at line 203 of file BeamEffectsAlgConfig.py.

◆ isMC

BeamEffectsAlgConfig.isMC

Definition at line 221 of file BeamEffectsAlgConfig.py.

◆ maxEvents

BeamEffectsAlgConfig.maxEvents

Definition at line 246 of file BeamEffectsAlgConfig.py.

◆ RunNumbers

BeamEffectsAlgConfig.RunNumbers

Definition at line 224 of file BeamEffectsAlgConfig.py.

◆ VertexSource

BeamEffectsAlgConfig.VertexSource

Definition at line 217 of file BeamEffectsAlgConfig.py.

◆ withDetails

BeamEffectsAlgConfig.withDetails

Definition at line 238 of file BeamEffectsAlgConfig.py.

BeamEffectsAlgConfig.ValidityCheckerCfg
def ValidityCheckerCfg(flags, name="GenEventValidityChecker", **kwargs)
Definition: BeamEffectsAlgConfig.py:17
python.JetAnalysisCommon.ComponentAccumulator
ComponentAccumulator
Definition: JetAnalysisCommon.py:302
BeamEffectsAlgConfig.LongBeamspotVertexPositionerCfg
def LongBeamspotVertexPositionerCfg(flags, name="LongBeamspotVertexPositioner", **kwargs)
Definition: BeamEffectsAlgConfig.py:92
python.OutputStreamConfig.OutputStreamCfg
def OutputStreamCfg(flags, streamName, ItemList=[], MetadataItemList=[], disableEventTag=False, trigNavThinningSvc=None, takeItemsFromInput=False, extendProvenanceRecord=True, AcceptAlgs=[], HelperTools=[])
Definition: OutputStreamConfig.py:12
python.BeamSpotConditionsConfig.BeamSpotCondAlgCfg
def BeamSpotCondAlgCfg(flags, name="BeamSpotCondAlg", **kwargs)
Definition: BeamSpotConditionsConfig.py:7
python.MainServicesConfig.OutputUsageIgnoreCfg
def OutputUsageIgnoreCfg(flags, algorithm)
Definition: MainServicesConfig.py:53
BeamEffectsAlgConfig.BeamEffectsAlgCfg
def BeamEffectsAlgCfg(flags, name="BeamEffectsAlg", **kwargs)
Definition: BeamEffectsAlgConfig.py:101
BeamEffectsAlgConfig.GenEventVertexPositionerCfg
def GenEventVertexPositionerCfg(flags, name="GenEventVertexPositioner", **kwargs)
Definition: BeamEffectsAlgConfig.py:39
BeamEffectsAlgConfig.CrabKissingVertexPositionerCfg
def CrabKissingVertexPositionerCfg(flags, name="CrabKissingVertexPositioner", **kwargs)
Definition: BeamEffectsAlgConfig.py:84
BeamEffectsAlgConfig.VertexPositionFromFileCfg
def VertexPositionFromFileCfg(flags, name="VertexPositionFromFile", **kwargs)
Definition: BeamEffectsAlgConfig.py:76
BeamEffectsAlgConfig.BeamSpotFixerAlgCfg
def BeamSpotFixerAlgCfg(flags, name="BeamSpotFixerAlg", **kwargs)
Definition: BeamEffectsAlgConfig.py:152
BeamEffectsAlgConfig.VertexBeamCondPositionerCfg
def VertexBeamCondPositionerCfg(flags, name="VertexBeamCondPositioner", **kwargs)
Definition: BeamEffectsAlgConfig.py:59
BeamEffectsAlgConfig.BeamEffectsAlgOutputCfg
def BeamEffectsAlgOutputCfg(flags, **kwargs)
Definition: BeamEffectsAlgConfig.py:141
BeamEffectsAlgConfig.BeamSpotReweightingAlgCfg
def BeamSpotReweightingAlgCfg(flags, name="BeamSpotReweightingAlg", **kwargs)
Definition: BeamEffectsAlgConfig.py:174
BeamEffectsAlgConfig.GenEventBeamEffectBoosterCfg
def GenEventBeamEffectBoosterCfg(flags, name="GenEventBeamEffectBooster", **kwargs)
Definition: BeamEffectsAlgConfig.py:31
BeamEffectsAlgConfig.GenEventRotatorCfg
def GenEventRotatorCfg(flags, name="GenEventRotator", **kwargs)
Definition: BeamEffectsAlgConfig.py:24
BeamEffectsAlgConfig.ZeroLifetimePositionerCfg
def ZeroLifetimePositionerCfg(flags, name="ZeroLifetimePositioner", **kwargs)
Definition: BeamEffectsAlgConfig.py:166
RngCompsConfig.AthRNGSvcCfg
def AthRNGSvcCfg(flags, name="AthRNGSvc")
Definition: RngCompsConfig.py:51