2014-01-24 10:37:15 +00:00
/*****************************************************************************
2021-11-23 06:46:06 +00:00
* This file is part of uvg266 VVC encoder .
2014-02-21 13:00:20 +00:00
*
2021-10-07 08:32:59 +00:00
* Copyright ( c ) 2021 , Tampere University , ITU / ISO / IEC , project contributors
* All rights reserved .
*
* Redistribution and use in source and binary forms , with or without modification ,
* are permitted provided that the following conditions are met :
*
* * Redistributions of source code must retain the above copyright notice , this
* list of conditions and the following disclaimer .
*
* * Redistributions in binary form must reproduce the above copyright notice , this
* list of conditions and the following disclaimer in the documentation and / or
* other materials provided with the distribution .
*
* * Neither the name of the Tampere University or ITU / ISO / IEC nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission .
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS " AS IS " AND
* ANY EXPRESS OR IMPLIED WARRANTIES , INCLUDING , BUT NOT LIMITED TO , THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED . IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
* ANY DIRECT , INDIRECT , INCIDENTAL , SPECIAL , EXEMPLARY , OR CONSEQUENTIAL DAMAGES
* INCLUDING , BUT NOT LIMITED TO , PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES ;
* LOSS OF USE , DATA , OR PROFITS ; OR BUSINESS INTERRUPTION HOWEVER CAUSED AND ON
* ANY THEORY OF LIABILITY , WHETHER IN CONTRACT , STRICT LIABILITY , OR TORT
* SOFTWARE , EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE .
* INCLUDING NEGLIGENCE OR OTHERWISE ARISING IN ANY WAY OUT OF THE USE OF THIS
2014-01-24 10:37:15 +00:00
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
2016-01-25 10:08:27 +00:00
# include "cfg.h"
2018-02-14 13:08:40 +00:00
# include "gop.h"
2020-10-13 14:43:03 +00:00
# include "alf.h"
2013-09-18 09:16:03 +00:00
2017-02-09 05:34:50 +00:00
# include <limits.h>
2012-05-30 12:10:23 +00:00
# include <stdio.h>
# include <stdlib.h>
# include <string.h>
2017-11-15 14:39:59 +00:00
# include <math.h>
2012-05-30 12:10:23 +00:00
2016-04-01 14:14:23 +00:00
2022-04-28 11:18:09 +00:00
static void parse_qp_map ( uvg_config * cfg , int index ) ;
2021-04-23 07:19:43 +00:00
2022-04-28 11:18:09 +00:00
uvg_config * uvg_config_alloc ( void )
2012-05-30 12:10:23 +00:00
{
2022-04-28 11:18:09 +00:00
return calloc ( 1 , sizeof ( uvg_config ) ) ;
2012-05-30 12:10:23 +00:00
}
2022-04-28 11:18:09 +00:00
int uvg_config_init ( uvg_config * cfg )
2014-02-21 13:00:20 +00:00
{
2014-03-10 13:53:23 +00:00
cfg - > width = 0 ;
cfg - > height = 0 ;
2017-11-09 14:53:31 +00:00
cfg - > framerate_num = 25 ;
2016-01-14 20:08:35 +00:00
cfg - > framerate_denom = 1 ;
2016-09-27 21:15:46 +00:00
cfg - > qp = 22 ;
2020-02-15 20:15:34 +00:00
cfg - > intra_qp_offset = 0 ;
2020-03-31 09:58:40 +00:00
cfg - > intra_qp_offset_auto = true ;
2016-09-27 21:15:46 +00:00
cfg - > intra_period = 64 ;
2015-02-18 11:41:03 +00:00
cfg - > vps_period = 0 ;
2021-04-30 06:51:41 +00:00
cfg - > lmcs_enable = 0 ;
2020-08-20 13:35:14 +00:00
cfg - > deblock_enable = 0 ;
2014-02-06 19:45:37 +00:00
cfg - > deblock_beta = 0 ;
cfg - > deblock_tc = 0 ;
2017-08-11 08:57:09 +00:00
cfg - > sao_type = 3 ;
2020-11-20 12:48:37 +00:00
cfg - > alf_type = 0 ;
2020-10-05 09:01:18 +00:00
cfg - > alf_info_in_ph_flag = 0 ;
2019-11-25 13:04:26 +00:00
cfg - > alf_non_linear_luma = 1 ;
cfg - > alf_non_linear_chroma = 1 ;
2020-12-14 17:02:43 +00:00
cfg - > alf_allow_predefined_filters = 1 ;
2014-03-05 14:56:00 +00:00
cfg - > rdoq_enable = 1 ;
2016-09-27 21:19:30 +00:00
cfg - > rdoq_skip = 1 ;
2015-01-24 18:10:21 +00:00
cfg - > signhide_enable = true ;
2014-04-07 11:36:01 +00:00
cfg - > rdo = 1 ;
2015-11-05 11:59:30 +00:00
cfg - > mv_rdo = 0 ;
2014-06-17 12:32:05 +00:00
cfg - > full_intra_search = 0 ;
2016-09-27 21:19:30 +00:00
cfg - > trskip_enable = 0 ;
2022-06-06 06:28:06 +00:00
cfg - > chroma_trskip_enable = 0 ;
2021-12-02 20:04:09 +00:00
cfg - > trskip_max_size = 2 ; //Default to 4x4
2020-12-18 15:40:47 +00:00
cfg - > mts = 0 ;
2021-02-11 13:11:15 +00:00
cfg - > mts_implicit = 0 ;
2015-03-19 16:48:10 +00:00
cfg - > ime_algorithm = 0 ; /* hexbs */
2016-07-12 12:39:01 +00:00
cfg - > fme_level = 4 ;
2015-08-13 09:53:14 +00:00
cfg - > source_scan_type = 0 ; /* progressive */
2014-02-06 19:45:37 +00:00
cfg - > vui . sar_width = 0 ;
cfg - > vui . sar_height = 0 ;
cfg - > vui . overscan = 0 ; /* undef */
cfg - > vui . videoformat = 5 ; /* undef */
cfg - > vui . fullrange = 0 ; /* limited range */
cfg - > vui . colorprim = 2 ; /* undef */
cfg - > vui . transfer = 2 ; /* undef */
cfg - > vui . colormatrix = 2 ; /* undef */
cfg - > vui . chroma_loc = 0 ; /* left center */
2014-02-06 22:35:15 +00:00
cfg - > aud_enable = 0 ;
config: Add --cqmfile to use custom quantization matrices from a file.
The coefficients in a matrix are stored in up-right diagonal order.
The following indicates the default matrices specified in the spec.
INTRA4X4_LUMA
16, 16, 16, 16,
16, 16, 16, 16,
16, 16, 16, 16,
16, 16, 16, 16
INTRA4X4_CHROMAU
16, 16, 16, 16,
16, 16, 16, 16,
16, 16, 16, 16,
16, 16, 16, 16
INTRA4X4_CHROMAV
16, 16, 16, 16,
16, 16, 16, 16,
16, 16, 16, 16,
16, 16, 16, 16
INTER4X4_LUMA
16, 16, 16, 16,
16, 16, 16, 16,
16, 16, 16, 16,
16, 16, 16, 16
INTER4X4_CHROMAU
16, 16, 16, 16,
16, 16, 16, 16,
16, 16, 16, 16,
16, 16, 16, 16
INTER4X4_CHROMAV
16, 16, 16, 16,
16, 16, 16, 16,
16, 16, 16, 16,
16, 16, 16, 16
INTRA8X8_LUMA
16, 16, 16, 16, 17, 18, 21, 24,
16, 16, 16, 16, 17, 19, 22, 25,
16, 16, 17, 18, 20, 22, 25, 29,
16, 16, 18, 21, 24, 27, 31, 36,
17, 17, 20, 24, 30, 35, 41, 47,
18, 19, 22, 27, 35, 44, 54, 65,
21, 22, 25, 31, 41, 54, 70, 88,
24, 25, 29, 36, 47, 65, 88, 115
INTRA8X8_CHROMAU
16, 16, 16, 16, 17, 18, 21, 24,
16, 16, 16, 16, 17, 19, 22, 25,
16, 16, 17, 18, 20, 22, 25, 29,
16, 16, 18, 21, 24, 27, 31, 36,
17, 17, 20, 24, 30, 35, 41, 47,
18, 19, 22, 27, 35, 44, 54, 65,
21, 22, 25, 31, 41, 54, 70, 88,
24, 25, 29, 36, 47, 65, 88, 115
INTRA8X8_CHROMAV
16, 16, 16, 16, 17, 18, 21, 24,
16, 16, 16, 16, 17, 19, 22, 25,
16, 16, 17, 18, 20, 22, 25, 29,
16, 16, 18, 21, 24, 27, 31, 36,
17, 17, 20, 24, 30, 35, 41, 47,
18, 19, 22, 27, 35, 44, 54, 65,
21, 22, 25, 31, 41, 54, 70, 88,
24, 25, 29, 36, 47, 65, 88, 115
INTER8X8_LUMA
16, 16, 16, 16, 17, 18, 20, 24,
16, 16, 16, 17, 18, 20, 24, 25,
16, 16, 17, 18, 20, 24, 25, 28,
16, 17, 18, 20, 24, 25, 28, 33,
17, 18, 20, 24, 25, 28, 33, 41,
18, 20, 24, 25, 28, 33, 41, 54,
20, 24, 25, 28, 33, 41, 54, 71,
24, 25, 28, 33, 41, 54, 71, 91
INTER8X8_CHROMAU
16, 16, 16, 16, 17, 18, 20, 24,
16, 16, 16, 17, 18, 20, 24, 25,
16, 16, 17, 18, 20, 24, 25, 28,
16, 17, 18, 20, 24, 25, 28, 33,
17, 18, 20, 24, 25, 28, 33, 41,
18, 20, 24, 25, 28, 33, 41, 54,
20, 24, 25, 28, 33, 41, 54, 71,
24, 25, 28, 33, 41, 54, 71, 91
INTER8X8_CHROMAV
16, 16, 16, 16, 17, 18, 20, 24,
16, 16, 16, 17, 18, 20, 24, 25,
16, 16, 17, 18, 20, 24, 25, 28,
16, 17, 18, 20, 24, 25, 28, 33,
17, 18, 20, 24, 25, 28, 33, 41,
18, 20, 24, 25, 28, 33, 41, 54,
20, 24, 25, 28, 33, 41, 54, 71,
24, 25, 28, 33, 41, 54, 71, 91
INTRA16X16_LUMA
16, 16, 16, 16, 17, 18, 21, 24,
16, 16, 16, 16, 17, 19, 22, 25,
16, 16, 17, 18, 20, 22, 25, 29,
16, 16, 18, 21, 24, 27, 31, 36,
17, 17, 20, 24, 30, 35, 41, 47,
18, 19, 22, 27, 35, 44, 54, 65,
21, 22, 25, 31, 41, 54, 70, 88,
24, 25, 29, 36, 47, 65, 88, 115
INTRA16X16_CHROMAU
16, 16, 16, 16, 17, 18, 21, 24,
16, 16, 16, 16, 17, 19, 22, 25,
16, 16, 17, 18, 20, 22, 25, 29,
16, 16, 18, 21, 24, 27, 31, 36,
17, 17, 20, 24, 30, 35, 41, 47,
18, 19, 22, 27, 35, 44, 54, 65,
21, 22, 25, 31, 41, 54, 70, 88,
24, 25, 29, 36, 47, 65, 88, 115
INTRA16X16_CHROMAV
16, 16, 16, 16, 17, 18, 21, 24,
16, 16, 16, 16, 17, 19, 22, 25,
16, 16, 17, 18, 20, 22, 25, 29,
16, 16, 18, 21, 24, 27, 31, 36,
17, 17, 20, 24, 30, 35, 41, 47,
18, 19, 22, 27, 35, 44, 54, 65,
21, 22, 25, 31, 41, 54, 70, 88,
24, 25, 29, 36, 47, 65, 88, 115
INTER16X16_LUMA
16, 16, 16, 16, 17, 18, 20, 24,
16, 16, 16, 17, 18, 20, 24, 25,
16, 16, 17, 18, 20, 24, 25, 28,
16, 17, 18, 20, 24, 25, 28, 33,
17, 18, 20, 24, 25, 28, 33, 41,
18, 20, 24, 25, 28, 33, 41, 54,
20, 24, 25, 28, 33, 41, 54, 71,
24, 25, 28, 33, 41, 54, 71, 91
INTER16X16_CHROMAU
16, 16, 16, 16, 17, 18, 20, 24,
16, 16, 16, 17, 18, 20, 24, 25,
16, 16, 17, 18, 20, 24, 25, 28,
16, 17, 18, 20, 24, 25, 28, 33,
17, 18, 20, 24, 25, 28, 33, 41,
18, 20, 24, 25, 28, 33, 41, 54,
20, 24, 25, 28, 33, 41, 54, 71,
24, 25, 28, 33, 41, 54, 71, 91
INTER16X16_CHROMAV
16, 16, 16, 16, 17, 18, 20, 24,
16, 16, 16, 17, 18, 20, 24, 25,
16, 16, 17, 18, 20, 24, 25, 28,
16, 17, 18, 20, 24, 25, 28, 33,
17, 18, 20, 24, 25, 28, 33, 41,
18, 20, 24, 25, 28, 33, 41, 54,
20, 24, 25, 28, 33, 41, 54, 71,
24, 25, 28, 33, 41, 54, 71, 91
INTRA32X32_LUMA
16, 16, 16, 16, 17, 18, 21, 24,
16, 16, 16, 16, 17, 19, 22, 25,
16, 16, 17, 18, 20, 22, 25, 29,
16, 16, 18, 21, 24, 27, 31, 36,
17, 17, 20, 24, 30, 35, 41, 47,
18, 19, 22, 27, 35, 44, 54, 65,
21, 22, 25, 31, 41, 54, 70, 88,
24, 25, 29, 36, 47, 65, 88, 115
INTER32X32_LUMA
16, 16, 16, 16, 17, 18, 20, 24,
16, 16, 16, 17, 18, 20, 24, 25,
16, 16, 17, 18, 20, 24, 25, 28,
16, 17, 18, 20, 24, 25, 28, 33,
17, 18, 20, 24, 25, 28, 33, 41,
18, 20, 24, 25, 28, 33, 41, 54,
20, 24, 25, 28, 33, 41, 54, 71,
24, 25, 28, 33, 41, 54, 71, 91
INTRA16X16_LUMA_DC
16
INTRA16X16_CHROMAU_DC
16
INTRA16X16_CHROMAV_DC
16
INTER16X16_LUMA_DC
16
INTER16X16_CHROMAU_DC
16
INTER16X16_CHROMAV_DC
16
INTRA32X32_LUMA_DC
16
INTER32X32_LUMA_DC
16
2014-02-11 10:55:21 +00:00
cfg - > cqmfile = NULL ;
2020-11-02 12:02:42 +00:00
cfg - > fast_coeff_table_fn = NULL ;
2017-02-13 15:34:03 +00:00
cfg - > ref_frames = 1 ;
2016-09-27 21:15:46 +00:00
cfg - > gop_len = 4 ;
cfg - > gop_lowdelay = true ;
2015-04-21 11:18:34 +00:00
cfg - > bipred = 0 ;
2015-03-13 14:31:00 +00:00
cfg - > target_bitrate = 0 ;
2022-04-28 11:18:09 +00:00
cfg - > hash = UVG_HASH_CHECKSUM ;
2016-05-26 07:38:45 +00:00
cfg - > lossless = false ;
2016-08-10 08:58:15 +00:00
cfg - > tmvp_enable = true ;
2016-08-03 05:18:56 +00:00
cfg - > implicit_rdpcm = false ;
2018-06-19 10:39:23 +00:00
cfg - > fast_residual_cost_limit = 0 ;
2012-05-30 12:10:23 +00:00
2022-04-28 11:18:09 +00:00
cfg - > cu_split_termination = UVG_CU_SPLIT_TERMINATION_ZERO ;
2016-05-10 11:15:41 +00:00
2016-03-07 15:21:40 +00:00
cfg - > tiles_width_count = 1 ;
cfg - > tiles_height_count = 1 ;
cfg - > tiles_width_split = NULL ;
cfg - > tiles_height_split = NULL ;
2018-03-19 08:20:31 +00:00
2016-09-27 19:55:12 +00:00
cfg - > wpp = 1 ;
2014-12-05 13:21:46 +00:00
cfg - > owf = - 1 ;
2014-05-05 13:17:22 +00:00
cfg - > slice_count = 1 ;
cfg - > slice_addresses_in_ts = MALLOC ( int32_t , 1 ) ;
cfg - > slice_addresses_in_ts [ 0 ] = 0 ;
2018-03-19 08:20:31 +00:00
2016-09-28 20:05:21 +00:00
cfg - > threads = - 1 ;
2014-10-14 09:01:56 +00:00
cfg - > cpuid = 1 ;
2014-04-02 08:38:03 +00:00
2015-01-09 10:04:23 +00:00
// Defaults for what sizes of PUs are tried.
2020-03-17 15:57:18 +00:00
memset ( cfg - > pu_depth_inter . min , - 1 , sizeof ( cfg - > pu_depth_inter . min ) ) ;
memset ( cfg - > pu_depth_inter . max , - 1 , sizeof ( cfg - > pu_depth_inter . max ) ) ;
memset ( cfg - > pu_depth_intra . min , - 1 , sizeof ( cfg - > pu_depth_intra . min ) ) ;
memset ( cfg - > pu_depth_intra . max , - 1 , sizeof ( cfg - > pu_depth_intra . max ) ) ;
* cfg - > pu_depth_inter . min = 2 ; // 0-3
* cfg - > pu_depth_inter . max = 3 ; // 0-3
* cfg - > pu_depth_intra . min = 2 ; // 0-4
* cfg - > pu_depth_intra . max = 3 ; // 0-4
2015-01-09 10:04:23 +00:00
2015-04-16 14:17:17 +00:00
cfg - > add_encoder_info = true ;
2016-01-21 13:08:34 +00:00
cfg - > calc_psnr = true ;
2015-04-16 14:17:17 +00:00
2022-04-28 11:18:09 +00:00
cfg - > mv_constraint = UVG_MV_CONSTRAIN_NONE ;
2016-02-24 10:39:03 +00:00
2016-09-27 21:19:30 +00:00
cfg - > me_early_termination = 1 ;
2018-04-13 14:54:11 +00:00
cfg - > intra_rdo_et = 0 ;
2016-08-17 07:07:40 +00:00
2022-04-28 11:18:09 +00:00
cfg - > input_format = UVG_FORMAT_P420 ;
2016-08-16 16:03:21 +00:00
cfg - > input_bitdepth = 8 ;
2016-09-27 21:15:46 +00:00
cfg - > gop_lp_definition . d = 3 ;
cfg - > gop_lp_definition . t = 1 ;
2018-10-02 06:55:30 +00:00
cfg - > open_gop = true ;
2016-09-27 19:12:02 +00:00
2022-02-06 18:08:28 +00:00
cfg - > roi . file_path = NULL ;
2022-05-30 09:11:48 +00:00
cfg - > roi . format = UVG_ROI_TXT ;
2022-02-06 18:08:28 +00:00
2018-08-21 13:37:55 +00:00
cfg - > set_qp_in_cu = false ;
2017-01-16 06:47:21 +00:00
2017-04-19 12:47:47 +00:00
cfg - > erp_aqp = false ;
2022-04-28 11:18:09 +00:00
cfg - > slices = UVG_SLICES_NONE ;
2017-01-31 13:44:23 +00:00
2017-11-09 14:53:31 +00:00
cfg - > level = 62 ; // default hevc level, 6.2 (the highest)
cfg - > force_level = true ; // don't care about level limits by-default
2017-11-20 14:05:28 +00:00
cfg - > high_tier = false ;
2017-11-03 13:47:35 +00:00
2017-12-22 10:10:41 +00:00
cfg - > me_max_steps = ( uint32_t ) - 1 ;
2020-03-19 20:13:53 +00:00
cfg - > vaq = 0 ;
2020-01-23 10:43:19 +00:00
2022-04-28 11:18:09 +00:00
cfg - > scaling_list = UVG_SCALING_LIST_OFF ;
2018-11-15 07:33:02 +00:00
2021-10-25 08:25:23 +00:00
cfg - > max_merge = 6 ;
2019-07-06 18:34:29 +00:00
cfg - > early_skip = true ;
2019-02-12 07:21:03 +00:00
2019-11-20 06:38:33 +00:00
cfg - > intra_smoothing_disabled = false ;
2020-01-27 11:11:31 +00:00
cfg - > ml_pu_depth_intra = false ;
2019-09-11 12:49:56 +00:00
2020-01-31 12:22:32 +00:00
cfg - > partial_coding . startCTU_x = 0 ;
cfg - > partial_coding . startCTU_y = 0 ;
cfg - > partial_coding . fullWidth = 0 ;
cfg - > partial_coding . fullHeight = 0 ;
2020-02-04 19:06:21 +00:00
cfg - > zero_coeff_rdo = true ;
2019-11-20 06:38:33 +00:00
2022-04-28 11:18:09 +00:00
cfg - > rc_algorithm = UVG_NO_RC ;
2020-03-19 08:47:28 +00:00
cfg - > intra_bit_allocation = false ;
cfg - > clip_neighbour = true ;
2020-02-05 09:06:56 +00:00
2022-04-28 11:18:09 +00:00
cfg - > file_format = UVG_FORMAT_AUTO ;
2020-09-07 11:04:48 +00:00
2020-09-09 09:35:47 +00:00
cfg - > stats_file_prefix = NULL ;
2021-03-16 08:22:40 +00:00
2020-12-22 09:05:53 +00:00
cfg - > log2_parallel_merge_level = 2 ;
2020-11-03 11:23:19 +00:00
cfg - > fastrd_sampling_on = 0 ;
cfg - > fastrd_accuracy_check_on = 0 ;
2020-11-05 17:47:47 +00:00
cfg - > fastrd_learning_outdir_fn = NULL ;
2021-08-16 12:39:14 +00:00
2021-04-23 07:19:43 +00:00
cfg - > chroma_scale_out [ 0 ] [ 0 ] = cfg - > chroma_scale_in [ 0 ] [ 0 ] = 17 ;
cfg - > chroma_scale_out [ 0 ] [ 1 ] = cfg - > chroma_scale_in [ 0 ] [ 1 ] = 27 ;
cfg - > chroma_scale_out [ 0 ] [ 2 ] = cfg - > chroma_scale_in [ 0 ] [ 2 ] = 32 ;
cfg - > chroma_scale_out [ 0 ] [ 3 ] = cfg - > chroma_scale_in [ 0 ] [ 3 ] = 44 ;
cfg - > chroma_scale_out [ 0 ] [ 4 ] = cfg - > chroma_scale_in [ 0 ] [ 4 ] = - 1 ;
cfg - > chroma_scale_out [ 1 ] [ 0 ] = cfg - > chroma_scale_in [ 1 ] [ 0 ] = - 1 ;
cfg - > chroma_scale_out [ 2 ] [ 0 ] = cfg - > chroma_scale_in [ 2 ] [ 0 ] = - 1 ;
2021-07-25 20:34:45 +00:00
cfg - > mrl = false ;
2021-12-15 14:25:08 +00:00
cfg - > mip = false ;
2021-04-23 07:19:43 +00:00
2022-02-22 18:06:53 +00:00
cfg - > lfnst = false ;
2022-07-14 12:39:07 +00:00
cfg - > isp = false ;
2021-04-23 07:19:43 +00:00
parse_qp_map ( cfg , 0 ) ;
2021-03-16 08:22:40 +00:00
2021-08-16 12:39:14 +00:00
cfg - > jccr = 0 ;
2021-11-16 15:38:38 +00:00
cfg - > amvr = 0 ;
2021-11-16 15:18:29 +00:00
2021-11-17 09:33:57 +00:00
cfg - > cclm = 0 ;
2022-06-03 10:28:40 +00:00
2021-12-02 08:42:30 +00:00
cfg - > combine_intra_cus = 1 ;
2022-03-17 12:48:08 +00:00
cfg - > force_inter = 0 ;
2022-06-03 10:28:40 +00:00
cfg - > cabac_debug_file_name = NULL ;
2022-06-07 12:32:55 +00:00
cfg - > dual_tree = 0 ;
2022-11-16 10:27:28 +00:00
cfg - > min_qt_size [ 0 ] = 4 ;
cfg - > min_qt_size [ 1 ] = 4 ;
cfg - > min_qt_size [ 2 ] = 4 ;
2022-12-01 12:38:35 +00:00
cfg - > max_btt_depth [ 0 ] = 0 ;
2022-11-16 10:27:28 +00:00
cfg - > max_btt_depth [ 1 ] = 0 ;
2022-12-01 12:38:35 +00:00
cfg - > max_btt_depth [ 2 ] = 0 ;
2022-11-16 10:27:28 +00:00
cfg - > max_tt_size [ 0 ] = 64 ;
cfg - > max_bt_size [ 0 ] = 64 ;
cfg - > max_tt_size [ 1 ] = 64 ;
cfg - > max_bt_size [ 1 ] = 64 ;
cfg - > max_tt_size [ 2 ] = 64 ;
cfg - > max_bt_size [ 2 ] = 64 ;
2022-09-28 05:49:38 +00:00
cfg - > intra_rough_search_levels = 2 ;
2022-06-17 06:15:01 +00:00
cfg - > ibc = 0 ;
2012-05-30 12:10:23 +00:00
return 1 ;
}
2022-04-28 11:18:09 +00:00
int uvg_config_destroy ( uvg_config * cfg )
2012-05-30 12:10:23 +00:00
{
2015-08-25 09:46:19 +00:00
if ( cfg ) {
FREE_POINTER ( cfg - > cqmfile ) ;
2022-02-06 18:08:28 +00:00
FREE_POINTER ( cfg - > roi . file_path ) ;
2020-11-02 12:02:42 +00:00
FREE_POINTER ( cfg - > fast_coeff_table_fn ) ;
2015-08-25 09:46:19 +00:00
FREE_POINTER ( cfg - > tiles_width_split ) ;
FREE_POINTER ( cfg - > tiles_height_split ) ;
FREE_POINTER ( cfg - > slice_addresses_in_ts ) ;
2020-11-05 17:47:47 +00:00
FREE_POINTER ( cfg - > fastrd_learning_outdir_fn ) ;
2015-08-25 09:46:19 +00:00
}
2012-05-30 12:10:23 +00:00
free ( cfg ) ;
return 1 ;
}
2014-02-03 22:16:42 +00:00
static int atobool ( const char * str )
{
if ( ! strcmp ( str , " 1 " ) | |
! strcmp ( str , " true " ) | |
! strcmp ( str , " yes " ) )
return 1 ;
if ( ! strcmp ( str , " 0 " ) | |
! strcmp ( str , " false " ) | |
! strcmp ( str , " no " ) )
return 0 ;
return 0 ;
}
2016-06-07 07:55:26 +00:00
static int parse_enum_n ( const char * arg , unsigned num_chars , const char * const * names , int8_t * dst )
2014-02-06 19:45:37 +00:00
{
2014-02-21 13:15:59 +00:00
int8_t i ;
for ( i = 0 ; names [ i ] ; i + + ) {
2016-06-07 07:55:26 +00:00
if ( ! strncmp ( arg , names [ i ] , num_chars ) ) {
2014-02-06 19:45:37 +00:00
* dst = i ;
return 1 ;
}
2014-02-21 13:15:59 +00:00
}
2014-02-06 19:45:37 +00:00
return 0 ;
}
2016-06-07 07:55:26 +00:00
static int parse_enum ( const char * arg , const char * const * names , int8_t * dst )
{
return parse_enum_n ( arg , 255 , names , dst ) ;
}
2014-04-02 08:38:03 +00:00
static int parse_tiles_specification ( const char * const arg , int32_t * const ntiles , int32_t * * const array ) {
const char * current_arg = NULL ;
int32_t current_value ;
2014-05-05 13:17:52 +00:00
int32_t values [ MAX_TILES_PER_DIM ] ;
2018-03-19 08:20:31 +00:00
2014-04-02 08:38:03 +00:00
int i ;
2018-03-19 08:20:31 +00:00
2014-04-02 08:38:03 +00:00
//Free pointer in any case
if ( * array ) {
FREE_POINTER ( * array ) ;
}
2018-03-19 08:20:31 +00:00
2014-04-02 08:38:03 +00:00
//If the arg starts with u, we want an uniform split
if ( arg [ 0 ] = = ' u ' ) {
2016-03-07 15:21:40 +00:00
* ntiles = atoi ( arg + 1 ) ;
if ( MAX_TILES_PER_DIM < = * ntiles | | 1 > = * ntiles ) {
fprintf ( stderr , " Invalid number of tiles (1 <= %d <= %d = MAX_TILES_PER_DIM)! \n " , * ntiles , MAX_TILES_PER_DIM ) ;
2014-04-02 08:38:03 +00:00
return 0 ;
}
//Done with parsing
return 1 ;
}
2018-03-19 08:20:31 +00:00
2014-04-02 08:38:03 +00:00
//We have a comma-separated list of int for the split...
current_arg = arg ;
2016-03-07 15:21:40 +00:00
* ntiles = 1 ;
2014-04-02 08:38:03 +00:00
do {
int ret = sscanf ( current_arg , " %d " , & current_value ) ;
if ( ret ! = 1 ) {
fprintf ( stderr , " Could not parse integer \" %s \" ! \n " , current_arg ) ;
return 0 ;
}
current_arg = strchr ( current_arg , ' , ' ) ;
//Skip the , if we found one
if ( current_arg ) + + current_arg ;
2016-03-07 15:21:40 +00:00
values [ * ntiles - 1 ] = current_value ;
2014-04-02 08:38:03 +00:00
+ + ( * ntiles ) ;
2014-05-05 13:17:52 +00:00
if ( MAX_TILES_PER_DIM < = * ntiles ) break ;
2014-04-02 08:38:03 +00:00
} while ( current_arg ) ;
2018-03-19 08:20:31 +00:00
2016-03-07 15:21:40 +00:00
if ( MAX_TILES_PER_DIM < = * ntiles | | 1 > = * ntiles ) {
fprintf ( stderr , " Invalid number of tiles (1 <= %d <= %d = MAX_TILES_PER_DIM)! \n " , * ntiles , MAX_TILES_PER_DIM ) ;
2014-04-02 08:38:03 +00:00
return 0 ;
}
2018-03-19 08:20:31 +00:00
2016-03-07 15:21:40 +00:00
* array = MALLOC ( int32_t , * ntiles - 1 ) ;
2014-04-02 08:38:03 +00:00
if ( ! * array ) {
fprintf ( stderr , " Could not allocate array for tiles \n " ) ;
return 0 ;
}
2018-03-19 08:20:31 +00:00
2014-04-02 08:38:03 +00:00
//TODO: memcpy?
2016-03-07 15:21:40 +00:00
for ( i = 0 ; i < * ntiles - 1 ; + + i ) {
2014-04-02 08:38:03 +00:00
( * array ) [ i ] = values [ i ] ;
}
2018-03-19 08:20:31 +00:00
2014-04-02 08:38:03 +00:00
return 1 ;
}
2022-12-01 12:38:35 +00:00
2018-03-19 08:20:31 +00:00
static int parse_uint8 ( const char * numstr , uint8_t * number , int min , int max )
2017-08-28 15:15:13 +00:00
{
char * tail ;
2017-08-31 13:14:31 +00:00
int d = strtol ( numstr , & tail , 10 ) ;
2017-08-31 09:51:41 +00:00
if ( * tail | | d < min | | d > max ) {
fprintf ( stderr , " Expected number between %d and %d \n " , min , max ) ;
if ( number )
* number = 0 ;
return 0 ;
} else {
if ( number )
* number = ( uint8_t ) d ;
return 1 ;
}
2017-08-28 15:15:13 +00:00
}
2022-12-01 12:38:35 +00:00
2021-04-23 07:19:43 +00:00
static int parse_int8 ( const char * numstr , int8_t * number , int min , int max )
{
char * tail ;
int d = strtol ( numstr , & tail , 10 ) ;
if ( * tail | | d < min | | d > max ) {
fprintf ( stderr , " Expected number between %d and %d \n " , min , max ) ;
if ( number )
* number = 0 ;
return 0 ;
} else {
if ( number )
* number = ( int8_t ) d ;
return 1 ;
}
}
2022-12-01 12:38:35 +00:00
2017-08-28 15:15:13 +00:00
static int parse_array ( const char * array , uint8_t * coeff_key , int size ,
int min , int max )
{
char * key = strdup ( array ) ;
2017-08-31 09:51:41 +00:00
const char delim [ ] = " ,;: " ;
2017-08-28 15:15:13 +00:00
char * token ;
int i = 0 ;
token = strtok ( key , delim ) ;
while ( token ! = NULL & & i < size ) {
2017-08-31 09:51:41 +00:00
if ( ! parse_uint8 ( token , & coeff_key [ i ] , min , max ) )
{
free ( key ) ;
return 0 ;
}
2017-08-28 15:15:13 +00:00
i + + ;
token = strtok ( NULL , delim ) ;
}
if ( i > = size & & ( token ! = NULL ) ) {
2017-08-31 09:51:41 +00:00
fprintf ( stderr , " parsing failed : too many members. \n " ) ;
free ( key ) ;
2017-08-28 15:15:13 +00:00
return 0 ;
}
2022-12-01 12:38:35 +00:00
//else if (i<size){
// fprintf(stderr, "parsing failed : too few members.\n");
// free(key);
// return 0;
//}
2017-08-31 09:51:41 +00:00
free ( key ) ;
2022-12-01 12:38:35 +00:00
return i ;
2017-08-28 15:15:13 +00:00
}
2022-12-01 12:38:35 +00:00
2017-08-28 15:15:13 +00:00
2021-04-23 07:19:43 +00:00
static int parse_qp_scale_array ( const char * array , int8_t * out )
{
const int size = 16 ;
char * key = strdup ( array ) ;
const char delim [ ] = " ,;: " ;
char * token ;
int i = 0 ;
token = strtok ( key , delim ) ;
while ( token ! = NULL & & i < size ) {
if ( ! parse_int8 ( token , & out [ i ] , 0 , 63 ) )
{
free ( key ) ;
return 0 ;
}
i + + ;
token = strtok ( NULL , delim ) ;
}
if ( i > = size ) {
fprintf ( stderr , " parsing failed : too many members. \n " ) ;
free ( key ) ;
return 0 ;
}
out [ i ] = - 1 ;
free ( key ) ;
return 1 ;
}
2022-04-28 11:18:09 +00:00
static void parse_qp_map ( uvg_config * cfg , int index ) {
2021-04-23 07:19:43 +00:00
int i = 0 ;
for ( ; cfg - > chroma_scale_in [ index ] [ i ] ! = - 1 ; i + + ) ;
if ( cfg - > chroma_scale_out [ index ] [ i ] ! = - 1 ) return ;
assert ( i < 17 ) ;
// TODO: Move this to somewhere else when we have more than one table
cfg - > num_used_table = 1 ;
cfg - > qp_table_length_minus1 [ index ] = i - 2 ;
cfg - > qp_table_start_minus26 [ index ] = cfg - > chroma_scale_in [ index ] [ 0 ] - 26 ;
for ( i = 0 ; i < cfg - > qp_table_length_minus1 [ 0 ] + 1 ; i + + ) {
cfg - > delta_qp_in_val_minus1 [ index ] [ i ] = cfg - > chroma_scale_in [ index ] [ i + 1 ] - cfg - > chroma_scale_in [ index ] [ i ] - ( int8_t ) 1 ;
cfg - > delta_qp_out_val [ index ] [ i ] = cfg - > chroma_scale_out [ index ] [ i + 1 ] - cfg - > chroma_scale_out [ index ] [ i ] ;
}
}
2020-03-17 15:57:18 +00:00
static int parse_pu_depth_list ( const char * array , int32_t * depths_min , int32_t * depths_max , int size )
{
char * list = strdup ( array ) ;
char * token ;
int i = 0 ;
int ptr = - 1 ;
2021-11-29 12:20:09 +00:00
int len = ( int ) strlen ( list ) ;
2020-03-17 15:57:18 +00:00
int retval = 1 ;
2020-03-31 13:39:56 +00:00
//Reset depths in case multiple pu depth parameters are given
2020-03-31 13:57:12 +00:00
if ( size > 1 ) memset ( depths_max + 1 , - 1 , ( size - 1 ) * sizeof ( * depths_max ) ) ;
if ( size > 1 ) memset ( depths_min + 1 , - 1 , ( size - 1 ) * sizeof ( * depths_min ) ) ;
2020-03-31 13:39:56 +00:00
2020-03-17 15:57:18 +00:00
token = strtok ( list , " , " ) ;
while ( ptr < len & & list [ ptr + 1 ] = = ' , ' )
{
i + + ;
ptr + + ;
}
while ( retval & & token ! = NULL & & i < size ) {
retval & = ( sscanf ( token , " %d-%d " , & depths_min [ i ] , & depths_max [ i ] ) = = 2 ) ;
ptr + = ( retval ? 4 : 0 ) ;
i + + ;
token = strtok ( NULL , " , " ) ;
while ( ptr < len & & list [ ptr + 1 ] = = ' , ' ) {
i + + ;
ptr + + ;
}
}
if ( i > = size & & ( token ! = NULL ) ) {
fprintf ( stderr , " parsing failed : too many values. \n " ) ;
retval = 0 ;
}
free ( list ) ;
return retval ;
}
2014-05-05 13:17:22 +00:00
static int parse_slice_specification ( const char * const arg , int32_t * const nslices , int32_t * * const array ) {
const char * current_arg = NULL ;
int32_t current_value ;
int32_t values [ MAX_SLICES ] ;
2018-03-19 08:20:31 +00:00
2014-05-05 13:17:22 +00:00
int i ;
2018-03-19 08:20:31 +00:00
2014-05-05 13:17:22 +00:00
//Free pointer in any case
if ( * array ) {
FREE_POINTER ( * array ) ;
}
2018-03-19 08:20:31 +00:00
2014-05-05 13:17:22 +00:00
//If the arg starts with u, we want an uniform split
if ( arg [ 0 ] = = ' u ' ) {
* nslices = atoi ( arg + 1 ) ;
if ( MAX_SLICES < = * nslices | | 0 > = * nslices ) {
fprintf ( stderr , " Invalid number of tiles (0 < %d <= %d = MAX_SLICES)! \n " , * nslices + 1 , MAX_SLICES ) ;
return 0 ;
}
//Done with parsing
return 1 ;
}
2018-03-19 08:20:31 +00:00
2014-05-05 13:17:22 +00:00
//We have a comma-separated list of int for the split...
current_arg = arg ;
//We always have a slice starting at 0
values [ 0 ] = 0 ;
* nslices = 1 ;
do {
int ret = sscanf ( current_arg , " %d " , & current_value ) ;
if ( ret ! = 1 ) {
fprintf ( stderr , " Could not parse integer \" %s \" ! \n " , current_arg ) ;
return 0 ;
}
current_arg = strchr ( current_arg , ' , ' ) ;
//Skip the , if we found one
if ( current_arg ) + + current_arg ;
values [ * nslices ] = current_value ;
+ + ( * nslices ) ;
if ( MAX_SLICES < = * nslices ) break ;
} while ( current_arg ) ;
2018-03-19 08:20:31 +00:00
2014-05-05 13:17:22 +00:00
if ( MAX_SLICES < = * nslices | | 0 > = * nslices ) {
fprintf ( stderr , " Invalid number of slices (0 < %d <= %d = MAX_SLICES)! \n " , * nslices , MAX_SLICES ) ;
return 0 ;
}
2018-03-19 08:20:31 +00:00
2014-05-05 13:17:22 +00:00
* array = MALLOC ( int32_t , * nslices ) ;
if ( ! * array ) {
fprintf ( stderr , " Could not allocate array for slices \n " ) ;
return 0 ;
}
2018-03-19 08:20:31 +00:00
2014-05-05 13:17:22 +00:00
//TODO: memcpy?
for ( i = 0 ; i < * nslices ; + + i ) {
( * array ) [ i ] = values [ i ] ;
}
2018-03-19 08:20:31 +00:00
2014-05-05 13:17:22 +00:00
return 1 ;
}
2022-04-28 11:18:09 +00:00
int uvg_config_parse ( uvg_config * cfg , const char * name , const char * value )
2014-02-03 20:58:23 +00:00
{
2017-12-12 12:37:55 +00:00
static const char * const me_names [ ] = { " hexbs " , " tz " , " full " , " full8 " , " full16 " , " full32 " , " full64 " , " dia " , NULL } ;
2015-08-13 09:53:14 +00:00
static const char * const source_scan_type_names [ ] = { " progressive " , " tff " , " bff " , NULL } ;
2015-03-19 16:48:10 +00:00
2014-02-06 19:45:37 +00:00
static const char * const overscan_names [ ] = { " undef " , " show " , " crop " , NULL } ;
static const char * const videoformat_names [ ] = { " component " , " pal " , " ntsc " , " secam " , " mac " , " undef " , NULL } ;
2014-02-08 02:29:50 +00:00
static const char * const range_names [ ] = { " tv " , " pc " , NULL } ;
2014-02-06 19:45:37 +00:00
static const char * const colorprim_names [ ] = { " " , " bt709 " , " undef " , " " , " bt470m " , " bt470bg " , " smpte170m " ,
" smpte240m " , " film " , " bt2020 " , NULL } ;
static const char * const transfer_names [ ] = { " " , " bt709 " , " undef " , " " , " bt470m " , " bt470bg " , " smpte170m " ,
" smpte240m " , " linear " , " log100 " , " log316 " , " iec61966-2-4 " ,
" bt1361e " , " iec61966-2-1 " , " bt2020-10 " , " bt2020-12 " , NULL } ;
static const char * const colormatrix_names [ ] = { " GBR " , " bt709 " , " undef " , " " , " fcc " , " bt470bg " , " smpte170m " ,
" smpte240m " , " YCgCo " , " bt2020nc " , " bt2020c " , NULL } ;
2016-02-29 16:39:21 +00:00
static const char * const mv_constraint_names [ ] = { " none " , " frame " , " tile " , " frametile " , " frametilemargin " , NULL } ;
2016-03-17 15:00:43 +00:00
static const char * const hash_names [ ] = { " none " , " checksum " , " md5 " , NULL } ;
2014-02-06 19:45:37 +00:00
2016-05-10 11:15:41 +00:00
static const char * const cu_split_termination_names [ ] = { " zero " , " off " , NULL } ;
2016-06-06 12:47:31 +00:00
static const char * const me_early_termination_names [ ] = { " off " , " on " , " sensitive " , NULL } ;
2017-08-11 08:57:09 +00:00
static const char * const sao_names [ ] = { " off " , " edge " , " band " , " full " , NULL } ;
2020-11-20 12:48:37 +00:00
static const char * const alf_names [ ] = { " off " , " no-cc " , " full " , NULL } ;
2021-02-10 11:05:18 +00:00
2021-02-11 13:11:15 +00:00
static const char * const mts_names [ ] = { " off " , " intra " , " inter " , " both " , " implicit " , NULL } ;
2020-12-18 15:40:47 +00:00
2018-11-15 07:33:02 +00:00
static const char * const scaling_list_names [ ] = { " off " , " custom " , " default " , NULL } ;
2019-11-18 12:20:43 +00:00
static const char * const rc_algorithm_names [ ] = { " no-rc " , " lambda " , " oba " , NULL } ;
2020-09-09 13:05:59 +00:00
static const char * const file_format_names [ ] = { " auto " , " y4m " , " yuv " , NULL } ;
2020-09-07 11:04:48 +00:00
2019-07-09 13:26:03 +00:00
static const char * const preset_values [ 11 ] [ 25 * 2 ] = {
2018-03-19 08:20:31 +00:00
{
" ultrafast " ,
" rd " , " 0 " ,
2015-11-03 13:26:34 +00:00
" pu-depth-intra " , " 2-3 " ,
2020-04-14 22:22:33 +00:00
" pu-depth-inter " , " 1-2 " ,
2015-11-03 13:26:34 +00:00
" me " , " hexbs " ,
2020-04-14 22:22:33 +00:00
" gop " , " 8 " ,
2015-11-03 13:26:34 +00:00
" ref " , " 1 " ,
2020-04-14 22:22:33 +00:00
" bipred " , " 1 " ,
2016-09-27 21:02:27 +00:00
" deblock " , " 0:0 " ,
2015-11-03 13:26:34 +00:00
" signhide " , " 0 " ,
2020-04-14 22:22:33 +00:00
" subme " , " 0 " ,
2017-08-11 10:12:22 +00:00
" sao " , " off " ,
2015-11-03 13:26:34 +00:00
" rdoq " , " 0 " ,
2018-03-19 08:20:31 +00:00
" rdoq-skip " , " 0 " ,
" transform-skip " , " 0 " ,
2015-11-05 11:59:30 +00:00
" mv-rdo " , " 0 " ,
2018-03-19 08:20:31 +00:00
" full-intra-search " , " 0 " ,
2016-09-27 21:02:27 +00:00
" cu-split-termination " , " zero " ,
" me-early-termination " , " sensitive " ,
2018-04-13 14:54:11 +00:00
" intra-rdo-et " , " 0 " ,
2019-07-06 18:34:29 +00:00
" early-skip " , " 1 " ,
2018-07-16 09:28:21 +00:00
" fast-residual-cost " , " 28 " ,
2021-10-25 08:25:23 +00:00
" max-merge " , " 6 " ,
2018-03-19 08:20:31 +00:00
NULL
2015-11-03 13:26:34 +00:00
} ,
2018-03-19 08:20:31 +00:00
{
2015-11-03 13:26:34 +00:00
" superfast " ,
2018-03-19 08:20:31 +00:00
" rd " , " 0 " ,
2016-09-27 21:02:27 +00:00
" pu-depth-intra " , " 2-3 " ,
2020-04-14 22:22:33 +00:00
" pu-depth-inter " , " 1-2 " ,
2015-11-03 13:26:34 +00:00
" me " , " hexbs " ,
2020-04-14 22:22:33 +00:00
" gop " , " 8 " ,
2015-11-03 13:26:34 +00:00
" ref " , " 1 " ,
2020-04-14 22:22:33 +00:00
" bipred " , " 1 " ,
2016-09-27 21:02:27 +00:00
" deblock " , " 0:0 " ,
2015-11-03 13:26:34 +00:00
" signhide " , " 0 " ,
2018-03-19 08:20:31 +00:00
" subme " , " 2 " ,
2017-08-11 10:12:22 +00:00
" sao " , " full " ,
2015-11-03 13:26:34 +00:00
" rdoq " , " 0 " ,
2018-03-19 08:20:31 +00:00
" rdoq-skip " , " 0 " ,
2015-11-03 13:26:34 +00:00
" transform-skip " , " 0 " ,
2015-11-05 11:59:30 +00:00
" mv-rdo " , " 0 " ,
2018-03-19 08:20:31 +00:00
" full-intra-search " , " 0 " ,
2016-09-27 21:02:27 +00:00
" cu-split-termination " , " zero " ,
" me-early-termination " , " sensitive " ,
2018-04-13 14:54:11 +00:00
" intra-rdo-et " , " 0 " ,
2019-07-06 18:34:29 +00:00
" early-skip " , " 1 " ,
2018-07-16 09:28:21 +00:00
" fast-residual-cost " , " 28 " ,
2021-10-25 08:25:23 +00:00
" max-merge " , " 6 " ,
2015-11-03 13:26:34 +00:00
NULL
} ,
{
" veryfast " ,
2016-09-27 21:02:27 +00:00
" rd " , " 0 " ,
2018-03-19 08:20:31 +00:00
" pu-depth-intra " , " 2-3 " ,
" pu-depth-inter " , " 1-3 " ,
2015-11-03 13:26:34 +00:00
" me " , " hexbs " ,
2020-04-14 22:22:33 +00:00
" gop " , " 8 " ,
2016-09-27 21:02:27 +00:00
" ref " , " 1 " ,
2020-04-14 22:22:33 +00:00
" bipred " , " 1 " ,
2016-09-27 21:02:27 +00:00
" deblock " , " 0:0 " ,
2015-11-03 13:26:34 +00:00
" signhide " , " 0 " ,
2016-09-27 21:02:27 +00:00
" subme " , " 2 " ,
2017-08-11 10:12:22 +00:00
" sao " , " full " ,
2015-11-03 13:26:34 +00:00
" rdoq " , " 0 " ,
2018-03-19 08:20:31 +00:00
" rdoq-skip " , " 0 " ,
2015-11-03 13:26:34 +00:00
" transform-skip " , " 0 " ,
2015-11-05 11:59:30 +00:00
" mv-rdo " , " 0 " ,
2018-03-19 08:20:31 +00:00
" full-intra-search " , " 0 " ,
2016-09-27 21:02:27 +00:00
" cu-split-termination " , " zero " ,
" me-early-termination " , " sensitive " ,
2018-04-13 14:54:11 +00:00
" intra-rdo-et " , " 0 " ,
2019-07-06 18:34:29 +00:00
" early-skip " , " 1 " ,
2018-07-16 09:28:21 +00:00
" fast-residual-cost " , " 28 " ,
2021-10-25 08:25:23 +00:00
" max-merge " , " 6 " ,
2015-11-03 13:26:34 +00:00
NULL
} ,
{
" faster " ,
2018-03-19 08:20:31 +00:00
" rd " , " 0 " ,
2016-09-27 21:02:27 +00:00
" pu-depth-intra " , " 2-3 " ,
" pu-depth-inter " , " 1-3 " ,
2015-11-03 13:26:34 +00:00
" me " , " hexbs " ,
2020-04-14 22:22:33 +00:00
" gop " , " 8 " ,
2016-09-27 21:02:27 +00:00
" ref " , " 1 " ,
2020-04-14 22:22:33 +00:00
" bipred " , " 1 " ,
2016-09-27 21:02:27 +00:00
" deblock " , " 0:0 " ,
" signhide " , " 0 " ,
2018-03-19 08:20:31 +00:00
" subme " , " 4 " ,
2017-08-11 10:12:22 +00:00
" sao " , " full " ,
2015-11-03 13:26:34 +00:00
" rdoq " , " 0 " ,
2018-03-19 08:20:31 +00:00
" rdoq-skip " , " 0 " ,
2015-11-03 13:26:34 +00:00
" transform-skip " , " 0 " ,
2015-11-05 11:59:30 +00:00
" mv-rdo " , " 0 " ,
2018-03-19 08:20:31 +00:00
" full-intra-search " , " 0 " ,
2016-09-27 21:02:27 +00:00
" cu-split-termination " , " zero " ,
" me-early-termination " , " sensitive " ,
2018-04-13 14:54:11 +00:00
" intra-rdo-et " , " 0 " ,
2019-07-06 18:34:29 +00:00
" early-skip " , " 1 " ,
2018-07-16 09:28:21 +00:00
" fast-residual-cost " , " 0 " ,
2021-10-25 08:25:23 +00:00
" max-merge " , " 6 " ,
2015-11-03 13:26:34 +00:00
NULL
} ,
{
" fast " ,
2018-03-19 08:20:31 +00:00
" rd " , " 0 " ,
" pu-depth-intra " , " 1-3 " ,
2016-09-27 21:02:27 +00:00
" pu-depth-inter " , " 1-3 " ,
2015-11-03 13:26:34 +00:00
" me " , " hexbs " ,
2020-04-14 22:22:33 +00:00
" gop " , " 8 " ,
2018-03-19 08:20:31 +00:00
" ref " , " 2 " ,
2020-04-14 22:22:33 +00:00
" bipred " , " 1 " ,
2016-09-27 21:02:27 +00:00
" deblock " , " 0:0 " ,
" signhide " , " 0 " ,
2016-07-12 12:39:01 +00:00
" subme " , " 4 " ,
2017-08-11 10:12:22 +00:00
" sao " , " full " ,
2016-09-29 12:44:12 +00:00
" rdoq " , " 0 " ,
2018-03-19 08:20:31 +00:00
" rdoq-skip " , " 0 " ,
2015-11-03 13:26:34 +00:00
" transform-skip " , " 0 " ,
2015-11-05 11:59:30 +00:00
" mv-rdo " , " 0 " ,
2018-03-19 08:20:31 +00:00
" full-intra-search " , " 0 " ,
2016-09-27 21:02:27 +00:00
" cu-split-termination " , " zero " ,
2018-03-19 08:20:31 +00:00
" me-early-termination " , " sensitive " ,
2018-04-13 14:54:11 +00:00
" intra-rdo-et " , " 0 " ,
2019-07-06 18:34:29 +00:00
" early-skip " , " 1 " ,
2018-07-16 09:28:21 +00:00
" fast-residual-cost " , " 0 " ,
2021-10-25 08:25:23 +00:00
" max-merge " , " 6 " ,
2015-11-03 13:26:34 +00:00
NULL
} ,
{
" medium " ,
2018-03-19 08:20:31 +00:00
" rd " , " 0 " ,
2022-05-05 11:28:07 +00:00
" pu-depth-intra " , " 1-4 " ,
2018-03-19 08:20:31 +00:00
" pu-depth-inter " , " 0-3 " ,
2015-11-03 13:26:34 +00:00
" me " , " hexbs " ,
2020-04-14 22:22:33 +00:00
" gop " , " 16 " ,
2018-03-19 08:20:31 +00:00
" ref " , " 4 " ,
2020-04-14 22:22:33 +00:00
" bipred " , " 1 " ,
2016-09-27 21:02:27 +00:00
" deblock " , " 0:0 " ,
2016-09-29 12:44:12 +00:00
" signhide " , " 0 " ,
2016-07-12 12:39:01 +00:00
" subme " , " 4 " ,
2017-08-11 10:12:22 +00:00
" sao " , " full " ,
2016-09-27 21:02:27 +00:00
" rdoq " , " 1 " ,
2018-03-19 08:20:31 +00:00
" rdoq-skip " , " 0 " ,
2015-11-03 13:26:34 +00:00
" transform-skip " , " 0 " ,
2015-11-05 11:59:30 +00:00
" mv-rdo " , " 0 " ,
2018-03-19 08:20:31 +00:00
" full-intra-search " , " 0 " ,
2016-09-27 21:02:27 +00:00
" cu-split-termination " , " zero " ,
" me-early-termination " , " on " ,
2018-04-13 14:54:11 +00:00
" intra-rdo-et " , " 0 " ,
2019-07-06 18:34:29 +00:00
" early-skip " , " 1 " ,
2018-07-16 09:28:21 +00:00
" fast-residual-cost " , " 0 " ,
2021-10-25 08:25:23 +00:00
" max-merge " , " 6 " ,
2015-11-03 13:26:34 +00:00
NULL
} ,
{
" slow " ,
2020-04-14 22:22:33 +00:00
" rd " , " 1 " ,
2022-05-05 11:28:07 +00:00
" pu-depth-intra " , " 1-4 " ,
2018-03-19 08:20:31 +00:00
" pu-depth-inter " , " 0-3 " ,
2015-11-03 13:26:34 +00:00
" me " , " hexbs " ,
2020-04-14 22:22:33 +00:00
" gop " , " 16 " ,
2018-03-19 08:20:31 +00:00
" ref " , " 4 " ,
" bipred " , " 1 " ,
2016-09-27 21:02:27 +00:00
" deblock " , " 0:0 " ,
2018-03-19 08:20:31 +00:00
" signhide " , " 0 " ,
2016-07-12 12:39:01 +00:00
" subme " , " 4 " ,
2017-08-11 10:12:22 +00:00
" sao " , " full " ,
2016-09-27 21:02:27 +00:00
" rdoq " , " 1 " ,
2018-03-19 08:20:31 +00:00
" rdoq-skip " , " 0 " ,
2015-11-03 13:26:34 +00:00
" transform-skip " , " 0 " ,
2015-11-05 11:59:30 +00:00
" mv-rdo " , " 0 " ,
2018-03-19 08:20:31 +00:00
" full-intra-search " , " 0 " ,
2016-09-27 21:02:27 +00:00
" cu-split-termination " , " zero " ,
" me-early-termination " , " on " ,
2018-04-13 14:54:11 +00:00
" intra-rdo-et " , " 0 " ,
2019-07-06 18:34:29 +00:00
" early-skip " , " 1 " ,
2018-07-16 09:28:21 +00:00
" fast-residual-cost " , " 0 " ,
2021-10-25 08:25:23 +00:00
" max-merge " , " 6 " ,
2015-11-03 13:26:34 +00:00
NULL
} ,
{
" slower " ,
2018-03-19 08:20:31 +00:00
" rd " , " 2 " ,
2022-05-05 11:28:07 +00:00
" pu-depth-intra " , " 1-4 " ,
2015-11-03 13:26:34 +00:00
" pu-depth-inter " , " 0-3 " ,
2016-09-27 21:02:27 +00:00
" me " , " hexbs " ,
2020-04-14 22:22:33 +00:00
" gop " , " 16 " ,
2018-03-19 08:20:31 +00:00
" ref " , " 4 " ,
" bipred " , " 1 " ,
2016-09-27 21:02:27 +00:00
" deblock " , " 0:0 " ,
2015-11-03 13:26:34 +00:00
" signhide " , " 1 " ,
2016-07-12 12:39:01 +00:00
" subme " , " 4 " ,
2017-08-11 10:12:22 +00:00
" sao " , " full " ,
2015-11-03 13:26:34 +00:00
" rdoq " , " 1 " ,
2018-03-19 08:20:31 +00:00
" rdoq-skip " , " 0 " ,
2015-11-03 13:26:34 +00:00
" transform-skip " , " 0 " ,
2015-11-05 11:59:30 +00:00
" mv-rdo " , " 0 " ,
2018-03-19 08:20:31 +00:00
" full-intra-search " , " 0 " ,
2016-09-27 21:02:27 +00:00
" cu-split-termination " , " zero " ,
2018-03-19 08:20:31 +00:00
" me-early-termination " , " off " ,
2018-04-13 14:54:11 +00:00
" intra-rdo-et " , " 0 " ,
2019-07-06 18:34:29 +00:00
" early-skip " , " 1 " ,
2018-07-16 09:28:21 +00:00
" fast-residual-cost " , " 0 " ,
2021-10-25 08:25:23 +00:00
" max-merge " , " 6 " ,
2015-11-03 13:26:34 +00:00
NULL
} ,
{
" veryslow " ,
2018-03-19 08:20:31 +00:00
" rd " , " 2 " ,
2022-05-05 11:28:07 +00:00
" pu-depth-intra " , " 1-4 " ,
2015-11-03 13:26:34 +00:00
" pu-depth-inter " , " 0-3 " ,
2020-03-02 16:37:44 +00:00
" me " , " tz " ,
" gop " , " 16 " ,
2018-03-19 08:20:31 +00:00
" ref " , " 4 " ,
" bipred " , " 1 " ,
2016-09-27 21:02:27 +00:00
" deblock " , " 0:0 " ,
2015-11-03 13:26:34 +00:00
" signhide " , " 1 " ,
2016-07-12 12:39:01 +00:00
" subme " , " 4 " ,
2017-08-11 10:12:22 +00:00
" sao " , " full " ,
2015-11-03 13:26:34 +00:00
" rdoq " , " 1 " ,
2018-03-19 08:20:31 +00:00
" rdoq-skip " , " 0 " ,
2020-03-02 16:37:44 +00:00
" transform-skip " , " 1 " ,
2016-09-27 21:02:27 +00:00
" mv-rdo " , " 0 " ,
2018-03-19 08:20:31 +00:00
" full-intra-search " , " 0 " ,
2016-09-29 12:44:12 +00:00
" cu-split-termination " , " zero " ,
2018-03-19 08:20:31 +00:00
" me-early-termination " , " off " ,
2018-04-13 14:54:11 +00:00
" intra-rdo-et " , " 0 " ,
2019-07-06 18:34:29 +00:00
" early-skip " , " 1 " ,
2018-07-16 09:28:21 +00:00
" fast-residual-cost " , " 0 " ,
2021-10-25 08:25:23 +00:00
" max-merge " , " 6 " ,
2015-11-03 13:26:34 +00:00
NULL
} ,
{
" placebo " ,
2018-03-19 08:20:31 +00:00
" rd " , " 2 " ,
2022-05-05 11:28:07 +00:00
" pu-depth-intra " , " 1-4 " ,
2015-11-03 13:26:34 +00:00
" pu-depth-inter " , " 0-3 " ,
" me " , " tz " ,
2020-03-02 16:37:44 +00:00
" gop " , " 16 " ,
2016-09-27 21:02:27 +00:00
" ref " , " 4 " ,
2018-03-19 08:20:31 +00:00
" bipred " , " 1 " ,
2016-09-27 21:02:27 +00:00
" deblock " , " 0:0 " ,
2015-11-03 13:26:34 +00:00
" signhide " , " 1 " ,
2016-07-12 12:39:01 +00:00
" subme " , " 4 " ,
2017-08-11 10:12:22 +00:00
" sao " , " full " ,
2015-11-03 13:26:34 +00:00
" rdoq " , " 1 " ,
2016-09-27 21:02:27 +00:00
" rdoq-skip " , " 0 " ,
2015-11-03 13:26:34 +00:00
" transform-skip " , " 1 " ,
2015-11-05 11:59:30 +00:00
" mv-rdo " , " 1 " ,
2018-03-19 08:20:31 +00:00
" full-intra-search " , " 0 " ,
2016-09-27 21:02:27 +00:00
" cu-split-termination " , " off " ,
" me-early-termination " , " off " ,
2018-04-13 14:54:11 +00:00
" intra-rdo-et " , " 0 " ,
2020-04-14 22:22:33 +00:00
" early-skip " , " 0 " ,
2018-07-16 09:28:21 +00:00
" fast-residual-cost " , " 0 " ,
2021-10-25 08:25:23 +00:00
" max-merge " , " 6 " ,
2015-11-03 13:26:34 +00:00
NULL
} ,
2015-11-03 10:04:22 +00:00
{ NULL }
} ;
2014-02-03 20:58:23 +00:00
if ( ! name )
return 0 ;
2015-06-29 12:50:15 +00:00
2014-02-03 22:16:42 +00:00
if ( ! value )
value = " true " ;
2014-04-04 08:39:25 +00:00
// Treat "--no-param" as --param 0
if ( ( ! strncmp ( name , " no- " , 3 ) ) ) {
name + = 3 ;
2014-02-03 22:16:42 +00:00
value = atobool ( value ) ? " false " : " true " ;
}
2014-02-03 20:58:23 +00:00
2014-04-04 09:20:17 +00:00
# define OPT(STR) (!strcmp(name, STR))
2021-12-02 20:57:44 +00:00
if OPT ( " input-res " )
2015-07-29 16:07:27 +00:00
if ( ! strcmp ( value , " auto " ) ) {
return 1 ;
} else {
return ( sscanf ( value , " %dx%d " , & cfg - > width , & cfg - > height ) = = 2 ) ;
}
2016-01-14 20:08:35 +00:00
else if OPT ( " input-fps " ) {
int32_t fps_num , fps_denom ;
if ( sscanf ( value , " %d/%d " , & fps_num , & fps_denom ) = = 2 ) {
cfg - > framerate_num = fps_num ;
cfg - > framerate_denom = fps_denom ;
} else {
// Accept decimal notation, making sure not to round 0 to 1.
cfg - > framerate_num = ( int ) ( atof ( value ) * 1000 + 0.49 ) ;
cfg - > framerate_denom = 1000 ;
}
}
2014-04-04 09:20:17 +00:00
else if OPT ( " qp " )
2014-02-03 20:58:23 +00:00
cfg - > qp = atoi ( value ) ;
2015-06-29 12:50:15 +00:00
else if OPT ( " period " )
2014-02-03 20:58:23 +00:00
cfg - > intra_period = atoi ( value ) ;
2015-02-18 11:41:03 +00:00
else if OPT ( " vps-period " )
cfg - > vps_period = atoi ( value ) ;
2015-06-29 12:50:15 +00:00
else if OPT ( " ref " )
2014-02-18 15:45:54 +00:00
cfg - > ref_frames = atoi ( value ) ;
2021-04-30 06:51:41 +00:00
else if OPT ( " lmcs " ) {
cfg - > lmcs_enable = atobool ( value ) ;
}
2014-04-04 09:20:17 +00:00
else if OPT ( " deblock " ) {
2014-02-03 22:31:24 +00:00
int beta , tc ;
if ( 2 = = sscanf ( value , " %d:%d " , & beta , & tc ) ) {
cfg - > deblock_enable = 1 ;
cfg - > deblock_beta = beta ;
cfg - > deblock_tc = tc ;
2015-06-29 12:50:15 +00:00
} else {
2014-02-03 22:31:24 +00:00
cfg - > deblock_enable = atobool ( value ) ;
2014-02-06 12:32:30 +00:00
}
2014-02-03 22:31:24 +00:00
}
2017-08-11 08:57:09 +00:00
else if OPT ( " sao " ) {
int8_t sao_type = 0 ;
2017-08-11 10:26:55 +00:00
if ( ! parse_enum ( value , sao_names , & sao_type ) ) sao_type = atobool ( value ) ? 3 : 0 ;
2017-08-11 08:57:09 +00:00
cfg - > sao_type = sao_type ;
}
2020-11-20 12:48:37 +00:00
else if OPT ( " alf " ) {
int8_t alf_type = 0 ;
2020-11-24 16:38:32 +00:00
if ( ! parse_enum ( value , alf_names , & alf_type ) ) alf_type = atobool ( value ) ? 2 : 0 ;
cfg - > alf_type = alf_type ;
2020-11-20 12:48:37 +00:00
}
2014-04-04 09:20:17 +00:00
else if OPT ( " rdoq " )
2014-03-05 14:56:00 +00:00
cfg - > rdoq_enable = atobool ( value ) ;
2015-01-24 18:10:21 +00:00
else if OPT ( " signhide " )
cfg - > signhide_enable = ( bool ) atobool ( value ) ;
2014-06-17 12:32:05 +00:00
else if OPT ( " rd " )
2015-06-29 12:50:15 +00:00
cfg - > rdo = atoi ( value ) ;
2014-06-17 12:32:05 +00:00
else if OPT ( " full-intra-search " )
cfg - > full_intra_search = atobool ( value ) ;
2014-04-04 09:20:17 +00:00
else if OPT ( " transform-skip " )
2014-04-02 11:41:40 +00:00
cfg - > trskip_enable = atobool ( value ) ;
2022-06-06 06:28:06 +00:00
else if OPT ( " chroma-transform-skip " )
cfg - > chroma_trskip_enable = atobool ( value ) ;
2021-12-02 20:04:09 +00:00
else if OPT ( " tr-skip-max-size " ) {
cfg - > trskip_max_size = atoi ( value ) ;
if ( cfg - > trskip_max_size < 2 | | cfg - > trskip_max_size > 5 ) {
fprintf ( stderr , " tr-skip-max-size not between 2 and 5. \n " ) ;
return 0 ;
}
}
2020-12-18 15:40:47 +00:00
else if OPT ( " mts " ) {
int8_t mts_type = 0 ;
if ( ! parse_enum ( value , mts_names , & mts_type ) ) mts_type = atobool ( value ) ? 3 : 0 ;
cfg - > mts = mts_type ;
2022-04-28 11:18:09 +00:00
cfg - > mts_implicit = ( mts_type = = UVG_MTS_IMPLICIT ) ;
2020-12-18 15:40:47 +00:00
}
2015-07-06 06:47:18 +00:00
else if OPT ( " me " ) {
int8_t ime_algorithm = 0 ;
if ( ! parse_enum ( value , me_names , & ime_algorithm ) ) return 0 ;
cfg - > ime_algorithm = ime_algorithm ;
}
2015-06-29 12:50:15 +00:00
else if OPT ( " subme " )
2014-11-20 12:59:04 +00:00
cfg - > fme_level = atoi ( value ) ;
2015-08-13 09:53:14 +00:00
else if OPT ( " source-scan-type " )
return parse_enum ( value , source_scan_type_names , & cfg - > source_scan_type ) ;
2016-02-24 10:39:03 +00:00
else if OPT ( " mv-constraint " )
{
2022-04-28 11:18:09 +00:00
int8_t constraint = UVG_MV_CONSTRAIN_NONE ;
2016-02-24 10:39:03 +00:00
int result = parse_enum ( value , mv_constraint_names , & constraint ) ;
cfg - > mv_constraint = constraint ;
return result ;
}
2015-06-29 12:50:15 +00:00
else if OPT ( " sar " )
return sscanf ( value , " %d:%d " , & cfg - > vui . sar_width , & cfg - > vui . sar_height ) = = 2 ;
2014-04-04 09:20:17 +00:00
else if OPT ( " overscan " )
2015-06-29 12:50:15 +00:00
return parse_enum ( value , overscan_names , & cfg - > vui . overscan ) ;
2014-04-04 09:20:17 +00:00
else if OPT ( " videoformat " )
2015-06-29 12:50:15 +00:00
return parse_enum ( value , videoformat_names , & cfg - > vui . videoformat ) ;
2014-04-04 09:20:17 +00:00
else if OPT ( " range " )
2015-06-29 12:50:15 +00:00
return parse_enum ( value , range_names , & cfg - > vui . fullrange ) ;
2014-04-04 09:20:17 +00:00
else if OPT ( " colorprim " )
2015-06-29 12:50:15 +00:00
return parse_enum ( value , colorprim_names , & cfg - > vui . colorprim ) ;
2014-04-04 09:20:17 +00:00
else if OPT ( " transfer " )
2015-06-29 12:50:15 +00:00
return parse_enum ( value , transfer_names , & cfg - > vui . transfer ) ;
2014-04-04 09:20:17 +00:00
else if OPT ( " colormatrix " )
2015-06-29 12:50:15 +00:00
return parse_enum ( value , colormatrix_names , & cfg - > vui . colormatrix ) ;
else if OPT ( " chromaloc " )
cfg - > vui . chroma_loc = atoi ( value ) ;
2014-04-04 09:20:17 +00:00
else if OPT ( " aud " )
2014-02-06 22:35:15 +00:00
cfg - > aud_enable = atobool ( value ) ;
2017-02-05 10:51:31 +00:00
else if OPT ( " cqmfile " ) {
char * cqmfile = strdup ( value ) ;
if ( ! cqmfile ) {
fprintf ( stderr , " Failed to allocate memory for CQM file name. \n " ) ;
return 0 ;
}
FREE_POINTER ( cfg - > cqmfile ) ;
cfg - > cqmfile = cqmfile ;
2022-04-28 11:18:09 +00:00
cfg - > scaling_list = UVG_SCALING_LIST_CUSTOM ;
2018-11-15 07:33:02 +00:00
}
2020-09-25 16:18:56 +00:00
else if OPT ( " fast-coeff-table " ) {
char * fast_coeff_table_fn = strdup ( value ) ;
if ( ! fast_coeff_table_fn ) {
fprintf ( stderr , " Failed to allocate memory for fast coeff table file name. \n " ) ;
return 0 ;
}
FREE_POINTER ( cfg - > fast_coeff_table_fn ) ;
cfg - > fast_coeff_table_fn = fast_coeff_table_fn ;
}
2020-11-03 11:23:19 +00:00
else if OPT ( " fastrd-sampling " ) {
cfg - > fastrd_sampling_on = 1 ;
}
else if OPT ( " fastrd-accuracy-check " ) {
cfg - > fastrd_accuracy_check_on = 1 ;
}
2020-11-05 17:47:47 +00:00
else if OPT ( " fastrd-outdir " ) {
char * fastrd_learning_outdir_fn = strdup ( value ) ;
if ( ! fastrd_learning_outdir_fn ) {
2020-11-03 11:23:19 +00:00
fprintf ( stderr , " Failed to allocate memory for fast RD learning outfile name. \n " ) ;
return 0 ;
}
2020-11-05 17:47:47 +00:00
FREE_POINTER ( cfg - > fastrd_learning_outdir_fn ) ;
cfg - > fastrd_learning_outdir_fn = fastrd_learning_outdir_fn ;
2020-11-03 11:23:19 +00:00
}
2018-11-15 07:33:02 +00:00
else if OPT ( " scaling-list " ) {
2022-04-28 11:18:09 +00:00
int8_t scaling_list = UVG_SCALING_LIST_OFF ;
2018-11-15 07:33:02 +00:00
int result = parse_enum ( value , scaling_list_names , & scaling_list ) ;
cfg - > scaling_list = scaling_list ;
return result ;
2017-02-05 10:51:31 +00:00
}
2016-08-10 11:51:41 +00:00
else if OPT ( " tiles-width-split " ) {
int retval = parse_tiles_specification ( value , & cfg - > tiles_width_count , & cfg - > tiles_width_split ) ;
2018-03-19 08:20:31 +00:00
2016-08-10 11:51:41 +00:00
if ( cfg - > tiles_width_count > 1 & & cfg - > tmvp_enable ) {
cfg - > tmvp_enable = false ;
fprintf ( stderr , " Disabling TMVP because tiles are used. \n " ) ;
}
2016-10-26 23:07:10 +00:00
if ( cfg - > wpp ) {
cfg - > wpp = false ;
fprintf ( stderr , " Disabling WPP because tiles were enabled. \n " ) ;
}
2016-08-10 11:51:41 +00:00
return retval ;
}
else if OPT ( " tiles-height-split " ) {
int retval = parse_tiles_specification ( value , & cfg - > tiles_height_count , & cfg - > tiles_height_split ) ;
2018-03-19 08:20:31 +00:00
2016-08-10 11:51:41 +00:00
if ( cfg - > tiles_height_count > 1 & & cfg - > tmvp_enable ) {
cfg - > tmvp_enable = false ;
fprintf ( stderr , " Disabling TMVP because tiles are used. \n " ) ;
}
2016-10-26 23:07:10 +00:00
if ( cfg - > wpp ) {
cfg - > wpp = false ;
fprintf ( stderr , " Disabling WPP because tiles were enabled. \n " ) ;
}
2016-08-10 11:51:41 +00:00
return retval ;
}
2016-03-07 13:54:35 +00:00
else if OPT ( " tiles " )
{
// A simpler interface for setting tiles, accepting only uniform split.
2016-03-07 15:21:40 +00:00
unsigned width ;
unsigned height ;
2016-03-07 13:54:35 +00:00
if ( 2 ! = sscanf ( value , " %ux%u " , & width , & height ) ) {
fprintf ( stderr , " Wrong format for tiles. Expected \" %%ux%%u \" , but got \" %s \" \n " , value ) ;
return 0 ;
}
2016-03-07 15:21:40 +00:00
if ( MAX_TILES_PER_DIM < = width | | 1 > width ) {
2016-03-07 13:54:35 +00:00
fprintf ( stderr , " Invalid number of tiles (0 < %d <= %d = MAX_TILES_PER_DIM)! \n " , width , MAX_TILES_PER_DIM ) ;
return 0 ;
}
2016-03-07 15:21:40 +00:00
if ( MAX_TILES_PER_DIM < = height | | 1 > height ) {
2016-03-07 13:54:35 +00:00
fprintf ( stderr , " Invalid number of tiles (0 < %d <= %d = MAX_TILES_PER_DIM)! \n " , height , MAX_TILES_PER_DIM ) ;
return 0 ;
}
// Free split arrays incase they have already been set by another parameter.
FREE_POINTER ( cfg - > tiles_width_split ) ;
FREE_POINTER ( cfg - > tiles_height_split ) ;
2016-03-07 15:21:40 +00:00
cfg - > tiles_width_count = width ;
cfg - > tiles_height_count = height ;
2016-08-10 11:51:41 +00:00
if ( cfg - > tmvp_enable ) {
cfg - > tmvp_enable = false ;
fprintf ( stderr , " Disabling TMVP because tiles are used. \n " ) ;
}
2016-10-26 23:07:10 +00:00
if ( cfg - > wpp ) {
cfg - > wpp = false ;
fprintf ( stderr , " Disabling WPP because tiles were enabled. \n " ) ;
}
2016-03-07 13:54:35 +00:00
return 1 ;
}
2014-05-05 13:17:22 +00:00
else if OPT ( " wpp " )
cfg - > wpp = atobool ( value ) ;
2014-06-16 06:26:24 +00:00
else if OPT ( " owf " ) {
cfg - > owf = atoi ( value ) ;
2015-06-29 12:50:15 +00:00
if ( cfg - > owf = = 0 & & ! strcmp ( value , " auto " ) ) {
// -1 means automatic selection
2014-11-17 00:19:25 +00:00
cfg - > owf = - 1 ;
2014-06-16 06:26:24 +00:00
}
2017-01-31 13:44:23 +00:00
} else if OPT ( " slices " ) {
if ( ! strcmp ( value , " tiles " ) ) {
2022-04-28 11:18:09 +00:00
cfg - > slices = UVG_SLICES_TILES ;
2017-01-31 13:44:23 +00:00
return 1 ;
} else if ( ! strcmp ( value , " wpp " ) ) {
2022-04-28 11:18:09 +00:00
cfg - > slices = UVG_SLICES_WPP ;
2017-01-31 13:44:23 +00:00
return 1 ;
} else if ( ! strcmp ( value , " tiles+wpp " ) ) {
2022-04-28 11:18:09 +00:00
cfg - > slices = UVG_SLICES_TILES | UVG_SLICES_WPP ;
2017-01-31 13:44:23 +00:00
return 1 ;
} else {
return parse_slice_specification ( value , & cfg - > slice_count , & cfg - > slice_addresses_in_ts ) ;
}
2017-01-08 08:56:02 +00:00
} else if OPT ( " threads " ) {
2014-05-13 09:52:10 +00:00
cfg - > threads = atoi ( value ) ;
2017-01-08 08:56:02 +00:00
if ( cfg - > threads = = 0 & & ! strcmp ( value , " auto " ) ) {
// -1 means automatic selection
cfg - > threads = - 1 ;
}
}
2014-10-14 09:01:56 +00:00
else if OPT ( " cpuid " )
2018-03-23 10:32:27 +00:00
cfg - > cpuid = atobool ( value ) ;
2015-01-09 10:04:23 +00:00
else if OPT ( " pu-depth-inter " )
2022-04-28 11:18:09 +00:00
return parse_pu_depth_list ( value , cfg - > pu_depth_inter . min , cfg - > pu_depth_inter . max , UVG_MAX_GOP_LAYERS ) ;
2015-01-09 10:04:23 +00:00
else if OPT ( " pu-depth-intra " )
2022-04-28 11:18:09 +00:00
return parse_pu_depth_list ( value , cfg - > pu_depth_intra . min , cfg - > pu_depth_intra . max , UVG_MAX_GOP_LAYERS ) ;
2015-04-16 14:17:17 +00:00
else if OPT ( " info " )
cfg - > add_encoder_info = atobool ( value ) ;
2015-03-30 06:54:22 +00:00
else if OPT ( " gop " ) {
2015-11-13 16:19:13 +00:00
if ( ! strncmp ( value , " lp- " , 3 ) ) { // Handle GOPs starting with "lp-".
struct {
unsigned g ; // length
unsigned d ; // depth
unsigned t ; // temporal
2016-09-27 19:12:02 +00:00
} gop = { 0 , 0 , 0 } ;
2015-11-13 16:19:13 +00:00
2016-09-27 19:12:02 +00:00
// Parse --gop=lp-g#d#t#
if ( sscanf ( value , " lp-g%ud%ut%u " , & gop . g , & gop . d , & gop . t ) ! = 3 ) {
fprintf ( stderr , " Error in GOP syntax. Example: lp-g8d4t2 \n " ) ;
2015-11-13 16:19:13 +00:00
return 0 ;
}
if ( gop . g < 1 | | gop . g > 32 ) {
fprintf ( stderr , " gop.g must be between 1 and 32. \n " ) ;
2016-09-27 19:12:02 +00:00
return 0 ;
2015-11-13 16:19:13 +00:00
}
if ( gop . d < 1 | | gop . d > 8 ) {
fprintf ( stderr , " gop.d must be between 1 and 8. \n " ) ;
2016-09-27 19:12:02 +00:00
return 0 ;
2015-11-13 16:19:13 +00:00
}
if ( gop . t < 1 | | gop . t > 15 ) {
2016-09-27 19:12:02 +00:00
fprintf ( stderr , " gop.t must be between 1 and 15. \n " ) ;
return 0 ;
2015-11-13 16:19:13 +00:00
}
2016-09-27 19:12:02 +00:00
cfg - > gop_lowdelay = true ;
2015-11-13 16:19:13 +00:00
cfg - > gop_len = gop . g ;
2016-09-27 19:12:02 +00:00
cfg - > gop_lp_definition . d = gop . d ;
cfg - > gop_lp_definition . t = gop . t ;
2019-11-18 12:20:43 +00:00
2020-03-19 08:47:28 +00:00
cfg - > intra_bit_allocation = true ;
cfg - > clip_neighbour = false ;
2015-11-13 16:19:13 +00:00
} else if ( atoi ( value ) = = 8 ) {
2015-11-02 10:22:25 +00:00
cfg - > gop_lowdelay = 0 ;
2022-04-28 11:18:09 +00:00
cfg - > gop_len = sizeof ( uvg_gop_ra8 ) / sizeof ( uvg_gop_ra8 [ 0 ] ) ;
memcpy ( cfg - > gop , uvg_gop_ra8 , sizeof ( uvg_gop_ra8 ) ) ;
2020-03-19 08:47:28 +00:00
cfg - > intra_bit_allocation = false ;
cfg - > clip_neighbour = true ;
2018-02-14 13:08:40 +00:00
2018-02-14 14:01:21 +00:00
} else if ( atoi ( value ) = = 16 ) {
cfg - > gop_lowdelay = 0 ;
2022-04-28 11:18:09 +00:00
cfg - > gop_len = sizeof ( uvg_gop_ra16 ) / sizeof ( uvg_gop_ra16 [ 0 ] ) ;
memcpy ( cfg - > gop , uvg_gop_ra16 , sizeof ( uvg_gop_ra16 ) ) ;
2020-03-19 08:47:28 +00:00
cfg - > intra_bit_allocation = false ;
cfg - > clip_neighbour = true ;
2018-02-14 14:01:21 +00:00
2016-12-20 12:25:13 +00:00
} else if ( atoi ( value ) = = 0 ) {
//Disable gop
cfg - > gop_len = 0 ;
cfg - > gop_lowdelay = 0 ;
cfg - > gop_lp_definition . d = 0 ;
cfg - > gop_lp_definition . t = 0 ;
2015-06-29 12:50:15 +00:00
} else if ( atoi ( value ) ) {
fprintf ( stderr , " Input error: unsupported gop length, must be 0 or 8 \n " ) ;
2015-03-30 06:54:22 +00:00
return 0 ;
}
}
2020-02-15 20:15:34 +00:00
else if OPT ( " intra-qp-offset " ) {
cfg - > intra_qp_offset = atoi ( value ) ;
2020-03-31 08:56:44 +00:00
if ( cfg - > intra_qp_offset = = 0 & & ! strcmp ( value , " auto " ) )
{
cfg - > intra_qp_offset_auto = true ;
2020-03-31 09:58:40 +00:00
} else {
cfg - > intra_qp_offset_auto = false ;
2020-03-31 08:56:44 +00:00
}
2020-02-15 20:15:34 +00:00
}
2018-10-02 06:55:30 +00:00
else if OPT ( " open-gop " ) {
cfg - > open_gop = ( bool ) atobool ( value ) ;
}
2015-04-21 11:18:34 +00:00
else if OPT ( " bipred " )
cfg - > bipred = atobool ( value ) ;
2019-11-18 12:20:43 +00:00
else if OPT ( " bitrate " ) {
2015-03-13 14:31:00 +00:00
cfg - > target_bitrate = atoi ( value ) ;
2019-11-18 12:20:43 +00:00
if ( ! cfg - > rc_algorithm ) {
2022-04-28 11:18:09 +00:00
cfg - > rc_algorithm = UVG_LAMBDA ;
2019-11-18 12:20:43 +00:00
}
}
2015-11-03 10:04:22 +00:00
else if OPT ( " preset " ) {
int preset_line = 0 ;
2015-11-03 13:46:23 +00:00
// Accept numbers from 0 to 9.
if ( ( atoi ( value ) = = 0 & & ! strcmp ( value , " 0 " ) ) | | ( atoi ( value ) > = 1 & & atoi ( value ) < = 9 ) ) {
preset_line = atoi ( value ) ;
} else {
// Find the selected preset from the list
while ( preset_values [ preset_line ] [ 0 ] ! = NULL ) {
if ( ! strcmp ( value , preset_values [ preset_line ] [ 0 ] ) ) {
break ;
2015-11-03 10:04:22 +00:00
}
2015-11-03 13:46:23 +00:00
preset_line + + ;
2015-11-03 10:04:22 +00:00
}
}
2015-11-03 13:46:23 +00:00
if ( preset_values [ preset_line ] [ 0 ] ! = NULL ) {
fprintf ( stderr , " Using preset %s: " , value ) ;
// Loop all the name and value pairs and push to the config parser
for ( int preset_value = 1 ; preset_values [ preset_line ] [ preset_value ] ! = NULL ; preset_value + = 2 ) {
fprintf ( stderr , " --%s=%s " , preset_values [ preset_line ] [ preset_value ] , preset_values [ preset_line ] [ preset_value + 1 ] ) ;
2022-04-28 11:18:09 +00:00
uvg_config_parse ( cfg , preset_values [ preset_line ] [ preset_value ] , preset_values [ preset_line ] [ preset_value + 1 ] ) ;
2015-11-03 13:46:23 +00:00
}
fprintf ( stderr , " \n " ) ;
} else {
2015-11-03 10:04:22 +00:00
fprintf ( stderr , " Input error: unknown preset \" %s \" \n " , value ) ;
return 0 ;
}
}
2015-11-05 11:59:30 +00:00
else if OPT ( " mv-rdo " )
cfg - > mv_rdo = atobool ( value ) ;
2016-01-21 13:08:34 +00:00
else if OPT ( " psnr " )
cfg - > calc_psnr = ( bool ) atobool ( value ) ;
2016-03-17 15:00:43 +00:00
else if OPT ( " hash " )
{
int8_t hash ;
int result ;
if ( ( result = parse_enum ( value , hash_names , & hash ) ) ) {
cfg - > hash = hash ;
}
return result ;
}
2016-05-10 11:15:41 +00:00
else if OPT ( " cu-split-termination " )
2016-05-10 11:34:46 +00:00
{
2022-04-28 11:18:09 +00:00
int8_t mode = UVG_CU_SPLIT_TERMINATION_ZERO ;
2016-05-10 11:34:46 +00:00
int result = parse_enum ( value , cu_split_termination_names , & mode ) ;
cfg - > cu_split_termination = mode ;
return result ;
}
2016-06-06 12:47:31 +00:00
else if OPT ( " me-early-termination " ) {
int8_t mode = 0 ;
int result = parse_enum ( value , me_early_termination_names , & mode ) ;
cfg - > me_early_termination = mode ;
return result ;
}
2018-04-13 14:54:11 +00:00
else if OPT ( " intra-rdo-et " )
cfg - > intra_rdo_et = ( bool ) atobool ( value ) ;
2016-05-26 07:38:45 +00:00
else if OPT ( " lossless " )
cfg - > lossless = ( bool ) atobool ( value ) ;
2016-08-10 08:58:15 +00:00
else if OPT ( " tmvp " ) {
cfg - > tmvp_enable = atobool ( value ) ;
2016-08-10 11:51:41 +00:00
if ( cfg - > tiles_width_count > 1 | | cfg - > tiles_height_count > 1 ) {
fprintf ( stderr , " Cannot enable TMVP because tiles are used. \n " ) ;
cfg - > tmvp_enable = false ;
}
2016-08-10 08:58:15 +00:00
}
2016-08-17 07:07:40 +00:00
else if OPT ( " rdoq-skip " ) {
cfg - > rdoq_skip = atobool ( value ) ;
}
2016-08-16 16:03:21 +00:00
else if OPT ( " input-format " ) {
2022-04-28 11:18:09 +00:00
static enum uvg_input_format const formats [ ] = { UVG_FORMAT_P400 , UVG_FORMAT_P420 } ;
2016-08-16 16:03:21 +00:00
static const char * const format_names [ ] = { " P400 " , " P420 " , NULL } ;
int8_t format = 0 ;
if ( ! parse_enum ( value , format_names , & format ) ) {
fprintf ( stderr , " input-format not recognized. \n " ) ;
return 0 ;
}
cfg - > input_format = formats [ format ] ;
}
else if OPT ( " input-bitdepth " ) {
cfg - > input_bitdepth = atoi ( value ) ;
if ( cfg - > input_bitdepth < 8 | | cfg - > input_bitdepth > 16 ) {
fprintf ( stderr , " input-bitdepth not between 8 and 16. \n " ) ;
return 0 ;
}
2022-04-28 11:18:09 +00:00
if ( cfg - > input_bitdepth > 8 & & UVG_BIT_DEPTH = = 8 ) {
2016-08-16 16:03:21 +00:00
// Because the image is read straight into the reference buffers,
2022-04-28 11:18:09 +00:00
// reading >8 bit samples doesn't work when sizeof(uvg_pixel)==1.
2016-08-16 16:03:21 +00:00
fprintf ( stderr , " input-bitdepth can't be set to larger than 8 because "
2022-06-18 10:49:09 +00:00
" uvg266 is compiled with UVG_BIT_DEPTH=8. \n " ) ;
2016-08-16 16:03:21 +00:00
return 0 ;
}
}
2016-08-03 05:18:56 +00:00
else if OPT ( " implicit-rdpcm " )
cfg - > implicit_rdpcm = ( bool ) atobool ( value ) ;
2017-01-29 05:22:26 +00:00
2022-02-06 18:08:28 +00:00
else if OPT ( " roi " ) {
2022-05-30 09:11:48 +00:00
static enum uvg_roi_format const formats [ ] = { UVG_ROI_TXT , UVG_ROI_BIN } ;
2022-02-06 18:08:28 +00:00
static const char * const format_names [ ] = { " txt " , " bin " , NULL } ;
2017-01-29 05:22:26 +00:00
2022-02-06 18:08:28 +00:00
char * roi_file = strdup ( value ) ;
if ( ! roi_file ) {
fprintf ( stderr , " Failed to allocate memory for ROI file name. \n " ) ;
2017-01-16 06:47:21 +00:00
return 0 ;
}
2022-02-06 18:08:28 +00:00
FREE_POINTER ( cfg - > roi . file_path ) ;
cfg - > roi . file_path = roi_file ;
2017-01-29 05:22:26 +00:00
2022-02-06 18:08:28 +00:00
// Get file extension or the substring after the last dot
char * maybe_extension = strrchr ( cfg - > roi . file_path , ' . ' ) ;
if ( ! maybe_extension ) {
2022-05-30 09:11:48 +00:00
cfg - > roi . format = UVG_ROI_TXT ;
2022-02-06 18:08:28 +00:00
} else {
maybe_extension + + ;
int8_t format ;
bool unknown_format = ! parse_enum ( maybe_extension , format_names , & format ) ;
2022-05-30 09:11:48 +00:00
cfg - > roi . format = unknown_format ? UVG_ROI_TXT : formats [ format ] ;
2017-01-16 06:47:21 +00:00
}
}
2018-08-21 13:37:55 +00:00
else if OPT ( " set-qp-in-cu " ) {
cfg - > set_qp_in_cu = ( bool ) atobool ( value ) ;
}
2017-11-03 13:47:35 +00:00
else if OPT ( " erp-aqp " ) {
2017-04-19 12:47:47 +00:00
cfg - > erp_aqp = ( bool ) atobool ( value ) ;
2017-11-03 13:47:35 +00:00
}
else if ( OPT ( " level " ) | | OPT ( " force-level " ) ) {
if OPT ( " force-level " ) {
cfg - > force_level = true ;
} else {
cfg - > force_level = false ;
}
2017-11-20 14:05:28 +00:00
unsigned int num_first , num_second , level ;
2017-11-03 13:47:35 +00:00
int matched_amount = sscanf ( value , " %u.%u " , & num_first , & num_second ) ;
if ( matched_amount = = 2 ) {
// of form x.y
level = num_first * 10 + num_second ;
2017-11-08 14:15:41 +00:00
} else if ( matched_amount = = 1 ) {
2017-11-03 13:47:35 +00:00
// no dot
if ( num_first < 10 ) {
// of form x
level = num_first * 10 ;
2017-11-08 14:15:41 +00:00
} else {
2017-11-03 13:47:35 +00:00
// of form xx
level = num_first ;
}
2017-11-08 14:15:41 +00:00
} else {
2017-11-20 14:05:28 +00:00
fprintf ( stderr , " Invalid level value: \" %s \" \n " , value ) ;
return 0 ;
}
if ( level < 10 | | level > 62 ) {
fprintf ( stderr , " Level value of %s is out of bounds \n " , value ) ;
2017-11-03 13:47:35 +00:00
return 0 ;
}
cfg - > level = level ;
}
2017-11-20 14:05:28 +00:00
else if ( OPT ( " high-tier " ) ) {
cfg - > high_tier = true ;
}
2017-12-22 10:10:41 +00:00
else if ( OPT ( " me-steps " ) ) {
2018-01-10 13:16:52 +00:00
char * tailptr = NULL ;
long steps = strtol ( value , & tailptr , 0 ) ;
if ( * tailptr ! = ' \0 ' ) {
fprintf ( stderr , " Invalid me-steps value: \" %s \" " , value ) ;
return 0 ;
}
2018-01-16 12:38:43 +00:00
if ( steps < - 1 | | steps > UINT32_MAX ) {
2018-01-10 13:16:52 +00:00
fprintf ( stderr , " me-steps value is out of bounds: \" %s \" " , value ) ;
return 0 ;
}
cfg - > me_max_steps = ( uint32_t ) steps ;
2017-12-22 10:10:41 +00:00
}
2018-06-19 10:39:23 +00:00
else if ( OPT ( " fast-residual-cost " ) )
cfg - > fast_residual_cost_limit = atoi ( value ) ;
2020-04-20 13:21:40 +00:00
else if ( OPT ( " vaq " ) ) {
cfg - > vaq = ( int ) atoi ( value ) ;
}
2019-02-19 11:53:42 +00:00
else if ( OPT ( " max-merge " ) ) {
int max_merge = atoi ( value ) ;
2021-10-25 08:25:23 +00:00
if ( max_merge < 1 | | max_merge > 6 ) {
fprintf ( stderr , " max-merge needs to be between 1 and 6 \n " ) ;
2019-02-19 11:53:42 +00:00
return 0 ;
}
cfg - > max_merge = ( uint8_t ) max_merge ;
}
2019-07-06 18:34:29 +00:00
else if OPT ( " early-skip " ) {
2020-01-27 11:11:31 +00:00
cfg - > early_skip = ( bool ) atobool ( value ) ;
}
else if OPT ( " ml-pu-depth-intra " ) {
cfg - > ml_pu_depth_intra = ( bool ) atobool ( value ) ;
}
2020-01-31 12:22:32 +00:00
else if OPT ( " partial-coding " ) {
2020-01-31 14:02:29 +00:00
uint32_t firstCTU_x ;
uint32_t firstCTU_y ;
uint32_t fullWidth ;
uint32_t fullHeight ;
2020-01-27 11:11:31 +00:00
if ( 4 ! = sscanf ( value , " %u!%u!%u!%u " , & firstCTU_x ,
& firstCTU_y , & fullWidth , & fullHeight ) ) {
2020-01-31 12:22:32 +00:00
fprintf ( stderr , " invalid partial-coding options. Expected \" %%u!%%u!%%u!%%u \" , but got \" %s \" \n " , value ) ;
2020-01-27 11:11:31 +00:00
return 0 ;
}
2020-01-31 12:22:32 +00:00
cfg - > partial_coding . startCTU_x = firstCTU_x ;
cfg - > partial_coding . startCTU_y = firstCTU_y ;
cfg - > partial_coding . fullWidth = fullWidth ;
cfg - > partial_coding . fullHeight = fullHeight ;
2019-07-06 18:34:29 +00:00
}
2020-02-04 19:06:21 +00:00
else if OPT ( " zero-coeff-rdo " ) {
cfg - > zero_coeff_rdo = ( bool ) atobool ( value ) ;
}
2019-11-18 12:20:43 +00:00
else if OPT ( " rc-algorithm " ) {
int8_t rc_algorithm = 0 ;
if ( ! parse_enum ( value , rc_algorithm_names , & rc_algorithm ) ) {
fprintf ( stderr , " Invalid rate control algorithm %s. Valid values include %s, %s, and %s \n " , value ,
rc_algorithm_names [ 0 ] ,
rc_algorithm_names [ 1 ] ,
rc_algorithm_names [ 2 ] ) ;
return 0 ;
}
cfg - > rc_algorithm = rc_algorithm ;
}
else if OPT ( " intra-bits " ) {
cfg - > intra_bit_allocation = atobool ( value ) ;
}
2019-12-06 07:51:04 +00:00
else if OPT ( " clip-neighbour " ) {
cfg - > clip_neighbour = atobool ( value ) ;
}
2020-09-07 11:04:48 +00:00
else if OPT ( " input-file-format " ) {
int8_t file_format = 0 ;
2020-09-10 09:57:25 +00:00
if ( ! parse_enum ( value , file_format_names , & file_format ) ) {
2020-09-09 13:05:59 +00:00
fprintf ( stderr , " Invalid input file format %s. Valid values include %s, %s, and %s \n " , value ,
file_format_names [ 0 ] ,
file_format_names [ 1 ] ,
file_format_names [ 2 ] ) ;
2020-09-07 11:04:48 +00:00
return 0 ;
}
cfg - > file_format = file_format ;
}
2020-09-09 09:35:47 +00:00
else if OPT ( " stats-file-prefix " ) {
cfg - > stats_file_prefix = strdup ( value ) ;
2019-07-06 18:34:29 +00:00
}
2021-04-23 07:19:43 +00:00
else if OPT ( " chroma-qp-in " ) {
memset ( cfg - > chroma_scale_in [ 0 ] , 0 , 17 ) ;
const bool success = parse_qp_scale_array ( value , cfg - > chroma_scale_in [ 0 ] ) ;
parse_qp_map ( cfg , 0 ) ;
return success ;
}
else if OPT ( " chroma-qp-out " ) {
memset ( cfg - > chroma_scale_out [ 0 ] , 0 , 17 ) ;
const bool success = parse_qp_scale_array ( value , cfg - > chroma_scale_out [ 0 ] ) ;
parse_qp_map ( cfg , 0 ) ;
return success ;
}
2021-07-25 20:34:45 +00:00
else if OPT ( " mrl " ) {
2021-08-31 08:20:41 +00:00
cfg - > mrl = atobool ( value ) ;
2021-07-25 20:34:45 +00:00
}
2021-12-15 14:25:08 +00:00
else if OPT ( " mip " ) {
cfg - > mip = atobool ( value ) ;
}
2022-02-22 18:06:53 +00:00
else if OPT ( " lfnst " ) {
cfg - > lfnst = atobool ( value ) ;
}
2022-07-14 12:39:07 +00:00
else if OPT ( " isp " ) {
cfg - > isp = atobool ( value ) ;
}
2021-08-17 06:31:01 +00:00
else if OPT ( " jccr " ) {
cfg - > jccr = ( bool ) atobool ( value ) ;
}
2021-11-17 12:14:27 +00:00
else if OPT ( " amvr " ) {
cfg - > amvr = ( bool ) atobool ( value ) ;
}
2021-11-17 09:33:57 +00:00
else if OPT ( " cclm " ) {
cfg - > cclm = ( bool ) atobool ( value ) ;
}
2021-12-02 08:42:30 +00:00
else if OPT ( " combine-intra-cus " ) {
cfg - > combine_intra_cus = atobool ( value ) ;
}
2022-03-17 12:48:08 +00:00
else if OPT ( " force-inter " ) {
cfg - > force_inter = atobool ( value ) ;
}
2022-06-03 10:28:40 +00:00
else if OPT ( " cabac-debug-file " ) {
cfg - > cabac_debug_file_name = strdup ( value ) ;
if ( cfg - > cabac_debug_file_name = = NULL ) {
fprintf ( stderr , " Failed to allocate memory for cabac debug file name. \n " ) ;
return 0 ;
}
}
2022-06-07 12:32:55 +00:00
else if OPT ( " dual-tree " ) {
cfg - > dual_tree = atobool ( value ) ;
}
2022-12-01 12:38:35 +00:00
else if OPT ( " mtt-depth-intra " ) {
cfg - > max_btt_depth [ 0 ] = atoi ( value ) ;
}
else if OPT ( " mtt-depth-intra-chroma " ) {
cfg - > max_btt_depth [ 2 ] = atoi ( value ) ;
}
else if OPT ( " mtt-depth-inter " ) {
cfg - > max_btt_depth [ 1 ] = atoi ( value ) ;
}
else if OPT ( " max-bt-size " ) {
uint8_t sizes [ 3 ] ;
const int got = parse_array ( value , sizes , 3 , 0 , 128 ) ;
if ( got = = 1 ) {
cfg - > max_bt_size [ 0 ] = sizes [ 0 ] ;
cfg - > max_bt_size [ 1 ] = sizes [ 0 ] ;
cfg - > max_bt_size [ 2 ] = sizes [ 0 ] ;
}
else if ( got = = 3 ) {
cfg - > max_bt_size [ 0 ] = sizes [ 0 ] ;
cfg - > max_bt_size [ 1 ] = sizes [ 1 ] ;
cfg - > max_bt_size [ 2 ] = sizes [ 2 ] ;
} else {
fprintf ( stderr , " Incorrect amount of values provided for max-bt-size \n " ) ;
return 0 ;
}
}
else if OPT ( " max-tt-size " ) {
uint8_t sizes [ 3 ] ;
const int got = parse_array ( value , sizes , 3 , 0 , 128 ) ;
if ( got = = 1 ) {
cfg - > max_tt_size [ 0 ] = sizes [ 0 ] ;
cfg - > max_tt_size [ 1 ] = sizes [ 0 ] ;
cfg - > max_tt_size [ 2 ] = sizes [ 0 ] ;
}
else if ( got = = 3 ) {
cfg - > max_tt_size [ 0 ] = sizes [ 0 ] ;
cfg - > max_tt_size [ 1 ] = sizes [ 1 ] ;
cfg - > max_tt_size [ 2 ] = sizes [ 2 ] ;
} else {
fprintf ( stderr , " Incorrect amount of values provided for max-tt-size \n " ) ;
return 0 ;
}
}
2022-09-28 05:49:38 +00:00
else if OPT ( " intra-rough-granularity " ) {
cfg - > intra_rough_search_levels = atoi ( value ) ;
}
2022-06-17 06:15:01 +00:00
else if OPT ( " ibc " ) {
int ibc_value = atoi ( value ) ;
if ( ibc_value < 0 | | ibc_value > 2 ) {
fprintf ( stderr , " ibc supports only range from 0 to 2 \n " ) ;
return 0 ;
}
cfg - > ibc = ( uint8_t ) ibc_value ;
2022-12-01 12:38:35 +00:00
}
else {
2014-02-03 20:58:23 +00:00
return 0 ;
2017-11-03 13:47:35 +00:00
}
2014-02-03 20:58:23 +00:00
# undef OPT
2015-06-29 12:50:15 +00:00
return 1 ;
2014-02-03 20:58:23 +00:00
}
2022-04-28 11:18:09 +00:00
void uvg_config_process_lp_gop ( uvg_config * cfg )
2016-09-27 19:12:02 +00:00
{
struct {
unsigned g ;
unsigned d ;
unsigned t ;
} gop ;
gop . g = cfg - > gop_len ;
gop . d = cfg - > gop_lp_definition . d ;
gop . t = cfg - > gop_lp_definition . t ;
// Initialize modulos for testing depth.
// The picture belong to the lowest depth in which (poc % modulo) == 0.
unsigned depth_modulos [ 8 ] = { 0 } ;
2022-06-17 08:27:19 +00:00
for ( uint32_t d = 0 ; d < gop . d ; + + d ) {
2016-09-27 19:12:02 +00:00
depth_modulos [ gop . d - 1 - d ] = 1 < < d ;
}
depth_modulos [ 0 ] = gop . g ;
cfg - > gop_lowdelay = 1 ;
cfg - > gop_len = gop . g ;
2022-06-17 08:27:19 +00:00
for ( uint32_t g = 1 ; g < = gop . g ; + + g ) {
2022-04-28 11:18:09 +00:00
uvg_gop_config * gop_pic = & cfg - > gop [ g - 1 ] ;
2016-09-27 19:12:02 +00:00
// Find gop depth for picture.
2022-06-17 08:27:19 +00:00
uint32_t gop_layer = 1 ;
2016-09-28 19:47:57 +00:00
while ( gop_layer < gop . d & & ( g % depth_modulos [ gop_layer - 1 ] ) ) {
2016-09-27 19:12:02 +00:00
+ + gop_layer ;
}
gop_pic - > poc_offset = g ;
2016-09-28 19:47:57 +00:00
gop_pic - > layer = gop_layer ;
gop_pic - > qp_offset = gop_layer ;
2016-09-27 19:12:02 +00:00
gop_pic - > ref_pos_count = 0 ;
gop_pic - > ref_neg_count = cfg - > ref_frames ;
gop_pic - > is_ref = 0 ;
// Set first ref to point to previous frame, and the rest to previous
// key-frames.
// If gop.t > 1, have (poc % gop.t) == 0 point gop.t frames away,
// instead of the previous frame. Set the frames in between to
// point to the nearest frame with a lower gop-depth.
if ( gop . t > 1 ) {
if ( gop_pic - > poc_offset % gop . t = = 0 ) {
gop_pic - > ref_neg [ 0 ] = gop . t ;
} else {
int r = gop_pic - > poc_offset - 1 ;
while ( r > 0 ) {
if ( cfg - > gop [ r ] . layer < gop_pic - > layer ) break ;
- - r ;
}
// Var r is now 0 or index of the pic with layer < depth.
if ( cfg - > gop [ r ] . layer < gop_pic - > layer ) {
gop_pic - > ref_neg [ 0 ] = gop_pic - > poc_offset - cfg - > gop [ r ] . poc_offset ;
cfg - > gop [ r ] . is_ref = 1 ;
} else {
// No ref was found, just refer to the previous key-frame.
gop_pic - > ref_neg [ 0 ] = gop_pic - > poc_offset % gop . g ;
}
}
} else {
gop_pic - > ref_neg [ 0 ] = 1 ;
if ( gop_pic - > poc_offset > = 2 ) {
cfg - > gop [ gop_pic - > poc_offset - 2 ] . is_ref = 1 ;
}
}
int keyframe = gop_pic - > poc_offset ;
for ( int i = 1 ; i < gop_pic - > ref_neg_count ; + + i ) {
while ( keyframe = = gop_pic - > ref_neg [ i - 1 ] ) {
keyframe + = gop . g ;
}
gop_pic - > ref_neg [ i ] = keyframe ;
}
gop_pic - > qp_factor = 0.4624 ; // from HM
}
2022-06-17 08:27:19 +00:00
for ( uint32_t g = 0 ; g < gop . g ; + + g ) {
2022-04-28 11:18:09 +00:00
uvg_gop_config * gop_pic = & cfg - > gop [ g ] ;
2016-09-27 19:12:02 +00:00
if ( ! gop_pic - > is_ref ) {
gop_pic - > qp_factor = 0.68 * 1.31 ; // derived from HM
}
}
// Key-frame is always a reference.
cfg - > gop [ gop . g - 1 ] . is_ref = 1 ;
cfg - > gop [ gop . g - 1 ] . qp_factor = 0.578 ; // from HM
}
2017-11-28 14:19:44 +00:00
// forward declaration
2022-04-28 11:18:09 +00:00
static int validate_hevc_level ( uvg_config * const cfg ) ;
2017-11-08 14:15:41 +00:00
2014-03-10 13:53:23 +00:00
/**
2015-06-29 12:50:15 +00:00
* \ brief Check that configuration is sensible .
2014-03-10 13:53:23 +00:00
*
2015-06-29 12:50:15 +00:00
* \ param cfg config to check
* \ return 1 if the config is ok , otherwise 1
2014-03-10 13:53:23 +00:00
*/
2022-04-28 11:18:09 +00:00
int uvg_config_validate ( const uvg_config * const cfg )
2014-03-10 13:53:23 +00:00
{
2015-06-29 12:50:15 +00:00
int error = 0 ;
2020-02-07 17:45:41 +00:00
if ( cfg - > vaq < 0 ) {
fprintf ( stderr , " vaq strength must be positive \n " ) ;
error = 1 ;
}
2015-06-29 12:50:15 +00:00
if ( cfg - > width < = 0 ) {
2015-07-15 08:37:37 +00:00
fprintf ( stderr , " Input error: width must be positive \n " ) ;
2015-06-29 12:50:15 +00:00
error = 1 ;
}
2015-07-15 08:37:37 +00:00
if ( cfg - > height < = 0 ) {
fprintf ( stderr , " Input error: height must be positive \n " ) ;
2015-06-29 12:50:15 +00:00
error = 1 ;
}
2015-07-15 08:37:37 +00:00
if ( cfg - > width % 2 ! = 0 ) {
fprintf ( stderr , " Input error: width must be a multiple of two \n " ) ;
error = 1 ;
}
if ( cfg - > height % 2 ! = 0 ) {
fprintf ( stderr , " Input error: height must be a multiple of two \n " ) ;
error = 1 ;
}
2017-02-09 05:34:50 +00:00
if ( cfg - > width > 0 & & cfg - > height > 0 ) {
// We must be able to store the total number of luma and chroma pixels
// in an int32_t. For 4:4:4 chroma mode, the number of pixels is
// 3 * width * height. Width and height are rounded up to a multiple of
// LCU size.
const uint32_t max_lcus = INT_MAX / ( 3 * LCU_WIDTH * LCU_WIDTH ) ;
const uint64_t num_lcus = CEILDIV ( ( uint64_t ) cfg - > width , LCU_WIDTH ) *
CEILDIV ( ( uint64_t ) cfg - > height , LCU_WIDTH ) ;
if ( num_lcus > max_lcus ) {
fprintf ( stderr , " Input error: resolution %dx%d too large (max %u CTUs) \n " ,
cfg - > width , cfg - > height , max_lcus ) ;
error = 1 ;
}
}
2016-01-14 20:08:35 +00:00
if ( cfg - > framerate_num < 0 ) {
fprintf ( stderr , " Input error: --input-fps must >=0 \n " ) ;
error = 1 ;
}
if ( cfg - > framerate_denom < = 0 ) {
fprintf ( stderr , " Input error: --input-fps denominator must be >0 \n " ) ;
error = 1 ;
}
2015-06-29 12:50:15 +00:00
2018-06-18 09:14:32 +00:00
if ( cfg - > gop_len & &
cfg - > intra_period > 1 & &
! cfg - > gop_lowdelay & &
2016-09-27 19:25:11 +00:00
cfg - > intra_period % cfg - > gop_len ! = 0 )
{
2015-06-29 12:50:15 +00:00
fprintf ( stderr ,
2016-09-27 19:25:11 +00:00
" Input error: intra period (%d) not a multiple of B-gop length (%d) \n " ,
2015-06-29 12:50:15 +00:00
cfg - > intra_period ,
cfg - > gop_len ) ;
error = 1 ;
}
if ( cfg - > ref_frames < 1 | | cfg - > ref_frames > = MAX_REF_PIC_COUNT ) {
fprintf ( stderr , " Input error: --ref out of range [1..%d] \n " , MAX_REF_PIC_COUNT - 1 ) ;
error = 1 ;
}
if ( cfg - > deblock_beta < - 6 | | cfg - > deblock_beta > 6 ) {
fprintf ( stderr , " Input error: deblock beta parameter out of range [-6..6] \n " ) ;
error = 1 ;
}
if ( cfg - > deblock_tc < - 6 | | cfg - > deblock_tc > 6 ) {
fprintf ( stderr , " Input error: deblock tc parameter out of range [-6..6] \n " ) ;
error = 1 ;
}
2015-10-04 23:28:10 +00:00
if ( cfg - > rdo < 0 | | cfg - > rdo > 3 ) {
fprintf ( stderr , " Input error: --rd parameter out of range [0..3] \n " ) ;
2015-06-29 12:50:15 +00:00
error = 1 ;
}
2016-07-12 12:39:01 +00:00
if ( cfg - > fme_level ! = 0 & & cfg - > fme_level > 4 ) {
fprintf ( stderr , " Input error: invalid --subme parameter (must be in range 0-4) \n " ) ;
2015-06-29 12:50:15 +00:00
error = 1 ;
}
if ( cfg - > vui . chroma_loc < 0 | | cfg - > vui . chroma_loc > 5 ) {
fprintf ( stderr , " Input error: --chromaloc parameter out of range [0..5] \n " ) ;
error = 1 ;
}
if ( cfg - > owf < - 1 ) {
fprintf ( stderr , " Input error: --owf must be nonnegative or -1 \n " ) ;
error = 1 ;
}
2017-05-17 09:41:51 +00:00
if ( cfg - > qp ! = CLIP_TO_QP ( cfg - > qp ) ) {
fprintf ( stderr , " Input error: --qp parameter out of range [0..51] \n " ) ;
error = 1 ;
}
2020-02-15 20:15:34 +00:00
if ( abs ( cfg - > intra_qp_offset ) > 51 ) {
fprintf ( stderr , " Input error: --intra-qp-offset out of range [-51..51] \n " ) ;
error = 1 ;
}
2015-06-29 12:50:15 +00:00
if ( cfg - > target_bitrate < 0 ) {
fprintf ( stderr , " Input error: --bitrate must be nonnegative \n " ) ;
error = 1 ;
}
2022-04-28 11:18:09 +00:00
for ( size_t i = 0 ; i < UVG_MAX_GOP_LAYERS ; i + + )
2015-06-29 12:50:15 +00:00
{
2020-03-17 15:57:18 +00:00
if ( cfg - > pu_depth_inter . min [ i ] < 0 | | cfg - > pu_depth_inter . max [ i ] < 0 ) continue ;
2015-06-29 12:50:15 +00:00
2020-03-17 15:57:18 +00:00
if ( ! WITHIN ( cfg - > pu_depth_inter . min [ i ] , PU_DEPTH_INTER_MIN , PU_DEPTH_INTER_MAX ) | |
! WITHIN ( cfg - > pu_depth_inter . max [ i ] , PU_DEPTH_INTER_MIN , PU_DEPTH_INTER_MAX ) )
{
fprintf ( stderr , " Input error: illegal value for --pu-depth-inter (%d-%d) \n " ,
cfg - > pu_depth_inter . min [ i ] , cfg - > pu_depth_inter . max [ i ] ) ;
error = 1 ;
}
else if ( cfg - > pu_depth_inter . min [ i ] > cfg - > pu_depth_inter . max [ i ] )
{
fprintf ( stderr , " Input error: Inter PU depth min (%d) > max (%d) \n " ,
cfg - > pu_depth_inter . min [ i ] , cfg - > pu_depth_inter . max [ i ] ) ;
error = 1 ;
}
if ( cfg - > pu_depth_intra . min [ i ] < 0 | | cfg - > pu_depth_intra . max [ i ] < 0 ) continue ;
if ( ! WITHIN ( cfg - > pu_depth_intra . min [ i ] , PU_DEPTH_INTRA_MIN , PU_DEPTH_INTRA_MAX ) | |
! WITHIN ( cfg - > pu_depth_intra . max [ i ] , PU_DEPTH_INTRA_MIN , PU_DEPTH_INTRA_MAX ) )
{
fprintf ( stderr , " Input error: illegal value for --pu-depth-intra (%d-%d) \n " ,
cfg - > pu_depth_intra . min [ i ] , cfg - > pu_depth_intra . max [ i ] ) ;
error = 1 ;
}
else if ( cfg - > pu_depth_intra . min [ i ] > cfg - > pu_depth_intra . max [ i ] )
{
fprintf ( stderr , " Input error: Intra PU depth min (%d) > max (%d) \n " ,
cfg - > pu_depth_intra . min [ i ] , cfg - > pu_depth_intra . max [ i ] ) ;
error = 1 ;
}
2014-03-10 13:53:23 +00:00
}
2015-06-29 12:50:15 +00:00
// Tile separation should be at round position in terms of LCU, should be monotonic, and should not start by 0
2014-04-02 08:38:03 +00:00
if ( cfg - > tiles_width_split ) {
int i ;
int32_t prev_tile_split = 0 ;
2016-03-07 15:21:40 +00:00
for ( i = 0 ; i < cfg - > tiles_width_count - 1 ; + + i ) {
2014-04-02 08:38:03 +00:00
if ( cfg - > tiles_width_split [ i ] < = prev_tile_split ) {
fprintf ( stderr , " Input error: tile separations in width should be strictly monotonic (%d <= %d) \n " , cfg - > tiles_width_split [ i ] , prev_tile_split ) ;
2015-06-29 12:50:15 +00:00
error = 1 ;
break ;
2014-04-02 08:38:03 +00:00
}
if ( ( cfg - > tiles_width_split [ i ] % LCU_WIDTH ) ! = 0 ) {
fprintf ( stderr , " Input error: tile separation in width %d (at %d) is not at a multiple of LCU_WIDTH (%d) \n " , i , cfg - > tiles_width_split [ i ] , LCU_WIDTH ) ;
2015-06-29 12:50:15 +00:00
error = 1 ;
break ;
2014-04-02 08:38:03 +00:00
}
prev_tile_split = cfg - > tiles_width_split [ i ] ;
}
2016-03-07 15:21:40 +00:00
if ( cfg - > tiles_width_split [ cfg - > tiles_width_count - 2 ] > = cfg - > width ) {
fprintf ( stderr , " Input error: last x tile separation in width (%d) should smaller than image width (%d) \n " , cfg - > tiles_width_split [ cfg - > tiles_width_count - 2 ] , cfg - > width ) ;
2015-06-29 12:50:15 +00:00
error = 1 ;
2014-04-02 08:38:03 +00:00
}
}
2015-06-29 12:50:15 +00:00
2014-04-02 08:38:03 +00:00
if ( cfg - > tiles_height_split ) {
int i ;
int32_t prev_tile_split = 0 ;
2016-03-07 15:21:40 +00:00
for ( i = 0 ; i < cfg - > tiles_height_count - 1 ; + + i ) {
2014-04-02 08:38:03 +00:00
if ( cfg - > tiles_height_split [ i ] < = prev_tile_split ) {
fprintf ( stderr , " Input error: tile separations in height should be strictly monotonic (%d <= %d) \n " , cfg - > tiles_height_split [ i ] , prev_tile_split ) ;
2015-06-29 12:50:15 +00:00
error = 1 ;
break ;
2014-04-02 08:38:03 +00:00
}
if ( ( cfg - > tiles_height_split [ i ] % LCU_WIDTH ) ! = 0 ) {
fprintf ( stderr , " Input error: tile separation in height %d (at %d) is not at a multiple of LCU_WIDTH (%d) \n " , i , cfg - > tiles_height_split [ i ] , LCU_WIDTH ) ;
2015-06-29 12:50:15 +00:00
error = 1 ;
break ;
2014-04-02 08:38:03 +00:00
}
prev_tile_split = cfg - > tiles_height_split [ i ] ;
}
2015-06-29 12:50:15 +00:00
2016-03-07 15:21:40 +00:00
if ( cfg - > tiles_height_split [ cfg - > tiles_height_count - 2 ] > = cfg - > height ) {
fprintf ( stderr , " Input error: last tile separation in height (%d) should smaller than image height (%d) \n " , cfg - > tiles_height_split [ cfg - > tiles_height_count - 2 ] , cfg - > height ) ;
2015-06-29 12:50:15 +00:00
error = 1 ;
2014-04-02 08:38:03 +00:00
}
}
2015-06-29 12:50:15 +00:00
2016-08-03 05:18:56 +00:00
if ( cfg - > implicit_rdpcm & & ! cfg - > lossless ) {
fprintf ( stderr , " Input error: --implicit-rdpcm is not suppoted without --lossless \n " ) ;
error = 1 ;
}
2022-04-28 11:18:09 +00:00
if ( ( cfg - > slices & UVG_SLICES_WPP ) & & ! cfg - > wpp ) {
2017-01-31 13:44:23 +00:00
fprintf ( stderr , " Input error: --slices=wpp does not work without --wpp. \n " ) ;
error = 1 ;
}
2022-04-28 11:18:09 +00:00
if ( ( cfg - > scaling_list = = UVG_SCALING_LIST_CUSTOM ) & & ! cfg - > cqmfile ) {
2018-11-15 07:33:02 +00:00
fprintf ( stderr , " Input error: --scaling-list=custom does not work without --cqmfile=<FILE>. \n " ) ;
error = 1 ;
}
2022-04-28 11:18:09 +00:00
if ( validate_hevc_level ( ( uvg_config * const ) cfg ) ) {
2017-11-08 14:15:41 +00:00
// a level error found and it's not okay
error = 1 ;
}
2022-04-28 11:18:09 +00:00
if ( cfg - > target_bitrate > 0 & & cfg - > rc_algorithm = = UVG_NO_RC ) {
2019-11-18 12:20:43 +00:00
fprintf ( stderr , " Bitrate set but rc-algorithm is turned off. \n " ) ;
error = 1 ;
}
2022-04-28 11:18:09 +00:00
if ( cfg - > target_bitrate = = 0 & & cfg - > rc_algorithm ! = UVG_NO_RC ) {
2020-03-19 08:47:28 +00:00
fprintf ( stderr , " Rate control algorithm set but bitrate not set. \n " ) ;
error = 1 ;
}
2021-04-23 07:19:43 +00:00
for ( int index = 0 ; index < 3 ; index + + ) {
int i = 0 ;
if ( cfg - > chroma_scale_in [ index ] [ i ] ! = cfg - > chroma_scale_out [ index ] [ i ] ) {
fprintf ( stderr , " The starting points of chroma qp scaling list %d do not match. %d != %d " ,
index , cfg - > chroma_scale_in [ index ] [ i ] , cfg - > chroma_scale_out [ index ] [ i ] ) ;
error = 1 ;
}
for ( ; cfg - > chroma_scale_in [ index ] [ i ] ! = - 1 ; i + + ) ;
if ( cfg - > chroma_scale_out [ index ] [ i ] ! = - 1 ) {
fprintf ( stderr , " The chroma qp scaling lists of index %d are different lengths. \n " , index ) ;
error = 1 ;
}
}
2022-06-03 10:28:40 +00:00
if ( cfg - > owf ! = 0 & & cfg - > cabac_debug_file_name ) {
fprintf ( stderr , " OWF and cabac debugging are not supported at the same time. \n " ) ;
error = 1 ;
}
2022-06-06 06:28:06 +00:00
if ( cfg - > chroma_trskip_enable & & ! cfg - > trskip_enable ) {
fprintf ( stderr , " Transform skip has to be enabled when chroma transform skip is enabled. \n " ) ;
error = 1 ;
}
2022-09-28 05:49:38 +00:00
if ( cfg - > intra_rough_search_levels > 4 ) {
fprintf ( stderr , " intra-rough-granularity must be between [0..4]. \n " ) ;
error = 1 ;
}
2015-06-29 12:50:15 +00:00
return ! error ;
2014-03-10 13:53:23 +00:00
}
2017-11-08 14:15:41 +00:00
2022-04-28 11:18:09 +00:00
static int validate_hevc_level ( uvg_config * const cfg ) {
2017-11-20 14:05:28 +00:00
static const struct { uint32_t lsr ; uint32_t lps ; uint32_t main_bitrate ; } LEVEL_CONSTRAINTS [ 13 ] = {
2017-12-08 14:09:00 +00:00
{ 552960 , 36864 , 128 } , // 1
2017-11-20 11:25:26 +00:00
2017-12-08 14:09:00 +00:00
{ 3686400 , 122880 , 1500 } , // 2
{ 7372800 , 245760 , 3000 } , // 2.1
2017-11-20 11:25:26 +00:00
2017-12-08 14:16:36 +00:00
{ 16588800 , 552960 , 6000 } , // 3
2017-12-08 14:09:00 +00:00
{ 33177600 , 983040 , 10000 } , // 3.1
2017-11-20 11:25:26 +00:00
2017-12-08 14:09:00 +00:00
{ 66846720 , 2228224 , 12000 } , // 4
{ 133693440 , 2228224 , 20000 } , // 4.1
2017-11-20 11:25:26 +00:00
2017-12-08 14:16:36 +00:00
{ 267386880 , 8912896 , 25000 } , // 5
{ 534773760 , 8912896 , 40000 } , // 5.1
2017-12-08 14:09:00 +00:00
{ 1069547520 , 8912896 , 60000 } , // 5.2
2017-11-20 14:05:28 +00:00
2017-12-08 14:09:00 +00:00
{ 1069547520 , 35651584 , 60000 } , // 6
{ 2139095040 , 35651584 , 120000 } , // 6.1
{ 4278190080 , 35651584 , 240000 } , // 6.2
2017-11-20 14:05:28 +00:00
} ;
2017-11-20 11:25:26 +00:00
2017-11-20 14:05:28 +00:00
// bit rates for the high-tiers of the levels from 4 to 6.2
static const uint32_t HIGH_TIER_BITRATES [ 8 ] = {
2017-12-08 14:09:00 +00:00
30000 , 50000 , 100000 , 160000 , 240000 , 240000 , 480000 , 800000
2017-11-20 11:25:26 +00:00
} ;
2017-11-08 14:15:41 +00:00
2017-11-20 11:25:26 +00:00
int level_error = 0 ;
2018-03-19 08:20:31 +00:00
2017-11-08 14:15:41 +00:00
const char * level_err_prefix ;
if ( cfg - > force_level ) {
level_err_prefix = " Level warning " ;
} else {
level_err_prefix = " Level error " ;
}
2017-12-08 14:16:36 +00:00
uint8_t lvl_idx ;
2017-11-08 14:15:41 +00:00
2017-11-20 14:05:28 +00:00
// for nicer error print
float lvl = ( ( float ) cfg - > level ) / 10.0f ;
2017-11-20 11:25:26 +00:00
// check if the level is valid and get it's lsr and lps values
2017-11-09 14:53:31 +00:00
switch ( cfg - > level ) {
2017-11-08 14:15:41 +00:00
case 10 :
2017-11-20 11:25:26 +00:00
lvl_idx = 0 ;
2017-11-08 14:15:41 +00:00
break ;
case 20 :
2017-11-20 11:25:26 +00:00
lvl_idx = 1 ;
2017-11-08 14:15:41 +00:00
break ;
case 21 :
2017-11-20 11:25:26 +00:00
lvl_idx = 2 ;
2017-11-08 14:15:41 +00:00
break ;
case 30 :
2017-11-20 11:25:26 +00:00
lvl_idx = 3 ;
2017-11-08 14:15:41 +00:00
break ;
case 31 :
2017-11-20 11:25:26 +00:00
lvl_idx = 4 ;
2017-11-08 14:15:41 +00:00
break ;
case 40 :
2017-11-20 11:25:26 +00:00
lvl_idx = 5 ;
break ;
2017-11-08 14:15:41 +00:00
case 41 :
2017-11-20 11:25:26 +00:00
lvl_idx = 6 ;
2017-11-08 14:15:41 +00:00
break ;
case 50 :
2017-11-20 11:25:26 +00:00
lvl_idx = 7 ;
break ;
2017-11-08 14:15:41 +00:00
case 51 :
2017-11-20 11:25:26 +00:00
lvl_idx = 8 ;
break ;
2017-11-08 14:15:41 +00:00
case 52 :
2017-11-20 11:25:26 +00:00
lvl_idx = 9 ;
2017-11-08 14:15:41 +00:00
break ;
case 60 :
2017-11-20 11:25:26 +00:00
lvl_idx = 10 ;
break ;
2017-11-08 14:15:41 +00:00
case 61 :
2017-11-20 11:25:26 +00:00
lvl_idx = 11 ;
break ;
2017-11-08 14:15:41 +00:00
case 62 :
2017-11-20 11:25:26 +00:00
lvl_idx = 12 ;
2017-11-08 14:15:41 +00:00
break ;
default :
2017-11-20 14:05:28 +00:00
fprintf ( stderr , " Input error: %g is an invalid level value \n " , lvl ) ;
return 1 ;
}
if ( cfg - > high_tier & & cfg - > level < 40 ) {
fprintf ( stderr , " Input error: high tier requires at least level 4 \n " ) ;
2017-11-08 14:15:41 +00:00
return 1 ;
}
2017-11-20 11:25:26 +00:00
// max luma sample rate
uint32_t max_lsr = LEVEL_CONSTRAINTS [ lvl_idx ] . lsr ;
// max luma picture size
uint32_t max_lps = LEVEL_CONSTRAINTS [ lvl_idx ] . lps ;
2017-11-20 14:05:28 +00:00
if ( cfg - > high_tier ) {
2017-11-28 14:19:44 +00:00
cfg - > max_bitrate = HIGH_TIER_BITRATES [ lvl_idx - 5 ] * 1000 ;
2017-11-20 14:05:28 +00:00
} else {
2017-11-28 14:19:44 +00:00
cfg - > max_bitrate = LEVEL_CONSTRAINTS [ lvl_idx ] . main_bitrate * 1000 ;
2017-11-20 14:05:28 +00:00
}
2022-06-17 08:27:19 +00:00
if ( cfg - > target_bitrate > ( int32_t ) cfg - > max_bitrate ) {
2017-11-20 14:05:28 +00:00
fprintf ( stderr , " %s: target bitrate exceeds %i, which is the maximum %s tier level %g bitrate \n " ,
2017-11-28 14:19:44 +00:00
level_err_prefix , cfg - > max_bitrate , cfg - > high_tier ? " high " : " main " , lvl ) ;
2017-11-20 14:05:28 +00:00
level_error = 1 ;
}
2017-11-08 14:15:41 +00:00
// check the conformance to the level limits
2017-11-15 14:39:59 +00:00
// luma samples
2017-11-20 11:25:26 +00:00
uint64_t cfg_samples = cfg - > width * cfg - > height ;
2017-11-08 14:15:41 +00:00
2017-11-15 14:39:59 +00:00
// luma sample rate
2017-11-08 14:15:41 +00:00
double framerate = ( ( double ) cfg - > framerate_num ) / ( ( double ) cfg - > framerate_denom ) ;
2017-11-20 11:25:26 +00:00
uint64_t cfg_sample_rate = cfg_samples * ( uint64_t ) framerate ;
2017-11-08 14:15:41 +00:00
2017-11-15 14:39:59 +00:00
// square of the maximum allowed dimension
2017-11-20 11:25:26 +00:00
uint32_t max_dimension_squared = 8 * max_lps ;
2017-11-15 14:39:59 +00:00
// check maximum dimensions
2022-06-17 08:27:19 +00:00
if ( cfg - > width * cfg - > width > ( int32_t ) max_dimension_squared ) {
uint32_t max_dim = ( uint32_t ) sqrtf ( ( float ) max_dimension_squared ) ;
2017-11-20 11:25:26 +00:00
fprintf ( stderr , " %s: picture width of %i is too large for this level (%g), maximum dimension is %i \n " ,
level_err_prefix , cfg - > width , lvl , max_dim ) ;
2017-11-15 14:39:59 +00:00
level_error = 1 ;
}
2022-06-17 08:27:19 +00:00
if ( cfg - > height * cfg - > height > ( int32_t ) max_dimension_squared ) {
uint32_t max_dim = ( uint32_t ) sqrtf ( ( float ) max_dimension_squared ) ;
2017-11-20 11:25:26 +00:00
fprintf ( stderr , " %s: picture height of %i is too large for this level (%g), maximum dimension is %i \n " ,
level_err_prefix , cfg - > height , lvl , max_dim ) ;
2017-11-15 14:39:59 +00:00
level_error = 1 ;
}
// check luma picture size
2017-11-08 14:15:41 +00:00
if ( cfg_samples > max_lps ) {
2017-11-20 14:05:28 +00:00
fprintf ( stderr , " %s: picture resolution of %ix%i is too large for this level (%g) (it has %llu samples, maximum is %u samples) \n " ,
2017-12-08 14:16:36 +00:00
level_err_prefix , cfg - > width , cfg - > height , lvl , ( unsigned long long ) cfg_samples , max_lps ) ;
2017-11-08 14:15:41 +00:00
level_error = 1 ;
}
2017-11-15 14:39:59 +00:00
// check luma sample rate
2017-11-08 14:15:41 +00:00
if ( cfg_sample_rate > max_lsr ) {
2017-11-20 14:05:28 +00:00
fprintf ( stderr , " %s: framerate of %g is too big for this level (%g) and picture resolution (it has the sample rate of %llu, maximum is %u \n " ,
2017-12-08 14:16:36 +00:00
level_err_prefix , framerate , lvl , ( unsigned long long ) cfg_sample_rate , max_lsr ) ;
2017-11-08 14:15:41 +00:00
level_error = 1 ;
}
if ( cfg - > force_level ) {
2017-11-28 14:19:44 +00:00
// we wanted to print warnings, not get errors
2017-11-08 14:15:41 +00:00
return 0 ;
} else {
return level_error ;
}
}