EMVertexBuilderConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaCommon.Logging import logging
 
 
def EMVertexBuilderCfg(flags, name="EMVertexBuilder", **kwargs):
    acc = ComponentAccumulator()
 
    if "ExtrapolationTool" not in kwargs:
        from egammaTrackTools.egammaTrackToolsConfig import (
            EMExtrapolationToolsCfg)
        kwargs["ExtrapolationTool"] = acc.popToolsAndMerge(
            EMExtrapolationToolsCfg(flags))
    if "VertexFinderTool" not in kwargs:
        from InDetConfig.InDetConversionFinderToolsConfig import (
            InDetConversionFinderToolsCfg)
        kwargs["VertexFinderTool"] = acc.popToolsAndMerge(
            InDetConversionFinderToolsCfg(flags))
 
    alg = CompFactory.EMVertexBuilder(name, **kwargs)
    acc.addEventAlgo(alg)
    return acc
 
 
if __name__ == "__main__":
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    from AthenaConfiguration.TestDefaults import defaultTestFiles
    from AthenaConfiguration.ComponentAccumulator import printProperties
    from AthenaConfiguration.MainServicesConfig import MainServicesCfg
    flags = initConfigFlags()
    flags.Input.Files = defaultTestFiles.RDO_RUN2
    flags.lock()
 
    acc = MainServicesCfg(flags)
    acc.merge(EMVertexBuilderCfg(flags))
    mlog = logging.getLogger("EMVertexBuilderConfigTest")
    mlog.info("Configuring  EMVertexBuilder: ")
    printProperties(mlog,
                    acc.getEventAlgo("EMVertexBuilder"),
                    nestLevel=1,
                    printDefaults=True)
    with open("vertexbuilder.pkl", "wb") as f:
        acc.store(f)