12ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian/* 22ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian * Copyright (c) 2012 The WebM project authors. All Rights Reserved. 32ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian * 42ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian * Use of this source code is governed by a BSD-style license 52ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian * that can be found in the LICENSE file in the root of the source 62ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian * tree. An additional intellectual property rights grant can be found 72ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian * in the file PATENTS. All contributing project authors may 82ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian * be found in the AUTHORS file in the root of the source tree. 92ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian */ 102ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 11ba6c59e9d7d7013b3906b6f4230b663422681848Vignesh Venkatasubramanian// This is an example demonstrating how to implement a multi-layer VPx 122ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian// encoding scheme based on temporal scalability for video applications 132ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian// that benefit from a scalable bitstream. 142ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 15da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#include <assert.h> 162ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian#include <math.h> 172ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian#include <stdio.h> 182ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian#include <stdlib.h> 192ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian#include <string.h> 202ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 212ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian#include "./vpx_config.h" 22da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#include "../vpx_ports/vpx_timer.h" 232ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian#include "vpx/vp8cx.h" 242ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian#include "vpx/vpx_encoder.h" 252ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 26da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#include "../tools_common.h" 27da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#include "../video_writer.h" 282ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 292ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanianstatic const char *exec_name; 302ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 31da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanianvoid usage_exit(void) { 322ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian exit(EXIT_FAILURE); 332ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian} 342ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 35ba6c59e9d7d7013b3906b6f4230b663422681848Vignesh Venkatasubramanian// Denoiser states, for temporal denoising. 36ba6c59e9d7d7013b3906b6f4230b663422681848Vignesh Venkatasubramanianenum denoiserState { 37ba6c59e9d7d7013b3906b6f4230b663422681848Vignesh Venkatasubramanian kDenoiserOff, 38ba6c59e9d7d7013b3906b6f4230b663422681848Vignesh Venkatasubramanian kDenoiserOnYOnly, 39ba6c59e9d7d7013b3906b6f4230b663422681848Vignesh Venkatasubramanian kDenoiserOnYUV, 40da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian kDenoiserOnYUVAggressive, 41da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian kDenoiserOnAdaptive 42ba6c59e9d7d7013b3906b6f4230b663422681848Vignesh Venkatasubramanian}; 43ba6c59e9d7d7013b3906b6f4230b663422681848Vignesh Venkatasubramanian 442ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanianstatic int mode_to_num_layers[12] = {1, 2, 2, 3, 3, 3, 3, 5, 2, 3, 3, 3}; 452ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 462ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian// For rate control encoding stats. 472ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanianstruct RateControlMetrics { 482ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Number of input frames per layer. 492ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int layer_input_frames[VPX_TS_MAX_LAYERS]; 502ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Total (cumulative) number of encoded frames per layer. 512ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int layer_tot_enc_frames[VPX_TS_MAX_LAYERS]; 522ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Number of encoded non-key frames per layer. 532ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int layer_enc_frames[VPX_TS_MAX_LAYERS]; 542ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Framerate per layer layer (cumulative). 552ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian double layer_framerate[VPX_TS_MAX_LAYERS]; 562ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Target average frame size per layer (per-frame-bandwidth per layer). 572ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian double layer_pfb[VPX_TS_MAX_LAYERS]; 582ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Actual average frame size per layer. 592ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian double layer_avg_frame_size[VPX_TS_MAX_LAYERS]; 602ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Average rate mismatch per layer (|target - actual| / target). 612ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian double layer_avg_rate_mismatch[VPX_TS_MAX_LAYERS]; 622ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Actual encoding bitrate per layer (cumulative). 632ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian double layer_encoding_bitrate[VPX_TS_MAX_LAYERS]; 64da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian // Average of the short-time encoder actual bitrate. 65da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian // TODO(marpan): Should we add these short-time stats for each layer? 66da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian double avg_st_encoding_bitrate; 67da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian // Variance of the short-time encoder actual bitrate. 68da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian double variance_st_encoding_bitrate; 69da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian // Window (number of frames) for computing short-timee encoding bitrate. 70da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian int window_size; 71da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian // Number of window measurements. 72da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian int window_count; 73da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian int layer_target_bitrate[VPX_MAX_LAYERS]; 742ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian}; 752ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 762ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian// Note: these rate control metrics assume only 1 key frame in the 772ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian// sequence (i.e., first frame only). So for temporal pattern# 7 782ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian// (which has key frame for every frame on base layer), the metrics 792ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian// computation will be off/wrong. 802ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian// TODO(marpan): Update these metrics to account for multiple key frames 812ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian// in the stream. 822ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanianstatic void set_rate_control_metrics(struct RateControlMetrics *rc, 832ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian vpx_codec_enc_cfg_t *cfg) { 842ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian unsigned int i = 0; 852ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Set the layer (cumulative) framerate and the target layer (non-cumulative) 862ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // per-frame-bandwidth, for the rate control encoding stats below. 872ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian const double framerate = cfg->g_timebase.den / cfg->g_timebase.num; 882ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_framerate[0] = framerate / cfg->ts_rate_decimator[0]; 89da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc->layer_pfb[0] = 1000.0 * rc->layer_target_bitrate[0] / 902ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_framerate[0]; 912ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian for (i = 0; i < cfg->ts_number_layers; ++i) { 922ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian if (i > 0) { 932ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_framerate[i] = framerate / cfg->ts_rate_decimator[i]; 942ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_pfb[i] = 1000.0 * 95da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian (rc->layer_target_bitrate[i] - rc->layer_target_bitrate[i - 1]) / 962ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian (rc->layer_framerate[i] - rc->layer_framerate[i - 1]); 972ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 982ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_input_frames[i] = 0; 992ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_enc_frames[i] = 0; 1002ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_tot_enc_frames[i] = 0; 1012ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_encoding_bitrate[i] = 0.0; 1022ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_avg_frame_size[i] = 0.0; 1032ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_avg_rate_mismatch[i] = 0.0; 1042ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 105da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc->window_count = 0; 106da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc->window_size = 15; 107da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc->avg_st_encoding_bitrate = 0.0; 108da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc->variance_st_encoding_bitrate = 0.0; 1092ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian} 1102ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 1112ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanianstatic void printout_rate_control_summary(struct RateControlMetrics *rc, 1122ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian vpx_codec_enc_cfg_t *cfg, 1132ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int frame_cnt) { 1142ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian unsigned int i = 0; 1152ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int tot_num_frames = 0; 116da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian double perc_fluctuation = 0.0; 1172ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian printf("Total number of processed frames: %d\n\n", frame_cnt -1); 1182ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian printf("Rate control layer stats for %d layer(s):\n\n", 1192ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_number_layers); 1202ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian for (i = 0; i < cfg->ts_number_layers; ++i) { 1212ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian const int num_dropped = (i > 0) ? 1222ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian (rc->layer_input_frames[i] - rc->layer_enc_frames[i]) : 1232ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian (rc->layer_input_frames[i] - rc->layer_enc_frames[i] - 1); 1242ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian tot_num_frames += rc->layer_input_frames[i]; 1252ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_encoding_bitrate[i] = 0.001 * rc->layer_framerate[i] * 1262ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_encoding_bitrate[i] / tot_num_frames; 1272ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_avg_frame_size[i] = rc->layer_avg_frame_size[i] / 1282ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_enc_frames[i]; 1292ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_avg_rate_mismatch[i] = 100.0 * rc->layer_avg_rate_mismatch[i] / 1302ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_enc_frames[i]; 1312ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian printf("For layer#: %d \n", i); 132da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian printf("Bitrate (target vs actual): %d %f \n", rc->layer_target_bitrate[i], 1332ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_encoding_bitrate[i]); 1342ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian printf("Average frame size (target vs actual): %f %f \n", rc->layer_pfb[i], 1352ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_avg_frame_size[i]); 1362ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian printf("Average rate_mismatch: %f \n", rc->layer_avg_rate_mismatch[i]); 1372ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian printf("Number of input frames, encoded (non-key) frames, " 1382ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian "and perc dropped frames: %d %d %f \n", rc->layer_input_frames[i], 1392ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc->layer_enc_frames[i], 1402ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 100.0 * num_dropped / rc->layer_input_frames[i]); 1412ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian printf("\n"); 1422ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 143da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc->avg_st_encoding_bitrate = rc->avg_st_encoding_bitrate / rc->window_count; 144da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc->variance_st_encoding_bitrate = 145da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc->variance_st_encoding_bitrate / rc->window_count - 146da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian (rc->avg_st_encoding_bitrate * rc->avg_st_encoding_bitrate); 147da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian perc_fluctuation = 100.0 * sqrt(rc->variance_st_encoding_bitrate) / 148da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc->avg_st_encoding_bitrate; 149da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian printf("Short-time stats, for window of %d frames: \n",rc->window_size); 150da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian printf("Average, rms-variance, and percent-fluct: %f %f %f \n", 151da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc->avg_st_encoding_bitrate, 152da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian sqrt(rc->variance_st_encoding_bitrate), 153da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian perc_fluctuation); 1542ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian if ((frame_cnt - 1) != tot_num_frames) 1552ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian die("Error: Number of input frames not equal to output! \n"); 1562ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian} 1572ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 1582ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian// Temporal scaling parameters: 1592ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian// NOTE: The 3 prediction frames cannot be used interchangeably due to 1602ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian// differences in the way they are handled throughout the code. The 1612ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian// frames should be allocated to layers in the order LAST, GF, ARF. 1622ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian// Other combinations work, but may produce slightly inferior results. 1632ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanianstatic void set_temporal_layer_pattern(int layering_mode, 1642ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian vpx_codec_enc_cfg_t *cfg, 1652ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int *layer_flags, 1662ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int *flag_periodicity) { 1672ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian switch (layering_mode) { 1682ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian case 0: { 1692ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 1-layer. 1702ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int ids[1] = {0}; 1712ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_periodicity = 1; 1722ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian *flag_periodicity = 1; 1732ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_number_layers = 1; 1742ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[0] = 1; 1752ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian memcpy(cfg->ts_layer_id, ids, sizeof(ids)); 1762ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Update L only. 1772ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[0] = VPX_EFLAG_FORCE_KF | VP8_EFLAG_NO_UPD_GF | 1782ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ARF; 1792ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian break; 1802ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 1812ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian case 1: { 1822ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 2-layers, 2-frame period. 1832ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int ids[2] = {0, 1}; 1842ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_periodicity = 2; 1852ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian *flag_periodicity = 2; 1862ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_number_layers = 2; 1872ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[0] = 2; 1882ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[1] = 1; 1892ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian memcpy(cfg->ts_layer_id, ids, sizeof(ids)); 1902ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian#if 1 1912ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 0=L, 1=GF, Intra-layer prediction enabled. 1922ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[0] = VPX_EFLAG_FORCE_KF | VP8_EFLAG_NO_UPD_GF | 1932ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ARF | VP8_EFLAG_NO_REF_GF | VP8_EFLAG_NO_REF_ARF; 1942ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[1] = VP8_EFLAG_NO_UPD_ARF | VP8_EFLAG_NO_UPD_LAST | 1952ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_REF_ARF; 1962ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian#else 1972ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 0=L, 1=GF, Intra-layer prediction disabled. 1982ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[0] = VPX_EFLAG_FORCE_KF | VP8_EFLAG_NO_UPD_GF | 1992ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ARF | VP8_EFLAG_NO_REF_GF | VP8_EFLAG_NO_REF_ARF; 2002ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[1] = VP8_EFLAG_NO_UPD_ARF | VP8_EFLAG_NO_UPD_LAST | 2012ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_REF_ARF | VP8_EFLAG_NO_REF_LAST; 2022ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian#endif 2032ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian break; 2042ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 2052ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian case 2: { 2062ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 2-layers, 3-frame period. 2072ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int ids[3] = {0, 1, 1}; 2082ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_periodicity = 3; 2092ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian *flag_periodicity = 3; 2102ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_number_layers = 2; 2112ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[0] = 3; 2122ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[1] = 1; 2132ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian memcpy(cfg->ts_layer_id, ids, sizeof(ids)); 2142ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 0=L, 1=GF, Intra-layer prediction enabled. 2152ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[0] = VPX_EFLAG_FORCE_KF | VP8_EFLAG_NO_REF_GF | 2162ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_REF_ARF | VP8_EFLAG_NO_UPD_GF | VP8_EFLAG_NO_UPD_ARF; 2172ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[1] = 2182ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[2] = VP8_EFLAG_NO_REF_GF | VP8_EFLAG_NO_REF_ARF | 2192ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ARF | VP8_EFLAG_NO_UPD_LAST; 2202ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian break; 2212ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 2222ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian case 3: { 2232ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 3-layers, 6-frame period. 2242ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int ids[6] = {0, 2, 2, 1, 2, 2}; 2252ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_periodicity = 6; 2262ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian *flag_periodicity = 6; 2272ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_number_layers = 3; 2282ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[0] = 6; 2292ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[1] = 3; 2302ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[2] = 1; 2312ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian memcpy(cfg->ts_layer_id, ids, sizeof(ids)); 2322ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 0=L, 1=GF, 2=ARF, Intra-layer prediction enabled. 2332ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[0] = VPX_EFLAG_FORCE_KF | VP8_EFLAG_NO_REF_GF | 2342ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_REF_ARF | VP8_EFLAG_NO_UPD_GF | VP8_EFLAG_NO_UPD_ARF; 2352ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[3] = VP8_EFLAG_NO_REF_ARF | VP8_EFLAG_NO_UPD_ARF | 2362ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_LAST; 2372ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[1] = 2382ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[2] = 2392ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[4] = 2402ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[5] = VP8_EFLAG_NO_UPD_GF | VP8_EFLAG_NO_UPD_LAST; 2412ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian break; 2422ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 2432ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian case 4: { 2442ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 3-layers, 4-frame period. 2452ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int ids[4] = {0, 2, 1, 2}; 2462ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_periodicity = 4; 2472ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian *flag_periodicity = 4; 2482ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_number_layers = 3; 2492ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[0] = 4; 2502ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[1] = 2; 2512ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[2] = 1; 2522ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian memcpy(cfg->ts_layer_id, ids, sizeof(ids)); 2532ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 0=L, 1=GF, 2=ARF, Intra-layer prediction disabled. 2542ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[0] = VPX_EFLAG_FORCE_KF | VP8_EFLAG_NO_REF_GF | 2552ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_REF_ARF | VP8_EFLAG_NO_UPD_GF | VP8_EFLAG_NO_UPD_ARF; 2562ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[2] = VP8_EFLAG_NO_REF_GF | VP8_EFLAG_NO_REF_ARF | 2572ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ARF | VP8_EFLAG_NO_UPD_LAST; 2582ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[1] = 2592ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[3] = VP8_EFLAG_NO_REF_ARF | VP8_EFLAG_NO_UPD_LAST | 2602ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_GF | VP8_EFLAG_NO_UPD_ARF; 2612ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian break; 2622ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 2632ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian case 5: { 2642ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 3-layers, 4-frame period. 2652ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int ids[4] = {0, 2, 1, 2}; 2662ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_periodicity = 4; 2672ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian *flag_periodicity = 4; 2682ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_number_layers = 3; 2692ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[0] = 4; 2702ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[1] = 2; 2712ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[2] = 1; 2722ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian memcpy(cfg->ts_layer_id, ids, sizeof(ids)); 2732ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 0=L, 1=GF, 2=ARF, Intra-layer prediction enabled in layer 1, disabled 2742ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // in layer 2. 2752ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[0] = VPX_EFLAG_FORCE_KF | VP8_EFLAG_NO_REF_GF | 2762ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_REF_ARF | VP8_EFLAG_NO_UPD_GF | VP8_EFLAG_NO_UPD_ARF; 2772ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[2] = VP8_EFLAG_NO_REF_ARF | VP8_EFLAG_NO_UPD_LAST | 2782ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ARF; 2792ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[1] = 2802ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[3] = VP8_EFLAG_NO_REF_ARF | VP8_EFLAG_NO_UPD_LAST | 2812ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_GF | VP8_EFLAG_NO_UPD_ARF; 2822ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian break; 2832ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 2842ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian case 6: { 2852ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 3-layers, 4-frame period. 2862ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int ids[4] = {0, 2, 1, 2}; 2872ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_periodicity = 4; 2882ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian *flag_periodicity = 4; 2892ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_number_layers = 3; 2902ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[0] = 4; 2912ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[1] = 2; 2922ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[2] = 1; 2932ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian memcpy(cfg->ts_layer_id, ids, sizeof(ids)); 2942ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 0=L, 1=GF, 2=ARF, Intra-layer prediction enabled. 2952ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[0] = VPX_EFLAG_FORCE_KF | VP8_EFLAG_NO_REF_GF | 2962ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_REF_ARF | VP8_EFLAG_NO_UPD_GF | VP8_EFLAG_NO_UPD_ARF; 2972ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[2] = VP8_EFLAG_NO_REF_ARF | VP8_EFLAG_NO_UPD_LAST | 2982ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ARF; 2992ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[1] = 3002ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[3] = VP8_EFLAG_NO_UPD_LAST | VP8_EFLAG_NO_UPD_GF; 3012ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian break; 3022ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 3032ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian case 7: { 3042ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // NOTE: Probably of academic interest only. 3052ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 5-layers, 16-frame period. 3062ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int ids[16] = {0, 4, 3, 4, 2, 4, 3, 4, 1, 4, 3, 4, 2, 4, 3, 4}; 3072ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_periodicity = 16; 3082ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian *flag_periodicity = 16; 3092ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_number_layers = 5; 3102ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[0] = 16; 3112ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[1] = 8; 3122ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[2] = 4; 3132ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[3] = 2; 3142ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[4] = 1; 3152ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian memcpy(cfg->ts_layer_id, ids, sizeof(ids)); 3162ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[0] = VPX_EFLAG_FORCE_KF; 3172ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[1] = 3182ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[3] = 3192ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[5] = 3202ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[7] = 3212ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[9] = 3222ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[11] = 3232ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[13] = 3242ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[15] = VP8_EFLAG_NO_UPD_LAST | VP8_EFLAG_NO_UPD_GF | 3252ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ARF; 3262ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[2] = 3272ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[6] = 3282ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[10] = 3292ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[14] = VP8_EFLAG_NO_UPD_ARF | VP8_EFLAG_NO_UPD_GF; 3302ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[4] = 3312ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[12] = VP8_EFLAG_NO_REF_LAST | VP8_EFLAG_NO_UPD_ARF; 3322ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[8] = VP8_EFLAG_NO_REF_LAST | VP8_EFLAG_NO_REF_GF; 3332ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian break; 3342ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 3352ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian case 8: { 3362ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 2-layers, with sync point at first frame of layer 1. 3372ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int ids[2] = {0, 1}; 3382ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_periodicity = 2; 3392ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian *flag_periodicity = 8; 3402ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_number_layers = 2; 3412ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[0] = 2; 3422ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[1] = 1; 3432ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian memcpy(cfg->ts_layer_id, ids, sizeof(ids)); 3442ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 0=L, 1=GF. 3452ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // ARF is used as predictor for all frames, and is only updated on 3462ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // key frame. Sync point every 8 frames. 3472ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 3482ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 0: predict from L and ARF, update L and G. 3492ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[0] = VPX_EFLAG_FORCE_KF | VP8_EFLAG_NO_REF_GF | 3502ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ARF; 3512ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 1: sync point: predict from L and ARF, and update G. 3522ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[1] = VP8_EFLAG_NO_REF_GF | VP8_EFLAG_NO_UPD_LAST | 3532ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ARF; 3542ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 0, predict from L and ARF, update L. 3552ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[2] = VP8_EFLAG_NO_REF_GF | VP8_EFLAG_NO_UPD_GF | 3562ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ARF; 3572ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 1: predict from L, G and ARF, and update G. 3582ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[3] = VP8_EFLAG_NO_UPD_ARF | VP8_EFLAG_NO_UPD_LAST | 3592ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ENTROPY; 3602ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 0. 3612ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[4] = layer_flags[2]; 3622ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 1. 3632ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[5] = layer_flags[3]; 3642ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 0. 3652ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[6] = layer_flags[4]; 3662ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 1. 3672ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[7] = layer_flags[5]; 3682ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian break; 3692ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 3702ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian case 9: { 3712ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 3-layers: Sync points for layer 1 and 2 every 8 frames. 3722ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int ids[4] = {0, 2, 1, 2}; 3732ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_periodicity = 4; 3742ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian *flag_periodicity = 8; 3752ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_number_layers = 3; 3762ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[0] = 4; 3772ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[1] = 2; 3782ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[2] = 1; 3792ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian memcpy(cfg->ts_layer_id, ids, sizeof(ids)); 3802ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 0=L, 1=GF, 2=ARF. 3812ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[0] = VPX_EFLAG_FORCE_KF | VP8_EFLAG_NO_REF_GF | 3822ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_REF_ARF | VP8_EFLAG_NO_UPD_GF | VP8_EFLAG_NO_UPD_ARF; 3832ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[1] = VP8_EFLAG_NO_REF_GF | VP8_EFLAG_NO_REF_ARF | 3842ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_LAST | VP8_EFLAG_NO_UPD_GF; 3852ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[2] = VP8_EFLAG_NO_REF_GF | VP8_EFLAG_NO_REF_ARF | 3862ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_LAST | VP8_EFLAG_NO_UPD_ARF; 3872ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[3] = 3882ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[5] = VP8_EFLAG_NO_UPD_LAST | VP8_EFLAG_NO_UPD_GF; 3892ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[4] = VP8_EFLAG_NO_REF_GF | VP8_EFLAG_NO_REF_ARF | 3902ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_GF | VP8_EFLAG_NO_UPD_ARF; 3912ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[6] = VP8_EFLAG_NO_REF_ARF | VP8_EFLAG_NO_UPD_LAST | 3922ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ARF; 3932ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[7] = VP8_EFLAG_NO_UPD_LAST | VP8_EFLAG_NO_UPD_GF | 3942ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ARF | VP8_EFLAG_NO_UPD_ENTROPY; 3952ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian break; 3962ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 3972ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian case 10: { 3982ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 3-layers structure where ARF is used as predictor for all frames, 3992ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // and is only updated on key frame. 4002ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Sync points for layer 1 and 2 every 8 frames. 4012ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 4022ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int ids[4] = {0, 2, 1, 2}; 4032ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_periodicity = 4; 4042ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian *flag_periodicity = 8; 4052ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_number_layers = 3; 4062ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[0] = 4; 4072ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[1] = 2; 4082ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[2] = 1; 4092ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian memcpy(cfg->ts_layer_id, ids, sizeof(ids)); 4102ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 0=L, 1=GF, 2=ARF. 4112ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 0: predict from L and ARF; update L and G. 4122ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[0] = VPX_EFLAG_FORCE_KF | VP8_EFLAG_NO_UPD_ARF | 4132ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_REF_GF; 4142ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 2: sync point: predict from L and ARF; update none. 4152ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[1] = VP8_EFLAG_NO_REF_GF | VP8_EFLAG_NO_UPD_GF | 4162ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ARF | VP8_EFLAG_NO_UPD_LAST | 4172ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_ENTROPY; 4182ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 1: sync point: predict from L and ARF; update G. 4192ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[2] = VP8_EFLAG_NO_REF_GF | VP8_EFLAG_NO_UPD_ARF | 4202ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_LAST; 4212ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 2: predict from L, G, ARF; update none. 4222ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[3] = VP8_EFLAG_NO_UPD_GF | VP8_EFLAG_NO_UPD_ARF | 4232ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_LAST | VP8_EFLAG_NO_UPD_ENTROPY; 4242ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 0: predict from L and ARF; update L. 4252ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[4] = VP8_EFLAG_NO_UPD_GF | VP8_EFLAG_NO_UPD_ARF | 4262ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_REF_GF; 4272ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 2: predict from L, G, ARF; update none. 4282ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[5] = layer_flags[3]; 4292ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 1: predict from L, G, ARF; update G. 4302ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[6] = VP8_EFLAG_NO_UPD_ARF | VP8_EFLAG_NO_UPD_LAST; 4312ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 2: predict from L, G, ARF; update none. 4322ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[7] = layer_flags[3]; 4332ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian break; 4342ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 4352ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian case 11: 4362ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian default: { 4372ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 3-layers structure as in case 10, but no sync/refresh points for 4382ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // layer 1 and 2. 4392ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int ids[4] = {0, 2, 1, 2}; 4402ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_periodicity = 4; 4412ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian *flag_periodicity = 8; 4422ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_number_layers = 3; 4432ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[0] = 4; 4442ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[1] = 2; 4452ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg->ts_rate_decimator[2] = 1; 4462ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian memcpy(cfg->ts_layer_id, ids, sizeof(ids)); 4472ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // 0=L, 1=GF, 2=ARF. 4482ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 0: predict from L and ARF; update L. 4492ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[0] = VP8_EFLAG_NO_UPD_GF | VP8_EFLAG_NO_UPD_ARF | 4502ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_REF_GF; 4512ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[4] = layer_flags[0]; 4522ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 1: predict from L, G, ARF; update G. 4532ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[2] = VP8_EFLAG_NO_UPD_ARF | VP8_EFLAG_NO_UPD_LAST; 4542ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[6] = layer_flags[2]; 4552ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Layer 2: predict from L, G, ARF; update none. 4562ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[1] = VP8_EFLAG_NO_UPD_GF | VP8_EFLAG_NO_UPD_ARF | 4572ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VP8_EFLAG_NO_UPD_LAST | VP8_EFLAG_NO_UPD_ENTROPY; 4582ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[3] = layer_flags[1]; 4592ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[5] = layer_flags[1]; 4602ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[7] = layer_flags[1]; 4612ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian break; 4622ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 4632ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 4642ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian} 4652ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 4662ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanianint main(int argc, char **argv) { 467da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian VpxVideoWriter *outfile[VPX_TS_MAX_LAYERS] = {NULL}; 4682ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian vpx_codec_ctx_t codec; 4692ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian vpx_codec_enc_cfg_t cfg; 4702ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int frame_cnt = 0; 4712ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian vpx_image_t raw; 4722ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian vpx_codec_err_t res; 4732ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian unsigned int width; 4742ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian unsigned int height; 4754fb68e5dd4e93c7599dc905d861de11ac39c5585hkuang int speed; 4762ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int frame_avail; 4772ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int got_data; 4782ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int flags = 0; 4792ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian unsigned int i; 4802ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int pts = 0; // PTS starts at 0. 4812ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int frame_duration = 1; // 1 timebase tick per frame. 4822ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int layering_mode = 0; 4832ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int layer_flags[VPX_TS_MAX_PERIODICITY] = {0}; 4842ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int flag_periodicity = 1; 485da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#if VPX_ENCODER_ABI_VERSION > (4 + VPX_CODEC_ABI_VERSION) 4862ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian vpx_svc_layer_id_t layer_id = {0, 0}; 487da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#else 488da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian vpx_svc_layer_id_t layer_id = {0}; 489da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#endif 4902ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian const VpxInterface *encoder = NULL; 4912ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian FILE *infile = NULL; 4922ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian struct RateControlMetrics rc; 4932ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian int64_t cx_time = 0; 494da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian const int min_args_base = 11; 495da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#if CONFIG_VP9_HIGHBITDEPTH 496da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian vpx_bit_depth_t bit_depth = VPX_BITS_8; 497da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian int input_bit_depth = 8; 498da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian const int min_args = min_args_base + 1; 499da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#else 500da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian const int min_args = min_args_base; 501da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#endif // CONFIG_VP9_HIGHBITDEPTH 502da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian double sum_bitrate = 0.0; 503da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian double sum_bitrate2 = 0.0; 504da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian double framerate = 30.0; 5052ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 5062ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian exec_name = argv[0]; 5072ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Check usage and arguments. 508da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian if (argc < min_args) { 509da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#if CONFIG_VP9_HIGHBITDEPTH 510da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian die("Usage: %s <infile> <outfile> <codec_type(vp8/vp9)> <width> <height> " 511da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian "<rate_num> <rate_den> <speed> <frame_drop_threshold> <mode> " 512da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian "<Rate_0> ... <Rate_nlayers-1> <bit-depth> \n", argv[0]); 513da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#else 5142ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian die("Usage: %s <infile> <outfile> <codec_type(vp8/vp9)> <width> <height> " 5154fb68e5dd4e93c7599dc905d861de11ac39c5585hkuang "<rate_num> <rate_den> <speed> <frame_drop_threshold> <mode> " 5162ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian "<Rate_0> ... <Rate_nlayers-1> \n", argv[0]); 517da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#endif // CONFIG_VP9_HIGHBITDEPTH 5182ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 5192ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 5202ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian encoder = get_vpx_encoder_by_name(argv[3]); 5212ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian if (!encoder) 5222ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian die("Unsupported codec."); 5232ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 524ba6c59e9d7d7013b3906b6f4230b663422681848Vignesh Venkatasubramanian printf("Using %s\n", vpx_codec_iface_name(encoder->codec_interface())); 5252ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 5262ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian width = strtol(argv[4], NULL, 0); 5272ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian height = strtol(argv[5], NULL, 0); 5282ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian if (width < 16 || width % 2 || height < 16 || height % 2) { 5292ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian die("Invalid resolution: %d x %d", width, height); 5302ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 5312ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 5324fb68e5dd4e93c7599dc905d861de11ac39c5585hkuang layering_mode = strtol(argv[10], NULL, 0); 5332ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian if (layering_mode < 0 || layering_mode > 12) { 5344fb68e5dd4e93c7599dc905d861de11ac39c5585hkuang die("Invalid layering mode (0..12) %s", argv[10]); 5352ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 5362ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 537da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian if (argc != min_args + mode_to_num_layers[layering_mode]) { 5382ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian die("Invalid number of arguments"); 5392ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 5402ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 541da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#if CONFIG_VP9_HIGHBITDEPTH 542da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian switch (strtol(argv[argc-1], NULL, 0)) { 543da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian case 8: 544da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian bit_depth = VPX_BITS_8; 545da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian input_bit_depth = 8; 546da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian break; 547da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian case 10: 548da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian bit_depth = VPX_BITS_10; 549da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian input_bit_depth = 10; 550da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian break; 551da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian case 12: 552da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian bit_depth = VPX_BITS_12; 553da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian input_bit_depth = 12; 554da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian break; 555da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian default: 556da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian die("Invalid bit depth (8, 10, 12) %s", argv[argc-1]); 557da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian } 558da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian if (!vpx_img_alloc(&raw, 559da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian bit_depth == VPX_BITS_8 ? VPX_IMG_FMT_I420 : 560da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian VPX_IMG_FMT_I42016, 561da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian width, height, 32)) { 562da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian die("Failed to allocate image", width, height); 563da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian } 564da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#else 5652ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian if (!vpx_img_alloc(&raw, VPX_IMG_FMT_I420, width, height, 32)) { 5662ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian die("Failed to allocate image", width, height); 5672ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 568da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#endif // CONFIG_VP9_HIGHBITDEPTH 5692ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 5702ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Populate encoder configuration. 571ba6c59e9d7d7013b3906b6f4230b663422681848Vignesh Venkatasubramanian res = vpx_codec_enc_config_default(encoder->codec_interface(), &cfg, 0); 5722ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian if (res) { 5732ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian printf("Failed to get config: %s\n", vpx_codec_err_to_string(res)); 5742ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian return EXIT_FAILURE; 5752ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 5762ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 5772ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Update the default configuration with our settings. 5782ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.g_w = width; 5792ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.g_h = height; 5802ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 581da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#if CONFIG_VP9_HIGHBITDEPTH 582da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian if (bit_depth != VPX_BITS_8) { 583da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian cfg.g_bit_depth = bit_depth; 584da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian cfg.g_input_bit_depth = input_bit_depth; 585da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian cfg.g_profile = 2; 586da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian } 587da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#endif // CONFIG_VP9_HIGHBITDEPTH 588da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian 5892ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Timebase format e.g. 30fps: numerator=1, demoninator = 30. 5902ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.g_timebase.num = strtol(argv[6], NULL, 0); 5912ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.g_timebase.den = strtol(argv[7], NULL, 0); 5922ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 5934fb68e5dd4e93c7599dc905d861de11ac39c5585hkuang speed = strtol(argv[8], NULL, 0); 5944fb68e5dd4e93c7599dc905d861de11ac39c5585hkuang if (speed < 0) { 5954fb68e5dd4e93c7599dc905d861de11ac39c5585hkuang die("Invalid speed setting: must be positive"); 5964fb68e5dd4e93c7599dc905d861de11ac39c5585hkuang } 5974fb68e5dd4e93c7599dc905d861de11ac39c5585hkuang 598da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian for (i = min_args_base; 599da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian (int)i < min_args_base + mode_to_num_layers[layering_mode]; 600da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian ++i) { 601da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc.layer_target_bitrate[i - 11] = strtol(argv[i], NULL, 0); 602da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian if (strncmp(encoder->name, "vp8", 3) == 0) 603da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian cfg.ts_target_bitrate[i - 11] = rc.layer_target_bitrate[i - 11]; 604da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian else if (strncmp(encoder->name, "vp9", 3) == 0) 605da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian cfg.layer_target_bitrate[i - 11] = rc.layer_target_bitrate[i - 11]; 6062ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 6072ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 6082ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Real time parameters. 6094fb68e5dd4e93c7599dc905d861de11ac39c5585hkuang cfg.rc_dropframe_thresh = strtol(argv[9], NULL, 0); 6102ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.rc_end_usage = VPX_CBR; 6112ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.rc_min_quantizer = 2; 6122ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.rc_max_quantizer = 56; 613da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian if (strncmp(encoder->name, "vp9", 3) == 0) 614da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian cfg.rc_max_quantizer = 52; 6152ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.rc_undershoot_pct = 50; 6162ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.rc_overshoot_pct = 50; 6172ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.rc_buf_initial_sz = 500; 6182ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.rc_buf_optimal_sz = 600; 6192ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.rc_buf_sz = 1000; 6202ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 621da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian // Disable dynamic resizing by default. 622da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian cfg.rc_resize_allowed = 0; 623da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian 624da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian // Use 1 thread as default. 625da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian cfg.g_threads = 1; 626da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian 6272ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Enable error resilient mode. 6282ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.g_error_resilient = 1; 6292ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.g_lag_in_frames = 0; 630ba6c59e9d7d7013b3906b6f4230b663422681848Vignesh Venkatasubramanian cfg.kf_mode = VPX_KF_AUTO; 6312ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 6322ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Disable automatic keyframe placement. 6332ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.kf_min_dist = cfg.kf_max_dist = 3000; 6342ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 635da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian cfg.temporal_layering_mode = VP9E_TEMPORAL_LAYERING_MODE_BYPASS; 636da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian 6372ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian set_temporal_layer_pattern(layering_mode, 6382ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian &cfg, 6392ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags, 6402ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian &flag_periodicity); 6412ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 6422ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian set_rate_control_metrics(&rc, &cfg); 6432ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 6442ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Target bandwidth for the whole stream. 645da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian // Set to layer_target_bitrate for highest layer (total bitrate). 646da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian cfg.rc_target_bitrate = rc.layer_target_bitrate[cfg.ts_number_layers - 1]; 6472ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 6482ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Open input file. 6492ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian if (!(infile = fopen(argv[1], "rb"))) { 6502ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian die("Failed to open %s for reading", argv[1]); 6512ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 6522ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 653da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian framerate = cfg.g_timebase.den / cfg.g_timebase.num; 6542ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Open an output file for each stream. 6552ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian for (i = 0; i < cfg.ts_number_layers; ++i) { 6562ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian char file_name[PATH_MAX]; 6572ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VpxVideoInfo info; 6582ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian info.codec_fourcc = encoder->fourcc; 6592ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian info.frame_width = cfg.g_w; 6602ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian info.frame_height = cfg.g_h; 6612ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian info.time_base.numerator = cfg.g_timebase.num; 6622ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian info.time_base.denominator = cfg.g_timebase.den; 6632ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 6642ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian snprintf(file_name, sizeof(file_name), "%s_%d.ivf", argv[2], i); 6652ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian outfile[i] = vpx_video_writer_open(file_name, kContainerIVF, &info); 6662ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian if (!outfile[i]) 6672ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian die("Failed to open %s for writing", file_name); 668da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian 669da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian assert(outfile[i] != NULL); 6702ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 6712ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // No spatial layers in this encoder. 6722ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.ss_number_layers = 1; 6732ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 6742ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Initialize codec. 675da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#if CONFIG_VP9_HIGHBITDEPTH 676da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian if (vpx_codec_enc_init( 677da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian &codec, encoder->codec_interface(), &cfg, 678da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian bit_depth == VPX_BITS_8 ? 0 : VPX_CODEC_USE_HIGHBITDEPTH)) 679da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#else 680ba6c59e9d7d7013b3906b6f4230b663422681848Vignesh Venkatasubramanian if (vpx_codec_enc_init(&codec, encoder->codec_interface(), &cfg, 0)) 681da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#endif // CONFIG_VP9_HIGHBITDEPTH 6822ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian die_codec(&codec, "Failed to initialize encoder"); 6832ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 6844fb68e5dd4e93c7599dc905d861de11ac39c5585hkuang if (strncmp(encoder->name, "vp8", 3) == 0) { 6854fb68e5dd4e93c7599dc905d861de11ac39c5585hkuang vpx_codec_control(&codec, VP8E_SET_CPUUSED, -speed); 686da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian vpx_codec_control(&codec, VP8E_SET_NOISE_SENSITIVITY, kDenoiserOff); 6872263fc984bdc858ee931d3e35c87c404de923950Johann vpx_codec_control(&codec, VP8E_SET_STATIC_THRESHOLD, 1); 6884fb68e5dd4e93c7599dc905d861de11ac39c5585hkuang } else if (strncmp(encoder->name, "vp9", 3) == 0) { 689da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian vpx_svc_extra_cfg_t svc_params; 690da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian vpx_codec_control(&codec, VP8E_SET_CPUUSED, speed); 691da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian vpx_codec_control(&codec, VP9E_SET_AQ_MODE, 3); 692da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian vpx_codec_control(&codec, VP9E_SET_FRAME_PERIODIC_BOOST, 0); 693da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian vpx_codec_control(&codec, VP9E_SET_NOISE_SENSITIVITY, 0); 6942263fc984bdc858ee931d3e35c87c404de923950Johann vpx_codec_control(&codec, VP8E_SET_STATIC_THRESHOLD, 1); 695da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian vpx_codec_control(&codec, VP9E_SET_TUNE_CONTENT, 0); 696da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian vpx_codec_control(&codec, VP9E_SET_TILE_COLUMNS, (cfg.g_threads >> 1)); 697da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian if (vpx_codec_control(&codec, VP9E_SET_SVC, layering_mode > 0 ? 1: 0)) 698da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian die_codec(&codec, "Failed to set SVC"); 699da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian for (i = 0; i < cfg.ts_number_layers; ++i) { 700da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian svc_params.max_quantizers[i] = cfg.rc_max_quantizer; 701da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian svc_params.min_quantizers[i] = cfg.rc_min_quantizer; 7022ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 703da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian svc_params.scaling_factor_num[0] = cfg.g_h; 704da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian svc_params.scaling_factor_den[0] = cfg.g_h; 705da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian vpx_codec_control(&codec, VP9E_SET_SVC_PARAMETERS, &svc_params); 706da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian } 707da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian if (strncmp(encoder->name, "vp8", 3) == 0) { 708da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian vpx_codec_control(&codec, VP8E_SET_SCREEN_CONTENT_MODE, 0); 7092ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 7102ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian vpx_codec_control(&codec, VP8E_SET_TOKEN_PARTITIONS, 1); 7112ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // This controls the maximum target size of the key frame. 7122ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // For generating smaller key frames, use a smaller max_intra_size_pct 7132ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // value, like 100 or 200. 714da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian { 715da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian const int max_intra_size_pct = 900; 716da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian vpx_codec_control(&codec, VP8E_SET_MAX_INTRA_BITRATE_PCT, 717da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian max_intra_size_pct); 718da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian } 7192ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 7202ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian frame_avail = 1; 7212ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian while (frame_avail || got_data) { 7222ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian struct vpx_usec_timer timer; 7232ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian vpx_codec_iter_t iter = NULL; 7242ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian const vpx_codec_cx_pkt_t *pkt; 725da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#if VPX_ENCODER_ABI_VERSION > (4 + VPX_CODEC_ABI_VERSION) 7262ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Update the temporal layer_id. No spatial layers in this test. 7272ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_id.spatial_layer_id = 0; 728da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian#endif 7292ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_id.temporal_layer_id = 7302ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cfg.ts_layer_id[frame_cnt % cfg.ts_periodicity]; 7312ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian if (strncmp(encoder->name, "vp9", 3) == 0) { 7322ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian vpx_codec_control(&codec, VP9E_SET_SVC_LAYER_ID, &layer_id); 733da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian } else if (strncmp(encoder->name, "vp8", 3) == 0) { 734da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian vpx_codec_control(&codec, VP8E_SET_TEMPORAL_LAYER_ID, 735da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian layer_id.temporal_layer_id); 7362ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 7372ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian flags = layer_flags[frame_cnt % flag_periodicity]; 738da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian if (layering_mode == 0) 739da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian flags = 0; 7402ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian frame_avail = vpx_img_read(&raw, infile); 7412ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian if (frame_avail) 7422ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian ++rc.layer_input_frames[layer_id.temporal_layer_id]; 7432ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian vpx_usec_timer_start(&timer); 7442ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian if (vpx_codec_encode(&codec, frame_avail? &raw : NULL, pts, 1, flags, 7452ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian VPX_DL_REALTIME)) { 7462ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian die_codec(&codec, "Failed to encode frame"); 7472ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 7482ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian vpx_usec_timer_mark(&timer); 7492ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian cx_time += vpx_usec_timer_elapsed(&timer); 7502ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Reset KF flag. 7512ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian if (layering_mode != 7) { 7522ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian layer_flags[0] &= ~VPX_EFLAG_FORCE_KF; 7532ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 7542ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian got_data = 0; 7552ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian while ( (pkt = vpx_codec_get_cx_data(&codec, &iter)) ) { 7562ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian got_data = 1; 7572ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian switch (pkt->kind) { 7582ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian case VPX_CODEC_CX_FRAME_PKT: 7592ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian for (i = cfg.ts_layer_id[frame_cnt % cfg.ts_periodicity]; 7602ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian i < cfg.ts_number_layers; ++i) { 7612ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian vpx_video_writer_write_frame(outfile[i], pkt->data.frame.buf, 7622ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian pkt->data.frame.sz, pts); 7632ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian ++rc.layer_tot_enc_frames[i]; 7642ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc.layer_encoding_bitrate[i] += 8.0 * pkt->data.frame.sz; 7652ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Keep count of rate control stats per layer (for non-key frames). 7662ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian if (i == cfg.ts_layer_id[frame_cnt % cfg.ts_periodicity] && 7672ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian !(pkt->data.frame.flags & VPX_FRAME_IS_KEY)) { 7682ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc.layer_avg_frame_size[i] += 8.0 * pkt->data.frame.sz; 7692ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc.layer_avg_rate_mismatch[i] += 7702ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian fabs(8.0 * pkt->data.frame.sz - rc.layer_pfb[i]) / 7712ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian rc.layer_pfb[i]; 7722ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian ++rc.layer_enc_frames[i]; 7732ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 7742ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 775da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian // Update for short-time encoding bitrate states, for moving window 776da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian // of size rc->window, shifted by rc->window / 2. 777da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian // Ignore first window segment, due to key frame. 778da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian if (frame_cnt > rc.window_size) { 779da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian sum_bitrate += 0.001 * 8.0 * pkt->data.frame.sz * framerate; 780da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian if (frame_cnt % rc.window_size == 0) { 781da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc.window_count += 1; 782da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc.avg_st_encoding_bitrate += sum_bitrate / rc.window_size; 783da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc.variance_st_encoding_bitrate += 784da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian (sum_bitrate / rc.window_size) * 785da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian (sum_bitrate / rc.window_size); 786da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian sum_bitrate = 0.0; 787da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian } 788da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian } 789da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian // Second shifted window. 790da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian if (frame_cnt > rc.window_size + rc.window_size / 2) { 791da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian sum_bitrate2 += 0.001 * 8.0 * pkt->data.frame.sz * framerate; 792da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian if (frame_cnt > 2 * rc.window_size && 793da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian frame_cnt % rc.window_size == 0) { 794da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc.window_count += 1; 795da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc.avg_st_encoding_bitrate += sum_bitrate2 / rc.window_size; 796da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian rc.variance_st_encoding_bitrate += 797da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian (sum_bitrate2 / rc.window_size) * 798da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian (sum_bitrate2 / rc.window_size); 799da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian sum_bitrate2 = 0.0; 800da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian } 801da49e34c1fb5e99681f4ad99c21d9cfd83eddb96Vignesh Venkatasubramanian } 8022ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian break; 8032ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian default: 8042ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian break; 8052ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 8062ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 8072ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian ++frame_cnt; 8082ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian pts += frame_duration; 8092ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian } 8102ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian fclose(infile); 8112ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian printout_rate_control_summary(&rc, &cfg, frame_cnt); 8122ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian printf("\n"); 8132ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian printf("Frame cnt and encoding time/FPS stats for encoding: %d %f %f \n", 8142ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian frame_cnt, 8152ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 1000 * (float)cx_time / (double)(frame_cnt * 1000000), 8162ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 1000000 * (double)frame_cnt / (double)cx_time); 8172ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 8182ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian if (vpx_codec_destroy(&codec)) 8192ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian die_codec(&codec, "Failed to destroy codec"); 8202ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 8212ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian // Try to rewrite the output file headers with the actual frame count. 8222ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian for (i = 0; i < cfg.ts_number_layers; ++i) 8232ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian vpx_video_writer_close(outfile[i]); 8242ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian 825ba6c59e9d7d7013b3906b6f4230b663422681848Vignesh Venkatasubramanian vpx_img_free(&raw); 8262ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian return EXIT_SUCCESS; 8272ec72e65689c948e92b826ae1e867bf369e72f13Vignesh Venkatasubramanian} 828