#Abundances of elements and comparison with abundances from Asplund 1997        

from numpy import *
import matplotlib
from matplotlib.pylab import *
import asciidata
import ppn

element=['H','C','N','O','F','Na','Mg','Al','Si','S','Ca','Ti','Fe','Ni','Zn','Y','Zr','Ba','La']

star=['V3795 Sgr','VZ Sgr','V Cr A','V 854 Cen']

z=[1,6,7,8,9,11,12,13,14,16,20,22,26,28,30,39,40,56,57]

E=empty([len(star),len(element)])
X_S=empty([len(star),len(element)])

h=[4.1,6.2,8.7,9.9]

c=[8.8,8.8,8.8,9.6]

n=[8.0,7.6,8.0,7.8]

o=[7.5,8.7,7.8,8.9]

f=[6.7,6.4,6.5,5.7]

na=[5.9,5.8,5.7,6.4]

mg=[6.1,0.0,6.6,6.2]

al=[5.6,5.4,5.4,5.7]

si=[7.5,7.3,7.6,7.0]

s=[7.4,6.7,7.2,6.4]

ca=[5.3,5.0,5.2,5.1]

ti=[3.5,0.0,3.3,4.1]

fe=[5.6,5.8,5.5,6.0]

ni=[5.8,5.2,4.9,5.9]

zn=[4.1,3.9,3.9,4.4]

y=[1.3,2.8,1.1,2.2]

zr=[0.0,2.6,1.4,2.1]

ba=[0.9,1.4,0.3,1.3]

la=[0.5,0.0,0.0,0.4]


E=empty([len(element),len(h)])
X_S=empty([len(element),len(h)])


E[0]=h
E[1]=c
E[2]=n
E[3]=o
E[4]=f
E[5]=na
E[6]=mg
E[7]=al
E[8]=si
E[9]=s
E[10]=ca
E[11]=ti
E[12]=fe
E[13]=ni
E[14]=zn
E[15]=y
E[16]=zr
E[17]=ba
E[18]=la

E_s=[12.0,8.6,8.1,8.0,4.6,6.3,7.6,6.5,7.6,7.2,6.4,5.0,7.5,6.3,4.6,2.2,2.6,2.1,1.2]

def x_s(e,i):
    return array(e)-array(s)-E_s[i]+7.2

X_S[0]=x_s(h,0)
X_S[1]=x_s(c,1)
X_S[2]=x_s(n,2)
X_S[3]=x_s(o,3)
X_S[4]=x_s(f,4)
X_S[5]=x_s(na,5)
X_S[6]=x_s(mg,6)
X_S[7]=x_s(al,7)
X_S[8]=x_s(si,8)
X_S[9]=x_s(s,9)
X_S[10]=x_s(ca,10)
X_S[11]=x_s(ti,11)
X_S[12]=x_s(fe,12)
X_S[13]=x_s(ni,13)
X_S[14]=x_s(zn,14)
X_S[15]=x_s(y,15)
X_S[16]=x_s(zr,16)
X_S[17]=x_s(ba,17)
X_S[18]=x_s(la,18)


for a in range(len(h)):
    for b in range(len(element)):
        if E[b][a]!=0.0:
            tmp_x.append(X_S[b][a])
            tmp_z.append(z[b])
    figure(1)
    plot(tmp_z,tmp_x,marker='s',color='b',linestyle='dashed',markersize=5,linestyle='None',mec='b',alpha=0.7,mfc='None')
    tmp_x
    tmp_z
    tmp_x=[]
    tmp_z=[]
    print a

axhline(0,color='black',linestyle='solid')
title('RCB minority stars and AGB model (12th TP) 49901, 2 Ms')
ylabel('[X/S]')
xlabel('Z')
for i in range(len(z)):
    text(z[i]-0.5,max(X_S[i].tolist())+0.09,element[i])



'''
from matplotlib.backends.backend_pdf import PdfPages

pp = PdfPages('min_abun1.pdf')

figure()
for a in range(len(star)):
    for b in range(len(z)):
        if E[a][b]!=0.0:
            tmp_E.append(E[a][b])
            tmp_z.append(z[b])
    scatter(z,E_s,marker='^',color='black',label='Solar')
    plot(tmp_z,tmp_E,marker='o',linestyle='--',label=star[a])
    legend()
    xlim((0,60))
    ylim((0,12.15))
    pp.savefig()
    tmp_E=[]
    tmp_z=[]
    print a

pp.close()


tmp_x=[]
tmp_z=[]
c=0

#pp = PdfPages('min_abun2.pdf')

figure(1)
for a in range(len(star)):
    for b in range(len(z)):
        if E[a][b]!=0.0:
            tmp_x.append(X_S[a][b])
            tmp_z.append(z[b])
    plot(tmp_z,tmp_x,markersize=10,linestyle='--',marker='x',color='y')
    legend()
    #xlim((0,60))
    #ylim((0,12.15))
    axhline(0,color='black',linestyle='solid')
    #pp.savefig()
    tmp_x=[]
    tmp_z=[]
    #print a

#pp.close()
'''