AthenaPoolExample_ReadWriteNext.py

Go to the documentation of this file.

#!/env/python
 
# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
 

 
from AthenaConfiguration.AllConfigFlags import initConfigFlags
from AthenaConfiguration.ComponentFactory import CompFactory
from AthenaCommon.Constants import DEBUG
from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg, outputStreamName
 
streamName = "Stream1"
outputFileName = "SimplePoolFile4.root"
noTag = True
                             
# Setup flags
flags = initConfigFlags()
flags.Input.Files = ["SimplePoolFile3.root"]
flags.addFlag(f"Output.{streamName}FileName", outputFileName)
flags.Exec.MaxEvents = -1
flags.Common.MsgSuppression = False
flags.Exec.DebugMessageComponents = [outputStreamName(streamName),
                                     "ReadData", "ReWriteData",
                                     "PoolSvc", "AthenaPoolCnvSvc","AthenaPoolAddressProviderSvc", "MetaDataSvc"]
 
#Run3 for now
from AthenaConfiguration.TestDefaults import defaultGeometryTags
flags.GeoModel.AtlasVersion = defaultGeometryTags.RUN3
 
flags.lock()
 
# Main services
from AthenaConfiguration.MainServicesConfig import MainServicesCfg
acc = MainServicesCfg( flags )
 
# Pool reading and writing
from AthenaPoolExampleAlgorithms.AthenaPoolExampleConfig import AthenaPoolExampleReadCfg, AthenaPoolExampleWriteCfg
acc.merge( AthenaPoolExampleReadCfg(flags, readCatalogs = ["file:Catalog1.xml"]) )
acc.merge( AthenaPoolExampleWriteCfg(flags, streamName,
                                     writeCatalog = "file:Catalog1.xml",
                                     disableEventTag = noTag) )
 
# Creata and attach the algorithms
acc.addEventAlgo( CompFactory.AthPoolEx.ReadData("ReadData", OutputLevel = DEBUG) )
acc.addEventAlgo( CompFactory.AthPoolEx.ReWriteData("ReWriteData", ExampleHitKey="", OutputLevel = DEBUG) )
 
# ----------------  Output Stream configuration
streamCA = OutputStreamCfg(flags, streamName, disableEventTag = noTag,
                           ItemList = ["EventInfo#*", "ExampleTrackContainer#MyTracks"] )
stream = streamCA.getEventAlgo( outputStreamName( streamName ) )
acc.merge( streamCA )
 
#--------------------------------------------------------------
# Set output level threshold (2=DEBUG, 3=INFO, 4=WARNING, 5=ERROR, 6=FATAL)
#--------------------------------------------------------------
stream.WritingTool.OutputLevel = 3
stream.HelperTools[0].OutputLevel = 3
 
# Run
import sys
sc = acc.run(flags.Exec.MaxEvents)
sys.exit(sc.isFailure())