From 377ac796f0a12124e9a58a97bdfbd9055951f784 Mon Sep 17 00:00:00 2001
From: Mark Borgerding <mark@borgerding.net>
Date: Tue, 30 Dec 2003 03:08:56 +0000
Subject: [PATCH] streamlined

---
 test/fft.py | 32 +++++++-------------------------
 1 file changed, 7 insertions(+), 25 deletions(-)

diff --git a/test/fft.py b/test/fft.py
index 80783be..e39b49c 100755
--- a/test/fft.py
+++ b/test/fft.py
@@ -174,33 +174,15 @@ def myfftnd(x):
 
 def fftndwork(x,dims):
     import Numeric
-    kk=1
+    gg=Numeric.product( dims )
+
     for k in range( len(dims) ):
         d=dims[ k ]
-        g=Numeric.product( dims ) / d
-        #print 'dimension %d==%d, g=%d' % ( k ,d,g )
-        #print 'x=%s' % str(x)
-        next_stage = []
-        chunk = 0
-        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[:]
+        g=gg/d
+        next_stage = [complex(0,0)]*len(x)
+        for i in range(g):
+            next_stage[i*d:(i+1)*d] = fft(x[i:(i+d)*g:g],0)
+        x = next_stage
     return x
 
 if __name__ == "__main__":