mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-18 16:17:50 +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}
99 lines
3.1 KiB
C++
99 lines
3.1 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 <memory>
|
|
#include <string>
|
|
|
|
#include "webrtc/api/array_view.h"
|
|
#include "webrtc/modules/audio_processing/agc2/digital_gain_applier.h"
|
|
#include "webrtc/modules/audio_processing/agc2/gain_controller2.h"
|
|
#include "webrtc/modules/audio_processing/audio_buffer.h"
|
|
#include "webrtc/test/gtest.h"
|
|
|
|
namespace webrtc {
|
|
namespace test {
|
|
|
|
namespace {
|
|
|
|
constexpr size_t kNumFrames = 480u;
|
|
constexpr size_t kStereo = 2u;
|
|
|
|
void SetAudioBufferSamples(float value, AudioBuffer* ab) {
|
|
for (size_t k = 0; k < ab->num_channels(); ++k) {
|
|
auto channel = rtc::ArrayView<float>(ab->channels_f()[k], ab->num_frames());
|
|
for (auto& sample : channel) { sample = value; }
|
|
}
|
|
}
|
|
|
|
template<typename Functor>
|
|
bool CheckAudioBufferSamples(Functor validator, AudioBuffer* ab) {
|
|
for (size_t k = 0; k < ab->num_channels(); ++k) {
|
|
auto channel = rtc::ArrayView<float>(ab->channels_f()[k], ab->num_frames());
|
|
for (auto& sample : channel) { if (!validator(sample)) { return false; } }
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool TestDigitalGainApplier(float sample_value, float gain, float expected) {
|
|
AudioBuffer ab(kNumFrames, kStereo, kNumFrames, kStereo, kNumFrames);
|
|
SetAudioBufferSamples(sample_value, &ab);
|
|
|
|
DigitalGainApplier gain_applier;
|
|
for (size_t k = 0; k < ab.num_channels(); ++k) {
|
|
auto channel_view = rtc::ArrayView<float>(
|
|
ab.channels_f()[k], ab.num_frames());
|
|
gain_applier.Process(gain, channel_view);
|
|
}
|
|
|
|
auto check_expectation = [expected](float sample) {
|
|
return sample == expected; };
|
|
return CheckAudioBufferSamples(check_expectation, &ab);
|
|
}
|
|
|
|
} // namespace
|
|
|
|
TEST(GainController2, Instance) {
|
|
std::unique_ptr<GainController2> gain_controller2;
|
|
gain_controller2.reset(new GainController2(
|
|
AudioProcessing::kSampleRate48kHz));
|
|
}
|
|
|
|
TEST(GainController2, ToString) {
|
|
AudioProcessing::Config config;
|
|
|
|
config.gain_controller2.enabled = false;
|
|
EXPECT_EQ("{enabled: false}",
|
|
GainController2::ToString(config.gain_controller2));
|
|
|
|
config.gain_controller2.enabled = true;
|
|
EXPECT_EQ("{enabled: true}",
|
|
GainController2::ToString(config.gain_controller2));
|
|
}
|
|
|
|
TEST(GainController2, DigitalGainApplierProcess) {
|
|
EXPECT_TRUE(TestDigitalGainApplier(1000.0f, 0.5, 500.0f));
|
|
}
|
|
|
|
TEST(GainController2, DigitalGainApplierCheckClipping) {
|
|
EXPECT_TRUE(TestDigitalGainApplier(30000.0f, 1.5, 32767.0f));
|
|
EXPECT_TRUE(TestDigitalGainApplier(-30000.0f, 1.5, -32767.0f));
|
|
}
|
|
|
|
TEST(GainController2, Usage) {
|
|
std::unique_ptr<GainController2> gain_controller2;
|
|
gain_controller2.reset(new GainController2(
|
|
AudioProcessing::kSampleRate48kHz));
|
|
AudioBuffer ab(kNumFrames, kStereo, kNumFrames, kStereo, kNumFrames);
|
|
SetAudioBufferSamples(1000.0f, &ab);
|
|
gain_controller2->Process(&ab);
|
|
}
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|