1

我试图让 FFTW 在 C 中工作。它曾经为另一个项目工作(在 JNI 中),我或多或少地从那个项目中复制代码,遗憾的是没有结果。

首先,我生成一个正弦信号,如下所示:

double* generateSignal() {
    int fs=44100;
    double fsd = 44100.0; // fs in double format
    double f1=1000.0;
    int i;
    double PI = 3.141592653589793238462643;

    double t[fs];
    double value = 0.0;
    for (i = 0; i < fs; i++) {
        t[i] = value;
        value += 1.0/fsd;
    }

    double* signal = (double*) malloc(sizeof(double) * fs);
    for (i = 0; i < fs; i++) {
        signal[i] = sqrt(2) * sin(2 * PI * f1 * t[i]);
    }

    return signal;
}    

这行得通,我只是为了完整性而发布它。

接下来,我想使用 FFTW 转换信号,我使用以下方法(基于FFTW 文档)执行此操作:

void processSignal(double* signal) {
    int size = 44100;
    int i;

    fftw_complex* in = fftw_malloc(sizeof(fftw_complex) * size);
    fftw_complex* out = fftw_malloc(sizeof(fftw_complex) * size);

    for (i = 0; i < size; i++) {
        double* ptr = in[i];
        *ptr = signal[i];       // set first double, real part
        *(ptr + 1) = 0.0;       // set second double, imaginary part
    }

    fftw_plan p = fftw_plan_dft_1d(size, in, out, FFTW_FORWARD, FFTW_ESTIMATE); 
    fftw_execute(p);

    for (i = 0; i < size; i++) {
        double* ptr = out[i];
        signal[i] = *ptr;       // get real part
    }

    fftw_destroy_plan(p);
    fftw_free(in);
    fftw_free(out);
}

请注意 FFTW 文档中的这一点:typedef double fftw_complex[2];

现在,这导致 -array 的所有值signal都是 -0.000000。我真的看不出这段代码的问题,所以请指出我做错了什么?

谢谢。

PS:为了清楚起见,从我的代码中删除了打印语句。

4

2 回答 2

0

我认为问题可能是您正在用当前迭代的实部覆盖上一次迭代的虚部。

for (i = 0; i < size; i++) {
        double* ptr = in[i];    //  <-- here's a problem
        *ptr = signal[i];       // set first double, real part
        *(ptr + 1) = 0.0;       // set second double, imaginary part
    }

i在每次迭代中递增,因此在第一次迭代时,ptr指向输入复数 [0] 的实部并ptr + 1指向复数 [0] 的虚部,但在第二次迭代时,ptr指向虚部复数[0]ptr + 1的一部分,指向复数[1]的实部。

解决此问题的一些建议可能是:

for (i = 0, j = 0; i < size; i++, j+= 2) {
    double* ptr = in[j];    // j increments by 2 making ptr alays point to real part 
    *ptr = signal[i];       // set first double, real part
    *(ptr + 1) = 0.0;       // set second double, imaginary part
}

或者

double* ptr = in[0]
for (i = 0; i < size; i++) {
        *ptr++ = signal[i]; // set first double, real part
        *ptr++ = 0.0;       // set second double, imaginary part
    }
于 2012-03-03T00:24:43.630 回答
0

天哪,这太愚蠢了。

这些值的大小类似于 -2.0E-9。结果四舍五入,因此打印了 -0.00000。

非常感谢弗雷德。他的问题“你的预期输出是什么?” 让我仔细查看了我之前提到的 JNI 项目的输出,并意识到我的错误。

也感谢其他答案!

于 2012-03-03T11:11:50.833 回答