Resolve "Add true decaying particle info onto event display"

Closes #43 (closed)

benchmarks/imaging_ecal/scripts/draw_cluster.py

@@ -149,11 +149,11 @@ if __name__ == '__main__':
 
     # Read all mc particles
     dfallmcp = get_all_mcp(args.file, args.iev, 'mcparticles2')
+    pdgbase = ROOT.TDatabasePDG()
     # Select decaying particles
     dftemp = dfallmcp[dfallmcp['g4Parent'] == 1.0]
     if len(dftemp) > 0:
         dfdecaymcp = dftemp.copy()
-        pdgbase = ROOT.TDatabasePDG()
         for iptl in [0, len(dfdecaymcp) - 1]:
             infoptl = pdgbase.GetParticle(int(dfdecaymcp['pid'].iloc[iptl]))
             print("{} Decaying particle = {}, pdgcode = {}, charge = {}, mass = {}"\
@@ -238,7 +238,7 @@ 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 
-    if len(dfdecaymcp) > 0:
+    if len(dftemp) > 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)