c - 如何在 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

您应该仔细阅读手册 4.6 New-array Execute Functions

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

fftwf_execute_dft(plan_forward2, in2, f2h);