Home | History | Annotate | Download | only in media 
      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
     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