from numpy import *
from matplotlib.pylab import *
from time import *
import wave
import struct
import sys

# .....

# Definerer funksjon for å lese lydfil
def fReadAudio(sFileName, iFrames, iChannels):
    fW = wave.open(sFileName, "r")
    sW = fW.readframes(iFrames)
    fW.close()
    aW = array(struct.unpack("%sh" %iFrames*iChannels,
sW)).reshape(iFrames,iChannels)
    return aW

# Leser lydfil
aFile = fReadAudio("Lydfiler\Granmeis2.wav", n, 2)
aWave = fft(aFile[:,0])

# Analyserer signalet
for j in range(ns):
    ......
    aA[j,:] = sqrt(abs(ifft(aFFT)))

# Plotter waveletanalysediagrammet
imshow(aA, origin="lower", extent=(0, n*dt, s1, s2), aspect=(n*dt*0.5)/(s2-s1))
xlabel("Time (s)")
ylabel("Scale")
title("Wavelet Analysis")
show()