Moving src/webrtc into src/.
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}
This commit is contained in:
committed by
Commit Bot
parent
6674846b4a
commit
bb547203bf
150
modules/audio_processing/rms_level_unittest.cc
Normal file
150
modules/audio_processing/rms_level_unittest.cc
Normal file
@ -0,0 +1,150 @@
|
||||
/*
|
||||
* 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 <cmath>
|
||||
#include <memory>
|
||||
#include <vector>
|
||||
|
||||
#include "webrtc/api/array_view.h"
|
||||
#include "webrtc/modules/audio_processing/rms_level.h"
|
||||
#include "webrtc/rtc_base/checks.h"
|
||||
#include "webrtc/rtc_base/mathutils.h"
|
||||
#include "webrtc/rtc_base/safe_conversions.h"
|
||||
#include "webrtc/test/gtest.h"
|
||||
|
||||
namespace webrtc {
|
||||
namespace {
|
||||
constexpr int kSampleRateHz = 48000;
|
||||
constexpr size_t kBlockSizeSamples = kSampleRateHz / 100;
|
||||
|
||||
std::unique_ptr<RmsLevel> RunTest(rtc::ArrayView<const int16_t> input) {
|
||||
std::unique_ptr<RmsLevel> level(new RmsLevel);
|
||||
for (size_t n = 0; n + kBlockSizeSamples <= input.size();
|
||||
n += kBlockSizeSamples) {
|
||||
level->Analyze(input.subview(n, kBlockSizeSamples));
|
||||
}
|
||||
return level;
|
||||
}
|
||||
|
||||
std::vector<int16_t> CreateSinusoid(int frequency_hz,
|
||||
int amplitude,
|
||||
size_t num_samples) {
|
||||
std::vector<int16_t> x(num_samples);
|
||||
for (size_t n = 0; n < num_samples; ++n) {
|
||||
x[n] = rtc::saturated_cast<int16_t>(
|
||||
amplitude * std::sin(2 * M_PI * n * frequency_hz / kSampleRateHz));
|
||||
}
|
||||
return x;
|
||||
}
|
||||
} // namespace
|
||||
|
||||
TEST(RmsLevelTest, Run1000HzFullScale) {
|
||||
auto x = CreateSinusoid(1000, INT16_MAX, kSampleRateHz);
|
||||
auto level = RunTest(x);
|
||||
EXPECT_EQ(3, level->Average()); // -3 dBFS
|
||||
}
|
||||
|
||||
TEST(RmsLevelTest, Run1000HzFullScaleAverageAndPeak) {
|
||||
auto x = CreateSinusoid(1000, INT16_MAX, kSampleRateHz);
|
||||
auto level = RunTest(x);
|
||||
auto stats = level->AverageAndPeak();
|
||||
EXPECT_EQ(3, stats.average); // -3 dBFS
|
||||
EXPECT_EQ(3, stats.peak);
|
||||
}
|
||||
|
||||
TEST(RmsLevelTest, Run1000HzHalfScale) {
|
||||
auto x = CreateSinusoid(1000, INT16_MAX / 2, kSampleRateHz);
|
||||
auto level = RunTest(x);
|
||||
EXPECT_EQ(9, level->Average()); // -9 dBFS
|
||||
}
|
||||
|
||||
TEST(RmsLevelTest, RunZeros) {
|
||||
std::vector<int16_t> x(kSampleRateHz, 0); // 1 second of pure silence.
|
||||
auto level = RunTest(x);
|
||||
EXPECT_EQ(127, level->Average());
|
||||
}
|
||||
|
||||
TEST(RmsLevelTest, RunZerosAverageAndPeak) {
|
||||
std::vector<int16_t> x(kSampleRateHz, 0); // 1 second of pure silence.
|
||||
auto level = RunTest(x);
|
||||
auto stats = level->AverageAndPeak();
|
||||
EXPECT_EQ(127, stats.average);
|
||||
EXPECT_EQ(127, stats.peak);
|
||||
}
|
||||
|
||||
TEST(RmsLevelTest, NoSamples) {
|
||||
RmsLevel level;
|
||||
EXPECT_EQ(127, level.Average()); // Return minimum if no samples are given.
|
||||
}
|
||||
|
||||
TEST(RmsLevelTest, NoSamplesAverageAndPeak) {
|
||||
RmsLevel level;
|
||||
auto stats = level.AverageAndPeak();
|
||||
EXPECT_EQ(127, stats.average);
|
||||
EXPECT_EQ(127, stats.peak);
|
||||
}
|
||||
|
||||
TEST(RmsLevelTest, PollTwice) {
|
||||
auto x = CreateSinusoid(1000, INT16_MAX, kSampleRateHz);
|
||||
auto level = RunTest(x);
|
||||
level->Average();
|
||||
EXPECT_EQ(127, level->Average()); // Stats should be reset at this point.
|
||||
}
|
||||
|
||||
TEST(RmsLevelTest, Reset) {
|
||||
auto x = CreateSinusoid(1000, INT16_MAX, kSampleRateHz);
|
||||
auto level = RunTest(x);
|
||||
level->Reset();
|
||||
EXPECT_EQ(127, level->Average()); // Stats should be reset at this point.
|
||||
}
|
||||
|
||||
// Inserts 1 second of full-scale sinusoid, followed by 1 second of muted.
|
||||
TEST(RmsLevelTest, ProcessMuted) {
|
||||
auto x = CreateSinusoid(1000, INT16_MAX, kSampleRateHz);
|
||||
auto level = RunTest(x);
|
||||
const size_t kBlocksPerSecond = rtc::CheckedDivExact(
|
||||
static_cast<size_t>(kSampleRateHz), kBlockSizeSamples);
|
||||
for (size_t i = 0; i < kBlocksPerSecond; ++i) {
|
||||
level->AnalyzeMuted(kBlockSizeSamples);
|
||||
}
|
||||
EXPECT_EQ(6, level->Average()); // Average RMS halved due to the silence.
|
||||
}
|
||||
|
||||
// Inserts 1 second of half-scale sinusoid, follwed by 10 ms of full-scale, and
|
||||
// finally 1 second of half-scale again. Expect the average to be -9 dBFS due
|
||||
// to the vast majority of the signal being half-scale, and the peak to be
|
||||
// -3 dBFS.
|
||||
TEST(RmsLevelTest, RunHalfScaleAndInsertFullScale) {
|
||||
auto half_scale = CreateSinusoid(1000, INT16_MAX / 2, kSampleRateHz);
|
||||
auto full_scale = CreateSinusoid(1000, INT16_MAX, kSampleRateHz / 100);
|
||||
auto x = half_scale;
|
||||
x.insert(x.end(), full_scale.begin(), full_scale.end());
|
||||
x.insert(x.end(), half_scale.begin(), half_scale.end());
|
||||
ASSERT_EQ(static_cast<size_t>(2 * kSampleRateHz + kSampleRateHz / 100),
|
||||
x.size());
|
||||
auto level = RunTest(x);
|
||||
auto stats = level->AverageAndPeak();
|
||||
EXPECT_EQ(9, stats.average);
|
||||
EXPECT_EQ(3, stats.peak);
|
||||
}
|
||||
|
||||
TEST(RmsLevelTest, ResetOnBlockSizeChange) {
|
||||
auto x = CreateSinusoid(1000, INT16_MAX, kSampleRateHz);
|
||||
auto level = RunTest(x);
|
||||
// Create a new signal with half amplitude, but double block length.
|
||||
auto y = CreateSinusoid(1000, INT16_MAX / 2, kBlockSizeSamples * 2);
|
||||
level->Analyze(y);
|
||||
auto stats = level->AverageAndPeak();
|
||||
// Expect all stats to only be influenced by the last signal (y), since the
|
||||
// changed block size should reset the stats.
|
||||
EXPECT_EQ(9, stats.average);
|
||||
EXPECT_EQ(9, stats.peak);
|
||||
}
|
||||
|
||||
} // namespace webrtc
|
||||
Reference in New Issue
Block a user