TileSimD3PDConfig.py

Go to the documentation of this file.

#
#  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
#
 
from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
 
'''
@file TileSimD3PDConfig.py
@brief Python configuration of Tile D3PD for simulaiton for the Run III
'''
 
 
def TileSimD3PDCfg(flags, outputFile=None):
    ''' Function to configure Tile D3PD for simulaiton.'''
 
    acc = ComponentAccumulator()
 
    from TileGeoModel.TileGMConfig import TileGMCfg
    acc.merge(TileGMCfg(flags))
 
    from TileConditions.TileCablingSvcConfig import TileCablingSvcCfg
    acc.merge( TileCablingSvcCfg(flags) )
 
    from xAODTruthCnv.xAODTruthCnvConfig import GEN_EVNT2xAODCfg
    acc.merge(GEN_EVNT2xAODCfg(flags, name='GEN_EVNT2xAOD', AODContainerName='TruthEvent'))
 
    if not outputFile:
        outputFile = 'tiletb.d3pd.root'
 
    from D3PDMakerCoreComps.MakerAlgConfig import MakerAlgConfig
    alg = MakerAlgConfig(flags, acc, 'truth', outputFile, ExistDataHeader=False)
 
    from TruthD3PDMaker.TruthParticleD3PDObject  import TruthParticleD3PDObject
    alg += TruthParticleD3PDObject(1)
 
    from CaloSysD3PDMaker.TileHitInfoD3PDObject import TileHitInfoD3PDObject
    alg += TileHitInfoD3PDObject(0, sgkey='TileHitVec', prefix='Tile_')
 
    from CaloSysD3PDMaker.TileHitD3PDObject import TileHitD3PDObject
    alg += TileHitD3PDObject(0, prefix='TileHit_')
 
    acc.addEventAlgo(alg.alg)
 
    acc.setAppProperty('HistogramPersistency', 'ROOT')
 
    return acc
 
 
if __name__ == '__main__':
 
    # Set the Athena configuration flags
    from AthenaConfiguration.AllConfigFlags import initConfigFlags
    flags = initConfigFlags()
 
    # Setup logs
    from AthenaCommon.Logging import log
    from AthenaCommon.Constants import INFO
    log.setLevel(INFO)
 
    from AthenaConfiguration.TestDefaults import defaultTestFiles
    flags.Input.Files = defaultTestFiles.HITS_RUN3
    flags.Exec.MaxEvents = 3
    flags.fillFromArgs()
 
    log.info('FINAL CONFIG FLAGS SETTINGS FOLLOW')
    flags.dump()
 
    flags.lock()
 
    # Initialize configuration object, add accumulator, merge, and run.
    from AthenaConfiguration.MainServicesConfig import MainServicesCfg
    cfg = MainServicesCfg(flags)
 
    from AthenaPoolCnvSvc.PoolReadConfig import PoolReadCfg
    cfg.merge(PoolReadCfg(flags))
 
    cfg.merge(TileSimD3PDCfg(flags))
 
    cfg.printConfig(withDetails=True, summariseProps=True)
 
    cfg.store( open('TileSimOutputConfig.pkl', 'wb') )
 
    sc = cfg.run()
 
    import sys
    # Success should be 0
    sys.exit(0 if sc.isSuccess() else 1)