diff --git a/benchmarks/dis/analysis/truth_reconstruction.py b/benchmarks/dis/analysis/truth_reconstruction.py index b985155e324824832d6a3c026b0403beea7054f3..946dbb740b706713571db8907a9763589db2850d 100644 --- a/benchmarks/dis/analysis/truth_reconstruction.py +++ b/benchmarks/dis/analysis/truth_reconstruction.py @@ -141,10 +141,7 @@ for i in range(len(MC_list)): axs[1].set_ylabel('%s_rc'%(title_list[i])) axs[1].set_title('%s Correlation'%(title_list[i])) - if benchmark_tag == 'single': - fig.suptitle('%s %s events'%(config,Nevents)) - else: - fig.suptitle('(%g x %g)GeV %gGeV minQ2 %s events'%(k,p,minq2,Nevents)) + fig.suptitle('%s %s events'%(config,Nevents)) plt.savefig(os.path.join(args.outdir, r_path, '%s_correlation_%s.png' % (title_list[i],config))) @@ -233,12 +230,7 @@ for i in range(len(MC_list)): ax6.set_xlabel('%s'%(title_list[i])) fig.set_figwidth(20) fig.set_figheight(10) - - if benchmark_tag == 'single': - ax1.set_title('%s %s %s %s events'%(title_list[i],tratio,config,Nevents)) - else: - ax1.set_title('%s %s (%g x %g)GeV %gGeV minQ2 %s events'%(title_list[i],tratio,k,p,minq2,Nevents)) - + ax1.set_title('%s %s %s %s events'%(title_list[i],tratio,config,Nevents)) plt.savefig(os.path.join(args.outdir, r_path, '%s_%s_%s.png' % (title_list[i],tratio,config))) @@ -317,10 +309,7 @@ for i in range(1,len(MC_list),1): fig.set_figwidth(20) fig.set_figheight(10) - if benchmark_tag == 'single': - ax1.set_title('%s Difference Vs Momentum %s %s events'%(title_list[i],config,Nevents)) - else: - ax1.set_title('%s Difference Vs Momentum (%g x %g)GeV %gGeV minQ2 %s events'%(title_list[i],k,p,minq2,Nevents)) + ax1.set_title('%s Difference Vs Momentum %s %s events'%(title_list[i],config,Nevents)) plt.savefig(os.path.join(args.outdir, r_path, '%s_difference_vs_momentum_%s.png' % (title_list[i],config))) @@ -363,11 +352,8 @@ ax4.set_xlabel('Theta rc') fig.set_figwidth(20) fig.set_figheight(10) -if benchmark_tag == 'single': - fig.suptitle('Photons in %s %s events'%(config,Nevents)) -else: - fig.suptitle('Photons in (%g x %g)GeV %gGeV minQ2 %s events'%(k,p,minq2,Nevents)) - +fig.suptitle('Photons in %s %s events'%(config,Nevents)) + plt.savefig(os.path.join(args.outdir, r_path, 'photons_%s.png' % (config)))