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 回答