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

streamlined

Browse Source
This commit is contained in:
Mark Borgerding 2003-12-30 03:08:56 +00:00
parent 22ff70b71d
commit 377ac796f0
1 changed files with 7 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
test/fft.py
View File

@ -174,33 +174,15 @@ def myfftnd(x):
def fftndwork(x,dims): def fftndwork(x,dims):
import Numeric import Numeric
kk=1 gg=Numeric.product( dims )
for k in range( len(dims) ): for k in range( len(dims) ):
d=dims[ k ] d=dims[ k ]
g=Numeric.product( dims ) / d g=gg/d
#print 'dimension %d==%d, g=%d' % ( k ,d,g ) next_stage = [complex(0,0)]*len(x)
#print 'x=%s' % str(x) for i in range(g):
next_stage = [] next_stage[i*d:(i+1)*d] = fft(x[i:(i+d)*g:g],0)
chunk = 0 x = next_stage
for ii in range(kk):
start = ii*d*g
for i in range(start,start+g):
xslice = x[i:(i+d)*g:g]
if len(xslice) != d:
print 'len(xslice)= %d' % len(xslice)
sys.exit(0)
fxslice = fft(xslice,0)
#print 'fft( x[%d:%d:%d] ) ' %( i,(i+d)*g,g),
#print 'fft( %s ) == %s ' % (str(xslice),str(fxslice))
next_stage.extend( fxslice )
if len(next_stage) != len(x):
print 'len(next_stage)= %d' % len(next_stage)
sys.exit(1)
#kk *= d
#print 'next_stage=%s' % str(next_stage)
x = next_stage[:]
return x return x
if __name__ == "__main__": if __name__ == "__main__":