正确的方法是在使用FFTW时,需要确保输入和输出数组的大小匹配,并且在进行FFT之前需要将输入数组进行适当的缩放。当使用ESTIMATE或MEASURE进行FFT计算时,需要确保使用的算法和它们的优先级正确,并且在执行FFT之前需要将输入数组进行适当的缩放。以下是使用FFTW计算FFT和逆FFT的示例代码:
#include <stdlib.h>
#include <stdio.h>
#include <math.h>
#include <fftw3.h>
#define N 16
int main(){
fftw_complex *in, *out;
fftw_plan p;
int i;
/* Allocate input and output arrays */
in = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * N);
out = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * N);
/* Initialize input array with a sine wave */
for(i=0; i<N; i++){
in[i][0] = sin(2*M_PI*i/N);
in[i][1] = 0.0;
}
/* Create FFT plan */
p = fftw_plan_dft_1d(N, in, out, FFTW_FORWARD, FFTW_ESTIMATE);
/* Execute FFT plan */
fftw_execute(p);
/* Print FFT result */
for(i=0; i<N; i++){
printf("%d %f %f\n", i, out[i][0], out[i][1]);
}
/* Scale output array and execute inverse FFT */
for(i=0; i<N; i++){
out[i][0] /= N;
out[i][1] /= N;
}
p = fftw_plan_dft_1d(N, out, in, FFTW_BACKWARD, FFTW_ESTIMATE);
fftw_execute(p);
/* Print inverse FFT result */
for(i=0; i<N; i++){
printf("%d %f %f\n", i, in[i][0], in[i][1]);
}
/* Free memory */
fftw_destroy_plan(p);
fftw_free(in);
fftw_free(out);
return 0;
}