mirror of
https://repo.dactyloidae.xyz/Dactyloidae/UXP.git
synced 2026-10-09 08:47:31 +09:00
Update aom to commit id e87fb2378f01103d5d6e477a4ef6892dc714e614
This commit is contained in:
parent
debbee1e2a
commit
992c6637e3
429 changed files with 76047 additions and 40937 deletions
295
third_party/aom/av1/common/entropymode.h
vendored
295
third_party/aom/av1/common/entropymode.h
vendored
|
|
@ -33,14 +33,11 @@ extern "C" {
|
|||
#define TX_SIZE_CONTEXTS 2
|
||||
|
||||
#define INTER_OFFSET(mode) ((mode)-NEARESTMV)
|
||||
#if CONFIG_EXT_INTER
|
||||
#if CONFIG_COMPOUND_SINGLEREF
|
||||
#define INTER_SINGLEREF_COMP_OFFSET(mode) ((mode)-SR_NEAREST_NEARMV)
|
||||
#endif // CONFIG_COMPOUND_SINGLEREF
|
||||
#define INTER_COMPOUND_OFFSET(mode) ((mode)-NEAREST_NEARESTMV)
|
||||
#endif // CONFIG_EXT_INTER
|
||||
|
||||
#if CONFIG_PALETTE
|
||||
// Number of possible contexts for a color index.
|
||||
// As can be seen from av1_get_palette_color_index_context(), the possible
|
||||
// contexts are (2,0,0), (2,2,1), (3,2,0), (4,1,0), (5,0,0). These are mapped to
|
||||
|
|
@ -70,11 +67,10 @@ extern "C" {
|
|||
#define PALETTE_UV_MODE_CONTEXTS 2
|
||||
|
||||
#define PALETTE_MAX_BLOCK_SIZE (64 * 64)
|
||||
#endif // CONFIG_PALETTE
|
||||
|
||||
#if CONFIG_INTRABC
|
||||
#define INTRABC_PROB_DEFAULT 192
|
||||
#endif // CONFIG_INTRABC
|
||||
#if CONFIG_KF_CTX
|
||||
#define KF_MODE_CONTEXTS 5
|
||||
#endif
|
||||
|
||||
struct AV1Common;
|
||||
|
||||
|
|
@ -98,12 +94,8 @@ typedef struct frame_contexts {
|
|||
#else
|
||||
aom_prob partition_prob[PARTITION_CONTEXTS][PARTITION_TYPES - 1];
|
||||
#endif
|
||||
av1_coeff_probs_model coef_probs[TX_SIZES][PLANE_TYPES];
|
||||
coeff_cdf_model coef_tail_cdfs[TX_SIZES][PLANE_TYPES];
|
||||
coeff_cdf_model coef_head_cdfs[TX_SIZES][PLANE_TYPES];
|
||||
aom_prob blockzero_probs[TX_SIZES][PLANE_TYPES][REF_TYPES][BLOCKZ_CONTEXTS];
|
||||
aom_prob switchable_interp_prob[SWITCHABLE_FILTER_CONTEXTS]
|
||||
[SWITCHABLE_FILTERS - 1];
|
||||
#if CONFIG_ADAPT_SCAN
|
||||
// TODO(angiebird): try aom_prob
|
||||
#if CONFIG_CHROMA_2X2
|
||||
|
|
@ -179,6 +171,38 @@ typedef struct frame_contexts {
|
|||
aom_prob coeff_base[TX_SIZES][PLANE_TYPES][NUM_BASE_LEVELS]
|
||||
[COEFF_BASE_CONTEXTS];
|
||||
aom_prob coeff_lps[TX_SIZES][PLANE_TYPES][LEVEL_CONTEXTS];
|
||||
#if BR_NODE
|
||||
aom_prob coeff_br[TX_SIZES][PLANE_TYPES][BASE_RANGE_SETS][LEVEL_CONTEXTS];
|
||||
#endif
|
||||
#if CONFIG_CTX1D
|
||||
aom_prob eob_mode[TX_SIZES][PLANE_TYPES][TX_CLASSES];
|
||||
aom_prob empty_line[TX_SIZES][PLANE_TYPES][TX_CLASSES][EMPTY_LINE_CONTEXTS];
|
||||
aom_prob hv_eob[TX_SIZES][PLANE_TYPES][TX_CLASSES][HV_EOB_CONTEXTS];
|
||||
#endif // CONFIG_CTX1D
|
||||
|
||||
#if LV_MAP_PROB
|
||||
aom_cdf_prob txb_skip_cdf[TX_SIZES][TXB_SKIP_CONTEXTS][CDF_SIZE(2)];
|
||||
aom_cdf_prob nz_map_cdf[TX_SIZES][PLANE_TYPES][SIG_COEF_CONTEXTS]
|
||||
[CDF_SIZE(2)];
|
||||
aom_cdf_prob eob_flag_cdf[TX_SIZES][PLANE_TYPES][EOB_COEF_CONTEXTS]
|
||||
[CDF_SIZE(2)];
|
||||
aom_cdf_prob dc_sign_cdf[PLANE_TYPES][DC_SIGN_CONTEXTS][CDF_SIZE(2)];
|
||||
aom_cdf_prob coeff_base_cdf[TX_SIZES][PLANE_TYPES][NUM_BASE_LEVELS]
|
||||
[COEFF_BASE_CONTEXTS][CDF_SIZE(2)];
|
||||
aom_cdf_prob coeff_lps_cdf[TX_SIZES][PLANE_TYPES][LEVEL_CONTEXTS]
|
||||
[CDF_SIZE(2)];
|
||||
#if BR_NODE
|
||||
aom_cdf_prob coeff_br_cdf[TX_SIZES][PLANE_TYPES][BASE_RANGE_SETS]
|
||||
[LEVEL_CONTEXTS][CDF_SIZE(2)];
|
||||
#endif
|
||||
#if CONFIG_CTX1D
|
||||
aom_cdf_prob eob_mode_cdf[TX_SIZES][PLANE_TYPES][TX_CLASSES][CDF_SIZE(2)];
|
||||
aom_cdf_prob empty_line_cdf[TX_SIZES][PLANE_TYPES][TX_CLASSES]
|
||||
[EMPTY_LINE_CONTEXTS][CDF_SIZE(2)];
|
||||
aom_cdf_prob hv_eob_cdf[TX_SIZES][PLANE_TYPES][TX_CLASSES][HV_EOB_CONTEXTS]
|
||||
[CDF_SIZE(2)];
|
||||
#endif // CONFIG_CTX1D
|
||||
#endif // LV_MAP_PROB
|
||||
#endif
|
||||
|
||||
aom_prob newmv_prob[NEWMV_MODE_CONTEXTS];
|
||||
|
|
@ -192,7 +216,6 @@ typedef struct frame_contexts {
|
|||
aom_cdf_prob drl_cdf[DRL_MODE_CONTEXTS][CDF_SIZE(2)];
|
||||
#endif
|
||||
|
||||
#if CONFIG_EXT_INTER
|
||||
aom_prob inter_compound_mode_probs[INTER_MODE_CONTEXTS]
|
||||
[INTER_COMPOUND_MODES - 1];
|
||||
aom_cdf_prob inter_compound_mode_cdf[INTER_MODE_CONTEXTS]
|
||||
|
|
@ -204,7 +227,9 @@ typedef struct frame_contexts {
|
|||
INTER_SINGLEREF_COMP_MODES)];
|
||||
#endif // CONFIG_COMPOUND_SINGLEREF
|
||||
aom_prob compound_type_prob[BLOCK_SIZES_ALL][COMPOUND_TYPES - 1];
|
||||
#if CONFIG_WEDGE || CONFIG_COMPOUND_SEGMENT
|
||||
aom_cdf_prob compound_type_cdf[BLOCK_SIZES_ALL][CDF_SIZE(COMPOUND_TYPES)];
|
||||
#endif // CONFIG_WEDGE || CONFIG_COMPOUND_SEGMENT
|
||||
#if CONFIG_INTERINTRA
|
||||
aom_prob interintra_prob[BLOCK_SIZE_GROUPS];
|
||||
aom_prob wedge_interintra_prob[BLOCK_SIZES_ALL];
|
||||
|
|
@ -216,7 +241,6 @@ typedef struct frame_contexts {
|
|||
aom_cdf_prob interintra_mode_cdf[BLOCK_SIZE_GROUPS]
|
||||
[CDF_SIZE(INTERINTRA_MODES)];
|
||||
#endif // CONFIG_INTERINTRA
|
||||
#endif // CONFIG_EXT_INTER
|
||||
#if CONFIG_MOTION_VAR || CONFIG_WARPED_MOTION
|
||||
aom_prob motion_mode_prob[BLOCK_SIZES_ALL][MOTION_MODES - 1];
|
||||
aom_cdf_prob motion_mode_cdf[BLOCK_SIZES_ALL][CDF_SIZE(MOTION_MODES)];
|
||||
|
|
@ -226,15 +250,18 @@ typedef struct frame_contexts {
|
|||
[CDF_SIZE(MAX_NCOBMC_MODES)];
|
||||
#endif
|
||||
#if CONFIG_MOTION_VAR && CONFIG_WARPED_MOTION
|
||||
#if CONFIG_NCOBMC_ADAPT_WEIGHT
|
||||
aom_prob ncobmc_prob[BLOCK_SIZES_ALL][OBMC_FAMILY_MODES - 1];
|
||||
aom_cdf_prob ncobmc_cdf[BLOCK_SIZES_ALL][CDF_SIZE(OBMC_FAMILY_MODES)];
|
||||
#endif // CONFIG_NCOBMC_ADAPT_WEIGHT
|
||||
aom_prob obmc_prob[BLOCK_SIZES_ALL];
|
||||
#if CONFIG_NEW_MULTISYMBOL
|
||||
#if CONFIG_NEW_MULTISYMBOL || CONFIG_NCOBMC_ADAPT_WEIGHT
|
||||
aom_cdf_prob obmc_cdf[BLOCK_SIZES_ALL][CDF_SIZE(2)];
|
||||
#endif // CONFIG_NEW_MULTISYMBOL
|
||||
#endif // CONFIG_MOTION_VAR && CONFIG_WARPED_MOTION
|
||||
#endif // CONFIG_MOTION_VAR || CONFIG_WARPED_MOTION
|
||||
aom_prob intra_inter_prob[INTRA_INTER_CONTEXTS];
|
||||
aom_prob comp_inter_prob[COMP_INTER_CONTEXTS];
|
||||
#if CONFIG_PALETTE
|
||||
aom_cdf_prob palette_y_size_cdf[PALETTE_BLOCK_SIZES][CDF_SIZE(PALETTE_SIZES)];
|
||||
aom_cdf_prob palette_uv_size_cdf[PALETTE_BLOCK_SIZES]
|
||||
[CDF_SIZE(PALETTE_SIZES)];
|
||||
|
|
@ -244,8 +271,16 @@ typedef struct frame_contexts {
|
|||
aom_cdf_prob palette_uv_color_index_cdf[PALETTE_SIZES]
|
||||
[PALETTE_COLOR_INDEX_CONTEXTS]
|
||||
[CDF_SIZE(PALETTE_COLORS)];
|
||||
#endif // CONFIG_PALETTE
|
||||
#if CONFIG_MRC_TX
|
||||
aom_cdf_prob mrc_mask_inter_cdf[PALETTE_SIZES][PALETTE_COLOR_INDEX_CONTEXTS]
|
||||
[CDF_SIZE(PALETTE_COLORS)];
|
||||
aom_cdf_prob mrc_mask_intra_cdf[PALETTE_SIZES][PALETTE_COLOR_INDEX_CONTEXTS]
|
||||
[CDF_SIZE(PALETTE_COLORS)];
|
||||
#endif // CONFIG_MRC_TX
|
||||
#if CONFIG_NEW_MULTISYMBOL
|
||||
aom_cdf_prob palette_y_mode_cdf[PALETTE_BLOCK_SIZES][PALETTE_Y_MODE_CONTEXTS]
|
||||
[CDF_SIZE(2)];
|
||||
aom_cdf_prob palette_uv_mode_cdf[PALETTE_UV_MODE_CONTEXTS][CDF_SIZE(2)];
|
||||
aom_cdf_prob comp_inter_cdf[COMP_INTER_CONTEXTS][CDF_SIZE(2)];
|
||||
aom_cdf_prob single_ref_cdf[REF_CONTEXTS][SINGLE_REFS - 1][CDF_SIZE(2)];
|
||||
#endif
|
||||
|
|
@ -273,12 +308,14 @@ typedef struct frame_contexts {
|
|||
aom_cdf_prob comp_ref_cdf[REF_CONTEXTS][COMP_REFS - 1][CDF_SIZE(2)];
|
||||
#endif // CONFIG_EXT_REFS
|
||||
#endif
|
||||
#if CONFIG_EXT_INTER && CONFIG_COMPOUND_SINGLEREF
|
||||
#if CONFIG_COMPOUND_SINGLEREF
|
||||
aom_prob comp_inter_mode_prob[COMP_INTER_MODE_CONTEXTS];
|
||||
#endif // CONFIG_EXT_INTER && CONFIG_COMPOUND_SINGLEREF
|
||||
aom_prob tx_size_probs[MAX_TX_DEPTH][TX_SIZE_CONTEXTS][MAX_TX_DEPTH];
|
||||
#endif // CONFIG_COMPOUND_SINGLEREF
|
||||
#if CONFIG_RECT_TX_EXT && (CONFIG_EXT_TX || CONFIG_VAR_TX)
|
||||
aom_prob quarter_tx_size_prob;
|
||||
#if CONFIG_NEW_MULTISYMBOL
|
||||
aom_cdf_prob quarter_tx_size_cdf[CDF_SIZE(2)];
|
||||
#endif
|
||||
#endif
|
||||
#if CONFIG_VAR_TX
|
||||
aom_prob txfm_partition_prob[TXFM_PARTITION_CONTEXTS];
|
||||
|
|
@ -294,17 +331,9 @@ typedef struct frame_contexts {
|
|||
nmv_context nmvc[NMV_CONTEXTS];
|
||||
#if CONFIG_INTRABC
|
||||
nmv_context ndvc;
|
||||
aom_prob intrabc_prob;
|
||||
aom_cdf_prob intrabc_cdf[CDF_SIZE(2)];
|
||||
#endif
|
||||
int initialized;
|
||||
#if CONFIG_EXT_TX
|
||||
aom_prob inter_ext_tx_prob[EXT_TX_SETS_INTER][EXT_TX_SIZES][TX_TYPES - 1];
|
||||
aom_prob intra_ext_tx_prob[EXT_TX_SETS_INTRA][EXT_TX_SIZES][INTRA_MODES]
|
||||
[TX_TYPES - 1];
|
||||
#else
|
||||
aom_prob intra_ext_tx_prob[EXT_TX_SIZES][TX_TYPES][TX_TYPES - 1];
|
||||
aom_prob inter_ext_tx_prob[EXT_TX_SIZES][TX_TYPES - 1];
|
||||
#endif // CONFIG_EXT_TX
|
||||
#if CONFIG_SUPERTX
|
||||
aom_prob supertx_prob[PARTITION_SUPERTX_CONTEXTS][TX_SIZES];
|
||||
#endif // CONFIG_SUPERTX
|
||||
|
|
@ -329,19 +358,25 @@ typedef struct frame_contexts {
|
|||
#endif
|
||||
aom_cdf_prob switchable_interp_cdf[SWITCHABLE_FILTER_CONTEXTS]
|
||||
[CDF_SIZE(SWITCHABLE_FILTERS)];
|
||||
/* kf_y_cdf is discarded after use, so does not require persistent storage.
|
||||
However, we keep it with the other CDFs in this struct since it needs to
|
||||
be copied to each tile to support parallelism just like the others.
|
||||
*/
|
||||
/* kf_y_cdf is discarded after use, so does not require persistent storage.
|
||||
However, we keep it with the other CDFs in this struct since it needs to
|
||||
be copied to each tile to support parallelism just like the others.
|
||||
*/
|
||||
#if CONFIG_KF_CTX
|
||||
aom_cdf_prob kf_y_cdf[KF_MODE_CONTEXTS][KF_MODE_CONTEXTS]
|
||||
[CDF_SIZE(INTRA_MODES)];
|
||||
#else
|
||||
aom_cdf_prob kf_y_cdf[INTRA_MODES][INTRA_MODES][CDF_SIZE(INTRA_MODES)];
|
||||
#endif
|
||||
aom_cdf_prob tx_size_cdf[MAX_TX_DEPTH][TX_SIZE_CONTEXTS]
|
||||
[CDF_SIZE(MAX_TX_DEPTH + 1)];
|
||||
#if CONFIG_DELTA_Q
|
||||
aom_cdf_prob delta_q_cdf[CDF_SIZE(DELTA_Q_PROBS + 1)];
|
||||
#if CONFIG_EXT_DELTA_Q
|
||||
#if CONFIG_LOOPFILTER_LEVEL
|
||||
aom_cdf_prob delta_lf_multi_cdf[FRAME_LF_COUNT][CDF_SIZE(DELTA_LF_PROBS + 1)];
|
||||
#endif // CONFIG_LOOPFILTER_LEVEL
|
||||
aom_cdf_prob delta_lf_cdf[CDF_SIZE(DELTA_LF_PROBS + 1)];
|
||||
#endif
|
||||
#endif // CONFIG_DELTA_Q
|
||||
#if CONFIG_EXT_TX
|
||||
aom_cdf_prob intra_ext_tx_cdf[EXT_TX_SETS_INTRA][EXT_TX_SIZES][INTRA_MODES]
|
||||
[CDF_SIZE(TX_TYPES)];
|
||||
|
|
@ -351,23 +386,34 @@ typedef struct frame_contexts {
|
|||
aom_cdf_prob intra_ext_tx_cdf[EXT_TX_SIZES][TX_TYPES][CDF_SIZE(TX_TYPES)];
|
||||
aom_cdf_prob inter_ext_tx_cdf[EXT_TX_SIZES][CDF_SIZE(TX_TYPES)];
|
||||
#endif // CONFIG_EXT_TX
|
||||
#if CONFIG_LGT_FROM_PRED
|
||||
aom_prob intra_lgt_prob[LGT_SIZES][INTRA_MODES];
|
||||
aom_prob inter_lgt_prob[LGT_SIZES];
|
||||
#endif // CONFIG_LGT_FROM_PRED
|
||||
#if CONFIG_EXT_INTRA && CONFIG_INTRA_INTERP
|
||||
aom_cdf_prob intra_filter_cdf[INTRA_FILTERS + 1][CDF_SIZE(INTRA_FILTERS)];
|
||||
#endif // CONFIG_EXT_INTRA && CONFIG_INTRA_INTERP
|
||||
#if CONFIG_DELTA_Q
|
||||
aom_prob delta_q_prob[DELTA_Q_PROBS];
|
||||
#if CONFIG_EXT_DELTA_Q
|
||||
#if CONFIG_LOOPFILTER_LEVEL
|
||||
aom_prob delta_lf_multi_prob[FRAME_LF_COUNT][DELTA_LF_PROBS];
|
||||
#endif // CONFIG_LOOPFILTER_LEVEL
|
||||
aom_prob delta_lf_prob[DELTA_LF_PROBS];
|
||||
#endif
|
||||
#endif
|
||||
#if CONFIG_PVQ
|
||||
// TODO(any): If PVQ is enabled, most of coefficient related cdf,
|
||||
// such as coef_cdfs[], coef_tail_cdfs[], and coef_heaf_cdfs[] can be removed.
|
||||
od_adapt_ctx pvq_context;
|
||||
#endif // CONFIG_PVQ
|
||||
#if CONFIG_CFL
|
||||
aom_cdf_prob cfl_alpha_cdf[CDF_SIZE(CFL_ALPHABET_SIZE)];
|
||||
aom_cdf_prob cfl_sign_cdf[CDF_SIZE(CFL_JOINT_SIGNS)];
|
||||
aom_cdf_prob cfl_alpha_cdf[CFL_ALPHA_CONTEXTS][CDF_SIZE(CFL_ALPHABET_SIZE)];
|
||||
#endif
|
||||
#if CONFIG_LPF_SB
|
||||
aom_cdf_prob lpf_reuse_cdf[LPF_REUSE_CONTEXT][CDF_SIZE(2)];
|
||||
aom_cdf_prob lpf_delta_cdf[LPF_DELTA_CONTEXT][CDF_SIZE(DELTA_RANGE)];
|
||||
aom_cdf_prob lpf_sign_cdf[LPF_REUSE_CONTEXT][LPF_SIGN_CONTEXT][CDF_SIZE(2)];
|
||||
#endif // CONFIG_LPF_SB
|
||||
} FRAME_CONTEXT;
|
||||
|
||||
typedef struct FRAME_COUNTS {
|
||||
|
|
@ -383,9 +429,6 @@ typedef struct FRAME_COUNTS {
|
|||
#else
|
||||
unsigned int partition[PARTITION_CONTEXTS][PARTITION_TYPES];
|
||||
#endif
|
||||
av1_coeff_count_model coef[TX_SIZES][PLANE_TYPES];
|
||||
unsigned int eob_branch[TX_SIZES][PLANE_TYPES][REF_TYPES][COEF_BANDS]
|
||||
[COEFF_CONTEXTS];
|
||||
unsigned int switchable_interp[SWITCHABLE_FILTER_CONTEXTS]
|
||||
[SWITCHABLE_FILTERS];
|
||||
#if CONFIG_ADAPT_SCAN
|
||||
|
|
@ -415,16 +458,26 @@ typedef struct FRAME_COUNTS {
|
|||
unsigned int coeff_base[TX_SIZES][PLANE_TYPES][NUM_BASE_LEVELS]
|
||||
[COEFF_BASE_CONTEXTS][2];
|
||||
unsigned int coeff_lps[TX_SIZES][PLANE_TYPES][LEVEL_CONTEXTS][2];
|
||||
unsigned int coeff_br[TX_SIZES][PLANE_TYPES][BASE_RANGE_SETS][LEVEL_CONTEXTS]
|
||||
[2];
|
||||
#if CONFIG_CTX1D
|
||||
unsigned int eob_mode[TX_SIZES][PLANE_TYPES][TX_CLASSES][2];
|
||||
unsigned int empty_line[TX_SIZES][PLANE_TYPES][TX_CLASSES]
|
||||
[EMPTY_LINE_CONTEXTS][2];
|
||||
unsigned int hv_eob[TX_SIZES][PLANE_TYPES][TX_CLASSES][HV_EOB_CONTEXTS][2];
|
||||
#endif // CONFIG_CTX1D
|
||||
#endif // CONFIG_LV_MAP
|
||||
|
||||
av1_blockz_count_model blockz_count[TX_SIZES][PLANE_TYPES];
|
||||
#if CONFIG_SYMBOLRATE
|
||||
unsigned int coeff_num[2]; // 0: zero coeff 1: non-zero coeff
|
||||
unsigned int symbol_num[2]; // 0: entropy symbol 1: non-entropy symbol
|
||||
#endif
|
||||
|
||||
unsigned int newmv_mode[NEWMV_MODE_CONTEXTS][2];
|
||||
unsigned int zeromv_mode[ZEROMV_MODE_CONTEXTS][2];
|
||||
unsigned int refmv_mode[REFMV_MODE_CONTEXTS][2];
|
||||
unsigned int drl_mode[DRL_MODE_CONTEXTS][2];
|
||||
|
||||
#if CONFIG_EXT_INTER
|
||||
unsigned int inter_compound_mode[INTER_MODE_CONTEXTS][INTER_COMPOUND_MODES];
|
||||
#if CONFIG_COMPOUND_SINGLEREF
|
||||
unsigned int inter_singleref_comp_mode[INTER_MODE_CONTEXTS]
|
||||
|
|
@ -436,13 +489,15 @@ typedef struct FRAME_COUNTS {
|
|||
unsigned int wedge_interintra[BLOCK_SIZES_ALL][2];
|
||||
#endif // CONFIG_INTERINTRA
|
||||
unsigned int compound_interinter[BLOCK_SIZES_ALL][COMPOUND_TYPES];
|
||||
#endif // CONFIG_EXT_INTER
|
||||
#if CONFIG_MOTION_VAR || CONFIG_WARPED_MOTION
|
||||
unsigned int motion_mode[BLOCK_SIZES_ALL][MOTION_MODES];
|
||||
#if CONFIG_NCOBMC_ADAPT_WEIGHT && CONFIG_MOTION_VAR
|
||||
unsigned int ncobmc_mode[ADAPT_OVERLAP_BLOCKS][MAX_NCOBMC_MODES];
|
||||
#endif
|
||||
#if CONFIG_MOTION_VAR && CONFIG_WARPED_MOTION
|
||||
#if CONFIG_NCOBMC_ADAPT_WEIGHT
|
||||
unsigned int ncobmc[BLOCK_SIZES_ALL][OBMC_FAMILY_MODES];
|
||||
#endif // CONFIG_NCOBMC_ADAPT_WEIGHT
|
||||
unsigned int obmc[BLOCK_SIZES_ALL][2];
|
||||
#endif // CONFIG_MOTION_VAR && CONFIG_WARPED_MOTION
|
||||
#endif // CONFIG_MOTION_VAR || CONFIG_WARPED_MOTION
|
||||
|
|
@ -459,13 +514,11 @@ typedef struct FRAME_COUNTS {
|
|||
#else
|
||||
unsigned int comp_ref[REF_CONTEXTS][COMP_REFS - 1][2];
|
||||
#endif // CONFIG_EXT_REFS
|
||||
#if CONFIG_EXT_INTER && CONFIG_COMPOUND_SINGLEREF
|
||||
#if CONFIG_COMPOUND_SINGLEREF
|
||||
unsigned int comp_inter_mode[COMP_INTER_MODE_CONTEXTS][2];
|
||||
#endif // CONFIG_EXT_INTER && CONFIG_COMPOUND_SINGLEREF
|
||||
// TODO(any): tx_size_totals is only used by the encoder to decide whether
|
||||
// to use forward updates for the coeff probs, and as such it does not really
|
||||
// belong into this structure.
|
||||
unsigned int tx_size_totals[TX_SIZES];
|
||||
#endif // CONFIG_COMPOUND_SINGLEREF
|
||||
// TODO(urvang): Only needed for !CONFIG_VAR_TX case. So can be removed when
|
||||
// CONFIG_VAR_TX flag is removed.
|
||||
unsigned int tx_size[MAX_TX_DEPTH][TX_SIZE_CONTEXTS][MAX_TX_DEPTH + 1];
|
||||
#if CONFIG_RECT_TX_EXT && (CONFIG_EXT_TX || CONFIG_VAR_TX)
|
||||
unsigned int quarter_tx_size[2];
|
||||
|
|
@ -479,16 +532,22 @@ typedef struct FRAME_COUNTS {
|
|||
unsigned int intrabc[2];
|
||||
nmv_context_counts dv;
|
||||
#endif
|
||||
#if CONFIG_DELTA_Q
|
||||
#if CONFIG_LGT_FROM_PRED
|
||||
unsigned int intra_lgt[LGT_SIZES][INTRA_MODES][2];
|
||||
unsigned int inter_lgt[LGT_SIZES][2];
|
||||
#endif // CONFIG_LGT_FROM_PRED
|
||||
unsigned int delta_q[DELTA_Q_PROBS][2];
|
||||
#if CONFIG_EXT_DELTA_Q
|
||||
#if CONFIG_LOOPFILTER_LEVEL
|
||||
unsigned int delta_lf_multi[FRAME_LF_COUNT][DELTA_LF_PROBS][2];
|
||||
#endif // CONFIG_LOOPFILTER_LEVEL
|
||||
unsigned int delta_lf[DELTA_LF_PROBS][2];
|
||||
#endif
|
||||
#endif
|
||||
#if CONFIG_EXT_TX
|
||||
#if CONFIG_RECT_TX
|
||||
#if CONFIG_EXT_TX && CONFIG_RECT_TX
|
||||
unsigned int tx_size_implied[TX_SIZES][TX_SIZES];
|
||||
#endif // CONFIG_RECT_TX
|
||||
#endif // CONFIG_EXT_TX && CONFIG_RECT_TX
|
||||
#if CONFIG_ENTROPY_STATS
|
||||
#if CONFIG_EXT_TX
|
||||
unsigned int inter_ext_tx[EXT_TX_SETS_INTER][EXT_TX_SIZES][TX_TYPES];
|
||||
unsigned int intra_ext_tx[EXT_TX_SETS_INTRA][EXT_TX_SIZES][INTRA_MODES]
|
||||
[TX_TYPES];
|
||||
|
|
@ -496,6 +555,7 @@ typedef struct FRAME_COUNTS {
|
|||
unsigned int intra_ext_tx[EXT_TX_SIZES][TX_TYPES][TX_TYPES];
|
||||
unsigned int inter_ext_tx[EXT_TX_SIZES][TX_TYPES];
|
||||
#endif // CONFIG_EXT_TX
|
||||
#endif // CONFIG_ENTROPY_STATS
|
||||
#if CONFIG_SUPERTX
|
||||
unsigned int supertx[PARTITION_SUPERTX_CONTEXTS][TX_SIZES][2];
|
||||
unsigned int supertx_size[TX_SIZES];
|
||||
|
|
@ -509,29 +569,103 @@ typedef struct FRAME_COUNTS {
|
|||
#if CONFIG_FILTER_INTRA
|
||||
unsigned int filter_intra[PLANE_TYPES][2];
|
||||
#endif // CONFIG_FILTER_INTRA
|
||||
#if CONFIG_LPF_SB
|
||||
unsigned int lpf_reuse[LPF_REUSE_CONTEXT][2];
|
||||
unsigned int lpf_delta[LPF_DELTA_CONTEXT][DELTA_RANGE];
|
||||
unsigned int lpf_sign[LPF_SIGN_CONTEXT][2];
|
||||
#endif // CONFIG_LPF_SB
|
||||
} FRAME_COUNTS;
|
||||
|
||||
// CDF version of 'av1_kf_y_mode_prob'.
|
||||
extern const aom_cdf_prob av1_kf_y_mode_cdf[INTRA_MODES][INTRA_MODES]
|
||||
[CDF_SIZE(INTRA_MODES)];
|
||||
#if CONFIG_KF_CTX
|
||||
extern const aom_cdf_prob default_kf_y_mode_cdf[KF_MODE_CONTEXTS]
|
||||
[KF_MODE_CONTEXTS]
|
||||
[CDF_SIZE(INTRA_MODES)];
|
||||
#else
|
||||
extern const aom_cdf_prob default_kf_y_mode_cdf[INTRA_MODES][INTRA_MODES]
|
||||
[CDF_SIZE(INTRA_MODES)];
|
||||
#endif
|
||||
|
||||
#if CONFIG_PALETTE
|
||||
extern const aom_prob av1_default_palette_y_mode_prob[PALETTE_BLOCK_SIZES]
|
||||
[PALETTE_Y_MODE_CONTEXTS];
|
||||
extern const aom_prob
|
||||
av1_default_palette_uv_mode_prob[PALETTE_UV_MODE_CONTEXTS];
|
||||
#endif // CONFIG_PALETTE
|
||||
|
||||
extern const int av1_intra_mode_ind[INTRA_MODES];
|
||||
extern const int av1_intra_mode_inv[INTRA_MODES];
|
||||
#if CONFIG_EXT_TX
|
||||
extern int av1_ext_tx_intra_ind[EXT_TX_SETS_INTRA][TX_TYPES];
|
||||
extern int av1_ext_tx_intra_inv[EXT_TX_SETS_INTRA][TX_TYPES];
|
||||
extern int av1_ext_tx_inter_ind[EXT_TX_SETS_INTER][TX_TYPES];
|
||||
extern int av1_ext_tx_inter_inv[EXT_TX_SETS_INTER][TX_TYPES];
|
||||
#endif
|
||||
static const int av1_ext_tx_ind[EXT_TX_SET_TYPES][TX_TYPES] = {
|
||||
{
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
||||
},
|
||||
{
|
||||
1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
||||
},
|
||||
#if CONFIG_MRC_TX
|
||||
{
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,
|
||||
},
|
||||
{
|
||||
1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2,
|
||||
},
|
||||
#endif // CONFIG_MRC_TX
|
||||
{
|
||||
1, 3, 4, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
||||
},
|
||||
{
|
||||
1, 5, 6, 4, 0, 0, 0, 0, 0, 0, 2, 3, 0, 0, 0, 0,
|
||||
},
|
||||
{
|
||||
3, 4, 5, 8, 6, 7, 9, 10, 11, 0, 1, 2, 0, 0, 0, 0,
|
||||
},
|
||||
{
|
||||
7, 8, 9, 12, 10, 11, 13, 14, 15, 0, 1, 2, 3, 4, 5, 6,
|
||||
},
|
||||
};
|
||||
|
||||
static const int av1_ext_tx_inv[EXT_TX_SET_TYPES][TX_TYPES] = {
|
||||
{
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
||||
},
|
||||
{
|
||||
9, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
||||
},
|
||||
#if CONFIG_MRC_TX
|
||||
{
|
||||
0, 16, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
||||
},
|
||||
{
|
||||
9, 0, 16, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
||||
},
|
||||
#endif // CONFIG_MRC_TX
|
||||
{
|
||||
9, 0, 3, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
||||
},
|
||||
{
|
||||
9, 0, 10, 11, 3, 1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
||||
},
|
||||
{
|
||||
9, 10, 11, 0, 1, 2, 4, 5, 3, 6, 7, 8, 0, 0, 0, 0,
|
||||
},
|
||||
{
|
||||
9, 10, 11, 12, 13, 14, 15, 0, 1, 2, 4, 5, 3, 6, 7, 8,
|
||||
},
|
||||
};
|
||||
#else
|
||||
#if CONFIG_MRC_TX
|
||||
static const int av1_ext_tx_ind[TX_TYPES] = {
|
||||
0, 3, 4, 2, 1,
|
||||
};
|
||||
static const int av1_ext_tx_inv[TX_TYPES] = {
|
||||
0, 4, 3, 1, 2,
|
||||
};
|
||||
#else
|
||||
static const int av1_ext_tx_ind[TX_TYPES] = {
|
||||
0, 2, 3, 1,
|
||||
};
|
||||
static const int av1_ext_tx_inv[TX_TYPES] = {
|
||||
0, 3, 1, 2,
|
||||
};
|
||||
#endif // CONFIG_MRC_TX
|
||||
#endif // CONFIG_EXT_TX
|
||||
|
||||
#if CONFIG_EXT_INTER
|
||||
#if CONFIG_INTERINTRA
|
||||
extern const aom_tree_index
|
||||
av1_interintra_mode_tree[TREE_SIZE(INTERINTRA_MODES)];
|
||||
|
|
@ -543,36 +677,31 @@ extern const aom_tree_index
|
|||
av1_inter_singleref_comp_mode_tree[TREE_SIZE(INTER_SINGLEREF_COMP_MODES)];
|
||||
#endif // CONFIG_COMPOUND_SINGLEREF
|
||||
extern const aom_tree_index av1_compound_type_tree[TREE_SIZE(COMPOUND_TYPES)];
|
||||
#endif // CONFIG_EXT_INTER
|
||||
extern const aom_tree_index av1_partition_tree[TREE_SIZE(PARTITION_TYPES)];
|
||||
#if CONFIG_EXT_PARTITION_TYPES
|
||||
extern const aom_tree_index
|
||||
av1_ext_partition_tree[TREE_SIZE(EXT_PARTITION_TYPES)];
|
||||
#endif
|
||||
extern const aom_tree_index
|
||||
av1_switchable_interp_tree[TREE_SIZE(SWITCHABLE_FILTERS)];
|
||||
#if CONFIG_PALETTE
|
||||
extern const aom_tree_index
|
||||
av1_palette_color_index_tree[PALETTE_SIZES][TREE_SIZE(PALETTE_COLORS)];
|
||||
#endif // CONFIG_PALETTE
|
||||
extern const aom_tree_index av1_tx_size_tree[MAX_TX_DEPTH][TREE_SIZE(TX_SIZES)];
|
||||
#if CONFIG_EXT_INTRA && CONFIG_INTRA_INTERP
|
||||
extern const aom_tree_index av1_intra_filter_tree[TREE_SIZE(INTRA_FILTERS)];
|
||||
#endif // CONFIG_EXT_INTRA && CONFIG_INTRA_INTERP
|
||||
#if CONFIG_EXT_TX
|
||||
extern const aom_tree_index av1_ext_tx_inter_tree[EXT_TX_SETS_INTER]
|
||||
[TREE_SIZE(TX_TYPES)];
|
||||
extern const aom_tree_index av1_ext_tx_intra_tree[EXT_TX_SETS_INTRA]
|
||||
[TREE_SIZE(TX_TYPES)];
|
||||
extern const aom_tree_index av1_ext_tx_tree[EXT_TX_SET_TYPES]
|
||||
[TREE_SIZE(TX_TYPES)];
|
||||
#else
|
||||
extern const aom_tree_index av1_ext_tx_tree[TREE_SIZE(TX_TYPES)];
|
||||
#endif // CONFIG_EXT_TX
|
||||
#if CONFIG_MOTION_VAR || CONFIG_WARPED_MOTION
|
||||
extern const aom_tree_index av1_motion_mode_tree[TREE_SIZE(MOTION_MODES)];
|
||||
#endif // CONFIG_MOTION_VAR || CONFIG_WARPED_MOTION
|
||||
#if CONFIG_NCOBMC_ADAPT_WEIGHT && CONFIG_MOTION_VAR
|
||||
#if CONFIG_NCOBMC_ADAPT_WEIGHT
|
||||
extern const aom_tree_index av1_ncobmc_mode_tree[TREE_SIZE(MAX_NCOBMC_MODES)];
|
||||
#endif
|
||||
#if CONFIG_WARPED_MOTION
|
||||
extern const aom_tree_index av1_ncobmc_tree[TREE_SIZE(OBMC_FAMILY_MODES)];
|
||||
#endif // CONFIG_WARPED_MOTION
|
||||
#endif // CONFIG_NCOBMC_ADAPT_WEIGHT
|
||||
#if CONFIG_LOOP_RESTORATION
|
||||
#define RESTORE_NONE_SGRPROJ_PROB 64
|
||||
#define RESTORE_NONE_BILATERAL_PROB 16
|
||||
|
|
@ -581,17 +710,11 @@ extern const aom_tree_index av1_ncobmc_mode_tree[TREE_SIZE(MAX_NCOBMC_MODES)];
|
|||
extern const aom_tree_index
|
||||
av1_switchable_restore_tree[TREE_SIZE(RESTORE_SWITCHABLE_TYPES)];
|
||||
#endif // CONFIG_LOOP_RESTORATION
|
||||
extern int av1_switchable_interp_ind[SWITCHABLE_FILTERS];
|
||||
extern int av1_switchable_interp_inv[SWITCHABLE_FILTERS];
|
||||
|
||||
void av1_setup_past_independence(struct AV1Common *cm);
|
||||
|
||||
void av1_adapt_intra_frame_probs(struct AV1Common *cm);
|
||||
void av1_adapt_inter_frame_probs(struct AV1Common *cm);
|
||||
#if !CONFIG_EXT_TX
|
||||
extern int av1_ext_tx_ind[TX_TYPES];
|
||||
extern int av1_ext_tx_inv[TX_TYPES];
|
||||
#endif
|
||||
|
||||
static INLINE int av1_ceil_log2(int n) {
|
||||
int i = 1, p = 2;
|
||||
|
|
@ -602,14 +725,12 @@ static INLINE int av1_ceil_log2(int n) {
|
|||
return i;
|
||||
}
|
||||
|
||||
#if CONFIG_PALETTE
|
||||
// Returns the context for palette color index at row 'r' and column 'c',
|
||||
// along with the 'color_order' of neighbors and the 'color_idx'.
|
||||
// The 'color_map' is a 2D array with the given 'stride'.
|
||||
int av1_get_palette_color_index_context(const uint8_t *color_map, int stride,
|
||||
int r, int c, int palette_size,
|
||||
uint8_t *color_order, int *color_idx);
|
||||
#endif // CONFIG_PALETTE
|
||||
|
||||
#ifdef __cplusplus
|
||||
} // extern "C"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue