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.
751 lines
27 KiB
751 lines
27 KiB
/******************************************************************************
|
|
*
|
|
* Copyright (C) 2012 Ittiam Systems Pvt Ltd, Bangalore
|
|
*
|
|
* 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.
|
|
*
|
|
******************************************************************************/
|
|
/**
|
|
*******************************************************************************
|
|
* @file
|
|
* ihevcd_mv_pred.c
|
|
*
|
|
* @brief
|
|
* Contains functions for motion vector prediction
|
|
*
|
|
* @author
|
|
* Ittiam
|
|
*
|
|
* @par List of Functions:
|
|
* - ihevcd_scale_mv()
|
|
* - ihevcd_mv_pred()
|
|
*
|
|
* @remarks
|
|
* None
|
|
*
|
|
*******************************************************************************
|
|
*/
|
|
/*****************************************************************************/
|
|
/* File Includes */
|
|
/*****************************************************************************/
|
|
|
|
#include <stdio.h>
|
|
#include <stddef.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "ihevc_typedefs.h"
|
|
#include "iv.h"
|
|
#include "ivd.h"
|
|
#include "ihevcd_cxa.h"
|
|
#include "ithread.h"
|
|
|
|
#include "ihevc_defs.h"
|
|
#include "ihevc_debug.h"
|
|
#include "ihevc_structs.h"
|
|
#include "ihevc_macros.h"
|
|
#include "ihevc_platform_macros.h"
|
|
#include "ihevc_cabac_tables.h"
|
|
#include "ihevc_disp_mgr.h"
|
|
#include "ihevc_buf_mgr.h"
|
|
#include "ihevc_dpb_mgr.h"
|
|
|
|
#include "ihevcd_defs.h"
|
|
#include "ihevcd_function_selector.h"
|
|
#include "ihevcd_structs.h"
|
|
#include "ihevcd_error.h"
|
|
#include "ihevcd_nal.h"
|
|
#include "ihevcd_bitstream.h"
|
|
#include "ihevcd_fmt_conv.h"
|
|
#include "ihevcd_job_queue.h"
|
|
#include "ihevcd_debug.h"
|
|
#include "ihevcd_mv_merge.h"
|
|
|
|
/**
|
|
*******************************************************************************
|
|
*
|
|
* @brief Function scaling motion vector
|
|
*
|
|
*
|
|
* @par Description:
|
|
* Scales mv based on difference between current POC and current
|
|
* reference POC and neighbour reference poc
|
|
*
|
|
* @param[inout] mv
|
|
* motion vector to be scaled
|
|
*
|
|
* @param[in] cur_ref_poc
|
|
* Current PU refernce pic poc
|
|
*
|
|
* @param[in] nbr_ref_poc
|
|
* Neighbor PU reference pic poc
|
|
*
|
|
* @param[in] cur_poc
|
|
* Picture order count of current pic
|
|
*
|
|
* @returns
|
|
* None
|
|
*
|
|
* @remarks
|
|
*
|
|
*******************************************************************************
|
|
*/
|
|
void ihevcd_scale_mv(mv_t *ps_mv,
|
|
WORD32 cur_ref_poc,
|
|
WORD32 nbr_ref_poc,
|
|
WORD32 cur_poc)
|
|
{
|
|
WORD32 td, tb, tx;
|
|
WORD32 dist_scale_factor;
|
|
WORD32 mvx, mvy;
|
|
|
|
td = CLIP_S8(cur_poc - nbr_ref_poc);
|
|
tb = CLIP_S8(cur_poc - cur_ref_poc);
|
|
|
|
if(0 != td)
|
|
{
|
|
tx = (16384 + (abs(td) >> 1)) / td;
|
|
|
|
dist_scale_factor = (tb * tx + 32) >> 6;
|
|
dist_scale_factor = CLIP3(dist_scale_factor, -4096, 4095);
|
|
|
|
mvx = ps_mv->i2_mvx;
|
|
mvy = ps_mv->i2_mvy;
|
|
|
|
mvx = SIGN(dist_scale_factor * mvx)
|
|
* ((abs(dist_scale_factor * mvx) + 127) >> 8);
|
|
mvy = SIGN(dist_scale_factor * mvy)
|
|
* ((abs(dist_scale_factor * mvy) + 127) >> 8);
|
|
|
|
ps_mv->i2_mvx = CLIP_S16(mvx);
|
|
ps_mv->i2_mvy = CLIP_S16(mvy);
|
|
}
|
|
}
|
|
|
|
/**
|
|
*******************************************************************************
|
|
*
|
|
* @brief Function scaling temporal motion vector
|
|
*
|
|
*
|
|
* @par Description:
|
|
* Scales mv based on difference between current POC and current
|
|
* reference POC and neighbour reference poc
|
|
*
|
|
* @param[inout] mv
|
|
* motion vector to be scaled
|
|
*
|
|
* @param[in] cur_ref_poc
|
|
* Current PU refernce pic poc
|
|
*
|
|
* @param[in] nbr_ref_poc
|
|
* Neighbor PU reference pic poc
|
|
*
|
|
* @param[in] cur_poc
|
|
* Picture order count of current pic
|
|
*
|
|
* @returns
|
|
* None
|
|
*
|
|
* @remarks
|
|
*
|
|
*******************************************************************************
|
|
*/
|
|
void ihevcd_scale_collocated_mv(mv_t *ps_mv,
|
|
WORD32 cur_ref_poc,
|
|
WORD32 col_ref_poc,
|
|
WORD32 col_poc,
|
|
WORD32 cur_poc)
|
|
{
|
|
WORD32 td, tb, tx;
|
|
WORD32 dist_scale_factor;
|
|
WORD32 mvx, mvy;
|
|
|
|
td = CLIP_S8(col_poc - col_ref_poc);
|
|
tb = CLIP_S8(cur_poc - cur_ref_poc);
|
|
|
|
tx = (16384 + (abs(td) >> 1)) / td;
|
|
|
|
dist_scale_factor = (tb * tx + 32) >> 6;
|
|
dist_scale_factor = CLIP3(dist_scale_factor, -4096, 4095);
|
|
|
|
mvx = ps_mv->i2_mvx;
|
|
mvy = ps_mv->i2_mvy;
|
|
|
|
mvx = SIGN(dist_scale_factor * mvx)
|
|
* ((abs(dist_scale_factor * mvx) + 127) >> 8);
|
|
mvy = SIGN(dist_scale_factor * mvy)
|
|
* ((abs(dist_scale_factor * mvy) + 127) >> 8);
|
|
|
|
ps_mv->i2_mvx = CLIP_S16(mvx);
|
|
ps_mv->i2_mvy = CLIP_S16(mvy);
|
|
}
|
|
|
|
#define CHECK_NBR_MV_ST(pi4_avail_flag, cur_ref_poc, u1_nbr_pred_flag, nbr_ref_poc, \
|
|
ps_mv, ps_nbr_mv ) \
|
|
{ \
|
|
if((u1_nbr_pred_flag) && (cur_ref_poc == nbr_ref_poc)) \
|
|
{ \
|
|
*pi4_avail_flag = 1; \
|
|
*ps_mv = *ps_nbr_mv; \
|
|
break ; \
|
|
} \
|
|
}
|
|
#define CHECK_NBR_MV_LT(pi4_avail_flag, u1_cur_ref_lt, cur_poc, cur_ref_poc, \
|
|
u1_nbr_pred_flag, u1_nbr_ref_lt, nbr_ref_poc, \
|
|
ps_mv, ps_nbr_mv ) \
|
|
{ \
|
|
WORD32 cur_lt, nbr_lt; \
|
|
cur_lt = (LONG_TERM_REF == (u1_cur_ref_lt)); \
|
|
nbr_lt = (LONG_TERM_REF == (u1_nbr_ref_lt)); \
|
|
if((u1_nbr_pred_flag) && (cur_lt == nbr_lt)) \
|
|
{ \
|
|
*pi4_avail_flag = 1; \
|
|
*ps_mv = *ps_nbr_mv; \
|
|
if(SHORT_TERM_REF == u1_nbr_ref_lt) \
|
|
{ \
|
|
ihevcd_scale_mv(ps_mv, cur_ref_poc, nbr_ref_poc, \
|
|
cur_poc); \
|
|
} \
|
|
break ; \
|
|
} \
|
|
}
|
|
|
|
|
|
void GET_MV_NBR_ST(ref_list_t **ps_ref_pic_list, WORD32 *pi4_avail_flag, pic_buf_t *ps_cur_pic_buf_lx, pu_t **aps_nbr_pu, mv_t *ps_mv, WORD32 num_nbrs, WORD32 lx)
|
|
{
|
|
WORD32 i, nbr_pred_lx;
|
|
pic_buf_t *ps_nbr_pic_buf_lx;
|
|
/* Short Term */
|
|
/* L0 */
|
|
if(0 == lx)
|
|
{
|
|
for(i = 0; i < num_nbrs; i++)
|
|
{
|
|
nbr_pred_lx = (PRED_L1 != aps_nbr_pu[i]->b2_pred_mode);
|
|
ps_nbr_pic_buf_lx = (pic_buf_t *)((ps_ref_pic_list[0][aps_nbr_pu[i]->mv.i1_l0_ref_idx].pv_pic_buf));
|
|
CHECK_NBR_MV_ST(pi4_avail_flag, ps_cur_pic_buf_lx->i4_abs_poc, nbr_pred_lx,
|
|
ps_nbr_pic_buf_lx->i4_abs_poc, ps_mv, &aps_nbr_pu[i]->mv.s_l0_mv);
|
|
nbr_pred_lx = (PRED_L0 != aps_nbr_pu[i]->b2_pred_mode);
|
|
|
|
nbr_pred_lx = (PRED_L0 != aps_nbr_pu[i]->b2_pred_mode);
|
|
ps_nbr_pic_buf_lx = (pic_buf_t *)((ps_ref_pic_list[1][aps_nbr_pu[i]->mv.i1_l1_ref_idx].pv_pic_buf));
|
|
CHECK_NBR_MV_ST(pi4_avail_flag, ps_cur_pic_buf_lx->i4_abs_poc, nbr_pred_lx,
|
|
ps_nbr_pic_buf_lx->i4_abs_poc, ps_mv, &aps_nbr_pu[i]->mv.s_l1_mv);
|
|
}
|
|
}
|
|
/* L1 */
|
|
else
|
|
{
|
|
for(i = 0; i < num_nbrs; i++)
|
|
{
|
|
nbr_pred_lx = (PRED_L0 != aps_nbr_pu[i]->b2_pred_mode);
|
|
ps_nbr_pic_buf_lx = (pic_buf_t *)((ps_ref_pic_list[1][aps_nbr_pu[i]->mv.i1_l1_ref_idx].pv_pic_buf));
|
|
CHECK_NBR_MV_ST(pi4_avail_flag, ps_cur_pic_buf_lx->i4_abs_poc, nbr_pred_lx,
|
|
ps_nbr_pic_buf_lx->i4_abs_poc, ps_mv, &aps_nbr_pu[i]->mv.s_l1_mv);
|
|
|
|
nbr_pred_lx = (PRED_L1 != aps_nbr_pu[i]->b2_pred_mode);
|
|
ps_nbr_pic_buf_lx = (pic_buf_t *)((ps_ref_pic_list[0][aps_nbr_pu[i]->mv.i1_l0_ref_idx].pv_pic_buf));
|
|
CHECK_NBR_MV_ST(pi4_avail_flag, ps_cur_pic_buf_lx->i4_abs_poc, nbr_pred_lx,
|
|
ps_nbr_pic_buf_lx->i4_abs_poc, ps_mv, &aps_nbr_pu[i]->mv.s_l0_mv);
|
|
}
|
|
}
|
|
}
|
|
|
|
void GET_MV_NBR_LT(ref_list_t **ps_ref_pic_list, slice_header_t *ps_slice_hdr, WORD32 *pi4_avail_flag, pic_buf_t *ps_cur_pic_buf_lx, pu_t **aps_nbr_pu, mv_t *ps_mv, WORD32 num_nbrs, WORD32 lx)
|
|
{
|
|
WORD32 i, nbr_pred_lx;
|
|
pic_buf_t *ps_nbr_pic_buf_lx;
|
|
/* Long Term*/
|
|
/* L0 */
|
|
if(0 == lx)
|
|
{
|
|
for(i = 0; i < num_nbrs; i++)
|
|
{
|
|
nbr_pred_lx = (PRED_L1 != aps_nbr_pu[i]->b2_pred_mode);
|
|
ps_nbr_pic_buf_lx = (pic_buf_t *)((ps_ref_pic_list[0][aps_nbr_pu[i]->mv.i1_l0_ref_idx].pv_pic_buf));
|
|
CHECK_NBR_MV_LT(pi4_avail_flag, ps_cur_pic_buf_lx->u1_used_as_ref, ps_slice_hdr->i4_abs_pic_order_cnt, ps_cur_pic_buf_lx->i4_abs_poc,
|
|
nbr_pred_lx,
|
|
ps_nbr_pic_buf_lx->u1_used_as_ref, ps_nbr_pic_buf_lx->i4_abs_poc,
|
|
ps_mv, &aps_nbr_pu[i]->mv.s_l0_mv);
|
|
|
|
nbr_pred_lx = (PRED_L0 != aps_nbr_pu[i]->b2_pred_mode);
|
|
ps_nbr_pic_buf_lx = (pic_buf_t *)((ps_ref_pic_list[1][aps_nbr_pu[i]->mv.i1_l1_ref_idx].pv_pic_buf));
|
|
CHECK_NBR_MV_LT(pi4_avail_flag, ps_cur_pic_buf_lx->u1_used_as_ref, ps_slice_hdr->i4_abs_pic_order_cnt, ps_cur_pic_buf_lx->i4_abs_poc,
|
|
nbr_pred_lx,
|
|
ps_nbr_pic_buf_lx->u1_used_as_ref, ps_nbr_pic_buf_lx->i4_abs_poc,
|
|
ps_mv, &aps_nbr_pu[i]->mv.s_l1_mv);
|
|
}
|
|
}
|
|
/* L1 */
|
|
else
|
|
{
|
|
for(i = 0; i < num_nbrs; i++)
|
|
{
|
|
nbr_pred_lx = (PRED_L0 != aps_nbr_pu[i]->b2_pred_mode);
|
|
ps_nbr_pic_buf_lx = (pic_buf_t *)((ps_ref_pic_list[1][aps_nbr_pu[i]->mv.i1_l1_ref_idx].pv_pic_buf));
|
|
CHECK_NBR_MV_LT(pi4_avail_flag, ps_cur_pic_buf_lx->u1_used_as_ref, ps_slice_hdr->i4_abs_pic_order_cnt, ps_cur_pic_buf_lx->i4_abs_poc,
|
|
nbr_pred_lx,
|
|
ps_nbr_pic_buf_lx->u1_used_as_ref, ps_nbr_pic_buf_lx->i4_abs_poc,
|
|
ps_mv, &aps_nbr_pu[i]->mv.s_l1_mv);
|
|
|
|
nbr_pred_lx = (PRED_L1 != aps_nbr_pu[i]->b2_pred_mode);
|
|
ps_nbr_pic_buf_lx = (pic_buf_t *)((ps_ref_pic_list[0][aps_nbr_pu[i]->mv.i1_l0_ref_idx].pv_pic_buf));
|
|
CHECK_NBR_MV_LT(pi4_avail_flag, ps_cur_pic_buf_lx->u1_used_as_ref, ps_slice_hdr->i4_abs_pic_order_cnt, ps_cur_pic_buf_lx->i4_abs_poc,
|
|
nbr_pred_lx,
|
|
ps_nbr_pic_buf_lx->u1_used_as_ref, ps_nbr_pic_buf_lx->i4_abs_poc,
|
|
ps_mv, &aps_nbr_pu[i]->mv.s_l0_mv);
|
|
}
|
|
}
|
|
}
|
|
/**
|
|
*******************************************************************************
|
|
*
|
|
* @brief
|
|
* This function performs Motion Vector prediction and return a list of mv
|
|
*
|
|
* @par Description:
|
|
* MV predictor list is computed using neighbor mvs and colocated mv
|
|
*
|
|
* @param[in] ps_ctxt
|
|
* pointer to mv predictor context
|
|
*
|
|
* @param[in] ps_top_nbr_4x4
|
|
* pointer to top 4x4 nbr structure
|
|
*
|
|
* @param[in] ps_left_nbr_4x4
|
|
* pointer to left 4x4 nbr structure
|
|
*
|
|
* @param[in] ps_top_left_nbr_4x4
|
|
* pointer to top left 4x4 nbr structure
|
|
*
|
|
* @param[in] left_nbr_4x4_strd
|
|
* left nbr buffer stride in terms of 4x4 units
|
|
*
|
|
* @param[in] ps_avail_flags
|
|
* Neighbor availability flags container
|
|
*
|
|
* @param[in] ps_col_mv
|
|
* Colocated MV pointer
|
|
*
|
|
* @param[in] ps_pu
|
|
* Current Partition PU strucrture pointer
|
|
*
|
|
* @param[inout] ps_pred_mv
|
|
* pointer to store predicted MV list
|
|
*
|
|
* @returns
|
|
* None
|
|
* @remarks
|
|
*
|
|
*
|
|
*******************************************************************************
|
|
*/
|
|
void ihevcd_mv_pred(mv_ctxt_t *ps_mv_ctxt,
|
|
UWORD32 *pu4_top_pu_idx,
|
|
UWORD32 *pu4_left_pu_idx,
|
|
UWORD32 *pu4_top_left_pu_idx,
|
|
WORD32 left_nbr_4x4_strd,
|
|
pu_t *ps_pu,
|
|
WORD32 lb_avail,
|
|
WORD32 l_avail,
|
|
WORD32 tr_avail,
|
|
WORD32 t_avail,
|
|
WORD32 tl_avail,
|
|
pu_mv_t *ps_pred_mv)
|
|
{
|
|
slice_header_t *ps_slice_hdr;
|
|
ref_list_t *ps_ref_pic_list[2];
|
|
pu_t *ps_pic_pu;
|
|
WORD32 max_l0_mvp_cand, max_l1_mvp_cand;
|
|
WORD32 l0_done_flag, l1_done_flag;
|
|
WORD32 num_l0_mvp_cand, num_l1_mvp_cand;
|
|
WORD32 is_scaled_flag_list /* Indicates whether A0 or A1 is available */;
|
|
WORD32 avail_a_flag[2];
|
|
mv_t as_mv_a[2];
|
|
WORD32 part_pos_x;
|
|
WORD32 part_pos_y;
|
|
WORD32 part_wd;
|
|
WORD32 part_ht;
|
|
pic_buf_t *ps_cur_pic_buf_l0, *ps_cur_pic_buf_l1;
|
|
WORD32 nbr_avail[3]; /*[A0/A1] */ /* [B0/B1/B2] */
|
|
pu_t *aps_nbr_pu[3]; /*[A0/A1] */ /* [B0/B1/B2] */
|
|
WORD32 num_nbrs = 0;
|
|
|
|
/*******************************************/
|
|
/* Neighbor location: Graphical indication */
|
|
/* */
|
|
/* B2 _____________B1 B0 */
|
|
/* | | */
|
|
/* | | */
|
|
/* | | */
|
|
/* | PU ht| */
|
|
/* | | */
|
|
/* | | */
|
|
/* A1|______wd_______| */
|
|
/* A0 */
|
|
/* */
|
|
/*******************************************/
|
|
|
|
ps_slice_hdr = ps_mv_ctxt->ps_slice_hdr;
|
|
ps_pic_pu = ps_mv_ctxt->ps_pic_pu;
|
|
max_l0_mvp_cand = ps_pu->b1_l0_mvp_idx + 1;
|
|
max_l1_mvp_cand = ps_pu->b1_l1_mvp_idx + 1;
|
|
num_l0_mvp_cand = 0;
|
|
num_l1_mvp_cand = 0;
|
|
|
|
/* Initializing reference list */
|
|
ps_ref_pic_list[0] = ps_slice_hdr->as_ref_pic_list0;
|
|
ps_ref_pic_list[1] = ps_slice_hdr->as_ref_pic_list1;
|
|
if(PSLICE == ps_slice_hdr->i1_slice_type)
|
|
ps_ref_pic_list[1] = ps_slice_hdr->as_ref_pic_list0;
|
|
|
|
ps_cur_pic_buf_l0 = (pic_buf_t *)((ps_ref_pic_list[0][ps_pu->mv.i1_l0_ref_idx].pv_pic_buf));
|
|
ps_cur_pic_buf_l1 = (pic_buf_t *)((ps_ref_pic_list[1][ps_pu->mv.i1_l1_ref_idx].pv_pic_buf));
|
|
|
|
is_scaled_flag_list = 0;
|
|
|
|
part_pos_x = ps_pu->b4_pos_x << 2;
|
|
part_pos_y = ps_pu->b4_pos_y << 2;
|
|
part_wd = (ps_pu->b4_wd + 1) << 2;
|
|
part_ht = (ps_pu->b4_ht + 1) << 2;
|
|
|
|
/************************************************************/
|
|
/* Calculating of motion vector A from neighbors A0 and A1 */
|
|
/************************************************************/
|
|
{
|
|
nbr_avail[0] = 0;
|
|
nbr_avail[1] = 0;
|
|
|
|
/* Pointers to A0 and A1 */
|
|
{
|
|
WORD32 y_a0, y_a1;
|
|
WORD32 pu_idx_a0, pu_idx_a1;
|
|
|
|
/* TODO: y_a0, y_a1 is coded assuming left nbr pointer starts at PU */
|
|
y_a0 = (part_ht >> 2);
|
|
y_a1 = ((part_ht - 1) >> 2);
|
|
|
|
pu_idx_a0 = *(pu4_left_pu_idx + (y_a0 * left_nbr_4x4_strd));
|
|
pu_idx_a1 = *(pu4_left_pu_idx + (y_a1 * left_nbr_4x4_strd));
|
|
|
|
if(lb_avail && (!ps_pic_pu[pu_idx_a0].b1_intra_flag))
|
|
{
|
|
aps_nbr_pu[num_nbrs] = &ps_pic_pu[pu_idx_a0];
|
|
num_nbrs++;
|
|
nbr_avail[0] = 1;
|
|
}
|
|
if(l_avail && (!ps_pic_pu[pu_idx_a1].b1_intra_flag))
|
|
{
|
|
aps_nbr_pu[num_nbrs] = &ps_pic_pu[pu_idx_a1];
|
|
num_nbrs++;
|
|
nbr_avail[1] = 1;
|
|
}
|
|
}
|
|
/* Setting is scaled flag based on availability of A0 and A1 */
|
|
if((nbr_avail[0] == 1) || (nbr_avail[1]))
|
|
{
|
|
is_scaled_flag_list = 1;
|
|
}
|
|
|
|
avail_a_flag[0] = 0;
|
|
avail_a_flag[1] = 0;
|
|
|
|
/* L0 */
|
|
GET_MV_NBR_ST(ps_ref_pic_list, &avail_a_flag[0], ps_cur_pic_buf_l0, aps_nbr_pu, &as_mv_a[0], num_nbrs, 0);
|
|
if(0 == avail_a_flag[0])
|
|
{
|
|
GET_MV_NBR_LT(ps_ref_pic_list, ps_slice_hdr, &avail_a_flag[0], ps_cur_pic_buf_l0, aps_nbr_pu, &as_mv_a[0], num_nbrs, 0);
|
|
}
|
|
|
|
/* L1 */
|
|
if(PRED_L0 != ps_pu->b2_pred_mode)
|
|
{
|
|
GET_MV_NBR_ST(ps_ref_pic_list, &avail_a_flag[1], ps_cur_pic_buf_l1, aps_nbr_pu, &as_mv_a[1], num_nbrs, 1);
|
|
if(0 == avail_a_flag[1])
|
|
{
|
|
GET_MV_NBR_LT(ps_ref_pic_list, ps_slice_hdr, &avail_a_flag[1], ps_cur_pic_buf_l1, aps_nbr_pu, &as_mv_a[1], num_nbrs, 1);
|
|
}
|
|
}
|
|
|
|
l0_done_flag = (PRED_L1 == ps_pu->b2_pred_mode);
|
|
l1_done_flag = (PRED_L0 == ps_pu->b2_pred_mode);
|
|
|
|
if(avail_a_flag[0])
|
|
{
|
|
num_l0_mvp_cand++;
|
|
if(max_l0_mvp_cand == num_l0_mvp_cand)
|
|
{
|
|
ps_pred_mv->s_l0_mv = as_mv_a[0];
|
|
l0_done_flag = 1;
|
|
}
|
|
}
|
|
if(avail_a_flag[1])
|
|
{
|
|
num_l1_mvp_cand++;
|
|
if(max_l1_mvp_cand == num_l1_mvp_cand)
|
|
{
|
|
ps_pred_mv->s_l1_mv = as_mv_a[1];
|
|
l1_done_flag = 1;
|
|
}
|
|
}
|
|
if(l0_done_flag && l1_done_flag)
|
|
return;
|
|
}
|
|
|
|
/************************************************************/
|
|
/* Calculating of motion vector B from neighbors B0 and B1 */
|
|
/************************************************************/
|
|
{
|
|
WORD32 avail_b_flag[2];
|
|
mv_t as_mv_b[2];
|
|
|
|
/* Pointers to B0, B1 and B2 */
|
|
{
|
|
WORD32 x_b0, x_b1, x_b2;
|
|
WORD32 pu_idx_b0, pu_idx_b1, pu_idx_b2;
|
|
|
|
/* Relative co-ordiante of Xp,Yp w.r.t CTB start will work */
|
|
/* as long as minCTB = 16 */
|
|
x_b0 = (part_pos_x + part_wd);
|
|
x_b1 = (part_pos_x + part_wd - 1);
|
|
x_b2 = (part_pos_x - 1);
|
|
/* Getting offset back to given pointer */
|
|
x_b0 = x_b0 - part_pos_x;
|
|
x_b1 = x_b1 - part_pos_x;
|
|
x_b2 = x_b2 - part_pos_x;
|
|
|
|
/* Below derivation are based on top pointer */
|
|
/* is pointing first pixel of PU */
|
|
pu_idx_b0 = *(pu4_top_pu_idx + (x_b0 >> 2));
|
|
pu_idx_b0 = pu_idx_b0 * tr_avail;
|
|
pu_idx_b1 = *(pu4_top_pu_idx + (x_b1 >> 2));
|
|
pu_idx_b1 = pu_idx_b1 * t_avail;
|
|
/* At CTB boundary, use top-left passed in */
|
|
if(part_pos_y)
|
|
{
|
|
pu_idx_b2 = *pu4_top_left_pu_idx;
|
|
}
|
|
else
|
|
{
|
|
/* Not at CTB boundary, use top and */
|
|
/* add correction to go to top-left */
|
|
pu_idx_b2 = *((pu4_top_pu_idx)+(x_b2 >> 2));
|
|
}
|
|
pu_idx_b2 = pu_idx_b2 * tl_avail;
|
|
|
|
num_nbrs = 0;
|
|
nbr_avail[0] = 0;
|
|
nbr_avail[1] = 0;
|
|
nbr_avail[2] = 0;
|
|
|
|
if(tr_avail && (!ps_pic_pu[pu_idx_b0].b1_intra_flag))
|
|
{
|
|
aps_nbr_pu[num_nbrs] = &ps_pic_pu[pu_idx_b0];
|
|
num_nbrs++;
|
|
nbr_avail[0] = 1;
|
|
}
|
|
if(t_avail && (!ps_pic_pu[pu_idx_b1].b1_intra_flag))
|
|
{
|
|
aps_nbr_pu[num_nbrs] = &ps_pic_pu[pu_idx_b1];
|
|
num_nbrs++;
|
|
nbr_avail[1] = 1;
|
|
}
|
|
if(tl_avail && (!ps_pic_pu[pu_idx_b2].b1_intra_flag))
|
|
{
|
|
aps_nbr_pu[num_nbrs] = &ps_pic_pu[pu_idx_b2];
|
|
num_nbrs++;
|
|
nbr_avail[2] = 1;
|
|
}
|
|
}
|
|
|
|
/* L0 */
|
|
avail_b_flag[0] = 0;
|
|
avail_b_flag[1] = 0;
|
|
|
|
GET_MV_NBR_ST(ps_ref_pic_list, &avail_b_flag[0], ps_cur_pic_buf_l0, aps_nbr_pu, &as_mv_b[0], num_nbrs, 0);
|
|
|
|
/* L1 */
|
|
if(PRED_L0 != ps_pu->b2_pred_mode)
|
|
{
|
|
/* B0 Short Term */
|
|
GET_MV_NBR_ST(ps_ref_pic_list, &avail_b_flag[1], ps_cur_pic_buf_l1, aps_nbr_pu, &as_mv_b[1], num_nbrs, 1);
|
|
}
|
|
|
|
if(avail_b_flag[0])
|
|
{
|
|
if(((0 == num_l0_mvp_cand)
|
|
|| (as_mv_a[0].i2_mvx != as_mv_b[0].i2_mvx)
|
|
|| (as_mv_a[0].i2_mvy != as_mv_b[0].i2_mvy)))
|
|
{
|
|
num_l0_mvp_cand++;
|
|
if(max_l0_mvp_cand == num_l0_mvp_cand)
|
|
{
|
|
ps_pred_mv->s_l0_mv = as_mv_b[0];
|
|
l0_done_flag = 1;
|
|
}
|
|
}
|
|
}
|
|
if(avail_b_flag[1])
|
|
{
|
|
if(((0 == num_l1_mvp_cand)
|
|
|| (as_mv_a[1].i2_mvx != as_mv_b[1].i2_mvx)
|
|
|| (as_mv_a[1].i2_mvy != as_mv_b[1].i2_mvy)))
|
|
{
|
|
num_l1_mvp_cand++;
|
|
if(max_l1_mvp_cand == num_l1_mvp_cand)
|
|
{
|
|
ps_pred_mv->s_l1_mv = as_mv_b[1];
|
|
l1_done_flag = 1;
|
|
}
|
|
}
|
|
}
|
|
if(l0_done_flag && l1_done_flag)
|
|
return;
|
|
|
|
if((is_scaled_flag_list == 0) && (avail_b_flag[0] == 1))
|
|
{
|
|
avail_a_flag[0] = 1;
|
|
as_mv_a[0] = as_mv_b[0];
|
|
}
|
|
if((is_scaled_flag_list == 0) && (avail_b_flag[1] == 1))
|
|
{
|
|
avail_a_flag[1] = 1;
|
|
as_mv_a[1] = as_mv_b[1];
|
|
}
|
|
|
|
if(0 == is_scaled_flag_list)
|
|
{
|
|
avail_b_flag[0] = avail_b_flag[1] = 0;
|
|
|
|
GET_MV_NBR_LT(ps_ref_pic_list, ps_slice_hdr, &avail_b_flag[0], ps_cur_pic_buf_l0, aps_nbr_pu, &as_mv_b[0], num_nbrs, 0);
|
|
|
|
if(PRED_L0 != ps_pu->b2_pred_mode)
|
|
{
|
|
GET_MV_NBR_LT(ps_ref_pic_list, ps_slice_hdr, &avail_b_flag[1], ps_cur_pic_buf_l1, aps_nbr_pu, &as_mv_b[1], num_nbrs, 1);
|
|
}
|
|
|
|
if(avail_b_flag[0])
|
|
{
|
|
if(((0 == num_l0_mvp_cand)
|
|
|| (as_mv_a[0].i2_mvx != as_mv_b[0].i2_mvx)
|
|
|| (as_mv_a[0].i2_mvy != as_mv_b[0].i2_mvy)))
|
|
{
|
|
num_l0_mvp_cand++;
|
|
if(max_l0_mvp_cand == num_l0_mvp_cand)
|
|
{
|
|
ps_pred_mv->s_l0_mv = as_mv_b[0];
|
|
l0_done_flag = 1;
|
|
}
|
|
}
|
|
}
|
|
if(avail_b_flag[1])
|
|
{
|
|
if(((0 == num_l1_mvp_cand)
|
|
|| (as_mv_a[1].i2_mvx != as_mv_b[1].i2_mvx)
|
|
|| (as_mv_a[1].i2_mvy != as_mv_b[1].i2_mvy)))
|
|
{
|
|
num_l1_mvp_cand++;
|
|
if(max_l1_mvp_cand == num_l1_mvp_cand)
|
|
{
|
|
ps_pred_mv->s_l1_mv = as_mv_b[1];
|
|
l1_done_flag = 1;
|
|
}
|
|
}
|
|
}
|
|
if(l0_done_flag && l1_done_flag)
|
|
return;
|
|
}
|
|
/***********************************************************/
|
|
/* Collocated MV prediction */
|
|
/***********************************************************/
|
|
if((2 != num_l0_mvp_cand) || (2 != num_l1_mvp_cand))
|
|
{
|
|
mv_t as_mv_col[2], s_mv_col_l0, s_mv_col_l1;
|
|
WORD32 avail_col_flag[2] = { 0 };
|
|
WORD32 x_col, y_col, avail_col_l0, avail_col_l1;
|
|
// ihevcd_collocated_mvp((mv_ctxt_t *)ps_mv_ctxt,ps_pu,part_pos_x,part_pos_y,part_wd,part_ht,as_mv_col,avail_col_flag,1);
|
|
x_col = part_pos_x + part_wd;
|
|
y_col = part_pos_y + part_ht;
|
|
ihevcd_collocated_mvp(ps_mv_ctxt, ps_pu, as_mv_col, avail_col_flag, 1, x_col, y_col);
|
|
|
|
avail_col_l0 = avail_col_flag[0];
|
|
avail_col_l1 = avail_col_flag[1];
|
|
if(avail_col_l0 || avail_col_l1)
|
|
{
|
|
s_mv_col_l0 = as_mv_col[0];
|
|
s_mv_col_l1 = as_mv_col[1];
|
|
}
|
|
|
|
if(avail_col_l0 == 0 || avail_col_l1 == 0)
|
|
{
|
|
/* Checking Collocated MV availability at Center of PU */
|
|
x_col = part_pos_x + (part_wd >> 1);
|
|
y_col = part_pos_y + (part_ht >> 1);
|
|
ihevcd_collocated_mvp(ps_mv_ctxt, ps_pu, as_mv_col, avail_col_flag, 1, x_col, y_col);
|
|
|
|
if(avail_col_l0 == 0)
|
|
{
|
|
s_mv_col_l0 = as_mv_col[0];
|
|
}
|
|
if(avail_col_l1 == 0)
|
|
{
|
|
s_mv_col_l1 = as_mv_col[1];
|
|
}
|
|
|
|
avail_col_l0 |= avail_col_flag[0];
|
|
avail_col_l1 |= avail_col_flag[1];
|
|
}
|
|
|
|
/* Checking if mvp index matches collocated mv */
|
|
if(avail_col_l0)
|
|
{
|
|
if(2 != num_l0_mvp_cand)
|
|
{
|
|
num_l0_mvp_cand++;
|
|
if(max_l0_mvp_cand == num_l0_mvp_cand)
|
|
{
|
|
ps_pred_mv->s_l0_mv = s_mv_col_l0;
|
|
l0_done_flag = 1;
|
|
}
|
|
}
|
|
}
|
|
if(avail_col_l1)
|
|
{
|
|
if(2 != num_l1_mvp_cand)
|
|
{
|
|
num_l1_mvp_cand++;
|
|
if(max_l1_mvp_cand == num_l1_mvp_cand)
|
|
{
|
|
ps_pred_mv->s_l1_mv = s_mv_col_l1;
|
|
l1_done_flag = 1;
|
|
}
|
|
}
|
|
}
|
|
if(l0_done_flag && l1_done_flag)
|
|
return;
|
|
}
|
|
|
|
if(0 == l0_done_flag)
|
|
{
|
|
ps_pred_mv->s_l0_mv.i2_mvx = 0;
|
|
ps_pred_mv->s_l0_mv.i2_mvy = 0;
|
|
}
|
|
if(0 == l1_done_flag)
|
|
{
|
|
ps_pred_mv->s_l1_mv.i2_mvx = 0;
|
|
ps_pred_mv->s_l1_mv.i2_mvy = 0;
|
|
}
|
|
}
|
|
}
|