mirror/kissfft
1
0
Fork 0
mirror of https://github.com/mborgerding/kissfft.git synced 2025-05-27 21:20:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

no real change

Browse Source
This commit is contained in:
Mark Borgerding 2003-12-30 00:09:50 +00:00
parent 62ecf69446
commit 5e6199c868
1 changed files with 12 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
test/fft.py
View File

@ -150,11 +150,10 @@ def test_fftnd(ndims=3):
import Numeric
x=randmat( ndims )
print 'dimensions=%s' % str( Numeric.shape(x) )
xver = FFT.fftnd(x)
x2=myfftnd(x)
err = xver-x2
#print xver
#print x2
err = xver - x2
errf = flatten(err)
xverf = flatten(xver)
errpow = Numeric.vdot(errf,errf)+1e-10
@ -173,16 +172,16 @@ def fftndwork(x,dims):
return fft(x,0)
else:
samples_per_chunk = Numeric.product( dims[1:] )
precomp=[]
curfftlen = dims[0]
for i in range(curfftlen):
xslice = x[i*samples_per_chunk:(i+1)*samples_per_chunk]
precomp.extend( fftndwork(xslice,dims[1:] ) )
xout=[ complex(0,0) ] * samples_per_chunk * dims[0]
for i in range(samples_per_chunk):
xout[i::samples_per_chunk] = fft( precomp[i::samples_per_chunk],0 )
return xout
precomp = [ ]
curfftlen = dims[ 0 ]
for i in range( curfftlen ):
xslice = x[ i*samples_per_chunk : (i+1)*samples_per_chunk ]
precomp.extend( fftndwork( xslice,dims[1:] ) )
xout = [ complex(0,0) ] * samples_per_chunk * dims[0]
for i in range( samples_per_chunk ):
pfx = fft( precomp[ i::samples_per_chunk ],0 )
xout[ i::samples_per_chunk ] = pfx
return xout
if __name__ == "__main__":
main()