# -*- coding: utf-8 -*-
"""
Éditeur de Spyder

Ceci est un script temporaire.
"""

import numpy as np		#pour certains calculs voir module cmath
import matplotlib.pyplot as plt
N=8192
fe=8192
tau=1/fe
f1=440.5
f2=3100.5

t=np.linspace(0,(N-1)*tau,N) 	#le temps en abscisse #axe horizontal

lapur=np.cos(2*np.pi*f1*t)
labruit=np.cos(2*np.pi*f1*t)+3.0*np.cos(2*np.pi*f2*t) 


#plot signal non bruité
plt.figure()
plt.plot(lapur,'o-r')
plt.grid()
plt.show()	

#plot signal bruité
plt.figure()
plt.plot(labruit,'o-r')
plt.grid()
plt.show()	

#fft du signal bruité
fourier=np.fft.fft(labruit)

#spectre du signal bruité
plt.figure()
plt.plot(abs(fourier),'o-r')
plt.grid()
plt.show()	

#mise à 0 du parasite HF
#fourier[3100]=0.0
#fourier[8192-3100]=0.0
for k in range(2100,4000):
    fourier[k]=0.0
for k in range((N-4000),(N-2100)):
    fourier[k]=0.0

    

#nouveau spectre
plt.figure()
plt.plot(abs(fourier),'o-r')
plt.grid()
plt.show()	

#retour dans le domaine temporel et plot parties réelle et imaginaire
retour=np.fft.ifft(fourier)
plt.figure()
plt.plot(np.real(retour),'o-r')
plt.grid()
plt.show()	

plt.figure()
plt.plot(np.imag(retour),'o-r')
plt.grid()
plt.show()