0

我正在使用 org.apache.commons.math3.transform 库进行从实数到复数的 FFT，如下所示：

private Complex[] fft(double[] values) {
   FastFourierTransformer ffTransformer = new FastFourierTransformer(DftNormalization.STANDARD);
   Complex[] result = ffTransformer.transform(values, TransformType.FORWARD); 
   return result;  
}

这给了我一个带有结果的 org.apache.commons.math3.complex 数组。这工作正常。

现在我想用 JCufft 库执行完全相同的操作。我试图这样做：

private Complex[] fft(double[] values) {
   double inputJCufft[] = values.clone();
   double outputJCufft[] = new double[values.length * 2];
   cufftHandle plan = new cufftHandle();
   JCufft.cufftPlan1d(plan, values.length, cufftType.CUFFT_D2Z, 1);
   JCufft.cufftExecD2Z(plan, inputJCufft, outputJCufft);
   JCufft.cufftDestroy(plan);
   Complex[] result = BaseHelper.getComplexArray(outputJCufft);
   return result;
}

public static Complex[] getComplexArray(double[] input) {
    List<Complex> result = new ArrayList<Complex>();
    for (int i = 0; i < input.length - 1; i = i + 2) {
        result.add(new Complex(input[i], input[i + 1]));
    }
    return result.toArray(new Complex[result.size()]);
}

但是，当我比较结果时，它们彼此不同。我没有考虑到什么，我做错了什么？

谢谢你的帮助。

4

1 回答 1

1

好吧，这是我对 FFT 缺乏了解...

我将 getComplexArray 方法更改为以下内容，现在它可以正常工作：

public static Complex[] getComplexArray(double[] input) {
    Deque<Complex> deque = new LinkedList<Complex>();
    int size = (input.length / 4 + 1) * 2;
    for (int i = 0; i < size; i = i + 2) {
        deque.add(new Complex(input[i], input[i + 1]));
    }
    List<Complex> result = new ArrayList<Complex>(deque);
    deque.removeLast();
    while (deque.size() > 1) {
        result.add(deque.removeLast().conjugate());
    }
    return result.toArray(new Complex[result.size()]);
}
于 2015-02-13T15:29:22.270 回答