IsoDensityConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 
__doc__ = """
          Instantiate the electron, photon and muon isolation
          """
 
from AthenaCommon.Logging import logging
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
from AthenaConfiguration.ComponentFactory import CompFactory
 
def EMTopoInputAlgCfg(flags, name='EMTopoInputAlg', **kwargs):
 
    mlog = logging.getLogger(name)
    mlog.info('Starting input building for isolation density configuration')
 
    acc = ComponentAccumulator()
 
    from JetRecConfig.StandardJetConstits import stdConstitDic as cst
    from JetRecConfig.JetRecConfig import JetInputCfg, getConstitPJGAlg
 
    # Get the input (from CaloCal to EM scale, no origin correction)
    acc.merge(JetInputCfg(flags, cst.EMTopo))
 
    # Get the pseudo jets for this input
    acc.addEventAlgo(getConstitPJGAlg(cst.EMTopo))
 
    return acc
 
def NFlowInputAlgCfg(flags, name='NFlowInputAlg', **kwargs):
 
    mlog = logging.getLogger(name)
    mlog.info('Starting input building for neutral eflow isolation density configuration')
 
    acc = ComponentAccumulator()
 
    from JetRecConfig.StandardJetConstits import stdConstitDic as cst
    from JetRecConfig.JetRecConfig import JetInputCfg
 
    # Get the input (FlowObject, with CHS and origin (for neutral) and weight (for charged) correction)
    if 'InputType' not in kwargs:
        kwargs['InputType'] = 'EMPFlow'
    cstI = cst[kwargs['InputType']]
    acc.merge(JetInputCfg(flags, cstI))
 
    cstO = cstI.containername[0:cstI.containername.find('ParticleFlowObjects')]
    mlog.info('Configuring density for type '+cstO)
    cstOs = cstO if cstO != 'CHS' else ''
 
    # Then transform into pseudo-jets for the neutral only
    constitpjalg = CompFactory.PseudoJetAlgorithm(
        name = "PseudoJetAlgForIso"+cstOs+"NFlow",
        InputContainer = cstO+"NeutralParticleFlowObjects",
        OutputContainer = "PseudoJet"+cstOs+"NFlow",
        Label = "EM"+cstOs+"NPFlow",
        SkipNegativeEnergy=True)
 
    # Add the algs to the sequence in the ComponentAccumulator
    acc.addEventAlgo(constitpjalg)
 
    return acc
 
 
def DensityForIsoAlgCfg(flags, name = "CentralDensityForTopoIso", **kwargs):
 
    mlog = logging.getLogger(name)
    mlog.info('Starting density alg for isolation configuration')
 
    acc = ComponentAccumulator()
    
    # And then the density tool and algs. By default the central one
    if name.find('Topo') >= 0:
        inputO = 'PseudoJetEMTopoClusters'
        outputS = 'TopoCluster'
    elif name.find('NFlow') >= 0:
        suff = 'CSSK' if name.find('CSSK') >= 0 else ''
        inputO = 'PseudoJet'+suff+'NFlow'
        outputS = 'NeutralParticle'+suff+'Flow'
    kwargs['InputContainer'] = inputO
    kwargs['JetRadius'] = 0.5
    kwargs['UseFourMomArea'] = True
    kwargs['VoronoiRfact'] = 0.9
    kwargs['JetAlgorithm'] = 'Kt'
    if name.find('Central') >= 0:
        kwargs['OutputContainer'] = outputS+'IsoCentralEventShape'
        kwargs['AbsRapidityMin'] = 0.0
        kwargs['AbsRapidityMax'] = 1.5
    elif name.find('Forward') >= 0:
        kwargs['OutputContainer'] = outputS+'IsoForwardEventShape'
        kwargs['AbsRapidityMin'] = 1.5
        kwargs['AbsRapidityMax'] = 3.0
    densityTool = CompFactory.EventDensityTool(
        name = name+'Tool',**kwargs)
    
    densityAlg = CompFactory.EventDensityAthAlg(
        name = name+'Alg',
        EventDensityTool = densityTool)
    acc.addEventAlgo(densityAlg)
 
    return acc
 
 
if __name__ == "__main__":
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    from AthenaConfiguration.ComponentAccumulator import printProperties
    from AthenaConfiguration.TestDefaults import defaultTestFiles
    from AthenaConfiguration.MainServicesConfig import MainServicesCfg
    flags = initConfigFlags()
    flags.Input.Files = defaultTestFiles.RDO_RUN2
    flags.Output.doWriteESD = True  # To test the ESD parts
    flags.Output.doWriteAOD = True  # To test the AOD parts
    flags.lock()
 
    mlog = logging.getLogger("densityForIsolationConfigTest")
    mlog.info("Configuring density for isolation: ")
 
    acc = MainServicesCfg(flags)
    acc.merge(EMTopoInputAlgCfg(flags))
    acc.merge(DensityForIsoAlgCfg(flags))
    acc.merge(DensityForIsoAlgCfg(flags, name='ForwardDensityForIso'))
    acc.merge(NFlowInputAlgCfg(flags))
    acc.merge(DensityForIsoAlgCfg(flags,name='CentralDensityForNFlowIso'))
    printProperties(mlog,
                    acc.getEventAlgo('CentralDensityForTopoIsoAlg'),
                    nestLevel=1,
                    printDefaults=True)
    printProperties(mlog,
                    acc.getEventAlgo('JetModAlgForIsoNFlow'),
                    nestLevel=1,
                    printDefaults=True)
 
    with open("isolationdensityconfig.pkl", "wb") as f:
        acc.store(f)