from numpy import *
import matplotlib
from matplotlib.pylab import *
import ppn
import utils
import mppnp as mp
import matplotlib.patches as mpatches


element=['C','N','O','F','Na','Mg','Al','Si','S','Ca','Ti','Fe','Ni','Zn','Y','Zr','Ba','La']
z=[6,7,8,9,11,12,13,14,16,20,22,26,28,30,39,40,56,57]

E_lims=[[8.6,9.2],[8.3,9.1],[7.5,9.0],[5.6,7.2],[5.6,6.8],[5.7,6.9],[5.2,6.3],[6.7,7.4],[6.2,7.7],[5.0,5.8],[3.7,4.2],[6.1,6.9],[5.4,6.2],[3.6,4.8],[1.3,3.1],[1.8,2.6],[0.3,2.6],[0.8,2.2]]

X_Fe_lims=[[ 1. ,  1.6],[ 1. ,  1.8],[ 0.1,  2.1],[ 1.7,  3.8],[ 0.7,  1.2],[-0.7,  0.5],[ 0.1,  1. ],[ 0.2,  0.8],[ 0.3,  1.2],[-0.3,  0.4],[-0.4,  0.3],[ 0. ,  0. ],[ 0.2,  0.8],[ 0.4,  1. ],[ 0.3,  1.6],[-0.1,  1. ],[-0.4,  1.3],[ 0.6,  1.8]]
 
E_s=[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]

#U Aqr has C,N,O,Na,S abundances normalized to E(Fe)=7.2 and E(C)=9.52 at 6000K (G-Hernandez 2010). The normaliztions are according to Asplund 1997.
#U Aqr has Zn,Y,Zr,Ba , [X/Fe] ratios from observations (Vanture 1999)
#U Aqr has C12/C13 = 110-120 at 5400 K and E(Fe)=6.5 and E(C)=9.2

X_Fe_uaqr=[1.2,1.6,1.6,0.0,7.8,0.0,5.4,0.0,0.0,0.0,6.5,0.0,0.0,7.2,0.0,0.7,3.3,3.0,2.1,0.0]

'''
tmp_E=[]
tmp_z=[]
tmp_x=[]

for a in range(len(element)):
    if X_Fe_uaqr[a]!=0.0:
        tmp_x.append(X_Fe_uaqr[a])
        tmp_z.append(z[a])

close(1)
figure(1)
plot(tmp_z,tmp_x,marker='o',color='green',linestyle='dashed',markersize=7,alpha=1)

for a in range(len(X_Fe_lims)):
    figure(1)
    xy=z[a]-0.4,X_Fe_lims[a][0]
    width=0.8
    height=X_Fe_lims[a][1]
    p1 = mpatches.Rectangle(xy, width, height, facecolor="yellow", edgecolor="red", alpha=0.65)
    plt.draw()
    plt.gca().add_patch(p1)
    plt.draw()
'''
stars1=['WX CrA','S Aps','SV Sge','ES Aql','Z Umi','U Aqr']
stars2=['GU Sgr','V482 Cyg','R CrB','UX Ant','RY Sgr','UW Cen']
stars=['WX CrA','S Aps','SV Sge','ES Aql','Z Umi','U Aqr','GU Sgr','V482 Cyg','R CrB','UX Ant','RY Sgr','UW Cen']
teff1=[5300,5400,4000,5000,5000,5400]
teff2=[6250,6500,6500,6750,7000,7250,7500]
teff3=[]
teff4=[]
#teff=[5300,5400,4000,5000,5000,5400,6250,6500,6750,7000,7250,7500]
o_numg=[0.3,16,16,20]
o_numc=[1,4,4,6,8,12]
f19=[0.0,0.0,0.0,0.0,0.0,0.0,7.2,6.6,6.9,6.2,6.95,7.1]
c_num=[0.0,0.0,0.0,0.0,0.0,110,14,100,40,14,0.0,0.0]
specs=[o_numg,o_numc,f19,c_num]
teff=[teff1,teff1,teff2,teff3]
markers=['^','^','d','s']
colors=['b','b','r','c']
labels=['$^{16}O/^{18}O$',None,'${\epsilon}(F19)$','$^{12}C/^{13}C$']

close(1)
for a in range(len(specs)):
    tmp_teff=[]
    tmp_abu=[]
    for b in range(len(teff)):
        if specs[a][b]!=0.0:
            tmp_abu.append(specs[a][b])
            tmp_teff.append(teff[b])
    semilogy(tmp_teff,tmp_abu,marker=markers[a],markersize=7,linestyle='None',color=colors[a],label=labels[a])