#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
print path+'/set1.3a/see_wind'
mesaset=set1.mesa_set(path+'/set1.3a/see_wind',multi_dir=['M3.000Z0.0060','M5.000Z0.0060'])#,extra_label=20*[''])
linestyle=['--','-','-.',':','-','-.','--',':']
symbs=['o','s','D','+','<','*','p','x']
mesaset.set_plot_hrd(fig=0,symbs_1=symbs,linestyle=linestyle,start_model=[800,650],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'])#,extra_label=20*[''])
linestyle=['--','-','-.',':','-','-.','--',':']
symbs=['+','<','*','p','x']
mesaset.set_plot_hrd(fig=0,symbs_1=symbs,linestyle=linestyle,start_model=[750,600],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'])#,extra_label=20*[''])

symbs=['p','x','^']
linestyle=['--','-','-.',':','-','-.','--',':']
mesaset.set_plot_hrd(fig=0,symbs_1=symbs,linestyle=linestyle,start_model=[720,580],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.xlim(4.445,3.45)
plt.ylim(1.35,5)
plt.ylim(1.9,5)
plt.savefig('hrd_Z_agb.pdf')