"""
Auteur : Clément de la Salle
"""


from pylab import *
from matplotlib import animation

dt = 0.01
w=2*pi
k=2*pi
kk=0.8*pi
delta=1

zed=linspace(0,10,1000)
zedneg=linspace(-5,0,500)

fig=figure()

subplot()

title(r"Cas : $\omega>\omega_p$")
xlabel('$z$')
ylabel(r'$E_x/E_0$')
ylim([-1.5,1.5])
xlim([-5,10])
line1,=plot([],[])
line2,=plot([],[])

# fonction à définir quand blit=True
# crée l'arrière de l'animation qui sera présent sur chaque image
def init():
    line1.set_data([],[])
    line2.set_data([],[])
    matplotlib.pyplot.xticks( [0, 2, 4, 6, 8, 10],
            [r'0',r'$2\delta$',r'$4\delta$',r'$6\delta$',r'$8\delta$',r'$10\delta$'])
    plt.plot([0,0],[-2,2],color='k')
    return line1,line2,

def animate(i): 
    t = i * dt
    E1=[cos(w*t-kk*z) for z in zed]
    E2=[cos(w*t-k*z) for z in zedneg]
    line1.set_data(zed,E1)
    line2.set_data(zedneg,E2)
    return line1,line2,
 
ani = animation.FuncAnimation(fig, animate, init_func=init, frames=2000, blit=True, interval=20, repeat=False)

subplot()

show()