egammaTrackToolsConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 
""" Tool configuration to instantiate all egammaCaloTools with default configuration"""
 
from AthenaCommon.Logging import logging
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from TrkConfig.AtlasExtrapolatorConfig import AtlasExtrapolatorCfg
from TrackToCalo.TrackToCaloConfig import EMParticleCaloExtensionToolCfg
 
 
def EMExtrapolationToolsCfg(flags, **kwargs):
    mlog = logging.getLogger("EMExtrapolationTools")
    mlog.debug("Start configuration")
 
    acc = ComponentAccumulator()
 
    if "Extrapolator" not in kwargs:
        extrapAcc = AtlasExtrapolatorCfg(flags)
        kwargs["Extrapolator"] = acc.popToolsAndMerge(extrapAcc)
 
    if "CaloExtensionTool" not in kwargs:
        kwargs["CaloExtensionTool"] = acc.popToolsAndMerge(
            EMParticleCaloExtensionToolCfg(flags)
        )
 
    kwargs["EnableTRT"] = flags.Detector.GeometryTRT
 
    emExtrapolationTools = CompFactory.EMExtrapolationTools(**kwargs)
    acc.setPrivateTools(emExtrapolationTools)
    return acc
 
 
def egammaTrkRefitterToolCfg(flags, name="GSFRefitterTool", **kwargs):
    acc = ComponentAccumulator()
    if "FitterTool" not in kwargs:
        if flags.Acts.GsfRefitLegacyTrk:
            from ActsConfig.ActsGaussianSumFitterConfig import (
                ActsTrkGaussianSumFitterToolCfg,
            )
 
            kwargs["FitterTool"] = acc.popToolsAndMerge(
                ActsTrkGaussianSumFitterToolCfg(flags, name="ActsGSFTrackFitter")
            )
        elif flags.Detector.GeometryITk:
            from TrkConfig.TrkGaussianSumFilterConfig import ITkGaussianSumFitterCfg
 
            kwargs["FitterTool"] = acc.popToolsAndMerge(
                ITkGaussianSumFitterCfg(flags, name="GSFTrackFitter")
            )
        else:
            from TrkConfig.TrkGaussianSumFilterConfig import GaussianSumFitterCfg
 
            kwargs["FitterTool"] = acc.popToolsAndMerge(
                GaussianSumFitterCfg(flags, name="GSFTrackFitter")
            )
 
    tool = CompFactory.egammaTrkRefitterTool(name, **kwargs)
    acc.setPrivateTools(tool)
    return acc
 
 
def CaloCluster_OnTrackBuilderCfg(flags, name="CaloCluster_OnTrackBuilder", **kwargs):
    acc = ComponentAccumulator()
    if "CaloSurfaceBuilder" not in kwargs:
        from CaloTrackingGeometry.CaloTrackingGeometryConfig import (
            CaloSurfaceBuilderMiddleCfg,
        )
 
        kwargs["CaloSurfaceBuilder"] = acc.popToolsAndMerge(
            CaloSurfaceBuilderMiddleCfg(flags)
        )
    tool = CompFactory.CaloCluster_OnTrackBuilder(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("egammaTrackToolsConfigTest")
    mlog.info("Configuring EMExtrapolationTools : ")
    printProperties(
        mlog,
        cfg.popToolsAndMerge(EMExtrapolationToolsCfg(flags)),
        nestLevel=1,
        printDefaults=True,
    )
    mlog.info("Configuring egammaTrkRefitterTool :")
    printProperties(
        mlog,
        cfg.popToolsAndMerge(egammaTrkRefitterToolCfg(flags)),
        nestLevel=1,
        printDefaults=True,
    )
    mlog.info("Configuring CaloCluster_OnTrackBuilder :")
    printProperties(
        mlog,
        cfg.popToolsAndMerge(CaloCluster_OnTrackBuilderCfg(flags)),
        nestLevel=1,
        printDefaults=True,
    )
 
    f = open("egtracktools.pkl", "wb")
    cfg.store(f)
    f.close()