IsoToolsConfig.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 isolationTools with default configuration
"""
 
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.Enums import BeamType
 
def TrackIsolationToolCfg(flags, **kwargs):
    acc = ComponentAccumulator()
        
    if 'TrackSelectionTool' not in kwargs:
        from InDetConfig.InDetTrackSelectionToolConfig import isoTrackSelectionToolCfg
        kwargs['TrackSelectionTool'] = acc.popToolsAndMerge(isoTrackSelectionToolCfg(flags))
    if 'TTVATool' not in kwargs:
        from TrackVertexAssociationTool.TrackVertexAssociationToolConfig import isoTTVAToolCfg
        kwargs['TTVATool'] = acc.popToolsAndMerge(isoTTVAToolCfg(flags))
    if flags.Beam.Type is BeamType.Cosmics:
        kwargs['VertexLocation'] = ''
 
    acc.setPrivateTools(CompFactory.xAOD.TrackIsolationTool(**kwargs))
    return acc
 
def CaloIsolationToolCfg(flags, **kwargs):
    acc = ComponentAccumulator()
        
    if 'IsoLeakCorrectionTool' not in kwargs:
        kwargs['IsoLeakCorrectionTool'] = CompFactory.CP.IsolationCorrectionTool(
            name = 'LeakageCorrTool')
 
    if 'ClustersInConeTool' not in kwargs:
        kwargs['ClustersInConeTool'] = CompFactory.xAOD.CaloClustersInConeTool(
            name = 'MyCaloClustersInConeTool',
            CaloClusterLocation = "CaloCalTopoClusters")
 
    if 'ParticleCaloExtensionTool' not in kwargs:
        from TrackToCalo.TrackToCaloConfig import ParticleCaloExtensionToolCfg
        kwargs['ParticleCaloExtensionTool'] = acc.popToolsAndMerge(
            ParticleCaloExtensionToolCfg(flags))
 
    kwargs.setdefault('name','CaloIsolationTool')
    kwargs.setdefault('ParticleCaloExtensionTool',None)
    kwargs.setdefault('ParticleCaloCellAssociationTool',None)
    kwargs.setdefault('isMC',flags.Input.isMC)
        
    acc.setPrivateTools(CompFactory.xAOD.CaloIsolationTool(**kwargs))
    return acc
 
def ElectronTrackIsolationToolCfg(flags, **kwargs):
    kwargs.setdefault('name','ElectronTrackIsolationTool')
    kwargs.setdefault('CoreTrackEtaRange',0.01)
    return TrackIsolationToolCfg(flags,**kwargs)
        
def EGammaCaloIsolationToolCfg(flags, **kwargs):
    acc = ComponentAccumulator()
 
    if 'IsoLeakCorrectionTool' not in kwargs:
        kwargs['IsoLeakCorrectionTool'] = CompFactory.CP.IsolationCorrectionTool(
            name = 'LeakageCorrTool',
            LogLogFitForLeakage = True)
 
    if 'CaloFillRectangularClusterTool' not in kwargs:
        kwargs['CaloFillRectangularClusterTool'] = CompFactory.CaloFillRectangularCluster(
            name="egamma_CaloFillRectangularCluster",
            eta_size=5,
            phi_size=7,
            cells_name=flags.Egamma.Keys.Input.CaloCells)
 
    # default is to read calocaltopoclusters.
    # In HI, if subtracted clusters, use them instead, and do not do pu correction
    if flags.HeavyIon.Egamma.doSubtractedClusters:
        kwargs['ClustersInConeTool'] = CompFactory.xAOD.CaloClustersInConeTool(
            name="topoiso_CaloClustersInConeTool",
            CaloClusterLocation=flags.Egamma.Keys.Input.TopoClusters)
        # No pileup correction, and do not save it
        kwargs['InitializeReadHandles'] = False
        kwargs['saveOnlyRequestedCorrections'] = True
        
    kwargs.setdefault('name','egCaloIsolationTool')
    kwargs.setdefault('ParticleCaloExtensionTool',None)
    kwargs.setdefault('ParticleCaloCellAssociationTool',None)
    kwargs.setdefault('isMC',flags.Input.isMC)
        
    acc.setPrivateTools(CompFactory.xAOD.CaloIsolationTool(**kwargs))
    return acc
 
def MuonCaloIsolationToolCfg(flags, **kwargs):
    acc = ComponentAccumulator()
 
    if 'ParticleCaloExtensionTool' not in kwargs:
        from TrackToCalo.TrackToCaloConfig import ParticleCaloExtensionToolCfg
        kwargs['ParticleCaloExtensionTool'] = acc.popToolsAndMerge(
            ParticleCaloExtensionToolCfg(flags))
    if 'FlowElementsInConeTool' not in kwargs and flags.Reco.EnablePFlow:
        kwargs['FlowElementsInConeTool'] = CompFactory.xAOD.FlowElementsInConeTool(
            name='FlowElementsInConeTool')
 
    # default is to read calocaltopoclusters.
    # In HI, if subtracted clusters, use them instead, and do not do pu correction
    if flags.HeavyIon.Egamma.doSubtractedClusters:
        ccict = CompFactory.xAOD.CaloClustersInConeTool(
            name="topoiso_CaloClustersInConeTool",
            CaloClusterLocation=flags.Egamma.Keys.Input.TopoClusters)
        kwargs['ClustersInConeTool'] = ccict
        # No pileup correction, and do not save it
        kwargs['InitializeReadHandles'] = False
        kwargs['saveOnlyRequestedCorrections'] = True
 
    kwargs.setdefault('ParticleCaloCellAssociationTool',None)
    kwargs.setdefault('UseEtaDepPUCorr',False)
    kwargs.setdefault('name','muonCaloIsolationTool')
 
    acc.setPrivateTools(CompFactory.xAOD.CaloIsolationTool(**kwargs))
    return acc