\r
\r
/* Didn't the Montium use Q15 instead of Q14? */\r
-#define FIXED_POINT 14\r
+#define FIXED_POINT 15\r
#define WORD_SIZE 16\r
\r
#define WORDS_PER_LINE 4\r
#define WORDS_PER_GROUP 1\r
\r
-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; \r
-\r
-\r
+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; \r
\r
int to_fixed(float n)\r
{\r
void pre_run()\r
{\r
int i;\r
-\r
+ /* Assign memories, at least for the first stage */\r
+ input_a_re = alloc_mem(P0M0);\r
+ input_a_im = alloc_mem(P1M0);\r
+ input_b_re = alloc_mem(P2M0);\r
+ input_b_im = alloc_mem(P3M0);\r
+ \r
+ twiddle_re = alloc_mem(P4M0);\r
+ twiddle_im = alloc_mem(P4M1);\r
+ \r
/* TODO: Init memory and twiddles */\r
- for (i=0;i<SIZE/2;i++)\r
+ for (i=0;i<PARAM_N_t/2;i++)\r
{\r
- set_mem(twiddle_re->id, i, to_fixed(cos(2*M_PI/SIZE*i)));\r
- set_mem(twiddle_im->id, i, to_fixed(sin(2*M_PI/SIZE*i)));\r
+ set_mem(twiddle_re->id, i, to_fixed(cos(i*2*M_PI/PARAM_N_t)));\r
+ set_mem(twiddle_im->id, i, to_fixed(sin(i*2*M_PI/PARAM_N_t)));\r
}\r
\r
- for (i=0;i<SIZE;i++)\r
+ for (i=0;i<PARAM_N_t;i++)\r
{\r
- int value = to_fixed(sin((float)i/SIZE*2*2*M_PI));\r
- if (i<SIZE/2)\r
+ /* We take the sine from 0 to 2*2*Pi, ie two periods. We divide \r
+ * the value by PARAM_N_t to prevent overflow. */\r
+ int value = to_fixed(sin((float)i*2*2*M_PI/PARAM_N_t)/PARAM_N_t);\r
+\r
+ if (i<PARAM_N_t/2)\r
{\r
- set_mem(input_a_re->id, i, value);\r
- set_mem(input_a_im->id, i, 0);\r
+ if (i % 2 == 0) {\r
+ set_mem(input_a_re->id, i, value);\r
+ set_mem(input_a_im->id, i, 0);\r
+ } else {\r
+ set_mem(input_b_re->id, i, value);\r
+ set_mem(input_b_im->id, i, 0);\r
+ }\r
}\r
else\r
{\r
- set_mem(input_a_re->id, i - SIZE / 2, value);\r
- set_mem(input_a_im->id, i - SIZE / 2, 0);\r
+ if (i % 2 == 0) {\r
+ set_mem(input_b_re->id, i - PARAM_N_t/2, value);\r
+ set_mem(input_b_im->id, i - PARAM_N_t/2, 0);\r
+ } else {\r
+ set_mem(input_a_re->id, i - PARAM_N_t/2, value);\r
+ set_mem(input_a_im->id, i - PARAM_N_t/2, 0);\r
+ }\r
}\r
}\r
-}\r
-\r
-void post_run()\r
-{\r
+ \r
printf("re(W)\n");\r
- print_mem(twiddle_re, 0, SIZE, true);\r
+ print_mem(twiddle_re, 0, PARAM_N_t/2, true);\r
printf("im(W)\n");\r
- print_mem(twiddle_im, 0, SIZE, true);\r
+ print_mem(twiddle_im, 0, PARAM_N_t/2, true);\r
printf("re(in_a)\n");\r
- print_mem(input_a_re, 0, SIZE, true);\r
+ print_mem(input_a_re, 0, PARAM_N_t/2, true);\r
printf("re(in_b)\n");\r
- print_mem(input_b_re, 0, SIZE, true);\r
+ print_mem(input_b_re, 0, PARAM_N_t/2, true);\r
+}\r
+\r
+void post_run()\r
+{\r
+ if (PARAM_n_t % 2 == 0) {\r
+ /* When the number of stages is odd, the \r
+ * outputs end up at the left memories again */\r
+ output_a_re = alloc_mem(P0M0);\r
+ output_a_im = alloc_mem(P1M0);\r
+ output_b_re = alloc_mem(P2M0);\r
+ output_b_im = alloc_mem(P3M0);\r
+ } else {\r
+ output_a_re = alloc_mem(P0M1);\r
+ output_a_im = alloc_mem(P1M1);\r
+ output_b_re = alloc_mem(P2M1);\r
+ output_b_im = alloc_mem(P3M1);\r
+ }\r
printf("re(out_a)\n");\r
- print_mem(output_a_re, 0, SIZE, true);\r
+ print_mem(output_a_re, 0, PARAM_N_t/2, true);\r
+ print_mem(output_b_re, 0, PARAM_N_t/2, true);\r
printf("im(out_a)\n");\r
- print_mem(output_a_im, 0, SIZE, true);\r
+ print_mem(output_a_im, 0, PARAM_N_t/2, true);\r
+ print_mem(output_b_im, 0, PARAM_N_t/2, true);\r
+\r
}\r