| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 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_SENDER_CODECS_VP8_VP8_ENCODER_H_ | 5 #ifndef MEDIA_CAST_SENDER_CODECS_VP8_VP8_ENCODER_H_ |
| 6 #define MEDIA_CAST_SENDER_CODECS_VP8_VP8_ENCODER_H_ | 6 #define MEDIA_CAST_SENDER_CODECS_VP8_VP8_ENCODER_H_ |
| 7 | 7 |
| 8 #include "base/basictypes.h" | 8 #include "base/basictypes.h" |
| 9 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
| 10 #include "base/threading/thread_checker.h" | 10 #include "base/threading/thread_checker.h" |
| 11 #include "media/cast/cast_config.h" | 11 #include "media/cast/cast_config.h" |
| 12 #include "media/cast/sender/software_video_encoder.h" | 12 #include "media/cast/sender/software_video_encoder.h" |
| 13 #include "third_party/libvpx/source/libvpx/vpx/vpx_encoder.h" | 13 #include "third_party/libvpx/source/libvpx/vpx/vpx_encoder.h" |
| 14 #include "ui/gfx/geometry/size.h" | |
| 15 | 14 |
| 16 namespace media { | 15 namespace media { |
| 17 class VideoFrame; | 16 class VideoFrame; |
| 18 } | 17 } |
| 19 | 18 |
| 20 namespace media { | 19 namespace media { |
| 21 namespace cast { | 20 namespace cast { |
| 22 | 21 |
| 23 class Vp8Encoder : public SoftwareVideoEncoder { | 22 class Vp8Encoder : public SoftwareVideoEncoder { |
| 24 public: | 23 public: |
| (...skipping 25 matching lines...) Expand all Loading... |
| 50 kBufferSent, | 49 kBufferSent, |
| 51 kBufferAcked | 50 kBufferAcked |
| 52 }; | 51 }; |
| 53 | 52 |
| 54 struct BufferState { | 53 struct BufferState { |
| 55 uint32 frame_id; | 54 uint32 frame_id; |
| 56 Vp8BufferState state; | 55 Vp8BufferState state; |
| 57 }; | 56 }; |
| 58 | 57 |
| 59 bool is_initialized() const { | 58 bool is_initialized() const { |
| 60 // ConfigureForNewFrameSize() sets the timebase denominator value to | 59 // Initialize() sets the timebase denominator value to non-zero if the |
| 61 // non-zero if the encoder is successfully initialized, and it is zero | 60 // encoder is successfully initialized, and it is zero otherwise. |
| 62 // otherwise. | |
| 63 return config_.g_timebase.den != 0; | 61 return config_.g_timebase.den != 0; |
| 64 } | 62 } |
| 65 | 63 |
| 66 // If the |encoder_| is live, attempt reconfiguration to allow it to encode | |
| 67 // frames at a new |frame_size|. Otherwise, tear it down and re-create a new | |
| 68 // |encoder_| instance. | |
| 69 void ConfigureForNewFrameSize(const gfx::Size& frame_size); | |
| 70 | |
| 71 // Calculate which next Vp8 buffers to update with the next frame. | 64 // Calculate which next Vp8 buffers to update with the next frame. |
| 72 Vp8Buffers GetNextBufferToUpdate(); | 65 Vp8Buffers GetNextBufferToUpdate(); |
| 73 | 66 |
| 74 // Get encoder flags for our referenced encoder buffers. | 67 // Get encoder flags for our referenced encoder buffers. |
| 75 // Return which previous frame to reference. | 68 // Return which previous frame to reference. |
| 76 uint32 GetCodecReferenceFlags(vpx_codec_flags_t* flags); | 69 uint32 GetCodecReferenceFlags(vpx_codec_flags_t* flags); |
| 77 | 70 |
| 78 // Get encoder flags for our encoder buffers to update with next frame. | 71 // Get encoder flags for our encoder buffers to update with next frame. |
| 79 void GetCodecUpdateFlags(Vp8Buffers buffer_to_update, | 72 void GetCodecUpdateFlags(Vp8Buffers buffer_to_update, |
| 80 vpx_codec_flags_t* flags); | 73 vpx_codec_flags_t* flags); |
| 81 | 74 |
| 82 const VideoSenderConfig cast_config_; | 75 const VideoSenderConfig cast_config_; |
| 83 const bool use_multiple_video_buffers_; | 76 const bool use_multiple_video_buffers_; |
| 84 | 77 |
| 85 // VP8 internal objects. These are valid for use only while is_initialized() | 78 // VP8 internal objects. These are valid for use only while is_initialized() |
| 86 // returns true. | 79 // returns true. |
| 87 vpx_codec_enc_cfg_t config_; | 80 vpx_codec_enc_cfg_t config_; |
| 88 vpx_codec_ctx_t encoder_; | 81 vpx_codec_ctx_t encoder_; |
| 89 | 82 |
| 83 // Wrapper for access to YUV data planes in a media::VideoFrame. |
| 84 vpx_image_t* raw_image_; |
| 85 |
| 90 // Set to true to request the next frame emitted by Vp8Encoder be a key frame. | 86 // Set to true to request the next frame emitted by Vp8Encoder be a key frame. |
| 91 bool key_frame_requested_; | 87 bool key_frame_requested_; |
| 92 | 88 |
| 93 // Saves the current bitrate setting, for when the |encoder_| is reconfigured | |
| 94 // for different frame sizes. | |
| 95 int bitrate_kbit_; | |
| 96 | |
| 97 // The |VideoFrame::timestamp()| of the last encoded frame. This is used to | 89 // The |VideoFrame::timestamp()| of the last encoded frame. This is used to |
| 98 // predict the duration of the next frame. | 90 // predict the duration of the next frame. |
| 99 base::TimeDelta last_frame_timestamp_; | 91 base::TimeDelta last_frame_timestamp_; |
| 100 | 92 |
| 101 // The last encoded frame's ID. | 93 // The last encoded frame's ID. |
| 102 uint32 last_encoded_frame_id_; | 94 uint32 last_encoded_frame_id_; |
| 103 | 95 |
| 104 // Used to track which buffers are old enough to be re-used. | 96 // Used to track which buffers are old enough to be re-used. |
| 105 uint32 last_acked_frame_id_; | 97 uint32 last_acked_frame_id_; |
| 106 | 98 |
| 107 // Used by GetNextBufferToUpdate() to track how many consecutive times the | 99 // Used by GetNextBufferToUpdate() to track how many consecutive times the |
| 108 // newest buffer had to be overwritten. | 100 // newest buffer had to be overwritten. |
| 109 int undroppable_frames_; | 101 int undroppable_frames_; |
| 110 | 102 |
| 111 // Tracks the lifecycle and dependency state of each of the three buffers. | 103 // Tracks the lifecycle and dependency state of each of the three buffers. |
| 112 BufferState buffer_state_[kNumberOfVp8VideoBuffers]; | 104 BufferState buffer_state_[kNumberOfVp8VideoBuffers]; |
| 113 | 105 |
| 114 // This is bound to the thread where Initialize() is called. | 106 // This is bound to the thread where Initialize() is called. |
| 115 base::ThreadChecker thread_checker_; | 107 base::ThreadChecker thread_checker_; |
| 116 | 108 |
| 117 DISALLOW_COPY_AND_ASSIGN(Vp8Encoder); | 109 DISALLOW_COPY_AND_ASSIGN(Vp8Encoder); |
| 118 }; | 110 }; |
| 119 | 111 |
| 120 } // namespace cast | 112 } // namespace cast |
| 121 } // namespace media | 113 } // namespace media |
| 122 | 114 |
| 123 #endif // MEDIA_CAST_SENDER_CODECS_VP8_VP8_ENCODER_H_ | 115 #endif // MEDIA_CAST_SENDER_CODECS_VP8_VP8_ENCODER_H_ |
| OLD | NEW |