Skip to content
Snippets Groups Projects

Add digitizationRunArgsToFlags helper function

Merged Tadej Novak requested to merge tadej/athena:digi/newtrf into master
1 file
+ 21
0
Compare changes
  • Side-by-side
  • Inline
@@ -38,6 +38,7 @@ def getSpecialConfiguration(flags):
# include(inc)
return out
def createDigitizationCfgFlags():
"""Return an AthConfigFlags object with required flags"""
flags = AthConfigFlags()
@@ -71,3 +72,23 @@ def createDigitizationCfgFlags():
flags.addFlag("Digitization.ExtraInputs", [("xAOD::EventInfo", "EventInfo")])
return flags
def digitizationRunArgsToFlags(runArgs, flags):
"""Fill digitization configuration flags from run arguments."""
if hasattr(runArgs, "PileUpPremixing"):
flags.Digitization.PileUpPremixing = runArgs.PileUpPremixing
if hasattr(runArgs, "doAllNoise"):
flags.Digitization.DoInnerDetectorNoise = runArgs.doAllNoise
flags.Digitization.DoCaloNoise = runArgs.doAllNoise
if hasattr(runArgs, "AddCaloDigi"):
flags.Digitization.AddCaloDigi = runArgs.AddCaloDigi
if hasattr(runArgs, "digiSeedOffset1"):
flags.Digitization.RandomSeedOffset = runArgs.digiSeedOffset1
# TODO: Not covered yet as no flag equivalents exist yet
# '--digiRndmSvc'
# '--digiSteeringConf'
# '--samplingFractionDbTag'
Loading