media/cast/sender/external_video_encoder.h

   1 // Copyright 2014 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 MEDIA_CAST_SENDER_EXTERNAL_VIDEO_ENCODER_H_
   6 #define MEDIA_CAST_SENDER_EXTERNAL_VIDEO_ENCODER_H_
   7
   8 #include "base/memory/scoped_ptr.h"
   9 #include "base/memory/weak_ptr.h"
  10 #include "media/cast/cast_config.h"
  11 #include "media/cast/cast_environment.h"
  12 #include "media/cast/sender/size_adaptable_video_encoder_base.h"
  13 #include "media/cast/sender/video_encoder.h"
  14 #include "media/video/video_encode_accelerator.h"
  15 #include "ui/gfx/geometry/size.h"
  16
  17 namespace media {
  18 namespace cast {
  19
  20 // Cast MAIN thread proxy to the internal media::VideoEncodeAccelerator
  21 // implementation running on a separate thread.  Encodes media::VideoFrames and
  22 // emits media::cast::EncodedFrames.
  23 class ExternalVideoEncoder : public VideoEncoder {
  24  public:
  25   // Returns true if the current platform and system configuration supports
  26   // using ExternalVideoEncoder with the given |video_config|.
  27   static bool IsSupported(const VideoSenderConfig& video_config);
  28
  29   ExternalVideoEncoder(
  30       const scoped_refptr<CastEnvironment>& cast_environment,
  31       const VideoSenderConfig& video_config,
  32       const gfx::Size& frame_size,
  33       uint32 first_frame_id,
  34       const StatusChangeCallback& status_change_cb,
  35       const CreateVideoEncodeAcceleratorCallback& create_vea_cb,
  36       const CreateVideoEncodeMemoryCallback& create_video_encode_memory_cb);
  37
  38   ~ExternalVideoEncoder() final;
  39
  40   // VideoEncoder implementation.
  41   bool EncodeVideoFrame(
  42       const scoped_refptr<media::VideoFrame>& video_frame,
  43       const base::TimeTicks& reference_time,
  44       const FrameEncodedCallback& frame_encoded_callback) final;
  45   void SetBitRate(int new_bit_rate) final;
  46   void GenerateKeyFrame() final;
  47   void LatestFrameIdToReference(uint32 frame_id) final;
  48
  49  private:
  50   class VEAClientImpl;
  51
  52   // Method invoked by the CreateVideoEncodeAcceleratorCallback to construct a
  53   // VEAClientImpl to own and interface with a new |vea|.  Upon return,
  54   // |client_| holds a reference to the new VEAClientImpl.
  55   void OnCreateVideoEncodeAccelerator(
  56       const VideoSenderConfig& video_config,
  57       uint32 first_frame_id,
  58       const StatusChangeCallback& status_change_cb,
  59       scoped_refptr<base::SingleThreadTaskRunner> encoder_task_runner,
  60       scoped_ptr<media::VideoEncodeAccelerator> vea);
  61
  62   const scoped_refptr<CastEnvironment> cast_environment_;
  63   const CreateVideoEncodeMemoryCallback create_video_encode_memory_cb_;
  64
  65   // The size of the visible region of the video frames to be encoded.
  66   const gfx::Size frame_size_;
  67
  68   int bit_rate_;
  69   bool key_frame_requested_;
  70
  71   scoped_refptr<VEAClientImpl> client_;
  72
  73   // Provides a weak pointer for the OnCreateVideoEncoderAccelerator() callback.
  74   // NOTE: Weak pointers must be invalidated before all other member variables.
  75   base::WeakPtrFactory<ExternalVideoEncoder> weak_factory_;
  76
  77   DISALLOW_COPY_AND_ASSIGN(ExternalVideoEncoder);
  78 };
  79
  80 // An implementation of SizeAdaptableVideoEncoderBase to proxy for
  81 // ExternalVideoEncoder instances.
  82 class SizeAdaptableExternalVideoEncoder : public SizeAdaptableVideoEncoderBase {
  83  public:
  84   SizeAdaptableExternalVideoEncoder(
  85       const scoped_refptr<CastEnvironment>& cast_environment,
  86       const VideoSenderConfig& video_config,
  87       const StatusChangeCallback& status_change_cb,
  88       const CreateVideoEncodeAcceleratorCallback& create_vea_cb,
  89       const CreateVideoEncodeMemoryCallback& create_video_encode_memory_cb);
  90
  91   ~SizeAdaptableExternalVideoEncoder() final;
  92
  93  protected:
  94   scoped_ptr<VideoEncoder> CreateEncoder() final;
  95
  96  private:
  97   // Special callbacks needed by media::cast::ExternalVideoEncoder.
  98   // TODO(miu): Remove these.  http://crbug.com/454029
  99   const CreateVideoEncodeAcceleratorCallback create_vea_cb_;
 100   const CreateVideoEncodeMemoryCallback create_video_encode_memory_cb_;
 101
 102   DISALLOW_COPY_AND_ASSIGN(SizeAdaptableExternalVideoEncoder);
 103 };
 104
 105 }  // namespace cast
 106 }  // namespace media
 107
 108 #endif  // MEDIA_CAST_SENDER_EXTERNAL_VIDEO_ENCODER_H_