import matplotlib.pyplot as pyplot
import scipy.io.wavfile
import pywt

rate, signal = scipy.io.wavfile.read('a0297.wav')


pyplot.figure(1)
pyplot.plot(signal)
pyplot.show()


tree = pywt.wavedec(data=signal[:5000], wavelet='db2', level=4, mode='symmetric')

for i in range(0, len(tree[1])): tree[1][i] = 0
for i in range(0, len(tree[2])): tree[2][i] = 0
for i in range(0, len(tree[3])): tree[3][i] = 0
for i in range(0, len(tree[4])): tree[4][i] = 0


recSignal = pywt.waverec(tree, 'db2')

a = 0
b = 0
c = 0
d = 0


for i in range(0, 1000):
    if recSignal[i] > recSignal[a]:
        a = i

#for i in range(2000, 2500):
#   if recSignal[i] > recSignal[c]:
#        c = i
#    elif recSignal[d] < recSignal[i] < recSignal[c]:
#       d = i


for i in range(0, 2200):
    if recSignal[i] > recSignal[b]:
         b = i

#a = a/2
#b = b/2
print('pierwszy', a)
print('pierwszy1', b)

print('test1 ', c/2)
print('test2 ', d/2)

suma = 0

suma = b - a
print('suma', suma)
print('liczba uderzen na minute', 60000/suma)

pyplot.figure(2)
pyplot.plot(recSignal[:5000])
pyplot.show()