tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2013 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 | */ |
kwiberg | 91d9756 | 2016-02-14 01:10:03 -0800 | [diff] [blame] | 10 | |
| 11 | #include <memory> |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 12 | #include <string> |
| 13 | |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 14 | #include "modules/audio_coding/codecs/opus/opus_inst.h" |
| 15 | #include "modules/audio_coding/codecs/opus/opus_interface.h" |
| 16 | #include "modules/audio_coding/neteq/tools/audio_loop.h" |
| 17 | #include "rtc_base/checks.h" |
Karl Wiberg | e40468b | 2017-11-22 10:42:26 +0100 | [diff] [blame] | 18 | #include "rtc_base/numerics/safe_conversions.h" |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 19 | #include "test/gtest.h" |
Steve Anton | 10542f2 | 2019-01-11 09:11:00 -0800 | [diff] [blame] | 20 | #include "test/testsupport/file_utils.h" |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 21 | |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 22 | namespace webrtc { |
| 23 | |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 24 | using test::AudioLoop; |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 25 | using ::testing::TestWithParam; |
| 26 | using ::testing::Values; |
| 27 | using ::testing::Combine; |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 28 | |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 29 | // Maximum number of bytes in output bitstream. |
Alex Loiko | 7a3e43a | 2019-01-29 12:27:08 +0100 | [diff] [blame] | 30 | const size_t kMaxBytes = 2000; |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 31 | // Sample rate of Opus. |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 32 | const size_t kOpusRateKhz = 48; |
minyue@webrtc.org | f563e85 | 2014-07-18 21:11:27 +0000 | [diff] [blame] | 33 | // Number of samples-per-channel in a 20 ms frame, sampled at 48 kHz. |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 34 | const size_t kOpus20msFrameSamples = kOpusRateKhz * 20; |
minyue@webrtc.org | f563e85 | 2014-07-18 21:11:27 +0000 | [diff] [blame] | 35 | // Number of samples-per-channel in a 10 ms frame, sampled at 48 kHz. |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 36 | const size_t kOpus10msFrameSamples = kOpusRateKhz * 10; |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 37 | |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 38 | class OpusTest : public TestWithParam<::testing::tuple<int, int>> { |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 39 | protected: |
| 40 | OpusTest(); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 41 | |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 42 | void TestDtxEffect(bool dtx, int block_length_ms); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 43 | |
soren | 28dc285 | 2017-04-06 05:48:36 -0700 | [diff] [blame] | 44 | void TestCbrEffect(bool dtx, int block_length_ms); |
| 45 | |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 46 | // Prepare |speech_data_| for encoding, read from a hard-coded file. |
| 47 | // After preparation, |speech_data_.GetNextBlock()| returns a pointer to a |
| 48 | // block of |block_length_ms| milliseconds. The data is looped every |
| 49 | // |loop_length_ms| milliseconds. |
Peter Kasting | 6955870 | 2016-01-12 16:26:35 -0800 | [diff] [blame] | 50 | void PrepareSpeechData(size_t channel, |
| 51 | int block_length_ms, |
| 52 | int loop_length_ms); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 53 | |
| 54 | int EncodeDecode(WebRtcOpusEncInst* encoder, |
kwiberg | 288886b | 2015-11-06 01:21:35 -0800 | [diff] [blame] | 55 | rtc::ArrayView<const int16_t> input_audio, |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 56 | WebRtcOpusDecInst* decoder, |
| 57 | int16_t* output_audio, |
| 58 | int16_t* audio_type); |
minyue@webrtc.org | 0040a6e | 2014-08-04 14:41:57 +0000 | [diff] [blame] | 59 | |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 60 | void SetMaxPlaybackRate(WebRtcOpusEncInst* encoder, |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 61 | opus_int32 expect, |
| 62 | int32_t set); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 63 | |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 64 | void CheckAudioBounded(const int16_t* audio, |
| 65 | size_t samples, |
| 66 | size_t channels, |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 67 | uint16_t bound) const; |
| 68 | |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 69 | WebRtcOpusEncInst* opus_encoder_; |
| 70 | WebRtcOpusDecInst* opus_decoder_; |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 71 | |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 72 | AudioLoop speech_data_; |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 73 | uint8_t bitstream_[kMaxBytes]; |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 74 | size_t encoded_bytes_; |
Peter Kasting | 6955870 | 2016-01-12 16:26:35 -0800 | [diff] [blame] | 75 | size_t channels_; |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 76 | int application_; |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 77 | }; |
| 78 | |
| 79 | OpusTest::OpusTest() |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 80 | : opus_encoder_(NULL), |
| 81 | opus_decoder_(NULL), |
| 82 | encoded_bytes_(0), |
Peter Kasting | 6955870 | 2016-01-12 16:26:35 -0800 | [diff] [blame] | 83 | channels_(static_cast<size_t>(::testing::get<0>(GetParam()))), |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 84 | application_(::testing::get<1>(GetParam())) {} |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 85 | |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 86 | void OpusTest::PrepareSpeechData(size_t channel, |
| 87 | int block_length_ms, |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 88 | int loop_length_ms) { |
Alex Loiko | 7a3e43a | 2019-01-29 12:27:08 +0100 | [diff] [blame] | 89 | std::map<int, std::string> channel_to_basename = { |
| 90 | {1, "audio_coding/testfile32kHz"}, |
| 91 | {2, "audio_coding/teststereo32kHz"}, |
| 92 | {4, "audio_coding/speech_4_channels_48k_one_second"}}; |
| 93 | std::map<int, std::string> channel_to_suffix = { |
| 94 | {1, "pcm"}, {2, "pcm"}, {4, "wav"}}; |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 95 | const std::string file_name = webrtc::test::ResourcePath( |
Alex Loiko | 7a3e43a | 2019-01-29 12:27:08 +0100 | [diff] [blame] | 96 | channel_to_basename[channel], channel_to_suffix[channel]); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 97 | if (loop_length_ms < block_length_ms) { |
| 98 | loop_length_ms = block_length_ms; |
| 99 | } |
| 100 | EXPECT_TRUE(speech_data_.Init(file_name, |
| 101 | loop_length_ms * kOpusRateKhz * channel, |
| 102 | block_length_ms * kOpusRateKhz * channel)); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 103 | } |
| 104 | |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 105 | void OpusTest::SetMaxPlaybackRate(WebRtcOpusEncInst* encoder, |
| 106 | opus_int32 expect, |
| 107 | int32_t set) { |
minyue@webrtc.org | 0040a6e | 2014-08-04 14:41:57 +0000 | [diff] [blame] | 108 | opus_int32 bandwidth; |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 109 | EXPECT_EQ(0, WebRtcOpus_SetMaxPlaybackRate(opus_encoder_, set)); |
Alex Loiko | 7a3e43a | 2019-01-29 12:27:08 +0100 | [diff] [blame] | 110 | EXPECT_EQ(0, WebRtcOpus_GetMaxPlaybackRate(opus_encoder_, &bandwidth)); |
minyue@webrtc.org | 0040a6e | 2014-08-04 14:41:57 +0000 | [diff] [blame] | 111 | EXPECT_EQ(expect, bandwidth); |
| 112 | } |
| 113 | |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 114 | void OpusTest::CheckAudioBounded(const int16_t* audio, |
| 115 | size_t samples, |
| 116 | size_t channels, |
| 117 | uint16_t bound) const { |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 118 | for (size_t i = 0; i < samples; ++i) { |
Peter Kasting | 6955870 | 2016-01-12 16:26:35 -0800 | [diff] [blame] | 119 | for (size_t c = 0; c < channels; ++c) { |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 120 | ASSERT_GE(audio[i * channels + c], -bound); |
| 121 | ASSERT_LE(audio[i * channels + c], bound); |
| 122 | } |
| 123 | } |
| 124 | } |
| 125 | |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 126 | int OpusTest::EncodeDecode(WebRtcOpusEncInst* encoder, |
kwiberg | 288886b | 2015-11-06 01:21:35 -0800 | [diff] [blame] | 127 | rtc::ArrayView<const int16_t> input_audio, |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 128 | WebRtcOpusDecInst* decoder, |
| 129 | int16_t* output_audio, |
| 130 | int16_t* audio_type) { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 131 | int encoded_bytes_int = |
| 132 | WebRtcOpus_Encode(encoder, input_audio.data(), |
| 133 | rtc::CheckedDivExact(input_audio.size(), channels_), |
| 134 | kMaxBytes, bitstream_); |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 135 | EXPECT_GE(encoded_bytes_int, 0); |
| 136 | encoded_bytes_ = static_cast<size_t>(encoded_bytes_int); |
minyuel | 6d92bf5 | 2015-09-23 15:20:39 +0200 | [diff] [blame] | 137 | int est_len = WebRtcOpus_DurationEst(decoder, bitstream_, encoded_bytes_); |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 138 | int act_len = WebRtcOpus_Decode(decoder, bitstream_, encoded_bytes_, |
| 139 | output_audio, audio_type); |
minyuel | 6d92bf5 | 2015-09-23 15:20:39 +0200 | [diff] [blame] | 140 | EXPECT_EQ(est_len, act_len); |
| 141 | return act_len; |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 142 | } |
| 143 | |
| 144 | // Test if encoder/decoder can enter DTX mode properly and do not enter DTX when |
| 145 | // they should not. This test is signal dependent. |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 146 | void OpusTest::TestDtxEffect(bool dtx, int block_length_ms) { |
| 147 | PrepareSpeechData(channels_, block_length_ms, 2000); |
| 148 | const size_t samples = kOpusRateKhz * block_length_ms; |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 149 | |
| 150 | // Create encoder memory. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 151 | EXPECT_EQ(0, |
| 152 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 153 | EXPECT_EQ(0, WebRtcOpus_DecoderCreate(&opus_decoder_, channels_)); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 154 | |
| 155 | // Set bitrate. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 156 | EXPECT_EQ( |
| 157 | 0, WebRtcOpus_SetBitRate(opus_encoder_, channels_ == 1 ? 32000 : 64000)); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 158 | |
| 159 | // Set input audio as silence. |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 160 | std::vector<int16_t> silence(samples * channels_, 0); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 161 | |
| 162 | // Setting DTX. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 163 | EXPECT_EQ(0, dtx ? WebRtcOpus_EnableDtx(opus_encoder_) |
| 164 | : WebRtcOpus_DisableDtx(opus_encoder_)); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 165 | |
| 166 | int16_t audio_type; |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 167 | int16_t* output_data_decode = new int16_t[samples * channels_]; |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 168 | |
| 169 | for (int i = 0; i < 100; ++i) { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 170 | EXPECT_EQ(samples, static_cast<size_t>(EncodeDecode( |
| 171 | opus_encoder_, speech_data_.GetNextBlock(), |
| 172 | opus_decoder_, output_data_decode, &audio_type))); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 173 | // If not DTX, it should never enter DTX mode. If DTX, we do not care since |
| 174 | // whether it enters DTX depends on the signal type. |
| 175 | if (!dtx) { |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 176 | EXPECT_GT(encoded_bytes_, 1U); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 177 | EXPECT_EQ(0, opus_encoder_->in_dtx_mode); |
| 178 | EXPECT_EQ(0, opus_decoder_->in_dtx_mode); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 179 | EXPECT_EQ(0, audio_type); // Speech. |
| 180 | } |
| 181 | } |
| 182 | |
| 183 | // We input some silent segments. In DTX mode, the encoder will stop sending. |
| 184 | // However, DTX may happen after a while. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 185 | for (int i = 0; i < 30; ++i) { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 186 | EXPECT_EQ(samples, static_cast<size_t>( |
| 187 | EncodeDecode(opus_encoder_, silence, opus_decoder_, |
| 188 | output_data_decode, &audio_type))); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 189 | if (!dtx) { |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 190 | EXPECT_GT(encoded_bytes_, 1U); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 191 | EXPECT_EQ(0, opus_encoder_->in_dtx_mode); |
| 192 | EXPECT_EQ(0, opus_decoder_->in_dtx_mode); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 193 | EXPECT_EQ(0, audio_type); // Speech. |
Peter Kasting | 728d903 | 2015-06-11 14:31:38 -0700 | [diff] [blame] | 194 | } else if (encoded_bytes_ == 1) { |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 195 | EXPECT_EQ(1, opus_encoder_->in_dtx_mode); |
| 196 | EXPECT_EQ(1, opus_decoder_->in_dtx_mode); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 197 | EXPECT_EQ(2, audio_type); // Comfort noise. |
| 198 | break; |
| 199 | } |
| 200 | } |
| 201 | |
Minyue Li | 092041c | 2015-05-11 12:19:35 +0200 | [diff] [blame] | 202 | // When Opus is in DTX, it wakes up in a regular basis. It sends two packets, |
| 203 | // one with an arbitrary size and the other of 1-byte, then stops sending for |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 204 | // a certain number of frames. |
| 205 | |
| 206 | // |max_dtx_frames| is the maximum number of frames Opus can stay in DTX. |
| 207 | const int max_dtx_frames = 400 / block_length_ms + 1; |
| 208 | |
| 209 | // We run |kRunTimeMs| milliseconds of pure silence. |
minyue | 58e08cb | 2016-02-24 03:49:19 -0800 | [diff] [blame] | 210 | const int kRunTimeMs = 4500; |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 211 | |
| 212 | // We check that, after a |kCheckTimeMs| milliseconds (given that the CNG in |
| 213 | // Opus needs time to adapt), the absolute values of DTX decoded signal are |
| 214 | // bounded by |kOutputValueBound|. |
minyue | 58e08cb | 2016-02-24 03:49:19 -0800 | [diff] [blame] | 215 | const int kCheckTimeMs = 4000; |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 216 | |
| 217 | #if defined(OPUS_FIXED_POINT) |
minyuel | 7e937e9 | 2016-02-29 10:24:15 +0100 | [diff] [blame] | 218 | // Fixed-point Opus generates a random (comfort) noise, which has a less |
| 219 | // predictable value bound than its floating-point Opus. This value depends on |
| 220 | // input signal, and the time window for checking the output values (between |
| 221 | // |kCheckTimeMs| and |kRunTimeMs|). |
| 222 | const uint16_t kOutputValueBound = 30; |
| 223 | |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 224 | #else |
minyue | 58e08cb | 2016-02-24 03:49:19 -0800 | [diff] [blame] | 225 | const uint16_t kOutputValueBound = 2; |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 226 | #endif |
| 227 | |
| 228 | int time = 0; |
| 229 | while (time < kRunTimeMs) { |
| 230 | // DTX mode is maintained for maximum |max_dtx_frames| frames. |
| 231 | int i = 0; |
| 232 | for (; i < max_dtx_frames; ++i) { |
| 233 | time += block_length_ms; |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 234 | EXPECT_EQ(samples, static_cast<size_t>( |
| 235 | EncodeDecode(opus_encoder_, silence, opus_decoder_, |
| 236 | output_data_decode, &audio_type))); |
Minyue Li | 092041c | 2015-05-11 12:19:35 +0200 | [diff] [blame] | 237 | if (dtx) { |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 238 | if (encoded_bytes_ > 1) |
| 239 | break; |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 240 | EXPECT_EQ(0U, encoded_bytes_) // Send 0 byte. |
Minyue Li | 092041c | 2015-05-11 12:19:35 +0200 | [diff] [blame] | 241 | << "Opus should have entered DTX mode."; |
| 242 | EXPECT_EQ(1, opus_encoder_->in_dtx_mode); |
| 243 | EXPECT_EQ(1, opus_decoder_->in_dtx_mode); |
| 244 | EXPECT_EQ(2, audio_type); // Comfort noise. |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 245 | if (time >= kCheckTimeMs) { |
| 246 | CheckAudioBounded(output_data_decode, samples, channels_, |
| 247 | kOutputValueBound); |
| 248 | } |
Minyue Li | 092041c | 2015-05-11 12:19:35 +0200 | [diff] [blame] | 249 | } else { |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 250 | EXPECT_GT(encoded_bytes_, 1U); |
Minyue Li | 092041c | 2015-05-11 12:19:35 +0200 | [diff] [blame] | 251 | EXPECT_EQ(0, opus_encoder_->in_dtx_mode); |
| 252 | EXPECT_EQ(0, opus_decoder_->in_dtx_mode); |
| 253 | EXPECT_EQ(0, audio_type); // Speech. |
| 254 | } |
| 255 | } |
| 256 | |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 257 | if (dtx) { |
| 258 | // With DTX, Opus must stop transmission for some time. |
| 259 | EXPECT_GT(i, 1); |
| 260 | } |
Minyue Li | 092041c | 2015-05-11 12:19:35 +0200 | [diff] [blame] | 261 | |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 262 | // We expect a normal payload. |
Minyue Li | 092041c | 2015-05-11 12:19:35 +0200 | [diff] [blame] | 263 | EXPECT_EQ(0, opus_encoder_->in_dtx_mode); |
| 264 | EXPECT_EQ(0, opus_decoder_->in_dtx_mode); |
| 265 | EXPECT_EQ(0, audio_type); // Speech. |
| 266 | |
| 267 | // Enters DTX again immediately. |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 268 | time += block_length_ms; |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 269 | EXPECT_EQ(samples, static_cast<size_t>( |
| 270 | EncodeDecode(opus_encoder_, silence, opus_decoder_, |
| 271 | output_data_decode, &audio_type))); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 272 | if (dtx) { |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 273 | EXPECT_EQ(1U, encoded_bytes_); // Send 1 byte. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 274 | EXPECT_EQ(1, opus_encoder_->in_dtx_mode); |
| 275 | EXPECT_EQ(1, opus_decoder_->in_dtx_mode); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 276 | EXPECT_EQ(2, audio_type); // Comfort noise. |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 277 | if (time >= kCheckTimeMs) { |
| 278 | CheckAudioBounded(output_data_decode, samples, channels_, |
| 279 | kOutputValueBound); |
| 280 | } |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 281 | } else { |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 282 | EXPECT_GT(encoded_bytes_, 1U); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 283 | EXPECT_EQ(0, opus_encoder_->in_dtx_mode); |
| 284 | EXPECT_EQ(0, opus_decoder_->in_dtx_mode); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 285 | EXPECT_EQ(0, audio_type); // Speech. |
| 286 | } |
| 287 | } |
| 288 | |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 289 | silence[0] = 10000; |
| 290 | if (dtx) { |
| 291 | // Verify that encoder/decoder can jump out from DTX mode. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 292 | EXPECT_EQ(samples, static_cast<size_t>( |
| 293 | EncodeDecode(opus_encoder_, silence, opus_decoder_, |
| 294 | output_data_decode, &audio_type))); |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 295 | EXPECT_GT(encoded_bytes_, 1U); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 296 | EXPECT_EQ(0, opus_encoder_->in_dtx_mode); |
| 297 | EXPECT_EQ(0, opus_decoder_->in_dtx_mode); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 298 | EXPECT_EQ(0, audio_type); // Speech. |
| 299 | } |
| 300 | |
| 301 | // Free memory. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 302 | delete[] output_data_decode; |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 303 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
| 304 | EXPECT_EQ(0, WebRtcOpus_DecoderFree(opus_decoder_)); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 305 | } |
| 306 | |
soren | 28dc285 | 2017-04-06 05:48:36 -0700 | [diff] [blame] | 307 | // Test if CBR does what we expect. |
| 308 | void OpusTest::TestCbrEffect(bool cbr, int block_length_ms) { |
| 309 | PrepareSpeechData(channels_, block_length_ms, 2000); |
| 310 | const size_t samples = kOpusRateKhz * block_length_ms; |
| 311 | |
| 312 | int32_t max_pkt_size_diff = 0; |
| 313 | int32_t prev_pkt_size = 0; |
| 314 | |
| 315 | // Create encoder memory. |
| 316 | EXPECT_EQ(0, |
| 317 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
| 318 | EXPECT_EQ(0, WebRtcOpus_DecoderCreate(&opus_decoder_, channels_)); |
| 319 | |
| 320 | // Set bitrate. |
| 321 | EXPECT_EQ( |
| 322 | 0, WebRtcOpus_SetBitRate(opus_encoder_, channels_ == 1 ? 32000 : 64000)); |
| 323 | |
| 324 | // Setting CBR. |
| 325 | EXPECT_EQ(0, cbr ? WebRtcOpus_EnableCbr(opus_encoder_) |
| 326 | : WebRtcOpus_DisableCbr(opus_encoder_)); |
| 327 | |
| 328 | int16_t audio_type; |
| 329 | std::vector<int16_t> audio_out(samples * channels_); |
| 330 | for (int i = 0; i < 100; ++i) { |
| 331 | EXPECT_EQ(samples, static_cast<size_t>(EncodeDecode( |
| 332 | opus_encoder_, speech_data_.GetNextBlock(), |
| 333 | opus_decoder_, audio_out.data(), &audio_type))); |
| 334 | |
| 335 | if (prev_pkt_size > 0) { |
| 336 | int32_t diff = std::abs((int32_t)encoded_bytes_ - prev_pkt_size); |
| 337 | max_pkt_size_diff = std::max(max_pkt_size_diff, diff); |
| 338 | } |
Mirko Bonadei | 737e073 | 2017-10-19 09:00:17 +0200 | [diff] [blame] | 339 | prev_pkt_size = rtc::checked_cast<int32_t>(encoded_bytes_); |
soren | 28dc285 | 2017-04-06 05:48:36 -0700 | [diff] [blame] | 340 | } |
| 341 | |
| 342 | if (cbr) { |
| 343 | EXPECT_EQ(max_pkt_size_diff, 0); |
| 344 | } else { |
| 345 | EXPECT_GT(max_pkt_size_diff, 0); |
| 346 | } |
| 347 | |
| 348 | // Free memory. |
| 349 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
| 350 | EXPECT_EQ(0, WebRtcOpus_DecoderFree(opus_decoder_)); |
| 351 | } |
| 352 | |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 353 | // Test failing Create. |
henrika | 1d34fe9 | 2015-06-16 10:04:20 +0200 | [diff] [blame] | 354 | TEST(OpusTest, OpusCreateFail) { |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 355 | WebRtcOpusEncInst* opus_encoder; |
| 356 | WebRtcOpusDecInst* opus_decoder; |
| 357 | |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 358 | // Test to see that an invalid pointer is caught. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 359 | EXPECT_EQ(-1, WebRtcOpus_EncoderCreate(NULL, 1, 0)); |
| 360 | // Invalid channel number. |
Alex Loiko | 7a3e43a | 2019-01-29 12:27:08 +0100 | [diff] [blame] | 361 | EXPECT_EQ(-1, WebRtcOpus_EncoderCreate(&opus_encoder, 257, 0)); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 362 | // Invalid applciation mode. |
| 363 | EXPECT_EQ(-1, WebRtcOpus_EncoderCreate(&opus_encoder, 1, 2)); |
| 364 | |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 365 | EXPECT_EQ(-1, WebRtcOpus_DecoderCreate(NULL, 1)); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 366 | // Invalid channel number. |
Alex Loiko | 7a3e43a | 2019-01-29 12:27:08 +0100 | [diff] [blame] | 367 | EXPECT_EQ(-1, WebRtcOpus_DecoderCreate(&opus_decoder, 257)); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 368 | } |
| 369 | |
| 370 | // Test failing Free. |
henrika | 1d34fe9 | 2015-06-16 10:04:20 +0200 | [diff] [blame] | 371 | TEST(OpusTest, OpusFreeFail) { |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 372 | // Test to see that an invalid pointer is caught. |
| 373 | EXPECT_EQ(-1, WebRtcOpus_EncoderFree(NULL)); |
| 374 | EXPECT_EQ(-1, WebRtcOpus_DecoderFree(NULL)); |
| 375 | } |
| 376 | |
| 377 | // Test normal Create and Free. |
henrika | 1d34fe9 | 2015-06-16 10:04:20 +0200 | [diff] [blame] | 378 | TEST_P(OpusTest, OpusCreateFree) { |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 379 | EXPECT_EQ(0, |
| 380 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 381 | EXPECT_EQ(0, WebRtcOpus_DecoderCreate(&opus_decoder_, channels_)); |
| 382 | EXPECT_TRUE(opus_encoder_ != NULL); |
| 383 | EXPECT_TRUE(opus_decoder_ != NULL); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 384 | // Free encoder and decoder memory. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 385 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
| 386 | EXPECT_EQ(0, WebRtcOpus_DecoderFree(opus_decoder_)); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 387 | } |
| 388 | |
Alex Loiko | 7a3e43a | 2019-01-29 12:27:08 +0100 | [diff] [blame] | 389 | #define ENCODER_CTL(inst, vargs) \ |
| 390 | inst->channels <= 2 \ |
| 391 | ? opus_encoder_ctl(inst->encoder.encoder, vargs) \ |
| 392 | : opus_multistream_encoder_ctl(inst->encoder.multistream_encoder, vargs) |
| 393 | |
henrika | 1d34fe9 | 2015-06-16 10:04:20 +0200 | [diff] [blame] | 394 | TEST_P(OpusTest, OpusEncodeDecode) { |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 395 | PrepareSpeechData(channels_, 20, 20); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 396 | |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 397 | // Create encoder memory. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 398 | EXPECT_EQ(0, |
| 399 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
| 400 | EXPECT_EQ(0, WebRtcOpus_DecoderCreate(&opus_decoder_, channels_)); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 401 | |
| 402 | // Set bitrate. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 403 | EXPECT_EQ( |
| 404 | 0, WebRtcOpus_SetBitRate(opus_encoder_, channels_ == 1 ? 32000 : 64000)); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 405 | |
| 406 | // Check number of channels for decoder. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 407 | EXPECT_EQ(channels_, WebRtcOpus_DecoderChannels(opus_decoder_)); |
| 408 | |
| 409 | // Check application mode. |
| 410 | opus_int32 app; |
Alex Loiko | 7a3e43a | 2019-01-29 12:27:08 +0100 | [diff] [blame] | 411 | ENCODER_CTL(opus_encoder_, OPUS_GET_APPLICATION(&app)); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 412 | EXPECT_EQ(application_ == 0 ? OPUS_APPLICATION_VOIP : OPUS_APPLICATION_AUDIO, |
| 413 | app); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 414 | |
| 415 | // Encode & decode. |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 416 | int16_t audio_type; |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 417 | int16_t* output_data_decode = new int16_t[kOpus20msFrameSamples * channels_]; |
minyue@webrtc.org | f563e85 | 2014-07-18 21:11:27 +0000 | [diff] [blame] | 418 | EXPECT_EQ(kOpus20msFrameSamples, |
kwiberg | 288886b | 2015-11-06 01:21:35 -0800 | [diff] [blame] | 419 | static_cast<size_t>( |
| 420 | EncodeDecode(opus_encoder_, speech_data_.GetNextBlock(), |
| 421 | opus_decoder_, output_data_decode, &audio_type))); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 422 | |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 423 | // Free memory. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 424 | delete[] output_data_decode; |
| 425 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
| 426 | EXPECT_EQ(0, WebRtcOpus_DecoderFree(opus_decoder_)); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 427 | } |
| 428 | |
henrika | 1d34fe9 | 2015-06-16 10:04:20 +0200 | [diff] [blame] | 429 | TEST_P(OpusTest, OpusSetBitRate) { |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 430 | // Test without creating encoder memory. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 431 | EXPECT_EQ(-1, WebRtcOpus_SetBitRate(opus_encoder_, 60000)); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 432 | |
| 433 | // Create encoder memory, try with different bitrates. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 434 | EXPECT_EQ(0, |
| 435 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 436 | EXPECT_EQ(0, WebRtcOpus_SetBitRate(opus_encoder_, 30000)); |
| 437 | EXPECT_EQ(0, WebRtcOpus_SetBitRate(opus_encoder_, 60000)); |
| 438 | EXPECT_EQ(0, WebRtcOpus_SetBitRate(opus_encoder_, 300000)); |
| 439 | EXPECT_EQ(0, WebRtcOpus_SetBitRate(opus_encoder_, 600000)); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 440 | |
| 441 | // Free memory. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 442 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 443 | } |
| 444 | |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 445 | TEST_P(OpusTest, OpusSetComplexity) { |
minyue@webrtc.org | 0454688 | 2014-03-07 08:55:48 +0000 | [diff] [blame] | 446 | // Test without creating encoder memory. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 447 | EXPECT_EQ(-1, WebRtcOpus_SetComplexity(opus_encoder_, 9)); |
minyue@webrtc.org | 0454688 | 2014-03-07 08:55:48 +0000 | [diff] [blame] | 448 | |
| 449 | // Create encoder memory, try with different complexities. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 450 | EXPECT_EQ(0, |
| 451 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
minyue@webrtc.org | 0454688 | 2014-03-07 08:55:48 +0000 | [diff] [blame] | 452 | |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 453 | EXPECT_EQ(0, WebRtcOpus_SetComplexity(opus_encoder_, 0)); |
| 454 | EXPECT_EQ(0, WebRtcOpus_SetComplexity(opus_encoder_, 10)); |
| 455 | EXPECT_EQ(-1, WebRtcOpus_SetComplexity(opus_encoder_, 11)); |
minyue@webrtc.org | 0454688 | 2014-03-07 08:55:48 +0000 | [diff] [blame] | 456 | |
| 457 | // Free memory. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 458 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
minyue@webrtc.org | 0454688 | 2014-03-07 08:55:48 +0000 | [diff] [blame] | 459 | } |
| 460 | |
Alex Luebs | eeb2765 | 2017-11-20 11:13:56 -0800 | [diff] [blame] | 461 | TEST_P(OpusTest, OpusSetBandwidth) { |
Alex Loiko | 7a3e43a | 2019-01-29 12:27:08 +0100 | [diff] [blame] | 462 | if (channels_ > 2) { |
| 463 | // TODO(webrtc:10217): investigate why multi-stream Opus reports |
| 464 | // narrowband when it's configured with FULLBAND. |
| 465 | return; |
| 466 | } |
Alex Luebs | eeb2765 | 2017-11-20 11:13:56 -0800 | [diff] [blame] | 467 | PrepareSpeechData(channels_, 20, 20); |
| 468 | |
| 469 | int16_t audio_type; |
| 470 | std::unique_ptr<int16_t[]> output_data_decode( |
| 471 | new int16_t[kOpus20msFrameSamples * channels_]()); |
| 472 | |
| 473 | // Test without creating encoder memory. |
| 474 | EXPECT_EQ(-1, |
| 475 | WebRtcOpus_SetBandwidth(opus_encoder_, OPUS_BANDWIDTH_NARROWBAND)); |
| 476 | EXPECT_EQ(-1, WebRtcOpus_GetBandwidth(opus_encoder_)); |
| 477 | |
| 478 | // Create encoder memory, try with different bandwidths. |
| 479 | EXPECT_EQ(0, |
| 480 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
| 481 | EXPECT_EQ(0, WebRtcOpus_DecoderCreate(&opus_decoder_, channels_)); |
| 482 | |
| 483 | EXPECT_EQ(-1, WebRtcOpus_SetBandwidth(opus_encoder_, |
| 484 | OPUS_BANDWIDTH_NARROWBAND - 1)); |
| 485 | EXPECT_EQ(0, |
| 486 | WebRtcOpus_SetBandwidth(opus_encoder_, OPUS_BANDWIDTH_NARROWBAND)); |
| 487 | EncodeDecode(opus_encoder_, speech_data_.GetNextBlock(), opus_decoder_, |
| 488 | output_data_decode.get(), &audio_type); |
| 489 | EXPECT_EQ(OPUS_BANDWIDTH_NARROWBAND, WebRtcOpus_GetBandwidth(opus_encoder_)); |
| 490 | EXPECT_EQ(0, WebRtcOpus_SetBandwidth(opus_encoder_, OPUS_BANDWIDTH_FULLBAND)); |
| 491 | EncodeDecode(opus_encoder_, speech_data_.GetNextBlock(), opus_decoder_, |
| 492 | output_data_decode.get(), &audio_type); |
| 493 | EXPECT_EQ(OPUS_BANDWIDTH_FULLBAND, WebRtcOpus_GetBandwidth(opus_encoder_)); |
| 494 | EXPECT_EQ( |
| 495 | -1, WebRtcOpus_SetBandwidth(opus_encoder_, OPUS_BANDWIDTH_FULLBAND + 1)); |
| 496 | EncodeDecode(opus_encoder_, speech_data_.GetNextBlock(), opus_decoder_, |
| 497 | output_data_decode.get(), &audio_type); |
| 498 | EXPECT_EQ(OPUS_BANDWIDTH_FULLBAND, WebRtcOpus_GetBandwidth(opus_encoder_)); |
| 499 | |
| 500 | // Free memory. |
| 501 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
| 502 | EXPECT_EQ(0, WebRtcOpus_DecoderFree(opus_decoder_)); |
| 503 | } |
| 504 | |
minyue | c8299f9 | 2016-09-27 02:08:47 -0700 | [diff] [blame] | 505 | TEST_P(OpusTest, OpusForceChannels) { |
| 506 | // Test without creating encoder memory. |
| 507 | EXPECT_EQ(-1, WebRtcOpus_SetForceChannels(opus_encoder_, 1)); |
| 508 | |
| 509 | ASSERT_EQ(0, |
| 510 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
| 511 | |
Alex Loiko | 7a3e43a | 2019-01-29 12:27:08 +0100 | [diff] [blame] | 512 | if (channels_ >= 2) { |
minyue | c8299f9 | 2016-09-27 02:08:47 -0700 | [diff] [blame] | 513 | EXPECT_EQ(-1, WebRtcOpus_SetForceChannels(opus_encoder_, 3)); |
| 514 | EXPECT_EQ(0, WebRtcOpus_SetForceChannels(opus_encoder_, 2)); |
| 515 | EXPECT_EQ(0, WebRtcOpus_SetForceChannels(opus_encoder_, 1)); |
| 516 | EXPECT_EQ(0, WebRtcOpus_SetForceChannels(opus_encoder_, 0)); |
| 517 | } else { |
| 518 | EXPECT_EQ(-1, WebRtcOpus_SetForceChannels(opus_encoder_, 2)); |
| 519 | EXPECT_EQ(0, WebRtcOpus_SetForceChannels(opus_encoder_, 1)); |
| 520 | EXPECT_EQ(0, WebRtcOpus_SetForceChannels(opus_encoder_, 0)); |
| 521 | } |
| 522 | |
| 523 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
| 524 | } |
| 525 | |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 526 | // Encode and decode one frame, initialize the decoder and |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 527 | // decode once more. |
henrika | 1d34fe9 | 2015-06-16 10:04:20 +0200 | [diff] [blame] | 528 | TEST_P(OpusTest, OpusDecodeInit) { |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 529 | PrepareSpeechData(channels_, 20, 20); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 530 | |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 531 | // Create encoder memory. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 532 | EXPECT_EQ(0, |
| 533 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 534 | EXPECT_EQ(0, WebRtcOpus_DecoderCreate(&opus_decoder_, channels_)); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 535 | |
| 536 | // Encode & decode. |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 537 | int16_t audio_type; |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 538 | int16_t* output_data_decode = new int16_t[kOpus20msFrameSamples * channels_]; |
minyue@webrtc.org | f563e85 | 2014-07-18 21:11:27 +0000 | [diff] [blame] | 539 | EXPECT_EQ(kOpus20msFrameSamples, |
kwiberg | 288886b | 2015-11-06 01:21:35 -0800 | [diff] [blame] | 540 | static_cast<size_t>( |
| 541 | EncodeDecode(opus_encoder_, speech_data_.GetNextBlock(), |
| 542 | opus_decoder_, output_data_decode, &audio_type))); |
minyue@webrtc.org | 52bc4f4 | 2014-12-04 11:00:50 +0000 | [diff] [blame] | 543 | |
Karl Wiberg | 4376648 | 2015-08-27 15:22:11 +0200 | [diff] [blame] | 544 | WebRtcOpus_DecoderInit(opus_decoder_); |
minyue@webrtc.org | 52bc4f4 | 2014-12-04 11:00:50 +0000 | [diff] [blame] | 545 | |
| 546 | EXPECT_EQ(kOpus20msFrameSamples, |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 547 | static_cast<size_t>( |
| 548 | WebRtcOpus_Decode(opus_decoder_, bitstream_, encoded_bytes_, |
| 549 | output_data_decode, &audio_type))); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 550 | |
| 551 | // Free memory. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 552 | delete[] output_data_decode; |
| 553 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
| 554 | EXPECT_EQ(0, WebRtcOpus_DecoderFree(opus_decoder_)); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 555 | } |
| 556 | |
henrika | 1d34fe9 | 2015-06-16 10:04:20 +0200 | [diff] [blame] | 557 | TEST_P(OpusTest, OpusEnableDisableFec) { |
minyue@webrtc.org | 46509c8 | 2014-03-07 11:49:11 +0000 | [diff] [blame] | 558 | // Test without creating encoder memory. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 559 | EXPECT_EQ(-1, WebRtcOpus_EnableFec(opus_encoder_)); |
| 560 | EXPECT_EQ(-1, WebRtcOpus_DisableFec(opus_encoder_)); |
minyue@webrtc.org | 46509c8 | 2014-03-07 11:49:11 +0000 | [diff] [blame] | 561 | |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 562 | // Create encoder memory. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 563 | EXPECT_EQ(0, |
| 564 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
minyue@webrtc.org | 46509c8 | 2014-03-07 11:49:11 +0000 | [diff] [blame] | 565 | |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 566 | EXPECT_EQ(0, WebRtcOpus_EnableFec(opus_encoder_)); |
| 567 | EXPECT_EQ(0, WebRtcOpus_DisableFec(opus_encoder_)); |
minyue@webrtc.org | 46509c8 | 2014-03-07 11:49:11 +0000 | [diff] [blame] | 568 | |
| 569 | // Free memory. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 570 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
minyue@webrtc.org | 46509c8 | 2014-03-07 11:49:11 +0000 | [diff] [blame] | 571 | } |
| 572 | |
henrika | 1d34fe9 | 2015-06-16 10:04:20 +0200 | [diff] [blame] | 573 | TEST_P(OpusTest, OpusEnableDisableDtx) { |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 574 | // Test without creating encoder memory. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 575 | EXPECT_EQ(-1, WebRtcOpus_EnableDtx(opus_encoder_)); |
| 576 | EXPECT_EQ(-1, WebRtcOpus_DisableDtx(opus_encoder_)); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 577 | |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 578 | // Create encoder memory. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 579 | EXPECT_EQ(0, |
| 580 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 581 | |
| 582 | opus_int32 dtx; |
| 583 | |
| 584 | // DTX is off by default. |
Alex Loiko | 7a3e43a | 2019-01-29 12:27:08 +0100 | [diff] [blame] | 585 | ENCODER_CTL(opus_encoder_, OPUS_GET_DTX(&dtx)); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 586 | EXPECT_EQ(0, dtx); |
| 587 | |
| 588 | // Test to enable DTX. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 589 | EXPECT_EQ(0, WebRtcOpus_EnableDtx(opus_encoder_)); |
Alex Loiko | 7a3e43a | 2019-01-29 12:27:08 +0100 | [diff] [blame] | 590 | ENCODER_CTL(opus_encoder_, OPUS_GET_DTX(&dtx)); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 591 | EXPECT_EQ(1, dtx); |
| 592 | |
| 593 | // Test to disable DTX. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 594 | EXPECT_EQ(0, WebRtcOpus_DisableDtx(opus_encoder_)); |
Alex Loiko | 7a3e43a | 2019-01-29 12:27:08 +0100 | [diff] [blame] | 595 | ENCODER_CTL(opus_encoder_, OPUS_GET_DTX(&dtx)); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 596 | EXPECT_EQ(0, dtx); |
| 597 | |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 598 | // Free memory. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 599 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 600 | } |
| 601 | |
henrika | 1d34fe9 | 2015-06-16 10:04:20 +0200 | [diff] [blame] | 602 | TEST_P(OpusTest, OpusDtxOff) { |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 603 | TestDtxEffect(false, 10); |
| 604 | TestDtxEffect(false, 20); |
| 605 | TestDtxEffect(false, 40); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 606 | } |
| 607 | |
henrika | 1d34fe9 | 2015-06-16 10:04:20 +0200 | [diff] [blame] | 608 | TEST_P(OpusTest, OpusDtxOn) { |
Alex Loiko | 7a3e43a | 2019-01-29 12:27:08 +0100 | [diff] [blame] | 609 | if (channels_ > 2) { |
| 610 | // TODO(webrtc:10218): adapt the test to the sizes and order of multi-stream |
| 611 | // DTX packets. |
| 612 | return; |
| 613 | } |
minyue | 3cea256 | 2015-11-10 03:49:26 -0800 | [diff] [blame] | 614 | TestDtxEffect(true, 10); |
| 615 | TestDtxEffect(true, 20); |
| 616 | TestDtxEffect(true, 40); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 617 | } |
| 618 | |
soren | 28dc285 | 2017-04-06 05:48:36 -0700 | [diff] [blame] | 619 | TEST_P(OpusTest, OpusCbrOff) { |
| 620 | TestCbrEffect(false, 10); |
| 621 | TestCbrEffect(false, 20); |
| 622 | TestCbrEffect(false, 40); |
| 623 | } |
| 624 | |
| 625 | TEST_P(OpusTest, OpusCbrOn) { |
| 626 | TestCbrEffect(true, 10); |
| 627 | TestCbrEffect(true, 20); |
| 628 | TestCbrEffect(true, 40); |
| 629 | } |
| 630 | |
henrika | 1d34fe9 | 2015-06-16 10:04:20 +0200 | [diff] [blame] | 631 | TEST_P(OpusTest, OpusSetPacketLossRate) { |
minyue@webrtc.org | 46509c8 | 2014-03-07 11:49:11 +0000 | [diff] [blame] | 632 | // Test without creating encoder memory. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 633 | EXPECT_EQ(-1, WebRtcOpus_SetPacketLossRate(opus_encoder_, 50)); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 634 | |
tina.legrand@webrtc.org | bd21fb5 | 2013-08-08 11:01:07 +0000 | [diff] [blame] | 635 | // Create encoder memory. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 636 | EXPECT_EQ(0, |
| 637 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 638 | |
| 639 | EXPECT_EQ(0, WebRtcOpus_SetPacketLossRate(opus_encoder_, 50)); |
| 640 | EXPECT_EQ(-1, WebRtcOpus_SetPacketLossRate(opus_encoder_, -1)); |
| 641 | EXPECT_EQ(-1, WebRtcOpus_SetPacketLossRate(opus_encoder_, 101)); |
| 642 | |
| 643 | // Free memory. |
| 644 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
| 645 | } |
| 646 | |
henrika | 1d34fe9 | 2015-06-16 10:04:20 +0200 | [diff] [blame] | 647 | TEST_P(OpusTest, OpusSetMaxPlaybackRate) { |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 648 | // Test without creating encoder memory. |
| 649 | EXPECT_EQ(-1, WebRtcOpus_SetMaxPlaybackRate(opus_encoder_, 20000)); |
| 650 | |
| 651 | // Create encoder memory. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 652 | EXPECT_EQ(0, |
| 653 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 654 | |
| 655 | SetMaxPlaybackRate(opus_encoder_, OPUS_BANDWIDTH_FULLBAND, 48000); |
| 656 | SetMaxPlaybackRate(opus_encoder_, OPUS_BANDWIDTH_FULLBAND, 24001); |
| 657 | SetMaxPlaybackRate(opus_encoder_, OPUS_BANDWIDTH_SUPERWIDEBAND, 24000); |
| 658 | SetMaxPlaybackRate(opus_encoder_, OPUS_BANDWIDTH_SUPERWIDEBAND, 16001); |
| 659 | SetMaxPlaybackRate(opus_encoder_, OPUS_BANDWIDTH_WIDEBAND, 16000); |
| 660 | SetMaxPlaybackRate(opus_encoder_, OPUS_BANDWIDTH_WIDEBAND, 12001); |
| 661 | SetMaxPlaybackRate(opus_encoder_, OPUS_BANDWIDTH_MEDIUMBAND, 12000); |
| 662 | SetMaxPlaybackRate(opus_encoder_, OPUS_BANDWIDTH_MEDIUMBAND, 8001); |
| 663 | SetMaxPlaybackRate(opus_encoder_, OPUS_BANDWIDTH_NARROWBAND, 8000); |
| 664 | SetMaxPlaybackRate(opus_encoder_, OPUS_BANDWIDTH_NARROWBAND, 4000); |
| 665 | |
| 666 | // Free memory. |
| 667 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
| 668 | } |
| 669 | |
| 670 | // Test PLC. |
henrika | 1d34fe9 | 2015-06-16 10:04:20 +0200 | [diff] [blame] | 671 | TEST_P(OpusTest, OpusDecodePlc) { |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 672 | PrepareSpeechData(channels_, 20, 20); |
| 673 | |
| 674 | // Create encoder memory. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 675 | EXPECT_EQ(0, |
| 676 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 677 | EXPECT_EQ(0, WebRtcOpus_DecoderCreate(&opus_decoder_, channels_)); |
tina.legrand@webrtc.org | bd21fb5 | 2013-08-08 11:01:07 +0000 | [diff] [blame] | 678 | |
| 679 | // Set bitrate. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 680 | EXPECT_EQ( |
| 681 | 0, WebRtcOpus_SetBitRate(opus_encoder_, channels_ == 1 ? 32000 : 64000)); |
tina.legrand@webrtc.org | bd21fb5 | 2013-08-08 11:01:07 +0000 | [diff] [blame] | 682 | |
| 683 | // Check number of channels for decoder. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 684 | EXPECT_EQ(channels_, WebRtcOpus_DecoderChannels(opus_decoder_)); |
tina.legrand@webrtc.org | bd21fb5 | 2013-08-08 11:01:07 +0000 | [diff] [blame] | 685 | |
| 686 | // Encode & decode. |
tina.legrand@webrtc.org | bd21fb5 | 2013-08-08 11:01:07 +0000 | [diff] [blame] | 687 | int16_t audio_type; |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 688 | int16_t* output_data_decode = new int16_t[kOpus20msFrameSamples * channels_]; |
minyue@webrtc.org | f563e85 | 2014-07-18 21:11:27 +0000 | [diff] [blame] | 689 | EXPECT_EQ(kOpus20msFrameSamples, |
kwiberg | 288886b | 2015-11-06 01:21:35 -0800 | [diff] [blame] | 690 | static_cast<size_t>( |
| 691 | EncodeDecode(opus_encoder_, speech_data_.GetNextBlock(), |
| 692 | opus_decoder_, output_data_decode, &audio_type))); |
tina.legrand@webrtc.org | bd21fb5 | 2013-08-08 11:01:07 +0000 | [diff] [blame] | 693 | |
minyue@webrtc.org | 33ccdfa | 2014-12-04 12:14:12 +0000 | [diff] [blame] | 694 | // Call decoder PLC. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 695 | int16_t* plc_buffer = new int16_t[kOpus20msFrameSamples * channels_]; |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 696 | EXPECT_EQ(kOpus20msFrameSamples, static_cast<size_t>(WebRtcOpus_DecodePlc( |
| 697 | opus_decoder_, plc_buffer, 1))); |
tina.legrand@webrtc.org | bd21fb5 | 2013-08-08 11:01:07 +0000 | [diff] [blame] | 698 | |
| 699 | // Free memory. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 700 | delete[] plc_buffer; |
| 701 | delete[] output_data_decode; |
| 702 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
| 703 | EXPECT_EQ(0, WebRtcOpus_DecoderFree(opus_decoder_)); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 704 | } |
| 705 | |
| 706 | // Duration estimation. |
henrika | 1d34fe9 | 2015-06-16 10:04:20 +0200 | [diff] [blame] | 707 | TEST_P(OpusTest, OpusDurationEstimation) { |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 708 | PrepareSpeechData(channels_, 20, 20); |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 709 | |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 710 | // Create. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 711 | EXPECT_EQ(0, |
| 712 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 713 | EXPECT_EQ(0, WebRtcOpus_DecoderCreate(&opus_decoder_, channels_)); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 714 | |
minyue@webrtc.org | 0ca768b | 2014-12-11 16:09:35 +0000 | [diff] [blame] | 715 | // 10 ms. We use only first 10 ms of a 20 ms block. |
kwiberg | 288886b | 2015-11-06 01:21:35 -0800 | [diff] [blame] | 716 | auto speech_block = speech_data_.GetNextBlock(); |
| 717 | int encoded_bytes_int = WebRtcOpus_Encode( |
| 718 | opus_encoder_, speech_block.data(), |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 719 | rtc::CheckedDivExact(speech_block.size(), 2 * channels_), kMaxBytes, |
| 720 | bitstream_); |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 721 | EXPECT_GE(encoded_bytes_int, 0); |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 722 | EXPECT_EQ( |
| 723 | kOpus10msFrameSamples, |
| 724 | static_cast<size_t>(WebRtcOpus_DurationEst( |
| 725 | opus_decoder_, bitstream_, static_cast<size_t>(encoded_bytes_int)))); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 726 | |
| 727 | // 20 ms |
kwiberg | 288886b | 2015-11-06 01:21:35 -0800 | [diff] [blame] | 728 | speech_block = speech_data_.GetNextBlock(); |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 729 | encoded_bytes_int = |
| 730 | WebRtcOpus_Encode(opus_encoder_, speech_block.data(), |
| 731 | rtc::CheckedDivExact(speech_block.size(), channels_), |
| 732 | kMaxBytes, bitstream_); |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 733 | EXPECT_GE(encoded_bytes_int, 0); |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 734 | EXPECT_EQ( |
| 735 | kOpus20msFrameSamples, |
| 736 | static_cast<size_t>(WebRtcOpus_DurationEst( |
| 737 | opus_decoder_, bitstream_, static_cast<size_t>(encoded_bytes_int)))); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 738 | |
| 739 | // Free memory. |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 740 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
| 741 | EXPECT_EQ(0, WebRtcOpus_DecoderFree(opus_decoder_)); |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 742 | } |
| 743 | |
henrika | 1d34fe9 | 2015-06-16 10:04:20 +0200 | [diff] [blame] | 744 | TEST_P(OpusTest, OpusDecodeRepacketized) { |
Alex Loiko | 7a3e43a | 2019-01-29 12:27:08 +0100 | [diff] [blame] | 745 | if (channels_ > 2) { |
| 746 | // As per the Opus documentation |
| 747 | // https://mf4.xiph.org/jenkins/view/opus/job/opus/ws/doc/html/group__opus__repacketizer.html#details, |
| 748 | // multiple streams are not supported. |
| 749 | return; |
| 750 | } |
minyue | a613eb6 | 2017-03-14 14:33:30 -0700 | [diff] [blame] | 751 | constexpr size_t kPackets = 6; |
minyue@webrtc.org | 7f7d7e3 | 2015-03-16 12:30:37 +0000 | [diff] [blame] | 752 | |
| 753 | PrepareSpeechData(channels_, 20, 20 * kPackets); |
| 754 | |
| 755 | // Create encoder memory. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 756 | ASSERT_EQ(0, |
| 757 | WebRtcOpus_EncoderCreate(&opus_encoder_, channels_, application_)); |
| 758 | ASSERT_EQ(0, WebRtcOpus_DecoderCreate(&opus_decoder_, channels_)); |
minyue@webrtc.org | 7f7d7e3 | 2015-03-16 12:30:37 +0000 | [diff] [blame] | 759 | |
| 760 | // Set bitrate. |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 761 | EXPECT_EQ( |
| 762 | 0, WebRtcOpus_SetBitRate(opus_encoder_, channels_ == 1 ? 32000 : 64000)); |
minyue@webrtc.org | 7f7d7e3 | 2015-03-16 12:30:37 +0000 | [diff] [blame] | 763 | |
| 764 | // Check number of channels for decoder. |
| 765 | EXPECT_EQ(channels_, WebRtcOpus_DecoderChannels(opus_decoder_)); |
| 766 | |
| 767 | // Encode & decode. |
| 768 | int16_t audio_type; |
kwiberg | 91d9756 | 2016-02-14 01:10:03 -0800 | [diff] [blame] | 769 | std::unique_ptr<int16_t[]> output_data_decode( |
minyue@webrtc.org | 7f7d7e3 | 2015-03-16 12:30:37 +0000 | [diff] [blame] | 770 | new int16_t[kPackets * kOpus20msFrameSamples * channels_]); |
| 771 | OpusRepacketizer* rp = opus_repacketizer_create(); |
| 772 | |
minyue | a613eb6 | 2017-03-14 14:33:30 -0700 | [diff] [blame] | 773 | size_t num_packets = 0; |
| 774 | constexpr size_t kMaxCycles = 100; |
| 775 | for (size_t idx = 0; idx < kMaxCycles; ++idx) { |
kwiberg | 288886b | 2015-11-06 01:21:35 -0800 | [diff] [blame] | 776 | auto speech_block = speech_data_.GetNextBlock(); |
| 777 | encoded_bytes_ = |
| 778 | WebRtcOpus_Encode(opus_encoder_, speech_block.data(), |
Peter Kasting | 6955870 | 2016-01-12 16:26:35 -0800 | [diff] [blame] | 779 | rtc::CheckedDivExact(speech_block.size(), channels_), |
kwiberg | 288886b | 2015-11-06 01:21:35 -0800 | [diff] [blame] | 780 | kMaxBytes, bitstream_); |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 781 | if (opus_repacketizer_cat(rp, bitstream_, |
| 782 | rtc::checked_cast<opus_int32>(encoded_bytes_)) == |
| 783 | OPUS_OK) { |
minyue | a613eb6 | 2017-03-14 14:33:30 -0700 | [diff] [blame] | 784 | ++num_packets; |
| 785 | if (num_packets == kPackets) { |
| 786 | break; |
| 787 | } |
| 788 | } else { |
| 789 | // Opus repacketizer cannot guarantee a success. We try again if it fails. |
| 790 | opus_repacketizer_init(rp); |
| 791 | num_packets = 0; |
| 792 | } |
minyue@webrtc.org | 7f7d7e3 | 2015-03-16 12:30:37 +0000 | [diff] [blame] | 793 | } |
minyue | a613eb6 | 2017-03-14 14:33:30 -0700 | [diff] [blame] | 794 | EXPECT_EQ(kPackets, num_packets); |
minyue@webrtc.org | 7f7d7e3 | 2015-03-16 12:30:37 +0000 | [diff] [blame] | 795 | |
| 796 | encoded_bytes_ = opus_repacketizer_out(rp, bitstream_, kMaxBytes); |
| 797 | |
| 798 | EXPECT_EQ(kOpus20msFrameSamples * kPackets, |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 799 | static_cast<size_t>(WebRtcOpus_DurationEst( |
| 800 | opus_decoder_, bitstream_, encoded_bytes_))); |
minyue@webrtc.org | 7f7d7e3 | 2015-03-16 12:30:37 +0000 | [diff] [blame] | 801 | |
| 802 | EXPECT_EQ(kOpus20msFrameSamples * kPackets, |
Yves Gerey | 665174f | 2018-06-19 15:03:05 +0200 | [diff] [blame] | 803 | static_cast<size_t>( |
| 804 | WebRtcOpus_Decode(opus_decoder_, bitstream_, encoded_bytes_, |
| 805 | output_data_decode.get(), &audio_type))); |
minyue@webrtc.org | 7f7d7e3 | 2015-03-16 12:30:37 +0000 | [diff] [blame] | 806 | |
| 807 | // Free memory. |
| 808 | opus_repacketizer_destroy(rp); |
| 809 | EXPECT_EQ(0, WebRtcOpus_EncoderFree(opus_encoder_)); |
| 810 | EXPECT_EQ(0, WebRtcOpus_DecoderFree(opus_decoder_)); |
| 811 | } |
| 812 | |
Mirko Bonadei | c84f661 | 2019-01-31 12:20:57 +0100 | [diff] [blame] | 813 | INSTANTIATE_TEST_SUITE_P(VariousMode, |
| 814 | OpusTest, |
| 815 | Combine(Values(1, 2, 4), Values(0, 1))); |
minyue@webrtc.org | 7dba786 | 2015-01-20 16:01:50 +0000 | [diff] [blame] | 816 | |
tina.legrand@webrtc.org | db11fab | 2013-04-17 10:39:41 +0000 | [diff] [blame] | 817 | } // namespace webrtc |