# As an example the Intiial-to final mass relaation

#Maybe add the massive parts
import nugrid_set as set

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

##ini_m=[1.,1.65,2.,3.,4.,5.,6.,7.]
#final_set1_3a=[0.548184954627,0.61571046429,0.617577613445,0.709715109378,0.861117892135,0.953300209621,1.05304109559,1.1640650839]
#final_set1_5a=[0.591853396404,0.637413707117,0.664808679075,0.852151651778,0.90451904883,0.991660321039,1.12514953187,1.27248905653]
#plt.plot(ini_m,final_set1_3a,label="Z=0.006",marker='o',linestyle='--')
#plt.plot(ini_m,final_set1_5a,label="Z=0.0001",marker='s',linestyle='-')

#plt.xlabel('$M_{Initial} [M_{\odot}]$',size=23)
#plt.ylabel('$M_{Final} [M_{\odot}]$',size=23)
rundirs=['M1.00Z0.020','M1.65Z2.0e-02','M2.00Z2.0e-02','M3.00Z2.0e-02','M4.00Z2.0e-02','M5.00Z2.0e-02','M6.00Z0.020','M7.00Z0.020']
setdir=path+'/set1.2/ppd_wind'
setse=set.mppnp_set(rundir=setdir,multi_dir=rundirs,extra_label=15*[" "])
setse.initial_finall_mass_relation(marker='o',linestyle='--')

rundirs=['M1.00Z0.010','M1.65Z1.0e-02','M2.00Z1.0e-02','M3.00Z1.0e-02','M4.00Z1.0e-02','M5.00Z1.0e-02','M6.00Z0.010','M7.00Z0.010']
setdir=path+'/set1.1/ppd_wind'
setse=set.mppnp_set(rundir=setdir,multi_dir=rundirs,extra_label=15*[" "])
setse.initial_finall_mass_relation(marker='s',linestyle='-')


rundirs=['M1.000Z0.0060','M1.650Z0.0060','M2.000Z0.0060','M3.000Z0.0060','M4.000Z0.0060','M5.000Z0.0060','M6.000Z0.0060','M7.000Z0.0060']
setdir=path+'/set1.3a/ppd_wind/'
setse=set.mppnp_set(rundir=setdir,multi_dir=rundirs,extra_label=15*[" "])
setse.initial_finall_mass_relation(marker='D',linestyle='-.')

rundirs=['M1.000Z0.0010','M1.650Z0.0010','M2.000Z0.0010','M3.000Z0.0010','M4.000Z0.0010','M5.000Z0.0010','M6.000Z0.0010','M7.000Z0.0010']
setdir=path+'/set1.4a/ppd_wind/'
setse=set.mppnp_set(rundir=setdir,multi_dir=rundirs,extra_label=15*[" "])
setse.initial_finall_mass_relation(marker='x',linestyle=':')

rundirs=['M1.000Z0.0001','M1.650Z0.0001','M2.000Z0.0001','M3.000Z0.0001','M4.000Z0.0001','M5.000Z0.0001','M6.000Z0.0001','M7.000Z0.0001']
expdir=path+'/set1.5a/ppd_exp/'
setdir=path+'/set1.5a/ppd_wind/'

setse=set.mppnp_set(rundir=setdir,multi_dir=rundirs,extra_label=15*[" "])
setse.initial_finall_mass_relation(marker='^',linestyle='-')


plt.legend(loc=2,prop={'size':17})
plt.rcParams.update({'font.size': 20})
plt.rc('xtick', labelsize=20)
plt.rc('ytick', labelsize=20)
plt.savefig('ini_final_massrelation.pdf')