blob: b70f44f1df3d450e0ec5b4bf6955056a7a1c7a82 [file] [log] [blame]
niklase@google.com470e71d2011-07-07 08:21:25 +00001/*
stefan@webrtc.org29794612012-02-08 08:58:55 +00002 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
niklase@google.com470e71d2011-07-07 08:21:25 +00003 *
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 */
Henrik Kjellander2557b862015-11-18 22:00:21 +010010#include "webrtc/modules/video_coding/jitter_buffer.h"
niklase@google.com470e71d2011-07-07 08:21:25 +000011
pbos@webrtc.org12dc1a32013-08-05 16:22:53 +000012#include <assert.h>
13
stefan@webrtc.org29794612012-02-08 08:58:55 +000014#include <algorithm>
philipel83f831a2016-03-12 03:30:23 -080015#include <limits>
agalusza@google.comd818dcb2013-07-29 21:48:11 +000016#include <utility>
stefan@webrtc.org29794612012-02-08 08:58:55 +000017
asapersson9a4cd872015-10-23 00:27:14 -070018#include "webrtc/base/checks.h"
pbos854e84c2015-11-16 16:39:06 -080019#include "webrtc/base/logging.h"
tommie4f96502015-10-20 23:00:48 -070020#include "webrtc/base/trace_event.h"
Henrik Kjellanderff761fb2015-11-04 08:31:52 +010021#include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h"
Henrik Kjellander2557b862015-11-18 22:00:21 +010022#include "webrtc/modules/video_coding/include/video_coding.h"
23#include "webrtc/modules/video_coding/frame_buffer.h"
24#include "webrtc/modules/video_coding/inter_frame_delay.h"
25#include "webrtc/modules/video_coding/internal_defines.h"
26#include "webrtc/modules/video_coding/jitter_buffer_common.h"
27#include "webrtc/modules/video_coding/jitter_estimator.h"
28#include "webrtc/modules/video_coding/packet.h"
Henrik Kjellander98f53512015-10-28 18:17:40 +010029#include "webrtc/system_wrappers/include/clock.h"
30#include "webrtc/system_wrappers/include/critical_section_wrapper.h"
31#include "webrtc/system_wrappers/include/event_wrapper.h"
philipel83f831a2016-03-12 03:30:23 -080032#include "webrtc/system_wrappers/include/field_trial.h"
Henrik Kjellander98f53512015-10-28 18:17:40 +010033#include "webrtc/system_wrappers/include/metrics.h"
niklase@google.com470e71d2011-07-07 08:21:25 +000034
niklase@google.com470e71d2011-07-07 08:21:25 +000035namespace webrtc {
asapersson9a4cd872015-10-23 00:27:14 -070036// Interval for updating SS data.
37static const uint32_t kSsCleanupIntervalSec = 60;
38
mflodman@webrtc.org7c894b72012-11-26 12:40:15 +000039// Use this rtt if no value has been reported.
pkasting@chromium.org16825b12015-01-12 21:51:21 +000040static const int64_t kDefaultRtt = 200;
mflodman@webrtc.org7c894b72012-11-26 12:40:15 +000041
jbauchdb81ffd2015-11-23 03:59:02 -080042// Request a keyframe if no continuous frame has been received for this
43// number of milliseconds and NACKs are disabled.
44static const int64_t kMaxDiscontinuousFramesTime = 1000;
45
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +000046typedef std::pair<uint32_t, VCMFrameBuffer*> FrameListPair;
niklase@google.com470e71d2011-07-07 08:21:25 +000047
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +000048bool IsKeyFrame(FrameListPair pair) {
49 return pair.second->FrameType() == kVideoFrameKey;
50}
stefan@webrtc.org29794612012-02-08 08:58:55 +000051
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +000052bool HasNonEmptyState(FrameListPair pair) {
53 return pair.second->GetState() != kStateEmpty;
stefan@webrtc.orga64300a2013-03-04 15:24:40 +000054}
55
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +000056void FrameList::InsertFrame(VCMFrameBuffer* frame) {
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +000057 insert(rbegin().base(), FrameListPair(frame->TimeStamp(), frame));
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +000058}
59
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +000060VCMFrameBuffer* FrameList::PopFrame(uint32_t timestamp) {
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +000061 FrameList::iterator it = find(timestamp);
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +000062 if (it == end())
63 return NULL;
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +000064 VCMFrameBuffer* frame = it->second;
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +000065 erase(it);
66 return frame;
67}
68
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +000069VCMFrameBuffer* FrameList::Front() const {
70 return begin()->second;
71}
72
73VCMFrameBuffer* FrameList::Back() const {
74 return rbegin()->second;
75}
76
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +000077int FrameList::RecycleFramesUntilKeyFrame(FrameList::iterator* key_frame_it,
78 UnorderedFrameList* free_frames) {
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +000079 int drop_count = 0;
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +000080 FrameList::iterator it = begin();
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +000081 while (!empty()) {
82 // Throw at least one frame.
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +000083 it->second->Reset();
84 free_frames->push_back(it->second);
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +000085 erase(it++);
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +000086 ++drop_count;
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +000087 if (it != end() && it->second->FrameType() == kVideoFrameKey) {
88 *key_frame_it = it;
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +000089 return drop_count;
90 }
91 }
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +000092 *key_frame_it = end();
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +000093 return drop_count;
94}
95
pbos@webrtc.org4f16c872014-11-24 09:06:48 +000096void FrameList::CleanUpOldOrEmptyFrames(VCMDecodingState* decoding_state,
philipel9d3ab612015-12-21 04:12:39 -080097 UnorderedFrameList* free_frames) {
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +000098 while (!empty()) {
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +000099 VCMFrameBuffer* oldest_frame = Front();
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000100 bool remove_frame = false;
101 if (oldest_frame->GetState() == kStateEmpty && size() > 1) {
102 // This frame is empty, try to update the last decoded state and drop it
103 // if successful.
104 remove_frame = decoding_state->UpdateEmptyFrame(oldest_frame);
105 } else {
106 remove_frame = decoding_state->IsOldFrame(oldest_frame);
107 }
108 if (!remove_frame) {
109 break;
110 }
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +0000111 free_frames->push_back(oldest_frame);
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000112 TRACE_EVENT_INSTANT1("webrtc", "JB::OldOrEmptyFrameDropped", "timestamp",
113 oldest_frame->TimeStamp());
114 erase(begin());
115 }
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000116}
117
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +0000118void FrameList::Reset(UnorderedFrameList* free_frames) {
119 while (!empty()) {
120 begin()->second->Reset();
121 free_frames->push_back(begin()->second);
122 erase(begin());
123 }
124}
125
asapersson9a4cd872015-10-23 00:27:14 -0700126bool Vp9SsMap::Insert(const VCMPacket& packet) {
127 if (!packet.codecSpecificHeader.codecHeader.VP9.ss_data_available)
128 return false;
129
130 ss_map_[packet.timestamp] = packet.codecSpecificHeader.codecHeader.VP9.gof;
131 return true;
132}
133
134void Vp9SsMap::Reset() {
135 ss_map_.clear();
136}
137
138bool Vp9SsMap::Find(uint32_t timestamp, SsMap::iterator* it_out) {
139 bool found = false;
140 for (SsMap::iterator it = ss_map_.begin(); it != ss_map_.end(); ++it) {
141 if (it->first == timestamp || IsNewerTimestamp(timestamp, it->first)) {
142 *it_out = it;
143 found = true;
144 }
145 }
146 return found;
147}
148
149void Vp9SsMap::RemoveOld(uint32_t timestamp) {
150 if (!TimeForCleanup(timestamp))
151 return;
152
153 SsMap::iterator it;
154 if (!Find(timestamp, &it))
155 return;
156
157 ss_map_.erase(ss_map_.begin(), it);
158 AdvanceFront(timestamp);
159}
160
161bool Vp9SsMap::TimeForCleanup(uint32_t timestamp) const {
162 if (ss_map_.empty() || !IsNewerTimestamp(timestamp, ss_map_.begin()->first))
163 return false;
164
165 uint32_t diff = timestamp - ss_map_.begin()->first;
166 return diff / kVideoPayloadTypeFrequency >= kSsCleanupIntervalSec;
167}
168
169void Vp9SsMap::AdvanceFront(uint32_t timestamp) {
170 RTC_DCHECK(!ss_map_.empty());
171 GofInfoVP9 gof = ss_map_.begin()->second;
172 ss_map_.erase(ss_map_.begin());
173 ss_map_[timestamp] = gof;
174}
175
asaperssonc253a1c2015-11-06 00:12:01 -0800176// TODO(asapersson): Update according to updates in RTP payload profile.
asapersson9a4cd872015-10-23 00:27:14 -0700177bool Vp9SsMap::UpdatePacket(VCMPacket* packet) {
178 uint8_t gof_idx = packet->codecSpecificHeader.codecHeader.VP9.gof_idx;
179 if (gof_idx == kNoGofIdx)
180 return false; // No update needed.
181
182 SsMap::iterator it;
183 if (!Find(packet->timestamp, &it))
184 return false; // Corresponding SS not yet received.
185
186 if (gof_idx >= it->second.num_frames_in_gof)
187 return false; // Assume corresponding SS not yet received.
188
189 RTPVideoHeaderVP9* vp9 = &packet->codecSpecificHeader.codecHeader.VP9;
190 vp9->temporal_idx = it->second.temporal_idx[gof_idx];
191 vp9->temporal_up_switch = it->second.temporal_up_switch[gof_idx];
192
193 // TODO(asapersson): Set vp9.ref_picture_id[i] and add usage.
194 vp9->num_ref_pics = it->second.num_ref_pics[gof_idx];
asaperssonc253a1c2015-11-06 00:12:01 -0800195 for (uint8_t i = 0; i < it->second.num_ref_pics[gof_idx]; ++i) {
asapersson9a4cd872015-10-23 00:27:14 -0700196 vp9->pid_diff[i] = it->second.pid_diff[gof_idx][i];
197 }
198 return true;
199}
200
201void Vp9SsMap::UpdateFrames(FrameList* frames) {
202 for (const auto& frame_it : *frames) {
203 uint8_t gof_idx =
204 frame_it.second->CodecSpecific()->codecSpecific.VP9.gof_idx;
205 if (gof_idx == kNoGofIdx) {
206 continue;
207 }
208 SsMap::iterator ss_it;
209 if (Find(frame_it.second->TimeStamp(), &ss_it)) {
210 if (gof_idx >= ss_it->second.num_frames_in_gof) {
211 continue; // Assume corresponding SS not yet received.
212 }
213 frame_it.second->SetGofInfo(ss_it->second, gof_idx);
214 }
215 }
216}
217
philipel83f831a2016-03-12 03:30:23 -0800218static NackModule*
219MaybeCreateNackModule(Clock* clock,
220 NackSender* nack_sender,
221 KeyFrameRequestSender* keyframe_request_sender) {
222if (field_trial::FindFullName("WebRTC-NewVideoJitterBuffer") == "Enabled") {
223 // Only create a nack module if a nack sender is available.
224 if (nack_sender) {
225 RTC_DCHECK(keyframe_request_sender);
226 return new NackModule(clock, nack_sender, keyframe_request_sender);
227 }
228 }
229 return nullptr;
230}
231
Qiang Chend4cec152015-06-19 09:17:00 -0700232VCMJitterBuffer::VCMJitterBuffer(Clock* clock,
philipel83f831a2016-03-12 03:30:23 -0800233 std::unique_ptr<EventWrapper> event,
234 NackSender* nack_sender,
235 KeyFrameRequestSender* keyframe_request_sender)
stefan@webrtc.org34c5da62014-04-11 14:08:35 +0000236 : clock_(clock),
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000237 running_(false),
238 crit_sect_(CriticalSectionWrapper::CreateCriticalSection()),
kwiberg0eb15ed2015-12-17 03:04:15 -0800239 frame_event_(std::move(event)),
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000240 max_number_of_frames_(kStartNumberOfFrames),
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +0000241 free_frames_(),
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000242 decodable_frames_(),
243 incomplete_frames_(),
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000244 last_decoded_state_(),
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000245 first_packet_since_reset_(true),
pbos@webrtc.org55707692014-12-19 15:45:03 +0000246 stats_callback_(NULL),
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000247 incoming_frame_rate_(0),
248 incoming_frame_count_(0),
249 time_last_incoming_frame_count_(0),
250 incoming_bit_count_(0),
251 incoming_bit_rate_(0),
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000252 num_consecutive_old_packets_(0),
asapersson@webrtc.org96dc6852014-11-03 14:40:38 +0000253 num_packets_(0),
254 num_duplicated_packets_(0),
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000255 num_discarded_packets_(0),
asapersson@webrtc.org83b52002014-11-28 10:17:13 +0000256 time_first_packet_ms_(0),
sprang@webrtc.org70e2d112014-09-24 14:06:56 +0000257 jitter_estimate_(clock),
stefan@webrtc.orga678a3b2013-01-21 07:42:11 +0000258 inter_frame_delay_(clock_->TimeInMilliseconds()),
mflodman@webrtc.org7c894b72012-11-26 12:40:15 +0000259 rtt_ms_(kDefaultRtt),
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000260 nack_mode_(kNoNack),
261 low_rtt_nack_threshold_ms_(-1),
262 high_rtt_nack_threshold_ms_(-1),
stefan@webrtc.orga64300a2013-03-04 15:24:40 +0000263 missing_sequence_numbers_(SequenceNumberLessThan()),
stefan@webrtc.orgbecf9c82013-02-01 15:09:57 +0000264 max_nack_list_size_(0),
265 max_packet_age_to_nack_(0),
stefan@webrtc.orgef144882013-05-07 19:16:33 +0000266 max_incomplete_time_ms_(0),
agalusza@google.coma7e360e2013-08-01 03:15:08 +0000267 decode_error_mode_(kNoErrors),
agalusza@google.comd818dcb2013-07-29 21:48:11 +0000268 average_packets_per_frame_(0.0f),
philipel83f831a2016-03-12 03:30:23 -0800269 frame_counter_(0),
270 nack_module_(MaybeCreateNackModule(clock,
271 nack_sender,
272 keyframe_request_sender)) {
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000273 for (int i = 0; i < kStartNumberOfFrames; i++)
274 free_frames_.push_back(new VCMFrameBuffer());
niklase@google.com470e71d2011-07-07 08:21:25 +0000275}
276
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000277VCMJitterBuffer::~VCMJitterBuffer() {
278 Stop();
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000279 for (UnorderedFrameList::iterator it = free_frames_.begin();
280 it != free_frames_.end(); ++it) {
281 delete *it;
282 }
283 for (FrameList::iterator it = incomplete_frames_.begin();
284 it != incomplete_frames_.end(); ++it) {
285 delete it->second;
286 }
287 for (FrameList::iterator it = decodable_frames_.begin();
288 it != decodable_frames_.end(); ++it) {
289 delete it->second;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000290 }
291 delete crit_sect_;
niklase@google.com470e71d2011-07-07 08:21:25 +0000292}
293
asapersson@webrtc.org96dc6852014-11-03 14:40:38 +0000294void VCMJitterBuffer::UpdateHistograms() {
Ã…sa Perssona96f02b2015-04-24 08:52:11 +0200295 if (num_packets_ <= 0 || !running_) {
asapersson@webrtc.org83b52002014-11-28 10:17:13 +0000296 return;
297 }
298 int64_t elapsed_sec =
299 (clock_->TimeInMilliseconds() - time_first_packet_ms_) / 1000;
300 if (elapsed_sec < metrics::kMinRunTimeInSeconds) {
301 return;
302 }
303
asapersson58d992e2016-03-29 02:15:06 -0700304 RTC_LOGGED_HISTOGRAM_PERCENTAGE("WebRTC.Video.DiscardedPacketsInPercent",
305 num_discarded_packets_ * 100 / num_packets_);
306 RTC_LOGGED_HISTOGRAM_PERCENTAGE("WebRTC.Video.DuplicatedPacketsInPercent",
307 num_duplicated_packets_ * 100 / num_packets_);
asapersson@webrtc.org83b52002014-11-28 10:17:13 +0000308
pbos@webrtc.orgce4e9a32014-12-18 13:50:16 +0000309 int total_frames =
310 receive_statistics_.key_frames + receive_statistics_.delta_frames;
asapersson@webrtc.org83b52002014-11-28 10:17:13 +0000311 if (total_frames > 0) {
asapersson58d992e2016-03-29 02:15:06 -0700312 RTC_LOGGED_HISTOGRAM_COUNTS_100(
asapersson53805322015-12-21 01:46:20 -0800313 "WebRTC.Video.CompleteFramesReceivedPerSecond",
asapersson@webrtc.org83b52002014-11-28 10:17:13 +0000314 static_cast<int>((total_frames / elapsed_sec) + 0.5f));
asapersson58d992e2016-03-29 02:15:06 -0700315 RTC_LOGGED_HISTOGRAM_COUNTS_1000(
pbos@webrtc.orgce4e9a32014-12-18 13:50:16 +0000316 "WebRTC.Video.KeyFramesReceivedInPermille",
317 static_cast<int>(
318 (receive_statistics_.key_frames * 1000.0f / total_frames) + 0.5f));
asapersson@webrtc.org96dc6852014-11-03 14:40:38 +0000319 }
320}
321
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000322void VCMJitterBuffer::Start() {
323 CriticalSectionScoped cs(crit_sect_);
324 running_ = true;
325 incoming_frame_count_ = 0;
326 incoming_frame_rate_ = 0;
327 incoming_bit_count_ = 0;
328 incoming_bit_rate_ = 0;
stefan@webrtc.orga678a3b2013-01-21 07:42:11 +0000329 time_last_incoming_frame_count_ = clock_->TimeInMilliseconds();
pbos@webrtc.orgce4e9a32014-12-18 13:50:16 +0000330 receive_statistics_ = FrameCounts();
niklase@google.com470e71d2011-07-07 08:21:25 +0000331
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000332 num_consecutive_old_packets_ = 0;
asapersson@webrtc.org96dc6852014-11-03 14:40:38 +0000333 num_packets_ = 0;
334 num_duplicated_packets_ = 0;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000335 num_discarded_packets_ = 0;
asapersson@webrtc.org83b52002014-11-28 10:17:13 +0000336 time_first_packet_ms_ = 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000337
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000338 // Start in a non-signaled state.
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000339 waiting_for_completion_.frame_size = 0;
340 waiting_for_completion_.timestamp = 0;
341 waiting_for_completion_.latest_packet_time = -1;
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000342 first_packet_since_reset_ = true;
mflodman@webrtc.org7c894b72012-11-26 12:40:15 +0000343 rtt_ms_ = kDefaultRtt;
mikhal@webrtc.org8392cd92013-04-25 21:30:50 +0000344 last_decoded_state_.Reset();
niklase@google.com470e71d2011-07-07 08:21:25 +0000345}
346
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000347void VCMJitterBuffer::Stop() {
348 crit_sect_->Enter();
asapersson@webrtc.org96dc6852014-11-03 14:40:38 +0000349 UpdateHistograms();
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000350 running_ = false;
351 last_decoded_state_.Reset();
philipel83f831a2016-03-12 03:30:23 -0800352 if (nack_module_)
353 nack_module_->Stop();
asaperssona9455ab2015-07-31 06:10:09 -0700354
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000355 // Make sure all frames are free and reset.
356 for (FrameList::iterator it = decodable_frames_.begin();
357 it != decodable_frames_.end(); ++it) {
358 free_frames_.push_back(it->second);
359 }
360 for (FrameList::iterator it = incomplete_frames_.begin();
361 it != incomplete_frames_.end(); ++it) {
362 free_frames_.push_back(it->second);
363 }
364 for (UnorderedFrameList::iterator it = free_frames_.begin();
365 it != free_frames_.end(); ++it) {
366 (*it)->Reset();
367 }
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000368 decodable_frames_.clear();
369 incomplete_frames_.clear();
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000370 crit_sect_->Leave();
371 // Make sure we wake up any threads waiting on these events.
stefan@webrtc.org2baf5f52013-03-13 08:46:25 +0000372 frame_event_->Set();
niklase@google.com470e71d2011-07-07 08:21:25 +0000373}
374
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000375bool VCMJitterBuffer::Running() const {
376 CriticalSectionScoped cs(crit_sect_);
377 return running_;
niklase@google.com470e71d2011-07-07 08:21:25 +0000378}
379
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000380void VCMJitterBuffer::Flush() {
381 CriticalSectionScoped cs(crit_sect_);
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +0000382 decodable_frames_.Reset(&free_frames_);
383 incomplete_frames_.Reset(&free_frames_);
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000384 last_decoded_state_.Reset(); // TODO(mikhal): sync reset.
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000385 num_consecutive_old_packets_ = 0;
386 // Also reset the jitter and delay estimates
387 jitter_estimate_.Reset();
stefan@webrtc.orga678a3b2013-01-21 07:42:11 +0000388 inter_frame_delay_.Reset(clock_->TimeInMilliseconds());
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000389 waiting_for_completion_.frame_size = 0;
390 waiting_for_completion_.timestamp = 0;
391 waiting_for_completion_.latest_packet_time = -1;
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000392 first_packet_since_reset_ = true;
stefan@webrtc.orga64300a2013-03-04 15:24:40 +0000393 missing_sequence_numbers_.clear();
philipel83f831a2016-03-12 03:30:23 -0800394 if (nack_module_)
395 nack_module_->Clear();
niklase@google.com470e71d2011-07-07 08:21:25 +0000396}
397
niklase@google.com470e71d2011-07-07 08:21:25 +0000398// Get received key and delta frames
pbos@webrtc.orgce4e9a32014-12-18 13:50:16 +0000399FrameCounts VCMJitterBuffer::FrameStatistics() const {
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000400 CriticalSectionScoped cs(crit_sect_);
sprang@webrtc.org71f055f2013-12-04 15:09:27 +0000401 return receive_statistics_;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000402}
403
asapersson@webrtc.org96dc6852014-11-03 14:40:38 +0000404int VCMJitterBuffer::num_packets() const {
405 CriticalSectionScoped cs(crit_sect_);
406 return num_packets_;
407}
408
409int VCMJitterBuffer::num_duplicated_packets() const {
410 CriticalSectionScoped cs(crit_sect_);
411 return num_duplicated_packets_;
412}
413
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000414int VCMJitterBuffer::num_discarded_packets() const {
415 CriticalSectionScoped cs(crit_sect_);
416 return num_discarded_packets_;
417}
418
419// Calculate framerate and bitrate.
420void VCMJitterBuffer::IncomingRateStatistics(unsigned int* framerate,
421 unsigned int* bitrate) {
422 assert(framerate);
423 assert(bitrate);
424 CriticalSectionScoped cs(crit_sect_);
stefan@webrtc.orga678a3b2013-01-21 07:42:11 +0000425 const int64_t now = clock_->TimeInMilliseconds();
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000426 int64_t diff = now - time_last_incoming_frame_count_;
427 if (diff < 1000 && incoming_frame_rate_ > 0 && incoming_bit_rate_ > 0) {
428 // Make sure we report something even though less than
429 // 1 second has passed since last update.
430 *framerate = incoming_frame_rate_;
431 *bitrate = incoming_bit_rate_;
432 } else if (incoming_frame_count_ != 0) {
433 // We have received frame(s) since last call to this function
434
435 // Prepare calculations
436 if (diff <= 0) {
437 diff = 1;
niklase@google.com470e71d2011-07-07 08:21:25 +0000438 }
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000439 // we add 0.5f for rounding
440 float rate = 0.5f + ((incoming_frame_count_ * 1000.0f) / diff);
441 if (rate < 1.0f) {
442 rate = 1.0f;
443 }
444
445 // Calculate frame rate
446 // Let r be rate.
447 // r(0) = 1000*framecount/delta_time.
448 // (I.e. frames per second since last calculation.)
449 // frame_rate = r(0)/2 + r(-1)/2
450 // (I.e. fr/s average this and the previous calculation.)
451 *framerate = (incoming_frame_rate_ + static_cast<unsigned int>(rate)) / 2;
452 incoming_frame_rate_ = static_cast<unsigned int>(rate);
453
454 // Calculate bit rate
455 if (incoming_bit_count_ == 0) {
456 *bitrate = 0;
457 } else {
philipel9d3ab612015-12-21 04:12:39 -0800458 *bitrate =
459 10 * ((100 * incoming_bit_count_) / static_cast<unsigned int>(diff));
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000460 }
461 incoming_bit_rate_ = *bitrate;
462
463 // Reset count
464 incoming_frame_count_ = 0;
465 incoming_bit_count_ = 0;
466 time_last_incoming_frame_count_ = now;
467
468 } else {
469 // No frames since last call
stefan@webrtc.orga678a3b2013-01-21 07:42:11 +0000470 time_last_incoming_frame_count_ = clock_->TimeInMilliseconds();
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000471 *framerate = 0;
stefan@webrtc.org49806792013-04-30 22:05:07 +0000472 *bitrate = 0;
473 incoming_frame_rate_ = 0;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000474 incoming_bit_rate_ = 0;
475 }
niklase@google.com470e71d2011-07-07 08:21:25 +0000476}
477
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000478// Answers the question:
479// Will the packet sequence be complete if the next frame is grabbed for
480// decoding right now? That is, have we lost a frame between the last decoded
481// frame and the next, or is the next
482// frame missing one or more packets?
483bool VCMJitterBuffer::CompleteSequenceWithNextFrame() {
484 CriticalSectionScoped cs(crit_sect_);
485 // Finding oldest frame ready for decoder, check sequence number and size
stefan@webrtc.orga64300a2013-03-04 15:24:40 +0000486 CleanUpOldOrEmptyFrames();
agalusza@google.comd177c102013-08-08 01:12:33 +0000487 if (!decodable_frames_.empty()) {
488 if (decodable_frames_.Front()->GetState() == kStateComplete) {
489 return true;
490 }
491 } else if (incomplete_frames_.size() <= 1) {
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000492 // Frame not ready to be decoded.
493 return true;
494 }
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000495 return false;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000496}
497
498// Returns immediately or a |max_wait_time_ms| ms event hang waiting for a
499// complete frame, |max_wait_time_ms| decided by caller.
philipel9d3ab612015-12-21 04:12:39 -0800500bool VCMJitterBuffer::NextCompleteTimestamp(uint32_t max_wait_time_ms,
501 uint32_t* timestamp) {
mikhal@webrtc.orgc1f243f2013-04-22 22:24:38 +0000502 crit_sect_->Enter();
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000503 if (!running_) {
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000504 crit_sect_->Leave();
mikhal@webrtc.org759b0412013-05-07 16:36:00 +0000505 return false;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000506 }
stefan@webrtc.orga64300a2013-03-04 15:24:40 +0000507 CleanUpOldOrEmptyFrames();
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000508
agalusza@google.comd177c102013-08-08 01:12:33 +0000509 if (decodable_frames_.empty() ||
510 decodable_frames_.Front()->GetState() != kStateComplete) {
philipel9d3ab612015-12-21 04:12:39 -0800511 const int64_t end_wait_time_ms =
512 clock_->TimeInMilliseconds() + max_wait_time_ms;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000513 int64_t wait_time_ms = max_wait_time_ms;
514 while (wait_time_ms > 0) {
515 crit_sect_->Leave();
516 const EventTypeWrapper ret =
philipel9d3ab612015-12-21 04:12:39 -0800517 frame_event_->Wait(static_cast<uint32_t>(wait_time_ms));
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000518 crit_sect_->Enter();
519 if (ret == kEventSignaled) {
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000520 // Are we shutting down the jitter buffer?
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000521 if (!running_) {
522 crit_sect_->Leave();
mikhal@webrtc.org759b0412013-05-07 16:36:00 +0000523 return false;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000524 }
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +0000525 // Finding oldest frame ready for decoder.
stefan@webrtc.orga64300a2013-03-04 15:24:40 +0000526 CleanUpOldOrEmptyFrames();
agalusza@google.comd177c102013-08-08 01:12:33 +0000527 if (decodable_frames_.empty() ||
528 decodable_frames_.Front()->GetState() != kStateComplete) {
stefan@webrtc.orga678a3b2013-01-21 07:42:11 +0000529 wait_time_ms = end_wait_time_ms - clock_->TimeInMilliseconds();
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000530 } else {
531 break;
532 }
533 } else {
mikhal@webrtc.org9c7685f2013-05-07 16:07:52 +0000534 break;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000535 }
536 }
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000537 }
agalusza@google.comd177c102013-08-08 01:12:33 +0000538 if (decodable_frames_.empty() ||
539 decodable_frames_.Front()->GetState() != kStateComplete) {
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000540 crit_sect_->Leave();
541 return false;
mikhal@webrtc.orgdc3cd212013-04-25 20:27:04 +0000542 }
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +0000543 *timestamp = decodable_frames_.Front()->TimeStamp();
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000544 crit_sect_->Leave();
mikhal@webrtc.org759b0412013-05-07 16:36:00 +0000545 return true;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000546}
547
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000548bool VCMJitterBuffer::NextMaybeIncompleteTimestamp(uint32_t* timestamp) {
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000549 CriticalSectionScoped cs(crit_sect_);
550 if (!running_) {
mikhal@webrtc.org759b0412013-05-07 16:36:00 +0000551 return false;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000552 }
agalusza@google.coma7e360e2013-08-01 03:15:08 +0000553 if (decode_error_mode_ == kNoErrors) {
mikhal@webrtc.orgdc3cd212013-04-25 20:27:04 +0000554 // No point to continue, as we are not decoding with errors.
mikhal@webrtc.org759b0412013-05-07 16:36:00 +0000555 return false;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000556 }
557
stefan@webrtc.orga64300a2013-03-04 15:24:40 +0000558 CleanUpOldOrEmptyFrames();
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000559
jbauchdb81ffd2015-11-23 03:59:02 -0800560 VCMFrameBuffer* oldest_frame;
agalusza@google.comd177c102013-08-08 01:12:33 +0000561 if (decodable_frames_.empty()) {
jbauchdb81ffd2015-11-23 03:59:02 -0800562 if (nack_mode_ != kNoNack || incomplete_frames_.size() <= 1) {
563 return false;
564 }
565 oldest_frame = incomplete_frames_.Front();
566 // Frame will only be removed from buffer if it is complete (or decodable).
567 if (oldest_frame->GetState() < kStateComplete) {
568 return false;
569 }
570 } else {
571 oldest_frame = decodable_frames_.Front();
572 // If we have exactly one frame in the buffer, release it only if it is
573 // complete. We know decodable_frames_ is not empty due to the previous
574 // check.
philipel9d3ab612015-12-21 04:12:39 -0800575 if (decodable_frames_.size() == 1 && incomplete_frames_.empty() &&
576 oldest_frame->GetState() != kStateComplete) {
jbauchdb81ffd2015-11-23 03:59:02 -0800577 return false;
578 }
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000579 }
mikhal@webrtc.orgd3cd5652013-05-03 17:54:18 +0000580
mikhal@webrtc.org759b0412013-05-07 16:36:00 +0000581 *timestamp = oldest_frame->TimeStamp();
582 return true;
583}
584
585VCMEncodedFrame* VCMJitterBuffer::ExtractAndSetDecode(uint32_t timestamp) {
mikhal@webrtc.org759b0412013-05-07 16:36:00 +0000586 CriticalSectionScoped cs(crit_sect_);
mikhal@webrtc.org759b0412013-05-07 16:36:00 +0000587 if (!running_) {
588 return NULL;
589 }
590 // Extract the frame with the desired timestamp.
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000591 VCMFrameBuffer* frame = decodable_frames_.PopFrame(timestamp);
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +0000592 bool continuous = true;
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000593 if (!frame) {
594 frame = incomplete_frames_.PopFrame(timestamp);
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +0000595 if (frame)
596 continuous = last_decoded_state_.ContinuousFrame(frame);
597 else
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000598 return NULL;
mikhal@webrtc.org759b0412013-05-07 16:36:00 +0000599 }
hclam@chromium.org1a7b9b92013-07-08 21:31:18 +0000600 TRACE_EVENT_ASYNC_STEP0("webrtc", "Video", timestamp, "Extract");
mikhal@webrtc.orgcb20a5b2013-05-15 17:10:44 +0000601 // Frame pulled out from jitter buffer, update the jitter estimate.
mikhal@webrtc.org759b0412013-05-07 16:36:00 +0000602 const bool retransmitted = (frame->GetNackCount() > 0);
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000603 if (retransmitted) {
604 jitter_estimate_.FrameNacked();
mikhal@webrtc.org759b0412013-05-07 16:36:00 +0000605 } else if (frame->Length() > 0) {
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000606 // Ignore retransmitted and empty frames.
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000607 if (waiting_for_completion_.latest_packet_time >= 0) {
608 UpdateJitterEstimate(waiting_for_completion_, true);
609 }
mikhal@webrtc.orgcb20a5b2013-05-15 17:10:44 +0000610 if (frame->GetState() == kStateComplete) {
611 UpdateJitterEstimate(*frame, false);
612 } else {
613 // Wait for this one to get complete.
614 waiting_for_completion_.frame_size = frame->Length();
philipel9d3ab612015-12-21 04:12:39 -0800615 waiting_for_completion_.latest_packet_time = frame->LatestPacketTimeMs();
mikhal@webrtc.orgcb20a5b2013-05-15 17:10:44 +0000616 waiting_for_completion_.timestamp = frame->TimeStamp();
617 }
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000618 }
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000619
620 // The state must be changed to decoding before cleaning up zero sized
621 // frames to avoid empty frames being cleaned up and then given to the
mikhal@webrtc.orgcb20a5b2013-05-15 17:10:44 +0000622 // decoder. Propagates the missing_frame bit.
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +0000623 frame->PrepareForDecode(continuous);
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000624
mikhal@webrtc.orgcb20a5b2013-05-15 17:10:44 +0000625 // We have a frame - update the last decoded state and nack list.
mikhal@webrtc.org759b0412013-05-07 16:36:00 +0000626 last_decoded_state_.SetState(frame);
stefan@webrtc.orga64300a2013-03-04 15:24:40 +0000627 DropPacketsFromNackList(last_decoded_state_.sequence_num());
agalusza@google.comd818dcb2013-07-29 21:48:11 +0000628
629 if ((*frame).IsSessionComplete())
630 UpdateAveragePacketsPerFrame(frame->NumPackets());
631
philipel83f831a2016-03-12 03:30:23 -0800632 if (nack_module_)
633 nack_module_->ClearUpTo(frame->GetHighSeqNum());
mikhal@webrtc.org759b0412013-05-07 16:36:00 +0000634 return frame;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000635}
636
637// Release frame when done with decoding. Should never be used to release
638// frames from within the jitter buffer.
639void VCMJitterBuffer::ReleaseFrame(VCMEncodedFrame* frame) {
640 CriticalSectionScoped cs(crit_sect_);
641 VCMFrameBuffer* frame_buffer = static_cast<VCMFrameBuffer*>(frame);
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +0000642 if (frame_buffer) {
643 free_frames_.push_back(frame_buffer);
644 }
stefan@webrtc.org791eec72011-10-11 07:53:43 +0000645}
646
niklase@google.com470e71d2011-07-07 08:21:25 +0000647// Gets frame to use for this timestamp. If no match, get empty frame.
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000648VCMFrameBufferEnum VCMJitterBuffer::GetFrame(const VCMPacket& packet,
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000649 VCMFrameBuffer** frame,
650 FrameList** frame_list) {
651 *frame = incomplete_frames_.PopFrame(packet.timestamp);
652 if (*frame != NULL) {
653 *frame_list = &incomplete_frames_;
654 return kNoError;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000655 }
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000656 *frame = decodable_frames_.PopFrame(packet.timestamp);
657 if (*frame != NULL) {
658 *frame_list = &decodable_frames_;
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000659 return kNoError;
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000660 }
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +0000661
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000662 *frame_list = NULL;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000663 // No match, return empty frame.
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000664 *frame = GetEmptyFrame();
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000665 if (*frame == NULL) {
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +0000666 // No free frame! Try to reclaim some...
stefan@webrtc.org34c5da62014-04-11 14:08:35 +0000667 LOG(LS_WARNING) << "Unable to get empty frame; Recycling.";
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +0000668 bool found_key_frame = RecycleFramesUntilKeyFrame();
669 *frame = GetEmptyFrame();
stefan@webrtc.org34c5da62014-04-11 14:08:35 +0000670 assert(*frame);
671 if (!found_key_frame) {
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000672 free_frames_.push_back(*frame);
673 return kFlushIndicator;
stefan@webrtc.org34c5da62014-04-11 14:08:35 +0000674 }
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000675 }
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +0000676 (*frame)->Reset();
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000677 return kNoError;
niklase@google.com470e71d2011-07-07 08:21:25 +0000678}
679
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000680int64_t VCMJitterBuffer::LastPacketTime(const VCMEncodedFrame* frame,
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000681 bool* retransmitted) const {
682 assert(retransmitted);
683 CriticalSectionScoped cs(crit_sect_);
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000684 const VCMFrameBuffer* frame_buffer =
685 static_cast<const VCMFrameBuffer*>(frame);
686 *retransmitted = (frame_buffer->GetNackCount() > 0);
687 return frame_buffer->LatestPacketTimeMs();
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000688}
689
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000690VCMFrameBufferEnum VCMJitterBuffer::InsertPacket(const VCMPacket& packet,
691 bool* retransmitted) {
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000692 CriticalSectionScoped cs(crit_sect_);
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +0000693
philipel83f831a2016-03-12 03:30:23 -0800694 if (nack_module_)
695 nack_module_->OnReceivedPacket(packet);
696
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000697 ++num_packets_;
asapersson@webrtc.org83b52002014-11-28 10:17:13 +0000698 if (num_packets_ == 1) {
699 time_first_packet_ms_ = clock_->TimeInMilliseconds();
700 }
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000701 // Does this packet belong to an old frame?
702 if (last_decoded_state_.IsOldPacket(&packet)) {
703 // Account only for media packets.
704 if (packet.sizeBytes > 0) {
705 num_discarded_packets_++;
706 num_consecutive_old_packets_++;
pbos@webrtc.org55707692014-12-19 15:45:03 +0000707 if (stats_callback_ != NULL)
708 stats_callback_->OnDiscardedPacketsUpdated(num_discarded_packets_);
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000709 }
710 // Update last decoded sequence number if the packet arrived late and
711 // belongs to a frame with a timestamp equal to the last decoded
712 // timestamp.
713 last_decoded_state_.UpdateOldPacket(&packet);
714 DropPacketsFromNackList(last_decoded_state_.sequence_num());
715
Noah Richardse4cb4e92015-05-22 14:03:00 -0700716 // Also see if this old packet made more incomplete frames continuous.
717 FindAndInsertContinuousFramesWithState(last_decoded_state_);
718
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000719 if (num_consecutive_old_packets_ > kMaxConsecutiveOldPackets) {
720 LOG(LS_WARNING)
721 << num_consecutive_old_packets_
722 << " consecutive old packets received. Flushing the jitter buffer.";
723 Flush();
724 return kFlushIndicator;
725 }
726 return kOldPacket;
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000727 }
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000728
asapersson9a4cd872015-10-23 00:27:14 -0700729 num_consecutive_old_packets_ = 0;
730
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000731 VCMFrameBuffer* frame;
732 FrameList* frame_list;
733 const VCMFrameBufferEnum error = GetFrame(packet, &frame, &frame_list);
734 if (error != kNoError)
735 return error;
736
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +0000737 int64_t now_ms = clock_->TimeInMilliseconds();
738 // We are keeping track of the first and latest seq numbers, and
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000739 // the number of wraps to be able to calculate how many packets we expect.
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000740 if (first_packet_since_reset_) {
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000741 // Now it's time to start estimating jitter
742 // reset the delay estimate.
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +0000743 inter_frame_delay_.Reset(now_ms);
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000744 }
745
746 // Empty packets may bias the jitter estimate (lacking size component),
747 // therefore don't let empty packet trigger the following updates:
pbos22993e12015-10-19 02:39:06 -0700748 if (packet.frameType != kEmptyFrame) {
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000749 if (waiting_for_completion_.timestamp == packet.timestamp) {
750 // This can get bad if we have a lot of duplicate packets,
751 // we will then count some packet multiple times.
752 waiting_for_completion_.frame_size += packet.sizeBytes;
753 waiting_for_completion_.latest_packet_time = now_ms;
754 } else if (waiting_for_completion_.latest_packet_time >= 0 &&
755 waiting_for_completion_.latest_packet_time + 2000 <= now_ms) {
756 // A packet should never be more than two seconds late
757 UpdateJitterEstimate(waiting_for_completion_, true);
758 waiting_for_completion_.latest_packet_time = -1;
759 waiting_for_completion_.frame_size = 0;
760 waiting_for_completion_.timestamp = 0;
761 }
762 }
763
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000764 VCMFrameBufferStateEnum previous_state = frame->GetState();
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000765 // Insert packet.
agalusza@google.comd818dcb2013-07-29 21:48:11 +0000766 FrameData frame_data;
767 frame_data.rtt_ms = rtt_ms_;
768 frame_data.rolling_average_packets_per_frame = average_packets_per_frame_;
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000769 VCMFrameBufferEnum buffer_state =
770 frame->InsertPacket(packet, now_ms, decode_error_mode_, frame_data);
771
772 if (previous_state != kStateComplete) {
philipel9d3ab612015-12-21 04:12:39 -0800773 TRACE_EVENT_ASYNC_BEGIN1("webrtc", "Video", frame->TimeStamp(), "timestamp",
774 frame->TimeStamp());
hclam@chromium.org1a7b9b92013-07-08 21:31:18 +0000775 }
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +0000776
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000777 if (buffer_state > 0) {
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000778 incoming_bit_count_ += packet.sizeBytes << 3;
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000779 if (first_packet_since_reset_) {
780 latest_received_sequence_number_ = packet.seqNum;
781 first_packet_since_reset_ = false;
782 } else {
783 if (IsPacketRetransmitted(packet)) {
784 frame->IncrementNackCount();
785 }
pbos@webrtc.orgebb467f2014-05-19 15:28:02 +0000786 if (!UpdateNackList(packet.seqNum) &&
787 packet.frameType != kVideoFrameKey) {
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000788 buffer_state = kFlushIndicator;
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000789 }
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000790
philipel9d3ab612015-12-21 04:12:39 -0800791 latest_received_sequence_number_ =
792 LatestSequenceNumber(latest_received_sequence_number_, packet.seqNum);
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000793 }
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000794 }
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +0000795
796 // Is the frame already in the decodable list?
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +0000797 bool continuous = IsContinuous(*frame);
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000798 switch (buffer_state) {
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000799 case kGeneralError:
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000800 case kTimeStampError:
801 case kSizeError: {
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000802 free_frames_.push_back(frame);
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000803 break;
804 }
805 case kCompleteSession: {
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000806 if (previous_state != kStateDecodable &&
807 previous_state != kStateComplete) {
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +0000808 CountFrame(*frame);
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +0000809 if (continuous) {
810 // Signal that we have a complete session.
811 frame_event_->Set();
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000812 }
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000813 }
kjellander@webrtc.org7d2b6a92015-01-28 18:37:58 +0000814 FALLTHROUGH();
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +0000815 }
816 // Note: There is no break here - continuing to kDecodableSession.
817 case kDecodableSession: {
818 *retransmitted = (frame->GetNackCount() > 0);
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +0000819 if (continuous) {
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +0000820 decodable_frames_.InsertFrame(frame);
821 FindAndInsertContinuousFrames(*frame);
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000822 } else {
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +0000823 incomplete_frames_.InsertFrame(frame);
jbauchdb81ffd2015-11-23 03:59:02 -0800824 // If NACKs are enabled, keyframes are triggered by |GetNackList|.
philipel9d3ab612015-12-21 04:12:39 -0800825 if (nack_mode_ == kNoNack &&
826 NonContinuousOrIncompleteDuration() >
827 90 * kMaxDiscontinuousFramesTime) {
jbauchdb81ffd2015-11-23 03:59:02 -0800828 return kFlushIndicator;
829 }
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +0000830 }
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000831 break;
832 }
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000833 case kIncomplete: {
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000834 if (frame->GetState() == kStateEmpty &&
835 last_decoded_state_.UpdateEmptyFrame(frame)) {
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +0000836 free_frames_.push_back(frame);
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +0000837 return kNoError;
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000838 } else {
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000839 incomplete_frames_.InsertFrame(frame);
jbauchdb81ffd2015-11-23 03:59:02 -0800840 // If NACKs are enabled, keyframes are triggered by |GetNackList|.
philipel9d3ab612015-12-21 04:12:39 -0800841 if (nack_mode_ == kNoNack &&
842 NonContinuousOrIncompleteDuration() >
843 90 * kMaxDiscontinuousFramesTime) {
jbauchdb81ffd2015-11-23 03:59:02 -0800844 return kFlushIndicator;
845 }
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000846 }
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000847 break;
848 }
849 case kNoError:
mikhal@webrtc.orgf31a47a2013-08-26 17:10:11 +0000850 case kOutOfBoundsPacket:
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000851 case kDuplicatePacket: {
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000852 // Put back the frame where it came from.
853 if (frame_list != NULL) {
854 frame_list->InsertFrame(frame);
855 } else {
856 free_frames_.push_back(frame);
857 }
asapersson@webrtc.org96dc6852014-11-03 14:40:38 +0000858 ++num_duplicated_packets_;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000859 break;
860 }
stefan@webrtc.org3417eb42013-05-21 15:25:53 +0000861 case kFlushIndicator:
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000862 free_frames_.push_back(frame);
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +0000863 return kFlushIndicator;
philipel9d3ab612015-12-21 04:12:39 -0800864 default:
865 assert(false);
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000866 }
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000867 return buffer_state;
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000868}
869
philipel9d3ab612015-12-21 04:12:39 -0800870bool VCMJitterBuffer::IsContinuousInState(
871 const VCMFrameBuffer& frame,
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000872 const VCMDecodingState& decoding_state) const {
agalusza@google.coma7e360e2013-08-01 03:15:08 +0000873 // Is this frame (complete or decodable) and continuous?
874 // kStateDecodable will never be set when decode_error_mode_ is false
875 // as SessionInfo determines this state based on the error mode (and frame
876 // completeness).
pbos@webrtc.org4f16c872014-11-24 09:06:48 +0000877 return (frame.GetState() == kStateComplete ||
878 frame.GetState() == kStateDecodable) &&
879 decoding_state.ContinuousFrame(&frame);
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000880}
881
882bool VCMJitterBuffer::IsContinuous(const VCMFrameBuffer& frame) const {
883 if (IsContinuousInState(frame, last_decoded_state_)) {
884 return true;
885 }
886 VCMDecodingState decoding_state;
887 decoding_state.CopyFrom(last_decoded_state_);
888 for (FrameList::const_iterator it = decodable_frames_.begin();
philipel9d3ab612015-12-21 04:12:39 -0800889 it != decodable_frames_.end(); ++it) {
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +0000890 VCMFrameBuffer* decodable_frame = it->second;
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000891 if (IsNewerTimestamp(decodable_frame->TimeStamp(), frame.TimeStamp())) {
892 break;
893 }
894 decoding_state.SetState(decodable_frame);
895 if (IsContinuousInState(frame, decoding_state)) {
896 return true;
897 }
898 }
899 return false;
900}
901
902void VCMJitterBuffer::FindAndInsertContinuousFrames(
903 const VCMFrameBuffer& new_frame) {
904 VCMDecodingState decoding_state;
905 decoding_state.CopyFrom(last_decoded_state_);
906 decoding_state.SetState(&new_frame);
Noah Richardse4cb4e92015-05-22 14:03:00 -0700907 FindAndInsertContinuousFramesWithState(decoding_state);
908}
909
910void VCMJitterBuffer::FindAndInsertContinuousFramesWithState(
911 const VCMDecodingState& original_decoded_state) {
912 // Copy original_decoded_state so we can move the state forward with each
913 // decodable frame we find.
914 VCMDecodingState decoding_state;
915 decoding_state.CopyFrom(original_decoded_state);
916
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000917 // When temporal layers are available, we search for a complete or decodable
918 // frame until we hit one of the following:
919 // 1. Continuous base or sync layer.
920 // 2. The end of the list was reached.
921 for (FrameList::iterator it = incomplete_frames_.begin();
philipel9d3ab612015-12-21 04:12:39 -0800922 it != incomplete_frames_.end();) {
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +0000923 VCMFrameBuffer* frame = it->second;
Noah Richardse4cb4e92015-05-22 14:03:00 -0700924 if (IsNewerTimestamp(original_decoded_state.time_stamp(),
925 frame->TimeStamp())) {
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000926 ++it;
927 continue;
928 }
929 if (IsContinuousInState(*frame, decoding_state)) {
930 decodable_frames_.InsertFrame(frame);
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +0000931 incomplete_frames_.erase(it++);
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +0000932 decoding_state.SetState(frame);
933 } else if (frame->TemporalId() <= 0) {
934 break;
935 } else {
936 ++it;
937 }
938 }
939}
940
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000941uint32_t VCMJitterBuffer::EstimatedJitterMs() {
942 CriticalSectionScoped cs(crit_sect_);
mikhal@webrtc.org119c67d2013-01-31 17:18:02 +0000943 // Compute RTT multiplier for estimation.
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000944 // low_rtt_nackThresholdMs_ == -1 means no FEC.
945 double rtt_mult = 1.0f;
stefan@webrtc.orga64300a2013-03-04 15:24:40 +0000946 if (low_rtt_nack_threshold_ms_ >= 0 &&
pkasting@chromium.org16825b12015-01-12 21:51:21 +0000947 rtt_ms_ >= low_rtt_nack_threshold_ms_) {
stefan@webrtc.orga64300a2013-03-04 15:24:40 +0000948 // For RTTs above low_rtt_nack_threshold_ms_ we don't apply extra delay
949 // when waiting for retransmissions.
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000950 rtt_mult = 0.0f;
951 }
mikhal@webrtc.org119c67d2013-01-31 17:18:02 +0000952 return jitter_estimate_.GetJitterEstimate(rtt_mult);
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000953}
954
pkasting@chromium.org16825b12015-01-12 21:51:21 +0000955void VCMJitterBuffer::UpdateRtt(int64_t rtt_ms) {
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000956 CriticalSectionScoped cs(crit_sect_);
957 rtt_ms_ = rtt_ms;
958 jitter_estimate_.UpdateRtt(rtt_ms);
philipel83f831a2016-03-12 03:30:23 -0800959 if (nack_module_)
960 nack_module_->UpdateRtt(rtt_ms);
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000961}
962
963void VCMJitterBuffer::SetNackMode(VCMNackMode mode,
pkasting@chromium.org16825b12015-01-12 21:51:21 +0000964 int64_t low_rtt_nack_threshold_ms,
965 int64_t high_rtt_nack_threshold_ms) {
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000966 CriticalSectionScoped cs(crit_sect_);
967 nack_mode_ = mode;
stefan@webrtc.orga64300a2013-03-04 15:24:40 +0000968 if (mode == kNoNack) {
969 missing_sequence_numbers_.clear();
970 }
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000971 assert(low_rtt_nack_threshold_ms >= -1 && high_rtt_nack_threshold_ms >= -1);
972 assert(high_rtt_nack_threshold_ms == -1 ||
973 low_rtt_nack_threshold_ms <= high_rtt_nack_threshold_ms);
974 assert(low_rtt_nack_threshold_ms > -1 || high_rtt_nack_threshold_ms == -1);
975 low_rtt_nack_threshold_ms_ = low_rtt_nack_threshold_ms;
976 high_rtt_nack_threshold_ms_ = high_rtt_nack_threshold_ms;
mflodman@webrtc.org7c894b72012-11-26 12:40:15 +0000977 // Don't set a high start rtt if high_rtt_nack_threshold_ms_ is used, to not
Wan-Teh Changf2912872015-06-05 13:16:45 -0700978 // disable NACK in |kNack| mode.
mflodman@webrtc.org7c894b72012-11-26 12:40:15 +0000979 if (rtt_ms_ == kDefaultRtt && high_rtt_nack_threshold_ms_ != -1) {
980 rtt_ms_ = 0;
981 }
stefan@webrtc.orga64300a2013-03-04 15:24:40 +0000982 if (!WaitForRetransmissions()) {
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000983 jitter_estimate_.ResetNackCount();
984 }
985}
986
stefan@webrtc.orgbecf9c82013-02-01 15:09:57 +0000987void VCMJitterBuffer::SetNackSettings(size_t max_nack_list_size,
stefan@webrtc.orgef144882013-05-07 19:16:33 +0000988 int max_packet_age_to_nack,
989 int max_incomplete_time_ms) {
stefan@webrtc.orgbecf9c82013-02-01 15:09:57 +0000990 CriticalSectionScoped cs(crit_sect_);
991 assert(max_packet_age_to_nack >= 0);
stefan@webrtc.orgef144882013-05-07 19:16:33 +0000992 assert(max_incomplete_time_ms_ >= 0);
stefan@webrtc.orgbecf9c82013-02-01 15:09:57 +0000993 max_nack_list_size_ = max_nack_list_size;
994 max_packet_age_to_nack_ = max_packet_age_to_nack;
stefan@webrtc.orgef144882013-05-07 19:16:33 +0000995 max_incomplete_time_ms_ = max_incomplete_time_ms;
stefan@webrtc.orgbecf9c82013-02-01 15:09:57 +0000996}
997
stefan@webrtc.org912981f2012-10-12 07:04:52 +0000998VCMNackMode VCMJitterBuffer::nack_mode() const {
999 CriticalSectionScoped cs(crit_sect_);
1000 return nack_mode_;
1001}
1002
stefan@webrtc.orgef144882013-05-07 19:16:33 +00001003int VCMJitterBuffer::NonContinuousOrIncompleteDuration() {
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001004 if (incomplete_frames_.empty()) {
stefan@webrtc.orgef144882013-05-07 19:16:33 +00001005 return 0;
1006 }
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +00001007 uint32_t start_timestamp = incomplete_frames_.Front()->TimeStamp();
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001008 if (!decodable_frames_.empty()) {
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +00001009 start_timestamp = decodable_frames_.Back()->TimeStamp();
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001010 }
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +00001011 return incomplete_frames_.Back()->TimeStamp() - start_timestamp;
stefan@webrtc.orgef144882013-05-07 19:16:33 +00001012}
1013
1014uint16_t VCMJitterBuffer::EstimatedLowSequenceNumber(
1015 const VCMFrameBuffer& frame) const {
1016 assert(frame.GetLowSeqNum() >= 0);
1017 if (frame.HaveFirstPacket())
1018 return frame.GetLowSeqNum();
hclam@chromium.orgfe307e12013-05-16 21:19:59 +00001019
1020 // This estimate is not accurate if more than one packet with lower sequence
1021 // number is lost.
stefan@webrtc.orgef144882013-05-07 19:16:33 +00001022 return frame.GetLowSeqNum() - 1;
1023}
1024
Wan-Teh Changb1825a42015-06-03 15:03:35 -07001025std::vector<uint16_t> VCMJitterBuffer::GetNackList(bool* request_key_frame) {
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001026 CriticalSectionScoped cs(crit_sect_);
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001027 *request_key_frame = false;
1028 if (nack_mode_ == kNoNack) {
Wan-Teh Changb1825a42015-06-03 15:03:35 -07001029 return std::vector<uint16_t>();
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001030 }
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001031 if (last_decoded_state_.in_initial_state()) {
pbos@webrtc.org4f16c872014-11-24 09:06:48 +00001032 VCMFrameBuffer* next_frame = NextFrame();
agalusza@google.comd177c102013-08-08 01:12:33 +00001033 const bool first_frame_is_key = next_frame &&
philipel9d3ab612015-12-21 04:12:39 -08001034 next_frame->FrameType() == kVideoFrameKey &&
1035 next_frame->HaveFirstPacket();
stefan@webrtc.org885cd132013-04-16 09:38:26 +00001036 if (!first_frame_is_key) {
philipel9d3ab612015-12-21 04:12:39 -08001037 bool have_non_empty_frame =
1038 decodable_frames_.end() != find_if(decodable_frames_.begin(),
1039 decodable_frames_.end(),
1040 HasNonEmptyState);
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001041 if (!have_non_empty_frame) {
philipel9d3ab612015-12-21 04:12:39 -08001042 have_non_empty_frame =
1043 incomplete_frames_.end() != find_if(incomplete_frames_.begin(),
1044 incomplete_frames_.end(),
1045 HasNonEmptyState);
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001046 }
stefan@webrtc.org885cd132013-04-16 09:38:26 +00001047 bool found_key_frame = RecycleFramesUntilKeyFrame();
1048 if (!found_key_frame) {
1049 *request_key_frame = have_non_empty_frame;
Wan-Teh Changb1825a42015-06-03 15:03:35 -07001050 return std::vector<uint16_t>();
stefan@webrtc.org885cd132013-04-16 09:38:26 +00001051 }
1052 }
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001053 }
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001054 if (TooLargeNackList()) {
1055 *request_key_frame = !HandleTooLargeNackList();
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001056 }
stefan@webrtc.orgef144882013-05-07 19:16:33 +00001057 if (max_incomplete_time_ms_ > 0) {
1058 int non_continuous_incomplete_duration =
1059 NonContinuousOrIncompleteDuration();
1060 if (non_continuous_incomplete_duration > 90 * max_incomplete_time_ms_) {
stefan@webrtc.org34c5da62014-04-11 14:08:35 +00001061 LOG_F(LS_WARNING) << "Too long non-decodable duration: "
1062 << non_continuous_incomplete_duration << " > "
1063 << 90 * max_incomplete_time_ms_;
philipel9d3ab612015-12-21 04:12:39 -08001064 FrameList::reverse_iterator rit = find_if(
1065 incomplete_frames_.rbegin(), incomplete_frames_.rend(), IsKeyFrame);
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001066 if (rit == incomplete_frames_.rend()) {
stefan@webrtc.orgef144882013-05-07 19:16:33 +00001067 // Request a key frame if we don't have one already.
1068 *request_key_frame = true;
Wan-Teh Changb1825a42015-06-03 15:03:35 -07001069 return std::vector<uint16_t>();
stefan@webrtc.orgef144882013-05-07 19:16:33 +00001070 } else {
1071 // Skip to the last key frame. If it's incomplete we will start
1072 // NACKing it.
hclam@chromium.orgfe307e12013-05-16 21:19:59 +00001073 // Note that the estimated low sequence number is correct for VP8
1074 // streams because only the first packet of a key frame is marked.
stefan@webrtc.orgef144882013-05-07 19:16:33 +00001075 last_decoded_state_.Reset();
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +00001076 DropPacketsFromNackList(EstimatedLowSequenceNumber(*rit->second));
stefan@webrtc.orgef144882013-05-07 19:16:33 +00001077 }
1078 }
1079 }
philipel83f831a2016-03-12 03:30:23 -08001080 // The experiment is running, the nack module will send Nacks.
1081 // The reason we allow the major part of the GetNackList function to
1082 // run even if we are running the experiment is because this
1083 // function is also used to signal that the jitter buffer wants to
1084 // request a keyframe.
1085 if (nack_module_)
1086 return std::vector<uint16_t>();
1087
Wan-Teh Changb1825a42015-06-03 15:03:35 -07001088 std::vector<uint16_t> nack_list(missing_sequence_numbers_.begin(),
1089 missing_sequence_numbers_.end());
1090 return nack_list;
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001091}
1092
mikhal@webrtc.orgdbf6a812013-08-21 20:40:47 +00001093void VCMJitterBuffer::SetDecodeErrorMode(VCMDecodeErrorMode error_mode) {
1094 CriticalSectionScoped cs(crit_sect_);
mikhal@webrtc.org3c5a9242013-09-03 20:45:36 +00001095 decode_error_mode_ = error_mode;
agalusza@google.comd177c102013-08-08 01:12:33 +00001096}
1097
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001098VCMFrameBuffer* VCMJitterBuffer::NextFrame() const {
1099 if (!decodable_frames_.empty())
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +00001100 return decodable_frames_.Front();
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001101 if (!incomplete_frames_.empty())
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +00001102 return incomplete_frames_.Front();
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001103 return NULL;
1104}
1105
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001106bool VCMJitterBuffer::UpdateNackList(uint16_t sequence_number) {
1107 if (nack_mode_ == kNoNack) {
1108 return true;
1109 }
stefan@webrtc.org4d2f5de2013-04-09 18:24:41 +00001110 // Make sure we don't add packets which are already too old to be decoded.
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001111 if (!last_decoded_state_.in_initial_state()) {
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001112 latest_received_sequence_number_ = LatestSequenceNumber(
philipel9d3ab612015-12-21 04:12:39 -08001113 latest_received_sequence_number_, last_decoded_state_.sequence_num());
stefan@webrtc.org4d2f5de2013-04-09 18:24:41 +00001114 }
stefan@webrtc.org7bc465b2013-04-11 17:48:02 +00001115 if (IsNewerSequenceNumber(sequence_number,
1116 latest_received_sequence_number_)) {
stefan@webrtc.org4d2f5de2013-04-09 18:24:41 +00001117 // Push any missing sequence numbers to the NACK list.
1118 for (uint16_t i = latest_received_sequence_number_ + 1;
stefan@webrtc.orga5dee332013-05-07 11:11:17 +00001119 IsNewerSequenceNumber(sequence_number, i); ++i) {
stefan@webrtc.org4d2f5de2013-04-09 18:24:41 +00001120 missing_sequence_numbers_.insert(missing_sequence_numbers_.end(), i);
sprang@webrtc.org0200f702015-02-16 12:06:00 +00001121 TRACE_EVENT_INSTANT1(TRACE_DISABLED_BY_DEFAULT("webrtc_rtp"), "AddNack",
1122 "seqnum", i);
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001123 }
stefan@webrtc.org4d2f5de2013-04-09 18:24:41 +00001124 if (TooLargeNackList() && !HandleTooLargeNackList()) {
stefan@webrtc.org34c5da62014-04-11 14:08:35 +00001125 LOG(LS_WARNING) << "Requesting key frame due to too large NACK list.";
stefan@webrtc.org4d2f5de2013-04-09 18:24:41 +00001126 return false;
1127 }
1128 if (MissingTooOldPacket(sequence_number) &&
1129 !HandleTooOldPackets(sequence_number)) {
stefan@webrtc.org34c5da62014-04-11 14:08:35 +00001130 LOG(LS_WARNING) << "Requesting key frame due to missing too old packets";
stefan@webrtc.org4d2f5de2013-04-09 18:24:41 +00001131 return false;
1132 }
1133 } else {
1134 missing_sequence_numbers_.erase(sequence_number);
sprang@webrtc.org0200f702015-02-16 12:06:00 +00001135 TRACE_EVENT_INSTANT1(TRACE_DISABLED_BY_DEFAULT("webrtc_rtp"), "RemoveNack",
1136 "seqnum", sequence_number);
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001137 }
1138 return true;
1139}
1140
1141bool VCMJitterBuffer::TooLargeNackList() const {
1142 return missing_sequence_numbers_.size() > max_nack_list_size_;
1143}
1144
1145bool VCMJitterBuffer::HandleTooLargeNackList() {
1146 // Recycle frames until the NACK list is small enough. It is likely cheaper to
1147 // request a key frame than to retransmit this many missing packets.
stefan@webrtc.org34c5da62014-04-11 14:08:35 +00001148 LOG_F(LS_WARNING) << "NACK list has grown too large: "
1149 << missing_sequence_numbers_.size() << " > "
1150 << max_nack_list_size_;
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001151 bool key_frame_found = false;
stefan@webrtc.org4d2f5de2013-04-09 18:24:41 +00001152 while (TooLargeNackList()) {
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001153 key_frame_found = RecycleFramesUntilKeyFrame();
1154 }
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001155 return key_frame_found;
1156}
1157
1158bool VCMJitterBuffer::MissingTooOldPacket(
1159 uint16_t latest_sequence_number) const {
1160 if (missing_sequence_numbers_.empty()) {
1161 return false;
1162 }
philipel9d3ab612015-12-21 04:12:39 -08001163 const uint16_t age_of_oldest_missing_packet =
1164 latest_sequence_number - *missing_sequence_numbers_.begin();
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001165 // Recycle frames if the NACK list contains too old sequence numbers as
1166 // the packets may have already been dropped by the sender.
1167 return age_of_oldest_missing_packet > max_packet_age_to_nack_;
1168}
1169
1170bool VCMJitterBuffer::HandleTooOldPackets(uint16_t latest_sequence_number) {
1171 bool key_frame_found = false;
philipel9d3ab612015-12-21 04:12:39 -08001172 const uint16_t age_of_oldest_missing_packet =
1173 latest_sequence_number - *missing_sequence_numbers_.begin();
stefan@webrtc.org34c5da62014-04-11 14:08:35 +00001174 LOG_F(LS_WARNING) << "NACK list contains too old sequence numbers: "
1175 << age_of_oldest_missing_packet << " > "
1176 << max_packet_age_to_nack_;
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001177 while (MissingTooOldPacket(latest_sequence_number)) {
1178 key_frame_found = RecycleFramesUntilKeyFrame();
1179 }
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001180 return key_frame_found;
1181}
1182
1183void VCMJitterBuffer::DropPacketsFromNackList(
1184 uint16_t last_decoded_sequence_number) {
1185 // Erase all sequence numbers from the NACK list which we won't need any
1186 // longer.
philipel9d3ab612015-12-21 04:12:39 -08001187 missing_sequence_numbers_.erase(
1188 missing_sequence_numbers_.begin(),
1189 missing_sequence_numbers_.upper_bound(last_decoded_sequence_number));
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001190}
1191
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001192int64_t VCMJitterBuffer::LastDecodedTimestamp() const {
1193 CriticalSectionScoped cs(crit_sect_);
1194 return last_decoded_state_.time_stamp();
1195}
1196
pbos@webrtc.org55707692014-12-19 15:45:03 +00001197void VCMJitterBuffer::RegisterStatsCallback(
1198 VCMReceiveStatisticsCallback* callback) {
pbos@webrtc.orgce4e9a32014-12-18 13:50:16 +00001199 CriticalSectionScoped cs(crit_sect_);
pbos@webrtc.org55707692014-12-19 15:45:03 +00001200 stats_callback_ = callback;
pbos@webrtc.orgce4e9a32014-12-18 13:50:16 +00001201}
1202
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001203VCMFrameBuffer* VCMJitterBuffer::GetEmptyFrame() {
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +00001204 if (free_frames_.empty()) {
1205 if (!TryToIncreaseJitterBufferSize()) {
1206 return NULL;
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001207 }
1208 }
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +00001209 VCMFrameBuffer* frame = free_frames_.front();
1210 free_frames_.pop_front();
1211 return frame;
1212}
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001213
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +00001214bool VCMJitterBuffer::TryToIncreaseJitterBufferSize() {
1215 if (max_number_of_frames_ >= kMaxNumberOfFrames)
1216 return false;
pbos@webrtc.org4f16c872014-11-24 09:06:48 +00001217 free_frames_.push_back(new VCMFrameBuffer());
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +00001218 ++max_number_of_frames_;
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +00001219 TRACE_COUNTER1("webrtc", "JBMaxFrames", max_number_of_frames_);
1220 return true;
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001221}
1222
1223// Recycle oldest frames up to a key frame, used if jitter buffer is completely
1224// full.
1225bool VCMJitterBuffer::RecycleFramesUntilKeyFrame() {
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001226 // First release incomplete frames, and only release decodable frames if there
1227 // are no incomplete ones.
1228 FrameList::iterator key_frame_it;
1229 bool key_frame_found = false;
1230 int dropped_frames = 0;
1231 dropped_frames += incomplete_frames_.RecycleFramesUntilKeyFrame(
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +00001232 &key_frame_it, &free_frames_);
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001233 key_frame_found = key_frame_it != incomplete_frames_.end();
1234 if (dropped_frames == 0) {
1235 dropped_frames += decodable_frames_.RecycleFramesUntilKeyFrame(
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +00001236 &key_frame_it, &free_frames_);
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001237 key_frame_found = key_frame_it != decodable_frames_.end();
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001238 }
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001239 TRACE_EVENT_INSTANT0("webrtc", "JB::RecycleFramesUntilKeyFrame");
1240 if (key_frame_found) {
stefan@webrtc.org34c5da62014-04-11 14:08:35 +00001241 LOG(LS_INFO) << "Found key frame while dropping frames.";
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001242 // Reset last decoded state to make sure the next frame decoded is a key
1243 // frame, and start NACKing from here.
1244 last_decoded_state_.Reset();
stefan@webrtc.org50fb4af2013-06-17 07:33:58 +00001245 DropPacketsFromNackList(EstimatedLowSequenceNumber(*key_frame_it->second));
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001246 } else if (decodable_frames_.empty()) {
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +00001247 // All frames dropped. Reset the decoding state and clear missing sequence
1248 // numbers as we're starting fresh.
1249 last_decoded_state_.Reset();
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001250 missing_sequence_numbers_.clear();
edjee@google.com79b02892013-04-04 19:43:34 +00001251 }
stefan@webrtc.orgc8b29a22013-06-17 07:13:16 +00001252 return key_frame_found;
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001253}
1254
1255// Must be called under the critical section |crit_sect_|.
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +00001256void VCMJitterBuffer::CountFrame(const VCMFrameBuffer& frame) {
pbos@webrtc.org4f16c872014-11-24 09:06:48 +00001257 incoming_frame_count_++;
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001258
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +00001259 if (frame.FrameType() == kVideoFrameKey) {
philipel9d3ab612015-12-21 04:12:39 -08001260 TRACE_EVENT_ASYNC_STEP0("webrtc", "Video", frame.TimeStamp(),
1261 "KeyComplete");
hclam@chromium.org806dc3b2013-04-09 19:54:10 +00001262 } else {
philipel9d3ab612015-12-21 04:12:39 -08001263 TRACE_EVENT_ASYNC_STEP0("webrtc", "Video", frame.TimeStamp(),
1264 "DeltaComplete");
hclam@chromium.org806dc3b2013-04-09 19:54:10 +00001265 }
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001266
1267 // Update receive statistics. We count all layers, thus when you use layers
1268 // adding all key and delta frames might differ from frame count.
stefan@webrtc.org4cf1a8a2013-06-27 15:20:14 +00001269 if (frame.IsSessionComplete()) {
pbos@webrtc.orgce4e9a32014-12-18 13:50:16 +00001270 if (frame.FrameType() == kVideoFrameKey) {
1271 ++receive_statistics_.key_frames;
skvlad98bb6642016-04-07 15:36:45 -07001272 if (receive_statistics_.key_frames == 1) {
1273 LOG(LS_INFO) << "Received first complete key frame";
1274 }
pbos@webrtc.orgce4e9a32014-12-18 13:50:16 +00001275 } else {
1276 ++receive_statistics_.delta_frames;
1277 }
skvlad98bb6642016-04-07 15:36:45 -07001278
pbos@webrtc.org55707692014-12-19 15:45:03 +00001279 if (stats_callback_ != NULL)
1280 stats_callback_->OnFrameCountsUpdated(receive_statistics_);
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001281 }
niklase@google.com470e71d2011-07-07 08:21:25 +00001282}
1283
agalusza@google.comd818dcb2013-07-29 21:48:11 +00001284void VCMJitterBuffer::UpdateAveragePacketsPerFrame(int current_number_packets) {
1285 if (frame_counter_ > kFastConvergeThreshold) {
philipel9d3ab612015-12-21 04:12:39 -08001286 average_packets_per_frame_ =
1287 average_packets_per_frame_ * (1 - kNormalConvergeMultiplier) +
1288 current_number_packets * kNormalConvergeMultiplier;
agalusza@google.comd818dcb2013-07-29 21:48:11 +00001289 } else if (frame_counter_ > 0) {
philipel9d3ab612015-12-21 04:12:39 -08001290 average_packets_per_frame_ =
1291 average_packets_per_frame_ * (1 - kFastConvergeMultiplier) +
1292 current_number_packets * kFastConvergeMultiplier;
agalusza@google.comd818dcb2013-07-29 21:48:11 +00001293 frame_counter_++;
1294 } else {
1295 average_packets_per_frame_ = current_number_packets;
1296 frame_counter_++;
1297 }
1298}
1299
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001300// Must be called under the critical section |crit_sect_|.
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001301void VCMJitterBuffer::CleanUpOldOrEmptyFrames() {
pbos@webrtc.org4f16c872014-11-24 09:06:48 +00001302 decodable_frames_.CleanUpOldOrEmptyFrames(&last_decoded_state_,
1303 &free_frames_);
1304 incomplete_frames_.CleanUpOldOrEmptyFrames(&last_decoded_state_,
1305 &free_frames_);
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001306 if (!last_decoded_state_.in_initial_state()) {
1307 DropPacketsFromNackList(last_decoded_state_.sequence_num());
1308 }
mikhal@webrtc.org832caca2011-12-13 21:15:05 +00001309}
1310
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001311// Must be called from within |crit_sect_|.
1312bool VCMJitterBuffer::IsPacketRetransmitted(const VCMPacket& packet) const {
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001313 return missing_sequence_numbers_.find(packet.seqNum) !=
philipel9d3ab612015-12-21 04:12:39 -08001314 missing_sequence_numbers_.end();
niklase@google.com470e71d2011-07-07 08:21:25 +00001315}
1316
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001317// Must be called under the critical section |crit_sect_|. Should never be
1318// called with retransmitted frames, they must be filtered out before this
1319// function is called.
1320void VCMJitterBuffer::UpdateJitterEstimate(const VCMJitterSample& sample,
1321 bool incomplete_frame) {
1322 if (sample.latest_packet_time == -1) {
1323 return;
1324 }
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001325 UpdateJitterEstimate(sample.latest_packet_time, sample.timestamp,
1326 sample.frame_size, incomplete_frame);
1327}
1328
1329// Must be called under the critical section crit_sect_. Should never be
1330// called with retransmitted frames, they must be filtered out before this
1331// function is called.
1332void VCMJitterBuffer::UpdateJitterEstimate(const VCMFrameBuffer& frame,
1333 bool incomplete_frame) {
1334 if (frame.LatestPacketTimeMs() == -1) {
1335 return;
1336 }
1337 // No retransmitted frames should be a part of the jitter
1338 // estimate.
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001339 UpdateJitterEstimate(frame.LatestPacketTimeMs(), frame.TimeStamp(),
1340 frame.Length(), incomplete_frame);
1341}
1342
1343// Must be called under the critical section |crit_sect_|. Should never be
1344// called with retransmitted frames, they must be filtered out before this
1345// function is called.
philipel9d3ab612015-12-21 04:12:39 -08001346void VCMJitterBuffer::UpdateJitterEstimate(int64_t latest_packet_time_ms,
1347 uint32_t timestamp,
1348 unsigned int frame_size,
1349 bool incomplete_frame) {
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001350 if (latest_packet_time_ms == -1) {
1351 return;
1352 }
1353 int64_t frame_delay;
philipel9d3ab612015-12-21 04:12:39 -08001354 bool not_reordered = inter_frame_delay_.CalculateDelay(
1355 timestamp, &frame_delay, latest_packet_time_ms);
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001356 // Filter out frames which have been reordered in time by the network
1357 if (not_reordered) {
1358 // Update the jitter estimate with the new samples
1359 jitter_estimate_.UpdateEstimate(frame_delay, frame_size, incomplete_frame);
1360 }
1361}
1362
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001363bool VCMJitterBuffer::WaitForRetransmissions() {
1364 if (nack_mode_ == kNoNack) {
1365 // NACK disabled -> don't wait for retransmissions.
1366 return false;
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001367 }
stefan@webrtc.orga64300a2013-03-04 15:24:40 +00001368 // Evaluate if the RTT is higher than |high_rtt_nack_threshold_ms_|, and in
1369 // that case we don't wait for retransmissions.
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001370 if (high_rtt_nack_threshold_ms_ >= 0 &&
pkasting@chromium.org16825b12015-01-12 21:51:21 +00001371 rtt_ms_ >= high_rtt_nack_threshold_ms_) {
stefan@webrtc.org912981f2012-10-12 07:04:52 +00001372 return false;
1373 }
1374 return true;
1375}
philipel83f831a2016-03-12 03:30:23 -08001376
1377int64_t VCMJitterBuffer::TimeUntilNextProcess() {
1378 if (nack_module_)
1379 return nack_module_->TimeUntilNextProcess();
1380 return std::numeric_limits<int64_t>::max();
1381}
1382
1383void VCMJitterBuffer::Process() {
1384 if (nack_module_)
1385 nack_module_->Process();
1386}
stefan@webrtc.org932ab182011-11-29 11:33:31 +00001387} // namespace webrtc