| OLD | NEW |
| (Empty) |
| 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 #include "media/cast/framer/frame_buffer.h" | |
| 6 | |
| 7 #include "base/logging.h" | |
| 8 | |
| 9 namespace media { | |
| 10 namespace cast { | |
| 11 | |
| 12 FrameBuffer::FrameBuffer() | |
| 13 : frame_id_(0), | |
| 14 max_packet_id_(0), | |
| 15 num_packets_received_(0), | |
| 16 is_key_frame_(false), | |
| 17 total_data_size_(0), | |
| 18 last_referenced_frame_id_(0), | |
| 19 packets_() {} | |
| 20 | |
| 21 FrameBuffer::~FrameBuffer() {} | |
| 22 | |
| 23 void FrameBuffer::InsertPacket(const uint8* payload_data, | |
| 24 size_t payload_size, | |
| 25 const RtpCastHeader& rtp_header) { | |
| 26 // Is this the first packet in the frame? | |
| 27 if (packets_.empty()) { | |
| 28 frame_id_ = rtp_header.frame_id; | |
| 29 max_packet_id_ = rtp_header.max_packet_id; | |
| 30 is_key_frame_ = rtp_header.is_key_frame; | |
| 31 if (is_key_frame_) | |
| 32 DCHECK_EQ(rtp_header.frame_id, rtp_header.reference_frame_id); | |
| 33 last_referenced_frame_id_ = rtp_header.reference_frame_id; | |
| 34 rtp_timestamp_ = rtp_header.rtp_timestamp; | |
| 35 } | |
| 36 // Is this the correct frame? | |
| 37 if (rtp_header.frame_id != frame_id_) | |
| 38 return; | |
| 39 | |
| 40 // Insert every packet only once. | |
| 41 if (packets_.find(rtp_header.packet_id) != packets_.end()) { | |
| 42 return; | |
| 43 } | |
| 44 | |
| 45 std::vector<uint8> data; | |
| 46 std::pair<PacketMap::iterator, bool> retval = | |
| 47 packets_.insert(make_pair(rtp_header.packet_id, data)); | |
| 48 | |
| 49 // Insert the packet. | |
| 50 retval.first->second.resize(payload_size); | |
| 51 std::copy( | |
| 52 payload_data, payload_data + payload_size, retval.first->second.begin()); | |
| 53 | |
| 54 ++num_packets_received_; | |
| 55 total_data_size_ += payload_size; | |
| 56 } | |
| 57 | |
| 58 bool FrameBuffer::Complete() const { | |
| 59 return num_packets_received_ - 1 == max_packet_id_; | |
| 60 } | |
| 61 | |
| 62 bool FrameBuffer::AssembleEncodedFrame(transport::EncodedFrame* frame) const { | |
| 63 if (!Complete()) | |
| 64 return false; | |
| 65 | |
| 66 // Frame is complete -> construct. | |
| 67 if (is_key_frame_) | |
| 68 frame->dependency = transport::EncodedFrame::KEY; | |
| 69 else if (frame_id_ == last_referenced_frame_id_) | |
| 70 frame->dependency = transport::EncodedFrame::INDEPENDENT; | |
| 71 else | |
| 72 frame->dependency = transport::EncodedFrame::DEPENDENT; | |
| 73 frame->frame_id = frame_id_; | |
| 74 frame->referenced_frame_id = last_referenced_frame_id_; | |
| 75 frame->rtp_timestamp = rtp_timestamp_; | |
| 76 | |
| 77 // Build the data vector. | |
| 78 frame->data.clear(); | |
| 79 frame->data.reserve(total_data_size_); | |
| 80 PacketMap::const_iterator it; | |
| 81 for (it = packets_.begin(); it != packets_.end(); ++it) | |
| 82 frame->data.insert(frame->data.end(), it->second.begin(), it->second.end()); | |
| 83 return true; | |
| 84 } | |
| 85 | |
| 86 } // namespace cast | |
| 87 } // namespace media | |
| OLD | NEW |