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 |