fftw_plan_dft_2d异常 使用技巧

fftw用着用着就异常了，前一个还挺好的，后一个就不行，程序也不报错，就是不能正常计算。

后来发现，fftw_plan_dft_2d有两个参数，一个代表傅里叶变换，一个代表傅里叶反变换。

一般要成对出现，不能连续创建两个傅里叶变换，然后创建一个傅里叶反变换，这样就有问题了。

改正后，都正常了

还发现，fftw_plan_dft_2d创建一次，后面更换数据源，可以反复计算。

fftw_plan_dft_2d创建以后，把输入数据换掉(不是重新实例化，把数据内容更新)，重新执行后，结果也会更新，

这样，输入输出的数组大小类型不变，就可以反复使用，不用每次执行时创建与释放，效率能提高好几倍乃至好几十倍。

加上参数FFTW_WISDOM_ONLY，分配内存为空。

 fftPlan  = fftwf_plan_dft_r2c_2d(row, col, (float *) xtv[0].data, 
                    (fftwf_complex *) xtfv[0].data, FFTW_WISDOM_ONLY|FFTW_PATIENT);

      #include "fftw3.h"
      #include <iostream>
      #include <string>
      #include <cassert>
      using namespace std;
      int main(int argc,char * argv[]) {
      int row=10;
      int col;
      float * realInput;
      fftwf_complex * complexOutput;
      fftwf_complex * complexInput;
      float * realOutput;
      for(int i= 1;i<argc;i++){
      row = col;
      realInput = (float *) fftwf_malloc(sizeof (float) * row * col );
      assert(realInput!=nullptr);
      complexOutput = (fftwf_complex *) fftwf_malloc(sizeof (fftwf_complex) * row * (col/2+1));
      assert(complexOutput!=nullptr);
      fftwf_plan r2c = fftwf_plan_dft_r2c_2d(row, col, realInput, complexOutput, FFTW_PATIENT);
      if(r2c == nullptr ){
      cout << "fftwf create r2c plan failed!" << endl;
      cout << "plan row: " << row << " col: " << col << endl;
      exit(1);
      } else {
      cout << "fftwf success to create fft r2c!" << endl;
      cout << "plan row: " << row << " col: " << col << endl;
      }
      complexInput = (fftwf_complex *) fftwf_malloc(sizeof (fftwf_complex) * row * (col/2+1) );
      assert(complexInput!=nullptr);
      realOutput = (float *) fftwf_malloc(sizeof (float) * row * col);
      assert(realOutput!=nullptr);
      fftwf_plan c2r = fftwf_plan_dft_c2r_2d(row, col, complexInput, realOutput, FFTW_PATIENT);
      if(c2r == nullptr){
      cout << "fftwf create c2r failed!" << endl;
      cout << "plan row: " << row << " col: " << col << endl;
      exit(1);
      } else {
      cout << "fftwf success to create c2r!" << endl;
      cout << "plan row: " << row << " col: " << col << endl;
      }
      }
      string wisdomFile = "wisdom";
      if(1==fftwf_export_wisdom_to_filename(wisdomFile.c_str()))
      cout << "fftwf_export_wisdom_to_filename wisdom success" << endl;
      else
      cout << "fftwf_export_wisdom_to_filename wisdom fail" << endl;
      }
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/64975458