/****************************************************************************** * * 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 */ #include #include "iv_datatypedef.h" #include "iv.h" #include "impeg2_buf_mgr.h" #include "impeg2_disp_mgr.h" #include "impeg2_defs.h" #include "impeg2_platform_macros.h" #include "impeg2_inter_pred.h" #include "impeg2_idct.h" #include "impeg2_globals.h" #include "impeg2_mem_func.h" #include "impeg2_format_conv.h" #include "impeg2_macros.h" #include "ivd.h" #include "impeg2d.h" #include "impeg2d_bitstream.h" #include "impeg2d_structs.h" #include "impeg2d_globals.h" #include "impeg2d_vld_tables.h" #include "impeg2d_pic_proc.h" #include "impeg2d_debug.h" #include "impeg2d_mv_dec.h" #include "impeg2d_mc.h" /******************************************************************************* * Function name : impeg2d_dec_1mv * * Description : Decodes a motion vector and updates the predictors * * Arguments : * stream : Bitstream * predMv : Prediction for the motion vectors * mv : Motion vectors * fCode : fcode to the used for the decoding * shift : Shift value to be used. This will be equal to * (mv_format == "field") && (picture_structure == "Frame picture") * i : 0 - MV_X and 1 - MV_Y * * Value Returned: None *******************************************************************************/ INLINE void impeg2d_dec_1mv(stream_t *ps_stream, WORD16 ai2_pred_mv[], WORD16 ai2_mv[],UWORD16 au2_fCode[], UWORD16 u2_mv_y_shift, WORD16 ai2_dmv[]) { WORD16 i2_f; WORD16 i2_r_size; WORD16 i2_high,i2_low,i2_range; UWORD32 u4_mv_code; WORD16 i2_delta; UWORD16 u2_first_bit; WORD32 i; WORD32 ai2_shifts[2]; UWORD32 u4_buf; UWORD32 u4_buf_nxt; UWORD32 u4_offset; UWORD32 *pu4_buf_aligned; ai2_shifts[0] = 0; ai2_shifts[1] = u2_mv_y_shift; GET_TEMP_STREAM_DATA(u4_buf,u4_buf_nxt,u4_offset,pu4_buf_aligned,ps_stream) for(i = 0; i < 2; i++) { WORD32 i4_shift = ai2_shifts[i]; /* Decode the motion_code */ IBITS_NXT(u4_buf, u4_buf_nxt, u4_offset, u4_mv_code, MV_CODE_LEN) u2_first_bit = (u4_mv_code >> (MV_CODE_LEN - 1)) & 0x01; if(u2_first_bit == 1) /* mvCode == 0 */ { i2_delta = 0; FLUSH_BITS(u4_offset,u4_buf,u4_buf_nxt,1,pu4_buf_aligned) ai2_mv[i] = (ai2_pred_mv[i] >> i4_shift); ai2_pred_mv[i] = (ai2_mv[i] << i4_shift); } else { UWORD16 u2_index; UWORD16 u2_value; UWORD16 u2_mv_len; UWORD16 u2_abs_mvcode_minus1; UWORD16 u2_sign_bit; i2_r_size = au2_fCode[i] - 1; i2_f = 1 << i2_r_size; i2_high = (16 * i2_f) - 1; i2_low = ((-16) * i2_f); i2_range = (32 * i2_f); u2_index = (u4_mv_code >> 1) & 0x1FF; u2_value = gau2_impeg2d_mv_code[u2_index]; u2_mv_len = (u2_value & 0x0F); u2_abs_mvcode_minus1 = (u2_value >> 8) & 0x0FF; u4_mv_code >>= (MV_CODE_LEN - u2_mv_len - 1); u2_sign_bit = u4_mv_code & 0x1; FLUSH_BITS(u4_offset,u4_buf,u4_buf_nxt,(u2_mv_len + 1),pu4_buf_aligned) i2_delta = u2_abs_mvcode_minus1 * i2_f + 1; if(i2_r_size) { UWORD32 val; IBITS_GET(u4_buf, u4_buf_nxt, u4_offset, val, pu4_buf_aligned, i2_r_size) i2_delta += val; } if(u2_sign_bit) i2_delta = -i2_delta; ai2_mv[i] = (ai2_pred_mv[i] >> i4_shift) + i2_delta; if(ai2_mv[i] < i2_low) { ai2_mv[i] += i2_range; } if(ai2_mv[i] > i2_high) { ai2_mv[i] -= i2_range; } ai2_pred_mv[i] = (ai2_mv[i] << i4_shift); } if(ai2_dmv) { UWORD32 u4_val; ai2_dmv[i] = 0; IBITS_GET(u4_buf, u4_buf_nxt, u4_offset, u4_val, pu4_buf_aligned, 1) if(u4_val) { IBITS_GET(u4_buf, u4_buf_nxt, u4_offset, u4_val, pu4_buf_aligned, 1) ai2_dmv[i] = gai2_impeg2d_dec_mv[u4_val]; } } } PUT_TEMP_STREAM_DATA(u4_buf, u4_buf_nxt, u4_offset, pu4_buf_aligned, ps_stream) } /******************************************************************************* * Function name : impeg2d_dec_mv * * Description : Decodes a motion vector and updates the predictors * * Arguments : * stream : Bitstream * predMv : Prediction for the motion vectors * mv : Motion vectors * fCode : fcode to the used for the decoding * shift : Shift value to be used. This will be equal to * (mv_format == "field") && (picture_structure == "Frame picture") * * Value Returned: None *******************************************************************************/ e_field_t impeg2d_dec_mv(stream_t *ps_stream, WORD16 ai2_pred_mv[], WORD16 ai2_mv[],UWORD16 au2_f_code[], UWORD16 u2_shift, UWORD16 u2_fld_sel) { e_field_t e_fld; if(u2_fld_sel) { e_fld = (e_field_t)impeg2d_bit_stream_get_bit(ps_stream); } else { e_fld = TOP; } impeg2d_dec_1mv(ps_stream,ai2_pred_mv,ai2_mv,au2_f_code,u2_shift,NULL); return(e_fld); } /***************************************************************************** * Function Name : impeg2d_dec_1mv_mb * * Description : Decodes mc params for 1 MV MB * * Arguments : * dec : Decoder state * * Values Returned : None *****************************************************************************/ void impeg2d_dec_1mv_mb(dec_state_t *ps_dec) { stream_t *ps_stream; WORD16 *pi2_mv; e_field_t e_fld; mb_mc_params_t *ps_mc; e_pred_direction_t e_ref_pic; ps_stream = &ps_dec->s_bit_stream; e_ref_pic = ps_dec->e_mb_pred; /************************************************************************/ /* Decode the motion vector */ /************************************************************************/ pi2_mv = (WORD16 *)&ps_dec->ai2_mv[FORW][FIRST]; e_fld = impeg2d_dec_mv(ps_stream,ps_dec->ai2_pred_mv[e_ref_pic][FIRST],pi2_mv, ps_dec->au2_f_code[e_ref_pic],0, ps_dec->u2_fld_pic); ps_dec->ai2_pred_mv[e_ref_pic][SECOND][MV_X] = ps_dec->ai2_pred_mv[e_ref_pic][FIRST][MV_X]; ps_dec->ai2_pred_mv[e_ref_pic][SECOND][MV_Y] = ps_dec->ai2_pred_mv[e_ref_pic][FIRST][MV_Y]; /************************************************************************/ /* Set the motion vector params */ /************************************************************************/ ps_mc = &ps_dec->as_mb_mc_params[e_ref_pic][FIRST]; ps_mc->s_ref = ps_dec->as_ref_buf[e_ref_pic][e_fld]; impeg2d_set_mc_params(&ps_mc->s_luma, &ps_mc->s_chroma, ps_dec->s_mb_type, 0, pi2_mv, ps_dec->u2_mb_x, ps_dec->u2_mb_y, ps_dec->u2_frame_width, ps_dec->u2_frame_height,ps_dec->u2_picture_width); } /***************************************************************************** * Function Name : impeg2d_dec_2mv_fw_or_bk_mb * * Description : Decodes first part of params for 2 MV Interpolated MB * * Arguments : * dec : Decoder state * * Values Returned : None *****************************************************************************/ void impeg2d_dec_2mv_fw_or_bk_mb(dec_state_t *ps_dec) { stream_t *ps_stream; WORD16 *pi2_mv; e_field_t e_fld; mb_mc_params_t *ps_mc; e_pred_direction_t e_ref_pic; UWORD16 i; ps_stream = &ps_dec->s_bit_stream; e_ref_pic = ps_dec->e_mb_pred; for(i = 0; i < 2; i++) { /********************************************************************/ /* Decode the first motion vector */ /********************************************************************/ pi2_mv = (WORD16 *)&ps_dec->ai2_mv[FORW][i]; e_fld = impeg2d_dec_mv(ps_stream,ps_dec->ai2_pred_mv[e_ref_pic][i],pi2_mv, ps_dec->au2_f_code[e_ref_pic],ps_dec->u2_frm_pic, 1); /********************************************************************/ /* Set the motion vector params */ /********************************************************************/ ps_mc = &ps_dec->as_mb_mc_params[FORW][i]; ps_mc->s_ref = ps_dec->as_ref_buf[e_ref_pic][e_fld]; impeg2d_set_mc_params(&ps_mc->s_luma, &ps_mc->s_chroma, ps_dec->s_mb_type, i, pi2_mv, ps_dec->u2_mb_x, ps_dec->u2_mb_y, ps_dec->u2_frame_width, ps_dec->u2_frame_height,ps_dec->u2_picture_width); } } /***************************************************************************** * Function Name : impeg2d_dec_frm_dual_prime * * Description : Decodes first part of params for 2 MV Interpolated MB * * Arguments : * dec : Decoder state * * Values Returned : None *****************************************************************************/ void impeg2d_dec_frm_dual_prime(dec_state_t *ps_dec) { stream_t *ps_stream; WORD16 *pi2_mv; mb_mc_params_t *ps_mc; WORD16 ai2_dmv[2]; WORD16 *pi2_mv1, *pi2_mv2, *pi2_mv3, *pi2_mv4; UWORD16 i,j; pi2_mv1 = (WORD16 *)&(ps_dec->ai2_mv[FORW][FIRST]); pi2_mv2 = (WORD16 *)&(ps_dec->ai2_mv[FORW][SECOND]); pi2_mv3 = (WORD16 *)&(ps_dec->ai2_mv[BACK][FIRST]); pi2_mv4 = (WORD16 *)&(ps_dec->ai2_mv[BACK][SECOND]); ps_stream = &ps_dec->s_bit_stream; /************************************************************************/ /* Decode the motion vector MV_X, MV_Y and dmv[0], dmv[1] */ /************************************************************************/ impeg2d_dec_1mv(ps_stream,ps_dec->ai2_pred_mv[FORW][FIRST],pi2_mv1,ps_dec->au2_f_code[FORW],ps_dec->u2_frm_pic,ai2_dmv); { WORD16 ai2_m[2][2]; if(ps_dec->u2_top_field_first) { ai2_m[1][0] = 1; ai2_m[0][1] = 3; } else { ai2_m[1][0] = 3; ai2_m[0][1] = 1; } pi2_mv2[MV_X] = pi2_mv1[MV_X]; pi2_mv2[MV_Y] = pi2_mv1[MV_Y]; pi2_mv3[MV_X] = ai2_dmv[0] + DIV_2_RND(pi2_mv1[MV_X] * ai2_m[1][0]); pi2_mv4[MV_X] = ai2_dmv[0] + DIV_2_RND(pi2_mv1[MV_X] * ai2_m[0][1]); pi2_mv3[MV_Y] = ai2_dmv[1] + DIV_2_RND(pi2_mv1[MV_Y] * ai2_m[1][0]) - 1; pi2_mv4[MV_Y] = ai2_dmv[1] + DIV_2_RND(pi2_mv1[MV_Y] * ai2_m[0][1]) + 1; } ps_dec->ai2_pred_mv[FORW][SECOND][MV_X] = ps_dec->ai2_pred_mv[FORW][FIRST][MV_X]; ps_dec->ai2_pred_mv[FORW][SECOND][MV_Y] = ps_dec->ai2_pred_mv[FORW][FIRST][MV_Y]; /************************************************************************/ /* Set the motion vector params */ /************************************************************************/ for(j = 0; j < 2; j++) { for(i = 0; i < 2; i++) { pi2_mv = (WORD16 *)&ps_dec->ai2_mv[j][i]; ps_mc = &ps_dec->as_mb_mc_params[j][i]; ps_mc->s_ref = ps_dec->as_ref_buf[FORW][(i ^ j) & 1]; impeg2d_set_mc_params(&ps_mc->s_luma, &ps_mc->s_chroma, ps_dec->s_mb_type, i, pi2_mv, ps_dec->u2_mb_x, ps_dec->u2_mb_y, ps_dec->u2_frame_width, ps_dec->u2_frame_height,ps_dec->u2_picture_width); } } } /***************************************************************************** * Function Name : impeg2d_dec_fld_dual_prime * * Description : Decodes first part of params for 2 MV Interpolated MB * * Arguments : * dec : Decoder state * * Values Returned : None *****************************************************************************/ void impeg2d_dec_fld_dual_prime(dec_state_t *ps_dec) { stream_t *ps_stream; WORD16 *pi2_mv; mb_mc_params_t *ps_mc; WORD16 *pi2_mv1, *pi2_mv2; WORD16 ai2_dmv[2]; pi2_mv1 = (WORD16 *)&(ps_dec->ai2_mv[FORW][FIRST]); pi2_mv2 = (WORD16 *)&(ps_dec->ai2_mv[FORW][SECOND]); ps_stream = &ps_dec->s_bit_stream; /************************************************************************/ /* Decode the motion vector MV_X, MV_Y and dmv[0], dmv[1] */ /************************************************************************/ impeg2d_dec_1mv(ps_stream,ps_dec->ai2_pred_mv[FORW][FIRST],pi2_mv1,ps_dec->au2_f_code[FORW],0,ai2_dmv); pi2_mv2[MV_X] = ai2_dmv[0] + DIV_2_RND(pi2_mv1[MV_X]); pi2_mv2[MV_Y] = ai2_dmv[1] + DIV_2_RND(pi2_mv1[MV_Y]); if(ps_dec->u2_picture_structure == TOP_FIELD) pi2_mv2[MV_Y] -= 1; else pi2_mv2[MV_Y] += 1; ps_dec->ai2_pred_mv[FORW][SECOND][MV_X] = ps_dec->ai2_pred_mv[FORW][FIRST][MV_X]; ps_dec->ai2_pred_mv[FORW][SECOND][MV_Y] = ps_dec->ai2_pred_mv[FORW][FIRST][MV_Y]; /************************************************************************/ /* Set the motion vector params */ /************************************************************************/ pi2_mv = (WORD16 *)&ps_dec->ai2_mv[FORW][0]; ps_mc = &ps_dec->as_mb_mc_params[FORW][0]; ps_mc->s_ref = ps_dec->as_ref_buf[FORW][ps_dec->u2_fld_parity]; impeg2d_set_mc_params(&ps_mc->s_luma, &ps_mc->s_chroma, ps_dec->s_mb_type, 0, pi2_mv, ps_dec->u2_mb_x, ps_dec->u2_mb_y, ps_dec->u2_frame_width, ps_dec->u2_frame_height,ps_dec->u2_picture_width); pi2_mv = (WORD16 *)&ps_dec->ai2_mv[FORW][1]; ps_mc = &ps_dec->as_mb_mc_params[FORW][1]; ps_mc->s_ref = ps_dec->as_ref_buf[FORW][!ps_dec->u2_fld_parity]; impeg2d_set_mc_params(&ps_mc->s_luma, &ps_mc->s_chroma, ps_dec->s_mb_type, 0, pi2_mv, ps_dec->u2_mb_x, ps_dec->u2_mb_y, ps_dec->u2_frame_width, ps_dec->u2_frame_height,ps_dec->u2_picture_width); } /***************************************************************************** * Function Name : impeg2d_dec_4mv_mb * * Description : Decodes first part of params for 2 MV Interpolated MB * * Arguments : * dec : Decoder state * * Values Returned : None *****************************************************************************/ void impeg2d_dec_4mv_mb(dec_state_t *ps_dec) { stream_t *ps_stream; WORD16 *pi2_mv; e_field_t e_fld; mb_mc_params_t *ps_mc; UWORD16 i,j; ps_stream = &ps_dec->s_bit_stream; /***********************************************/ /* loop for FW & BK */ /***********************************************/ for(j = 0; j < 2; j++) { /***********************************************/ /* loop for decoding 2 mvs of same reference frame*/ /***********************************************/ for(i = 0; i < 2; i++) { /****************************************************************/ /* Decode the first motion vector */ /****************************************************************/ pi2_mv = (WORD16 *)&ps_dec->ai2_mv[j][i]; e_fld = impeg2d_dec_mv(ps_stream,ps_dec->ai2_pred_mv[j][i],pi2_mv, ps_dec->au2_f_code[j],ps_dec->u2_frm_pic, 1); /****************************************************************/ /* Set the motion vector params */ /****************************************************************/ ps_mc = &ps_dec->as_mb_mc_params[j][i]; ps_mc->s_ref = ps_dec->as_ref_buf[j][e_fld]; impeg2d_set_mc_params(&ps_mc->s_luma, &ps_mc->s_chroma, ps_dec->s_mb_type, i, pi2_mv, ps_dec->u2_mb_x, ps_dec->u2_mb_y, ps_dec->u2_frame_width, ps_dec->u2_frame_height,ps_dec->u2_picture_width); } } } /******************************************************************************* * Function Name : impeg2d_dec_2mv_interp_mb * * Description : Decodes first part of params for 2 MV Interpolated MB * * Arguments : * dec : Decoder state * * Values Returned : None *******************************************************************************/ void impeg2d_dec_2mv_interp_mb(dec_state_t *ps_dec) { stream_t *ps_stream; WORD16 *pi2_mv; e_field_t e_fld; mb_mc_params_t *ps_mc; UWORD16 i; ps_stream = &ps_dec->s_bit_stream; for(i = 0; i < 2; i++) { /********************************************************************/ /* Decode the first motion vector */ /********************************************************************/ pi2_mv = (WORD16 *)&ps_dec->ai2_mv[i][FIRST]; e_fld = impeg2d_dec_mv(ps_stream,ps_dec->ai2_pred_mv[i][FIRST],pi2_mv, ps_dec->au2_f_code[i],0, ps_dec->u2_fld_pic); ps_dec->ai2_pred_mv[i][SECOND][MV_X] = ps_dec->ai2_pred_mv[i][FIRST][MV_X]; ps_dec->ai2_pred_mv[i][SECOND][MV_Y] = ps_dec->ai2_pred_mv[i][FIRST][MV_Y]; /********************************************************************/ /* Set the motion vector params */ /********************************************************************/ ps_mc = &ps_dec->as_mb_mc_params[i][FIRST]; ps_mc->s_ref = ps_dec->as_ref_buf[i][e_fld]; impeg2d_set_mc_params(&ps_mc->s_luma, &ps_mc->s_chroma, ps_dec->s_mb_type,i, pi2_mv, ps_dec->u2_mb_x, ps_dec->u2_mb_y, ps_dec->u2_frame_width, ps_dec->u2_frame_height,ps_dec->u2_picture_width); } }