1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150
|
/*
* 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 API_TRANSPORT_RTP_DEPENDENCY_DESCRIPTOR_H_
#define API_TRANSPORT_RTP_DEPENDENCY_DESCRIPTOR_H_
#include <stdint.h>
#include <initializer_list>
#include <memory>
#include <optional>
#include <vector>
#include "absl/container/inlined_vector.h"
#include "absl/strings/string_view.h"
#include "api/video/render_resolution.h"
namespace webrtc {
// Structures to build and parse dependency descriptor as described in
// https://aomediacodec.github.io/av1-rtp-spec/#dependency-descriptor-rtp-header-extension
// Relationship of a frame to a Decode target.
enum class DecodeTargetIndication {
kNotPresent = 0, // DecodeTargetInfo symbol '-'
kDiscardable = 1, // DecodeTargetInfo symbol 'D'
kSwitch = 2, // DecodeTargetInfo symbol 'S'
kRequired = 3 // DecodeTargetInfo symbol 'R'
};
struct FrameDependencyTemplate {
// Setters are named briefly to chain them when building the template.
FrameDependencyTemplate& S(int spatial_layer);
FrameDependencyTemplate& T(int temporal_layer);
FrameDependencyTemplate& Dtis(absl::string_view dtis);
FrameDependencyTemplate& FrameDiffs(std::initializer_list<int> diffs);
FrameDependencyTemplate& ChainDiffs(std::initializer_list<int> diffs);
friend bool operator==(const FrameDependencyTemplate& lhs,
const FrameDependencyTemplate& rhs) {
return lhs.spatial_id == rhs.spatial_id &&
lhs.temporal_id == rhs.temporal_id &&
lhs.decode_target_indications == rhs.decode_target_indications &&
lhs.frame_diffs == rhs.frame_diffs &&
lhs.chain_diffs == rhs.chain_diffs;
}
int spatial_id = 0;
int temporal_id = 0;
absl::InlinedVector<DecodeTargetIndication, 10> decode_target_indications;
absl::InlinedVector<int, 4> frame_diffs;
absl::InlinedVector<int, 4> chain_diffs;
};
struct FrameDependencyStructure {
friend bool operator==(const FrameDependencyStructure& lhs,
const FrameDependencyStructure& rhs) {
return lhs.num_decode_targets == rhs.num_decode_targets &&
lhs.num_chains == rhs.num_chains &&
lhs.decode_target_protected_by_chain ==
rhs.decode_target_protected_by_chain &&
lhs.resolutions == rhs.resolutions && lhs.templates == rhs.templates;
}
int structure_id = 0;
int num_decode_targets = 0;
int num_chains = 0;
// If chains are used (num_chains > 0), maps decode target index into index of
// the chain protecting that target.
absl::InlinedVector<int, 10> decode_target_protected_by_chain;
absl::InlinedVector<RenderResolution, 4> resolutions;
std::vector<FrameDependencyTemplate> templates;
};
class DependencyDescriptorMandatory {
public:
void set_frame_number(int frame_number) { frame_number_ = frame_number; }
int frame_number() const { return frame_number_; }
void set_template_id(int template_id) { template_id_ = template_id; }
int template_id() const { return template_id_; }
void set_first_packet_in_frame(bool first) { first_packet_in_frame_ = first; }
bool first_packet_in_frame() const { return first_packet_in_frame_; }
void set_last_packet_in_frame(bool last) { last_packet_in_frame_ = last; }
bool last_packet_in_frame() const { return last_packet_in_frame_; }
private:
int frame_number_;
int template_id_;
bool first_packet_in_frame_;
bool last_packet_in_frame_;
};
struct DependencyDescriptor {
static constexpr int kMaxSpatialIds = 4;
static constexpr int kMaxTemporalIds = 8;
static constexpr int kMaxDecodeTargets = 32;
static constexpr int kMaxTemplates = 64;
bool first_packet_in_frame = true;
bool last_packet_in_frame = true;
int frame_number = 0;
FrameDependencyTemplate frame_dependencies;
std::optional<RenderResolution> resolution;
std::optional<uint32_t> active_decode_targets_bitmask;
std::unique_ptr<FrameDependencyStructure> attached_structure;
};
// Below are implementation details.
namespace webrtc_impl {
absl::InlinedVector<DecodeTargetIndication, 10> StringToDecodeTargetIndications(
absl::string_view indication_symbols);
} // namespace webrtc_impl
inline FrameDependencyTemplate& FrameDependencyTemplate::S(int spatial_layer) {
this->spatial_id = spatial_layer;
return *this;
}
inline FrameDependencyTemplate& FrameDependencyTemplate::T(int temporal_layer) {
this->temporal_id = temporal_layer;
return *this;
}
inline FrameDependencyTemplate& FrameDependencyTemplate::Dtis(
absl::string_view dtis) {
this->decode_target_indications =
webrtc_impl::StringToDecodeTargetIndications(dtis);
return *this;
}
inline FrameDependencyTemplate& FrameDependencyTemplate::FrameDiffs(
std::initializer_list<int> diffs) {
this->frame_diffs.assign(diffs.begin(), diffs.end());
return *this;
}
inline FrameDependencyTemplate& FrameDependencyTemplate::ChainDiffs(
std::initializer_list<int> diffs) {
this->chain_diffs.assign(diffs.begin(), diffs.end());
return *this;
}
} // namespace webrtc
#endif // API_TRANSPORT_RTP_DEPENDENCY_DESCRIPTOR_H_
|