blob: 42cf4188fcbf326f9d14fc72cc318c8817d78f04 [file] [log] [blame]
niklase@google.com470e71d2011-07-07 08:21:25 +00001/*
andrew@webrtc.org293d22b2012-01-30 22:04:26 +00002 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
niklase@google.com470e71d2011-07-07 08:21:25 +00003 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +000011#include <math.h>
ajm@google.com59e41402011-07-28 17:34:04 +000012#include <stdio.h>
kwiberg62eaacf2016-02-17 06:39:05 -080013
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +000014#include <algorithm>
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +000015#include <limits>
kwiberg62eaacf2016-02-17 06:39:05 -080016#include <memory>
bjornv@webrtc.org3e102492013-02-14 15:29:09 +000017#include <queue>
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +000018
pkasting25702cb2016-01-08 13:50:27 -080019#include "webrtc/base/arraysize.h"
kwiberg9e2be5f2016-09-14 05:23:22 -070020#include "webrtc/base/checks.h"
peahc19f3122016-10-07 14:54:10 -070021#include "webrtc/base/gtest_prod_util.h"
kwiberg77eab702016-09-28 17:42:01 -070022#include "webrtc/base/ignore_wundef.h"
mbonadei7c2c8432017-04-07 00:59:12 -070023#include "webrtc/base/protobuf_utils.h"
kwiberg7885d3f2017-04-25 12:35:07 -070024#include "webrtc/base/safe_minmax.h"
andrew@webrtc.org27c69802014-02-18 20:24:56 +000025#include "webrtc/common_audio/include/audio_util.h"
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +000026#include "webrtc/common_audio/resampler/include/push_resampler.h"
27#include "webrtc/common_audio/resampler/push_sinc_resampler.h"
bjornv@webrtc.org3e102492013-02-14 15:29:09 +000028#include "webrtc/common_audio/signal_processing/include/signal_processing_library.h"
peahc19f3122016-10-07 14:54:10 -070029#include "webrtc/modules/audio_processing/audio_processing_impl.h"
mgraczyk@chromium.org0f663de2015-03-13 00:13:32 +000030#include "webrtc/modules/audio_processing/beamformer/mock_nonlinear_beamformer.h"
aluebs@webrtc.org87893762014-11-27 23:40:25 +000031#include "webrtc/modules/audio_processing/common.h"
bjornv@webrtc.org3e102492013-02-14 15:29:09 +000032#include "webrtc/modules/audio_processing/include/audio_processing.h"
peahc19f3122016-10-07 14:54:10 -070033#include "webrtc/modules/audio_processing/level_controller/level_controller_constants.h"
Andrew MacDonaldcb05b722015-05-07 22:17:51 -070034#include "webrtc/modules/audio_processing/test/protobuf_utils.h"
andrew@webrtc.org60730cf2014-01-07 17:45:09 +000035#include "webrtc/modules/audio_processing/test/test_utils.h"
Henrik Kjellanderff761fb2015-11-04 08:31:52 +010036#include "webrtc/modules/include/module_common_types.h"
Henrik Kjellander98f53512015-10-28 18:17:40 +010037#include "webrtc/system_wrappers/include/event_wrapper.h"
38#include "webrtc/system_wrappers/include/trace.h"
kwiberg77eab702016-09-28 17:42:01 -070039#include "webrtc/test/gtest.h"
bjornv@webrtc.org3e102492013-02-14 15:29:09 +000040#include "webrtc/test/testsupport/fileutils.h"
kwiberg77eab702016-09-28 17:42:01 -070041
42RTC_PUSH_IGNORING_WUNDEF()
leozwang@webrtc.orga3736342012-03-16 21:36:00 +000043#ifdef WEBRTC_ANDROID_PLATFORM_BUILD
leozwang@webrtc.org534e4952012-10-22 21:21:52 +000044#include "external/webrtc/webrtc/modules/audio_processing/test/unittest.pb.h"
leozwang@webrtc.orga3736342012-03-16 21:36:00 +000045#else
kjellandere3e902e2017-02-28 08:01:46 -080046#include "webrtc/modules/audio_processing/test/unittest.pb.h"
leozwang@webrtc.orga3736342012-03-16 21:36:00 +000047#endif
kwiberg77eab702016-09-28 17:42:01 -070048RTC_POP_IGNORING_WUNDEF()
niklase@google.com470e71d2011-07-07 08:21:25 +000049
andrew@webrtc.org27c69802014-02-18 20:24:56 +000050namespace webrtc {
niklase@google.com470e71d2011-07-07 08:21:25 +000051namespace {
andrew@webrtc.org17e40642014-03-04 20:58:13 +000052
ekmeyerson60d9b332015-08-14 10:35:55 -070053// TODO(ekmeyerson): Switch to using StreamConfig and ProcessingConfig where
54// applicable.
55
bjornv@webrtc.orgbbd47fc2014-01-13 08:54:34 +000056// TODO(bjornv): This is not feasible until the functionality has been
bjornv@webrtc.org8dd60cc2014-09-11 08:36:35 +000057// re-implemented; see comment at the bottom of this file. For now, the user has
58// to hard code the |write_ref_data| value.
ajm@google.com59e41402011-07-28 17:34:04 +000059// When false, this will compare the output data with the results stored to
niklase@google.com470e71d2011-07-07 08:21:25 +000060// file. This is the typical case. When the file should be updated, it can
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +000061// be set to true with the command-line switch --write_ref_data.
62bool write_ref_data = false;
mbonadei7c2c8432017-04-07 00:59:12 -070063const int32_t kChannels[] = {1, 2};
Alejandro Luebs47748742015-05-22 12:00:21 -070064const int kSampleRates[] = {8000, 16000, 32000, 48000};
andrew@webrtc.orgecac9b72012-05-02 00:04:10 +000065
aluebseb3603b2016-04-20 15:27:58 -070066#if defined(WEBRTC_AUDIOPROC_FIXED_PROFILE)
67// Android doesn't support 48kHz.
68const int kProcessSampleRates[] = {8000, 16000, 32000};
69#elif defined(WEBRTC_AUDIOPROC_FLOAT_PROFILE)
Alejandro Luebs47748742015-05-22 12:00:21 -070070const int kProcessSampleRates[] = {8000, 16000, 32000, 48000};
aluebseb3603b2016-04-20 15:27:58 -070071#endif
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +000072
ekmeyerson60d9b332015-08-14 10:35:55 -070073enum StreamDirection { kForward = 0, kReverse };
74
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +000075void ConvertToFloat(const int16_t* int_data, ChannelBuffer<float>* cb) {
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +000076 ChannelBuffer<int16_t> cb_int(cb->num_frames(),
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +000077 cb->num_channels());
78 Deinterleave(int_data,
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +000079 cb->num_frames(),
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +000080 cb->num_channels(),
andrew@webrtc.orga8b97372014-03-10 22:26:12 +000081 cb_int.channels());
Peter Kasting69558702016-01-12 16:26:35 -080082 for (size_t i = 0; i < cb->num_channels(); ++i) {
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +000083 S16ToFloat(cb_int.channels()[i],
84 cb->num_frames(),
85 cb->channels()[i]);
86 }
andrew@webrtc.orga8b97372014-03-10 22:26:12 +000087}
andrew@webrtc.org17e40642014-03-04 20:58:13 +000088
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +000089void ConvertToFloat(const AudioFrame& frame, ChannelBuffer<float>* cb) {
90 ConvertToFloat(frame.data_, cb);
91}
92
andrew@webrtc.org103657b2014-04-24 18:28:56 +000093// Number of channels including the keyboard channel.
Peter Kasting69558702016-01-12 16:26:35 -080094size_t TotalChannelsFromLayout(AudioProcessing::ChannelLayout layout) {
andrew@webrtc.org103657b2014-04-24 18:28:56 +000095 switch (layout) {
96 case AudioProcessing::kMono:
97 return 1;
98 case AudioProcessing::kMonoAndKeyboard:
99 case AudioProcessing::kStereo:
100 return 2;
101 case AudioProcessing::kStereoAndKeyboard:
102 return 3;
103 }
kwiberg9e2be5f2016-09-14 05:23:22 -0700104 RTC_NOTREACHED();
pkasting25702cb2016-01-08 13:50:27 -0800105 return 0;
andrew@webrtc.org103657b2014-04-24 18:28:56 +0000106}
107
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000108int TruncateToMultipleOf10(int value) {
109 return (value / 10) * 10;
110}
111
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000112void MixStereoToMono(const float* stereo, float* mono,
pkasting25702cb2016-01-08 13:50:27 -0800113 size_t samples_per_channel) {
114 for (size_t i = 0; i < samples_per_channel; ++i)
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000115 mono[i] = (stereo[i * 2] + stereo[i * 2 + 1]) / 2;
andrew@webrtc.org81865342012-10-27 00:28:27 +0000116}
117
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000118void MixStereoToMono(const int16_t* stereo, int16_t* mono,
pkasting25702cb2016-01-08 13:50:27 -0800119 size_t samples_per_channel) {
120 for (size_t i = 0; i < samples_per_channel; ++i)
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000121 mono[i] = (stereo[i * 2] + stereo[i * 2 + 1]) >> 1;
122}
123
pkasting25702cb2016-01-08 13:50:27 -0800124void CopyLeftToRightChannel(int16_t* stereo, size_t samples_per_channel) {
125 for (size_t i = 0; i < samples_per_channel; i++) {
andrew@webrtc.org81865342012-10-27 00:28:27 +0000126 stereo[i * 2 + 1] = stereo[i * 2];
127 }
128}
129
pkasting25702cb2016-01-08 13:50:27 -0800130void VerifyChannelsAreEqual(int16_t* stereo, size_t samples_per_channel) {
131 for (size_t i = 0; i < samples_per_channel; i++) {
andrew@webrtc.org81865342012-10-27 00:28:27 +0000132 EXPECT_EQ(stereo[i * 2 + 1], stereo[i * 2]);
133 }
134}
135
136void SetFrameTo(AudioFrame* frame, int16_t value) {
Peter Kastingdce40cf2015-08-24 14:52:23 -0700137 for (size_t i = 0; i < frame->samples_per_channel_ * frame->num_channels_;
138 ++i) {
andrew@webrtc.org81865342012-10-27 00:28:27 +0000139 frame->data_[i] = value;
140 }
141}
142
143void SetFrameTo(AudioFrame* frame, int16_t left, int16_t right) {
Peter Kasting69558702016-01-12 16:26:35 -0800144 ASSERT_EQ(2u, frame->num_channels_);
Peter Kastingdce40cf2015-08-24 14:52:23 -0700145 for (size_t i = 0; i < frame->samples_per_channel_ * 2; i += 2) {
andrew@webrtc.org81865342012-10-27 00:28:27 +0000146 frame->data_[i] = left;
147 frame->data_[i + 1] = right;
148 }
149}
150
andrew@webrtc.org27c69802014-02-18 20:24:56 +0000151void ScaleFrame(AudioFrame* frame, float scale) {
Peter Kastingdce40cf2015-08-24 14:52:23 -0700152 for (size_t i = 0; i < frame->samples_per_channel_ * frame->num_channels_;
153 ++i) {
andrew@webrtc.org4fc4add2014-10-30 03:40:10 +0000154 frame->data_[i] = FloatS16ToS16(frame->data_[i] * scale);
andrew@webrtc.org27c69802014-02-18 20:24:56 +0000155 }
156}
157
andrew@webrtc.org81865342012-10-27 00:28:27 +0000158bool FrameDataAreEqual(const AudioFrame& frame1, const AudioFrame& frame2) {
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000159 if (frame1.samples_per_channel_ != frame2.samples_per_channel_) {
andrew@webrtc.org81865342012-10-27 00:28:27 +0000160 return false;
161 }
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000162 if (frame1.num_channels_ != frame2.num_channels_) {
andrew@webrtc.org81865342012-10-27 00:28:27 +0000163 return false;
164 }
165 if (memcmp(frame1.data_, frame2.data_,
166 frame1.samples_per_channel_ * frame1.num_channels_ *
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000167 sizeof(int16_t))) {
andrew@webrtc.org81865342012-10-27 00:28:27 +0000168 return false;
169 }
170 return true;
171}
172
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000173void EnableAllAPComponents(AudioProcessing* ap) {
174#if defined(WEBRTC_AUDIOPROC_FIXED_PROFILE)
175 EXPECT_NOERR(ap->echo_control_mobile()->Enable(true));
176
177 EXPECT_NOERR(ap->gain_control()->set_mode(GainControl::kAdaptiveDigital));
178 EXPECT_NOERR(ap->gain_control()->Enable(true));
179#elif defined(WEBRTC_AUDIOPROC_FLOAT_PROFILE)
180 EXPECT_NOERR(ap->echo_cancellation()->enable_drift_compensation(true));
181 EXPECT_NOERR(ap->echo_cancellation()->enable_metrics(true));
182 EXPECT_NOERR(ap->echo_cancellation()->enable_delay_logging(true));
183 EXPECT_NOERR(ap->echo_cancellation()->Enable(true));
184
185 EXPECT_NOERR(ap->gain_control()->set_mode(GainControl::kAdaptiveAnalog));
186 EXPECT_NOERR(ap->gain_control()->set_analog_level_limits(0, 255));
187 EXPECT_NOERR(ap->gain_control()->Enable(true));
188#endif
189
peah8271d042016-11-22 07:24:52 -0800190 AudioProcessing::Config apm_config;
191 apm_config.high_pass_filter.enabled = true;
192 ap->ApplyConfig(apm_config);
193
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000194 EXPECT_NOERR(ap->level_estimator()->Enable(true));
195 EXPECT_NOERR(ap->noise_suppression()->Enable(true));
196
197 EXPECT_NOERR(ap->voice_detection()->Enable(true));
198}
199
bjornv@webrtc.org8dd60cc2014-09-11 08:36:35 +0000200// These functions are only used by ApmTest.Process.
andrew@webrtc.orgd7696c42013-12-03 23:39:16 +0000201template <class T>
202T AbsValue(T a) {
203 return a > 0 ? a: -a;
204}
205
206int16_t MaxAudioFrame(const AudioFrame& frame) {
pkasting25702cb2016-01-08 13:50:27 -0800207 const size_t length = frame.samples_per_channel_ * frame.num_channels_;
andrew@webrtc.orgd7696c42013-12-03 23:39:16 +0000208 int16_t max_data = AbsValue(frame.data_[0]);
pkasting25702cb2016-01-08 13:50:27 -0800209 for (size_t i = 1; i < length; i++) {
andrew@webrtc.orgd7696c42013-12-03 23:39:16 +0000210 max_data = std::max(max_data, AbsValue(frame.data_[i]));
211 }
212
213 return max_data;
214}
215
fischman@webrtc.orgf8be8df2013-12-17 23:46:39 +0000216#if defined(WEBRTC_AUDIOPROC_FLOAT_PROFILE)
andrew@webrtc.org81865342012-10-27 00:28:27 +0000217void TestStats(const AudioProcessing::Statistic& test,
andrew@webrtc.org27c69802014-02-18 20:24:56 +0000218 const audioproc::Test::Statistic& reference) {
minyue58530ed2016-05-24 05:50:12 -0700219 EXPECT_EQ(reference.instant(), test.instant);
220 EXPECT_EQ(reference.average(), test.average);
221 EXPECT_EQ(reference.maximum(), test.maximum);
222 EXPECT_EQ(reference.minimum(), test.minimum);
andrew@webrtc.org81865342012-10-27 00:28:27 +0000223}
224
225void WriteStatsMessage(const AudioProcessing::Statistic& output,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000226 audioproc::Test::Statistic* msg) {
227 msg->set_instant(output.instant);
228 msg->set_average(output.average);
229 msg->set_maximum(output.maximum);
230 msg->set_minimum(output.minimum);
andrew@webrtc.org81865342012-10-27 00:28:27 +0000231}
fischman@webrtc.orgf8be8df2013-12-17 23:46:39 +0000232#endif
andrew@webrtc.org81865342012-10-27 00:28:27 +0000233
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000234void OpenFileAndWriteMessage(const std::string filename,
mbonadei7c2c8432017-04-07 00:59:12 -0700235 const MessageLite& msg) {
andrew@webrtc.org81865342012-10-27 00:28:27 +0000236 FILE* file = fopen(filename.c_str(), "wb");
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000237 ASSERT_TRUE(file != NULL);
238
239 int32_t size = msg.ByteSize();
andrew@webrtc.org81865342012-10-27 00:28:27 +0000240 ASSERT_GT(size, 0);
kwiberg62eaacf2016-02-17 06:39:05 -0800241 std::unique_ptr<uint8_t[]> array(new uint8_t[size]);
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000242 ASSERT_TRUE(msg.SerializeToArray(array.get(), size));
andrew@webrtc.org81865342012-10-27 00:28:27 +0000243
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000244 ASSERT_EQ(1u, fwrite(&size, sizeof(size), 1, file));
andrew@webrtc.org81865342012-10-27 00:28:27 +0000245 ASSERT_EQ(static_cast<size_t>(size),
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000246 fwrite(array.get(), sizeof(array[0]), size, file));
andrew@webrtc.org81865342012-10-27 00:28:27 +0000247 fclose(file);
248}
249
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000250std::string ResourceFilePath(std::string name, int sample_rate_hz) {
251 std::ostringstream ss;
252 // Resource files are all stereo.
253 ss << name << sample_rate_hz / 1000 << "_stereo";
254 return test::ResourcePath(ss.str(), "pcm");
255}
256
pbos@webrtc.orga525c982015-01-12 17:31:18 +0000257// Temporary filenames unique to this process. Used to be able to run these
258// tests in parallel as each process needs to be running in isolation they can't
259// have competing filenames.
260std::map<std::string, std::string> temp_filenames;
261
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000262std::string OutputFilePath(std::string name,
andrew@webrtc.orgf26c9e82014-04-24 03:46:46 +0000263 int input_rate,
264 int output_rate,
ekmeyerson60d9b332015-08-14 10:35:55 -0700265 int reverse_input_rate,
266 int reverse_output_rate,
Peter Kasting69558702016-01-12 16:26:35 -0800267 size_t num_input_channels,
268 size_t num_output_channels,
269 size_t num_reverse_input_channels,
270 size_t num_reverse_output_channels,
ekmeyerson60d9b332015-08-14 10:35:55 -0700271 StreamDirection file_direction) {
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000272 std::ostringstream ss;
ekmeyerson60d9b332015-08-14 10:35:55 -0700273 ss << name << "_i" << num_input_channels << "_" << input_rate / 1000 << "_ir"
274 << num_reverse_input_channels << "_" << reverse_input_rate / 1000 << "_";
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000275 if (num_output_channels == 1) {
276 ss << "mono";
277 } else if (num_output_channels == 2) {
278 ss << "stereo";
279 } else {
kwiberg9e2be5f2016-09-14 05:23:22 -0700280 RTC_NOTREACHED();
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000281 }
ekmeyerson60d9b332015-08-14 10:35:55 -0700282 ss << output_rate / 1000;
283 if (num_reverse_output_channels == 1) {
284 ss << "_rmono";
285 } else if (num_reverse_output_channels == 2) {
286 ss << "_rstereo";
287 } else {
kwiberg9e2be5f2016-09-14 05:23:22 -0700288 RTC_NOTREACHED();
ekmeyerson60d9b332015-08-14 10:35:55 -0700289 }
290 ss << reverse_output_rate / 1000;
291 ss << "_d" << file_direction << "_pcm";
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000292
pbos@webrtc.orga525c982015-01-12 17:31:18 +0000293 std::string filename = ss.str();
pbosbb36fdf2015-07-09 07:48:14 -0700294 if (temp_filenames[filename].empty())
pbos@webrtc.orga525c982015-01-12 17:31:18 +0000295 temp_filenames[filename] = test::TempFilename(test::OutputPath(), filename);
296 return temp_filenames[filename];
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000297}
298
pbos@webrtc.org200ac002015-02-03 14:14:01 +0000299void ClearTempFiles() {
300 for (auto& kv : temp_filenames)
301 remove(kv.second.c_str());
302}
303
mbonadei7c2c8432017-04-07 00:59:12 -0700304void OpenFileAndReadMessage(std::string filename, MessageLite* msg) {
andrew@webrtc.org81865342012-10-27 00:28:27 +0000305 FILE* file = fopen(filename.c_str(), "rb");
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000306 ASSERT_TRUE(file != NULL);
307 ReadMessageFromFile(file, msg);
andrew@webrtc.org81865342012-10-27 00:28:27 +0000308 fclose(file);
309}
310
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +0000311// Reads a 10 ms chunk of int16 interleaved audio from the given (assumed
312// stereo) file, converts to deinterleaved float (optionally downmixing) and
313// returns the result in |cb|. Returns false if the file ended (or on error) and
314// true otherwise.
315//
316// |int_data| and |float_data| are just temporary space that must be
317// sufficiently large to hold the 10 ms chunk.
318bool ReadChunk(FILE* file, int16_t* int_data, float* float_data,
319 ChannelBuffer<float>* cb) {
320 // The files always contain stereo audio.
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +0000321 size_t frame_size = cb->num_frames() * 2;
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +0000322 size_t read_count = fread(int_data, sizeof(int16_t), frame_size, file);
323 if (read_count != frame_size) {
324 // Check that the file really ended.
kwiberg9e2be5f2016-09-14 05:23:22 -0700325 RTC_DCHECK(feof(file));
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +0000326 return false; // This is expected.
327 }
328
329 S16ToFloat(int_data, frame_size, float_data);
330 if (cb->num_channels() == 1) {
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +0000331 MixStereoToMono(float_data, cb->channels()[0], cb->num_frames());
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +0000332 } else {
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +0000333 Deinterleave(float_data, cb->num_frames(), 2,
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +0000334 cb->channels());
335 }
336
337 return true;
338}
339
niklase@google.com470e71d2011-07-07 08:21:25 +0000340class ApmTest : public ::testing::Test {
341 protected:
342 ApmTest();
343 virtual void SetUp();
344 virtual void TearDown();
andrew@webrtc.org755b04a2011-11-15 16:57:56 +0000345
346 static void SetUpTestCase() {
347 Trace::CreateTrace();
andrew@webrtc.org755b04a2011-11-15 16:57:56 +0000348 }
349
350 static void TearDownTestCase() {
351 Trace::ReturnTrace();
pbos@webrtc.org200ac002015-02-03 14:14:01 +0000352 ClearTempFiles();
andrew@webrtc.org755b04a2011-11-15 16:57:56 +0000353 }
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +0000354
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000355 // Used to select between int and float interface tests.
356 enum Format {
357 kIntFormat,
358 kFloatFormat
359 };
360
361 void Init(int sample_rate_hz,
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000362 int output_sample_rate_hz,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000363 int reverse_sample_rate_hz,
Peter Kasting69558702016-01-12 16:26:35 -0800364 size_t num_input_channels,
365 size_t num_output_channels,
366 size_t num_reverse_channels,
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +0000367 bool open_output_file);
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000368 void Init(AudioProcessing* ap);
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +0000369 void EnableAllComponents();
370 bool ReadFrame(FILE* file, AudioFrame* frame);
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000371 bool ReadFrame(FILE* file, AudioFrame* frame, ChannelBuffer<float>* cb);
andrew@webrtc.org27c69802014-02-18 20:24:56 +0000372 void ReadFrameWithRewind(FILE* file, AudioFrame* frame);
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000373 void ReadFrameWithRewind(FILE* file, AudioFrame* frame,
374 ChannelBuffer<float>* cb);
andrew@webrtc.org81865342012-10-27 00:28:27 +0000375 void ProcessWithDefaultStreamParameters(AudioFrame* frame);
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000376 void ProcessDelayVerificationTest(int delay_ms, int system_delay_ms,
377 int delay_min, int delay_max);
Michael Graczyk86c6d332015-07-23 11:41:39 -0700378 void TestChangingChannelsInt16Interface(
Peter Kasting69558702016-01-12 16:26:35 -0800379 size_t num_channels,
Michael Graczyk86c6d332015-07-23 11:41:39 -0700380 AudioProcessing::Error expected_return);
Peter Kasting69558702016-01-12 16:26:35 -0800381 void TestChangingForwardChannels(size_t num_in_channels,
382 size_t num_out_channels,
Michael Graczyk86c6d332015-07-23 11:41:39 -0700383 AudioProcessing::Error expected_return);
Peter Kasting69558702016-01-12 16:26:35 -0800384 void TestChangingReverseChannels(size_t num_rev_channels,
Michael Graczyk86c6d332015-07-23 11:41:39 -0700385 AudioProcessing::Error expected_return);
andrew@webrtc.org27c69802014-02-18 20:24:56 +0000386 void RunQuantizedVolumeDoesNotGetStuckTest(int sample_rate);
387 void RunManualVolumeChangeIsPossibleTest(int sample_rate);
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000388 void StreamParametersTest(Format format);
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000389 int ProcessStreamChooser(Format format);
390 int AnalyzeReverseStreamChooser(Format format);
391 void ProcessDebugDump(const std::string& in_filename,
392 const std::string& out_filename,
ivocd66b44d2016-01-15 03:06:36 -0800393 Format format,
394 int max_size_bytes);
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000395 void VerifyDebugDumpTest(Format format);
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +0000396
397 const std::string output_path_;
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +0000398 const std::string ref_filename_;
kwiberg62eaacf2016-02-17 06:39:05 -0800399 std::unique_ptr<AudioProcessing> apm_;
andrew@webrtc.orgf3930e92013-09-18 22:37:32 +0000400 AudioFrame* frame_;
401 AudioFrame* revframe_;
kwiberg62eaacf2016-02-17 06:39:05 -0800402 std::unique_ptr<ChannelBuffer<float> > float_cb_;
403 std::unique_ptr<ChannelBuffer<float> > revfloat_cb_;
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000404 int output_sample_rate_hz_;
Peter Kasting69558702016-01-12 16:26:35 -0800405 size_t num_output_channels_;
niklase@google.com470e71d2011-07-07 08:21:25 +0000406 FILE* far_file_;
407 FILE* near_file_;
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +0000408 FILE* out_file_;
niklase@google.com470e71d2011-07-07 08:21:25 +0000409};
410
411ApmTest::ApmTest()
andrew@webrtc.org27c69802014-02-18 20:24:56 +0000412 : output_path_(test::OutputPath()),
andrew@webrtc.org293d22b2012-01-30 22:04:26 +0000413#if defined(WEBRTC_AUDIOPROC_FIXED_PROFILE)
ehmaldonadodedaf1c2016-11-18 04:52:22 -0800414 ref_filename_(test::ResourcePath("audio_processing/output_data_fixed",
415 "pb")),
andrew@webrtc.org293d22b2012-01-30 22:04:26 +0000416#elif defined(WEBRTC_AUDIOPROC_FLOAT_PROFILE)
aluebs@webrtc.orgf17ee9c2015-01-29 00:03:53 +0000417#if defined(WEBRTC_MAC)
418 // A different file for Mac is needed because on this platform the AEC
419 // constant |kFixedDelayMs| value is 20 and not 50 as it is on the rest.
ehmaldonadodedaf1c2016-11-18 04:52:22 -0800420 ref_filename_(test::ResourcePath("audio_processing/output_data_mac",
421 "pb")),
aluebs@webrtc.orgf17ee9c2015-01-29 00:03:53 +0000422#else
ehmaldonadodedaf1c2016-11-18 04:52:22 -0800423 ref_filename_(test::ResourcePath("audio_processing/output_data_float",
424 "pb")),
kjellander@webrtc.org61f07c32011-10-18 06:54:58 +0000425#endif
aluebs@webrtc.orgf17ee9c2015-01-29 00:03:53 +0000426#endif
niklase@google.com470e71d2011-07-07 08:21:25 +0000427 frame_(NULL),
ajm@google.com22e65152011-07-18 18:03:01 +0000428 revframe_(NULL),
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000429 output_sample_rate_hz_(0),
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000430 num_output_channels_(0),
ajm@google.com22e65152011-07-18 18:03:01 +0000431 far_file_(NULL),
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +0000432 near_file_(NULL),
aluebs@webrtc.orgc9ee4122014-02-03 14:41:57 +0000433 out_file_(NULL) {
434 Config config;
435 config.Set<ExperimentalAgc>(new ExperimentalAgc(false));
436 apm_.reset(AudioProcessing::Create(config));
437}
niklase@google.com470e71d2011-07-07 08:21:25 +0000438
439void ApmTest::SetUp() {
andrew@webrtc.orgf3930e92013-09-18 22:37:32 +0000440 ASSERT_TRUE(apm_.get() != NULL);
niklase@google.com470e71d2011-07-07 08:21:25 +0000441
442 frame_ = new AudioFrame();
443 revframe_ = new AudioFrame();
444
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000445 Init(32000, 32000, 32000, 2, 2, 2, false);
niklase@google.com470e71d2011-07-07 08:21:25 +0000446}
447
448void ApmTest::TearDown() {
449 if (frame_) {
450 delete frame_;
451 }
452 frame_ = NULL;
453
454 if (revframe_) {
455 delete revframe_;
456 }
457 revframe_ = NULL;
458
459 if (far_file_) {
460 ASSERT_EQ(0, fclose(far_file_));
461 }
462 far_file_ = NULL;
463
464 if (near_file_) {
465 ASSERT_EQ(0, fclose(near_file_));
466 }
467 near_file_ = NULL;
468
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +0000469 if (out_file_) {
470 ASSERT_EQ(0, fclose(out_file_));
471 }
472 out_file_ = NULL;
niklase@google.com470e71d2011-07-07 08:21:25 +0000473}
474
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000475void ApmTest::Init(AudioProcessing* ap) {
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000476 ASSERT_EQ(kNoErr,
Michael Graczyk86c6d332015-07-23 11:41:39 -0700477 ap->Initialize(
478 {{{frame_->sample_rate_hz_, frame_->num_channels_},
479 {output_sample_rate_hz_, num_output_channels_},
ekmeyerson60d9b332015-08-14 10:35:55 -0700480 {revframe_->sample_rate_hz_, revframe_->num_channels_},
Michael Graczyk86c6d332015-07-23 11:41:39 -0700481 {revframe_->sample_rate_hz_, revframe_->num_channels_}}}));
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000482}
483
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000484void ApmTest::Init(int sample_rate_hz,
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000485 int output_sample_rate_hz,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000486 int reverse_sample_rate_hz,
Peter Kasting69558702016-01-12 16:26:35 -0800487 size_t num_input_channels,
488 size_t num_output_channels,
489 size_t num_reverse_channels,
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +0000490 bool open_output_file) {
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000491 SetContainerFormat(sample_rate_hz, num_input_channels, frame_, &float_cb_);
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000492 output_sample_rate_hz_ = output_sample_rate_hz;
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000493 num_output_channels_ = num_output_channels;
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +0000494
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000495 SetContainerFormat(reverse_sample_rate_hz, num_reverse_channels, revframe_,
496 &revfloat_cb_);
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000497 Init(apm_.get());
andrew@webrtc.org60730cf2014-01-07 17:45:09 +0000498
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +0000499 if (far_file_) {
500 ASSERT_EQ(0, fclose(far_file_));
501 }
502 std::string filename = ResourceFilePath("far", sample_rate_hz);
503 far_file_ = fopen(filename.c_str(), "rb");
504 ASSERT_TRUE(far_file_ != NULL) << "Could not open file " <<
505 filename << "\n";
506
507 if (near_file_) {
508 ASSERT_EQ(0, fclose(near_file_));
509 }
510 filename = ResourceFilePath("near", sample_rate_hz);
511 near_file_ = fopen(filename.c_str(), "rb");
512 ASSERT_TRUE(near_file_ != NULL) << "Could not open file " <<
513 filename << "\n";
514
515 if (open_output_file) {
516 if (out_file_) {
517 ASSERT_EQ(0, fclose(out_file_));
518 }
ekmeyerson60d9b332015-08-14 10:35:55 -0700519 filename = OutputFilePath(
520 "out", sample_rate_hz, output_sample_rate_hz, reverse_sample_rate_hz,
521 reverse_sample_rate_hz, num_input_channels, num_output_channels,
522 num_reverse_channels, num_reverse_channels, kForward);
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +0000523 out_file_ = fopen(filename.c_str(), "wb");
524 ASSERT_TRUE(out_file_ != NULL) << "Could not open file " <<
525 filename << "\n";
526 }
527}
528
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +0000529void ApmTest::EnableAllComponents() {
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000530 EnableAllAPComponents(apm_.get());
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +0000531}
532
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000533bool ApmTest::ReadFrame(FILE* file, AudioFrame* frame,
534 ChannelBuffer<float>* cb) {
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +0000535 // The files always contain stereo audio.
536 size_t frame_size = frame->samples_per_channel_ * 2;
537 size_t read_count = fread(frame->data_,
538 sizeof(int16_t),
539 frame_size,
540 file);
541 if (read_count != frame_size) {
542 // Check that the file really ended.
543 EXPECT_NE(0, feof(file));
544 return false; // This is expected.
545 }
546
547 if (frame->num_channels_ == 1) {
548 MixStereoToMono(frame->data_, frame->data_,
549 frame->samples_per_channel_);
550 }
551
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000552 if (cb) {
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000553 ConvertToFloat(*frame, cb);
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000554 }
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +0000555 return true;
ajm@google.coma769fa52011-07-13 21:57:58 +0000556}
557
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000558bool ApmTest::ReadFrame(FILE* file, AudioFrame* frame) {
559 return ReadFrame(file, frame, NULL);
560}
561
andrew@webrtc.org27c69802014-02-18 20:24:56 +0000562// If the end of the file has been reached, rewind it and attempt to read the
563// frame again.
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000564void ApmTest::ReadFrameWithRewind(FILE* file, AudioFrame* frame,
565 ChannelBuffer<float>* cb) {
566 if (!ReadFrame(near_file_, frame_, cb)) {
andrew@webrtc.org27c69802014-02-18 20:24:56 +0000567 rewind(near_file_);
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000568 ASSERT_TRUE(ReadFrame(near_file_, frame_, cb));
andrew@webrtc.org27c69802014-02-18 20:24:56 +0000569 }
570}
571
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000572void ApmTest::ReadFrameWithRewind(FILE* file, AudioFrame* frame) {
573 ReadFrameWithRewind(file, frame, NULL);
574}
575
andrew@webrtc.org81865342012-10-27 00:28:27 +0000576void ApmTest::ProcessWithDefaultStreamParameters(AudioFrame* frame) {
577 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(0));
andrew@webrtc.org6be1e932013-03-01 18:47:28 +0000578 apm_->echo_cancellation()->set_stream_drift_samples(0);
andrew@webrtc.org81865342012-10-27 00:28:27 +0000579 EXPECT_EQ(apm_->kNoError,
580 apm_->gain_control()->set_stream_analog_level(127));
581 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame));
andrew@webrtc.org755b04a2011-11-15 16:57:56 +0000582}
583
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000584int ApmTest::ProcessStreamChooser(Format format) {
585 if (format == kIntFormat) {
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000586 return apm_->ProcessStream(frame_);
587 }
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000588 return apm_->ProcessStream(float_cb_->channels(),
589 frame_->samples_per_channel_,
590 frame_->sample_rate_hz_,
591 LayoutFromChannels(frame_->num_channels_),
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000592 output_sample_rate_hz_,
593 LayoutFromChannels(num_output_channels_),
594 float_cb_->channels());
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000595}
596
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000597int ApmTest::AnalyzeReverseStreamChooser(Format format) {
598 if (format == kIntFormat) {
aluebsb0319552016-03-17 20:39:53 -0700599 return apm_->ProcessReverseStream(revframe_);
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000600 }
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000601 return apm_->AnalyzeReverseStream(
602 revfloat_cb_->channels(),
603 revframe_->samples_per_channel_,
604 revframe_->sample_rate_hz_,
605 LayoutFromChannels(revframe_->num_channels_));
606}
607
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000608void ApmTest::ProcessDelayVerificationTest(int delay_ms, int system_delay_ms,
609 int delay_min, int delay_max) {
610 // The |revframe_| and |frame_| should include the proper frame information,
611 // hence can be used for extracting information.
andrew@webrtc.org27c69802014-02-18 20:24:56 +0000612 AudioFrame tmp_frame;
613 std::queue<AudioFrame*> frame_queue;
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000614 bool causal = true;
615
616 tmp_frame.CopyFrom(*revframe_);
617 SetFrameTo(&tmp_frame, 0);
618
619 EXPECT_EQ(apm_->kNoError, apm_->Initialize());
620 // Initialize the |frame_queue| with empty frames.
621 int frame_delay = delay_ms / 10;
622 while (frame_delay < 0) {
andrew@webrtc.org27c69802014-02-18 20:24:56 +0000623 AudioFrame* frame = new AudioFrame();
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000624 frame->CopyFrom(tmp_frame);
625 frame_queue.push(frame);
626 frame_delay++;
627 causal = false;
628 }
629 while (frame_delay > 0) {
andrew@webrtc.org27c69802014-02-18 20:24:56 +0000630 AudioFrame* frame = new AudioFrame();
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000631 frame->CopyFrom(tmp_frame);
632 frame_queue.push(frame);
633 frame_delay--;
634 }
bjornv@webrtc.orgbbd47fc2014-01-13 08:54:34 +0000635 // Run for 4.5 seconds, skipping statistics from the first 2.5 seconds. We
636 // need enough frames with audio to have reliable estimates, but as few as
637 // possible to keep processing time down. 4.5 seconds seemed to be a good
638 // compromise for this recording.
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000639 for (int frame_count = 0; frame_count < 450; ++frame_count) {
andrew@webrtc.org27c69802014-02-18 20:24:56 +0000640 AudioFrame* frame = new AudioFrame();
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000641 frame->CopyFrom(tmp_frame);
642 // Use the near end recording, since that has more speech in it.
643 ASSERT_TRUE(ReadFrame(near_file_, frame));
644 frame_queue.push(frame);
andrew@webrtc.org27c69802014-02-18 20:24:56 +0000645 AudioFrame* reverse_frame = frame;
646 AudioFrame* process_frame = frame_queue.front();
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000647 if (!causal) {
648 reverse_frame = frame_queue.front();
649 // When we call ProcessStream() the frame is modified, so we can't use the
650 // pointer directly when things are non-causal. Use an intermediate frame
651 // and copy the data.
652 process_frame = &tmp_frame;
653 process_frame->CopyFrom(*frame);
654 }
aluebsb0319552016-03-17 20:39:53 -0700655 EXPECT_EQ(apm_->kNoError, apm_->ProcessReverseStream(reverse_frame));
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000656 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(system_delay_ms));
657 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(process_frame));
658 frame = frame_queue.front();
659 frame_queue.pop();
660 delete frame;
661
bjornv@webrtc.orgbbd47fc2014-01-13 08:54:34 +0000662 if (frame_count == 250) {
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000663 int median;
664 int std;
bjornv@webrtc.orgb1786db2015-02-03 06:06:26 +0000665 float poor_fraction;
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000666 // Discard the first delay metrics to avoid convergence effects.
667 EXPECT_EQ(apm_->kNoError,
bjornv@webrtc.orgb1786db2015-02-03 06:06:26 +0000668 apm_->echo_cancellation()->GetDelayMetrics(&median, &std,
669 &poor_fraction));
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000670 }
671 }
672
673 rewind(near_file_);
674 while (!frame_queue.empty()) {
andrew@webrtc.org27c69802014-02-18 20:24:56 +0000675 AudioFrame* frame = frame_queue.front();
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000676 frame_queue.pop();
677 delete frame;
678 }
679 // Calculate expected delay estimate and acceptable regions. Further,
680 // limit them w.r.t. AEC delay estimation support.
Peter Kastingdce40cf2015-08-24 14:52:23 -0700681 const size_t samples_per_ms =
kwiberg7885d3f2017-04-25 12:35:07 -0700682 rtc::SafeMin<size_t>(16u, frame_->samples_per_channel_ / 10);
kwiberg07038562017-06-12 11:40:47 -0700683 const int expected_median =
684 rtc::SafeClamp<int>(delay_ms - system_delay_ms, delay_min, delay_max);
685 const int expected_median_high = rtc::SafeClamp<int>(
686 expected_median + rtc::dchecked_cast<int>(96 / samples_per_ms), delay_min,
Peter Kastingdce40cf2015-08-24 14:52:23 -0700687 delay_max);
kwiberg07038562017-06-12 11:40:47 -0700688 const int expected_median_low = rtc::SafeClamp<int>(
689 expected_median - rtc::dchecked_cast<int>(96 / samples_per_ms), delay_min,
Peter Kastingdce40cf2015-08-24 14:52:23 -0700690 delay_max);
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000691 // Verify delay metrics.
692 int median;
693 int std;
bjornv@webrtc.orgb1786db2015-02-03 06:06:26 +0000694 float poor_fraction;
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000695 EXPECT_EQ(apm_->kNoError,
bjornv@webrtc.orgb1786db2015-02-03 06:06:26 +0000696 apm_->echo_cancellation()->GetDelayMetrics(&median, &std,
697 &poor_fraction));
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000698 EXPECT_GE(expected_median_high, median);
699 EXPECT_LE(expected_median_low, median);
700}
701
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000702void ApmTest::StreamParametersTest(Format format) {
niklase@google.com470e71d2011-07-07 08:21:25 +0000703 // No errors when the components are disabled.
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000704 EXPECT_EQ(apm_->kNoError, ProcessStreamChooser(format));
niklase@google.com470e71d2011-07-07 08:21:25 +0000705
andrew@webrtc.org1e916932011-11-29 18:28:57 +0000706 // -- Missing AGC level --
niklase@google.com470e71d2011-07-07 08:21:25 +0000707 EXPECT_EQ(apm_->kNoError, apm_->gain_control()->Enable(true));
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000708 EXPECT_EQ(apm_->kStreamParameterNotSetError,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000709 ProcessStreamChooser(format));
niklase@google.com470e71d2011-07-07 08:21:25 +0000710
andrew@webrtc.org1e916932011-11-29 18:28:57 +0000711 // Resets after successful ProcessStream().
niklase@google.com470e71d2011-07-07 08:21:25 +0000712 EXPECT_EQ(apm_->kNoError,
713 apm_->gain_control()->set_stream_analog_level(127));
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000714 EXPECT_EQ(apm_->kNoError, ProcessStreamChooser(format));
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000715 EXPECT_EQ(apm_->kStreamParameterNotSetError,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000716 ProcessStreamChooser(format));
niklase@google.com470e71d2011-07-07 08:21:25 +0000717
andrew@webrtc.org1e916932011-11-29 18:28:57 +0000718 // Other stream parameters set correctly.
719 EXPECT_EQ(apm_->kNoError, apm_->echo_cancellation()->Enable(true));
niklase@google.com470e71d2011-07-07 08:21:25 +0000720 EXPECT_EQ(apm_->kNoError,
721 apm_->echo_cancellation()->enable_drift_compensation(true));
andrew@webrtc.org1e916932011-11-29 18:28:57 +0000722 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(100));
andrew@webrtc.org6be1e932013-03-01 18:47:28 +0000723 apm_->echo_cancellation()->set_stream_drift_samples(0);
niklase@google.com470e71d2011-07-07 08:21:25 +0000724 EXPECT_EQ(apm_->kStreamParameterNotSetError,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000725 ProcessStreamChooser(format));
andrew@webrtc.org1e916932011-11-29 18:28:57 +0000726 EXPECT_EQ(apm_->kNoError, apm_->gain_control()->Enable(false));
727 EXPECT_EQ(apm_->kNoError,
728 apm_->echo_cancellation()->enable_drift_compensation(false));
729
730 // -- Missing delay --
andrew@webrtc.org1e916932011-11-29 18:28:57 +0000731 EXPECT_EQ(apm_->kNoError, apm_->echo_cancellation()->Enable(true));
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000732 EXPECT_EQ(apm_->kNoError, ProcessStreamChooser(format));
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000733 EXPECT_EQ(apm_->kStreamParameterNotSetError,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000734 ProcessStreamChooser(format));
andrew@webrtc.org1e916932011-11-29 18:28:57 +0000735
736 // Resets after successful ProcessStream().
737 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(100));
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000738 EXPECT_EQ(apm_->kNoError, ProcessStreamChooser(format));
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000739 EXPECT_EQ(apm_->kStreamParameterNotSetError,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000740 ProcessStreamChooser(format));
andrew@webrtc.org1e916932011-11-29 18:28:57 +0000741
742 // Other stream parameters set correctly.
743 EXPECT_EQ(apm_->kNoError, apm_->gain_control()->Enable(true));
744 EXPECT_EQ(apm_->kNoError,
745 apm_->echo_cancellation()->enable_drift_compensation(true));
andrew@webrtc.org6be1e932013-03-01 18:47:28 +0000746 apm_->echo_cancellation()->set_stream_drift_samples(0);
andrew@webrtc.org1e916932011-11-29 18:28:57 +0000747 EXPECT_EQ(apm_->kNoError,
748 apm_->gain_control()->set_stream_analog_level(127));
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000749 EXPECT_EQ(apm_->kStreamParameterNotSetError,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000750 ProcessStreamChooser(format));
andrew@webrtc.org1e916932011-11-29 18:28:57 +0000751 EXPECT_EQ(apm_->kNoError, apm_->gain_control()->Enable(false));
752
753 // -- Missing drift --
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000754 EXPECT_EQ(apm_->kStreamParameterNotSetError,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000755 ProcessStreamChooser(format));
andrew@webrtc.org1e916932011-11-29 18:28:57 +0000756
757 // Resets after successful ProcessStream().
758 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(100));
andrew@webrtc.org6be1e932013-03-01 18:47:28 +0000759 apm_->echo_cancellation()->set_stream_drift_samples(0);
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000760 EXPECT_EQ(apm_->kNoError, ProcessStreamChooser(format));
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000761 EXPECT_EQ(apm_->kStreamParameterNotSetError,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000762 ProcessStreamChooser(format));
andrew@webrtc.org1e916932011-11-29 18:28:57 +0000763
764 // Other stream parameters set correctly.
niklase@google.com470e71d2011-07-07 08:21:25 +0000765 EXPECT_EQ(apm_->kNoError, apm_->gain_control()->Enable(true));
766 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(100));
767 EXPECT_EQ(apm_->kNoError,
768 apm_->gain_control()->set_stream_analog_level(127));
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000769 EXPECT_EQ(apm_->kStreamParameterNotSetError,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000770 ProcessStreamChooser(format));
niklase@google.com470e71d2011-07-07 08:21:25 +0000771
andrew@webrtc.org1e916932011-11-29 18:28:57 +0000772 // -- No stream parameters --
niklase@google.com470e71d2011-07-07 08:21:25 +0000773 EXPECT_EQ(apm_->kNoError,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000774 AnalyzeReverseStreamChooser(format));
niklase@google.com470e71d2011-07-07 08:21:25 +0000775 EXPECT_EQ(apm_->kStreamParameterNotSetError,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000776 ProcessStreamChooser(format));
niklase@google.com470e71d2011-07-07 08:21:25 +0000777
andrew@webrtc.org1e916932011-11-29 18:28:57 +0000778 // -- All there --
niklase@google.com470e71d2011-07-07 08:21:25 +0000779 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(100));
andrew@webrtc.org6be1e932013-03-01 18:47:28 +0000780 apm_->echo_cancellation()->set_stream_drift_samples(0);
niklase@google.com470e71d2011-07-07 08:21:25 +0000781 EXPECT_EQ(apm_->kNoError,
782 apm_->gain_control()->set_stream_analog_level(127));
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000783 EXPECT_EQ(apm_->kNoError, ProcessStreamChooser(format));
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000784}
785
786TEST_F(ApmTest, StreamParametersInt) {
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000787 StreamParametersTest(kIntFormat);
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000788}
789
790TEST_F(ApmTest, StreamParametersFloat) {
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000791 StreamParametersTest(kFloatFormat);
niklase@google.com470e71d2011-07-07 08:21:25 +0000792}
793
andrew@webrtc.org5f23d642012-05-29 21:14:06 +0000794TEST_F(ApmTest, DefaultDelayOffsetIsZero) {
795 EXPECT_EQ(0, apm_->delay_offset_ms());
796 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(50));
797 EXPECT_EQ(50, apm_->stream_delay_ms());
798}
799
800TEST_F(ApmTest, DelayOffsetWithLimitsIsSetProperly) {
801 // High limit of 500 ms.
andrew@webrtc.org6f9f8172012-03-06 19:03:39 +0000802 apm_->set_delay_offset_ms(100);
803 EXPECT_EQ(100, apm_->delay_offset_ms());
804 EXPECT_EQ(apm_->kBadStreamParameterWarning, apm_->set_stream_delay_ms(450));
andrew@webrtc.org5f23d642012-05-29 21:14:06 +0000805 EXPECT_EQ(500, apm_->stream_delay_ms());
andrew@webrtc.org6f9f8172012-03-06 19:03:39 +0000806 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(100));
807 EXPECT_EQ(200, apm_->stream_delay_ms());
808
andrew@webrtc.org5f23d642012-05-29 21:14:06 +0000809 // Low limit of 0 ms.
andrew@webrtc.org6f9f8172012-03-06 19:03:39 +0000810 apm_->set_delay_offset_ms(-50);
811 EXPECT_EQ(-50, apm_->delay_offset_ms());
andrew@webrtc.org5f23d642012-05-29 21:14:06 +0000812 EXPECT_EQ(apm_->kBadStreamParameterWarning, apm_->set_stream_delay_ms(20));
813 EXPECT_EQ(0, apm_->stream_delay_ms());
andrew@webrtc.org6f9f8172012-03-06 19:03:39 +0000814 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(100));
815 EXPECT_EQ(50, apm_->stream_delay_ms());
816}
817
Michael Graczyk86c6d332015-07-23 11:41:39 -0700818void ApmTest::TestChangingChannelsInt16Interface(
Peter Kasting69558702016-01-12 16:26:35 -0800819 size_t num_channels,
Michael Graczyk86c6d332015-07-23 11:41:39 -0700820 AudioProcessing::Error expected_return) {
andrew@webrtc.org60730cf2014-01-07 17:45:09 +0000821 frame_->num_channels_ = num_channels;
822 EXPECT_EQ(expected_return, apm_->ProcessStream(frame_));
aluebsb0319552016-03-17 20:39:53 -0700823 EXPECT_EQ(expected_return, apm_->ProcessReverseStream(frame_));
andrew@webrtc.org60730cf2014-01-07 17:45:09 +0000824}
825
Michael Graczyk86c6d332015-07-23 11:41:39 -0700826void ApmTest::TestChangingForwardChannels(
Peter Kasting69558702016-01-12 16:26:35 -0800827 size_t num_in_channels,
828 size_t num_out_channels,
Michael Graczyk86c6d332015-07-23 11:41:39 -0700829 AudioProcessing::Error expected_return) {
830 const StreamConfig input_stream = {frame_->sample_rate_hz_, num_in_channels};
831 const StreamConfig output_stream = {output_sample_rate_hz_, num_out_channels};
832
833 EXPECT_EQ(expected_return,
834 apm_->ProcessStream(float_cb_->channels(), input_stream,
835 output_stream, float_cb_->channels()));
836}
837
838void ApmTest::TestChangingReverseChannels(
Peter Kasting69558702016-01-12 16:26:35 -0800839 size_t num_rev_channels,
Michael Graczyk86c6d332015-07-23 11:41:39 -0700840 AudioProcessing::Error expected_return) {
841 const ProcessingConfig processing_config = {
ekmeyerson60d9b332015-08-14 10:35:55 -0700842 {{frame_->sample_rate_hz_, apm_->num_input_channels()},
843 {output_sample_rate_hz_, apm_->num_output_channels()},
844 {frame_->sample_rate_hz_, num_rev_channels},
845 {frame_->sample_rate_hz_, num_rev_channels}}};
Michael Graczyk86c6d332015-07-23 11:41:39 -0700846
ekmeyerson60d9b332015-08-14 10:35:55 -0700847 EXPECT_EQ(
848 expected_return,
849 apm_->ProcessReverseStream(
850 float_cb_->channels(), processing_config.reverse_input_stream(),
851 processing_config.reverse_output_stream(), float_cb_->channels()));
Michael Graczyk86c6d332015-07-23 11:41:39 -0700852}
853
854TEST_F(ApmTest, ChannelsInt16Interface) {
855 // Testing number of invalid and valid channels.
856 Init(16000, 16000, 16000, 4, 4, 4, false);
857
858 TestChangingChannelsInt16Interface(0, apm_->kBadNumberChannelsError);
859
Peter Kasting69558702016-01-12 16:26:35 -0800860 for (size_t i = 1; i < 4; i++) {
Michael Graczyk86c6d332015-07-23 11:41:39 -0700861 TestChangingChannelsInt16Interface(i, kNoErr);
niklase@google.com470e71d2011-07-07 08:21:25 +0000862 EXPECT_EQ(i, apm_->num_input_channels());
niklase@google.com470e71d2011-07-07 08:21:25 +0000863 }
864}
865
Michael Graczyk86c6d332015-07-23 11:41:39 -0700866TEST_F(ApmTest, Channels) {
867 // Testing number of invalid and valid channels.
868 Init(16000, 16000, 16000, 4, 4, 4, false);
869
870 TestChangingForwardChannels(0, 1, apm_->kBadNumberChannelsError);
871 TestChangingReverseChannels(0, apm_->kBadNumberChannelsError);
872
Peter Kasting69558702016-01-12 16:26:35 -0800873 for (size_t i = 1; i < 4; ++i) {
874 for (size_t j = 0; j < 1; ++j) {
Michael Graczyk86c6d332015-07-23 11:41:39 -0700875 // Output channels much be one or match input channels.
876 if (j == 1 || i == j) {
877 TestChangingForwardChannels(i, j, kNoErr);
878 TestChangingReverseChannels(i, kNoErr);
879
880 EXPECT_EQ(i, apm_->num_input_channels());
881 EXPECT_EQ(j, apm_->num_output_channels());
882 // The number of reverse channels used for processing to is always 1.
Peter Kasting69558702016-01-12 16:26:35 -0800883 EXPECT_EQ(1u, apm_->num_reverse_channels());
Michael Graczyk86c6d332015-07-23 11:41:39 -0700884 } else {
885 TestChangingForwardChannels(i, j,
886 AudioProcessing::kBadNumberChannelsError);
887 }
888 }
889 }
890}
891
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000892TEST_F(ApmTest, SampleRatesInt) {
niklase@google.com470e71d2011-07-07 08:21:25 +0000893 // Testing invalid sample rates
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000894 SetContainerFormat(10000, 2, frame_, &float_cb_);
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000895 EXPECT_EQ(apm_->kBadSampleRateError, ProcessStreamChooser(kIntFormat));
niklase@google.com470e71d2011-07-07 08:21:25 +0000896 // Testing valid sample rates
Alejandro Luebs47748742015-05-22 12:00:21 -0700897 int fs[] = {8000, 16000, 32000, 48000};
pkasting25702cb2016-01-08 13:50:27 -0800898 for (size_t i = 0; i < arraysize(fs); i++) {
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000899 SetContainerFormat(fs[i], 2, frame_, &float_cb_);
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000900 EXPECT_NOERR(ProcessStreamChooser(kIntFormat));
niklase@google.com470e71d2011-07-07 08:21:25 +0000901 }
902}
903
niklase@google.com470e71d2011-07-07 08:21:25 +0000904TEST_F(ApmTest, EchoCancellation) {
905 EXPECT_EQ(apm_->kNoError,
906 apm_->echo_cancellation()->enable_drift_compensation(true));
907 EXPECT_TRUE(apm_->echo_cancellation()->is_drift_compensation_enabled());
908 EXPECT_EQ(apm_->kNoError,
909 apm_->echo_cancellation()->enable_drift_compensation(false));
910 EXPECT_FALSE(apm_->echo_cancellation()->is_drift_compensation_enabled());
911
niklase@google.com470e71d2011-07-07 08:21:25 +0000912 EchoCancellation::SuppressionLevel level[] = {
913 EchoCancellation::kLowSuppression,
914 EchoCancellation::kModerateSuppression,
915 EchoCancellation::kHighSuppression,
916 };
pkasting25702cb2016-01-08 13:50:27 -0800917 for (size_t i = 0; i < arraysize(level); i++) {
niklase@google.com470e71d2011-07-07 08:21:25 +0000918 EXPECT_EQ(apm_->kNoError,
919 apm_->echo_cancellation()->set_suppression_level(level[i]));
920 EXPECT_EQ(level[i],
921 apm_->echo_cancellation()->suppression_level());
922 }
923
924 EchoCancellation::Metrics metrics;
925 EXPECT_EQ(apm_->kNotEnabledError,
926 apm_->echo_cancellation()->GetMetrics(&metrics));
927
ivoc3e9a5372016-10-28 07:55:33 -0700928 EXPECT_EQ(apm_->kNoError, apm_->echo_cancellation()->Enable(true));
929 EXPECT_TRUE(apm_->echo_cancellation()->is_enabled());
930
niklase@google.com470e71d2011-07-07 08:21:25 +0000931 EXPECT_EQ(apm_->kNoError,
932 apm_->echo_cancellation()->enable_metrics(true));
933 EXPECT_TRUE(apm_->echo_cancellation()->are_metrics_enabled());
934 EXPECT_EQ(apm_->kNoError,
935 apm_->echo_cancellation()->enable_metrics(false));
936 EXPECT_FALSE(apm_->echo_cancellation()->are_metrics_enabled());
937
ivoc48dfab52016-10-28 03:29:31 -0700938 EXPECT_EQ(apm_->kNoError,
939 apm_->echo_cancellation()->enable_delay_logging(true));
940 EXPECT_TRUE(apm_->echo_cancellation()->is_delay_logging_enabled());
941 EXPECT_EQ(apm_->kNoError,
942 apm_->echo_cancellation()->enable_delay_logging(false));
943 EXPECT_FALSE(apm_->echo_cancellation()->is_delay_logging_enabled());
944
ivoc3e9a5372016-10-28 07:55:33 -0700945 EXPECT_EQ(apm_->kNoError, apm_->echo_cancellation()->Enable(false));
946 EXPECT_FALSE(apm_->echo_cancellation()->is_enabled());
947
948 int median = 0;
949 int std = 0;
950 float poor_fraction = 0;
951 EXPECT_EQ(apm_->kNotEnabledError, apm_->echo_cancellation()->GetDelayMetrics(
952 &median, &std, &poor_fraction));
953
niklase@google.com470e71d2011-07-07 08:21:25 +0000954 EXPECT_EQ(apm_->kNoError, apm_->echo_cancellation()->Enable(true));
955 EXPECT_TRUE(apm_->echo_cancellation()->is_enabled());
956 EXPECT_EQ(apm_->kNoError, apm_->echo_cancellation()->Enable(false));
957 EXPECT_FALSE(apm_->echo_cancellation()->is_enabled());
bjornv@webrtc.org91d11b32013-03-05 16:53:09 +0000958
959 EXPECT_EQ(apm_->kNoError, apm_->echo_cancellation()->Enable(true));
960 EXPECT_TRUE(apm_->echo_cancellation()->is_enabled());
961 EXPECT_TRUE(apm_->echo_cancellation()->aec_core() != NULL);
962 EXPECT_EQ(apm_->kNoError, apm_->echo_cancellation()->Enable(false));
963 EXPECT_FALSE(apm_->echo_cancellation()->is_enabled());
964 EXPECT_FALSE(apm_->echo_cancellation()->aec_core() != NULL);
niklase@google.com470e71d2011-07-07 08:21:25 +0000965}
966
bjornv@webrtc.org84f8ec12014-06-19 12:14:33 +0000967TEST_F(ApmTest, DISABLED_EchoCancellationReportsCorrectDelays) {
bjornv@webrtc.orgbac00122015-01-02 09:23:49 +0000968 // TODO(bjornv): Fix this test to work with DA-AEC.
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000969 // Enable AEC only.
970 EXPECT_EQ(apm_->kNoError,
971 apm_->echo_cancellation()->enable_drift_compensation(false));
972 EXPECT_EQ(apm_->kNoError,
973 apm_->echo_cancellation()->enable_metrics(false));
974 EXPECT_EQ(apm_->kNoError,
975 apm_->echo_cancellation()->enable_delay_logging(true));
976 EXPECT_EQ(apm_->kNoError, apm_->echo_cancellation()->Enable(true));
bjornv@webrtc.org5c3f4e32014-06-19 09:51:29 +0000977 Config config;
henrik.lundin0f133b92015-07-02 00:17:55 -0700978 config.Set<DelayAgnostic>(new DelayAgnostic(false));
bjornv@webrtc.org5c3f4e32014-06-19 09:51:29 +0000979 apm_->SetExtraOptions(config);
bjornv@webrtc.org3e102492013-02-14 15:29:09 +0000980
981 // Internally in the AEC the amount of lookahead the delay estimation can
982 // handle is 15 blocks and the maximum delay is set to 60 blocks.
983 const int kLookaheadBlocks = 15;
984 const int kMaxDelayBlocks = 60;
985 // The AEC has a startup time before it actually starts to process. This
986 // procedure can flush the internal far-end buffer, which of course affects
987 // the delay estimation. Therefore, we set a system_delay high enough to
988 // avoid that. The smallest system_delay you can report without flushing the
989 // buffer is 66 ms in 8 kHz.
990 //
991 // It is known that for 16 kHz (and 32 kHz) sampling frequency there is an
992 // additional stuffing of 8 ms on the fly, but it seems to have no impact on
993 // delay estimation. This should be noted though. In case of test failure,
994 // this could be the cause.
995 const int kSystemDelayMs = 66;
996 // Test a couple of corner cases and verify that the estimated delay is
997 // within a valid region (set to +-1.5 blocks). Note that these cases are
998 // sampling frequency dependent.
pkasting25702cb2016-01-08 13:50:27 -0800999 for (size_t i = 0; i < arraysize(kProcessSampleRates); i++) {
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00001000 Init(kProcessSampleRates[i],
1001 kProcessSampleRates[i],
1002 kProcessSampleRates[i],
1003 2,
1004 2,
1005 2,
1006 false);
bjornv@webrtc.org3e102492013-02-14 15:29:09 +00001007 // Sampling frequency dependent variables.
Peter Kastingdce40cf2015-08-24 14:52:23 -07001008 const int num_ms_per_block =
1009 std::max(4, static_cast<int>(640 / frame_->samples_per_channel_));
bjornv@webrtc.org3e102492013-02-14 15:29:09 +00001010 const int delay_min_ms = -kLookaheadBlocks * num_ms_per_block;
1011 const int delay_max_ms = (kMaxDelayBlocks - 1) * num_ms_per_block;
1012
1013 // 1) Verify correct delay estimate at lookahead boundary.
1014 int delay_ms = TruncateToMultipleOf10(kSystemDelayMs + delay_min_ms);
1015 ProcessDelayVerificationTest(delay_ms, kSystemDelayMs, delay_min_ms,
1016 delay_max_ms);
1017 // 2) A delay less than maximum lookahead should give an delay estimate at
1018 // the boundary (= -kLookaheadBlocks * num_ms_per_block).
1019 delay_ms -= 20;
1020 ProcessDelayVerificationTest(delay_ms, kSystemDelayMs, delay_min_ms,
1021 delay_max_ms);
1022 // 3) Three values around zero delay. Note that we need to compensate for
1023 // the fake system_delay.
1024 delay_ms = TruncateToMultipleOf10(kSystemDelayMs - 10);
1025 ProcessDelayVerificationTest(delay_ms, kSystemDelayMs, delay_min_ms,
1026 delay_max_ms);
1027 delay_ms = TruncateToMultipleOf10(kSystemDelayMs);
1028 ProcessDelayVerificationTest(delay_ms, kSystemDelayMs, delay_min_ms,
1029 delay_max_ms);
1030 delay_ms = TruncateToMultipleOf10(kSystemDelayMs + 10);
1031 ProcessDelayVerificationTest(delay_ms, kSystemDelayMs, delay_min_ms,
1032 delay_max_ms);
1033 // 4) Verify correct delay estimate at maximum delay boundary.
1034 delay_ms = TruncateToMultipleOf10(kSystemDelayMs + delay_max_ms);
1035 ProcessDelayVerificationTest(delay_ms, kSystemDelayMs, delay_min_ms,
1036 delay_max_ms);
1037 // 5) A delay above the maximum delay should give an estimate at the
1038 // boundary (= (kMaxDelayBlocks - 1) * num_ms_per_block).
1039 delay_ms += 20;
1040 ProcessDelayVerificationTest(delay_ms, kSystemDelayMs, delay_min_ms,
1041 delay_max_ms);
1042 }
1043}
1044
niklase@google.com470e71d2011-07-07 08:21:25 +00001045TEST_F(ApmTest, EchoControlMobile) {
niklase@google.com470e71d2011-07-07 08:21:25 +00001046 // Turn AECM on (and AEC off)
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00001047 Init(16000, 16000, 16000, 2, 2, 2, false);
niklase@google.com470e71d2011-07-07 08:21:25 +00001048 EXPECT_EQ(apm_->kNoError, apm_->echo_control_mobile()->Enable(true));
1049 EXPECT_TRUE(apm_->echo_control_mobile()->is_enabled());
1050
niklase@google.com470e71d2011-07-07 08:21:25 +00001051 // Toggle routing modes
1052 EchoControlMobile::RoutingMode mode[] = {
1053 EchoControlMobile::kQuietEarpieceOrHeadset,
1054 EchoControlMobile::kEarpiece,
1055 EchoControlMobile::kLoudEarpiece,
1056 EchoControlMobile::kSpeakerphone,
1057 EchoControlMobile::kLoudSpeakerphone,
1058 };
pkasting25702cb2016-01-08 13:50:27 -08001059 for (size_t i = 0; i < arraysize(mode); i++) {
niklase@google.com470e71d2011-07-07 08:21:25 +00001060 EXPECT_EQ(apm_->kNoError,
1061 apm_->echo_control_mobile()->set_routing_mode(mode[i]));
1062 EXPECT_EQ(mode[i],
1063 apm_->echo_control_mobile()->routing_mode());
1064 }
1065 // Turn comfort noise off/on
1066 EXPECT_EQ(apm_->kNoError,
1067 apm_->echo_control_mobile()->enable_comfort_noise(false));
1068 EXPECT_FALSE(apm_->echo_control_mobile()->is_comfort_noise_enabled());
1069 EXPECT_EQ(apm_->kNoError,
1070 apm_->echo_control_mobile()->enable_comfort_noise(true));
1071 EXPECT_TRUE(apm_->echo_control_mobile()->is_comfort_noise_enabled());
bjornv@google.comc4b939c2011-07-13 08:09:56 +00001072 // Set and get echo path
ajm@google.com22e65152011-07-18 18:03:01 +00001073 const size_t echo_path_size =
1074 apm_->echo_control_mobile()->echo_path_size_bytes();
kwiberg62eaacf2016-02-17 06:39:05 -08001075 std::unique_ptr<char[]> echo_path_in(new char[echo_path_size]);
1076 std::unique_ptr<char[]> echo_path_out(new char[echo_path_size]);
bjornv@google.comc4b939c2011-07-13 08:09:56 +00001077 EXPECT_EQ(apm_->kNullPointerError,
1078 apm_->echo_control_mobile()->SetEchoPath(NULL, echo_path_size));
1079 EXPECT_EQ(apm_->kNullPointerError,
1080 apm_->echo_control_mobile()->GetEchoPath(NULL, echo_path_size));
1081 EXPECT_EQ(apm_->kBadParameterError,
andrew@webrtc.org3119ecf2011-11-01 17:00:18 +00001082 apm_->echo_control_mobile()->GetEchoPath(echo_path_out.get(), 1));
bjornv@google.comc4b939c2011-07-13 08:09:56 +00001083 EXPECT_EQ(apm_->kNoError,
andrew@webrtc.org3119ecf2011-11-01 17:00:18 +00001084 apm_->echo_control_mobile()->GetEchoPath(echo_path_out.get(),
bjornv@google.comc4b939c2011-07-13 08:09:56 +00001085 echo_path_size));
ajm@google.com22e65152011-07-18 18:03:01 +00001086 for (size_t i = 0; i < echo_path_size; i++) {
bjornv@google.comc4b939c2011-07-13 08:09:56 +00001087 echo_path_in[i] = echo_path_out[i] + 1;
1088 }
1089 EXPECT_EQ(apm_->kBadParameterError,
andrew@webrtc.org3119ecf2011-11-01 17:00:18 +00001090 apm_->echo_control_mobile()->SetEchoPath(echo_path_in.get(), 1));
bjornv@google.comc4b939c2011-07-13 08:09:56 +00001091 EXPECT_EQ(apm_->kNoError,
andrew@webrtc.org3119ecf2011-11-01 17:00:18 +00001092 apm_->echo_control_mobile()->SetEchoPath(echo_path_in.get(),
1093 echo_path_size));
bjornv@google.comc4b939c2011-07-13 08:09:56 +00001094 EXPECT_EQ(apm_->kNoError,
andrew@webrtc.org3119ecf2011-11-01 17:00:18 +00001095 apm_->echo_control_mobile()->GetEchoPath(echo_path_out.get(),
1096 echo_path_size));
ajm@google.com22e65152011-07-18 18:03:01 +00001097 for (size_t i = 0; i < echo_path_size; i++) {
bjornv@google.comc4b939c2011-07-13 08:09:56 +00001098 EXPECT_EQ(echo_path_in[i], echo_path_out[i]);
1099 }
andrew@webrtc.org75f19482012-02-09 17:16:18 +00001100
1101 // Process a few frames with NS in the default disabled state. This exercises
1102 // a different codepath than with it enabled.
1103 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(0));
1104 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1105 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(0));
1106 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1107
niklase@google.com470e71d2011-07-07 08:21:25 +00001108 // Turn AECM off
1109 EXPECT_EQ(apm_->kNoError, apm_->echo_control_mobile()->Enable(false));
1110 EXPECT_FALSE(apm_->echo_control_mobile()->is_enabled());
1111}
1112
aluebs@webrtc.orgc9ee4122014-02-03 14:41:57 +00001113TEST_F(ApmTest, GainControl) {
niklase@google.com470e71d2011-07-07 08:21:25 +00001114 // Testing gain modes
niklase@google.com470e71d2011-07-07 08:21:25 +00001115 EXPECT_EQ(apm_->kNoError,
1116 apm_->gain_control()->set_mode(
1117 apm_->gain_control()->mode()));
1118
1119 GainControl::Mode mode[] = {
1120 GainControl::kAdaptiveAnalog,
1121 GainControl::kAdaptiveDigital,
1122 GainControl::kFixedDigital
1123 };
pkasting25702cb2016-01-08 13:50:27 -08001124 for (size_t i = 0; i < arraysize(mode); i++) {
niklase@google.com470e71d2011-07-07 08:21:25 +00001125 EXPECT_EQ(apm_->kNoError,
1126 apm_->gain_control()->set_mode(mode[i]));
1127 EXPECT_EQ(mode[i], apm_->gain_control()->mode());
1128 }
1129 // Testing invalid target levels
1130 EXPECT_EQ(apm_->kBadParameterError,
1131 apm_->gain_control()->set_target_level_dbfs(-3));
1132 EXPECT_EQ(apm_->kBadParameterError,
1133 apm_->gain_control()->set_target_level_dbfs(-40));
1134 // Testing valid target levels
1135 EXPECT_EQ(apm_->kNoError,
1136 apm_->gain_control()->set_target_level_dbfs(
1137 apm_->gain_control()->target_level_dbfs()));
1138
1139 int level_dbfs[] = {0, 6, 31};
pkasting25702cb2016-01-08 13:50:27 -08001140 for (size_t i = 0; i < arraysize(level_dbfs); i++) {
niklase@google.com470e71d2011-07-07 08:21:25 +00001141 EXPECT_EQ(apm_->kNoError,
1142 apm_->gain_control()->set_target_level_dbfs(level_dbfs[i]));
1143 EXPECT_EQ(level_dbfs[i], apm_->gain_control()->target_level_dbfs());
1144 }
1145
1146 // Testing invalid compression gains
1147 EXPECT_EQ(apm_->kBadParameterError,
1148 apm_->gain_control()->set_compression_gain_db(-1));
1149 EXPECT_EQ(apm_->kBadParameterError,
1150 apm_->gain_control()->set_compression_gain_db(100));
1151
1152 // Testing valid compression gains
1153 EXPECT_EQ(apm_->kNoError,
1154 apm_->gain_control()->set_compression_gain_db(
1155 apm_->gain_control()->compression_gain_db()));
1156
1157 int gain_db[] = {0, 10, 90};
pkasting25702cb2016-01-08 13:50:27 -08001158 for (size_t i = 0; i < arraysize(gain_db); i++) {
niklase@google.com470e71d2011-07-07 08:21:25 +00001159 EXPECT_EQ(apm_->kNoError,
1160 apm_->gain_control()->set_compression_gain_db(gain_db[i]));
1161 EXPECT_EQ(gain_db[i], apm_->gain_control()->compression_gain_db());
1162 }
1163
1164 // Testing limiter off/on
1165 EXPECT_EQ(apm_->kNoError, apm_->gain_control()->enable_limiter(false));
1166 EXPECT_FALSE(apm_->gain_control()->is_limiter_enabled());
1167 EXPECT_EQ(apm_->kNoError, apm_->gain_control()->enable_limiter(true));
1168 EXPECT_TRUE(apm_->gain_control()->is_limiter_enabled());
1169
1170 // Testing invalid level limits
1171 EXPECT_EQ(apm_->kBadParameterError,
1172 apm_->gain_control()->set_analog_level_limits(-1, 512));
1173 EXPECT_EQ(apm_->kBadParameterError,
1174 apm_->gain_control()->set_analog_level_limits(100000, 512));
1175 EXPECT_EQ(apm_->kBadParameterError,
1176 apm_->gain_control()->set_analog_level_limits(512, -1));
1177 EXPECT_EQ(apm_->kBadParameterError,
1178 apm_->gain_control()->set_analog_level_limits(512, 100000));
1179 EXPECT_EQ(apm_->kBadParameterError,
1180 apm_->gain_control()->set_analog_level_limits(512, 255));
1181
1182 // Testing valid level limits
1183 EXPECT_EQ(apm_->kNoError,
1184 apm_->gain_control()->set_analog_level_limits(
1185 apm_->gain_control()->analog_level_minimum(),
1186 apm_->gain_control()->analog_level_maximum()));
1187
1188 int min_level[] = {0, 255, 1024};
pkasting25702cb2016-01-08 13:50:27 -08001189 for (size_t i = 0; i < arraysize(min_level); i++) {
niklase@google.com470e71d2011-07-07 08:21:25 +00001190 EXPECT_EQ(apm_->kNoError,
1191 apm_->gain_control()->set_analog_level_limits(min_level[i], 1024));
1192 EXPECT_EQ(min_level[i], apm_->gain_control()->analog_level_minimum());
1193 }
1194
1195 int max_level[] = {0, 1024, 65535};
pkasting25702cb2016-01-08 13:50:27 -08001196 for (size_t i = 0; i < arraysize(min_level); i++) {
niklase@google.com470e71d2011-07-07 08:21:25 +00001197 EXPECT_EQ(apm_->kNoError,
1198 apm_->gain_control()->set_analog_level_limits(0, max_level[i]));
1199 EXPECT_EQ(max_level[i], apm_->gain_control()->analog_level_maximum());
1200 }
1201
1202 // TODO(ajm): stream_is_saturated() and stream_analog_level()
1203
1204 // Turn AGC off
1205 EXPECT_EQ(apm_->kNoError, apm_->gain_control()->Enable(false));
1206 EXPECT_FALSE(apm_->gain_control()->is_enabled());
1207}
1208
andrew@webrtc.org27c69802014-02-18 20:24:56 +00001209void ApmTest::RunQuantizedVolumeDoesNotGetStuckTest(int sample_rate) {
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00001210 Init(sample_rate, sample_rate, sample_rate, 2, 2, 2, false);
andrew@webrtc.org27c69802014-02-18 20:24:56 +00001211 EXPECT_EQ(apm_->kNoError,
1212 apm_->gain_control()->set_mode(GainControl::kAdaptiveAnalog));
1213 EXPECT_EQ(apm_->kNoError, apm_->gain_control()->Enable(true));
1214
1215 int out_analog_level = 0;
1216 for (int i = 0; i < 2000; ++i) {
1217 ReadFrameWithRewind(near_file_, frame_);
1218 // Ensure the audio is at a low level, so the AGC will try to increase it.
1219 ScaleFrame(frame_, 0.25);
1220
1221 // Always pass in the same volume.
1222 EXPECT_EQ(apm_->kNoError,
1223 apm_->gain_control()->set_stream_analog_level(100));
1224 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1225 out_analog_level = apm_->gain_control()->stream_analog_level();
1226 }
1227
1228 // Ensure the AGC is still able to reach the maximum.
1229 EXPECT_EQ(255, out_analog_level);
1230}
1231
1232// Verifies that despite volume slider quantization, the AGC can continue to
1233// increase its volume.
1234TEST_F(ApmTest, QuantizedVolumeDoesNotGetStuck) {
pkasting25702cb2016-01-08 13:50:27 -08001235 for (size_t i = 0; i < arraysize(kSampleRates); ++i) {
andrew@webrtc.org27c69802014-02-18 20:24:56 +00001236 RunQuantizedVolumeDoesNotGetStuckTest(kSampleRates[i]);
1237 }
1238}
1239
1240void ApmTest::RunManualVolumeChangeIsPossibleTest(int sample_rate) {
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00001241 Init(sample_rate, sample_rate, sample_rate, 2, 2, 2, false);
andrew@webrtc.org27c69802014-02-18 20:24:56 +00001242 EXPECT_EQ(apm_->kNoError,
1243 apm_->gain_control()->set_mode(GainControl::kAdaptiveAnalog));
1244 EXPECT_EQ(apm_->kNoError, apm_->gain_control()->Enable(true));
1245
1246 int out_analog_level = 100;
1247 for (int i = 0; i < 1000; ++i) {
1248 ReadFrameWithRewind(near_file_, frame_);
1249 // Ensure the audio is at a low level, so the AGC will try to increase it.
1250 ScaleFrame(frame_, 0.25);
1251
1252 EXPECT_EQ(apm_->kNoError,
1253 apm_->gain_control()->set_stream_analog_level(out_analog_level));
1254 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1255 out_analog_level = apm_->gain_control()->stream_analog_level();
1256 }
1257
1258 // Ensure the volume was raised.
1259 EXPECT_GT(out_analog_level, 100);
1260 int highest_level_reached = out_analog_level;
1261 // Simulate a user manual volume change.
1262 out_analog_level = 100;
1263
1264 for (int i = 0; i < 300; ++i) {
1265 ReadFrameWithRewind(near_file_, frame_);
1266 ScaleFrame(frame_, 0.25);
1267
1268 EXPECT_EQ(apm_->kNoError,
1269 apm_->gain_control()->set_stream_analog_level(out_analog_level));
1270 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1271 out_analog_level = apm_->gain_control()->stream_analog_level();
1272 // Check that AGC respected the manually adjusted volume.
1273 EXPECT_LT(out_analog_level, highest_level_reached);
1274 }
1275 // Check that the volume was still raised.
1276 EXPECT_GT(out_analog_level, 100);
1277}
1278
1279TEST_F(ApmTest, ManualVolumeChangeIsPossible) {
pkasting25702cb2016-01-08 13:50:27 -08001280 for (size_t i = 0; i < arraysize(kSampleRates); ++i) {
andrew@webrtc.org27c69802014-02-18 20:24:56 +00001281 RunManualVolumeChangeIsPossibleTest(kSampleRates[i]);
1282 }
1283}
1284
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +00001285#if !defined(WEBRTC_ANDROID) && !defined(WEBRTC_IOS)
1286TEST_F(ApmTest, AgcOnlyAdaptsWhenTargetSignalIsPresent) {
1287 const int kSampleRateHz = 16000;
pkasting25702cb2016-01-08 13:50:27 -08001288 const size_t kSamplesPerChannel =
1289 static_cast<size_t>(AudioProcessing::kChunkSizeMs * kSampleRateHz / 1000);
Peter Kasting69558702016-01-12 16:26:35 -08001290 const size_t kNumInputChannels = 2;
1291 const size_t kNumOutputChannels = 1;
pkasting25702cb2016-01-08 13:50:27 -08001292 const size_t kNumChunks = 700;
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +00001293 const float kScaleFactor = 0.25f;
1294 Config config;
1295 std::vector<webrtc::Point> geometry;
1296 geometry.push_back(webrtc::Point(0.f, 0.f, 0.f));
1297 geometry.push_back(webrtc::Point(0.05f, 0.f, 0.f));
1298 config.Set<Beamforming>(new Beamforming(true, geometry));
mgraczyk@chromium.org0f663de2015-03-13 00:13:32 +00001299 testing::NiceMock<MockNonlinearBeamformer>* beamformer =
Alejandro Luebsf4022ff2016-07-01 17:19:09 -07001300 new testing::NiceMock<MockNonlinearBeamformer>(geometry, 1u);
kwiberg62eaacf2016-02-17 06:39:05 -08001301 std::unique_ptr<AudioProcessing> apm(
kwiberg@webrtc.org00b8f6b2015-02-26 14:34:55 +00001302 AudioProcessing::Create(config, beamformer));
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +00001303 EXPECT_EQ(kNoErr, apm->gain_control()->Enable(true));
1304 ChannelBuffer<float> src_buf(kSamplesPerChannel, kNumInputChannels);
1305 ChannelBuffer<float> dest_buf(kSamplesPerChannel, kNumOutputChannels);
pkasting25702cb2016-01-08 13:50:27 -08001306 const size_t max_length = kSamplesPerChannel * std::max(kNumInputChannels,
1307 kNumOutputChannels);
kwiberg62eaacf2016-02-17 06:39:05 -08001308 std::unique_ptr<int16_t[]> int_data(new int16_t[max_length]);
1309 std::unique_ptr<float[]> float_data(new float[max_length]);
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +00001310 std::string filename = ResourceFilePath("far", kSampleRateHz);
1311 FILE* far_file = fopen(filename.c_str(), "rb");
1312 ASSERT_TRUE(far_file != NULL) << "Could not open file " << filename << "\n";
1313 const int kDefaultVolume = apm->gain_control()->stream_analog_level();
1314 const int kDefaultCompressionGain =
1315 apm->gain_control()->compression_gain_db();
1316 bool is_target = false;
1317 EXPECT_CALL(*beamformer, is_target_present())
1318 .WillRepeatedly(testing::ReturnPointee(&is_target));
pkasting25702cb2016-01-08 13:50:27 -08001319 for (size_t i = 0; i < kNumChunks; ++i) {
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +00001320 ASSERT_TRUE(ReadChunk(far_file,
1321 int_data.get(),
1322 float_data.get(),
1323 &src_buf));
Peter Kasting69558702016-01-12 16:26:35 -08001324 for (size_t j = 0; j < kNumInputChannels; ++j) {
pkasting25702cb2016-01-08 13:50:27 -08001325 for (size_t k = 0; k < kSamplesPerChannel; ++k) {
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +00001326 src_buf.channels()[j][k] *= kScaleFactor;
1327 }
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +00001328 }
1329 EXPECT_EQ(kNoErr,
1330 apm->ProcessStream(src_buf.channels(),
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +00001331 src_buf.num_frames(),
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +00001332 kSampleRateHz,
1333 LayoutFromChannels(src_buf.num_channels()),
1334 kSampleRateHz,
1335 LayoutFromChannels(dest_buf.num_channels()),
1336 dest_buf.channels()));
1337 }
1338 EXPECT_EQ(kDefaultVolume,
1339 apm->gain_control()->stream_analog_level());
1340 EXPECT_EQ(kDefaultCompressionGain,
1341 apm->gain_control()->compression_gain_db());
1342 rewind(far_file);
1343 is_target = true;
pkasting25702cb2016-01-08 13:50:27 -08001344 for (size_t i = 0; i < kNumChunks; ++i) {
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +00001345 ASSERT_TRUE(ReadChunk(far_file,
1346 int_data.get(),
1347 float_data.get(),
1348 &src_buf));
Peter Kasting69558702016-01-12 16:26:35 -08001349 for (size_t j = 0; j < kNumInputChannels; ++j) {
pkasting25702cb2016-01-08 13:50:27 -08001350 for (size_t k = 0; k < kSamplesPerChannel; ++k) {
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +00001351 src_buf.channels()[j][k] *= kScaleFactor;
1352 }
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +00001353 }
1354 EXPECT_EQ(kNoErr,
1355 apm->ProcessStream(src_buf.channels(),
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +00001356 src_buf.num_frames(),
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +00001357 kSampleRateHz,
1358 LayoutFromChannels(src_buf.num_channels()),
1359 kSampleRateHz,
1360 LayoutFromChannels(dest_buf.num_channels()),
1361 dest_buf.channels()));
1362 }
1363 EXPECT_LT(kDefaultVolume,
1364 apm->gain_control()->stream_analog_level());
1365 EXPECT_LT(kDefaultCompressionGain,
1366 apm->gain_control()->compression_gain_db());
1367 ASSERT_EQ(0, fclose(far_file));
1368}
1369#endif
1370
niklase@google.com470e71d2011-07-07 08:21:25 +00001371TEST_F(ApmTest, NoiseSuppression) {
andrew@webrtc.org648af742012-02-08 01:57:29 +00001372 // Test valid suppression levels.
niklase@google.com470e71d2011-07-07 08:21:25 +00001373 NoiseSuppression::Level level[] = {
1374 NoiseSuppression::kLow,
1375 NoiseSuppression::kModerate,
1376 NoiseSuppression::kHigh,
1377 NoiseSuppression::kVeryHigh
1378 };
pkasting25702cb2016-01-08 13:50:27 -08001379 for (size_t i = 0; i < arraysize(level); i++) {
niklase@google.com470e71d2011-07-07 08:21:25 +00001380 EXPECT_EQ(apm_->kNoError,
1381 apm_->noise_suppression()->set_level(level[i]));
1382 EXPECT_EQ(level[i], apm_->noise_suppression()->level());
1383 }
1384
andrew@webrtc.org648af742012-02-08 01:57:29 +00001385 // Turn NS on/off
niklase@google.com470e71d2011-07-07 08:21:25 +00001386 EXPECT_EQ(apm_->kNoError, apm_->noise_suppression()->Enable(true));
1387 EXPECT_TRUE(apm_->noise_suppression()->is_enabled());
1388 EXPECT_EQ(apm_->kNoError, apm_->noise_suppression()->Enable(false));
1389 EXPECT_FALSE(apm_->noise_suppression()->is_enabled());
1390}
1391
1392TEST_F(ApmTest, HighPassFilter) {
andrew@webrtc.org648af742012-02-08 01:57:29 +00001393 // Turn HP filter on/off
peah8271d042016-11-22 07:24:52 -08001394 AudioProcessing::Config apm_config;
1395 apm_config.high_pass_filter.enabled = true;
1396 apm_->ApplyConfig(apm_config);
1397 apm_config.high_pass_filter.enabled = false;
1398 apm_->ApplyConfig(apm_config);
niklase@google.com470e71d2011-07-07 08:21:25 +00001399}
1400
1401TEST_F(ApmTest, LevelEstimator) {
andrew@webrtc.org648af742012-02-08 01:57:29 +00001402 // Turn level estimator on/off
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001403 EXPECT_EQ(apm_->kNoError, apm_->level_estimator()->Enable(false));
niklase@google.com470e71d2011-07-07 08:21:25 +00001404 EXPECT_FALSE(apm_->level_estimator()->is_enabled());
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001405
1406 EXPECT_EQ(apm_->kNotEnabledError, apm_->level_estimator()->RMS());
1407
1408 EXPECT_EQ(apm_->kNoError, apm_->level_estimator()->Enable(true));
1409 EXPECT_TRUE(apm_->level_estimator()->is_enabled());
1410
1411 // Run this test in wideband; in super-wb, the splitting filter distorts the
1412 // audio enough to cause deviation from the expectation for small values.
andrew@webrtc.org63a50982012-05-02 23:56:37 +00001413 frame_->samples_per_channel_ = 160;
1414 frame_->num_channels_ = 2;
1415 frame_->sample_rate_hz_ = 16000;
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001416
1417 // Min value if no frames have been processed.
1418 EXPECT_EQ(127, apm_->level_estimator()->RMS());
1419
1420 // Min value on zero frames.
1421 SetFrameTo(frame_, 0);
1422 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1423 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1424 EXPECT_EQ(127, apm_->level_estimator()->RMS());
1425
1426 // Try a few RMS values.
1427 // (These also test that the value resets after retrieving it.)
1428 SetFrameTo(frame_, 32767);
1429 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1430 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1431 EXPECT_EQ(0, apm_->level_estimator()->RMS());
1432
1433 SetFrameTo(frame_, 30000);
1434 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1435 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1436 EXPECT_EQ(1, apm_->level_estimator()->RMS());
1437
1438 SetFrameTo(frame_, 10000);
1439 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1440 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1441 EXPECT_EQ(10, apm_->level_estimator()->RMS());
1442
1443 SetFrameTo(frame_, 10);
1444 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1445 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1446 EXPECT_EQ(70, apm_->level_estimator()->RMS());
1447
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001448 // Verify reset after enable/disable.
1449 SetFrameTo(frame_, 32767);
1450 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1451 EXPECT_EQ(apm_->kNoError, apm_->level_estimator()->Enable(false));
1452 EXPECT_EQ(apm_->kNoError, apm_->level_estimator()->Enable(true));
1453 SetFrameTo(frame_, 1);
1454 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1455 EXPECT_EQ(90, apm_->level_estimator()->RMS());
1456
1457 // Verify reset after initialize.
1458 SetFrameTo(frame_, 32767);
1459 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1460 EXPECT_EQ(apm_->kNoError, apm_->Initialize());
1461 SetFrameTo(frame_, 1);
1462 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1463 EXPECT_EQ(90, apm_->level_estimator()->RMS());
niklase@google.com470e71d2011-07-07 08:21:25 +00001464}
1465
1466TEST_F(ApmTest, VoiceDetection) {
1467 // Test external VAD
1468 EXPECT_EQ(apm_->kNoError,
1469 apm_->voice_detection()->set_stream_has_voice(true));
1470 EXPECT_TRUE(apm_->voice_detection()->stream_has_voice());
1471 EXPECT_EQ(apm_->kNoError,
1472 apm_->voice_detection()->set_stream_has_voice(false));
1473 EXPECT_FALSE(apm_->voice_detection()->stream_has_voice());
1474
andrew@webrtc.org648af742012-02-08 01:57:29 +00001475 // Test valid likelihoods
niklase@google.com470e71d2011-07-07 08:21:25 +00001476 VoiceDetection::Likelihood likelihood[] = {
1477 VoiceDetection::kVeryLowLikelihood,
1478 VoiceDetection::kLowLikelihood,
1479 VoiceDetection::kModerateLikelihood,
1480 VoiceDetection::kHighLikelihood
1481 };
pkasting25702cb2016-01-08 13:50:27 -08001482 for (size_t i = 0; i < arraysize(likelihood); i++) {
niklase@google.com470e71d2011-07-07 08:21:25 +00001483 EXPECT_EQ(apm_->kNoError,
1484 apm_->voice_detection()->set_likelihood(likelihood[i]));
1485 EXPECT_EQ(likelihood[i], apm_->voice_detection()->likelihood());
1486 }
1487
1488 /* TODO(bjornv): Enable once VAD supports other frame lengths than 10 ms
andrew@webrtc.org648af742012-02-08 01:57:29 +00001489 // Test invalid frame sizes
niklase@google.com470e71d2011-07-07 08:21:25 +00001490 EXPECT_EQ(apm_->kBadParameterError,
1491 apm_->voice_detection()->set_frame_size_ms(12));
1492
andrew@webrtc.org648af742012-02-08 01:57:29 +00001493 // Test valid frame sizes
niklase@google.com470e71d2011-07-07 08:21:25 +00001494 for (int i = 10; i <= 30; i += 10) {
1495 EXPECT_EQ(apm_->kNoError,
1496 apm_->voice_detection()->set_frame_size_ms(i));
1497 EXPECT_EQ(i, apm_->voice_detection()->frame_size_ms());
1498 }
1499 */
1500
andrew@webrtc.org648af742012-02-08 01:57:29 +00001501 // Turn VAD on/off
niklase@google.com470e71d2011-07-07 08:21:25 +00001502 EXPECT_EQ(apm_->kNoError, apm_->voice_detection()->Enable(true));
1503 EXPECT_TRUE(apm_->voice_detection()->is_enabled());
1504 EXPECT_EQ(apm_->kNoError, apm_->voice_detection()->Enable(false));
1505 EXPECT_FALSE(apm_->voice_detection()->is_enabled());
1506
andrew@webrtc.orged083d42011-09-19 15:28:51 +00001507 // Test that AudioFrame activity is maintained when VAD is disabled.
1508 EXPECT_EQ(apm_->kNoError, apm_->voice_detection()->Enable(false));
1509 AudioFrame::VADActivity activity[] = {
1510 AudioFrame::kVadActive,
1511 AudioFrame::kVadPassive,
1512 AudioFrame::kVadUnknown
1513 };
pkasting25702cb2016-01-08 13:50:27 -08001514 for (size_t i = 0; i < arraysize(activity); i++) {
andrew@webrtc.org63a50982012-05-02 23:56:37 +00001515 frame_->vad_activity_ = activity[i];
andrew@webrtc.orged083d42011-09-19 15:28:51 +00001516 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
andrew@webrtc.org63a50982012-05-02 23:56:37 +00001517 EXPECT_EQ(activity[i], frame_->vad_activity_);
andrew@webrtc.orged083d42011-09-19 15:28:51 +00001518 }
1519
1520 // Test that AudioFrame activity is set when VAD is enabled.
1521 EXPECT_EQ(apm_->kNoError, apm_->voice_detection()->Enable(true));
andrew@webrtc.org63a50982012-05-02 23:56:37 +00001522 frame_->vad_activity_ = AudioFrame::kVadUnknown;
andrew@webrtc.orged083d42011-09-19 15:28:51 +00001523 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
andrew@webrtc.org63a50982012-05-02 23:56:37 +00001524 EXPECT_NE(AudioFrame::kVadUnknown, frame_->vad_activity_);
andrew@webrtc.orged083d42011-09-19 15:28:51 +00001525
niklase@google.com470e71d2011-07-07 08:21:25 +00001526 // TODO(bjornv): Add tests for streamed voice; stream_has_voice()
1527}
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001528
andrew@webrtc.orgecac9b72012-05-02 00:04:10 +00001529TEST_F(ApmTest, AllProcessingDisabledByDefault) {
1530 EXPECT_FALSE(apm_->echo_cancellation()->is_enabled());
1531 EXPECT_FALSE(apm_->echo_control_mobile()->is_enabled());
1532 EXPECT_FALSE(apm_->gain_control()->is_enabled());
1533 EXPECT_FALSE(apm_->high_pass_filter()->is_enabled());
1534 EXPECT_FALSE(apm_->level_estimator()->is_enabled());
1535 EXPECT_FALSE(apm_->noise_suppression()->is_enabled());
1536 EXPECT_FALSE(apm_->voice_detection()->is_enabled());
1537}
1538
1539TEST_F(ApmTest, NoProcessingWhenAllComponentsDisabled) {
pkasting25702cb2016-01-08 13:50:27 -08001540 for (size_t i = 0; i < arraysize(kSampleRates); i++) {
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00001541 Init(kSampleRates[i], kSampleRates[i], kSampleRates[i], 2, 2, 2, false);
andrew@webrtc.orgecac9b72012-05-02 00:04:10 +00001542 SetFrameTo(frame_, 1000, 2000);
andrew@webrtc.orgae1a58b2013-01-22 04:44:30 +00001543 AudioFrame frame_copy;
1544 frame_copy.CopyFrom(*frame_);
andrew@webrtc.orgecac9b72012-05-02 00:04:10 +00001545 for (int j = 0; j < 1000; j++) {
1546 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1547 EXPECT_TRUE(FrameDataAreEqual(*frame_, frame_copy));
ekmeyerson60d9b332015-08-14 10:35:55 -07001548 EXPECT_EQ(apm_->kNoError, apm_->ProcessReverseStream(frame_));
1549 EXPECT_TRUE(FrameDataAreEqual(*frame_, frame_copy));
andrew@webrtc.orgecac9b72012-05-02 00:04:10 +00001550 }
1551 }
1552}
1553
mgraczyk@chromium.orgd6e84d92015-01-14 01:33:54 +00001554TEST_F(ApmTest, NoProcessingWhenAllComponentsDisabledFloat) {
1555 // Test that ProcessStream copies input to output even with no processing.
1556 const size_t kSamples = 80;
1557 const int sample_rate = 8000;
1558 const float src[kSamples] = {
1559 -1.0f, 0.0f, 1.0f
1560 };
1561 float dest[kSamples] = {};
1562
1563 auto src_channels = &src[0];
1564 auto dest_channels = &dest[0];
1565
1566 apm_.reset(AudioProcessing::Create());
1567 EXPECT_NOERR(apm_->ProcessStream(
1568 &src_channels, kSamples, sample_rate, LayoutFromChannels(1),
1569 sample_rate, LayoutFromChannels(1), &dest_channels));
1570
1571 for (size_t i = 0; i < kSamples; ++i) {
1572 EXPECT_EQ(src[i], dest[i]);
1573 }
ekmeyerson60d9b332015-08-14 10:35:55 -07001574
1575 // Same for ProcessReverseStream.
1576 float rev_dest[kSamples] = {};
1577 auto rev_dest_channels = &rev_dest[0];
1578
1579 StreamConfig input_stream = {sample_rate, 1};
1580 StreamConfig output_stream = {sample_rate, 1};
1581 EXPECT_NOERR(apm_->ProcessReverseStream(&src_channels, input_stream,
1582 output_stream, &rev_dest_channels));
1583
1584 for (size_t i = 0; i < kSamples; ++i) {
1585 EXPECT_EQ(src[i], rev_dest[i]);
1586 }
mgraczyk@chromium.orgd6e84d92015-01-14 01:33:54 +00001587}
1588
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +00001589TEST_F(ApmTest, IdenticalInputChannelsResultInIdenticalOutputChannels) {
1590 EnableAllComponents();
1591
pkasting25702cb2016-01-08 13:50:27 -08001592 for (size_t i = 0; i < arraysize(kProcessSampleRates); i++) {
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00001593 Init(kProcessSampleRates[i],
1594 kProcessSampleRates[i],
1595 kProcessSampleRates[i],
1596 2,
1597 2,
1598 2,
1599 false);
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +00001600 int analog_level = 127;
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00001601 ASSERT_EQ(0, feof(far_file_));
1602 ASSERT_EQ(0, feof(near_file_));
andrew@webrtc.org17e40642014-03-04 20:58:13 +00001603 while (ReadFrame(far_file_, revframe_) && ReadFrame(near_file_, frame_)) {
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +00001604 CopyLeftToRightChannel(revframe_->data_, revframe_->samples_per_channel_);
1605
aluebsb0319552016-03-17 20:39:53 -07001606 ASSERT_EQ(kNoErr, apm_->ProcessReverseStream(revframe_));
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +00001607
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +00001608 CopyLeftToRightChannel(frame_->data_, frame_->samples_per_channel_);
1609 frame_->vad_activity_ = AudioFrame::kVadUnknown;
1610
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00001611 ASSERT_EQ(kNoErr, apm_->set_stream_delay_ms(0));
andrew@webrtc.org6be1e932013-03-01 18:47:28 +00001612 apm_->echo_cancellation()->set_stream_drift_samples(0);
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00001613 ASSERT_EQ(kNoErr,
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +00001614 apm_->gain_control()->set_stream_analog_level(analog_level));
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00001615 ASSERT_EQ(kNoErr, apm_->ProcessStream(frame_));
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +00001616 analog_level = apm_->gain_control()->stream_analog_level();
1617
1618 VerifyChannelsAreEqual(frame_->data_, frame_->samples_per_channel_);
1619 }
bjornv@webrtc.org3e102492013-02-14 15:29:09 +00001620 rewind(far_file_);
1621 rewind(near_file_);
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +00001622 }
1623}
1624
bjornv@webrtc.orgcb0ea432014-06-09 08:21:52 +00001625TEST_F(ApmTest, SplittingFilter) {
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001626 // Verify the filter is not active through undistorted audio when:
1627 // 1. No components are enabled...
1628 SetFrameTo(frame_, 1000);
andrew@webrtc.orgae1a58b2013-01-22 04:44:30 +00001629 AudioFrame frame_copy;
1630 frame_copy.CopyFrom(*frame_);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001631 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1632 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1633 EXPECT_TRUE(FrameDataAreEqual(*frame_, frame_copy));
1634
1635 // 2. Only the level estimator is enabled...
1636 SetFrameTo(frame_, 1000);
andrew@webrtc.orgae1a58b2013-01-22 04:44:30 +00001637 frame_copy.CopyFrom(*frame_);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001638 EXPECT_EQ(apm_->kNoError, apm_->level_estimator()->Enable(true));
1639 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1640 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1641 EXPECT_TRUE(FrameDataAreEqual(*frame_, frame_copy));
1642 EXPECT_EQ(apm_->kNoError, apm_->level_estimator()->Enable(false));
1643
1644 // 3. Only VAD is enabled...
1645 SetFrameTo(frame_, 1000);
andrew@webrtc.orgae1a58b2013-01-22 04:44:30 +00001646 frame_copy.CopyFrom(*frame_);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001647 EXPECT_EQ(apm_->kNoError, apm_->voice_detection()->Enable(true));
1648 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1649 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1650 EXPECT_TRUE(FrameDataAreEqual(*frame_, frame_copy));
1651 EXPECT_EQ(apm_->kNoError, apm_->voice_detection()->Enable(false));
1652
1653 // 4. Both VAD and the level estimator are enabled...
1654 SetFrameTo(frame_, 1000);
andrew@webrtc.orgae1a58b2013-01-22 04:44:30 +00001655 frame_copy.CopyFrom(*frame_);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001656 EXPECT_EQ(apm_->kNoError, apm_->level_estimator()->Enable(true));
1657 EXPECT_EQ(apm_->kNoError, apm_->voice_detection()->Enable(true));
1658 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1659 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1660 EXPECT_TRUE(FrameDataAreEqual(*frame_, frame_copy));
1661 EXPECT_EQ(apm_->kNoError, apm_->level_estimator()->Enable(false));
1662 EXPECT_EQ(apm_->kNoError, apm_->voice_detection()->Enable(false));
1663
1664 // 5. Not using super-wb.
andrew@webrtc.org63a50982012-05-02 23:56:37 +00001665 frame_->samples_per_channel_ = 160;
1666 frame_->num_channels_ = 2;
1667 frame_->sample_rate_hz_ = 16000;
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001668 // Enable AEC, which would require the filter in super-wb. We rely on the
1669 // first few frames of data being unaffected by the AEC.
1670 // TODO(andrew): This test, and the one below, rely rather tenuously on the
1671 // behavior of the AEC. Think of something more robust.
1672 EXPECT_EQ(apm_->kNoError, apm_->echo_cancellation()->Enable(true));
bjornv@webrtc.orgcb0ea432014-06-09 08:21:52 +00001673 // Make sure we have extended filter enabled. This makes sure nothing is
1674 // touched until we have a farend frame.
1675 Config config;
Henrik Lundin441f6342015-06-09 16:03:13 +02001676 config.Set<ExtendedFilter>(new ExtendedFilter(true));
bjornv@webrtc.orgcb0ea432014-06-09 08:21:52 +00001677 apm_->SetExtraOptions(config);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001678 SetFrameTo(frame_, 1000);
andrew@webrtc.orgae1a58b2013-01-22 04:44:30 +00001679 frame_copy.CopyFrom(*frame_);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001680 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(0));
andrew@webrtc.org6be1e932013-03-01 18:47:28 +00001681 apm_->echo_cancellation()->set_stream_drift_samples(0);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001682 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1683 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(0));
andrew@webrtc.org6be1e932013-03-01 18:47:28 +00001684 apm_->echo_cancellation()->set_stream_drift_samples(0);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001685 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1686 EXPECT_TRUE(FrameDataAreEqual(*frame_, frame_copy));
1687
1688 // Check the test is valid. We should have distortion from the filter
1689 // when AEC is enabled (which won't affect the audio).
andrew@webrtc.org63a50982012-05-02 23:56:37 +00001690 frame_->samples_per_channel_ = 320;
1691 frame_->num_channels_ = 2;
1692 frame_->sample_rate_hz_ = 32000;
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001693 SetFrameTo(frame_, 1000);
andrew@webrtc.orgae1a58b2013-01-22 04:44:30 +00001694 frame_copy.CopyFrom(*frame_);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001695 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(0));
andrew@webrtc.org6be1e932013-03-01 18:47:28 +00001696 apm_->echo_cancellation()->set_stream_drift_samples(0);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00001697 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1698 EXPECT_FALSE(FrameDataAreEqual(*frame_, frame_copy));
1699}
1700
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001701#ifdef WEBRTC_AUDIOPROC_DEBUG_DUMP
1702void ApmTest::ProcessDebugDump(const std::string& in_filename,
1703 const std::string& out_filename,
ivocd66b44d2016-01-15 03:06:36 -08001704 Format format,
1705 int max_size_bytes) {
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001706 FILE* in_file = fopen(in_filename.c_str(), "rb");
1707 ASSERT_TRUE(in_file != NULL);
1708 audioproc::Event event_msg;
1709 bool first_init = true;
1710
1711 while (ReadMessageFromFile(in_file, &event_msg)) {
1712 if (event_msg.type() == audioproc::Event::INIT) {
1713 const audioproc::Init msg = event_msg.init();
1714 int reverse_sample_rate = msg.sample_rate();
1715 if (msg.has_reverse_sample_rate()) {
1716 reverse_sample_rate = msg.reverse_sample_rate();
1717 }
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00001718 int output_sample_rate = msg.sample_rate();
1719 if (msg.has_output_sample_rate()) {
1720 output_sample_rate = msg.output_sample_rate();
1721 }
1722
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001723 Init(msg.sample_rate(),
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00001724 output_sample_rate,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001725 reverse_sample_rate,
1726 msg.num_input_channels(),
1727 msg.num_output_channels(),
1728 msg.num_reverse_channels(),
1729 false);
1730 if (first_init) {
1731 // StartDebugRecording() writes an additional init message. Don't start
1732 // recording until after the first init to avoid the extra message.
ivocd66b44d2016-01-15 03:06:36 -08001733 EXPECT_NOERR(
1734 apm_->StartDebugRecording(out_filename.c_str(), max_size_bytes));
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001735 first_init = false;
1736 }
1737
1738 } else if (event_msg.type() == audioproc::Event::REVERSE_STREAM) {
1739 const audioproc::ReverseStream msg = event_msg.reverse_stream();
1740
1741 if (msg.channel_size() > 0) {
Peter Kasting69558702016-01-12 16:26:35 -08001742 ASSERT_EQ(revframe_->num_channels_,
1743 static_cast<size_t>(msg.channel_size()));
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001744 for (int i = 0; i < msg.channel_size(); ++i) {
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +00001745 memcpy(revfloat_cb_->channels()[i],
1746 msg.channel(i).data(),
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001747 msg.channel(i).size());
1748 }
1749 } else {
1750 memcpy(revframe_->data_, msg.data().data(), msg.data().size());
1751 if (format == kFloatFormat) {
1752 // We're using an int16 input file; convert to float.
1753 ConvertToFloat(*revframe_, revfloat_cb_.get());
1754 }
1755 }
1756 AnalyzeReverseStreamChooser(format);
1757
1758 } else if (event_msg.type() == audioproc::Event::STREAM) {
1759 const audioproc::Stream msg = event_msg.stream();
1760 // ProcessStream could have changed this for the output frame.
1761 frame_->num_channels_ = apm_->num_input_channels();
1762
1763 EXPECT_NOERR(apm_->gain_control()->set_stream_analog_level(msg.level()));
1764 EXPECT_NOERR(apm_->set_stream_delay_ms(msg.delay()));
1765 apm_->echo_cancellation()->set_stream_drift_samples(msg.drift());
1766 if (msg.has_keypress()) {
1767 apm_->set_stream_key_pressed(msg.keypress());
1768 } else {
1769 apm_->set_stream_key_pressed(true);
1770 }
1771
1772 if (msg.input_channel_size() > 0) {
Peter Kasting69558702016-01-12 16:26:35 -08001773 ASSERT_EQ(frame_->num_channels_,
1774 static_cast<size_t>(msg.input_channel_size()));
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001775 for (int i = 0; i < msg.input_channel_size(); ++i) {
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +00001776 memcpy(float_cb_->channels()[i],
1777 msg.input_channel(i).data(),
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001778 msg.input_channel(i).size());
1779 }
1780 } else {
1781 memcpy(frame_->data_, msg.input_data().data(), msg.input_data().size());
1782 if (format == kFloatFormat) {
1783 // We're using an int16 input file; convert to float.
1784 ConvertToFloat(*frame_, float_cb_.get());
1785 }
1786 }
1787 ProcessStreamChooser(format);
1788 }
1789 }
1790 EXPECT_NOERR(apm_->StopDebugRecording());
1791 fclose(in_file);
1792}
1793
1794void ApmTest::VerifyDebugDumpTest(Format format) {
1795 const std::string in_filename = test::ResourcePath("ref03", "aecdump");
henrik.lundin@webrtc.org1092ea02014-04-02 07:46:49 +00001796 std::string format_string;
1797 switch (format) {
1798 case kIntFormat:
1799 format_string = "_int";
1800 break;
1801 case kFloatFormat:
1802 format_string = "_float";
1803 break;
1804 }
pbos@webrtc.orga525c982015-01-12 17:31:18 +00001805 const std::string ref_filename = test::TempFilename(
1806 test::OutputPath(), std::string("ref") + format_string + "_aecdump");
1807 const std::string out_filename = test::TempFilename(
1808 test::OutputPath(), std::string("out") + format_string + "_aecdump");
ivocd66b44d2016-01-15 03:06:36 -08001809 const std::string limited_filename = test::TempFilename(
1810 test::OutputPath(), std::string("limited") + format_string + "_aecdump");
1811 const size_t logging_limit_bytes = 100000;
1812 // We expect at least this many bytes in the created logfile.
1813 const size_t logging_expected_bytes = 95000;
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001814 EnableAllComponents();
ivocd66b44d2016-01-15 03:06:36 -08001815 ProcessDebugDump(in_filename, ref_filename, format, -1);
1816 ProcessDebugDump(ref_filename, out_filename, format, -1);
1817 ProcessDebugDump(ref_filename, limited_filename, format, logging_limit_bytes);
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001818
1819 FILE* ref_file = fopen(ref_filename.c_str(), "rb");
1820 FILE* out_file = fopen(out_filename.c_str(), "rb");
ivocd66b44d2016-01-15 03:06:36 -08001821 FILE* limited_file = fopen(limited_filename.c_str(), "rb");
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001822 ASSERT_TRUE(ref_file != NULL);
1823 ASSERT_TRUE(out_file != NULL);
ivocd66b44d2016-01-15 03:06:36 -08001824 ASSERT_TRUE(limited_file != NULL);
kwiberg62eaacf2016-02-17 06:39:05 -08001825 std::unique_ptr<uint8_t[]> ref_bytes;
1826 std::unique_ptr<uint8_t[]> out_bytes;
1827 std::unique_ptr<uint8_t[]> limited_bytes;
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001828
1829 size_t ref_size = ReadMessageBytesFromFile(ref_file, &ref_bytes);
1830 size_t out_size = ReadMessageBytesFromFile(out_file, &out_bytes);
ivocd66b44d2016-01-15 03:06:36 -08001831 size_t limited_size = ReadMessageBytesFromFile(limited_file, &limited_bytes);
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001832 size_t bytes_read = 0;
ivocd66b44d2016-01-15 03:06:36 -08001833 size_t bytes_read_limited = 0;
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001834 while (ref_size > 0 && out_size > 0) {
1835 bytes_read += ref_size;
ivocd66b44d2016-01-15 03:06:36 -08001836 bytes_read_limited += limited_size;
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001837 EXPECT_EQ(ref_size, out_size);
ivocd66b44d2016-01-15 03:06:36 -08001838 EXPECT_GE(ref_size, limited_size);
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001839 EXPECT_EQ(0, memcmp(ref_bytes.get(), out_bytes.get(), ref_size));
ivocd66b44d2016-01-15 03:06:36 -08001840 EXPECT_EQ(0, memcmp(ref_bytes.get(), limited_bytes.get(), limited_size));
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001841 ref_size = ReadMessageBytesFromFile(ref_file, &ref_bytes);
1842 out_size = ReadMessageBytesFromFile(out_file, &out_bytes);
ivocd66b44d2016-01-15 03:06:36 -08001843 limited_size = ReadMessageBytesFromFile(limited_file, &limited_bytes);
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001844 }
1845 EXPECT_GT(bytes_read, 0u);
ivocd66b44d2016-01-15 03:06:36 -08001846 EXPECT_GT(bytes_read_limited, logging_expected_bytes);
1847 EXPECT_LE(bytes_read_limited, logging_limit_bytes);
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001848 EXPECT_NE(0, feof(ref_file));
1849 EXPECT_NE(0, feof(out_file));
ivocd66b44d2016-01-15 03:06:36 -08001850 EXPECT_NE(0, feof(limited_file));
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001851 ASSERT_EQ(0, fclose(ref_file));
1852 ASSERT_EQ(0, fclose(out_file));
ivocd66b44d2016-01-15 03:06:36 -08001853 ASSERT_EQ(0, fclose(limited_file));
Peter Boströmfade1792015-05-12 10:44:11 +02001854 remove(ref_filename.c_str());
1855 remove(out_filename.c_str());
ivocd66b44d2016-01-15 03:06:36 -08001856 remove(limited_filename.c_str());
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001857}
1858
pbosc7a65692016-05-06 12:50:04 -07001859TEST_F(ApmTest, VerifyDebugDumpInt) {
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001860 VerifyDebugDumpTest(kIntFormat);
1861}
1862
pbosc7a65692016-05-06 12:50:04 -07001863TEST_F(ApmTest, VerifyDebugDumpFloat) {
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001864 VerifyDebugDumpTest(kFloatFormat);
1865}
1866#endif
1867
andrew@webrtc.org7bf26462011-12-03 00:03:31 +00001868// TODO(andrew): expand test to verify output.
pbosc7a65692016-05-06 12:50:04 -07001869TEST_F(ApmTest, DebugDump) {
pbos@webrtc.orga525c982015-01-12 17:31:18 +00001870 const std::string filename =
1871 test::TempFilename(test::OutputPath(), "debug_aec");
henrikg@webrtc.org863b5362013-12-06 16:05:17 +00001872 EXPECT_EQ(apm_->kNullPointerError,
ivocd66b44d2016-01-15 03:06:36 -08001873 apm_->StartDebugRecording(static_cast<const char*>(NULL), -1));
andrew@webrtc.org7bf26462011-12-03 00:03:31 +00001874
1875#ifdef WEBRTC_AUDIOPROC_DEBUG_DUMP
1876 // Stopping without having started should be OK.
1877 EXPECT_EQ(apm_->kNoError, apm_->StopDebugRecording());
1878
ivocd66b44d2016-01-15 03:06:36 -08001879 EXPECT_EQ(apm_->kNoError, apm_->StartDebugRecording(filename.c_str(), -1));
andrew@webrtc.org7bf26462011-12-03 00:03:31 +00001880 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
aluebsb0319552016-03-17 20:39:53 -07001881 EXPECT_EQ(apm_->kNoError, apm_->ProcessReverseStream(revframe_));
andrew@webrtc.org7bf26462011-12-03 00:03:31 +00001882 EXPECT_EQ(apm_->kNoError, apm_->StopDebugRecording());
1883
1884 // Verify the file has been written.
andrew@webrtc.orgf5d8c3b2012-01-24 21:35:39 +00001885 FILE* fid = fopen(filename.c_str(), "r");
1886 ASSERT_TRUE(fid != NULL);
1887
andrew@webrtc.org7bf26462011-12-03 00:03:31 +00001888 // Clean it up.
andrew@webrtc.orgf5d8c3b2012-01-24 21:35:39 +00001889 ASSERT_EQ(0, fclose(fid));
andrew@webrtc.org7bf26462011-12-03 00:03:31 +00001890 ASSERT_EQ(0, remove(filename.c_str()));
1891#else
1892 EXPECT_EQ(apm_->kUnsupportedFunctionError,
ivocd66b44d2016-01-15 03:06:36 -08001893 apm_->StartDebugRecording(filename.c_str(), -1));
andrew@webrtc.org7bf26462011-12-03 00:03:31 +00001894 EXPECT_EQ(apm_->kUnsupportedFunctionError, apm_->StopDebugRecording());
1895
1896 // Verify the file has NOT been written.
1897 ASSERT_TRUE(fopen(filename.c_str(), "r") == NULL);
1898#endif // WEBRTC_AUDIOPROC_DEBUG_DUMP
1899}
1900
henrikg@webrtc.org863b5362013-12-06 16:05:17 +00001901// TODO(andrew): expand test to verify output.
pbosc7a65692016-05-06 12:50:04 -07001902TEST_F(ApmTest, DebugDumpFromFileHandle) {
henrikg@webrtc.org863b5362013-12-06 16:05:17 +00001903 FILE* fid = NULL;
ivocd66b44d2016-01-15 03:06:36 -08001904 EXPECT_EQ(apm_->kNullPointerError, apm_->StartDebugRecording(fid, -1));
pbos@webrtc.orga525c982015-01-12 17:31:18 +00001905 const std::string filename =
1906 test::TempFilename(test::OutputPath(), "debug_aec");
henrikg@webrtc.org863b5362013-12-06 16:05:17 +00001907 fid = fopen(filename.c_str(), "w");
1908 ASSERT_TRUE(fid);
1909
1910#ifdef WEBRTC_AUDIOPROC_DEBUG_DUMP
1911 // Stopping without having started should be OK.
1912 EXPECT_EQ(apm_->kNoError, apm_->StopDebugRecording());
1913
ivocd66b44d2016-01-15 03:06:36 -08001914 EXPECT_EQ(apm_->kNoError, apm_->StartDebugRecording(fid, -1));
aluebsb0319552016-03-17 20:39:53 -07001915 EXPECT_EQ(apm_->kNoError, apm_->ProcessReverseStream(revframe_));
henrikg@webrtc.org863b5362013-12-06 16:05:17 +00001916 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
1917 EXPECT_EQ(apm_->kNoError, apm_->StopDebugRecording());
1918
1919 // Verify the file has been written.
1920 fid = fopen(filename.c_str(), "r");
1921 ASSERT_TRUE(fid != NULL);
1922
1923 // Clean it up.
1924 ASSERT_EQ(0, fclose(fid));
1925 ASSERT_EQ(0, remove(filename.c_str()));
1926#else
1927 EXPECT_EQ(apm_->kUnsupportedFunctionError,
ivocd66b44d2016-01-15 03:06:36 -08001928 apm_->StartDebugRecording(fid, -1));
henrikg@webrtc.org863b5362013-12-06 16:05:17 +00001929 EXPECT_EQ(apm_->kUnsupportedFunctionError, apm_->StopDebugRecording());
1930
1931 ASSERT_EQ(0, fclose(fid));
1932#endif // WEBRTC_AUDIOPROC_DEBUG_DUMP
1933}
1934
andrew@webrtc.org8328e7c2014-10-31 04:58:14 +00001935TEST_F(ApmTest, FloatAndIntInterfacesGiveSimilarResults) {
andrew@webrtc.org17e40642014-03-04 20:58:13 +00001936 audioproc::OutputData ref_data;
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00001937 OpenFileAndReadMessage(ref_filename_, &ref_data);
andrew@webrtc.org17e40642014-03-04 20:58:13 +00001938
1939 Config config;
1940 config.Set<ExperimentalAgc>(new ExperimentalAgc(false));
kwiberg62eaacf2016-02-17 06:39:05 -08001941 std::unique_ptr<AudioProcessing> fapm(AudioProcessing::Create(config));
andrew@webrtc.org17e40642014-03-04 20:58:13 +00001942 EnableAllComponents();
1943 EnableAllAPComponents(fapm.get());
1944 for (int i = 0; i < ref_data.test_size(); i++) {
1945 printf("Running test %d of %d...\n", i + 1, ref_data.test_size());
1946
1947 audioproc::Test* test = ref_data.mutable_test(i);
1948 // TODO(ajm): Restore downmixing test cases.
1949 if (test->num_input_channels() != test->num_output_channels())
1950 continue;
1951
Peter Kasting69558702016-01-12 16:26:35 -08001952 const size_t num_render_channels =
1953 static_cast<size_t>(test->num_reverse_channels());
1954 const size_t num_input_channels =
1955 static_cast<size_t>(test->num_input_channels());
1956 const size_t num_output_channels =
1957 static_cast<size_t>(test->num_output_channels());
pkasting25702cb2016-01-08 13:50:27 -08001958 const size_t samples_per_channel = static_cast<size_t>(
1959 test->sample_rate() * AudioProcessing::kChunkSizeMs / 1000);
andrew@webrtc.org17e40642014-03-04 20:58:13 +00001960
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00001961 Init(test->sample_rate(), test->sample_rate(), test->sample_rate(),
1962 num_input_channels, num_output_channels, num_render_channels, true);
andrew@webrtc.org17e40642014-03-04 20:58:13 +00001963 Init(fapm.get());
1964
1965 ChannelBuffer<int16_t> output_cb(samples_per_channel, num_input_channels);
andrew@webrtc.org8328e7c2014-10-31 04:58:14 +00001966 ChannelBuffer<int16_t> output_int16(samples_per_channel,
1967 num_input_channels);
andrew@webrtc.org17e40642014-03-04 20:58:13 +00001968
1969 int analog_level = 127;
aluebs776593b2016-03-15 14:04:58 -07001970 size_t num_bad_chunks = 0;
andrew@webrtc.org17e40642014-03-04 20:58:13 +00001971 while (ReadFrame(far_file_, revframe_, revfloat_cb_.get()) &&
1972 ReadFrame(near_file_, frame_, float_cb_.get())) {
1973 frame_->vad_activity_ = AudioFrame::kVadUnknown;
1974
aluebsb0319552016-03-17 20:39:53 -07001975 EXPECT_NOERR(apm_->ProcessReverseStream(revframe_));
andrew@webrtc.org17e40642014-03-04 20:58:13 +00001976 EXPECT_NOERR(fapm->AnalyzeReverseStream(
1977 revfloat_cb_->channels(),
1978 samples_per_channel,
1979 test->sample_rate(),
1980 LayoutFromChannels(num_render_channels)));
1981
1982 EXPECT_NOERR(apm_->set_stream_delay_ms(0));
1983 EXPECT_NOERR(fapm->set_stream_delay_ms(0));
1984 apm_->echo_cancellation()->set_stream_drift_samples(0);
1985 fapm->echo_cancellation()->set_stream_drift_samples(0);
1986 EXPECT_NOERR(apm_->gain_control()->set_stream_analog_level(analog_level));
1987 EXPECT_NOERR(fapm->gain_control()->set_stream_analog_level(analog_level));
1988
1989 EXPECT_NOERR(apm_->ProcessStream(frame_));
andrew@webrtc.org8328e7c2014-10-31 04:58:14 +00001990 Deinterleave(frame_->data_, samples_per_channel, num_output_channels,
1991 output_int16.channels());
1992
andrew@webrtc.org17e40642014-03-04 20:58:13 +00001993 EXPECT_NOERR(fapm->ProcessStream(
1994 float_cb_->channels(),
1995 samples_per_channel,
1996 test->sample_rate(),
1997 LayoutFromChannels(num_input_channels),
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00001998 test->sample_rate(),
1999 LayoutFromChannels(num_output_channels),
2000 float_cb_->channels()));
Peter Kasting69558702016-01-12 16:26:35 -08002001 for (size_t j = 0; j < num_output_channels; ++j) {
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +00002002 FloatToS16(float_cb_->channels()[j],
2003 samples_per_channel,
2004 output_cb.channels()[j]);
andrew@webrtc.org8328e7c2014-10-31 04:58:14 +00002005 float variance = 0;
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +00002006 float snr = ComputeSNR(output_int16.channels()[j],
2007 output_cb.channels()[j],
andrew@webrtc.org8328e7c2014-10-31 04:58:14 +00002008 samples_per_channel, &variance);
aluebs776593b2016-03-15 14:04:58 -07002009
andrew@webrtc.org8328e7c2014-10-31 04:58:14 +00002010 const float kVarianceThreshold = 20;
2011 const float kSNRThreshold = 20;
aluebs776593b2016-03-15 14:04:58 -07002012
andrew@webrtc.org8328e7c2014-10-31 04:58:14 +00002013 // Skip frames with low energy.
aluebs776593b2016-03-15 14:04:58 -07002014 if (sqrt(variance) > kVarianceThreshold && snr < kSNRThreshold) {
2015 ++num_bad_chunks;
andrew@webrtc.org8328e7c2014-10-31 04:58:14 +00002016 }
2017 }
andrew@webrtc.org17e40642014-03-04 20:58:13 +00002018
2019 analog_level = fapm->gain_control()->stream_analog_level();
2020 EXPECT_EQ(apm_->gain_control()->stream_analog_level(),
2021 fapm->gain_control()->stream_analog_level());
2022 EXPECT_EQ(apm_->echo_cancellation()->stream_has_echo(),
2023 fapm->echo_cancellation()->stream_has_echo());
andrew@webrtc.org8328e7c2014-10-31 04:58:14 +00002024 EXPECT_NEAR(apm_->noise_suppression()->speech_probability(),
2025 fapm->noise_suppression()->speech_probability(),
Alejandro Luebs47748742015-05-22 12:00:21 -07002026 0.01);
andrew@webrtc.org17e40642014-03-04 20:58:13 +00002027
2028 // Reset in case of downmixing.
Peter Kasting69558702016-01-12 16:26:35 -08002029 frame_->num_channels_ = static_cast<size_t>(test->num_input_channels());
andrew@webrtc.org17e40642014-03-04 20:58:13 +00002030 }
aluebs776593b2016-03-15 14:04:58 -07002031
2032#if defined(WEBRTC_AUDIOPROC_FLOAT_PROFILE)
2033 const size_t kMaxNumBadChunks = 0;
2034#elif defined(WEBRTC_AUDIOPROC_FIXED_PROFILE)
2035 // There are a few chunks in the fixed-point profile that give low SNR.
2036 // Listening confirmed the difference is acceptable.
2037 const size_t kMaxNumBadChunks = 60;
2038#endif
2039 EXPECT_LE(num_bad_chunks, kMaxNumBadChunks);
2040
andrew@webrtc.org17e40642014-03-04 20:58:13 +00002041 rewind(far_file_);
2042 rewind(near_file_);
2043 }
2044}
2045
andrew@webrtc.org75f19482012-02-09 17:16:18 +00002046// TODO(andrew): Add a test to process a few frames with different combinations
2047// of enabled components.
2048
bjornv@webrtc.orgdc0b37d2014-09-23 05:03:44 +00002049TEST_F(ApmTest, Process) {
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002050 GOOGLE_PROTOBUF_VERIFY_VERSION;
andrew@webrtc.org27c69802014-02-18 20:24:56 +00002051 audioproc::OutputData ref_data;
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002052
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +00002053 if (!write_ref_data) {
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00002054 OpenFileAndReadMessage(ref_filename_, &ref_data);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002055 } else {
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +00002056 // Write the desired tests to the protobuf reference file.
pkasting25702cb2016-01-08 13:50:27 -08002057 for (size_t i = 0; i < arraysize(kChannels); i++) {
2058 for (size_t j = 0; j < arraysize(kChannels); j++) {
2059 for (size_t l = 0; l < arraysize(kProcessSampleRates); l++) {
andrew@webrtc.org27c69802014-02-18 20:24:56 +00002060 audioproc::Test* test = ref_data.add_test();
andrew@webrtc.org60730cf2014-01-07 17:45:09 +00002061 test->set_num_reverse_channels(kChannels[i]);
2062 test->set_num_input_channels(kChannels[j]);
2063 test->set_num_output_channels(kChannels[j]);
2064 test->set_sample_rate(kProcessSampleRates[l]);
aluebs@webrtc.orgf17ee9c2015-01-29 00:03:53 +00002065 test->set_use_aec_extended_filter(false);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002066 }
2067 }
2068 }
aluebs@webrtc.orgf17ee9c2015-01-29 00:03:53 +00002069#if defined(WEBRTC_AUDIOPROC_FLOAT_PROFILE)
2070 // To test the extended filter mode.
2071 audioproc::Test* test = ref_data.add_test();
2072 test->set_num_reverse_channels(2);
2073 test->set_num_input_channels(2);
2074 test->set_num_output_channels(2);
2075 test->set_sample_rate(AudioProcessing::kSampleRate32kHz);
2076 test->set_use_aec_extended_filter(true);
2077#endif
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002078 }
2079
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +00002080 for (int i = 0; i < ref_data.test_size(); i++) {
2081 printf("Running test %d of %d...\n", i + 1, ref_data.test_size());
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002082
andrew@webrtc.org27c69802014-02-18 20:24:56 +00002083 audioproc::Test* test = ref_data.mutable_test(i);
andrew@webrtc.org60730cf2014-01-07 17:45:09 +00002084 // TODO(ajm): We no longer allow different input and output channels. Skip
2085 // these tests for now, but they should be removed from the set.
2086 if (test->num_input_channels() != test->num_output_channels())
2087 continue;
2088
aluebs@webrtc.orgf17ee9c2015-01-29 00:03:53 +00002089 Config config;
2090 config.Set<ExperimentalAgc>(new ExperimentalAgc(false));
Henrik Lundin441f6342015-06-09 16:03:13 +02002091 config.Set<ExtendedFilter>(
2092 new ExtendedFilter(test->use_aec_extended_filter()));
aluebs@webrtc.orgf17ee9c2015-01-29 00:03:53 +00002093 apm_.reset(AudioProcessing::Create(config));
2094
2095 EnableAllComponents();
2096
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002097 Init(test->sample_rate(),
2098 test->sample_rate(),
2099 test->sample_rate(),
Peter Kasting69558702016-01-12 16:26:35 -08002100 static_cast<size_t>(test->num_input_channels()),
2101 static_cast<size_t>(test->num_output_channels()),
2102 static_cast<size_t>(test->num_reverse_channels()),
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002103 true);
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +00002104
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002105 int frame_count = 0;
2106 int has_echo_count = 0;
2107 int has_voice_count = 0;
2108 int is_saturated_count = 0;
2109 int analog_level = 127;
2110 int analog_level_average = 0;
2111 int max_output_average = 0;
bjornv@webrtc.org08329f42012-07-12 21:00:43 +00002112 float ns_speech_prob_average = 0.0f;
minyue58530ed2016-05-24 05:50:12 -07002113#if defined(WEBRTC_AUDIOPROC_FLOAT_PROFILE)
2114 int stats_index = 0;
2115#endif
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002116
andrew@webrtc.org17e40642014-03-04 20:58:13 +00002117 while (ReadFrame(far_file_, revframe_) && ReadFrame(near_file_, frame_)) {
aluebsb0319552016-03-17 20:39:53 -07002118 EXPECT_EQ(apm_->kNoError, apm_->ProcessReverseStream(revframe_));
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002119
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +00002120 frame_->vad_activity_ = AudioFrame::kVadUnknown;
2121
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002122 EXPECT_EQ(apm_->kNoError, apm_->set_stream_delay_ms(0));
andrew@webrtc.org6be1e932013-03-01 18:47:28 +00002123 apm_->echo_cancellation()->set_stream_drift_samples(0);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002124 EXPECT_EQ(apm_->kNoError,
2125 apm_->gain_control()->set_stream_analog_level(analog_level));
2126
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002127 EXPECT_EQ(apm_->kNoError, apm_->ProcessStream(frame_));
andrew@webrtc.org17e40642014-03-04 20:58:13 +00002128
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +00002129 // Ensure the frame was downmixed properly.
Peter Kasting69558702016-01-12 16:26:35 -08002130 EXPECT_EQ(static_cast<size_t>(test->num_output_channels()),
2131 frame_->num_channels_);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002132
2133 max_output_average += MaxAudioFrame(*frame_);
2134
2135 if (apm_->echo_cancellation()->stream_has_echo()) {
2136 has_echo_count++;
2137 }
2138
2139 analog_level = apm_->gain_control()->stream_analog_level();
2140 analog_level_average += analog_level;
2141 if (apm_->gain_control()->stream_is_saturated()) {
2142 is_saturated_count++;
2143 }
2144 if (apm_->voice_detection()->stream_has_voice()) {
2145 has_voice_count++;
andrew@webrtc.org63a50982012-05-02 23:56:37 +00002146 EXPECT_EQ(AudioFrame::kVadActive, frame_->vad_activity_);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002147 } else {
andrew@webrtc.org63a50982012-05-02 23:56:37 +00002148 EXPECT_EQ(AudioFrame::kVadPassive, frame_->vad_activity_);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002149 }
2150
bjornv@webrtc.org08329f42012-07-12 21:00:43 +00002151 ns_speech_prob_average += apm_->noise_suppression()->speech_probability();
2152
andrew@webrtc.org07bf9a02012-05-05 00:32:00 +00002153 size_t frame_size = frame_->samples_per_channel_ * frame_->num_channels_;
andrew@webrtc.org63a50982012-05-02 23:56:37 +00002154 size_t write_count = fwrite(frame_->data_,
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +00002155 sizeof(int16_t),
2156 frame_size,
2157 out_file_);
2158 ASSERT_EQ(frame_size, write_count);
2159
2160 // Reset in case of downmixing.
Peter Kasting69558702016-01-12 16:26:35 -08002161 frame_->num_channels_ = static_cast<size_t>(test->num_input_channels());
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002162 frame_count++;
minyue58530ed2016-05-24 05:50:12 -07002163
2164#if defined(WEBRTC_AUDIOPROC_FLOAT_PROFILE)
2165 const int kStatsAggregationFrameNum = 100; // 1 second.
2166 if (frame_count % kStatsAggregationFrameNum == 0) {
2167 // Get echo metrics.
2168 EchoCancellation::Metrics echo_metrics;
2169 EXPECT_EQ(apm_->kNoError,
2170 apm_->echo_cancellation()->GetMetrics(&echo_metrics));
2171
2172 // Get delay metrics.
2173 int median = 0;
2174 int std = 0;
2175 float fraction_poor_delays = 0;
2176 EXPECT_EQ(apm_->kNoError,
2177 apm_->echo_cancellation()->GetDelayMetrics(
2178 &median, &std, &fraction_poor_delays));
2179
2180 // Get RMS.
2181 int rms_level = apm_->level_estimator()->RMS();
2182 EXPECT_LE(0, rms_level);
2183 EXPECT_GE(127, rms_level);
2184
2185 if (!write_ref_data) {
2186 const audioproc::Test::EchoMetrics& reference =
2187 test->echo_metrics(stats_index);
2188 TestStats(echo_metrics.residual_echo_return_loss,
2189 reference.residual_echo_return_loss());
2190 TestStats(echo_metrics.echo_return_loss,
2191 reference.echo_return_loss());
2192 TestStats(echo_metrics.echo_return_loss_enhancement,
2193 reference.echo_return_loss_enhancement());
2194 TestStats(echo_metrics.a_nlp,
2195 reference.a_nlp());
2196 EXPECT_EQ(echo_metrics.divergent_filter_fraction,
2197 reference.divergent_filter_fraction());
2198
2199 const audioproc::Test::DelayMetrics& reference_delay =
2200 test->delay_metrics(stats_index);
2201 EXPECT_EQ(reference_delay.median(), median);
2202 EXPECT_EQ(reference_delay.std(), std);
2203 EXPECT_EQ(reference_delay.fraction_poor_delays(),
2204 fraction_poor_delays);
2205
2206 EXPECT_EQ(test->rms_level(stats_index), rms_level);
2207
2208 ++stats_index;
2209 } else {
2210 audioproc::Test::EchoMetrics* message =
2211 test->add_echo_metrics();
2212 WriteStatsMessage(echo_metrics.residual_echo_return_loss,
2213 message->mutable_residual_echo_return_loss());
2214 WriteStatsMessage(echo_metrics.echo_return_loss,
2215 message->mutable_echo_return_loss());
2216 WriteStatsMessage(echo_metrics.echo_return_loss_enhancement,
2217 message->mutable_echo_return_loss_enhancement());
2218 WriteStatsMessage(echo_metrics.a_nlp,
2219 message->mutable_a_nlp());
2220 message->set_divergent_filter_fraction(
2221 echo_metrics.divergent_filter_fraction);
2222
2223 audioproc::Test::DelayMetrics* message_delay =
2224 test->add_delay_metrics();
2225 message_delay->set_median(median);
2226 message_delay->set_std(std);
2227 message_delay->set_fraction_poor_delays(fraction_poor_delays);
2228
2229 test->add_rms_level(rms_level);
2230 }
2231 }
2232#endif // defined(WEBRTC_AUDIOPROC_FLOAT_PROFILE).
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002233 }
2234 max_output_average /= frame_count;
2235 analog_level_average /= frame_count;
bjornv@webrtc.org08329f42012-07-12 21:00:43 +00002236 ns_speech_prob_average /= frame_count;
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002237
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +00002238 if (!write_ref_data) {
bjornv@webrtc.org8dd60cc2014-09-11 08:36:35 +00002239 const int kIntNear = 1;
bjornv@webrtc.orgdc0b37d2014-09-23 05:03:44 +00002240 // When running the test on a N7 we get a {2, 6} difference of
2241 // |has_voice_count| and |max_output_average| is up to 18 higher.
2242 // All numbers being consistently higher on N7 compare to ref_data.
2243 // TODO(bjornv): If we start getting more of these offsets on Android we
2244 // should consider a different approach. Either using one slack for all,
2245 // or generate a separate android reference.
2246#if defined(WEBRTC_ANDROID)
2247 const int kHasVoiceCountOffset = 3;
Alejandro Luebs2a5609d2016-04-05 18:16:54 -07002248 const int kHasVoiceCountNear = 4;
bjornv@webrtc.orgdc0b37d2014-09-23 05:03:44 +00002249 const int kMaxOutputAverageOffset = 9;
2250 const int kMaxOutputAverageNear = 9;
2251#else
2252 const int kHasVoiceCountOffset = 0;
2253 const int kHasVoiceCountNear = kIntNear;
2254 const int kMaxOutputAverageOffset = 0;
2255 const int kMaxOutputAverageNear = kIntNear;
2256#endif
bjornv@webrtc.org8dd60cc2014-09-11 08:36:35 +00002257 EXPECT_NEAR(test->has_echo_count(), has_echo_count, kIntNear);
bjornv@webrtc.orgdc0b37d2014-09-23 05:03:44 +00002258 EXPECT_NEAR(test->has_voice_count(),
2259 has_voice_count - kHasVoiceCountOffset,
2260 kHasVoiceCountNear);
bjornv@webrtc.org8dd60cc2014-09-11 08:36:35 +00002261 EXPECT_NEAR(test->is_saturated_count(), is_saturated_count, kIntNear);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002262
bjornv@webrtc.org8dd60cc2014-09-11 08:36:35 +00002263 EXPECT_NEAR(test->analog_level_average(), analog_level_average, kIntNear);
bjornv@webrtc.orgdc0b37d2014-09-23 05:03:44 +00002264 EXPECT_NEAR(test->max_output_average(),
2265 max_output_average - kMaxOutputAverageOffset,
2266 kMaxOutputAverageNear);
andrew@webrtc.org293d22b2012-01-30 22:04:26 +00002267#if defined(WEBRTC_AUDIOPROC_FLOAT_PROFILE)
bjornv@webrtc.org8dd60cc2014-09-11 08:36:35 +00002268 const double kFloatNear = 0.0005;
bjornv@webrtc.org8dd60cc2014-09-11 08:36:35 +00002269 EXPECT_NEAR(test->ns_speech_probability_average(),
2270 ns_speech_prob_average,
2271 kFloatNear);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002272#endif
2273 } else {
2274 test->set_has_echo_count(has_echo_count);
2275 test->set_has_voice_count(has_voice_count);
2276 test->set_is_saturated_count(is_saturated_count);
2277
2278 test->set_analog_level_average(analog_level_average);
2279 test->set_max_output_average(max_output_average);
2280
andrew@webrtc.org293d22b2012-01-30 22:04:26 +00002281#if defined(WEBRTC_AUDIOPROC_FLOAT_PROFILE)
bjornv@webrtc.org08329f42012-07-12 21:00:43 +00002282 EXPECT_LE(0.0f, ns_speech_prob_average);
2283 EXPECT_GE(1.0f, ns_speech_prob_average);
2284 test->set_ns_speech_probability_average(ns_speech_prob_average);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002285#endif
2286 }
2287
2288 rewind(far_file_);
2289 rewind(near_file_);
2290 }
2291
andrew@webrtc.orgdaacee82012-02-07 00:01:04 +00002292 if (write_ref_data) {
andrew@webrtc.orga8b97372014-03-10 22:26:12 +00002293 OpenFileAndWriteMessage(ref_filename_, ref_data);
andrew@webrtc.org755b04a2011-11-15 16:57:56 +00002294 }
2295}
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002296
andrew@webrtc.org103657b2014-04-24 18:28:56 +00002297TEST_F(ApmTest, NoErrorsWithKeyboardChannel) {
2298 struct ChannelFormat {
2299 AudioProcessing::ChannelLayout in_layout;
2300 AudioProcessing::ChannelLayout out_layout;
2301 };
2302 ChannelFormat cf[] = {
2303 {AudioProcessing::kMonoAndKeyboard, AudioProcessing::kMono},
2304 {AudioProcessing::kStereoAndKeyboard, AudioProcessing::kMono},
2305 {AudioProcessing::kStereoAndKeyboard, AudioProcessing::kStereo},
2306 };
andrew@webrtc.org103657b2014-04-24 18:28:56 +00002307
kwiberg62eaacf2016-02-17 06:39:05 -08002308 std::unique_ptr<AudioProcessing> ap(AudioProcessing::Create());
andrew@webrtc.org103657b2014-04-24 18:28:56 +00002309 // Enable one component just to ensure some processing takes place.
2310 ap->noise_suppression()->Enable(true);
pkasting25702cb2016-01-08 13:50:27 -08002311 for (size_t i = 0; i < arraysize(cf); ++i) {
andrew@webrtc.org103657b2014-04-24 18:28:56 +00002312 const int in_rate = 44100;
2313 const int out_rate = 48000;
2314 ChannelBuffer<float> in_cb(SamplesFromRate(in_rate),
2315 TotalChannelsFromLayout(cf[i].in_layout));
2316 ChannelBuffer<float> out_cb(SamplesFromRate(out_rate),
2317 ChannelsFromLayout(cf[i].out_layout));
2318
2319 // Run over a few chunks.
2320 for (int j = 0; j < 10; ++j) {
2321 EXPECT_NOERR(ap->ProcessStream(
2322 in_cb.channels(),
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +00002323 in_cb.num_frames(),
andrew@webrtc.org103657b2014-04-24 18:28:56 +00002324 in_rate,
2325 cf[i].in_layout,
2326 out_rate,
2327 cf[i].out_layout,
2328 out_cb.channels()));
2329 }
2330 }
2331}
2332
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002333// Compares the reference and test arrays over a region around the expected
2334// delay. Finds the highest SNR in that region and adds the variance and squared
2335// error results to the supplied accumulators.
2336void UpdateBestSNR(const float* ref,
2337 const float* test,
pkasting25702cb2016-01-08 13:50:27 -08002338 size_t length,
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002339 int expected_delay,
2340 double* variance_acc,
2341 double* sq_error_acc) {
2342 double best_snr = std::numeric_limits<double>::min();
2343 double best_variance = 0;
2344 double best_sq_error = 0;
2345 // Search over a region of eight samples around the expected delay.
2346 for (int delay = std::max(expected_delay - 4, 0); delay <= expected_delay + 4;
2347 ++delay) {
2348 double sq_error = 0;
2349 double variance = 0;
pkasting25702cb2016-01-08 13:50:27 -08002350 for (size_t i = 0; i < length - delay; ++i) {
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002351 double error = test[i + delay] - ref[i];
2352 sq_error += error * error;
2353 variance += ref[i] * ref[i];
2354 }
2355
2356 if (sq_error == 0) {
2357 *variance_acc += variance;
2358 return;
2359 }
2360 double snr = variance / sq_error;
2361 if (snr > best_snr) {
2362 best_snr = snr;
2363 best_variance = variance;
2364 best_sq_error = sq_error;
2365 }
2366 }
2367
2368 *variance_acc += best_variance;
2369 *sq_error_acc += best_sq_error;
2370}
2371
2372// Used to test a multitude of sample rate and channel combinations. It works
2373// by first producing a set of reference files (in SetUpTestCase) that are
2374// assumed to be correct, as the used parameters are verified by other tests
2375// in this collection. Primarily the reference files are all produced at
2376// "native" rates which do not involve any resampling.
2377
2378// Each test pass produces an output file with a particular format. The output
2379// is matched against the reference file closest to its internal processing
2380// format. If necessary the output is resampled back to its process format.
2381// Due to the resampling distortion, we don't expect identical results, but
2382// enforce SNR thresholds which vary depending on the format. 0 is a special
2383// case SNR which corresponds to inf, or zero error.
ekmeyerson60d9b332015-08-14 10:35:55 -07002384typedef std::tr1::tuple<int, int, int, int, double, double>
2385 AudioProcessingTestData;
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002386class AudioProcessingTest
2387 : public testing::TestWithParam<AudioProcessingTestData> {
2388 public:
2389 AudioProcessingTest()
2390 : input_rate_(std::tr1::get<0>(GetParam())),
2391 output_rate_(std::tr1::get<1>(GetParam())),
ekmeyerson60d9b332015-08-14 10:35:55 -07002392 reverse_input_rate_(std::tr1::get<2>(GetParam())),
2393 reverse_output_rate_(std::tr1::get<3>(GetParam())),
2394 expected_snr_(std::tr1::get<4>(GetParam())),
2395 expected_reverse_snr_(std::tr1::get<5>(GetParam())) {}
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002396
2397 virtual ~AudioProcessingTest() {}
2398
2399 static void SetUpTestCase() {
2400 // Create all needed output reference files.
Alejandro Luebs47748742015-05-22 12:00:21 -07002401 const int kNativeRates[] = {8000, 16000, 32000, 48000};
Peter Kasting69558702016-01-12 16:26:35 -08002402 const size_t kNumChannels[] = {1, 2};
pkasting25702cb2016-01-08 13:50:27 -08002403 for (size_t i = 0; i < arraysize(kNativeRates); ++i) {
2404 for (size_t j = 0; j < arraysize(kNumChannels); ++j) {
2405 for (size_t k = 0; k < arraysize(kNumChannels); ++k) {
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002406 // The reference files always have matching input and output channels.
ekmeyerson60d9b332015-08-14 10:35:55 -07002407 ProcessFormat(kNativeRates[i], kNativeRates[i], kNativeRates[i],
2408 kNativeRates[i], kNumChannels[j], kNumChannels[j],
2409 kNumChannels[k], kNumChannels[k], "ref");
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002410 }
2411 }
2412 }
2413 }
2414
pbos@webrtc.org200ac002015-02-03 14:14:01 +00002415 static void TearDownTestCase() {
2416 ClearTempFiles();
2417 }
ekmeyerson60d9b332015-08-14 10:35:55 -07002418
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002419 // Runs a process pass on files with the given parameters and dumps the output
ekmeyerson60d9b332015-08-14 10:35:55 -07002420 // to a file specified with |output_file_prefix|. Both forward and reverse
2421 // output streams are dumped.
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002422 static void ProcessFormat(int input_rate,
2423 int output_rate,
ekmeyerson60d9b332015-08-14 10:35:55 -07002424 int reverse_input_rate,
2425 int reverse_output_rate,
Peter Kasting69558702016-01-12 16:26:35 -08002426 size_t num_input_channels,
2427 size_t num_output_channels,
2428 size_t num_reverse_input_channels,
2429 size_t num_reverse_output_channels,
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002430 std::string output_file_prefix) {
andrew@webrtc.org8328e7c2014-10-31 04:58:14 +00002431 Config config;
2432 config.Set<ExperimentalAgc>(new ExperimentalAgc(false));
kwiberg62eaacf2016-02-17 06:39:05 -08002433 std::unique_ptr<AudioProcessing> ap(AudioProcessing::Create(config));
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002434 EnableAllAPComponents(ap.get());
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002435
ekmeyerson60d9b332015-08-14 10:35:55 -07002436 ProcessingConfig processing_config = {
2437 {{input_rate, num_input_channels},
2438 {output_rate, num_output_channels},
2439 {reverse_input_rate, num_reverse_input_channels},
2440 {reverse_output_rate, num_reverse_output_channels}}};
2441 ap->Initialize(processing_config);
2442
2443 FILE* far_file =
2444 fopen(ResourceFilePath("far", reverse_input_rate).c_str(), "rb");
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002445 FILE* near_file = fopen(ResourceFilePath("near", input_rate).c_str(), "rb");
ekmeyerson60d9b332015-08-14 10:35:55 -07002446 FILE* out_file =
2447 fopen(OutputFilePath(output_file_prefix, input_rate, output_rate,
2448 reverse_input_rate, reverse_output_rate,
2449 num_input_channels, num_output_channels,
2450 num_reverse_input_channels,
2451 num_reverse_output_channels, kForward).c_str(),
2452 "wb");
2453 FILE* rev_out_file =
2454 fopen(OutputFilePath(output_file_prefix, input_rate, output_rate,
2455 reverse_input_rate, reverse_output_rate,
2456 num_input_channels, num_output_channels,
2457 num_reverse_input_channels,
2458 num_reverse_output_channels, kReverse).c_str(),
2459 "wb");
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002460 ASSERT_TRUE(far_file != NULL);
2461 ASSERT_TRUE(near_file != NULL);
2462 ASSERT_TRUE(out_file != NULL);
ekmeyerson60d9b332015-08-14 10:35:55 -07002463 ASSERT_TRUE(rev_out_file != NULL);
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002464
2465 ChannelBuffer<float> fwd_cb(SamplesFromRate(input_rate),
2466 num_input_channels);
ekmeyerson60d9b332015-08-14 10:35:55 -07002467 ChannelBuffer<float> rev_cb(SamplesFromRate(reverse_input_rate),
2468 num_reverse_input_channels);
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002469 ChannelBuffer<float> out_cb(SamplesFromRate(output_rate),
2470 num_output_channels);
ekmeyerson60d9b332015-08-14 10:35:55 -07002471 ChannelBuffer<float> rev_out_cb(SamplesFromRate(reverse_output_rate),
2472 num_reverse_output_channels);
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002473
2474 // Temporary buffers.
2475 const int max_length =
ekmeyerson60d9b332015-08-14 10:35:55 -07002476 2 * std::max(std::max(out_cb.num_frames(), rev_out_cb.num_frames()),
2477 std::max(fwd_cb.num_frames(), rev_cb.num_frames()));
kwiberg62eaacf2016-02-17 06:39:05 -08002478 std::unique_ptr<float[]> float_data(new float[max_length]);
2479 std::unique_ptr<int16_t[]> int_data(new int16_t[max_length]);
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002480
2481 int analog_level = 127;
2482 while (ReadChunk(far_file, int_data.get(), float_data.get(), &rev_cb) &&
2483 ReadChunk(near_file, int_data.get(), float_data.get(), &fwd_cb)) {
ekmeyerson60d9b332015-08-14 10:35:55 -07002484 EXPECT_NOERR(ap->ProcessReverseStream(
2485 rev_cb.channels(), processing_config.reverse_input_stream(),
2486 processing_config.reverse_output_stream(), rev_out_cb.channels()));
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002487
2488 EXPECT_NOERR(ap->set_stream_delay_ms(0));
2489 ap->echo_cancellation()->set_stream_drift_samples(0);
2490 EXPECT_NOERR(ap->gain_control()->set_stream_analog_level(analog_level));
2491
2492 EXPECT_NOERR(ap->ProcessStream(
2493 fwd_cb.channels(),
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +00002494 fwd_cb.num_frames(),
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002495 input_rate,
2496 LayoutFromChannels(num_input_channels),
2497 output_rate,
2498 LayoutFromChannels(num_output_channels),
2499 out_cb.channels()));
2500
ekmeyerson60d9b332015-08-14 10:35:55 -07002501 // Dump forward output to file.
2502 Interleave(out_cb.channels(), out_cb.num_frames(), out_cb.num_channels(),
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002503 float_data.get());
pkasting25702cb2016-01-08 13:50:27 -08002504 size_t out_length = out_cb.num_channels() * out_cb.num_frames();
ekmeyerson60d9b332015-08-14 10:35:55 -07002505
pkasting25702cb2016-01-08 13:50:27 -08002506 ASSERT_EQ(out_length,
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002507 fwrite(float_data.get(), sizeof(float_data[0]),
aluebs@webrtc.orgd35a5c32015-02-10 22:52:15 +00002508 out_length, out_file));
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002509
ekmeyerson60d9b332015-08-14 10:35:55 -07002510 // Dump reverse output to file.
2511 Interleave(rev_out_cb.channels(), rev_out_cb.num_frames(),
2512 rev_out_cb.num_channels(), float_data.get());
pkasting25702cb2016-01-08 13:50:27 -08002513 size_t rev_out_length =
2514 rev_out_cb.num_channels() * rev_out_cb.num_frames();
ekmeyerson60d9b332015-08-14 10:35:55 -07002515
pkasting25702cb2016-01-08 13:50:27 -08002516 ASSERT_EQ(rev_out_length,
ekmeyerson60d9b332015-08-14 10:35:55 -07002517 fwrite(float_data.get(), sizeof(float_data[0]), rev_out_length,
2518 rev_out_file));
2519
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002520 analog_level = ap->gain_control()->stream_analog_level();
2521 }
2522 fclose(far_file);
2523 fclose(near_file);
2524 fclose(out_file);
ekmeyerson60d9b332015-08-14 10:35:55 -07002525 fclose(rev_out_file);
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002526 }
2527
2528 protected:
2529 int input_rate_;
2530 int output_rate_;
ekmeyerson60d9b332015-08-14 10:35:55 -07002531 int reverse_input_rate_;
2532 int reverse_output_rate_;
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002533 double expected_snr_;
ekmeyerson60d9b332015-08-14 10:35:55 -07002534 double expected_reverse_snr_;
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002535};
2536
bjornv@webrtc.org2812b592014-06-02 11:27:29 +00002537TEST_P(AudioProcessingTest, Formats) {
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002538 struct ChannelFormat {
2539 int num_input;
2540 int num_output;
ekmeyerson60d9b332015-08-14 10:35:55 -07002541 int num_reverse_input;
2542 int num_reverse_output;
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002543 };
2544 ChannelFormat cf[] = {
ekmeyerson60d9b332015-08-14 10:35:55 -07002545 {1, 1, 1, 1},
2546 {1, 1, 2, 1},
2547 {2, 1, 1, 1},
2548 {2, 1, 2, 1},
2549 {2, 2, 1, 1},
2550 {2, 2, 2, 2},
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002551 };
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002552
pkasting25702cb2016-01-08 13:50:27 -08002553 for (size_t i = 0; i < arraysize(cf); ++i) {
ekmeyerson60d9b332015-08-14 10:35:55 -07002554 ProcessFormat(input_rate_, output_rate_, reverse_input_rate_,
2555 reverse_output_rate_, cf[i].num_input, cf[i].num_output,
2556 cf[i].num_reverse_input, cf[i].num_reverse_output, "out");
Alejandro Luebs47748742015-05-22 12:00:21 -07002557
ekmeyerson60d9b332015-08-14 10:35:55 -07002558 // Verify output for both directions.
2559 std::vector<StreamDirection> stream_directions;
2560 stream_directions.push_back(kForward);
2561 stream_directions.push_back(kReverse);
2562 for (StreamDirection file_direction : stream_directions) {
2563 const int in_rate = file_direction ? reverse_input_rate_ : input_rate_;
2564 const int out_rate = file_direction ? reverse_output_rate_ : output_rate_;
2565 const int out_num =
2566 file_direction ? cf[i].num_reverse_output : cf[i].num_output;
2567 const double expected_snr =
2568 file_direction ? expected_reverse_snr_ : expected_snr_;
2569
2570 const int min_ref_rate = std::min(in_rate, out_rate);
2571 int ref_rate;
2572
2573 if (min_ref_rate > 32000) {
2574 ref_rate = 48000;
2575 } else if (min_ref_rate > 16000) {
2576 ref_rate = 32000;
2577 } else if (min_ref_rate > 8000) {
2578 ref_rate = 16000;
2579 } else {
2580 ref_rate = 8000;
2581 }
aluebs776593b2016-03-15 14:04:58 -07002582#ifdef WEBRTC_ARCH_ARM_FAMILY
perkjdfc28702016-03-09 16:23:23 -08002583 if (file_direction == kForward) {
aluebs776593b2016-03-15 14:04:58 -07002584 ref_rate = std::min(ref_rate, 32000);
perkjdfc28702016-03-09 16:23:23 -08002585 }
2586#endif
ekmeyerson60d9b332015-08-14 10:35:55 -07002587 FILE* out_file = fopen(
2588 OutputFilePath("out", input_rate_, output_rate_, reverse_input_rate_,
2589 reverse_output_rate_, cf[i].num_input,
2590 cf[i].num_output, cf[i].num_reverse_input,
2591 cf[i].num_reverse_output, file_direction).c_str(),
2592 "rb");
2593 // The reference files always have matching input and output channels.
2594 FILE* ref_file = fopen(
2595 OutputFilePath("ref", ref_rate, ref_rate, ref_rate, ref_rate,
2596 cf[i].num_output, cf[i].num_output,
2597 cf[i].num_reverse_output, cf[i].num_reverse_output,
2598 file_direction).c_str(),
2599 "rb");
2600 ASSERT_TRUE(out_file != NULL);
2601 ASSERT_TRUE(ref_file != NULL);
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002602
pkasting25702cb2016-01-08 13:50:27 -08002603 const size_t ref_length = SamplesFromRate(ref_rate) * out_num;
2604 const size_t out_length = SamplesFromRate(out_rate) * out_num;
ekmeyerson60d9b332015-08-14 10:35:55 -07002605 // Data from the reference file.
kwiberg62eaacf2016-02-17 06:39:05 -08002606 std::unique_ptr<float[]> ref_data(new float[ref_length]);
ekmeyerson60d9b332015-08-14 10:35:55 -07002607 // Data from the output file.
kwiberg62eaacf2016-02-17 06:39:05 -08002608 std::unique_ptr<float[]> out_data(new float[out_length]);
ekmeyerson60d9b332015-08-14 10:35:55 -07002609 // Data from the resampled output, in case the reference and output rates
2610 // don't match.
kwiberg62eaacf2016-02-17 06:39:05 -08002611 std::unique_ptr<float[]> cmp_data(new float[ref_length]);
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002612
ekmeyerson60d9b332015-08-14 10:35:55 -07002613 PushResampler<float> resampler;
2614 resampler.InitializeIfNeeded(out_rate, ref_rate, out_num);
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002615
ekmeyerson60d9b332015-08-14 10:35:55 -07002616 // Compute the resampling delay of the output relative to the reference,
2617 // to find the region over which we should search for the best SNR.
2618 float expected_delay_sec = 0;
2619 if (in_rate != ref_rate) {
2620 // Input resampling delay.
2621 expected_delay_sec +=
2622 PushSincResampler::AlgorithmicDelaySeconds(in_rate);
2623 }
2624 if (out_rate != ref_rate) {
2625 // Output resampling delay.
2626 expected_delay_sec +=
2627 PushSincResampler::AlgorithmicDelaySeconds(ref_rate);
2628 // Delay of converting the output back to its processing rate for
2629 // testing.
2630 expected_delay_sec +=
2631 PushSincResampler::AlgorithmicDelaySeconds(out_rate);
2632 }
2633 int expected_delay =
2634 floor(expected_delay_sec * ref_rate + 0.5f) * out_num;
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002635
ekmeyerson60d9b332015-08-14 10:35:55 -07002636 double variance = 0;
2637 double sq_error = 0;
2638 while (fread(out_data.get(), sizeof(out_data[0]), out_length, out_file) &&
2639 fread(ref_data.get(), sizeof(ref_data[0]), ref_length, ref_file)) {
2640 float* out_ptr = out_data.get();
2641 if (out_rate != ref_rate) {
2642 // Resample the output back to its internal processing rate if
2643 // necssary.
pkasting25702cb2016-01-08 13:50:27 -08002644 ASSERT_EQ(ref_length,
2645 static_cast<size_t>(resampler.Resample(
2646 out_ptr, out_length, cmp_data.get(), ref_length)));
ekmeyerson60d9b332015-08-14 10:35:55 -07002647 out_ptr = cmp_data.get();
2648 }
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002649
ekmeyerson60d9b332015-08-14 10:35:55 -07002650 // Update the |sq_error| and |variance| accumulators with the highest
2651 // SNR of reference vs output.
2652 UpdateBestSNR(ref_data.get(), out_ptr, ref_length, expected_delay,
2653 &variance, &sq_error);
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002654 }
2655
ekmeyerson60d9b332015-08-14 10:35:55 -07002656 std::cout << "(" << input_rate_ << ", " << output_rate_ << ", "
2657 << reverse_input_rate_ << ", " << reverse_output_rate_ << ", "
2658 << cf[i].num_input << ", " << cf[i].num_output << ", "
2659 << cf[i].num_reverse_input << ", " << cf[i].num_reverse_output
2660 << ", " << file_direction << "): ";
2661 if (sq_error > 0) {
2662 double snr = 10 * log10(variance / sq_error);
2663 EXPECT_GE(snr, expected_snr);
2664 EXPECT_NE(0, expected_snr);
2665 std::cout << "SNR=" << snr << " dB" << std::endl;
2666 } else {
aluebs776593b2016-03-15 14:04:58 -07002667 std::cout << "SNR=inf dB" << std::endl;
ekmeyerson60d9b332015-08-14 10:35:55 -07002668 }
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002669
ekmeyerson60d9b332015-08-14 10:35:55 -07002670 fclose(out_file);
2671 fclose(ref_file);
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002672 }
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002673 }
2674}
2675
2676#if defined(WEBRTC_AUDIOPROC_FLOAT_PROFILE)
2677INSTANTIATE_TEST_CASE_P(
ekmeyerson60d9b332015-08-14 10:35:55 -07002678 CommonFormats,
2679 AudioProcessingTest,
2680 testing::Values(std::tr1::make_tuple(48000, 48000, 48000, 48000, 0, 0),
peah0bf612b2016-04-06 02:47:46 -07002681 std::tr1::make_tuple(48000, 48000, 32000, 48000, 40, 30),
2682 std::tr1::make_tuple(48000, 48000, 16000, 48000, 40, 20),
ekmeyerson60d9b332015-08-14 10:35:55 -07002683 std::tr1::make_tuple(48000, 44100, 48000, 44100, 20, 20),
2684 std::tr1::make_tuple(48000, 44100, 32000, 44100, 20, 15),
2685 std::tr1::make_tuple(48000, 44100, 16000, 44100, 20, 15),
2686 std::tr1::make_tuple(48000, 32000, 48000, 32000, 30, 35),
2687 std::tr1::make_tuple(48000, 32000, 32000, 32000, 30, 0),
2688 std::tr1::make_tuple(48000, 32000, 16000, 32000, 30, 20),
2689 std::tr1::make_tuple(48000, 16000, 48000, 16000, 25, 20),
2690 std::tr1::make_tuple(48000, 16000, 32000, 16000, 25, 20),
2691 std::tr1::make_tuple(48000, 16000, 16000, 16000, 25, 0),
Alejandro Luebs47748742015-05-22 12:00:21 -07002692
ekmeyerson60d9b332015-08-14 10:35:55 -07002693 std::tr1::make_tuple(44100, 48000, 48000, 48000, 30, 0),
2694 std::tr1::make_tuple(44100, 48000, 32000, 48000, 30, 30),
2695 std::tr1::make_tuple(44100, 48000, 16000, 48000, 30, 20),
2696 std::tr1::make_tuple(44100, 44100, 48000, 44100, 20, 20),
2697 std::tr1::make_tuple(44100, 44100, 32000, 44100, 20, 15),
2698 std::tr1::make_tuple(44100, 44100, 16000, 44100, 20, 15),
2699 std::tr1::make_tuple(44100, 32000, 48000, 32000, 30, 35),
2700 std::tr1::make_tuple(44100, 32000, 32000, 32000, 30, 0),
2701 std::tr1::make_tuple(44100, 32000, 16000, 32000, 30, 20),
2702 std::tr1::make_tuple(44100, 16000, 48000, 16000, 25, 20),
2703 std::tr1::make_tuple(44100, 16000, 32000, 16000, 25, 20),
2704 std::tr1::make_tuple(44100, 16000, 16000, 16000, 25, 0),
Alejandro Luebs47748742015-05-22 12:00:21 -07002705
ekmeyerson60d9b332015-08-14 10:35:55 -07002706 std::tr1::make_tuple(32000, 48000, 48000, 48000, 30, 0),
2707 std::tr1::make_tuple(32000, 48000, 32000, 48000, 35, 30),
2708 std::tr1::make_tuple(32000, 48000, 16000, 48000, 30, 20),
2709 std::tr1::make_tuple(32000, 44100, 48000, 44100, 20, 20),
2710 std::tr1::make_tuple(32000, 44100, 32000, 44100, 20, 15),
2711 std::tr1::make_tuple(32000, 44100, 16000, 44100, 20, 15),
2712 std::tr1::make_tuple(32000, 32000, 48000, 32000, 40, 35),
2713 std::tr1::make_tuple(32000, 32000, 32000, 32000, 0, 0),
2714 std::tr1::make_tuple(32000, 32000, 16000, 32000, 40, 20),
2715 std::tr1::make_tuple(32000, 16000, 48000, 16000, 25, 20),
2716 std::tr1::make_tuple(32000, 16000, 32000, 16000, 25, 20),
2717 std::tr1::make_tuple(32000, 16000, 16000, 16000, 25, 0),
Alejandro Luebs47748742015-05-22 12:00:21 -07002718
ekmeyerson60d9b332015-08-14 10:35:55 -07002719 std::tr1::make_tuple(16000, 48000, 48000, 48000, 25, 0),
2720 std::tr1::make_tuple(16000, 48000, 32000, 48000, 25, 30),
2721 std::tr1::make_tuple(16000, 48000, 16000, 48000, 25, 20),
2722 std::tr1::make_tuple(16000, 44100, 48000, 44100, 15, 20),
2723 std::tr1::make_tuple(16000, 44100, 32000, 44100, 15, 15),
2724 std::tr1::make_tuple(16000, 44100, 16000, 44100, 15, 15),
2725 std::tr1::make_tuple(16000, 32000, 48000, 32000, 25, 35),
2726 std::tr1::make_tuple(16000, 32000, 32000, 32000, 25, 0),
2727 std::tr1::make_tuple(16000, 32000, 16000, 32000, 25, 20),
2728 std::tr1::make_tuple(16000, 16000, 48000, 16000, 40, 20),
aluebseb3603b2016-04-20 15:27:58 -07002729 std::tr1::make_tuple(16000, 16000, 32000, 16000, 40, 20),
ekmeyerson60d9b332015-08-14 10:35:55 -07002730 std::tr1::make_tuple(16000, 16000, 16000, 16000, 0, 0)));
Alejandro Luebs47748742015-05-22 12:00:21 -07002731
2732#elif defined(WEBRTC_AUDIOPROC_FIXED_PROFILE)
2733INSTANTIATE_TEST_CASE_P(
ekmeyerson60d9b332015-08-14 10:35:55 -07002734 CommonFormats,
2735 AudioProcessingTest,
perkjdfc28702016-03-09 16:23:23 -08002736 testing::Values(std::tr1::make_tuple(48000, 48000, 48000, 48000, 20, 0),
2737 std::tr1::make_tuple(48000, 48000, 32000, 48000, 20, 30),
2738 std::tr1::make_tuple(48000, 48000, 16000, 48000, 20, 20),
2739 std::tr1::make_tuple(48000, 44100, 48000, 44100, 15, 20),
2740 std::tr1::make_tuple(48000, 44100, 32000, 44100, 15, 15),
2741 std::tr1::make_tuple(48000, 44100, 16000, 44100, 15, 15),
ekmeyerson60d9b332015-08-14 10:35:55 -07002742 std::tr1::make_tuple(48000, 32000, 48000, 32000, 20, 35),
2743 std::tr1::make_tuple(48000, 32000, 32000, 32000, 20, 0),
2744 std::tr1::make_tuple(48000, 32000, 16000, 32000, 20, 20),
2745 std::tr1::make_tuple(48000, 16000, 48000, 16000, 20, 20),
2746 std::tr1::make_tuple(48000, 16000, 32000, 16000, 20, 20),
2747 std::tr1::make_tuple(48000, 16000, 16000, 16000, 20, 0),
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002748
aluebs776593b2016-03-15 14:04:58 -07002749 std::tr1::make_tuple(44100, 48000, 48000, 48000, 15, 0),
2750 std::tr1::make_tuple(44100, 48000, 32000, 48000, 15, 30),
2751 std::tr1::make_tuple(44100, 48000, 16000, 48000, 15, 20),
ekmeyerson60d9b332015-08-14 10:35:55 -07002752 std::tr1::make_tuple(44100, 44100, 48000, 44100, 15, 20),
2753 std::tr1::make_tuple(44100, 44100, 32000, 44100, 15, 15),
2754 std::tr1::make_tuple(44100, 44100, 16000, 44100, 15, 15),
2755 std::tr1::make_tuple(44100, 32000, 48000, 32000, 20, 35),
2756 std::tr1::make_tuple(44100, 32000, 32000, 32000, 20, 0),
2757 std::tr1::make_tuple(44100, 32000, 16000, 32000, 20, 20),
2758 std::tr1::make_tuple(44100, 16000, 48000, 16000, 20, 20),
2759 std::tr1::make_tuple(44100, 16000, 32000, 16000, 20, 20),
2760 std::tr1::make_tuple(44100, 16000, 16000, 16000, 20, 0),
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002761
aluebs776593b2016-03-15 14:04:58 -07002762 std::tr1::make_tuple(32000, 48000, 48000, 48000, 35, 0),
2763 std::tr1::make_tuple(32000, 48000, 32000, 48000, 65, 30),
2764 std::tr1::make_tuple(32000, 48000, 16000, 48000, 40, 20),
2765 std::tr1::make_tuple(32000, 44100, 48000, 44100, 20, 20),
2766 std::tr1::make_tuple(32000, 44100, 32000, 44100, 20, 15),
2767 std::tr1::make_tuple(32000, 44100, 16000, 44100, 20, 15),
2768 std::tr1::make_tuple(32000, 32000, 48000, 32000, 35, 35),
2769 std::tr1::make_tuple(32000, 32000, 32000, 32000, 0, 0),
2770 std::tr1::make_tuple(32000, 32000, 16000, 32000, 40, 20),
ekmeyerson60d9b332015-08-14 10:35:55 -07002771 std::tr1::make_tuple(32000, 16000, 48000, 16000, 20, 20),
2772 std::tr1::make_tuple(32000, 16000, 32000, 16000, 20, 20),
2773 std::tr1::make_tuple(32000, 16000, 16000, 16000, 20, 0),
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002774
ekmeyerson60d9b332015-08-14 10:35:55 -07002775 std::tr1::make_tuple(16000, 48000, 48000, 48000, 25, 0),
2776 std::tr1::make_tuple(16000, 48000, 32000, 48000, 25, 30),
2777 std::tr1::make_tuple(16000, 48000, 16000, 48000, 25, 20),
2778 std::tr1::make_tuple(16000, 44100, 48000, 44100, 15, 20),
2779 std::tr1::make_tuple(16000, 44100, 32000, 44100, 15, 15),
2780 std::tr1::make_tuple(16000, 44100, 16000, 44100, 15, 15),
2781 std::tr1::make_tuple(16000, 32000, 48000, 32000, 25, 35),
2782 std::tr1::make_tuple(16000, 32000, 32000, 32000, 25, 0),
2783 std::tr1::make_tuple(16000, 32000, 16000, 32000, 25, 20),
2784 std::tr1::make_tuple(16000, 16000, 48000, 16000, 35, 20),
aluebseb3603b2016-04-20 15:27:58 -07002785 std::tr1::make_tuple(16000, 16000, 32000, 16000, 35, 20),
ekmeyerson60d9b332015-08-14 10:35:55 -07002786 std::tr1::make_tuple(16000, 16000, 16000, 16000, 0, 0)));
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +00002787#endif
2788
niklase@google.com470e71d2011-07-07 08:21:25 +00002789} // namespace
peahc19f3122016-10-07 14:54:10 -07002790
2791TEST(ApmConfiguration, DefaultBehavior) {
2792 // Verify that the level controller is default off, it can be activated using
2793 // the config, and that the default initial level is maintained after the
2794 // config has been applied.
2795 std::unique_ptr<AudioProcessingImpl> apm(
2796 new AudioProcessingImpl(webrtc::Config()));
2797 AudioProcessing::Config config;
2798 EXPECT_FALSE(apm->config_.level_controller.enabled);
2799 // TODO(peah): Add test for the existence of the level controller object once
2800 // that is created only when that is specified in the config.
2801 // TODO(peah): Remove the testing for
2802 // apm->capture_nonlocked_.level_controller_enabled once the value in config_
2803 // is instead used to activate the level controller.
2804 EXPECT_FALSE(apm->capture_nonlocked_.level_controller_enabled);
2805 EXPECT_NEAR(kTargetLcPeakLeveldBFS,
2806 apm->config_.level_controller.initial_peak_level_dbfs,
2807 std::numeric_limits<float>::epsilon());
2808 config.level_controller.enabled = true;
2809 apm->ApplyConfig(config);
2810 EXPECT_TRUE(apm->config_.level_controller.enabled);
2811 // TODO(peah): Add test for the existence of the level controller object once
2812 // that is created only when the that is specified in the config.
2813 // TODO(peah): Remove the testing for
2814 // apm->capture_nonlocked_.level_controller_enabled once the value in config_
2815 // is instead used to activate the level controller.
2816 EXPECT_TRUE(apm->capture_nonlocked_.level_controller_enabled);
2817 EXPECT_NEAR(kTargetLcPeakLeveldBFS,
2818 apm->config_.level_controller.initial_peak_level_dbfs,
2819 std::numeric_limits<float>::epsilon());
2820}
2821
2822TEST(ApmConfiguration, ValidConfigBehavior) {
2823 // Verify that the initial level can be specified and is retained after the
2824 // config has been applied.
2825 std::unique_ptr<AudioProcessingImpl> apm(
2826 new AudioProcessingImpl(webrtc::Config()));
2827 AudioProcessing::Config config;
2828 config.level_controller.initial_peak_level_dbfs = -50.f;
2829 apm->ApplyConfig(config);
2830 EXPECT_FALSE(apm->config_.level_controller.enabled);
2831 // TODO(peah): Add test for the existence of the level controller object once
2832 // that is created only when the that is specified in the config.
2833 // TODO(peah): Remove the testing for
2834 // apm->capture_nonlocked_.level_controller_enabled once the value in config_
2835 // is instead used to activate the level controller.
2836 EXPECT_FALSE(apm->capture_nonlocked_.level_controller_enabled);
2837 EXPECT_NEAR(-50.f, apm->config_.level_controller.initial_peak_level_dbfs,
2838 std::numeric_limits<float>::epsilon());
2839}
2840
2841TEST(ApmConfiguration, InValidConfigBehavior) {
2842 // Verify that the config is properly reset when nonproper values are applied
2843 // for the initial level.
2844
2845 // Verify that the config is properly reset when the specified initial peak
2846 // level is too low.
2847 std::unique_ptr<AudioProcessingImpl> apm(
2848 new AudioProcessingImpl(webrtc::Config()));
2849 AudioProcessing::Config config;
2850 config.level_controller.enabled = true;
2851 config.level_controller.initial_peak_level_dbfs = -101.f;
2852 apm->ApplyConfig(config);
2853 EXPECT_FALSE(apm->config_.level_controller.enabled);
2854 // TODO(peah): Add test for the existence of the level controller object once
2855 // that is created only when the that is specified in the config.
2856 // TODO(peah): Remove the testing for
2857 // apm->capture_nonlocked_.level_controller_enabled once the value in config_
2858 // is instead used to activate the level controller.
2859 EXPECT_FALSE(apm->capture_nonlocked_.level_controller_enabled);
2860 EXPECT_NEAR(kTargetLcPeakLeveldBFS,
2861 apm->config_.level_controller.initial_peak_level_dbfs,
2862 std::numeric_limits<float>::epsilon());
2863
2864 // Verify that the config is properly reset when the specified initial peak
2865 // level is too high.
2866 apm.reset(new AudioProcessingImpl(webrtc::Config()));
2867 config = AudioProcessing::Config();
2868 config.level_controller.enabled = true;
2869 config.level_controller.initial_peak_level_dbfs = 1.f;
2870 apm->ApplyConfig(config);
2871 EXPECT_FALSE(apm->config_.level_controller.enabled);
2872 // TODO(peah): Add test for the existence of the level controller object once
2873 // that is created only when that is specified in the config.
2874 // TODO(peah): Remove the testing for
2875 // apm->capture_nonlocked_.level_controller_enabled once the value in config_
2876 // is instead used to activate the level controller.
2877 EXPECT_FALSE(apm->capture_nonlocked_.level_controller_enabled);
2878 EXPECT_NEAR(kTargetLcPeakLeveldBFS,
2879 apm->config_.level_controller.initial_peak_level_dbfs,
2880 std::numeric_limits<float>::epsilon());
2881}
2882
andrew@webrtc.org27c69802014-02-18 20:24:56 +00002883} // namespace webrtc