egammaSelectedTrackCopyConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 
__doc__ = "Instantiate egammaSelectedTrackCopy with default configuration"
 
from AthenaCommon.Logging import logging
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
 
 
def egammaSelectedTrackCopyCfg(flags, name="egammaSelectedTrackCopy", **kwargs):
    acc = ComponentAccumulator()
 
    if "egammaCaloClusterSelector" not in kwargs:
        from egammaCaloTools.egammaCaloToolsConfig import (
            egammaCaloClusterSelectorGSFCfg,
        )
 
        kwargs["egammaCaloClusterSelector"] = acc.popToolsAndMerge(
            egammaCaloClusterSelectorGSFCfg(flags)
        )
 
    if "ExtrapolationTool" not in kwargs:
        from egammaTrackTools.egammaTrackToolsConfig import (
            EMExtrapolationToolsCfg,
        )
 
        extraptool = EMExtrapolationToolsCfg(
            flags, name="EMExtrapolationTools")
        kwargs["ExtrapolationTool"] = acc.popToolsAndMerge(extraptool)
 
    kwargs.setdefault("ClusterContainerName",
                      flags.Egamma.Keys.Internal.EgammaTopoClusters)
    kwargs.setdefault("TrackParticleContainerName",
                      flags.Egamma.Keys.Input.TrackParticles)
    kwargs.setdefault("OutputTrkPartContainerName",
                      flags.Egamma.Keys.Output.TrkPartContainerName)
 
    if flags.Reco.EnableHGTDExtension:
        kwargs.setdefault("TrackParticleTimeDecoration",
                          flags.Egamma.Keys.Input.TrackParticles+".time")
 
    doFwd = flags.Detector.GeometryITk and flags.Egamma.doForward
    kwargs.setdefault("doFwdTracks", doFwd)
    if doFwd:
        kwargs.setdefault("FwdClusterContainerName",
                          flags.Egamma.Keys.Internal.ForwardTopoClusters)
 
    # P->T conversion extra dependencies
    if flags.Detector.GeometryITk:
        kwargs.setdefault(
            "ExtraInputs",
            [
                (
                    "InDetDD::SiDetectorElementCollection",
                    "ConditionStore+ITkPixelDetectorElementCollection",
                ),
                (
                    "InDetDD::SiDetectorElementCollection",
                    "ConditionStore+ITkStripDetectorElementCollection",
                ),
            ],
        )
    else:
        kwargs.setdefault(
            "ExtraInputs",
            [
                (
                    "InDetDD::SiDetectorElementCollection",
                    "ConditionStore+PixelDetectorElementCollection",
                ),
                (
                    "InDetDD::SiDetectorElementCollection",
                    "ConditionStore+SCT_DetectorElementCollection",
                ),
            ],
        )
 
    egseltrkcpAlg = CompFactory.egammaSelectedTrackCopy(name, **kwargs)
 
    acc.addEventAlgo(egseltrkcpAlg)
    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(egammaSelectedTrackCopyCfg(flags))
    mlog = logging.getLogger("egammaSelectedTrackCopyConfigTest")
    mlog.info("Configuring  egammaSelectedTrackCopy: ")
    printProperties(
        mlog,
        acc.getEventAlgo("egammaSelectedTrackCopy"),
        nestLevel=1,
        printDefaults=True,
    )
    with open("egammaselectedtrackCopy.pkl", "wb") as f:
        acc.store(f)