这个代码用g++编译OK，gcc编译出错，谁帮我改改？

prolj

1. #include <stdio.h>
   
2. #include <stdlib.h>
   
3. #include <math.h>
   
4. 
   
5. const int N = 1024;
   
6. const float PI = 3.1416;
   
7. 
   
8. inline void swap(float &a, float &b)
   
9. {
   
10.   float t;
    
11.   t = a;
    
12.   a = b;
    
13.   b = t;
    
14. }
    
15. 
    
16. void bitrp (float xreal[], float ximag[], int n)
    
17. {
    
18.   // 位反转置换 Bit-reversal Permutation
    
19.   int i, j, a, b, p;
    
20. 
    
21.   for (i = 1, p = 0; i < n; i *= 2) {
    
22.     p++;
    
23.   }
    
24.   for (i = 0; i < n; i++) {
    
25.     a = i;
    
26.     b = 0;
    
27.     for (j = 0; j < p; j++) {
    
28.       b = (b << 1) + (a & 1);     // b = b * 2 + a % 2;
    
29.       a >>= 1;         // a = a / 2;
    
30.     }
    
31.     if ( b > i) {
    
32.       swap (xreal[i], xreal[b]);
    
33.       swap (ximag[i], ximag[b]);
    
34.     }
    
35.   }
    
36. }
    
37. 
    
38. void FFT(float xreal [], float ximag [], int n)
    
39. {
    
40.   // 快速傅立叶变换，将复数 x 变换后仍保存在 x 中，xreal, ximag 分别是 x 的实部和虚部
    
41.   float wreal[N / 2], wimag[N / 2], treal, timag, ureal, uimag, arg;
    
42.   int m, k, j, t, index1, index2;
    
43. 
    
44.   bitrp (xreal, ximag, n);
    
45. 
    
46.   // 计算 1 的前 n / 2 个 n 次方根的共轭复数 W'j = wreal[j] + i * wimag[j] , j = 0, 1,   , n / 2 - 1
    
47.   arg = - 2 * PI / n;
    
48.   treal = cos (arg);
    
49.   timag = sin (arg);
    
50.   wreal [0] = 1.0;
    
51.   wimag [0] = 0.0;
    
52.   for (j = 1; j < n / 2; j++) {
    
53.     wreal[j] = wreal[j - 1] * treal - wimag[j - 1] * timag;
    
54.     wimag[j] = wreal[j - 1] * timag + wimag[j - 1] * treal;
    
55.   }
    
56. 
    
57.   for (m = 2; m <= n; m *= 2) {
    
58.     for (k = 0; k < n; k += m) {
    
59.       for (j = 0; j < m / 2; j++) {
    
60.         index1 = k + j;
    
61.         index2 = index1 + m / 2;
    
62.         t = n * j / m;     // 旋转因子 w 的实部在 wreal [] 中的下标为 t
    
63.         treal = wreal[t] * xreal[index2] - wimag[t] * ximag[index2];
    
64.         timag = wreal[t] * ximag[index2] + wimag[t] * xreal[index2];
    
65.         ureal = xreal[index1];
    
66.         uimag = ximag[index1];
    
67.         xreal[index1] = ureal + treal;
    
68.         ximag[index1] = uimag + timag;
    
69.         xreal[index2] = ureal - treal;
    
70.         ximag[index2] = uimag - timag;
    
71.       }
    
72.     }
    
73.   }
    
74. }
    
75. 
    
76. void   IFFT (float xreal [], float ximag [], int n)
    
77. {
    
78.   // 快速傅立叶逆变换
    
79.   float wreal[N / 2], wimag[N / 2], treal, timag, ureal, uimag, arg;
    
80.   int m, k, j, t, index1, index2;
    
81. 
    
82.   bitrp (xreal, ximag, n);
    
83. 
    
84.   // 计算 1 的前 n / 2 个 n 次方根 Wj = wreal [j] + i * wimag [j] , j = 0, 1,   , n / 2 - 1
    
85.   arg = 2 * PI / n;
    
86.   treal = cos (arg);
    
87.   timag = sin (arg);
    
88.   wreal [0] = 1.0;
    
89.   wimag [0] = 0.0;
    
90.   for (j = 1; j < n / 2; j++) {
    
91.     wreal[j] = wreal[j - 1] * treal - wimag[j - 1] * timag;
    
92.     wimag[j] = wreal[j - 1] * timag + wimag[j - 1] * treal;
    
93.   }
    
94. 
    
95.   for (m = 2; m <= n; m *= 2) {
    
96.     for (k = 0; k < n; k += m) {
    
97.       for (j = 0; j < m / 2; j++) {
    
98.         index1 = k + j;
    
99.         index2 = index1 + m / 2;
    
100.         t = n * j / m;     // 旋转因子 w 的实部在 wreal [] 中的下标为 t
     
101.         treal = wreal[t] * xreal[index2] - wimag[t] * ximag[index2];
     
102.         timag = wreal[t] * ximag[index2] + wimag[t] * xreal[index2];
     
103.         ureal = xreal[index1];
     
104.         uimag = ximag[index1];
     
105.         xreal[index1] = ureal + treal;
     
106.         ximag[index1] = uimag + timag;
     
107.         xreal[index2] = ureal - treal;
     
108.         ximag[index2] = uimag - timag;
     
109.       }
     
110.     }
     
111.   }
     
112. 
     
113.   for (j=0; j < n; j++) {
     
114.     xreal[j] /= n;
     
115.     ximag[j] /= n;
     
116.   }
     
117. }
     
118. 
     
119. void FFT_test ()
     
120. {
     
121.   char inputfile[] = "input.txt";
     
122.   char outputfile[] = "output.txt";
     
123.   float xreal[N] = {}, ximag[N] = {};
     
124.   int n, i;
     
125.   FILE *input, *output;
     
126. 
     
127.   if (!(input = fopen (inputfile, "r"))) {
     
128.     printf ("Cannot open file.\n");
     
129.     exit (1);
     
130.   }
     
131.   if (!(output = fopen (outputfile, "w"))) {
     
132.     printf ("Cannot open file.\n");
     
133.     exit (1);
     
134.   }
     
135. 
     
136.   i = 0;
     
137.   while ((fscanf (input, "%f%f", xreal + i, ximag + i)) != EOF) {
     
138.     i++;
     
139.   }
     
140.   n = i;     // 要求 n 为 2 的整数幂
     
141.   while (i > 1) {
     
142.     if (i % 2) {
     
143.       fprintf (output, "%d is not a power of 2!\n", n);
     
144.       exit (1);
     
145.     }
     
146.     i /= 2;
     
147.   }
     
148. 
     
149.   FFT (xreal, ximag, n);
     
150.   fprintf (output, "FFT:     i      real imag\n");
     
151.   for (i = 0; i < n; i++) {
     
152.     fprintf (output, "%4d     %8.4f     %8.4f\n", i, xreal[i], ximag[i]);
     
153.   }
     
154.   fprintf (output, "=================================\n");
     
155. 
     
156.   IFFT (xreal, ximag, n);
     
157.   fprintf (output, "IFFT:     i      real imag\n");
     
158.   for (i = 0; i < n; i++) {
     
159.     fprintf (output, "%4d     %8.4f     %8.4f\n", i, xreal[i], ximag[i]);
     
160.   }
     
161. 
     
162.   if ( fclose (input)) {
     
163.     printf ("File close error.\n");
     
164.     exit (1);
     
165.   }
     
166.   if ( fclose (output)) {
     
167.     printf ("File close error.\n");
     
168.     exit (1);
     
169.   }
     
170. }
     
171. 
     
172. int main (void)
     
173. {
     
174.   FFT_test ();
     
175. 
     
176.   return 0;
     
177. }
     

复制代码

该死的FFT

hellioncu

const int 定义改define

float &引用改成指针

prolj

float &引用改成指针可以么？
传进去的参数就是数组的下标啊。我改了不行的。

hellioncu

inline void swap(float *a, float *b)
{
  float t;

  t = *a;
  *a = *b;
  *b = t;
}

      swap (&xreal[i], &xreal[b]);

      swap (&ximag[i], &ximag[b]);

prolj

特别坏，你编译通过了么？

hellioncu

特别坏，你编译通过了么？
prolj 发表于 2010-11-12 10:52

    编译没问题，链接错，加上 -lm就可以了


gcc fft.c -lm

pinecrane

回复 6# hellioncu


    数学运算库不是默认加载的。需要显示指定。

  这个大牛就是大牛哦

prolj

-lm这个我肯定知道啊。
可是我这里还是不对啊。

hellioncu

回复 8# prolj

1. #include <stdio.h>
   
2. #include <stdlib.h>
   
3. #include <math.h>
   
4. 
   
5. #define N        1024
   
6. #define PI        3.1416
   
7. 
   
8. inline void swap(float *a, float *b)
   
9. {
   
10.   float t;
    
11.   t = *a;
    
12.   *a = *b;
    
13.   *b = t;
    
14. }
    
15. 
    
16. void bitrp (float xreal[], float ximag[], int n)
    
17. {
    
18.   // 位反转置换 Bit-reversal Permutation
    
19.   int i, j, a, b, p;
    
20. 
    
21.   for (i = 1, p = 0; i < n; i *= 2) {
    
22.     p++;
    
23.   }
    
24.   for (i = 0; i < n; i++) {
    
25.     a = i;
    
26.     b = 0;
    
27.     for (j = 0; j < p; j++) {
    
28.       b = (b << 1) + (a & 1);     // b = b * 2 + a % 2;
    
29.       a >>= 1;         // a = a / 2;
    
30.     }
    
31.     if ( b > i) {
    
32.       swap (&xreal[i], &xreal[b]);
    
33.       swap (&ximag[i], &ximag[b]);
    
34.     }
    
35.   }
    
36. }
    
37. 
    
38. void FFT(float xreal [], float ximag [], int n)
    
39. {
    
40.   // 快速傅立叶变换，将复数 x 变换后仍保存在 x 中，xreal, ximag 分别是 x 的实部和虚部
    
41.   float wreal[N / 2], wimag[N / 2], treal, timag, ureal, uimag, arg;
    
42.   int m, k, j, t, index1, index2;
    
43. 
    
44.   bitrp (xreal, ximag, n);
    
45. 
    
46.   // 计算 1 的前 n / 2 个 n 次方根的共轭复数 W'j = wreal[j] + i * wimag[j] , j = 0, 1,   , n / 2 - 1
    
47.   arg = - 2 * PI / n;
    
48.   treal = cos (arg);
    
49.   timag = sin (arg);
    
50.   wreal [0] = 1.0;
    
51.   wimag [0] = 0.0;
    
52.   for (j = 1; j < n / 2; j++) {
    
53.     wreal[j] = wreal[j - 1] * treal - wimag[j - 1] * timag;
    
54.     wimag[j] = wreal[j - 1] * timag + wimag[j - 1] * treal;
    
55.   }
    
56. 
    
57.   for (m = 2; m <= n; m *= 2) {
    
58.     for (k = 0; k < n; k += m) {
    
59.       for (j = 0; j < m / 2; j++) {
    
60.         index1 = k + j;
    
61.         index2 = index1 + m / 2;
    
62.         t = n * j / m;     // 旋转因子 w 的实部在 wreal [] 中的下标为 t
    
63.         treal = wreal[t] * xreal[index2] - wimag[t] * ximag[index2];
    
64.         timag = wreal[t] * ximag[index2] + wimag[t] * xreal[index2];
    
65.         ureal = xreal[index1];
    
66.         uimag = ximag[index1];
    
67.         xreal[index1] = ureal + treal;
    
68.         ximag[index1] = uimag + timag;
    
69.         xreal[index2] = ureal - treal;
    
70.         ximag[index2] = uimag - timag;
    
71.       }
    
72.     }
    
73.   }
    
74. }
    
75. 
    
76. void   IFFT (float xreal [], float ximag [], int n)
    
77. {
    
78.   // 快速傅立叶逆变换
    
79.   float wreal[N / 2], wimag[N / 2], treal, timag, ureal, uimag, arg;
    
80.   int m, k, j, t, index1, index2;
    
81. 
    
82.   bitrp (xreal, ximag, n);
    
83. 
    
84.   // 计算 1 的前 n / 2 个 n 次方根 Wj = wreal [j] + i * wimag [j] , j = 0, 1,   , n / 2 - 1
    
85.   arg = 2 * PI / n;
    
86.   treal = cos (arg);
    
87.   timag = sin (arg);
    
88.   wreal [0] = 1.0;
    
89.   wimag [0] = 0.0;
    
90.   for (j = 1; j < n / 2; j++) {
    
91.     wreal[j] = wreal[j - 1] * treal - wimag[j - 1] * timag;
    
92.     wimag[j] = wreal[j - 1] * timag + wimag[j - 1] * treal;
    
93.   }
    
94. 
    
95.   for (m = 2; m <= n; m *= 2) {
    
96.     for (k = 0; k < n; k += m) {
    
97.       for (j = 0; j < m / 2; j++) {
    
98.         index1 = k + j;
    
99.         index2 = index1 + m / 2;
    
100.         t = n * j / m;     // 旋转因子 w 的实部在 wreal [] 中的下标为 t
     
101.         treal = wreal[t] * xreal[index2] - wimag[t] * ximag[index2];
     
102.         timag = wreal[t] * ximag[index2] + wimag[t] * xreal[index2];
     
103.         ureal = xreal[index1];
     
104.         uimag = ximag[index1];
     
105.         xreal[index1] = ureal + treal;
     
106.         ximag[index1] = uimag + timag;
     
107.         xreal[index2] = ureal - treal;
     
108.         ximag[index2] = uimag - timag;
     
109.       }
     
110.     }
     
111.   }
     
112. 
     
113.   for (j=0; j < n; j++) {
     
114.     xreal[j] /= n;
     
115.     ximag[j] /= n;
     
116.   }
     
117. }
     
118. 
     
119. void FFT_test ()
     
120. {
     
121.   char inputfile[] = "input.txt";
     
122.   char outputfile[] = "output.txt";
     
123.   float xreal[N] = {}, ximag[N] = {};
     
124.   int n, i;
     
125.   FILE *input, *output;
     
126. 
     
127.   if (!(input = fopen (inputfile, "r"))) {
     
128.     printf ("Cannot open file.\n");
     
129.     exit (1);
     
130.   }
     
131.   if (!(output = fopen (outputfile, "w"))) {
     
132.     printf ("Cannot open file.\n");
     
133.     exit (1);
     
134.   }
     
135. 
     
136.   i = 0;
     
137.   while ((fscanf (input, "%f%f", xreal + i, ximag + i)) != EOF) {
     
138.     i++;
     
139.   }
     
140.   n = i;     // 要求 n 为 2 的整数幂
     
141.   while (i > 1) {
     
142.     if (i % 2) {
     
143.       fprintf (output, "%d is not a power of 2!\n", n);
     
144.       exit (1);
     
145.     }
     
146.     i /= 2;
     
147.   }
     
148. 
     
149.   FFT (xreal, ximag, n);
     
150.   fprintf (output, "FFT:     i      real imag\n");
     
151.   for (i = 0; i < n; i++) {
     
152.     fprintf (output, "%4d     %8.4f     %8.4f\n", i, xreal[i], ximag[i]);
     
153.   }
     
154.   fprintf (output, "=================================\n");
     
155. 
     
156.   IFFT (xreal, ximag, n);
     
157.   fprintf (output, "IFFT:     i      real imag\n");
     
158.   for (i = 0; i < n; i++) {
     
159.     fprintf (output, "%4d     %8.4f     %8.4f\n", i, xreal[i], ximag[i]);
     
160.   }
     
161. 
     
162.   if ( fclose (input)) {
     
163.     printf ("File close error.\n");
     
164.     exit (1);
     
165.   }
     
166.   if ( fclose (output)) {
     
167.     printf ("File close error.\n");
     
168.     exit (1);
     
169.   }
     
170. }
     
171. 
     
172. int main (void)
     
173. {
     
174.   FFT_test ();
     
175. 
     
176.   return 0;
     
177. }
     
178. 
     

复制代码

prolj

哦，不知道我哪里改错了。