import mesa as ms
import utils as u

cb = u.colourblind

# read the data:

models=['/Volumes/jazz/work/data/iprocess/new_fs/Z_study/M8.20Z0.02/',
    '/Volumes/jazz/work/data/iprocess/new_fs/Z_study/M8.15Z0.01/',
    '/Volumes/jazz/work/data/iprocess/new_fs/Z_study/M7.0Z0.001/',
    '/Volumes/jazz/work/data/iprocess/new_fs/Z_study/M7.0Z0.0001/',
    '/Volumes/jazz/work/data/iprocess/new_fs/Z_study/M7.2Z0.00001/']

s=[]

for i in range(len(models)):
    sx=ms.star_log(models[i]+'LOGS')
    idx=sx.find_first_TP()
    t0=sx.get('star_age')[idx]
    s.append([sx.get('star_age')-t0,sx.get('h1_boundary_mass')])


markers=['o','s','d','*','v']
labels=['Z2.e-2','Z1.e-2','Z1.e-3','Z1.e-4','Z1.e-5']
me=5000

for i in range(len(s)):
    plot(s[i][0],s[i][1],color=cb(i),
         marker=markers[i],label=labels[i],
         markevery=me)

ylabel('\mathrm{Mass}\,/\,\mathrm{M}_\odot')
xlabel('\mathrm{time}\,/\,\mathrm{yr}$')
xlim(0.,20000.)