EGammaToolsConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
 
# ==============================================================================
# Provides configs for the tools used for e-gamma decorations used in DAOD
# building
# ==============================================================================
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
 
 
# PhotonsDirectionTool
def PhotonsDirectionToolCfg(flags, name, **kwargs):
    """Configure the PhotonsDirectionTool"""
    acc = ComponentAccumulator()
    PhotonsDirectionTool = CompFactory.DerivationFramework.PhotonsDirectionTool
    acc.addPublicTool(PhotonsDirectionTool(name, **kwargs), primary=True)
    return acc
 
 
# E-gamma selection tool wrapper
def EGSelectionToolWrapperCfg(flags, name, **kwargs):
    """Configure the E-gamma selection tool wrapper"""
    acc = ComponentAccumulator()
    EGSelectionToolWrapper = CompFactory.DerivationFramework.EGSelectionToolWrapper
    acc.addPublicTool(EGSelectionToolWrapper(name, **kwargs), primary=True)
    return acc
 
 
# Electron likelihood tool wrapper
def EGElectronLikelihoodToolWrapperCfg(flags, name, **kwargs):
    """Configure the electron likelihood tool wrapper"""
    acc = ComponentAccumulator()
    EGElectronLikelihoodToolWrapper = (
        CompFactory.DerivationFramework.EGElectronLikelihoodToolWrapper
    )
    acc.addPublicTool(EGElectronLikelihoodToolWrapper(name, **kwargs), primary=True)
    return acc
 
 
# Photon cleaning tool wrapper
def EGPhotonCleaningWrapperCfg(flags, name, **kwargs):
    """Configure the photon cleaning tool wrapper"""
    acc = ComponentAccumulator()
    EGPhotonCleaningWrapper = CompFactory.DerivationFramework.EGPhotonCleaningWrapper
    acc.addPublicTool(EGPhotonCleaningWrapper(name, **kwargs), primary=True)
    return acc
 
 
# Electron ambiguity tool
def EGElectronAmbiguityToolCfg(flags, name, **kwargs):
    """Configure the electron ambiguity tool"""
    acc = ComponentAccumulator()
    EGElectronAmbiguityTool = CompFactory.DerivationFramework.EGElectronAmbiguityTool
    acc.addPublicTool(EGElectronAmbiguityTool(name, **kwargs), primary=True)
    return acc
 
 
# Background electron classification tool
def BkgElectronClassificationCfg(flags, name, **kwargs):
    """Configure the background electron classification tool"""
    acc = ComponentAccumulator()
    BkgElectronClassification = (
        CompFactory.DerivationFramework.BkgElectronClassification
    )
    acc.addPublicTool(BkgElectronClassification(name, **kwargs), primary=True)
    return acc
 
 
# Standard + LRT electron collection merger
def ElectronMergerCfg(flags, name, **kwargs):
    """Configure the track particle merger tool"""
    acc = ComponentAccumulator()
    ElectronMerger = CompFactory.DerivationFramework.ElectronMergerTool
    acc.addPublicTool(ElectronMerger(name, **kwargs), primary=True)
    return acc
 
 
def PhotonVertexSelectionWrapperCfg(
        flags, name="PhotonVertexSelectionWrapper", **kwargs):
    acc = ComponentAccumulator()
 
    if "PhotonPointingTool" not in kwargs:
        from PhotonVertexSelection.PhotonVertexSelectionConfig import (
            PhotonPointingToolCfg)
        kwargs.setdefault("PhotonPointingTool", acc.popToolsAndMerge(
            PhotonPointingToolCfg(flags)))
 
    acc.setPrivateTools(
        CompFactory.DerivationFramework.PhotonVertexSelectionWrapper(
            name, **kwargs))
    return acc
 
 
def PhotonVertexSelectionWrapperKernelCfg(
        flags, name="PhotonVertexSelectionWrapperKernel", **kwargs):
    acc = ComponentAccumulator()
 
    augmentationTools = [
        acc.addPublicTool(acc.popToolsAndMerge(PhotonVertexSelectionWrapperCfg(flags)))
    ]
    kwargs.setdefault("AugmentationTools", augmentationTools)
 
    acc.addEventAlgo(
        CompFactory.DerivationFramework.DerivationKernel(name, **kwargs))
    return acc
 
def EGammaCookieCutClusterToolCfg(flags, name = 'EGCookieCutTool', **kwargs):
    acc = ComponentAccumulator()
    # needed for reading cells, do not rely on other config to do that
    from LArGeoAlgsNV.LArGMConfig import LArGMCfg
    acc.merge(LArGMCfg(flags))
    from TileGeoModel.TileGMConfig import TileGMCfg
    acc.merge(TileGMCfg(flags))
    #
    kwargs.setdefault('StoreCookedMoments',False)
    kwargs.setdefault('StoreInputMoments',False)
    kwargs.setdefault("SGKey_electrons", flags.Egamma.Keys.Output.ForwardElectrons)
 
    from CaloTools.CaloNoiseCondAlgConfig import CaloNoiseCondAlgCfg
    acc.merge(CaloNoiseCondAlgCfg(flags,"totalNoise"))
    from CaloRec.CaloTopoClusterConfig import getTopoMoments
    momentsMaker = acc.popToolsAndMerge(getTopoMoments(flags))
    kwargs.setdefault("ClusterMomentMaker",[momentsMaker])
 
    acc.setPrivateTools(
        CompFactory.DerivationFramework.EGammaCookieCutClusterTool(
            name, **kwargs))
    return acc