OLD | NEW |
---|---|
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef REMOTING_PROTOCOL_WEBRTC_VIDEO_ENCODER_H_ | 5 #ifndef REMOTING_PROTOCOL_WEBRTC_VIDEO_ENCODER_H_ |
6 #define REMOTING_PROTOCOL_WEBRTC_VIDEO_ENCODER_H_ | 6 #define REMOTING_PROTOCOL_WEBRTC_VIDEO_ENCODER_H_ |
7 | 7 |
8 #include <memory> | 8 #include <memory> |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
11 #include "base/callback.h" | 11 #include "base/callback.h" |
12 #include "base/synchronization/lock.h" | 12 #include "base/synchronization/lock.h" |
13 #include "remoting/proto/video.pb.h" | 13 #include "remoting/proto/video.pb.h" |
14 #include "third_party/webrtc/media/engine/webrtcvideoencoderfactory.h" | 14 #include "third_party/webrtc/media/engine/webrtcvideoencoderfactory.h" |
15 #include "third_party/webrtc/modules/video_coding/include/video_codec_interface. h" | 15 #include "third_party/webrtc/modules/video_coding/include/video_codec_interface. h" |
16 | 16 |
17 namespace remoting { | 17 namespace remoting { |
18 | 18 |
19 using TargetBitrateCallback = base::Callback<void(uint32_t)>; | |
Sergey Ulanov
2016/05/10 00:00:58
s/uint32_t/int bitrate_kbps/
Irfan
2016/05/10 16:30:45
Done.
| |
20 | |
19 // This is the interface between the WebRtc engine and the external encoder | 21 // This is the interface between the WebRtc engine and the external encoder |
20 // provided by remoting. WebRtc provides feedback on network bandwidth, latency | 22 // provided by remoting. WebRtc provides feedback on network bandwidth, latency |
21 // & RTT and in turn remoting passes encoded frames as they get encoded | 23 // & RTT and in turn remoting passes encoded frames as they get encoded |
22 // through the capture pipeline. | 24 // through the capture pipeline. |
23 class WebRtcVideoEncoder : public webrtc::VideoEncoder { | 25 class WebRtcVideoEncoder : public webrtc::VideoEncoder { |
Sergey Ulanov
2016/05/10 00:00:58
maybe move this to the .cc file.
Irfan
2016/05/10 16:30:45
hm, think i prefer it here unless you strongly fee
| |
24 public: | 26 public: |
25 enum State { kUninitialized = 0, kInitialized }; | 27 enum State { kUninitialized = 0, kInitialized }; |
26 explicit WebRtcVideoEncoder(webrtc::VideoCodecType codec); | 28 explicit WebRtcVideoEncoder(webrtc::VideoCodecType codec); |
27 ~WebRtcVideoEncoder() override; | 29 ~WebRtcVideoEncoder() override; |
28 | 30 |
29 // webrtc::VideoEncoder overrides | 31 // webrtc::VideoEncoder overrides |
30 int32_t InitEncode(const webrtc::VideoCodec* codec_settings, | 32 int32_t InitEncode(const webrtc::VideoCodec* codec_settings, |
31 int32_t number_of_cores, | 33 int32_t number_of_cores, |
32 size_t max_payload_size) override; | 34 size_t max_payload_size) override; |
33 int32_t RegisterEncodeCompleteCallback( | 35 int32_t RegisterEncodeCompleteCallback( |
34 webrtc::EncodedImageCallback* callback) override; | 36 webrtc::EncodedImageCallback* callback) override; |
35 int32_t Release() override; | 37 int32_t Release() override; |
36 int32_t Encode(const webrtc::VideoFrame& frame, | 38 int32_t Encode(const webrtc::VideoFrame& frame, |
37 const webrtc::CodecSpecificInfo* codec_specific_info, | 39 const webrtc::CodecSpecificInfo* codec_specific_info, |
38 const std::vector<webrtc::FrameType>* frame_types) override; | 40 const std::vector<webrtc::FrameType>* frame_types) override; |
39 int32_t SetChannelParameters(uint32_t packet_loss, int64_t rtt) override; | 41 int32_t SetChannelParameters(uint32_t packet_loss, int64_t rtt) override; |
40 int32_t SetRates(uint32_t bitrate, uint32_t framerate) override; | 42 int32_t SetRates(uint32_t bitrate, uint32_t framerate) override; |
41 | 43 |
42 int SendEncodedFrame(int64_t capture_timestamp_ms, | 44 int SendEncodedFrame(std::unique_ptr<VideoPacket> pkt); |
43 std::unique_ptr<VideoPacket> pkt); | |
44 void SetKeyFrameRequestCallback(const base::Closure& key_frame_request); | 45 void SetKeyFrameRequestCallback(const base::Closure& key_frame_request); |
46 void SetTargetBitrateCallback(const TargetBitrateCallback& target_bitrate_cb); | |
45 | 47 |
46 private: | 48 private: |
47 // Protects |encoded_callback_|, |key_frame_request_| and |state_|. | 49 // Protects |encoded_callback_|, |key_frame_request_|, |
50 // |target_bitrate_cb_| and |state_|. | |
48 base::Lock lock_; | 51 base::Lock lock_; |
49 State state_; | 52 State state_; |
50 webrtc::EncodedImageCallback* encoded_callback_; | 53 webrtc::EncodedImageCallback* encoded_callback_; |
51 | 54 |
52 base::Closure key_frame_request_; | 55 base::Closure key_frame_request_; |
56 TargetBitrateCallback target_bitrate_cb_; | |
53 webrtc::VideoCodecType video_codec_type_; | 57 webrtc::VideoCodecType video_codec_type_; |
54 uint32_t target_bitrate_; | |
55 }; | 58 }; |
56 | 59 |
57 // This is the external encoder factory implementation that is passed to | 60 // This is the external encoder factory implementation that is passed to |
58 // WebRTC at the time of creation of peer connection. The external encoder | 61 // WebRTC at the time of creation of peer connection. The external encoder |
59 // factory primarily manages creation and destruction of encoder. | 62 // factory primarily manages creation and destruction of encoder. |
60 class WebRtcVideoEncoderFactory : public cricket::WebRtcVideoEncoderFactory { | 63 class WebRtcVideoEncoderFactory : public cricket::WebRtcVideoEncoderFactory { |
61 public: | 64 public: |
62 WebRtcVideoEncoderFactory(); | 65 WebRtcVideoEncoderFactory(); |
63 ~WebRtcVideoEncoderFactory() override; | 66 ~WebRtcVideoEncoderFactory() override; |
64 | 67 |
65 webrtc::VideoEncoder* CreateVideoEncoder( | 68 webrtc::VideoEncoder* CreateVideoEncoder( |
66 webrtc::VideoCodecType type) override; | 69 webrtc::VideoCodecType type) override; |
67 const std::vector<cricket::WebRtcVideoEncoderFactory::VideoCodec>& codecs() | 70 const std::vector<cricket::WebRtcVideoEncoderFactory::VideoCodec>& codecs() |
68 const override; | 71 const override; |
69 | 72 |
70 // Returns true to directly provide encoded frames to webrtc | 73 // Returns true to directly provide encoded frames to webrtc |
71 bool EncoderTypeHasInternalSource(webrtc::VideoCodecType type) const override; | 74 bool EncoderTypeHasInternalSource(webrtc::VideoCodecType type) const override; |
72 void DestroyVideoEncoder(webrtc::VideoEncoder* encoder) override; | 75 void DestroyVideoEncoder(webrtc::VideoEncoder* encoder) override; |
73 | 76 |
74 int SendEncodedFrame(int64_t capture_timestamp_ms, | 77 int SendEncodedFrame(std::unique_ptr<VideoPacket> pkt); |
75 std::unique_ptr<VideoPacket> pkt); | |
76 | 78 |
77 void SetKeyFrameRequestCallback(const base::Closure& key_frame_request); | 79 void SetKeyFrameRequestCallback(const base::Closure& key_frame_request); |
80 void SetTargetBitrateCallback(const TargetBitrateCallback& target_bitrate_cb); | |
78 | 81 |
79 private: | 82 private: |
80 // Protects |key_frame_request_|. | 83 // Protects |key_frame_request_| and |target_bitrate_cb_|. |
81 base::Lock lock_; | 84 base::Lock lock_; |
82 base::Closure key_frame_request_; | 85 base::Closure key_frame_request_; |
86 TargetBitrateCallback target_bitrate_cb_; | |
83 std::vector<cricket::WebRtcVideoEncoderFactory::VideoCodec> codecs_; | 87 std::vector<cricket::WebRtcVideoEncoderFactory::VideoCodec> codecs_; |
84 std::vector<std::unique_ptr<WebRtcVideoEncoder>> encoders_; | 88 std::vector<std::unique_ptr<WebRtcVideoEncoder>> encoders_; |
85 }; | 89 }; |
86 | 90 |
87 } // namespace remoting | 91 } // namespace remoting |
88 | 92 |
89 #endif // REMOTING_PROTOCOL_WEBRTC_VIDEO_ENCODER_H_ | 93 #endif // REMOTING_PROTOCOL_WEBRTC_VIDEO_ENCODER_H_ |
OLD | NEW |