音频DSP波形折叠失真：32位整数转16位值的高效包裹方法

问题背景 #

我正在编写实现输入信号波形折叠失真的DSP代码：先对输入施加幅度增益（输入乘以增益值），随后执行波形折叠操作，让最终幅度落在**-32768至32767**之间（对应16位定点音频的-1到~1范围）——超过32767的值回绕到-32768，低于-32768的值则反向回绕。

之前在STM32硬件上，大增益下16位值会自动回滚，但现在要适配NXP硬件，必须主动控制该操作，不能依赖未充分文档化的硬件副作用。由于是实时DSP处理，对性能要求极高，原代码的while循环效率太低，急需更优实现。

原实现代码：

void wavefolding_distortion::update(void)
{
  audio_block_t *block = receiveWritable(0);

  if(block){
    for (int i=0; i < AUDIO_BLOCK_SAMPLES; i++) {
      int32_t in = block->data[i];

      in *= gain;

      while(in > 32767){
        in -= 65536;
      }

      while(in < -32768){
        in += 65536;
      }

      block->data[i] = in;
    }
    transmit(block);
    release(block);
  }

}

高效解决方案 #

方法1：位操作（性能最优，针对2^N范围） #

由于目标范围是2^16，直接对int32_t值保留低16位，再转换为int16_t即可利用补码特性自动完成回绕，这是最快的实现方式：

void wavefolding_distortion::update(void)
{
  audio_block_t *block = receiveWritable(0);

  if(block){
    for (int i=0; i < AUDIO_BLOCK_SAMPLES; i++) {
      int32_t in = block->data[i];
      in *= gain;

      // 保留低16位，通过int16_t补码特性自动完成波形折叠
      block->data[i] = static_cast<int16_t>(in & 0xFFFF);
    }
    transmit(block);
    release(block);
  }
}

原理：int16_t的补码表示中，超过32767的值会自动回绕到-32768开始，低于-32768的值反向回绕，完全符合波形折叠的逻辑。位运算无分支、单周期完成，适合实时DSP的高性能需求。

方法2：整数模运算（兼容性好） #

利用65536（2^16）作为模值，一次性完成回绕，需额外处理负数模运算的编译器差异：

void wavefolding_distortion::update(void)
{
  audio_block_t *block = receiveWritable(0);
  const int32_t MOD = 65536;
  const int32_t MAX = 32767;
  const int32_t MIN = -32768;

  if(block){
    for (int i=0; i < AUDIO_BLOCK_SAMPLES; i++) {
      int32_t in = block->data[i];
      in *= gain;

      // 取模后调整到目标区间
      in %= MOD;
      if (in > MAX) {
        in -= MOD;
      } else if (in < MIN) {
        in += MOD;
      }

      block->data[i] = static_cast<int16_t>(in);
    }
    transmit(block);
    release(block);
  }
}

注：部分编译器中负数取模会得到负数结果，因此需要额外判断调整，确保结果落在[-32768, 32767]区间。

方法3：分支判断+一次性偏移计算 #

通过计算超出范围的总倍数，一次性完成调整，避免循环迭代开销：

void wavefolding_distortion::update(void)
{
  audio_block_t *block = receiveWritable(0);
  const int32_t RANGE = 65536;
  const int32_t MAX = 32767;
  const int32_t MIN = -32768;

  if(block){
    for (int i=0; i < AUDIO_BLOCK_SAMPLES; i++) {
      int32_t in = block->data[i];
      in *= gain;

      if (in > MAX) {
        in -= ((in - MIN) / RANGE) * RANGE;
      } else if (in < MIN) {
        in += ((MIN - in) / RANGE) * RANGE;
      }

      block->data[i] = static_cast<int16_t>(in);
    }
    transmit(block);
    release(block);
  }
}

性能对比 #

• 原while循环：大增益下每个样本可能多次迭代，性能最差；
• 位操作法：无分支、单周期运算，性能最优，是实时场景的首选；
• 模运算/分支调整法：性能介于两者之间，兼容性更强。

内容的提问来源于stack exchange，提问作者Emmett Palaima