from numpy import *
import matplotlib
from matplotlib.pylab import *
import ppne
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]
z=[6,7,8,9,11,12,13,14,16,20,22,26,28,30,39,40,56,57]

#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]

X_Fe_RCrB=[]
'''
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)
'''
'''
X_Fe_1_1=array([ 1.13,  1.36,  1.7 ,  2.37,  0.83,  0.15, -0.09,  0.09,  0.08, -0.11,  0.37,  0.  ,  0.41,  1.74,  2.41,  2.08,  0.91,  0.63])

X_Fe_1_2=array([ 1.17,  1.24,  1.76,  2.6 ,  0.89,  0.17, -0.1 ,  0.12,  0.09,-0.12,  0.43,  0.  ,  0.47,  1.82,  2.59,  2.26,  1.04,  0.74])

X_Fe_2_1=array([ 0.91,  1.31,  2.02,  2.38,  1.  ,  0.02,  0.36,  0.07,  0.03, -0.12,  0.16,  0.  ,  0.25,  1.25,  0.52,  0.26,  0.53,  0.26])

X_Fe_2_2=array([ 0.93,  1.25,  2.03,  2.46,  1.01,  0.03,  0.35,  0.07,  0.03, -0.12,  0.17,  0.  ,  0.28,  1.32,  0.6 ,  0.31,  0.57,  0.29])
'''

X_Fe_1=array([ 1.11,  1.49,  1.48,  2.  ,  0.83,  0.  , -0.05, -0.03,  0.01, -0.07,  0.04,  0.  , -0.02,  0.25,  0.08,  0.02,  0.24,  0.02])


close(30)
figure(30)

plot(z,X_Fe_1,marker='*',mfc='m',markersize=6,linestyle='None',mec='m')

'''
plot(z,X_Fe_2_1,marker='o',mfc='b',markersize=6,linestyle='None',mec='b')
plot(z,X_Fe_2_2,marker='o',markersize=6,linestyle='None',mec='b',mfc='None',mew=1.5)

plot(z,X_Fe_1_1,marker='o',mfc='m',markersize=6,linestyle='None',mec='m')
plot(z,X_Fe_1_2,marker='o',markersize=6,linestyle='None',mec='m',mfc='None',mew=1.5)
'''

for a in range(len(X_Fe_lims)):
    figure(30)
    xy=z[a]-0.4,X_Fe_lims[a][0]
    width=0.8
    height=X_Fe_lims[a][1]-X_Fe_lims[a][0]
    print height
    p1 = mpatches.Rectangle(xy, width, height, facecolor="black", edgecolor="black", alpha=0.3)
    plt.draw()
    plt.gca().add_patch(p1)
    plt.draw()

for i in range(len(z)):
    text(z[i]-0.8,X_Fe_lims[i][1]+0.165,element[i],fontsize=11,weight='bold')

'''
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]
teff=[5300,5400,4000,5000,5000,5400,6250,6500,6750,7000,7250,7500]
o_numg=[0.3,16,16,20,0.0,0.0,0.0,0.0,0.0,0.0,0.0,0.0]
o_numc=[1,4,4,6,8,12,0.0,0.0,0.0,0.0,0.0,0.0]
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]
markers=['^','^','o','s']
colors=['k','k','k','k']
labels=['$^{16}O/^{18}O$',None,'${\epsilon}(F19)$','$^{12}C/^{13}C$']

close(10)
figure(10)
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])
'''