mirror/kissfft
1
0
Fork 0
mirror of https://github.com/mborgerding/kissfft.git synced 2025-06-04 01:28:23 -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 import Numeric
x=randmat( ndims ) x=randmat( ndims )
print 'dimensions=%s' % str( Numeric.shape(x) )
xver = FFT.fftnd(x) xver = FFT.fftnd(x)
x2=myfftnd(x) x2=myfftnd(x)
err = xver-x2 err = xver - x2
#print xver
#print x2
errf = flatten(err) errf = flatten(err)
xverf = flatten(xver) xverf = flatten(xver)
errpow = Numeric.vdot(errf,errf)+1e-10 errpow = Numeric.vdot(errf,errf)+1e-10
@ -173,16 +172,16 @@ def fftndwork(x,dims):
return fft(x,0) return fft(x,0)
else: else:
samples_per_chunk = Numeric.product( dims[1:] ) samples_per_chunk = Numeric.product( dims[1:] )
precomp=[] precomp = [ ]
curfftlen = dims[0] curfftlen = dims[ 0 ]
for i in range(curfftlen): for i in range( curfftlen ):
xslice = x[i*samples_per_chunk:(i+1)*samples_per_chunk] xslice = x[ i*samples_per_chunk : (i+1)*samples_per_chunk ]
precomp.extend( fftndwork(xslice,dims[1:] ) ) precomp.extend( fftndwork( xslice,dims[1:] ) )
xout = [ complex(0,0) ] * samples_per_chunk * dims[0]
xout=[ complex(0,0) ] * samples_per_chunk * dims[0] for i in range( samples_per_chunk ):
for i in range(samples_per_chunk): pfx = fft( precomp[ i::samples_per_chunk ],0 )
xout[i::samples_per_chunk] = fft( precomp[i::samples_per_chunk],0 ) xout[ i::samples_per_chunk ] = pfx
return xout return xout
if __name__ == "__main__": if __name__ == "__main__":
main() main()