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;
  }