mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-13 13:50:40 +01:00
bb547203bf
In order to eliminate the WebRTC Subtree mirror in Chromium, WebRTC is moving the content of the src/webrtc directory up to the src/ directory. NOPRESUBMIT=true NOTREECHECKS=true NOTRY=true TBR=tommi@webrtc.org Bug: chromium:611808 Change-Id: Iac59c5b51b950f174119565bac87955a7994bc38 Reviewed-on: https://webrtc-review.googlesource.com/1560 Commit-Queue: Mirko Bonadei <mbonadei@webrtc.org> Reviewed-by: Henrik Kjellander <kjellander@webrtc.org> Cr-Commit-Position: refs/heads/master@{#19845}
122 lines
4.3 KiB
C++
122 lines
4.3 KiB
C++
/*
|
|
* Copyright (c) 2017 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 "webrtc/modules/audio_processing/aec3/render_signal_analyzer.h"
|
|
|
|
#include <math.h>
|
|
#include <array>
|
|
#include <vector>
|
|
|
|
#include "webrtc/api/array_view.h"
|
|
#include "webrtc/modules/audio_processing/aec3/aec3_common.h"
|
|
#include "webrtc/modules/audio_processing/aec3/aec3_fft.h"
|
|
#include "webrtc/modules/audio_processing/aec3/fft_data.h"
|
|
#include "webrtc/modules/audio_processing/aec3/render_buffer.h"
|
|
#include "webrtc/modules/audio_processing/test/echo_canceller_test_tools.h"
|
|
#include "webrtc/rtc_base/random.h"
|
|
#include "webrtc/test/gtest.h"
|
|
|
|
namespace webrtc {
|
|
namespace {
|
|
|
|
constexpr float kPi = 3.141592f;
|
|
|
|
void ProduceSinusoid(int sample_rate_hz,
|
|
float sinusoidal_frequency_hz,
|
|
size_t* sample_counter,
|
|
rtc::ArrayView<float> x) {
|
|
// Produce a sinusoid of the specified frequency.
|
|
for (size_t k = *sample_counter, j = 0; k < (*sample_counter + kBlockSize);
|
|
++k, ++j) {
|
|
x[j] =
|
|
32767.f * sin(2.f * kPi * sinusoidal_frequency_hz * k / sample_rate_hz);
|
|
}
|
|
*sample_counter = *sample_counter + kBlockSize;
|
|
}
|
|
|
|
} // namespace
|
|
|
|
#if RTC_DCHECK_IS_ON && GTEST_HAS_DEATH_TEST && !defined(WEBRTC_ANDROID)
|
|
// Verifies that the check for non-null output parameter works.
|
|
TEST(RenderSignalAnalyzer, NullMaskOutput) {
|
|
RenderSignalAnalyzer analyzer;
|
|
EXPECT_DEATH(analyzer.MaskRegionsAroundNarrowBands(nullptr), "");
|
|
}
|
|
|
|
#endif
|
|
|
|
// Verify that no narrow bands are detected in a Gaussian noise signal.
|
|
TEST(RenderSignalAnalyzer, NoFalseDetectionOfNarrowBands) {
|
|
RenderSignalAnalyzer analyzer;
|
|
Random random_generator(42U);
|
|
std::vector<std::vector<float>> x(3, std::vector<float>(kBlockSize, 0.f));
|
|
std::array<float, kBlockSize> x_old;
|
|
FftData X;
|
|
Aec3Fft fft;
|
|
RenderBuffer render_buffer(Aec3Optimization::kNone, 3, 1,
|
|
std::vector<size_t>(1, 1));
|
|
std::array<float, kFftLengthBy2Plus1> mask;
|
|
x_old.fill(0.f);
|
|
|
|
for (size_t k = 0; k < 100; ++k) {
|
|
RandomizeSampleVector(&random_generator, x[0]);
|
|
fft.PaddedFft(x[0], x_old, &X);
|
|
render_buffer.Insert(x);
|
|
analyzer.Update(render_buffer, rtc::Optional<size_t>(0));
|
|
}
|
|
|
|
mask.fill(1.f);
|
|
analyzer.MaskRegionsAroundNarrowBands(&mask);
|
|
EXPECT_TRUE(
|
|
std::all_of(mask.begin(), mask.end(), [](float a) { return a == 1.f; }));
|
|
EXPECT_FALSE(analyzer.PoorSignalExcitation());
|
|
}
|
|
|
|
// Verify that a sinusiod signal is detected as narrow bands.
|
|
TEST(RenderSignalAnalyzer, NarrowBandDetection) {
|
|
RenderSignalAnalyzer analyzer;
|
|
Random random_generator(42U);
|
|
std::vector<std::vector<float>> x(3, std::vector<float>(kBlockSize, 0.f));
|
|
std::array<float, kBlockSize> x_old;
|
|
Aec3Fft fft;
|
|
RenderBuffer render_buffer(Aec3Optimization::kNone, 3, 1,
|
|
std::vector<size_t>(1, 1));
|
|
std::array<float, kFftLengthBy2Plus1> mask;
|
|
x_old.fill(0.f);
|
|
constexpr int kSinusFrequencyBin = 32;
|
|
|
|
auto generate_sinusoid_test = [&](bool known_delay) {
|
|
size_t sample_counter = 0;
|
|
for (size_t k = 0; k < 100; ++k) {
|
|
ProduceSinusoid(16000, 16000 / 2 * kSinusFrequencyBin / kFftLengthBy2,
|
|
&sample_counter, x[0]);
|
|
render_buffer.Insert(x);
|
|
analyzer.Update(render_buffer, known_delay ? rtc::Optional<size_t>(0)
|
|
: rtc::Optional<size_t>());
|
|
}
|
|
};
|
|
|
|
generate_sinusoid_test(true);
|
|
mask.fill(1.f);
|
|
analyzer.MaskRegionsAroundNarrowBands(&mask);
|
|
for (int k = 0; k < static_cast<int>(mask.size()); ++k) {
|
|
EXPECT_EQ(abs(k - kSinusFrequencyBin) <= 2 ? 0.f : 1.f, mask[k]);
|
|
}
|
|
EXPECT_TRUE(analyzer.PoorSignalExcitation());
|
|
|
|
// Verify that no bands are detected as narrow when the delay is unknown.
|
|
generate_sinusoid_test(false);
|
|
mask.fill(1.f);
|
|
analyzer.MaskRegionsAroundNarrowBands(&mask);
|
|
std::for_each(mask.begin(), mask.end(), [](float a) { EXPECT_EQ(1.f, a); });
|
|
EXPECT_FALSE(analyzer.PoorSignalExcitation());
|
|
}
|
|
|
|
} // namespace webrtc
|