1
0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-12-02 03:06:28 +02:00
FFmpeg/libavcodec/vvc/cabac.c
James Almer 1b9af306da avcodec: use the renamed av_zero_extend
Signed-off-by: James Almer <jamrial@gmail.com>
2024-06-13 20:36:09 -03:00

2479 lines
79 KiB
C

/*
* VVC CABAC decoder
*
* Copyright (C) 2021 Nuo Mi
*
* This file is part of FFmpeg.
*
* FFmpeg is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* FFmpeg is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with FFmpeg; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
#include "libavcodec/cabac_functions.h"
#include "cabac.h"
#include "ctu.h"
#include "data.h"
#define CABAC_MAX_BIN 31
#define CNU 35
enum SyntaxElement {
ALF_CTB_FLAG = 0,
ALF_USE_APS_FLAG = ALF_CTB_FLAG + 9,
ALF_CTB_CC_CB_IDC,
ALF_CTB_CC_CR_IDC = ALF_CTB_CC_CB_IDC + 3,
ALF_CTB_FILTER_ALT_IDX = ALF_CTB_CC_CR_IDC + 3,
SAO_MERGE_FLAG = ALF_CTB_FILTER_ALT_IDX + 2,
SAO_TYPE_IDX,
SPLIT_CU_FLAG,
SPLIT_QT_FLAG = SPLIT_CU_FLAG + 9,
MTT_SPLIT_CU_VERTICAL_FLAG = SPLIT_QT_FLAG + 6,
MTT_SPLIT_CU_BINARY_FLAG = MTT_SPLIT_CU_VERTICAL_FLAG + 5,
NON_INTER_FLAG = MTT_SPLIT_CU_BINARY_FLAG + 4,
CU_SKIP_FLAG = NON_INTER_FLAG + 2,
PRED_MODE_IBC_FLAG = CU_SKIP_FLAG + 3,
PRED_MODE_FLAG = PRED_MODE_IBC_FLAG + 3,
PRED_MODE_PLT_FLAG = PRED_MODE_FLAG + 2,
CU_ACT_ENABLED_FLAG,
INTRA_BDPCM_LUMA_FLAG,
INTRA_BDPCM_LUMA_DIR_FLAG,
INTRA_MIP_FLAG,
INTRA_LUMA_REF_IDX = INTRA_MIP_FLAG + 4,
INTRA_SUBPARTITIONS_MODE_FLAG = INTRA_LUMA_REF_IDX + 2,
INTRA_SUBPARTITIONS_SPLIT_FLAG,
INTRA_LUMA_MPM_FLAG,
INTRA_LUMA_NOT_PLANAR_FLAG,
INTRA_BDPCM_CHROMA_FLAG = INTRA_LUMA_NOT_PLANAR_FLAG + 2,
INTRA_BDPCM_CHROMA_DIR_FLAG,
CCLM_MODE_FLAG,
CCLM_MODE_IDX,
INTRA_CHROMA_PRED_MODE,
GENERAL_MERGE_FLAG,
INTER_PRED_IDC,
INTER_AFFINE_FLAG = INTER_PRED_IDC + 6,
CU_AFFINE_TYPE_FLAG = INTER_AFFINE_FLAG + 3,
SYM_MVD_FLAG,
REF_IDX_LX,
MVP_LX_FLAG = REF_IDX_LX + 2,
AMVR_FLAG,
AMVR_PRECISION_IDX = AMVR_FLAG + 2,
BCW_IDX = AMVR_PRECISION_IDX + 3,
CU_CODED_FLAG,
CU_SBT_FLAG,
CU_SBT_QUAD_FLAG = CU_SBT_FLAG + 2,
CU_SBT_HORIZONTAL_FLAG,
CU_SBT_POS_FLAG = CU_SBT_HORIZONTAL_FLAG + 3,
LFNST_IDX,
MTS_IDX = LFNST_IDX + 3,
COPY_ABOVE_PALETTE_INDICES_FLAG = MTS_IDX + 4,
PALETTE_TRANSPOSE_FLAG,
RUN_COPY_FLAG,
REGULAR_MERGE_FLAG = RUN_COPY_FLAG + 8,
MMVD_MERGE_FLAG = REGULAR_MERGE_FLAG + 2,
MMVD_CAND_FLAG,
MMVD_DISTANCE_IDX,
CIIP_FLAG,
MERGE_SUBBLOCK_FLAG,
MERGE_SUBBLOCK_IDX = MERGE_SUBBLOCK_FLAG + 3,
MERGE_IDX,
ABS_MVD_GREATER0_FLAG,
ABS_MVD_GREATER1_FLAG,
TU_Y_CODED_FLAG,
TU_CB_CODED_FLAG = TU_Y_CODED_FLAG + 4,
TU_CR_CODED_FLAG = TU_CB_CODED_FLAG + 2,
CU_QP_DELTA_ABS = TU_CR_CODED_FLAG + 3,
CU_CHROMA_QP_OFFSET_FLAG = CU_QP_DELTA_ABS + 2,
CU_CHROMA_QP_OFFSET_IDX,
TRANSFORM_SKIP_FLAG,
TU_JOINT_CBCR_RESIDUAL_FLAG = TRANSFORM_SKIP_FLAG + 2,
LAST_SIG_COEFF_X_PREFIX = TU_JOINT_CBCR_RESIDUAL_FLAG + 3,
LAST_SIG_COEFF_Y_PREFIX = LAST_SIG_COEFF_X_PREFIX +23,
SB_CODED_FLAG = LAST_SIG_COEFF_Y_PREFIX +23,
SIG_COEFF_FLAG = SB_CODED_FLAG + 7,
PAR_LEVEL_FLAG = SIG_COEFF_FLAG +63,
ABS_LEVEL_GTX_FLAG = PAR_LEVEL_FLAG +33,
COEFF_SIGN_FLAG = ABS_LEVEL_GTX_FLAG +72,
SYNTAX_ELEMENT_LAST = COEFF_SIGN_FLAG + 6,
};
static const uint8_t init_values[4][SYNTAX_ELEMENT_LAST] = {
{
//alf_ctb_flag
62, 39, 39, 54, 39, 39, 31, 39, 39,
//alf_use_aps_flag
46,
//alf_ctb_cc_cb_idc
18, 30, 31,
//alf_ctb_cc_cr_idc
18, 30, 31,
//alf_ctb_filter_alt_idx
11, 11,
//sao_merge_left_flag and sao_merge_up_flag
60,
//sao_type_idx_luma and sao_type_idx_chroma
13,
//split_cu_flag
19, 28, 38, 27, 29, 38, 20, 30, 31,
//split_qt_flag
27, 6, 15, 25, 19, 37,
//mtt_split_cu_vertical_flag
43, 42, 29, 27, 44,
//mtt_split_cu_binary_flag
36, 45, 36, 45,
//non_inter_flag
CNU, CNU,
//cu_skip_flag
0, 26, 28,
//pred_mode_ibc_flag
17, 42, 36,
//pred_mode_flag
CNU, CNU,
//pred_mode_plt_flag
25,
//cu_act_enabled_flag
52,
//intra_bdpcm_luma_flag
19,
//intra_bdpcm_luma_dir_flag
35,
//intra_mip_flag
33, 49, 50, 25,
//intra_luma_ref_idx
25, 60,
//intra_subpartitions_mode_flag
33,
//intra_subpartitions_split_flag
43,
//intra_luma_mpm_flag
45,
//intra_luma_not_planar_flag
13, 28,
//intra_bdpcm_chroma_flag
1,
//intra_bdpcm_chroma_dir_flag
27,
//cclm_mode_flag
59,
//cclm_mode_idx
27,
//intra_chroma_pred_mode
34,
//general_merge_flag
26,
//inter_pred_idc
CNU, CNU, CNU, CNU, CNU, CNU,
//inter_affine_flag
CNU, CNU, CNU,
//cu_affine_type_flag
CNU,
//sym_mvd_flag
CNU,
//ref_idx_l0 and ref_idx_l1
CNU, CNU,
//mvp_l0_flag and mvp_l1_flag
42,
//amvr_flag
CNU, CNU,
//amvr_precision_idx
35, 34, 35,
//bcw_idx
CNU,
//cu_coded_flag
6,
//cu_sbt_flag
CNU, CNU,
//cu_sbt_quad_flag
CNU,
//cu_sbt_horizontal_flag
CNU, CNU, CNU,
//cu_sbt_pos_flag
CNU,
//lfnst_idx
28, 52, 42,
//mts_idx
29, 0, 28, 0,
//copy_above_palette_indices_flag
42,
//palette_transpose_flag
42,
//run_copy_flag
50, 37, 45, 30, 46, 45, 38, 46,
//regular_merge_flag
CNU, CNU,
//mmvd_merge_flag
CNU,
//mmvd_cand_flag
CNU,
//mmvd_distance_idx
CNU,
//ciip_flag
CNU,
//merge_subblock_flag
CNU, CNU, CNU,
//merge_subblock_idx
CNU,
//merge_idx, merge_gpm_idx0, and merge_gpm_idx1
34,
//abs_mvd_greater0_flag
14,
//abs_mvd_greater1_flag
45,
//tu_y_coded_flag
15, 12, 5, 7,
//tu_cb_coded_flag
12, 21,
//tu_cr_coded_flag
33, 28, 36,
//cu_qp_delta_abs
CNU, CNU,
//cu_chroma_qp_offset_flag
CNU,
//cu_chroma_qp_offset_idx
CNU,
//transform_skip_flag
25, 9,
//tu_joint_cbcr_residual_flag
12, 21, 35,
//last_sig_coeff_x_prefix
13, 5, 4, 21, 14, 4, 6, 14, 21, 11, 14, 7, 14, 5, 11, 21,
30, 22, 13, 42, 12, 4, 3,
//last_sig_coeff_y_prefix
13, 5, 4, 6, 13, 11, 14, 6, 5, 3, 14, 22, 6, 4, 3, 6,
22, 29, 20, 34, 12, 4, 3,
//sb_coded_flag
18, 31, 25, 15, 18, 20, 38,
//sig_coeff_flag
25, 19, 28, 14, 25, 20, 29, 30, 19, 37, 30, 38, 11, 38, 46, 54,
27, 39, 39, 39, 44, 39, 39, 39, 18, 39, 39, 39, 27, 39, 39, 39,
0, 39, 39, 39, 25, 27, 28, 37, 34, 53, 53, 46, 19, 46, 38, 39,
52, 39, 39, 39, 11, 39, 39, 39, 19, 39, 39, 39, 25, 28, 38,
//par_level_flag
33, 25, 18, 26, 34, 27, 25, 26, 19, 42, 35, 33, 19, 27, 35, 35,
34, 42, 20, 43, 20, 33, 25, 26, 42, 19, 27, 26, 50, 35, 20, 43,
11,
//abs_level_gtx_flag
25, 25, 11, 27, 20, 21, 33, 12, 28, 21, 22, 34, 28, 29, 29, 30,
36, 29, 45, 30, 23, 40, 33, 27, 28, 21, 37, 36, 37, 45, 38, 46,
25, 1, 40, 25, 33, 11, 17, 25, 25, 18, 4, 17, 33, 26, 19, 13,
33, 19, 20, 28, 22, 40, 9, 25, 18, 26, 35, 25, 26, 35, 28, 37,
11, 5, 5, 14, 10, 3, 3, 3,
//coeff_sign_flag
12, 17, 46, 28, 25, 46,
},
{
//alf_ctb_flag
13, 23, 46, 4, 61, 54, 19, 46, 54,
//alf_use_aps_flag
46,
//alf_ctb_cc_cb_idc
18, 21, 38,
//alf_ctb_cc_cr_idc
18, 21, 38,
//alf_ctb_filter_alt_idx
20, 12,
//sao_merge_left_flag and sao_merge_up_flag
60,
//sao_type_idx_luma and sao_type_idx_chroma
5,
//split_cu_flag
11, 35, 53, 12, 6, 30, 13, 15, 31,
//split_qt_flag
20, 14, 23, 18, 19, 6,
//mtt_split_cu_vertical_flag
43, 35, 37, 34, 52,
//mtt_split_cu_binary_flag
43, 37, 21, 22,
//non_inter_flag
25, 12,
//cu_skip_flag
57, 59, 45,
//pred_mode_ibc_flag
0, 57, 44,
//pred_mode_flag
40, 35,
//pred_mode_plt_flag
0,
//cu_act_enabled_flag
46,
//intra_bdpcm_luma_flag
40,
//intra_bdpcm_luma_dir_flag
36,
//intra_mip_flag
41, 57, 58, 26,
//intra_luma_ref_idx
25, 58,
//intra_subpartitions_mode_flag
33,
//intra_subpartitions_split_flag
36,
//intra_luma_mpm_flag
36,
//intra_luma_not_planar_flag
12, 20,
//intra_bdpcm_chroma_flag
0,
//intra_bdpcm_chroma_dir_flag
13,
//cclm_mode_flag
34,
//cclm_mode_idx
27,
//intra_chroma_pred_mode
25,
//general_merge_flag
21,
//inter_pred_idc
7, 6, 5, 12, 4, 40,
//inter_affine_flag
12, 13, 14,
//cu_affine_type_flag
35,
//sym_mvd_flag
28,
//ref_idx_l0 and ref_idx_l1
20, 35,
//mvp_l0_flag and mvp_l1_flag
34,
//amvr_flag
59, 58,
//amvr_precision_idx
60, 48, 60,
//bcw_idx
4,
//cu_coded_flag
5,
//cu_sbt_flag
56, 57,
//cu_sbt_quad_flag
42,
//cu_sbt_horizontal_flag
20, 43, 12,
//cu_sbt_pos_flag
28,
//lfnst_idx
37, 45, 27,
//mts_idx
45, 40, 27, 0,
//copy_above_palette_indices_flag
59,
//palette_transpose_flag
42,
//run_copy_flag
51, 30, 30, 38, 23, 38, 53, 46,
//regular_merge_flag
38, 7,
//mmvd_merge_flag
26,
//mmvd_cand_flag
43,
//mmvd_distance_idx
60,
//ciip_flag
57,
//merge_subblock_flag
48, 57, 44,
//merge_subblock_idx
5,
//merge_idx, merge_gpm_idx0, and merge_gpm_idx1
20,
//abs_mvd_greater0_flag
44,
//abs_mvd_greater1_flag
43,
//tu_y_coded_flag
23, 5, 20, 7,
//tu_cb_coded_flag
25, 28,
//tu_cr_coded_flag
25, 29, 45,
//cu_qp_delta_abs
CNU, CNU,
//cu_chroma_qp_offset_flag
CNU,
//cu_chroma_qp_offset_idx
CNU,
//transform_skip_flag
25, 9,
//tu_joint_cbcr_residual_flag
27, 36, 45,
//last_sig_coeff_x_prefix
6, 13, 12, 6, 6, 12, 14, 14, 13, 12, 29, 7, 6, 13, 36, 28,
14, 13, 5, 26, 12, 4, 18,
//last_sig_coeff_y_prefix
5, 5, 12, 6, 6, 4, 6, 14, 5, 12, 14, 7, 13, 5, 13, 21,
14, 20, 12, 34, 11, 4, 18,
//sb_coded_flag
25, 30, 25, 45, 18, 12, 29,
//sig_coeff_flag
17, 41, 42, 29, 25, 49, 43, 37, 33, 58, 51, 30, 19, 38, 38, 46,
34, 54, 54, 39, 6, 39, 39, 39, 19, 39, 54, 39, 19, 39, 39, 39,
56, 39, 39, 39, 17, 34, 35, 21, 41, 59, 60, 38, 35, 45, 53, 54,
44, 39, 39, 39, 34, 38, 62, 39, 26, 39, 39, 39, 40, 35, 44,
//par_level_flag
18, 17, 33, 18, 26, 42, 25, 33, 26, 42, 27, 25, 34, 42, 42, 35,
26, 27, 42, 20, 20, 25, 25, 26, 11, 19, 27, 33, 42, 35, 35, 43,
3,
//abs_level_gtx_flag
0, 17, 26, 19, 35, 21, 25, 34, 20, 28, 29, 33, 27, 28, 29, 22,
34, 28, 44, 37, 38, 0, 25, 19, 20, 13, 14, 57, 44, 30, 30, 23,
17, 0, 1, 17, 25, 18, 0, 9, 25, 33, 34, 9, 25, 18, 26, 20,
25, 18, 19, 27, 29, 17, 9, 25, 10, 18, 4, 17, 33, 19, 20, 29,
18, 11, 4, 28, 2, 10, 3, 3,
//coeff_sign_flag
5, 10, 53, 43, 25, 46,
},
{
//alf_ctb_flag
33, 52, 46, 25, 61, 54, 25, 61, 54,
//alf_use_aps_flag
46,
//alf_ctb_cc_cb_idc
25, 35, 38,
//alf_ctb_cc_cr_idc
25, 28, 38,
//alf_ctb_filter_alt_idx
11, 26,
//sao_merge_left_flag and sao_merge_up_flag
2,
//sao_type_idx_luma and sao_type_idx_chroma
2,
//split_cu_flag
18, 27, 15, 18, 28, 45, 26, 7, 23,
//split_qt_flag
26, 36, 38, 18, 34, 21,
//mtt_split_cu_vertical_flag
43, 42, 37, 42, 44,
//mtt_split_cu_binary_flag
28, 29, 28, 29,
//non_inter_flag
25, 20,
//cu_skip_flag
57, 60, 46,
//pred_mode_ibc_flag
0, 43, 45,
//pred_mode_flag
40, 35,
//pred_mode_plt_flag
17,
//cu_act_enabled_flag
46,
//intra_bdpcm_luma_flag
19,
//intra_bdpcm_luma_dir_flag
21,
//intra_mip_flag
56, 57, 50, 26,
//intra_luma_ref_idx
25, 59,
//intra_subpartitions_mode_flag
33,
//intra_subpartitions_split_flag
43,
//intra_luma_mpm_flag
44,
//intra_luma_not_planar_flag
13, 6,
//intra_bdpcm_chroma_flag
0,
//intra_bdpcm_chroma_dir_flag
28,
//cclm_mode_flag
26,
//cclm_mode_idx
27,
//intra_chroma_pred_mode
25,
//general_merge_flag
6,
//inter_pred_idc
14, 13, 5, 4, 3, 40,
//inter_affine_flag
19, 13, 6,
//cu_affine_type_flag
35,
//sym_mvd_flag
28,
//ref_idx_l0 and ref_idx_l1
5, 35,
//mvp_l0_flag and mvp_l1_flag
34,
//amvr_flag
59, 50,
//amvr_precision_idx
38, 26, 60,
//bcw_idx
5,
//cu_coded_flag
12,
//cu_sbt_flag
41, 57,
//cu_sbt_quad_flag
42,
//cu_sbt_horizontal_flag
35, 51, 27,
//cu_sbt_pos_flag
28,
//lfnst_idx
52, 37, 27,
//mts_idx
45, 25, 27, 0,
//copy_above_palette_indices_flag
50,
//palette_transpose_flag
35,
//run_copy_flag
58, 45, 45, 30, 38, 45, 38, 46,
//regular_merge_flag
46, 15,
//mmvd_merge_flag
25,
//mmvd_cand_flag
43,
//mmvd_distance_idx
59,
//ciip_flag
57,
//merge_subblock_flag
25, 58, 45,
//merge_subblock_idx
4,
//merge_idx, merge_gpm_idx0, and merge_gpm_idx1
18,
//abs_mvd_greater0_flag
51,
//abs_mvd_greater1_flag
36,
//tu_y_coded_flag
15, 6, 5, 14,
//tu_cb_coded_flag
25, 37,
//tu_cr_coded_flag
9, 36, 45,
//cu_qp_delta_abs
CNU, CNU,
//cu_chroma_qp_offset_flag
CNU,
//cu_chroma_qp_offset_idx
CNU,
//transform_skip_flag
25, 17,
//tu_joint_cbcr_residual_flag
42, 43, 52,
//last_sig_coeff_x_prefix
6, 6, 12, 14, 6, 4, 14, 7, 6, 4, 29, 7, 6, 6, 12, 28,
7, 13, 13, 35, 19, 5, 4,
//last_sig_coeff_y_prefix
5, 5, 20, 13, 13, 19, 21, 6, 12, 12, 14, 14, 5, 4, 12, 13,
7, 13, 12, 41, 11, 5, 27,
//sb_coded_flag
25, 45, 25, 14, 18, 35, 45,
//sig_coeff_flag
17, 41, 49, 36, 1, 49, 50, 37, 48, 51, 58, 45, 26, 45, 53, 46,
49, 54, 61, 39, 35, 39, 39, 39, 19, 54, 39, 39, 50, 39, 39, 39,
0, 39, 39, 39, 9, 49, 50, 36, 48, 59, 59, 38, 34, 45, 38, 31,
58, 39, 39, 39, 34, 38, 54, 39, 41, 39, 39, 39, 25, 50, 37,
//par_level_flag
33, 40, 25, 41, 26, 42, 25, 33, 26, 34, 27, 25, 41, 42, 42, 35,
33, 27, 35, 42, 43, 33, 25, 26, 34, 19, 27, 33, 42, 43, 35, 43,
11,
//abs_level_gtx_flag
0, 0, 33, 34, 35, 21, 25, 34, 35, 28, 29, 40, 42, 43, 29, 30,
49, 36, 37, 45, 38, 0, 40, 34, 43, 36, 37, 57, 52, 45, 38, 46,
25, 0, 0, 17, 25, 26, 0, 9, 25, 33, 19, 0, 25, 33, 26, 20,
25, 33, 27, 35, 22, 25, 1, 25, 33, 26, 12, 25, 33, 27, 28, 37,
19, 11, 4, 6, 3, 4, 4, 5,
//coeff_sign_flag
35, 25, 46, 28, 33, 38,
},
//shiftIdx
{
//alf_ctb_flag
0, 0, 0, 4, 0, 0, 1, 0, 0,
//alf_use_aps_flag
0,
//alf_ctb_cc_cb_idc
4, 1, 4,
//alf_ctb_cc_cr_idc
4, 1, 4,
//alf_ctb_filter_alt_idx
0, 0,
//sao_merge_left_flag and sao_merge_up_flag
0,
//sao_type_idx_luma and sao_type_idx_chroma
4,
//split_cu_flag
12, 13, 8, 8, 13, 12, 5, 9, 9,
//split_qt_flag
0, 8, 8, 12, 12, 8,
//mtt_split_cu_vertical_flag
9, 8, 9, 8, 5,
//mtt_split_cu_binary_flag
12, 13, 12, 13,
//non_inter_flag
1, 0,
//cu_skip_flag
5, 4, 8,
//pred_mode_ibc_flag
1, 5, 8,
//pred_mode_flag
5, 1,
//pred_mode_plt_flag
1,
//cu_act_enabled_flag
1,
//intra_bdpcm_luma_flag
1,
//intra_bdpcm_luma_dir_flag
4,
//intra_mip_flag
9, 10, 9, 6,
//intra_luma_ref_idx
5, 8,
//intra_subpartitions_mode_flag
9,
//intra_subpartitions_split_flag
2,
//intra_luma_mpm_flag
6,
//intra_luma_not_planar_flag
1, 5,
//intra_bdpcm_chroma_flag
1,
//intra_bdpcm_chroma_dir_flag
0,
//cclm_mode_flag
4,
//cclm_mode_idx
9,
//intra_chroma_pred_mode
5,
//general_merge_flag
4,
//inter_pred_idc
0, 0, 1, 4, 4, 0,
//inter_affine_flag
4, 0, 0,
//cu_affine_type_flag
4,
//sym_mvd_flag
5,
//ref_idx_l0 and ref_idx_l1
0, 4,
//mvp_l0_flag and mvp_l1_flag
12,
//amvr_flag
0, 0,
//amvr_precision_idx
4, 5, 0,
//bcw_idx
1,
//cu_coded_flag
4,
//cu_sbt_flag
1, 5,
//cu_sbt_quad_flag
10,
//cu_sbt_horizontal_flag
8, 4, 1,
//cu_sbt_pos_flag
13,
//lfnst_idx
9, 9, 10,
//mts_idx
8, 0, 9, 0,
//copy_above_palette_indices_flag
9,
//palette_transpose_flag
5,
//run_copy_flag
9, 6, 9, 10, 5, 0, 9, 5,
//regular_merge_flag
5, 5,
//mmvd_merge_flag
4,
//mmvd_cand_flag
10,
//mmvd_distance_idx
0,
//ciip_flag
1,
//merge_subblock_flag
4, 4, 4,
//merge_subblock_idx
0,
//merge_idx, merge_gpm_idx0, and merge_gpm_idx1
4,
//abs_mvd_greater0_flag
9,
//abs_mvd_greater1_flag
5,
//tu_y_coded_flag
5, 1, 8, 9,
//tu_cb_coded_flag
5, 0,
//tu_cr_coded_flag
2, 1, 0,
//cu_qp_delta_abs
8, 8,
//cu_chroma_qp_offset_flag
8,
//cu_chroma_qp_offset_idx
8,
//transform_skip_flag
1, 1,
//tu_joint_cbcr_residual_flag
1, 1, 0,
//last_sig_coeff_x_prefix
8, 5, 4, 5, 4, 4, 5, 4, 1, 0, 4, 1, 0, 0, 0, 0,
1, 0, 0, 0, 5, 4, 4,
//last_sig_coeff_y_prefix
8, 5, 8, 5, 5, 4, 5, 5, 4, 0, 5, 4, 1, 0, 0, 1,
4, 0, 0, 0, 6, 5, 5,
//sb_coded_flag
8, 5, 5, 8, 5, 8, 8,
//sig_coeff_flag
12, 9, 9, 10, 9, 9, 9, 10, 8, 8, 8, 10, 9, 13, 8, 8,
8, 8, 8, 5, 8, 0, 0, 0, 8, 8, 8, 8, 8, 0, 4, 4,
0, 0, 0, 0, 12, 12, 9, 13, 4, 5, 8, 9, 8, 12, 12, 8,
4, 0, 0, 0, 8, 8, 8, 8, 4, 0, 0, 0, 13, 13, 8,
//par_level_flag
8, 9, 12, 13, 13, 13, 10, 13, 13, 13, 13, 13, 13, 13, 13, 13,
10, 13, 13, 13, 13, 8, 12, 12, 12, 13, 13, 13, 13, 13, 13, 13,
6,
//abs_level_gtx_flag
9, 5, 10, 13, 13, 10, 9, 10, 13, 13, 13, 9, 10, 10, 10, 13,
8, 9, 10, 10, 13, 8, 8, 9, 12, 12, 10, 5, 9, 9, 9, 13,
1, 5, 9, 9, 9, 6, 5, 9, 10, 10, 9, 9, 9, 9, 9, 9,
6, 8, 9, 9, 10, 1, 5, 8, 8, 9, 6, 6, 9, 8, 8, 9,
4, 2, 1, 6, 1, 1, 1, 1,
//coeff_sign_flag
1, 4, 4, 5, 8, 8,
}
};
#define MAX_SUB_BLOCKS 16
#define MAX_SUB_BLOCK_SIZE 4
#define MAX_TB_SIZE 64
typedef struct ResidualCoding {
//common for ts and non ts
TransformBlock *tb;
int log2_sb_w;
int log2_sb_h;
int last_sub_block;
int hist_value;
int update_hist;
int num_sb_coeff;
int rem_bins_pass1;
int width_in_sbs;
int height_in_sbs;
int nb_sbs;
const uint8_t *sb_scan_x_off;
const uint8_t *sb_scan_y_off;
const uint8_t *scan_x_off;
const uint8_t *scan_y_off;
uint8_t sb_coded_flag[MAX_SUB_BLOCKS * MAX_SUB_BLOCKS];
int sig_coeff_flag[MAX_TB_SIZE * MAX_TB_SIZE];
int abs_level_pass1[MAX_TB_SIZE * MAX_TB_SIZE]; ///< AbsLevelPass1[][]
int abs_level[MAX_TB_SIZE * MAX_TB_SIZE];
//for ts only
uint8_t infer_sb_cbf;
int coeff_sign_level[MAX_TB_SIZE * MAX_TB_SIZE]; ///< CoeffSignLevel[][]
//for non ts only
int qstate;
int last_scan_pos;
int last_significant_coeff_x;
int last_significant_coeff_y;
} ResidualCoding;
static int cabac_reinit(VVCLocalContext *lc)
{
return skip_bytes(&lc->ep->cc, 0) == NULL ? AVERROR_INVALIDDATA : 0;
}
static void cabac_init_state(VVCLocalContext *lc)
{
const VVCSPS *sps = lc->fc->ps.sps;
const H266RawSliceHeader *rsh = lc->sc->sh.r;
const int qp = av_clip_uintp2(lc->sc->sh.slice_qp_y, 6);
int init_type = 2 - rsh->sh_slice_type;
av_assert0(VVC_CONTEXTS == SYNTAX_ELEMENT_LAST);
ff_vvc_ep_init_stat_coeff(lc->ep, sps->bit_depth, sps->r->sps_persistent_rice_adaptation_enabled_flag);
if (rsh->sh_cabac_init_flag && !IS_I(rsh))
init_type ^= 3;
for (int i = 0; i < VVC_CONTEXTS; i++) {
VVCCabacState *state = &lc->ep->cabac_state[i];
const int init_value = init_values[init_type][i];
const int shift_idx = init_values[3][i];
const int m = (init_value >> 3) - 4;
const int n = ((init_value & 7) * 18) + 1;
const int pre = av_clip(((m * (qp - 16)) >> 1) + n, 1, 127);
state->state[0] = pre << 3;
state->state[1] = pre << 7;
state->shift[0] = (shift_idx >> 2 ) + 2;
state->shift[1] = (shift_idx & 3 ) + 3 + state->shift[0];
}
}
int ff_vvc_cabac_init(VVCLocalContext *lc,
const int ctu_idx, const int rx, const int ry)
{
int ret = 0;
const VVCPPS *pps = lc->fc->ps.pps;
const int first_ctb_in_slice = !ctu_idx;
const int first_ctb_in_tile = rx == pps->ctb_to_col_bd[rx] && ry == pps->ctb_to_row_bd[ry];
if (first_ctb_in_slice|| first_ctb_in_tile) {
if (lc->sc->nb_eps == 1 && !first_ctb_in_slice)
ret = cabac_reinit(lc);
if (!ret)
cabac_init_state(lc);
}
return ret;
}
//fixme
static void vvc_refill2(CABACContext* c) {
int i;
unsigned x;
#if !HAVE_FAST_CLZ
x = c->low ^ (c->low - 1);
i = 7 - ff_h264_norm_shift[x >> (CABAC_BITS - 1)];
#else
i = ff_ctz(c->low) - CABAC_BITS;
#endif
x = -CABAC_MASK;
#if CABAC_BITS == 16
x += (c->bytestream[0] << 9) + (c->bytestream[1] << 1);
#else
x += c->bytestream[0] << 1;
#endif
c->low += x << i;
#if !UNCHECKED_BITSTREAM_READER
if (c->bytestream < c->bytestream_end)
#endif
c->bytestream += CABAC_BITS / 8;
}
static int inline vvc_get_cabac(CABACContext *c, VVCCabacState* base, const int ctx)
{
VVCCabacState *s = base + ctx;
const int qRangeIdx = c->range >> 5;
const int pState = s->state[1] + (s->state[0] << 4);
const int valMps = pState >> 14;
const int RangeLPS = (qRangeIdx * ((valMps ? 32767 - pState : pState) >> 9 ) >> 1) + 4;
int bit, lps_mask;
c->range -= RangeLPS;
lps_mask = ((c->range<<(CABAC_BITS+1)) - c->low)>>31;
c->low -= (c->range<<(CABAC_BITS+1)) & lps_mask;
c->range += (RangeLPS - c->range) & lps_mask;
bit = valMps ^ (lps_mask & 1);
lps_mask = ff_h264_norm_shift[c->range];
c->range <<= lps_mask;
c->low <<= lps_mask;
if (!(c->low & CABAC_MASK))
vvc_refill2(c);
s->state[0] = s->state[0] - (s->state[0] >> s->shift[0]) + (1023 * bit >> s->shift[0]);
s->state[1] = s->state[1] - (s->state[1] >> s->shift[1]) + (16383 * bit >> s->shift[1]);
return bit;
}
#define GET_CABAC(ctx) vvc_get_cabac(&lc->ep->cc, lc->ep->cabac_state, ctx)
//9.3.3.4 Truncated binary (TB) binarization process
static int truncated_binary_decode(VVCLocalContext *lc, const int c_max)
{
const int n = c_max + 1;
const int k = av_log2(n);
const int u = (1 << (k+1)) - n;
int v = 0;
for (int i = 0; i < k; i++)
v = (v << 1) | get_cabac_bypass(&lc->ep->cc);
if (v >= u) {
v = (v << 1) | get_cabac_bypass(&lc->ep->cc);
v -= u;
}
return v;
}
// 9.3.3.6 Limited k-th order Exp-Golomb binarization process
static int limited_kth_order_egk_decode(CABACContext *c, const int k, const int max_pre_ext_len, const int trunc_suffix_len)
{
int pre_ext_len = 0;
int escape_length;
int val = 0;
while ((pre_ext_len < max_pre_ext_len) && get_cabac_bypass(c))
pre_ext_len++;
if (pre_ext_len == max_pre_ext_len)
escape_length = trunc_suffix_len;
else
escape_length = pre_ext_len + k;
while (escape_length-- > 0) {
val = (val << 1) + get_cabac_bypass(c);
}
val += ((1 << pre_ext_len) - 1) << k;
return val;
}
static av_always_inline
void get_left_top(const VVCLocalContext *lc, uint8_t *left, uint8_t *top,
const int x0, const int y0, const uint8_t *left_ctx, const uint8_t *top_ctx)
{
const VVCFrameContext *fc = lc->fc;
const VVCSPS *sps = fc->ps.sps;
const int min_cb_width = fc->ps.pps->min_cb_width;
const int x0b = av_zero_extend(x0, sps->ctb_log2_size_y);
const int y0b = av_zero_extend(y0, sps->ctb_log2_size_y);
const int x_cb = x0 >> sps->min_cb_log2_size_y;
const int y_cb = y0 >> sps->min_cb_log2_size_y;
if (lc->ctb_left_flag || x0b)
*left = SAMPLE_CTB(left_ctx, x_cb - 1, y_cb);
if (lc->ctb_up_flag || y0b)
*top = SAMPLE_CTB(top_ctx, x_cb, y_cb - 1);
}
static av_always_inline
uint8_t get_inc(VVCLocalContext *lc, const uint8_t *ctx)
{
uint8_t left = 0, top = 0;
get_left_top(lc, &left, &top, lc->cu->x0, lc->cu->y0, ctx, ctx);
return left + top;
}
int ff_vvc_sao_merge_flag_decode(VVCLocalContext *lc)
{
return GET_CABAC(SAO_MERGE_FLAG);
}
int ff_vvc_sao_type_idx_decode(VVCLocalContext *lc)
{
if (!GET_CABAC(SAO_TYPE_IDX))
return SAO_NOT_APPLIED;
if (!get_cabac_bypass(&lc->ep->cc))
return SAO_BAND;
return SAO_EDGE;
}
int ff_vvc_sao_band_position_decode(VVCLocalContext *lc)
{
int value = get_cabac_bypass(&lc->ep->cc);
for (int i = 0; i < 4; i++)
value = (value << 1) | get_cabac_bypass(&lc->ep->cc);
return value;
}
int ff_vvc_sao_offset_abs_decode(VVCLocalContext *lc)
{
int i = 0;
const int length = (1 << (FFMIN(lc->fc->ps.sps->bit_depth, 10) - 5)) - 1;
while (i < length && get_cabac_bypass(&lc->ep->cc))
i++;
return i;
}
int ff_vvc_sao_offset_sign_decode(VVCLocalContext *lc)
{
return get_cabac_bypass(&lc->ep->cc);
}
int ff_vvc_sao_eo_class_decode(VVCLocalContext *lc)
{
int ret = get_cabac_bypass(&lc->ep->cc) << 1;
ret |= get_cabac_bypass(&lc->ep->cc);
return ret;
}
int ff_vvc_alf_ctb_flag(VVCLocalContext *lc, const int rx, const int ry, const int c_idx)
{
int inc = c_idx * 3;
const VVCFrameContext *fc = lc->fc;
if (lc->ctb_left_flag) {
const ALFParams *left = &CTB(fc->tab.alf, rx - 1, ry);
inc += left->ctb_flag[c_idx];
}
if (lc->ctb_up_flag) {
const ALFParams *above = &CTB(fc->tab.alf, rx, ry - 1);
inc += above->ctb_flag[c_idx];
}
return GET_CABAC(ALF_CTB_FLAG + inc);
}
int ff_vvc_alf_use_aps_flag(VVCLocalContext *lc)
{
return GET_CABAC(ALF_USE_APS_FLAG);
}
int ff_vvc_alf_luma_prev_filter_idx(VVCLocalContext *lc)
{
return truncated_binary_decode(lc, lc->sc->sh.r->sh_num_alf_aps_ids_luma - 1);
}
int ff_vvc_alf_luma_fixed_filter_idx(VVCLocalContext *lc)
{
return truncated_binary_decode(lc, 15);
}
int ff_vvc_alf_ctb_filter_alt_idx(VVCLocalContext *lc, const int c_idx, const int num_chroma_filters)
{
int i = 0;
const int length = num_chroma_filters - 1;
while (i < length && GET_CABAC(ALF_CTB_FILTER_ALT_IDX + c_idx - 1))
i++;
return i;
}
int ff_vvc_alf_ctb_cc_idc(VVCLocalContext *lc, const int rx, const int ry, const int idx, const int cc_filters_signalled)
{
int inc = !idx ? ALF_CTB_CC_CB_IDC : ALF_CTB_CC_CR_IDC;
int i = 0;
const VVCFrameContext *fc = lc->fc;
if (lc->ctb_left_flag) {
const ALFParams *left = &CTB(fc->tab.alf, rx - 1, ry);
inc += left->ctb_cc_idc[idx] != 0;
}
if (lc->ctb_up_flag) {
const ALFParams *above = &CTB(fc->tab.alf, rx, ry - 1);
inc += above->ctb_cc_idc[idx] != 0;
}
if (!GET_CABAC(inc))
return 0;
i++;
while (i < cc_filters_signalled && get_cabac_bypass(&lc->ep->cc))
i++;
return i;
}
int ff_vvc_split_cu_flag(VVCLocalContext *lc, const int x0, const int y0,
const int cb_width, const int cb_height, const int is_chroma, const VVCAllowedSplit *a)
{
const VVCFrameContext *fc = lc->fc;
const VVCPPS *pps = fc->ps.pps;
const int is_inside = (x0 + cb_width <= pps->width) && (y0 + cb_height <= pps->height);
if ((a->btv || a->bth || a->ttv || a->tth || a->qt) && is_inside)
{
uint8_t inc = 0, left_height = cb_height, top_width = cb_width;
get_left_top(lc, &left_height, &top_width, x0, y0, fc->tab.cb_height[is_chroma], fc->tab.cb_width[is_chroma]);
inc += left_height < cb_height;
inc += top_width < cb_width;
inc += (a->btv + a->bth + a->ttv + a->tth + 2 * a->qt - 1) / 2 * 3;
return GET_CABAC(SPLIT_CU_FLAG + inc);
}
return !is_inside;
}
static int split_qt_flag_decode(VVCLocalContext *lc, const int x0, const int y0, const int ch_type, const int cqt_depth)
{
const VVCFrameContext *fc = lc->fc;
int inc = 0;
uint8_t depth_left = 0, depth_top = 0;
get_left_top(lc, &depth_left, &depth_top, x0, y0, fc->tab.cqt_depth[ch_type], fc->tab.cqt_depth[ch_type]);
inc += depth_left > cqt_depth;
inc += depth_top > cqt_depth;
inc += (cqt_depth >= 2) * 3;
return GET_CABAC(SPLIT_QT_FLAG + inc);
}
static int mtt_split_cu_vertical_flag_decode(VVCLocalContext *lc, const int x0, const int y0,
const int cb_width, const int cb_height, const int ch_type, const VVCAllowedSplit* a)
{
if ((a->bth || a->tth) && (a->btv || a->ttv)) {
int inc;
const int v = a->btv + a->ttv;
const int h = a->bth + a->tth;
if (v > h)
inc = 4;
else if (v < h)
inc = 3;
else {
const VVCFrameContext *fc = lc->fc;
const VVCSPS *sps = fc->ps.sps;
const int min_cb_width = fc->ps.pps->min_cb_width;
const int x0b = av_zero_extend(x0, sps->ctb_log2_size_y);
const int y0b = av_zero_extend(y0, sps->ctb_log2_size_y);
const int x_cb = x0 >> sps->min_cb_log2_size_y;
const int y_cb = y0 >> sps->min_cb_log2_size_y;
const int available_a = lc->ctb_up_flag || y0b;
const int available_l = lc->ctb_left_flag || x0b;
const int da = cb_width / (available_a ? SAMPLE_CTB(fc->tab.cb_width[ch_type], x_cb, y_cb - 1) : 1);
const int dl = cb_height / (available_l ? SAMPLE_CTB(fc->tab.cb_height[ch_type], x_cb - 1, y_cb) : 1);
if (da == dl || !available_a || !available_l)
inc = 0;
else if (da < dl)
inc = 1;
else
inc = 2;
}
return GET_CABAC(MTT_SPLIT_CU_VERTICAL_FLAG + inc);
}
return !(a->bth || a->tth);
}
static int mtt_split_cu_binary_flag_decode(VVCLocalContext *lc, const int mtt_split_cu_vertical_flag, const int mtt_depth)
{
const int inc = (2 * mtt_split_cu_vertical_flag) + ((mtt_depth <= 1) ? 1 : 0);
return GET_CABAC(MTT_SPLIT_CU_BINARY_FLAG + inc);
}
VVCSplitMode ff_vvc_split_mode(VVCLocalContext *lc, const int x0, const int y0, const int cb_width, const int cb_height,
const int cqt_depth, const int mtt_depth, const int ch_type, const VVCAllowedSplit *a)
{
const int allow_no_qt = a->btv || a->bth || a->ttv || a->tth;
int split_qt_flag;
int mtt_split_cu_vertical_flag;
int mtt_split_cu_binary_flag;
const VVCSplitMode mtt_split_modes[] = {
SPLIT_TT_HOR, SPLIT_BT_HOR, SPLIT_TT_VER, SPLIT_BT_VER,
};
if (allow_no_qt && a->qt) {
split_qt_flag = split_qt_flag_decode(lc, x0, y0, ch_type, cqt_depth);
} else {
split_qt_flag = !allow_no_qt || a->qt;
}
if (split_qt_flag)
return SPLIT_QT;
mtt_split_cu_vertical_flag = mtt_split_cu_vertical_flag_decode(lc, x0, y0, cb_width, cb_height, ch_type, a);
if ((a->btv && a->ttv && mtt_split_cu_vertical_flag) ||
(a->bth && a->tth && !mtt_split_cu_vertical_flag)) {
mtt_split_cu_binary_flag = mtt_split_cu_binary_flag_decode(lc, mtt_split_cu_vertical_flag, mtt_depth);
} else {
if (!a->btv && !a->bth)
mtt_split_cu_binary_flag = 0;
else if (!a->ttv && !a->tth)
mtt_split_cu_binary_flag = 1;
else if (a->bth && a->ttv)
mtt_split_cu_binary_flag = 1 - mtt_split_cu_vertical_flag;
else
mtt_split_cu_binary_flag = mtt_split_cu_vertical_flag;
}
return mtt_split_modes[(mtt_split_cu_vertical_flag << 1) + mtt_split_cu_binary_flag];
}
int ff_vvc_non_inter_flag(VVCLocalContext *lc, const int x0, const int y0, const int ch_type)
{
const VVCFrameContext *fc = lc->fc;
uint8_t inc, left = MODE_INTER, top = MODE_INTER;
get_left_top(lc, &left, &top, x0, y0, fc->tab.cpm[ch_type], fc->tab.cpm[ch_type]);
inc = left == MODE_INTRA || top == MODE_INTRA;
return GET_CABAC(NON_INTER_FLAG + inc);
}
int ff_vvc_pred_mode_flag(VVCLocalContext *lc, const int is_chroma)
{
const VVCFrameContext *fc = lc->fc;
const CodingUnit *cu = lc->cu;
uint8_t inc, left = MODE_INTER, top = MODE_INTER;
get_left_top(lc, &left, &top, cu->x0, cu->y0, fc->tab.cpm[is_chroma], fc->tab.cpm[is_chroma]);
inc = left == MODE_INTRA || top == MODE_INTRA;
return GET_CABAC(PRED_MODE_FLAG + inc);
}
int ff_vvc_pred_mode_plt_flag(VVCLocalContext *lc)
{
return GET_CABAC(PRED_MODE_PLT_FLAG);
}
int ff_vvc_intra_bdpcm_luma_flag(VVCLocalContext *lc)
{
return GET_CABAC(INTRA_BDPCM_LUMA_FLAG);
}
int ff_vvc_intra_bdpcm_luma_dir_flag(VVCLocalContext *lc)
{
return GET_CABAC(INTRA_BDPCM_LUMA_DIR_FLAG);
}
int ff_vvc_intra_bdpcm_chroma_flag(VVCLocalContext *lc)
{
return GET_CABAC(INTRA_BDPCM_CHROMA_FLAG);
}
int ff_vvc_intra_bdpcm_chroma_dir_flag(VVCLocalContext *lc)
{
return GET_CABAC(INTRA_BDPCM_CHROMA_DIR_FLAG);
}
int ff_vvc_cu_skip_flag(VVCLocalContext *lc, const uint8_t *cu_skip_flag)
{
const int inc = get_inc(lc, cu_skip_flag);
return GET_CABAC(CU_SKIP_FLAG + inc);
}
int ff_vvc_pred_mode_ibc_flag(VVCLocalContext *lc, const int is_chroma)
{
const VVCFrameContext *fc = lc->fc;
const CodingUnit *cu = lc->cu;
uint8_t left_mode = MODE_INTER, top_mode = MODE_INTER;
int inc;
get_left_top(lc, &left_mode, &top_mode, cu->x0, cu->y0, fc->tab.cpm[is_chroma], fc->tab.cpm[is_chroma]);
inc = (left_mode == MODE_IBC) + (top_mode == MODE_IBC);
return GET_CABAC(PRED_MODE_IBC_FLAG + inc);
}
int ff_vvc_intra_mip_flag(VVCLocalContext *lc, const uint8_t *intra_mip_flag)
{
const int w = lc->cu->cb_width;
const int h = lc->cu->cb_height;
const int inc = (w > h * 2 || h > w * 2) ? 3 : get_inc(lc, intra_mip_flag);
return GET_CABAC(INTRA_MIP_FLAG + inc);
}
int ff_vvc_intra_mip_transposed_flag(VVCLocalContext *lc)
{
return get_cabac_bypass(&lc->ep->cc);
}
int ff_vvc_intra_mip_mode(VVCLocalContext *lc)
{
const int w = lc->cu->cb_width;
const int h = lc->cu->cb_height;
const int c_max = (w == 4 && h == 4) ? 15 :
((w == 4 || h == 4) || (w == 8 && h == 8)) ? 7: 5;
return truncated_binary_decode(lc, c_max);
}
int ff_vvc_intra_luma_ref_idx(VVCLocalContext *lc)
{
int i;
for (i = 0; i < 2; i++) {
if (!GET_CABAC(INTRA_LUMA_REF_IDX + i))
return i;
}
return i;
}
int ff_vvc_intra_subpartitions_mode_flag(VVCLocalContext *lc)
{
return GET_CABAC(INTRA_SUBPARTITIONS_MODE_FLAG);
}
enum IspType ff_vvc_isp_split_type(VVCLocalContext *lc, const int intra_subpartitions_mode_flag)
{
if (!intra_subpartitions_mode_flag)
return ISP_NO_SPLIT;
return 1 + GET_CABAC(INTRA_SUBPARTITIONS_SPLIT_FLAG);
}
int ff_vvc_intra_luma_mpm_flag(VVCLocalContext *lc)
{
return GET_CABAC(INTRA_LUMA_MPM_FLAG);
}
int ff_vvc_intra_luma_not_planar_flag(VVCLocalContext *lc, const int intra_subpartitions_mode_flag)
{
return GET_CABAC(INTRA_LUMA_NOT_PLANAR_FLAG + !intra_subpartitions_mode_flag);
}
int ff_vvc_intra_luma_mpm_idx(VVCLocalContext *lc)
{
int i;
for (i = 0; i < 4 && get_cabac_bypass(&lc->ep->cc); i++)
/* nothing */;
return i;
}
int ff_vvc_intra_luma_mpm_remainder(VVCLocalContext *lc)
{
return truncated_binary_decode(lc, 60);
}
int ff_vvc_cclm_mode_flag(VVCLocalContext *lc)
{
return GET_CABAC(CCLM_MODE_FLAG);
}
int ff_vvc_cclm_mode_idx(VVCLocalContext *lc)
{
if (!GET_CABAC(CCLM_MODE_IDX))
return 0;
return get_cabac_bypass(&lc->ep->cc) + 1;
}
int ff_vvc_intra_chroma_pred_mode(VVCLocalContext *lc)
{
if (!GET_CABAC(INTRA_CHROMA_PRED_MODE))
return 4;
return (get_cabac_bypass(&lc->ep->cc) << 1) | get_cabac_bypass(&lc->ep->cc);
}
int ff_vvc_general_merge_flag(VVCLocalContext *lc)
{
return GET_CABAC(GENERAL_MERGE_FLAG);
}
static int get_inter_flag_inc(VVCLocalContext *lc, const int x0, const int y0)
{
uint8_t left_merge = 0, top_merge = 0;
uint8_t left_affine = 0, top_affine = 0;
const VVCFrameContext *fc = lc->fc;
get_left_top(lc, &left_merge, &top_merge, x0, y0, fc->tab.msf, fc->tab.msf);
get_left_top(lc, &left_affine, &top_affine, x0, y0, fc->tab.iaf, fc->tab.iaf);
return (left_merge || left_affine) + (top_merge + top_affine);
}
int ff_vvc_merge_subblock_flag(VVCLocalContext *lc)
{
const int inc = get_inter_flag_inc(lc, lc->cu->x0, lc->cu->y0);
return GET_CABAC(MERGE_SUBBLOCK_FLAG + inc);
}
int ff_vvc_merge_subblock_idx(VVCLocalContext *lc, const int max_num_subblock_merge_cand)
{
int i;
if (!GET_CABAC(MERGE_SUBBLOCK_IDX))
return 0;
for (i = 1; i < max_num_subblock_merge_cand - 1 && get_cabac_bypass(&lc->ep->cc); i++)
/* nothing */;
return i;
}
int ff_vvc_regular_merge_flag(VVCLocalContext *lc, const int cu_skip_flag)
{
int inc = !cu_skip_flag;
return GET_CABAC(REGULAR_MERGE_FLAG + inc);
}
int ff_vvc_mmvd_merge_flag(VVCLocalContext *lc)
{
return GET_CABAC(MMVD_MERGE_FLAG);
}
int ff_vvc_mmvd_cand_flag(VVCLocalContext *lc)
{
return GET_CABAC(MMVD_CAND_FLAG);
}
static int mmvd_distance_idx_decode(VVCLocalContext *lc)
{
int i;
if (!GET_CABAC(MMVD_DISTANCE_IDX))
return 0;
for (i = 1; i < 7 && get_cabac_bypass(&lc->ep->cc); i++)
/* nothing */;
return i;
}
static int mmvd_direction_idx_decode(VVCLocalContext *lc)
{
return (get_cabac_bypass(&lc->ep->cc) << 1) | get_cabac_bypass(&lc->ep->cc);
}
void ff_vvc_mmvd_offset_coding(VVCLocalContext *lc, Mv *mmvd_offset, const int ph_mmvd_fullpel_only_flag)
{
const int shift = ph_mmvd_fullpel_only_flag ? 4 : 2;
const int mmvd_distance = 1 << (mmvd_distance_idx_decode(lc) + shift);
const int mmvd_direction_idx = mmvd_direction_idx_decode(lc);
const int mmvd_signs[][2] = { {1, 0}, {-1, 0}, {0, 1}, {0, -1} };
mmvd_offset->x = mmvd_distance * mmvd_signs[mmvd_direction_idx][0];
mmvd_offset->y = mmvd_distance * mmvd_signs[mmvd_direction_idx][1];
}
static PredMode get_luma_pred_mode(VVCLocalContext *lc)
{
const VVCFrameContext *fc = lc->fc;
const CodingUnit *cu = lc->cu;
PredMode pred_mode;
if (cu->tree_type != DUAL_TREE_CHROMA) {
pred_mode = cu->pred_mode;
} else {
const int x_cb = cu->x0 >> fc->ps.sps->min_cb_log2_size_y;
const int y_cb = cu->y0 >> fc->ps.sps->min_cb_log2_size_y;
const int min_cb_width = fc->ps.pps->min_cb_width;
pred_mode = SAMPLE_CTB(fc->tab.cpm[0], x_cb, y_cb);
}
return pred_mode;
}
int ff_vvc_merge_idx(VVCLocalContext *lc)
{
const VVCSPS *sps = lc->fc->ps.sps;
const int is_ibc = get_luma_pred_mode(lc) == MODE_IBC;
const int c_max = (is_ibc ? sps->max_num_ibc_merge_cand : sps->max_num_merge_cand) - 1;
int i;
if (!GET_CABAC(MERGE_IDX))
return 0;
for (i = 1; i < c_max && get_cabac_bypass(&lc->ep->cc); i++)
/* nothing */;
return i;
}
int ff_vvc_merge_gpm_partition_idx(VVCLocalContext *lc)
{
int i = 0;
for (int j = 0; j < 6; j++)
i = (i << 1) | get_cabac_bypass(&lc->ep->cc);
return i;
}
int ff_vvc_merge_gpm_idx(VVCLocalContext *lc, const int idx)
{
const int c_max = lc->fc->ps.sps->max_num_gpm_merge_cand - idx - 1;
int i;
if (!GET_CABAC(MERGE_IDX))
return 0;
for (i = 1; i < c_max && get_cabac_bypass(&lc->ep->cc); i++)
/* nothing */;
return i;
}
int ff_vvc_ciip_flag(VVCLocalContext *lc)
{
return GET_CABAC(CIIP_FLAG);
}
PredFlag ff_vvc_pred_flag(VVCLocalContext *lc, const int is_b)
{
const int w = lc->cu->cb_width;
const int h = lc->cu->cb_height;
if (!is_b)
return PF_L0;
if (w + h > 12) {
const int log2 = av_log2(w) + av_log2(h);
const int inc = 7 - ((1 + log2)>>1);
if (GET_CABAC(INTER_PRED_IDC + inc))
return PF_BI;
}
return PF_L0 + GET_CABAC(INTER_PRED_IDC + 5);
}
int ff_vvc_inter_affine_flag(VVCLocalContext *lc)
{
const int inc = get_inter_flag_inc(lc, lc->cu->x0, lc->cu->y0);
return GET_CABAC(INTER_AFFINE_FLAG + inc);
}
int ff_vvc_cu_affine_type_flag(VVCLocalContext *lc)
{
return GET_CABAC(CU_AFFINE_TYPE_FLAG);
}
int ff_vvc_sym_mvd_flag(VVCLocalContext *lc)
{
return GET_CABAC(SYM_MVD_FLAG);
}
int ff_vvc_ref_idx_lx(VVCLocalContext *lc, const uint8_t nb_refs)
{
const int c_max = nb_refs - 1;
const int max_ctx = FFMIN(c_max, 2);
int i = 0;
while (i < max_ctx && GET_CABAC(REF_IDX_LX + i))
i++;
if (i == 2) {
while (i < c_max && get_cabac_bypass(&lc->ep->cc))
i++;
}
return i;
}
int ff_vvc_abs_mvd_greater0_flag(VVCLocalContext *lc)
{
return GET_CABAC(ABS_MVD_GREATER0_FLAG);
}
int ff_vvc_abs_mvd_greater1_flag(VVCLocalContext *lc)
{
return GET_CABAC(ABS_MVD_GREATER1_FLAG);
}
int ff_vvc_abs_mvd_minus2(VVCLocalContext *lc)
{
return limited_kth_order_egk_decode(&lc->ep->cc, 1, 15, 17);
}
int ff_vvc_mvd_sign_flag(VVCLocalContext *lc)
{
return get_cabac_bypass(&lc->ep->cc);
}
int ff_vvc_mvp_lx_flag(VVCLocalContext *lc)
{
return GET_CABAC(MVP_LX_FLAG);
}
static int amvr_flag(VVCLocalContext *lc, const int inter_affine_flag)
{
return GET_CABAC(AMVR_FLAG + inter_affine_flag);
}
static int amvr_precision_idx(VVCLocalContext *lc, const int inc, const int c_max)
{
int i = 0;
if (!GET_CABAC(AMVR_PRECISION_IDX + inc))
return 0;
i++;
if (i < c_max && GET_CABAC(AMVR_PRECISION_IDX + 1))
i++;
return i;
}
int ff_vvc_amvr_shift(VVCLocalContext *lc, const int inter_affine_flag,
const PredMode pred_mode, const int has_amvr_flag)
{
int amvr_shift = 2;
if (has_amvr_flag) {
if (pred_mode == MODE_IBC || amvr_flag(lc, inter_affine_flag)) {
int idx;
if (inter_affine_flag) {
idx = amvr_precision_idx(lc, 2, 1);
amvr_shift = idx * 4;
} else if (pred_mode == MODE_IBC) {
idx = amvr_precision_idx(lc, 1, 1);
amvr_shift = 4 + idx * 2;
} else {
static const int shifts[] = {3, 4, 6};
idx = amvr_precision_idx(lc, 0, 2);
amvr_shift = shifts[idx];
}
}
}
return amvr_shift;
}
int ff_vvc_bcw_idx(VVCLocalContext *lc, const int no_backward_pred_flag)
{
const int c_max = no_backward_pred_flag ? 4 : 2;
int i = 1;
if (!GET_CABAC(BCW_IDX))
return 0;
while (i < c_max && get_cabac_bypass(&lc->ep->cc))
i++;
return i;
}
int ff_vvc_tu_cb_coded_flag(VVCLocalContext *lc)
{
return GET_CABAC(TU_CB_CODED_FLAG + lc->cu->bdpcm_flag[1]);
}
int ff_vvc_tu_cr_coded_flag(VVCLocalContext *lc, int tu_cb_coded_flag)
{
return GET_CABAC(TU_CR_CODED_FLAG + (lc->cu->bdpcm_flag[1] ? 2 : tu_cb_coded_flag));
}
int ff_vvc_tu_y_coded_flag(VVCLocalContext *lc)
{
const CodingUnit *cu = lc->cu;
int inc;
if (cu->bdpcm_flag[0])
inc = 1;
else if (cu->isp_split_type == ISP_NO_SPLIT)
inc = 0;
else
inc = 2 + lc->parse.prev_tu_cbf_y;
lc->parse.prev_tu_cbf_y = GET_CABAC(TU_Y_CODED_FLAG + inc);
return lc->parse.prev_tu_cbf_y;
}
int ff_vvc_cu_qp_delta_abs(VVCLocalContext *lc)
{
int v, i, k;
if (!GET_CABAC(CU_QP_DELTA_ABS))
return 0;
// prefixVal
for (v = 1; v < 5 && GET_CABAC(CU_QP_DELTA_ABS + 1); v++)
/* nothing */;
if (v < 5)
return v;
// 9.3.3.5 k-th order Exp-Golomb binarization process
// suffixVal
// CuQpDeltaVal shall in the range of −( 32 + QpBdOffset / 2 ) to +( 31 + QpBdOffset / 2 )
// so k = 6 should enough
for (k = 0; k < 6 && get_cabac_bypass(&lc->ep->cc); k++)
/* nothing */;
i = (1 << k) - 1;
v = 0;
while (k--)
v = (v << 1) + get_cabac_bypass(&lc->ep->cc);
v += i;
return v + 5;
}
int ff_vvc_cu_qp_delta_sign_flag(VVCLocalContext *lc)
{
return get_cabac_bypass(&lc->ep->cc);
}
int ff_vvc_cu_chroma_qp_offset_flag(VVCLocalContext *lc)
{
return GET_CABAC(CU_CHROMA_QP_OFFSET_FLAG);
}
int ff_vvc_cu_chroma_qp_offset_idx(VVCLocalContext *lc)
{
const int c_max = lc->fc->ps.pps->r->pps_chroma_qp_offset_list_len_minus1;
int i;
for (i = 0; i < c_max && GET_CABAC(CU_CHROMA_QP_OFFSET_IDX); i++)
/* nothing */;
return i;
}
static av_always_inline int last_significant_coeff_xy_prefix(VVCLocalContext *lc,
const int log2_tb_size, const int log2_zo_tb_size, const int c_idx, const int ctx)
{
int i = 0;
int max = (log2_zo_tb_size << 1) - 1;
int ctx_offset, ctx_shift;
if (!log2_tb_size)
return 0;
if (!c_idx) {
const int offset_y[] = {0, 0, 3, 6, 10, 15};
ctx_offset = offset_y[log2_tb_size - 1];
ctx_shift = (log2_tb_size + 1) >> 2;
} else {
const int shifts[] = {0, 0, 0, 1, 2, 2, 2};
ctx_offset = 20;
ctx_shift = shifts[log2_tb_size];
}
while (i < max && GET_CABAC(ctx + (i >> ctx_shift) + ctx_offset))
i++;
return i;
}
static av_always_inline int last_significant_coeff_x_prefix_decode(VVCLocalContext *lc,
const int log2_tb_width, const int log2_zo_tb_width, const int c_idx)
{
return last_significant_coeff_xy_prefix(lc, log2_tb_width, log2_zo_tb_width, c_idx, LAST_SIG_COEFF_X_PREFIX);
}
static av_always_inline int last_significant_coeff_y_prefix_decode(VVCLocalContext *lc,
const int log2_tb_height, const int log2_zo_tb_height, const int c_idx)
{
return last_significant_coeff_xy_prefix(lc, log2_tb_height, log2_zo_tb_height, c_idx, LAST_SIG_COEFF_Y_PREFIX);
}
static av_always_inline int last_sig_coeff_suffix_decode(VVCLocalContext *lc,
const int last_significant_coeff_y_prefix)
{
const int length = (last_significant_coeff_y_prefix >> 1) - 1;
int value = get_cabac_bypass(&lc->ep->cc);
for (int i = 1; i < length; i++)
value = (value << 1) | get_cabac_bypass(&lc->ep->cc);
return value;
}
int ff_vvc_tu_joint_cbcr_residual_flag(VVCLocalContext *lc, const int tu_cb_coded_flag, const int tu_cr_coded_flag)
{
return GET_CABAC(TU_JOINT_CBCR_RESIDUAL_FLAG + 2 * tu_cb_coded_flag + tu_cr_coded_flag - 1);
}
int ff_vvc_transform_skip_flag(VVCLocalContext *lc, const int inc)
{
return GET_CABAC(TRANSFORM_SKIP_FLAG + inc);
}
//9.3.4.2.7 Derivation process for the variables locNumSig, locSumAbsPass1
static int get_local_sum(const int *level, const int w, const int h,
const int xc, const int yc, const int hist_value)
{
int loc_sum = 3 * hist_value;
level += w * yc + xc;
if (xc < w - 1) {
loc_sum += level[1];
if (xc < w - 2)
loc_sum += level[2] - hist_value;
if (yc < h - 1)
loc_sum += level[w + 1] - hist_value;
}
if (yc < h - 1) {
loc_sum += level[w];
if (yc < h - 2)
loc_sum += level[w << 1] - hist_value;
}
return loc_sum;
}
//9.3.4.2.7 Derivation process for the variables locNumSig, locSumAbsPass1
static int get_local_sum_ts(const int *level, const int w, const int h, const int xc, const int yc)
{
int loc_sum = 0;
level += w * yc + xc;
if (xc > 0)
loc_sum += level[-1];
if (yc > 0)
loc_sum += level[-w];
return loc_sum;
}
static int get_gtx_flag_inc(const ResidualCoding* rc, const int xc, const int yc, const int last)
{
const TransformBlock *tb = rc->tb;
int inc;
if (last) {
const int incs[] = {0, 21, 21};
inc = incs[tb->c_idx];
} else {
const int d = xc + yc;
const int local_sum_sig = get_local_sum(rc->sig_coeff_flag,
tb->tb_width,tb->tb_height, xc, yc, rc->hist_value);
const int loc_sum_abs_pass1 = get_local_sum(rc->abs_level_pass1,
tb->tb_width, tb->tb_height, xc, yc, rc->hist_value);
const int offset = FFMIN(loc_sum_abs_pass1 - local_sum_sig, 4);
if (!tb->c_idx)
inc = 1 + offset + (!d ? 15 : (d < 3 ? 10 : (d < 10 ? 5 : 0)));
else
inc = 22 + offset + (!d ? 5 : 0);
}
return inc;
}
static int abs_level_gtx_flag_decode(VVCLocalContext *lc, const int inc)
{
return GET_CABAC(ABS_LEVEL_GTX_FLAG + inc);
}
static int par_level_flag_decode(VVCLocalContext *lc, const int inc)
{
return GET_CABAC(PAR_LEVEL_FLAG + inc);
}
static int par_level_flag_ts_decode(VVCLocalContext *lc)
{
const int inc = 32;
return GET_CABAC(PAR_LEVEL_FLAG + inc);
}
static int sb_coded_flag_decode(VVCLocalContext *lc, const uint8_t *sb_coded_flag,
const ResidualCoding *rc, const int xs, const int ys)
{
const H266RawSliceHeader *rsh = lc->sc->sh.r;
const TransformBlock *tb = rc->tb;
const int w = rc->width_in_sbs;
const int h = rc->height_in_sbs;
int inc;
if (tb->ts && !rsh->sh_ts_residual_coding_disabled_flag) {
const int left = xs > 0 ? sb_coded_flag[-1] : 0;
const int above = ys > 0 ? sb_coded_flag[-w] : 0;
inc = left + above + 4;
} else {
const int right = (xs < w - 1) ? sb_coded_flag[1] : 0;
const int bottom = (ys < h - 1) ? sb_coded_flag[w] : 0;
inc = (right | bottom) + (tb->c_idx ? 2 : 0);
}
return GET_CABAC(SB_CODED_FLAG + inc);
}
static int sig_coeff_flag_decode(VVCLocalContext *lc, const ResidualCoding* rc, const int xc, const int yc)
{
const H266RawSliceHeader *rsh = lc->sc->sh.r;
const TransformBlock *tb = rc->tb;
int inc;
if (tb->ts && !rsh->sh_ts_residual_coding_disabled_flag) {
const int local_num_sig = get_local_sum_ts(rc->sig_coeff_flag, tb->tb_width, tb->tb_height, xc, yc);
inc = 60 + local_num_sig;
} else {
const int d = xc + yc;
const int loc_sum_abs_pass1 = get_local_sum(rc->abs_level_pass1,
tb->tb_width, tb->tb_height, xc, yc, 0);
if (!tb->c_idx) {
inc = 12 * FFMAX(0, rc->qstate - 1) + FFMIN((loc_sum_abs_pass1 + 1) >> 1, 3) + ((d < 2) ? 8 : (d < 5 ? 4 : 0));
} else {
inc = 36 + 8 * FFMAX(0, rc->qstate - 1) + FFMIN((loc_sum_abs_pass1 + 1) >> 1, 3) + (d < 2 ? 4 : 0);
}
}
return GET_CABAC(SIG_COEFF_FLAG + inc);
}
static int abs_get_rice_param(VVCLocalContext *lc, const ResidualCoding* rc,
const int xc, const int yc, const int base_level)
{
const VVCSPS *sps = lc->fc->ps.sps;
const TransformBlock* tb = rc->tb;
const int rice_params[] = {
0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 2, 2,
2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3,
};
int loc_sum_abs;
int shift_val;
loc_sum_abs = get_local_sum(rc->abs_level, tb->tb_width, tb->tb_height, xc,
yc, rc->hist_value);
if (!sps->r->sps_rrc_rice_extension_flag) {
shift_val = 0;
} else {
shift_val = (av_log2(FFMAX(FFMIN(loc_sum_abs, 2048), 8)) - 3) & ~1;
}
loc_sum_abs = av_clip_uintp2((loc_sum_abs >> shift_val) - base_level * 5, 5);
return rice_params[loc_sum_abs] + shift_val;
}
static int abs_decode(VVCLocalContext *lc, const int c_rice_param)
{
const VVCSPS *sps = lc->fc->ps.sps;
const int MAX_BIN = 6;
int prefix = 0;
int suffix = 0;
while (prefix < MAX_BIN && get_cabac_bypass(&lc->ep->cc))
prefix++;
if (prefix < MAX_BIN) {
for (int i = 0; i < c_rice_param; i++) {
suffix = (suffix << 1) | get_cabac_bypass(&lc->ep->cc);
}
} else {
suffix = limited_kth_order_egk_decode(&lc->ep->cc,
c_rice_param + 1,
26 - sps->log2_transform_range,
sps->log2_transform_range);
}
return suffix + (prefix << c_rice_param);
}
static int abs_remainder_decode(VVCLocalContext *lc, const ResidualCoding* rc, const int xc, const int yc)
{
const VVCSPS *sps = lc->fc->ps.sps;
const H266RawSliceHeader *rsh = lc->sc->sh.r;
const int base_level[][2][2] = {
{ {4, 4}, {4, 4} },
{ {3, 2}, {2, 1} }
};
const int c_rice_param = abs_get_rice_param(lc, rc, xc, yc,
base_level[sps->r->sps_rrc_rice_extension_flag][sps->bit_depth > 12][IS_I(rsh)]);
const int rem = abs_decode(lc, c_rice_param);
return rem;
}
static int abs_remainder_ts_decode(VVCLocalContext *lc, const ResidualCoding* rc, const int xc, const int yc)
{
const H266RawSliceHeader *rsh = lc->sc->sh.r;
const int c_rice_param = rsh->sh_ts_residual_coding_rice_idx_minus1 + 1;
const int rem = abs_decode(lc, c_rice_param);
return rem;
}
static int coeff_sign_flag_decode(VVCLocalContext *lc)
{
return get_cabac_bypass(&lc->ep->cc);
}
//9.3.4.2.10 Derivation process of ctxInc for the syntax element coeff_sign_flag for transform skip mode
static int coeff_sign_flag_ts_decode(VVCLocalContext *lc, const CodingUnit *cu, const ResidualCoding *rc, const int xc, const int yc)
{
const TransformBlock *tb = rc->tb;
const int w = tb->tb_width;
const int *level = rc->coeff_sign_level + yc * w + xc;
const int left_sign = xc ? level[-1] : 0;
const int above_sign = yc ? level[-w] : 0;
const int bdpcm_flag = cu->bdpcm_flag[tb->c_idx];
int inc;
if (left_sign == -above_sign)
inc = bdpcm_flag ? 3 : 0;
else if (left_sign >= 0 && above_sign >= 0)
inc = bdpcm_flag ? 4 : 1;
else
inc = bdpcm_flag ? 5 : 2;
return GET_CABAC(COEFF_SIGN_FLAG + inc);
}
static int abs_level_gt1_flag_ts_decode(VVCLocalContext *lc, const CodingUnit *cu, const ResidualCoding *rc, const int xc, const int yc)
{
const TransformBlock *tb = rc->tb;
const int *sig_coeff_flag = rc->sig_coeff_flag + yc * tb->tb_width + xc;
int inc;
if (cu->bdpcm_flag[tb->c_idx]) {
inc = 67;
} else {
const int l = xc > 0 ? sig_coeff_flag[-1] : 0;
const int a = yc > 0 ? sig_coeff_flag[-tb->tb_width] : 0;
inc = 64 + a + l;
}
return GET_CABAC(ABS_LEVEL_GTX_FLAG + inc);
}
static int abs_level_gtx_flag_ts_decode(VVCLocalContext *lc, const int j)
{
const int inc = 67 + j;
return GET_CABAC(ABS_LEVEL_GTX_FLAG + inc);
}
static const uint8_t qstate_translate_table[][2] = {
{ 0, 2 }, { 2, 0 }, { 1, 3 }, { 3, 1 }
};
static int dec_abs_level_decode(VVCLocalContext *lc, const ResidualCoding *rc,
const int xc, const int yc, int *abs_level)
{
const int c_rice_param = abs_get_rice_param(lc, rc, xc, yc, 0);
const int dec_abs_level = abs_decode(lc, c_rice_param);
const int zero_pos = (rc->qstate < 2 ? 1 : 2) << c_rice_param;
*abs_level = 0;
if (dec_abs_level != zero_pos) {
*abs_level = dec_abs_level;
if (dec_abs_level < zero_pos)
*abs_level += 1;
}
return dec_abs_level;
}
static void ep_update_hist(EntryPoint *ep, ResidualCoding *rc,
const int remainder, const int addin)
{
int *stat = ep->stat_coeff + rc->tb->c_idx;
if (rc->update_hist && remainder > 0) {
*stat = (*stat + av_log2(remainder) + addin) >> 1;
rc->update_hist = 0;
}
}
static void init_residual_coding(const VVCLocalContext *lc, ResidualCoding *rc,
const int log2_zo_tb_width, const int log2_zo_tb_height,
TransformBlock *tb)
{
const VVCSPS *sps = lc->fc->ps.sps;
int log2_sb_w = (FFMIN(log2_zo_tb_width, log2_zo_tb_height ) < 2 ? 1 : 2 );
int log2_sb_h = log2_sb_w;
if ( log2_zo_tb_width + log2_zo_tb_height > 3 ) {
if ( log2_zo_tb_width < 2 ) {
log2_sb_w = log2_zo_tb_width;
log2_sb_h = 4 - log2_sb_w;
} else if ( log2_zo_tb_height < 2 ) {
log2_sb_h = log2_zo_tb_height;
log2_sb_w = 4 - log2_sb_h;
}
}
rc->log2_sb_w = log2_sb_w;
rc->log2_sb_h = log2_sb_h;
rc->num_sb_coeff = 1 << (log2_sb_w + log2_sb_h);
rc->last_sub_block = ( 1 << ( log2_zo_tb_width + log2_zo_tb_height - (log2_sb_w + log2_sb_h))) - 1;
rc->hist_value = sps->r->sps_persistent_rice_adaptation_enabled_flag ? (1 << lc->ep->stat_coeff[tb->c_idx]) : 0;
rc->update_hist = sps->r->sps_persistent_rice_adaptation_enabled_flag ? 1 : 0;
rc->rem_bins_pass1 = (( 1 << ( log2_zo_tb_width + log2_zo_tb_height)) * 7 ) >> 2;
rc->sb_scan_x_off = ff_vvc_diag_scan_x[log2_zo_tb_width - log2_sb_w][log2_zo_tb_height - log2_sb_h];
rc->sb_scan_y_off = ff_vvc_diag_scan_y[log2_zo_tb_width - log2_sb_w][log2_zo_tb_height - log2_sb_h];
rc->scan_x_off = ff_vvc_diag_scan_x[log2_sb_w][log2_sb_h];
rc->scan_y_off = ff_vvc_diag_scan_y[log2_sb_w][log2_sb_h];
rc->infer_sb_cbf = 1;
rc->width_in_sbs = (1 << (log2_zo_tb_width - log2_sb_w));
rc->height_in_sbs = (1 << (log2_zo_tb_height - log2_sb_h));
rc->nb_sbs = rc->width_in_sbs * rc->height_in_sbs;
rc->last_scan_pos = rc->num_sb_coeff;
rc->qstate = 0;
rc->tb = tb;
}
static int residual_ts_coding_subblock(VVCLocalContext *lc, ResidualCoding* rc, const int i)
{
const CodingUnit *cu = lc->cu;
TransformBlock *tb = rc->tb;
const int bdpcm_flag = cu->bdpcm_flag[tb->c_idx];
const int xs = rc->sb_scan_x_off[i];
const int ys = rc->sb_scan_y_off[i];
uint8_t *sb_coded_flag = rc->sb_coded_flag + ys * rc->width_in_sbs + xs;
int infer_sb_sig_coeff_flag = 1;
int last_scan_pos_pass1 = -1, last_scan_pos_pass2 = -1, n;
int abs_level_gtx_flag[MAX_SUB_BLOCK_SIZE * MAX_SUB_BLOCK_SIZE];
int abs_level_pass2[MAX_SUB_BLOCK_SIZE * MAX_SUB_BLOCK_SIZE]; ///< AbsLevelPass2
if (i != rc->last_sub_block || !rc->infer_sb_cbf)
*sb_coded_flag = sb_coded_flag_decode(lc, sb_coded_flag, rc, xs, ys);
else
*sb_coded_flag = 1;
if (*sb_coded_flag && i < rc->last_sub_block)
rc->infer_sb_cbf = 0;
//first scan pass
for (n = 0; n < rc->num_sb_coeff && rc->rem_bins_pass1 >= 4; n++) {
const int xc = (xs << rc->log2_sb_w) + rc->scan_x_off[n];
const int yc = (ys << rc->log2_sb_h) + rc->scan_y_off[n];
const int off = yc * tb->tb_width + xc;
int *sig_coeff_flag = rc->sig_coeff_flag + off;
int *abs_level_pass1 = rc->abs_level_pass1 + off;
int *coeff_sign_level = rc->coeff_sign_level + off;
int par_level_flag = 0;
abs_level_gtx_flag[n] = 0;
last_scan_pos_pass1 = n;
if (*sb_coded_flag && (n != rc->num_sb_coeff - 1 || !infer_sb_sig_coeff_flag)) {
*sig_coeff_flag = sig_coeff_flag_decode(lc, rc, xc, yc);
rc->rem_bins_pass1--;
if (*sig_coeff_flag)
infer_sb_sig_coeff_flag = 0;
} else {
*sig_coeff_flag = (n == rc->num_sb_coeff - 1) && infer_sb_sig_coeff_flag && *sb_coded_flag;
}
*coeff_sign_level = 0;
if (*sig_coeff_flag) {
*coeff_sign_level = 1 - 2 * coeff_sign_flag_ts_decode(lc, cu, rc, xc, yc);
abs_level_gtx_flag[n] = abs_level_gt1_flag_ts_decode(lc, cu, rc, xc, yc);
rc->rem_bins_pass1 -= 2;
if (abs_level_gtx_flag[n]) {
par_level_flag = par_level_flag_ts_decode(lc);
rc->rem_bins_pass1--;
}
}
*abs_level_pass1 = *sig_coeff_flag + par_level_flag + abs_level_gtx_flag[n];
}
//greater than x scan pass
for (n = 0; n < rc->num_sb_coeff && rc->rem_bins_pass1 >= 4; n++) {
const int xc = (xs << rc->log2_sb_w) + rc->scan_x_off[n];
const int yc = (ys << rc->log2_sb_h) + rc->scan_y_off[n];
const int off = yc * tb->tb_width + xc;
abs_level_pass2[n] = rc->abs_level_pass1[off];
for (int j = 1; j < 5 && abs_level_gtx_flag[n]; j++) {
abs_level_gtx_flag[n] = abs_level_gtx_flag_ts_decode(lc, j);
abs_level_pass2[n] += abs_level_gtx_flag[n] << 1;
rc->rem_bins_pass1--;
}
last_scan_pos_pass2 = n;
}
/* remainder scan pass */
for (n = 0; n < rc->num_sb_coeff; n++) {
const int xc = (xs << rc->log2_sb_w) + rc->scan_x_off[n];
const int yc = (ys << rc->log2_sb_h) + rc->scan_y_off[n];
const int off = yc * tb->tb_width + xc;
const int *abs_level_pass1 = rc->abs_level_pass1 + off;
int *abs_level = rc->abs_level + off;
int *coeff_sign_level = rc->coeff_sign_level + off;
int abs_remainder = 0;
if ((n <= last_scan_pos_pass2 && abs_level_pass2[n] >= 10) ||
(n > last_scan_pos_pass2 && n <= last_scan_pos_pass1 &&
*abs_level_pass1 >= 2) ||
(n > last_scan_pos_pass1 && *sb_coded_flag))
abs_remainder = abs_remainder_ts_decode(lc, rc, xc, yc);
if (n <= last_scan_pos_pass2) {
*abs_level = abs_level_pass2[n] + 2 * abs_remainder;
} else if (n <= last_scan_pos_pass1) {
*abs_level = *abs_level_pass1 + 2 * abs_remainder;
} else {
*abs_level = abs_remainder;
if (abs_remainder) {
//n > lastScanPosPass1
*coeff_sign_level = 1 - 2 * coeff_sign_flag_decode(lc);
}
}
if (!bdpcm_flag && n <= last_scan_pos_pass1) {
const int left = xc > 0 ? abs_level[-1] : 0;
const int above = yc > 0 ? abs_level[-tb->tb_width] : 0;
const int pred = FFMAX(left, above);
if (*abs_level == 1 && pred > 0)
*abs_level = pred;
else if (*abs_level > 0 && *abs_level <= pred)
(*abs_level)--;
}
if (*abs_level) {
tb->coeffs[off] = *coeff_sign_level * *abs_level;
tb->max_scan_x = FFMAX(xc, tb->max_scan_x);
tb->max_scan_y = FFMAX(yc, tb->max_scan_y);
tb->min_scan_x = FFMIN(xc, tb->min_scan_x);
tb->min_scan_y = FFMIN(yc, tb->min_scan_y);
} else {
tb->coeffs[off] = 0;
}
}
return 0;
}
static int hls_residual_ts_coding(VVCLocalContext *lc, TransformBlock *tb)
{
ResidualCoding rc;
tb->min_scan_x = tb->min_scan_y = INT_MAX;
init_residual_coding(lc, &rc, tb->log2_tb_width, tb->log2_tb_height, tb);
for (int i = 0; i <= rc.last_sub_block; i++) {
int ret = residual_ts_coding_subblock(lc, &rc, i);
if (ret < 0)
return ret;
}
return 0;
}
static inline int residual_coding_subblock(VVCLocalContext *lc, ResidualCoding *rc, const int i)
{
const H266RawSliceHeader *rsh = lc->sc->sh.r;
TransformBlock *tb = rc->tb;
int first_sig_scan_pos_sb, last_sig_scan_pos_sb;
int first_pos_mode0, first_pos_mode1;
int infer_sb_dc_sig_coeff_flag = 0;
int n, sig_hidden_flag, sum = 0;
int abs_level_gt2_flag[MAX_SUB_BLOCK_SIZE * MAX_SUB_BLOCK_SIZE];
const int start_qstate_sb = rc->qstate;
const int xs = rc->sb_scan_x_off[i];
const int ys = rc->sb_scan_y_off[i];
uint8_t *sb_coded_flag = rc->sb_coded_flag + ys * rc->width_in_sbs + xs;
av_assert0(rc->num_sb_coeff <= MAX_SUB_BLOCK_SIZE * MAX_SUB_BLOCK_SIZE);
if (i < rc->last_sub_block && i > 0) {
*sb_coded_flag = sb_coded_flag_decode(lc, sb_coded_flag, rc, xs, ys);
infer_sb_dc_sig_coeff_flag = 1;
} else {
*sb_coded_flag = 1;
}
if (*sb_coded_flag && (xs > 3 || ys > 3) && !tb->c_idx)
lc->parse.mts_zero_out_sig_coeff_flag = 0;
if (!*sb_coded_flag)
return 0;
first_sig_scan_pos_sb = rc->num_sb_coeff;
last_sig_scan_pos_sb = -1;
first_pos_mode0 = (i == rc->last_sub_block ? rc->last_scan_pos : rc->num_sb_coeff -1);
first_pos_mode1 = first_pos_mode0;
for (n = first_pos_mode0; n >= 0 && rc->rem_bins_pass1 >= 4; n--) {
const int xc = (xs << rc->log2_sb_w) + rc->scan_x_off[n];
const int yc = (ys << rc->log2_sb_h) + rc->scan_y_off[n];
const int last = (xc == rc->last_significant_coeff_x && yc == rc->last_significant_coeff_y);
int *abs_level_pass1 = rc->abs_level_pass1 + yc * tb->tb_width + xc;
int *sig_coeff_flag = rc->sig_coeff_flag + yc * tb->tb_width + xc;
if ((n > 0 || !infer_sb_dc_sig_coeff_flag ) && !last) {
*sig_coeff_flag = sig_coeff_flag_decode(lc, rc, xc, yc);
rc->rem_bins_pass1--;
if (*sig_coeff_flag)
infer_sb_dc_sig_coeff_flag = 0;
} else {
*sig_coeff_flag = last || (!rc->scan_x_off[n] && !rc ->scan_y_off[n] &&
infer_sb_dc_sig_coeff_flag);
}
*abs_level_pass1 = 0;
if (*sig_coeff_flag) {
int abs_level_gt1_flag, par_level_flag = 0;
const int inc = get_gtx_flag_inc(rc, xc, yc, last);
abs_level_gt1_flag = abs_level_gtx_flag_decode(lc, inc);
rc->rem_bins_pass1--;
if (abs_level_gt1_flag) {
par_level_flag = par_level_flag_decode(lc, inc);
abs_level_gt2_flag[n] = abs_level_gtx_flag_decode(lc, inc + 32);
rc->rem_bins_pass1 -= 2;
} else {
abs_level_gt2_flag[n] = 0;
}
if (last_sig_scan_pos_sb == -1)
last_sig_scan_pos_sb = n;
first_sig_scan_pos_sb = n;
*abs_level_pass1 =
1 + par_level_flag + abs_level_gt1_flag + (abs_level_gt2_flag[n] << 1);
} else {
abs_level_gt2_flag[n] = 0;
}
if (rsh->sh_dep_quant_used_flag)
rc->qstate = qstate_translate_table[rc->qstate][*abs_level_pass1 & 1];
first_pos_mode1 = n - 1;
}
for (n = first_pos_mode0; n > first_pos_mode1; n--) {
const int xc = (xs << rc->log2_sb_w) + rc->scan_x_off[n];
const int yc = (ys << rc->log2_sb_h) + rc->scan_y_off[n];
const int *abs_level_pass1 = rc->abs_level_pass1 + yc * tb->tb_width + xc;
int *abs_level = rc->abs_level + yc * tb->tb_width + xc;
*abs_level = *abs_level_pass1;
if (abs_level_gt2_flag[n]) {
const int abs_remainder = abs_remainder_decode(lc, rc, xc, yc);
ep_update_hist(lc->ep, rc, abs_remainder, 2);
*abs_level += 2 * abs_remainder;
}
}
for (n = first_pos_mode1; n >= 0; n--) {
const int xc = (xs << rc->log2_sb_w) + rc->scan_x_off[n];
const int yc = (ys << rc->log2_sb_h) + rc->scan_y_off[n];
int *abs_level = rc->abs_level + yc * tb->tb_width + xc;
if (*sb_coded_flag) {
const int dec_abs_level = dec_abs_level_decode(lc, rc, xc, yc, abs_level);
ep_update_hist(lc->ep, rc, dec_abs_level, 0);
}
if (*abs_level > 0) {
if (last_sig_scan_pos_sb == -1)
last_sig_scan_pos_sb = n;
first_sig_scan_pos_sb = n;
}
if (rsh->sh_dep_quant_used_flag)
rc->qstate = qstate_translate_table[rc->qstate][*abs_level & 1];
}
sig_hidden_flag = rsh->sh_sign_data_hiding_used_flag &&
(last_sig_scan_pos_sb - first_sig_scan_pos_sb > 3 ? 1 : 0);
if (rsh->sh_dep_quant_used_flag)
rc->qstate = start_qstate_sb;
n = (i == rc->last_sub_block ? rc->last_scan_pos : rc->num_sb_coeff -1);
for (/* nothing */; n >= 0; n--) {
int trans_coeff_level;
const int xc = (xs << rc->log2_sb_w) + rc->scan_x_off[n];
const int yc = (ys << rc->log2_sb_h) + rc->scan_y_off[n];
const int off = yc * tb->tb_width + xc;
const int *abs_level = rc->abs_level + off;
if (*abs_level > 0) {
int sign = 1;
if (!sig_hidden_flag || (n != first_sig_scan_pos_sb))
sign = 1 - 2 * coeff_sign_flag_decode(lc);
if (rsh->sh_dep_quant_used_flag) {
trans_coeff_level = (2 * *abs_level - (rc->qstate > 1)) * sign;
} else {
trans_coeff_level = *abs_level * sign;
if (sig_hidden_flag) {
sum += *abs_level;
if (n == first_sig_scan_pos_sb && (sum % 2))
trans_coeff_level = -trans_coeff_level;
}
}
tb->coeffs[off] = trans_coeff_level;
tb->max_scan_x = FFMAX(xc, tb->max_scan_x);
tb->max_scan_y = FFMAX(yc, tb->max_scan_y);
}
if (rsh->sh_dep_quant_used_flag)
rc->qstate = qstate_translate_table[rc->qstate][*abs_level & 1];
}
return 0;
}
static void derive_last_scan_pos(ResidualCoding *rc)
{
int xc, yc, xs, ys;
do {
if (!rc->last_scan_pos) {
rc->last_scan_pos = rc->num_sb_coeff;
rc->last_sub_block--;
}
rc->last_scan_pos--;
xs = rc->sb_scan_x_off[rc->last_sub_block];
ys = rc->sb_scan_y_off[rc->last_sub_block];
xc = (xs << rc->log2_sb_w) + rc->scan_x_off[rc->last_scan_pos];
yc = (ys << rc->log2_sb_h) + rc->scan_y_off[rc->last_scan_pos];
} while ((xc != rc->last_significant_coeff_x) || (yc != rc->last_significant_coeff_y));
}
static void last_significant_coeff_x_y_decode(ResidualCoding *rc, VVCLocalContext *lc,
const int log2_zo_tb_width, const int log2_zo_tb_height)
{
const H266RawSliceHeader *rsh = lc->sc->sh.r;
const TransformBlock *tb = rc->tb;
int last_significant_coeff_x, last_significant_coeff_y;
last_significant_coeff_x = last_significant_coeff_x_prefix_decode(lc,
tb->log2_tb_width, log2_zo_tb_width, tb->c_idx);
last_significant_coeff_y = last_significant_coeff_y_prefix_decode(lc,
tb->log2_tb_height, log2_zo_tb_height, tb->c_idx);
if (last_significant_coeff_x > 3) {
int suffix = last_sig_coeff_suffix_decode(lc, last_significant_coeff_x);
last_significant_coeff_x = (1 << ((last_significant_coeff_x >> 1) - 1)) *
(2 + (last_significant_coeff_x & 1)) + suffix;
}
if (last_significant_coeff_y > 3) {
int suffix = last_sig_coeff_suffix_decode(lc, last_significant_coeff_y);
last_significant_coeff_y = (1 << ((last_significant_coeff_y >> 1) - 1)) *
(2 + (last_significant_coeff_y & 1)) + suffix;
}
if (rsh->sh_reverse_last_sig_coeff_flag) {
last_significant_coeff_x = (1 << log2_zo_tb_width) - 1 - last_significant_coeff_x;
last_significant_coeff_y = (1 << log2_zo_tb_height) - 1 - last_significant_coeff_y;
}
rc->last_significant_coeff_x = last_significant_coeff_x;
rc->last_significant_coeff_y = last_significant_coeff_y;
}
static int hls_residual_coding(VVCLocalContext *lc, TransformBlock *tb)
{
const VVCSPS *sps = lc->fc->ps.sps;
const CodingUnit *cu = lc->cu;
const int log2_tb_width = tb->log2_tb_width;
const int log2_tb_height = tb->log2_tb_height;
const int c_idx = tb->c_idx;
int log2_zo_tb_width, log2_zo_tb_height;
ResidualCoding rc;
if (sps->r->sps_mts_enabled_flag && cu->sbt_flag && !c_idx && log2_tb_width == 5 && log2_tb_height < 6)
log2_zo_tb_width = 4;
else
log2_zo_tb_width = FFMIN(log2_tb_width, 5 );
if (sps->r->sps_mts_enabled_flag && cu->sbt_flag && !c_idx && log2_tb_width < 6 && log2_tb_height == 5 )
log2_zo_tb_height = 4;
else
log2_zo_tb_height = FFMIN(log2_tb_height, 5);
init_residual_coding(lc, &rc, log2_zo_tb_width, log2_zo_tb_height, tb);
last_significant_coeff_x_y_decode(&rc, lc, log2_zo_tb_width, log2_zo_tb_height);
derive_last_scan_pos(&rc);
if (!rc.last_sub_block && log2_tb_width >= 2 && log2_tb_height >= 2 && !tb->ts && rc.last_scan_pos > 0)
lc->parse.lfnst_dc_only = 0;
if ((rc.last_sub_block > 0 && log2_tb_width >= 2 && log2_tb_height >= 2 ) ||
(rc.last_scan_pos > 7 && (log2_tb_width == 2 || log2_tb_width == 3 ) &&
log2_tb_width == log2_tb_height))
lc->parse.lfnst_zero_out_sig_coeff_flag = 0;
if ((rc.last_sub_block > 0 || rc.last_scan_pos > 0 ) && !c_idx)
lc->parse.mts_dc_only = 0;
memset(tb->coeffs, 0, tb->tb_width * tb->tb_height * sizeof(*tb->coeffs));
memset(rc.abs_level, 0, tb->tb_width * tb->tb_height * sizeof(rc.abs_level[0]));
memset(rc.sb_coded_flag, 0, rc.nb_sbs);
memset(rc.abs_level_pass1, 0, tb->tb_width * tb->tb_height * sizeof(rc.abs_level_pass1[0]));
memset(rc.sig_coeff_flag, 0, tb->tb_width * tb->tb_height * sizeof(rc.sig_coeff_flag[0]));
for (int i = rc.last_sub_block; i >= 0; i--) {
int ret = residual_coding_subblock(lc, &rc, i);
if (ret < 0)
return ret;
}
return 0;
}
int ff_vvc_residual_coding(VVCLocalContext *lc, TransformBlock *tb)
{
const H266RawSliceHeader *rsh = lc->sc->sh.r;
const int ts = !rsh->sh_ts_residual_coding_disabled_flag && tb->ts;
return ts ? hls_residual_ts_coding(lc, tb) : hls_residual_coding(lc, tb);
}
int ff_vvc_cu_coded_flag(VVCLocalContext *lc)
{
return GET_CABAC(CU_CODED_FLAG);
}
int ff_vvc_sbt_flag(VVCLocalContext *lc)
{
const int w = lc->cu->cb_width;
const int h = lc->cu->cb_height;
const int inc = w * h <= 256;
return GET_CABAC(CU_SBT_FLAG + inc);
}
int ff_vvc_sbt_quad_flag(VVCLocalContext *lc)
{
return GET_CABAC(CU_SBT_QUAD_FLAG);
}
int ff_vvc_sbt_horizontal_flag(VVCLocalContext *lc)
{
const int w = lc->cu->cb_width;
const int h = lc->cu->cb_height;
const int inc = (w == h) ? 0 : ((w < h) ? 1 : 2);
return GET_CABAC(CU_SBT_HORIZONTAL_FLAG + inc);
}
int ff_vvc_sbt_pos_flag(VVCLocalContext *lc)
{
return GET_CABAC(CU_SBT_POS_FLAG);
}
int ff_vvc_lfnst_idx(VVCLocalContext *lc, const int inc)
{
if (!GET_CABAC(LFNST_IDX + inc))
return 0;
if (!GET_CABAC(LFNST_IDX + 2))
return 1;
return 2;
}
int ff_vvc_mts_idx(VVCLocalContext *lc)
{
int i;
for (i = 0; i < 4; i++) {
if (!GET_CABAC(MTS_IDX + i))
return i;
}
return i;
}
int ff_vvc_end_of_slice_flag_decode(VVCLocalContext *lc)
{
return get_cabac_terminate(&lc->ep->cc);
}
int ff_vvc_end_of_tile_one_bit(VVCLocalContext *lc)
{
return get_cabac_terminate(&lc->ep->cc);
}
int ff_vvc_end_of_subset_one_bit(VVCLocalContext *lc)
{
return get_cabac_terminate(&lc->ep->cc);
}