| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| 11 #include "webrtc/video/payload_router.h" | 11 #include "webrtc/video/payload_router.h" |
| 12 | 12 |
| 13 #include "webrtc/base/checks.h" | 13 #include "webrtc/base/checks.h" |
| 14 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp.h" | 14 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp.h" |
| 15 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" | 15 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" |
| 16 #include "webrtc/modules/video_coding/include/video_codec_interface.h" | |
| 17 | 16 |
| 18 namespace webrtc { | 17 namespace webrtc { |
| 19 | 18 |
| 20 namespace { | 19 PayloadRouter::PayloadRouter(const std::vector<RtpRtcp*>& rtp_modules) |
| 21 // Map information from info into rtp. | 20 : active_(false), num_sending_modules_(1), rtp_modules_(rtp_modules) { |
| 22 void CopyCodecSpecific(const CodecSpecificInfo* info, RTPVideoHeader* rtp) { | |
| 23 RTC_DCHECK(info); | |
| 24 switch (info->codecType) { | |
| 25 case kVideoCodecVP8: { | |
| 26 rtp->codec = kRtpVideoVp8; | |
| 27 rtp->codecHeader.VP8.InitRTPVideoHeaderVP8(); | |
| 28 rtp->codecHeader.VP8.pictureId = info->codecSpecific.VP8.pictureId; | |
| 29 rtp->codecHeader.VP8.nonReference = info->codecSpecific.VP8.nonReference; | |
| 30 rtp->codecHeader.VP8.temporalIdx = info->codecSpecific.VP8.temporalIdx; | |
| 31 rtp->codecHeader.VP8.layerSync = info->codecSpecific.VP8.layerSync; | |
| 32 rtp->codecHeader.VP8.tl0PicIdx = info->codecSpecific.VP8.tl0PicIdx; | |
| 33 rtp->codecHeader.VP8.keyIdx = info->codecSpecific.VP8.keyIdx; | |
| 34 rtp->simulcastIdx = info->codecSpecific.VP8.simulcastIdx; | |
| 35 return; | |
| 36 } | |
| 37 case kVideoCodecVP9: { | |
| 38 rtp->codec = kRtpVideoVp9; | |
| 39 rtp->codecHeader.VP9.InitRTPVideoHeaderVP9(); | |
| 40 rtp->codecHeader.VP9.inter_pic_predicted = | |
| 41 info->codecSpecific.VP9.inter_pic_predicted; | |
| 42 rtp->codecHeader.VP9.flexible_mode = | |
| 43 info->codecSpecific.VP9.flexible_mode; | |
| 44 rtp->codecHeader.VP9.ss_data_available = | |
| 45 info->codecSpecific.VP9.ss_data_available; | |
| 46 rtp->codecHeader.VP9.picture_id = info->codecSpecific.VP9.picture_id; | |
| 47 rtp->codecHeader.VP9.tl0_pic_idx = info->codecSpecific.VP9.tl0_pic_idx; | |
| 48 rtp->codecHeader.VP9.temporal_idx = info->codecSpecific.VP9.temporal_idx; | |
| 49 rtp->codecHeader.VP9.spatial_idx = info->codecSpecific.VP9.spatial_idx; | |
| 50 rtp->codecHeader.VP9.temporal_up_switch = | |
| 51 info->codecSpecific.VP9.temporal_up_switch; | |
| 52 rtp->codecHeader.VP9.inter_layer_predicted = | |
| 53 info->codecSpecific.VP9.inter_layer_predicted; | |
| 54 rtp->codecHeader.VP9.gof_idx = info->codecSpecific.VP9.gof_idx; | |
| 55 rtp->codecHeader.VP9.num_spatial_layers = | |
| 56 info->codecSpecific.VP9.num_spatial_layers; | |
| 57 | |
| 58 if (info->codecSpecific.VP9.ss_data_available) { | |
| 59 rtp->codecHeader.VP9.spatial_layer_resolution_present = | |
| 60 info->codecSpecific.VP9.spatial_layer_resolution_present; | |
| 61 if (info->codecSpecific.VP9.spatial_layer_resolution_present) { | |
| 62 for (size_t i = 0; i < info->codecSpecific.VP9.num_spatial_layers; | |
| 63 ++i) { | |
| 64 rtp->codecHeader.VP9.width[i] = info->codecSpecific.VP9.width[i]; | |
| 65 rtp->codecHeader.VP9.height[i] = info->codecSpecific.VP9.height[i]; | |
| 66 } | |
| 67 } | |
| 68 rtp->codecHeader.VP9.gof.CopyGofInfoVP9(info->codecSpecific.VP9.gof); | |
| 69 } | |
| 70 | |
| 71 rtp->codecHeader.VP9.num_ref_pics = info->codecSpecific.VP9.num_ref_pics; | |
| 72 for (int i = 0; i < info->codecSpecific.VP9.num_ref_pics; ++i) | |
| 73 rtp->codecHeader.VP9.pid_diff[i] = info->codecSpecific.VP9.p_diff[i]; | |
| 74 return; | |
| 75 } | |
| 76 case kVideoCodecH264: | |
| 77 rtp->codec = kRtpVideoH264; | |
| 78 return; | |
| 79 case kVideoCodecGeneric: | |
| 80 rtp->codec = kRtpVideoGeneric; | |
| 81 rtp->simulcastIdx = info->codecSpecific.generic.simulcast_idx; | |
| 82 return; | |
| 83 default: | |
| 84 return; | |
| 85 } | |
| 86 } | |
| 87 } // namespace | |
| 88 | |
| 89 PayloadRouter::PayloadRouter(const std::vector<RtpRtcp*>& rtp_modules, | |
| 90 int payload_type) | |
| 91 : active_(false), | |
| 92 num_sending_modules_(1), | |
| 93 rtp_modules_(rtp_modules), | |
| 94 payload_type_(payload_type) { | |
| 95 UpdateModuleSendingState(); | 21 UpdateModuleSendingState(); |
| 96 } | 22 } |
| 97 | 23 |
| 98 PayloadRouter::~PayloadRouter() {} | 24 PayloadRouter::~PayloadRouter() {} |
| 99 | 25 |
| 100 size_t PayloadRouter::DefaultMaxPayloadLength() { | 26 size_t PayloadRouter::DefaultMaxPayloadLength() { |
| 101 const size_t kIpUdpSrtpLength = 44; | 27 const size_t kIpUdpSrtpLength = 44; |
| 102 return IP_PACKET_SIZE - kIpUdpSrtpLength; | 28 return IP_PACKET_SIZE - kIpUdpSrtpLength; |
| 103 } | 29 } |
| 104 | 30 |
| (...skipping 22 matching lines...) Expand all Loading... |
| 127 rtp_modules_[i]->SetSendingStatus(active_); | 53 rtp_modules_[i]->SetSendingStatus(active_); |
| 128 rtp_modules_[i]->SetSendingMediaStatus(active_); | 54 rtp_modules_[i]->SetSendingMediaStatus(active_); |
| 129 } | 55 } |
| 130 // Disable inactive modules. | 56 // Disable inactive modules. |
| 131 for (size_t i = num_sending_modules_; i < rtp_modules_.size(); ++i) { | 57 for (size_t i = num_sending_modules_; i < rtp_modules_.size(); ++i) { |
| 132 rtp_modules_[i]->SetSendingStatus(false); | 58 rtp_modules_[i]->SetSendingStatus(false); |
| 133 rtp_modules_[i]->SetSendingMediaStatus(false); | 59 rtp_modules_[i]->SetSendingMediaStatus(false); |
| 134 } | 60 } |
| 135 } | 61 } |
| 136 | 62 |
| 137 int32_t PayloadRouter::Encoded(const EncodedImage& encoded_image, | 63 bool PayloadRouter::RoutePayload(FrameType frame_type, |
| 138 const CodecSpecificInfo* codec_specific_info, | 64 int8_t payload_type, |
| 139 const RTPFragmentationHeader* fragmentation) { | 65 uint32_t time_stamp, |
| 66 int64_t capture_time_ms, |
| 67 const uint8_t* payload_data, |
| 68 size_t payload_length, |
| 69 const RTPFragmentationHeader* fragmentation, |
| 70 const RTPVideoHeader* rtp_video_hdr) { |
| 140 rtc::CritScope lock(&crit_); | 71 rtc::CritScope lock(&crit_); |
| 141 RTC_DCHECK(!rtp_modules_.empty()); | 72 RTC_DCHECK(!rtp_modules_.empty()); |
| 142 if (!active_ || num_sending_modules_ == 0) | 73 if (!active_ || num_sending_modules_ == 0) |
| 143 return -1; | 74 return false; |
| 144 | 75 |
| 145 int stream_idx = 0; | 76 int stream_idx = 0; |
| 146 | 77 if (rtp_video_hdr) { |
| 147 RTPVideoHeader rtp_video_header; | 78 RTC_DCHECK_LT(rtp_video_hdr->simulcastIdx, rtp_modules_.size()); |
| 148 memset(&rtp_video_header, 0, sizeof(RTPVideoHeader)); | 79 // The simulcast index might actually be larger than the number of modules |
| 149 if (codec_specific_info) | 80 // in case the encoder was processing a frame during a codec reconfig. |
| 150 CopyCodecSpecific(codec_specific_info, &rtp_video_header); | 81 if (rtp_video_hdr->simulcastIdx >= num_sending_modules_) |
| 151 rtp_video_header.rotation = encoded_image.rotation_; | 82 return false; |
| 152 | 83 stream_idx = rtp_video_hdr->simulcastIdx; |
| 153 RTC_DCHECK_LT(rtp_video_header.simulcastIdx, rtp_modules_.size()); | 84 } |
| 154 // The simulcast index might actually be larger than the number of modules | |
| 155 // in case the encoder was processing a frame during a codec reconfig. | |
| 156 if (rtp_video_header.simulcastIdx >= num_sending_modules_) | |
| 157 return -1; | |
| 158 stream_idx = rtp_video_header.simulcastIdx; | |
| 159 | |
| 160 return rtp_modules_[stream_idx]->SendOutgoingData( | 85 return rtp_modules_[stream_idx]->SendOutgoingData( |
| 161 encoded_image._frameType, payload_type_, encoded_image._timeStamp, | 86 frame_type, payload_type, time_stamp, capture_time_ms, payload_data, |
| 162 encoded_image.capture_time_ms_, encoded_image._buffer, | 87 payload_length, fragmentation, rtp_video_hdr) == 0 ? true : false; |
| 163 encoded_image._length, fragmentation, &rtp_video_header); | |
| 164 } | 88 } |
| 165 | 89 |
| 166 void PayloadRouter::SetTargetSendBitrates( | 90 void PayloadRouter::SetTargetSendBitrates( |
| 167 const std::vector<uint32_t>& stream_bitrates) { | 91 const std::vector<uint32_t>& stream_bitrates) { |
| 168 rtc::CritScope lock(&crit_); | 92 rtc::CritScope lock(&crit_); |
| 169 RTC_DCHECK_LE(stream_bitrates.size(), rtp_modules_.size()); | 93 RTC_DCHECK_LE(stream_bitrates.size(), rtp_modules_.size()); |
| 170 for (size_t i = 0; i < stream_bitrates.size(); ++i) { | 94 for (size_t i = 0; i < stream_bitrates.size(); ++i) { |
| 171 rtp_modules_[i]->SetTargetSendBitrate(stream_bitrates[i]); | 95 rtp_modules_[i]->SetTargetSendBitrate(stream_bitrates[i]); |
| 172 } | 96 } |
| 173 } | 97 } |
| 174 | 98 |
| 175 size_t PayloadRouter::MaxPayloadLength() const { | 99 size_t PayloadRouter::MaxPayloadLength() const { |
| 176 size_t min_payload_length = DefaultMaxPayloadLength(); | 100 size_t min_payload_length = DefaultMaxPayloadLength(); |
| 177 rtc::CritScope lock(&crit_); | 101 rtc::CritScope lock(&crit_); |
| 178 for (size_t i = 0; i < num_sending_modules_; ++i) { | 102 for (size_t i = 0; i < num_sending_modules_; ++i) { |
| 179 size_t module_payload_length = rtp_modules_[i]->MaxDataPayloadLength(); | 103 size_t module_payload_length = rtp_modules_[i]->MaxDataPayloadLength(); |
| 180 if (module_payload_length < min_payload_length) | 104 if (module_payload_length < min_payload_length) |
| 181 min_payload_length = module_payload_length; | 105 min_payload_length = module_payload_length; |
| 182 } | 106 } |
| 183 return min_payload_length; | 107 return min_payload_length; |
| 184 } | 108 } |
| 185 | 109 |
| 186 } // namespace webrtc | 110 } // namespace webrtc |
| OLD | NEW |