mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-14 22:30:40 +01:00

This is a space efficient way to store more records about decoded frames, which is needed for long term references. Bug: webrtc:9710 Change-Id: I051d59d34a966d48db011142466d9cd15304b7d9 Reviewed-on: https://webrtc-review.googlesource.com/c/116792 Commit-Queue: Ilya Nikolaevskiy <ilnik@webrtc.org> Reviewed-by: Philip Eliasson <philipel@webrtc.org> Cr-Commit-Position: refs/heads/master@{#26240}
108 lines
3.7 KiB
C++
108 lines
3.7 KiB
C++
/*
|
|
* Copyright (c) 2019 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#include "modules/video_coding/utility/decoded_frames_history.h"
|
|
|
|
#include <algorithm>
|
|
|
|
#include "rtc_base/checks.h"
|
|
#include "rtc_base/logging.h"
|
|
|
|
namespace webrtc {
|
|
namespace video_coding {
|
|
|
|
DecodedFramesHistory::LayerHistory::LayerHistory() = default;
|
|
DecodedFramesHistory::LayerHistory::~LayerHistory() = default;
|
|
|
|
DecodedFramesHistory::DecodedFramesHistory(size_t window_size)
|
|
: window_size_(window_size) {}
|
|
|
|
DecodedFramesHistory::~DecodedFramesHistory() = default;
|
|
|
|
void DecodedFramesHistory::InsertDecoded(const VideoLayerFrameId& frameid,
|
|
uint32_t timestamp) {
|
|
last_decoded_frame_ = frameid;
|
|
last_decoded_frame_timestamp_ = timestamp;
|
|
if (static_cast<int>(layers_.size()) < frameid.spatial_layer + 1) {
|
|
layers_.resize(frameid.spatial_layer + 1);
|
|
layers_[frameid.spatial_layer].buffer.resize(window_size_);
|
|
layers_[frameid.spatial_layer].last_stored_index = frameid.picture_id;
|
|
layers_[frameid.spatial_layer].buffer[frameid.picture_id % window_size_] =
|
|
true;
|
|
return;
|
|
}
|
|
|
|
LayerHistory& history = layers_[frameid.spatial_layer];
|
|
|
|
RTC_DCHECK_LT(history.last_stored_index, frameid.picture_id);
|
|
|
|
int64_t id_jump = frameid.picture_id - history.last_stored_index;
|
|
int last_index = history.last_stored_index % window_size_;
|
|
int new_index = frameid.picture_id % window_size_;
|
|
|
|
// Need to clear indexes between last_index + 1 and new_index as they fall
|
|
// out of the history window and now represent new unseen picture ids.
|
|
if (id_jump >= window_size_) {
|
|
// Wraps around whole buffer - clear it all.
|
|
std::fill(history.buffer.begin(), history.buffer.end(), false);
|
|
} else if (new_index > last_index) {
|
|
std::fill(history.buffer.begin() + last_index + 1,
|
|
history.buffer.begin() + new_index, false);
|
|
} else {
|
|
std::fill(history.buffer.begin() + last_index + 1, history.buffer.end(),
|
|
false);
|
|
std::fill(history.buffer.begin(), history.buffer.begin() + new_index,
|
|
false);
|
|
}
|
|
|
|
history.buffer[new_index] = true;
|
|
history.last_stored_index = frameid.picture_id;
|
|
}
|
|
|
|
bool DecodedFramesHistory::WasDecoded(const VideoLayerFrameId& frameid) {
|
|
// Unseen before spatial layer.
|
|
if (static_cast<int>(layers_.size()) < frameid.spatial_layer + 1)
|
|
return false;
|
|
|
|
LayerHistory& history = layers_[frameid.spatial_layer];
|
|
|
|
// Reference to the picture_id out of the stored history should happen.
|
|
if (frameid.picture_id <= history.last_stored_index - window_size_) {
|
|
RTC_LOG(LS_WARNING) << "Referencing a frame out of the history window. "
|
|
"Assuming it was undecoded to avoid artifacts.";
|
|
return false;
|
|
}
|
|
|
|
if (frameid.picture_id > history.last_stored_index)
|
|
return false;
|
|
|
|
return history.buffer[frameid.picture_id % window_size_];
|
|
}
|
|
|
|
void DecodedFramesHistory::Clear() {
|
|
for (LayerHistory& layer : layers_) {
|
|
std::fill(layer.buffer.begin(), layer.buffer.end(), false);
|
|
layer.last_stored_index = 0;
|
|
}
|
|
last_decoded_frame_timestamp_.reset();
|
|
last_decoded_frame_.reset();
|
|
}
|
|
|
|
absl::optional<VideoLayerFrameId>
|
|
DecodedFramesHistory::GetLastDecodedFrameId() {
|
|
return last_decoded_frame_;
|
|
}
|
|
|
|
absl::optional<uint32_t> DecodedFramesHistory::GetLastDecodedFrameTimestamp() {
|
|
return last_decoded_frame_timestamp_;
|
|
}
|
|
|
|
} // namespace video_coding
|
|
} // namespace webrtc
|