| 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_packet_count_(0), | 41 send_packet_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 // TODO(mikhal): Switch to pass data with a const_ref. | 53 void RtpPacketizer::SendFrameAsPackets(const EncodedFrame& frame) { |
| 84 void RtpPacketizer::Cast(bool is_key, | |
| 85 uint32 frame_id, | |
| 86 uint32 reference_frame_id, | |
| 87 uint32 timestamp, | |
| 88 const std::string& data, | |
| 89 const base::TimeTicks& capture_time) { | |
| 90 uint16 rtp_header_length = kCommonRtpHeaderLength + kCastRtpHeaderLength; | 54 uint16 rtp_header_length = kCommonRtpHeaderLength + kCastRtpHeaderLength; |
| 91 uint16 max_length = config_.max_payload_length - rtp_header_length - 1; | 55 uint16 max_length = config_.max_payload_length - rtp_header_length - 1; |
| 92 rtp_timestamp_ = timestamp; | 56 rtp_timestamp_ = frame.rtp_timestamp; |
| 93 | 57 |
| 94 // Split the payload evenly (round number up). | 58 // Split the payload evenly (round number up). |
| 95 size_t num_packets = (data.size() + max_length) / max_length; | 59 size_t num_packets = (frame.data.size() + max_length) / max_length; |
| 96 size_t payload_length = (data.size() + num_packets) / num_packets; | 60 size_t payload_length = (frame.data.size() + num_packets) / num_packets; |
| 97 DCHECK_LE(payload_length, max_length) << "Invalid argument"; | 61 DCHECK_LE(payload_length, max_length) << "Invalid argument"; |
| 98 | 62 |
| 99 SendPacketVector packets; | 63 SendPacketVector packets; |
| 100 | 64 |
| 101 size_t remaining_size = data.size(); | 65 size_t remaining_size = frame.data.size(); |
| 102 std::string::const_iterator data_iter = data.begin(); | 66 std::string::const_iterator data_iter = frame.data.begin(); |
| 103 while (remaining_size > 0) { | 67 while (remaining_size > 0) { |
| 104 PacketRef packet(new base::RefCountedData<Packet>); | 68 PacketRef packet(new base::RefCountedData<Packet>); |
| 105 | 69 |
| 106 if (remaining_size < payload_length) { | 70 if (remaining_size < payload_length) { |
| 107 payload_length = remaining_size; | 71 payload_length = remaining_size; |
| 108 } | 72 } |
| 109 remaining_size -= payload_length; | 73 remaining_size -= payload_length; |
| 110 BuildCommonRTPheader(&packet->data, remaining_size == 0, timestamp); | 74 BuildCommonRTPheader( |
| 75 &packet->data, remaining_size == 0, frame.rtp_timestamp); |
| 111 | 76 |
| 112 // Build Cast header. | 77 // Build Cast header. |
| 113 packet->data.push_back((is_key ? kCastKeyFrameBitMask : 0) | | 78 // TODO(miu): Should we always set the ref frame bit and the ref_frame_id? |
| 114 kCastReferenceFrameIdBitMask); | 79 DCHECK_NE(frame.dependency, EncodedFrame::UNKNOWN_DEPENDENCY); |
| 115 packet->data.push_back(frame_id); | 80 packet->data.push_back( |
| 81 ((frame.dependency == EncodedFrame::KEY) ? kCastKeyFrameBitMask : 0) | |
| 82 kCastReferenceFrameIdBitMask); |
| 83 packet->data.push_back(static_cast<uint8>(frame.frame_id)); |
| 116 size_t start_size = packet->data.size(); | 84 size_t start_size = packet->data.size(); |
| 117 packet->data.resize(start_size + 4); | 85 packet->data.resize(start_size + 4); |
| 118 base::BigEndianWriter big_endian_writer( | 86 base::BigEndianWriter big_endian_writer( |
| 119 reinterpret_cast<char*>(&(packet->data[start_size])), 4); | 87 reinterpret_cast<char*>(&(packet->data[start_size])), 4); |
| 120 big_endian_writer.WriteU16(packet_id_); | 88 big_endian_writer.WriteU16(packet_id_); |
| 121 big_endian_writer.WriteU16(static_cast<uint16>(num_packets - 1)); | 89 big_endian_writer.WriteU16(static_cast<uint16>(num_packets - 1)); |
| 122 packet->data.push_back(static_cast<uint8>(reference_frame_id)); | 90 packet->data.push_back(static_cast<uint8>(frame.referenced_frame_id)); |
| 123 | 91 |
| 124 // Copy payload data. | 92 // Copy payload data. |
| 125 packet->data.insert(packet->data.end(), | 93 packet->data.insert(packet->data.end(), |
| 126 data_iter, | 94 data_iter, |
| 127 data_iter + payload_length); | 95 data_iter + payload_length); |
| 128 | 96 |
| 129 PacketKey key = PacedPacketSender::MakePacketKey(capture_time, | 97 PacketKey key = PacedPacketSender::MakePacketKey(frame.reference_time, |
| 130 config_.ssrc, | 98 config_.ssrc, |
| 131 packet_id_); | 99 packet_id_); |
| 132 | 100 |
| 133 // Store packet. | 101 // Store packet. |
| 134 packet_storage_->StorePacket(frame_id, packet_id_, key, packet); | 102 packet_storage_->StorePacket(frame.frame_id, packet_id_, key, packet); |
| 135 ++packet_id_; | 103 ++packet_id_; |
| 136 data_iter += payload_length; | 104 data_iter += payload_length; |
| 137 | 105 |
| 138 // Update stats. | 106 // Update stats. |
| 139 ++send_packet_count_; | 107 ++send_packet_count_; |
| 140 send_octet_count_ += payload_length; | 108 send_octet_count_ += payload_length; |
| 141 packets.push_back(make_pair(key, packet)); | 109 packets.push_back(make_pair(key, packet)); |
| 142 } | 110 } |
| 143 DCHECK(packet_id_ == num_packets) << "Invalid state"; | 111 DCHECK(packet_id_ == num_packets) << "Invalid state"; |
| 144 | 112 |
| (...skipping 16 matching lines...) Expand all Loading... |
| 161 reinterpret_cast<char*>(&((*packet)[start_size])), 10); | 129 reinterpret_cast<char*>(&((*packet)[start_size])), 10); |
| 162 big_endian_writer.WriteU16(sequence_number_); | 130 big_endian_writer.WriteU16(sequence_number_); |
| 163 big_endian_writer.WriteU32(time_stamp); | 131 big_endian_writer.WriteU32(time_stamp); |
| 164 big_endian_writer.WriteU32(config_.ssrc); | 132 big_endian_writer.WriteU32(config_.ssrc); |
| 165 ++sequence_number_; | 133 ++sequence_number_; |
| 166 } | 134 } |
| 167 | 135 |
| 168 } // namespace transport | 136 } // namespace transport |
| 169 } // namespace cast | 137 } // namespace cast |
| 170 } // namespace media | 138 } // namespace media |
| OLD | NEW |