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 media { 27 28 class GpuVideoAcceleratorFactories; 29 30 } // namespace media 31 32 namespace content { 33 34 // RTCVideoEncoder uses a media::VideoEncodeAccelerator to implement a 35 // webrtc::VideoEncoder class for WebRTC. Internally, VEA methods are 36 // trampolined to a private RTCVideoEncoder::Impl instance. The Impl class runs 37 // on the worker thread queried from the |gpu_factories_|, which is presently 38 // the media thread. RTCVideoEncoder is sychronized by webrtc::VideoSender. 39 // webrtc::VideoEncoder methods do not run concurrently. RTCVideoEncoder is run 40 // and destroyed on the thread it is constructed on, which is presently the 41 // libjingle worker thread. Encode is run on ViECaptureThread. SetRates and 42 // SetChannelParameters are run on ProcessThread or the libjingle worker thread. 43 // Callbacks from the Impl due to its VEA::Client notifications are posted back 44 // to RTCVideoEncoder on the libjingle worker thread. 45 class CONTENT_EXPORT RTCVideoEncoder NON_EXPORTED_BASE(public webrtc::VideoEncoder)46 : NON_EXPORTED_BASE(public webrtc::VideoEncoder) { 47 public: 48 RTCVideoEncoder( 49 webrtc::VideoCodecType type, 50 media::VideoCodecProfile profile, 51 const scoped_refptr<media::GpuVideoAcceleratorFactories>& gpu_factories); 52 virtual ~RTCVideoEncoder(); 53 54 // webrtc::VideoEncoder implementation. Tasks are posted to |impl_| using the 55 // appropriate VEA methods. 56 virtual int32_t InitEncode(const webrtc::VideoCodec* codec_settings, 57 int32_t number_of_cores, 58 uint32_t max_payload_size) OVERRIDE; 59 virtual int32_t Encode( 60 const webrtc::I420VideoFrame& input_image, 61 const webrtc::CodecSpecificInfo* codec_specific_info, 62 const std::vector<webrtc::VideoFrameType>* frame_types) OVERRIDE; 63 virtual int32_t RegisterEncodeCompleteCallback( 64 webrtc::EncodedImageCallback* callback) OVERRIDE; 65 virtual int32_t Release() OVERRIDE; 66 virtual int32_t SetChannelParameters(uint32_t packet_loss, int rtt) OVERRIDE; 67 virtual int32_t SetRates(uint32_t new_bit_rate, uint32_t frame_rate) OVERRIDE; 68 69 private: 70 class Impl; 71 friend class RTCVideoEncoder::Impl; 72 73 // Return an encoded output buffer to WebRTC. 74 void ReturnEncodedImage(scoped_ptr<webrtc::EncodedImage> image, 75 int32 bitstream_buffer_id, 76 uint16 picture_id); 77 78 void NotifyError(int32_t error); 79 80 void RecordInitEncodeUMA(int32_t init_retval); 81 82 base::ThreadChecker thread_checker_; 83 84 // The video codec type, as reported to WebRTC. 85 const webrtc::VideoCodecType video_codec_type_; 86 87 // The video codec profile, to configure the encoder to encode to. 88 const media::VideoCodecProfile video_codec_profile_; 89 90 // Factory for creating VEAs, shared memory buffers, etc. 91 scoped_refptr<media::GpuVideoAcceleratorFactories> gpu_factories_; 92 93 // webrtc::VideoEncoder encode complete callback. 94 webrtc::EncodedImageCallback* encoded_image_callback_; 95 96 // The RTCVideoEncoder::Impl that does all the work. 97 scoped_refptr<Impl> impl_; 98 99 // We cannot immediately return error conditions to the WebRTC user of this 100 // class, as there is no error callback in the webrtc::VideoEncoder interface. 101 // Instead, we cache an error status here and return it the next time an 102 // interface entry point is called. 103 int32_t impl_status_; 104 105 // Weak pointer factory for posting back VEA::Client notifications to 106 // RTCVideoEncoder. 107 // NOTE: Weak pointers must be invalidated before all other member variables. 108 base::WeakPtrFactory<RTCVideoEncoder> weak_factory_; 109 110 DISALLOW_COPY_AND_ASSIGN(RTCVideoEncoder); 111 }; 112 113 } // namespace content 114 115 #endif // CONTENT_RENDERER_MEDIA_RTC_VIDEO_ENCODER_H_ 116