blob: 6c93c84018c17c73f1d47c95689d43dbc69774f1 [file] [log] [blame]
philipel02447bc2016-05-13 06:01:03 -07001/*
2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved.
3 *
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
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020011#include "modules/video_coding/rtp_frame_reference_finder.h"
philipel02447bc2016-05-13 06:01:03 -070012
13#include <algorithm>
14#include <limits>
15
philipel1a4746a2018-07-09 15:52:29 +020016#include "absl/types/variant.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020017#include "modules/video_coding/frame_object.h"
18#include "modules/video_coding/packet_buffer.h"
19#include "rtc_base/checks.h"
20#include "rtc_base/logging.h"
Karl Wiberg80ba3332018-02-05 10:33:35 +010021#include "rtc_base/system/fallthrough.h"
philipel02447bc2016-05-13 06:01:03 -070022
23namespace webrtc {
24namespace video_coding {
25
26RtpFrameReferenceFinder::RtpFrameReferenceFinder(
27 OnCompleteFrameCallback* frame_callback)
28 : last_picture_id_(-1),
philipel02447bc2016-05-13 06:01:03 -070029 current_ss_idx_(0),
philipel463d3012016-09-09 03:32:44 -070030 cleared_to_seq_num_(-1),
philipel02447bc2016-05-13 06:01:03 -070031 frame_callback_(frame_callback) {}
32
Mirko Bonadei8fdcac32018-08-28 16:30:18 +020033RtpFrameReferenceFinder::~RtpFrameReferenceFinder() = default;
34
philipel02447bc2016-05-13 06:01:03 -070035void RtpFrameReferenceFinder::ManageFrame(
36 std::unique_ptr<RtpFrameObject> frame) {
37 rtc::CritScope lock(&crit_);
philipel463d3012016-09-09 03:32:44 -070038
39 // If we have cleared past this frame, drop it.
40 if (cleared_to_seq_num_ != -1 &&
41 AheadOf<uint16_t>(cleared_to_seq_num_, frame->first_seq_num())) {
42 return;
43 }
44
philipelafcf7f52017-04-26 08:17:35 -070045 FrameDecision decision = ManageFrameInternal(frame.get());
46
47 switch (decision) {
48 case kStash:
49 if (stashed_frames_.size() > kMaxStashedFrames)
50 stashed_frames_.pop_back();
51 stashed_frames_.push_front(std::move(frame));
52 break;
53 case kHandOff:
54 frame_callback_->OnCompleteFrame(std::move(frame));
55 RetryStashedFrames();
56 break;
57 case kDrop:
58 break;
59 }
60}
61
62void RtpFrameReferenceFinder::RetryStashedFrames() {
63 bool complete_frame = false;
64 do {
65 complete_frame = false;
66 for (auto frame_it = stashed_frames_.begin();
67 frame_it != stashed_frames_.end();) {
68 FrameDecision decision = ManageFrameInternal(frame_it->get());
69
70 switch (decision) {
71 case kStash:
72 ++frame_it;
73 break;
74 case kHandOff:
75 complete_frame = true;
76 frame_callback_->OnCompleteFrame(std::move(*frame_it));
Karl Wiberg80ba3332018-02-05 10:33:35 +010077 RTC_FALLTHROUGH();
philipelafcf7f52017-04-26 08:17:35 -070078 case kDrop:
79 frame_it = stashed_frames_.erase(frame_it);
80 }
81 }
82 } while (complete_frame);
83}
84
85RtpFrameReferenceFinder::FrameDecision
86RtpFrameReferenceFinder::ManageFrameInternal(RtpFrameObject* frame) {
philipel02447bc2016-05-13 06:01:03 -070087 switch (frame->codec_type()) {
philipel02447bc2016-05-13 06:01:03 -070088 case kVideoCodecVP8:
philipelafcf7f52017-04-26 08:17:35 -070089 return ManageFrameVp8(frame);
philipel02447bc2016-05-13 06:01:03 -070090 case kVideoCodecVP9:
philipelafcf7f52017-04-26 08:17:35 -070091 return ManageFrameVp9(frame);
Sami Kalliomäki98824952018-08-28 14:39:21 +020092 default: {
93 // Use 15 first bits of frame ID as picture ID if available.
philipel5470f402018-09-07 13:38:53 +020094 absl::optional<RTPVideoHeader> video_header = frame->GetRtpVideoHeader();
Sami Kalliomäki98824952018-08-28 14:39:21 +020095 absl::optional<RTPVideoHeader::GenericDescriptorInfo> generic_info =
96 video_header ? video_header->generic : absl::nullopt;
97 return ManageFrameGeneric(
98 frame, generic_info ? generic_info->frame_id & 0x7fff : kNoPictureId);
99 }
philipel02447bc2016-05-13 06:01:03 -0700100 }
101}
102
philipel9b2ce6b2016-07-05 05:04:46 -0700103void RtpFrameReferenceFinder::PaddingReceived(uint16_t seq_num) {
104 rtc::CritScope lock(&crit_);
105 auto clean_padding_to =
106 stashed_padding_.lower_bound(seq_num - kMaxPaddingAge);
107 stashed_padding_.erase(stashed_padding_.begin(), clean_padding_to);
108 stashed_padding_.insert(seq_num);
109 UpdateLastPictureIdWithPadding(seq_num);
110 RetryStashedFrames();
111}
112
philipel463d3012016-09-09 03:32:44 -0700113void RtpFrameReferenceFinder::ClearTo(uint16_t seq_num) {
114 rtc::CritScope lock(&crit_);
115 cleared_to_seq_num_ = seq_num;
116
117 auto it = stashed_frames_.begin();
118 while (it != stashed_frames_.end()) {
119 if (AheadOf<uint16_t>(cleared_to_seq_num_, (*it)->first_seq_num())) {
120 it = stashed_frames_.erase(it);
121 } else {
122 ++it;
123 }
124 }
125}
126
philipel9b2ce6b2016-07-05 05:04:46 -0700127void RtpFrameReferenceFinder::UpdateLastPictureIdWithPadding(uint16_t seq_num) {
128 auto gop_seq_num_it = last_seq_num_gop_.upper_bound(seq_num);
129
130 // If this padding packet "belongs" to a group of pictures that we don't track
131 // anymore, do nothing.
132 if (gop_seq_num_it == last_seq_num_gop_.begin())
133 return;
134 --gop_seq_num_it;
135
136 // Calculate the next contiuous sequence number and search for it in
137 // the padding packets we have stashed.
138 uint16_t next_seq_num_with_padding = gop_seq_num_it->second.second + 1;
139 auto padding_seq_num_it =
140 stashed_padding_.lower_bound(next_seq_num_with_padding);
141
142 // While there still are padding packets and those padding packets are
143 // continuous, then advance the "last-picture-id-with-padding" and remove
144 // the stashed padding packet.
145 while (padding_seq_num_it != stashed_padding_.end() &&
146 *padding_seq_num_it == next_seq_num_with_padding) {
147 gop_seq_num_it->second.second = next_seq_num_with_padding;
148 ++next_seq_num_with_padding;
149 padding_seq_num_it = stashed_padding_.erase(padding_seq_num_it);
150 }
philipel41bb7922017-02-20 07:53:23 -0800151
152 // In the case where the stream has been continuous without any new keyframes
153 // for a while there is a risk that new frames will appear to be older than
154 // the keyframe they belong to due to wrapping sequence number. In order
155 // to prevent this we advance the picture id of the keyframe every so often.
156 if (ForwardDiff(gop_seq_num_it->first, seq_num) > 10000) {
157 RTC_DCHECK_EQ(1ul, last_seq_num_gop_.size());
158 last_seq_num_gop_[seq_num] = gop_seq_num_it->second;
159 last_seq_num_gop_.erase(gop_seq_num_it);
160 }
philipel9b2ce6b2016-07-05 05:04:46 -0700161}
162
philipelafcf7f52017-04-26 08:17:35 -0700163RtpFrameReferenceFinder::FrameDecision
164RtpFrameReferenceFinder::ManageFrameGeneric(RtpFrameObject* frame,
165 int picture_id) {
philipel647998c2016-06-03 09:40:16 -0700166 // If |picture_id| is specified then we use that to set the frame references,
167 // otherwise we use sequence number.
168 if (picture_id != kNoPictureId) {
philipel0fa82a62018-03-19 15:34:53 +0100169 frame->id.picture_id = unwrapper_.Unwrap(picture_id);
philipel647998c2016-06-03 09:40:16 -0700170 frame->num_references = frame->frame_type() == kVideoFrameKey ? 0 : 1;
philipel0fa82a62018-03-19 15:34:53 +0100171 frame->references[0] = frame->id.picture_id - 1;
philipelafcf7f52017-04-26 08:17:35 -0700172 return kHandOff;
philipel647998c2016-06-03 09:40:16 -0700173 }
174
philipel9b2ce6b2016-07-05 05:04:46 -0700175 if (frame->frame_type() == kVideoFrameKey) {
176 last_seq_num_gop_.insert(std::make_pair(
177 frame->last_seq_num(),
178 std::make_pair(frame->last_seq_num(), frame->last_seq_num())));
179 }
philipel02447bc2016-05-13 06:01:03 -0700180
181 // We have received a frame but not yet a keyframe, stash this frame.
philipelafcf7f52017-04-26 08:17:35 -0700182 if (last_seq_num_gop_.empty())
183 return kStash;
philipel02447bc2016-05-13 06:01:03 -0700184
185 // Clean up info for old keyframes but make sure to keep info
186 // for the last keyframe.
187 auto clean_to = last_seq_num_gop_.lower_bound(frame->last_seq_num() - 100);
philipel41bb7922017-02-20 07:53:23 -0800188 for (auto it = last_seq_num_gop_.begin();
189 it != clean_to && last_seq_num_gop_.size() > 1;) {
190 it = last_seq_num_gop_.erase(it);
191 }
philipel02447bc2016-05-13 06:01:03 -0700192
193 // Find the last sequence number of the last frame for the keyframe
194 // that this frame indirectly references.
195 auto seq_num_it = last_seq_num_gop_.upper_bound(frame->last_seq_num());
philipel9b2ce6b2016-07-05 05:04:46 -0700196 if (seq_num_it == last_seq_num_gop_.begin()) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100197 RTC_LOG(LS_WARNING) << "Generic frame with packet range ["
198 << frame->first_seq_num() << ", "
199 << frame->last_seq_num()
200 << "] has no GoP, dropping frame.";
philipelafcf7f52017-04-26 08:17:35 -0700201 return kDrop;
philipel9b2ce6b2016-07-05 05:04:46 -0700202 }
philipel02447bc2016-05-13 06:01:03 -0700203 seq_num_it--;
204
205 // Make sure the packet sequence numbers are continuous, otherwise stash
206 // this frame.
philipel9b2ce6b2016-07-05 05:04:46 -0700207 uint16_t last_picture_id_gop = seq_num_it->second.first;
208 uint16_t last_picture_id_with_padding_gop = seq_num_it->second.second;
philipel02447bc2016-05-13 06:01:03 -0700209 if (frame->frame_type() == kVideoFrameDelta) {
philipel9b2ce6b2016-07-05 05:04:46 -0700210 uint16_t prev_seq_num = frame->first_seq_num() - 1;
philipelafcf7f52017-04-26 08:17:35 -0700211
212 if (prev_seq_num != last_picture_id_with_padding_gop)
213 return kStash;
philipel02447bc2016-05-13 06:01:03 -0700214 }
215
216 RTC_DCHECK(AheadOrAt(frame->last_seq_num(), seq_num_it->first));
217
218 // Since keyframes can cause reordering we can't simply assign the
219 // picture id according to some incrementing counter.
philipel0fa82a62018-03-19 15:34:53 +0100220 frame->id.picture_id = frame->last_seq_num();
philipel02447bc2016-05-13 06:01:03 -0700221 frame->num_references = frame->frame_type() == kVideoFrameDelta;
philipeld4fac692017-09-04 07:03:46 -0700222 frame->references[0] = generic_unwrapper_.Unwrap(last_picture_id_gop);
philipel0fa82a62018-03-19 15:34:53 +0100223 if (AheadOf<uint16_t>(frame->id.picture_id, last_picture_id_gop)) {
224 seq_num_it->second.first = frame->id.picture_id;
225 seq_num_it->second.second = frame->id.picture_id;
philipel9b2ce6b2016-07-05 05:04:46 -0700226 }
philipel02447bc2016-05-13 06:01:03 -0700227
philipel0fa82a62018-03-19 15:34:53 +0100228 last_picture_id_ = frame->id.picture_id;
229 UpdateLastPictureIdWithPadding(frame->id.picture_id);
230 frame->id.picture_id = generic_unwrapper_.Unwrap(frame->id.picture_id);
philipelafcf7f52017-04-26 08:17:35 -0700231 return kHandOff;
philipel02447bc2016-05-13 06:01:03 -0700232}
233
philipelafcf7f52017-04-26 08:17:35 -0700234RtpFrameReferenceFinder::FrameDecision RtpFrameReferenceFinder::ManageFrameVp8(
235 RtpFrameObject* frame) {
philipel5470f402018-09-07 13:38:53 +0200236 absl::optional<RTPVideoHeader> video_header = frame->GetRtpVideoHeader();
Sami Kalliomäki98824952018-08-28 14:39:21 +0200237 if (!video_header) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100238 RTC_LOG(LS_WARNING)
239 << "Failed to get codec header from frame, dropping frame.";
philipelafcf7f52017-04-26 08:17:35 -0700240 return kDrop;
philipeld4fac692017-09-04 07:03:46 -0700241 }
Sami Kalliomäki98824952018-08-28 14:39:21 +0200242 RTPVideoTypeHeader rtp_codec_header = video_header->video_type_header;
philipel02447bc2016-05-13 06:01:03 -0700243
philipel1a4746a2018-07-09 15:52:29 +0200244 const RTPVideoHeaderVP8& codec_header =
Sami Kalliomäki98824952018-08-28 14:39:21 +0200245 absl::get<RTPVideoHeaderVP8>(rtp_codec_header);
philipel02447bc2016-05-13 06:01:03 -0700246
247 if (codec_header.pictureId == kNoPictureId ||
248 codec_header.temporalIdx == kNoTemporalIdx ||
249 codec_header.tl0PicIdx == kNoTl0PicIdx) {
philipelafcf7f52017-04-26 08:17:35 -0700250 return ManageFrameGeneric(std::move(frame), codec_header.pictureId);
philipel02447bc2016-05-13 06:01:03 -0700251 }
252
philipel0fa82a62018-03-19 15:34:53 +0100253 frame->id.picture_id = codec_header.pictureId % kPicIdLength;
philipel02447bc2016-05-13 06:01:03 -0700254
philipel02447bc2016-05-13 06:01:03 -0700255 if (last_picture_id_ == -1)
philipel0fa82a62018-03-19 15:34:53 +0100256 last_picture_id_ = frame->id.picture_id;
philipel02447bc2016-05-13 06:01:03 -0700257
258 // Find if there has been a gap in fully received frames and save the picture
259 // id of those frames in |not_yet_received_frames_|.
philipel0fa82a62018-03-19 15:34:53 +0100260 if (AheadOf<uint16_t, kPicIdLength>(frame->id.picture_id, last_picture_id_)) {
philipel9bd1d662017-07-14 04:52:01 -0700261 do {
philipel02447bc2016-05-13 06:01:03 -0700262 last_picture_id_ = Add<kPicIdLength>(last_picture_id_, 1);
philipel9bd1d662017-07-14 04:52:01 -0700263 not_yet_received_frames_.insert(last_picture_id_);
philipel0fa82a62018-03-19 15:34:53 +0100264 } while (last_picture_id_ != frame->id.picture_id);
philipel02447bc2016-05-13 06:01:03 -0700265 }
266
philipel57ec6852018-07-03 18:09:32 +0200267 int64_t unwrapped_tl0 = tl0_unwrapper_.Unwrap(codec_header.tl0PicIdx);
268
philipel02447bc2016-05-13 06:01:03 -0700269 // Clean up info for base layers that are too old.
philipel57ec6852018-07-03 18:09:32 +0200270 int64_t old_tl0_pic_idx = unwrapped_tl0 - kMaxLayerInfo;
philipel02447bc2016-05-13 06:01:03 -0700271 auto clean_layer_info_to = layer_info_.lower_bound(old_tl0_pic_idx);
272 layer_info_.erase(layer_info_.begin(), clean_layer_info_to);
273
274 // Clean up info about not yet received frames that are too old.
275 uint16_t old_picture_id =
philipel0fa82a62018-03-19 15:34:53 +0100276 Subtract<kPicIdLength>(frame->id.picture_id, kMaxNotYetReceivedFrames);
philipel02447bc2016-05-13 06:01:03 -0700277 auto clean_frames_to = not_yet_received_frames_.lower_bound(old_picture_id);
278 not_yet_received_frames_.erase(not_yet_received_frames_.begin(),
279 clean_frames_to);
280
281 if (frame->frame_type() == kVideoFrameKey) {
282 frame->num_references = 0;
philipel57ec6852018-07-03 18:09:32 +0200283 layer_info_[unwrapped_tl0].fill(-1);
284 UpdateLayerInfoVp8(frame, unwrapped_tl0, codec_header.temporalIdx);
philipelafcf7f52017-04-26 08:17:35 -0700285 return kHandOff;
philipel02447bc2016-05-13 06:01:03 -0700286 }
287
philipel57ec6852018-07-03 18:09:32 +0200288 auto layer_info_it = layer_info_.find(
289 codec_header.temporalIdx == 0 ? unwrapped_tl0 - 1 : unwrapped_tl0);
philipel02447bc2016-05-13 06:01:03 -0700290
291 // If we don't have the base layer frame yet, stash this frame.
philipelafcf7f52017-04-26 08:17:35 -0700292 if (layer_info_it == layer_info_.end())
293 return kStash;
philipel02447bc2016-05-13 06:01:03 -0700294
295 // A non keyframe base layer frame has been received, copy the layer info
296 // from the previous base layer frame and set a reference to the previous
297 // base layer frame.
298 if (codec_header.temporalIdx == 0) {
299 layer_info_it =
philipel57ec6852018-07-03 18:09:32 +0200300 layer_info_.emplace(unwrapped_tl0, layer_info_it->second).first;
philipel02447bc2016-05-13 06:01:03 -0700301 frame->num_references = 1;
302 frame->references[0] = layer_info_it->second[0];
philipel57ec6852018-07-03 18:09:32 +0200303 UpdateLayerInfoVp8(frame, unwrapped_tl0, codec_header.temporalIdx);
philipelafcf7f52017-04-26 08:17:35 -0700304 return kHandOff;
philipel02447bc2016-05-13 06:01:03 -0700305 }
306
307 // Layer sync frame, this frame only references its base layer frame.
308 if (codec_header.layerSync) {
309 frame->num_references = 1;
310 frame->references[0] = layer_info_it->second[0];
311
philipel57ec6852018-07-03 18:09:32 +0200312 UpdateLayerInfoVp8(frame, unwrapped_tl0, codec_header.temporalIdx);
philipelafcf7f52017-04-26 08:17:35 -0700313 return kHandOff;
philipel02447bc2016-05-13 06:01:03 -0700314 }
315
316 // Find all references for this frame.
317 frame->num_references = 0;
318 for (uint8_t layer = 0; layer <= codec_header.temporalIdx; ++layer) {
philipeld268d6f2016-09-15 13:43:13 +0200319 // If we have not yet received a previous frame on this temporal layer,
320 // stash this frame.
philipelafcf7f52017-04-26 08:17:35 -0700321 if (layer_info_it->second[layer] == -1)
322 return kStash;
philipel02447bc2016-05-13 06:01:03 -0700323
philipel86b92e02016-10-24 07:11:53 -0700324 // If the last frame on this layer is ahead of this frame it means that
325 // a layer sync frame has been received after this frame for the same
326 // base layer frame, drop this frame.
327 if (AheadOf<uint16_t, kPicIdLength>(layer_info_it->second[layer],
philipel0fa82a62018-03-19 15:34:53 +0100328 frame->id.picture_id)) {
philipelafcf7f52017-04-26 08:17:35 -0700329 return kDrop;
philipel86b92e02016-10-24 07:11:53 -0700330 }
331
philipel02447bc2016-05-13 06:01:03 -0700332 // If we have not yet received a frame between this frame and the referenced
333 // frame then we have to wait for that frame to be completed first.
334 auto not_received_frame_it =
335 not_yet_received_frames_.upper_bound(layer_info_it->second[layer]);
336 if (not_received_frame_it != not_yet_received_frames_.end() &&
philipel0fa82a62018-03-19 15:34:53 +0100337 AheadOf<uint16_t, kPicIdLength>(frame->id.picture_id,
philipel02447bc2016-05-13 06:01:03 -0700338 *not_received_frame_it)) {
philipelafcf7f52017-04-26 08:17:35 -0700339 return kStash;
philipel02447bc2016-05-13 06:01:03 -0700340 }
341
philipel0fa82a62018-03-19 15:34:53 +0100342 if (!(AheadOf<uint16_t, kPicIdLength>(frame->id.picture_id,
philipel57f19cc2017-03-07 03:54:05 -0800343 layer_info_it->second[layer]))) {
philipel0fa82a62018-03-19 15:34:53 +0100344 RTC_LOG(LS_WARNING) << "Frame with picture id " << frame->id.picture_id
Mirko Bonadei675513b2017-11-09 11:09:25 +0100345 << " and packet range [" << frame->first_seq_num()
346 << ", " << frame->last_seq_num()
347 << "] already received, "
348 << " dropping frame.";
philipelafcf7f52017-04-26 08:17:35 -0700349 return kDrop;
philipel57f19cc2017-03-07 03:54:05 -0800350 }
351
philipel02447bc2016-05-13 06:01:03 -0700352 ++frame->num_references;
353 frame->references[layer] = layer_info_it->second[layer];
354 }
355
philipel57ec6852018-07-03 18:09:32 +0200356 UpdateLayerInfoVp8(frame, unwrapped_tl0, codec_header.temporalIdx);
philipelafcf7f52017-04-26 08:17:35 -0700357 return kHandOff;
philipel02447bc2016-05-13 06:01:03 -0700358}
359
philipel57ec6852018-07-03 18:09:32 +0200360void RtpFrameReferenceFinder::UpdateLayerInfoVp8(RtpFrameObject* frame,
361 int64_t unwrapped_tl0,
362 uint8_t temporal_idx) {
363 auto layer_info_it = layer_info_.find(unwrapped_tl0);
philipel02447bc2016-05-13 06:01:03 -0700364
365 // Update this layer info and newer.
366 while (layer_info_it != layer_info_.end()) {
philipel57ec6852018-07-03 18:09:32 +0200367 if (layer_info_it->second[temporal_idx] != -1 &&
368 AheadOf<uint16_t, kPicIdLength>(layer_info_it->second[temporal_idx],
philipel0fa82a62018-03-19 15:34:53 +0100369 frame->id.picture_id)) {
philipel02447bc2016-05-13 06:01:03 -0700370 // The frame was not newer, then no subsequent layer info have to be
371 // update.
372 break;
373 }
374
philipel57ec6852018-07-03 18:09:32 +0200375 layer_info_it->second[temporal_idx] = frame->id.picture_id;
376 ++unwrapped_tl0;
377 layer_info_it = layer_info_.find(unwrapped_tl0);
philipel02447bc2016-05-13 06:01:03 -0700378 }
philipel0fa82a62018-03-19 15:34:53 +0100379 not_yet_received_frames_.erase(frame->id.picture_id);
philipel02447bc2016-05-13 06:01:03 -0700380
philipelafcf7f52017-04-26 08:17:35 -0700381 UnwrapPictureIds(frame);
philipel02447bc2016-05-13 06:01:03 -0700382}
383
philipelafcf7f52017-04-26 08:17:35 -0700384RtpFrameReferenceFinder::FrameDecision RtpFrameReferenceFinder::ManageFrameVp9(
385 RtpFrameObject* frame) {
philipel5470f402018-09-07 13:38:53 +0200386 absl::optional<RTPVideoHeader> video_header = frame->GetRtpVideoHeader();
Sami Kalliomäki98824952018-08-28 14:39:21 +0200387 if (!video_header) {
Mirko Bonadei675513b2017-11-09 11:09:25 +0100388 RTC_LOG(LS_WARNING)
389 << "Failed to get codec header from frame, dropping frame.";
philipel4c140092017-08-31 08:31:45 -0700390 return kDrop;
philipeld4fac692017-09-04 07:03:46 -0700391 }
Sami Kalliomäki98824952018-08-28 14:39:21 +0200392 RTPVideoTypeHeader rtp_codec_header = video_header->video_type_header;
philipel4c140092017-08-31 08:31:45 -0700393
philipel1a4746a2018-07-09 15:52:29 +0200394 const RTPVideoHeaderVP9& codec_header =
Sami Kalliomäki98824952018-08-28 14:39:21 +0200395 absl::get<RTPVideoHeaderVP9>(rtp_codec_header);
philipel02447bc2016-05-13 06:01:03 -0700396
philipel647998c2016-06-03 09:40:16 -0700397 if (codec_header.picture_id == kNoPictureId ||
Sergey Silkind34a1882018-08-20 16:46:05 +0200398 codec_header.temporal_idx == kNoTemporalIdx) {
philipelafcf7f52017-04-26 08:17:35 -0700399 return ManageFrameGeneric(std::move(frame), codec_header.picture_id);
philipel02447bc2016-05-13 06:01:03 -0700400 }
401
philipel0fa82a62018-03-19 15:34:53 +0100402 frame->id.spatial_layer = codec_header.spatial_idx;
philipel02447bc2016-05-13 06:01:03 -0700403 frame->inter_layer_predicted = codec_header.inter_layer_predicted;
philipel0fa82a62018-03-19 15:34:53 +0100404 frame->id.picture_id = codec_header.picture_id % kPicIdLength;
philipel02447bc2016-05-13 06:01:03 -0700405
philipel02447bc2016-05-13 06:01:03 -0700406 if (last_picture_id_ == -1)
philipel0fa82a62018-03-19 15:34:53 +0100407 last_picture_id_ = frame->id.picture_id;
philipel02447bc2016-05-13 06:01:03 -0700408
409 if (codec_header.flexible_mode) {
410 frame->num_references = codec_header.num_ref_pics;
411 for (size_t i = 0; i < frame->num_references; ++i) {
philipel0fa82a62018-03-19 15:34:53 +0100412 frame->references[i] = Subtract<kPicIdLength>(frame->id.picture_id,
413 codec_header.pid_diff[i]);
philipel02447bc2016-05-13 06:01:03 -0700414 }
415
philipelafcf7f52017-04-26 08:17:35 -0700416 UnwrapPictureIds(frame);
417 return kHandOff;
philipel02447bc2016-05-13 06:01:03 -0700418 }
419
Sergey Silkind34a1882018-08-20 16:46:05 +0200420 if (codec_header.tl0_pic_idx == kNoTl0PicIdx) {
421 RTC_LOG(LS_WARNING) << "TL0PICIDX is expected to be present in "
422 "non-flexible mode.";
423 return kDrop;
424 }
425
philipel15643602018-05-03 16:14:13 +0200426 GofInfo* info;
philipel57ec6852018-07-03 18:09:32 +0200427 int64_t unwrapped_tl0 = tl0_unwrapper_.Unwrap(codec_header.tl0_pic_idx);
philipel02447bc2016-05-13 06:01:03 -0700428 if (codec_header.ss_data_available) {
philipel02447bc2016-05-13 06:01:03 -0700429 if (codec_header.temporal_idx != 0) {
philipel15643602018-05-03 16:14:13 +0200430 RTC_LOG(LS_WARNING) << "Received scalability structure on a non base "
431 "layer frame. Scalability structure ignored.";
philipel02447bc2016-05-13 06:01:03 -0700432 } else {
Sergey Silkin2f864fb2018-09-07 11:49:38 +0200433 if (codec_header.gof.num_frames_in_gof > kMaxVp9FramesInGof) {
Natalie Silvanovich3ea3e302018-05-16 11:03:12 -0700434 return kDrop;
435 }
436
Sergey Silkin2f864fb2018-09-07 11:49:38 +0200437 GofInfoVP9 gof = codec_header.gof;
438 if (gof.num_frames_in_gof == 0) {
439 RTC_LOG(LS_WARNING) << "Number of frames in GOF is zero. Assume "
440 "that stream has only one temporal layer.";
441 gof.SetGofInfoVP9(kTemporalStructureMode1);
442 }
443
444 current_ss_idx_ = Add<kMaxGofSaved>(current_ss_idx_, 1);
445 scalability_structures_[current_ss_idx_] = gof;
philipel0fa82a62018-03-19 15:34:53 +0100446 scalability_structures_[current_ss_idx_].pid_start = frame->id.picture_id;
philipel57ec6852018-07-03 18:09:32 +0200447 gof_info_.emplace(unwrapped_tl0,
philipel15643602018-05-03 16:14:13 +0200448 GofInfo(&scalability_structures_[current_ss_idx_],
449 frame->id.picture_id));
philipel02447bc2016-05-13 06:01:03 -0700450 }
philipel15643602018-05-03 16:14:13 +0200451
philipel57ec6852018-07-03 18:09:32 +0200452 const auto gof_info_it = gof_info_.find(unwrapped_tl0);
philipel15643602018-05-03 16:14:13 +0200453 if (gof_info_it == gof_info_.end())
454 return kStash;
455
456 info = &gof_info_it->second;
457
458 if (frame->frame_type() == kVideoFrameKey) {
459 frame->num_references = 0;
460 FrameReceivedVp9(frame->id.picture_id, info);
461 UnwrapPictureIds(frame);
462 return kHandOff;
463 }
464 } else {
465 if (frame->frame_type() == kVideoFrameKey) {
466 RTC_LOG(LS_WARNING) << "Received keyframe without scalability structure";
467 return kDrop;
468 }
469
philipel57ec6852018-07-03 18:09:32 +0200470 auto gof_info_it = gof_info_.find(
471 (codec_header.temporal_idx == 0) ? unwrapped_tl0 - 1 : unwrapped_tl0);
philipel15643602018-05-03 16:14:13 +0200472
473 // Gof info for this frame is not available yet, stash this frame.
474 if (gof_info_it == gof_info_.end())
475 return kStash;
476
477 if (codec_header.temporal_idx == 0) {
philipel57ec6852018-07-03 18:09:32 +0200478 gof_info_it = gof_info_
479 .emplace(unwrapped_tl0, GofInfo(gof_info_it->second.gof,
480 frame->id.picture_id))
481 .first;
philipel15643602018-05-03 16:14:13 +0200482 }
483
484 info = &gof_info_it->second;
philipel02447bc2016-05-13 06:01:03 -0700485 }
486
487 // Clean up info for base layers that are too old.
philipel57ec6852018-07-03 18:09:32 +0200488 int64_t old_tl0_pic_idx = unwrapped_tl0 - kMaxGofSaved;
philipel02447bc2016-05-13 06:01:03 -0700489 auto clean_gof_info_to = gof_info_.lower_bound(old_tl0_pic_idx);
490 gof_info_.erase(gof_info_.begin(), clean_gof_info_to);
491
philipel0fa82a62018-03-19 15:34:53 +0100492 FrameReceivedVp9(frame->id.picture_id, info);
philipel02447bc2016-05-13 06:01:03 -0700493
494 // Make sure we don't miss any frame that could potentially have the
495 // up switch flag set.
philipel0fa82a62018-03-19 15:34:53 +0100496 if (MissingRequiredFrameVp9(frame->id.picture_id, *info))
philipelafcf7f52017-04-26 08:17:35 -0700497 return kStash;
philipel02447bc2016-05-13 06:01:03 -0700498
philipel15643602018-05-03 16:14:13 +0200499 if (codec_header.temporal_up_switch)
500 up_switch_.emplace(frame->id.picture_id, codec_header.temporal_idx);
philipel02447bc2016-05-13 06:01:03 -0700501
502 // Clean out old info about up switch frames.
philipel0fa82a62018-03-19 15:34:53 +0100503 uint16_t old_picture_id = Subtract<kPicIdLength>(frame->id.picture_id, 50);
philipel02447bc2016-05-13 06:01:03 -0700504 auto up_switch_erase_to = up_switch_.lower_bound(old_picture_id);
505 up_switch_.erase(up_switch_.begin(), up_switch_erase_to);
506
philipelc9b27d52016-07-15 06:50:27 -0700507 size_t diff = ForwardDiff<uint16_t, kPicIdLength>(info->gof->pid_start,
philipel0fa82a62018-03-19 15:34:53 +0100508 frame->id.picture_id);
philipelc9b27d52016-07-15 06:50:27 -0700509 size_t gof_idx = diff % info->gof->num_frames_in_gof;
philipel02447bc2016-05-13 06:01:03 -0700510
511 // Populate references according to the scalability structure.
philipelc9b27d52016-07-15 06:50:27 -0700512 frame->num_references = info->gof->num_ref_pics[gof_idx];
philipel02447bc2016-05-13 06:01:03 -0700513 for (size_t i = 0; i < frame->num_references; ++i) {
philipelc9b27d52016-07-15 06:50:27 -0700514 frame->references[i] = Subtract<kPicIdLength>(
philipel0fa82a62018-03-19 15:34:53 +0100515 frame->id.picture_id, info->gof->pid_diff[gof_idx][i]);
philipel02447bc2016-05-13 06:01:03 -0700516
517 // If this is a reference to a frame earlier than the last up switch point,
518 // then ignore this reference.
philipel0fa82a62018-03-19 15:34:53 +0100519 if (UpSwitchInIntervalVp9(frame->id.picture_id, codec_header.temporal_idx,
philipel02447bc2016-05-13 06:01:03 -0700520 frame->references[i])) {
521 --frame->num_references;
522 }
523 }
524
philipelafcf7f52017-04-26 08:17:35 -0700525 UnwrapPictureIds(frame);
526 return kHandOff;
philipel02447bc2016-05-13 06:01:03 -0700527}
528
529bool RtpFrameReferenceFinder::MissingRequiredFrameVp9(uint16_t picture_id,
philipelc9b27d52016-07-15 06:50:27 -0700530 const GofInfo& info) {
531 size_t diff =
532 ForwardDiff<uint16_t, kPicIdLength>(info.gof->pid_start, picture_id);
533 size_t gof_idx = diff % info.gof->num_frames_in_gof;
534 size_t temporal_idx = info.gof->temporal_idx[gof_idx];
philipel02447bc2016-05-13 06:01:03 -0700535
philipela157e082018-05-02 15:19:01 +0200536 if (temporal_idx >= kMaxTemporalLayers) {
537 RTC_LOG(LS_WARNING) << "At most " << kMaxTemporalLayers << " temporal "
538 << "layers are supported.";
539 return true;
540 }
541
philipel02447bc2016-05-13 06:01:03 -0700542 // For every reference this frame has, check if there is a frame missing in
543 // the interval (|ref_pid|, |picture_id|) in any of the lower temporal
544 // layers. If so, we are missing a required frame.
philipelc9b27d52016-07-15 06:50:27 -0700545 uint8_t num_references = info.gof->num_ref_pics[gof_idx];
philipel02447bc2016-05-13 06:01:03 -0700546 for (size_t i = 0; i < num_references; ++i) {
547 uint16_t ref_pid =
philipelc9b27d52016-07-15 06:50:27 -0700548 Subtract<kPicIdLength>(picture_id, info.gof->pid_diff[gof_idx][i]);
philipel02447bc2016-05-13 06:01:03 -0700549 for (size_t l = 0; l < temporal_idx; ++l) {
550 auto missing_frame_it = missing_frames_for_layer_[l].lower_bound(ref_pid);
551 if (missing_frame_it != missing_frames_for_layer_[l].end() &&
552 AheadOf<uint16_t, kPicIdLength>(picture_id, *missing_frame_it)) {
553 return true;
554 }
555 }
556 }
557 return false;
558}
559
560void RtpFrameReferenceFinder::FrameReceivedVp9(uint16_t picture_id,
philipelc9b27d52016-07-15 06:50:27 -0700561 GofInfo* info) {
562 int last_picture_id = info->last_picture_id;
philipel459f4e32018-03-02 10:55:12 +0100563 size_t gof_size = std::min(info->gof->num_frames_in_gof, kMaxVp9FramesInGof);
philipel02447bc2016-05-13 06:01:03 -0700564
565 // If there is a gap, find which temporal layer the missing frames
566 // belong to and add the frame as missing for that temporal layer.
567 // Otherwise, remove this frame from the set of missing frames.
philipelc9b27d52016-07-15 06:50:27 -0700568 if (AheadOf<uint16_t, kPicIdLength>(picture_id, last_picture_id)) {
569 size_t diff = ForwardDiff<uint16_t, kPicIdLength>(info->gof->pid_start,
570 last_picture_id);
philipel459f4e32018-03-02 10:55:12 +0100571 size_t gof_idx = diff % gof_size;
philipel02447bc2016-05-13 06:01:03 -0700572
philipelc9b27d52016-07-15 06:50:27 -0700573 last_picture_id = Add<kPicIdLength>(last_picture_id, 1);
574 while (last_picture_id != picture_id) {
Yves Gerey665174f2018-06-19 15:03:05 +0200575 gof_idx = (gof_idx + 1) % gof_size;
philipel459f4e32018-03-02 10:55:12 +0100576 RTC_CHECK(gof_idx < kMaxVp9FramesInGof);
577
philipelc9b27d52016-07-15 06:50:27 -0700578 size_t temporal_idx = info->gof->temporal_idx[gof_idx];
philipel459f4e32018-03-02 10:55:12 +0100579 if (temporal_idx >= kMaxTemporalLayers) {
580 RTC_LOG(LS_WARNING) << "At most " << kMaxTemporalLayers << " temporal "
581 << "layers are supported.";
582 return;
583 }
584
philipelc9b27d52016-07-15 06:50:27 -0700585 missing_frames_for_layer_[temporal_idx].insert(last_picture_id);
586 last_picture_id = Add<kPicIdLength>(last_picture_id, 1);
philipel02447bc2016-05-13 06:01:03 -0700587 }
philipel459f4e32018-03-02 10:55:12 +0100588
philipelc9b27d52016-07-15 06:50:27 -0700589 info->last_picture_id = last_picture_id;
philipel02447bc2016-05-13 06:01:03 -0700590 } else {
591 size_t diff =
philipelc9b27d52016-07-15 06:50:27 -0700592 ForwardDiff<uint16_t, kPicIdLength>(info->gof->pid_start, picture_id);
philipel459f4e32018-03-02 10:55:12 +0100593 size_t gof_idx = diff % gof_size;
594 RTC_CHECK(gof_idx < kMaxVp9FramesInGof);
595
philipelc9b27d52016-07-15 06:50:27 -0700596 size_t temporal_idx = info->gof->temporal_idx[gof_idx];
philipel459f4e32018-03-02 10:55:12 +0100597 if (temporal_idx >= kMaxTemporalLayers) {
598 RTC_LOG(LS_WARNING) << "At most " << kMaxTemporalLayers << " temporal "
599 << "layers are supported.";
600 return;
601 }
602
philipel02447bc2016-05-13 06:01:03 -0700603 missing_frames_for_layer_[temporal_idx].erase(picture_id);
604 }
605}
606
607bool RtpFrameReferenceFinder::UpSwitchInIntervalVp9(uint16_t picture_id,
608 uint8_t temporal_idx,
609 uint16_t pid_ref) {
610 for (auto up_switch_it = up_switch_.upper_bound(pid_ref);
611 up_switch_it != up_switch_.end() &&
612 AheadOf<uint16_t, kPicIdLength>(picture_id, up_switch_it->first);
613 ++up_switch_it) {
614 if (up_switch_it->second < temporal_idx)
615 return true;
616 }
617
618 return false;
619}
620
philipelafcf7f52017-04-26 08:17:35 -0700621void RtpFrameReferenceFinder::UnwrapPictureIds(RtpFrameObject* frame) {
philipel02447bc2016-05-13 06:01:03 -0700622 for (size_t i = 0; i < frame->num_references; ++i)
philipeld4fac692017-09-04 07:03:46 -0700623 frame->references[i] = unwrapper_.Unwrap(frame->references[i]);
philipel0fa82a62018-03-19 15:34:53 +0100624 frame->id.picture_id = unwrapper_.Unwrap(frame->id.picture_id);
philipel02447bc2016-05-13 06:01:03 -0700625}
626
philipel02447bc2016-05-13 06:01:03 -0700627} // namespace video_coding
628} // namespace webrtc