improvements

[my-code/api.git] / fourier / fourier.h

diff --git a/fourier/fourier.h b/fourier/fourier.h
index f3cab6b..fcaa3e9 100644 (file)
--- a/fourier/fourier.h
+++ b/fourier/fourier.h
@@ -4,7 +4,11 @@
 #define FOURIER_H
 
 /* includes */
+#define _GNU_SOURCE
 #include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <math.h>
 
 /* defines */
 #define F_SUCCESS 1
@@ -26,16 +30,18 @@ typedef struct s_fourier {
 #define FFT (1<<1)
 #define FWD (1<<2)
 #define BWD (1<<3)
-  int dim;
 #define MAX_DIM 3
-  t_complex *data[MAX_DIM];
-  t_complex *ftdata[MAX_DIM];
+  int dim;
+  t_complex *data;
+  t_complex *ftdata;
   int data_len[MAX_DIM];
 } t_fourier;
 
 /* function prototypes */
 int fourier_init(t_fourier *fourier,int outfd);
+int fourier_alloc_data(t_fourier *fourier);
 int fourier_dft_1d(t_fourier *fourier);
-int fourier_calc(t_fourier *fourier);
+int fourier_dft_2d(t_fourier *fourier);
+int fourier_dft_3d(t_fourier *fourier);
 
 #endif