import mesa as ms
import utils
import matplotlib.patches as mpatches

lines=utils.symbol_list('lines2')

s1=ms.star_log('/nfs/rpod2/athira/mesa/co+he_0.7/old_run3/LOGS_n14_0.30_8d51')
teff1=s1.get('log_Teff')
l1=s1.get('log_L')
age1=s1.get('star_age')
s2=ms.star_log('/nfs/rpod2/athira/mesa/co+he_0.6/LOGS')
teff2=s2.get('log_Teff')
l2=s2.get('log_L')
age2=s2.get('star_age')
s3=ms.star_log('/nfs/rpod2/athira/mesa/co+he_0.5/old_run3/LOGS_n14_0.30_8d51')
teff3=s3.get('log_Teff')
l3=s3.get('log_L')
age3=s3.get('star_age')

logs=[s1,s3]
specs=['log_Teff','log_L']
legends=['q=0.7','q=0.5']
labels=['log Teff', 'log L']


for num,spec in enumerate(specs):
    i=0
    close(num);figure(num)
    for ind,log in enumerate(logs):
        age=[]
        age = log.get('star_age')-log.get('star_age')[0]
        plot(age,log.get(spec),lines[i],label=legends[ind])
        xlabel('age(years)')
        ylabel(labels[num])
        i+=1

legend(loc=2)

rcb_teff=[]
rcb_model=[]
rcb_age=[]
rcb_teffu=[]
rcb_modelu=[]
rcb_ageu=[]


i=0
close(100)
for ind,log in enumerate(logs):
    age=[]
    teff=[]
    model=[]
    figure(100)
    print i
    log.plot(specs[0],specs[1],shape=lines[i],legend=legends[ind],labelx=labels[0],labely=labels[1],title='')
    i+=1
    age=log.get('star_age')
    teff=log.get(specs[0])
    model=log.get('model_number')
    diff1=[]
    for a,b in enumerate(teff):
        res1=abs(b-3.9030)
        diff1.append(res1)
    rcb_teff.append(teff[diff1.index(min(diff1))])
    rcb_age.append(age[diff1.index(min(diff1))])
    rcb_model.append(model[diff1.index(min(diff1))])

log.xlimrev()    
legend(loc=4)
xlim(5.0,3.5)
ylim(2.2,4.1)

xy=3.9,3.6
width=-0.3
height=0.4
p1 = mpatches.Rectangle(xy, width, height, facecolor="orange", edgecolor="red", alpha=0.65)
plt.draw()
plt.gca().add_patch(p1)
plt.draw()

text(x=3.8,y=3.5,s='RCB',fontsize=17,color='k')