summaryrefslogtreecommitdiff
path: root/src/signalProcessing/fft/zfftma.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/signalProcessing/fft/zfftma.c')
-rw-r--r--src/signalProcessing/fft/zfftma.c143
1 files changed, 143 insertions, 0 deletions
diff --git a/src/signalProcessing/fft/zfftma.c b/src/signalProcessing/fft/zfftma.c
new file mode 100644
index 00000000..2b0a7510
--- /dev/null
+++ b/src/signalProcessing/fft/zfftma.c
@@ -0,0 +1,143 @@
+/*
+ * Scilab ( http://www.scilab.org/ ) - This file is part of Scilab
+ * Copyright (C) 2008 - INRIA - Allan SIMON
+ *
+ * This file must be used under the terms of the CeCILL.
+ * This source file is licensed as described in the file COPYING, which
+ * you should have received as part of this distribution. The terms
+ * are also available at
+ * http://www.cecill.info/licences/Licence_CeCILL_V2-en.txt
+ *
+ */
+
+
+#include "fft.h"
+#include <stdio.h>
+
+void zfftma ( doubleComplex* in , int rows, int cols, doubleComplex* out)
+{
+
+int size = rows*cols ;
+int sizeTemp = 0;
+int sizeWorkSpace = 2* size;
+
+int rowsTemp = 0 ;
+int colsTemp = 0 ;
+
+int ierr = 0 ;
+int isn = -1;
+int i = 0;
+
+double* realIn = (double*) malloc ( sizeof (double) * (unsigned int) size );
+double* imagIn = (double*) malloc ( sizeof (double) * (unsigned int) size );
+
+
+doubleComplex* inTemp = (doubleComplex*) malloc ( sizeof (doubleComplex) * (unsigned int) size );
+
+double* workSpace = (double*) malloc ( sizeof (double) * (unsigned int) sizeWorkSpace );
+
+
+
+zimaga ( in , size , imagIn) ;
+zreala ( in , size , realIn) ;
+
+if ( rows == 1 || cols == 1 )
+{
+ printf ( "it'a vector \n" ) ;
+
+ sizeTemp = (int) pow ( 2 , (int ) (log( size + 0.5 ) /log ( 2 ))) ;
+ printf ("pow %e , temp %d \n" , pow ( 2 , (int )(log( size +0.5 ) /log ( 2 ))), sizeTemp);
+
+ if ( size == sizeTemp )
+ {
+ if ( size <= pow ( 2 , 15 ))
+ {
+ printf ( "we call fft842 \n" ) ;
+ fft842 ( in , size , 0 );
+ }
+ else
+ {
+ printf ( "we call dfft2 \n" ) ;
+ dfft2 ( realIn , imagIn , 1 , size , 1 , isn , ierr , workSpace , sizeWorkSpace );
+ }
+
+
+ }
+ else
+ {
+ printf ( "we call dfft2 2\n" ) ;
+ dfft2 ( realIn , imagIn , 1 , size , 1 , isn , ierr , workSpace , sizeWorkSpace );
+ }
+
+}
+else
+ {
+ printf ( "it'a matrix \n" ) ;
+ rowsTemp = (int) pow ( 2 , log( rows + 0.5) /log ( 2 )) ;
+ colsTemp = (int) pow ( 2 , log( cols + 0.5) /log ( 2 )) ;
+
+ if ( cols == colsTemp)
+ {
+ if ( cols <= pow ( 2 , 15 ))
+ {
+ for ( i = 0 ; i < rows ; i++ )
+ {
+ fft842 ( &in[ cols*i] , cols , 0);
+ }
+ }
+ else
+ {
+ dfft2 ( realIn, imagIn ,rows , cols , 1 , isn , ierr ,workSpace , sizeWorkSpace );
+ }
+ }
+ else
+ {
+ dfft2 ( realIn, imagIn ,rows , cols , 1 , isn , ierr ,workSpace , sizeWorkSpace );
+ }
+
+ /*second call*/
+
+ if ( 2*rows <= sizeWorkSpace )
+ {
+ if ( rowsTemp == rows )
+ {
+ if ( rows <= pow ( 2 ,15) )
+ {
+ /*compute the fft on each line of the matrix */
+ for (i = 0 ; i < cols ; i++ )
+ {
+ C2F(zcopy) ( rows, in + i, cols, inTemp , 1 );
+
+ fft842( inTemp , rows , 0);
+
+ C2F(zcopy) ( rows, inTemp , cols, in + i, 1 );
+
+ }
+ }
+ else
+ {
+ dfft2 ( realIn, imagIn, 1, rows, cols, isn, ierr, workSpace, sizeWorkSpace);
+ }
+ }
+ else
+ {
+ dfft2 ( realIn, imagIn, 1, rows, cols, isn, ierr, workSpace, sizeWorkSpace);
+ }
+ }
+ else
+ {
+ dfft2 ( realIn, imagIn, 1, rows, cols, isn, ierr, workSpace, sizeWorkSpace);
+ }
+ }
+
+
+
+
+
+for ( i = 0 ; i < size ; i++)
+{
+out[i] = DoubleComplex ( zreals(in[i]) , zimags(in[i]) );
+}
+
+
+}