#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=['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,start_model=[800],end_model=20*[-1],markevery=10000,single_plot=True)



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


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

symbs=['p','x','^']
linestyle=['--','-','-.',':','-','-.','--',':']
mesaset.set_plot_hrd(fig=0,symbs_1=symbs,linestyle=linestyle,start_model=[750],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.legend(loc=2,prop={'size':11})
plt.ylim(3.65,5.2)
plt.xlim(4.65,3.5)
plt.ylim(3.5,5.5)

plt.ylim(4.25,5.2)
plt.savefig('hrd_Z_massive.pdf')