egammaxAODThinningConfig.py

Go to the documentation of this file.

# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 
""" Instantiate the Egamma related xAOD Thinning
"""
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
 
 
def egammaxAODThinningCfg(flags, name="EGammaxAODThinning"):
 
    acc = ComponentAccumulator()
    # Add e/gamma track thinning
    if flags.Egamma.doTrackThinning:
        from egammaAlgs.egammaTrackThinnerConfig import (
            egammaTrackThinnerCfg)
        acc.merge(egammaTrackThinnerCfg(flags))
 
    # keep cells for egamma and egammaLargeClusters
    if flags.Egamma.keepCaloCellsAOD:
        outFlags = flags.Egamma.Keys.Output
        allClusters = []
        if flags.Egamma.doCentral:
            allClusters.append(outFlags.CaloClusters)
            allClusters.append(outFlags.EgammaLargeClusters)
 
        if flags.Egamma.doForward:
            allClusters.append(outFlags.ForwardClusters)
            allClusters.append(outFlags.EgammaLargeFWDClusters)
 
        if flags.Tracking.doLargeD0:
            allClusters.append(f"LRT{outFlags.CaloClusters}")
 
        if flags.HeavyIon.Egamma.doSubtractedClusters:
            allClusters.append(flags.HeavyIon.Egamma.CaloTopoCluster)
 
        samplings = [
            "TileGap1",
            "TileGap2",
            "TileGap3",
            "TileBar0",
            "TileExt0",
            "HEC0",
        ]
        from CaloRec.CaloThinCellsByClusterAlgConfig import (
            CaloThinCellsByClusterAlgCfg)
 
        cellsName = flags.Egamma.Keys.Input.CaloCells
        for clus in allClusters:
            acc.merge(CaloThinCellsByClusterAlgCfg(
                flags,
                streamName="StreamAOD",
                clusters=clus,
                samplings=samplings,
                cells=cellsName
            ))
 
    return acc
 
 
if __name__ == "__main__":
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    from AthenaConfiguration.TestDefaults import defaultTestFiles
    from AthenaConfiguration.MainServicesConfig import MainServicesCfg
    flags = initConfigFlags()
    flags.Input.Files = defaultTestFiles.ESD
    flags.Output.doWriteAOD = True  # To test the AOD parts
    flags.lock()
    acc = MainServicesCfg(flags)
    acc.merge(egammaxAODThinningCfg(flags))
    acc.printConfig(withDetails=True,
                    printDefaults=True)
 
    with open("egammaxaodthinningconfig.pkl", "wb") as f:
        acc.store(f)