Chromium Code Reviews| Index: media/cast/video_sender/codecs/vp8/vp8_encoder.cc |
| diff --git a/media/cast/video_sender/codecs/vp8/vp8_encoder.cc b/media/cast/video_sender/codecs/vp8/vp8_encoder.cc |
| index 38c7dfccaf62335f30ecdc8fab3231e961ec5f1f..2a82664a6c907fc447f910b0769188f57c94a7ef 100644 |
| --- a/media/cast/video_sender/codecs/vp8/vp8_encoder.cc |
| +++ b/media/cast/video_sender/codecs/vp8/vp8_encoder.cc |
| @@ -135,7 +135,7 @@ void Vp8Encoder::InitEncode(int number_of_encode_threads) { |
| } |
| bool Vp8Encoder::Encode(const scoped_refptr<media::VideoFrame>& video_frame, |
| - transport::EncodedVideoFrame* encoded_image) { |
| + transport::EncodedFrame* encoded_image) { |
| DCHECK(thread_checker_.CalledOnValidThread()); |
| // Image in vpx_image_t format. |
| // Input image is const. VP8's raw image is not defined as const. |
| @@ -199,35 +199,35 @@ bool Vp8Encoder::Encode(const scoped_refptr<media::VideoFrame>& video_frame, |
| // Get encoded frame. |
| const vpx_codec_cx_pkt_t* pkt = NULL; |
| vpx_codec_iter_t iter = NULL; |
| - size_t total_size = 0; |
| + bool is_key_frame = false; |
| + encoded_image->data.clear(); |
| while ((pkt = vpx_codec_get_cx_data(encoder_.get(), &iter)) != NULL) { |
| - if (pkt->kind == VPX_CODEC_CX_FRAME_PKT) { |
| - total_size += pkt->data.frame.sz; |
| - encoded_image->data.reserve(total_size); |
| - encoded_image->data.insert( |
| - encoded_image->data.end(), |
| - static_cast<const uint8*>(pkt->data.frame.buf), |
| - static_cast<const uint8*>(pkt->data.frame.buf) + pkt->data.frame.sz); |
| - if (pkt->data.frame.flags & VPX_FRAME_IS_KEY) { |
| - encoded_image->key_frame = true; |
| - } else { |
| - encoded_image->key_frame = false; |
| - } |
| - } |
| + if (pkt->kind != VPX_CODEC_CX_FRAME_PKT) |
| + continue; |
| + encoded_image->data.insert( |
| + encoded_image->data.end(), |
| + static_cast<const uint8*>(pkt->data.frame.buf), |
| + static_cast<const uint8*>(pkt->data.frame.buf) + pkt->data.frame.sz); |
| + is_key_frame |= !!(pkt->data.frame.flags & VPX_FRAME_IS_KEY); |
|
Alpha Left Google
2014/05/16 18:35:22
There is only one packet of kind VPX_CODEC_CX_FRAM
miu
2014/05/16 21:28:46
Done.
|
| } |
| // Don't update frame_id for zero size frames. |
| - if (total_size == 0) |
| + if (encoded_image->data.empty()) |
| return true; |
| // Populate the encoded frame. |
| - encoded_image->codec = transport::kVp8; |
| - encoded_image->last_referenced_frame_id = latest_frame_id_to_reference; |
| encoded_image->frame_id = ++last_encoded_frame_id_; |
| + if (is_key_frame) { |
| + encoded_image->relationship = transport::EncodedFrame::KEY; |
| + encoded_image->referenced_frame_id = encoded_image->frame_id; |
| + } else { |
| + encoded_image->relationship = transport::EncodedFrame::DEPENDENT; |
| + encoded_image->referenced_frame_id = latest_frame_id_to_reference; |
| + } |
| - VLOG(1) << "VP8 encoded frame:" << static_cast<int>(encoded_image->frame_id) |
| - << " sized:" << total_size; |
| + DVLOG(1) << "VP8 encoded frame_id " << encoded_image->frame_id |
| + << ", sized:" << encoded_image->data.size(); |
| - if (encoded_image->key_frame) { |
| + if (is_key_frame) { |
| key_frame_requested_ = false; |
| for (int i = 0; i < kNumberOfVp8VideoBuffers; ++i) { |