1 /*
2 * Copyright (c) 2014 The WebM project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include <math.h>
12
13 #include "vp9/encoder/vp9_aq_cyclicrefresh.h"
14 #include "vp9/encoder/vp9_encoder.h"
15 #include "vp9/encoder/vp9_svc_layercontext.h"
16 #include "vp9/encoder/vp9_extend.h"
17 #include "vpx_dsp/vpx_dsp_common.h"
18
19 #define SMALL_FRAME_WIDTH 32
20 #define SMALL_FRAME_HEIGHT 16
21
swap_ptr(void * a,void * b)22 static void swap_ptr(void *a, void *b) {
23 void **a_p = (void **)a;
24 void **b_p = (void **)b;
25 void *c = *a_p;
26 *a_p = *b_p;
27 *b_p = c;
28 }
29
vp9_init_layer_context(VP9_COMP * const cpi)30 void vp9_init_layer_context(VP9_COMP *const cpi) {
31 SVC *const svc = &cpi->svc;
32 const VP9EncoderConfig *const oxcf = &cpi->oxcf;
33 int mi_rows = cpi->common.mi_rows;
34 int mi_cols = cpi->common.mi_cols;
35 int sl, tl, i;
36 int alt_ref_idx = svc->number_spatial_layers;
37
38 svc->spatial_layer_id = 0;
39 svc->temporal_layer_id = 0;
40 svc->force_zero_mode_spatial_ref = 0;
41 svc->use_base_mv = 0;
42 svc->use_partition_reuse = 0;
43 svc->use_gf_temporal_ref = 1;
44 svc->use_gf_temporal_ref_current_layer = 0;
45 svc->scaled_temp_is_alloc = 0;
46 svc->scaled_one_half = 0;
47 svc->current_superframe = 0;
48 svc->non_reference_frame = 0;
49 svc->skip_enhancement_layer = 0;
50 svc->disable_inter_layer_pred = INTER_LAYER_PRED_ON;
51 svc->framedrop_mode = CONSTRAINED_LAYER_DROP;
52 svc->set_intra_only_frame = 0;
53 svc->previous_frame_is_intra_only = 0;
54 svc->superframe_has_layer_sync = 0;
55 svc->use_set_ref_frame_config = 0;
56 svc->num_encoded_top_layer = 0;
57
58 for (i = 0; i < REF_FRAMES; ++i) {
59 svc->fb_idx_spatial_layer_id[i] = -1;
60 svc->fb_idx_temporal_layer_id[i] = -1;
61 svc->fb_idx_base[i] = 0;
62 }
63 for (sl = 0; sl < oxcf->ss_number_layers; ++sl) {
64 svc->last_layer_dropped[sl] = 0;
65 svc->drop_spatial_layer[sl] = 0;
66 svc->ext_frame_flags[sl] = 0;
67 svc->lst_fb_idx[sl] = 0;
68 svc->gld_fb_idx[sl] = 1;
69 svc->alt_fb_idx[sl] = 2;
70 svc->downsample_filter_type[sl] = BILINEAR;
71 svc->downsample_filter_phase[sl] = 8; // Set to 8 for averaging filter.
72 svc->framedrop_thresh[sl] = oxcf->drop_frames_water_mark;
73 svc->fb_idx_upd_tl0[sl] = -1;
74 svc->drop_count[sl] = 0;
75 svc->spatial_layer_sync[sl] = 0;
76 }
77 svc->max_consec_drop = INT_MAX;
78
79 svc->buffer_gf_temporal_ref[1].idx = 7;
80 svc->buffer_gf_temporal_ref[0].idx = 6;
81 svc->buffer_gf_temporal_ref[1].is_used = 0;
82 svc->buffer_gf_temporal_ref[0].is_used = 0;
83
84 if (cpi->oxcf.error_resilient_mode == 0 && cpi->oxcf.pass == 2) {
85 if (vpx_realloc_frame_buffer(&cpi->svc.empty_frame.img, SMALL_FRAME_WIDTH,
86 SMALL_FRAME_HEIGHT, cpi->common.subsampling_x,
87 cpi->common.subsampling_y,
88 #if CONFIG_VP9_HIGHBITDEPTH
89 cpi->common.use_highbitdepth,
90 #endif
91 VP9_ENC_BORDER_IN_PIXELS,
92 cpi->common.byte_alignment, NULL, NULL, NULL))
93 vpx_internal_error(&cpi->common.error, VPX_CODEC_MEM_ERROR,
94 "Failed to allocate empty frame for multiple frame "
95 "contexts");
96
97 memset(cpi->svc.empty_frame.img.buffer_alloc, 0x80,
98 cpi->svc.empty_frame.img.buffer_alloc_sz);
99 }
100
101 for (sl = 0; sl < oxcf->ss_number_layers; ++sl) {
102 for (tl = 0; tl < oxcf->ts_number_layers; ++tl) {
103 int layer = LAYER_IDS_TO_IDX(sl, tl, oxcf->ts_number_layers);
104 LAYER_CONTEXT *const lc = &svc->layer_context[layer];
105 RATE_CONTROL *const lrc = &lc->rc;
106 int i;
107 lc->current_video_frame_in_layer = 0;
108 lc->layer_size = 0;
109 lc->frames_from_key_frame = 0;
110 lc->last_frame_type = FRAME_TYPES;
111 lrc->ni_av_qi = oxcf->worst_allowed_q;
112 lrc->total_actual_bits = 0;
113 lrc->total_target_vs_actual = 0;
114 lrc->ni_tot_qi = 0;
115 lrc->tot_q = 0.0;
116 lrc->avg_q = 0.0;
117 lrc->ni_frames = 0;
118 lrc->decimation_count = 0;
119 lrc->decimation_factor = 0;
120 lrc->worst_quality = oxcf->worst_allowed_q;
121 lrc->best_quality = oxcf->best_allowed_q;
122
123 for (i = 0; i < RATE_FACTOR_LEVELS; ++i) {
124 lrc->rate_correction_factors[i] = 1.0;
125 }
126
127 if (cpi->oxcf.rc_mode == VPX_CBR) {
128 lc->target_bandwidth = oxcf->layer_target_bitrate[layer];
129 lrc->last_q[INTER_FRAME] = oxcf->worst_allowed_q;
130 lrc->avg_frame_qindex[INTER_FRAME] = oxcf->worst_allowed_q;
131 lrc->avg_frame_qindex[KEY_FRAME] = oxcf->worst_allowed_q;
132 } else {
133 lc->target_bandwidth = oxcf->layer_target_bitrate[layer];
134 lrc->last_q[KEY_FRAME] = oxcf->best_allowed_q;
135 lrc->last_q[INTER_FRAME] = oxcf->best_allowed_q;
136 lrc->avg_frame_qindex[KEY_FRAME] =
137 (oxcf->worst_allowed_q + oxcf->best_allowed_q) / 2;
138 lrc->avg_frame_qindex[INTER_FRAME] =
139 (oxcf->worst_allowed_q + oxcf->best_allowed_q) / 2;
140 if (oxcf->ss_enable_auto_arf[sl])
141 lc->alt_ref_idx = alt_ref_idx++;
142 else
143 lc->alt_ref_idx = INVALID_IDX;
144 lc->gold_ref_idx = INVALID_IDX;
145 }
146
147 lrc->buffer_level =
148 oxcf->starting_buffer_level_ms * lc->target_bandwidth / 1000;
149 lrc->bits_off_target = lrc->buffer_level;
150
151 // Initialize the cyclic refresh parameters. If spatial layers are used
152 // (i.e., ss_number_layers > 1), these need to be updated per spatial
153 // layer.
154 // Cyclic refresh is only applied on base temporal layer.
155 if (oxcf->ss_number_layers > 1 && tl == 0) {
156 size_t last_coded_q_map_size;
157 size_t consec_zero_mv_size;
158 VP9_COMMON *const cm = &cpi->common;
159 lc->sb_index = 0;
160 lc->actual_num_seg1_blocks = 0;
161 lc->actual_num_seg2_blocks = 0;
162 lc->counter_encode_maxq_scene_change = 0;
163 CHECK_MEM_ERROR(cm, lc->map,
164 vpx_malloc(mi_rows * mi_cols * sizeof(*lc->map)));
165 memset(lc->map, 0, mi_rows * mi_cols);
166 last_coded_q_map_size =
167 mi_rows * mi_cols * sizeof(*lc->last_coded_q_map);
168 CHECK_MEM_ERROR(cm, lc->last_coded_q_map,
169 vpx_malloc(last_coded_q_map_size));
170 assert(MAXQ <= 255);
171 memset(lc->last_coded_q_map, MAXQ, last_coded_q_map_size);
172 consec_zero_mv_size = mi_rows * mi_cols * sizeof(*lc->consec_zero_mv);
173 CHECK_MEM_ERROR(cm, lc->consec_zero_mv,
174 vpx_malloc(consec_zero_mv_size));
175 memset(lc->consec_zero_mv, 0, consec_zero_mv_size);
176 }
177 }
178 }
179
180 // Still have extra buffer for base layer golden frame
181 if (!(svc->number_temporal_layers > 1 && cpi->oxcf.rc_mode == VPX_CBR) &&
182 alt_ref_idx < REF_FRAMES)
183 svc->layer_context[0].gold_ref_idx = alt_ref_idx;
184 }
185
186 // Update the layer context from a change_config() call.
vp9_update_layer_context_change_config(VP9_COMP * const cpi,const int target_bandwidth)187 void vp9_update_layer_context_change_config(VP9_COMP *const cpi,
188 const int target_bandwidth) {
189 SVC *const svc = &cpi->svc;
190 const VP9EncoderConfig *const oxcf = &cpi->oxcf;
191 const RATE_CONTROL *const rc = &cpi->rc;
192 int sl, tl, layer = 0, spatial_layer_target;
193 float bitrate_alloc = 1.0;
194
195 cpi->svc.temporal_layering_mode = oxcf->temporal_layering_mode;
196
197 if (svc->temporal_layering_mode != VP9E_TEMPORAL_LAYERING_MODE_NOLAYERING) {
198 for (sl = 0; sl < oxcf->ss_number_layers; ++sl) {
199 for (tl = 0; tl < oxcf->ts_number_layers; ++tl) {
200 layer = LAYER_IDS_TO_IDX(sl, tl, oxcf->ts_number_layers);
201 svc->layer_context[layer].target_bandwidth =
202 oxcf->layer_target_bitrate[layer];
203 }
204
205 layer = LAYER_IDS_TO_IDX(
206 sl,
207 ((oxcf->ts_number_layers - 1) < 0 ? 0 : (oxcf->ts_number_layers - 1)),
208 oxcf->ts_number_layers);
209 spatial_layer_target = svc->layer_context[layer].target_bandwidth =
210 oxcf->layer_target_bitrate[layer];
211
212 for (tl = 0; tl < oxcf->ts_number_layers; ++tl) {
213 LAYER_CONTEXT *const lc =
214 &svc->layer_context[sl * oxcf->ts_number_layers + tl];
215 RATE_CONTROL *const lrc = &lc->rc;
216
217 lc->spatial_layer_target_bandwidth = spatial_layer_target;
218 bitrate_alloc = (float)lc->target_bandwidth / target_bandwidth;
219 lrc->starting_buffer_level =
220 (int64_t)(rc->starting_buffer_level * bitrate_alloc);
221 lrc->optimal_buffer_level =
222 (int64_t)(rc->optimal_buffer_level * bitrate_alloc);
223 lrc->maximum_buffer_size =
224 (int64_t)(rc->maximum_buffer_size * bitrate_alloc);
225 lrc->bits_off_target =
226 VPXMIN(lrc->bits_off_target, lrc->maximum_buffer_size);
227 lrc->buffer_level = VPXMIN(lrc->buffer_level, lrc->maximum_buffer_size);
228 lc->framerate = cpi->framerate / oxcf->ts_rate_decimator[tl];
229 lrc->avg_frame_bandwidth = (int)(lc->target_bandwidth / lc->framerate);
230 lrc->max_frame_bandwidth = rc->max_frame_bandwidth;
231 lrc->worst_quality = rc->worst_quality;
232 lrc->best_quality = rc->best_quality;
233 }
234 }
235 } else {
236 int layer_end;
237
238 if (svc->number_temporal_layers > 1 && cpi->oxcf.rc_mode == VPX_CBR) {
239 layer_end = svc->number_temporal_layers;
240 } else {
241 layer_end = svc->number_spatial_layers;
242 }
243
244 for (layer = 0; layer < layer_end; ++layer) {
245 LAYER_CONTEXT *const lc = &svc->layer_context[layer];
246 RATE_CONTROL *const lrc = &lc->rc;
247
248 lc->target_bandwidth = oxcf->layer_target_bitrate[layer];
249
250 bitrate_alloc = (float)lc->target_bandwidth / target_bandwidth;
251 // Update buffer-related quantities.
252 lrc->starting_buffer_level =
253 (int64_t)(rc->starting_buffer_level * bitrate_alloc);
254 lrc->optimal_buffer_level =
255 (int64_t)(rc->optimal_buffer_level * bitrate_alloc);
256 lrc->maximum_buffer_size =
257 (int64_t)(rc->maximum_buffer_size * bitrate_alloc);
258 lrc->bits_off_target =
259 VPXMIN(lrc->bits_off_target, lrc->maximum_buffer_size);
260 lrc->buffer_level = VPXMIN(lrc->buffer_level, lrc->maximum_buffer_size);
261 // Update framerate-related quantities.
262 if (svc->number_temporal_layers > 1 && cpi->oxcf.rc_mode == VPX_CBR) {
263 lc->framerate = cpi->framerate / oxcf->ts_rate_decimator[layer];
264 } else {
265 lc->framerate = cpi->framerate;
266 }
267 lrc->avg_frame_bandwidth = (int)(lc->target_bandwidth / lc->framerate);
268 lrc->max_frame_bandwidth = rc->max_frame_bandwidth;
269 // Update qp-related quantities.
270 lrc->worst_quality = rc->worst_quality;
271 lrc->best_quality = rc->best_quality;
272 }
273 }
274 }
275
get_layer_context(VP9_COMP * const cpi)276 static LAYER_CONTEXT *get_layer_context(VP9_COMP *const cpi) {
277 if (is_one_pass_cbr_svc(cpi))
278 return &cpi->svc.layer_context[cpi->svc.spatial_layer_id *
279 cpi->svc.number_temporal_layers +
280 cpi->svc.temporal_layer_id];
281 else
282 return (cpi->svc.number_temporal_layers > 1 && cpi->oxcf.rc_mode == VPX_CBR)
283 ? &cpi->svc.layer_context[cpi->svc.temporal_layer_id]
284 : &cpi->svc.layer_context[cpi->svc.spatial_layer_id];
285 }
286
vp9_update_temporal_layer_framerate(VP9_COMP * const cpi)287 void vp9_update_temporal_layer_framerate(VP9_COMP *const cpi) {
288 SVC *const svc = &cpi->svc;
289 const VP9EncoderConfig *const oxcf = &cpi->oxcf;
290 LAYER_CONTEXT *const lc = get_layer_context(cpi);
291 RATE_CONTROL *const lrc = &lc->rc;
292 // Index into spatial+temporal arrays.
293 const int st_idx = svc->spatial_layer_id * svc->number_temporal_layers +
294 svc->temporal_layer_id;
295 const int tl = svc->temporal_layer_id;
296
297 lc->framerate = cpi->framerate / oxcf->ts_rate_decimator[tl];
298 lrc->avg_frame_bandwidth = (int)(lc->target_bandwidth / lc->framerate);
299 lrc->max_frame_bandwidth = cpi->rc.max_frame_bandwidth;
300 // Update the average layer frame size (non-cumulative per-frame-bw).
301 if (tl == 0) {
302 lc->avg_frame_size = lrc->avg_frame_bandwidth;
303 } else {
304 const double prev_layer_framerate =
305 cpi->framerate / oxcf->ts_rate_decimator[tl - 1];
306 const int prev_layer_target_bandwidth =
307 oxcf->layer_target_bitrate[st_idx - 1];
308 lc->avg_frame_size =
309 (int)((lc->target_bandwidth - prev_layer_target_bandwidth) /
310 (lc->framerate - prev_layer_framerate));
311 }
312 }
313
vp9_update_spatial_layer_framerate(VP9_COMP * const cpi,double framerate)314 void vp9_update_spatial_layer_framerate(VP9_COMP *const cpi, double framerate) {
315 const VP9EncoderConfig *const oxcf = &cpi->oxcf;
316 LAYER_CONTEXT *const lc = get_layer_context(cpi);
317 RATE_CONTROL *const lrc = &lc->rc;
318
319 lc->framerate = framerate;
320 lrc->avg_frame_bandwidth = (int)(lc->target_bandwidth / lc->framerate);
321 lrc->min_frame_bandwidth =
322 (int)(lrc->avg_frame_bandwidth * oxcf->two_pass_vbrmin_section / 100);
323 lrc->max_frame_bandwidth = (int)(((int64_t)lrc->avg_frame_bandwidth *
324 oxcf->two_pass_vbrmax_section) /
325 100);
326 vp9_rc_set_gf_interval_range(cpi, lrc);
327 }
328
vp9_restore_layer_context(VP9_COMP * const cpi)329 void vp9_restore_layer_context(VP9_COMP *const cpi) {
330 LAYER_CONTEXT *const lc = get_layer_context(cpi);
331 const int old_frame_since_key = cpi->rc.frames_since_key;
332 const int old_frame_to_key = cpi->rc.frames_to_key;
333 const int old_ext_use_post_encode_drop = cpi->rc.ext_use_post_encode_drop;
334
335 cpi->rc = lc->rc;
336 cpi->twopass = lc->twopass;
337 cpi->oxcf.target_bandwidth = lc->target_bandwidth;
338 cpi->alt_ref_source = lc->alt_ref_source;
339 // Check if it is one_pass_cbr_svc mode and lc->speed > 0 (real-time mode
340 // does not use speed = 0).
341 if (is_one_pass_cbr_svc(cpi) && lc->speed > 0) {
342 cpi->oxcf.speed = lc->speed;
343 }
344 // Reset the frames_since_key and frames_to_key counters to their values
345 // before the layer restore. Keep these defined for the stream (not layer).
346 if (cpi->svc.number_temporal_layers > 1 ||
347 cpi->svc.number_spatial_layers > 1) {
348 cpi->rc.frames_since_key = old_frame_since_key;
349 cpi->rc.frames_to_key = old_frame_to_key;
350 }
351 cpi->rc.ext_use_post_encode_drop = old_ext_use_post_encode_drop;
352 // For spatial-svc, allow cyclic-refresh to be applied on the spatial layers,
353 // for the base temporal layer.
354 if (cpi->oxcf.aq_mode == CYCLIC_REFRESH_AQ &&
355 cpi->svc.number_spatial_layers > 1 && cpi->svc.temporal_layer_id == 0) {
356 CYCLIC_REFRESH *const cr = cpi->cyclic_refresh;
357 swap_ptr(&cr->map, &lc->map);
358 swap_ptr(&cr->last_coded_q_map, &lc->last_coded_q_map);
359 swap_ptr(&cpi->consec_zero_mv, &lc->consec_zero_mv);
360 cr->sb_index = lc->sb_index;
361 cr->actual_num_seg1_blocks = lc->actual_num_seg1_blocks;
362 cr->actual_num_seg2_blocks = lc->actual_num_seg2_blocks;
363 cr->counter_encode_maxq_scene_change = lc->counter_encode_maxq_scene_change;
364 }
365 }
366
vp9_save_layer_context(VP9_COMP * const cpi)367 void vp9_save_layer_context(VP9_COMP *const cpi) {
368 const VP9EncoderConfig *const oxcf = &cpi->oxcf;
369 LAYER_CONTEXT *const lc = get_layer_context(cpi);
370
371 lc->rc = cpi->rc;
372 lc->twopass = cpi->twopass;
373 lc->target_bandwidth = (int)oxcf->target_bandwidth;
374 lc->alt_ref_source = cpi->alt_ref_source;
375
376 // For spatial-svc, allow cyclic-refresh to be applied on the spatial layers,
377 // for the base temporal layer.
378 if (cpi->oxcf.aq_mode == CYCLIC_REFRESH_AQ &&
379 cpi->svc.number_spatial_layers > 1 && cpi->svc.temporal_layer_id == 0) {
380 CYCLIC_REFRESH *const cr = cpi->cyclic_refresh;
381 signed char *temp = lc->map;
382 uint8_t *temp2 = lc->last_coded_q_map;
383 uint8_t *temp3 = lc->consec_zero_mv;
384 lc->map = cr->map;
385 cr->map = temp;
386 lc->last_coded_q_map = cr->last_coded_q_map;
387 cr->last_coded_q_map = temp2;
388 lc->consec_zero_mv = cpi->consec_zero_mv;
389 cpi->consec_zero_mv = temp3;
390 lc->sb_index = cr->sb_index;
391 lc->actual_num_seg1_blocks = cr->actual_num_seg1_blocks;
392 lc->actual_num_seg2_blocks = cr->actual_num_seg2_blocks;
393 lc->counter_encode_maxq_scene_change = cr->counter_encode_maxq_scene_change;
394 }
395 }
396
397 #if !CONFIG_REALTIME_ONLY
vp9_init_second_pass_spatial_svc(VP9_COMP * cpi)398 void vp9_init_second_pass_spatial_svc(VP9_COMP *cpi) {
399 SVC *const svc = &cpi->svc;
400 int i;
401
402 for (i = 0; i < svc->number_spatial_layers; ++i) {
403 TWO_PASS *const twopass = &svc->layer_context[i].twopass;
404
405 svc->spatial_layer_id = i;
406 vp9_init_second_pass(cpi);
407
408 twopass->total_stats.spatial_layer_id = i;
409 twopass->total_left_stats.spatial_layer_id = i;
410 }
411 svc->spatial_layer_id = 0;
412 }
413 #endif // !CONFIG_REALTIME_ONLY
414
vp9_inc_frame_in_layer(VP9_COMP * const cpi)415 void vp9_inc_frame_in_layer(VP9_COMP *const cpi) {
416 LAYER_CONTEXT *const lc =
417 &cpi->svc.layer_context[cpi->svc.spatial_layer_id *
418 cpi->svc.number_temporal_layers];
419 ++lc->current_video_frame_in_layer;
420 ++lc->frames_from_key_frame;
421 if (cpi->svc.spatial_layer_id == cpi->svc.number_spatial_layers - 1)
422 ++cpi->svc.current_superframe;
423 }
424
get_layer_resolution(const int width_org,const int height_org,const int num,const int den,int * width_out,int * height_out)425 void get_layer_resolution(const int width_org, const int height_org,
426 const int num, const int den, int *width_out,
427 int *height_out) {
428 int w, h;
429
430 if (width_out == NULL || height_out == NULL || den == 0) return;
431
432 w = width_org * num / den;
433 h = height_org * num / den;
434
435 // make height and width even to make chrome player happy
436 w += w % 2;
437 h += h % 2;
438
439 *width_out = w;
440 *height_out = h;
441 }
442
reset_fb_idx_unused(VP9_COMP * const cpi)443 static void reset_fb_idx_unused(VP9_COMP *const cpi) {
444 // If a reference frame is not referenced or refreshed, then set the
445 // fb_idx for that reference to the first one used/referenced.
446 // This is to avoid setting fb_idx for a reference to a slot that is not
447 // used/needed (i.e., since that reference is not referenced or refreshed).
448 static const int flag_list[4] = { 0, VP9_LAST_FLAG, VP9_GOLD_FLAG,
449 VP9_ALT_FLAG };
450 MV_REFERENCE_FRAME ref_frame;
451 MV_REFERENCE_FRAME first_ref = 0;
452 int first_fb_idx = 0;
453 int fb_idx[3] = { cpi->lst_fb_idx, cpi->gld_fb_idx, cpi->alt_fb_idx };
454 for (ref_frame = LAST_FRAME; ref_frame <= ALTREF_FRAME; ref_frame++) {
455 if (cpi->ref_frame_flags & flag_list[ref_frame]) {
456 first_ref = ref_frame;
457 first_fb_idx = fb_idx[ref_frame - 1];
458 break;
459 }
460 }
461 if (first_ref > 0) {
462 if (first_ref != LAST_FRAME &&
463 !(cpi->ref_frame_flags & flag_list[LAST_FRAME]) &&
464 !cpi->ext_refresh_last_frame)
465 cpi->lst_fb_idx = first_fb_idx;
466 else if (first_ref != GOLDEN_FRAME &&
467 !(cpi->ref_frame_flags & flag_list[GOLDEN_FRAME]) &&
468 !cpi->ext_refresh_golden_frame)
469 cpi->gld_fb_idx = first_fb_idx;
470 else if (first_ref != ALTREF_FRAME &&
471 !(cpi->ref_frame_flags & flag_list[ALTREF_FRAME]) &&
472 !cpi->ext_refresh_alt_ref_frame)
473 cpi->alt_fb_idx = first_fb_idx;
474 }
475 }
476
477 // The function sets proper ref_frame_flags, buffer indices, and buffer update
478 // variables for temporal layering mode 3 - that does 0-2-1-2 temporal layering
479 // scheme.
set_flags_and_fb_idx_for_temporal_mode3(VP9_COMP * const cpi)480 static void set_flags_and_fb_idx_for_temporal_mode3(VP9_COMP *const cpi) {
481 int frame_num_within_temporal_struct = 0;
482 int spatial_id, temporal_id;
483 spatial_id = cpi->svc.spatial_layer_id = cpi->svc.spatial_layer_to_encode;
484 frame_num_within_temporal_struct =
485 cpi->svc
486 .layer_context[cpi->svc.spatial_layer_id *
487 cpi->svc.number_temporal_layers]
488 .current_video_frame_in_layer %
489 4;
490 temporal_id = cpi->svc.temporal_layer_id =
491 (frame_num_within_temporal_struct & 1)
492 ? 2
493 : (frame_num_within_temporal_struct >> 1);
494 cpi->ext_refresh_last_frame = cpi->ext_refresh_golden_frame =
495 cpi->ext_refresh_alt_ref_frame = 0;
496 if (!temporal_id) {
497 cpi->ext_refresh_frame_flags_pending = 1;
498 cpi->ext_refresh_last_frame = 1;
499 if (!spatial_id) {
500 cpi->ref_frame_flags = VP9_LAST_FLAG;
501 } else if (cpi->svc.layer_context[temporal_id].is_key_frame) {
502 // base layer is a key frame.
503 cpi->ref_frame_flags = VP9_LAST_FLAG;
504 cpi->ext_refresh_last_frame = 0;
505 cpi->ext_refresh_golden_frame = 1;
506 } else {
507 cpi->ref_frame_flags = VP9_LAST_FLAG | VP9_GOLD_FLAG;
508 }
509 } else if (temporal_id == 1) {
510 cpi->ext_refresh_frame_flags_pending = 1;
511 cpi->ext_refresh_alt_ref_frame = 1;
512 if (!spatial_id) {
513 cpi->ref_frame_flags = VP9_LAST_FLAG;
514 } else {
515 cpi->ref_frame_flags = VP9_LAST_FLAG | VP9_GOLD_FLAG;
516 }
517 } else {
518 if (frame_num_within_temporal_struct == 1) {
519 // the first tl2 picture
520 if (spatial_id == cpi->svc.number_spatial_layers - 1) { // top layer
521 cpi->ext_refresh_frame_flags_pending = 1;
522 if (!spatial_id)
523 cpi->ref_frame_flags = VP9_LAST_FLAG;
524 else
525 cpi->ref_frame_flags = VP9_LAST_FLAG | VP9_GOLD_FLAG;
526 } else if (!spatial_id) {
527 cpi->ext_refresh_frame_flags_pending = 1;
528 cpi->ext_refresh_alt_ref_frame = 1;
529 cpi->ref_frame_flags = VP9_LAST_FLAG;
530 } else if (spatial_id < cpi->svc.number_spatial_layers - 1) {
531 cpi->ext_refresh_frame_flags_pending = 1;
532 cpi->ext_refresh_alt_ref_frame = 1;
533 cpi->ref_frame_flags = VP9_LAST_FLAG | VP9_GOLD_FLAG;
534 }
535 } else {
536 // The second tl2 picture
537 if (spatial_id == cpi->svc.number_spatial_layers - 1) { // top layer
538 cpi->ext_refresh_frame_flags_pending = 1;
539 if (!spatial_id)
540 cpi->ref_frame_flags = VP9_LAST_FLAG;
541 else
542 cpi->ref_frame_flags = VP9_LAST_FLAG | VP9_GOLD_FLAG;
543 } else if (!spatial_id) {
544 cpi->ext_refresh_frame_flags_pending = 1;
545 cpi->ref_frame_flags = VP9_LAST_FLAG;
546 cpi->ext_refresh_alt_ref_frame = 1;
547 } else { // top layer
548 cpi->ext_refresh_frame_flags_pending = 1;
549 cpi->ref_frame_flags = VP9_LAST_FLAG | VP9_GOLD_FLAG;
550 cpi->ext_refresh_alt_ref_frame = 1;
551 }
552 }
553 }
554 if (temporal_id == 0) {
555 cpi->lst_fb_idx = spatial_id;
556 if (spatial_id) {
557 if (cpi->svc.layer_context[temporal_id].is_key_frame) {
558 cpi->lst_fb_idx = spatial_id - 1;
559 cpi->gld_fb_idx = spatial_id;
560 } else {
561 cpi->gld_fb_idx = spatial_id - 1;
562 }
563 } else {
564 cpi->gld_fb_idx = 0;
565 }
566 cpi->alt_fb_idx = 0;
567 } else if (temporal_id == 1) {
568 cpi->lst_fb_idx = spatial_id;
569 cpi->gld_fb_idx = cpi->svc.number_spatial_layers + spatial_id - 1;
570 cpi->alt_fb_idx = cpi->svc.number_spatial_layers + spatial_id;
571 } else if (frame_num_within_temporal_struct == 1) {
572 cpi->lst_fb_idx = spatial_id;
573 cpi->gld_fb_idx = cpi->svc.number_spatial_layers + spatial_id - 1;
574 cpi->alt_fb_idx = cpi->svc.number_spatial_layers + spatial_id;
575 } else {
576 cpi->lst_fb_idx = cpi->svc.number_spatial_layers + spatial_id;
577 cpi->gld_fb_idx = cpi->svc.number_spatial_layers + spatial_id - 1;
578 cpi->alt_fb_idx = cpi->svc.number_spatial_layers + spatial_id;
579 }
580
581 reset_fb_idx_unused(cpi);
582 }
583
584 // The function sets proper ref_frame_flags, buffer indices, and buffer update
585 // variables for temporal layering mode 2 - that does 0-1-0-1 temporal layering
586 // scheme.
set_flags_and_fb_idx_for_temporal_mode2(VP9_COMP * const cpi)587 static void set_flags_and_fb_idx_for_temporal_mode2(VP9_COMP *const cpi) {
588 int spatial_id, temporal_id;
589 spatial_id = cpi->svc.spatial_layer_id = cpi->svc.spatial_layer_to_encode;
590 temporal_id = cpi->svc.temporal_layer_id =
591 cpi->svc
592 .layer_context[cpi->svc.spatial_layer_id *
593 cpi->svc.number_temporal_layers]
594 .current_video_frame_in_layer &
595 1;
596 cpi->ext_refresh_last_frame = cpi->ext_refresh_golden_frame =
597 cpi->ext_refresh_alt_ref_frame = 0;
598 if (!temporal_id) {
599 cpi->ext_refresh_frame_flags_pending = 1;
600 cpi->ext_refresh_last_frame = 1;
601 if (!spatial_id) {
602 cpi->ref_frame_flags = VP9_LAST_FLAG;
603 } else if (cpi->svc.layer_context[temporal_id].is_key_frame) {
604 // base layer is a key frame.
605 cpi->ref_frame_flags = VP9_LAST_FLAG;
606 cpi->ext_refresh_last_frame = 0;
607 cpi->ext_refresh_golden_frame = 1;
608 } else {
609 cpi->ref_frame_flags = VP9_LAST_FLAG | VP9_GOLD_FLAG;
610 }
611 } else if (temporal_id == 1) {
612 cpi->ext_refresh_frame_flags_pending = 1;
613 cpi->ext_refresh_alt_ref_frame = 1;
614 if (!spatial_id) {
615 cpi->ref_frame_flags = VP9_LAST_FLAG;
616 } else {
617 if (spatial_id == cpi->svc.number_spatial_layers - 1)
618 cpi->ext_refresh_alt_ref_frame = 0;
619 cpi->ref_frame_flags = VP9_LAST_FLAG | VP9_GOLD_FLAG;
620 }
621 }
622
623 if (temporal_id == 0) {
624 cpi->lst_fb_idx = spatial_id;
625 if (spatial_id) {
626 if (cpi->svc.layer_context[temporal_id].is_key_frame) {
627 cpi->lst_fb_idx = spatial_id - 1;
628 cpi->gld_fb_idx = spatial_id;
629 } else {
630 cpi->gld_fb_idx = spatial_id - 1;
631 }
632 } else {
633 cpi->gld_fb_idx = 0;
634 }
635 cpi->alt_fb_idx = 0;
636 } else if (temporal_id == 1) {
637 cpi->lst_fb_idx = spatial_id;
638 cpi->gld_fb_idx = cpi->svc.number_spatial_layers + spatial_id - 1;
639 cpi->alt_fb_idx = cpi->svc.number_spatial_layers + spatial_id;
640 }
641
642 reset_fb_idx_unused(cpi);
643 }
644
645 // The function sets proper ref_frame_flags, buffer indices, and buffer update
646 // variables for temporal layering mode 0 - that has no temporal layering.
set_flags_and_fb_idx_for_temporal_mode_noLayering(VP9_COMP * const cpi)647 static void set_flags_and_fb_idx_for_temporal_mode_noLayering(
648 VP9_COMP *const cpi) {
649 int spatial_id;
650 spatial_id = cpi->svc.spatial_layer_id = cpi->svc.spatial_layer_to_encode;
651 cpi->ext_refresh_last_frame = cpi->ext_refresh_golden_frame =
652 cpi->ext_refresh_alt_ref_frame = 0;
653 cpi->ext_refresh_frame_flags_pending = 1;
654 cpi->ext_refresh_last_frame = 1;
655 if (!spatial_id) {
656 cpi->ref_frame_flags = VP9_LAST_FLAG;
657 } else if (cpi->svc.layer_context[0].is_key_frame) {
658 cpi->ref_frame_flags = VP9_LAST_FLAG;
659 cpi->ext_refresh_last_frame = 0;
660 cpi->ext_refresh_golden_frame = 1;
661 } else {
662 cpi->ref_frame_flags = VP9_LAST_FLAG | VP9_GOLD_FLAG;
663 }
664 cpi->lst_fb_idx = spatial_id;
665 if (spatial_id) {
666 if (cpi->svc.layer_context[0].is_key_frame) {
667 cpi->lst_fb_idx = spatial_id - 1;
668 cpi->gld_fb_idx = spatial_id;
669 } else {
670 cpi->gld_fb_idx = spatial_id - 1;
671 }
672 } else {
673 cpi->gld_fb_idx = 0;
674 }
675
676 reset_fb_idx_unused(cpi);
677 }
678
set_flags_and_fb_idx_bypass_via_set_ref_frame_config(VP9_COMP * const cpi)679 static void set_flags_and_fb_idx_bypass_via_set_ref_frame_config(
680 VP9_COMP *const cpi) {
681 SVC *const svc = &cpi->svc;
682 int sl = svc->spatial_layer_id = svc->spatial_layer_to_encode;
683 cpi->svc.temporal_layer_id = cpi->svc.temporal_layer_id_per_spatial[sl];
684 cpi->ext_refresh_frame_flags_pending = 1;
685 cpi->lst_fb_idx = svc->lst_fb_idx[sl];
686 cpi->gld_fb_idx = svc->gld_fb_idx[sl];
687 cpi->alt_fb_idx = svc->alt_fb_idx[sl];
688 cpi->ext_refresh_last_frame = 0;
689 cpi->ext_refresh_golden_frame = 0;
690 cpi->ext_refresh_alt_ref_frame = 0;
691 cpi->ref_frame_flags = 0;
692 if (svc->reference_last[sl]) cpi->ref_frame_flags |= VP9_LAST_FLAG;
693 if (svc->reference_golden[sl]) cpi->ref_frame_flags |= VP9_GOLD_FLAG;
694 if (svc->reference_altref[sl]) cpi->ref_frame_flags |= VP9_ALT_FLAG;
695 }
696
vp9_copy_flags_ref_update_idx(VP9_COMP * const cpi)697 void vp9_copy_flags_ref_update_idx(VP9_COMP *const cpi) {
698 SVC *const svc = &cpi->svc;
699 static const int flag_list[4] = { 0, VP9_LAST_FLAG, VP9_GOLD_FLAG,
700 VP9_ALT_FLAG };
701 int sl = svc->spatial_layer_id;
702 svc->lst_fb_idx[sl] = cpi->lst_fb_idx;
703 svc->gld_fb_idx[sl] = cpi->gld_fb_idx;
704 svc->alt_fb_idx[sl] = cpi->alt_fb_idx;
705 // For the fixed SVC mode: pass the refresh_lst/gld/alt_frame flags to the
706 // update_buffer_slot, this is needed for the GET_SVC_REF_FRAME_CONFIG api.
707 if (svc->temporal_layering_mode != VP9E_TEMPORAL_LAYERING_MODE_BYPASS) {
708 int ref;
709 for (ref = 0; ref < REF_FRAMES; ++ref) {
710 svc->update_buffer_slot[sl] &= ~(1 << ref);
711 if ((ref == svc->lst_fb_idx[sl] && cpi->refresh_last_frame) ||
712 (ref == svc->gld_fb_idx[sl] && cpi->refresh_golden_frame) ||
713 (ref == svc->alt_fb_idx[sl] && cpi->refresh_alt_ref_frame))
714 svc->update_buffer_slot[sl] |= (1 << ref);
715 }
716 }
717 // TODO(jianj): Remove these 3, deprecated.
718 svc->update_last[sl] = (uint8_t)cpi->refresh_last_frame;
719 svc->update_golden[sl] = (uint8_t)cpi->refresh_golden_frame;
720 svc->update_altref[sl] = (uint8_t)cpi->refresh_alt_ref_frame;
721
722 svc->reference_last[sl] =
723 (uint8_t)(cpi->ref_frame_flags & flag_list[LAST_FRAME]);
724 svc->reference_golden[sl] =
725 (uint8_t)(cpi->ref_frame_flags & flag_list[GOLDEN_FRAME]);
726 svc->reference_altref[sl] =
727 (uint8_t)(cpi->ref_frame_flags & flag_list[ALTREF_FRAME]);
728 }
729
vp9_one_pass_cbr_svc_start_layer(VP9_COMP * const cpi)730 int vp9_one_pass_cbr_svc_start_layer(VP9_COMP *const cpi) {
731 int width = 0, height = 0;
732 SVC *const svc = &cpi->svc;
733 LAYER_CONTEXT *lc = NULL;
734 svc->skip_enhancement_layer = 0;
735 if (svc->number_spatial_layers > 1) {
736 svc->use_base_mv = 1;
737 svc->use_partition_reuse = 1;
738 }
739 svc->force_zero_mode_spatial_ref = 1;
740 svc->mi_stride[svc->spatial_layer_id] = cpi->common.mi_stride;
741 svc->mi_rows[svc->spatial_layer_id] = cpi->common.mi_rows;
742 svc->mi_cols[svc->spatial_layer_id] = cpi->common.mi_cols;
743
744 if (svc->temporal_layering_mode == VP9E_TEMPORAL_LAYERING_MODE_0212) {
745 set_flags_and_fb_idx_for_temporal_mode3(cpi);
746 } else if (svc->temporal_layering_mode ==
747 VP9E_TEMPORAL_LAYERING_MODE_NOLAYERING) {
748 set_flags_and_fb_idx_for_temporal_mode_noLayering(cpi);
749 } else if (svc->temporal_layering_mode == VP9E_TEMPORAL_LAYERING_MODE_0101) {
750 set_flags_and_fb_idx_for_temporal_mode2(cpi);
751 } else if (svc->temporal_layering_mode ==
752 VP9E_TEMPORAL_LAYERING_MODE_BYPASS &&
753 svc->use_set_ref_frame_config) {
754 set_flags_and_fb_idx_bypass_via_set_ref_frame_config(cpi);
755 }
756
757 if (cpi->lst_fb_idx == svc->buffer_gf_temporal_ref[0].idx ||
758 cpi->gld_fb_idx == svc->buffer_gf_temporal_ref[0].idx ||
759 cpi->alt_fb_idx == svc->buffer_gf_temporal_ref[0].idx)
760 svc->buffer_gf_temporal_ref[0].is_used = 1;
761 if (cpi->lst_fb_idx == svc->buffer_gf_temporal_ref[1].idx ||
762 cpi->gld_fb_idx == svc->buffer_gf_temporal_ref[1].idx ||
763 cpi->alt_fb_idx == svc->buffer_gf_temporal_ref[1].idx)
764 svc->buffer_gf_temporal_ref[1].is_used = 1;
765
766 // For the fixed (non-flexible/bypass) SVC mode:
767 // If long term temporal reference is enabled at the sequence level
768 // (use_gf_temporal_ref == 1), and inter_layer is disabled (on inter-frames),
769 // we can use golden as a second temporal reference
770 // (since the spatial/inter-layer reference is disabled).
771 // We check that the fb_idx for this reference (buffer_gf_temporal_ref.idx) is
772 // unused (slot 7 and 6 should be available for 3-3 layer system).
773 // For now usage of this second temporal reference will only be used for
774 // highest and next to highest spatial layer (i.e., top and middle layer for
775 // 3 spatial layers).
776 svc->use_gf_temporal_ref_current_layer = 0;
777 if (svc->use_gf_temporal_ref && !svc->buffer_gf_temporal_ref[0].is_used &&
778 !svc->buffer_gf_temporal_ref[1].is_used &&
779 svc->temporal_layering_mode != VP9E_TEMPORAL_LAYERING_MODE_BYPASS &&
780 svc->disable_inter_layer_pred != INTER_LAYER_PRED_ON &&
781 svc->number_spatial_layers <= 3 && svc->number_temporal_layers <= 3 &&
782 svc->spatial_layer_id >= svc->number_spatial_layers - 2) {
783 // Enable the second (long-term) temporal reference at the frame-level.
784 svc->use_gf_temporal_ref_current_layer = 1;
785 }
786
787 // Check if current superframe has any layer sync, only check once on
788 // base layer.
789 if (svc->spatial_layer_id == 0) {
790 int sl = 0;
791 // Default is no sync.
792 svc->superframe_has_layer_sync = 0;
793 for (sl = 0; sl < svc->number_spatial_layers; ++sl) {
794 if (cpi->svc.spatial_layer_sync[sl]) svc->superframe_has_layer_sync = 1;
795 }
796 }
797
798 // Reset the drop flags for all spatial layers, on the base layer.
799 if (svc->spatial_layer_id == 0) {
800 vp9_zero(svc->drop_spatial_layer);
801 // TODO(jianj/marpan): Investigate why setting svc->lst/gld/alt_fb_idx
802 // causes an issue with frame dropping and temporal layers, when the frame
803 // flags are passed via the encode call (bypass mode). Issue is that we're
804 // resetting ext_refresh_frame_flags_pending to 0 on frame drops.
805 if (svc->temporal_layering_mode != VP9E_TEMPORAL_LAYERING_MODE_BYPASS) {
806 memset(&svc->lst_fb_idx, -1, sizeof(svc->lst_fb_idx));
807 memset(&svc->gld_fb_idx, -1, sizeof(svc->lst_fb_idx));
808 memset(&svc->alt_fb_idx, -1, sizeof(svc->lst_fb_idx));
809 // These are set by API before the superframe is encoded and they are
810 // passed to encoder layer by layer. Don't reset them on layer 0 in bypass
811 // mode.
812 vp9_zero(svc->update_buffer_slot);
813 vp9_zero(svc->reference_last);
814 vp9_zero(svc->reference_golden);
815 vp9_zero(svc->reference_altref);
816 // TODO(jianj): Remove these 3, deprecated.
817 vp9_zero(svc->update_last);
818 vp9_zero(svc->update_golden);
819 vp9_zero(svc->update_altref);
820 }
821 }
822
823 lc = &svc->layer_context[svc->spatial_layer_id * svc->number_temporal_layers +
824 svc->temporal_layer_id];
825
826 // Setting the worst/best_quality via the encoder control: SET_SVC_PARAMETERS,
827 // only for non-BYPASS mode for now.
828 if (svc->temporal_layering_mode != VP9E_TEMPORAL_LAYERING_MODE_BYPASS ||
829 svc->use_set_ref_frame_config) {
830 RATE_CONTROL *const lrc = &lc->rc;
831 lrc->worst_quality = vp9_quantizer_to_qindex(lc->max_q);
832 lrc->best_quality = vp9_quantizer_to_qindex(lc->min_q);
833 }
834
835 get_layer_resolution(cpi->oxcf.width, cpi->oxcf.height,
836 lc->scaling_factor_num, lc->scaling_factor_den, &width,
837 &height);
838
839 // Use Eightap_smooth for low resolutions.
840 if (width * height <= 320 * 240)
841 svc->downsample_filter_type[svc->spatial_layer_id] = EIGHTTAP_SMOOTH;
842 // For scale factors > 0.75, set the phase to 0 (aligns decimated pixel
843 // to source pixel).
844 lc = &svc->layer_context[svc->spatial_layer_id * svc->number_temporal_layers +
845 svc->temporal_layer_id];
846 if (lc->scaling_factor_num > (3 * lc->scaling_factor_den) >> 2)
847 svc->downsample_filter_phase[svc->spatial_layer_id] = 0;
848
849 // The usage of use_base_mv or partition_reuse assumes down-scale of 2x2.
850 // For now, turn off use of base motion vectors and partition reuse if the
851 // spatial scale factors for any layers are not 2,
852 // keep the case of 3 spatial layers with scale factor of 4x4 for base layer.
853 // TODO(marpan): Fix this to allow for use_base_mv for scale factors != 2.
854 if (svc->number_spatial_layers > 1) {
855 int sl;
856 for (sl = 0; sl < svc->number_spatial_layers - 1; ++sl) {
857 lc = &svc->layer_context[sl * svc->number_temporal_layers +
858 svc->temporal_layer_id];
859 if ((lc->scaling_factor_num != lc->scaling_factor_den >> 1) &&
860 !(lc->scaling_factor_num == lc->scaling_factor_den >> 2 && sl == 0 &&
861 svc->number_spatial_layers == 3)) {
862 svc->use_base_mv = 0;
863 svc->use_partition_reuse = 0;
864 break;
865 }
866 }
867 // For non-zero spatial layers: if the previous spatial layer was dropped
868 // disable the base_mv and partition_reuse features.
869 if (svc->spatial_layer_id > 0 &&
870 svc->drop_spatial_layer[svc->spatial_layer_id - 1]) {
871 svc->use_base_mv = 0;
872 svc->use_partition_reuse = 0;
873 }
874 }
875
876 svc->non_reference_frame = 0;
877 if (cpi->common.frame_type != KEY_FRAME && !cpi->ext_refresh_last_frame &&
878 !cpi->ext_refresh_golden_frame && !cpi->ext_refresh_alt_ref_frame)
879 svc->non_reference_frame = 1;
880 // For non-flexible mode, where update_buffer_slot is used, need to check if
881 // all buffer slots are not refreshed.
882 if (svc->temporal_layering_mode == VP9E_TEMPORAL_LAYERING_MODE_BYPASS) {
883 if (svc->update_buffer_slot[svc->spatial_layer_id] != 0)
884 svc->non_reference_frame = 0;
885 }
886
887 if (svc->spatial_layer_id == 0) {
888 svc->high_source_sad_superframe = 0;
889 svc->high_num_blocks_with_motion = 0;
890 }
891
892 if (svc->temporal_layering_mode != VP9E_TEMPORAL_LAYERING_MODE_BYPASS &&
893 svc->last_layer_dropped[svc->spatial_layer_id] &&
894 svc->fb_idx_upd_tl0[svc->spatial_layer_id] != -1 &&
895 !svc->layer_context[svc->temporal_layer_id].is_key_frame) {
896 // For fixed/non-flexible mode, if the previous frame (same spatial layer
897 // from previous superframe) was dropped, make sure the lst_fb_idx
898 // for this frame corresponds to the buffer index updated on (last) encoded
899 // TL0 frame (with same spatial layer).
900 cpi->lst_fb_idx = svc->fb_idx_upd_tl0[svc->spatial_layer_id];
901 }
902
903 if (vp9_set_size_literal(cpi, width, height) != 0)
904 return VPX_CODEC_INVALID_PARAM;
905
906 return 0;
907 }
908
vp9_svc_lookahead_pop(VP9_COMP * const cpi,struct lookahead_ctx * ctx,int drain)909 struct lookahead_entry *vp9_svc_lookahead_pop(VP9_COMP *const cpi,
910 struct lookahead_ctx *ctx,
911 int drain) {
912 struct lookahead_entry *buf = NULL;
913 if (ctx->sz && (drain || ctx->sz == ctx->max_sz - MAX_PRE_FRAMES)) {
914 buf = vp9_lookahead_peek(ctx, 0);
915 if (buf != NULL) {
916 // Only remove the buffer when pop the highest layer.
917 if (cpi->svc.spatial_layer_id == cpi->svc.number_spatial_layers - 1) {
918 vp9_lookahead_pop(ctx, drain);
919 }
920 }
921 }
922 return buf;
923 }
924
vp9_free_svc_cyclic_refresh(VP9_COMP * const cpi)925 void vp9_free_svc_cyclic_refresh(VP9_COMP *const cpi) {
926 int sl, tl;
927 SVC *const svc = &cpi->svc;
928 const VP9EncoderConfig *const oxcf = &cpi->oxcf;
929 for (sl = 0; sl < oxcf->ss_number_layers; ++sl) {
930 for (tl = 0; tl < oxcf->ts_number_layers; ++tl) {
931 int layer = LAYER_IDS_TO_IDX(sl, tl, oxcf->ts_number_layers);
932 LAYER_CONTEXT *const lc = &svc->layer_context[layer];
933 if (lc->map) vpx_free(lc->map);
934 if (lc->last_coded_q_map) vpx_free(lc->last_coded_q_map);
935 if (lc->consec_zero_mv) vpx_free(lc->consec_zero_mv);
936 }
937 }
938 }
939
940 // Reset on key frame: reset counters, references and buffer updates.
vp9_svc_reset_temporal_layers(VP9_COMP * const cpi,int is_key)941 void vp9_svc_reset_temporal_layers(VP9_COMP *const cpi, int is_key) {
942 int sl, tl;
943 SVC *const svc = &cpi->svc;
944 LAYER_CONTEXT *lc = NULL;
945 for (sl = 0; sl < svc->number_spatial_layers; ++sl) {
946 for (tl = 0; tl < svc->number_temporal_layers; ++tl) {
947 lc = &cpi->svc.layer_context[sl * svc->number_temporal_layers + tl];
948 lc->current_video_frame_in_layer = 0;
949 if (is_key) lc->frames_from_key_frame = 0;
950 }
951 }
952 if (svc->temporal_layering_mode == VP9E_TEMPORAL_LAYERING_MODE_0212) {
953 set_flags_and_fb_idx_for_temporal_mode3(cpi);
954 } else if (svc->temporal_layering_mode ==
955 VP9E_TEMPORAL_LAYERING_MODE_NOLAYERING) {
956 set_flags_and_fb_idx_for_temporal_mode_noLayering(cpi);
957 } else if (svc->temporal_layering_mode == VP9E_TEMPORAL_LAYERING_MODE_0101) {
958 set_flags_and_fb_idx_for_temporal_mode2(cpi);
959 }
960 vp9_update_temporal_layer_framerate(cpi);
961 vp9_restore_layer_context(cpi);
962 }
963
vp9_svc_check_reset_layer_rc_flag(VP9_COMP * const cpi)964 void vp9_svc_check_reset_layer_rc_flag(VP9_COMP *const cpi) {
965 SVC *svc = &cpi->svc;
966 int sl, tl;
967 for (sl = 0; sl < svc->number_spatial_layers; ++sl) {
968 // Check for reset based on avg_frame_bandwidth for spatial layer sl.
969 int layer = LAYER_IDS_TO_IDX(sl, svc->number_temporal_layers - 1,
970 svc->number_temporal_layers);
971 LAYER_CONTEXT *lc = &svc->layer_context[layer];
972 RATE_CONTROL *lrc = &lc->rc;
973 if (lrc->avg_frame_bandwidth > (3 * lrc->last_avg_frame_bandwidth >> 1) ||
974 lrc->avg_frame_bandwidth < (lrc->last_avg_frame_bandwidth >> 1)) {
975 // Reset for all temporal layers with spatial layer sl.
976 for (tl = 0; tl < svc->number_temporal_layers; ++tl) {
977 int layer = LAYER_IDS_TO_IDX(sl, tl, svc->number_temporal_layers);
978 LAYER_CONTEXT *lc = &svc->layer_context[layer];
979 RATE_CONTROL *lrc = &lc->rc;
980 lrc->rc_1_frame = 0;
981 lrc->rc_2_frame = 0;
982 lrc->bits_off_target = lrc->optimal_buffer_level;
983 lrc->buffer_level = lrc->optimal_buffer_level;
984 }
985 }
986 }
987 }
988
vp9_svc_constrain_inter_layer_pred(VP9_COMP * const cpi)989 void vp9_svc_constrain_inter_layer_pred(VP9_COMP *const cpi) {
990 VP9_COMMON *const cm = &cpi->common;
991 SVC *const svc = &cpi->svc;
992 // Check for disabling inter-layer (spatial) prediction, if
993 // svc.disable_inter_layer_pred is set. If the previous spatial layer was
994 // dropped then disable the prediction from this (scaled) reference.
995 // For INTER_LAYER_PRED_OFF_NONKEY: inter-layer prediction is disabled
996 // on key frames or if any spatial layer is a sync layer.
997 if ((svc->disable_inter_layer_pred == INTER_LAYER_PRED_OFF_NONKEY &&
998 !svc->layer_context[svc->temporal_layer_id].is_key_frame &&
999 !svc->superframe_has_layer_sync) ||
1000 svc->disable_inter_layer_pred == INTER_LAYER_PRED_OFF ||
1001 svc->drop_spatial_layer[svc->spatial_layer_id - 1]) {
1002 MV_REFERENCE_FRAME ref_frame;
1003 static const int flag_list[4] = { 0, VP9_LAST_FLAG, VP9_GOLD_FLAG,
1004 VP9_ALT_FLAG };
1005 for (ref_frame = LAST_FRAME; ref_frame <= ALTREF_FRAME; ++ref_frame) {
1006 const YV12_BUFFER_CONFIG *yv12 = get_ref_frame_buffer(cpi, ref_frame);
1007 if (yv12 != NULL && (cpi->ref_frame_flags & flag_list[ref_frame])) {
1008 const struct scale_factors *const scale_fac =
1009 &cm->frame_refs[ref_frame - 1].sf;
1010 if (vp9_is_scaled(scale_fac))
1011 cpi->ref_frame_flags &= (~flag_list[ref_frame]);
1012 }
1013 }
1014 }
1015 // For fixed/non-flexible SVC: check for disabling inter-layer prediction.
1016 // If the reference for inter-layer prediction (the reference that is scaled)
1017 // is not the previous spatial layer from the same superframe, then we disable
1018 // inter-layer prediction. Only need to check when inter_layer prediction is
1019 // not set to OFF mode.
1020 if (svc->temporal_layering_mode != VP9E_TEMPORAL_LAYERING_MODE_BYPASS &&
1021 svc->disable_inter_layer_pred != INTER_LAYER_PRED_OFF) {
1022 // We only use LAST and GOLDEN for prediction in real-time mode, so we
1023 // check both here.
1024 MV_REFERENCE_FRAME ref_frame;
1025 for (ref_frame = LAST_FRAME; ref_frame <= GOLDEN_FRAME; ref_frame++) {
1026 struct scale_factors *scale_fac = &cm->frame_refs[ref_frame - 1].sf;
1027 if (vp9_is_scaled(scale_fac)) {
1028 // If this reference was updated on the previous spatial layer of the
1029 // current superframe, then we keep this reference (don't disable).
1030 // Otherwise we disable the inter-layer prediction.
1031 // This condition is verified by checking if the current frame buffer
1032 // index is equal to any of the slots for the previous spatial layer,
1033 // and if so, check if that slot was updated/refreshed. If that is the
1034 // case, then this reference is valid for inter-layer prediction under
1035 // the mode INTER_LAYER_PRED_ON_CONSTRAINED.
1036 int fb_idx =
1037 ref_frame == LAST_FRAME ? cpi->lst_fb_idx : cpi->gld_fb_idx;
1038 int ref_flag = ref_frame == LAST_FRAME ? VP9_LAST_FLAG : VP9_GOLD_FLAG;
1039 int sl = svc->spatial_layer_id;
1040 int disable = 1;
1041 if (fb_idx < 0) continue;
1042 if ((fb_idx == svc->lst_fb_idx[sl - 1] &&
1043 (svc->update_buffer_slot[sl - 1] & (1 << fb_idx))) ||
1044 (fb_idx == svc->gld_fb_idx[sl - 1] &&
1045 (svc->update_buffer_slot[sl - 1] & (1 << fb_idx))) ||
1046 (fb_idx == svc->alt_fb_idx[sl - 1] &&
1047 (svc->update_buffer_slot[sl - 1] & (1 << fb_idx))))
1048 disable = 0;
1049 if (disable) cpi->ref_frame_flags &= (~ref_flag);
1050 }
1051 }
1052 }
1053 }
1054
vp9_svc_assert_constraints_pattern(VP9_COMP * const cpi)1055 void vp9_svc_assert_constraints_pattern(VP9_COMP *const cpi) {
1056 SVC *const svc = &cpi->svc;
1057 // For fixed/non-flexible mode, the following constraint are expected,
1058 // when inter-layer prediciton is on (default).
1059 if (svc->temporal_layering_mode != VP9E_TEMPORAL_LAYERING_MODE_BYPASS &&
1060 svc->disable_inter_layer_pred == INTER_LAYER_PRED_ON &&
1061 svc->framedrop_mode != LAYER_DROP) {
1062 if (!svc->layer_context[svc->temporal_layer_id].is_key_frame) {
1063 // On non-key frames: LAST is always temporal reference, GOLDEN is
1064 // spatial reference.
1065 if (svc->temporal_layer_id == 0)
1066 // Base temporal only predicts from base temporal.
1067 assert(svc->fb_idx_temporal_layer_id[cpi->lst_fb_idx] == 0);
1068 else
1069 // Non-base temporal only predicts from lower temporal layer.
1070 assert(svc->fb_idx_temporal_layer_id[cpi->lst_fb_idx] <
1071 svc->temporal_layer_id);
1072 if (svc->spatial_layer_id > 0 && cpi->ref_frame_flags & VP9_GOLD_FLAG &&
1073 svc->spatial_layer_id > svc->first_spatial_layer_to_encode) {
1074 // Non-base spatial only predicts from lower spatial layer with same
1075 // temporal_id.
1076 assert(svc->fb_idx_spatial_layer_id[cpi->gld_fb_idx] ==
1077 svc->spatial_layer_id - 1);
1078 assert(svc->fb_idx_temporal_layer_id[cpi->gld_fb_idx] ==
1079 svc->temporal_layer_id);
1080 }
1081 } else if (svc->spatial_layer_id > 0 &&
1082 svc->spatial_layer_id > svc->first_spatial_layer_to_encode) {
1083 // Only 1 reference for frame whose base is key; reference may be LAST
1084 // or GOLDEN, so we check both.
1085 if (cpi->ref_frame_flags & VP9_LAST_FLAG) {
1086 assert(svc->fb_idx_spatial_layer_id[cpi->lst_fb_idx] ==
1087 svc->spatial_layer_id - 1);
1088 assert(svc->fb_idx_temporal_layer_id[cpi->lst_fb_idx] ==
1089 svc->temporal_layer_id);
1090 } else if (cpi->ref_frame_flags & VP9_GOLD_FLAG) {
1091 assert(svc->fb_idx_spatial_layer_id[cpi->gld_fb_idx] ==
1092 svc->spatial_layer_id - 1);
1093 assert(svc->fb_idx_temporal_layer_id[cpi->gld_fb_idx] ==
1094 svc->temporal_layer_id);
1095 }
1096 }
1097 } else if (svc->use_gf_temporal_ref_current_layer &&
1098 !svc->layer_context[svc->temporal_layer_id].is_key_frame) {
1099 // For the usage of golden as second long term reference: the
1100 // temporal_layer_id of that reference must be base temporal layer 0, and
1101 // spatial_layer_id of that reference must be same as current
1102 // spatial_layer_id. If not, disable feature.
1103 // TODO(marpan): Investigate when this can happen, and maybe put this check
1104 // and reset in a different place.
1105 if (svc->fb_idx_spatial_layer_id[cpi->gld_fb_idx] !=
1106 svc->spatial_layer_id ||
1107 svc->fb_idx_temporal_layer_id[cpi->gld_fb_idx] != 0)
1108 svc->use_gf_temporal_ref_current_layer = 0;
1109 }
1110 }
1111
1112 #if CONFIG_VP9_TEMPORAL_DENOISING
vp9_denoise_svc_non_key(VP9_COMP * const cpi)1113 int vp9_denoise_svc_non_key(VP9_COMP *const cpi) {
1114 int layer =
1115 LAYER_IDS_TO_IDX(cpi->svc.spatial_layer_id, cpi->svc.temporal_layer_id,
1116 cpi->svc.number_temporal_layers);
1117 LAYER_CONTEXT *lc = &cpi->svc.layer_context[layer];
1118 return denoise_svc(cpi) && !lc->is_key_frame;
1119 }
1120 #endif
1121
vp9_svc_check_spatial_layer_sync(VP9_COMP * const cpi)1122 void vp9_svc_check_spatial_layer_sync(VP9_COMP *const cpi) {
1123 SVC *const svc = &cpi->svc;
1124 // Only for superframes whose base is not key, as those are
1125 // already sync frames.
1126 if (!svc->layer_context[svc->temporal_layer_id].is_key_frame) {
1127 if (svc->spatial_layer_id == 0) {
1128 // On base spatial layer: if the current superframe has a layer sync then
1129 // reset the pattern counters and reset to base temporal layer.
1130 if (svc->superframe_has_layer_sync)
1131 vp9_svc_reset_temporal_layers(cpi, cpi->common.frame_type == KEY_FRAME);
1132 }
1133 // If the layer sync is set for this current spatial layer then
1134 // disable the temporal reference.
1135 if (svc->spatial_layer_id > 0 &&
1136 svc->spatial_layer_sync[svc->spatial_layer_id]) {
1137 cpi->ref_frame_flags &= (~VP9_LAST_FLAG);
1138 if (svc->use_gf_temporal_ref_current_layer) {
1139 int index = svc->spatial_layer_id;
1140 // If golden is used as second reference: need to remove it from
1141 // prediction, reset refresh period to 0, and update the reference.
1142 svc->use_gf_temporal_ref_current_layer = 0;
1143 cpi->rc.baseline_gf_interval = 0;
1144 cpi->rc.frames_till_gf_update_due = 0;
1145 // On layer sync frame we must update the buffer index used for long
1146 // term reference. Use the alt_ref since it is not used or updated on
1147 // sync frames.
1148 if (svc->number_spatial_layers == 3) index = svc->spatial_layer_id - 1;
1149 assert(index >= 0);
1150 cpi->alt_fb_idx = svc->buffer_gf_temporal_ref[index].idx;
1151 cpi->ext_refresh_alt_ref_frame = 1;
1152 }
1153 }
1154 }
1155 }
1156
vp9_svc_update_ref_frame_buffer_idx(VP9_COMP * const cpi)1157 void vp9_svc_update_ref_frame_buffer_idx(VP9_COMP *const cpi) {
1158 SVC *const svc = &cpi->svc;
1159 // Update the usage of frame buffer index for base spatial layers.
1160 if (svc->spatial_layer_id == 0) {
1161 if ((cpi->ref_frame_flags & VP9_LAST_FLAG) || cpi->refresh_last_frame)
1162 svc->fb_idx_base[cpi->lst_fb_idx] = 1;
1163 if ((cpi->ref_frame_flags & VP9_GOLD_FLAG) || cpi->refresh_golden_frame)
1164 svc->fb_idx_base[cpi->gld_fb_idx] = 1;
1165 if ((cpi->ref_frame_flags & VP9_ALT_FLAG) || cpi->refresh_alt_ref_frame)
1166 svc->fb_idx_base[cpi->alt_fb_idx] = 1;
1167 }
1168 }
1169
vp9_svc_update_ref_frame_bypass_mode(VP9_COMP * const cpi)1170 static void vp9_svc_update_ref_frame_bypass_mode(VP9_COMP *const cpi) {
1171 // For non-flexible/bypass SVC mode: check for refreshing other buffer
1172 // slots.
1173 SVC *const svc = &cpi->svc;
1174 VP9_COMMON *const cm = &cpi->common;
1175 BufferPool *const pool = cm->buffer_pool;
1176 int i;
1177 for (i = 0; i < REF_FRAMES; i++) {
1178 if (cm->frame_type == KEY_FRAME ||
1179 svc->update_buffer_slot[svc->spatial_layer_id] & (1 << i)) {
1180 ref_cnt_fb(pool->frame_bufs, &cm->ref_frame_map[i], cm->new_fb_idx);
1181 svc->fb_idx_spatial_layer_id[i] = svc->spatial_layer_id;
1182 svc->fb_idx_temporal_layer_id[i] = svc->temporal_layer_id;
1183 }
1184 }
1185 }
1186
vp9_svc_update_ref_frame(VP9_COMP * const cpi)1187 void vp9_svc_update_ref_frame(VP9_COMP *const cpi) {
1188 VP9_COMMON *const cm = &cpi->common;
1189 SVC *const svc = &cpi->svc;
1190 BufferPool *const pool = cm->buffer_pool;
1191
1192 if (svc->temporal_layering_mode == VP9E_TEMPORAL_LAYERING_MODE_BYPASS &&
1193 svc->use_set_ref_frame_config) {
1194 vp9_svc_update_ref_frame_bypass_mode(cpi);
1195 } else if (cm->frame_type == KEY_FRAME) {
1196 // Keep track of frame index for each reference frame.
1197 int i;
1198 // On key frame update all reference frame slots.
1199 for (i = 0; i < REF_FRAMES; i++) {
1200 svc->fb_idx_spatial_layer_id[i] = svc->spatial_layer_id;
1201 svc->fb_idx_temporal_layer_id[i] = svc->temporal_layer_id;
1202 // LAST/GOLDEN/ALTREF is already updated above.
1203 if (i != cpi->lst_fb_idx && i != cpi->gld_fb_idx && i != cpi->alt_fb_idx)
1204 ref_cnt_fb(pool->frame_bufs, &cm->ref_frame_map[i], cm->new_fb_idx);
1205 }
1206 } else {
1207 if (cpi->refresh_last_frame) {
1208 svc->fb_idx_spatial_layer_id[cpi->lst_fb_idx] = svc->spatial_layer_id;
1209 svc->fb_idx_temporal_layer_id[cpi->lst_fb_idx] = svc->temporal_layer_id;
1210 }
1211 if (cpi->refresh_golden_frame) {
1212 svc->fb_idx_spatial_layer_id[cpi->gld_fb_idx] = svc->spatial_layer_id;
1213 svc->fb_idx_temporal_layer_id[cpi->gld_fb_idx] = svc->temporal_layer_id;
1214 }
1215 if (cpi->refresh_alt_ref_frame) {
1216 svc->fb_idx_spatial_layer_id[cpi->alt_fb_idx] = svc->spatial_layer_id;
1217 svc->fb_idx_temporal_layer_id[cpi->alt_fb_idx] = svc->temporal_layer_id;
1218 }
1219 }
1220 // Copy flags from encoder to SVC struct.
1221 vp9_copy_flags_ref_update_idx(cpi);
1222 vp9_svc_update_ref_frame_buffer_idx(cpi);
1223 }
1224
vp9_svc_adjust_frame_rate(VP9_COMP * const cpi)1225 void vp9_svc_adjust_frame_rate(VP9_COMP *const cpi) {
1226 int64_t this_duration =
1227 cpi->svc.timebase_fac * cpi->svc.duration[cpi->svc.spatial_layer_id];
1228 vp9_new_framerate(cpi, 10000000.0 / this_duration);
1229 }
1230