1*d9f75844SAndroid Build Coastguard Worker /*
2*d9f75844SAndroid Build Coastguard Worker * Copyright (c) 2018 The WebRTC project authors. All Rights Reserved.
3*d9f75844SAndroid Build Coastguard Worker *
4*d9f75844SAndroid Build Coastguard Worker * Use of this source code is governed by a BSD-style license
5*d9f75844SAndroid Build Coastguard Worker * that can be found in the LICENSE file in the root of the source
6*d9f75844SAndroid Build Coastguard Worker * tree. An additional intellectual property rights grant can be found
7*d9f75844SAndroid Build Coastguard Worker * in the file PATENTS. All contributing project authors may
8*d9f75844SAndroid Build Coastguard Worker * be found in the AUTHORS file in the root of the source tree.
9*d9f75844SAndroid Build Coastguard Worker */
10*d9f75844SAndroid Build Coastguard Worker
11*d9f75844SAndroid Build Coastguard Worker #include "call/rtp_payload_params.h"
12*d9f75844SAndroid Build Coastguard Worker
13*d9f75844SAndroid Build Coastguard Worker #include <stddef.h>
14*d9f75844SAndroid Build Coastguard Worker
15*d9f75844SAndroid Build Coastguard Worker #include <algorithm>
16*d9f75844SAndroid Build Coastguard Worker
17*d9f75844SAndroid Build Coastguard Worker #include "absl/container/inlined_vector.h"
18*d9f75844SAndroid Build Coastguard Worker #include "absl/strings/match.h"
19*d9f75844SAndroid Build Coastguard Worker #include "absl/types/variant.h"
20*d9f75844SAndroid Build Coastguard Worker #include "api/video/video_timing.h"
21*d9f75844SAndroid Build Coastguard Worker #include "modules/video_coding/codecs/h264/include/h264_globals.h"
22*d9f75844SAndroid Build Coastguard Worker #include "modules/video_coding/codecs/interface/common_constants.h"
23*d9f75844SAndroid Build Coastguard Worker #include "modules/video_coding/codecs/vp8/include/vp8_globals.h"
24*d9f75844SAndroid Build Coastguard Worker #include "modules/video_coding/codecs/vp9/include/vp9_globals.h"
25*d9f75844SAndroid Build Coastguard Worker #include "modules/video_coding/frame_dependencies_calculator.h"
26*d9f75844SAndroid Build Coastguard Worker #include "rtc_base/arraysize.h"
27*d9f75844SAndroid Build Coastguard Worker #include "rtc_base/checks.h"
28*d9f75844SAndroid Build Coastguard Worker #include "rtc_base/logging.h"
29*d9f75844SAndroid Build Coastguard Worker #include "rtc_base/random.h"
30*d9f75844SAndroid Build Coastguard Worker #include "rtc_base/time_utils.h"
31*d9f75844SAndroid Build Coastguard Worker
32*d9f75844SAndroid Build Coastguard Worker namespace webrtc {
33*d9f75844SAndroid Build Coastguard Worker namespace {
34*d9f75844SAndroid Build Coastguard Worker
35*d9f75844SAndroid Build Coastguard Worker constexpr int kMaxSimulatedSpatialLayers = 3;
36*d9f75844SAndroid Build Coastguard Worker
PopulateRtpWithCodecSpecifics(const CodecSpecificInfo & info,absl::optional<int> spatial_index,RTPVideoHeader * rtp)37*d9f75844SAndroid Build Coastguard Worker void PopulateRtpWithCodecSpecifics(const CodecSpecificInfo& info,
38*d9f75844SAndroid Build Coastguard Worker absl::optional<int> spatial_index,
39*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader* rtp) {
40*d9f75844SAndroid Build Coastguard Worker rtp->codec = info.codecType;
41*d9f75844SAndroid Build Coastguard Worker rtp->is_last_frame_in_picture = info.end_of_picture;
42*d9f75844SAndroid Build Coastguard Worker switch (info.codecType) {
43*d9f75844SAndroid Build Coastguard Worker case kVideoCodecVP8: {
44*d9f75844SAndroid Build Coastguard Worker auto& vp8_header = rtp->video_type_header.emplace<RTPVideoHeaderVP8>();
45*d9f75844SAndroid Build Coastguard Worker vp8_header.InitRTPVideoHeaderVP8();
46*d9f75844SAndroid Build Coastguard Worker vp8_header.nonReference = info.codecSpecific.VP8.nonReference;
47*d9f75844SAndroid Build Coastguard Worker vp8_header.temporalIdx = info.codecSpecific.VP8.temporalIdx;
48*d9f75844SAndroid Build Coastguard Worker vp8_header.layerSync = info.codecSpecific.VP8.layerSync;
49*d9f75844SAndroid Build Coastguard Worker vp8_header.keyIdx = info.codecSpecific.VP8.keyIdx;
50*d9f75844SAndroid Build Coastguard Worker rtp->simulcastIdx = spatial_index.value_or(0);
51*d9f75844SAndroid Build Coastguard Worker return;
52*d9f75844SAndroid Build Coastguard Worker }
53*d9f75844SAndroid Build Coastguard Worker case kVideoCodecVP9: {
54*d9f75844SAndroid Build Coastguard Worker auto& vp9_header = rtp->video_type_header.emplace<RTPVideoHeaderVP9>();
55*d9f75844SAndroid Build Coastguard Worker vp9_header.InitRTPVideoHeaderVP9();
56*d9f75844SAndroid Build Coastguard Worker vp9_header.inter_pic_predicted =
57*d9f75844SAndroid Build Coastguard Worker info.codecSpecific.VP9.inter_pic_predicted;
58*d9f75844SAndroid Build Coastguard Worker vp9_header.flexible_mode = info.codecSpecific.VP9.flexible_mode;
59*d9f75844SAndroid Build Coastguard Worker vp9_header.ss_data_available = info.codecSpecific.VP9.ss_data_available;
60*d9f75844SAndroid Build Coastguard Worker vp9_header.non_ref_for_inter_layer_pred =
61*d9f75844SAndroid Build Coastguard Worker info.codecSpecific.VP9.non_ref_for_inter_layer_pred;
62*d9f75844SAndroid Build Coastguard Worker vp9_header.temporal_idx = info.codecSpecific.VP9.temporal_idx;
63*d9f75844SAndroid Build Coastguard Worker vp9_header.temporal_up_switch = info.codecSpecific.VP9.temporal_up_switch;
64*d9f75844SAndroid Build Coastguard Worker vp9_header.inter_layer_predicted =
65*d9f75844SAndroid Build Coastguard Worker info.codecSpecific.VP9.inter_layer_predicted;
66*d9f75844SAndroid Build Coastguard Worker vp9_header.gof_idx = info.codecSpecific.VP9.gof_idx;
67*d9f75844SAndroid Build Coastguard Worker vp9_header.num_spatial_layers = info.codecSpecific.VP9.num_spatial_layers;
68*d9f75844SAndroid Build Coastguard Worker vp9_header.first_active_layer = info.codecSpecific.VP9.first_active_layer;
69*d9f75844SAndroid Build Coastguard Worker if (vp9_header.num_spatial_layers > 1) {
70*d9f75844SAndroid Build Coastguard Worker vp9_header.spatial_idx = spatial_index.value_or(kNoSpatialIdx);
71*d9f75844SAndroid Build Coastguard Worker } else {
72*d9f75844SAndroid Build Coastguard Worker vp9_header.spatial_idx = kNoSpatialIdx;
73*d9f75844SAndroid Build Coastguard Worker }
74*d9f75844SAndroid Build Coastguard Worker if (info.codecSpecific.VP9.ss_data_available) {
75*d9f75844SAndroid Build Coastguard Worker vp9_header.spatial_layer_resolution_present =
76*d9f75844SAndroid Build Coastguard Worker info.codecSpecific.VP9.spatial_layer_resolution_present;
77*d9f75844SAndroid Build Coastguard Worker if (info.codecSpecific.VP9.spatial_layer_resolution_present) {
78*d9f75844SAndroid Build Coastguard Worker for (size_t i = 0; i < info.codecSpecific.VP9.num_spatial_layers;
79*d9f75844SAndroid Build Coastguard Worker ++i) {
80*d9f75844SAndroid Build Coastguard Worker vp9_header.width[i] = info.codecSpecific.VP9.width[i];
81*d9f75844SAndroid Build Coastguard Worker vp9_header.height[i] = info.codecSpecific.VP9.height[i];
82*d9f75844SAndroid Build Coastguard Worker }
83*d9f75844SAndroid Build Coastguard Worker }
84*d9f75844SAndroid Build Coastguard Worker vp9_header.gof.CopyGofInfoVP9(info.codecSpecific.VP9.gof);
85*d9f75844SAndroid Build Coastguard Worker }
86*d9f75844SAndroid Build Coastguard Worker
87*d9f75844SAndroid Build Coastguard Worker vp9_header.num_ref_pics = info.codecSpecific.VP9.num_ref_pics;
88*d9f75844SAndroid Build Coastguard Worker for (int i = 0; i < info.codecSpecific.VP9.num_ref_pics; ++i) {
89*d9f75844SAndroid Build Coastguard Worker vp9_header.pid_diff[i] = info.codecSpecific.VP9.p_diff[i];
90*d9f75844SAndroid Build Coastguard Worker }
91*d9f75844SAndroid Build Coastguard Worker vp9_header.end_of_picture = info.end_of_picture;
92*d9f75844SAndroid Build Coastguard Worker return;
93*d9f75844SAndroid Build Coastguard Worker }
94*d9f75844SAndroid Build Coastguard Worker case kVideoCodecH264: {
95*d9f75844SAndroid Build Coastguard Worker auto& h264_header = rtp->video_type_header.emplace<RTPVideoHeaderH264>();
96*d9f75844SAndroid Build Coastguard Worker h264_header.packetization_mode =
97*d9f75844SAndroid Build Coastguard Worker info.codecSpecific.H264.packetization_mode;
98*d9f75844SAndroid Build Coastguard Worker rtp->simulcastIdx = spatial_index.value_or(0);
99*d9f75844SAndroid Build Coastguard Worker return;
100*d9f75844SAndroid Build Coastguard Worker }
101*d9f75844SAndroid Build Coastguard Worker case kVideoCodecMultiplex:
102*d9f75844SAndroid Build Coastguard Worker case kVideoCodecGeneric:
103*d9f75844SAndroid Build Coastguard Worker rtp->codec = kVideoCodecGeneric;
104*d9f75844SAndroid Build Coastguard Worker rtp->simulcastIdx = spatial_index.value_or(0);
105*d9f75844SAndroid Build Coastguard Worker return;
106*d9f75844SAndroid Build Coastguard Worker default:
107*d9f75844SAndroid Build Coastguard Worker return;
108*d9f75844SAndroid Build Coastguard Worker }
109*d9f75844SAndroid Build Coastguard Worker }
110*d9f75844SAndroid Build Coastguard Worker
SetVideoTiming(const EncodedImage & image,VideoSendTiming * timing)111*d9f75844SAndroid Build Coastguard Worker void SetVideoTiming(const EncodedImage& image, VideoSendTiming* timing) {
112*d9f75844SAndroid Build Coastguard Worker if (image.timing_.flags == VideoSendTiming::TimingFrameFlags::kInvalid ||
113*d9f75844SAndroid Build Coastguard Worker image.timing_.flags == VideoSendTiming::TimingFrameFlags::kNotTriggered) {
114*d9f75844SAndroid Build Coastguard Worker timing->flags = VideoSendTiming::TimingFrameFlags::kInvalid;
115*d9f75844SAndroid Build Coastguard Worker return;
116*d9f75844SAndroid Build Coastguard Worker }
117*d9f75844SAndroid Build Coastguard Worker
118*d9f75844SAndroid Build Coastguard Worker timing->encode_start_delta_ms = VideoSendTiming::GetDeltaCappedMs(
119*d9f75844SAndroid Build Coastguard Worker image.capture_time_ms_, image.timing_.encode_start_ms);
120*d9f75844SAndroid Build Coastguard Worker timing->encode_finish_delta_ms = VideoSendTiming::GetDeltaCappedMs(
121*d9f75844SAndroid Build Coastguard Worker image.capture_time_ms_, image.timing_.encode_finish_ms);
122*d9f75844SAndroid Build Coastguard Worker timing->packetization_finish_delta_ms = 0;
123*d9f75844SAndroid Build Coastguard Worker timing->pacer_exit_delta_ms = 0;
124*d9f75844SAndroid Build Coastguard Worker timing->network_timestamp_delta_ms = 0;
125*d9f75844SAndroid Build Coastguard Worker timing->network2_timestamp_delta_ms = 0;
126*d9f75844SAndroid Build Coastguard Worker timing->flags = image.timing_.flags;
127*d9f75844SAndroid Build Coastguard Worker }
128*d9f75844SAndroid Build Coastguard Worker
129*d9f75844SAndroid Build Coastguard Worker // Returns structure that aligns with simulated generic info. The templates
130*d9f75844SAndroid Build Coastguard Worker // allow to produce valid dependency descriptor for any stream where
131*d9f75844SAndroid Build Coastguard Worker // `num_spatial_layers` * `num_temporal_layers` <= 32 (limited by
132*d9f75844SAndroid Build Coastguard Worker // https://aomediacodec.github.io/av1-rtp-spec/#a82-syntax, see
133*d9f75844SAndroid Build Coastguard Worker // template_fdiffs()). The set of the templates is not tuned for any paricular
134*d9f75844SAndroid Build Coastguard Worker // structure thus dependency descriptor would use more bytes on the wire than
135*d9f75844SAndroid Build Coastguard Worker // with tuned templates.
MinimalisticStructure(int num_spatial_layers,int num_temporal_layers)136*d9f75844SAndroid Build Coastguard Worker FrameDependencyStructure MinimalisticStructure(int num_spatial_layers,
137*d9f75844SAndroid Build Coastguard Worker int num_temporal_layers) {
138*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LE(num_spatial_layers, DependencyDescriptor::kMaxSpatialIds);
139*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LE(num_temporal_layers, DependencyDescriptor::kMaxTemporalIds);
140*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LE(num_spatial_layers * num_temporal_layers, 32);
141*d9f75844SAndroid Build Coastguard Worker FrameDependencyStructure structure;
142*d9f75844SAndroid Build Coastguard Worker structure.num_decode_targets = num_spatial_layers * num_temporal_layers;
143*d9f75844SAndroid Build Coastguard Worker structure.num_chains = num_spatial_layers;
144*d9f75844SAndroid Build Coastguard Worker structure.templates.reserve(num_spatial_layers * num_temporal_layers);
145*d9f75844SAndroid Build Coastguard Worker for (int sid = 0; sid < num_spatial_layers; ++sid) {
146*d9f75844SAndroid Build Coastguard Worker for (int tid = 0; tid < num_temporal_layers; ++tid) {
147*d9f75844SAndroid Build Coastguard Worker FrameDependencyTemplate a_template;
148*d9f75844SAndroid Build Coastguard Worker a_template.spatial_id = sid;
149*d9f75844SAndroid Build Coastguard Worker a_template.temporal_id = tid;
150*d9f75844SAndroid Build Coastguard Worker for (int s = 0; s < num_spatial_layers; ++s) {
151*d9f75844SAndroid Build Coastguard Worker for (int t = 0; t < num_temporal_layers; ++t) {
152*d9f75844SAndroid Build Coastguard Worker // Prefer kSwitch indication for frames that is part of the decode
153*d9f75844SAndroid Build Coastguard Worker // target because dependency descriptor information generated in this
154*d9f75844SAndroid Build Coastguard Worker // class use kSwitch indications more often that kRequired, increasing
155*d9f75844SAndroid Build Coastguard Worker // the chance of a good (or complete) template match.
156*d9f75844SAndroid Build Coastguard Worker a_template.decode_target_indications.push_back(
157*d9f75844SAndroid Build Coastguard Worker sid <= s && tid <= t ? DecodeTargetIndication::kSwitch
158*d9f75844SAndroid Build Coastguard Worker : DecodeTargetIndication::kNotPresent);
159*d9f75844SAndroid Build Coastguard Worker }
160*d9f75844SAndroid Build Coastguard Worker }
161*d9f75844SAndroid Build Coastguard Worker a_template.frame_diffs.push_back(tid == 0 ? num_spatial_layers *
162*d9f75844SAndroid Build Coastguard Worker num_temporal_layers
163*d9f75844SAndroid Build Coastguard Worker : num_spatial_layers);
164*d9f75844SAndroid Build Coastguard Worker a_template.chain_diffs.assign(structure.num_chains, 1);
165*d9f75844SAndroid Build Coastguard Worker structure.templates.push_back(a_template);
166*d9f75844SAndroid Build Coastguard Worker
167*d9f75844SAndroid Build Coastguard Worker structure.decode_target_protected_by_chain.push_back(sid);
168*d9f75844SAndroid Build Coastguard Worker }
169*d9f75844SAndroid Build Coastguard Worker }
170*d9f75844SAndroid Build Coastguard Worker return structure;
171*d9f75844SAndroid Build Coastguard Worker }
172*d9f75844SAndroid Build Coastguard Worker } // namespace
173*d9f75844SAndroid Build Coastguard Worker
RtpPayloadParams(const uint32_t ssrc,const RtpPayloadState * state,const FieldTrialsView & trials)174*d9f75844SAndroid Build Coastguard Worker RtpPayloadParams::RtpPayloadParams(const uint32_t ssrc,
175*d9f75844SAndroid Build Coastguard Worker const RtpPayloadState* state,
176*d9f75844SAndroid Build Coastguard Worker const FieldTrialsView& trials)
177*d9f75844SAndroid Build Coastguard Worker : ssrc_(ssrc),
178*d9f75844SAndroid Build Coastguard Worker generic_picture_id_experiment_(
179*d9f75844SAndroid Build Coastguard Worker absl::StartsWith(trials.Lookup("WebRTC-GenericPictureId"),
180*d9f75844SAndroid Build Coastguard Worker "Enabled")),
181*d9f75844SAndroid Build Coastguard Worker simulate_generic_structure_(absl::StartsWith(
182*d9f75844SAndroid Build Coastguard Worker trials.Lookup("WebRTC-GenericCodecDependencyDescriptor"),
183*d9f75844SAndroid Build Coastguard Worker "Enabled")) {
184*d9f75844SAndroid Build Coastguard Worker for (auto& spatial_layer : last_shared_frame_id_)
185*d9f75844SAndroid Build Coastguard Worker spatial_layer.fill(-1);
186*d9f75844SAndroid Build Coastguard Worker
187*d9f75844SAndroid Build Coastguard Worker chain_last_frame_id_.fill(-1);
188*d9f75844SAndroid Build Coastguard Worker buffer_id_to_frame_id_.fill(-1);
189*d9f75844SAndroid Build Coastguard Worker
190*d9f75844SAndroid Build Coastguard Worker Random random(rtc::TimeMicros());
191*d9f75844SAndroid Build Coastguard Worker state_.picture_id =
192*d9f75844SAndroid Build Coastguard Worker state ? state->picture_id : (random.Rand<int16_t>() & 0x7FFF);
193*d9f75844SAndroid Build Coastguard Worker state_.tl0_pic_idx = state ? state->tl0_pic_idx : (random.Rand<uint8_t>());
194*d9f75844SAndroid Build Coastguard Worker }
195*d9f75844SAndroid Build Coastguard Worker
196*d9f75844SAndroid Build Coastguard Worker RtpPayloadParams::RtpPayloadParams(const RtpPayloadParams& other) = default;
197*d9f75844SAndroid Build Coastguard Worker
~RtpPayloadParams()198*d9f75844SAndroid Build Coastguard Worker RtpPayloadParams::~RtpPayloadParams() {}
199*d9f75844SAndroid Build Coastguard Worker
GetRtpVideoHeader(const EncodedImage & image,const CodecSpecificInfo * codec_specific_info,int64_t shared_frame_id)200*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader RtpPayloadParams::GetRtpVideoHeader(
201*d9f75844SAndroid Build Coastguard Worker const EncodedImage& image,
202*d9f75844SAndroid Build Coastguard Worker const CodecSpecificInfo* codec_specific_info,
203*d9f75844SAndroid Build Coastguard Worker int64_t shared_frame_id) {
204*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader rtp_video_header;
205*d9f75844SAndroid Build Coastguard Worker if (codec_specific_info) {
206*d9f75844SAndroid Build Coastguard Worker PopulateRtpWithCodecSpecifics(*codec_specific_info, image.SpatialIndex(),
207*d9f75844SAndroid Build Coastguard Worker &rtp_video_header);
208*d9f75844SAndroid Build Coastguard Worker }
209*d9f75844SAndroid Build Coastguard Worker rtp_video_header.frame_type = image._frameType;
210*d9f75844SAndroid Build Coastguard Worker rtp_video_header.rotation = image.rotation_;
211*d9f75844SAndroid Build Coastguard Worker rtp_video_header.content_type = image.content_type_;
212*d9f75844SAndroid Build Coastguard Worker rtp_video_header.playout_delay = image.playout_delay_;
213*d9f75844SAndroid Build Coastguard Worker rtp_video_header.width = image._encodedWidth;
214*d9f75844SAndroid Build Coastguard Worker rtp_video_header.height = image._encodedHeight;
215*d9f75844SAndroid Build Coastguard Worker rtp_video_header.color_space = image.ColorSpace()
216*d9f75844SAndroid Build Coastguard Worker ? absl::make_optional(*image.ColorSpace())
217*d9f75844SAndroid Build Coastguard Worker : absl::nullopt;
218*d9f75844SAndroid Build Coastguard Worker rtp_video_header.video_frame_tracking_id = image.VideoFrameTrackingId();
219*d9f75844SAndroid Build Coastguard Worker SetVideoTiming(image, &rtp_video_header.video_timing);
220*d9f75844SAndroid Build Coastguard Worker
221*d9f75844SAndroid Build Coastguard Worker const bool is_keyframe = image._frameType == VideoFrameType::kVideoFrameKey;
222*d9f75844SAndroid Build Coastguard Worker const bool first_frame_in_picture =
223*d9f75844SAndroid Build Coastguard Worker (codec_specific_info && codec_specific_info->codecType == kVideoCodecVP9)
224*d9f75844SAndroid Build Coastguard Worker ? codec_specific_info->codecSpecific.VP9.first_frame_in_picture
225*d9f75844SAndroid Build Coastguard Worker : true;
226*d9f75844SAndroid Build Coastguard Worker
227*d9f75844SAndroid Build Coastguard Worker SetCodecSpecific(&rtp_video_header, first_frame_in_picture);
228*d9f75844SAndroid Build Coastguard Worker
229*d9f75844SAndroid Build Coastguard Worker SetGeneric(codec_specific_info, shared_frame_id, is_keyframe,
230*d9f75844SAndroid Build Coastguard Worker &rtp_video_header);
231*d9f75844SAndroid Build Coastguard Worker
232*d9f75844SAndroid Build Coastguard Worker return rtp_video_header;
233*d9f75844SAndroid Build Coastguard Worker }
234*d9f75844SAndroid Build Coastguard Worker
ssrc() const235*d9f75844SAndroid Build Coastguard Worker uint32_t RtpPayloadParams::ssrc() const {
236*d9f75844SAndroid Build Coastguard Worker return ssrc_;
237*d9f75844SAndroid Build Coastguard Worker }
238*d9f75844SAndroid Build Coastguard Worker
state() const239*d9f75844SAndroid Build Coastguard Worker RtpPayloadState RtpPayloadParams::state() const {
240*d9f75844SAndroid Build Coastguard Worker return state_;
241*d9f75844SAndroid Build Coastguard Worker }
242*d9f75844SAndroid Build Coastguard Worker
SetCodecSpecific(RTPVideoHeader * rtp_video_header,bool first_frame_in_picture)243*d9f75844SAndroid Build Coastguard Worker void RtpPayloadParams::SetCodecSpecific(RTPVideoHeader* rtp_video_header,
244*d9f75844SAndroid Build Coastguard Worker bool first_frame_in_picture) {
245*d9f75844SAndroid Build Coastguard Worker // Always set picture id. Set tl0_pic_idx iff temporal index is set.
246*d9f75844SAndroid Build Coastguard Worker if (first_frame_in_picture) {
247*d9f75844SAndroid Build Coastguard Worker state_.picture_id = (static_cast<uint16_t>(state_.picture_id) + 1) & 0x7FFF;
248*d9f75844SAndroid Build Coastguard Worker }
249*d9f75844SAndroid Build Coastguard Worker if (rtp_video_header->codec == kVideoCodecVP8) {
250*d9f75844SAndroid Build Coastguard Worker auto& vp8_header =
251*d9f75844SAndroid Build Coastguard Worker absl::get<RTPVideoHeaderVP8>(rtp_video_header->video_type_header);
252*d9f75844SAndroid Build Coastguard Worker vp8_header.pictureId = state_.picture_id;
253*d9f75844SAndroid Build Coastguard Worker
254*d9f75844SAndroid Build Coastguard Worker if (vp8_header.temporalIdx != kNoTemporalIdx) {
255*d9f75844SAndroid Build Coastguard Worker if (vp8_header.temporalIdx == 0) {
256*d9f75844SAndroid Build Coastguard Worker ++state_.tl0_pic_idx;
257*d9f75844SAndroid Build Coastguard Worker }
258*d9f75844SAndroid Build Coastguard Worker vp8_header.tl0PicIdx = state_.tl0_pic_idx;
259*d9f75844SAndroid Build Coastguard Worker }
260*d9f75844SAndroid Build Coastguard Worker }
261*d9f75844SAndroid Build Coastguard Worker if (rtp_video_header->codec == kVideoCodecVP9) {
262*d9f75844SAndroid Build Coastguard Worker auto& vp9_header =
263*d9f75844SAndroid Build Coastguard Worker absl::get<RTPVideoHeaderVP9>(rtp_video_header->video_type_header);
264*d9f75844SAndroid Build Coastguard Worker vp9_header.picture_id = state_.picture_id;
265*d9f75844SAndroid Build Coastguard Worker
266*d9f75844SAndroid Build Coastguard Worker // Note that in the case that we have no temporal layers but we do have
267*d9f75844SAndroid Build Coastguard Worker // spatial layers, packets will carry layering info with a temporal_idx of
268*d9f75844SAndroid Build Coastguard Worker // zero, and we then have to set and increment tl0_pic_idx.
269*d9f75844SAndroid Build Coastguard Worker if (vp9_header.temporal_idx != kNoTemporalIdx ||
270*d9f75844SAndroid Build Coastguard Worker vp9_header.spatial_idx != kNoSpatialIdx) {
271*d9f75844SAndroid Build Coastguard Worker if (first_frame_in_picture &&
272*d9f75844SAndroid Build Coastguard Worker (vp9_header.temporal_idx == 0 ||
273*d9f75844SAndroid Build Coastguard Worker vp9_header.temporal_idx == kNoTemporalIdx)) {
274*d9f75844SAndroid Build Coastguard Worker ++state_.tl0_pic_idx;
275*d9f75844SAndroid Build Coastguard Worker }
276*d9f75844SAndroid Build Coastguard Worker vp9_header.tl0_pic_idx = state_.tl0_pic_idx;
277*d9f75844SAndroid Build Coastguard Worker }
278*d9f75844SAndroid Build Coastguard Worker }
279*d9f75844SAndroid Build Coastguard Worker if (generic_picture_id_experiment_ &&
280*d9f75844SAndroid Build Coastguard Worker rtp_video_header->codec == kVideoCodecGeneric) {
281*d9f75844SAndroid Build Coastguard Worker rtp_video_header->video_type_header.emplace<RTPVideoHeaderLegacyGeneric>()
282*d9f75844SAndroid Build Coastguard Worker .picture_id = state_.picture_id;
283*d9f75844SAndroid Build Coastguard Worker }
284*d9f75844SAndroid Build Coastguard Worker }
285*d9f75844SAndroid Build Coastguard Worker
286*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader::GenericDescriptorInfo
GenericDescriptorFromFrameInfo(const GenericFrameInfo & frame_info,int64_t frame_id)287*d9f75844SAndroid Build Coastguard Worker RtpPayloadParams::GenericDescriptorFromFrameInfo(
288*d9f75844SAndroid Build Coastguard Worker const GenericFrameInfo& frame_info,
289*d9f75844SAndroid Build Coastguard Worker int64_t frame_id) {
290*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader::GenericDescriptorInfo generic;
291*d9f75844SAndroid Build Coastguard Worker generic.frame_id = frame_id;
292*d9f75844SAndroid Build Coastguard Worker generic.dependencies = dependencies_calculator_.FromBuffersUsage(
293*d9f75844SAndroid Build Coastguard Worker frame_id, frame_info.encoder_buffers);
294*d9f75844SAndroid Build Coastguard Worker generic.chain_diffs =
295*d9f75844SAndroid Build Coastguard Worker chains_calculator_.From(frame_id, frame_info.part_of_chain);
296*d9f75844SAndroid Build Coastguard Worker generic.spatial_index = frame_info.spatial_id;
297*d9f75844SAndroid Build Coastguard Worker generic.temporal_index = frame_info.temporal_id;
298*d9f75844SAndroid Build Coastguard Worker generic.decode_target_indications = frame_info.decode_target_indications;
299*d9f75844SAndroid Build Coastguard Worker generic.active_decode_targets = frame_info.active_decode_targets;
300*d9f75844SAndroid Build Coastguard Worker return generic;
301*d9f75844SAndroid Build Coastguard Worker }
302*d9f75844SAndroid Build Coastguard Worker
SetGeneric(const CodecSpecificInfo * codec_specific_info,int64_t frame_id,bool is_keyframe,RTPVideoHeader * rtp_video_header)303*d9f75844SAndroid Build Coastguard Worker void RtpPayloadParams::SetGeneric(const CodecSpecificInfo* codec_specific_info,
304*d9f75844SAndroid Build Coastguard Worker int64_t frame_id,
305*d9f75844SAndroid Build Coastguard Worker bool is_keyframe,
306*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader* rtp_video_header) {
307*d9f75844SAndroid Build Coastguard Worker if (codec_specific_info && codec_specific_info->generic_frame_info &&
308*d9f75844SAndroid Build Coastguard Worker !codec_specific_info->generic_frame_info->encoder_buffers.empty()) {
309*d9f75844SAndroid Build Coastguard Worker if (is_keyframe) {
310*d9f75844SAndroid Build Coastguard Worker // Key frame resets all chains it is in.
311*d9f75844SAndroid Build Coastguard Worker chains_calculator_.Reset(
312*d9f75844SAndroid Build Coastguard Worker codec_specific_info->generic_frame_info->part_of_chain);
313*d9f75844SAndroid Build Coastguard Worker }
314*d9f75844SAndroid Build Coastguard Worker rtp_video_header->generic = GenericDescriptorFromFrameInfo(
315*d9f75844SAndroid Build Coastguard Worker *codec_specific_info->generic_frame_info, frame_id);
316*d9f75844SAndroid Build Coastguard Worker return;
317*d9f75844SAndroid Build Coastguard Worker }
318*d9f75844SAndroid Build Coastguard Worker
319*d9f75844SAndroid Build Coastguard Worker switch (rtp_video_header->codec) {
320*d9f75844SAndroid Build Coastguard Worker case VideoCodecType::kVideoCodecGeneric:
321*d9f75844SAndroid Build Coastguard Worker GenericToGeneric(frame_id, is_keyframe, rtp_video_header);
322*d9f75844SAndroid Build Coastguard Worker return;
323*d9f75844SAndroid Build Coastguard Worker case VideoCodecType::kVideoCodecVP8:
324*d9f75844SAndroid Build Coastguard Worker if (codec_specific_info) {
325*d9f75844SAndroid Build Coastguard Worker Vp8ToGeneric(codec_specific_info->codecSpecific.VP8, frame_id,
326*d9f75844SAndroid Build Coastguard Worker is_keyframe, rtp_video_header);
327*d9f75844SAndroid Build Coastguard Worker }
328*d9f75844SAndroid Build Coastguard Worker return;
329*d9f75844SAndroid Build Coastguard Worker case VideoCodecType::kVideoCodecVP9:
330*d9f75844SAndroid Build Coastguard Worker if (codec_specific_info != nullptr) {
331*d9f75844SAndroid Build Coastguard Worker Vp9ToGeneric(codec_specific_info->codecSpecific.VP9, frame_id,
332*d9f75844SAndroid Build Coastguard Worker *rtp_video_header);
333*d9f75844SAndroid Build Coastguard Worker }
334*d9f75844SAndroid Build Coastguard Worker return;
335*d9f75844SAndroid Build Coastguard Worker case VideoCodecType::kVideoCodecAV1:
336*d9f75844SAndroid Build Coastguard Worker // TODO(philipel): Implement AV1 to generic descriptor.
337*d9f75844SAndroid Build Coastguard Worker return;
338*d9f75844SAndroid Build Coastguard Worker case VideoCodecType::kVideoCodecH264:
339*d9f75844SAndroid Build Coastguard Worker if (codec_specific_info) {
340*d9f75844SAndroid Build Coastguard Worker H264ToGeneric(codec_specific_info->codecSpecific.H264, frame_id,
341*d9f75844SAndroid Build Coastguard Worker is_keyframe, rtp_video_header);
342*d9f75844SAndroid Build Coastguard Worker }
343*d9f75844SAndroid Build Coastguard Worker return;
344*d9f75844SAndroid Build Coastguard Worker case VideoCodecType::kVideoCodecMultiplex:
345*d9f75844SAndroid Build Coastguard Worker return;
346*d9f75844SAndroid Build Coastguard Worker }
347*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_NOTREACHED() << "Unsupported codec.";
348*d9f75844SAndroid Build Coastguard Worker }
349*d9f75844SAndroid Build Coastguard Worker
GenericStructure(const CodecSpecificInfo * codec_specific_info)350*d9f75844SAndroid Build Coastguard Worker absl::optional<FrameDependencyStructure> RtpPayloadParams::GenericStructure(
351*d9f75844SAndroid Build Coastguard Worker const CodecSpecificInfo* codec_specific_info) {
352*d9f75844SAndroid Build Coastguard Worker if (codec_specific_info == nullptr) {
353*d9f75844SAndroid Build Coastguard Worker return absl::nullopt;
354*d9f75844SAndroid Build Coastguard Worker }
355*d9f75844SAndroid Build Coastguard Worker // This helper shouldn't be used when template structure is specified
356*d9f75844SAndroid Build Coastguard Worker // explicetly.
357*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(!codec_specific_info->template_structure.has_value());
358*d9f75844SAndroid Build Coastguard Worker switch (codec_specific_info->codecType) {
359*d9f75844SAndroid Build Coastguard Worker case VideoCodecType::kVideoCodecGeneric:
360*d9f75844SAndroid Build Coastguard Worker if (simulate_generic_structure_) {
361*d9f75844SAndroid Build Coastguard Worker return MinimalisticStructure(/*num_spatial_layers=*/1,
362*d9f75844SAndroid Build Coastguard Worker /*num_temporal_layer=*/1);
363*d9f75844SAndroid Build Coastguard Worker }
364*d9f75844SAndroid Build Coastguard Worker return absl::nullopt;
365*d9f75844SAndroid Build Coastguard Worker case VideoCodecType::kVideoCodecVP8:
366*d9f75844SAndroid Build Coastguard Worker return MinimalisticStructure(/*num_spatial_layers=*/1,
367*d9f75844SAndroid Build Coastguard Worker /*num_temporal_layer=*/kMaxTemporalStreams);
368*d9f75844SAndroid Build Coastguard Worker case VideoCodecType::kVideoCodecVP9: {
369*d9f75844SAndroid Build Coastguard Worker absl::optional<FrameDependencyStructure> structure =
370*d9f75844SAndroid Build Coastguard Worker MinimalisticStructure(
371*d9f75844SAndroid Build Coastguard Worker /*num_spatial_layers=*/kMaxSimulatedSpatialLayers,
372*d9f75844SAndroid Build Coastguard Worker /*num_temporal_layer=*/kMaxTemporalStreams);
373*d9f75844SAndroid Build Coastguard Worker const CodecSpecificInfoVP9& vp9 = codec_specific_info->codecSpecific.VP9;
374*d9f75844SAndroid Build Coastguard Worker if (vp9.ss_data_available && vp9.spatial_layer_resolution_present) {
375*d9f75844SAndroid Build Coastguard Worker RenderResolution first_valid;
376*d9f75844SAndroid Build Coastguard Worker RenderResolution last_valid;
377*d9f75844SAndroid Build Coastguard Worker for (size_t i = 0; i < vp9.num_spatial_layers; ++i) {
378*d9f75844SAndroid Build Coastguard Worker RenderResolution r(vp9.width[i], vp9.height[i]);
379*d9f75844SAndroid Build Coastguard Worker if (r.Valid()) {
380*d9f75844SAndroid Build Coastguard Worker if (!first_valid.Valid()) {
381*d9f75844SAndroid Build Coastguard Worker first_valid = r;
382*d9f75844SAndroid Build Coastguard Worker }
383*d9f75844SAndroid Build Coastguard Worker last_valid = r;
384*d9f75844SAndroid Build Coastguard Worker }
385*d9f75844SAndroid Build Coastguard Worker structure->resolutions.push_back(r);
386*d9f75844SAndroid Build Coastguard Worker }
387*d9f75844SAndroid Build Coastguard Worker if (!last_valid.Valid()) {
388*d9f75844SAndroid Build Coastguard Worker // No valid resolution found. Do not send resolutions.
389*d9f75844SAndroid Build Coastguard Worker structure->resolutions.clear();
390*d9f75844SAndroid Build Coastguard Worker } else {
391*d9f75844SAndroid Build Coastguard Worker structure->resolutions.resize(kMaxSimulatedSpatialLayers, last_valid);
392*d9f75844SAndroid Build Coastguard Worker // VP9 encoder wrapper may disable first few spatial layers by
393*d9f75844SAndroid Build Coastguard Worker // setting invalid resolution (0,0). `structure->resolutions`
394*d9f75844SAndroid Build Coastguard Worker // doesn't support invalid resolution, so reset them to something
395*d9f75844SAndroid Build Coastguard Worker // valid.
396*d9f75844SAndroid Build Coastguard Worker for (RenderResolution& r : structure->resolutions) {
397*d9f75844SAndroid Build Coastguard Worker if (!r.Valid()) {
398*d9f75844SAndroid Build Coastguard Worker r = first_valid;
399*d9f75844SAndroid Build Coastguard Worker }
400*d9f75844SAndroid Build Coastguard Worker }
401*d9f75844SAndroid Build Coastguard Worker }
402*d9f75844SAndroid Build Coastguard Worker }
403*d9f75844SAndroid Build Coastguard Worker return structure;
404*d9f75844SAndroid Build Coastguard Worker }
405*d9f75844SAndroid Build Coastguard Worker case VideoCodecType::kVideoCodecAV1:
406*d9f75844SAndroid Build Coastguard Worker case VideoCodecType::kVideoCodecH264:
407*d9f75844SAndroid Build Coastguard Worker case VideoCodecType::kVideoCodecMultiplex:
408*d9f75844SAndroid Build Coastguard Worker return absl::nullopt;
409*d9f75844SAndroid Build Coastguard Worker }
410*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_NOTREACHED() << "Unsupported codec.";
411*d9f75844SAndroid Build Coastguard Worker }
412*d9f75844SAndroid Build Coastguard Worker
GenericToGeneric(int64_t shared_frame_id,bool is_keyframe,RTPVideoHeader * rtp_video_header)413*d9f75844SAndroid Build Coastguard Worker void RtpPayloadParams::GenericToGeneric(int64_t shared_frame_id,
414*d9f75844SAndroid Build Coastguard Worker bool is_keyframe,
415*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader* rtp_video_header) {
416*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader::GenericDescriptorInfo& generic =
417*d9f75844SAndroid Build Coastguard Worker rtp_video_header->generic.emplace();
418*d9f75844SAndroid Build Coastguard Worker
419*d9f75844SAndroid Build Coastguard Worker generic.frame_id = shared_frame_id;
420*d9f75844SAndroid Build Coastguard Worker generic.decode_target_indications.push_back(DecodeTargetIndication::kSwitch);
421*d9f75844SAndroid Build Coastguard Worker
422*d9f75844SAndroid Build Coastguard Worker if (is_keyframe) {
423*d9f75844SAndroid Build Coastguard Worker generic.chain_diffs.push_back(0);
424*d9f75844SAndroid Build Coastguard Worker last_shared_frame_id_[0].fill(-1);
425*d9f75844SAndroid Build Coastguard Worker } else {
426*d9f75844SAndroid Build Coastguard Worker int64_t frame_id = last_shared_frame_id_[0][0];
427*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_NE(frame_id, -1);
428*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LT(frame_id, shared_frame_id);
429*d9f75844SAndroid Build Coastguard Worker generic.chain_diffs.push_back(shared_frame_id - frame_id);
430*d9f75844SAndroid Build Coastguard Worker generic.dependencies.push_back(frame_id);
431*d9f75844SAndroid Build Coastguard Worker }
432*d9f75844SAndroid Build Coastguard Worker
433*d9f75844SAndroid Build Coastguard Worker last_shared_frame_id_[0][0] = shared_frame_id;
434*d9f75844SAndroid Build Coastguard Worker }
435*d9f75844SAndroid Build Coastguard Worker
H264ToGeneric(const CodecSpecificInfoH264 & h264_info,int64_t shared_frame_id,bool is_keyframe,RTPVideoHeader * rtp_video_header)436*d9f75844SAndroid Build Coastguard Worker void RtpPayloadParams::H264ToGeneric(const CodecSpecificInfoH264& h264_info,
437*d9f75844SAndroid Build Coastguard Worker int64_t shared_frame_id,
438*d9f75844SAndroid Build Coastguard Worker bool is_keyframe,
439*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader* rtp_video_header) {
440*d9f75844SAndroid Build Coastguard Worker const int temporal_index =
441*d9f75844SAndroid Build Coastguard Worker h264_info.temporal_idx != kNoTemporalIdx ? h264_info.temporal_idx : 0;
442*d9f75844SAndroid Build Coastguard Worker
443*d9f75844SAndroid Build Coastguard Worker if (temporal_index >= RtpGenericFrameDescriptor::kMaxTemporalLayers) {
444*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_WARNING) << "Temporal and/or spatial index is too high to be "
445*d9f75844SAndroid Build Coastguard Worker "used with generic frame descriptor.";
446*d9f75844SAndroid Build Coastguard Worker return;
447*d9f75844SAndroid Build Coastguard Worker }
448*d9f75844SAndroid Build Coastguard Worker
449*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader::GenericDescriptorInfo& generic =
450*d9f75844SAndroid Build Coastguard Worker rtp_video_header->generic.emplace();
451*d9f75844SAndroid Build Coastguard Worker
452*d9f75844SAndroid Build Coastguard Worker generic.frame_id = shared_frame_id;
453*d9f75844SAndroid Build Coastguard Worker generic.temporal_index = temporal_index;
454*d9f75844SAndroid Build Coastguard Worker
455*d9f75844SAndroid Build Coastguard Worker if (is_keyframe) {
456*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_EQ(temporal_index, 0);
457*d9f75844SAndroid Build Coastguard Worker last_shared_frame_id_[/*spatial index*/ 0].fill(-1);
458*d9f75844SAndroid Build Coastguard Worker last_shared_frame_id_[/*spatial index*/ 0][temporal_index] =
459*d9f75844SAndroid Build Coastguard Worker shared_frame_id;
460*d9f75844SAndroid Build Coastguard Worker return;
461*d9f75844SAndroid Build Coastguard Worker }
462*d9f75844SAndroid Build Coastguard Worker
463*d9f75844SAndroid Build Coastguard Worker if (h264_info.base_layer_sync) {
464*d9f75844SAndroid Build Coastguard Worker int64_t tl0_frame_id = last_shared_frame_id_[/*spatial index*/ 0][0];
465*d9f75844SAndroid Build Coastguard Worker
466*d9f75844SAndroid Build Coastguard Worker for (int i = 1; i < RtpGenericFrameDescriptor::kMaxTemporalLayers; ++i) {
467*d9f75844SAndroid Build Coastguard Worker if (last_shared_frame_id_[/*spatial index*/ 0][i] < tl0_frame_id) {
468*d9f75844SAndroid Build Coastguard Worker last_shared_frame_id_[/*spatial index*/ 0][i] = -1;
469*d9f75844SAndroid Build Coastguard Worker }
470*d9f75844SAndroid Build Coastguard Worker }
471*d9f75844SAndroid Build Coastguard Worker
472*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_GE(tl0_frame_id, 0);
473*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LT(tl0_frame_id, shared_frame_id);
474*d9f75844SAndroid Build Coastguard Worker generic.dependencies.push_back(tl0_frame_id);
475*d9f75844SAndroid Build Coastguard Worker } else {
476*d9f75844SAndroid Build Coastguard Worker for (int i = 0; i <= temporal_index; ++i) {
477*d9f75844SAndroid Build Coastguard Worker int64_t frame_id = last_shared_frame_id_[/*spatial index*/ 0][i];
478*d9f75844SAndroid Build Coastguard Worker
479*d9f75844SAndroid Build Coastguard Worker if (frame_id != -1) {
480*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LT(frame_id, shared_frame_id);
481*d9f75844SAndroid Build Coastguard Worker generic.dependencies.push_back(frame_id);
482*d9f75844SAndroid Build Coastguard Worker }
483*d9f75844SAndroid Build Coastguard Worker }
484*d9f75844SAndroid Build Coastguard Worker }
485*d9f75844SAndroid Build Coastguard Worker
486*d9f75844SAndroid Build Coastguard Worker last_shared_frame_id_[/*spatial_index*/ 0][temporal_index] = shared_frame_id;
487*d9f75844SAndroid Build Coastguard Worker }
488*d9f75844SAndroid Build Coastguard Worker
Vp8ToGeneric(const CodecSpecificInfoVP8 & vp8_info,int64_t shared_frame_id,bool is_keyframe,RTPVideoHeader * rtp_video_header)489*d9f75844SAndroid Build Coastguard Worker void RtpPayloadParams::Vp8ToGeneric(const CodecSpecificInfoVP8& vp8_info,
490*d9f75844SAndroid Build Coastguard Worker int64_t shared_frame_id,
491*d9f75844SAndroid Build Coastguard Worker bool is_keyframe,
492*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader* rtp_video_header) {
493*d9f75844SAndroid Build Coastguard Worker const auto& vp8_header =
494*d9f75844SAndroid Build Coastguard Worker absl::get<RTPVideoHeaderVP8>(rtp_video_header->video_type_header);
495*d9f75844SAndroid Build Coastguard Worker const int spatial_index = 0;
496*d9f75844SAndroid Build Coastguard Worker const int temporal_index =
497*d9f75844SAndroid Build Coastguard Worker vp8_header.temporalIdx != kNoTemporalIdx ? vp8_header.temporalIdx : 0;
498*d9f75844SAndroid Build Coastguard Worker
499*d9f75844SAndroid Build Coastguard Worker if (temporal_index >= RtpGenericFrameDescriptor::kMaxTemporalLayers ||
500*d9f75844SAndroid Build Coastguard Worker spatial_index >= RtpGenericFrameDescriptor::kMaxSpatialLayers) {
501*d9f75844SAndroid Build Coastguard Worker RTC_LOG(LS_WARNING) << "Temporal and/or spatial index is too high to be "
502*d9f75844SAndroid Build Coastguard Worker "used with generic frame descriptor.";
503*d9f75844SAndroid Build Coastguard Worker return;
504*d9f75844SAndroid Build Coastguard Worker }
505*d9f75844SAndroid Build Coastguard Worker
506*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader::GenericDescriptorInfo& generic =
507*d9f75844SAndroid Build Coastguard Worker rtp_video_header->generic.emplace();
508*d9f75844SAndroid Build Coastguard Worker
509*d9f75844SAndroid Build Coastguard Worker generic.frame_id = shared_frame_id;
510*d9f75844SAndroid Build Coastguard Worker generic.spatial_index = spatial_index;
511*d9f75844SAndroid Build Coastguard Worker generic.temporal_index = temporal_index;
512*d9f75844SAndroid Build Coastguard Worker
513*d9f75844SAndroid Build Coastguard Worker // Generate decode target indications.
514*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LT(temporal_index, kMaxTemporalStreams);
515*d9f75844SAndroid Build Coastguard Worker generic.decode_target_indications.resize(kMaxTemporalStreams);
516*d9f75844SAndroid Build Coastguard Worker auto it = std::fill_n(generic.decode_target_indications.begin(),
517*d9f75844SAndroid Build Coastguard Worker temporal_index, DecodeTargetIndication::kNotPresent);
518*d9f75844SAndroid Build Coastguard Worker std::fill(it, generic.decode_target_indications.end(),
519*d9f75844SAndroid Build Coastguard Worker DecodeTargetIndication::kSwitch);
520*d9f75844SAndroid Build Coastguard Worker
521*d9f75844SAndroid Build Coastguard Worker // Frame dependencies.
522*d9f75844SAndroid Build Coastguard Worker if (vp8_info.useExplicitDependencies) {
523*d9f75844SAndroid Build Coastguard Worker SetDependenciesVp8New(vp8_info, shared_frame_id, is_keyframe,
524*d9f75844SAndroid Build Coastguard Worker vp8_header.layerSync, &generic);
525*d9f75844SAndroid Build Coastguard Worker } else {
526*d9f75844SAndroid Build Coastguard Worker SetDependenciesVp8Deprecated(vp8_info, shared_frame_id, is_keyframe,
527*d9f75844SAndroid Build Coastguard Worker spatial_index, temporal_index,
528*d9f75844SAndroid Build Coastguard Worker vp8_header.layerSync, &generic);
529*d9f75844SAndroid Build Coastguard Worker }
530*d9f75844SAndroid Build Coastguard Worker
531*d9f75844SAndroid Build Coastguard Worker // Calculate chains.
532*d9f75844SAndroid Build Coastguard Worker generic.chain_diffs = {
533*d9f75844SAndroid Build Coastguard Worker (is_keyframe || chain_last_frame_id_[0] < 0)
534*d9f75844SAndroid Build Coastguard Worker ? 0
535*d9f75844SAndroid Build Coastguard Worker : static_cast<int>(shared_frame_id - chain_last_frame_id_[0])};
536*d9f75844SAndroid Build Coastguard Worker if (temporal_index == 0) {
537*d9f75844SAndroid Build Coastguard Worker chain_last_frame_id_[0] = shared_frame_id;
538*d9f75844SAndroid Build Coastguard Worker }
539*d9f75844SAndroid Build Coastguard Worker }
540*d9f75844SAndroid Build Coastguard Worker
Vp9ToGeneric(const CodecSpecificInfoVP9 & vp9_info,int64_t shared_frame_id,RTPVideoHeader & rtp_video_header)541*d9f75844SAndroid Build Coastguard Worker void RtpPayloadParams::Vp9ToGeneric(const CodecSpecificInfoVP9& vp9_info,
542*d9f75844SAndroid Build Coastguard Worker int64_t shared_frame_id,
543*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader& rtp_video_header) {
544*d9f75844SAndroid Build Coastguard Worker const auto& vp9_header =
545*d9f75844SAndroid Build Coastguard Worker absl::get<RTPVideoHeaderVP9>(rtp_video_header.video_type_header);
546*d9f75844SAndroid Build Coastguard Worker const int num_spatial_layers = kMaxSimulatedSpatialLayers;
547*d9f75844SAndroid Build Coastguard Worker const int num_active_spatial_layers = vp9_header.num_spatial_layers;
548*d9f75844SAndroid Build Coastguard Worker const int num_temporal_layers = kMaxTemporalStreams;
549*d9f75844SAndroid Build Coastguard Worker static_assert(num_spatial_layers <=
550*d9f75844SAndroid Build Coastguard Worker RtpGenericFrameDescriptor::kMaxSpatialLayers);
551*d9f75844SAndroid Build Coastguard Worker static_assert(num_temporal_layers <=
552*d9f75844SAndroid Build Coastguard Worker RtpGenericFrameDescriptor::kMaxTemporalLayers);
553*d9f75844SAndroid Build Coastguard Worker static_assert(num_spatial_layers <= DependencyDescriptor::kMaxSpatialIds);
554*d9f75844SAndroid Build Coastguard Worker static_assert(num_temporal_layers <= DependencyDescriptor::kMaxTemporalIds);
555*d9f75844SAndroid Build Coastguard Worker
556*d9f75844SAndroid Build Coastguard Worker int spatial_index =
557*d9f75844SAndroid Build Coastguard Worker vp9_header.spatial_idx != kNoSpatialIdx ? vp9_header.spatial_idx : 0;
558*d9f75844SAndroid Build Coastguard Worker int temporal_index =
559*d9f75844SAndroid Build Coastguard Worker vp9_header.temporal_idx != kNoTemporalIdx ? vp9_header.temporal_idx : 0;
560*d9f75844SAndroid Build Coastguard Worker
561*d9f75844SAndroid Build Coastguard Worker if (spatial_index >= num_spatial_layers ||
562*d9f75844SAndroid Build Coastguard Worker temporal_index >= num_temporal_layers ||
563*d9f75844SAndroid Build Coastguard Worker num_active_spatial_layers > num_spatial_layers) {
564*d9f75844SAndroid Build Coastguard Worker // Prefer to generate no generic layering than an inconsistent one.
565*d9f75844SAndroid Build Coastguard Worker return;
566*d9f75844SAndroid Build Coastguard Worker }
567*d9f75844SAndroid Build Coastguard Worker
568*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader::GenericDescriptorInfo& result =
569*d9f75844SAndroid Build Coastguard Worker rtp_video_header.generic.emplace();
570*d9f75844SAndroid Build Coastguard Worker
571*d9f75844SAndroid Build Coastguard Worker result.frame_id = shared_frame_id;
572*d9f75844SAndroid Build Coastguard Worker result.spatial_index = spatial_index;
573*d9f75844SAndroid Build Coastguard Worker result.temporal_index = temporal_index;
574*d9f75844SAndroid Build Coastguard Worker
575*d9f75844SAndroid Build Coastguard Worker result.decode_target_indications.reserve(num_spatial_layers *
576*d9f75844SAndroid Build Coastguard Worker num_temporal_layers);
577*d9f75844SAndroid Build Coastguard Worker for (int sid = 0; sid < num_spatial_layers; ++sid) {
578*d9f75844SAndroid Build Coastguard Worker for (int tid = 0; tid < num_temporal_layers; ++tid) {
579*d9f75844SAndroid Build Coastguard Worker DecodeTargetIndication dti;
580*d9f75844SAndroid Build Coastguard Worker if (sid < spatial_index || tid < temporal_index) {
581*d9f75844SAndroid Build Coastguard Worker dti = DecodeTargetIndication::kNotPresent;
582*d9f75844SAndroid Build Coastguard Worker } else if (spatial_index != sid &&
583*d9f75844SAndroid Build Coastguard Worker vp9_header.non_ref_for_inter_layer_pred) {
584*d9f75844SAndroid Build Coastguard Worker dti = DecodeTargetIndication::kNotPresent;
585*d9f75844SAndroid Build Coastguard Worker } else if (sid == spatial_index && tid == temporal_index) {
586*d9f75844SAndroid Build Coastguard Worker // Assume that if frame is decodable, all of its own layer is decodable.
587*d9f75844SAndroid Build Coastguard Worker dti = DecodeTargetIndication::kSwitch;
588*d9f75844SAndroid Build Coastguard Worker } else if (sid == spatial_index && vp9_header.temporal_up_switch) {
589*d9f75844SAndroid Build Coastguard Worker dti = DecodeTargetIndication::kSwitch;
590*d9f75844SAndroid Build Coastguard Worker } else if (!vp9_header.inter_pic_predicted) {
591*d9f75844SAndroid Build Coastguard Worker // Key frame or spatial upswitch
592*d9f75844SAndroid Build Coastguard Worker dti = DecodeTargetIndication::kSwitch;
593*d9f75844SAndroid Build Coastguard Worker } else {
594*d9f75844SAndroid Build Coastguard Worker // Make no other assumptions. That should be safe, though suboptimal.
595*d9f75844SAndroid Build Coastguard Worker // To provide more accurate dti, encoder wrapper should fill in
596*d9f75844SAndroid Build Coastguard Worker // CodecSpecificInfo::generic_frame_info
597*d9f75844SAndroid Build Coastguard Worker dti = DecodeTargetIndication::kRequired;
598*d9f75844SAndroid Build Coastguard Worker }
599*d9f75844SAndroid Build Coastguard Worker result.decode_target_indications.push_back(dti);
600*d9f75844SAndroid Build Coastguard Worker }
601*d9f75844SAndroid Build Coastguard Worker }
602*d9f75844SAndroid Build Coastguard Worker
603*d9f75844SAndroid Build Coastguard Worker // Calculate frame dependencies.
604*d9f75844SAndroid Build Coastguard Worker static constexpr int kPictureDiffLimit = 128;
605*d9f75844SAndroid Build Coastguard Worker if (last_vp9_frame_id_.empty()) {
606*d9f75844SAndroid Build Coastguard Worker // Create the array only if it is ever used.
607*d9f75844SAndroid Build Coastguard Worker last_vp9_frame_id_.resize(kPictureDiffLimit);
608*d9f75844SAndroid Build Coastguard Worker }
609*d9f75844SAndroid Build Coastguard Worker if (vp9_header.inter_layer_predicted && spatial_index > 0) {
610*d9f75844SAndroid Build Coastguard Worker result.dependencies.push_back(
611*d9f75844SAndroid Build Coastguard Worker last_vp9_frame_id_[vp9_header.picture_id % kPictureDiffLimit]
612*d9f75844SAndroid Build Coastguard Worker [spatial_index - 1]);
613*d9f75844SAndroid Build Coastguard Worker }
614*d9f75844SAndroid Build Coastguard Worker if (vp9_header.inter_pic_predicted) {
615*d9f75844SAndroid Build Coastguard Worker for (size_t i = 0; i < vp9_header.num_ref_pics; ++i) {
616*d9f75844SAndroid Build Coastguard Worker // picture_id is 15 bit number that wraps around. Though undeflow may
617*d9f75844SAndroid Build Coastguard Worker // produce picture that exceeds 2^15, it is ok because in this
618*d9f75844SAndroid Build Coastguard Worker // code block only last 7 bits of the picture_id are used.
619*d9f75844SAndroid Build Coastguard Worker uint16_t depend_on = vp9_header.picture_id - vp9_header.pid_diff[i];
620*d9f75844SAndroid Build Coastguard Worker result.dependencies.push_back(
621*d9f75844SAndroid Build Coastguard Worker last_vp9_frame_id_[depend_on % kPictureDiffLimit][spatial_index]);
622*d9f75844SAndroid Build Coastguard Worker }
623*d9f75844SAndroid Build Coastguard Worker }
624*d9f75844SAndroid Build Coastguard Worker last_vp9_frame_id_[vp9_header.picture_id % kPictureDiffLimit][spatial_index] =
625*d9f75844SAndroid Build Coastguard Worker shared_frame_id;
626*d9f75844SAndroid Build Coastguard Worker
627*d9f75844SAndroid Build Coastguard Worker result.active_decode_targets =
628*d9f75844SAndroid Build Coastguard Worker ((uint32_t{1} << num_temporal_layers * num_active_spatial_layers) - 1);
629*d9f75844SAndroid Build Coastguard Worker
630*d9f75844SAndroid Build Coastguard Worker // Calculate chains, asuming chain includes all frames with temporal_id = 0
631*d9f75844SAndroid Build Coastguard Worker if (!vp9_header.inter_pic_predicted && !vp9_header.inter_layer_predicted) {
632*d9f75844SAndroid Build Coastguard Worker // Assume frames without dependencies also reset chains.
633*d9f75844SAndroid Build Coastguard Worker for (int sid = spatial_index; sid < num_spatial_layers; ++sid) {
634*d9f75844SAndroid Build Coastguard Worker chain_last_frame_id_[sid] = -1;
635*d9f75844SAndroid Build Coastguard Worker }
636*d9f75844SAndroid Build Coastguard Worker }
637*d9f75844SAndroid Build Coastguard Worker result.chain_diffs.resize(num_spatial_layers, 0);
638*d9f75844SAndroid Build Coastguard Worker for (int sid = 0; sid < num_active_spatial_layers; ++sid) {
639*d9f75844SAndroid Build Coastguard Worker if (chain_last_frame_id_[sid] == -1) {
640*d9f75844SAndroid Build Coastguard Worker result.chain_diffs[sid] = 0;
641*d9f75844SAndroid Build Coastguard Worker continue;
642*d9f75844SAndroid Build Coastguard Worker }
643*d9f75844SAndroid Build Coastguard Worker result.chain_diffs[sid] = shared_frame_id - chain_last_frame_id_[sid];
644*d9f75844SAndroid Build Coastguard Worker }
645*d9f75844SAndroid Build Coastguard Worker
646*d9f75844SAndroid Build Coastguard Worker if (temporal_index == 0) {
647*d9f75844SAndroid Build Coastguard Worker chain_last_frame_id_[spatial_index] = shared_frame_id;
648*d9f75844SAndroid Build Coastguard Worker if (!vp9_header.non_ref_for_inter_layer_pred) {
649*d9f75844SAndroid Build Coastguard Worker for (int sid = spatial_index + 1; sid < num_spatial_layers; ++sid) {
650*d9f75844SAndroid Build Coastguard Worker chain_last_frame_id_[sid] = shared_frame_id;
651*d9f75844SAndroid Build Coastguard Worker }
652*d9f75844SAndroid Build Coastguard Worker }
653*d9f75844SAndroid Build Coastguard Worker }
654*d9f75844SAndroid Build Coastguard Worker }
655*d9f75844SAndroid Build Coastguard Worker
SetDependenciesVp8Deprecated(const CodecSpecificInfoVP8 & vp8_info,int64_t shared_frame_id,bool is_keyframe,int spatial_index,int temporal_index,bool layer_sync,RTPVideoHeader::GenericDescriptorInfo * generic)656*d9f75844SAndroid Build Coastguard Worker void RtpPayloadParams::SetDependenciesVp8Deprecated(
657*d9f75844SAndroid Build Coastguard Worker const CodecSpecificInfoVP8& vp8_info,
658*d9f75844SAndroid Build Coastguard Worker int64_t shared_frame_id,
659*d9f75844SAndroid Build Coastguard Worker bool is_keyframe,
660*d9f75844SAndroid Build Coastguard Worker int spatial_index,
661*d9f75844SAndroid Build Coastguard Worker int temporal_index,
662*d9f75844SAndroid Build Coastguard Worker bool layer_sync,
663*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader::GenericDescriptorInfo* generic) {
664*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(!vp8_info.useExplicitDependencies);
665*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(!new_version_used_.has_value() || !new_version_used_.value());
666*d9f75844SAndroid Build Coastguard Worker new_version_used_ = false;
667*d9f75844SAndroid Build Coastguard Worker
668*d9f75844SAndroid Build Coastguard Worker if (is_keyframe) {
669*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_EQ(temporal_index, 0);
670*d9f75844SAndroid Build Coastguard Worker last_shared_frame_id_[spatial_index].fill(-1);
671*d9f75844SAndroid Build Coastguard Worker last_shared_frame_id_[spatial_index][temporal_index] = shared_frame_id;
672*d9f75844SAndroid Build Coastguard Worker return;
673*d9f75844SAndroid Build Coastguard Worker }
674*d9f75844SAndroid Build Coastguard Worker
675*d9f75844SAndroid Build Coastguard Worker if (layer_sync) {
676*d9f75844SAndroid Build Coastguard Worker int64_t tl0_frame_id = last_shared_frame_id_[spatial_index][0];
677*d9f75844SAndroid Build Coastguard Worker
678*d9f75844SAndroid Build Coastguard Worker for (int i = 1; i < RtpGenericFrameDescriptor::kMaxTemporalLayers; ++i) {
679*d9f75844SAndroid Build Coastguard Worker if (last_shared_frame_id_[spatial_index][i] < tl0_frame_id) {
680*d9f75844SAndroid Build Coastguard Worker last_shared_frame_id_[spatial_index][i] = -1;
681*d9f75844SAndroid Build Coastguard Worker }
682*d9f75844SAndroid Build Coastguard Worker }
683*d9f75844SAndroid Build Coastguard Worker
684*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_GE(tl0_frame_id, 0);
685*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LT(tl0_frame_id, shared_frame_id);
686*d9f75844SAndroid Build Coastguard Worker generic->dependencies.push_back(tl0_frame_id);
687*d9f75844SAndroid Build Coastguard Worker } else {
688*d9f75844SAndroid Build Coastguard Worker for (int i = 0; i <= temporal_index; ++i) {
689*d9f75844SAndroid Build Coastguard Worker int64_t frame_id = last_shared_frame_id_[spatial_index][i];
690*d9f75844SAndroid Build Coastguard Worker
691*d9f75844SAndroid Build Coastguard Worker if (frame_id != -1) {
692*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LT(frame_id, shared_frame_id);
693*d9f75844SAndroid Build Coastguard Worker generic->dependencies.push_back(frame_id);
694*d9f75844SAndroid Build Coastguard Worker }
695*d9f75844SAndroid Build Coastguard Worker }
696*d9f75844SAndroid Build Coastguard Worker }
697*d9f75844SAndroid Build Coastguard Worker
698*d9f75844SAndroid Build Coastguard Worker last_shared_frame_id_[spatial_index][temporal_index] = shared_frame_id;
699*d9f75844SAndroid Build Coastguard Worker }
700*d9f75844SAndroid Build Coastguard Worker
SetDependenciesVp8New(const CodecSpecificInfoVP8 & vp8_info,int64_t shared_frame_id,bool is_keyframe,bool layer_sync,RTPVideoHeader::GenericDescriptorInfo * generic)701*d9f75844SAndroid Build Coastguard Worker void RtpPayloadParams::SetDependenciesVp8New(
702*d9f75844SAndroid Build Coastguard Worker const CodecSpecificInfoVP8& vp8_info,
703*d9f75844SAndroid Build Coastguard Worker int64_t shared_frame_id,
704*d9f75844SAndroid Build Coastguard Worker bool is_keyframe,
705*d9f75844SAndroid Build Coastguard Worker bool layer_sync,
706*d9f75844SAndroid Build Coastguard Worker RTPVideoHeader::GenericDescriptorInfo* generic) {
707*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(vp8_info.useExplicitDependencies);
708*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK(!new_version_used_.has_value() || new_version_used_.value());
709*d9f75844SAndroid Build Coastguard Worker new_version_used_ = true;
710*d9f75844SAndroid Build Coastguard Worker
711*d9f75844SAndroid Build Coastguard Worker if (is_keyframe) {
712*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_EQ(vp8_info.referencedBuffersCount, 0u);
713*d9f75844SAndroid Build Coastguard Worker buffer_id_to_frame_id_.fill(shared_frame_id);
714*d9f75844SAndroid Build Coastguard Worker return;
715*d9f75844SAndroid Build Coastguard Worker }
716*d9f75844SAndroid Build Coastguard Worker
717*d9f75844SAndroid Build Coastguard Worker constexpr size_t kBuffersCountVp8 = CodecSpecificInfoVP8::kBuffersCount;
718*d9f75844SAndroid Build Coastguard Worker
719*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_GT(vp8_info.referencedBuffersCount, 0u);
720*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LE(vp8_info.referencedBuffersCount,
721*d9f75844SAndroid Build Coastguard Worker arraysize(vp8_info.referencedBuffers));
722*d9f75844SAndroid Build Coastguard Worker
723*d9f75844SAndroid Build Coastguard Worker for (size_t i = 0; i < vp8_info.referencedBuffersCount; ++i) {
724*d9f75844SAndroid Build Coastguard Worker const size_t referenced_buffer = vp8_info.referencedBuffers[i];
725*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LT(referenced_buffer, kBuffersCountVp8);
726*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LT(referenced_buffer, buffer_id_to_frame_id_.size());
727*d9f75844SAndroid Build Coastguard Worker
728*d9f75844SAndroid Build Coastguard Worker const int64_t dependency_frame_id =
729*d9f75844SAndroid Build Coastguard Worker buffer_id_to_frame_id_[referenced_buffer];
730*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_GE(dependency_frame_id, 0);
731*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LT(dependency_frame_id, shared_frame_id);
732*d9f75844SAndroid Build Coastguard Worker
733*d9f75844SAndroid Build Coastguard Worker const bool is_new_dependency =
734*d9f75844SAndroid Build Coastguard Worker std::find(generic->dependencies.begin(), generic->dependencies.end(),
735*d9f75844SAndroid Build Coastguard Worker dependency_frame_id) == generic->dependencies.end();
736*d9f75844SAndroid Build Coastguard Worker if (is_new_dependency) {
737*d9f75844SAndroid Build Coastguard Worker generic->dependencies.push_back(dependency_frame_id);
738*d9f75844SAndroid Build Coastguard Worker }
739*d9f75844SAndroid Build Coastguard Worker }
740*d9f75844SAndroid Build Coastguard Worker
741*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LE(vp8_info.updatedBuffersCount, kBuffersCountVp8);
742*d9f75844SAndroid Build Coastguard Worker for (size_t i = 0; i < vp8_info.updatedBuffersCount; ++i) {
743*d9f75844SAndroid Build Coastguard Worker const size_t updated_id = vp8_info.updatedBuffers[i];
744*d9f75844SAndroid Build Coastguard Worker buffer_id_to_frame_id_[updated_id] = shared_frame_id;
745*d9f75844SAndroid Build Coastguard Worker }
746*d9f75844SAndroid Build Coastguard Worker
747*d9f75844SAndroid Build Coastguard Worker RTC_DCHECK_LE(buffer_id_to_frame_id_.size(), kBuffersCountVp8);
748*d9f75844SAndroid Build Coastguard Worker }
749*d9f75844SAndroid Build Coastguard Worker
750*d9f75844SAndroid Build Coastguard Worker } // namespace webrtc
751