diff --git a/benchmarks/dis/analysis/kinematics_correlations.py b/benchmarks/dis/analysis/kinematics_correlations.py index c578be734f6a31f9d5676e0833d80bc41abbc925..f175e8c6858c3c51785d8893fd787e792ee37a78 100644 --- a/benchmarks/dis/analysis/kinematics_correlations.py +++ b/benchmarks/dis/analysis/kinematics_correlations.py @@ -89,7 +89,7 @@ def Q2correlation(minq2,method): #minq2 can be 1,10,100, or 1000; method can be plt.xlabel('$Q^2$ [$GeV^2$] Truth') plt.ylabel('$Q^2$ [$GeV^2$] {}'.format(method_dict['{}'.format(method)])) plt.title('{} $Q^2$ correlation {}x{} $minQ^2=${}$GeV^2$\n {} events DETECTOR_CONFIG: {} '.format(method_dict['{}'.format(method)],k,p,minq2,Nevents,Dconfig)) - plt.savefig(os.path.join(r_path, 'Q2_correlation_%s_%s.png' %(method,config))) + plt.savefig(os.path.join(r_path, 'Q2_correlation_%s_%s.png' %(method_dict_n[method],config))) ########################################################################################## @@ -147,7 +147,7 @@ def Xcorrelation(minq2,method): #minq2 can be 1,10,100, or 1000; method can be ' plt.xlabel('x Truth') plt.ylabel('$x$ {}'.format(method_dict['{}'.format(method)])) plt.title('{} $x$ correlation {}x{} $minQ^2=${}$GeV^2$\n {} events DETECTOR_CONFIG: {} '.format(method_dict['{}'.format(method)],k,p,minq2,Nevents,Dconfig)) - plt.savefig(os.path.join(r_path, 'x_correlation_%s_%s.png' %(method,config))) + plt.savefig(os.path.join(r_path, 'x_correlation_%s_%s.png' %(method_dict_n[method],config))) @@ -178,6 +178,7 @@ Xvalues_S = Sigma[1] Xvalues_eS = eSigma[1] method_dict = {'e':'Electron','DA':'Double-Angle','JB':'Jacquet-Blondel','S':'Sigma','eS':'eSigma'} +method_dict_n = {'e':'8','DA':'9','JB':'10','S':'11','eS':'12'} method_Q2values_dict = {'e':Q2values_E,'DA':Q2values_DA,'JB':Q2values_JB,'S':Q2values_S,'eS':Q2values_eS} method_Xvalues_dict = {'e':Xvalues_E,'DA':Xvalues_DA,'JB':Xvalues_JB,'S':Xvalues_S,'eS':Xvalues_eS} @@ -197,7 +198,7 @@ Xcorrelation(minq2,'eS') #Kinematic coverage plot ########################################################################################## -fig = plt.figure() +fig = plt.figure(figsize=(20,10)) gs = fig.add_gridspec(2, 3, wspace=0.09, hspace = 0.2) (ax1, ax2, ax3), (ax4, ax5, ax6) = gs.subplots() @@ -220,8 +221,5 @@ ax4.set_title('Electron') ax5.set_title('eSigma') ax6.set_title('Jacquet Blondel') -fig.set_figwidth(30) -fig.set_figheight(20) - -plt.title('Kinematic Coverage {}x{} $minQ^2=${}$GeV^2$\n {} events DETECTOR_CONFIG: {} '.format(k,p,minq2,Nevents,Dconfig)) -plt.savefig(os.path.join(r_path, 'kinematic_coverage_%s.png' %(config))) +fig.suptitle('Kinematic Coverage {}x{} $minQ^2=${}$GeV^2$\n {} events DETECTOR_CONFIG: {} '.format(k,p,minq2,Nevents,Dconfig)) +plt.savefig(os.path.join(r_path, 'z_kinematic_coverage_%s.png' %(config))) diff --git a/benchmarks/dis/analysis/truth_reconstruction.py b/benchmarks/dis/analysis/truth_reconstruction.py index 5bb19de30e4961b9e5caa92290f7765d41297ece..cd74d45c58be891c70adeae063611b69322d4c74 100644 --- a/benchmarks/dis/analysis/truth_reconstruction.py +++ b/benchmarks/dis/analysis/truth_reconstruction.py @@ -17,7 +17,7 @@ kwargs = vars(args) rec_file = args.rec_file Nevents = int(args.nevents) -r_path = args.results_path + '/truth_reconstruction/' #Path for output figures and file. +r_path = args.results_path #Path for output figures and file. Dconfig = 'epic' + args.config.split('_epic')[1].strip() #Detector config config = args.config.split('_epic')[0].strip() @@ -71,6 +71,8 @@ RC_list = [ak.Array(momentum_rc[recID][booll]), #Momentum ak.Array(phi_rc[recID][booll]), #Phi -np.log(np.tan((ak.Array(theta_rc[recID][booll]))/2))] #Eta title_list = ['Momentum','Theta','Phi','Eta'] +title_list_n = ['1','2','3','4'] +particle_name_n = {'Electron':'5','Pions':'6','Photons':'7'} ### MC Momentum for different particles list M_list = [ak.Array(momentum_mc[simID][booll]), ak.Array(momentum_mc[simID][booll][boolean_pion]), @@ -175,7 +177,7 @@ for i in range(len(MC_list)): #Repeat the following steps for each variable (mom fig.set_figwidth(20) fig.set_figheight(10) ax1.set_title('%s %s %s %s events\n DETECTOR_CONFIG: %s'%(title_list[i],title,config,Nevents,Dconfig)) - plt.savefig(os.path.join(r_path, '%s_%s_%s.png' % (title_list[i],title,config))) + plt.savefig(os.path.join(r_path, '%s_%s_%s.png' % (title_list_n[i],title,config))) plt.close() @@ -219,7 +221,7 @@ for i in range(len(MC_list)): #Repeat the following steps for each variable (mom fig.set_figwidth(20) fig.set_figheight(10) ax1.set_title('%s Difference Vs Momentum %s %s events\n DETECTOR_CONFIG: %s'%(title_list[i],config,Nevents,Dconfig)) - plt.savefig(os.path.join(r_path, '%s_difference_vs_momentum_%s.png' % (title_list[i],config))) + plt.savefig(os.path.join(r_path, '%s_difference_vs_momentum_%s.png' % (title_list_n[i],config))) ################################################################################################### @@ -269,7 +271,7 @@ for i in range(len(MC_list)): #Repeat the following steps for each variable (mom axs[1].set_ylabel('%s_rc'%(title_list[i])) axs[1].set_title('%s Correlation'%(title_list[i])) fig.suptitle('%s %s events\n DETECTOR_CONFIG: %s'%(config,Nevents,Dconfig)) - plt.savefig(os.path.join(r_path, '%s_correlation_%s.png' % (title_list[i],config))) + plt.savefig(os.path.join(r_path, '%s_correlation_%s.png' % (title_list_n[i],config))) ################################################################################################### @@ -320,7 +322,7 @@ if particle in particle_dict.keys(): particle_plots(boolean_particle) plt.suptitle('%s in %s %s events\n DETECTOR_CONFIG: %s'%(particle_name,config,Nevents,Dconfig)) - plt.savefig(os.path.join(r_path, '%s_%s.png' % (particle_name,config))) + plt.savefig(os.path.join(r_path, '%s_%s.png' % (particle_name_n[particle_name],config))) else: for i in [[boolean_photon,'Photons'],[boolean_electron,'Electrons'],[boolean_pion,'Pions']]: boolean_particle = i[0] @@ -328,7 +330,7 @@ else: particle_plots(boolean_particle) plt.suptitle('%s in %s %s events\n DETECTOR_CONFIG: %s'%(particle_name,config,Nevents,Dconfig)) - plt.savefig(os.path.join(r_path, '%s_%s.png' % (particle_name,config))) + plt.savefig(os.path.join(r_path, '%s_%s.png' % (particle_name_n[particle_name],config)))