AthenaPoolExample_Copy.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 OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg, outputStreamName
 
streamName = "Copy"
outputFileName =  "SimplePoolReplica1.root"
noTag = True
 
# Setup flags
flags = initConfigFlags()
flags.Input.Files = [ "SimplePoolFile1.root" ]
flags.addFlag(f"Output.{streamName}FileName", outputFileName)
flags.Exec.MaxEvents = -1
flags.Common.MsgSuppression = False
flags.Exec.DebugMessageComponents = [ outputStreamName(streamName), "EventSelector",
                                      "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) )
 
# Produce xAOD::EventInfo from EventInfo
from xAODEventInfoCnv.xAODEventInfoCnvConfig import EventInfoCnvAlgCfg
acc.merge( EventInfoCnvAlgCfg(flags, disableBeamSpot = True) )
 
streamCA = OutputStreamCfg(flags, streamName, disableEventTag = noTag,
                           ItemList = ["EventInfo#*", "ExampleHitContainer#MyHits"])
stream = streamCA.getEventAlgo( outputStreamName( streamName ) )
stream.ExtendProvenanceRecord = False
acc.merge( streamCA )
 
# Run
import sys
sc = acc.run(flags.Exec.MaxEvents)
sys.exit(sc.isFailure())