#For plotting the HRDs

import nugrid_set as set1

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


linestyle=['--','-','-.',':','-','-.','--',':','--','-','-.',':','--','-','-.',':']
symbs=['^','o','s','D','+','<','*','p','x','d','>','v']

#For set 1.3a

mesaset=set1.mesa_set(path+'/set1.3a/see_wind',multi_dir=['M3.000Z0.0060','M5.000Z0.0060','M15.0Z6.0e-03'])#,extra_label=20*[''])
linestyle=['--','-','-.',':','-','-.','--',':']
symbs=['o','s','D','+','<','*','p','x']
mesaset.set_plot_hrd(fig=0,symbs_1=symbs,linestyle=linestyle,end_model=20*[-1],markevery=10000,single_plot=True)



#For set 1.4a
mesaset=set1.mesa_set(path+'/set1.4a/see_wind',multi_dir=['M3.000Z0.0010','M5.000Z0.0010','M15.0Z1.0e-03'])#,extra_label=20*[''])
linestyle=['--','-','-.',':','-','-.','--',':']
symbs=['+','<','*','p','x']
mesaset.set_plot_hrd(fig=0,symbs_1=symbs,linestyle=linestyle,end_model=20*[-1],markevery=10000,single_plot=True)


#For set 1.5a..
mesaset=set1.mesa_set(path+'/set1.5a/see_wind',multi_dir=['M3.000Z0.0001','M5.000Z0.0001','M15.0Z1.0e-04'])#,extra_label=20*[''])

symbs=['p','x','^']
linestyle=['--','-','-.',':','-','-.','--',':']
mesaset.set_plot_hrd(fig=0,symbs_1=symbs,linestyle=linestyle,end_model=20*[-1],markevery=10000,single_plot=True)


plt.ylim(1.2,5)
plt.ylabel('log $L$ [$L_{\odot}$]',fontsize=18)
plt.minorticks_on()
plt.legend(loc=3,prop={'size':11})
plt.xlim(4.9,3.45)
plt.xlabel('log $T_{eff}$ [K]',fontsize=18,labelpad=2.5)
plt.savefig('hrd_Z.png')