RunRDOAnalysis.py

Go to the documentation of this file.

#!/usr/bin/env python
"""Run RDOAnalysis
 
Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
"""
import sys
from argparse import ArgumentParser
 
from AthenaConfiguration.AllConfigFlags import initConfigFlags
from AthenaConfiguration.Enums import ProductionStep
 
# Argument parsing
parser = ArgumentParser("RunRDOAnalysis.py")
parser.add_argument("detectors", metavar="detectors", type=str, nargs="*",
                    help="Specify the list of detectors")
parser.add_argument("--localgeo", default=False, action="store_true",
                    help="Use local geometry XML files")
parser.add_argument("-n", "--maxEvents", default=-1, type=int,
                    help="The number of events to run. -1 runs all events.")
parser.add_argument("-i", "--input",
                    help="The input RDO file to use")
parser.add_argument("-p", "--presampling",  default=False, action="store_true",
                    help="Run on presampling RDO")
args = parser.parse_args()
 
 
# Some info about the job
print("----RDOAnalysis----")
print()
if args.localgeo:
    print("Using local Geometry XML files")
print(f"Input file: {args.input}")
if not args.detectors:
    print("Running complete detector")
else:
    print("Running with: {}".format(", ".join(args.detectors)))
print()
 
# Configure
flags = initConfigFlags()
flags.Input.Files = [args.input]
flags.Concurrency.NumThreads = 1
flags.Concurrency.NumConcurrentEvents = 1
flags.Scheduler.CheckDependencies = True
flags.Scheduler.ShowDataDeps = True
flags.Scheduler.ShowDataFlow = True
flags.Scheduler.ShowControlFlow = True
 
if args.localgeo:
    flags.ITk.Geometry.AllLocal = True
if args.presampling:
    flags.Common.ProductionStep = ProductionStep.PileUpPresampling
flags.Output.HISTFileName = 'RDOAnalysis.root'
 
from AthenaConfiguration.DetectorConfigFlags import setupDetectorFlags
setupDetectorFlags(flags, args.detectors, use_metadata=True, toggle_geometry=True)
 
flags.lock()
 
# Construct our accumulator to run
from AthenaConfiguration.MainServicesConfig import MainServicesCfg
acc = MainServicesCfg(flags)
from AthenaPoolCnvSvc.PoolReadConfig import PoolReadCfg
acc.merge(PoolReadCfg(flags))
from RDOAnalysis.RDOAnalysisConfig import RDOAnalysisCfg
acc.merge(RDOAnalysisCfg(flags))
 
# Execute and finish
sc = acc.run(maxEvents=args.maxEvents)
 
# Success should be 0
sys.exit(not sc.isSuccess())