#For plotting the HRDs

import nugrid_set as set1

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


xaxis='model'
xaxis='logtimerev'
xaxis='time'

savepath='all_logtimerev'

#set1.5a
mesaset=set1.mesa_set(path+'/set1.5a/see_wind',multi_dir=['M5.000Z0.0001'])
#mesaset.set_plot_kipp(xaxis=xaxis,startfirstTP=True,savepath='')
import mesa as ms
beginTP=mesaset.set_find_first_TP()[0]
historydata=ms.history_data(path+'/set1.5a/see_wind/M5.000Z0.0001/LOGS')
historydata.kip_cont(ifig=1122, modstart=beginTP, modstop=beginTP+10000, \
		 t0_model = beginTP, \
                 xlims=[0.,0.], ylims=[0.,2.],outfile='', \
                 xres=20000, yres=20000, ixaxis='age', \
                 mix_zones=20, burn_zones=20, plot_radius=False, \
                 engenPlus=False, engenMinus=False, \
                 landscape_plot=False, rad_lines=False, profiles=[], \
                 showfig=True, outlines=True, boundaries=True, \
                 c12_boundary=False, rasterise=False, yscale='1.')

plt.figure(1122)
plt.xlim(0,64000)
plt.title('')
plt.minorticks_on()
ax=plt.gca()
ax.set_yticks([0,4,8,12])
plt.yticks([0.978,0.982,0.986,0.99])
plt.ylim(0.978,0.992)
ax.set_yticklabels([0,4,8,12])
plt.xlabel('$t-t_0\ [10^4yr]$',fontsize=20,labelpad=1)
ax.set_xticklabels([0,1,2,3,4,5,6])
plt.tick_params(axis='both', which='major', labelsize=16)
plt.legend().set_visible(False)
plt.ylabel('$M-0.978\ [10^{-3}M_{\odot}]$',fontsize=20,labelpad=1)
plt.savefig('kipp_TPs_M5_0_Z0_0001.png')


'''
######################Fig. 2 /7Msun
mesaset=set1.mesa_set(path+'/set1.5a/see_wind',multi_dir=['M7.000Z0.0001'])
#mesaset.set_plot_kipp(xaxis=xaxis,startfirstTP=True,savepath='')
import mesa as ms
historydata=ms.history_data(path+'/set1.5a/see_wind/M7.000Z0.0001/LOGS')
historydata.kip_cont(ifig=1123, modstart=mesaset.set_find_first_TP(), modstop=-1, \
		 t0_model = mesaset.set_find_first_TP(), \
                 xlims=[0.,0.], ylims=[0.,0.],outfile='', \
                 xres=20000, yres=20000, ixaxis='age', \
                 mix_zones=20, burn_zones=20, plot_radius=False, \
                 engenPlus=False, engenMinus=False, \
                 landscape_plot=False, rad_lines=False, profiles=[], \
                 showfig=True, outlines=True, boundaries=True, \
                 c12_boundary=False, rasterise=False, yscale='1.')

plt.figure(1123)
plt.xlim(0,3500)
plt.title('')
ax=plt.gca()
plt.ylabel('$M-1.2714\ [10^{-4}M_{\odot}]$',fontsize=20,labelpad=1)
plt.ylim(1.2714,1.27225)
ax.set_yticks([1.2714,1.2716,1.2718,1.272,1.2722])
ax.set_yticklabels([0,2,4,6,8])
plt.xlabel('$t-t_0\ [10^2yr]$',fontsize=20,labelpad=1)
ax.set_xticklabels([5,10,15,20,25,30,35])
plt.tick_params(axis='both', which='major', labelsize=16)
plt.xlabel('$t-t_0\ [10^2yr]$',fontsize=20,labelpad=1)
plt.legend().set_visible(False)
plt.minorticks_on()
plt.savefig('kipp_TPs_M7_0_Z0_0001.png')
'''