1 /*
2 * Copyright (c) 2018 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/pcm16b/audio_encoder_pcm16b.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 = 48;
28 static const int kOutputSampleRateKhz = 48;
29 } // namespace
30
31 class NetEqPcm16bQualityTest : public NetEqQualityTest {
32 protected:
NetEqPcm16bQualityTest()33 NetEqPcm16bQualityTest()
34 : NetEqQualityTest(absl::GetFlag(FLAGS_frame_size_ms),
35 kInputSampleRateKhz,
36 kOutputSampleRateKhz,
37 SdpAudioFormat("l16", 48000, 1)) {
38 // Flag validation
39 RTC_CHECK(absl::GetFlag(FLAGS_frame_size_ms) >= 10 &&
40 absl::GetFlag(FLAGS_frame_size_ms) <= 60 &&
41 (absl::GetFlag(FLAGS_frame_size_ms) % 10) == 0)
42 << "Invalid frame size, should be 10, 20, ..., 60 ms.";
43 }
44
SetUp()45 void SetUp() override {
46 AudioEncoderPcm16B::Config config;
47 config.frame_size_ms = absl::GetFlag(FLAGS_frame_size_ms);
48 config.sample_rate_hz = 48000;
49 config.num_channels = channels_;
50 encoder_.reset(new AudioEncoderPcm16B(config));
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 = 480; // 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<AudioEncoderPcm16B> encoder_;
74 };
75
TEST_F(NetEqPcm16bQualityTest,Test)76 TEST_F(NetEqPcm16bQualityTest, Test) {
77 Simulate();
78 }
79
80 } // namespace test
81 } // namespace webrtc
82