[Scipy-svn] r5355 - trunk/scipy/fftpack

[scipy-svn at scipy.org]

Author: cdavid
Date: 2009-01-07 08:49:15 -0600 (Wed, 07 Jan 2009)
New Revision: 5355

Modified:
   trunk/scipy/fftpack/basic.py
Log:
Handle single prec complex input for fft and ifft.

Modified: trunk/scipy/fftpack/basic.py
===================================================================
--- trunk/scipy/fftpack/basic.py	2009-01-07 14:48:50 UTC (rev 5354)
+++ trunk/scipy/fftpack/basic.py	2009-01-07 14:49:15 UTC (rev 5355)
@@ -109,9 +109,9 @@
                                       hasattr(x,'__array__'))
         work_function = fftpack.zfft
     elif istype(tmp, numpy.complex64):
-        raise NotImplementedError
-    elif istype(tmp, numpy.float32):
-        raise NotImplementedError
+        overwrite_x = overwrite_x or (tmp is not x and not \
+                                      hasattr(x,'__array__'))
+        work_function = fftpack.cfft
     else:
         overwrite_x = 1
         work_function = fftpack.zrfft
@@ -149,7 +149,9 @@
                                       hasattr(x,'__array__'))
         work_function = fftpack.zfft
     elif istype(tmp, numpy.complex64):
-        raise NotImplementedError
+        overwrite_x = overwrite_x or (tmp is not x and not \
+                                      hasattr(x,'__array__'))
+        work_function = fftpack.cfft
     else:
         overwrite_x = 1
         work_function = fftpack.zrfft