/******************************************************************************
|
*
|
* 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_inter_pred.c
|
*
|
* @brief
|
* Contains funtions for giving out prediction samples for a given pu
|
*
|
* @author
|
* Ittiam
|
*
|
* @par List of Functions:
|
* - ihevc_inter_pred()
|
*
|
*
|
*******************************************************************************
|
*/
|
/* System include files */
|
#include <stdio.h>
|
#include <string.h>
|
#include <stdlib.h>
|
#include <assert.h>
|
#include <stdarg.h>
|
#include <math.h>
|
|
/* User include files */
|
#include "ihevc_typedefs.h"
|
#include "itt_video_api.h"
|
#include "ihevce_api.h"
|
|
#include "rc_cntrl_param.h"
|
#include "rc_frame_info_collector.h"
|
#include "rc_look_ahead_params.h"
|
|
#include "ihevc_debug.h"
|
#include "ihevc_defs.h"
|
#include "ihevc_structs.h"
|
#include "ihevc_platform_macros.h"
|
#include "ihevc_deblk.h"
|
#include "ihevc_itrans_recon.h"
|
#include "ihevc_chroma_itrans_recon.h"
|
#include "ihevc_chroma_intra_pred.h"
|
#include "ihevc_intra_pred.h"
|
#include "ihevc_inter_pred.h"
|
#include "ihevc_mem_fns.h"
|
#include "ihevc_padding.h"
|
#include "ihevc_weighted_pred.h"
|
#include "ihevc_sao.h"
|
#include "ihevc_resi_trans.h"
|
#include "ihevc_quant_iquant_ssd.h"
|
#include "ihevc_cabac_tables.h"
|
|
#include "ihevce_defs.h"
|
#include "ihevce_lap_enc_structs.h"
|
#include "ihevce_multi_thrd_structs.h"
|
#include "ihevce_me_common_defs.h"
|
#include "ihevce_had_satd.h"
|
#include "ihevce_error_codes.h"
|
#include "ihevce_bitstream.h"
|
#include "ihevce_cabac.h"
|
#include "ihevce_rdoq_macros.h"
|
#include "ihevce_function_selector.h"
|
#include "ihevce_enc_structs.h"
|
#include "ihevce_entropy_structs.h"
|
#include "ihevce_cmn_utils_instr_set_router.h"
|
#include "ihevce_enc_loop_structs.h"
|
#include "ihevce_inter_pred.h"
|
#include "ihevc_weighted_pred.h"
|
|
/*****************************************************************************/
|
/* Global tables */
|
/*****************************************************************************/
|
|
/**
|
******************************************************************************
|
* @brief Table of filter tap coefficients for HEVC luma inter prediction
|
* input : sub pel mv position (dx/dy = 0:3)
|
* output : filter coeffs to be used for that position
|
*
|
* @remarks See section 8.5.2.2.2.1 Luma sample interpolation process of HEVC
|
******************************************************************************
|
*/
|
WORD8 gai1_hevc_luma_filter_taps[4][NTAPS_LUMA] = { { 0, 0, 0, 64, 0, 0, 0, 0 },
|
{ -1, 4, -10, 58, 17, -5, 1, 0 },
|
{ -1, 4, -11, 40, 40, -11, 4, -1 },
|
{ 0, 1, -5, 17, 58, -10, 4, -1 } };
|
|
/**
|
******************************************************************************
|
* @brief Table of filter tap coefficients for HEVC chroma inter prediction
|
* input : chroma sub pel mv position (dx/dy = 0:7)
|
* output : filter coeffs to be used for that position
|
*
|
* @remarks See section 8.5.2.2.2.2 Chroma sample interpolation process of HEVC
|
The filter uses only the first four elements in each array
|
******************************************************************************
|
*/
|
WORD8 gai1_hevc_chroma_filter_taps[8][NTAPS_CHROMA] = { { 0, 64, 0, 0 }, { -2, 58, 10, -2 },
|
{ -4, 54, 16, -2 }, { -6, 46, 28, -4 },
|
{ -4, 36, 36, -4 }, { -4, 28, 46, -6 },
|
{ -2, 16, 54, -4 }, { -2, 10, 58, -2 } };
|
|
/*****************************************************************************/
|
/* Function Definitions */
|
/*****************************************************************************/
|
|
/**
|
*******************************************************************************
|
*
|
* @brief
|
* Performs Luma inter pred based on sub pel position dxdy and store the result
|
* in a 16 bit destination buffer
|
*
|
* @param[in] pu1_src
|
* pointer to the source correspoding to integer pel position of a mv (left and
|
* top justified integer position)
|
*
|
* @param[out] pi2_dst
|
* WORD16 pointer to the destination
|
*
|
* @param[in] src_strd
|
* source buffer stride
|
*
|
* @param[in] dst_strd
|
* destination buffer stride
|
*
|
* @param[in] pi2_hdst_scratch
|
* scratch buffer for intermediate storage of horizontal filter output; used as
|
* input for vertical filtering when sub pel components (dx != 0) && (dy != 0)
|
*
|
* Max scratch buffer required is w * (h + 7) * sizeof(WORD16)
|
*
|
* @param[in] ht
|
* width of the prediction unit
|
*
|
* @param[in] wd
|
* width of the prediction unit
|
*
|
* @param[in] dx
|
* qpel position[0:3] of mv in x direction
|
*
|
* @param[in] dy
|
* qpel position[0:3] of mv in y direction
|
*
|
* @returns
|
* none
|
*
|
* @remarks
|
*
|
*******************************************************************************
|
*/
|
void ihevce_luma_interpolate_16bit_dxdy(
|
UWORD8 *pu1_src,
|
WORD16 *pi2_dst,
|
WORD32 src_strd,
|
WORD32 dst_strd,
|
WORD16 *pi2_hdst_scratch,
|
WORD32 ht,
|
WORD32 wd,
|
WORD32 dy,
|
WORD32 dx,
|
func_selector_t *ps_func_selector)
|
{
|
if((0 == dx) && (0 == dy))
|
{
|
/*--------- full pel position : copy input by upscaling-------*/
|
|
ps_func_selector->ihevc_inter_pred_luma_copy_w16out_fptr(
|
pu1_src, pi2_dst, src_strd, dst_strd, &gai1_hevc_luma_filter_taps[0][0], ht, wd);
|
}
|
else if((0 != dx) && (0 != dy))
|
{
|
/*----------sub pel in both x and y direction---------*/
|
|
UWORD8 *pu1_horz_src = pu1_src - (3 * src_strd);
|
WORD32 hdst_buf_stride = wd;
|
WORD16 *pi2_vert_src = pi2_hdst_scratch + (3 * hdst_buf_stride);
|
|
/* horizontal filtering of source done in a scratch buffer first */
|
ps_func_selector->ihevc_inter_pred_luma_horz_w16out_fptr(
|
pu1_horz_src,
|
pi2_hdst_scratch,
|
src_strd,
|
hdst_buf_stride,
|
&gai1_hevc_luma_filter_taps[dx][0],
|
(ht + NTAPS_LUMA - 1),
|
wd);
|
|
/* vertical filtering on scratch buffer and stored in desitnation */
|
ps_func_selector->ihevc_inter_pred_luma_vert_w16inp_w16out_fptr(
|
pi2_vert_src,
|
pi2_dst,
|
hdst_buf_stride,
|
dst_strd,
|
&gai1_hevc_luma_filter_taps[dy][0],
|
ht,
|
wd);
|
}
|
else if(0 == dy)
|
{
|
/*----------sub pel in x direction only ---------*/
|
|
ps_func_selector->ihevc_inter_pred_luma_horz_w16out_fptr(
|
pu1_src, pi2_dst, src_strd, dst_strd, &gai1_hevc_luma_filter_taps[dx][0], ht, wd);
|
}
|
else /* if (0 == dx) */
|
{
|
/*----------sub pel in y direction only ---------*/
|
|
ps_func_selector->ihevc_inter_pred_luma_vert_w16out_fptr(
|
pu1_src, pi2_dst, src_strd, dst_strd, &gai1_hevc_luma_filter_taps[dy][0], ht, wd);
|
}
|
}
|
|
/**
|
*******************************************************************************
|
*
|
* @brief
|
* Performs Luma inter pred based on sub pel position dxdy and store the result
|
* in a 8 bit destination buffer
|
*
|
* @param[in] pu1_src
|
* pointer to the source correspoding to integer pel position of a mv (left and
|
* top justified integer position)
|
*
|
* @param[out] pu1_dst
|
* UWORD8 pointer to the destination
|
*
|
* @param[in] src_strd
|
* source buffer stride
|
*
|
* @param[in] dst_strd
|
* destination buffer stride
|
*
|
* @param[in] pi2_hdst_scratch
|
* scratch buffer for intermediate storage of horizontal filter output; used as
|
* input for vertical filtering when sub pel components (dx != 0) && (dy != 0)
|
*
|
* Max scratch buffer required is w * (h + 7) * sizeof(WORD16)
|
*
|
* @param[in] ht
|
* width of the prediction unit
|
*
|
* @param[in] wd
|
* width of the prediction unit
|
*
|
* @param[in] dx
|
* qpel position[0:3] of mv in x direction
|
*
|
* @param[in] dy
|
* qpel position[0:3] of mv in y direction
|
*
|
* @returns
|
* none
|
*
|
* @remarks
|
*
|
*******************************************************************************
|
*/
|
void ihevce_luma_interpolate_8bit_dxdy(
|
UWORD8 *pu1_src,
|
UWORD8 *pu1_dst,
|
WORD32 src_strd,
|
WORD32 dst_strd,
|
WORD16 *pi2_hdst_scratch,
|
WORD32 ht,
|
WORD32 wd,
|
WORD32 dy,
|
WORD32 dx,
|
func_selector_t *ps_func_selector)
|
{
|
if((0 == dx) && (0 == dy))
|
{
|
/*--------- full pel position : copy input as is -------*/
|
|
ps_func_selector->ihevc_inter_pred_luma_copy_fptr(
|
pu1_src, pu1_dst, src_strd, dst_strd, &gai1_hevc_luma_filter_taps[0][0], ht, wd);
|
}
|
else if((0 != dx) && (0 != dy))
|
{
|
/*----------sub pel in both x and y direction---------*/
|
|
UWORD8 *pu1_horz_src = pu1_src - (3 * src_strd);
|
WORD32 hdst_buf_stride = wd;
|
WORD16 *pi2_vert_src = pi2_hdst_scratch + (3 * hdst_buf_stride);
|
|
/* horizontal filtering of source done in a scratch buffer first */
|
ps_func_selector->ihevc_inter_pred_luma_horz_w16out_fptr(
|
pu1_horz_src,
|
pi2_hdst_scratch,
|
src_strd,
|
hdst_buf_stride,
|
&gai1_hevc_luma_filter_taps[dx][0],
|
(ht + NTAPS_LUMA - 1),
|
wd);
|
|
/* vertical filtering on scratch buffer and stored in desitnation */
|
ps_func_selector->ihevc_inter_pred_luma_vert_w16inp_fptr(
|
pi2_vert_src,
|
pu1_dst,
|
hdst_buf_stride,
|
dst_strd,
|
&gai1_hevc_luma_filter_taps[dy][0],
|
ht,
|
wd);
|
}
|
else if(0 == dy)
|
{
|
/*----------sub pel in x direction only ---------*/
|
|
ps_func_selector->ihevc_inter_pred_luma_horz_fptr(
|
pu1_src, pu1_dst, src_strd, dst_strd, &gai1_hevc_luma_filter_taps[dx][0], ht, wd);
|
}
|
else /* if (0 == dx) */
|
{
|
/*----------sub pel in y direction only ---------*/
|
|
ps_func_selector->ihevc_inter_pred_luma_vert_fptr(
|
pu1_src, pu1_dst, src_strd, dst_strd, &gai1_hevc_luma_filter_taps[dy][0], ht, wd);
|
}
|
}
|
|
/**
|
*******************************************************************************
|
*
|
* @brief
|
* Performs Luma prediction for a inter prediction unit(PU)
|
*
|
* @par Description:
|
* For a given PU, Inter prediction followed by weighted prediction (if
|
* required)
|
*
|
* @param[in] ps_inter_pred_ctxt
|
* context for inter prediction; contains ref list, weight offsets, ctb offsets
|
*
|
* @param[in] ps_pu
|
* pointer to PU structure whose inter prediction needs to be done
|
*
|
* @param[in] pu1_dst_buf
|
* pointer to destination buffer where the inter prediction is done
|
*
|
* @param[in] dst_stride
|
* pitch of the destination buffer
|
*
|
* @returns
|
* IV_FAIL for mvs going outside ref frame padded limits
|
* IV_SUCCESS after completing mc for given inter pu
|
*
|
* @remarks
|
*
|
*******************************************************************************
|
*/
|
IV_API_CALL_STATUS_T ihevce_luma_inter_pred_pu(
|
void *pv_inter_pred_ctxt,
|
pu_t *ps_pu,
|
void *pv_dst_buf,
|
WORD32 dst_stride,
|
WORD32 i4_flag_inter_pred_source)
|
{
|
inter_pred_ctxt_t *ps_inter_pred_ctxt = (inter_pred_ctxt_t *)pv_inter_pred_ctxt;
|
func_selector_t *ps_func_selector = ps_inter_pred_ctxt->ps_func_selector;
|
|
WORD32 inter_pred_idc = ps_pu->b2_pred_mode;
|
UWORD8 *pu1_dst_buf = (UWORD8 *)pv_dst_buf;
|
WORD32 pu_wd = (ps_pu->b4_wd + 1) << 2;
|
WORD32 pu_ht = (ps_pu->b4_ht + 1) << 2;
|
|
WORD32 wp_flag = ps_inter_pred_ctxt->i1_weighted_pred_flag ||
|
ps_inter_pred_ctxt->i1_weighted_bipred_flag;
|
|
/* 16bit dest required for interpolate if weighted pred is on or bipred */
|
WORD32 store_16bit_output;
|
|
recon_pic_buf_t *ps_ref_pic_l0, *ps_ref_pic_l1;
|
UWORD8 *pu1_ref_pic, *pu1_ref_int_pel;
|
WORD32 ref_pic_stride;
|
|
/* offset of reference block in integer pel units */
|
WORD32 frm_x_ofst, frm_y_ofst;
|
WORD32 frm_x_pu, frm_y_pu;
|
|
/* scratch 16 bit buffers for interpolation in l0 and l1 direction */
|
WORD16 *pi2_scr_buf_l0 = &ps_inter_pred_ctxt->ai2_scratch_buf_l0[0];
|
WORD16 *pi2_scr_buf_l1 = &ps_inter_pred_ctxt->ai2_scratch_buf_l1[0];
|
|
/* scratch buffer for horizontal interpolation destination */
|
WORD16 *pi2_horz_scratch = &ps_inter_pred_ctxt->ai2_horz_scratch[0];
|
|
WORD32 wgt0, wgt1, off0, off1, shift, lvl_shift0, lvl_shift1;
|
|
/* get PU's frm x and frm y offset */
|
frm_x_pu = ps_inter_pred_ctxt->i4_ctb_frm_pos_x + (ps_pu->b4_pos_x << 2);
|
frm_y_pu = ps_inter_pred_ctxt->i4_ctb_frm_pos_y + (ps_pu->b4_pos_y << 2);
|
|
/* sanity checks */
|
ASSERT((wp_flag == 0) || (wp_flag == 1));
|
ASSERT(dst_stride >= pu_wd);
|
ASSERT(ps_pu->b1_intra_flag == 0);
|
|
lvl_shift0 = 0;
|
lvl_shift1 = 0;
|
|
if(wp_flag)
|
{
|
UWORD8 u1_is_wgt_pred_L0, u1_is_wgt_pred_L1;
|
|
if(inter_pred_idc != PRED_L1)
|
{
|
ps_ref_pic_l0 = ps_inter_pred_ctxt->ps_ref_list[0][ps_pu->mv.i1_l0_ref_idx];
|
u1_is_wgt_pred_L0 = ps_ref_pic_l0->s_weight_offset.u1_luma_weight_enable_flag;
|
}
|
if(inter_pred_idc != PRED_L0)
|
{
|
ps_ref_pic_l1 = ps_inter_pred_ctxt->ps_ref_list[1][ps_pu->mv.i1_l1_ref_idx];
|
u1_is_wgt_pred_L1 = ps_ref_pic_l1->s_weight_offset.u1_luma_weight_enable_flag;
|
}
|
if(inter_pred_idc == PRED_BI)
|
{
|
wp_flag = (u1_is_wgt_pred_L0 || u1_is_wgt_pred_L1);
|
}
|
else if(inter_pred_idc == PRED_L0)
|
{
|
wp_flag = u1_is_wgt_pred_L0;
|
}
|
else if(inter_pred_idc == PRED_L1)
|
{
|
wp_flag = u1_is_wgt_pred_L1;
|
}
|
else
|
{
|
/*other values are not allowed*/
|
assert(0);
|
}
|
}
|
store_16bit_output = (inter_pred_idc == PRED_BI) || (wp_flag);
|
|
if(inter_pred_idc != PRED_L1)
|
{
|
/*****************************************************/
|
/* L0 inter prediction */
|
/*****************************************************/
|
|
/* motion vecs in qpel precision */
|
WORD32 mv_x = ps_pu->mv.s_l0_mv.i2_mvx;
|
WORD32 mv_y = ps_pu->mv.s_l0_mv.i2_mvy;
|
|
/* sub pel offsets in x and y direction w.r.t integer pel */
|
WORD32 dx = mv_x & 0x3;
|
WORD32 dy = mv_y & 0x3;
|
|
/* ref idx is currently stored in the lower 4bits */
|
WORD32 ref_idx = (ps_pu->mv.i1_l0_ref_idx);
|
|
/* x and y integer offsets w.r.t frame start */
|
frm_x_ofst = (frm_x_pu + (mv_x >> 2));
|
frm_y_ofst = (frm_y_pu + (mv_y >> 2));
|
|
ps_ref_pic_l0 = ps_inter_pred_ctxt->ps_ref_list[0][ref_idx];
|
|
/* picture buffer start and stride */
|
if(i4_flag_inter_pred_source == 1)
|
{
|
pu1_ref_pic = (UWORD8 *)ps_ref_pic_l0->s_yuv_buf_desc_src.pv_y_buf;
|
}
|
else
|
{
|
pu1_ref_pic = (UWORD8 *)ps_ref_pic_l0->s_yuv_buf_desc.pv_y_buf;
|
}
|
ref_pic_stride = ps_ref_pic_l0->s_yuv_buf_desc.i4_y_strd;
|
|
/* Error check for mvs going out of ref frame padded limits */
|
{
|
WORD32 min_x, max_x = ps_ref_pic_l0->s_yuv_buf_desc.i4_y_wd;
|
WORD32 min_y, max_y = ps_ref_pic_l0->s_yuv_buf_desc.i4_y_ht;
|
|
min_x =
|
-(ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_LEFT]
|
? (ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_LEFT] - 4)
|
: (PAD_HORZ - 4));
|
|
max_x += ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_RIGHT]
|
? (ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_RIGHT] - 4)
|
: (PAD_HORZ - 4);
|
|
min_y =
|
-(ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_TOP]
|
? (ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_TOP] - 4)
|
: (PAD_VERT - 4));
|
|
max_y += ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_BOT]
|
? (ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_BOT] - 4)
|
: (PAD_VERT - 4);
|
|
if((frm_x_ofst < min_x) || (frm_x_ofst + pu_wd) > max_x)
|
//ASSERT(0);
|
return (IV_FAIL);
|
|
if((frm_y_ofst < min_y) || (frm_y_ofst + pu_ht) > max_y)
|
//ASSERT(0);
|
return (IV_FAIL);
|
}
|
|
/* point to reference start location in ref frame */
|
/* Assuming clipping of mv is not required here as ME would */
|
/* take care of mv access not going beyond padded data */
|
pu1_ref_int_pel = pu1_ref_pic + frm_x_ofst + (ref_pic_stride * frm_y_ofst);
|
|
/* level shifted for subpel with both x and y componenet being non 0 */
|
/* this is because the interpolate function subtract this to contain */
|
/* the resulting data in 16 bits */
|
lvl_shift0 = (dx != 0) && (dy != 0) ? OFFSET14 : 0;
|
|
if(store_16bit_output)
|
{
|
/* do interpolation in 16bit L0 scratch buffer */
|
ihevce_luma_interpolate_16bit_dxdy(
|
pu1_ref_int_pel,
|
pi2_scr_buf_l0,
|
ref_pic_stride,
|
pu_wd,
|
pi2_horz_scratch,
|
pu_ht,
|
pu_wd,
|
dy,
|
dx,
|
ps_func_selector);
|
}
|
else
|
{
|
/* do interpolation in 8bit destination buffer and return */
|
ihevce_luma_interpolate_8bit_dxdy(
|
pu1_ref_int_pel,
|
pu1_dst_buf,
|
ref_pic_stride,
|
dst_stride,
|
pi2_horz_scratch,
|
pu_ht,
|
pu_wd,
|
dy,
|
dx,
|
ps_func_selector);
|
|
return (IV_SUCCESS);
|
}
|
}
|
|
if(inter_pred_idc != PRED_L0)
|
{
|
/*****************************************************/
|
/* L1 inter prediction */
|
/*****************************************************/
|
|
/* motion vecs in qpel precision */
|
WORD32 mv_x = ps_pu->mv.s_l1_mv.i2_mvx;
|
WORD32 mv_y = ps_pu->mv.s_l1_mv.i2_mvy;
|
|
/* sub pel offsets in x and y direction w.r.t integer pel */
|
WORD32 dx = mv_x & 0x3;
|
WORD32 dy = mv_y & 0x3;
|
|
/* ref idx is currently stored in the lower 4bits */
|
WORD32 ref_idx = (ps_pu->mv.i1_l1_ref_idx);
|
|
/* x and y integer offsets w.r.t frame start */
|
frm_x_ofst = (frm_x_pu + (mv_x >> 2));
|
frm_y_ofst = (frm_y_pu + (mv_y >> 2));
|
|
ps_ref_pic_l1 = ps_inter_pred_ctxt->ps_ref_list[1][ref_idx];
|
|
/* picture buffer start and stride */
|
|
if(i4_flag_inter_pred_source == 1)
|
{
|
pu1_ref_pic = (UWORD8 *)ps_ref_pic_l1->s_yuv_buf_desc_src.pv_y_buf;
|
}
|
else
|
{
|
pu1_ref_pic = (UWORD8 *)ps_ref_pic_l1->s_yuv_buf_desc.pv_y_buf;
|
}
|
ref_pic_stride = ps_ref_pic_l1->s_yuv_buf_desc.i4_y_strd;
|
|
/* Error check for mvs going out of ref frame padded limits */
|
{
|
WORD32 min_x, max_x = ps_ref_pic_l1->s_yuv_buf_desc.i4_y_wd;
|
WORD32 min_y, max_y = ps_ref_pic_l1->s_yuv_buf_desc.i4_y_ht;
|
|
min_x =
|
-(ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_LEFT]
|
? (ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_LEFT] - 4)
|
: (PAD_HORZ - 4));
|
|
max_x += ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_RIGHT]
|
? (ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_RIGHT] - 4)
|
: (PAD_HORZ - 4);
|
|
min_y =
|
-(ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_TOP]
|
? (ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_TOP] - 4)
|
: (PAD_VERT - 4));
|
|
max_y += ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_BOT]
|
? (ps_inter_pred_ctxt->ai4_tile_xtra_pel[E_BOT] - 4)
|
: (PAD_VERT - 4);
|
|
if((frm_x_ofst < min_x) || (frm_x_ofst + pu_wd) > max_x)
|
//ASSERT(0);
|
return (IV_FAIL);
|
|
if((frm_y_ofst < min_y) || (frm_y_ofst + pu_ht) > max_y)
|
//ASSERT(0);
|
return (IV_FAIL);
|
}
|
|
/* point to reference start location in ref frame */
|
/* Assuming clipping of mv is not required here as ME would */
|
/* take care of mv access not going beyond padded data */
|
pu1_ref_int_pel = pu1_ref_pic + frm_x_ofst + (ref_pic_stride * frm_y_ofst);
|
|
/* level shifted for subpel with both x and y componenet being non 0 */
|
/* this is because the interpolate function subtract this to contain */
|
/* the resulting data in 16 bits */
|
lvl_shift1 = (dx != 0) && (dy != 0) ? OFFSET14 : 0;
|
|
if(store_16bit_output)
|
{
|
/* do interpolation in 16bit L1 scratch buffer */
|
ihevce_luma_interpolate_16bit_dxdy(
|
pu1_ref_int_pel,
|
pi2_scr_buf_l1,
|
ref_pic_stride,
|
pu_wd,
|
pi2_horz_scratch,
|
pu_ht,
|
pu_wd,
|
dy,
|
dx,
|
ps_func_selector);
|
}
|
else
|
{
|
/* do interpolation in 8bit destination buffer and return */
|
ihevce_luma_interpolate_8bit_dxdy(
|
pu1_ref_int_pel,
|
pu1_dst_buf,
|
ref_pic_stride,
|
dst_stride,
|
pi2_horz_scratch,
|
pu_ht,
|
pu_wd,
|
dy,
|
dx,
|
ps_func_selector);
|
|
return (IV_SUCCESS);
|
}
|
}
|
|
if((inter_pred_idc != PRED_BI) && wp_flag)
|
{
|
/*****************************************************/
|
/* unidirection weighted prediction */
|
/*****************************************************/
|
ihevce_wght_offst_t *ps_weight_offset;
|
WORD16 *pi2_src;
|
WORD32 lvl_shift;
|
|
/* intialize the weight, offsets and ref based on l0/l1 mode */
|
if(inter_pred_idc == PRED_L0)
|
{
|
pi2_src = pi2_scr_buf_l0;
|
ps_weight_offset = &ps_ref_pic_l0->s_weight_offset;
|
lvl_shift = lvl_shift0;
|
}
|
else
|
{
|
pi2_src = pi2_scr_buf_l1;
|
ps_weight_offset = &ps_ref_pic_l1->s_weight_offset;
|
lvl_shift = lvl_shift1;
|
}
|
|
wgt0 = ps_weight_offset->i2_luma_weight;
|
off0 = ps_weight_offset->i2_luma_offset;
|
shift = ps_inter_pred_ctxt->i4_log2_luma_wght_denom + SHIFT_14_MINUS_BIT_DEPTH;
|
|
/* do the uni directional weighted prediction */
|
ps_func_selector->ihevc_weighted_pred_uni_fptr(
|
pi2_src, pu1_dst_buf, pu_wd, dst_stride, wgt0, off0, shift, lvl_shift, pu_ht, pu_wd);
|
}
|
else
|
{
|
/*****************************************************/
|
/* Bipred prediction */
|
/*****************************************************/
|
|
if(wp_flag)
|
{
|
/*****************************************************/
|
/* Bi pred weighted prediction */
|
/*****************************************************/
|
wgt0 = ps_ref_pic_l0->s_weight_offset.i2_luma_weight;
|
off0 = ps_ref_pic_l0->s_weight_offset.i2_luma_offset;
|
|
wgt1 = ps_ref_pic_l1->s_weight_offset.i2_luma_weight;
|
off1 = ps_ref_pic_l1->s_weight_offset.i2_luma_offset;
|
|
shift = ps_inter_pred_ctxt->i4_log2_luma_wght_denom + SHIFT_14_MINUS_BIT_DEPTH + 1;
|
|
ps_func_selector->ihevc_weighted_pred_bi_fptr(
|
pi2_scr_buf_l0,
|
pi2_scr_buf_l1,
|
pu1_dst_buf,
|
pu_wd,
|
pu_wd,
|
dst_stride,
|
wgt0,
|
off0,
|
wgt1,
|
off1,
|
shift,
|
lvl_shift0,
|
lvl_shift1,
|
pu_ht,
|
pu_wd);
|
}
|
else
|
{
|
/*****************************************************/
|
/* Default Bi pred prediction */
|
/*****************************************************/
|
ps_func_selector->ihevc_weighted_pred_bi_default_fptr(
|
pi2_scr_buf_l0,
|
pi2_scr_buf_l1,
|
pu1_dst_buf,
|
pu_wd,
|
pu_wd,
|
dst_stride,
|
lvl_shift0,
|
lvl_shift1,
|
pu_ht,
|
pu_wd);
|
}
|
}
|
|
return (IV_SUCCESS);
|
}
|
|
/**
|
*******************************************************************************
|
*
|
* @brief
|
* Performs Chroma inter pred based on sub pel position dxdy and store the
|
* result in a 16 bit destination buffer
|
*
|
* @param[in] pu1_src
|
* pointer to the source correspoding to integer pel position of a mv (left and
|
* top justified integer position)
|
*
|
* @param[out] pi2_dst
|
* WORD16 pointer to the destination
|
*
|
* @param[in] src_strd
|
* source buffer stride
|
*
|
* @param[in] dst_strd
|
* destination buffer stride
|
*
|
* @param[in] pi2_hdst_scratch
|
* scratch buffer for intermediate storage of horizontal filter output; used as
|
* input for vertical filtering when sub pel components (dx != 0) && (dy != 0)
|
*
|
* Max scratch buffer required is w * (h + 3) * sizeof(WORD16)
|
*
|
* @param[in] ht
|
* width of the prediction unit
|
*
|
* @param[in] wd
|
* width of the prediction unit
|
*
|
* @param[in] dx
|
* 1/8th pel position[0:7] of mv in x direction
|
*
|
* @param[in] dy
|
* 1/8th pel position[0:7] of mv in y direction
|
*
|
* @returns
|
* none
|
*
|
* @remarks
|
*
|
*******************************************************************************
|
*/
|
void ihevce_chroma_interpolate_16bit_dxdy(
|
UWORD8 *pu1_src,
|
WORD16 *pi2_dst,
|
WORD32 src_strd,
|
WORD32 dst_strd,
|
WORD16 *pi2_hdst_scratch,
|
WORD32 ht,
|
WORD32 wd,
|
WORD32 dy,
|
WORD32 dx,
|
func_selector_t *ps_func_selector)
|
{
|
if((0 == dx) && (0 == dy))
|
{
|
/*--------- full pel position : copy input by upscaling-------*/
|
|
ps_func_selector->ihevc_inter_pred_chroma_copy_w16out_fptr(
|
pu1_src, pi2_dst, src_strd, dst_strd, &gai1_hevc_chroma_filter_taps[0][0], ht, wd);
|
}
|
else if((0 != dx) && (0 != dy))
|
{
|
/*----------sub pel in both x and y direction---------*/
|
|
UWORD8 *pu1_horz_src = pu1_src - src_strd;
|
WORD32 hdst_buf_stride = (wd << 1); /* uv interleave */
|
WORD16 *pi2_vert_src = pi2_hdst_scratch + hdst_buf_stride;
|
|
/* horizontal filtering of source done in a scratch buffer first */
|
ps_func_selector->ihevc_inter_pred_chroma_horz_w16out_fptr(
|
pu1_horz_src,
|
pi2_hdst_scratch,
|
src_strd,
|
hdst_buf_stride,
|
&gai1_hevc_chroma_filter_taps[dx][0],
|
(ht + NTAPS_CHROMA - 1),
|
wd);
|
|
/* vertical filtering on scratch buffer and stored in desitnation */
|
ps_func_selector->ihevc_inter_pred_chroma_vert_w16inp_w16out_fptr(
|
pi2_vert_src,
|
pi2_dst,
|
hdst_buf_stride,
|
dst_strd,
|
&gai1_hevc_chroma_filter_taps[dy][0],
|
ht,
|
wd);
|
}
|
else if(0 == dy)
|
{
|
/*----------sub pel in x direction only ---------*/
|
|
ps_func_selector->ihevc_inter_pred_chroma_horz_w16out_fptr(
|
pu1_src, pi2_dst, src_strd, dst_strd, &gai1_hevc_chroma_filter_taps[dx][0], ht, wd);
|
}
|
else /* if (0 == dx) */
|
{
|
/*----------sub pel in y direction only ---------*/
|
|
ps_func_selector->ihevc_inter_pred_chroma_vert_w16out_fptr(
|
pu1_src, pi2_dst, src_strd, dst_strd, &gai1_hevc_chroma_filter_taps[dy][0], ht, wd);
|
}
|
}
|
|
/**
|
*******************************************************************************
|
*
|
* @brief
|
* Performs Chroma inter pred based on sub pel position dxdy and store the
|
* result in a 8 bit destination buffer
|
*
|
* @param[in] pu1_src
|
* pointer to the source correspoding to integer pel position of a mv (left and
|
* top justified integer position)
|
*
|
* @param[out] pu1_dst
|
* UWORD8 pointer to the destination
|
*
|
* @param[in] src_strd
|
* source buffer stride
|
*
|
* @param[in] dst_strd
|
* destination buffer stride
|
*
|
* @param[in] pi2_hdst_scratch
|
* scratch buffer for intermediate storage of horizontal filter output; used as
|
* input for vertical filtering when sub pel components (dx != 0) && (dy != 0)
|
*
|
* Max scratch buffer required is w * (h + 3) * sizeof(WORD16)
|
*
|
* @param[in] ht
|
* width of the prediction unit
|
*
|
* @param[in] wd
|
* width of the prediction unit
|
*
|
* @param[in] dx
|
* 1/8th pel position[0:7] of mv in x direction
|
*
|
* @param[in] dy
|
* 1/8th pel position[0:7] of mv in y direction
|
*
|
* @returns
|
* none
|
*
|
* @remarks
|
*
|
*******************************************************************************
|
*/
|
void ihevce_chroma_interpolate_8bit_dxdy(
|
UWORD8 *pu1_src,
|
UWORD8 *pu1_dst,
|
WORD32 src_strd,
|
WORD32 dst_strd,
|
WORD16 *pi2_hdst_scratch,
|
WORD32 ht,
|
WORD32 wd,
|
WORD32 dy,
|
WORD32 dx,
|
func_selector_t *ps_func_selector)
|
{
|
if((0 == dx) && (0 == dy))
|
{
|
/*--------- full pel position : copy input as is -------*/
|
ps_func_selector->ihevc_inter_pred_chroma_copy_fptr(
|
pu1_src, pu1_dst, src_strd, dst_strd, &gai1_hevc_chroma_filter_taps[0][0], ht, wd);
|
}
|
else if((0 != dx) && (0 != dy))
|
{
|
/*----------sub pel in both x and y direction---------*/
|
UWORD8 *pu1_horz_src = pu1_src - src_strd;
|
WORD32 hdst_buf_stride = (wd << 1); /* uv interleave */
|
WORD16 *pi2_vert_src = pi2_hdst_scratch + hdst_buf_stride;
|
|
/* horizontal filtering of source done in a scratch buffer first */
|
ps_func_selector->ihevc_inter_pred_chroma_horz_w16out_fptr(
|
pu1_horz_src,
|
pi2_hdst_scratch,
|
src_strd,
|
hdst_buf_stride,
|
&gai1_hevc_chroma_filter_taps[dx][0],
|
(ht + NTAPS_CHROMA - 1),
|
wd);
|
|
/* vertical filtering on scratch buffer and stored in desitnation */
|
ps_func_selector->ihevc_inter_pred_chroma_vert_w16inp_fptr(
|
pi2_vert_src,
|
pu1_dst,
|
hdst_buf_stride,
|
dst_strd,
|
&gai1_hevc_chroma_filter_taps[dy][0],
|
ht,
|
wd);
|
}
|
else if(0 == dy)
|
{
|
/*----------sub pel in x direction only ---------*/
|
ps_func_selector->ihevc_inter_pred_chroma_horz_fptr(
|
pu1_src, pu1_dst, src_strd, dst_strd, &gai1_hevc_chroma_filter_taps[dx][0], ht, wd);
|
}
|
else /* if (0 == dx) */
|
{
|
/*----------sub pel in y direction only ---------*/
|
ps_func_selector->ihevc_inter_pred_chroma_vert_fptr(
|
pu1_src, pu1_dst, src_strd, dst_strd, &gai1_hevc_chroma_filter_taps[dy][0], ht, wd);
|
}
|
}
|
|
/**
|
*******************************************************************************
|
*
|
* @brief
|
* Performs Chroma prediction for a inter prediction unit(PU)
|
*
|
* @par Description:
|
* For a given PU, Inter prediction followed by weighted prediction (if
|
* required). The reference and destination buffers are uv interleaved
|
*
|
* @param[in] ps_inter_pred_ctxt
|
* context for inter prediction; contains ref list, weight offsets, ctb offsets
|
*
|
* @param[in] ps_pu
|
* pointer to PU structure whose inter prediction needs to be done
|
*
|
* @param[in] pu1_dst_buf
|
* pointer to destination buffer where the inter prediction is done
|
*
|
* @param[in] dst_stride
|
* pitch of the destination buffer
|
*
|
* @returns
|
* none
|
*
|
* @remarks
|
*
|
*******************************************************************************
|
*/
|
void ihevce_chroma_inter_pred_pu(
|
void *pv_inter_pred_ctxt, pu_t *ps_pu, UWORD8 *pu1_dst_buf, WORD32 dst_stride)
|
{
|
inter_pred_ctxt_t *ps_inter_pred_ctxt = (inter_pred_ctxt_t *)pv_inter_pred_ctxt;
|
func_selector_t *ps_func_selector = ps_inter_pred_ctxt->ps_func_selector;
|
|
WORD32 inter_pred_idc = ps_pu->b2_pred_mode;
|
UWORD8 u1_is_422 = (ps_inter_pred_ctxt->u1_chroma_array_type == 2);
|
/* chroma width and height are half of luma width and height */
|
WORD32 pu_wd_chroma = (ps_pu->b4_wd + 1) << 1;
|
WORD32 pu_ht_chroma = (ps_pu->b4_ht + 1) << (u1_is_422 + 1);
|
|
WORD32 wp_flag = ps_inter_pred_ctxt->i1_weighted_pred_flag ||
|
ps_inter_pred_ctxt->i1_weighted_bipred_flag;
|
|
/* 16bit dest required for interpolate if weighted pred is on or bipred */
|
WORD32 store_16bit_output;
|
|
recon_pic_buf_t *ps_ref_pic_l0, *ps_ref_pic_l1;
|
UWORD8 *pu1_ref_pic, *pu1_ref_int_pel;
|
WORD32 ref_pic_stride;
|
|
/* offset of reference block in integer pel units */
|
WORD32 frm_x_ofst, frm_y_ofst;
|
WORD32 frm_x_pu, frm_y_pu;
|
|
/* scratch 16 bit buffers for interpolation in l0 and l1 direction */
|
WORD16 *pi2_scr_buf_l0 = &ps_inter_pred_ctxt->ai2_scratch_buf_l0[0];
|
WORD16 *pi2_scr_buf_l1 = &ps_inter_pred_ctxt->ai2_scratch_buf_l1[0];
|
|
/* scratch buffer for horizontal interpolation destination */
|
WORD16 *pi2_horz_scratch = &ps_inter_pred_ctxt->ai2_horz_scratch[0];
|
|
/* get PU's frm x and frm y offset : Note uv is interleaved */
|
frm_x_pu = ps_inter_pred_ctxt->i4_ctb_frm_pos_x + (ps_pu->b4_pos_x << 2);
|
frm_y_pu = (ps_inter_pred_ctxt->i4_ctb_frm_pos_y >> (u1_is_422 == 0)) +
|
(ps_pu->b4_pos_y << (u1_is_422 + 1));
|
|
/* sanity checks */
|
ASSERT((wp_flag == 0) || (wp_flag == 1));
|
ASSERT(dst_stride >= (pu_wd_chroma << 1)); /* uv interleaved */
|
ASSERT(ps_pu->b1_intra_flag == 0);
|
|
if(wp_flag)
|
{
|
UWORD8 u1_is_wgt_pred_L0, u1_is_wgt_pred_L1;
|
|
if(inter_pred_idc != PRED_L1)
|
{
|
ps_ref_pic_l0 = ps_inter_pred_ctxt->ps_ref_list[0][ps_pu->mv.i1_l0_ref_idx];
|
u1_is_wgt_pred_L0 = ps_ref_pic_l0->s_weight_offset.u1_chroma_weight_enable_flag;
|
}
|
if(inter_pred_idc != PRED_L0)
|
{
|
ps_ref_pic_l1 = ps_inter_pred_ctxt->ps_ref_list[1][ps_pu->mv.i1_l1_ref_idx];
|
u1_is_wgt_pred_L1 = ps_ref_pic_l1->s_weight_offset.u1_chroma_weight_enable_flag;
|
}
|
if(inter_pred_idc == PRED_BI)
|
{
|
wp_flag = (u1_is_wgt_pred_L0 || u1_is_wgt_pred_L1);
|
}
|
else if(inter_pred_idc == PRED_L0)
|
{
|
wp_flag = u1_is_wgt_pred_L0;
|
}
|
else if(inter_pred_idc == PRED_L1)
|
{
|
wp_flag = u1_is_wgt_pred_L1;
|
}
|
else
|
{
|
/*other values are not allowed*/
|
assert(0);
|
}
|
}
|
store_16bit_output = (inter_pred_idc == PRED_BI) || (wp_flag);
|
|
if(inter_pred_idc != PRED_L1)
|
{
|
/*****************************************************/
|
/* L0 inter prediction(Chroma ) */
|
/*****************************************************/
|
|
/* motion vecs in qpel precision */
|
WORD32 mv_x = ps_pu->mv.s_l0_mv.i2_mvx;
|
WORD32 mv_y = ps_pu->mv.s_l0_mv.i2_mvy;
|
|
/* sub pel offsets in x and y direction w.r.t integer pel */
|
WORD32 dx = mv_x & 0x7;
|
WORD32 dy = (mv_y & ((1 << (!u1_is_422 + 2)) - 1)) << u1_is_422;
|
|
/* ref idx is currently stored in the lower 4bits */
|
WORD32 ref_idx = (ps_pu->mv.i1_l0_ref_idx);
|
|
/* x and y integer offsets w.r.t frame start */
|
|
frm_x_ofst = (frm_x_pu + ((mv_x >> 3) << 1)); /* uv interleaved */
|
frm_y_ofst = (frm_y_pu + ((mv_y >> (3 - u1_is_422))));
|
|
ps_ref_pic_l0 = ps_inter_pred_ctxt->ps_ref_list[0][ref_idx];
|
|
/* picture buffer start and stride */
|
pu1_ref_pic = (UWORD8 *)ps_ref_pic_l0->s_yuv_buf_desc.pv_u_buf;
|
ref_pic_stride = ps_ref_pic_l0->s_yuv_buf_desc.i4_uv_strd;
|
|
/* point to reference start location in ref frame */
|
/* Assuming clipping of mv is not required here as ME would */
|
/* take care of mv access not going beyond padded data */
|
pu1_ref_int_pel = pu1_ref_pic + frm_x_ofst + (ref_pic_stride * frm_y_ofst);
|
|
if(store_16bit_output)
|
{
|
/* do interpolation in 16bit L0 scratch buffer */
|
ihevce_chroma_interpolate_16bit_dxdy(
|
pu1_ref_int_pel,
|
pi2_scr_buf_l0,
|
ref_pic_stride,
|
(pu_wd_chroma << 1),
|
pi2_horz_scratch,
|
pu_ht_chroma,
|
pu_wd_chroma,
|
dy,
|
dx,
|
ps_func_selector);
|
}
|
else
|
{
|
/* do interpolation in 8bit destination buffer and return */
|
ihevce_chroma_interpolate_8bit_dxdy(
|
pu1_ref_int_pel,
|
pu1_dst_buf,
|
ref_pic_stride,
|
dst_stride,
|
pi2_horz_scratch,
|
pu_ht_chroma,
|
pu_wd_chroma,
|
dy,
|
dx,
|
ps_func_selector);
|
|
return;
|
}
|
}
|
|
if(inter_pred_idc != PRED_L0)
|
{
|
/*****************************************************/
|
/* L1 inter prediction(Chroma) */
|
/*****************************************************/
|
|
/* motion vecs in qpel precision */
|
WORD32 mv_x = ps_pu->mv.s_l1_mv.i2_mvx;
|
WORD32 mv_y = ps_pu->mv.s_l1_mv.i2_mvy;
|
|
/* sub pel offsets in x and y direction w.r.t integer pel */
|
WORD32 dx = mv_x & 0x7;
|
WORD32 dy = (mv_y & ((1 << (!u1_is_422 + 2)) - 1)) << u1_is_422;
|
|
/* ref idx is currently stored in the lower 4bits */
|
WORD32 ref_idx = (ps_pu->mv.i1_l1_ref_idx);
|
|
/* x and y integer offsets w.r.t frame start */
|
frm_x_ofst = (frm_x_pu + ((mv_x >> 3) << 1)); /* uv interleaved */
|
frm_y_ofst = (frm_y_pu + ((mv_y >> (3 - u1_is_422))));
|
|
ps_ref_pic_l1 = ps_inter_pred_ctxt->ps_ref_list[1][ref_idx];
|
|
/* picture buffer start and stride */
|
pu1_ref_pic = (UWORD8 *)ps_ref_pic_l1->s_yuv_buf_desc.pv_u_buf;
|
ref_pic_stride = ps_ref_pic_l1->s_yuv_buf_desc.i4_uv_strd;
|
|
/* point to reference start location in ref frame */
|
/* Assuming clipping of mv is not required here as ME would */
|
/* take care of mv access not going beyond padded data */
|
pu1_ref_int_pel = pu1_ref_pic + frm_x_ofst + (ref_pic_stride * frm_y_ofst);
|
|
if(store_16bit_output)
|
{
|
/* do interpolation in 16bit L1 scratch buffer */
|
ihevce_chroma_interpolate_16bit_dxdy(
|
pu1_ref_int_pel,
|
pi2_scr_buf_l1,
|
ref_pic_stride,
|
(pu_wd_chroma << 1),
|
pi2_horz_scratch,
|
pu_ht_chroma,
|
pu_wd_chroma,
|
dy,
|
dx,
|
ps_func_selector);
|
}
|
else
|
{
|
/* do interpolation in 8bit destination buffer and return */
|
ihevce_chroma_interpolate_8bit_dxdy(
|
pu1_ref_int_pel,
|
pu1_dst_buf,
|
ref_pic_stride,
|
dst_stride,
|
pi2_horz_scratch,
|
pu_ht_chroma,
|
pu_wd_chroma,
|
dy,
|
dx,
|
ps_func_selector);
|
|
return;
|
}
|
}
|
|
if((inter_pred_idc != PRED_BI) && wp_flag)
|
{
|
/*****************************************************/
|
/* unidirection weighted prediction(Chroma) */
|
/*****************************************************/
|
ihevce_wght_offst_t *ps_weight_offset;
|
WORD16 *pi2_src;
|
WORD32 lvl_shift = 0;
|
WORD32 wgt_cb, wgt_cr, off_cb, off_cr;
|
WORD32 shift;
|
|
/* intialize the weight, offsets and ref based on l0/l1 mode */
|
if(inter_pred_idc == PRED_L0)
|
{
|
pi2_src = pi2_scr_buf_l0;
|
ps_weight_offset = &ps_ref_pic_l0->s_weight_offset;
|
}
|
else
|
{
|
pi2_src = pi2_scr_buf_l1;
|
ps_weight_offset = &ps_ref_pic_l1->s_weight_offset;
|
}
|
|
wgt_cb = ps_weight_offset->i2_cb_weight;
|
off_cb = ps_weight_offset->i2_cb_offset;
|
wgt_cr = ps_weight_offset->i2_cr_weight;
|
off_cr = ps_weight_offset->i2_cr_offset;
|
|
shift = ps_inter_pred_ctxt->i4_log2_chroma_wght_denom + SHIFT_14_MINUS_BIT_DEPTH;
|
|
/* do the uni directional weighted prediction */
|
ps_func_selector->ihevc_weighted_pred_chroma_uni_fptr(
|
pi2_src,
|
pu1_dst_buf,
|
(pu_wd_chroma << 1),
|
dst_stride,
|
wgt_cb,
|
wgt_cr,
|
off_cb,
|
off_cr,
|
shift,
|
lvl_shift,
|
pu_ht_chroma,
|
pu_wd_chroma);
|
}
|
else
|
{
|
/*****************************************************/
|
/* Bipred prediction(Chroma) */
|
/*****************************************************/
|
if(wp_flag)
|
{
|
WORD32 wgt0_cb, wgt1_cb, wgt0_cr, wgt1_cr;
|
WORD32 off0_cb, off1_cb, off0_cr, off1_cr;
|
WORD32 shift;
|
|
/*****************************************************/
|
/* Bi pred weighted prediction (Chroma) */
|
/*****************************************************/
|
wgt0_cb = ps_ref_pic_l0->s_weight_offset.i2_cb_weight;
|
off0_cb = ps_ref_pic_l0->s_weight_offset.i2_cb_offset;
|
|
wgt0_cr = ps_ref_pic_l0->s_weight_offset.i2_cr_weight;
|
off0_cr = ps_ref_pic_l0->s_weight_offset.i2_cr_offset;
|
|
wgt1_cb = ps_ref_pic_l1->s_weight_offset.i2_cb_weight;
|
off1_cb = ps_ref_pic_l1->s_weight_offset.i2_cb_offset;
|
|
wgt1_cr = ps_ref_pic_l1->s_weight_offset.i2_cr_weight;
|
off1_cr = ps_ref_pic_l1->s_weight_offset.i2_cr_offset;
|
|
shift = ps_inter_pred_ctxt->i4_log2_chroma_wght_denom + SHIFT_14_MINUS_BIT_DEPTH + 1;
|
|
ps_func_selector->ihevc_weighted_pred_chroma_bi_fptr(
|
pi2_scr_buf_l0,
|
pi2_scr_buf_l1,
|
pu1_dst_buf,
|
(pu_wd_chroma << 1),
|
(pu_wd_chroma << 1),
|
dst_stride,
|
wgt0_cb,
|
wgt0_cr,
|
off0_cb,
|
off0_cr,
|
wgt1_cb,
|
wgt1_cr,
|
off1_cb,
|
off1_cr,
|
shift,
|
0,
|
0,
|
pu_ht_chroma,
|
pu_wd_chroma);
|
}
|
else
|
{
|
/*****************************************************/
|
/* Default Bi pred prediction (Chroma) */
|
/*****************************************************/
|
ps_func_selector->ihevc_weighted_pred_chroma_bi_default_fptr(
|
pi2_scr_buf_l0,
|
pi2_scr_buf_l1,
|
pu1_dst_buf,
|
(pu_wd_chroma << 1),
|
(pu_wd_chroma << 1),
|
dst_stride,
|
0,
|
0,
|
pu_ht_chroma,
|
pu_wd_chroma);
|
}
|
}
|
}
|
