Java中可靠,快速的FFT

时间:2010-07-20 06:25:02

标签: java fft

因为我不想自己做,所以我正在为java寻找一个好的FFT实现。首先我在这里FFT Princeton使用了这个,但是它使用了对象,而我的分析器告诉我,由于这个事实它并不是很快。所以我再次搜索并找到了这个:FFT Columbia这更快。也许你们其中一个人知道另一个FFT实现?我想拥有“最好的”,因为我的应用程序必须处理大量的声音数据,用户不喜欢等待......; - )

问候。

5 个答案:

答案 0 :(得分:31)

FFTW是“西方最快的傅立叶变换”,并且有一些Java包装器:

  

http://www.fftw.org/download.html

希望有所帮助!

答案 1 :(得分:18)

聚会晚了 - 这里是一个纯粹的Java解决方案,适用于JNI无法选择的人。JTransforms

答案 2 :(得分:11)

我在Java中编写了一个FFT函数:http://www.wikijava.org/wiki/The_Fast_Fourier_Transform_in_Java_%28part_1%29

它位于公共域中,因此您可以在任何地方使用这些功能(个人或商业项目)。只需在信用卡中引用我,然后向我发送一份工作链接,您就可以了。

完全可靠。我已经根据Mathematica的FFT检查了它的输出,它们总是正确的,直到第15个十进制数字。我认为这是一个非常好的Java实现。我在J2SE 1.6版本上编写了它,并在J2SE 1.5-1.6版本上进行了测试。

如果计算指令的数量(它比完美的计算复杂度函数估计简单得多),你可以清楚地看到这个版本很好,即使它根本没有优化。如果有足够的请求,我打算发布优化版本。

让我知道它是否有用,并告诉我你喜欢的任何评论。

我在这里分享相同的代码:

/**
* @author Orlando Selenu
*
*/
public class FFTbase {
/**
 * The Fast Fourier Transform (generic version, with NO optimizations).
 *
 * @param inputReal
 *            an array of length n, the real part
 * @param inputImag
 *            an array of length n, the imaginary part
 * @param DIRECT
 *            TRUE = direct transform, FALSE = inverse transform
 * @return a new array of length 2n
 */
public static double[] fft(final double[] inputReal, double[] inputImag,
                           boolean DIRECT) {
    // - n is the dimension of the problem
    // - nu is its logarithm in base e
    int n = inputReal.length;

    // If n is a power of 2, then ld is an integer (_without_ decimals)
    double ld = Math.log(n) / Math.log(2.0);

    // Here I check if n is a power of 2. If exist decimals in ld, I quit
    // from the function returning null.
    if (((int) ld) - ld != 0) {
        System.out.println("The number of elements is not a power of 2.");
        return null;
    }

    // Declaration and initialization of the variables
    // ld should be an integer, actually, so I don't lose any information in
    // the cast
    int nu = (int) ld;
    int n2 = n / 2;
    int nu1 = nu - 1;
    double[] xReal = new double[n];
    double[] xImag = new double[n];
    double tReal, tImag, p, arg, c, s;

    // Here I check if I'm going to do the direct transform or the inverse
    // transform.
    double constant;
    if (DIRECT)
        constant = -2 * Math.PI;
    else
        constant = 2 * Math.PI;

    // I don't want to overwrite the input arrays, so here I copy them. This
    // choice adds \Theta(2n) to the complexity.
    for (int i = 0; i < n; i++) {
        xReal[i] = inputReal[i];
        xImag[i] = inputImag[i];
    }

    // First phase - calculation
    int k = 0;
    for (int l = 1; l <= nu; l++) {
        while (k < n) {
            for (int i = 1; i <= n2; i++) {
                p = bitreverseReference(k >> nu1, nu);
                // direct FFT or inverse FFT
                arg = constant * p / n;
                c = Math.cos(arg);
                s = Math.sin(arg);
                tReal = xReal[k + n2] * c + xImag[k + n2] * s;
                tImag = xImag[k + n2] * c - xReal[k + n2] * s;
                xReal[k + n2] = xReal[k] - tReal;
                xImag[k + n2] = xImag[k] - tImag;
                xReal[k] += tReal;
                xImag[k] += tImag;
                k++;
            }
            k += n2;
        }
        k = 0;
        nu1--;
        n2 /= 2;
    }

    // Second phase - recombination
    k = 0;
    int r;
    while (k < n) {
        r = bitreverseReference(k, nu);
        if (r > k) {
            tReal = xReal[k];
            tImag = xImag[k];
            xReal[k] = xReal[r];
            xImag[k] = xImag[r];
            xReal[r] = tReal;
            xImag[r] = tImag;
        }
        k++;
    }

    // Here I have to mix xReal and xImag to have an array (yes, it should
    // be possible to do this stuff in the earlier parts of the code, but
    // it's here to readibility).
    double[] newArray = new double[xReal.length * 2];
    double radice = 1 / Math.sqrt(n);
    for (int i = 0; i < newArray.length; i += 2) {
        int i2 = i / 2;
        // I used Stephen Wolfram's Mathematica as a reference so I'm going
        // to normalize the output while I'm copying the elements.
        newArray[i] = xReal[i2] * radice;
        newArray[i + 1] = xImag[i2] * radice;
    }
    return newArray;
}

/**
 * The reference bitreverse function.
 */
private static int bitreverseReference(int j, int nu) {
    int j2;
    int j1 = j;
    int k = 0;
    for (int i = 1; i <= nu; i++) {
        j2 = j1 / 2;
        k = 2 * k + j1 - 2 * j2;
        j1 = j2;
    }
    return k;
  }
}

答案 3 :(得分:4)

我正在考虑在Java中使用SSTJ进行FFT。如果库可用,它可以通过JNI重定向到FFTW,否则将使用纯Java实现。

答案 4 :(得分:4)

我想这取决于你正在处理什么。如果您在很长的时间内计算FFT,您可能会发现它确实需要一段时间,具体取决于您想要的频率点数。然而,在大多数情况下,对于音频,它被认为是非平稳的(即信号均值和方差随时间变化很大),因此采用一个大的FFT(Periodogram PSD估计)不是准确的表示。或者,您可以使用短时傅立叶变换,从而将信号分解为更小的帧并计算FFT。帧大小取决于统计数据的变化速度,对于语音通常为20-40ms,对于音乐,我认为它略高一些。

如果您从麦克风采样,这种方法很好,因为它允许您一次缓冲每个帧,计算fft并给出用户感觉“实时”交互的内容。因为20ms很快,因为我们无法真正感知到时间差异那么小。

我开发了一个小基准测试来测试语音信号上FFTW和KissFFT c库之间的差异。是FFTW是高度优化的,但是当你只拍摄短帧,为用户更新数据,并且只使用很小的fft时,它们都非常相似。这是一个关于如何通过badlogic游戏使用LibGdx实现KissFFT libraries in Android的示例。我在几个月前开发的一款名为Speech Enhancement for Android的Android App中使用重叠帧实现了这个库。