EMShowerBuilderConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 
""" Configuration for EMShowerBuilder """
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
 
from AthenaCommon.Logging import logging
from CaloIdentifier import SUBCALO
 
 
def EMShowerBuilderCfg(flags, name='EMShowerBuilder', **kwargs):
 
    mlog = logging.getLogger(name)
    mlog.debug('Start configuration')
 
    acc = ComponentAccumulator()
    egammaIso, egammaShowerShape = CompFactory.getComps(
        "egammaIso", "egammaShowerShape",)
    kwargs.setdefault("CellsName", flags.Egamma.Keys.Input.CaloCells)
    kwargs.setdefault(
        "CaloNums", [SUBCALO.LAREM, SUBCALO.LARHEC, SUBCALO.TILE])
    kwargs.setdefault("ShowerShapeTool", egammaShowerShape())
    kwargs.setdefault("HadronicLeakageTool", egammaIso())
 
    tool = CompFactory.EMShowerBuilder(name, **kwargs)
 
    acc.setPrivateTools(tool)
    return acc
 
 
if __name__ == "__main__":
 
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    from AthenaConfiguration.ComponentAccumulator import printProperties
    from AthenaConfiguration.TestDefaults import defaultTestFiles
    flags = initConfigFlags()
    flags.Input.Files = defaultTestFiles.RDO_RUN2
    flags.fillFromArgs()
    flags.lock()
    flags.dump()
 
    cfg = ComponentAccumulator()
    mlog = logging.getLogger("EMShowerBuilderConfigTest")
    mlog.info("Configuring  EMShowerBuilder: ")
    printProperties(mlog, cfg.popToolsAndMerge(
        EMShowerBuilderCfg(flags)),
        nestLevel=1,
        printDefaults=True)
 
    f = open("emshowerbuilder.pkl", "wb")
    cfg.store(f)
    f.close()