• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Copyright 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4 
5 #ifndef CONTENT_RENDERER_MEDIA_RTC_VIDEO_ENCODER_H_
6 #define CONTENT_RENDERER_MEDIA_RTC_VIDEO_ENCODER_H_
7 
8 #include <vector>
9 
10 #include "base/memory/ref_counted.h"
11 #include "base/memory/scoped_ptr.h"
12 #include "base/memory/weak_ptr.h"
13 #include "base/threading/thread_checker.h"
14 #include "base/time/time.h"
15 #include "content/common/content_export.h"
16 #include "media/base/video_decoder_config.h"
17 #include "third_party/webrtc/modules/video_coding/codecs/interface/video_codec_interface.h"
18 #include "ui/gfx/size.h"
19 
20 namespace base {
21 
22 class MessageLoopProxy;
23 
24 }  // namespace base
25 
26 namespace content {
27 
28 class RendererGpuVideoAcceleratorFactories;
29 
30 // RTCVideoEncoder uses a media::VideoEncodeAccelerator to implement a
31 // webrtc::VideoEncoder class for WebRTC.  Internally, VEA methods are
32 // trampolined to a private RTCVideoEncoder::Impl instance.  The Impl class runs
33 // on the worker thread queried from the |gpu_factories_|, which is presently
34 // the media thread.  RTCVideoEncoder itself is run and destroyed on the thread
35 // it is constructed on, which is presently the libjingle worker thread.
36 // Callbacks from the Impl due to its VEA::Client notifications are also posted
37 // back to RTCVideoEncoder on this thread.
38 class CONTENT_EXPORT RTCVideoEncoder
NON_EXPORTED_BASE(public webrtc::VideoEncoder)39     : NON_EXPORTED_BASE(public webrtc::VideoEncoder) {
40  public:
41   RTCVideoEncoder(
42       webrtc::VideoCodecType type,
43       media::VideoCodecProfile profile,
44       const scoped_refptr<RendererGpuVideoAcceleratorFactories>& gpu_factories);
45   virtual ~RTCVideoEncoder();
46 
47   // webrtc::VideoEncoder implementation.  Tasks are posted to |impl_| using the
48   // appropriate VEA methods.
49   virtual int32_t InitEncode(const webrtc::VideoCodec* codec_settings,
50                              int32_t number_of_cores,
51                              uint32_t max_payload_size) OVERRIDE;
52   virtual int32_t Encode(
53       const webrtc::I420VideoFrame& input_image,
54       const webrtc::CodecSpecificInfo* codec_specific_info,
55       const std::vector<webrtc::VideoFrameType>* frame_types) OVERRIDE;
56   virtual int32_t RegisterEncodeCompleteCallback(
57       webrtc::EncodedImageCallback* callback) OVERRIDE;
58   virtual int32_t Release() OVERRIDE;
59   virtual int32_t SetChannelParameters(uint32_t packet_loss, int rtt) OVERRIDE;
60   virtual int32_t SetRates(uint32_t new_bit_rate, uint32_t frame_rate) OVERRIDE;
61 
62  private:
63   class Impl;
64   friend class RTCVideoEncoder::Impl;
65 
66   // Return an encoded output buffer to WebRTC.
67   void ReturnEncodedImage(scoped_ptr<webrtc::EncodedImage> image,
68                           int32 bitstream_buffer_id);
69 
70   void NotifyError(int32_t error);
71 
72   void RecordInitEncodeUMA(int32_t init_retval);
73 
74   base::ThreadChecker thread_checker_;
75 
76   // The video codec type, as reported to WebRTC.
77   const webrtc::VideoCodecType video_codec_type_;
78 
79   // The video codec profile, to configure the encoder to encode to.
80   const media::VideoCodecProfile video_codec_profile_;
81 
82   // Factory for creating VEAs, shared memory buffers, etc.
83   scoped_refptr<RendererGpuVideoAcceleratorFactories> gpu_factories_;
84 
85   // webrtc::VideoEncoder encode complete callback.
86   webrtc::EncodedImageCallback* encoded_image_callback_;
87 
88   // The RTCVideoEncoder::Impl that does all the work.
89   scoped_refptr<Impl> impl_;
90 
91   // We cannot immediately return error conditions to the WebRTC user of this
92   // class, as there is no error callback in the webrtc::VideoEncoder interface.
93   // Instead, we cache an error status here and return it the next time an
94   // interface entry point is called.
95   int32_t impl_status_;
96 
97   // Weak pointer factory for posting back VEA::Client notifications to
98   // RTCVideoEncoder.
99   base::WeakPtrFactory<RTCVideoEncoder> weak_this_factory_;
100 
101   DISALLOW_COPY_AND_ASSIGN(RTCVideoEncoder);
102 };
103 
104 }  // namespace content
105 
106 #endif  // CONTENT_RENDERER_MEDIA_RTC_VIDEO_ENCODER_H_
107