| /****************************************************************************** |
| * |
| * 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; |
| } |
| } |
| } |
