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