mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-14 14:20:45 +01:00
92ea95e34a
In https://webrtc-review.googlesource.com/c/src/+/1560 we moved WebRTC from src/webrtc to src/ (in order to preserve an healthy git history). This CL takes care of fixing header guards, #include paths, etc... NOPRESUBMIT=true NOTREECHECKS=true NOTRY=true TBR=tommi@webrtc.org Bug: chromium:611808 Change-Id: Iea91618212bee0af16aa3f05071eab8f93706578 Reviewed-on: https://webrtc-review.googlesource.com/1561 Reviewed-by: Mirko Bonadei <mbonadei@webrtc.org> Reviewed-by: Henrik Kjellander <kjellander@webrtc.org> Commit-Queue: Mirko Bonadei <mbonadei@webrtc.org> Cr-Commit-Position: refs/heads/master@{#19846}
106 lines
3 KiB
C++
106 lines
3 KiB
C++
/*
|
|
* Copyright (c) 2012 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.
|
|
*/
|
|
|
|
// Use CreateHistUnittestFile.m to generate the input file.
|
|
|
|
#include "modules/audio_processing/agc/loudness_histogram.h"
|
|
|
|
#include <stdio.h>
|
|
#include <algorithm>
|
|
#include <cmath>
|
|
#include <memory>
|
|
|
|
#include "modules/audio_processing/agc/utility.h"
|
|
#include "test/gtest.h"
|
|
#include "test/testsupport/fileutils.h"
|
|
|
|
namespace webrtc {
|
|
|
|
struct InputOutput {
|
|
double rms;
|
|
double activity_probability;
|
|
double audio_content;
|
|
double loudness;
|
|
};
|
|
|
|
const double kRelativeErrTol = 1e-10;
|
|
|
|
class LoudnessHistogramTest : public ::testing::Test {
|
|
protected:
|
|
void RunTest(bool enable_circular_buff, const char* filename);
|
|
|
|
private:
|
|
void TestClean();
|
|
std::unique_ptr<LoudnessHistogram> hist_;
|
|
};
|
|
|
|
void LoudnessHistogramTest::TestClean() {
|
|
EXPECT_EQ(hist_->CurrentRms(), 7.59621091765857e-02);
|
|
EXPECT_EQ(hist_->AudioContent(), 0);
|
|
EXPECT_EQ(hist_->num_updates(), 0);
|
|
}
|
|
|
|
void LoudnessHistogramTest::RunTest(bool enable_circular_buff,
|
|
const char* filename) {
|
|
FILE* in_file = fopen(filename, "rb");
|
|
ASSERT_TRUE(in_file != NULL);
|
|
if (enable_circular_buff) {
|
|
int buffer_size;
|
|
EXPECT_EQ(fread(&buffer_size, sizeof(buffer_size), 1, in_file), 1u);
|
|
hist_.reset(LoudnessHistogram::Create(buffer_size));
|
|
} else {
|
|
hist_.reset(LoudnessHistogram::Create());
|
|
}
|
|
TestClean();
|
|
|
|
InputOutput io;
|
|
int num_updates = 0;
|
|
int num_reset = 0;
|
|
while (fread(&io, sizeof(InputOutput), 1, in_file) == 1) {
|
|
if (io.rms < 0) {
|
|
// We have to reset.
|
|
hist_->Reset();
|
|
TestClean();
|
|
num_updates = 0;
|
|
num_reset++;
|
|
// Read the next chunk of input.
|
|
if (fread(&io, sizeof(InputOutput), 1, in_file) != 1)
|
|
break;
|
|
}
|
|
hist_->Update(io.rms, io.activity_probability);
|
|
num_updates++;
|
|
EXPECT_EQ(hist_->num_updates(), num_updates);
|
|
double audio_content = hist_->AudioContent();
|
|
|
|
double abs_err =
|
|
std::min(audio_content, io.audio_content) * kRelativeErrTol;
|
|
|
|
ASSERT_NEAR(audio_content, io.audio_content, abs_err);
|
|
double current_loudness = Linear2Loudness(hist_->CurrentRms());
|
|
abs_err =
|
|
std::min(fabs(current_loudness), fabs(io.loudness)) * kRelativeErrTol;
|
|
ASSERT_NEAR(current_loudness, io.loudness, abs_err);
|
|
}
|
|
fclose(in_file);
|
|
}
|
|
|
|
TEST_F(LoudnessHistogramTest, ActiveCircularBuffer) {
|
|
RunTest(true, test::ResourcePath(
|
|
"audio_processing/agc/agc_with_circular_buffer", "dat")
|
|
.c_str());
|
|
}
|
|
|
|
TEST_F(LoudnessHistogramTest, InactiveCircularBuffer) {
|
|
RunTest(false, test::ResourcePath(
|
|
"audio_processing/agc/agc_no_circular_buffer", "dat")
|
|
.c_str());
|
|
}
|
|
|
|
} // namespace webrtc
|