import numpy as np
from matplotlib import pyplot as plt
#%% Définissons les valeurs
Q=10
w0=1#On normalise tout par w0
N=500# nombre d epoints par graphique, à prendre élevé si on veut dériver
facteur=0.01#NV
"""On veut juste les valeurs asymptotiques et la dérivée, on normalise par w0 et les préfacteurs, 
qui reviendront dans les titres. Attention, si vous redéfinissez les grandeurs, 
relancez la cellule suivante avant de tracer les graphiques pour actualiser les valeurs :)"""
#%% Définition des tableaux de valeurs
omegas=np.linspace(0,2*w0,N)# Bornes partant de w0 cntré sur w0
def ni(w):
    return 1.5*facteur*(w/(w0*Q))/((1-(w/w0)**2)**2+(w/(w0*Q))**2)
def nr(w):
    return 1.5*facteur*(1-(w/w0)**2)/((1-(w/w0)**2)**2+(w/(w0*Q))**2)
#%% Tracé
val_nr=nr(omegas)
val_ni=ni(omegas)
plt.figure(figsize=(12,9))
plt.plot(omegas,val_nr,label="n'-1")
plt.plot(omegas,val_ni,label='n"')
plt.xlabel('w/w0',fontsize=15)
plt.ylabel('Indice optique',fontsize=15)
plt.legend(fontsize=12)
plt.title('Indice optique pour Q='+str(Q)+' et 3NV/2='+str(3*facteur/2))
plt.show()