Extract bitrate allocation of spatial/temporal layers out of codec impl.

webrtc/common_types.cc

 /*
  *  Copyright (c) 2012 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 "webrtc/common_types.h"
 
+#include "webrtc/base/checks.h"
+
 #include <string.h>
 
 namespace webrtc {
 
 StreamDataCounters::StreamDataCounters() : first_packet_time_ms(-1) {}
 
 RTPHeaderExtension::RTPHeaderExtension()
     : hasTransmissionTimeOffset(false),
       transmissionTimeOffset(0),
       hasAbsoluteSendTime(false),
@@ skipping 12 matching lines @@
       sequenceNumber(0),
       timestamp(0),
       ssrc(0),
       numCSRCs(0),
       arrOfCSRCs(),
       paddingLength(0),
       headerLength(0),
       payload_type_frequency(0),
       extension() {}
 
+BitrateAllocation::BitrateAllocation() : sum_(0), bitrates_{} {}
+
+bool BitrateAllocation::SetBitrate(size_t spatial_index,
+                                   size_t temporal_index,
+                                   uint32_t bitrate_bps) {
+  RTC_DCHECK_LT(spatial_index, static_cast<size_t>(kMaxSpatialLayers));
+  RTC_DCHECK_LT(temporal_index, static_cast<size_t>(kMaxTemporalStreams));
+  if (bitrate_bps > kMaxBitrateBps ||
+      sum_ - bitrates_[spatial_index][temporal_index] + bitrate_bps >
+          kMaxBitrateBps) {
+    return false;
+  }
+  sum_ -= bitrates_[spatial_index][temporal_index];
+  bitrates_[spatial_index][temporal_index] = bitrate_bps;
+  sum_ += bitrate_bps;
+  return true;
+}
+
+uint32_t BitrateAllocation::GetBitrate(size_t spatial_index,
+                                       size_t temporal_index) const {
+  RTC_DCHECK_LT(spatial_index, static_cast<size_t>(kMaxSpatialLayers));
+  RTC_DCHECK_LT(temporal_index, static_cast<size_t>(kMaxTemporalStreams));
+  return bitrates_[spatial_index][temporal_index];
+}
+
+// Get the sum of all the temporal layer for a specific spatial layer.
+uint32_t BitrateAllocation::GetSpatialLayerSum(size_t spatial_index) const {
+  RTC_DCHECK_LT(spatial_index, static_cast<size_t>(kMaxSpatialLayers));
+  uint32_t sum = 0;
+  for (int i = 0; i < kMaxTemporalStreams; ++i)
+    sum += bitrates_[spatial_index][i];
+  return sum;
+}
+
 }  // namespace webrtc