我在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);
答案 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_dft
。 in this answer详细描述了@Dylan。