From 07d3bc9656a7aee7a676e8e39a6cb7d8e32b92f4 Mon Sep 17 00:00:00 2001
From: "jihee.kim" <jihee.kim@anl.gov>
Date: Fri, 13 Aug 2021 17:37:44 -0500
Subject: [PATCH] fixed typo
---
benchmarks/imaging_ecal/scripts/draw_cluster.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/benchmarks/imaging_ecal/scripts/draw_cluster.py b/benchmarks/imaging_ecal/scripts/draw_cluster.py
index 35dbf0c9..2e72941d 100644
--- a/benchmarks/imaging_ecal/scripts/draw_cluster.py
+++ b/benchmarks/imaging_ecal/scripts/draw_cluster.py
@@ -151,7 +151,7 @@ if __name__ == '__main__':
dfallmcp = get_all_mcp(args.file, args.iev, 'mcparticles2')
# Select decaying particles
dftemp = dfallmcp[dfallmcp['g4Parent'] == 1.0]
- if len(dfdecaymcp) > 0:
+ if len(dftemp) > 0:
dfdecaymcp = dftemp.copy()
pdgbase = ROOT.TDatabasePDG()
for iptl in [0, len(dfdecaymcp) - 1]:
@@ -238,7 +238,8 @@ if __name__ == '__main__':
bins=(np.arange(*eta_rg, step=args.topo_size/1000.), np.arange(*phi_rg, step=args.topo_size)),
cmap=cmap, cmin=0., pc_kw=dict(alpha=0.8, edgecolor='k'))
# draw true decaying particle position
- ax.scatter(dfdecaymcp['eta'].values, dfdecaymcp['phi'].values, marker='x', color='red', s=22**2, linewidth=5.0)
+ if len(dfdecaymcp) > 0:
+ ax.scatter(dfdecaymcp['eta'].values, dfdecaymcp['phi'].values, marker='x', color='red', s=22**2, linewidth=5.0)
ax.set_ylabel(r'$\phi$ (mrad)', fontsize=32)
ax.set_xlabel(r'$\eta$', fontsize=32)
--
GitLab