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) { |