# Dispersion d'un paquet d'onde constitué d'une superposition d'OPPH monochromatiques
# !!!Données à modifier!!!


import numpy as np
import matplotlib.pyplot as plt
from matplotlib.widgets import Slider, Button #outils pour faire des curseurs

N=500 #!!!Résolution
z=np.linspace(-0.5e-5,4e-5,N) #!!!axe des z (en mètre)
c=2e8
omega0=2*np.pi*c/(800e-9) #!!!pulsation centrale
omegaphi=omega0/2 #!!!pulsation plasma pour ma relation de dispersion


def OPPH(omega,z,t) :
    k=np.sqrt(omega**2-omegaphi**2)/c #!!! Relation de dispersion dans le plasma pour omegaphi<omega
    return np.cos(omega*t-k*z)

#OPPH considérés :
n=50 #!!!Nombre d'OPPH
deltaomega=omega0/3
omegalist=np.linspace(omega0-deltaomega,omega0+deltaomega,n)

def amplitudezt(z,t) :
    # Amplitude en un point z et un temps t
    amplitude=0
    for i in omegalist :
        amplitude+=OPPH(i,z,t)
    return amplitude


#Courbe amplitude=f(z)

t_init=0

def amplitudet(t):
    #Amplitude=f(z) pour t fixé
    ampl=[]
    for i in range(len(z)) :
        ampl+=[amplitudezt(z[i],t)]
    return ampl


fig=plt.figure()
ax=plt.subplot()
shape,=ax.plot(z,amplitudet(t_init))
ax.set_ylabel('amplitude (unités arbitraires')
ax.set_xlabel('z (en m)')

# Ajuste le graphique pour faire de la place pour le curseur
plt.subplots_adjust(bottom=0.25)

axi = plt.axes([0.25, 0.1, 0.65, 0.03])
i_slider = Slider(
    ax=axi,
    label='t (en s)',
    valmin=0,
    valmax=2e-13,
    valinit=0)

# Fonction appelée dès que la valeur de x change
def update(val):
    ampl=amplitudet(i_slider.val)
    shape.set_ydata(ampl)
    fig.canvas.draw_idle()

i_slider.on_changed(update)