From 038082c8ec71f505adbc0d247f4741adf9470405 Mon Sep 17 00:00:00 2001
From: Tobias Boeckh <tobias.boeckh@cern.ch>
Date: Mon, 13 Jun 2022 00:03:29 +0200
Subject: [PATCH] set input file and max number of events from commandline

---
 Tracking/Acts/FaserActsKalmanFilter/test/CKF2.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/Tracking/Acts/FaserActsKalmanFilter/test/CKF2.py b/Tracking/Acts/FaserActsKalmanFilter/test/CKF2.py
index 98059c9e..afcef75d 100644
--- a/Tracking/Acts/FaserActsKalmanFilter/test/CKF2.py
+++ b/Tracking/Acts/FaserActsKalmanFilter/test/CKF2.py
@@ -13,11 +13,17 @@ from TrackerSpacePointFormation.TrackerSpacePointFormationConfig import TrackerS
 from TrackerSegmentFit.TrackerSegmentFitConfig import SegmentFitAlgCfg
 from FaserActsKalmanFilter.GhostBustersConfig import GhostBustersCfg
 from FaserActsKalmanFilter.CKF2Config import CKF2Cfg
+import argparse
+
+parser = argparse.ArgumentParser()
+parser.add_argument("file", nargs="+", help="full path to input file")
+parser.add_argument("--nevents", "-n", default=-1, type=int, help="Number of events to process")
+args = parser.parse_args()
 
 log.setLevel(DEBUG)
 Configurable.configurableRun3Behavior = True
 
-ConfigFlags.Input.Files = ['my.RDO.pool.root']
+ConfigFlags.Input.Files = args.file
 ConfigFlags.addFlag("Output.xAODFileName", f"CKF.xAOD.root")
 ConfigFlags.Output.ESDFileName = "CKF.ESD.pool.root"
 ConfigFlags.IOVDb.GlobalTag = "OFLCOND-FASER-02"
@@ -59,5 +65,7 @@ itemList = [
 ]
 acc.merge(OutputStreamCfg(ConfigFlags, "xAOD", itemList))
 
-sc = acc.run(maxEvents=-1)
+acc.getService("MessageSvc").Format = "% F%40W%S%7W%R%T %0W%M"
+
+sc = acc.run(maxEvents=args.nevents)
 sys.exit(not sc.isSuccess())
-- 
GitLab