#For plotting the HRDs
import nugrid_set as set1

props = dict(boxstyle='square', facecolor='w', alpha=1)
symbs=['o','s','D','+','<','*','p','x']
end_model=20*[-1]
linestyle=['--','-','-.',':','-','-.','--',':']
color=['r','m','g','k','b']

#dir to save
savefig='all_model'
xaxis='cycles'
#xaxis='mass'

end_model=20*[-1] #29900,33500,48200,25300,51000,56000,57400,77100]
symbs=['o','s','D','+','<','^','p','x','>','>']
linestyle=['--','-','-.',':','-','-.','--',':']

'''

#for set1.2
mesaset1=set1.mesa_set('/astro/critter/critter1/Results/DATA/Set1_extension/set1.2/see_wind',multi_dir=['M1.65Z2.0e-02','M2.00Z2.0e-02','M3.00Z2.0e-02','M4.00Z2.0e-02','M5.00Z2.0e-02']) #,'M6.00Z2.0e-02','M7.00Z2.0e-02'])
#mesaset1.set_plot_kipp_CO(savefig='')
mesaset1.set_plot_CO(fig=3123,startfirstTP=False,withoutZlabel=True,xtime=False,symbs_1=symbs,color=color,linestyle=linestyle,markevery=10000,end_model=end_model)
plt.xlim(5,0.5)
plt.xlabel( '$M/M_{\odot}$')
plt.ylabel('C/O ratio', fontsize=23)
ax=plt.gca()
ax.text(0.85, 0.85, 'set1.2', transform=ax.transAxes, fontsize=16,verticalalignment='top', bbox=props)


#set1.1

mesaset2=set1.mesa_set('/astro/critter/critter1/Results/DATA/Set1_extension/set1.1/see_wind',multi_dir=['M1.65Z1.0e-02','M2.00Z1.0e-02','M3.00Z1.0e-02','M4.00Z1.0e-02','M5.00Z1.0e-02']) #,'M6.00Z1.0e-02','M7.00Z1.0e-02'])
#plt.figure(0);plt.xlim(7,0.5)
#mesaset2.set_plot_kipp_CO(savefig='')
mesaset2.set_plot_CO(fig=3112,startfirstTP=False,withoutZlabel=True,xtime=False,symbs_1=symbs,color=color,linestyle=linestyle,markevery=10000,end_model=end_model)


plt.xlim(5,0.5)
plt.xlabel( '$M/M_{\odot}$')
plt.ylabel('C/O ratio', fontsize=23)
ax=plt.gca()
ax.text(0.85, 0.85, 'set1.1', transform=ax.transAxes, fontsize=16,verticalalignment='top', bbox=props)



#For set 1.3a
mesaset=set1.mesa_set('/nfs/rpod3/critter/Results/DATA/Set1_extension/set1.3a/see_wind',multi_dir=['M1.000Z0.0060','M1.650Z0.0060','M2.000Z0.0060','M3.000Z0.0060','M4.000Z0.0060','M5.000Z0.0060','M6.000Z0.0060','M7.000Z0.0060'])#,extra_label=20*[''])
end_model=20*[-1]#[29i900,33500,48200,25300,51000,56000,57400,77100]
#mesaset.set_plot_hrd(symbs_1=symbs,linestyle=linestyle,end_model=end_model,markevery=10000)
#plt.figure(0);plt.xlim(7,0.5)
mesaset.set_plot_kipp_CO(savefig='all_co_kip')

#For set 1.4a
mesaset=set1.mesa_set('/nfs/rpod3/critter/Results/DATA/Set1_extension/set1.4a/see_wind',multi_dir=['M1.000Z0.0010','M1.650Z0.0010','M2.000Z0.0010','M3.000Z0.0010','M4.000Z0.0010','M5.000Z0.0010','M6.000Z0.0010','M7.000Z0.0010'])#,extra_label=20*[''])
end_model=[29900,33500,48200,25300,51000,56000,57400,77100]
symbs=['o','s','D','+','<','*','p','x']
end_model=20*[-1]
linestyle=['--','-','-.',':','-','-.','--',':']
mesaset.set_plot_kipp_CO(savefig='all_co_kip')

'''
#set1.4a
#plt.figure(0);plt.xlim(7,0.5)

path='/apod2/NuGrid/data/set1ext/'

###for set1.5a
mesaset=set1.mesa_set(path+'/set1.5a/see_wind',multi_dir=['M1.000Z0.0001','M1.650Z0.0001','M2.000Z0.0001','M3.000Z0.0001','M4.000Z0.0001','M5.000Z0.0001','M6.000Z0.0001','M7.000Z0.0001'])#,extra_label=20*[''])
#end_model=[29900,33500,48200,25300,51000,56000,57400,77100]
end_model=20*[-1]
symbs=['o','s','D','+','<','*','p','x']
linestyle=['--','-','-.',':','-','-.','--',':']
color=['c','m','g','k','b','r','gray','brown']
#mesaset.set_plot_kipp_CO(savefig='all_co_kip')
mesaset.set_plot_CO(fig=3123,startfirstTP=False,withoutZlabel=True,xtime=False,symbs_1=symbs,color=color,linestyle=linestyle,markevery=10000,end_model=end_model,singlefigures=False)