使用FFTW的2D复数阵列的FFT

时间:2018-04-23 08:19:13

标签: c fftw

我在C中有一个大小为1001(行)* 144(列)的2D双复数组。我想在每一行上应用FFT,最后想要输出为4096 * 144格式。这里N点= 4096.最后将结果与matlab输出进行比较。 我正在使用着名的FFTW C库。我已阅读tutorials但无法理解如何正确使用。我应该使用哪种程序,如1D例程或2D例程然后如何?

#Update

double complex reshaped_calib[1001][144]={{1.0 + 0.1 * I}};

double complex** input; //[4096][144];
 // have to take dynamic array as I was getting segmentation fault here   
input = malloc(4096 * sizeof(double complex*));
for (i = 0; i < 4096; i++) {
  input[i] = malloc(144* sizeof(double complex));
}
// input is array I am sending to fftw to apply fft

for (i= 0; i< 1001; i++)
{
  for (j= 0; j< 144; j++)
  {
    input[i][j]=reshaped_calib[i][j];
  }
}

// Pad the extra rows
for (i= 1001; i< 4096; i++)
{
  for (j= 0; j< 144; j++)
  {
    input[i][j] = 0.0;
  }
}

 int N=144, howmany=4096;
 fftw_complex* data = (fftw_complex*)  fftw_malloc(N*howmany*sizeof(fftw_complex));

i=0,j=0;
int dataCount=0;
for(i=0;i<4096;i++)
    {
        for(j=0;j<144;j++)
        {
            data[dataCount++]=CMPLX(creal(input[i][j]),cimag(input[i][j]));
        }
    }
int istride=1, idist=N;// as in C data as row major
// ... if data is column-major, set istride=howmany, idist=1
//    if data is row-major, set istride=1, idist=N
fftw_plan p = fftw_plan_many_dft(1,&N,howmany,data,NULL,howmany,1,data,NULL,howmany,1,FFTW_FORWARD,FFTW_MEASURE);
fftw_execute(p);

1 个答案:

答案 0 :(得分:0)

您尝试使用

填充数组
int pad = 4096;
memset(reshaped_calib, 0.0, pad * sizeof(double complex)); //zero padding

基本上会覆盖数组reshaped_calib的前4096个值。为了正确填充,您需要将2D数组的大小扩展到所需的4096 x 144大小,并将输入的1001 x 144范围之外的条目设置为零。

由于您只是扩展行数,因此可以使用以下内容:

double complex input[1001][144]={{1.0 + 0.1 * I}};

// Allocate storage for the larger 4096x144 2D size, and copy the input.
double complex reshaped_calib[4096][144];
for (int row = 0; row < 1001; row++)
{
  for (int col = 0; col < 144; col++)
  {
    reshaped_calib[row][col] = input[row][col];
  }
}
// Pad the extra rows
for (int row = 1001; row < 4996; row++)
{
  for (int col = 0; col < 144; col++)
  {
    reshaped_calib[row][col] = 0.0;
  }
}

也就是说,如果你想分别对每一行进行1D FFT,你应该使用fftw_plan_dft_1d并多次调用fftw_execute,否则使用fftw_plan_many_dftin this answer详细描述了@Dylan