egammaSteeringConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 
__doc__ = """
          Instantiate the
          Things needed upstream the main egamma Reconstruction,
          EGamma Reconstruction,
          Output Item Lists,
          xAOD related Thinning
          """
 
from AthenaCommon.Logging import logging
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
 
 
def EGammaSteeringCfg(flags,
                      name="EGammaSteering"):
 
    mlog = logging.getLogger(name)
    mlog.info('Starting EGamma Steering')
 
    acc = ComponentAccumulator()
 
    # e/gamma main Reconstruction
    from egammaConfig.egammaReconstructionConfig import (
        egammaReconstructionCfg)
    acc.merge(egammaReconstructionCfg(flags))
 
    # Add e/gamma related containers to the output stream
    if flags.Output.doWriteESD or flags.Output.doWriteAOD:
        from egammaConfig.egammaOutputConfig import (
            egammaOutputCfg)
        acc.merge(egammaOutputCfg(flags))
 
    # LRT Reconstruction
    if flags.Tracking.doLargeD0:
        from egammaConfig.egammaLRTReconstructionConfig import (
            egammaLRTReconstructionCfg)
        acc.merge(egammaLRTReconstructionCfg(flags))
 
        # LRT output
        if flags.Output.doWriteESD or flags.Output.doWriteAOD:
            from egammaConfig.egammaLRTOutputConfig import (
                egammaLRTOutputCfg)
            acc.merge(egammaLRTOutputCfg(flags))
 
    # Add e/gamma xAOD thinning
    if flags.Output.doWriteAOD:
        from egammaConfig.egammaxAODThinningConfig import (
            egammaxAODThinningCfg)
        acc.merge(egammaxAODThinningCfg(flags))
 
    mlog.info("EGamma Steering done")
    return acc
 
# Run with python -m egammaConfig.egammaSteeringConfig
def egammaSteeringConfigTest(flags=None):
 
    if flags is None:
        from AthenaConfiguration.AllConfigFlags import initConfigFlags
        flags = initConfigFlags()
        flags.addFlag('egamma.configOnly', False, help='custom option for egammaSteeringConfig to not run, only output config file')
 
        from AthenaConfiguration.TestDefaults import defaultTestFiles, defaultConditionsTags
        flags.Input.Files = defaultTestFiles.RDO_RUN3
        flags.IOVDb.GlobalTag = defaultConditionsTags.RUN3_MC
        flags.Exec.MaxEvents = 1
 
        flags.Output.doWriteESD = True  # To test the ESD parts
        flags.Output.doWriteAOD = True  # To test the AOD parts
        flags.Output.ESDFileName = "myESD.pool.root"
        flags.Output.AODFileName = "myAOD.pool.root"
        
        from egammaConfig.ConfigurationHelpers import egammaOnlyFromRaw
        egammaOnlyFromRaw(flags)
        flags.fillFromArgs()
 
        flags.lock()
 
    from RecJobTransforms.RecoSteering import RecoSteering
    acc = RecoSteering(flags)
 
    # Special message service configuration
    from DigitizationConfig.DigitizationSteering import DigitizationMessageSvcCfg
    acc.merge(DigitizationMessageSvcCfg(flags))
 
    from AthenaConfiguration.Utils import setupLoggingLevels
    setupLoggingLevels(flags, acc)
 
    # Print reco domain status
    from RecJobTransforms.RecoConfigFlags import printRecoFlags
    printRecoFlags(flags)
 
    with open("egammasteeringconfig.pkl", "wb") as f:
        acc.store(f)
 
    if hasattr(flags, 'egamma') and flags.egamma.configOnly:
        return None  # returns statusCode of None
    else:
        statusCode = acc.run()
        return statusCode
 
if __name__ == "__main__":
    egammaSteeringConfigTest()