Home | History | Annotate | Download | only in video_sender 
      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 MEDIA_CAST_VIDEO_SENDER_VIDEO_ENCODER_H_
      6 #define MEDIA_CAST_VIDEO_SENDER_VIDEO_ENCODER_H_
      7 
      8 #include "base/memory/scoped_ptr.h"
      9 #include "base/memory/weak_ptr.h"
     10 #include "base/message_loop/message_loop.h"
     11 #include "media/cast/cast_config.h"
     12 #include "media/cast/cast_environment.h"
     13 #include "media/cast/video_sender/codecs/vp8/vp8_encoder.h"
     14 
     15 namespace media {
     16 class VideoFrame;
     17 }
     18 
     19 namespace media {
     20 namespace cast {
     21 
     22 // This object is called external from the main cast thread and internally from
     23 // the video encoder thread.
     24 class VideoEncoder : public VideoEncoderController {
     25  public:
     26   typedef base::Callback<void(scoped_ptr<EncodedVideoFrame>,
     27                               const base::TimeTicks&)> FrameEncodedCallback;
     28 
     29   VideoEncoder(scoped_refptr<CastEnvironment> cast_environment,
     30                const VideoSenderConfig& video_config,
     31                uint8 max_unacked_frames);
     32 
     33   virtual ~VideoEncoder();
     34 
     35   // Called from the main cast thread. This function post the encode task to the
     36   // video encoder thread;
     37   // The video_frame must be valid until the closure callback is called.
     38   // The closure callback is called from the video encoder thread as soon as
     39   // the encoder is done with the frame; it does not mean that the encoded frame
     40   // has been sent out.
     41   // Once the encoded frame is ready the frame_encoded_callback is called.
     42   bool EncodeVideoFrame(const scoped_refptr<media::VideoFrame>& video_frame,
     43                         const base::TimeTicks& capture_time,
     44                         const FrameEncodedCallback& frame_encoded_callback);
     45 
     46  protected:
     47   struct CodecDynamicConfig {
     48     bool key_frame_requested;
     49     uint32 latest_frame_id_to_reference;
     50     int bit_rate;
     51   };
     52 
     53   // The actual encode, called from the video encoder thread.
     54   void EncodeVideoFrameEncoderThread(
     55       const scoped_refptr<media::VideoFrame>& video_frame,
     56       const base::TimeTicks& capture_time,
     57       const CodecDynamicConfig& dynamic_config,
     58       const FrameEncodedCallback& frame_encoded_callback);
     59 
     60   // The following functions are called from the main cast thread.
     61   virtual void SetBitRate(int new_bit_rate) OVERRIDE;
     62   virtual void SkipNextFrame(bool skip_next_frame) OVERRIDE;
     63   virtual void GenerateKeyFrame() OVERRIDE;
     64   virtual void LatestFrameIdToReference(uint32 frame_id) OVERRIDE;
     65   virtual int NumberOfSkippedFrames() const OVERRIDE;
     66 
     67  private:
     68   friend class base::RefCountedThreadSafe<VideoEncoder>;
     69 
     70   const VideoSenderConfig video_config_;
     71   scoped_refptr<CastEnvironment> cast_environment_;
     72   scoped_ptr<Vp8Encoder> vp8_encoder_;
     73   CodecDynamicConfig dynamic_config_;
     74   bool skip_next_frame_;
     75   int skip_count_;
     76 
     77   DISALLOW_COPY_AND_ASSIGN(VideoEncoder);
     78 };
     79 
     80 }  // namespace cast
     81 }  // namespace media
     82 
     83 #endif  // MEDIA_CAST_VIDEO_SENDER_VIDEO_ENCODER_H_
     84