手动为双二阶 IIR 滤波器重新实现 vDSP_deq22

Reimplement vDSP_deq22 for Biquad IIR Filter by hand

我正在将当前使用 Apple 特定(Accelerate)vDSP 函数 vDSP_deq22 的滤波器组移植到 Android(其中 Accelerate 不可用)。滤波器组是一组带通滤波器,每个 return 各自频带的 RMS 幅度。目前代码(ObjectiveC++,改编自 NVDSP)如下所示:

- (float) filterContiguousData: (float *)data numFrames:(UInt32)numFrames channel:(UInt32)channel {

    // Init float to store RMS volume
    float rmsVolume = 0.0f;

    // Provide buffer for processing
    float tInputBuffer[numFrames + 2];
    float tOutputBuffer[numFrames + 2];

    // Copy the two frames we stored into the start of the inputBuffer, filling the rest with the current buffer data 
    memcpy(tInputBuffer, gInputKeepBuffer[channel], 2 * sizeof(float));
    memcpy(tOutputBuffer, gOutputKeepBuffer[channel], 2 * sizeof(float));
    memcpy(&(tInputBuffer[2]), data, numFrames * sizeof(float));

    // Do the processing
    vDSP_deq22(tInputBuffer, 1, coefficients, tOutputBuffer, 1, numFrames);
    vDSP_rmsqv(tOutputBuffer, 1, &rmsVolume, numFrames);

    // Copy the last two data points of each array to be put at the start of the next buffer.
    memcpy(gInputKeepBuffer[channel], &(tInputBuffer[numFrames]), 2 * sizeof(float));
    memcpy(gOutputKeepBuffer[channel], &(tOutputBuffer[numFrames]), 2 * sizeof(float));

    return rmsVolume;
}

here 所示,deq22 通过递归函数在给定的输入向量上实现双二阶滤波器。这是文档中函数的描述:

这是我目前所拥有的(它在 Swift 中,就像我已经在 Android 上 运行 的代码库的其余部分):

// N is fixed on init to be the same size as buffer.count, below
// 'input' and 'output' are initialised with (N+2) length and filled with 0s

func getFilteredRMSMagnitudeFromBuffer(var buffer: [Float]) -> Float {
    let inputStride = 1 // hardcoded for now
    let outputStride = 1

    input[0] = input[N]
    input[1] = input[N+1]
    output[0] = output[N]
    output[1] = output[N+1]

    // copy the current buffer into input
    input[2 ... N+1] = buffer[0 ..< N]

    // Not sure if this is neccessary, just here to duplicate NVDSP behaviour:
    output[2 ... N+1] = [Float](count: N, repeatedValue: 0)[0 ..< N]

    // Again duplicating NVDSP behaviour, can probably just start at 0:
    var sumOfSquares = (input[0] * input[0]) + (input[1] * input[1])

    for n in (2 ... N+1) {
        let sumG = (0...2).reduce(Float(0)) { total, p in
            return total + input[(n - p) * inputStride] * coefficients[p]
        }

        let sumH = (3...4).reduce(Float(0)) { total, p in
            return total + output[(n - p + 2) * outputStride] * coefficients[p]
        }

        let filteredFrame = sumG - sumH
        output[n] = filteredFrame
        sumOfSquares = filteredFrame * filteredFrame
    }

    let meanSquare = sumOfSquares / Float(N + 2) // we added 2 values by hand, before the loop
    let rootMeanSquare = sqrt(meanSquare)
    return rootMeanSquare
}

尽管滤波器为 deq22 提供了不同幅度的输出,并且其中似乎有一个循环的摆动循环 'noise'(在恒定输入音调下,该频率的幅度会上下波动)。

我已经检查以确保每个实现之间的系数数组相同。每个滤波器实际上似乎 "work" 因为它拾取了正确的频率(并且只有那个频率),它只是这个泵浦,并且 RMS 幅值输出比 vDSP 安静很多,通常是数量级:

   Naive    |    vDSP
3.24305e-06   0.000108608
1.57104e-06   5.53645e-05
1.96445e-06   4.33506e-05
2.05422e-06   2.09781e-05
1.44778e-06   1.8729e-05
4.28997e-07   2.72648e-05

有人能看出我的逻辑有问题吗?

编辑:这是一个具有恒定 440Hz 音调的结果的 gif 视频。各种绿色条是单独的滤波器带。第三波段(此处显示)是调谐到 440Hz 的波段。

NVDSP 版本仅显示与输入音量成比例的恒定(非波动)幅度读数,正如预期的那样。

好的,sumOfSquares = filteredFrame * filteredFrame 行应该是 +=,而不是赋值。所以只计算了最后一帧,解释了很多 ;)

如果您想在 Swift 中进行一些双二阶滤波,请随意使用它。 MIT License 就像之前的 NVDSP。