/****************************************************************************** * * 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_ */