From 980d349116e598d6f98281b680a6665cbe94ac4c Mon Sep 17 00:00:00 2001
From: Wouter Deconinck <wdconinc@gmail.com>
Date: Wed, 21 Feb 2024 17:04:07 +0000
Subject: [PATCH] fix: avoid reading unused outputTrackParameters columns
---
benchmarks/tracking/scripts/tracking_performance.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/benchmarks/tracking/scripts/tracking_performance.py b/benchmarks/tracking/scripts/tracking_performance.py
index 7a2cb76a..b776097a 100644
--- a/benchmarks/tracking/scripts/tracking_performance.py
+++ b/benchmarks/tracking/scripts/tracking_performance.py
@@ -116,6 +116,8 @@ if __name__ == '__main__':
help='Collection name of MC particles truth info.')
parser.add_argument('--tracking-collection', dest='coll', default='outputTrackParameters',
help='Collection name of clusters to plot')
+ parser.add_argument('--tracking-collection-columns', dest='cols', default=['theta', 'phi', 'qOverP'],
+ help='Collection columes of clusters to load')
args = parser.parse_args()
# multi-threading for RDataFrame
@@ -134,7 +136,7 @@ if __name__ == '__main__':
rdf_rec = ROOT.RDataFrame('events', args.rec_file)
dfm = thrown_particles_figure(rdf_rec, save=os.path.join(args.outdir, '{}_thrown_particles.png'.format(args.nametag)), mcbranch=args.mc)
- df = flatten_collection(rdf_rec, args.coll)
+ df = flatten_collection(rdf_rec, args.coll, args.cols)
df.rename(columns={c: c.replace(args.coll + '.', '') for c in df.columns}, inplace=True)
fig, axs = plt.subplots(3, 2, figsize=(16, 18), dpi=120)
--
GitLab