X-Git-Url: https://git.stderr.nl/gitweb?p=matthijs%2Fprojects%2Fmontium-fft.git;a=blobdiff_plain;f=FFT_support.cpp;fp=FFT_support.cpp;h=6f21e5a24afa1e9cca6fe74a37bfa4965ff6d2ea;hp=1876e36ce4302e6cae6511d7113c976b637bcdc6;hb=1efd07a9676d63cd5ffe6535ee45101e87f89c86;hpb=7b57bd2ad91ccd7b8faff015adf1cbe87ddff3ea

diff --git a/FFT_support.cpp b/FFT_support.cpp
index 1876e36..6f21e5a 100644
--- a/FFT_support.cpp
+++ b/FFT_support.cpp
@@ -10,9 +10,7 @@
 #define WORDS_PER_LINE 4
 #define WORDS_PER_GROUP 1
 
-extern 	mem input_a_re, input_a_im, input_b_re, input_b_im, output_a_re, output_a_im, output_b_re, output_b_im, twiddle_re, twiddle_im; 
-
-
+mem input_a_re, input_a_im, input_b_re, input_b_im, output_a_re, output_a_im, output_b_re, output_b_im, twiddle_re, twiddle_im; 
 
 int to_fixed(float n)
 {
@@ -52,7 +50,16 @@ void print_mem(mem m, int offset, int size, bool fixed)
 void pre_run()
 {
 	int i;
-
+	/* Assign memories, at least for the first stage */
+	input_a_re   = alloc_mem(P0M0);
+	input_a_im   = alloc_mem(P1M0);
+	input_b_re   = alloc_mem(P2M0);
+	input_b_im   = alloc_mem(P3M0);
+	output_a_re  = alloc_mem(P0M1);
+	output_a_im  = alloc_mem(P1M1);
+	output_b_re  = alloc_mem(P2M1);
+	output_b_im  = alloc_mem(P3M1);
+	
 	/* TODO: Init memory and twiddles */
 	for (i=0;i<SIZE/2;i++)
 	{