blob: 47d8bf531fb4de3099a540d6f37fa02f59e4a99c [file] [log] [blame]
/*
* Copyright (c) 2018 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/agc2/rnn_vad/lp_residual.h"
#include <array>
#include "modules/audio_processing/agc2/rnn_vad/common.h"
#include "modules/audio_processing/agc2/rnn_vad/test_utils.h"
// TODO(bugs.webrtc.org/8948): Add when the issue is fixed.
// #include "test/fpe_observer.h"
#include "test/gtest.h"
namespace webrtc {
namespace rnn_vad {
namespace test {
TEST(RnnVadTest, LpResidualOfEmptyFrame) {
// TODO(bugs.webrtc.org/8948): Add when the issue is fixed.
// FloatingPointExceptionObserver fpe_observer;
// Input frame (empty, i.e., all samples set to 0).
std::array<float, kFrameSize10ms24kHz> empty_frame;
empty_frame.fill(0.f);
// Compute inverse filter coefficients.
std::array<float, kNumLpcCoefficients> lpc_coeffs;
ComputeAndPostProcessLpcCoefficients(empty_frame, lpc_coeffs);
// Compute LP residual.
std::array<float, kFrameSize10ms24kHz> lp_residual;
ComputeLpResidual(lpc_coeffs, empty_frame, lp_residual);
}
// TODO(bugs.webrtc.org/9076): Remove when the issue is fixed.
TEST(RnnVadTest, LpResidualPipelineBitExactness) {
// Pitch buffer 24 kHz data reader.
auto pitch_buf_24kHz_reader = CreatePitchBuffer24kHzReader();
const size_t num_frames = pitch_buf_24kHz_reader.second;
std::array<float, kBufSize24kHz> pitch_buf_data;
// Read ground-truth.
auto lp_residual_reader = CreateLpResidualAndPitchPeriodGainReader();
ASSERT_EQ(num_frames, lp_residual_reader.second);
std::array<float, kBufSize24kHz> expected_lp_residual;
rtc::ArrayView<float, kBufSize24kHz> expected_lp_residual_view(
expected_lp_residual.data(), expected_lp_residual.size());
// Init pipeline.
std::array<float, kNumLpcCoefficients> lpc_coeffs;
rtc::ArrayView<float, kNumLpcCoefficients> lpc_coeffs_view(
lpc_coeffs.data(), kNumLpcCoefficients);
std::array<float, kBufSize24kHz> computed_lp_residual;
rtc::ArrayView<float, kBufSize24kHz> computed_lp_residual_view(
computed_lp_residual.data(), computed_lp_residual.size());
{
// TODO(bugs.webrtc.org/8948): Add when the issue is fixed.
// FloatingPointExceptionObserver fpe_observer;
for (size_t i = 0; i < num_frames; ++i) {
SCOPED_TRACE(i);
// Read input and expected output.
pitch_buf_24kHz_reader.first->ReadChunk(pitch_buf_data);
lp_residual_reader.first->ReadChunk(expected_lp_residual_view);
// Skip pitch gain and period.
float unused;
lp_residual_reader.first->ReadValue(&unused);
lp_residual_reader.first->ReadValue(&unused);
// Run pipeline.
ComputeAndPostProcessLpcCoefficients(pitch_buf_data, lpc_coeffs_view);
ComputeLpResidual(lpc_coeffs_view, pitch_buf_data,
computed_lp_residual_view);
// Compare.
ExpectNearAbsolute(expected_lp_residual_view, computed_lp_residual_view,
kFloatMin);
}
}
}
} // namespace test
} // namespace rnn_vad
} // namespace webrtc