fftw_plan_dft_2d优化

fftw_plan_dft_2d优化方法，

fftw_plan_dft_2d的输入和输出都是

fftwf_complex * *
 

经过测试发现，

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/65446955