def addOverlayArguments(parser):
from SimuJobTransforms.simTrfArgs import addBasicDigiArgs, addForwardDetTrfArgs
from EventOverlayJobTransforms.overlayTrfArgs import addOverlayTrfArgs, addOverlayPoolTrfArgs
addBasicDigiArgs(parser)
addForwardDetTrfArgs(parser)
addOverlayTrfArgs(parser)
addOverlayPoolTrfArgs(parser)
def getTransform(RAWtoALL=False):
executorSet = set()
addRecoSubsteps(executorSet)
trf = transform(
executor=executorSet,
description=
'General purpose ATLAS digitisation and reconstruction transform'
' Inputs can be HITS, RDO, BS, ESD or AOD, with outputs of RDO, ESD, AOD or DAODs.'
' See https://twiki.cern.ch/twiki/bin/view/AtlasComputing/RecoTf for more details.'
)
addAthenaArguments(trf.parser)
addDetectorArguments(trf.parser)
addTriggerArguments(trf.parser)
addAllRecoArgs(trf, RAWtoALL)
# For digi step - make sure we can add the digitisation/simulation arguments
# before we add this substep; allows Reco_tf to work without AtlasSimulation
try:
from SimuJobTransforms.simTrfArgs import addForwardDetTrfArgs, addBasicDigiArgs, addPileUpTrfArgs, addCommonSimDigTrfArgs
from SimuJobTransforms.SimTransformUtils import addDigitizationSubstep
addBasicDigiArgs(trf.parser)
addForwardDetTrfArgs(trf.parser)
addPileUpTrfArgs(trf.parser)
addCommonSimDigTrfArgs(trf.parser)
simStepSet = set()
addDigitizationSubstep(simStepSet)
trf.appendToExecutorSet(list(simStepSet)[0])
except ImportError as e:
msg.warning(
'Failed to import digitisation arguments ({0}). Digitisation substep will not be available.'
.format(e))
# Again, protect core functionality from too tight a dependence on EventOverlay
try:
from EventOverlayJobTransforms.overlayTrfArgs import addOverlayTrfArgs, addOverlayPoolTrfArgs
from EventOverlayJobTransforms.overlayTransformUtils import appendOverlay_PoolSubstep
addOverlayTrfArgs(trf.parser)
addOverlayPoolTrfArgs(trf.parser)
appendOverlay_PoolSubstep(trf, True)
except ImportError as e:
msg.warning(
'Failed to import overlay arguments ({0}). Event overlay substep will not be available.'
.format(e))
# Again, protect core functionality from too tight a dependence on PATJobTransforms
try:
from PATJobTransforms.PATTransformUtils import addPhysValidationFiles, addValidationArguments, appendPhysValidationSubstep
addPhysValidationFiles(trf.parser)
addValidationArguments(trf.parser)
appendPhysValidationSubstep(trf)
except ImportError:
msg.warning(
'Failed to import PAT arguments. Physics validation substep will not be available.'
)
return trf
from SimuJobTransforms.SimTransformUtils import addDigitizationSubstep
addBasicDigiArgs(trf.parser)
addForwardDetTrfArgs(trf.parser)
addPileUpTrfArgs(trf.parser)
addCommonSimDigTrfArgs(trf.parser)
simStepSet = set()
addDigitizationSubstep(simStepSet)
trf.appendToExecutorSet(list(simStepSet)[0])
except ImportError, e:
msg.warning('Failed to import digitisation arguments ({0}). Digitisation substep will not be available.'.format(e))
# Again, protect core functionality from too tight a dependence on EventOverlay
try:
from EventOverlayJobTransforms.overlayTrfArgs import addOverlayTrfArgs, addOverlayPoolTrfArgs
from EventOverlayJobTransforms.overlayTransformUtils import appendOverlay_PoolSubstep
addOverlayTrfArgs(trf.parser)
addOverlayPoolTrfArgs(trf.parser)
appendOverlay_PoolSubstep(trf, True)
except ImportError, e:
msg.warning('Failed to import overlay arguments ({0}). Event overlay substep will not be available.'.format(e))
# Again, protect core functionality from too tight a dependence on PATJobTransforms
try:
from PATJobTransforms.PATTransformUtils import addPhysValidationFiles, addValidationArguments, appendPhysValidationSubstep
addPhysValidationFiles(trf.parser)
addValidationArguments(trf.parser)
appendPhysValidationSubstep(trf)
except ImportError:
msg.warning('Failed to import PAT arguments. Physics validation substep will not be available.')
return trf
def addUniqueOverlay_BSArguments(parser):
from SimuJobTransforms.simTrfArgs import addBasicDigiArgs
from EventOverlayJobTransforms.overlayTrfArgs import addOverlayTrfArgs, addOverlayBSTrfArgs
addBasicDigiArgs(parser)
addOverlayTrfArgs(parser)
addOverlayBSTrfArgs(parser)