mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-16 07:10:38 +01:00
411b49be17
FrameConfig is not specific to temporal layers. Anything that can control referenced/updated buffers could potentially use it. Bug: webrtc:10259 Change-Id: I04ed177ee884693798c3b69e35fd4255ce1e9062 Reviewed-on: https://webrtc-review.googlesource.com/c/120355 Reviewed-by: Erik Språng <sprang@webrtc.org> Commit-Queue: Elad Alon <eladalon@webrtc.org> Cr-Commit-Position: refs/heads/master@{#26448}
126 lines
4.5 KiB
C++
126 lines
4.5 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 <stddef.h>
|
|
#include <stdint.h>
|
|
#include <limits>
|
|
#include <map>
|
|
#include <memory>
|
|
#include <set>
|
|
#include <vector>
|
|
|
|
#include "absl/types/optional.h"
|
|
#include "api/video_codecs/vp8_frame_config.h"
|
|
#include "api/video_codecs/vp8_temporal_layers.h"
|
|
#include "modules/video_coding/codecs/vp8/include/temporal_layers_checker.h"
|
|
#include "modules/video_coding/include/video_codec_interface.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.
|
|
Vp8FrameConfig 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<Vp8FrameConfig> GetTemporalPattern(size_t num_layers);
|
|
bool IsSyncFrame(const Vp8FrameConfig& config) const;
|
|
void ValidateReferences(Vp8FrameConfig::BufferFlags* flags,
|
|
Vp8FrameConfig::Vp8BufferReference ref) const;
|
|
void UpdateSearchOrder(Vp8FrameConfig* config);
|
|
|
|
const size_t num_layers_;
|
|
const std::vector<unsigned int> temporal_ids_;
|
|
const std::vector<Vp8FrameConfig> temporal_pattern_;
|
|
// Set of buffers that are never updated except by keyframes.
|
|
const std::set<Vp8FrameConfig::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 Vp8FrameConfig& 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.
|
|
Vp8FrameConfig 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<Vp8FrameConfig::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 Vp8FrameConfig& 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_
|