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