import mppnp as mp
import utils
import mesa as ms



logdir='/rpod2/athira/mesa/case1/LOGS'
#logdir='/rpod2/athira/mesa/case1/LOGS_6'

#logdir='/rpod2/athira/mesa/co+he_0.5/LOGS'

p=ms.mesa_profile(logdir,100,num_type='model')
cycles_1=p.log_ind.keys()
cycles_1.sort()
m=mp.se('/rpod2/athira/mppnp/case1/H5_out')
#m=mp.se('/rpod2/athira/mppnp/RCB_0.7_1e-02/H5_out')

#index=[2000]
matplotlib.rc('xtick',labelsize=15,color='k')
matplotlib.rc('ytick',labelsize=12)
matplotlib.rc('ytick.major',pad=15)
matplotlib.rc('axes',labelsize=15,labelcolor='b')

lines=utils.symbol_list('lines2')
specs1=['H-1','He-4','C-12','C-13','N-14','N-15','O-16','O-18','F-19']
specs2=['Si-28','S-32','Fe-56','Ni-58','Zn-64','Y-89','Ba-138','La-139']

neut_1=[]
mass_1=[]
neut_den_1=[]
rho_1=[]

c12_c13_1=[]
o16_o18_1=[]
age_1=[]
E_n_1=[]
E_c_1=[]
E_o_1=[]
E_f_1=[]
E_fe_1=[]
c_n_1=[]
n_o_1=[]
f_fe_1=[]
ratio_1=[]
c_o_1=[]
#index_1=arange(1,3500,50)
index_1=arange(1,2300,50)
#index_1=[2000]
#index_1=[cycles_1[-1]]

for ind in index_1:
    c12=m.get(ind,'C-12')
    c13=m.get(ind,'C-13')
    n14=m.get(ind,'N-14')
    n15=m.get(ind,'N-15')
    o16=m.get(ind,'O-16')
    o17=m.get(ind,'O-17')
    o18=m.get(ind,'O-18')
    c12_c13_1.append((c12[-1]/c13[-1])*(13.0/12.0))
    o16_o18_1.append((o16[-1]/o18[-1])*(18.0/16.0))
    n_s=log10((n14[-1]/14.0)+(n15[-1]/15.0))+12.0
    c_s=log10((c12[-1]/12.0)+(c13[-1]/13.0))+12.0
    o_s=log10((o16[-1]/16.0)+(o17[-1]/17.0)+(o18[-1]/18.0))+12.0
    c_n_temp=10**(c_s-n_s)
    n_o_temp=10**(n_s-o_s)
    c_o_temp=10**(c_s-o_s)
    c_n_1.append(c_n_temp)
    n_o_1.append(n_o_temp)
    c_o_1.append(c_o_temp)
    ratio_1.append(n_o_temp/c_n_temp)

close(210)
ifig=figure(210,figsize=(8.25,11.6))
ax1=ifig.add_subplot(111)
ax1.plot(index_1,ratio_1,linestyle='-',color='m')
ax1.plot(index_1,n_o_1,linestyle='-',color='r')
ax1.plot(index_1,c_n_1,linestyle='-',color='b')
axvline(y=2,linestyle='--',color='k')

'''
for ind in index_1:
    c12=m.get(ind,'C-12')
    c13=m.get(ind,'C-13')
    n14=m.get(ind,'N-14')
    n15=m.get(ind,'N-15')
    o16=m.get(ind,'O-16')
    o17=m.get(ind,'O-17')
    o18=m.get(ind,'O-18')
    f19=m.get(ind,'F-19')
    fe54=m.get(ind,'Fe-54')
    fe56=m.get(ind,'Fe-56')
    fe57=m.get(ind,'Fe-57')
    fe58=m.get(ind,'Fe-58')
    c12_c13_1.append((c12[-1]/c13[-1])*(13.0/12.0))
    o16_o18_1.append((o16[-1]/o18[-1])*(18.0/16.0))
    n_s=log10((n14[-1]/14.0)+(n15[-1]/15.0))+12.0
    c_s=log10((c12[-1]/12.0)+(c13[-1]/13.0))+12.0
    o_s=log10((o16[-1]/16.0)+(o17[-1]/17.0)+(o18[-1]/18.0))+12.0
    f_s=log10(f19[-1]/19.0)+12.0
    c_n_temp=10**(c_s-n_s)
    n_o_temp=10**(n_s-o_s)
    c_o_temp=10**(c_s-o_s)
    fe_s = log10((fe54[-1]/54.0)+(fe56[-1]/56.0)+(fe57[-1]/57.0)+(fe58[-1]/58.0))+12.0
    E_f_1.append(f_s)
    E_fe_1.append(fe_s)
    E_n_1.append(n_s)
    E_c_1.append(c_s)
    E_o_1.append(o_s)
    E_fe_1.append(fe_s)
    c_n_1.append(c_s-n_s)
    n_o_1.append(n_s-o_s)
    f_fe_temp=f_s-fe_s-4.6+7.5
    #f_fe_temp=log10((f19[-1]/19.0)/((fe54[-1]/54.0)+(fe56[-1]/56.0)+(fe57[-1]/57.0)+(fe58[-1]/58.0)))+2.889
    f_fe_1.append(f_fe_temp)
    #time=m.get(ind,'age')/(3.17*10**7)-m.get(0,'age')/(3.17*10**7)
    #age_1.append(time)
'''

'''
for ind in index_1:
    mass_1=[]
    neut_1=[]
    rho_1=[]
    close(ind+10)
    ifig=figure(ind+10,figsize=(17,12))
    i=0
    for spec in specs1:
        i += 1
        m.plot('mass',spec,fname=ind,shape=lines[i], legend=spec,logy=True)
        ylim(-8,0)
        xlim(0.0,0.90)
    legend(loc=3)
    savefig('mppnp_profile1_'+'0'+str(ind)+'.png')
    close(ifig)
    close(ind+110);
    ifig=figure(ind+110,figsize=(20,15))
    ax1=ifig.add_subplot(111)
    i=0
    neut_1=m.get(ind,'Neutron-1')
    mass_1=m.get(ind,'mass')
    rho_1 = m.get(ind,'rho')
    for a in range(len(neut_1)):
        res=(neut_1[a]*6*10**23*rho_1[a])
        neut_den_1.append(res)
    for spec in specs2:
        i += 1
        m.plot('mass',spec,fname=ind,shape=lines[i], legend=spec,logy=True)
        #m.plot('mass',neut_den_1,fname=ind,shape=lines[i], legend=spec,logy=True)
        legend(loc=3)
        ylim(-8,0)
        #ax2=ax1.twinx()
        #ax2.semilogy(mass_1,neut_den_1,'k--',linewidth=2)
        xlim(0.0,0.90)
        legend(loc=2)
    savefig('mppnp_profile2_'+'0'+str(ind)+'.png')
    close(ifig)

'''