# -*- coding: utf-8 -*-

"""
Courbe intéractive : exemplde de l'influence des différents paramètres d'un circuit RLC série sur la bande passante.

On trace u = f(omega) en donnant une valeur initiale de R, C et L et on peut suivre l'évolution de la courbe en "bougeant" à l'aide d'un curseur les valeurs de R, C et L dans des domaines finis.

A changer : Le nom et le nombre de variables ; leur domaine de variation et leur valeurs initiales ; la formule de la courbe à tracer
"""

## Importation des bibliotheques

import matplotlib.pyplot as plt
import numpy as np
import widgets
import scipy.constants as constants
from matplotlib import rc
## Definition des parametres fixes initiaux

eo = 5                      


##Paramètres
parameters = {'R' : widgets.IntSlider(value=35, description='$R (ohm)$', min=10, max=10000),
              'L' : widgets.IntSlider(value=11 , description='$L (mH)$', min=10, max=100),
              'C' : widgets.FloatSlider(value=0.05, description='$C (microF)$', min=0.01, max=0.2)}


## Creation de la fenetre

fig=plt.figure(figsize=(12,6))
ax = fig.add_axes([0.2, 0.2, 0.35, 0.7])
ax.set_xlim(1000,50000)
ax.set_ylim(0,90)
plt.xlabel("pulsation (rad/s)")                     # titre de l'axe des abscisses
plt.ylabel("U/E0")                                  # titre de l'axe des ordonnées
plt.title("Courbe de la résonance en tension d'un RLC série.")
plt.legend()
plt.grid(True)                                          # quadrille le graphique


omega = np.arange(1e3, 1e5, 1e1)                    # omega allant de 1000 Ã  100000 par pas de 10

                
##Défintion des fonctions

def Fonction_transfert(omega,R,L,C):
    w0 = 1/np.sqrt(L*1E-3 *C*1E-6) 
    Q = 1/R*np.sqrt(L*1E-3 /(C*1E-6))
    return Q, eo/(np.sqrt((1-(omega/w0)**2)**2+(((omega/w0)/Q))**2))



def plot_data(R,L,C):
    lines.set_data(omega,Fonction_transfert(omega, R, L, C)[1])
    texts.set_text('Facteur de qualité : Q = {:.3e}'.format(Fonction_transfert(omega, R, L, C)[0]))
    fig.canvas.draw_idle()


lines = {}
lines, = ax.plot([], [],color='red',lw=2)
texts=  fig.text(0.32, 0.8, '$x=$', fontsize=10,ha='center')
ax.legend()

##Définition d'un slider

param_widgets = widgets.make_param_widgets(parameters, plot_data, slider_box=[0.65, 0.15, 0.30, 0.15])

## Définition d'un bouton reset

reset_button = widgets.make_reset_button(param_widgets,box=[0.32, 0.05, 0.10, 0.05])

## Affichage du tout


if __name__=='__main__':
    plt.show()