webrtc/modules/video_coding/chain_diff_calculator.cc
Danil Chapovalov 61bc0d1ed3 Introduce ChainDiffCalculator
to convert flags which chains a video frame part of into chain_diffs

Bug: webrtc:10342
Change-Id: I6fb899eae934078223b101c9f85e2ac101980d4c
Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/175108
Commit-Queue: Danil Chapovalov <danilchap@webrtc.org>
Reviewed-by: Philip Eliasson <philipel@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#31306}
2020-05-18 14:22:44 +00:00

62 lines
1.9 KiB
C++

/*
* Copyright (c) 2020 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/chain_diff_calculator.h"
#include <stddef.h>
#include <stdint.h>
#include <algorithm>
#include <vector>
#include "absl/container/inlined_vector.h"
#include "absl/types/optional.h"
#include "rtc_base/logging.h"
namespace webrtc {
void ChainDiffCalculator::Reset(const std::vector<bool>& chains) {
last_frame_in_chain_.resize(chains.size());
for (size_t i = 0; i < chains.size(); ++i) {
if (chains[i]) {
last_frame_in_chain_[i] = absl::nullopt;
}
}
}
absl::InlinedVector<int, 4> ChainDiffCalculator::ChainDiffs(
int64_t frame_id) const {
absl::InlinedVector<int, 4> result;
result.reserve(last_frame_in_chain_.size());
for (const auto& frame_id_in_chain : last_frame_in_chain_) {
result.push_back(frame_id_in_chain ? (frame_id - *frame_id_in_chain) : 0);
}
return result;
}
absl::InlinedVector<int, 4> ChainDiffCalculator::From(
int64_t frame_id,
const std::vector<bool>& chains) {
auto result = ChainDiffs(frame_id);
if (chains.size() != last_frame_in_chain_.size()) {
RTC_LOG(LS_ERROR) << "Insconsistent chain configuration for frame#"
<< frame_id << ": expected "
<< last_frame_in_chain_.size() << " chains, found "
<< chains.size();
}
size_t num_chains = std::min(last_frame_in_chain_.size(), chains.size());
for (size_t i = 0; i < num_chains; ++i) {
if (chains[i]) {
last_frame_in_chain_[i] = frame_id;
}
}
return result;
}
} // namespace webrtc