ATLAS Offline Software
testPrepDataWrite.py
Go to the documentation of this file.
1 # Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
2 from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
3 
4 def setupTestOutputCfg(flags,**kwargs):
5 
6  kwargs.setdefault("AcceptAlgs",[])
7 
8  result = ComponentAccumulator()
9 
13  container_items = ["xAOD::MdtDriftCircleContainer#*", "xAOD::MdtDriftCircleAuxContainer#*",
14  "xAOD::sTgcStripContainer#*", "xAOD::sTgcStripAuxContainer#*",
15  "xAOD::MMClusterContainer#*", "xAOD::MMClusterAuxContainer#*",
16  "xAOD::TgcStripContainer#*", "xAOD::TgcStripAuxContainer#*",
17  "xAOD::RpcStripContainer#*", "xAOD::RpcStripAuxContainer#*",
18  "xAOD::MdtTwinDriftCircleContainer#*", "xAOD::MdtDriftCircleAuxContainer#*",
19  "xAOD::RpcStrip2DContainer#*", "xAOD::RpcStrip2DAuxContainer#*",
20  "xAOD::sTgcPadContainer#*", "xAOD::sTgcPadAuxContainer#*",
21  "xAOD::sTgcStripContainer#*", "xAOD::sTgcStripAuxContainer#*",
22  "xAOD::sTgcWireContainer#*", "xAOD::sTgcWireAuxContainer#*",
23 
24  ]
25 
26 
27  from xAODMetaDataCnv.InfileMetaDataConfig import SetupMetaDataForStreamCfg
28  result.merge(SetupMetaDataForStreamCfg(flags, kwargs["streamName"]))
29 
30  from OutputStreamAthenaPool.OutputStreamConfig import OutputStreamCfg
31  kwargs.setdefault("ItemList", container_items)
32  result.merge(OutputStreamCfg(flags, **kwargs))
33  return result
34 
35 if __name__=="__main__":
36  from MuonGeoModelTestR4.testGeoModel import setupGeoR4TestCfg, SetupArgParser, executeTest
37  parser = SetupArgParser()
38  parser.add_argument("--saveTestNtuple", help="Schedule the SimHits tester n-tuple", action='store_true',
39  default=False)
40  parser.set_defaults(nEvents = 150)
41  parser.set_defaults(outRootFile="MuonPrepDataTest.root")
42  parser.set_defaults(inputFile=["/cvmfs/atlas-nightlies.cern.ch/repo/data/data-art/MuonGeomRTT/R4SimHits.pool.root"])
43  parser.set_defaults(defaultGeoFile="RUN4")
44  parser.set_defaults(eventPrintoutLevel = 50)
45 
46  args = parser.parse_args()
47  from AthenaConfiguration.AllConfigFlags import initConfigFlags
48  flags = initConfigFlags()
49 
50 
51  streamName = "MuonPrepDataTest"
52  flags.addFlag(f"Output.{streamName}FileName", args.outRootFile)
53  flags.addFlag(f"Output.doWrite{streamName}", True)
54 
55  flags, cfg = setupGeoR4TestCfg(args, flags)
56  from MuonConfig.MuonDataPrepConfig import xAODUncalibMeasPrepCfg
57  cfg.merge(xAODUncalibMeasPrepCfg(flags))
58 
59  cfg.merge(setupTestOutputCfg(flags, streamName=streamName))
60 
61  executeTest(cfg)
62 
python.JetAnalysisCommon.ComponentAccumulator
ComponentAccumulator
Definition: JetAnalysisCommon.py:302
python.OutputStreamConfig.OutputStreamCfg
def OutputStreamCfg(flags, streamName, ItemList=[], MetadataItemList=[], disableEventTag=False, trigNavThinningSvc=None, takeItemsFromInput=False, extendProvenanceRecord=True, keepProvenanceTagsRegEx=None, AcceptAlgs=[], HelperTools=[])
Definition: OutputStreamConfig.py:16
testGeoModel.executeTest
def executeTest(cfg)
Definition: MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py:227
MdtCablingTester.SetupArgParser
def SetupArgParser()
Definition: MdtCablingTester.py:3
testGeoModel.setupGeoR4TestCfg
def setupGeoR4TestCfg(args, flags=None)
Definition: MuonPhaseII/MuonDetDescr/MuonGeoModelTestR4/python/testGeoModel.py:126
MuonDataPrepConfig.xAODUncalibMeasPrepCfg
def xAODUncalibMeasPrepCfg(flags)
Configuriation snippet to schedule all algorithms providing the Uncalibrated measurements.
Definition: MuonDataPrepConfig.py:29
python.AllConfigFlags.initConfigFlags
def initConfigFlags()
Definition: AllConfigFlags.py:19
testPrepDataWrite.setupTestOutputCfg
def setupTestOutputCfg(flags, **kwargs)
Definition: testPrepDataWrite.py:4
InfileMetaDataConfig.SetupMetaDataForStreamCfg
def SetupMetaDataForStreamCfg(flags, streamName="", AcceptAlgs=None, createMetadata=None, propagateMetadataFromInput=True, *args, **kwargs)
Definition: InfileMetaDataConfig.py:222