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.
2818 lines
70 KiB
2818 lines
70 KiB
/****************************************************************************** |
|
* |
|
* Copyright (C) 2015 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 |
|
* ih264e_structs.h |
|
* |
|
* @brief |
|
* Structure definitions used in the encoder |
|
* |
|
* @author |
|
* Harish |
|
* |
|
* @remarks |
|
* None |
|
* |
|
******************************************************************************* |
|
*/ |
|
|
|
#ifndef IH264E_STRUCTS_H_ |
|
#define IH264E_STRUCTS_H_ |
|
|
|
/*****************************************************************************/ |
|
/* Structure definitions */ |
|
/*****************************************************************************/ |
|
|
|
/* Early declaration of structs */ |
|
typedef struct _codec_t codec_t; |
|
typedef struct _proc_t process_ctxt_t; |
|
|
|
|
|
/*****************************************************************************/ |
|
/* Extern Function type definitions */ |
|
/*****************************************************************************/ |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief intra prediction filters leaf level |
|
****************************************************************************** |
|
*/ |
|
typedef void (*pf_intra_pred)(UWORD8 *pu1_src, UWORD8 *pu1_dst, |
|
WORD32 src_strd, WORD32 dst_strd, |
|
WORD32 ui_neighboravailability); |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief inter prediction filters leaf level |
|
****************************************************************************** |
|
*/ |
|
|
|
typedef void (*pf_inter_pred_luma_bilinear)(UWORD8 *pu1_src1, UWORD8 *pu1_src2, UWORD8 *pu1_dst, |
|
WORD32 src_strd1, WORD32 src_strd2, WORD32 dst_strd, |
|
WORD32 height, WORD32 width); |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief fwd transform leaf level |
|
****************************************************************************** |
|
*/ |
|
typedef void (*pf_trans_quant)(UWORD8*pu1_src, UWORD8 *pu1_pred, WORD16 *pi2_out, |
|
WORD32 i4_src_stride, UWORD32 u4_pred_stride, UWORD32 u4_dst_stride, |
|
const UWORD16 *pu2_scale_mat, const UWORD16 *pu2_thresh_mat, |
|
UWORD32 u4_qbit, UWORD32 u4_round_fact, UWORD8 *pu1_nnz); |
|
|
|
typedef void (*pf_iquant_itrans)(WORD16 *pi2_src, UWORD8 *pu1_pred, UWORD8 *pu1_out, |
|
WORD32 i4_src_stride, UWORD32 u4_pred_stride, UWORD32 u4_out_stride, |
|
const UWORD16 *pu2_iscale_mat, const UWORD16 *pu2_weigh_mat, |
|
UWORD32 qp_div, WORD32 *pi4_tmp); |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief Padding leaf level |
|
****************************************************************************** |
|
*/ |
|
typedef void (*pf_pad)(UWORD8 *pu1_src, WORD32 src_strd, WORD32 wd, WORD32 pad_size); |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief memory handling leaf level |
|
****************************************************************************** |
|
*/ |
|
typedef void (*pf_memcpy)(UWORD8 *pu1_dst, UWORD8 *pu1_src, UWORD32 num_bytes); |
|
|
|
typedef void (*pf_memset)(UWORD8 *pu1_dst, UWORD8 value, UWORD32 num_bytes); |
|
|
|
typedef void (*pf_memcpy_mul8)(UWORD8 *pu1_dst, UWORD8 *pu1_src, UWORD32 num_bytes); |
|
|
|
typedef void (*pf_memset_mul8)(UWORD8 *pu1_dst, UWORD8 value, UWORD32 num_bytes); |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief Sad computation |
|
****************************************************************************** |
|
*/ |
|
typedef void (*pf_compute_sad)(UWORD8 *pu1_src, UWORD8 *pu1_est, |
|
UWORD32 src_strd, UWORD32 est_strd, |
|
WORD32 i4_max_sad, WORD32 *pi4_mb_distortion); |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief Intra mode eval:encoder level |
|
****************************************************************************** |
|
*/ |
|
typedef void (*pf_evaluate_intra_modes)(UWORD8 *pu1_src, UWORD8 *pu1_ngbr_pels_i16, UWORD8 *pu1_dst, |
|
UWORD32 src_strd, UWORD32 dst_strd, |
|
WORD32 u4_n_avblty, UWORD32 *u4_intra_mode, |
|
WORD32 *pu4_sadmin, |
|
UWORD32 u4_valid_intra_modes); |
|
|
|
typedef void (*pf_evaluate_intra_4x4_modes)(UWORD8 *pu1_src, UWORD8 *pu1_ngbr_pels, UWORD8 *pu1_dst, |
|
UWORD32 src_strd, UWORD32 dst_strd, |
|
WORD32 u4_n_avblty, UWORD32 *u4_intra_mode, |
|
WORD32 *pu4_sadmin, |
|
UWORD32 u4_valid_intra_modes, UWORD32 u4_lambda, |
|
UWORD32 u4_predictd_mode); |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief half_pel generation :encoder level |
|
****************************************************************************** |
|
*/ |
|
typedef void (*pf_sixtapfilter_horz)(UWORD8 *pu1_src, UWORD8 *pu1_dst, |
|
WORD32 src_strd, WORD32 dst_strd); |
|
|
|
typedef void (*pf_sixtap_filter_2dvh_vert)(UWORD8 *pu1_src, UWORD8 *pu1_dst1, UWORD8 *pu1_dst2, |
|
WORD32 src_strd, WORD32 dst_strd, |
|
WORD32 *pi16_pred1, |
|
WORD32 pi16_pred1_strd); |
|
/** |
|
****************************************************************************** |
|
* @brief color space conversion |
|
****************************************************************************** |
|
*/ |
|
typedef void (*pf_fmt_conv_420p_to_420sp)(UWORD8 *pu1_y_src, UWORD8 *pu1_u_src, UWORD8 *pu1_v_src, |
|
UWORD8 *pu1_y_dst, UWORD8 *pu1_uv_dst, |
|
UWORD16 u2_height, UWORD16 u2_width, |
|
UWORD16 src_y_strd, UWORD16 src_u_strd, UWORD16 src_v_strd, |
|
UWORD16 dst_y_strd, UWORD16 dst_uv_strd, |
|
UWORD32 convert_uv_only); |
|
|
|
typedef void (*pf_fmt_conv_422ile_to_420sp)(UWORD8 *pu1_y_buf, UWORD8 *pu1_u_buf, UWORD8 *pu1_v_buf, |
|
UWORD8 *pu1_422i_buf, |
|
WORD32 u4_y_width, WORD32 u4_y_height, WORD32 u4_y_stride, |
|
WORD32 u4_u_stride, WORD32 u4_v_stride, |
|
WORD32 u4_422i_stride); |
|
|
|
|
|
|
|
/** |
|
****************************************************************************** |
|
* @brief ME evaluation |
|
****************************************************************************** |
|
*/ |
|
typedef void ih264e_compute_me_ft(process_ctxt_t *); |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief SKIP decision |
|
****************************************************************************** |
|
*/ |
|
typedef WORD32 ih264e_skip_params_ft(process_ctxt_t *, WORD32); |
|
|
|
|
|
/*****************************************************************************/ |
|
/* Enums */ |
|
/*****************************************************************************/ |
|
|
|
/** |
|
****************************************************************************** |
|
* @enum CODEC_STATE_T |
|
* @brief codec state |
|
****************************************************************************** |
|
*/ |
|
typedef enum |
|
{ |
|
INIT_DONE, |
|
HEADER_DONE, |
|
FIRST_FRAME_DONE, |
|
} CODEC_STATE_T; |
|
|
|
|
|
/** |
|
****************************************************************************** |
|
* @enum JOBQ_CMD_T |
|
* @brief list of job commands (used during job instantiation) |
|
****************************************************************************** |
|
*/ |
|
typedef enum |
|
{ |
|
CMD_PROCESS, |
|
CMD_ENTROPY, |
|
CMD_FMTCONV, |
|
CMD_ME, |
|
}JOBQ_CMD_T; |
|
|
|
|
|
/*****************************************************************************/ |
|
/* Structures */ |
|
/*****************************************************************************/ |
|
|
|
/** |
|
* PU information |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* Motion Vector |
|
*/ |
|
mv_t s_mv; |
|
|
|
/** |
|
* Ref index |
|
*/ |
|
WORD8 i1_ref_idx; |
|
|
|
} enc_pu_mv_t; |
|
|
|
|
|
/* |
|
* Total Pu info for an MB |
|
*/ |
|
typedef struct |
|
{ |
|
|
|
/* Array with ME info for all lists */ |
|
enc_pu_mv_t s_me_info[2]; |
|
|
|
/** |
|
* PU X position in terms of min PU (4x4) units |
|
*/ |
|
UWORD32 b4_pos_x : 4; |
|
|
|
/** |
|
* PU Y position in terms of min PU (4x4) units |
|
*/ |
|
UWORD32 b4_pos_y : 4; |
|
|
|
/** |
|
* PU width in pixels = (b4_wd + 1) << 2 |
|
*/ |
|
UWORD32 b4_wd : 2; |
|
|
|
/** |
|
* PU height in pixels = (b4_ht + 1) << 2 |
|
*/ |
|
UWORD32 b4_ht : 2; |
|
|
|
/** |
|
* Intra or Inter flag for each partition - 0 or 1 |
|
*/ |
|
UWORD32 b1_intra_flag : 1; |
|
|
|
/** |
|
* PRED_L0, PRED_L1, PRED_BI |
|
*/ |
|
UWORD32 b2_pred_mode : 2; |
|
|
|
|
|
} enc_pu_t; |
|
|
|
|
|
typedef struct |
|
{ |
|
/** Descriptor of raw buffer */ |
|
iv_raw_buf_t s_raw_buf; |
|
|
|
/** Lower 32bits of time stamp corresponding to the above buffer */ |
|
UWORD32 u4_timestamp_low; |
|
|
|
/** Upper 32bits of time stamp corresponding to the above buffer */ |
|
UWORD32 u4_timestamp_high; |
|
|
|
/** Flag to indicate if the current buffer is last buffer */ |
|
UWORD32 u4_is_last; |
|
|
|
/** Flag to indicate if mb info is sent along with input buffer */ |
|
UWORD32 u4_mb_info_type; |
|
|
|
/** Flag to indicate the size of mb info structure */ |
|
UWORD32 u4_mb_info_size; |
|
|
|
/** Buffer containing mb info if mb_info_type is non-zero */ |
|
void *pv_mb_info; |
|
|
|
/** Flag to indicate if pic info is sent along with input buffer */ |
|
UWORD32 u4_pic_info_type; |
|
|
|
/** Buffer containing pic info if mb_info_type is non-zero */ |
|
void *pv_pic_info; |
|
|
|
}inp_buf_t; |
|
|
|
typedef struct |
|
{ |
|
/** Descriptor of bitstream buffer */ |
|
iv_bits_buf_t s_bits_buf; |
|
|
|
/** Lower 32bits of time stamp corresponding to the above buffer */ |
|
UWORD32 u4_timestamp_low; |
|
|
|
/** Upper 32bits of time stamp corresponding to the above buffer */ |
|
UWORD32 u4_timestamp_high; |
|
|
|
/** Flag to indicate if the current buffer is last buffer */ |
|
UWORD32 u4_is_last; |
|
|
|
}out_buf_t; |
|
|
|
typedef struct |
|
{ |
|
/** Descriptor of picture buffer */ |
|
pic_buf_t s_pic_buf; |
|
|
|
/** Lower 32bits of time stamp corresponding to the above buffer */ |
|
UWORD32 u4_timestamp_low; |
|
|
|
/** Upper 32bits of time stamp corresponding to the above buffer */ |
|
UWORD32 u4_timestamp_high; |
|
|
|
/** Flag to indicate if the current buffer is last buffer */ |
|
UWORD32 u4_is_last; |
|
|
|
/** Picture count corresponding to current picture */ |
|
WORD32 i4_pic_cnt; |
|
|
|
}rec_buf_t; |
|
|
|
typedef struct |
|
{ |
|
/** maximum width for which codec should request memory requirements */ |
|
UWORD32 u4_max_wd; |
|
|
|
/** maximum height for which codec should request memory requirements */ |
|
UWORD32 u4_max_ht; |
|
|
|
/** Maximum number of reference frames */ |
|
UWORD32 u4_max_ref_cnt; |
|
|
|
/** Maximum number of reorder frames */ |
|
UWORD32 u4_max_reorder_cnt; |
|
|
|
/** Maximum level supported */ |
|
UWORD32 u4_max_level; |
|
|
|
/** Input color format */ |
|
IV_COLOR_FORMAT_T e_inp_color_fmt; |
|
|
|
/** Flag to enable/disable - To be used only for debugging/testing */ |
|
UWORD32 u4_enable_recon; |
|
|
|
/** Recon color format */ |
|
IV_COLOR_FORMAT_T e_recon_color_fmt; |
|
|
|
/** Encoder Speed preset - Value between 0 (slowest) and 100 (fastest) */ |
|
IVE_SPEED_CONFIG u4_enc_speed_preset; |
|
|
|
/** Rate control mode */ |
|
IVE_RC_MODE_T e_rc_mode; |
|
|
|
/** Maximum frame rate to be supported */ |
|
UWORD32 u4_max_framerate; |
|
|
|
/** Maximum bitrate to be supported */ |
|
UWORD32 u4_max_bitrate; |
|
|
|
/** Maximum number of consecutive B frames */ |
|
UWORD32 u4_num_bframes; |
|
|
|
/** Content type Interlaced/Progressive */ |
|
IV_CONTENT_TYPE_T e_content_type; |
|
|
|
/** Maximum search range to be used in X direction */ |
|
UWORD32 u4_max_srch_rng_x; |
|
|
|
/** Maximum search range to be used in Y direction */ |
|
UWORD32 u4_max_srch_rng_y; |
|
|
|
/** Slice Mode */ |
|
IVE_SLICE_MODE_T e_slice_mode; |
|
|
|
/** Slice parameter */ |
|
UWORD32 u4_slice_param; |
|
|
|
/** Processor architecture */ |
|
IV_ARCH_T e_arch; |
|
|
|
/** SOC details */ |
|
IV_SOC_T e_soc; |
|
|
|
/** Input width to be sent in bitstream */ |
|
UWORD32 u4_disp_wd; |
|
|
|
/** Input height to be sent in bitstream */ |
|
UWORD32 u4_disp_ht; |
|
|
|
/** Input width */ |
|
UWORD32 u4_wd; |
|
|
|
/** Input height */ |
|
UWORD32 u4_ht; |
|
|
|
/** Input stride */ |
|
UWORD32 u4_strd; |
|
|
|
/** Source frame rate */ |
|
UWORD32 u4_src_frame_rate; |
|
|
|
/** Target frame rate */ |
|
UWORD32 u4_tgt_frame_rate; |
|
|
|
/** Target bitrate in kilobits per second */ |
|
UWORD32 u4_target_bitrate; |
|
|
|
/** Force current frame type */ |
|
IV_PICTURE_CODING_TYPE_T e_frame_type; |
|
|
|
/** Encoder mode */ |
|
IVE_ENC_MODE_T e_enc_mode; |
|
|
|
/** Set initial Qp for I pictures */ |
|
UWORD32 u4_i_qp; |
|
|
|
/** Set initial Qp for P pictures */ |
|
UWORD32 u4_p_qp; |
|
|
|
/** Set initial Qp for B pictures */ |
|
UWORD32 u4_b_qp; |
|
|
|
/** Set minimum Qp for I pictures */ |
|
UWORD32 u4_i_qp_min; |
|
|
|
/** Set maximum Qp for I pictures */ |
|
UWORD32 u4_i_qp_max; |
|
|
|
/** Set minimum Qp for P pictures */ |
|
UWORD32 u4_p_qp_min; |
|
|
|
/** Set maximum Qp for P pictures */ |
|
UWORD32 u4_p_qp_max; |
|
|
|
/** Set minimum Qp for B pictures */ |
|
UWORD32 u4_b_qp_min; |
|
|
|
/** Set maximum Qp for B pictures */ |
|
UWORD32 u4_b_qp_max; |
|
|
|
/** Adaptive intra refresh mode */ |
|
IVE_AIR_MODE_T e_air_mode; |
|
|
|
/** Adaptive intra refresh period in frames */ |
|
UWORD32 u4_air_refresh_period; |
|
|
|
/** VBV buffer delay */ |
|
UWORD32 u4_vbv_buffer_delay; |
|
|
|
/** VBV buffer size */ |
|
UWORD32 u4_vbv_buf_size; |
|
|
|
/** Number of cores to be used */ |
|
UWORD32 u4_num_cores; |
|
|
|
/** ME speed preset - Value between 0 (slowest) and 100 (fastest) */ |
|
UWORD32 u4_me_speed_preset; |
|
|
|
/** Flag to enable/disable half pel motion estimation */ |
|
UWORD32 u4_enable_hpel; |
|
|
|
/** Flag to enable/disable quarter pel motion estimation */ |
|
UWORD32 u4_enable_qpel; |
|
|
|
/** Flag to enable/disable intra 4x4 analysis */ |
|
UWORD32 u4_enable_intra_4x4; |
|
|
|
/** Flag to enable/disable intra 8x8 analysis */ |
|
UWORD32 u4_enable_intra_8x8; |
|
|
|
/** Flag to enable/disable intra 16x16 analysis */ |
|
UWORD32 u4_enable_intra_16x16; |
|
|
|
/** Flag to enable/disable fast SAD approximation */ |
|
UWORD32 u4_enable_fast_sad; |
|
|
|
/*flag to enable/disable alternate reference frames */ |
|
UWORD32 u4_enable_alt_ref; |
|
|
|
/*Flag to enable/disable computation of SATDQ in ME*/ |
|
UWORD32 u4_enable_satqd; |
|
|
|
/*Minimum SAD to search for*/ |
|
WORD32 i4_min_sad; |
|
|
|
/** Maximum search range in X direction for farthest reference */ |
|
UWORD32 u4_srch_rng_x; |
|
|
|
/** Maximum search range in Y direction for farthest reference */ |
|
UWORD32 u4_srch_rng_y; |
|
|
|
/** I frame interval */ |
|
UWORD32 u4_i_frm_interval; |
|
|
|
/** IDR frame interval */ |
|
UWORD32 u4_idr_frm_interval; |
|
|
|
/** Disable deblock level (0: Enable completely, 3: Disable completely */ |
|
UWORD32 u4_disable_deblock_level; |
|
|
|
/** Profile */ |
|
IV_PROFILE_T e_profile; |
|
|
|
/** Lower 32bits of time stamp corresponding to input buffer, |
|
* from which this command takes effect */ |
|
UWORD32 u4_timestamp_low; |
|
|
|
/** Upper 32bits of time stamp corresponding to input buffer, |
|
* from which this command takes effect */ |
|
UWORD32 u4_timestamp_high; |
|
|
|
/** Flag to say if the current config parameter set is valid |
|
* Will be zero to start with and will be set to 1, when configured |
|
* Once encoder uses the parameter set, this will be set to zero */ |
|
UWORD32 u4_is_valid; |
|
|
|
/** Command associated with this config param set */ |
|
IVE_CONTROL_API_COMMAND_TYPE_T e_cmd; |
|
|
|
/** Input width in mbs */ |
|
UWORD32 i4_wd_mbs; |
|
|
|
/** Input height in mbs */ |
|
UWORD32 i4_ht_mbs; |
|
|
|
/** entropy coding mode flag */ |
|
UWORD32 u4_entropy_coding_mode; |
|
|
|
/** enable weighted prediction */ |
|
UWORD32 u4_weighted_prediction; |
|
|
|
/** enable constrained intra prediction */ |
|
UWORD32 u4_constrained_intra_pred; |
|
|
|
/** Pic info type */ |
|
UWORD32 u4_pic_info_type; |
|
/** |
|
* MB info type |
|
*/ |
|
UWORD32 u4_mb_info_type; |
|
|
|
/** VUI structure */ |
|
vui_t s_vui; |
|
|
|
}cfg_params_t; |
|
|
|
|
|
|
|
/** Structure to hold format conversion context */ |
|
typedef struct |
|
{ |
|
/** Current row for which format conversion should be done */ |
|
WORD32 i4_cur_row; |
|
|
|
/** Number of rows for which format conversion should be done */ |
|
WORD32 i4_num_rows; |
|
|
|
}fmt_conv_t; |
|
|
|
|
|
/** |
|
* Structure to represent a processing job entry |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* Command |
|
*/ |
|
WORD32 i4_cmd; |
|
|
|
/** |
|
* MB x of the starting MB |
|
*/ |
|
WORD16 i2_mb_x; |
|
|
|
/** |
|
* MB y of the starting MB |
|
*/ |
|
|
|
WORD16 i2_mb_y; |
|
|
|
/** |
|
* Number of MBs that need to be processed in this job |
|
*/ |
|
WORD16 i2_mb_cnt; |
|
|
|
/** |
|
* Process contexts base index |
|
* Will toggle between 0 and MAX_PROCESS_THREADS |
|
*/ |
|
WORD16 i2_proc_base_idx; |
|
|
|
} job_t; |
|
|
|
|
|
/** |
|
* Structure to represent a MV Bank buffer |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* Pointer to hold num PUs each MB in a picture |
|
*/ |
|
UWORD32 *pu4_mb_pu_cnt; |
|
|
|
/** |
|
* Pointer to hold enc_pu_t for each PU in a picture |
|
*/ |
|
enc_pu_t *ps_pic_pu; |
|
|
|
/** |
|
* Pointer to hold PU map for each MB in a picture |
|
*/ |
|
UWORD8 *pu1_pic_pu_map; |
|
|
|
/** |
|
* Pointer to hold the Slice map |
|
*/ |
|
UWORD16 *pu1_pic_slice_map; |
|
|
|
/** |
|
* Absolute POC for the current MV Bank |
|
*/ |
|
WORD32 i4_abs_poc; |
|
|
|
/** |
|
* Buffer Id |
|
*/ |
|
WORD32 i4_buf_id; |
|
|
|
} mv_buf_t; |
|
|
|
|
|
/** |
|
* Reference set containing pointers to MV buf and pic buf |
|
*/ |
|
typedef struct |
|
{ |
|
/** Picture count */ |
|
WORD32 i4_pic_cnt; |
|
|
|
/** POC */ |
|
WORD32 i4_poc; |
|
|
|
/** picture buffer */ |
|
pic_buf_t *ps_pic_buf; |
|
|
|
/** mv buffer */ |
|
mv_buf_t *ps_mv_buf; |
|
|
|
}ref_set_t; |
|
|
|
typedef struct |
|
{ |
|
|
|
/** |
|
* Pointer to current PPS |
|
*/ |
|
pps_t *ps_pps; |
|
|
|
/** |
|
* Pointer to current SPS |
|
*/ |
|
sps_t *ps_sps; |
|
|
|
/** |
|
* Pointer to current slice header structure |
|
*/ |
|
slice_header_t *ps_slice_hdr; |
|
|
|
/** |
|
* MB's x position within a picture in raster scan in MB units |
|
*/ |
|
WORD32 i4_mb_x; |
|
|
|
/** |
|
* MB's y position within a picture in raster scan in MB units |
|
*/ |
|
|
|
WORD32 i4_mb_y; |
|
|
|
/** |
|
* Current PU structure - set to MB enc_pu_t pointer at the start of MB processing and incremented |
|
* for every TU |
|
*/ |
|
enc_pu_t *ps_pu; |
|
|
|
/** |
|
* Pointer to frame level enc_pu_t for the current frame being parsed |
|
* where MVs and Intra pred modes will be updated |
|
*/ |
|
enc_pu_t *ps_pic_pu; |
|
|
|
/** |
|
* Pointer to hold num PUs each MB in a picture |
|
*/ |
|
UWORD32 *pu4_mb_pu_cnt; |
|
|
|
/** PU Index map per MB. The indices in this map are w.r.t picture pu array and not |
|
* w.r.t MB pu array. |
|
* This will be used during mv prediction and since neighbors will have different MB pu map |
|
* it will be easier if they all have indices w.r.t picture level PU array rather than MB level |
|
* PU array. |
|
* pu1_pic_pu_map is map w.r.t MB's enc_pu_t array |
|
*/ |
|
UWORD32 *pu4_pic_pu_idx_map; |
|
|
|
/** |
|
* Pointer to pu_map for the current frame being parsed |
|
* where MVs and Intra pred modes will be updated |
|
*/ |
|
UWORD8 *pu1_pic_pu_map; |
|
|
|
/** |
|
* PU count in current MB |
|
*/ |
|
WORD32 i4_mb_pu_cnt; |
|
|
|
/** |
|
* PU count in current MB |
|
*/ |
|
WORD32 i4_mb_start_pu_idx; |
|
|
|
/** |
|
* Top availability for current MB level |
|
*/ |
|
UWORD8 u1_top_mb_avail; |
|
|
|
/** |
|
* Top right availability for current MB level |
|
*/ |
|
UWORD8 u1_top_rt_mb_avail; |
|
/** |
|
* Top left availability for current MB level |
|
*/ |
|
UWORD8 u1_top_lt_mb_avail; |
|
/** |
|
* left availability for current MB level |
|
*/ |
|
UWORD8 u1_left_mb_avail; |
|
|
|
}mv_ctxt_t; |
|
|
|
typedef struct |
|
{ |
|
/** |
|
* MB's x position within a picture in raster scan in MB units |
|
*/ |
|
WORD32 i4_mb_x; |
|
|
|
/** |
|
* MB's y position within a picture in raster scan in MB units |
|
*/ |
|
WORD32 i4_mb_y; |
|
|
|
/** |
|
* MB's x position within a Slice in raster scan in MB units |
|
*/ |
|
WORD32 i4_mb_slice_x; |
|
|
|
/** |
|
* MB's y position within a Slice in raster scan in MB units |
|
*/ |
|
WORD32 i4_mb_slice_y; |
|
|
|
/** |
|
* Vertical strength, Two bits per edge. |
|
* Stored in format. BS[15] | BS[14] | .. |BS[0] |
|
*/ |
|
UWORD32 *pu4_pic_vert_bs; |
|
|
|
/** |
|
* Boundary strength, Two bits per edge. |
|
* Stored in format. BS[15] | BS[14] | .. |BS[0] |
|
*/ |
|
UWORD32 *pu4_pic_horz_bs; |
|
|
|
/** |
|
* Qp array stored for each mb |
|
*/ |
|
UWORD8 *pu1_pic_qp; |
|
|
|
}bs_ctxt_t; |
|
|
|
typedef struct |
|
{ |
|
/** |
|
* MB's x position within a picture in raster scan in MB units |
|
*/ |
|
WORD32 i4_mb_x; |
|
|
|
/** |
|
* MB's y position within a picture in raster scan in MB units |
|
*/ |
|
WORD32 i4_mb_y; |
|
|
|
/** |
|
* structure that contains BS and QP frame level arrays |
|
*/ |
|
bs_ctxt_t s_bs_ctxt; |
|
|
|
/** |
|
* Pointer to 0th luma pixel in current pic |
|
*/ |
|
UWORD8 *pu1_cur_pic_luma; |
|
|
|
/** |
|
* Pointer to 0th chroma pixel in current pic |
|
*/ |
|
UWORD8 *pu1_cur_pic_chroma; |
|
|
|
/** |
|
* Points to the array of slice indices which is used to identify the slice |
|
* to which each MB in a frame belongs. |
|
*/ |
|
UWORD8 *pu1_slice_idx; |
|
|
|
}deblk_ctxt_t; |
|
|
|
|
|
/** |
|
****************************************************************************** |
|
* @brief Structure to hold data and flags for 'n' mb processing for |
|
* deblocking , padding and half pel generation. |
|
****************************************************************************** |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* MB's x position last processed + 1 |
|
*/ |
|
WORD32 i4_mb_x; |
|
|
|
/** |
|
* MB's y position ,current processing. |
|
*/ |
|
WORD32 i4_mb_y; |
|
|
|
/** |
|
* Number of MBs processed in a stretch |
|
*/ |
|
WORD32 i4_n_mbs; |
|
|
|
}n_mb_process_ctxt_t; |
|
|
|
|
|
/** |
|
****************************************************************************** |
|
* @brief Structure to hold coefficient info for a 4x4 subblock. |
|
* The following can be used to type-cast coefficient data that is stored |
|
* per subblock. Note that though i2_level is shown as an array that |
|
* holds 16 coefficients, only the first few entries will be valid. Next |
|
* subblocks data starts after the valid number of coefficients. Number |
|
* of non-zero coefficients will be derived using number of non-zero bits |
|
* in sig coeff map |
|
****************************************************************************** |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* significant coefficient map and nnz are packed in |
|
* to msb (2 bytes) and lsb (2 bytes) respectively |
|
*/ |
|
WORD32 i4_sig_map_nnz; |
|
|
|
/** |
|
* array of non zero residue coefficients |
|
*/ |
|
WORD16 ai2_residue[16]; |
|
|
|
}tu_sblk_coeff_data_t; |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief Structure contains few common state variables such as MB indices, |
|
* current SPS, PPS etc which are to be used in the entropy thread. By keeping |
|
* it a different structure it is being explicitly signaled that these |
|
* variables are specific to entropy threads context and other threads should |
|
* not update these elements |
|
****************************************************************************** |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* Pointer to the cabac context |
|
*/ |
|
cabac_ctxt_t *ps_cabac; |
|
|
|
/** |
|
* start of frame / start of slice flag |
|
*/ |
|
WORD32 i4_sof; |
|
|
|
/** |
|
* end of frame / end of slice flag |
|
*/ |
|
WORD32 i4_eof; |
|
|
|
/** |
|
* generate header upon request |
|
*/ |
|
WORD32 i4_gen_header; |
|
|
|
/** |
|
* seq_parameter_set_id |
|
*/ |
|
UWORD32 u4_sps_id; |
|
|
|
/** |
|
* Pointer to base of sequence parameter set structure array |
|
*/ |
|
sps_t *ps_sps_base; |
|
|
|
/** |
|
* pic_parameter_set_id |
|
*/ |
|
UWORD32 u4_pps_id; |
|
|
|
/** |
|
* Pointer to base of Picture parameter set structure array |
|
*/ |
|
pps_t *ps_pps_base; |
|
|
|
/** |
|
* Current slice idx |
|
*/ |
|
WORD32 i4_cur_slice_idx; |
|
|
|
/** |
|
* Points to the array of slice indices which is used to identify the independent slice |
|
* to which each MB in a frame belongs. |
|
*/ |
|
UWORD8 *pu1_slice_idx; |
|
|
|
/** |
|
* Pointer to base of slice header structure array |
|
*/ |
|
slice_header_t *ps_slice_hdr_base; |
|
|
|
/** |
|
* entropy status |
|
*/ |
|
UWORD8 *pu1_entropy_map; |
|
|
|
/** |
|
* MB's x position within a picture in raster scan in MB units |
|
*/ |
|
WORD32 i4_mb_x; |
|
|
|
/** |
|
* MB's y position within a picture in raster scan in MB units |
|
*/ |
|
WORD32 i4_mb_y; |
|
|
|
/** |
|
* MB start address |
|
*/ |
|
WORD32 i4_mb_cnt; |
|
|
|
/** |
|
* MB start address |
|
*/ |
|
WORD32 i4_mb_start_add; |
|
|
|
/** |
|
* MB end address |
|
*/ |
|
WORD32 i4_mb_end_add; |
|
|
|
/** |
|
* Input width in mbs |
|
*/ |
|
WORD32 i4_wd_mbs; |
|
|
|
/** |
|
* Input height in mbs |
|
*/ |
|
WORD32 i4_ht_mbs; |
|
|
|
/** |
|
* Bitstream structure |
|
*/ |
|
bitstrm_t *ps_bitstrm; |
|
|
|
/** |
|
* transform_8x8_mode_flag |
|
*/ |
|
WORD8 i1_transform_8x8_mode_flag; |
|
|
|
/** |
|
* entropy_coding_mode_flag |
|
*/ |
|
WORD8 u1_entropy_coding_mode_flag; |
|
|
|
/** |
|
* Pointer to the top row nnz for luma |
|
*/ |
|
UWORD8 (*pu1_top_nnz_luma)[4]; |
|
|
|
/** |
|
* left nnz for luma |
|
*/ |
|
UWORD32 u4_left_nnz_luma; |
|
|
|
/** |
|
* Pointer to zero runs before for the mb |
|
*/ |
|
UWORD8 au1_zero_run[16]; |
|
|
|
/** |
|
* Pointer to the top row nnz for chroma |
|
*/ |
|
UWORD8 (*pu1_top_nnz_cbcr)[4]; |
|
|
|
/** |
|
* left nnz for chroma |
|
*/ |
|
UWORD8 u4_left_nnz_cbcr; |
|
|
|
/** |
|
* Pointer frame level mb subblock coeff data |
|
*/ |
|
void *pv_pic_mb_coeff_data; |
|
|
|
/** |
|
* Pointer to mb subblock coeff data and number of subblocks and scan idx |
|
* Incremented each time a coded subblock is processed |
|
*/ |
|
void *pv_mb_coeff_data; |
|
|
|
/** |
|
* Pointer frame level mb header data |
|
*/ |
|
void *pv_pic_mb_header_data; |
|
|
|
/** |
|
* Pointer to mb header data and |
|
* incremented each time a coded mb is encoded |
|
*/ |
|
void *pv_mb_header_data; |
|
|
|
/** |
|
* Error code during parse stage |
|
*/ |
|
IH264E_ERROR_T i4_error_code; |
|
|
|
/** |
|
* Void pointer to job context |
|
*/ |
|
void *pv_proc_jobq, *pv_entropy_jobq; |
|
|
|
/** |
|
* Flag to signal end of frame |
|
*/ |
|
WORD32 i4_end_of_frame; |
|
|
|
/** |
|
* Abs POC count of the frame |
|
*/ |
|
WORD32 i4_abs_pic_order_cnt; |
|
|
|
/** |
|
* mb skip run |
|
*/ |
|
WORD32 *pi4_mb_skip_run; |
|
|
|
/** |
|
* Flag to signal end of sequence |
|
*/ |
|
UWORD32 u4_is_last; |
|
|
|
/** |
|
* Lower 32bits of time-stamp corresponding to the buffer being encoded |
|
*/ |
|
UWORD32 u4_timestamp_low; |
|
|
|
/** |
|
* Upper 32bits of time-stamp corresponding to the buffer being encoded |
|
*/ |
|
UWORD32 u4_timestamp_high; |
|
|
|
/** |
|
* Current Picture count - used for synchronization |
|
*/ |
|
WORD32 i4_pic_cnt; |
|
|
|
/** |
|
* Number of bits consumed by header for I and P mb types |
|
*/ |
|
UWORD32 u4_header_bits[MAX_MB_TYPE]; |
|
|
|
/** |
|
* Number of bits consumed by residue for I and P mb types |
|
*/ |
|
UWORD32 u4_residue_bits[MAX_MB_TYPE]; |
|
|
|
} entropy_ctxt_t; |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief macro block info. |
|
****************************************************************************** |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* mb type |
|
*/ |
|
UWORD16 u2_is_intra; |
|
|
|
/** |
|
* mb type |
|
*/ |
|
UWORD16 u2_mb_type; |
|
|
|
/** |
|
* csbp |
|
*/ |
|
UWORD32 u4_csbp; |
|
|
|
/** |
|
* mb distortion |
|
*/ |
|
WORD32 i4_mb_distortion; |
|
|
|
}mb_info_t; |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief mb_hdr structures to access first few common elements of above |
|
* structures |
|
****************************************************************************** |
|
*/ |
|
|
|
typedef struct |
|
{ |
|
/** |
|
* mb type and mode |
|
*/ |
|
UWORD8 u1_mb_type_mode; |
|
|
|
/** |
|
* CBP |
|
*/ |
|
UWORD8 u1_cbp; |
|
|
|
/** |
|
* MB qp delta |
|
*/ |
|
UWORD8 u1_mb_qp_delta; |
|
|
|
/** |
|
* Element to align structure to 2 byte boundary |
|
*/ |
|
UWORD8 u1_pad; |
|
}mb_hdr_common_t; |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief macro block info for I4x4 MB |
|
****************************************************************************** |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* Common MB header params |
|
*/ |
|
mb_hdr_common_t common; |
|
|
|
/** |
|
* Sub block modes, 2 modes per byte |
|
*/ |
|
UWORD8 au1_sub_blk_modes[8]; |
|
}mb_hdr_i4x4_t; |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief macro block info for I8x8 MB |
|
****************************************************************************** |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* Common MB header params |
|
*/ |
|
mb_hdr_common_t common; |
|
|
|
|
|
/** |
|
* Sub block modes, 2 modes per byte |
|
*/ |
|
UWORD8 au1_sub_blk_modes[2]; |
|
}mb_hdr_i8x8_t; |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief macro block info for I16x16 MB |
|
****************************************************************************** |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* Common MB header params |
|
*/ |
|
mb_hdr_common_t common; |
|
|
|
}mb_hdr_i16x16_t; |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief macro block info for P16x16 MB |
|
****************************************************************************** |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* Common MB header params |
|
*/ |
|
mb_hdr_common_t common; |
|
|
|
/** |
|
* MV |
|
*/ |
|
WORD16 ai2_mv[2]; |
|
}mb_hdr_p16x16_t; |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief macro block info for PSKIP MB |
|
****************************************************************************** |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* Common MB header params |
|
*/ |
|
mb_hdr_common_t common; |
|
|
|
}mb_hdr_pskip_t; |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief macro block info for B16x16 MB |
|
****************************************************************************** |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* Common MB header params |
|
*/ |
|
mb_hdr_common_t common; |
|
|
|
|
|
/** |
|
* MV |
|
*/ |
|
WORD16 ai2_mv[2][2]; |
|
}mb_hdr_b16x16_t; |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief macro block info for BDIRECT MB |
|
****************************************************************************** |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* Common MB header params |
|
*/ |
|
mb_hdr_common_t common; |
|
|
|
}mb_hdr_bdirect_t; |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief macro block info for PSKIP MB |
|
****************************************************************************** |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* Common MB header params |
|
*/ |
|
mb_hdr_common_t common; |
|
|
|
}mb_hdr_bskip_t; |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief Union of mb_hdr structures for size calculation |
|
* and to access first few common elements |
|
****************************************************************************** |
|
*/ |
|
|
|
typedef union |
|
{ |
|
mb_hdr_i4x4_t mb_hdr_i4x4; |
|
mb_hdr_i8x8_t mb_hdr_i8x8; |
|
mb_hdr_i16x16_t mb_hdr_i16x16; |
|
mb_hdr_p16x16_t mb_hdr_p16x16; |
|
mb_hdr_pskip_t mb_hdr_pskip; |
|
mb_hdr_b16x16_t mb_hdr_b16x16; |
|
mb_hdr_bdirect_t mb_hdr_bdirect; |
|
mb_hdr_bskip_t mb_hdr_bskip; |
|
}mb_hdr_t; |
|
/** |
|
****************************************************************************** |
|
* @brief structure presenting the neighbor availability of a mb |
|
* or subblk or any other partition |
|
****************************************************************************** |
|
*/ |
|
typedef struct |
|
{ |
|
/** |
|
* left blk/subblk/partition |
|
*/ |
|
UWORD8 u1_mb_a; |
|
|
|
/** |
|
* top blk/subblk/partition |
|
*/ |
|
UWORD8 u1_mb_b; |
|
|
|
/** |
|
* topright blk/subblk/partition |
|
*/ |
|
UWORD8 u1_mb_c; |
|
|
|
/** |
|
* topleft blk/subblk/partition |
|
*/ |
|
UWORD8 u1_mb_d; |
|
|
|
}block_neighbors_t; |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief MB info related variables used during NMB processing |
|
****************************************************************************** |
|
*/ |
|
typedef struct |
|
{ |
|
UWORD32 u4_mb_type; |
|
UWORD32 u4_min_sad; |
|
UWORD32 u4_min_sad_reached; |
|
WORD32 i4_mb_cost; |
|
WORD32 i4_mb_distortion; |
|
|
|
enc_pu_mv_t as_skip_mv[4]; |
|
|
|
enc_pu_mv_t as_pred_mv[2]; |
|
|
|
block_neighbors_t s_ngbr_avbl; |
|
|
|
/* |
|
* Buffer to hold best subpel buffer in each MB of NMB |
|
*/ |
|
UWORD8 *pu1_best_sub_pel_buf; |
|
|
|
/* |
|
* Stride for subpel buffer |
|
*/ |
|
UWORD32 u4_bst_spel_buf_strd; |
|
|
|
}mb_info_nmb_t; |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief Pixel processing thread context |
|
****************************************************************************** |
|
*/ |
|
struct _proc_t |
|
{ |
|
/** |
|
* entropy context |
|
*/ |
|
entropy_ctxt_t s_entropy; |
|
|
|
/** |
|
* me context |
|
*/ |
|
me_ctxt_t s_me_ctxt; |
|
|
|
/** |
|
* Pointer to codec context |
|
*/ |
|
codec_t *ps_codec; |
|
|
|
/** |
|
* N mb process contest |
|
*/ |
|
n_mb_process_ctxt_t s_n_mb_ctxt; |
|
|
|
/** |
|
* Source pointer to current MB luma |
|
*/ |
|
UWORD8 *pu1_src_buf_luma; |
|
|
|
/** |
|
* Source pointer to current MB chroma |
|
*/ |
|
UWORD8 *pu1_src_buf_chroma; |
|
|
|
/** |
|
* Recon pointer to current MB luma |
|
*/ |
|
UWORD8 *pu1_rec_buf_luma; |
|
|
|
/** |
|
* Recon pointer to current MB chroma |
|
*/ |
|
UWORD8 *pu1_rec_buf_chroma; |
|
|
|
/** |
|
* Ref pointer to current MB luma |
|
*/ |
|
UWORD8 *apu1_ref_buf_luma[MAX_REF_PIC_CNT]; |
|
|
|
/** |
|
* Ref pointer to current MB chroma |
|
*/ |
|
UWORD8 *apu1_ref_buf_chroma[MAX_REF_PIC_CNT]; |
|
|
|
/** |
|
* pointer to luma plane of input buffer (base :: mb (0,0)) |
|
*/ |
|
UWORD8 *pu1_src_buf_luma_base; |
|
|
|
/** |
|
* pointer to luma plane of reconstructed buffer (base :: mb (0,0)) |
|
*/ |
|
UWORD8 *pu1_rec_buf_luma_base; |
|
|
|
/** |
|
* pointer to luma plane of ref buffer (base :: mb (0,0)) |
|
*/ |
|
UWORD8 *apu1_ref_buf_luma_base[MAX_REF_PIC_CNT]; |
|
|
|
/** |
|
* pointer to chroma plane of input buffer (base :: mb (0,0)) |
|
*/ |
|
UWORD8 *pu1_src_buf_chroma_base; |
|
|
|
/* |
|
* Buffer for color space conversion of luma |
|
*/ |
|
UWORD8 *pu1_y_csc_buf; |
|
|
|
/* |
|
* Buffer for color space conversion of luma |
|
*/ |
|
|
|
UWORD8 *pu1_uv_csc_buf; |
|
|
|
/** |
|
* pointer to chroma plane of reconstructed buffer (base :: mb (0,0)) |
|
*/ |
|
UWORD8 *pu1_rec_buf_chroma_base; |
|
|
|
/** |
|
* pointer to chroma plane of reconstructed buffer (base :: mb (0,0)) |
|
*/ |
|
UWORD8 *apu1_ref_buf_chroma_base[MAX_REF_PIC_CNT]; |
|
|
|
/** |
|
* Pointer to ME NMB info |
|
*/ |
|
mb_info_nmb_t *ps_nmb_info; |
|
|
|
mb_info_nmb_t *ps_cur_mb; |
|
|
|
/** |
|
* source luma stride |
|
*/ |
|
WORD32 i4_src_strd; |
|
|
|
/** |
|
* source chroma stride |
|
*/ |
|
WORD32 i4_src_chroma_strd; |
|
|
|
/** |
|
* recon stride & ref stride |
|
* (strides for luma and chroma are the same) |
|
*/ |
|
WORD32 i4_rec_strd; |
|
|
|
/** |
|
* Offset for half pel x plane from the pic buf |
|
*/ |
|
UWORD32 u4_half_x_offset; |
|
|
|
/** |
|
* Offset for half pel y plane from half x plane |
|
*/ |
|
UWORD32 u4_half_y_offset; |
|
|
|
/** |
|
* Offset for half pel xy plane from half y plane |
|
*/ |
|
UWORD32 u4_half_xy_offset; |
|
|
|
/** |
|
* pred buffer pointer (temp buffer 1) |
|
*/ |
|
UWORD8 *pu1_pred_mb; |
|
|
|
/** |
|
* pred buffer pointer (prediction buffer for intra 16x16 |
|
*/ |
|
UWORD8 *pu1_pred_mb_intra_16x16; |
|
|
|
/** |
|
* pred buffer pointer (prediction buffer for intra 16x16_plane |
|
*/ |
|
UWORD8 *pu1_pred_mb_intra_16x16_plane; |
|
|
|
/** |
|
* pred buffer pointer (prediction buffer for intra chroma |
|
*/ |
|
UWORD8 *pu1_pred_mb_intra_chroma; |
|
|
|
/** |
|
* pred buffer pointer (prediction buffer for intra chroma plane |
|
*/ |
|
UWORD8 *pu1_pred_mb_intra_chroma_plane; |
|
|
|
/** |
|
* temp. reference buffer ptr for intra 4x4 when rdopt is on |
|
*/ |
|
UWORD8 *pu1_ref_mb_intra_4x4; |
|
|
|
/** |
|
* prediction buffer stride |
|
*/ |
|
WORD32 i4_pred_strd; |
|
|
|
/** |
|
* transform buffer pointer (temp buffer 2) |
|
*/ |
|
WORD16 *pi2_res_buf; |
|
|
|
/** |
|
* temp. transform buffer ptr for intra 4x4 when rdopt is on |
|
*/ |
|
WORD16 *pi2_res_buf_intra_4x4; |
|
|
|
/** |
|
* transform buffer stride |
|
*/ |
|
WORD32 i4_res_strd; |
|
|
|
/** |
|
* scratch buffer for inverse transform (temp buffer 3) |
|
*/ |
|
void *pv_scratch_buff; |
|
|
|
/** |
|
* frame num |
|
*/ |
|
WORD32 i4_frame_num; |
|
|
|
/** |
|
* start address of frame / sub-frame |
|
*/ |
|
WORD32 i4_frame_strt_add; |
|
|
|
/** |
|
* IDR pic |
|
*/ |
|
UWORD32 u4_is_idr; |
|
|
|
/** |
|
* idr_pic_id |
|
*/ |
|
UWORD32 u4_idr_pic_id; |
|
|
|
/** |
|
* Input width in mbs |
|
*/ |
|
WORD32 i4_wd_mbs; |
|
|
|
/** |
|
* Input height in mbs |
|
*/ |
|
WORD32 i4_ht_mbs; |
|
|
|
/** |
|
* slice_type |
|
*/ |
|
WORD32 i4_slice_type; |
|
|
|
/** |
|
* Current slice idx |
|
*/ |
|
WORD32 i4_cur_slice_idx; |
|
|
|
/** |
|
* MB's x position within a picture in raster scan in MB units |
|
*/ |
|
WORD32 i4_mb_x; |
|
|
|
/** |
|
* MB's y position within a picture in raster scan in MB units |
|
*/ |
|
WORD32 i4_mb_y; |
|
|
|
/** |
|
* MB's x position within a Slice in raster scan in MB units |
|
*/ |
|
WORD32 i4_mb_slice_x; |
|
|
|
/** |
|
* MB's y position within a Slice in raster scan in MB units |
|
*/ |
|
WORD32 i4_mb_slice_y; |
|
|
|
/** |
|
* mb type |
|
*/ |
|
UWORD32 u4_mb_type; |
|
|
|
/** |
|
* is intra |
|
*/ |
|
UWORD32 u4_is_intra; |
|
|
|
/** |
|
* mb neighbor availability pointer |
|
*/ |
|
block_neighbors_t *ps_ngbr_avbl; |
|
|
|
/** |
|
* lambda (lagrange multiplier for cost computation) |
|
*/ |
|
UWORD32 u4_lambda; |
|
|
|
/** |
|
* mb distortion |
|
*/ |
|
WORD32 i4_mb_distortion; |
|
|
|
/** |
|
* mb cost |
|
*/ |
|
WORD32 i4_mb_cost; |
|
|
|
/********************************************************************/ |
|
/* i4_ngbr_avbl_mb_16 - ngbr avbl of curr mb */ |
|
/* i4_ngbr_avbl_sb_8 - ngbr avbl of all 8x8 sub blocks of curr mb */ |
|
/* i4_ngbr_avbl_sb_4 - ngbr avbl of all 4x4 sub blocks of curr mb */ |
|
/* i4_ngbr_avbl_mb_c - chroma ngbr avbl of curr mb */ |
|
/********************************************************************/ |
|
WORD32 i4_ngbr_avbl_16x16_mb; |
|
WORD32 ai4_neighbor_avail_8x8_subblks[4]; |
|
UWORD8 au1_ngbr_avbl_4x4_subblks[16]; |
|
WORD32 i4_chroma_neighbor_avail_8x8_mb; |
|
|
|
/** |
|
* array to store the mode of mb sub blocks |
|
*/ |
|
UWORD8 au1_intra_luma_mb_4x4_modes[16]; |
|
|
|
/** |
|
* array to store the predicted mode of mb sub blks |
|
*/ |
|
UWORD8 au1_predicted_intra_luma_mb_4x4_modes[16]; |
|
|
|
/** |
|
* macro block intra 16x16 mode |
|
*/ |
|
UWORD8 u1_l_i16_mode; |
|
|
|
/** |
|
* array to store the mode of the macro block intra 8x8 4 modes |
|
*/ |
|
UWORD8 au1_intra_luma_mb_8x8_modes[4]; |
|
|
|
/** |
|
* intra chroma mb mode |
|
*/ |
|
UWORD8 u1_c_i8_mode; |
|
|
|
/********************************************************************/ |
|
/* array to store pixels from the neighborhood for intra prediction */ |
|
/* i16 - 16 left pels + 1 top left pel + 16 top pels = 33 pels */ |
|
/* i8 - 8 lpels + 1 tlpels + 8 tpels + 8 tr pels = 25 pels */ |
|
/* i4 - 4 lpels + 1 tlpels + 4 tpels + 4 tr pels = 13 pels */ |
|
/* ic - 8 left pels + 1 top left pel + 8 top pels )*2 */ |
|
/********************************************************************/ |
|
UWORD8 au1_ngbr_pels[34]; |
|
|
|
/** |
|
* array for 8x8 intra pels filtering (temp buff 4) |
|
*/ |
|
UWORD8 au1_neighbor_pels_i8x8_unfiltered[25]; |
|
|
|
/** |
|
* Number of sub partitons in the inter pred MB |
|
*/ |
|
UWORD32 u4_num_sub_partitions; |
|
|
|
/** |
|
* Pointer to hold num PUs each MB in a picture |
|
*/ |
|
UWORD32 *pu4_mb_pu_cnt; |
|
|
|
/** |
|
* Pointer to the array of structures having motion vectors, size |
|
* and position of sub partitions |
|
*/ |
|
enc_pu_t *ps_pu; |
|
|
|
/** |
|
* Pointer to the pu of current co-located MB in list 1 |
|
*/ |
|
enc_pu_t *ps_colpu; |
|
|
|
/** |
|
* predicted motion vector |
|
*/ |
|
enc_pu_mv_t *ps_skip_mv; |
|
|
|
/** |
|
* predicted motion vector |
|
*/ |
|
enc_pu_mv_t *ps_pred_mv; |
|
|
|
/** |
|
* top row mb syntax information base |
|
* In normal working scenarios, for a given context set, |
|
* the mb syntax info pointer is identical across all process threads. |
|
* But when the hard bound on slices are enabled, in multi core, frame |
|
* is partitioned in to sections equal to set number of cores and each |
|
* partition is run independently. In this scenario, a ctxt set will alone |
|
* appear to run multiple frames at a time. For this to occur, the common |
|
* pointers across the proc ctxt should disappear. |
|
* |
|
* This is done by allocating MAX_PROCESS_THREADS memory and distributing |
|
* across individual ctxts when byte bnd per slice is enabled. |
|
*/ |
|
mb_info_t *ps_top_row_mb_syntax_ele_base; |
|
|
|
/** |
|
* top row mb syntax information |
|
*/ |
|
mb_info_t *ps_top_row_mb_syntax_ele; |
|
|
|
/** |
|
* left mb syntax information |
|
*/ |
|
mb_info_t s_left_mb_syntax_ele; |
|
|
|
/** |
|
* top left mb syntax information |
|
*/ |
|
mb_info_t s_top_left_mb_syntax_ele; |
|
|
|
/** |
|
* top left mb syntax information |
|
*/ |
|
|
|
mb_info_t s_top_left_mb_syntax_ME; |
|
|
|
/** |
|
* left mb motion vector |
|
*/ |
|
enc_pu_t s_left_mb_pu_ME; |
|
|
|
/** |
|
* top left mb motion vector |
|
*/ |
|
enc_pu_t s_top_left_mb_pu_ME; |
|
|
|
/** |
|
* mb neighbor availability pointer |
|
*/ |
|
block_neighbors_t s_ngbr_avbl; |
|
|
|
/** |
|
* In case the macroblock type is intra, the intra modes of all |
|
* partitions for the left mb are stored in the array below |
|
*/ |
|
UWORD8 au1_left_mb_intra_modes[16]; |
|
|
|
/** |
|
* In case the macroblock type is intra, the intra modes of all |
|
* partitions for the top mb are stored in the array below |
|
* |
|
* In normal working scenarios, for a given context set, |
|
* the mb syntax info pointer is identical across all process threads. |
|
* But when the hard bound on slices are enabled, in multi core, frame |
|
* is partitioned in to sections equal to set number of cores and each |
|
* partition is run independently. In this scenario, a ctxt set will alone |
|
* appear to run multiple frames at a time. For this to occur, the common |
|
* pointers across the proc ctxt should disappear. |
|
* |
|
* This is done by allocating MAX_PROCESS_THREADS memory and distributing |
|
* across individual ctxts when byte bnd per slice is enabled. |
|
*/ |
|
UWORD8 *pu1_top_mb_intra_modes_base; |
|
|
|
/** |
|
* In case the macroblock type is intra, the intra modes of all |
|
* partitions for the top mb are stored in the array below |
|
*/ |
|
UWORD8 *pu1_top_mb_intra_modes; |
|
|
|
/** |
|
* left mb motion vector |
|
*/ |
|
enc_pu_t s_left_mb_pu; |
|
|
|
/** |
|
* top left mb motion vector |
|
*/ |
|
enc_pu_t s_top_left_mb_pu; |
|
|
|
/** |
|
* top row motion vector info |
|
* |
|
* In normal working scenarios, for a given context set, |
|
* the top row pu pointer is identical across all process threads. |
|
* But when the hard bound on slices are enabled, in multi core, frame |
|
* is partitioned in to sections equal to set number of cores and each |
|
* partition is run independently. In this scenario, a ctxt set will alone |
|
* appear to run multiple frames at a time. For this to occur, the common |
|
* pointers across the proc ctxt should disappear. |
|
* |
|
* This is done by allocating MAX_PROCESS_THREADS memory and distributing |
|
* across individual ctxts when byte bnd per slice is enabled. |
|
*/ |
|
enc_pu_t *ps_top_row_pu_base; |
|
|
|
/** |
|
* top row motion vector info |
|
*/ |
|
enc_pu_t *ps_top_row_pu; |
|
|
|
enc_pu_t *ps_top_row_pu_ME; |
|
|
|
/** |
|
* coded block pattern |
|
*/ |
|
UWORD32 u4_cbp; |
|
|
|
/** |
|
* csbp |
|
*/ |
|
UWORD32 u4_csbp; |
|
|
|
/** |
|
* number of non zero coeffs |
|
*/ |
|
UWORD32 au4_nnz[5]; |
|
|
|
/** |
|
* number of non zero coeffs for intra 4x4 when rdopt is on |
|
*/ |
|
UWORD32 au4_nnz_intra_4x4[4]; |
|
|
|
/** |
|
* frame qp & mb qp |
|
*/ |
|
UWORD32 u4_frame_qp, u4_mb_qp; |
|
|
|
/** |
|
* mb qp previous |
|
*/ |
|
UWORD32 u4_mb_qp_prev; |
|
|
|
/** |
|
* quantization parameters for luma & chroma planes |
|
*/ |
|
quant_params_t *ps_qp_params[3]; |
|
|
|
/** |
|
* Pointer frame level mb subblock coeff data |
|
*/ |
|
void *pv_pic_mb_coeff_data; |
|
|
|
/** |
|
* Pointer to mb subblock coeff data and number of subblocks and scan idx |
|
* Incremented each time a coded subblock is processed |
|
*/ |
|
void *pv_mb_coeff_data; |
|
|
|
/** |
|
* Pointer frame level mb header data |
|
*/ |
|
void *pv_pic_mb_header_data; |
|
|
|
/** |
|
* Pointer to mb header data and |
|
* incremented each time a coded mb is encoded |
|
*/ |
|
void *pv_mb_header_data; |
|
|
|
/** |
|
* Signal that pic_init is called first time |
|
*/ |
|
WORD32 i4_first_pic_init; |
|
|
|
/** |
|
* Current MV Bank's buffer ID |
|
*/ |
|
WORD32 i4_cur_mv_bank_buf_id; |
|
|
|
/** |
|
* Void pointer to job context |
|
*/ |
|
void *pv_proc_jobq, *pv_entropy_jobq; |
|
|
|
/** |
|
* Number of MBs to be processed in the current Job |
|
*/ |
|
WORD32 i4_mb_cnt; |
|
|
|
/** |
|
* ID for the current context - Used for debugging |
|
*/ |
|
WORD32 i4_id; |
|
|
|
/** |
|
* Pointer to current picture buffer structure |
|
*/ |
|
pic_buf_t *ps_cur_pic; |
|
|
|
/** |
|
* Pointer to current picture's mv buffer structure |
|
*/ |
|
mv_buf_t *ps_cur_mv_buf; |
|
|
|
/** |
|
* Flag to indicate if ps_proc was initialized at least once in a frame. |
|
* This is needed to handle cases where a core starts to handle format |
|
* conversion jobs directly |
|
*/ |
|
WORD32 i4_init_done; |
|
|
|
/** |
|
* Process status: one byte per MB |
|
*/ |
|
UWORD8 *pu1_proc_map; |
|
|
|
/** |
|
* Deblk status: one byte per MB |
|
*/ |
|
UWORD8 *pu1_deblk_map; |
|
|
|
/** |
|
* Process status: one byte per MB |
|
*/ |
|
UWORD8 *pu1_me_map; |
|
|
|
/* |
|
* Intra refresh mask. |
|
* Indicates if an Mb is coded in intra mode within the current AIR interval |
|
* NOTE Refreshes after each AIR period |
|
* NOTE The map is shared between process |
|
*/ |
|
UWORD8 *pu1_is_intra_coded; |
|
|
|
/** |
|
* Disable deblock level (0: Enable completely, 3: Disable completely |
|
*/ |
|
UWORD32 u4_disable_deblock_level; |
|
|
|
/** |
|
* Pointer to the structure that contains deblock context |
|
*/ |
|
deblk_ctxt_t s_deblk_ctxt; |
|
|
|
/** |
|
* Points to the array of slice indices which is used to identify the independent |
|
* slice to which each MB in a frame belongs. |
|
*/ |
|
UWORD8 *pu1_slice_idx; |
|
|
|
/** |
|
* Pointer to base of slice header structure array |
|
*/ |
|
slice_header_t *ps_slice_hdr_base; |
|
|
|
/** |
|
* Number of mb's to process in one loop |
|
*/ |
|
WORD32 i4_nmb_ntrpy; |
|
|
|
/** |
|
* Number of mb's to process in one loop |
|
*/ |
|
UWORD32 u4_nmb_me; |
|
|
|
/** |
|
* Structure for current input buffer |
|
*/ |
|
inp_buf_t s_inp_buf; |
|
|
|
/** |
|
* api call cnt |
|
*/ |
|
WORD32 i4_encode_api_call_cnt; |
|
|
|
/** |
|
* Current Picture count - used for synchronization |
|
*/ |
|
WORD32 i4_pic_cnt; |
|
|
|
/** |
|
* Intermediate buffer for interpred leaf level functions |
|
*/ |
|
WORD32 ai16_pred1[HP_BUFF_WD * HP_BUFF_HT]; |
|
|
|
/** |
|
* Reference picture for the current picture |
|
* TODO: Only 2 reference assumed currently |
|
*/ |
|
pic_buf_t *aps_ref_pic[MAX_REF_PIC_CNT]; |
|
|
|
/** |
|
* Reference MV buff for the current picture |
|
*/ |
|
mv_buf_t *aps_mv_buf[MAX_REF_PIC_CNT]; |
|
|
|
/** |
|
* frame info used by RC |
|
*/ |
|
frame_info_t s_frame_info; |
|
|
|
/* |
|
* NOTE NOT PERSISTANT INSIDE FUNCTIONS |
|
* Min sad for current MB |
|
* will be populated initially |
|
* Once a sad less than eq to u4_min_sad is reached, the value will be copied to the cariable |
|
*/ |
|
UWORD32 u4_min_sad; |
|
|
|
/* |
|
* indicates weather we have rached minimum sa or not |
|
*/ |
|
UWORD32 u4_min_sad_reached; |
|
|
|
/** |
|
* Current error code |
|
*/ |
|
WORD32 i4_error_code; |
|
|
|
/* |
|
* Enables or disables computation of recon |
|
*/ |
|
UWORD32 u4_compute_recon; |
|
|
|
/* |
|
* Temporary buffers to be used for subpel computation |
|
*/ |
|
UWORD8 *apu1_subpel_buffs[SUBPEL_BUFF_CNT]; |
|
|
|
/* |
|
* Buffer holding best sub pel values |
|
*/ |
|
UWORD8 *pu1_best_subpel_buf; |
|
|
|
/* |
|
* Stride for buffer holding best sub pel |
|
*/ |
|
UWORD32 u4_bst_spel_buf_strd; |
|
|
|
}; |
|
|
|
/** |
|
****************************************************************************** |
|
* @brief Rate control related variables |
|
****************************************************************************** |
|
*/ |
|
typedef struct |
|
{ |
|
void *pps_rate_control_api; |
|
|
|
void *pps_frame_time; |
|
|
|
void *pps_time_stamp; |
|
|
|
void *pps_pd_frm_rate; |
|
|
|
/** |
|
* frame rate pull down |
|
*/ |
|
WORD32 pre_encode_skip[MAX_CTXT_SETS]; |
|
|
|
/** |
|
* skip frame (cbr) |
|
*/ |
|
WORD32 post_encode_skip[MAX_CTXT_SETS]; |
|
|
|
/** |
|
* rate control type |
|
*/ |
|
rc_type_e e_rc_type; |
|
|
|
/** |
|
* pic type |
|
*/ |
|
picture_type_e e_pic_type; |
|
|
|
/** |
|
* intra cnt in previous frame |
|
*/ |
|
WORD32 num_intra_in_prev_frame; |
|
|
|
/** |
|
* avg activity of prev frame |
|
*/ |
|
WORD32 i4_avg_activity; |
|
|
|
}rate_control_ctxt_t; |
|
|
|
/** |
|
* Codec context |
|
*/ |
|
struct _codec_t |
|
{ |
|
/** |
|
* Id of current pic (input order) |
|
*/ |
|
WORD32 i4_poc; |
|
|
|
/** |
|
* Number of encode frame API calls made |
|
* This variable must only be used for context selection [Read only] |
|
*/ |
|
WORD32 i4_encode_api_call_cnt; |
|
|
|
/** |
|
* Number of pictures encoded |
|
*/ |
|
WORD32 i4_pic_cnt; |
|
|
|
/** |
|
* Number of threads created |
|
*/ |
|
WORD32 i4_proc_thread_cnt; |
|
|
|
/** |
|
* Mutex used to keep the control calls thread-safe |
|
*/ |
|
void *pv_ctl_mutex; |
|
|
|
/** |
|
* Current active config parameters |
|
*/ |
|
cfg_params_t s_cfg; |
|
|
|
/** |
|
* Array containing the config parameter sets |
|
*/ |
|
cfg_params_t as_cfg[MAX_ACTIVE_CONFIG_PARAMS]; |
|
|
|
/** |
|
* Color format used by encoder internally |
|
*/ |
|
IV_COLOR_FORMAT_T e_codec_color_format; |
|
|
|
/** |
|
* recon stride |
|
* (strides for luma and chroma are the same) |
|
*/ |
|
WORD32 i4_rec_strd; |
|
|
|
/** |
|
* Flag to enable/disable deblocking of a frame |
|
*/ |
|
WORD32 i4_disable_deblk_pic; |
|
|
|
/** |
|
* Number of continuous frames where deblocking was disabled |
|
*/ |
|
WORD32 i4_disable_deblk_pic_cnt; |
|
|
|
/** |
|
* frame type |
|
*/ |
|
PIC_TYPE_T pic_type; |
|
|
|
/** |
|
* frame qp |
|
*/ |
|
UWORD32 u4_frame_qp; |
|
|
|
/** |
|
* frame num |
|
*/ |
|
WORD32 i4_frame_num; |
|
|
|
/** |
|
* slice_type |
|
*/ |
|
WORD32 i4_slice_type; |
|
|
|
/* |
|
* Force current frame to specific type |
|
*/ |
|
IV_PICTURE_CODING_TYPE_T force_curr_frame_type; |
|
|
|
/** |
|
* IDR pic |
|
*/ |
|
UWORD32 u4_is_idr; |
|
|
|
/** |
|
* idr_pic_id |
|
*/ |
|
WORD32 i4_idr_pic_id; |
|
|
|
/** |
|
* Flush mode |
|
*/ |
|
WORD32 i4_flush_mode; |
|
|
|
/** |
|
* Encode header mode |
|
*/ |
|
WORD32 i4_header_mode; |
|
|
|
/** |
|
* Flag to indicate if header has already |
|
* been generated when i4_api_call_cnt 0 |
|
*/ |
|
UWORD32 u4_header_generated; |
|
|
|
/** |
|
* Encode generate header |
|
*/ |
|
WORD32 i4_gen_header; |
|
|
|
/** |
|
* To signal successful completion of init |
|
*/ |
|
WORD32 i4_init_done; |
|
|
|
/** |
|
* To signal that at least one picture was decoded |
|
*/ |
|
WORD32 i4_first_pic_done; |
|
|
|
/** |
|
* Reset flag - Codec is reset if this flag is set |
|
*/ |
|
WORD32 i4_reset_flag; |
|
|
|
/** |
|
* Current error code |
|
*/ |
|
WORD32 i4_error_code; |
|
|
|
/** |
|
* threshold residue |
|
*/ |
|
WORD32 u4_thres_resi; |
|
|
|
/** |
|
* disable intra inter gating |
|
*/ |
|
UWORD32 u4_inter_gate; |
|
|
|
/** |
|
* Holds mem records passed during init. |
|
* This will be used to return the mem records during retrieve call |
|
*/ |
|
iv_mem_rec_t *ps_mem_rec_backup; |
|
|
|
/** |
|
* Flag to determine if the entropy thread is active |
|
*/ |
|
volatile UWORD32 au4_entropy_thread_active[MAX_CTXT_SETS]; |
|
|
|
/** |
|
* Mutex used to keep the entropy calls thread-safe |
|
*/ |
|
void *pv_entropy_mutex; |
|
|
|
/** |
|
* Job queue buffer base |
|
*/ |
|
void *pv_proc_jobq_buf, *pv_entropy_jobq_buf; |
|
|
|
/** |
|
* Job Queue mem tab size |
|
*/ |
|
WORD32 i4_proc_jobq_buf_size, i4_entropy_jobq_buf_size; |
|
|
|
/** |
|
* Memory for MV Bank buffer manager |
|
*/ |
|
void *pv_mv_buf_mgr_base; |
|
|
|
/** |
|
* MV Bank buffer manager |
|
*/ |
|
void *pv_mv_buf_mgr; |
|
|
|
/** |
|
* Pointer to MV Buf structure array |
|
*/ |
|
void *ps_mv_buf; |
|
|
|
/** |
|
* Base address for Motion Vector bank buffer |
|
*/ |
|
void *pv_mv_bank_buf_base; |
|
|
|
/** |
|
* MV Bank size allocated |
|
*/ |
|
WORD32 i4_total_mv_bank_size; |
|
|
|
/** |
|
* Memory for Picture buffer manager for reference pictures |
|
*/ |
|
void *pv_ref_buf_mgr_base; |
|
|
|
/** |
|
* Picture buffer manager for reference pictures |
|
*/ |
|
void *pv_ref_buf_mgr; |
|
|
|
/** |
|
* Number of reference buffers added to the buffer manager |
|
*/ |
|
WORD32 i4_ref_buf_cnt; |
|
|
|
/** |
|
* Pointer to Pic Buf structure array |
|
*/ |
|
void *ps_pic_buf; |
|
|
|
/** |
|
* Base address for Picture buffer |
|
*/ |
|
void *pv_pic_buf_base; |
|
|
|
/** |
|
* Total pic buffer size allocated |
|
*/ |
|
WORD32 i4_total_pic_buf_size; |
|
|
|
/** |
|
* Memory for Buffer manager for output buffers |
|
*/ |
|
void *pv_out_buf_mgr_base; |
|
|
|
/** |
|
* Buffer manager for output buffers |
|
*/ |
|
void *pv_out_buf_mgr; |
|
|
|
/** |
|
* Current output buffer's buffer ID |
|
*/ |
|
WORD32 i4_out_buf_id; |
|
|
|
/** |
|
* Number of output buffers added to the buffer manager |
|
*/ |
|
WORD32 i4_out_buf_cnt; |
|
|
|
/** |
|
* Memory for Picture buffer manager for input buffers |
|
*/ |
|
void *pv_inp_buf_mgr_base; |
|
|
|
/** |
|
* Picture buffer manager for input buffers |
|
*/ |
|
void *pv_inp_buf_mgr; |
|
|
|
/** |
|
* Current input buffer's buffer ID |
|
*/ |
|
WORD32 i4_inp_buf_id; |
|
|
|
/** |
|
* Number of input buffers added to the buffer manager |
|
*/ |
|
WORD32 i4_inp_buf_cnt; |
|
|
|
/** |
|
* Current input buffer |
|
*/ |
|
pic_buf_t *ps_inp_buf; |
|
|
|
/** |
|
* Pointer to dpb manager structure |
|
*/ |
|
void *pv_dpb_mgr; |
|
|
|
/** |
|
* Pointer to base of Sequence parameter set structure array |
|
*/ |
|
sps_t *ps_sps_base; |
|
|
|
/** |
|
* Pointer to base of Picture parameter set structure array |
|
*/ |
|
pps_t *ps_pps_base; |
|
|
|
/** |
|
* seq_parameter_set_id |
|
*/ |
|
WORD32 i4_sps_id; |
|
|
|
/** |
|
* pic_parameter_set_id |
|
*/ |
|
WORD32 i4_pps_id; |
|
|
|
/** |
|
* Pointer to base of slice header structure array |
|
*/ |
|
slice_header_t *ps_slice_hdr_base; |
|
|
|
/** |
|
* packed residue coeff data size for 1 row of mbs |
|
*/ |
|
UWORD32 u4_size_coeff_data; |
|
|
|
/** |
|
* packed header data size for 1 row of mbs |
|
*/ |
|
UWORD32 u4_size_header_data; |
|
|
|
/** |
|
* Processing context - One for each processing thread |
|
* Create two sets, each set used for alternate frames |
|
*/ |
|
process_ctxt_t as_process[MAX_PROCESS_CTXT]; |
|
|
|
/** |
|
* Thread handle for each of the processing threads |
|
*/ |
|
void *apv_proc_thread_handle[MAX_PROCESS_THREADS]; |
|
|
|
/** |
|
* Thread created flag for each of the processing threads |
|
*/ |
|
WORD32 ai4_process_thread_created[MAX_PROCESS_THREADS]; |
|
|
|
/** |
|
* Void pointer to process job context |
|
*/ |
|
void *pv_proc_jobq, *pv_entropy_jobq; |
|
|
|
/** |
|
* Number of MBs processed together for better instruction cache handling |
|
*/ |
|
WORD32 i4_proc_nmb; |
|
|
|
/** |
|
* Previous POC lsb |
|
*/ |
|
WORD32 i4_prev_poc_lsb; |
|
|
|
/** |
|
* Previous POC msb |
|
*/ |
|
WORD32 i4_prev_poc_msb; |
|
|
|
/** |
|
* Max POC lsb that has arrived till now |
|
*/ |
|
WORD32 i4_max_prev_poc_lsb; |
|
|
|
/** |
|
* Context for format conversion |
|
*/ |
|
fmt_conv_t s_fmt_conv; |
|
|
|
/** |
|
* Absolute pic order count |
|
*/ |
|
WORD32 i4_abs_pic_order_cnt; |
|
|
|
/** |
|
* Pic order count of lsb |
|
*/ |
|
WORD32 i4_pic_order_cnt_lsb; |
|
|
|
/** |
|
* Array giving current picture being processed in each context set |
|
*/ |
|
WORD32 ai4_pic_cnt[MAX_CTXT_SETS]; |
|
|
|
/* |
|
* Min sad to search for |
|
*/ |
|
UWORD32 u4_min_sad; |
|
|
|
/** |
|
* Reference picture set |
|
*/ |
|
ref_set_t as_ref_set[MAX_DPB_SIZE + MAX_CTXT_SETS]; |
|
|
|
|
|
/* |
|
* Air pic cnt |
|
* Contains the number of pictures that have been encoded with air |
|
* This value is moudulo air refresh period |
|
*/ |
|
WORD32 i4_air_pic_cnt; |
|
|
|
/* |
|
* Intra refresh map |
|
* Stores the frames at which intra refresh should occur for a MB |
|
*/ |
|
UWORD16 *pu2_intr_rfrsh_map; |
|
|
|
/* |
|
* Indicates if the current frame is used as a reference frame |
|
*/ |
|
UWORD32 u4_is_curr_frm_ref; |
|
|
|
/* |
|
* Indicates if there can be non reference frames in the stream |
|
*/ |
|
WORD32 i4_non_ref_frames_in_stream; |
|
|
|
/* |
|
* Memory for color space conversion for luma plane |
|
*/ |
|
UWORD8 *pu1_y_csc_buf_base; |
|
|
|
/* |
|
* Memory for color space conversion foe chroma plane |
|
*/ |
|
UWORD8 *pu1_uv_csc_buf_base; |
|
|
|
/** |
|
* Function pointers for intra pred leaf level functions luma |
|
*/ |
|
pf_intra_pred apf_intra_pred_16_l[MAX_I16x16]; |
|
pf_intra_pred apf_intra_pred_8_l[MAX_I8x8]; |
|
pf_intra_pred apf_intra_pred_4_l[MAX_I4x4]; |
|
|
|
/** |
|
* Function pointers for intra pred leaf level functions chroma |
|
*/ |
|
pf_intra_pred apf_intra_pred_c[MAX_CH_I8x8]; |
|
|
|
/** |
|
* luma core coding function pointer |
|
*/ |
|
UWORD8 (*luma_energy_compaction[4])(process_ctxt_t *ps_proc); |
|
|
|
/** |
|
* chroma core coding function pointer |
|
*/ |
|
UWORD8 (*chroma_energy_compaction[2])(process_ctxt_t *ps_proc); |
|
|
|
/** |
|
* forward transform for intra blk of mb type 16x16 |
|
*/ |
|
ih264_luma_16x16_resi_trans_dctrans_quant_ft *pf_resi_trans_dctrans_quant_16x16; |
|
|
|
/** |
|
* inverse transform for intra blk of mb type 16x16 |
|
*/ |
|
ih264_luma_16x16_idctrans_iquant_itrans_recon_ft *pf_idctrans_iquant_itrans_recon_16x16; |
|
|
|
/** |
|
* forward transform for 4x4 blk luma |
|
*/ |
|
ih264_resi_trans_quant_ft *pf_resi_trans_quant_4x4; |
|
|
|
/** |
|
* forward transform for 4x4 blk luma |
|
*/ |
|
ih264_resi_trans_quant_ft *pf_resi_trans_quant_chroma_4x4; |
|
|
|
/* |
|
* hadamard transform and quant for a 4x4 block |
|
*/ |
|
ih264_hadamard_quant_ft *pf_hadamard_quant_4x4; |
|
|
|
/* |
|
* hadamard transform and quant for a 4x4 block |
|
*/ |
|
ih264_hadamard_quant_ft *pf_hadamard_quant_2x2_uv; |
|
|
|
/** |
|
* inverse transform for 4x4 blk |
|
*/ |
|
ih264_iquant_itrans_recon_ft *pf_iquant_itrans_recon_4x4; |
|
|
|
/** |
|
* inverse transform for chroma 4x4 blk |
|
*/ |
|
ih264_iquant_itrans_recon_chroma_ft *pf_iquant_itrans_recon_chroma_4x4; |
|
|
|
/** |
|
* inverse transform for 4x4 blk with only single dc coeff |
|
*/ |
|
ih264_iquant_itrans_recon_ft *pf_iquant_itrans_recon_4x4_dc; |
|
|
|
/** |
|
* inverse transform for chroma 4x4 blk with only single dc coeff |
|
*/ |
|
ih264_iquant_itrans_recon_chroma_ft *pf_iquant_itrans_recon_chroma_4x4_dc; |
|
|
|
/* |
|
* Inverse hadamard transform and iquant for a 4x4 block |
|
*/ |
|
ih264_ihadamard_scaling_ft *pf_ihadamard_scaling_4x4; |
|
|
|
/* |
|
* Inverse hadamard transform and iquant for a 4x4 block |
|
*/ |
|
ih264_ihadamard_scaling_ft *pf_ihadamard_scaling_2x2_uv; |
|
|
|
/* |
|
* Function for interleave copy* |
|
*/ |
|
ih264_interleave_copy_ft *pf_interleave_copy; |
|
|
|
/** |
|
* forward transform for 8x8 blk |
|
*/ |
|
ih264_resi_trans_quant_ft *pf_resi_trans_quant_8x8; |
|
|
|
/** |
|
* inverse transform for 8x8 blk |
|
*/ |
|
/** |
|
* inverse transform for 4x4 blk |
|
*/ |
|
ih264_iquant_itrans_recon_ft *pf_iquant_itrans_recon_8x8; |
|
|
|
/** |
|
* forward transform for chroma MB |
|
*/ |
|
ih264_chroma_8x8_resi_trans_dctrans_quant_ft *pf_resi_trans_dctrans_quant_8x8_chroma; |
|
|
|
/** |
|
* inverse transform for chroma MB |
|
*/ |
|
ih264_idctrans_iquant_itrans_recon_ft *pf_idctrans_iquant_itrans_recon_8x8_chroma; |
|
|
|
/** |
|
* deblock vertical luma edge with blocking strength 4 |
|
*/ |
|
ih264_deblk_edge_bs4_ft *pf_deblk_luma_vert_bs4; |
|
|
|
/** |
|
* deblock vertical chroma edge with blocking strength 4 |
|
*/ |
|
ih264_deblk_chroma_edge_bs4_ft *pf_deblk_chroma_vert_bs4; |
|
|
|
/** |
|
* deblock vertical luma edge with blocking strength less than 4 |
|
*/ |
|
ih264_deblk_edge_bslt4_ft *pf_deblk_luma_vert_bslt4; |
|
|
|
/** |
|
* deblock vertical chroma edge with blocking strength less than 4 |
|
*/ |
|
ih264_deblk_chroma_edge_bslt4_ft *pf_deblk_chroma_vert_bslt4; |
|
|
|
/** |
|
* deblock horizontal luma edge with blocking strength 4 |
|
*/ |
|
ih264_deblk_edge_bs4_ft *pf_deblk_luma_horz_bs4; |
|
|
|
/** |
|
* deblock horizontal chroma edge with blocking strength 4 |
|
*/ |
|
ih264_deblk_chroma_edge_bs4_ft *pf_deblk_chroma_horz_bs4; |
|
|
|
/** |
|
* deblock horizontal luma edge with blocking strength less than 4 |
|
*/ |
|
ih264_deblk_edge_bslt4_ft *pf_deblk_luma_horz_bslt4; |
|
|
|
/** |
|
* deblock horizontal chroma edge with blocking strength less than 4 |
|
*/ |
|
ih264_deblk_chroma_edge_bslt4_ft *pf_deblk_chroma_horz_bslt4; |
|
|
|
|
|
/** |
|
* functions for padding |
|
*/ |
|
pf_pad pf_pad_top; |
|
pf_pad pf_pad_bottom; |
|
pf_pad pf_pad_left_luma; |
|
pf_pad pf_pad_left_chroma; |
|
pf_pad pf_pad_right_luma; |
|
pf_pad pf_pad_right_chroma; |
|
|
|
/** |
|
* Inter pred leaf level functions |
|
*/ |
|
ih264_inter_pred_luma_ft *pf_inter_pred_luma_copy; |
|
ih264_inter_pred_luma_ft *pf_inter_pred_luma_horz; |
|
ih264_inter_pred_luma_ft *pf_inter_pred_luma_vert; |
|
pf_inter_pred_luma_bilinear pf_inter_pred_luma_bilinear; |
|
ih264_inter_pred_chroma_ft *pf_inter_pred_chroma; |
|
|
|
/** |
|
* fn ptrs for compute sad routines |
|
*/ |
|
ime_compute_sad_ft *apf_compute_sad_16x16[2]; |
|
ime_compute_sad_ft *pf_compute_sad_16x8; |
|
|
|
|
|
/** |
|
* Function pointer for computing ME |
|
* 1 for PSLICE and 1 for BSLICE |
|
*/ |
|
ih264e_compute_me_ft *apf_compute_me[2]; |
|
|
|
/** |
|
* Function pointers for computing SKIP parameters |
|
*/ |
|
ih264e_skip_params_ft *apf_find_skip_params_me[2]; |
|
|
|
/** |
|
* fn ptrs for memory handling operations |
|
*/ |
|
pf_memcpy pf_mem_cpy; |
|
pf_memset pf_mem_set; |
|
pf_memcpy_mul8 pf_mem_cpy_mul8; |
|
pf_memset_mul8 pf_mem_set_mul8; |
|
|
|
/** |
|
* intra mode eval -encoder level function |
|
*/ |
|
pf_evaluate_intra_modes pf_ih264e_evaluate_intra16x16_modes; |
|
pf_evaluate_intra_modes pf_ih264e_evaluate_intra_chroma_modes; |
|
pf_evaluate_intra_4x4_modes pf_ih264e_evaluate_intra_4x4_modes; |
|
|
|
/* Half pel generation function - encoder level |
|
* |
|
*/ |
|
pf_sixtapfilter_horz pf_ih264e_sixtapfilter_horz; |
|
pf_sixtap_filter_2dvh_vert pf_ih264e_sixtap_filter_2dvh_vert; |
|
|
|
/** |
|
* color space conversion form YUV 420P to YUV 420Sp |
|
*/ |
|
pf_fmt_conv_420p_to_420sp pf_ih264e_conv_420p_to_420sp; |
|
|
|
|
|
/** |
|
* color space conversion form YUV 420P to YUV 420Sp |
|
*/ |
|
pf_fmt_conv_422ile_to_420sp pf_ih264e_fmt_conv_422i_to_420sp; |
|
|
|
/** |
|
* write mb layer for a given slice I, P, B |
|
*/ |
|
IH264E_ERROR_T (*pf_write_mb_syntax_layer[2][3]) ( entropy_ctxt_t *ps_ent_ctxt ); |
|
|
|
/** |
|
* Output buffer |
|
*/ |
|
out_buf_t as_out_buf[MAX_CTXT_SETS]; |
|
|
|
/** |
|
* recon buffer |
|
*/ |
|
rec_buf_t as_rec_buf[MAX_CTXT_SETS]; |
|
|
|
/** |
|
* rate control context |
|
*/ |
|
rate_control_ctxt_t s_rate_control; |
|
|
|
/** |
|
* input buffer queue |
|
*/ |
|
inp_buf_t as_inp_list[MAX_NUM_BFRAMES]; |
|
|
|
/** |
|
* Flag to indicate if any IDR requests are pending |
|
*/ |
|
WORD32 i4_pending_idr_flag; |
|
|
|
/* |
|
*Flag to indicate if we have recived the last input frame |
|
*/ |
|
WORD32 i4_last_inp_buff_received; |
|
|
|
}; |
|
|
|
#endif /* IH264E_STRUCTS_H_ */
|
|
|