mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-17 15:47:53 +01:00
4529fbcfab
Also renaming it Vp8TemporalLayers to show that it is codec specific. Bug: webrtc:9012 Change-Id: I18187538b8142cdd7538f1a4ed1bada09d040f1f Reviewed-on: https://webrtc-review.googlesource.com/c/104643 Commit-Queue: Erik Språng <sprang@webrtc.org> Reviewed-by: Per Kjellander <perkj@webrtc.org> Reviewed-by: Sebastian Jansson <srte@webrtc.org> Reviewed-by: Niels Moller <nisse@webrtc.org> Cr-Commit-Position: refs/heads/master@{#25137}
123 lines
4.3 KiB
C++
123 lines
4.3 KiB
C++
/* Copyright (c) 2013 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.
|
|
*/
|
|
/*
|
|
* This file defines classes for doing temporal layers with VP8.
|
|
*/
|
|
#ifndef MODULES_VIDEO_CODING_CODECS_VP8_DEFAULT_TEMPORAL_LAYERS_H_
|
|
#define MODULES_VIDEO_CODING_CODECS_VP8_DEFAULT_TEMPORAL_LAYERS_H_
|
|
|
|
#include <limits>
|
|
#include <map>
|
|
#include <memory>
|
|
#include <set>
|
|
#include <vector>
|
|
|
|
#include "absl/types/optional.h"
|
|
|
|
#include "api/video_codecs/vp8_temporal_layers.h"
|
|
#include "modules/video_coding/codecs/vp8/include/temporal_layers_checker.h"
|
|
|
|
namespace webrtc {
|
|
|
|
class DefaultTemporalLayers : public Vp8TemporalLayers {
|
|
public:
|
|
explicit DefaultTemporalLayers(int number_of_temporal_layers);
|
|
~DefaultTemporalLayers() override;
|
|
|
|
bool SupportsEncoderFrameDropping() const override;
|
|
|
|
// Returns the recommended VP8 encode flags needed. May refresh the decoder
|
|
// and/or update the reference buffers.
|
|
Vp8TemporalLayers::FrameConfig UpdateLayerConfig(uint32_t timestamp) override;
|
|
|
|
// New target bitrate, per temporal layer.
|
|
void OnRatesUpdated(const std::vector<uint32_t>& bitrates_bps,
|
|
int framerate_fps) override;
|
|
|
|
bool UpdateConfiguration(Vp8EncoderConfig* cfg) override;
|
|
|
|
void OnEncodeDone(uint32_t rtp_timestamp,
|
|
size_t size_bytes,
|
|
bool is_keyframe,
|
|
int qp,
|
|
CodecSpecificInfoVP8* vp8_info) override;
|
|
|
|
private:
|
|
static constexpr size_t kKeyframeBuffer = std::numeric_limits<size_t>::max();
|
|
static std::vector<Vp8TemporalLayers::FrameConfig> GetTemporalPattern(
|
|
size_t num_layers);
|
|
bool IsSyncFrame(const FrameConfig& config) const;
|
|
void ValidateReferences(BufferFlags* flags, Vp8BufferReference ref) const;
|
|
void UpdateSearchOrder(FrameConfig* config);
|
|
|
|
const size_t num_layers_;
|
|
const std::vector<unsigned int> temporal_ids_;
|
|
const std::vector<Vp8TemporalLayers::FrameConfig> temporal_pattern_;
|
|
// Set of buffers that are never updated except by keyframes.
|
|
const std::set<Vp8BufferReference> kf_buffers_;
|
|
|
|
uint8_t pattern_idx_;
|
|
// Updated cumulative bitrates, per temporal layer.
|
|
absl::optional<std::vector<uint32_t>> new_bitrates_bps_;
|
|
|
|
struct PendingFrame {
|
|
PendingFrame();
|
|
PendingFrame(bool expired,
|
|
uint8_t updated_buffers_mask,
|
|
const FrameConfig& frame_config);
|
|
// Flag indicating if this frame has expired, ie it belongs to a previous
|
|
// iteration of the temporal pattern.
|
|
bool expired = false;
|
|
// Bitmask of Vp8BufferReference flags, indicating which buffers this frame
|
|
// updates.
|
|
uint8_t updated_buffer_mask = 0;
|
|
// The frame config return by UpdateLayerConfig() for this frame.
|
|
FrameConfig frame_config;
|
|
};
|
|
// Map from rtp timestamp to pending frame status. Reset on pattern loop.
|
|
std::map<uint32_t, PendingFrame> pending_frames_;
|
|
|
|
// One counter per Vp8BufferReference, indicating number of frames since last
|
|
// refresh. For non-base-layer frames (ie golden, altref buffers), this is
|
|
// reset when the pattern loops.
|
|
std::map<Vp8BufferReference, size_t> frames_since_buffer_refresh_;
|
|
|
|
// Optional utility used to verify reference validity.
|
|
std::unique_ptr<TemporalLayersChecker> checker_;
|
|
};
|
|
|
|
class DefaultTemporalLayersChecker : public TemporalLayersChecker {
|
|
public:
|
|
explicit DefaultTemporalLayersChecker(int number_of_temporal_layers);
|
|
~DefaultTemporalLayersChecker() override;
|
|
|
|
bool CheckTemporalConfig(
|
|
bool frame_is_keyframe,
|
|
const Vp8TemporalLayers::FrameConfig& frame_config) override;
|
|
|
|
private:
|
|
struct BufferState {
|
|
BufferState()
|
|
: is_updated_this_cycle(false), is_keyframe(true), pattern_idx(0) {}
|
|
|
|
bool is_updated_this_cycle;
|
|
bool is_keyframe;
|
|
uint8_t pattern_idx;
|
|
};
|
|
const size_t num_layers_;
|
|
std::vector<unsigned int> temporal_ids_;
|
|
const std::vector<std::set<uint8_t>> temporal_dependencies_;
|
|
BufferState last_;
|
|
BufferState arf_;
|
|
BufferState golden_;
|
|
uint8_t pattern_idx_;
|
|
};
|
|
|
|
} // namespace webrtc
|
|
#endif // MODULES_VIDEO_CODING_CODECS_VP8_DEFAULT_TEMPORAL_LAYERS_H_
|