#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','M7.000Z0.0001'])
mesaset.set_plot_kipp(xaxis=xaxis,startfirstTP=True,savepath='')

#historydata=ms.history_data(path+'/set1.5a/see_wind/M5.000Z0.0001/LOGS')

########################Fig. 1/ 5Msun

plt.figure(0)
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',dpi=300) #changed for now

######################Fig. 2 /7Msun
plt.figure(1)
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',dpi=300) #changed for now