egammaCaloToolsConfig.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 AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
 
 
def CaloFillRectangularClusterCfg(flags, **kwargs):
    result = ComponentAccumulator()
 
    kwargs.setdefault("eta_size", 5)
    kwargs.setdefault("phi_size", 7)
    kwargs.setdefault("cells_name", flags.Egamma.Keys.Input.CaloCells)
 
    result.setPrivateTools(CompFactory.CaloFillRectangularCluster(**kwargs))
    return result
 
 
def egammaCheckEnergyDepositToolcfg(
    flags, name="egammaCheckEnergyDepositTool", **kwargs
):
    result = ComponentAccumulator()
    kwargs.setdefault("ThrE2min", 50)
    kwargs.setdefault("ThrF0max", 0.9)
    kwargs.setdefault("ThrF1max", 0.9)
    kwargs.setdefault("ThrF2max", 0.999)
    kwargs.setdefault("ThrF3max", 0.8)
    result.setPrivateTools(CompFactory.egammaCheckEnergyDepositTool(name, **kwargs))
    return result
 
 
def egammaCaloClusterSelectorGSFCfg(flags, name="caloClusterGSFSelector", **kwargs):
    result = ComponentAccumulator()
 
    if "egammaCheckEnergyDepositTool" not in kwargs:
        if not flags.Egamma.doLowMu:
            kwargs["egammaCheckEnergyDepositTool"] = result.popToolsAndMerge(
                egammaCheckEnergyDepositToolcfg(flags)
            )
 
    kwargs.setdefault("EMEtCut", 2250.0 if not flags.Egamma.doLowMu else 300.0)
    kwargs.setdefault("EMEtSplittingFraction", 0.7)
    kwargs.setdefault("EMFCut", 0.5)
    kwargs.setdefault("CellContainerName", flags.Egamma.Keys.Input.CaloCells)
    result.setPrivateTools(CompFactory.egammaCaloClusterSelector(name, **kwargs))
    return result
 
 
def egammaCaloClusterSelectorCfg(flags, name="caloClusterROISelector", **kwargs):
    result = ComponentAccumulator()
 
    if "egammaCheckEnergyDepositTool" not in kwargs:
        if not flags.Egamma.doLowMu:
            kwargs["egammaCheckEnergyDepositTool"] = result.popToolsAndMerge(
                egammaCheckEnergyDepositToolcfg(flags)
            )
    kwargs.setdefault("EMEtCut", 2250.0 if not flags.Egamma.doLowMu else 300.0)
    kwargs.setdefault("EMEtSplittingFraction", 0.7)
    kwargs.setdefault("EMFCut", 0.7)
    kwargs.setdefault("RetaCut", 0.65 if not flags.Egamma.doLowMu else 0.0)
    kwargs.setdefault("HadLeakCut", 0.15)
    result.setPrivateTools(CompFactory.egammaCaloClusterSelector(name, **kwargs))
    return result
 
 
def egammaHadCaloClusterSelectorCfg(
    flags, name="caloClusterHadROISelector", **kwargs
):
    result = ComponentAccumulator()
    kwargs.setdefault("egammaCheckEnergyDepositTool", "")
    kwargs.setdefault("ClusterEtCut", 150e3)
    result.setPrivateTools(CompFactory.egammaCaloClusterSelector(name, **kwargs))
    return result
 
def egammaLargeClusterCellRecoveryToolCfg(
    flags, name="egammaLargeClusterCellRecoveryTool", **kwargs
):
    result = ComponentAccumulator()
 
    # Configure to fill 7x11 cluster with cells
    toolArgs = dict(
        eta_size = 7,
        phi_size = 11,
        fill_cluster = True,
    )
 
    kwargs["CaloFillRectangularClusterTool"] = result.popToolsAndMerge(
        CaloFillRectangularClusterCfg(flags, **toolArgs)
    )
 
    result.setPrivateTools(
        CompFactory.egammaLargeClusterCellRecoveryTool(name, **kwargs)
    )
 
    return result
 
if __name__ == "__main__":
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    from AthenaConfiguration.ComponentAccumulator import printProperties
    from AthenaConfiguration.TestDefaults import defaultTestFiles
    from AthenaCommon.Logging import logging
    flags = initConfigFlags()
    flags.Input.Files = defaultTestFiles.RDO_RUN2
    flags.fillFromArgs()
    flags.lock()
    flags.dump()
 
    cfg = ComponentAccumulator()
    mlog = logging.getLogger("egammaCaloToolsConfigTest")
    mlog.info("Configuring egammaCaloClusterSelector : ")
    printProperties(
        mlog,
        cfg.popToolsAndMerge(egammaCaloClusterSelectorCfg(flags)),
        nestLevel=1,
        printDefaults=True,
    )
    mlog.info("Configuring egammaCaloClusterSelectorGSF :")
    printProperties(
        mlog,
        cfg.popToolsAndMerge(egammaCaloClusterSelectorGSFCfg(flags)),
        nestLevel=1,
        printDefaults=True,
    )
    mlog.info("Configuring egammaHadCaloClusterSelector :")
    printProperties(
        mlog,
        cfg.popToolsAndMerge(egammaHadCaloClusterSelectorCfg(flags)),
        nestLevel=1,
        printDefaults=True,
    )
    f = open("egtracktools.pkl", "wb")
    cfg.store(f)
    f.close()