如何在C中批量FFT

Question

我正在尝试使用以下代码段对50张图片进行批量FFT：

pix3 = n*pix1*pix2;
fftwf_complex *in2, *f2h; //input for FFT2
in2 = (fftwf_complex*)fftwf_malloc(sizeof(fftwf_complex) * pix3);
f2h = (fftwf_complex*)fftwf_malloc(sizeof(fftwf_complex) * pix3);

for (i = 0; i < pix3; i++)
{
in2[i][0] = ref20[i]; //ref20 is an array of real values 
in2[i][1] = 0;
}    

for(i = 0; i < n; ++i)
{
  plan_forward2 = fftwf_plan_dft_2d (pix1, pix2, in2, f2h, FFTW_FORWARD, FFTW_ESTIMATE);
  fftwf_execute (plan_forward2);
  in2 += pix1*pix2;
  f2h += pix1*pix2;
}

但是，我只能获得第一张图像的FFT（即第一个pix1 * pix2元素）。关于如何做到这一点的任何建议？

提前致谢

Answer 1

您应该仔细阅读the manual 4.6 New-array Execute Functions

您只需要在循环外创建一次计划，然后将其用作：

fftwf_execute_dft(plan_forward2, in2, f2h);