mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-17 07:37:51 +01:00
f67bb271c2
Bug: webrtc:11999 Change-Id: Iacbb3e5d782987ee504b0fd1042a5e7fad2e2e50 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/186561 Reviewed-by: Philip Eliasson <philipel@webrtc.org> Commit-Queue: Danil Chapovalov <danilchap@webrtc.org> Cr-Commit-Position: refs/heads/master@{#32311}
70 lines
2.5 KiB
C++
70 lines
2.5 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.
|
|
*/
|
|
#ifndef MODULES_VIDEO_CODING_CODECS_AV1_SCALABILITY_STRUCTURE_FULL_SVC_H_
|
|
#define MODULES_VIDEO_CODING_CODECS_AV1_SCALABILITY_STRUCTURE_FULL_SVC_H_
|
|
|
|
#include <bitset>
|
|
#include <vector>
|
|
|
|
#include "absl/types/optional.h"
|
|
#include "api/transport/rtp/dependency_descriptor.h"
|
|
#include "common_video/generic_frame_descriptor/generic_frame_info.h"
|
|
#include "modules/video_coding/codecs/av1/scalable_video_controller.h"
|
|
|
|
namespace webrtc {
|
|
|
|
class ScalabilityStructureFullSvc : public ScalableVideoController {
|
|
public:
|
|
ScalabilityStructureFullSvc(int num_spatial_layers, int num_temporal_layers);
|
|
~ScalabilityStructureFullSvc() override;
|
|
|
|
StreamLayersConfig StreamConfig() const override;
|
|
|
|
std::vector<LayerFrameConfig> NextFrameConfig(bool restart) override;
|
|
absl::optional<GenericFrameInfo> OnEncodeDone(
|
|
LayerFrameConfig config) override;
|
|
void OnRatesUpdated(const VideoBitrateAllocation& bitrates) override;
|
|
|
|
private:
|
|
enum FramePattern {
|
|
kKey,
|
|
kDeltaT1,
|
|
kDeltaT0,
|
|
};
|
|
static constexpr int kMaxNumSpatialLayers = 3;
|
|
// TODO(bugs.webrtc.org/11999): Support up to 3 temporal layers.
|
|
static constexpr int kMaxNumTemporalLayers = 2;
|
|
|
|
// Index of the buffer to store last frame for layer (`sid`, `tid`)
|
|
int BufferIndex(int sid, int tid) const {
|
|
return tid * num_spatial_layers_ + sid;
|
|
}
|
|
bool DecodeTargetIsActive(int sid, int tid) const {
|
|
return active_decode_targets_[sid * num_temporal_layers_ + tid];
|
|
}
|
|
void SetDecodeTargetIsActive(int sid, int tid, bool value) {
|
|
active_decode_targets_.set(sid * num_temporal_layers_ + tid, value);
|
|
}
|
|
bool TemporalLayerIsActive(int tid) const;
|
|
static DecodeTargetIndication Dti(int sid,
|
|
int tid,
|
|
const LayerFrameConfig& frame);
|
|
|
|
const int num_spatial_layers_;
|
|
const int num_temporal_layers_;
|
|
|
|
FramePattern next_pattern_ = kKey;
|
|
std::bitset<kMaxNumSpatialLayers> can_depend_on_t0_frame_for_spatial_id_ = 0;
|
|
std::bitset<32> active_decode_targets_;
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // MODULES_VIDEO_CODING_CODECS_AV1_SCALABILITY_STRUCTURE_FULL_SVC_H_
|