You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
495 lines
16 KiB
495 lines
16 KiB
/******************************************************************************
|
|
*
|
|
* Copyright (C) 2018 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at:
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*
|
|
*****************************************************************************
|
|
* Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
|
|
*/
|
|
/*!
|
|
******************************************************************************
|
|
* \file ihevce_ipe_structs.h
|
|
*
|
|
* \brief
|
|
* This file contains strcutures of ipe pass
|
|
*
|
|
* \date
|
|
* 18/09/2012
|
|
*
|
|
* \author
|
|
* Ittiam
|
|
*
|
|
******************************************************************************
|
|
*/
|
|
|
|
#ifndef _IHEVCE_IPE_STRUCTS_H_
|
|
#define _IHEVCE_IPE_STRUCTS_H_
|
|
|
|
/*****************************************************************************/
|
|
/* Constant Macros */
|
|
/*****************************************************************************/
|
|
#define MAX_FAST_IP_MODES 23
|
|
#define NUM_INTRA_RDOPT_MODES 1
|
|
#if 1 // FAST_PART_WITH_OPTION_4
|
|
#define MAX_TREE_NODES \
|
|
((MAX_CTB_SIZE == MIN_TU_SIZE) \
|
|
? 1 \
|
|
: (MAX_CTB_SIZE == (MIN_TU_SIZE << 1) \
|
|
? 5 \
|
|
: (MAX_CTB_SIZE == (MIN_TU_SIZE << 2) \
|
|
? 21 \
|
|
: (MAX_CTB_SIZE == (MIN_TU_SIZE << 3) ? 37 : 53))))
|
|
#else // FAST_PART_WITH_OPTION_4
|
|
#define MAX_TREE_NODES \
|
|
((MAX_CTB_SIZE == MIN_TU_SIZE) \
|
|
? 1 \
|
|
: (MAX_CTB_SIZE == (MIN_TU_SIZE << 1) \
|
|
? 5 \
|
|
: (MAX_CTB_SIZE == (MIN_TU_SIZE << 2) \
|
|
? 9 \
|
|
: (MAX_CTB_SIZE == (MIN_TU_SIZE << 3) ? 13 : 17))))
|
|
#endif // FAST_PART_WITH_OPTION_4
|
|
#define BOTTOM_LEFT_FLAG 0x0000000F
|
|
#define LEFT_FLAG 0x000000F0
|
|
#define TOP_LEFT_FLAG 0x00010000
|
|
#define TOP_FLAG 0x00000F00
|
|
#define TOP_RIGHT_FLAG 0x0000F000
|
|
#define MAX_UWORD8 0xFF
|
|
#define MAX_DOUBLE 1.7e+308 ///< max. value of double-type value
|
|
#define MAX_INTRA_COST_IPE 0x0F7F7F7F
|
|
|
|
#define MAX_TU_ROW_IN_CTB (MAX_CTB_SIZE >> 2)
|
|
#define MAX_TU_COL_IN_CTB (MAX_CTB_SIZE >> 2)
|
|
|
|
#define BIT_DEPTH 8
|
|
|
|
#define FAST_PARTITION_WITH_TRANSFORM 1
|
|
|
|
#define IHEVCE_INTRA_REF_FILTERING C
|
|
#define IHEVCE_INTRA_LUMA_REF_SUBSTITUTION C
|
|
/*****************************************************************************/
|
|
/* Constant Macros */
|
|
/*****************************************************************************/
|
|
/** /breif 4x4 DST, 4x4, 8x8, 16x16, 32x32 */
|
|
#define NUM_TRANS_TYPES 5
|
|
#define INTRA_PLANAR 0
|
|
#define INTRA_DC 1
|
|
|
|
/*****************************************************************************/
|
|
/* Function Macros */
|
|
/*****************************************************************************/
|
|
#define INTRA_ANGULAR(x) (x)
|
|
|
|
/** @breif max 30bit value */
|
|
#define MAX30 ((1 << 30) - 1)
|
|
|
|
/* @bried macro to clip a data to max of 30bits (assuming unsgined) */
|
|
#define CLIP30(x) ((x) > MAX30 ? MAX30 : (x))
|
|
|
|
/* @bried compute the (lambda * rate) with a qshift and clip result to 30bits */
|
|
#define COMPUTE_RATE_COST_CLIP30(r, l, qshift) ((WORD32)CLIP30((ULWORD64)((r) * (l)) >> (qshift)))
|
|
|
|
/*****************************************************************************/
|
|
/* Typedefs */
|
|
/*****************************************************************************/
|
|
typedef UWORD32 (*pf_res_trans_luma_had)(
|
|
UWORD8 *pu1_origin,
|
|
WORD32 src_strd,
|
|
UWORD8 *pu1_pred_buf,
|
|
WORD32 pred_strd,
|
|
WORD16 *pi2_dst,
|
|
WORD32 dst_strd,
|
|
WORD32 size);
|
|
|
|
typedef void (*pf_ipe_intra_pred)(
|
|
UWORD8 *pu1_ref, WORD32 src_strd, UWORD8 *pu1_dst, WORD32 dst_strd, WORD32 nt, WORD32 mode);
|
|
|
|
typedef UWORD32 (*pf_ipe_res_trans)(
|
|
UWORD8 *pu1_src,
|
|
UWORD8 *pu1_pred,
|
|
WORD16 *pi2_tmp,
|
|
WORD16 *pi2_dst,
|
|
WORD32 src_strd,
|
|
WORD32 pred_strd,
|
|
WORD32 dst_strd,
|
|
WORD32 chroma_flag);
|
|
|
|
typedef FT_CALC_HAD_SATD_8BIT *pf_ipe_res_trans_had;
|
|
/*****************************************************************************/
|
|
/* Enums */
|
|
/*****************************************************************************/
|
|
|
|
typedef enum
|
|
{
|
|
|
|
IPE_CTXT = 0,
|
|
IPE_THRDS_CTXT,
|
|
|
|
/* should be last entry */
|
|
NUM_IPE_MEM_RECS
|
|
|
|
} IPE_MEM_TABS_T;
|
|
|
|
typedef enum
|
|
{
|
|
IPE_FUNC_MODE_0 = 0,
|
|
IPE_FUNC_MODE_1,
|
|
IPE_FUNC_MODE_2,
|
|
IPE_FUNC_MODE_3TO9,
|
|
IPE_FUNC_MODE_10,
|
|
IPE_FUNC_MODE_11TO17,
|
|
IPE_FUNC_MODE_18_34,
|
|
IPE_FUNC_MODE_19TO25,
|
|
IPE_FUNC_MODE_26,
|
|
IPE_FUNC_MODE_27TO33,
|
|
|
|
NUM_IPE_FUNCS
|
|
|
|
} IPE_FUNCS_T;
|
|
|
|
/*****************************************************************************/
|
|
/* Structure */
|
|
/*****************************************************************************/
|
|
/**
|
|
******************************************************************************
|
|
* @brief IPE CTB to CU and TU Quadtree Recursive Structure
|
|
******************************************************************************
|
|
*/
|
|
|
|
typedef struct ihevce_ipe_cu_tree_t ihevce_ipe_cu_tree_t;
|
|
|
|
typedef struct ihevce_ipe_cu_tree_t
|
|
{
|
|
/**
|
|
* Origin of current coding unit relative to top-left of CTB
|
|
*/
|
|
UWORD16 u2_x0;
|
|
|
|
UWORD16 u2_y0;
|
|
|
|
/**
|
|
* Origin of current coding unit relative to top-left of Picture
|
|
*/
|
|
UWORD16 u2_orig_x;
|
|
|
|
UWORD16 u2_orig_y;
|
|
|
|
/**
|
|
* Size of current coding unit in luma pixels
|
|
*/
|
|
UWORD8 u1_cu_size;
|
|
|
|
UWORD8 u1_width;
|
|
|
|
UWORD8 u1_height;
|
|
|
|
UWORD8 u1_depth;
|
|
|
|
UWORD8 u1_part_flag_pos;
|
|
|
|
UWORD8 u1_log2_nt;
|
|
|
|
WORD32 i4_nbr_flag;
|
|
|
|
/**
|
|
* Recursive Bracketing Parameters
|
|
*/
|
|
UWORD8 best_mode;
|
|
|
|
WORD32 best_satd;
|
|
|
|
WORD32 best_cost;
|
|
|
|
/**
|
|
* Number of pixels available in these neighbors
|
|
*/
|
|
UWORD8 u1_num_left_avail;
|
|
|
|
UWORD8 u1_num_top_avail;
|
|
|
|
UWORD8 u1_num_top_right_avail;
|
|
|
|
UWORD8 u1_num_bottom_left_avail;
|
|
|
|
UWORD8 au1_best_mode_1tu[NUM_BEST_MODES];
|
|
|
|
WORD32 au4_best_cost_1tu[NUM_BEST_MODES];
|
|
|
|
UWORD8 au1_best_mode_4tu[NUM_BEST_MODES];
|
|
|
|
WORD32 au4_best_cost_4tu[NUM_BEST_MODES];
|
|
|
|
ihevce_ipe_cu_tree_t *ps_parent;
|
|
|
|
ihevce_ipe_cu_tree_t *ps_sub_cu[4];
|
|
|
|
/* best mode bits cost */
|
|
UWORD16 u2_mode_bits_cost;
|
|
|
|
} ihevce_ipe_cu_tree_node_t;
|
|
|
|
/**
|
|
******************************************************************************
|
|
* @brief IPE module context memory
|
|
******************************************************************************
|
|
*/
|
|
typedef struct
|
|
{
|
|
ihevce_ipe_cu_tree_t *ps_ipe_cu_tree;
|
|
|
|
/* one parent and four children */
|
|
ihevce_ipe_cu_tree_t as_ipe_cu_tree[5];
|
|
|
|
UWORD8 au1_ctb_mode_map[MAX_TU_ROW_IN_CTB + 1][MAX_TU_COL_IN_CTB + 1];
|
|
|
|
UWORD8 au1_cand_mode_list[3];
|
|
|
|
/** Pointer to structure containing function pointers of common*/
|
|
func_selector_t *ps_func_selector;
|
|
|
|
/**
|
|
* CU level Qp / 6
|
|
*/
|
|
WORD32 i4_cu_qp_div6;
|
|
|
|
/**
|
|
* CU level Qp % 6
|
|
*/
|
|
WORD32 i4_cu_qp_mod6;
|
|
|
|
/** array of luma intra prediction function pointers */
|
|
pf_ipe_intra_pred apf_ipe_lum_ip[NUM_IPE_FUNCS];
|
|
|
|
/** array of function pointers for residual and
|
|
* forward transform for all transform sizes
|
|
*/
|
|
pf_res_trans_luma apf_resd_trns[NUM_TRANS_TYPES];
|
|
|
|
/** array of function pointers for residual and
|
|
* forward transform for all transform sizes
|
|
*/
|
|
pf_res_trans_luma_had apf_resd_trns_had[NUM_TRANS_TYPES];
|
|
|
|
/** array of pointer to store the scaling matrices for
|
|
* all transform sizes and qp % 6 (pre computed)
|
|
*/
|
|
WORD16 *api2_scal_mat[NUM_TRANS_TYPES * 2];
|
|
|
|
/** array of pointer to store the re-scaling matrices for
|
|
* all transform sizes and qp % 6 (pre computed)
|
|
*/
|
|
WORD16 *api2_rescal_mat[NUM_TRANS_TYPES * 2];
|
|
|
|
/** Qunatization rounding factor for inter and intra CUs */
|
|
WORD32 i4_quant_rnd_factor[2];
|
|
|
|
UWORD8 u1_ctb_size;
|
|
|
|
UWORD8 u1_min_cu_size;
|
|
|
|
UWORD8 u1_min_tu_size;
|
|
|
|
UWORD16 u2_ctb_row_num;
|
|
|
|
UWORD16 u2_ctb_num_in_row;
|
|
|
|
WORD8 i1_QP;
|
|
|
|
UWORD8 u1_num_b_frames;
|
|
|
|
UWORD8 b_sad_type;
|
|
|
|
UWORD8 u1_ipe_step_size;
|
|
|
|
WORD32 i4_ol_satd_lambda;
|
|
|
|
WORD32 i4_ol_sad_lambda;
|
|
|
|
UWORD8 au1_nbr_ctb_map[MAX_PU_IN_CTB_ROW + 1 + 8][MAX_PU_IN_CTB_ROW + 1 + 8];
|
|
|
|
/**
|
|
* Pointer to (1,1) location in au1_nbr_ctb_map
|
|
*/
|
|
UWORD8 *pu1_ctb_nbr_map;
|
|
|
|
/**
|
|
* neigbour map buffer stride;
|
|
*/
|
|
WORD32 i4_nbr_map_strd;
|
|
|
|
/** CTB neighbour availability flags */
|
|
nbr_avail_flags_t s_ctb_nbr_avail_flags;
|
|
|
|
/** Slice Type of the current picture being processed */
|
|
WORD32 i4_slice_type;
|
|
|
|
/** Temporal ID of the current picture being processed */
|
|
WORD32 i4_temporal_lyr_id;
|
|
|
|
WORD32 i4_ol_sad_lambda_qf_array[MAX_HEVC_QP_10bit + 1];
|
|
WORD32 i4_ol_satd_lambda_qf_array[MAX_HEVC_QP_10bit + 1];
|
|
|
|
/************************************************************************/
|
|
/* The fields with the string 'type2' in their names are required */
|
|
/* when both 8bit and hbd lambdas are needed. The lambdas corresponding */
|
|
/* to the bit_depth != internal_bit_depth are stored in these fields */
|
|
/************************************************************************/
|
|
WORD32 i4_ol_sad_type2_lambda_qf_array[MAX_HEVC_QP_10bit + 1];
|
|
WORD32 i4_ol_satd_type2_lambda_qf_array[MAX_HEVC_QP_10bit + 1];
|
|
|
|
/*Store the HEVC frame level qp for level modulation*/
|
|
WORD32 i4_hevc_qp;
|
|
/*Store the frame level qscale for level modulation*/
|
|
WORD32 i4_qscale;
|
|
#if POW_OPT
|
|
/* Averge activity of 8x8 blocks from previous frame
|
|
* If L1, maps to 16*16 in L0
|
|
*/
|
|
long double ld_curr_frame_8x8_log_avg[2];
|
|
|
|
/* Averge activity of 16x16 blocks from previous frame
|
|
* If L1, maps to 32*32 in L0
|
|
*/
|
|
long double ld_curr_frame_16x16_log_avg[3];
|
|
|
|
/* Averge activity of 32x32 blocks from previous frame
|
|
* If L1, maps to 64*64 in L0
|
|
*/
|
|
long double ld_curr_frame_32x32_log_avg[3];
|
|
#else
|
|
/* Averge activity of 8x8 blocks from previous frame
|
|
* If L1, maps to 16*16 in L0
|
|
*/
|
|
LWORD64 i8_curr_frame_8x8_avg_act[2];
|
|
|
|
/* Averge activity of 16x16 blocks from previous frame
|
|
* If L1, maps to 32*32 in L0
|
|
*/
|
|
LWORD64 i8_curr_frame_16x16_avg_act[3];
|
|
|
|
/* Averge activity of 32x32 blocks from previous frame
|
|
* If L1, maps to 64*64 in L0
|
|
*/
|
|
LWORD64 i8_curr_frame_32x32_avg_act[3];
|
|
#endif
|
|
/** Frame-levelSATD cost accumalator */
|
|
LWORD64 i8_frame_acc_satd_cost;
|
|
|
|
/** Frame-levelSATD accumalator */
|
|
LWORD64 i8_frame_acc_satd;
|
|
|
|
/** Frame-level activity factor for CU 8x8 accumalator */
|
|
LWORD64 i8_frame_acc_act_factor;
|
|
|
|
/** Frame-level Mode Bits cost accumalator */
|
|
LWORD64 i8_frame_acc_mode_bits_cost;
|
|
|
|
/** Encoder quality preset : See IHEVCE_QUALITY_CONFIG_T for presets */
|
|
WORD32 i4_quality_preset;
|
|
|
|
/** Frame-level SATD/qp accumulator in q10 format*/
|
|
LWORD64 i8_frame_acc_satd_by_modqp_q10;
|
|
|
|
/** For testing EIID only. */
|
|
UWORD32 u4_num_16x16_skips_at_L0_IPE;
|
|
|
|
/** Reference sample array. Used as local variable in mode_eval_filtering */
|
|
UWORD8 au1_ref_samples[1028];
|
|
/** filtered reference sample array. Used as local variable in mode_eval_filtering */
|
|
UWORD8 au1_filt_ref_samples[1028];
|
|
/** array for the modes to be evaluated. Used as local variable in mode_eval_filtering */
|
|
UWORD8 au1_modes_to_eval[MAX_NUM_IP_MODES];
|
|
/** temp array for the modes to be evaluated. Used as local variable in mode_eval_filtering */
|
|
UWORD8 au1_modes_to_eval_temp[MAX_NUM_IP_MODES];
|
|
/** pred samples array. Used as local variable in mode_eval_filtering */
|
|
MEM_ALIGN32 UWORD8 au1_pred_samples[4096];
|
|
/** array for storing satd cost. Used as local variable in mode_eval_filtering*/
|
|
UWORD16 au2_mode_bits_satd_cost[MAX_NUM_IP_MODES];
|
|
/** array for storing satd values. used as local variable in mode_eval_filtering */
|
|
UWORD16 au2_mode_bits_satd[MAX_NUM_IP_MODES];
|
|
|
|
/** reference data, local for pu_calc_8x8 */
|
|
UWORD8 au1_ref_8x8pu[4][18];
|
|
/** mode_bits_cost, local for pu_calc_8x8 */
|
|
UWORD16 au2_mode_bits_cost_8x8pu[4][MAX_NUM_IP_MODES];
|
|
/** mode_bits, local for pu_calc_8x8 */
|
|
UWORD16 au2_mode_bits_8x8_pu[MAX_NUM_IP_MODES];
|
|
|
|
/** tranform coeff temp, local to ihevce_pu_calc_4x4_blk */
|
|
WORD16 *pi2_trans_tmp; //this memory is overlayed with au1_pred_samples[4096]. First half.
|
|
|
|
/** tranform coeff out, local to ihevce_pu_calc_4x4_blk */
|
|
WORD16 *pi2_trans_out; //this memory is overlayed with au1_pred_samples[4096]. Second half.
|
|
|
|
UWORD8 u1_use_lambda_derived_from_min_8x8_act_in_ctb;
|
|
|
|
UWORD8 u1_bit_depth;
|
|
|
|
rc_quant_t *ps_rc_quant_ctxt;
|
|
/** Flag that specifies whether to use SATD or SAD in L0 IPE */
|
|
UWORD8 u1_use_satd;
|
|
|
|
/** Flag that specifies level of refinement */
|
|
UWORD8 u1_level_1_refine_on;
|
|
|
|
/** Flag indicates that child mode decision is disabled in L0 IPE recur bracketing */
|
|
UWORD8 u1_disable_child_cu_decide;
|
|
|
|
/*Modulation factor*/
|
|
WORD32 ai4_mod_factor_derived_by_variance[2];
|
|
float f_strength;
|
|
WORD32 i4_l0ipe_qp_mod;
|
|
|
|
WORD32 i4_frm_qp;
|
|
WORD32 i4_temporal_layer;
|
|
WORD32 i4_pass;
|
|
|
|
double f_i_pic_lamda_modifier;
|
|
WORD32 i4_use_const_lamda_modifier;
|
|
WORD32 i4_is_ref_pic;
|
|
LWORD64 i8_curr_frame_avg_mean_act;
|
|
WORD32 i4_enable_noise_detection;
|
|
|
|
ihevce_ipe_optimised_function_list_t s_ipe_optimised_function_list;
|
|
|
|
ihevce_cmn_opt_func_t s_cmn_opt_func;
|
|
|
|
} ihevce_ipe_ctxt_t;
|
|
|
|
/**
|
|
******************************************************************************
|
|
* @brief IPE module overall context
|
|
******************************************************************************
|
|
*/
|
|
typedef struct
|
|
{
|
|
/*array of ipe ctxt */
|
|
ihevce_ipe_ctxt_t *aps_ipe_thrd_ctxt[MAX_NUM_FRM_PROC_THRDS_PRE_ENC];
|
|
|
|
/** Number of processing threads created run time */
|
|
WORD32 i4_num_proc_thrds;
|
|
|
|
} ihevce_ipe_master_ctxt_t;
|
|
|
|
/*****************************************************************************/
|
|
/* Extern Variable Declarations */
|
|
/*****************************************************************************/
|
|
|
|
/*****************************************************************************/
|
|
/* Extern Function Declarations */
|
|
/*****************************************************************************/
|
|
void ihevce_ipe_analyse_update_cost(
|
|
ihevce_ipe_cu_tree_t *ps_cu_node, UWORD8 u1_mode, DOUBLE lf_cost);
|
|
#endif /* _IHEVCE_IPE_STRUCTS_H_ */
|