Adapt encoder behavior to target bitrate

remoting/protocol/webrtc_video_encoder.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "remoting/protocol/webrtc_video_encoder.h"
 
 #include <algorithm>
 #include <vector>
 
 #include "base/callback.h"
@@ skipping 26 matching lines @@
   int streamCount = codec_settings->numberOfSimulcastStreams;
   // Validate request is to support a single stream.
   if (streamCount > 1) {
     for (int i = 0; i < streamCount; ++i) {
       if (codec_settings->simulcastStream[i].maxBitrate != 0) {
         LOG(ERROR) << "Simulcast unsupported";
         return WEBRTC_VIDEO_CODEC_ERR_PARAMETER;
       }
     }
   }
-  target_bitrate_ = codec_settings->startBitrate;
   state_ = kInitialized;
   return WEBRTC_VIDEO_CODEC_OK;
 }
 
 int32_t WebRtcVideoEncoder::RegisterEncodeCompleteCallback(
     webrtc::EncodedImageCallback* callback) {
   base::AutoLock lock(lock_);
   DCHECK(callback);
   encoded_callback_ = callback;
   return WEBRTC_VIDEO_CODEC_OK;
@@ skipping 19 matching lines @@
                                                  int64_t rtt) {
   VLOG(1) << "WebRtcVideoEncoder::SetChannelParameters "
           << "loss:RTT " << packet_loss << ":" << rtt;
   // Unused right now.
   return WEBRTC_VIDEO_CODEC_OK;
 }
 
 int32_t WebRtcVideoEncoder::SetRates(uint32_t bitrate, uint32_t framerate) {
   VLOG(1) << "WebRtcVideoEncoder::SetRates bitrate:framerate " << bitrate << ":"
           << framerate;
-  target_bitrate_ = bitrate;
+  if (!target_bitrate_cb_.is_null())
+    target_bitrate_cb_.Run(bitrate);
   // framerate is not expected to be valid given we never report captured
   // frames
   return WEBRTC_VIDEO_CODEC_OK;
 }
 
 int WebRtcVideoEncoder::SendEncodedFrame(int64_t capture_timestamp_ms,
                                          std::unique_ptr<VideoPacket> frame) {
   uint8_t* buffer =
       reinterpret_cast<uint8_t*>(const_cast<char*>(frame->data().data()));
   size_t buffer_size = frame->data().size();
@@ skipping 39 matching lines @@
   }
   return result;
 }
 
 void WebRtcVideoEncoder::SetKeyFrameRequestCallback(
     const base::Closure& key_frame_request) {
   base::AutoLock lock(lock_);
   key_frame_request_ = key_frame_request;
 }
 
+void WebRtcVideoEncoder::SetTargetBitrateCallback(
+    const TargetBitrateCallback& target_bitrate_cb) {
+  base::AutoLock lock(lock_);
+  target_bitrate_cb_ = target_bitrate_cb;
+}
+
 WebRtcVideoEncoderFactory::WebRtcVideoEncoderFactory() {
   // TODO(isheriff): These do not really affect anything internally
   // in webrtc.
   codecs_.push_back(cricket::WebRtcVideoEncoderFactory::VideoCodec(
       webrtc::kVideoCodecVP8, "VP8", 1280, 720, 30));
 }
 
 WebRtcVideoEncoderFactory::~WebRtcVideoEncoderFactory() {
   DCHECK(encoders_.empty());
 }
 
 webrtc::VideoEncoder* WebRtcVideoEncoderFactory::CreateVideoEncoder(
     webrtc::VideoCodecType type) {
   VLOG(2) << "WebRtcVideoEncoderFactory::CreateVideoEncoder " << type;
   DCHECK(type == webrtc::kVideoCodecVP8);
   WebRtcVideoEncoder* encoder = new WebRtcVideoEncoder(type);
   base::AutoLock lock(lock_);
   encoder->SetKeyFrameRequestCallback(key_frame_request_);
+  encoder->SetTargetBitrateCallback(target_bitrate_cb_);
   VLOG(1) << "Created " << encoder;
   encoders_.push_back(base::WrapUnique(encoder));
   return encoder;
 }
 
 const std::vector<cricket::WebRtcVideoEncoderFactory::VideoCodec>&
 WebRtcVideoEncoderFactory::codecs() const {
   VLOG(2) << "WebRtcVideoEncoderFactory::codecs";
   return codecs_;
 }
@@ skipping 36 matching lines @@
   base::AutoLock lock(lock_);
   key_frame_request_ = key_frame_request;
   if (encoders_.size() == 1) {
     encoders_.front()->SetKeyFrameRequestCallback(key_frame_request);
   } else {
     LOG(ERROR) << "Dropping key frame request callback with unexpected"
                   " number of encoders"
                << encoders_.size();
   }
 }
+
+void WebRtcVideoEncoderFactory::SetTargetBitrateCallback(
+    const TargetBitrateCallback& target_bitrate_cb) {
+  base::AutoLock lock(lock_);
+  target_bitrate_cb_ = target_bitrate_cb;
+  if (encoders_.size() == 1) {
+    encoders_.front()->SetTargetBitrateCallback(target_bitrate_cb);
+  } else {
+    LOG(ERROR) << "Dropping target bitrate request callback with unexpected"
+                  " number of encoders"
+               << encoders_.size();
+  }
+}
 }  // namespace remoting