1 /*
2 * Copyright (c) 2015 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 #include <memory>
12
13 #include "absl/flags/flag.h"
14 #include "modules/audio_coding/codecs/ilbc/audio_encoder_ilbc.h"
15 #include "modules/audio_coding/neteq/tools/neteq_quality_test.h"
16 #include "rtc_base/checks.h"
17 #include "rtc_base/numerics/safe_conversions.h"
18 #include "test/testsupport/file_utils.h"
19
20 ABSL_FLAG(int, frame_size_ms, 20, "Codec frame size (milliseconds).");
21
22 using ::testing::InitGoogleTest;
23
24 namespace webrtc {
25 namespace test {
26 namespace {
27 static const int kInputSampleRateKhz = 8;
28 static const int kOutputSampleRateKhz = 8;
29 } // namespace
30
31 class NetEqIlbcQualityTest : public NetEqQualityTest {
32 protected:
NetEqIlbcQualityTest()33 NetEqIlbcQualityTest()
34 : NetEqQualityTest(absl::GetFlag(FLAGS_frame_size_ms),
35 kInputSampleRateKhz,
36 kOutputSampleRateKhz,
37 SdpAudioFormat("ilbc", 8000, 1)) {
38 // Flag validation
39 RTC_CHECK(absl::GetFlag(FLAGS_frame_size_ms) == 20 ||
40 absl::GetFlag(FLAGS_frame_size_ms) == 30 ||
41 absl::GetFlag(FLAGS_frame_size_ms) == 40 ||
42 absl::GetFlag(FLAGS_frame_size_ms) == 60)
43 << "Invalid frame size, should be 20, 30, 40, or 60 ms.";
44 }
45
SetUp()46 void SetUp() override {
47 ASSERT_EQ(1u, channels_) << "iLBC supports only mono audio.";
48 AudioEncoderIlbcConfig config;
49 config.frame_size_ms = absl::GetFlag(FLAGS_frame_size_ms);
50 encoder_.reset(new AudioEncoderIlbcImpl(config, 102));
51 NetEqQualityTest::SetUp();
52 }
53
EncodeBlock(int16_t * in_data,size_t block_size_samples,rtc::Buffer * payload,size_t max_bytes)54 int EncodeBlock(int16_t* in_data,
55 size_t block_size_samples,
56 rtc::Buffer* payload,
57 size_t max_bytes) override {
58 const size_t kFrameSizeSamples = 80; // Samples per 10 ms.
59 size_t encoded_samples = 0;
60 uint32_t dummy_timestamp = 0;
61 AudioEncoder::EncodedInfo info;
62 do {
63 info = encoder_->Encode(dummy_timestamp,
64 rtc::ArrayView<const int16_t>(
65 in_data + encoded_samples, kFrameSizeSamples),
66 payload);
67 encoded_samples += kFrameSizeSamples;
68 } while (info.encoded_bytes == 0);
69 return rtc::checked_cast<int>(info.encoded_bytes);
70 }
71
72 private:
73 std::unique_ptr<AudioEncoderIlbcImpl> encoder_;
74 };
75
TEST_F(NetEqIlbcQualityTest,Test)76 TEST_F(NetEqIlbcQualityTest, Test) {
77 Simulate();
78 }
79
80 } // namespace test
81 } // namespace webrtc
82