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 |