Index: remoting/protocol/webrtc_dummy_video_encoder.h |
diff --git a/remoting/protocol/webrtc_dummy_video_encoder.h b/remoting/protocol/webrtc_dummy_video_encoder.h |
index 0f1a7344d761f76fdb0fea2208e9686e1a3d81e3..4f1bfdce29b65b1ff377736ef32755cce47815ef 100644 |
--- a/remoting/protocol/webrtc_dummy_video_encoder.h |
+++ b/remoting/protocol/webrtc_dummy_video_encoder.h |
@@ -2,13 +2,15 @@ |
// Use of this source code is governed by a BSD-style license that can be |
// found in the LICENSE file. |
-#ifndef REMOTING_PROTOCOL_DUMMY_WEBRTC_VIDEO_ENCODER_H_ |
-#define REMOTING_PROTOCOL_DUMMY_WEBRTC_VIDEO_ENCODER_H_ |
+#ifndef REMOTING_PROTOCOL_WEBRTC_DUMMY_VIDEO_ENCODER_H_ |
+#define REMOTING_PROTOCOL_WEBRTC_DUMMY_VIDEO_ENCODER_H_ |
#include <memory> |
#include <vector> |
#include "base/callback.h" |
+#include "base/memory/ref_counted.h" |
+#include "base/memory/weak_ptr.h" |
#include "base/synchronization/lock.h" |
#include "remoting/codec/webrtc_video_encoder.h" |
#include "third_party/webrtc/media/engine/webrtcvideoencoderfactory.h" |
@@ -17,7 +19,7 @@ |
namespace remoting { |
namespace protocol { |
-using TargetBitrateCallback = base::Callback<void(int)>; |
+class NetworkStateObserver; |
// WebrtcDummyVideoEncoder implements webrtc::VideoEncoder interface for WebRTC. |
// It's responsible for getting feedback on network bandwidth, latency & RTT |
@@ -28,7 +30,10 @@ using TargetBitrateCallback = base::Callback<void(int)>; |
class WebrtcDummyVideoEncoder : public webrtc::VideoEncoder { |
public: |
enum State { kUninitialized = 0, kInitialized }; |
- explicit WebrtcDummyVideoEncoder(webrtc::VideoCodecType codec); |
+ |
+ WebrtcDummyVideoEncoder( |
+ scoped_refptr<base::SingleThreadTaskRunner> main_task_runner, |
+ base::WeakPtr<NetworkStateObserver> network_state_observer); |
~WebrtcDummyVideoEncoder() override; |
// webrtc::VideoEncoder overrides |
@@ -47,19 +52,16 @@ class WebrtcDummyVideoEncoder : public webrtc::VideoEncoder { |
webrtc::EncodedImageCallback::Result SendEncodedFrame( |
const WebrtcVideoEncoder::EncodedFrame& frame, |
base::TimeTicks capture_time); |
- void SetKeyFrameRequestCallback(const base::Closure& key_frame_request); |
- void SetTargetBitrateCallback(const TargetBitrateCallback& target_bitrate_cb); |
private: |
- // Protects |encoded_callback_|, |key_frame_request_|, |
- // |target_bitrate_cb_| and |state_|. |
+ scoped_refptr<base::SingleThreadTaskRunner> main_task_runner_; |
+ |
+ // Protects |encoded_callback_| and |state_|. |
base::Lock lock_; |
State state_; |
- webrtc::EncodedImageCallback* encoded_callback_; |
+ webrtc::EncodedImageCallback* encoded_callback_ = nullptr; |
- base::Closure key_frame_request_; |
- TargetBitrateCallback target_bitrate_cb_; |
- webrtc::VideoCodecType video_codec_type_; |
+ base::WeakPtr<NetworkStateObserver> network_state_observer_; |
}; |
// This is the external encoder factory implementation that is passed to |
@@ -71,12 +73,11 @@ class WebrtcDummyVideoEncoderFactory |
WebrtcDummyVideoEncoderFactory(); |
~WebrtcDummyVideoEncoderFactory() override; |
+ // cricket::WebRtcVideoEncoderFactory interface. |
webrtc::VideoEncoder* CreateVideoEncoder( |
webrtc::VideoCodecType type) override; |
const std::vector<cricket::WebRtcVideoEncoderFactory::VideoCodec>& codecs() |
const override; |
- |
- // Returns true to directly provide encoded frames to webrtc |
bool EncoderTypeHasInternalSource(webrtc::VideoCodecType type) const override; |
void DestroyVideoEncoder(webrtc::VideoEncoder* encoder) override; |
@@ -84,19 +85,21 @@ class WebrtcDummyVideoEncoderFactory |
const WebrtcVideoEncoder::EncodedFrame& packet, |
base::TimeTicks capture_time); |
- void SetKeyFrameRequestCallback(const base::Closure& key_frame_request); |
- void SetTargetBitrateCallback(const TargetBitrateCallback& target_bitrate_cb); |
+ void SetNetworkStateObserver( |
+ base::WeakPtr<NetworkStateObserver> network_state_observer); |
private: |
- // Protects |key_frame_request_| and |target_bitrate_cb_|. |
- base::Lock lock_; |
- base::Closure key_frame_request_; |
- TargetBitrateCallback target_bitrate_cb_; |
+ scoped_refptr<base::SingleThreadTaskRunner> main_task_runner_; |
+ |
std::vector<cricket::WebRtcVideoEncoderFactory::VideoCodec> codecs_; |
+ |
+ // Protects |network_state_observer_| and |encoders_|. |
+ base::Lock lock_; |
+ base::WeakPtr<NetworkStateObserver> network_state_observer_; |
std::vector<std::unique_ptr<WebrtcDummyVideoEncoder>> encoders_; |
}; |
} // namespace protocol |
} // namespace remoting |
-#endif // REMOTING_PROTOCOL_DUMMY_WEBRTC_VIDEO_ENCODER_H_ |
+#endif // REMOTING_PROTOCOL_WEBRTC_DUMMY_VIDEO_ENCODER_H_ |