from numpy import * import matplotlib from matplotlib.pylab import * import ppn import mppnp as mp import data_plot import utils ''' mod1_1=1921 mod1_2=4797 mod1_env=1 mod1_rcb=1700 #mod2_1=2110 #mod2_2=4490 #cycles=arange(mod1,mod2+1,1) cycles=[mod1_1] h5=mp.se('/nfs/rpod2/athira/mppnp/case9/H5_out') #h5=mp.se('/nfs/rpod2/athira/mppnp/RCB_0.5/H5_out') #h5=mp.se('/nfs/rpod2/athira/mppnp/RCB_0.7/old_run3/H5_out_0.30_8d51') element=['H','Li','C','N','O','F','Na','Mg','Al','Si','S','Ca','Ti','Fe','Ni','Zn','Y','Zr','Ba','La'] specs1=['H-1'] specs2=['Li-7'] specs3=['C-12','C-13'] specs4=['O-16','O-17','O-18'] specs5=['Ne-20','Ne-21','Ne-22'] specs6=['Na-23'] specs7=['Mg-24','Mg-25','Mg-26'] specs8=['Al-27'] specs9=['Si-28','Si-29','Si-30'] specs10=['S-32','S-33','S-34','S-36'] specs11=['Ca-40','Ca-42','Ca-43','Ca-44','Ca-46','Ca-48'] specs12=['Ti-46','Ti-47','Ti-48','Ti-49','Ti-50'] specs13=['Fe-54','Fe-56','Fe-57','Fe-58'] specs14=['Ni-58','Ni-60','Ni-61','Ni-62','Ni-64'] specs15=['Zn-64','Zn-66','Zn-67','Zn-68','Zn-70'] specs16=['Y-89'] specs17=['Zr-90','Zr-91','Zr-92','Zr-94','Zr-96'] specs18=['Ba-130','Ba-132','Ba-134','Ba-135','Ba-136','Ba-137','Ba-138'] specs19=['La-138','La-139'] h=['H-1'] li=['Li-7'] c=['C-12','C-13'] n=['N-14','N-15'] o=['O-16','O-17','O-18'] f=['F-19'] ne=['Ne-20','Ne-21','Ne-22'] na=['Na-23'] mg=['Mg-24','Mg-25','Mg-26'] al=['Al-27'] si=['Si-28','Si-29','Si-30'] s=['S-32','S-33','S-34','S-36'] ca=['Ca-40','Ca-42','Ca-43','Ca-44','Ca-46','Ca-48'] ti=['Ti-46','Ti-47','Ti-48','Ti-49','Ti-50'] fe=['Fe-54','Fe-56','Fe-57','Fe-58'] ni=['Ni-58','Ni-60','Ni-61','Ni-62','Ni-64'] zn=['Zn-64','Zn-66','Zn-67','Zn-68','Zn-70'] y=['Y-89'] zr=['Zr-90','Zr-91','Zr-92','Zr-94','Zr-96'] ba=['Ba-130','Ba-132','Ba-134','Ba-135','Ba-136','Ba-137','Ba-138'] la=['La-138','La-139'] specs=[h,li,c,n,o,f,na,mg,al,si,s,ca,ti,fe,ni,zn,y,zr,ba,la] n_frac1=[] H=[];Li=[];C=[];N=[];O=[];F=[];Na=[];Mg=[];Al=[];Si=[];S=[];Ca=[];Ti=[];Fe=[];Ni=[];Zn=[];Y=[];Zr=[];Ba=[];La=[] E_s=[12.0,1.2,8.6,8.1,8.0,4.6,6.3,7.6,6.5,7.6,7.2,6.4,5.0,7.5,6.3,4.6,2.2,2.6,2.1,1.2] z=[1,3,6,7,8,9,11,12,13,14,16,20,22,26,28,30,39,40,56,57] def E(y): return log10(y)+12.0 def x_fe(e,i): return e[i]-e[13]-E_s[i]+7.5 def x_s(e,i): return e[i]-e[10]-E_s[i]+7.2 for cycle in cycles: for i in range(len(element)): mass=[] n_temp=[] for j,item in enumerate(specs[i]): m=float(''.join(d for d in item if d.isdigit())) el=h5.get(cycle,item) n=el/m #print n[len(el)-1] n_temp.append(n[len(el)-1]) #print n_temp eval(element[i]).append(sum(n_temp)) n_frac1.append(sum(n_temp)) #eval(element[i]).append(sum(n_temp)+12.0-z[i]+7.5) E_x1_3=empty([len(element)]) E_x2_3=empty([len(element)]) E_x=empty([len(element)]) for a in range(len(n_frac1)): E_x1_3[a]=E((n_frac1[a])) #-----------------------------------------------------------------------------------# cycles=[mod1_2] n_frac2=[] H=[];Li=[];C=[];N=[];O=[];F=[];Na=[];Mg=[];Al=[];Si=[];S=[];Ca=[];Ti=[];Fe=[];Ni=[];Zn=[];Y=[];Zr=[];Ba=[];La=[] E_s=[12.0,1.2,8.6,8.1,8.0,4.6,6.3,7.6,6.5,7.6,7.2,6.4,5.0,7.5,6.3,4.6,2.2,2.6,2.1,1.2] z=[1,3,6,7,8,9,11,12,13,14,16,20,22,26,28,30,39,40,56,57] def E(y): return log10(y)+12.0 def x_fe(e,i): return e[i]-e[13]-E_s[i]+7.5 def x_s(e,i): return e[i]-e[10]-E_s[i]+7.2 for cycle in cycles: for i in range(len(element)): mass=[] n_temp=[] for j,item in enumerate(specs[i]): m=float(''.join(d for d in item if d.isdigit())) el=h5.get(cycle,item) n=el/m #print n[len(el)-1] n_temp.append(n[len(el)-1]) #print n_temp eval(element[i]).append(sum(n_temp)) n_frac2.append(sum(n_temp)) cycles=[mod1_env] n_frac3=[] for cycle in cycles: for i in range(len(element)): mass=[] n_temp=[] for j,item in enumerate(specs[i]): m=float(''.join(d for d in item if d.isdigit())) el=h5.get(cycle,item) n=el/m #print n[len(el)-1] n_temp.append(n[len(el)-1]) #print n_temp eval(element[i]).append(sum(n_temp)) n_frac3.append(sum(n_temp)) cycles=[mod1_rcb] n_frac4=[] for cycle in cycles: for i in range(len(element)): mass=[] n_temp=[] for j,item in enumerate(specs[i]): m=float(''.join(d for d in item if d.isdigit())) el=h5.get(cycle,item) n=el/m #print n[len(el)-1] n_temp.append(n[len(el)-1]) #print n_temp eval(element[i]).append(sum(n_temp)) n_frac4.append(sum(n_temp)) E_x2_3=empty([len(element)]) E_x3_3=empty([len(element)]) E_x4_3=empty([len(element)]) for a in range(len(n_frac2)): E_x2_3[a]=E((n_frac2[a])) E_x3_3[a]=E((n_frac3[a])) E_x4_3[a]=E((n_frac4[a])) for a in range(len(n_frac2)): E_x[a]=E((n_frac1[a]+n_frac2[a])/2.0) X_Fe3=empty([len(element)]) X_S3=empty([len(element)]) X_Fe3_1=empty([len(element)]) X_S3_1=empty([len(element)]) X_Fe3_2=empty([len(element)]) X_S3_2=empty([len(element)]) X_Fe3_env=empty([len(element)]) X_S3_env=empty([len(element)]) X_Fe3_rcb=empty([len(element)]) X_S3_rcb=empty([len(element)]) for a in range(len(E_x)): X_Fe3[a]=round(x_fe((E_x),a),2) X_S3[a]=round(x_s((E_x),a),2) X_Fe3_1[a]=round(x_fe((E_x1_3),a),2) X_S3_1[a]=round(x_s((E_x1_3),a),2) X_Fe3_2[a]=round(x_fe((E_x2_3),a),2) X_S3_2[a]=round(x_s((E_x2_3),a),2) X_Fe3_env[a]=round(x_fe((E_x3_3),a),2) X_S3_env[a]=round(x_s((E_x3_3),a),2) X_Fe3_rcb[a]=round(x_fe((E_x4_3),a),2) X_S3_rcb[a]=round(x_s((E_x4_3),a),2) def x(e,i): return e[i]-E_s[i] X3=empty([len(element)]) X3_1=empty([len(element)]) X3_2=empty([len(element)]) X3_env=empty([len(element)]) X3_rcb=empty([len(element)]) for a in range(len(E_x)): X3[a]=round(x((E_x),a),2) X3_1[a]=round(x((E_x1_3),a),2) X3_2[a]=round(x((E_x2_3),a),2) X3_env[a]=round(x((E_x3_3),a),2) X3_rcb[a]=round(x((E_x4_3),a),2) ''' figure(120) co_n_31=((10**(E_x4_3[2]-12)+10**(E_x4_3[4]-12))/10**(E_x4_3[3]-12)) c_o_31= 10**(E_x4_3[2]-E_x4_3[4]) plot(c_o_31,co_n_31,marker='o',mfc='None',mec='blue',markersize=6,linewidth=2) co_n_32=((10**(E_x1_3[2]-12)+10**(E_x1_3[4]-12))/10**(E_x1_3[3]-12)) c_o_32= 10**(E_x1_3[2]-E_x1_3[4]) plot(c_o_32,co_n_32,marker='^',mfc='None',mec='blue',markersize=6,linewidth=2) co_n_33=((10**(E_x2_3[2]-12)+10**(E_x2_3[4]-12))/10**(E_x2_3[3]-12)) c_o_33= 10**(E_x2_3[2]-E_x2_3[4]) plot(c_o_33,co_n_33,marker='^',mfc='blue',mec='blue',markersize=6,linewidth=2) figure(130) c_n_31= 10**(E_x4_3[2]-E_x4_3[3]) n_o_31= 10**(E_x4_3[3]-E_x4_3[4]) plot(c_n_31,n_o_31,marker='o',mfc='None',mec='blue',markersize=6,linewidth=2) figure(130) c_n_32= 10**(E_x1_3[2]-E_x1_3[3]) n_o_32= 10**(E_x1_3[3]-E_x1_3[4]) plot(c_n_32,n_o_32,marker='^',mfc='None',mec='blue',markersize=6,linewidth=2) figure(130) c_n_33= 10**(E_x2_3[2]-E_x2_3[3]) n_o_33= 10**(E_x2_3[3]-E_x2_3[4]) plot(c_n_33,n_o_33,marker='^',mfc='blue',mec='blue',markersize=6,linewidth=2) ''' figure(120) co_n_3=((10**(E_x4_3[2]-12)+10**(E_x4_3[4]-12))/10**(E_x4_3[3]-12)) c_o_3= 10**(E_x4_3[2]-E_x4_3[4]) plot(c_o_3,co_n_3,marker='o',mfc='None',mec='blue',markersize=6,linewidth=2) figure(130) c_n_3= 10**(E_x4_3[2]-E_x4_3[3]) n_o_3= 10**(E_x4_3[3]-E_x4_3[4]) plot(c_n_3,n_o_3,marker='o',mfc='None',mec='blue',markersize=6,linewidth=2) ''' #X_Fe1=X_Fe[2:] ''' for a in range(len(h)): for b in range(len(element)): if E[b][a]!=0.0: tmp_x.append(X_Fe[b][a]) tmp_z.append(z[b]) #c=%3.0 #if c==0.0: #print c #figure(1) figure(1) plot(tmp_z,tmp_x,marker='o',color='m',linestyle='dashed',markersize=5,alpha=0.4) #xlim((0,60)) #ylim((0,12.15)) #pp.savefig() tmp_x tmp_z tmp_x=[] tmp_z=[] print a axhline(0,color='black',linestyle='solid') for i in range(len(z)): text(z[i]-0.5,max(X_Fe[i].tolist())+0.09,element[i]) #close(200) figure(200) plot(z,X_S3_1,markersize=6.5,marker='h',linestyle='',color='k',alpha=0.6) plot(z,X_S3_2,markersize=6.5,marker='h',linestyle='',color='k',mfc='None',mec='k') plot(z,X_Fe3_1,markersize=6.5,marker='^',linestyle='',color='orange',alpha=0.6) plot(z,X_Fe3_2,markersize=6.5,marker='^',linestyle='',color='orange',mfc='None',mec='orange') #ylim(-4,4) #axhline(0,color='black',linestyle='solid') #for i in range(len(z)): #text(z[i]-0.5,max(X_Fe3_2[i].tolist())+0.09,element[i]) for i in range(len(z)): text(z[i]-0.5,X_S[i].tolist()+1.2,element[i]) figure(300) plot(z,X_Fe3_1,markersize=12,marker='',linestyle='--',color='b') ylim(-4,4) ''' ''' figure(1) plot(z,E_x,markersize=12,marker='*',linestyle='--',color='b',label='q=0.7') ylim(0,12) figure(2) plot(z,E_x,markersize=12,marker='*',linestyle='--',color='m') ylim(0,12) for i in range(len(z)): text(z[i]-0.5,E_x[i].tolist()+1.2,element[i]) '''