You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
293 lines
17 KiB
293 lines
17 KiB
/****************************************************************************** |
|
* |
|
* 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_ */
|
|
|