blob: 0a85466db0165e73bff4cc302a0b4fbba1f9c2c8 [file] [log] [blame]
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +00001/*
2 * Copyright (c) 2011 The WebRTC project authors. All Rights Reserved.
3 *
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
11/*
12 * This file includes unit tests for NetEQ.
13 */
14
Henrik Kjellander74640892015-10-29 11:31:02 +010015#include "webrtc/modules/audio_coding/neteq/include/neteq.h"
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +000016
pbos@webrtc.org3ecc1622014-03-07 15:23:34 +000017#include <math.h>
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +000018#include <stdlib.h>
19#include <string.h> // memset
20
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +000021#include <algorithm>
kwiberg2d0c3322016-02-14 09:28:33 -080022#include <memory>
turaj@webrtc.org78b41a02013-11-22 20:27:07 +000023#include <set>
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +000024#include <string>
25#include <vector>
26
turaj@webrtc.orga6101d72013-10-01 22:01:09 +000027#include "gflags/gflags.h"
kjellander@webrtc.org3c0aae12014-09-04 09:55:40 +000028#include "testing/gtest/include/gtest/gtest.h"
henrik.lundin@webrtc.org9b8102c2014-08-21 08:27:44 +000029#include "webrtc/modules/audio_coding/neteq/tools/audio_loop.h"
henrik.lundin@webrtc.org966a7082014-11-17 09:08:38 +000030#include "webrtc/modules/audio_coding/neteq/tools/rtp_file_source.h"
kjellander@webrtc.org3c652b62015-11-18 23:07:57 +010031#include "webrtc/modules/audio_coding/codecs/pcm16b/pcm16b.h"
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +000032#include "webrtc/test/testsupport/fileutils.h"
33#include "webrtc/typedefs.h"
34
minyue5f026d02015-12-16 07:36:04 -080035#ifdef WEBRTC_NETEQ_UNITTEST_BITEXACT
36#ifdef WEBRTC_ANDROID_PLATFORM_BUILD
37#include "external/webrtc/webrtc/modules/audio_coding/neteq/neteq_unittest.pb.h"
38#else
39#include "webrtc/audio_coding/neteq/neteq_unittest.pb.h"
40#endif
41#endif
42
turaj@webrtc.orga6101d72013-10-01 22:01:09 +000043DEFINE_bool(gen_ref, false, "Generate reference files.");
44
minyue5f026d02015-12-16 07:36:04 -080045namespace {
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +000046
minyue5f026d02015-12-16 07:36:04 -080047bool IsAllZero(const int16_t* buf, size_t buf_length) {
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +000048 bool all_zero = true;
Peter Kastingdce40cf2015-08-24 14:52:23 -070049 for (size_t n = 0; n < buf_length && all_zero; ++n)
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +000050 all_zero = buf[n] == 0;
51 return all_zero;
52}
53
minyue5f026d02015-12-16 07:36:04 -080054bool IsAllNonZero(const int16_t* buf, size_t buf_length) {
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +000055 bool all_non_zero = true;
Peter Kastingdce40cf2015-08-24 14:52:23 -070056 for (size_t n = 0; n < buf_length && all_non_zero; ++n)
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +000057 all_non_zero = buf[n] != 0;
58 return all_non_zero;
59}
60
minyue5f026d02015-12-16 07:36:04 -080061#ifdef WEBRTC_NETEQ_UNITTEST_BITEXACT
62void Convert(const webrtc::NetEqNetworkStatistics& stats_raw,
63 webrtc::neteq_unittest::NetEqNetworkStatistics* stats) {
64 stats->set_current_buffer_size_ms(stats_raw.current_buffer_size_ms);
65 stats->set_preferred_buffer_size_ms(stats_raw.preferred_buffer_size_ms);
66 stats->set_jitter_peaks_found(stats_raw.jitter_peaks_found);
67 stats->set_packet_loss_rate(stats_raw.packet_loss_rate);
68 stats->set_packet_discard_rate(stats_raw.packet_discard_rate);
69 stats->set_expand_rate(stats_raw.expand_rate);
70 stats->set_speech_expand_rate(stats_raw.speech_expand_rate);
71 stats->set_preemptive_rate(stats_raw.preemptive_rate);
72 stats->set_accelerate_rate(stats_raw.accelerate_rate);
73 stats->set_secondary_decoded_rate(stats_raw.secondary_decoded_rate);
74 stats->set_clockdrift_ppm(stats_raw.clockdrift_ppm);
75 stats->set_added_zero_samples(stats_raw.added_zero_samples);
76 stats->set_mean_waiting_time_ms(stats_raw.mean_waiting_time_ms);
77 stats->set_median_waiting_time_ms(stats_raw.median_waiting_time_ms);
78 stats->set_min_waiting_time_ms(stats_raw.min_waiting_time_ms);
79 stats->set_max_waiting_time_ms(stats_raw.max_waiting_time_ms);
80}
81
82void Convert(const webrtc::RtcpStatistics& stats_raw,
83 webrtc::neteq_unittest::RtcpStatistics* stats) {
84 stats->set_fraction_lost(stats_raw.fraction_lost);
85 stats->set_cumulative_lost(stats_raw.cumulative_lost);
86 stats->set_extended_max_sequence_number(
87 stats_raw.extended_max_sequence_number);
88 stats->set_jitter(stats_raw.jitter);
89}
90
91void WriteMessage(FILE* file, const std::string& message) {
92 int32_t size = message.length();
93 ASSERT_EQ(1u, fwrite(&size, sizeof(size), 1, file));
94 if (size <= 0)
95 return;
96 ASSERT_EQ(static_cast<size_t>(size),
97 fwrite(message.data(), sizeof(char), size, file));
98}
99
100void ReadMessage(FILE* file, std::string* message) {
101 int32_t size;
102 ASSERT_EQ(1u, fread(&size, sizeof(size), 1, file));
103 if (size <= 0)
104 return;
kwiberg2d0c3322016-02-14 09:28:33 -0800105 std::unique_ptr<char[]> buffer(new char[size]);
minyue5f026d02015-12-16 07:36:04 -0800106 ASSERT_EQ(static_cast<size_t>(size),
107 fread(buffer.get(), sizeof(char), size, file));
108 message->assign(buffer.get(), size);
109}
110#endif // WEBRTC_NETEQ_UNITTEST_BITEXACT
111
112} // namespace
113
114namespace webrtc {
115
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000116class RefFiles {
117 public:
118 RefFiles(const std::string& input_file, const std::string& output_file);
119 ~RefFiles();
120 template<class T> void ProcessReference(const T& test_results);
121 template<typename T, size_t n> void ProcessReference(
122 const T (&test_results)[n],
123 size_t length);
124 template<typename T, size_t n> void WriteToFile(
125 const T (&test_results)[n],
126 size_t length);
127 template<typename T, size_t n> void ReadFromFileAndCompare(
128 const T (&test_results)[n],
129 size_t length);
130 void WriteToFile(const NetEqNetworkStatistics& stats);
131 void ReadFromFileAndCompare(const NetEqNetworkStatistics& stats);
132 void WriteToFile(const RtcpStatistics& stats);
133 void ReadFromFileAndCompare(const RtcpStatistics& stats);
134
135 FILE* input_fp_;
136 FILE* output_fp_;
137};
138
139RefFiles::RefFiles(const std::string &input_file,
140 const std::string &output_file)
141 : input_fp_(NULL),
142 output_fp_(NULL) {
143 if (!input_file.empty()) {
144 input_fp_ = fopen(input_file.c_str(), "rb");
145 EXPECT_TRUE(input_fp_ != NULL);
146 }
147 if (!output_file.empty()) {
148 output_fp_ = fopen(output_file.c_str(), "wb");
149 EXPECT_TRUE(output_fp_ != NULL);
150 }
151}
152
153RefFiles::~RefFiles() {
154 if (input_fp_) {
155 EXPECT_EQ(EOF, fgetc(input_fp_)); // Make sure that we reached the end.
156 fclose(input_fp_);
157 }
158 if (output_fp_) fclose(output_fp_);
159}
160
161template<class T>
162void RefFiles::ProcessReference(const T& test_results) {
163 WriteToFile(test_results);
164 ReadFromFileAndCompare(test_results);
165}
166
167template<typename T, size_t n>
168void RefFiles::ProcessReference(const T (&test_results)[n], size_t length) {
169 WriteToFile(test_results, length);
170 ReadFromFileAndCompare(test_results, length);
171}
172
173template<typename T, size_t n>
174void RefFiles::WriteToFile(const T (&test_results)[n], size_t length) {
175 if (output_fp_) {
176 ASSERT_EQ(length, fwrite(&test_results, sizeof(T), length, output_fp_));
177 }
178}
179
180template<typename T, size_t n>
181void RefFiles::ReadFromFileAndCompare(const T (&test_results)[n],
182 size_t length) {
183 if (input_fp_) {
184 // Read from ref file.
185 T* ref = new T[length];
186 ASSERT_EQ(length, fread(ref, sizeof(T), length, input_fp_));
187 // Compare
188 ASSERT_EQ(0, memcmp(&test_results, ref, sizeof(T) * length));
189 delete [] ref;
190 }
191}
192
minyue5f026d02015-12-16 07:36:04 -0800193void RefFiles::WriteToFile(const NetEqNetworkStatistics& stats_raw) {
194#ifdef WEBRTC_NETEQ_UNITTEST_BITEXACT
195 if (!output_fp_)
196 return;
197 neteq_unittest::NetEqNetworkStatistics stats;
198 Convert(stats_raw, &stats);
199
200 std::string stats_string;
201 ASSERT_TRUE(stats.SerializeToString(&stats_string));
202 WriteMessage(output_fp_, stats_string);
203#else
204 FAIL() << "Writing to reference file requires Proto Buffer.";
205#endif // WEBRTC_NETEQ_UNITTEST_BITEXACT
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000206}
207
208void RefFiles::ReadFromFileAndCompare(
209 const NetEqNetworkStatistics& stats) {
minyue5f026d02015-12-16 07:36:04 -0800210#ifdef WEBRTC_NETEQ_UNITTEST_BITEXACT
211 if (!input_fp_)
212 return;
213
214 std::string stats_string;
215 ReadMessage(input_fp_, &stats_string);
216 neteq_unittest::NetEqNetworkStatistics ref_stats;
217 ASSERT_TRUE(ref_stats.ParseFromString(stats_string));
218
219 // Compare
220 ASSERT_EQ(stats.current_buffer_size_ms, ref_stats.current_buffer_size_ms());
221 ASSERT_EQ(stats.preferred_buffer_size_ms,
222 ref_stats.preferred_buffer_size_ms());
223 ASSERT_EQ(stats.jitter_peaks_found, ref_stats.jitter_peaks_found());
224 ASSERT_EQ(stats.packet_loss_rate, ref_stats.packet_loss_rate());
225 ASSERT_EQ(stats.packet_discard_rate, ref_stats.packet_discard_rate());
226 ASSERT_EQ(stats.expand_rate, ref_stats.expand_rate());
227 ASSERT_EQ(stats.preemptive_rate, ref_stats.preemptive_rate());
228 ASSERT_EQ(stats.accelerate_rate, ref_stats.accelerate_rate());
229 ASSERT_EQ(stats.clockdrift_ppm, ref_stats.clockdrift_ppm());
230 ASSERT_EQ(stats.added_zero_samples, ref_stats.added_zero_samples());
minyue93c08b72015-12-22 09:57:41 -0800231 ASSERT_EQ(stats.secondary_decoded_rate, ref_stats.secondary_decoded_rate());
minyue5f026d02015-12-16 07:36:04 -0800232 ASSERT_LE(stats.speech_expand_rate, ref_stats.expand_rate());
233#else
234 FAIL() << "Reading from reference file requires Proto Buffer.";
235#endif // WEBRTC_NETEQ_UNITTEST_BITEXACT
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000236}
237
minyue5f026d02015-12-16 07:36:04 -0800238void RefFiles::WriteToFile(const RtcpStatistics& stats_raw) {
239#ifdef WEBRTC_NETEQ_UNITTEST_BITEXACT
240 if (!output_fp_)
241 return;
242 neteq_unittest::RtcpStatistics stats;
243 Convert(stats_raw, &stats);
244
245 std::string stats_string;
246 ASSERT_TRUE(stats.SerializeToString(&stats_string));
247 WriteMessage(output_fp_, stats_string);
248#else
249 FAIL() << "Writing to reference file requires Proto Buffer.";
250#endif // WEBRTC_NETEQ_UNITTEST_BITEXACT
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000251}
252
minyue5f026d02015-12-16 07:36:04 -0800253void RefFiles::ReadFromFileAndCompare(const RtcpStatistics& stats) {
254#ifdef WEBRTC_NETEQ_UNITTEST_BITEXACT
255 if (!input_fp_)
256 return;
257 std::string stats_string;
258 ReadMessage(input_fp_, &stats_string);
259 neteq_unittest::RtcpStatistics ref_stats;
260 ASSERT_TRUE(ref_stats.ParseFromString(stats_string));
261
262 // Compare
263 ASSERT_EQ(stats.fraction_lost, ref_stats.fraction_lost());
264 ASSERT_EQ(stats.cumulative_lost, ref_stats.cumulative_lost());
265 ASSERT_EQ(stats.extended_max_sequence_number,
266 ref_stats.extended_max_sequence_number());
267 ASSERT_EQ(stats.jitter, ref_stats.jitter());
268#else
269 FAIL() << "Reading from reference file requires Proto Buffer.";
270#endif // WEBRTC_NETEQ_UNITTEST_BITEXACT
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000271}
272
273class NetEqDecodingTest : public ::testing::Test {
274 protected:
275 // NetEQ must be polled for data once every 10 ms. Thus, neither of the
276 // constants below can be changed.
277 static const int kTimeStepMs = 10;
Peter Kastingdce40cf2015-08-24 14:52:23 -0700278 static const size_t kBlockSize8kHz = kTimeStepMs * 8;
279 static const size_t kBlockSize16kHz = kTimeStepMs * 16;
280 static const size_t kBlockSize32kHz = kTimeStepMs * 32;
minyue93c08b72015-12-22 09:57:41 -0800281 static const size_t kBlockSize48kHz = kTimeStepMs * 48;
282 static const size_t kMaxBlockSize = kBlockSize48kHz;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000283 static const int kInitSampleRateHz = 8000;
284
285 NetEqDecodingTest();
286 virtual void SetUp();
287 virtual void TearDown();
288 void SelectDecoders(NetEqDecoder* used_codec);
289 void LoadDecoders();
290 void OpenInputFile(const std::string &rtp_file);
Peter Kastingdce40cf2015-08-24 14:52:23 -0700291 void Process(size_t* out_len);
minyue5f026d02015-12-16 07:36:04 -0800292
henrik.lundin@webrtc.org4e4b0982014-08-11 14:48:49 +0000293 void DecodeAndCompare(const std::string& rtp_file,
294 const std::string& ref_file,
295 const std::string& stat_ref_file,
296 const std::string& rtcp_ref_file);
minyue5f026d02015-12-16 07:36:04 -0800297
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000298 static void PopulateRtpInfo(int frame_index,
299 int timestamp,
300 WebRtcRTPHeader* rtp_info);
301 static void PopulateCng(int frame_index,
302 int timestamp,
303 WebRtcRTPHeader* rtp_info,
304 uint8_t* payload,
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000305 size_t* payload_len);
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000306
turaj@webrtc.org78b41a02013-11-22 20:27:07 +0000307 void WrapTest(uint16_t start_seq_no, uint32_t start_timestamp,
308 const std::set<uint16_t>& drop_seq_numbers,
309 bool expect_seq_no_wrap, bool expect_timestamp_wrap);
310
henrik.lundin@webrtc.org24779fe2014-03-14 12:40:05 +0000311 void LongCngWithClockDrift(double drift_factor,
312 double network_freeze_ms,
313 bool pull_audio_during_freeze,
314 int delay_tolerance_ms,
315 int max_time_to_speech_ms);
316
henrik.lundin@webrtc.orgca8cb952014-03-12 10:26:52 +0000317 void DuplicateCng();
henrik.lundin@webrtc.orgfcfc6a92014-02-13 11:42:28 +0000318
wu@webrtc.org94454b72014-06-05 20:34:08 +0000319 uint32_t PlayoutTimestamp();
320
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000321 NetEq* neteq_;
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +0000322 NetEq::Config config_;
kwiberg2d0c3322016-02-14 09:28:33 -0800323 std::unique_ptr<test::RtpFileSource> rtp_source_;
324 std::unique_ptr<test::Packet> packet_;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000325 unsigned int sim_clock_;
326 int16_t out_data_[kMaxBlockSize];
327 int output_sample_rate_;
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +0000328 int algorithmic_delay_ms_;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000329};
330
331// Allocating the static const so that it can be passed by reference.
332const int NetEqDecodingTest::kTimeStepMs;
Peter Kastingdce40cf2015-08-24 14:52:23 -0700333const size_t NetEqDecodingTest::kBlockSize8kHz;
334const size_t NetEqDecodingTest::kBlockSize16kHz;
335const size_t NetEqDecodingTest::kBlockSize32kHz;
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000336const size_t NetEqDecodingTest::kMaxBlockSize;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000337const int NetEqDecodingTest::kInitSampleRateHz;
338
339NetEqDecodingTest::NetEqDecodingTest()
340 : neteq_(NULL),
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +0000341 config_(),
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000342 sim_clock_(0),
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +0000343 output_sample_rate_(kInitSampleRateHz),
344 algorithmic_delay_ms_(0) {
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +0000345 config_.sample_rate_hz = kInitSampleRateHz;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000346 memset(out_data_, 0, sizeof(out_data_));
347}
348
349void NetEqDecodingTest::SetUp() {
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +0000350 neteq_ = NetEq::Create(config_);
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +0000351 NetEqNetworkStatistics stat;
352 ASSERT_EQ(0, neteq_->NetworkStatistics(&stat));
353 algorithmic_delay_ms_ = stat.current_buffer_size_ms;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000354 ASSERT_TRUE(neteq_);
355 LoadDecoders();
356}
357
358void NetEqDecodingTest::TearDown() {
359 delete neteq_;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000360}
361
362void NetEqDecodingTest::LoadDecoders() {
363 // Load PCMu.
henrik.lundin4cf61dd2015-12-09 06:20:58 -0800364 ASSERT_EQ(0,
365 neteq_->RegisterPayloadType(NetEqDecoder::kDecoderPCMu, "pcmu", 0));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000366 // Load PCMa.
henrik.lundin4cf61dd2015-12-09 06:20:58 -0800367 ASSERT_EQ(0,
368 neteq_->RegisterPayloadType(NetEqDecoder::kDecoderPCMa, "pcma", 8));
kwiberg98ab3a42015-09-30 21:54:21 -0700369#ifdef WEBRTC_CODEC_ILBC
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000370 // Load iLBC.
henrik.lundin4cf61dd2015-12-09 06:20:58 -0800371 ASSERT_EQ(
372 0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderILBC, "ilbc", 102));
kwiberg98ab3a42015-09-30 21:54:21 -0700373#endif
374#if defined(WEBRTC_CODEC_ISAC) || defined(WEBRTC_CODEC_ISACFX)
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000375 // Load iSAC.
henrik.lundin4cf61dd2015-12-09 06:20:58 -0800376 ASSERT_EQ(
377 0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderISAC, "isac", 103));
kwiberg98ab3a42015-09-30 21:54:21 -0700378#endif
379#ifdef WEBRTC_CODEC_ISAC
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000380 // Load iSAC SWB.
henrik.lundin4cf61dd2015-12-09 06:20:58 -0800381 ASSERT_EQ(0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderISACswb,
382 "isac-swb", 104));
kwiberg98ab3a42015-09-30 21:54:21 -0700383#endif
minyue93c08b72015-12-22 09:57:41 -0800384#ifdef WEBRTC_CODEC_OPUS
385 ASSERT_EQ(0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderOpus,
386 "opus", 111));
387#endif
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000388 // Load PCM16B nb.
henrik.lundin4cf61dd2015-12-09 06:20:58 -0800389 ASSERT_EQ(0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderPCM16B,
390 "pcm16-nb", 93));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000391 // Load PCM16B wb.
henrik.lundin4cf61dd2015-12-09 06:20:58 -0800392 ASSERT_EQ(0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderPCM16Bwb,
393 "pcm16-wb", 94));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000394 // Load PCM16B swb32.
henrik.lundin4cf61dd2015-12-09 06:20:58 -0800395 ASSERT_EQ(0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderPCM16Bswb32kHz,
396 "pcm16-swb32", 95));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000397 // Load CNG 8 kHz.
henrik.lundin4cf61dd2015-12-09 06:20:58 -0800398 ASSERT_EQ(0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderCNGnb,
399 "cng-nb", 13));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000400 // Load CNG 16 kHz.
henrik.lundin4cf61dd2015-12-09 06:20:58 -0800401 ASSERT_EQ(0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderCNGwb,
402 "cng-wb", 98));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000403}
404
405void NetEqDecodingTest::OpenInputFile(const std::string &rtp_file) {
henrik.lundin@webrtc.org966a7082014-11-17 09:08:38 +0000406 rtp_source_.reset(test::RtpFileSource::Create(rtp_file));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000407}
408
Peter Kastingdce40cf2015-08-24 14:52:23 -0700409void NetEqDecodingTest::Process(size_t* out_len) {
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000410 // Check if time to receive.
henrik.lundin@webrtc.org966a7082014-11-17 09:08:38 +0000411 while (packet_ && sim_clock_ >= packet_->time_ms()) {
412 if (packet_->payload_length_bytes() > 0) {
413 WebRtcRTPHeader rtp_header;
414 packet_->ConvertHeader(&rtp_header);
ivoc72c08ed2016-01-20 07:26:24 -0800415#ifndef WEBRTC_CODEC_ISAC
416 // Ignore payload type 104 (iSAC-swb) if ISAC is not supported.
417 if (rtp_header.header.payloadType != 104)
418#endif
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000419 ASSERT_EQ(0, neteq_->InsertPacket(
kwibergee2bac22015-11-11 10:34:00 -0800420 rtp_header,
421 rtc::ArrayView<const uint8_t>(
422 packet_->payload(), packet_->payload_length_bytes()),
423 static_cast<uint32_t>(packet_->time_ms() *
424 (output_sample_rate_ / 1000))));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000425 }
426 // Get next packet.
henrik.lundin@webrtc.org966a7082014-11-17 09:08:38 +0000427 packet_.reset(rtp_source_->NextPacket());
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000428 }
429
henrik.lundin@webrtc.orge1d468c2013-01-30 07:37:20 +0000430 // Get audio from NetEq.
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000431 NetEqOutputType type;
Peter Kasting69558702016-01-12 16:26:35 -0800432 size_t num_channels;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000433 ASSERT_EQ(0, neteq_->GetAudio(kMaxBlockSize, out_data_, out_len,
434 &num_channels, &type));
435 ASSERT_TRUE((*out_len == kBlockSize8kHz) ||
436 (*out_len == kBlockSize16kHz) ||
minyue93c08b72015-12-22 09:57:41 -0800437 (*out_len == kBlockSize32kHz) ||
438 (*out_len == kBlockSize48kHz));
Peter Kastingdce40cf2015-08-24 14:52:23 -0700439 output_sample_rate_ = static_cast<int>(*out_len / 10 * 1000);
henrik.lundind89814b2015-11-23 06:49:25 -0800440 EXPECT_EQ(output_sample_rate_, neteq_->last_output_sample_rate_hz());
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000441
442 // Increase time.
443 sim_clock_ += kTimeStepMs;
444}
445
henrik.lundin@webrtc.org4e4b0982014-08-11 14:48:49 +0000446void NetEqDecodingTest::DecodeAndCompare(const std::string& rtp_file,
447 const std::string& ref_file,
448 const std::string& stat_ref_file,
449 const std::string& rtcp_ref_file) {
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000450 OpenInputFile(rtp_file);
451
452 std::string ref_out_file = "";
453 if (ref_file.empty()) {
turaj@webrtc.orga6101d72013-10-01 22:01:09 +0000454 ref_out_file = webrtc::test::OutputPath() + "neteq_universal_ref.pcm";
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000455 }
456 RefFiles ref_files(ref_file, ref_out_file);
457
henrik.lundin@webrtc.org4e4b0982014-08-11 14:48:49 +0000458 std::string stat_out_file = "";
459 if (stat_ref_file.empty()) {
460 stat_out_file = webrtc::test::OutputPath() + "neteq_network_stats.dat";
461 }
462 RefFiles network_stat_files(stat_ref_file, stat_out_file);
463
464 std::string rtcp_out_file = "";
465 if (rtcp_ref_file.empty()) {
466 rtcp_out_file = webrtc::test::OutputPath() + "neteq_rtcp_stats.dat";
467 }
468 RefFiles rtcp_stat_files(rtcp_ref_file, rtcp_out_file);
469
henrik.lundin@webrtc.org966a7082014-11-17 09:08:38 +0000470 packet_.reset(rtp_source_->NextPacket());
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000471 int i = 0;
henrik.lundin@webrtc.org966a7082014-11-17 09:08:38 +0000472 while (packet_) {
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000473 std::ostringstream ss;
474 ss << "Lap number " << i++ << " in DecodeAndCompare while loop";
475 SCOPED_TRACE(ss.str()); // Print out the parameter values on failure.
Peter Kastingdce40cf2015-08-24 14:52:23 -0700476 size_t out_len = 0;
henrik.lundin@webrtc.org966a7082014-11-17 09:08:38 +0000477 ASSERT_NO_FATAL_FAILURE(Process(&out_len));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000478 ASSERT_NO_FATAL_FAILURE(ref_files.ProcessReference(out_data_, out_len));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000479
480 // Query the network statistics API once per second
481 if (sim_clock_ % 1000 == 0) {
482 // Process NetworkStatistics.
483 NetEqNetworkStatistics network_stats;
484 ASSERT_EQ(0, neteq_->NetworkStatistics(&network_stats));
henrik.lundin@webrtc.orgb4e80e02014-05-15 07:14:00 +0000485 ASSERT_NO_FATAL_FAILURE(
486 network_stat_files.ProcessReference(network_stats));
henrik.lundin9c3efd02015-08-27 13:12:22 -0700487 // Compare with CurrentDelay, which should be identical.
488 EXPECT_EQ(network_stats.current_buffer_size_ms, neteq_->CurrentDelayMs());
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000489
490 // Process RTCPstat.
491 RtcpStatistics rtcp_stats;
492 neteq_->GetRtcpStatistics(&rtcp_stats);
henrik.lundin@webrtc.orgb4e80e02014-05-15 07:14:00 +0000493 ASSERT_NO_FATAL_FAILURE(rtcp_stat_files.ProcessReference(rtcp_stats));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000494 }
495 }
496}
497
498void NetEqDecodingTest::PopulateRtpInfo(int frame_index,
499 int timestamp,
500 WebRtcRTPHeader* rtp_info) {
501 rtp_info->header.sequenceNumber = frame_index;
502 rtp_info->header.timestamp = timestamp;
503 rtp_info->header.ssrc = 0x1234; // Just an arbitrary SSRC.
504 rtp_info->header.payloadType = 94; // PCM16b WB codec.
505 rtp_info->header.markerBit = 0;
506}
507
508void NetEqDecodingTest::PopulateCng(int frame_index,
509 int timestamp,
510 WebRtcRTPHeader* rtp_info,
511 uint8_t* payload,
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000512 size_t* payload_len) {
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000513 rtp_info->header.sequenceNumber = frame_index;
514 rtp_info->header.timestamp = timestamp;
515 rtp_info->header.ssrc = 0x1234; // Just an arbitrary SSRC.
516 rtp_info->header.payloadType = 98; // WB CNG.
517 rtp_info->header.markerBit = 0;
518 payload[0] = 64; // Noise level -64 dBov, quite arbitrarily chosen.
519 *payload_len = 1; // Only noise level, no spectral parameters.
520}
521
ivoc72c08ed2016-01-20 07:26:24 -0800522#if !defined(WEBRTC_IOS) && defined(WEBRTC_NETEQ_UNITTEST_BITEXACT) && \
523 (defined(WEBRTC_CODEC_ISAC) || defined(WEBRTC_CODEC_ISACFX)) && \
524 defined(WEBRTC_CODEC_ILBC) && defined(WEBRTC_CODEC_G722) && \
525 !defined(WEBRTC_ARCH_ARM64)
minyue5f026d02015-12-16 07:36:04 -0800526#define MAYBE_TestBitExactness TestBitExactness
kwiberg98ab3a42015-09-30 21:54:21 -0700527#else
minyue5f026d02015-12-16 07:36:04 -0800528#define MAYBE_TestBitExactness DISABLED_TestBitExactness
kwiberg98ab3a42015-09-30 21:54:21 -0700529#endif
minyue5f026d02015-12-16 07:36:04 -0800530TEST_F(NetEqDecodingTest, MAYBE_TestBitExactness) {
minyue49c454e2016-01-08 11:30:14 -0800531 const std::string input_rtp_file =
532 webrtc::test::ResourcePath("audio_coding/neteq_universal_new", "rtp");
henrik.lundin@webrtc.org48438c22014-05-20 16:07:43 +0000533 // Note that neteq4_universal_ref.pcm and neteq4_universal_ref_win_32.pcm
534 // are identical. The latter could have been removed, but if clients still
535 // have a copy of the file, the test will fail.
andrew@webrtc.orgf6a638e2014-02-04 01:31:28 +0000536 const std::string input_ref_file =
turaj@webrtc.orga6101d72013-10-01 22:01:09 +0000537 webrtc::test::ResourcePath("audio_coding/neteq4_universal_ref", "pcm");
henrik.lundin@webrtc.org6e3968f2013-01-31 15:07:30 +0000538#if defined(_MSC_VER) && (_MSC_VER >= 1700)
539 // For Visual Studio 2012 and later, we will have to use the generic reference
540 // file, rather than the windows-specific one.
andrew@webrtc.orgf6a638e2014-02-04 01:31:28 +0000541 const std::string network_stat_ref_file = webrtc::test::ProjectRootPath() +
turaj@webrtc.orga6101d72013-10-01 22:01:09 +0000542 "resources/audio_coding/neteq4_network_stats.dat";
henrik.lundin@webrtc.org6e3968f2013-01-31 15:07:30 +0000543#else
andrew@webrtc.orgf6a638e2014-02-04 01:31:28 +0000544 const std::string network_stat_ref_file =
turaj@webrtc.orga6101d72013-10-01 22:01:09 +0000545 webrtc::test::ResourcePath("audio_coding/neteq4_network_stats", "dat");
henrik.lundin@webrtc.org6e3968f2013-01-31 15:07:30 +0000546#endif
andrew@webrtc.orgf6a638e2014-02-04 01:31:28 +0000547 const std::string rtcp_stat_ref_file =
turaj@webrtc.orga6101d72013-10-01 22:01:09 +0000548 webrtc::test::ResourcePath("audio_coding/neteq4_rtcp_stats", "dat");
henrik.lundin@webrtc.org4e4b0982014-08-11 14:48:49 +0000549
turaj@webrtc.orga6101d72013-10-01 22:01:09 +0000550 if (FLAGS_gen_ref) {
henrik.lundin@webrtc.org4e4b0982014-08-11 14:48:49 +0000551 DecodeAndCompare(input_rtp_file, "", "", "");
turaj@webrtc.orga6101d72013-10-01 22:01:09 +0000552 } else {
henrik.lundin@webrtc.org4e4b0982014-08-11 14:48:49 +0000553 DecodeAndCompare(input_rtp_file,
554 input_ref_file,
555 network_stat_ref_file,
556 rtcp_stat_ref_file);
turaj@webrtc.orga6101d72013-10-01 22:01:09 +0000557 }
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000558}
559
minyue93c08b72015-12-22 09:57:41 -0800560#if !defined(WEBRTC_IOS) && !defined(WEBRTC_ANDROID) && \
561 defined(WEBRTC_NETEQ_UNITTEST_BITEXACT) && \
562 defined(WEBRTC_CODEC_OPUS)
563#define MAYBE_TestOpusBitExactness TestOpusBitExactness
564#else
565#define MAYBE_TestOpusBitExactness DISABLED_TestOpusBitExactness
566#endif
567TEST_F(NetEqDecodingTest, MAYBE_TestOpusBitExactness) {
568 const std::string input_rtp_file =
569 webrtc::test::ResourcePath("audio_coding/neteq_opus", "rtp");
570 const std::string input_ref_file =
kjellanderc3a09832016-02-02 13:18:33 -0800571 // The pcm files were generated by using Opus v1.1.2 to decode the RTC
572 // file generated by Opus v1.1
minyue93c08b72015-12-22 09:57:41 -0800573 webrtc::test::ResourcePath("audio_coding/neteq4_opus_ref", "pcm");
574 const std::string network_stat_ref_file =
kjellanderc3a09832016-02-02 13:18:33 -0800575 // The network stats file was generated when using Opus v1.1.2 to decode
576 // the RTC file generated by Opus v1.1
minyue93c08b72015-12-22 09:57:41 -0800577 webrtc::test::ResourcePath("audio_coding/neteq4_opus_network_stats",
578 "dat");
579 const std::string rtcp_stat_ref_file =
580 webrtc::test::ResourcePath("audio_coding/neteq4_opus_rtcp_stats", "dat");
581
582 if (FLAGS_gen_ref) {
583 DecodeAndCompare(input_rtp_file, "", "", "");
584 } else {
585 DecodeAndCompare(input_rtp_file,
586 input_ref_file,
587 network_stat_ref_file,
588 rtcp_stat_ref_file);
589 }
590}
591
henrik.lundin@webrtc.org7cbc4f92014-10-07 06:37:39 +0000592// Use fax mode to avoid time-scaling. This is to simplify the testing of
593// packet waiting times in the packet buffer.
594class NetEqDecodingTestFaxMode : public NetEqDecodingTest {
595 protected:
596 NetEqDecodingTestFaxMode() : NetEqDecodingTest() {
597 config_.playout_mode = kPlayoutFax;
598 }
599};
600
601TEST_F(NetEqDecodingTestFaxMode, TestFrameWaitingTimeStatistics) {
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000602 // Insert 30 dummy packets at once. Each packet contains 10 ms 16 kHz audio.
603 size_t num_frames = 30;
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000604 const size_t kSamples = 10 * 16;
605 const size_t kPayloadBytes = kSamples * 2;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000606 for (size_t i = 0; i < num_frames; ++i) {
kwibergee2bac22015-11-11 10:34:00 -0800607 const uint8_t payload[kPayloadBytes] = {0};
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000608 WebRtcRTPHeader rtp_info;
609 rtp_info.header.sequenceNumber = i;
610 rtp_info.header.timestamp = i * kSamples;
611 rtp_info.header.ssrc = 0x1234; // Just an arbitrary SSRC.
612 rtp_info.header.payloadType = 94; // PCM16b WB codec.
613 rtp_info.header.markerBit = 0;
kwibergee2bac22015-11-11 10:34:00 -0800614 ASSERT_EQ(0, neteq_->InsertPacket(rtp_info, payload, 0));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000615 }
616 // Pull out all data.
617 for (size_t i = 0; i < num_frames; ++i) {
Peter Kastingdce40cf2015-08-24 14:52:23 -0700618 size_t out_len;
Peter Kasting69558702016-01-12 16:26:35 -0800619 size_t num_channels;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000620 NetEqOutputType type;
621 ASSERT_EQ(0, neteq_->GetAudio(kMaxBlockSize, out_data_, &out_len,
622 &num_channels, &type));
623 ASSERT_EQ(kBlockSize16kHz, out_len);
624 }
625
Henrik Lundin1bb8cf82015-08-25 13:08:04 +0200626 NetEqNetworkStatistics stats;
627 EXPECT_EQ(0, neteq_->NetworkStatistics(&stats));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000628 // Since all frames are dumped into NetEQ at once, but pulled out with 10 ms
629 // spacing (per definition), we expect the delay to increase with 10 ms for
Henrik Lundin1bb8cf82015-08-25 13:08:04 +0200630 // each packet. Thus, we are calculating the statistics for a series from 10
631 // to 300, in steps of 10 ms.
632 EXPECT_EQ(155, stats.mean_waiting_time_ms);
633 EXPECT_EQ(155, stats.median_waiting_time_ms);
634 EXPECT_EQ(10, stats.min_waiting_time_ms);
635 EXPECT_EQ(300, stats.max_waiting_time_ms);
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000636
637 // Check statistics again and make sure it's been reset.
Henrik Lundin1bb8cf82015-08-25 13:08:04 +0200638 EXPECT_EQ(0, neteq_->NetworkStatistics(&stats));
639 EXPECT_EQ(-1, stats.mean_waiting_time_ms);
640 EXPECT_EQ(-1, stats.median_waiting_time_ms);
641 EXPECT_EQ(-1, stats.min_waiting_time_ms);
642 EXPECT_EQ(-1, stats.max_waiting_time_ms);
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000643}
644
henrik.lundin@webrtc.orgb4e80e02014-05-15 07:14:00 +0000645TEST_F(NetEqDecodingTest, TestAverageInterArrivalTimeNegative) {
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000646 const int kNumFrames = 3000; // Needed for convergence.
647 int frame_index = 0;
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000648 const size_t kSamples = 10 * 16;
649 const size_t kPayloadBytes = kSamples * 2;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000650 while (frame_index < kNumFrames) {
651 // Insert one packet each time, except every 10th time where we insert two
652 // packets at once. This will create a negative clock-drift of approx. 10%.
653 int num_packets = (frame_index % 10 == 0 ? 2 : 1);
654 for (int n = 0; n < num_packets; ++n) {
655 uint8_t payload[kPayloadBytes] = {0};
656 WebRtcRTPHeader rtp_info;
657 PopulateRtpInfo(frame_index, frame_index * kSamples, &rtp_info);
kwibergee2bac22015-11-11 10:34:00 -0800658 ASSERT_EQ(0, neteq_->InsertPacket(rtp_info, payload, 0));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000659 ++frame_index;
660 }
661
662 // Pull out data once.
Peter Kastingdce40cf2015-08-24 14:52:23 -0700663 size_t out_len;
Peter Kasting69558702016-01-12 16:26:35 -0800664 size_t num_channels;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000665 NetEqOutputType type;
666 ASSERT_EQ(0, neteq_->GetAudio(kMaxBlockSize, out_data_, &out_len,
667 &num_channels, &type));
668 ASSERT_EQ(kBlockSize16kHz, out_len);
669 }
670
671 NetEqNetworkStatistics network_stats;
672 ASSERT_EQ(0, neteq_->NetworkStatistics(&network_stats));
673 EXPECT_EQ(-103196, network_stats.clockdrift_ppm);
674}
675
henrik.lundin@webrtc.orgb4e80e02014-05-15 07:14:00 +0000676TEST_F(NetEqDecodingTest, TestAverageInterArrivalTimePositive) {
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000677 const int kNumFrames = 5000; // Needed for convergence.
678 int frame_index = 0;
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000679 const size_t kSamples = 10 * 16;
680 const size_t kPayloadBytes = kSamples * 2;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000681 for (int i = 0; i < kNumFrames; ++i) {
682 // Insert one packet each time, except every 10th time where we don't insert
683 // any packet. This will create a positive clock-drift of approx. 11%.
684 int num_packets = (i % 10 == 9 ? 0 : 1);
685 for (int n = 0; n < num_packets; ++n) {
686 uint8_t payload[kPayloadBytes] = {0};
687 WebRtcRTPHeader rtp_info;
688 PopulateRtpInfo(frame_index, frame_index * kSamples, &rtp_info);
kwibergee2bac22015-11-11 10:34:00 -0800689 ASSERT_EQ(0, neteq_->InsertPacket(rtp_info, payload, 0));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000690 ++frame_index;
691 }
692
693 // Pull out data once.
Peter Kastingdce40cf2015-08-24 14:52:23 -0700694 size_t out_len;
Peter Kasting69558702016-01-12 16:26:35 -0800695 size_t num_channels;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000696 NetEqOutputType type;
697 ASSERT_EQ(0, neteq_->GetAudio(kMaxBlockSize, out_data_, &out_len,
698 &num_channels, &type));
699 ASSERT_EQ(kBlockSize16kHz, out_len);
700 }
701
702 NetEqNetworkStatistics network_stats;
703 ASSERT_EQ(0, neteq_->NetworkStatistics(&network_stats));
704 EXPECT_EQ(110946, network_stats.clockdrift_ppm);
705}
706
henrik.lundin@webrtc.org24779fe2014-03-14 12:40:05 +0000707void NetEqDecodingTest::LongCngWithClockDrift(double drift_factor,
708 double network_freeze_ms,
709 bool pull_audio_during_freeze,
710 int delay_tolerance_ms,
711 int max_time_to_speech_ms) {
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000712 uint16_t seq_no = 0;
713 uint32_t timestamp = 0;
714 const int kFrameSizeMs = 30;
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000715 const size_t kSamples = kFrameSizeMs * 16;
716 const size_t kPayloadBytes = kSamples * 2;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000717 double next_input_time_ms = 0.0;
718 double t_ms;
Peter Kastingdce40cf2015-08-24 14:52:23 -0700719 size_t out_len;
Peter Kasting69558702016-01-12 16:26:35 -0800720 size_t num_channels;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000721 NetEqOutputType type;
722
723 // Insert speech for 5 seconds.
724 const int kSpeechDurationMs = 5000;
725 for (t_ms = 0; t_ms < kSpeechDurationMs; t_ms += 10) {
726 // Each turn in this for loop is 10 ms.
727 while (next_input_time_ms <= t_ms) {
728 // Insert one 30 ms speech frame.
729 uint8_t payload[kPayloadBytes] = {0};
730 WebRtcRTPHeader rtp_info;
731 PopulateRtpInfo(seq_no, timestamp, &rtp_info);
kwibergee2bac22015-11-11 10:34:00 -0800732 ASSERT_EQ(0, neteq_->InsertPacket(rtp_info, payload, 0));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000733 ++seq_no;
734 timestamp += kSamples;
henrik.lundin@webrtc.orgfcfc6a92014-02-13 11:42:28 +0000735 next_input_time_ms += static_cast<double>(kFrameSizeMs) * drift_factor;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000736 }
737 // Pull out data once.
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000738 ASSERT_EQ(0, neteq_->GetAudio(kMaxBlockSize, out_data_, &out_len,
739 &num_channels, &type));
740 ASSERT_EQ(kBlockSize16kHz, out_len);
741 }
742
743 EXPECT_EQ(kOutputNormal, type);
wu@webrtc.org94454b72014-06-05 20:34:08 +0000744 int32_t delay_before = timestamp - PlayoutTimestamp();
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000745
746 // Insert CNG for 1 minute (= 60000 ms).
747 const int kCngPeriodMs = 100;
748 const int kCngPeriodSamples = kCngPeriodMs * 16; // Period in 16 kHz samples.
749 const int kCngDurationMs = 60000;
750 for (; t_ms < kSpeechDurationMs + kCngDurationMs; t_ms += 10) {
751 // Each turn in this for loop is 10 ms.
752 while (next_input_time_ms <= t_ms) {
753 // Insert one CNG frame each 100 ms.
754 uint8_t payload[kPayloadBytes];
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000755 size_t payload_len;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000756 WebRtcRTPHeader rtp_info;
757 PopulateCng(seq_no, timestamp, &rtp_info, payload, &payload_len);
kwibergee2bac22015-11-11 10:34:00 -0800758 ASSERT_EQ(0, neteq_->InsertPacket(
759 rtp_info,
760 rtc::ArrayView<const uint8_t>(payload, payload_len), 0));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000761 ++seq_no;
762 timestamp += kCngPeriodSamples;
henrik.lundin@webrtc.orgfcfc6a92014-02-13 11:42:28 +0000763 next_input_time_ms += static_cast<double>(kCngPeriodMs) * drift_factor;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000764 }
765 // Pull out data once.
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000766 ASSERT_EQ(0, neteq_->GetAudio(kMaxBlockSize, out_data_, &out_len,
767 &num_channels, &type));
768 ASSERT_EQ(kBlockSize16kHz, out_len);
769 }
770
771 EXPECT_EQ(kOutputCNG, type);
772
henrik.lundin@webrtc.org24779fe2014-03-14 12:40:05 +0000773 if (network_freeze_ms > 0) {
774 // First keep pulling audio for |network_freeze_ms| without inserting
775 // any data, then insert CNG data corresponding to |network_freeze_ms|
776 // without pulling any output audio.
777 const double loop_end_time = t_ms + network_freeze_ms;
778 for (; t_ms < loop_end_time; t_ms += 10) {
779 // Pull out data once.
780 ASSERT_EQ(0,
781 neteq_->GetAudio(
782 kMaxBlockSize, out_data_, &out_len, &num_channels, &type));
783 ASSERT_EQ(kBlockSize16kHz, out_len);
784 EXPECT_EQ(kOutputCNG, type);
785 }
786 bool pull_once = pull_audio_during_freeze;
787 // If |pull_once| is true, GetAudio will be called once half-way through
788 // the network recovery period.
789 double pull_time_ms = (t_ms + next_input_time_ms) / 2;
790 while (next_input_time_ms <= t_ms) {
791 if (pull_once && next_input_time_ms >= pull_time_ms) {
792 pull_once = false;
793 // Pull out data once.
794 ASSERT_EQ(
795 0,
796 neteq_->GetAudio(
797 kMaxBlockSize, out_data_, &out_len, &num_channels, &type));
798 ASSERT_EQ(kBlockSize16kHz, out_len);
799 EXPECT_EQ(kOutputCNG, type);
800 t_ms += 10;
801 }
802 // Insert one CNG frame each 100 ms.
803 uint8_t payload[kPayloadBytes];
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000804 size_t payload_len;
henrik.lundin@webrtc.org24779fe2014-03-14 12:40:05 +0000805 WebRtcRTPHeader rtp_info;
806 PopulateCng(seq_no, timestamp, &rtp_info, payload, &payload_len);
kwibergee2bac22015-11-11 10:34:00 -0800807 ASSERT_EQ(0, neteq_->InsertPacket(
808 rtp_info,
809 rtc::ArrayView<const uint8_t>(payload, payload_len), 0));
henrik.lundin@webrtc.org24779fe2014-03-14 12:40:05 +0000810 ++seq_no;
811 timestamp += kCngPeriodSamples;
812 next_input_time_ms += kCngPeriodMs * drift_factor;
813 }
814 }
815
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000816 // Insert speech again until output type is speech.
henrik.lundin@webrtc.org24779fe2014-03-14 12:40:05 +0000817 double speech_restart_time_ms = t_ms;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000818 while (type != kOutputNormal) {
819 // Each turn in this for loop is 10 ms.
820 while (next_input_time_ms <= t_ms) {
821 // Insert one 30 ms speech frame.
822 uint8_t payload[kPayloadBytes] = {0};
823 WebRtcRTPHeader rtp_info;
824 PopulateRtpInfo(seq_no, timestamp, &rtp_info);
kwibergee2bac22015-11-11 10:34:00 -0800825 ASSERT_EQ(0, neteq_->InsertPacket(rtp_info, payload, 0));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000826 ++seq_no;
827 timestamp += kSamples;
henrik.lundin@webrtc.org24779fe2014-03-14 12:40:05 +0000828 next_input_time_ms += kFrameSizeMs * drift_factor;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000829 }
830 // Pull out data once.
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000831 ASSERT_EQ(0, neteq_->GetAudio(kMaxBlockSize, out_data_, &out_len,
832 &num_channels, &type));
833 ASSERT_EQ(kBlockSize16kHz, out_len);
834 // Increase clock.
835 t_ms += 10;
836 }
837
henrik.lundin@webrtc.org24779fe2014-03-14 12:40:05 +0000838 // Check that the speech starts again within reasonable time.
839 double time_until_speech_returns_ms = t_ms - speech_restart_time_ms;
840 EXPECT_LT(time_until_speech_returns_ms, max_time_to_speech_ms);
wu@webrtc.org94454b72014-06-05 20:34:08 +0000841 int32_t delay_after = timestamp - PlayoutTimestamp();
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000842 // Compare delay before and after, and make sure it differs less than 20 ms.
henrik.lundin@webrtc.org24779fe2014-03-14 12:40:05 +0000843 EXPECT_LE(delay_after, delay_before + delay_tolerance_ms * 16);
844 EXPECT_GE(delay_after, delay_before - delay_tolerance_ms * 16);
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000845}
846
henrik.lundin@webrtc.orgb4e80e02014-05-15 07:14:00 +0000847TEST_F(NetEqDecodingTest, LongCngWithNegativeClockDrift) {
henrik.lundin@webrtc.orgfcfc6a92014-02-13 11:42:28 +0000848 // Apply a clock drift of -25 ms / s (sender faster than receiver).
849 const double kDriftFactor = 1000.0 / (1000.0 + 25.0);
henrik.lundin@webrtc.org24779fe2014-03-14 12:40:05 +0000850 const double kNetworkFreezeTimeMs = 0.0;
851 const bool kGetAudioDuringFreezeRecovery = false;
852 const int kDelayToleranceMs = 20;
853 const int kMaxTimeToSpeechMs = 100;
854 LongCngWithClockDrift(kDriftFactor,
855 kNetworkFreezeTimeMs,
856 kGetAudioDuringFreezeRecovery,
857 kDelayToleranceMs,
858 kMaxTimeToSpeechMs);
henrik.lundin@webrtc.orgfcfc6a92014-02-13 11:42:28 +0000859}
860
henrik.lundin@webrtc.orgb4e80e02014-05-15 07:14:00 +0000861TEST_F(NetEqDecodingTest, LongCngWithPositiveClockDrift) {
henrik.lundin@webrtc.orgfcfc6a92014-02-13 11:42:28 +0000862 // Apply a clock drift of +25 ms / s (sender slower than receiver).
863 const double kDriftFactor = 1000.0 / (1000.0 - 25.0);
henrik.lundin@webrtc.org24779fe2014-03-14 12:40:05 +0000864 const double kNetworkFreezeTimeMs = 0.0;
865 const bool kGetAudioDuringFreezeRecovery = false;
866 const int kDelayToleranceMs = 20;
867 const int kMaxTimeToSpeechMs = 100;
868 LongCngWithClockDrift(kDriftFactor,
869 kNetworkFreezeTimeMs,
870 kGetAudioDuringFreezeRecovery,
871 kDelayToleranceMs,
872 kMaxTimeToSpeechMs);
873}
874
henrik.lundin@webrtc.orgb4e80e02014-05-15 07:14:00 +0000875TEST_F(NetEqDecodingTest, LongCngWithNegativeClockDriftNetworkFreeze) {
henrik.lundin@webrtc.org24779fe2014-03-14 12:40:05 +0000876 // Apply a clock drift of -25 ms / s (sender faster than receiver).
877 const double kDriftFactor = 1000.0 / (1000.0 + 25.0);
878 const double kNetworkFreezeTimeMs = 5000.0;
879 const bool kGetAudioDuringFreezeRecovery = false;
880 const int kDelayToleranceMs = 50;
881 const int kMaxTimeToSpeechMs = 200;
882 LongCngWithClockDrift(kDriftFactor,
883 kNetworkFreezeTimeMs,
884 kGetAudioDuringFreezeRecovery,
885 kDelayToleranceMs,
886 kMaxTimeToSpeechMs);
887}
888
henrik.lundin@webrtc.orgb4e80e02014-05-15 07:14:00 +0000889TEST_F(NetEqDecodingTest, LongCngWithPositiveClockDriftNetworkFreeze) {
henrik.lundin@webrtc.org24779fe2014-03-14 12:40:05 +0000890 // Apply a clock drift of +25 ms / s (sender slower than receiver).
891 const double kDriftFactor = 1000.0 / (1000.0 - 25.0);
892 const double kNetworkFreezeTimeMs = 5000.0;
893 const bool kGetAudioDuringFreezeRecovery = false;
894 const int kDelayToleranceMs = 20;
895 const int kMaxTimeToSpeechMs = 100;
896 LongCngWithClockDrift(kDriftFactor,
897 kNetworkFreezeTimeMs,
898 kGetAudioDuringFreezeRecovery,
899 kDelayToleranceMs,
900 kMaxTimeToSpeechMs);
901}
902
henrik.lundin@webrtc.orgb4e80e02014-05-15 07:14:00 +0000903TEST_F(NetEqDecodingTest, LongCngWithPositiveClockDriftNetworkFreezeExtraPull) {
henrik.lundin@webrtc.org24779fe2014-03-14 12:40:05 +0000904 // Apply a clock drift of +25 ms / s (sender slower than receiver).
905 const double kDriftFactor = 1000.0 / (1000.0 - 25.0);
906 const double kNetworkFreezeTimeMs = 5000.0;
907 const bool kGetAudioDuringFreezeRecovery = true;
908 const int kDelayToleranceMs = 20;
909 const int kMaxTimeToSpeechMs = 100;
910 LongCngWithClockDrift(kDriftFactor,
911 kNetworkFreezeTimeMs,
912 kGetAudioDuringFreezeRecovery,
913 kDelayToleranceMs,
914 kMaxTimeToSpeechMs);
915}
916
henrik.lundin@webrtc.orgb4e80e02014-05-15 07:14:00 +0000917TEST_F(NetEqDecodingTest, LongCngWithoutClockDrift) {
henrik.lundin@webrtc.org24779fe2014-03-14 12:40:05 +0000918 const double kDriftFactor = 1.0; // No drift.
919 const double kNetworkFreezeTimeMs = 0.0;
920 const bool kGetAudioDuringFreezeRecovery = false;
921 const int kDelayToleranceMs = 10;
922 const int kMaxTimeToSpeechMs = 50;
923 LongCngWithClockDrift(kDriftFactor,
924 kNetworkFreezeTimeMs,
925 kGetAudioDuringFreezeRecovery,
926 kDelayToleranceMs,
927 kMaxTimeToSpeechMs);
henrik.lundin@webrtc.orgfcfc6a92014-02-13 11:42:28 +0000928}
929
henrik.lundin@webrtc.orgb4e80e02014-05-15 07:14:00 +0000930TEST_F(NetEqDecodingTest, UnknownPayloadType) {
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000931 const size_t kPayloadBytes = 100;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000932 uint8_t payload[kPayloadBytes] = {0};
933 WebRtcRTPHeader rtp_info;
934 PopulateRtpInfo(0, 0, &rtp_info);
935 rtp_info.header.payloadType = 1; // Not registered as a decoder.
kwibergee2bac22015-11-11 10:34:00 -0800936 EXPECT_EQ(NetEq::kFail, neteq_->InsertPacket(rtp_info, payload, 0));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000937 EXPECT_EQ(NetEq::kUnknownRtpPayloadType, neteq_->LastError());
938}
939
Peter Boströme2976c82016-01-04 22:44:05 +0100940#if defined(WEBRTC_CODEC_ISAC) || defined(WEBRTC_CODEC_ISACFX)
ivoc72c08ed2016-01-20 07:26:24 -0800941#define MAYBE_DecoderError DecoderError
942#else
943#define MAYBE_DecoderError DISABLED_DecoderError
944#endif
945
Peter Boströme2976c82016-01-04 22:44:05 +0100946TEST_F(NetEqDecodingTest, MAYBE_DecoderError) {
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000947 const size_t kPayloadBytes = 100;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000948 uint8_t payload[kPayloadBytes] = {0};
949 WebRtcRTPHeader rtp_info;
950 PopulateRtpInfo(0, 0, &rtp_info);
951 rtp_info.header.payloadType = 103; // iSAC, but the payload is invalid.
kwibergee2bac22015-11-11 10:34:00 -0800952 EXPECT_EQ(0, neteq_->InsertPacket(rtp_info, payload, 0));
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000953 NetEqOutputType type;
954 // Set all of |out_data_| to 1, and verify that it was set to 0 by the call
955 // to GetAudio.
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000956 for (size_t i = 0; i < kMaxBlockSize; ++i) {
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000957 out_data_[i] = 1;
958 }
Peter Kasting69558702016-01-12 16:26:35 -0800959 size_t num_channels;
Peter Kastingdce40cf2015-08-24 14:52:23 -0700960 size_t samples_per_channel;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000961 EXPECT_EQ(NetEq::kFail,
962 neteq_->GetAudio(kMaxBlockSize, out_data_,
963 &samples_per_channel, &num_channels, &type));
964 // Verify that there is a decoder error to check.
965 EXPECT_EQ(NetEq::kDecoderErrorCode, neteq_->LastError());
ivoc72c08ed2016-01-20 07:26:24 -0800966
967 enum NetEqDecoderError {
968 ISAC_LENGTH_MISMATCH = 6730,
969 ISAC_RANGE_ERROR_DECODE_FRAME_LENGTH = 6640
970 };
971#if defined(WEBRTC_CODEC_ISAC)
972 EXPECT_EQ(ISAC_LENGTH_MISMATCH, neteq_->LastDecoderError());
973#elif defined(WEBRTC_CODEC_ISACFX)
974 EXPECT_EQ(ISAC_RANGE_ERROR_DECODE_FRAME_LENGTH, neteq_->LastDecoderError());
975#endif
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000976 // Verify that the first 160 samples are set to 0, and that the remaining
977 // samples are left unmodified.
978 static const int kExpectedOutputLength = 160; // 10 ms at 16 kHz sample rate.
979 for (int i = 0; i < kExpectedOutputLength; ++i) {
980 std::ostringstream ss;
981 ss << "i = " << i;
982 SCOPED_TRACE(ss.str()); // Print out the parameter values on failure.
983 EXPECT_EQ(0, out_data_[i]);
984 }
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000985 for (size_t i = kExpectedOutputLength; i < kMaxBlockSize; ++i) {
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000986 std::ostringstream ss;
987 ss << "i = " << i;
988 SCOPED_TRACE(ss.str()); // Print out the parameter values on failure.
989 EXPECT_EQ(1, out_data_[i]);
990 }
991}
992
henrik.lundin@webrtc.orgb4e80e02014-05-15 07:14:00 +0000993TEST_F(NetEqDecodingTest, GetAudioBeforeInsertPacket) {
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000994 NetEqOutputType type;
995 // Set all of |out_data_| to 1, and verify that it was set to 0 by the call
996 // to GetAudio.
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +0000997 for (size_t i = 0; i < kMaxBlockSize; ++i) {
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +0000998 out_data_[i] = 1;
999 }
Peter Kasting69558702016-01-12 16:26:35 -08001000 size_t num_channels;
Peter Kastingdce40cf2015-08-24 14:52:23 -07001001 size_t samples_per_channel;
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +00001002 EXPECT_EQ(0, neteq_->GetAudio(kMaxBlockSize, out_data_,
1003 &samples_per_channel,
1004 &num_channels, &type));
1005 // Verify that the first block of samples is set to 0.
1006 static const int kExpectedOutputLength =
1007 kInitSampleRateHz / 100; // 10 ms at initial sample rate.
1008 for (int i = 0; i < kExpectedOutputLength; ++i) {
1009 std::ostringstream ss;
1010 ss << "i = " << i;
1011 SCOPED_TRACE(ss.str()); // Print out the parameter values on failure.
1012 EXPECT_EQ(0, out_data_[i]);
1013 }
henrik.lundind89814b2015-11-23 06:49:25 -08001014 // Verify that the sample rate did not change from the initial configuration.
1015 EXPECT_EQ(config_.sample_rate_hz, neteq_->last_output_sample_rate_hz());
henrik.lundin@webrtc.orgd94659d2013-01-29 12:09:21 +00001016}
turaj@webrtc.orgff43c852013-09-25 00:07:27 +00001017
henrik.lundin@webrtc.org9b8102c2014-08-21 08:27:44 +00001018class NetEqBgnTest : public NetEqDecodingTest {
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +00001019 protected:
henrik.lundin@webrtc.org9b8102c2014-08-21 08:27:44 +00001020 virtual void TestCondition(double sum_squared_noise,
1021 bool should_be_faded) = 0;
turaj@webrtc.orgff43c852013-09-25 00:07:27 +00001022
henrik.lundin@webrtc.org9b8102c2014-08-21 08:27:44 +00001023 void CheckBgn(int sampling_rate_hz) {
Peter Kastingdce40cf2015-08-24 14:52:23 -07001024 size_t expected_samples_per_channel = 0;
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +00001025 uint8_t payload_type = 0xFF; // Invalid.
1026 if (sampling_rate_hz == 8000) {
1027 expected_samples_per_channel = kBlockSize8kHz;
1028 payload_type = 93; // PCM 16, 8 kHz.
1029 } else if (sampling_rate_hz == 16000) {
1030 expected_samples_per_channel = kBlockSize16kHz;
1031 payload_type = 94; // PCM 16, 16 kHZ.
1032 } else if (sampling_rate_hz == 32000) {
1033 expected_samples_per_channel = kBlockSize32kHz;
1034 payload_type = 95; // PCM 16, 32 kHz.
1035 } else {
1036 ASSERT_TRUE(false); // Unsupported test case.
1037 }
turaj@webrtc.orgff43c852013-09-25 00:07:27 +00001038
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +00001039 NetEqOutputType type;
1040 int16_t output[kBlockSize32kHz]; // Maximum size is chosen.
henrik.lundin@webrtc.org9b8102c2014-08-21 08:27:44 +00001041 test::AudioLoop input;
1042 // We are using the same 32 kHz input file for all tests, regardless of
1043 // |sampling_rate_hz|. The output may sound weird, but the test is still
1044 // valid.
1045 ASSERT_TRUE(input.Init(
1046 webrtc::test::ResourcePath("audio_coding/testfile32kHz", "pcm"),
1047 10 * sampling_rate_hz, // Max 10 seconds loop length.
Peter Kastingdce40cf2015-08-24 14:52:23 -07001048 expected_samples_per_channel));
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +00001049
1050 // Payload of 10 ms of PCM16 32 kHz.
1051 uint8_t payload[kBlockSize32kHz * sizeof(int16_t)];
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +00001052 WebRtcRTPHeader rtp_info;
1053 PopulateRtpInfo(0, 0, &rtp_info);
1054 rtp_info.header.payloadType = payload_type;
1055
Peter Kasting69558702016-01-12 16:26:35 -08001056 size_t number_channels = 0;
Peter Kastingdce40cf2015-08-24 14:52:23 -07001057 size_t samples_per_channel = 0;
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +00001058
1059 uint32_t receive_timestamp = 0;
1060 for (int n = 0; n < 10; ++n) { // Insert few packets and get audio.
kwiberg288886b2015-11-06 01:21:35 -08001061 auto block = input.GetNextBlock();
1062 ASSERT_EQ(expected_samples_per_channel, block.size());
1063 size_t enc_len_bytes =
1064 WebRtcPcm16b_Encode(block.data(), block.size(), payload);
henrik.lundin@webrtc.org9b8102c2014-08-21 08:27:44 +00001065 ASSERT_EQ(enc_len_bytes, expected_samples_per_channel * 2);
1066
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +00001067 number_channels = 0;
1068 samples_per_channel = 0;
kwibergee2bac22015-11-11 10:34:00 -08001069 ASSERT_EQ(0, neteq_->InsertPacket(rtp_info, rtc::ArrayView<const uint8_t>(
1070 payload, enc_len_bytes),
1071 receive_timestamp));
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +00001072 ASSERT_EQ(0,
1073 neteq_->GetAudio(kBlockSize32kHz,
1074 output,
1075 &samples_per_channel,
1076 &number_channels,
1077 &type));
Peter Kasting69558702016-01-12 16:26:35 -08001078 ASSERT_EQ(1u, number_channels);
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +00001079 ASSERT_EQ(expected_samples_per_channel, samples_per_channel);
1080 ASSERT_EQ(kOutputNormal, type);
1081
1082 // Next packet.
1083 rtp_info.header.timestamp += expected_samples_per_channel;
1084 rtp_info.header.sequenceNumber++;
1085 receive_timestamp += expected_samples_per_channel;
1086 }
1087
1088 number_channels = 0;
1089 samples_per_channel = 0;
1090
1091 // Get audio without inserting packets, expecting PLC and PLC-to-CNG. Pull
1092 // one frame without checking speech-type. This is the first frame pulled
1093 // without inserting any packet, and might not be labeled as PLC.
1094 ASSERT_EQ(0,
1095 neteq_->GetAudio(kBlockSize32kHz,
1096 output,
1097 &samples_per_channel,
1098 &number_channels,
1099 &type));
Peter Kasting69558702016-01-12 16:26:35 -08001100 ASSERT_EQ(1u, number_channels);
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +00001101 ASSERT_EQ(expected_samples_per_channel, samples_per_channel);
1102
1103 // To be able to test the fading of background noise we need at lease to
1104 // pull 611 frames.
1105 const int kFadingThreshold = 611;
1106
1107 // Test several CNG-to-PLC packet for the expected behavior. The number 20
1108 // is arbitrary, but sufficiently large to test enough number of frames.
1109 const int kNumPlcToCngTestFrames = 20;
1110 bool plc_to_cng = false;
1111 for (int n = 0; n < kFadingThreshold + kNumPlcToCngTestFrames; ++n) {
1112 number_channels = 0;
1113 samples_per_channel = 0;
1114 memset(output, 1, sizeof(output)); // Set to non-zero.
1115 ASSERT_EQ(0,
1116 neteq_->GetAudio(kBlockSize32kHz,
1117 output,
1118 &samples_per_channel,
1119 &number_channels,
1120 &type));
Peter Kasting69558702016-01-12 16:26:35 -08001121 ASSERT_EQ(1u, number_channels);
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +00001122 ASSERT_EQ(expected_samples_per_channel, samples_per_channel);
1123 if (type == kOutputPLCtoCNG) {
1124 plc_to_cng = true;
1125 double sum_squared = 0;
Peter Kastingdce40cf2015-08-24 14:52:23 -07001126 for (size_t k = 0; k < number_channels * samples_per_channel; ++k)
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +00001127 sum_squared += output[k] * output[k];
henrik.lundin@webrtc.org9b8102c2014-08-21 08:27:44 +00001128 TestCondition(sum_squared, n > kFadingThreshold);
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +00001129 } else {
1130 EXPECT_EQ(kOutputPLC, type);
1131 }
1132 }
1133 EXPECT_TRUE(plc_to_cng); // Just to be sure that PLC-to-CNG has occurred.
1134 }
1135};
1136
henrik.lundin@webrtc.org9b8102c2014-08-21 08:27:44 +00001137class NetEqBgnTestOn : public NetEqBgnTest {
1138 protected:
1139 NetEqBgnTestOn() : NetEqBgnTest() {
1140 config_.background_noise_mode = NetEq::kBgnOn;
1141 }
1142
1143 void TestCondition(double sum_squared_noise, bool /*should_be_faded*/) {
1144 EXPECT_NE(0, sum_squared_noise);
1145 }
1146};
1147
1148class NetEqBgnTestOff : public NetEqBgnTest {
1149 protected:
1150 NetEqBgnTestOff() : NetEqBgnTest() {
1151 config_.background_noise_mode = NetEq::kBgnOff;
1152 }
1153
1154 void TestCondition(double sum_squared_noise, bool /*should_be_faded*/) {
1155 EXPECT_EQ(0, sum_squared_noise);
1156 }
1157};
1158
1159class NetEqBgnTestFade : public NetEqBgnTest {
1160 protected:
1161 NetEqBgnTestFade() : NetEqBgnTest() {
1162 config_.background_noise_mode = NetEq::kBgnFade;
1163 }
1164
1165 void TestCondition(double sum_squared_noise, bool should_be_faded) {
1166 if (should_be_faded)
1167 EXPECT_EQ(0, sum_squared_noise);
1168 }
1169};
1170
henrika1d34fe92015-06-16 10:04:20 +02001171TEST_F(NetEqBgnTestOn, RunTest) {
henrik.lundin@webrtc.org9b8102c2014-08-21 08:27:44 +00001172 CheckBgn(8000);
1173 CheckBgn(16000);
1174 CheckBgn(32000);
turaj@webrtc.orgff43c852013-09-25 00:07:27 +00001175}
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001176
henrika1d34fe92015-06-16 10:04:20 +02001177TEST_F(NetEqBgnTestOff, RunTest) {
henrik.lundin@webrtc.org9b8102c2014-08-21 08:27:44 +00001178 CheckBgn(8000);
1179 CheckBgn(16000);
1180 CheckBgn(32000);
1181}
1182
henrika1d34fe92015-06-16 10:04:20 +02001183TEST_F(NetEqBgnTestFade, RunTest) {
henrik.lundin@webrtc.org9b8102c2014-08-21 08:27:44 +00001184 CheckBgn(8000);
1185 CheckBgn(16000);
1186 CheckBgn(32000);
1187}
henrik.lundin@webrtc.orgea257842014-08-07 12:27:37 +00001188
Peter Boströme2976c82016-01-04 22:44:05 +01001189#if defined(WEBRTC_CODEC_ISAC) || defined(WEBRTC_CODEC_ISACFX)
ivoc72c08ed2016-01-20 07:26:24 -08001190#define MAYBE_SyncPacketInsert SyncPacketInsert
1191#else
1192#define MAYBE_SyncPacketInsert DISABLED_SyncPacketInsert
1193#endif
1194TEST_F(NetEqDecodingTest, MAYBE_SyncPacketInsert) {
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001195 WebRtcRTPHeader rtp_info;
1196 uint32_t receive_timestamp = 0;
1197 // For the readability use the following payloads instead of the defaults of
1198 // this test.
1199 uint8_t kPcm16WbPayloadType = 1;
1200 uint8_t kCngNbPayloadType = 2;
1201 uint8_t kCngWbPayloadType = 3;
1202 uint8_t kCngSwb32PayloadType = 4;
1203 uint8_t kCngSwb48PayloadType = 5;
1204 uint8_t kAvtPayloadType = 6;
1205 uint8_t kRedPayloadType = 7;
1206 uint8_t kIsacPayloadType = 9; // Payload type 8 is already registered.
1207
1208 // Register decoders.
kwibergee1879c2015-10-29 06:20:28 -07001209 ASSERT_EQ(0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderPCM16Bwb,
henrik.lundin4cf61dd2015-12-09 06:20:58 -08001210 "pcm16-wb", kPcm16WbPayloadType));
kwibergee1879c2015-10-29 06:20:28 -07001211 ASSERT_EQ(0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderCNGnb,
henrik.lundin4cf61dd2015-12-09 06:20:58 -08001212 "cng-nb", kCngNbPayloadType));
kwibergee1879c2015-10-29 06:20:28 -07001213 ASSERT_EQ(0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderCNGwb,
henrik.lundin4cf61dd2015-12-09 06:20:58 -08001214 "cng-wb", kCngWbPayloadType));
kwibergee1879c2015-10-29 06:20:28 -07001215 ASSERT_EQ(0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderCNGswb32kHz,
henrik.lundin4cf61dd2015-12-09 06:20:58 -08001216 "cng-swb32", kCngSwb32PayloadType));
kwibergee1879c2015-10-29 06:20:28 -07001217 ASSERT_EQ(0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderCNGswb48kHz,
henrik.lundin4cf61dd2015-12-09 06:20:58 -08001218 "cng-swb48", kCngSwb48PayloadType));
1219 ASSERT_EQ(0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderAVT, "avt",
kwibergee1879c2015-10-29 06:20:28 -07001220 kAvtPayloadType));
henrik.lundin4cf61dd2015-12-09 06:20:58 -08001221 ASSERT_EQ(0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderRED, "red",
kwibergee1879c2015-10-29 06:20:28 -07001222 kRedPayloadType));
henrik.lundin4cf61dd2015-12-09 06:20:58 -08001223 ASSERT_EQ(0, neteq_->RegisterPayloadType(NetEqDecoder::kDecoderISAC, "isac",
kwibergee1879c2015-10-29 06:20:28 -07001224 kIsacPayloadType));
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001225
1226 PopulateRtpInfo(0, 0, &rtp_info);
1227 rtp_info.header.payloadType = kPcm16WbPayloadType;
1228
1229 // The first packet injected cannot be sync-packet.
1230 EXPECT_EQ(-1, neteq_->InsertSyncPacket(rtp_info, receive_timestamp));
1231
1232 // Payload length of 10 ms PCM16 16 kHz.
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +00001233 const size_t kPayloadBytes = kBlockSize16kHz * sizeof(int16_t);
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001234 uint8_t payload[kPayloadBytes] = {0};
kwibergee2bac22015-11-11 10:34:00 -08001235 ASSERT_EQ(0, neteq_->InsertPacket(rtp_info, payload, receive_timestamp));
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001236
1237 // Next packet. Last packet contained 10 ms audio.
1238 rtp_info.header.sequenceNumber++;
1239 rtp_info.header.timestamp += kBlockSize16kHz;
1240 receive_timestamp += kBlockSize16kHz;
1241
1242 // Unacceptable payload types CNG, AVT (DTMF), RED.
1243 rtp_info.header.payloadType = kCngNbPayloadType;
1244 EXPECT_EQ(-1, neteq_->InsertSyncPacket(rtp_info, receive_timestamp));
1245
1246 rtp_info.header.payloadType = kCngWbPayloadType;
1247 EXPECT_EQ(-1, neteq_->InsertSyncPacket(rtp_info, receive_timestamp));
1248
1249 rtp_info.header.payloadType = kCngSwb32PayloadType;
1250 EXPECT_EQ(-1, neteq_->InsertSyncPacket(rtp_info, receive_timestamp));
1251
1252 rtp_info.header.payloadType = kCngSwb48PayloadType;
1253 EXPECT_EQ(-1, neteq_->InsertSyncPacket(rtp_info, receive_timestamp));
1254
1255 rtp_info.header.payloadType = kAvtPayloadType;
1256 EXPECT_EQ(-1, neteq_->InsertSyncPacket(rtp_info, receive_timestamp));
1257
1258 rtp_info.header.payloadType = kRedPayloadType;
1259 EXPECT_EQ(-1, neteq_->InsertSyncPacket(rtp_info, receive_timestamp));
1260
1261 // Change of codec cannot be initiated with a sync packet.
1262 rtp_info.header.payloadType = kIsacPayloadType;
1263 EXPECT_EQ(-1, neteq_->InsertSyncPacket(rtp_info, receive_timestamp));
1264
1265 // Change of SSRC is not allowed with a sync packet.
1266 rtp_info.header.payloadType = kPcm16WbPayloadType;
1267 ++rtp_info.header.ssrc;
1268 EXPECT_EQ(-1, neteq_->InsertSyncPacket(rtp_info, receive_timestamp));
1269
1270 --rtp_info.header.ssrc;
1271 EXPECT_EQ(0, neteq_->InsertSyncPacket(rtp_info, receive_timestamp));
1272}
1273
1274// First insert several noise like packets, then sync-packets. Decoding all
1275// packets should not produce error, statistics should not show any packet loss
1276// and sync-packets should decode to zero.
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +00001277// TODO(turajs) we will have a better test if we have a referece NetEq, and
1278// when Sync packets are inserted in "test" NetEq we insert all-zero payload
1279// in reference NetEq and compare the output of those two.
henrik.lundin@webrtc.orgb4e80e02014-05-15 07:14:00 +00001280TEST_F(NetEqDecodingTest, SyncPacketDecode) {
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001281 WebRtcRTPHeader rtp_info;
1282 PopulateRtpInfo(0, 0, &rtp_info);
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +00001283 const size_t kPayloadBytes = kBlockSize16kHz * sizeof(int16_t);
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001284 uint8_t payload[kPayloadBytes];
1285 int16_t decoded[kBlockSize16kHz];
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +00001286 int algorithmic_frame_delay = algorithmic_delay_ms_ / 10 + 1;
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +00001287 for (size_t n = 0; n < kPayloadBytes; ++n) {
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001288 payload[n] = (rand() & 0xF0) + 1; // Non-zero random sequence.
1289 }
1290 // Insert some packets which decode to noise. We are not interested in
1291 // actual decoded values.
1292 NetEqOutputType output_type;
Peter Kasting69558702016-01-12 16:26:35 -08001293 size_t num_channels;
Peter Kastingdce40cf2015-08-24 14:52:23 -07001294 size_t samples_per_channel;
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001295 uint32_t receive_timestamp = 0;
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001296 for (int n = 0; n < 100; ++n) {
kwibergee2bac22015-11-11 10:34:00 -08001297 ASSERT_EQ(0, neteq_->InsertPacket(rtp_info, payload, receive_timestamp));
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001298 ASSERT_EQ(0, neteq_->GetAudio(kBlockSize16kHz, decoded,
1299 &samples_per_channel, &num_channels,
1300 &output_type));
1301 ASSERT_EQ(kBlockSize16kHz, samples_per_channel);
Peter Kasting69558702016-01-12 16:26:35 -08001302 ASSERT_EQ(1u, num_channels);
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001303
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001304 rtp_info.header.sequenceNumber++;
1305 rtp_info.header.timestamp += kBlockSize16kHz;
1306 receive_timestamp += kBlockSize16kHz;
1307 }
1308 const int kNumSyncPackets = 10;
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +00001309
1310 // Make sure sufficient number of sync packets are inserted that we can
1311 // conduct a test.
1312 ASSERT_GT(kNumSyncPackets, algorithmic_frame_delay);
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001313 // Insert sync-packets, the decoded sequence should be all-zero.
1314 for (int n = 0; n < kNumSyncPackets; ++n) {
1315 ASSERT_EQ(0, neteq_->InsertSyncPacket(rtp_info, receive_timestamp));
1316 ASSERT_EQ(0, neteq_->GetAudio(kBlockSize16kHz, decoded,
1317 &samples_per_channel, &num_channels,
1318 &output_type));
1319 ASSERT_EQ(kBlockSize16kHz, samples_per_channel);
Peter Kasting69558702016-01-12 16:26:35 -08001320 ASSERT_EQ(1u, num_channels);
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +00001321 if (n > algorithmic_frame_delay) {
1322 EXPECT_TRUE(IsAllZero(decoded, samples_per_channel * num_channels));
1323 }
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001324 rtp_info.header.sequenceNumber++;
1325 rtp_info.header.timestamp += kBlockSize16kHz;
1326 receive_timestamp += kBlockSize16kHz;
1327 }
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +00001328
1329 // We insert regular packets, if sync packet are not correctly buffered then
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001330 // network statistics would show some packet loss.
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +00001331 for (int n = 0; n <= algorithmic_frame_delay + 10; ++n) {
kwibergee2bac22015-11-11 10:34:00 -08001332 ASSERT_EQ(0, neteq_->InsertPacket(rtp_info, payload, receive_timestamp));
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +00001333 ASSERT_EQ(0, neteq_->GetAudio(kBlockSize16kHz, decoded,
1334 &samples_per_channel, &num_channels,
1335 &output_type));
1336 if (n >= algorithmic_frame_delay + 1) {
1337 // Expect that this frame contain samples from regular RTP.
1338 EXPECT_TRUE(IsAllNonZero(decoded, samples_per_channel * num_channels));
1339 }
1340 rtp_info.header.sequenceNumber++;
1341 rtp_info.header.timestamp += kBlockSize16kHz;
1342 receive_timestamp += kBlockSize16kHz;
1343 }
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001344 NetEqNetworkStatistics network_stats;
1345 ASSERT_EQ(0, neteq_->NetworkStatistics(&network_stats));
1346 // Expecting a "clean" network.
1347 EXPECT_EQ(0, network_stats.packet_loss_rate);
1348 EXPECT_EQ(0, network_stats.expand_rate);
1349 EXPECT_EQ(0, network_stats.accelerate_rate);
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +00001350 EXPECT_LE(network_stats.preemptive_rate, 150);
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001351}
1352
1353// Test if the size of the packet buffer reported correctly when containing
1354// sync packets. Also, test if network packets override sync packets. That is to
1355// prefer decoding a network packet to a sync packet, if both have same sequence
1356// number and timestamp.
henrik.lundin@webrtc.orgb4e80e02014-05-15 07:14:00 +00001357TEST_F(NetEqDecodingTest, SyncPacketBufferSizeAndOverridenByNetworkPackets) {
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001358 WebRtcRTPHeader rtp_info;
1359 PopulateRtpInfo(0, 0, &rtp_info);
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +00001360 const size_t kPayloadBytes = kBlockSize16kHz * sizeof(int16_t);
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001361 uint8_t payload[kPayloadBytes];
1362 int16_t decoded[kBlockSize16kHz];
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +00001363 for (size_t n = 0; n < kPayloadBytes; ++n) {
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001364 payload[n] = (rand() & 0xF0) + 1; // Non-zero random sequence.
1365 }
1366 // Insert some packets which decode to noise. We are not interested in
1367 // actual decoded values.
1368 NetEqOutputType output_type;
Peter Kasting69558702016-01-12 16:26:35 -08001369 size_t num_channels;
Peter Kastingdce40cf2015-08-24 14:52:23 -07001370 size_t samples_per_channel;
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001371 uint32_t receive_timestamp = 0;
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +00001372 int algorithmic_frame_delay = algorithmic_delay_ms_ / 10 + 1;
1373 for (int n = 0; n < algorithmic_frame_delay; ++n) {
kwibergee2bac22015-11-11 10:34:00 -08001374 ASSERT_EQ(0, neteq_->InsertPacket(rtp_info, payload, receive_timestamp));
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001375 ASSERT_EQ(0, neteq_->GetAudio(kBlockSize16kHz, decoded,
1376 &samples_per_channel, &num_channels,
1377 &output_type));
1378 ASSERT_EQ(kBlockSize16kHz, samples_per_channel);
Peter Kasting69558702016-01-12 16:26:35 -08001379 ASSERT_EQ(1u, num_channels);
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001380 rtp_info.header.sequenceNumber++;
1381 rtp_info.header.timestamp += kBlockSize16kHz;
1382 receive_timestamp += kBlockSize16kHz;
1383 }
1384 const int kNumSyncPackets = 10;
1385
1386 WebRtcRTPHeader first_sync_packet_rtp_info;
1387 memcpy(&first_sync_packet_rtp_info, &rtp_info, sizeof(rtp_info));
1388
1389 // Insert sync-packets, but no decoding.
1390 for (int n = 0; n < kNumSyncPackets; ++n) {
1391 ASSERT_EQ(0, neteq_->InsertSyncPacket(rtp_info, receive_timestamp));
1392 rtp_info.header.sequenceNumber++;
1393 rtp_info.header.timestamp += kBlockSize16kHz;
1394 receive_timestamp += kBlockSize16kHz;
1395 }
1396 NetEqNetworkStatistics network_stats;
1397 ASSERT_EQ(0, neteq_->NetworkStatistics(&network_stats));
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +00001398 EXPECT_EQ(kNumSyncPackets * 10 + algorithmic_delay_ms_,
1399 network_stats.current_buffer_size_ms);
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001400
1401 // Rewind |rtp_info| to that of the first sync packet.
1402 memcpy(&rtp_info, &first_sync_packet_rtp_info, sizeof(rtp_info));
1403
1404 // Insert.
1405 for (int n = 0; n < kNumSyncPackets; ++n) {
kwibergee2bac22015-11-11 10:34:00 -08001406 ASSERT_EQ(0, neteq_->InsertPacket(rtp_info, payload, receive_timestamp));
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001407 rtp_info.header.sequenceNumber++;
1408 rtp_info.header.timestamp += kBlockSize16kHz;
1409 receive_timestamp += kBlockSize16kHz;
1410 }
1411
1412 // Decode.
1413 for (int n = 0; n < kNumSyncPackets; ++n) {
1414 ASSERT_EQ(0, neteq_->GetAudio(kBlockSize16kHz, decoded,
1415 &samples_per_channel, &num_channels,
1416 &output_type));
1417 ASSERT_EQ(kBlockSize16kHz, samples_per_channel);
Peter Kasting69558702016-01-12 16:26:35 -08001418 ASSERT_EQ(1u, num_channels);
turaj@webrtc.org7b75ac62013-09-26 00:27:56 +00001419 EXPECT_TRUE(IsAllNonZero(decoded, samples_per_channel * num_channels));
1420 }
1421}
1422
turaj@webrtc.org78b41a02013-11-22 20:27:07 +00001423void NetEqDecodingTest::WrapTest(uint16_t start_seq_no,
1424 uint32_t start_timestamp,
1425 const std::set<uint16_t>& drop_seq_numbers,
1426 bool expect_seq_no_wrap,
1427 bool expect_timestamp_wrap) {
1428 uint16_t seq_no = start_seq_no;
1429 uint32_t timestamp = start_timestamp;
1430 const int kBlocksPerFrame = 3; // Number of 10 ms blocks per frame.
1431 const int kFrameSizeMs = kBlocksPerFrame * kTimeStepMs;
1432 const int kSamples = kBlockSize16kHz * kBlocksPerFrame;
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +00001433 const size_t kPayloadBytes = kSamples * sizeof(int16_t);
turaj@webrtc.org78b41a02013-11-22 20:27:07 +00001434 double next_input_time_ms = 0.0;
1435 int16_t decoded[kBlockSize16kHz];
Peter Kasting69558702016-01-12 16:26:35 -08001436 size_t num_channels;
Peter Kastingdce40cf2015-08-24 14:52:23 -07001437 size_t samples_per_channel;
turaj@webrtc.org78b41a02013-11-22 20:27:07 +00001438 NetEqOutputType output_type;
1439 uint32_t receive_timestamp = 0;
1440
henrik.lundin@webrtc.orgca8cb952014-03-12 10:26:52 +00001441 // Insert speech for 2 seconds.
turaj@webrtc.org78b41a02013-11-22 20:27:07 +00001442 const int kSpeechDurationMs = 2000;
1443 int packets_inserted = 0;
1444 uint16_t last_seq_no;
1445 uint32_t last_timestamp;
1446 bool timestamp_wrapped = false;
1447 bool seq_no_wrapped = false;
1448 for (double t_ms = 0; t_ms < kSpeechDurationMs; t_ms += 10) {
1449 // Each turn in this for loop is 10 ms.
1450 while (next_input_time_ms <= t_ms) {
1451 // Insert one 30 ms speech frame.
1452 uint8_t payload[kPayloadBytes] = {0};
1453 WebRtcRTPHeader rtp_info;
1454 PopulateRtpInfo(seq_no, timestamp, &rtp_info);
1455 if (drop_seq_numbers.find(seq_no) == drop_seq_numbers.end()) {
1456 // This sequence number was not in the set to drop. Insert it.
1457 ASSERT_EQ(0,
kwibergee2bac22015-11-11 10:34:00 -08001458 neteq_->InsertPacket(rtp_info, payload, receive_timestamp));
turaj@webrtc.org78b41a02013-11-22 20:27:07 +00001459 ++packets_inserted;
1460 }
1461 NetEqNetworkStatistics network_stats;
1462 ASSERT_EQ(0, neteq_->NetworkStatistics(&network_stats));
1463
1464 // Due to internal NetEq logic, preferred buffer-size is about 4 times the
1465 // packet size for first few packets. Therefore we refrain from checking
1466 // the criteria.
1467 if (packets_inserted > 4) {
1468 // Expect preferred and actual buffer size to be no more than 2 frames.
1469 EXPECT_LE(network_stats.preferred_buffer_size_ms, kFrameSizeMs * 2);
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +00001470 EXPECT_LE(network_stats.current_buffer_size_ms, kFrameSizeMs * 2 +
1471 algorithmic_delay_ms_);
turaj@webrtc.org78b41a02013-11-22 20:27:07 +00001472 }
1473 last_seq_no = seq_no;
1474 last_timestamp = timestamp;
1475
1476 ++seq_no;
1477 timestamp += kSamples;
1478 receive_timestamp += kSamples;
1479 next_input_time_ms += static_cast<double>(kFrameSizeMs);
1480
1481 seq_no_wrapped |= seq_no < last_seq_no;
1482 timestamp_wrapped |= timestamp < last_timestamp;
1483 }
1484 // Pull out data once.
1485 ASSERT_EQ(0, neteq_->GetAudio(kBlockSize16kHz, decoded,
1486 &samples_per_channel, &num_channels,
1487 &output_type));
1488 ASSERT_EQ(kBlockSize16kHz, samples_per_channel);
Peter Kasting69558702016-01-12 16:26:35 -08001489 ASSERT_EQ(1u, num_channels);
turaj@webrtc.org78b41a02013-11-22 20:27:07 +00001490
1491 // Expect delay (in samples) to be less than 2 packets.
wu@webrtc.org94454b72014-06-05 20:34:08 +00001492 EXPECT_LE(timestamp - PlayoutTimestamp(),
turaj@webrtc.org78b41a02013-11-22 20:27:07 +00001493 static_cast<uint32_t>(kSamples * 2));
turaj@webrtc.org78b41a02013-11-22 20:27:07 +00001494 }
1495 // Make sure we have actually tested wrap-around.
1496 ASSERT_EQ(expect_seq_no_wrap, seq_no_wrapped);
1497 ASSERT_EQ(expect_timestamp_wrap, timestamp_wrapped);
1498}
1499
1500TEST_F(NetEqDecodingTest, SequenceNumberWrap) {
1501 // Start with a sequence number that will soon wrap.
1502 std::set<uint16_t> drop_seq_numbers; // Don't drop any packets.
1503 WrapTest(0xFFFF - 10, 0, drop_seq_numbers, true, false);
1504}
1505
1506TEST_F(NetEqDecodingTest, SequenceNumberWrapAndDrop) {
1507 // Start with a sequence number that will soon wrap.
1508 std::set<uint16_t> drop_seq_numbers;
1509 drop_seq_numbers.insert(0xFFFF);
1510 drop_seq_numbers.insert(0x0);
1511 WrapTest(0xFFFF - 10, 0, drop_seq_numbers, true, false);
1512}
1513
1514TEST_F(NetEqDecodingTest, TimestampWrap) {
1515 // Start with a timestamp that will soon wrap.
1516 std::set<uint16_t> drop_seq_numbers;
1517 WrapTest(0, 0xFFFFFFFF - 3000, drop_seq_numbers, false, true);
1518}
1519
1520TEST_F(NetEqDecodingTest, TimestampAndSequenceNumberWrap) {
1521 // Start with a timestamp and a sequence number that will wrap at the same
1522 // time.
1523 std::set<uint16_t> drop_seq_numbers;
1524 WrapTest(0xFFFF - 10, 0xFFFFFFFF - 5000, drop_seq_numbers, true, true);
1525}
1526
henrik.lundin@webrtc.orgca8cb952014-03-12 10:26:52 +00001527void NetEqDecodingTest::DuplicateCng() {
1528 uint16_t seq_no = 0;
1529 uint32_t timestamp = 0;
1530 const int kFrameSizeMs = 10;
1531 const int kSampleRateKhz = 16;
1532 const int kSamples = kFrameSizeMs * kSampleRateKhz;
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +00001533 const size_t kPayloadBytes = kSamples * 2;
henrik.lundin@webrtc.orgca8cb952014-03-12 10:26:52 +00001534
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +00001535 const int algorithmic_delay_samples = std::max(
1536 algorithmic_delay_ms_ * kSampleRateKhz, 5 * kSampleRateKhz / 8);
henrik.lundin@webrtc.orgc93437e2014-12-01 11:42:42 +00001537 // Insert three speech packets. Three are needed to get the frame length
henrik.lundin@webrtc.orgca8cb952014-03-12 10:26:52 +00001538 // correct.
Peter Kastingdce40cf2015-08-24 14:52:23 -07001539 size_t out_len;
Peter Kasting69558702016-01-12 16:26:35 -08001540 size_t num_channels;
henrik.lundin@webrtc.orgca8cb952014-03-12 10:26:52 +00001541 NetEqOutputType type;
1542 uint8_t payload[kPayloadBytes] = {0};
1543 WebRtcRTPHeader rtp_info;
1544 for (int i = 0; i < 3; ++i) {
1545 PopulateRtpInfo(seq_no, timestamp, &rtp_info);
kwibergee2bac22015-11-11 10:34:00 -08001546 ASSERT_EQ(0, neteq_->InsertPacket(rtp_info, payload, 0));
henrik.lundin@webrtc.orgca8cb952014-03-12 10:26:52 +00001547 ++seq_no;
1548 timestamp += kSamples;
1549
1550 // Pull audio once.
1551 ASSERT_EQ(0,
1552 neteq_->GetAudio(
1553 kMaxBlockSize, out_data_, &out_len, &num_channels, &type));
1554 ASSERT_EQ(kBlockSize16kHz, out_len);
1555 }
1556 // Verify speech output.
1557 EXPECT_EQ(kOutputNormal, type);
1558
1559 // Insert same CNG packet twice.
1560 const int kCngPeriodMs = 100;
1561 const int kCngPeriodSamples = kCngPeriodMs * kSampleRateKhz;
pkasting@chromium.org4591fbd2014-11-20 22:28:14 +00001562 size_t payload_len;
henrik.lundin@webrtc.orgca8cb952014-03-12 10:26:52 +00001563 PopulateCng(seq_no, timestamp, &rtp_info, payload, &payload_len);
1564 // This is the first time this CNG packet is inserted.
kwibergee2bac22015-11-11 10:34:00 -08001565 ASSERT_EQ(
1566 0, neteq_->InsertPacket(
1567 rtp_info, rtc::ArrayView<const uint8_t>(payload, payload_len), 0));
henrik.lundin@webrtc.orgca8cb952014-03-12 10:26:52 +00001568
1569 // Pull audio once and make sure CNG is played.
1570 ASSERT_EQ(0,
1571 neteq_->GetAudio(
1572 kMaxBlockSize, out_data_, &out_len, &num_channels, &type));
1573 ASSERT_EQ(kBlockSize16kHz, out_len);
1574 EXPECT_EQ(kOutputCNG, type);
wu@webrtc.org94454b72014-06-05 20:34:08 +00001575 EXPECT_EQ(timestamp - algorithmic_delay_samples, PlayoutTimestamp());
henrik.lundin@webrtc.orgca8cb952014-03-12 10:26:52 +00001576
1577 // Insert the same CNG packet again. Note that at this point it is old, since
1578 // we have already decoded the first copy of it.
kwibergee2bac22015-11-11 10:34:00 -08001579 ASSERT_EQ(
1580 0, neteq_->InsertPacket(
1581 rtp_info, rtc::ArrayView<const uint8_t>(payload, payload_len), 0));
henrik.lundin@webrtc.orgca8cb952014-03-12 10:26:52 +00001582
1583 // Pull audio until we have played |kCngPeriodMs| of CNG. Start at 10 ms since
1584 // we have already pulled out CNG once.
1585 for (int cng_time_ms = 10; cng_time_ms < kCngPeriodMs; cng_time_ms += 10) {
1586 ASSERT_EQ(0,
1587 neteq_->GetAudio(
1588 kMaxBlockSize, out_data_, &out_len, &num_channels, &type));
1589 ASSERT_EQ(kBlockSize16kHz, out_len);
1590 EXPECT_EQ(kOutputCNG, type);
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +00001591 EXPECT_EQ(timestamp - algorithmic_delay_samples,
wu@webrtc.org94454b72014-06-05 20:34:08 +00001592 PlayoutTimestamp());
henrik.lundin@webrtc.orgca8cb952014-03-12 10:26:52 +00001593 }
1594
1595 // Insert speech again.
1596 ++seq_no;
1597 timestamp += kCngPeriodSamples;
1598 PopulateRtpInfo(seq_no, timestamp, &rtp_info);
kwibergee2bac22015-11-11 10:34:00 -08001599 ASSERT_EQ(0, neteq_->InsertPacket(rtp_info, payload, 0));
henrik.lundin@webrtc.orgca8cb952014-03-12 10:26:52 +00001600
1601 // Pull audio once and verify that the output is speech again.
1602 ASSERT_EQ(0,
1603 neteq_->GetAudio(
1604 kMaxBlockSize, out_data_, &out_len, &num_channels, &type));
1605 ASSERT_EQ(kBlockSize16kHz, out_len);
1606 EXPECT_EQ(kOutputNormal, type);
turaj@webrtc.org8d1cdaa2014-04-11 18:47:55 +00001607 EXPECT_EQ(timestamp + kSamples - algorithmic_delay_samples,
wu@webrtc.org94454b72014-06-05 20:34:08 +00001608 PlayoutTimestamp());
1609}
1610
1611uint32_t NetEqDecodingTest::PlayoutTimestamp() {
1612 uint32_t playout_timestamp = 0;
1613 EXPECT_TRUE(neteq_->GetPlayoutTimestamp(&playout_timestamp));
1614 return playout_timestamp;
henrik.lundin@webrtc.orgca8cb952014-03-12 10:26:52 +00001615}
1616
1617TEST_F(NetEqDecodingTest, DiscardDuplicateCng) { DuplicateCng(); }
henrik.lundin@webrtc.orgc93437e2014-12-01 11:42:42 +00001618
1619TEST_F(NetEqDecodingTest, CngFirst) {
1620 uint16_t seq_no = 0;
1621 uint32_t timestamp = 0;
1622 const int kFrameSizeMs = 10;
1623 const int kSampleRateKhz = 16;
1624 const int kSamples = kFrameSizeMs * kSampleRateKhz;
1625 const int kPayloadBytes = kSamples * 2;
1626 const int kCngPeriodMs = 100;
1627 const int kCngPeriodSamples = kCngPeriodMs * kSampleRateKhz;
1628 size_t payload_len;
1629
1630 uint8_t payload[kPayloadBytes] = {0};
1631 WebRtcRTPHeader rtp_info;
1632
1633 PopulateCng(seq_no, timestamp, &rtp_info, payload, &payload_len);
kwibergee2bac22015-11-11 10:34:00 -08001634 ASSERT_EQ(
1635 NetEq::kOK,
1636 neteq_->InsertPacket(
1637 rtp_info, rtc::ArrayView<const uint8_t>(payload, payload_len), 0));
henrik.lundin@webrtc.orgc93437e2014-12-01 11:42:42 +00001638 ++seq_no;
1639 timestamp += kCngPeriodSamples;
1640
1641 // Pull audio once and make sure CNG is played.
Peter Kastingdce40cf2015-08-24 14:52:23 -07001642 size_t out_len;
Peter Kasting69558702016-01-12 16:26:35 -08001643 size_t num_channels;
henrik.lundin@webrtc.orgc93437e2014-12-01 11:42:42 +00001644 NetEqOutputType type;
1645 ASSERT_EQ(0, neteq_->GetAudio(kMaxBlockSize, out_data_, &out_len,
1646 &num_channels, &type));
1647 ASSERT_EQ(kBlockSize16kHz, out_len);
1648 EXPECT_EQ(kOutputCNG, type);
1649
1650 // Insert some speech packets.
1651 for (int i = 0; i < 3; ++i) {
1652 PopulateRtpInfo(seq_no, timestamp, &rtp_info);
kwibergee2bac22015-11-11 10:34:00 -08001653 ASSERT_EQ(0, neteq_->InsertPacket(rtp_info, payload, 0));
henrik.lundin@webrtc.orgc93437e2014-12-01 11:42:42 +00001654 ++seq_no;
1655 timestamp += kSamples;
1656
1657 // Pull audio once.
1658 ASSERT_EQ(0, neteq_->GetAudio(kMaxBlockSize, out_data_, &out_len,
1659 &num_channels, &type));
1660 ASSERT_EQ(kBlockSize16kHz, out_len);
1661 }
1662 // Verify speech output.
1663 EXPECT_EQ(kOutputNormal, type);
1664}
1665
henrik.lundin@webrtc.orge7ce4372014-01-09 14:01:55 +00001666} // namespace webrtc