Adding BitrateController to audio network adaptor.

webrtc/modules/audio_coding/audio_network_adaptor/bitrate_controller_unittest.cc

Index: webrtc/modules/audio_coding/audio_network_adaptor/bitrate_controller_unittest.cc
diff --git a/webrtc/modules/audio_coding/audio_network_adaptor/bitrate_controller_unittest.cc b/webrtc/modules/audio_coding/audio_network_adaptor/bitrate_controller_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..5a3a74ad3e2eedf5770658ca193f1cc8f5e1da29
--- /dev/null
+++ b/webrtc/modules/audio_coding/audio_network_adaptor/bitrate_controller_unittest.cc
@@ -0,0 +1,142 @@
+/*
+ *  Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
+ *
+ *  Use of this source code is governed by a BSD-style license
+ *  that can be found in the LICENSE file in the root of the source
+ *  tree. An additional intellectual property rights grant can be found
+ *  in the file PATENTS.  All contributing project authors may
+ *  be found in the AUTHORS file in the root of the source tree.
+ */
+
+#include "testing/gtest/include/gtest/gtest.h"
+#include "webrtc/modules/audio_coding/audio_network_adaptor/bitrate_controller.h"
+
+namespace webrtc {
+
+namespace {
+
+// L2(14B) + IPv4(20B) + UDP(8B) + RTP(12B) + SRTP_AUTH(10B) = 64B = 512 bits
+constexpr int kPacketOverheadBits = 512;
+
+void CheckDecision(BitrateController* controller,
+                   const rtc::Optional<int>& target_audio_bitrate_bps,
+                   const rtc::Optional<int>& frame_length_ms,
+                   int expected_bitrate_bps) {
+  Controller::NetworkMetrics metrics;
+  metrics.target_audio_bitrate_bps = target_audio_bitrate_bps;
+  AudioNetworkAdaptor::EncoderRuntimeConfig config;
+  config.frame_length_ms = frame_length_ms;
+  controller->MakeDecision(metrics, &config);
+  EXPECT_EQ(rtc::Optional<int>(expected_bitrate_bps), config.bitrate_bps);
+}
+
+}  // namespace
+
+// These tests are named AnaBitrateControllerTest to distinguish from
+// BitrateControllerTest in
+// modules/bitrate_controller/bitrate_controller_unittest.cc.
+
+TEST(AnaBitrateControllerTest, OutputInitValueWhenTargetBitrateUnknown) {
+  constexpr int kInitialBitrateBps = 32000;
+  constexpr int kInitialFrameLengthMs = 20;
+  BitrateController controller(
+      BitrateController::Config(kInitialBitrateBps, kInitialFrameLengthMs));
+  CheckDecision(&controller, rtc::Optional<int>(),
+                rtc::Optional<int>(kInitialFrameLengthMs * 2),
+                kInitialBitrateBps);
+}
+
+TEST(AnaBitrateControllerTest, ChangeBitrateOnTargetBitrateChanged) {
+  constexpr int kInitialBitrateBps = 32000;
+  constexpr int kInitialFrameLengthMs = 20;
+  BitrateController controller(
+      BitrateController::Config(kInitialBitrateBps, kInitialFrameLengthMs));
+  constexpr int kTargetBitrateBps = 48000;
+  // Frame length unchanged, bitrate changes in accordance with
+  // |metrics.target_audio_bitrate_bps|
+  CheckDecision(&controller, rtc::Optional<int>(kTargetBitrateBps),
+                rtc::Optional<int>(kInitialFrameLengthMs), kTargetBitrateBps);
+}
+
+TEST(AnaBitrateControllerTest, TreatUnknownFrameLengthAsFrameLengthUnchanged) {
+  constexpr int kInitialBitrateBps = 32000;
+  constexpr int kInitialFrameLengthMs = 20;
+  BitrateController controller(
+      BitrateController::Config(kInitialBitrateBps, kInitialFrameLengthMs));
+  constexpr int kTargetBitrateBps = 48000;
+  CheckDecision(&controller, rtc::Optional<int>(kTargetBitrateBps),
+                rtc::Optional<int>(), kTargetBitrateBps);
+}
+
+TEST(AnaBitrateControllerTest, IncreaseBitrateOnFrameLengthIncreased) {
+  constexpr int kInitialBitrateBps = 32000;
+  constexpr int kInitialFrameLengthMs = 20;
+  BitrateController controller(
+      BitrateController::Config(kInitialBitrateBps, kInitialFrameLengthMs));
+  constexpr int kFrameLengthMs = 60;
+  constexpr int kPacketOverheadRateDiff =
+      kPacketOverheadBits * 1000 / kInitialFrameLengthMs -
+      kPacketOverheadBits * 1000 / kFrameLengthMs;
+  CheckDecision(&controller, rtc::Optional<int>(kInitialBitrateBps),
+                rtc::Optional<int>(kFrameLengthMs),
+                kInitialBitrateBps + kPacketOverheadRateDiff);
+}
+
+TEST(AnaBitrateControllerTest, DecreaseBitrateOnFrameLengthDecreased) {
+  constexpr int kInitialBitrateBps = 32000;
+  constexpr int kInitialFrameLengthMs = 60;
+  BitrateController controller(
+      BitrateController::Config(kInitialBitrateBps, kInitialFrameLengthMs));
+  constexpr int kFrameLengthMs = 20;
+  constexpr int kPacketOverheadRateDiff =
+      kPacketOverheadBits * 1000 / kInitialFrameLengthMs -
+      kPacketOverheadBits * 1000 / kFrameLengthMs;
+  CheckDecision(&controller, rtc::Optional<int>(kInitialBitrateBps),
+                rtc::Optional<int>(kFrameLengthMs),
+                kInitialBitrateBps + kPacketOverheadRateDiff);
+}
+
+TEST(AnaBitrateControllerTest, CheckBehaviorOnChangingCondition) {
+  constexpr int kInitialBitrateBps = 32000;
+  constexpr int kInitialFrameLengthMs = 20;
+  BitrateController controller(
+      BitrateController::Config(kInitialBitrateBps, kInitialFrameLengthMs));
+
+  int last_overhead_bitrate =
+      kPacketOverheadBits * 1000 / kInitialFrameLengthMs;
+  int current_overhead_bitrate = kPacketOverheadBits * 1000 / 20;
+  // Start from an arbitrary overall bitrate.
+  int overall_bitrate = 34567;
+  CheckDecision(
+      &controller, rtc::Optional<int>(overall_bitrate - last_overhead_bitrate),
+      rtc::Optional<int>(20), overall_bitrate - current_overhead_bitrate);
+
+  // Next: increase overall bitrate.
+  overall_bitrate += 100;
+  CheckDecision(
+      &controller, rtc::Optional<int>(overall_bitrate - last_overhead_bitrate),
+      rtc::Optional<int>(20), overall_bitrate - current_overhead_bitrate);
+
+  // Next: change frame length.
+  current_overhead_bitrate = kPacketOverheadBits * 1000 / 60;
+  CheckDecision(
+      &controller, rtc::Optional<int>(overall_bitrate - last_overhead_bitrate),
+      rtc::Optional<int>(60), overall_bitrate - current_overhead_bitrate);
+  last_overhead_bitrate = current_overhead_bitrate;
+
+  // Next: change frame length.
+  current_overhead_bitrate = kPacketOverheadBits * 1000 / 20;
+  CheckDecision(
+      &controller, rtc::Optional<int>(overall_bitrate - last_overhead_bitrate),
+      rtc::Optional<int>(20), overall_bitrate - current_overhead_bitrate);
+  last_overhead_bitrate = current_overhead_bitrate;
+
+  // Next: decrease overall bitrate and frame length.
+  overall_bitrate -= 100;
+  current_overhead_bitrate = kPacketOverheadBits * 1000 / 60;
+  CheckDecision(
+      &controller, rtc::Optional<int>(overall_bitrate - last_overhead_bitrate),
+      rtc::Optional<int>(60), overall_bitrate - current_overhead_bitrate);
+}
+
+}  // namespace webrtc