OLD | NEW |
---|---|
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 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 #include "media/cast/transport/rtp_sender/rtp_packetizer/rtp_packetizer.h" | 5 #include "media/cast/transport/rtp_sender/rtp_packetizer/rtp_packetizer.h" |
6 | 6 |
7 #include "base/big_endian.h" | 7 #include "base/big_endian.h" |
8 #include "base/logging.h" | 8 #include "base/logging.h" |
9 #include "media/cast/transport/pacing/paced_sender.h" | 9 #include "media/cast/transport/pacing/paced_sender.h" |
10 | 10 |
(...skipping 27 matching lines...) Expand all Loading... | |
38 sequence_number_(config_.sequence_number), | 38 sequence_number_(config_.sequence_number), |
39 rtp_timestamp_(0), | 39 rtp_timestamp_(0), |
40 packet_id_(0), | 40 packet_id_(0), |
41 send_packets_count_(0), | 41 send_packets_count_(0), |
42 send_octet_count_(0) { | 42 send_octet_count_(0) { |
43 DCHECK(transport) << "Invalid argument"; | 43 DCHECK(transport) << "Invalid argument"; |
44 } | 44 } |
45 | 45 |
46 RtpPacketizer::~RtpPacketizer() {} | 46 RtpPacketizer::~RtpPacketizer() {} |
47 | 47 |
48 void RtpPacketizer::IncomingEncodedVideoFrame( | |
49 const EncodedVideoFrame* video_frame, | |
50 const base::TimeTicks& capture_time) { | |
51 DCHECK(!config_.audio) << "Invalid state"; | |
52 if (config_.audio) | |
53 return; | |
54 | |
55 Cast(video_frame->key_frame, | |
56 video_frame->frame_id, | |
57 video_frame->last_referenced_frame_id, | |
58 video_frame->rtp_timestamp, | |
59 video_frame->data, | |
60 capture_time); | |
61 } | |
62 | |
63 void RtpPacketizer::IncomingEncodedAudioFrame( | |
64 const EncodedAudioFrame* audio_frame, | |
65 const base::TimeTicks& recorded_time) { | |
66 DCHECK(config_.audio) << "Invalid state"; | |
67 if (!config_.audio) | |
68 return; | |
69 | |
70 Cast(true, | |
71 audio_frame->frame_id, | |
72 0, | |
73 audio_frame->rtp_timestamp, | |
74 audio_frame->data, | |
75 recorded_time); | |
76 } | |
77 | |
78 uint16 RtpPacketizer::NextSequenceNumber() { | 48 uint16 RtpPacketizer::NextSequenceNumber() { |
79 ++sequence_number_; | 49 ++sequence_number_; |
80 return sequence_number_ - 1; | 50 return sequence_number_ - 1; |
81 } | 51 } |
82 | 52 |
83 bool RtpPacketizer::LastSentTimestamp(base::TimeTicks* time_sent, | 53 bool RtpPacketizer::LastSentTimestamp(base::TimeTicks* time_sent, |
84 uint32* rtp_timestamp) const { | 54 uint32* rtp_timestamp) const { |
85 if (time_last_sent_rtp_timestamp_.is_null()) | 55 if (time_last_sent_rtp_timestamp_.is_null()) |
86 return false; | 56 return false; |
87 | 57 |
88 *time_sent = time_last_sent_rtp_timestamp_; | 58 *time_sent = time_last_sent_rtp_timestamp_; |
89 *rtp_timestamp = rtp_timestamp_; | 59 *rtp_timestamp = rtp_timestamp_; |
90 return true; | 60 return true; |
91 } | 61 } |
92 | 62 |
93 // TODO(mikhal): Switch to pass data with a const_ref. | 63 void RtpPacketizer::SendFrameAsPackets(const EncodedFrame& frame) { |
94 void RtpPacketizer::Cast(bool is_key, | 64 time_last_sent_rtp_timestamp_ = frame.reference_time; |
95 uint32 frame_id, | |
96 uint32 reference_frame_id, | |
97 uint32 timestamp, | |
98 const std::string& data, | |
99 const base::TimeTicks& capture_time) { | |
100 time_last_sent_rtp_timestamp_ = capture_time; | |
101 uint16 rtp_header_length = kCommonRtpHeaderLength + kCastRtpHeaderLength; | 65 uint16 rtp_header_length = kCommonRtpHeaderLength + kCastRtpHeaderLength; |
102 uint16 max_length = config_.max_payload_length - rtp_header_length - 1; | 66 uint16 max_length = config_.max_payload_length - rtp_header_length - 1; |
103 rtp_timestamp_ = timestamp; | 67 rtp_timestamp_ = frame.rtp_timestamp; |
104 | 68 |
105 // Split the payload evenly (round number up). | 69 // Split the payload evenly (round number up). |
106 size_t num_packets = (data.size() + max_length) / max_length; | 70 size_t num_packets = (frame.data.size() + max_length) / max_length; |
107 size_t payload_length = (data.size() + num_packets) / num_packets; | 71 size_t payload_length = (frame.data.size() + num_packets) / num_packets; |
108 DCHECK_LE(payload_length, max_length) << "Invalid argument"; | 72 DCHECK_LE(payload_length, max_length) << "Invalid argument"; |
109 | 73 |
110 SendPacketVector packets; | 74 SendPacketVector packets; |
111 | 75 |
112 size_t remaining_size = data.size(); | 76 size_t remaining_size = frame.data.size(); |
113 std::string::const_iterator data_iter = data.begin(); | 77 std::string::const_iterator data_iter = frame.data.begin(); |
114 while (remaining_size > 0) { | 78 while (remaining_size > 0) { |
115 PacketRef packet(new base::RefCountedData<Packet>); | 79 PacketRef packet(new base::RefCountedData<Packet>); |
116 | 80 |
117 if (remaining_size < payload_length) { | 81 if (remaining_size < payload_length) { |
118 payload_length = remaining_size; | 82 payload_length = remaining_size; |
119 } | 83 } |
120 remaining_size -= payload_length; | 84 remaining_size -= payload_length; |
121 BuildCommonRTPheader(&packet->data, remaining_size == 0, timestamp); | 85 BuildCommonRTPheader( |
86 &packet->data, remaining_size == 0, frame.rtp_timestamp); | |
122 | 87 |
123 // Build Cast header. | 88 // Build Cast header. |
89 const bool is_key = (frame.frame_id == frame.base_frame_id); | |
Alpha Left Google
2014/05/15 22:11:23
As described before frame_id == base_frame_id does
miu
2014/05/16 04:27:25
Done.
| |
90 // TODO(miu): Should we *really* always send the reference_frame_id? | |
124 packet->data.push_back((is_key ? kCastKeyFrameBitMask : 0) | | 91 packet->data.push_back((is_key ? kCastKeyFrameBitMask : 0) | |
125 kCastReferenceFrameIdBitMask); | 92 kCastReferenceFrameIdBitMask); |
126 packet->data.push_back(frame_id); | 93 packet->data.push_back(static_cast<uint8>(frame.frame_id)); |
127 size_t start_size = packet->data.size(); | 94 size_t start_size = packet->data.size(); |
128 packet->data.resize(start_size + 4); | 95 packet->data.resize(start_size + 4); |
129 base::BigEndianWriter big_endian_writer( | 96 base::BigEndianWriter big_endian_writer( |
130 reinterpret_cast<char*>(&(packet->data[start_size])), 4); | 97 reinterpret_cast<char*>(&(packet->data[start_size])), 4); |
131 big_endian_writer.WriteU16(packet_id_); | 98 big_endian_writer.WriteU16(packet_id_); |
132 big_endian_writer.WriteU16(static_cast<uint16>(num_packets - 1)); | 99 big_endian_writer.WriteU16(static_cast<uint16>(num_packets - 1)); |
133 packet->data.push_back(static_cast<uint8>(reference_frame_id)); | 100 packet->data.push_back(static_cast<uint8>(frame.base_frame_id)); |
134 | 101 |
135 // Copy payload data. | 102 // Copy payload data. |
136 packet->data.insert(packet->data.end(), | 103 packet->data.insert(packet->data.end(), |
137 data_iter, | 104 data_iter, |
138 data_iter + payload_length); | 105 data_iter + payload_length); |
139 | 106 |
140 PacketKey key = PacedPacketSender::MakePacketKey(capture_time, | 107 PacketKey key = PacedPacketSender::MakePacketKey(frame.reference_time, |
141 config_.ssrc, | 108 config_.ssrc, |
142 packet_id_); | 109 packet_id_); |
143 | 110 |
144 // Store packet. | 111 // Store packet. |
145 packet_storage_->StorePacket(frame_id, packet_id_, key, packet); | 112 packet_storage_->StorePacket(frame.frame_id, packet_id_, key, packet); |
146 ++packet_id_; | 113 ++packet_id_; |
147 data_iter += payload_length; | 114 data_iter += payload_length; |
148 | 115 |
149 // Update stats. | 116 // Update stats. |
150 ++send_packets_count_; | 117 ++send_packets_count_; |
151 send_octet_count_ += payload_length; | 118 send_octet_count_ += payload_length; |
152 packets.push_back(make_pair(key, packet)); | 119 packets.push_back(make_pair(key, packet)); |
153 } | 120 } |
154 DCHECK(packet_id_ == num_packets) << "Invalid state"; | 121 DCHECK(packet_id_ == num_packets) << "Invalid state"; |
155 | 122 |
(...skipping 16 matching lines...) Expand all Loading... | |
172 reinterpret_cast<char*>(&((*packet)[start_size])), 10); | 139 reinterpret_cast<char*>(&((*packet)[start_size])), 10); |
173 big_endian_writer.WriteU16(sequence_number_); | 140 big_endian_writer.WriteU16(sequence_number_); |
174 big_endian_writer.WriteU32(time_stamp); | 141 big_endian_writer.WriteU32(time_stamp); |
175 big_endian_writer.WriteU32(config_.ssrc); | 142 big_endian_writer.WriteU32(config_.ssrc); |
176 ++sequence_number_; | 143 ++sequence_number_; |
177 } | 144 } |
178 | 145 |
179 } // namespace transport | 146 } // namespace transport |
180 } // namespace cast | 147 } // namespace cast |
181 } // namespace media | 148 } // namespace media |
OLD | NEW |