modules/audio_processing/utility/cascaded_biquad_filter_unittest.cc

/*
 *  Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */

#include "modules/audio_processing/utility/cascaded_biquad_filter.h"

#include <array>
#include <cstddef>
#include <vector>

#include "api/array_view.h"
#include "rtc_base/checks.h"
#include "test/gtest.h"

namespace webrtc {

namespace {

// Coefficients for a second order Butterworth high-pass filter with cutoff
// frequency 100 Hz.
constexpr std::array<CascadedBiQuadFilter::BiQuadCoefficients, 1>
    kHighPassFilterCoefficients = {{
        {{0.97261f, -1.94523f, 0.97261f}, {-1.94448f, 0.94598f}},
    }};

constexpr std::array<CascadedBiQuadFilter::BiQuadCoefficients, 1>
    kTransparentCoefficients = {{
        {{1.f, 0.f, 0.f}, {0.f, 0.f}},
    }};

constexpr std::array<CascadedBiQuadFilter::BiQuadCoefficients, 1>
    kBlockingCoefficients = {{
        {{0.f, 0.f, 0.f}, {0.f, 0.f}},
    }};

constexpr std::array<CascadedBiQuadFilter::BiQuadCoefficients, 2>
    kCascadedCoefficients = {{
        {{-1.f, 0.f, 0.f}, {0.f, 0.f}},
        {{0.f, 0.f, -1.f}, {0.f, 0.f}},
    }};

std::vector<float> CreateInputWithIncreasingValues(size_t vector_length) {
  std::vector<float> v(vector_length);
  for (size_t k = 0; k < v.size(); ++k) {
    v[k] = k;
  }
  return v;
}

}  // namespace

// Verifies that the filter applies an effect which removes the input signal.
// The test also verifies that the in-place Process API call works as intended.
TEST(CascadedBiquadFilter, BlockingConfiguration) {
  std::vector<float> values = CreateInputWithIncreasingValues(1000);

  CascadedBiQuadFilter filter(
      (ArrayView<const CascadedBiQuadFilter::BiQuadCoefficients>(
          kBlockingCoefficients)));
  filter.Process(values);

  EXPECT_EQ(std::vector<float>(1000, 0.f), values);
}

// Verifies that the filter is able to form a zero-mean output from a
// non-zeromean input signal when coefficients for a high-pass filter are
// applied. The test also verifies that the filter works with multiple biquads.
TEST(CascadedBiquadFilter, HighPassConfiguration) {
  std::vector<float> values(1000);
  for (size_t k = 0; k < values.size(); ++k) {
    values[k] = 1.f;
  }

  CascadedBiQuadFilter filter(
      (ArrayView<const CascadedBiQuadFilter::BiQuadCoefficients>(
          kHighPassFilterCoefficients)));
  filter.Process(values);

  for (size_t k = values.size() / 2; k < values.size(); ++k) {
    EXPECT_NEAR(0.f, values[k], 1e-2);
  }
}

// Verifies that the reset functionality works as intended.
TEST(CascadedBiquadFilter, HighPassConfigurationResetFunctionality) {
  CascadedBiQuadFilter filter(
      (ArrayView<const CascadedBiQuadFilter::BiQuadCoefficients>(
          kHighPassFilterCoefficients)));

  std::vector<float> values1(100, 1.f);
  filter.Process(values1);

  filter.Reset();

  std::vector<float> values2(100, 1.f);
  filter.Process(values2);

  for (size_t k = 0; k < values1.size(); ++k) {
    EXPECT_EQ(values1[k], values2[k]);
  }
}

// Verifies that the filter is able to produce a transparent effect with no
// impact on the data when the proper coefficients are applied. The test also
// verifies that the non-in-place Process API call works as intended.
TEST(CascadedBiquadFilter, TransparentConfiguration) {
  const std::vector<float> input = CreateInputWithIncreasingValues(1000);
  std::vector<float> output(input.size());

  CascadedBiQuadFilter filter(
      (ArrayView<const CascadedBiQuadFilter::BiQuadCoefficients>(
          kTransparentCoefficients)));
  filter.Process(input, output);

  EXPECT_EQ(input, output);
}

// Verifies that cascading multiple biquads works as expected.
TEST(CascadedBiquadFilter, CascadedConfiguration) {
  const std::vector<float> input = CreateInputWithIncreasingValues(1000);
  std::vector<float> output(input.size());

  CascadedBiQuadFilter filter(
      (ArrayView<const CascadedBiQuadFilter::BiQuadCoefficients>(
          kCascadedCoefficients)));
  filter.Process(input, output);

  // The cascaded biquad is expected to result in a delay of two samples.
  for (size_t i = 2; i < output.size(); ++i) {
    EXPECT_EQ(output[i], input[i - 2]);
  }
}

#if RTC_DCHECK_IS_ON && GTEST_HAS_DEATH_TEST && !defined(WEBRTC_ANDROID)
// Verifies that the check of the lengths for the input and output works for the
// non-in-place call.
TEST(CascadedBiquadFilterDeathTest, InputSizeCheckVerification) {
  const std::vector<float> input = CreateInputWithIncreasingValues(10);
  std::vector<float> output(input.size() - 1);

  CascadedBiQuadFilter filter(
      (ArrayView<const CascadedBiQuadFilter::BiQuadCoefficients>(
          kTransparentCoefficients)));
  EXPECT_DEATH(filter.Process(input, output), "");
}
#endif
}  // namespace webrtc