| /****************************************************************************** |
| * |
| * 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 |
| */ |
| #ifndef _IH264D_MB_UTILS_H_ |
| #define _IH264D_MB_UTILS_H_ |
| /*! |
| ************************************************************************** |
| * \file ih264d_mb_utils.h |
| * |
| * \brief |
| * Contains declarations of the utility functions needed to decode MB |
| * |
| * \date |
| * 18/12/2002 |
| * |
| * \author AI |
| ************************************************************************** |
| */ |
| #include "ih264_typedefs.h" |
| #include "ih264_macros.h" |
| #include "ih264_platform_macros.h" |
| #include "ih264d_structs.h" |
| |
| /*--------------------------------------------------------------------*/ |
| /* Macros to get raster scan position of a block[8x8] / sub block[4x4]*/ |
| /*--------------------------------------------------------------------*/ |
| |
| #define GET_BLK_RASTER_POS_X(x) ((x & 0x01) << 1) |
| #define GET_BLK_RASTER_POS_Y(y) ((y >> 1) << 1) |
| #define GET_SUB_BLK_RASTER_POS_X(x) ((x & 0x01)) |
| #define GET_SUB_BLK_RASTER_POS_Y(y) ((y >> 1)) |
| |
| /*--------------------------------------------------------------------*/ |
| /* Masks used in decoding of Macroblock */ |
| /*--------------------------------------------------------------------*/ |
| |
| #define LEFT_MB_AVAILABLE_MASK 0x01 |
| #define TOP_LEFT_MB_AVAILABLE_MASK 0x02 |
| #define TOP_MB_AVAILABLE_MASK 0x04 |
| #define TOP_RIGHT_MB_AVAILABLE_MASK 0x08 |
| |
| #define TOP_RT_SUBBLOCK_MASK_MOD 0xFFF7 |
| |
| #define TOP_RIGHT_DEFAULT_AVAILABLE 0x5750 |
| #define TOP_RIGHT_TOPR_AVAILABLE 0x0008 |
| #define TOP_RIGHT_TOP_AVAILABLE 0x0007 |
| |
| #define TOP_LEFT_DEFAULT_AVAILABLE 0xEEE0 |
| #define TOP_LEFT_TOPL_AVAILABLE 0x0001 |
| #define TOP_LEFT_TOP_AVAILABLE 0x000E |
| #define TOP_LEFT_LEFT_AVAILABLE 0x1110 |
| |
| #define CHECK_MB_MAP(u4_mb_num, mb_map, u4_cond) \ |
| { \ |
| UWORD32 u4_bit_number; \ |
| volatile UWORD8 *pu1_mb_flag; \ |
| \ |
| u4_bit_number = u4_mb_num & 0x07; \ |
| pu1_mb_flag = (UWORD8 *)mb_map + (u4_mb_num >> 3); \ |
| \ |
| u4_cond = CHECKBIT((*pu1_mb_flag), u4_bit_number); \ |
| } |
| |
| #define CHECK_MB_MAP_BYTE(u4_mb_num, mb_map, u4_cond) \ |
| { \ |
| volatile UWORD8 *pu1_mb_flag; \ |
| \ |
| pu1_mb_flag = (UWORD8 *)mb_map + (u4_mb_num ); \ |
| \ |
| u4_cond = (*pu1_mb_flag); \ |
| } |
| |
| #define UPDATE_MB_MAP(u2_frm_wd_in_mbs, u2_mbx, u2_mby, mb_map, mb_count) \ |
| { \ |
| UWORD32 u4_bit_number; \ |
| UWORD32 u4_mb_number; \ |
| \ |
| u4_mb_number = u2_frm_wd_in_mbs * (u2_mby >> u1_mbaff) + u2_mbx; \ |
| \ |
| u4_bit_number = u4_mb_number & 0x07; \ |
| /* \ |
| * In case of MbAff, update the mb_map only if the entire MB is done. We can check that \ |
| * by checking if Y is odd, implying that this is the second row in the MbAff MB \ |
| */ \ |
| SET_BIT(mb_map[u4_mb_number >> 3], u4_bit_number); \ |
| \ |
| if (1 == u1_mbaff) \ |
| { \ |
| /* \ |
| * If MBAFF u4_flag is set, set this MB and the MB just below this. \ |
| * So, add frame width to the MB number and set that bit. \ |
| */ \ |
| /* \ |
| u4_mb_number += u2_frm_wd_in_mbs; \ |
| \ |
| u4_bit_number = u4_mb_number & 0x07; \ |
| \ |
| SET_BIT(mb_map[u4_mb_number >> 3], u4_bit_number); \ |
| */ \ |
| } \ |
| \ |
| /*H264_DEC_DEBUG_PRINT("SETBIT: %d\n", u4_mb_number);*/ \ |
| mb_count++; \ |
| } |
| |
| #define UPDATE_MB_MAP_MBNUM(mb_map, u4_mb_number) \ |
| { \ |
| UWORD32 u4_bit_number; \ |
| volatile UWORD8 *pu1_mb_flag; \ |
| \ |
| u4_bit_number = u4_mb_number & 0x07; \ |
| pu1_mb_flag = (UWORD8 *)mb_map + (u4_mb_number >> 3); \ |
| /* \ |
| * In case of MbAff, update the mb_map only if the entire MB is done. We can check that \ |
| * by checking if Y is odd, implying that this is the second row in the MbAff MB \ |
| */ \ |
| SET_BIT((*pu1_mb_flag), u4_bit_number); \ |
| } |
| |
| #define UPDATE_MB_MAP_MBNUM_BYTE(mb_map, u4_mb_number) \ |
| { \ |
| volatile UWORD8 *pu1_mb_flag; \ |
| \ |
| pu1_mb_flag = (UWORD8 *)mb_map + (u4_mb_number); \ |
| /* \ |
| * In case of MbAff, update the mb_map only if the entire MB is done. We can check that \ |
| * by checking if Y is odd, implying that this is the second row in the MbAff MB \ |
| */ \ |
| (*pu1_mb_flag) = 1; \ |
| } |
| |
| #define UPDATE_SLICE_NUM_MAP(slice_map, u4_mb_number,u2_slice_num) \ |
| { \ |
| volatile UWORD16 *pu2_slice_map; \ |
| \ |
| pu2_slice_map = (UWORD16 *)slice_map + (u4_mb_number); \ |
| (*pu2_slice_map) = u2_slice_num; \ |
| } |
| |
| #define GET_SLICE_NUM_MAP(slice_map, mb_number,u2_slice_num) \ |
| { \ |
| volatile UWORD16 *pu2_slice_map; \ |
| \ |
| pu2_slice_map = (UWORD16 *)slice_map + (mb_number); \ |
| u2_slice_num = (*pu2_slice_map) ; \ |
| } |
| |
| |
| #define GET_XPOS_PRED(u1_out,pkd_info) \ |
| { \ |
| WORD32 bit_field; \ |
| bit_field = pkd_info & 0x3; \ |
| u1_out = bit_field; \ |
| } |
| |
| |
| #define GET_YPOS_PRED(u1_out,pkd_info) \ |
| { \ |
| WORD32 bit_field; \ |
| bit_field = pkd_info >> 2; \ |
| u1_out = bit_field & 0x3; \ |
| } |
| |
| |
| |
| #define GET_WIDTH_PRED(u1_out,pkd_info) \ |
| { \ |
| WORD32 bit_field; \ |
| bit_field = pkd_info >> 4; \ |
| bit_field = (bit_field & 0x3) << 1 ; \ |
| u1_out = (bit_field == 0)?1:bit_field; \ |
| } |
| |
| #define GET_HEIGHT_PRED(u1_out,pkd_info) \ |
| { \ |
| WORD32 bit_field; \ |
| bit_field = pkd_info >> 6; \ |
| bit_field = (bit_field & 0x3) << 1 ; \ |
| u1_out = (bit_field == 0)?1:bit_field; \ |
| } |
| |
| /*! |
| ************************************************************************** |
| * \brief Masks for elements present in the first column but not on the |
| * first row. |
| ************************************************************************** |
| */ |
| #define FIRST_COL_NOT_FIRST_ROW 0xFAFB |
| #define FIRST_ROW_MASK 0xFFCC |
| /*! |
| ************************************************************************** |
| * \brief Mask for elements presen in the first row but not in the |
| * last column. |
| ************************************************************************** |
| */ |
| #define FIRST_ROW_NOT_LAST_COL 0xFFEC |
| /*! |
| ************************************************************************** |
| * \brief Mask for elements presen in the first row but not in the |
| * first column. |
| ************************************************************************** |
| */ |
| #define FIRST_ROW_NOT_FIRST_COL 0xFFCD |
| /*! |
| ************************************************************************** |
| * \brief Masks for the top right subMB of a 4x4 block |
| ************************************************************************** |
| */ |
| #define TOP_RT_SUBBLOCK_MASK 0xFFDF |
| /*! |
| ************************************************************************** |
| * \brief Masks for the top left subMB of a 4x4 block |
| ************************************************************************** |
| */ |
| #define TOP_LT_SUBBLOCK_MASK 0xFFFE |
| /*! |
| ************************************************************************** |
| * \brief Indicates if a subMB has a top right subMB available |
| ************************************************************************** |
| */ |
| #define TOP_RT_SUBBLOCK_MB_MASK 0x5F4C |
| |
| #define FIRST_COL_MASK 0xFAFA |
| |
| /*--------------------------------------------------------------------*/ |
| /* Macros to calculate the current position of a MB wrt picture */ |
| /*--------------------------------------------------------------------*/ |
| #define MB_LUMA_PIC_OFFSET(mb_x,mb_y,frmWidthY) (((mb_y)*(frmWidthY) + (mb_x))<<4) |
| #define MB_CHROMA_PIC_OFFSET(mb_x,mb_y,frmWidthUV) (((mb_y)*(frmWidthUV) + (mb_x))<<3) |
| |
| /*--------------------------------------------------------------------*/ |
| /* Macros to calculate the current position of a MB wrt N[ Num coeff] Array */ |
| /*--------------------------------------------------------------------*/ |
| #define MB_PARAM_OFFSET(mb_x,mb_y,frmWidthInMbs,u1_mbaff,u1_topmb) \ |
| ((mb_x << u1_mbaff) + (1 - u1_topmb) + (mb_y * frmWidthInMbs)) |
| |
| UWORD32 ih264d_get_mb_info_cavlc_mbaff(dec_struct_t * ps_dec, |
| const UWORD16 ui16_curMbAddress, |
| dec_mb_info_t * ps_cur_mb_info, |
| UWORD32 u4_mbskip_run); |
| UWORD32 ih264d_get_mb_info_cavlc_nonmbaff(dec_struct_t * ps_dec, |
| const UWORD16 ui16_curMbAddress, |
| dec_mb_info_t * ps_cur_mb_info, |
| UWORD32 u4_mbskip_run); |
| |
| UWORD32 ih264d_get_mb_info_cabac_mbaff(dec_struct_t * ps_dec, |
| const UWORD16 ui16_curMbAddress, |
| dec_mb_info_t * ps_cur_mb_info, |
| UWORD32 u4_mbskip_run); |
| |
| UWORD32 ih264d_get_mb_info_cabac_nonmbaff(dec_struct_t * ps_dec, |
| const UWORD16 ui16_curMbAddress, |
| dec_mb_info_t * ps_cur_mb_info, |
| UWORD32 u4_mbskip_run); |
| |
| UWORD8 get_cabac_context_non_mbaff(dec_struct_t * ps_dec, UWORD16 u2_mbskip); |
| |
| UWORD32 ih264d_get_cabac_context_mbaff(dec_struct_t * ps_dec, |
| dec_mb_info_t * ps_cur_mb_info, |
| UWORD32 u4_mbskip); |
| |
| WORD32 PutMbToFrame(dec_struct_t * ps_dec); |
| void ih264d_get_mbaff_neighbours(dec_struct_t * ps_dec, |
| dec_mb_info_t * ps_cur_mb_info, |
| UWORD8 uc_curMbFldDecFlag); |
| |
| void ih264d_update_mbaff_left_nnz(dec_struct_t * ps_dec, |
| dec_mb_info_t * ps_cur_mb_info); |
| void ih264d_transfer_mb_group_data(dec_struct_t * ps_dec, |
| const UWORD8 u1_num_mbs, |
| const UWORD8 u1_end_of_row, /* Cur n-Mb End of Row Flag */ |
| const UWORD8 u1_end_of_row_next /* Next n-Mb End of Row Flag */ |
| ); |
| |
| //void FillRandomData(UWORD8 *pu1_buf, WORD32 u4_bufSize); |
| |
| #endif /* _MB_UTILS_H_ */ |