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

import scipy.io.wavfile as impwav
import matplotlib.pyplot as plt
import numpy as np

lecture=impwav.read('/Users/yvestalbourdel/Desktop/PT Benjam 2021/IPT/IMSP 0/B3.wav',mmap=False)
fe=lecture[0]
valeurs=lecture[1]
N=len(valeurs)
print ("la fréquence d'échantillonnage du fichier WAV est de ",fe/1000," kHz")
print ("soit une durée d'enregistrement de ",round((N-1)/fe,1)," secondes")

moyenne=np.mean(valeurs)
ecarttype=np.std(valeurs,0,ddof=1)

plt.figure(1)
plt.plot(valeurs,'k',alpha=1,linewidth=0.5,label="bruit étudié")
plt.legend(loc='lower right')
plt.figure(2)
plt.subplot(211)
plt.hist(valeurs,bins=500,normed=True,color='r',alpha=0.2,label="distribution")
plt.legend(loc='upper right')
x=np.arange(moyenne-3*ecarttype,moyenne+3*ecarttype,10)
gauss = (1/(ecarttype*np.sqrt(2*np.pi)))*np.exp(-((x)/ecarttype)*((x)/ecarttype)/2)
plt.fill(x,gauss,'b',alpha=0.2)
spectre=(abs(np.fft.fft(valeurs)))
freq=np.arange(0,fe,fe/N)
plt.subplot(212)
plt.vlines(freq,0,spectre,colors='g',alpha=0.1,label="spectre")
plt.legend(loc='upper center')
plt.show()