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

# Programme de simulation d'un circuit RLC en oscillations libres
# Dominique Lefebvre pour TangenteX.com
# 8 janvier 2015
#

# importation des librairies
from numpy import array,arange,pi
from scipy.integrate import odeint 
import matplotlib.pyplot as plt
       
# définition des constantes
Omega0 = 5      # Pulsation propre du circuit  

# fonction de définition du circuit RLC
def RLC(y,t):
    u,u_dot = y
    return [u_dot,-K*u_dot - Omega0**2*u]   

# saisie du coefficent d'amortissement

K = float(input("Coefficient d'amortissement : "))
# conditions initiales
u0 = 1      
du0 = 0
C0 = array([u0,du0])

# définition du vecteur temps de l'expérience
t0 = 0.0
tmax = 10*pi
pastemps = 0.01
time = arange(t0, tmax, pastemps)

# intégration du système
y = odeint(RLC,C0,time)

# affichage de l'évolution de la tension aux bornes de C
plt.figure()
plt.xlim(t0,tmax)
plt.ylim(y[:,0].min(),y[:,0].max())
plt.plot(time, y[:,0])
plt.title('Oscillations libres - Amortissement = ' + str(K))
plt.xlabel('Temps')
plt.ylabel('Amplitude')
plt.show()