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" |
16 | 17 |
17 namespace webrtc { | 18 namespace webrtc { |
18 | 19 |
19 PayloadRouter::PayloadRouter(const std::vector<RtpRtcp*>& rtp_modules) | 20 namespace { |
20 : active_(false), num_sending_modules_(1), rtp_modules_(rtp_modules) { | 21 // Map information from info into rtp. |
| 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) { |
21 UpdateModuleSendingState(); | 95 UpdateModuleSendingState(); |
22 } | 96 } |
23 | 97 |
24 PayloadRouter::~PayloadRouter() {} | 98 PayloadRouter::~PayloadRouter() {} |
25 | 99 |
26 size_t PayloadRouter::DefaultMaxPayloadLength() { | 100 size_t PayloadRouter::DefaultMaxPayloadLength() { |
27 const size_t kIpUdpSrtpLength = 44; | 101 const size_t kIpUdpSrtpLength = 44; |
28 return IP_PACKET_SIZE - kIpUdpSrtpLength; | 102 return IP_PACKET_SIZE - kIpUdpSrtpLength; |
29 } | 103 } |
30 | 104 |
(...skipping 22 matching lines...) Expand all Loading... |
53 rtp_modules_[i]->SetSendingStatus(active_); | 127 rtp_modules_[i]->SetSendingStatus(active_); |
54 rtp_modules_[i]->SetSendingMediaStatus(active_); | 128 rtp_modules_[i]->SetSendingMediaStatus(active_); |
55 } | 129 } |
56 // Disable inactive modules. | 130 // Disable inactive modules. |
57 for (size_t i = num_sending_modules_; i < rtp_modules_.size(); ++i) { | 131 for (size_t i = num_sending_modules_; i < rtp_modules_.size(); ++i) { |
58 rtp_modules_[i]->SetSendingStatus(false); | 132 rtp_modules_[i]->SetSendingStatus(false); |
59 rtp_modules_[i]->SetSendingMediaStatus(false); | 133 rtp_modules_[i]->SetSendingMediaStatus(false); |
60 } | 134 } |
61 } | 135 } |
62 | 136 |
63 bool PayloadRouter::RoutePayload(FrameType frame_type, | 137 int32_t PayloadRouter::Encoded(const EncodedImage& encoded_image, |
64 int8_t payload_type, | 138 const CodecSpecificInfo* codec_specific_info, |
65 uint32_t time_stamp, | 139 const RTPFragmentationHeader* fragmentation) { |
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) { | |
71 rtc::CritScope lock(&crit_); | 140 rtc::CritScope lock(&crit_); |
72 RTC_DCHECK(!rtp_modules_.empty()); | 141 RTC_DCHECK(!rtp_modules_.empty()); |
73 if (!active_ || num_sending_modules_ == 0) | 142 if (!active_ || num_sending_modules_ == 0) |
74 return false; | 143 return -1; |
75 | 144 |
76 int stream_idx = 0; | 145 int stream_idx = 0; |
77 if (rtp_video_hdr) { | 146 |
78 RTC_DCHECK_LT(rtp_video_hdr->simulcastIdx, rtp_modules_.size()); | 147 RTPVideoHeader rtp_video_header; |
79 // The simulcast index might actually be larger than the number of modules | 148 memset(&rtp_video_header, 0, sizeof(RTPVideoHeader)); |
80 // in case the encoder was processing a frame during a codec reconfig. | 149 if (codec_specific_info) |
81 if (rtp_video_hdr->simulcastIdx >= num_sending_modules_) | 150 CopyCodecSpecific(codec_specific_info, &rtp_video_header); |
82 return false; | 151 rtp_video_header.rotation = encoded_image.rotation_; |
83 stream_idx = rtp_video_hdr->simulcastIdx; | 152 |
84 } | 153 RTC_DCHECK_LT(rtp_video_header.simulcastIdx, rtp_modules_.size()); |
| 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 |
85 return rtp_modules_[stream_idx]->SendOutgoingData( | 160 return rtp_modules_[stream_idx]->SendOutgoingData( |
86 frame_type, payload_type, time_stamp, capture_time_ms, payload_data, | 161 encoded_image._frameType, payload_type_, encoded_image._timeStamp, |
87 payload_length, fragmentation, rtp_video_hdr) == 0 ? true : false; | 162 encoded_image.capture_time_ms_, encoded_image._buffer, |
| 163 encoded_image._length, fragmentation, &rtp_video_header); |
88 } | 164 } |
89 | 165 |
90 void PayloadRouter::SetTargetSendBitrates( | 166 void PayloadRouter::SetTargetSendBitrates( |
91 const std::vector<uint32_t>& stream_bitrates) { | 167 const std::vector<uint32_t>& stream_bitrates) { |
92 rtc::CritScope lock(&crit_); | 168 rtc::CritScope lock(&crit_); |
93 RTC_DCHECK_LE(stream_bitrates.size(), rtp_modules_.size()); | 169 RTC_DCHECK_LE(stream_bitrates.size(), rtp_modules_.size()); |
94 for (size_t i = 0; i < stream_bitrates.size(); ++i) { | 170 for (size_t i = 0; i < stream_bitrates.size(); ++i) { |
95 rtp_modules_[i]->SetTargetSendBitrate(stream_bitrates[i]); | 171 rtp_modules_[i]->SetTargetSendBitrate(stream_bitrates[i]); |
96 } | 172 } |
97 } | 173 } |
98 | 174 |
99 size_t PayloadRouter::MaxPayloadLength() const { | 175 size_t PayloadRouter::MaxPayloadLength() const { |
100 size_t min_payload_length = DefaultMaxPayloadLength(); | 176 size_t min_payload_length = DefaultMaxPayloadLength(); |
101 rtc::CritScope lock(&crit_); | 177 rtc::CritScope lock(&crit_); |
102 for (size_t i = 0; i < num_sending_modules_; ++i) { | 178 for (size_t i = 0; i < num_sending_modules_; ++i) { |
103 size_t module_payload_length = rtp_modules_[i]->MaxDataPayloadLength(); | 179 size_t module_payload_length = rtp_modules_[i]->MaxDataPayloadLength(); |
104 if (module_payload_length < min_payload_length) | 180 if (module_payload_length < min_payload_length) |
105 min_payload_length = module_payload_length; | 181 min_payload_length = module_payload_length; |
106 } | 182 } |
107 return min_payload_length; | 183 return min_payload_length; |
108 } | 184 } |
109 | 185 |
110 } // namespace webrtc | 186 } // namespace webrtc |
OLD | NEW |