This commit is contained in:
Mark Borgerding 2003-10-10 21:24:46 +00:00
parent 93de2a9410
commit edf93e8540

19
fft.py
View File

@ -22,14 +22,17 @@ def fft(f):
Fp = fft( fp )
Fm.extend( Fp )
Fout=[0]*n
for k in range(n):
km = (k%m)
val = Fm[ km ]
for q in range(1,p):
t = e ** ( j*2*pi*k*q/n )
val += Fm[ q*m + km ] * t
Fout[k] = val
Fout = [ 0 ] * n
for u in range(m):
scratch = Fm[u::m] # u to end in strides of m
for q1 in range(p):
k = q1*m + u # indices to Fout above that became scratch
val = Fm[ u ]
for q in range(1,p):
t = e ** ( j*2*pi*k*q/n )
#Fout[ k ] += scratch[q] * t
val += Fm[ q*m + u ] * t
Fout[ k ] = val
return Fout