diff --git a/benchmarks/imaging_ecal/scripts/draw_cluster.py b/benchmarks/imaging_ecal/scripts/draw_cluster.py index 2e72941d3df7747857ee8a92c87311fc87b9843d..a2925f2bd9291ddc3dd58341bbc809e0f52deaef 100644 --- a/benchmarks/imaging_ecal/scripts/draw_cluster.py +++ b/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 = {}"\