/****************************************************************************** * * Copyright (C) 2018 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 */ /** ****************************************************************************** * @file ihevce_nbr_avail.c * * @brief * This file contains function definitions and look up tables for various * neigbour avail flags in HEVC encoder * * @author * Ittiam * * List of Functions * * ****************************************************************************** */ /*****************************************************************************/ /* File Includes */ /*****************************************************************************/ /* System include files */ #include #include #include #include #include #include /* User include files */ #include "ihevc_typedefs.h" #include "itt_video_api.h" #include "ihevce_api.h" #include "rc_cntrl_param.h" #include "rc_frame_info_collector.h" #include "rc_look_ahead_params.h" #include "ihevc_defs.h" #include "ihevc_structs.h" #include "ihevc_platform_macros.h" #include "ihevc_deblk.h" #include "ihevc_itrans_recon.h" #include "ihevc_chroma_itrans_recon.h" #include "ihevc_chroma_intra_pred.h" #include "ihevc_intra_pred.h" #include "ihevc_inter_pred.h" #include "ihevc_mem_fns.h" #include "ihevc_padding.h" #include "ihevc_weighted_pred.h" #include "ihevc_sao.h" #include "ihevc_resi_trans.h" #include "ihevc_quant_iquant_ssd.h" #include "ihevc_cabac_tables.h" #include "ihevce_defs.h" #include "ihevce_lap_enc_structs.h" #include "ihevce_multi_thrd_structs.h" #include "ihevce_multi_thrd_funcs.h" #include "ihevce_me_common_defs.h" #include "ihevce_had_satd.h" #include "ihevce_error_codes.h" #include "ihevce_bitstream.h" #include "ihevce_cabac.h" #include "ihevce_rdoq_macros.h" #include "ihevce_function_selector.h" #include "ihevce_enc_structs.h" #include "ihevce_nbr_avail.h" /*****************************************************************************/ /* Function Definitions */ /*****************************************************************************/ /*! ****************************************************************************** * \if Function name : ihevce_set_ctb_nbr \endif * * \brief * This function sets the neighbour availability flags of ctb based on the * CTB position * * \date * 18/09/2012 * * \author * Ittiam * * \return * none * ****************************************************************************** */ void ihevce_set_ctb_nbr( nbr_avail_flags_t *ps_nbr, UWORD8 *pu1_nbr_map, WORD32 nbr_map_strd, WORD32 ctb_pos_x, WORD32 ctb_pos_y, frm_ctb_ctxt_t *ps_frm_ctb_prms) { WORD32 ctr; WORD32 *pi4_cur_ctb_tile_id; WORD32 i4_curr_ctb_tile_id, i4_top_ctb_tile_id; WORD32 i4_left_ctb_tile_id, i4_right_ctb_tile_id; WORD32 ctb_size = ps_frm_ctb_prms->i4_ctb_size; WORD32 num_ctb_horz = ps_frm_ctb_prms->i4_num_ctbs_horz; WORD32 num_ctb_vert = ps_frm_ctb_prms->i4_num_ctbs_vert; WORD32 cu_aligned_pic_wd = ps_frm_ctb_prms->i4_cu_aligned_pic_wd; WORD32 cu_aligned_pic_ht = ps_frm_ctb_prms->i4_cu_aligned_pic_ht; UWORD8 *pu1_top_nbr_map = pu1_nbr_map - nbr_map_strd; UWORD8 *pu1_left_nbr_map = pu1_nbr_map - 1; UWORD8 *pu1_top_lt_nbr_map = pu1_top_nbr_map - 1; UWORD8 *pu1_top_rt_nbr_map = pu1_top_nbr_map + (ctb_size >> 2); WORD32 num_4x4_ctb_x = (ctb_size >> 2); WORD32 num_4x4_ctb_y = (ctb_size >> 2); /* Conditionally update num_4x4_ctb_x and num_4x4_ctb_y */ if(ctb_pos_y == (num_ctb_vert - 1)) { num_4x4_ctb_y = (cu_aligned_pic_ht - ((num_ctb_vert - 1) * ctb_size)) / 4; } if(ctb_pos_x == (num_ctb_horz - 1)) { num_4x4_ctb_x = (cu_aligned_pic_wd - ((num_ctb_horz - 1) * ctb_size)) / 4; } /* Get Tile-ids of top, left and current CTBs */ pi4_cur_ctb_tile_id = ps_frm_ctb_prms->pi4_tile_id_map + ctb_pos_y * ps_frm_ctb_prms->i4_tile_id_ctb_map_stride + ctb_pos_x; i4_curr_ctb_tile_id = *pi4_cur_ctb_tile_id; i4_left_ctb_tile_id = *(pi4_cur_ctb_tile_id - 1); i4_right_ctb_tile_id = *(pi4_cur_ctb_tile_id + 1); i4_top_ctb_tile_id = *(pi4_cur_ctb_tile_id - ps_frm_ctb_prms->i4_tile_id_ctb_map_stride); /*********** Update Nbr availability in ps_nbr **********/ ps_nbr->u1_left_avail = (i4_left_ctb_tile_id == i4_curr_ctb_tile_id); ps_nbr->u1_top_avail = (i4_top_ctb_tile_id == i4_curr_ctb_tile_id); ps_nbr->u1_top_lt_avail = (ps_nbr->u1_left_avail && ps_nbr->u1_top_avail); ps_nbr->u1_top_rt_avail = ps_nbr->u1_top_avail && (i4_right_ctb_tile_id == i4_curr_ctb_tile_id); ps_nbr->u1_bot_lt_avail = 0; /* at ctb level bottom left is always not available */ /*********** Update Nbr availability in pu1_nbr_map **********/ /* NOTE: entire Nbr availability map is by default set to 0 */ *pu1_top_lt_nbr_map = ps_nbr->u1_top_lt_avail; /* Top-Left*/ memset(pu1_top_nbr_map, ps_nbr->u1_top_avail, num_4x4_ctb_x); /* Top */ for(ctr = 0; ctr < num_4x4_ctb_y; ctr++) /* Left */ { *pu1_left_nbr_map = ps_nbr->u1_left_avail; pu1_left_nbr_map += nbr_map_strd; } if((num_ctb_horz - 2) == ctb_pos_x) /* Top-Right */ { /* For the last but 1 ctb, if the last ctb is non-multiple of 64, then set the map accordingly */ WORD32 last_ctb_x = cu_aligned_pic_wd - ((num_ctb_horz - 1) * ctb_size); num_4x4_ctb_x = MIN(last_ctb_x, MAX_TU_SIZE) / 4; memset(pu1_top_rt_nbr_map, ps_nbr->u1_top_rt_avail, num_4x4_ctb_x); } else { memset(pu1_top_rt_nbr_map, ps_nbr->u1_top_rt_avail, (MAX_TU_SIZE / 4)); } return; } /*! ****************************************************************************** * \if Function name : ihevce_get_nbr_intra \endif * * \brief * This function sets the neighbour availability flags of given unit * based on the position and size * * \date * 18/09/2012 * * \author * Ittiam * * \return * none * ****************************************************************************** */ WORD32 ihevce_get_nbr_intra( nbr_avail_flags_t *ps_cu_nbr, UWORD8 *pu1_nbr_map, WORD32 nbr_map_strd, WORD32 unit_4x4_pos_x, WORD32 unit_4x4_pos_y, WORD32 unit_4x4_size) { WORD32 nbr_tem_flags = 0; WORD32 i; UWORD8 *pu1_bot_lt_map; UWORD8 *pu1_top_rt_map; UWORD8 *pu1_top_lt_map; UWORD8 *pu1_left_map; UWORD8 *pu1_top_map; /* map is stored at 4x4 level increment to point to current cu 4x4 */ pu1_nbr_map += (unit_4x4_pos_x); pu1_nbr_map += (unit_4x4_pos_y)*nbr_map_strd; pu1_top_map = pu1_nbr_map - nbr_map_strd; pu1_top_lt_map = pu1_top_map - 1; pu1_left_map = (pu1_nbr_map - 1); /* use map to get top right availablility */ pu1_top_rt_map = pu1_nbr_map - nbr_map_strd; pu1_top_rt_map += unit_4x4_size; /* use map to get bot left availablility */ pu1_bot_lt_map = pu1_nbr_map - 1; pu1_bot_lt_map += unit_4x4_size * nbr_map_strd; /* Top flag */ ps_cu_nbr->u1_top_avail = *pu1_top_map; /* left flag */ ps_cu_nbr->u1_left_avail = *pu1_left_map; /* top left flag */ ps_cu_nbr->u1_top_lt_avail = *pu1_top_lt_map; /* top right flag */ ps_cu_nbr->u1_top_rt_avail = *pu1_top_rt_map; /* bottom left flag */ ps_cu_nbr->u1_bot_lt_avail = (*pu1_bot_lt_map); /* Update the neighbor availiblity flag according to the nbr_map */ nbr_tem_flags = 0; for(i = 0; i < 4; i++) { nbr_tem_flags |= ((*pu1_bot_lt_map) << (3 - i)); pu1_bot_lt_map += (nbr_map_strd * 2); } for(i = 0; i < 4; i++) { nbr_tem_flags |= ((*pu1_left_map) << (7 - i)); pu1_left_map += (nbr_map_strd * 2); } for(i = 0; i < 4; i++) { nbr_tem_flags |= ((*pu1_top_map) << (i + 8)); pu1_top_map += 2; } for(i = 0; i < 4; i++) { nbr_tem_flags |= ((*pu1_top_rt_map) << (i + 12)); pu1_top_rt_map += 2; } nbr_tem_flags |= (*pu1_top_lt_map << 16); return nbr_tem_flags; } /*! ****************************************************************************** * \if Function name : ihevce_get_nbr_intra_mxn_tu \endif * * \brief * This function sets the neighbour availability flags of given unit * based on the position and size * * \date * 24/06/2014 * * \author * Ittiam * * \return * none * ****************************************************************************** */ WORD32 ihevce_get_nbr_intra_mxn_tu( UWORD8 *pu1_nbr_map, WORD32 nbr_map_strd, WORD32 unit_4x4_pos_x, WORD32 unit_4x4_pos_y, WORD32 unit_4x4_size_horz, WORD32 unit_4x4_size_vert) { WORD32 nbr_tem_flags = 0; WORD32 i; UWORD8 *pu1_bot_lt_map; UWORD8 *pu1_top_rt_map; UWORD8 *pu1_top_lt_map; UWORD8 *pu1_left_map; UWORD8 *pu1_top_map; /* map is stored at 4x4 level increment to point to current cu 4x4 */ pu1_nbr_map += (unit_4x4_pos_x); pu1_nbr_map += (unit_4x4_pos_y)*nbr_map_strd; pu1_top_map = pu1_nbr_map - nbr_map_strd; pu1_top_lt_map = pu1_top_map - 1; pu1_left_map = (pu1_nbr_map - 1); /* use map to get top right availablility */ pu1_top_rt_map = pu1_nbr_map - nbr_map_strd; pu1_top_rt_map += unit_4x4_size_horz; /* use map to get bot left availablility */ pu1_bot_lt_map = pu1_nbr_map - 1; pu1_bot_lt_map += unit_4x4_size_vert * nbr_map_strd; /* Update the neighbor availiblity flag according to the nbr_map */ nbr_tem_flags = 0; for(i = 0; i < 4; i++) { nbr_tem_flags |= ((*pu1_bot_lt_map) << (3 - i)); pu1_bot_lt_map += (nbr_map_strd * 2); } for(i = 0; i < 4; i++) { nbr_tem_flags |= ((*pu1_left_map) << (7 - i)); pu1_left_map += (nbr_map_strd * 2); } for(i = 0; i < 4; i++) { nbr_tem_flags |= ((*pu1_top_map) << (i + 8)); pu1_top_map += 2; } for(i = 0; i < 4; i++) { nbr_tem_flags |= ((*pu1_top_rt_map) << (i + 12)); pu1_top_rt_map += 2; } nbr_tem_flags |= (*pu1_top_lt_map << 16); return nbr_tem_flags; } /*! ****************************************************************************** * \if Function name : ihevce_get_intra_chroma_tu_nbr \endif * * \brief * This function sets the neighbour availability flags of a chroma * subTU based on luma availability and chroma format * * \date * 04/07/2014 * * \author * Ittiam * * \return * none * ****************************************************************************** */ WORD32 ihevce_get_intra_chroma_tu_nbr( WORD32 i4_luma_nbr_flags, WORD32 i4_subtu_idx, WORD32 i4_trans_size, UWORD8 u1_is_422) { /* TOP LEFT | TOP-RIGHT | TOP | LEFT | BOTTOM LEFT*/ /* (1 bit) (4 bits) (4 bits) (4 bits) (4 bits) */ /* With reference to the above bit arrangement - */ /* BL0 - Bit 3 */ /* BL1 - Bit 2 */ /* BL2 - Bit 1 */ /* BL3 - Bit 0 */ /* L0 - Bit 7 */ /* L1 - Bit 6 */ /* L2 - Bit 5 */ /* L3 - Bit 4 */ /* T0 - Bit 8 */ /* T1 - Bit 9 */ /* T2 - Bit 10 */ /* T3 - Bit 11 */ /* TR0 - Bit 12 */ /* TR1 - Bit 13 */ /* TR2 - Bit 14 */ /* TR3 - Bit 15 */ if(u1_is_422) { if(0 == i4_subtu_idx) { /* If left is available for luma, then */ if(i4_luma_nbr_flags & 0xf0) { switch(i4_trans_size) { case 4: { /* BL0 - 1 */ /* BL1-3 - Luma_BL0-2 */ /*i4_luma_nbr_flags |= (i4_luma_nbr_flags & 0xe) >> 1;*/ i4_luma_nbr_flags |= 0x8; /* L0-1 - 11 */ /* L2-3 - Luma_L2-3 */ i4_luma_nbr_flags |= 0xc0; break; } case 8: { /* BL0-1 - 11 */ /* BL1-3 - Luma_BL0-1 */ /*i4_luma_nbr_flags |= (i4_luma_nbr_flags & 0xc) >> 2;*/ i4_luma_nbr_flags |= 0xc; /* L0-3 - 1111 */ i4_luma_nbr_flags |= 0xf0; break; } case 16: { /* BL0-3 - 1111 */ i4_luma_nbr_flags |= 0xf; /* L0-3 - 1111 */ i4_luma_nbr_flags |= 0xf0; break; } } } } else { /* Top right is always unavailable */ /* Top is always available */ i4_luma_nbr_flags &= (0xffff0fff); /* Top left is marked as available if */ /* luma left is available */ if(i4_luma_nbr_flags & 0xf0) { i4_luma_nbr_flags |= (1 << 16); } switch(i4_trans_size) { case 4: { /* T0 - 1 */ /* T1-3 - 000 */ i4_luma_nbr_flags |= 0x100; i4_luma_nbr_flags &= 0xfffff1ff; if(i4_luma_nbr_flags & 0xf0) { i4_luma_nbr_flags |= 0x80; } if(i4_luma_nbr_flags & 0x8) { i4_luma_nbr_flags |= 0x8; } break; } case 8: { /* T0-1 - 11 */ /* T2-3 - 00 */ i4_luma_nbr_flags |= 0x300; i4_luma_nbr_flags &= 0xfffff3ff; if(i4_luma_nbr_flags & 0xf0) { i4_luma_nbr_flags |= 0xc0; } if((i4_luma_nbr_flags & 0xc) == 0x8) { i4_luma_nbr_flags |= 0xc; } else if((i4_luma_nbr_flags & 0xc) == 0xc) { i4_luma_nbr_flags |= 0xf; } else if((i4_luma_nbr_flags & 0xf) == 0xe) { i4_luma_nbr_flags |= 0xf; } break; } case 16: { /* T0-3 - 1111 */ i4_luma_nbr_flags |= 0xf00; if(i4_luma_nbr_flags & 0xf0) { i4_luma_nbr_flags |= 0xf0; } if((i4_luma_nbr_flags & 0xf) == 0x8) { i4_luma_nbr_flags |= 0xc; } else if((i4_luma_nbr_flags & 0xf) == 0xc) { i4_luma_nbr_flags |= 0xf; } else if((i4_luma_nbr_flags & 0xf) == 0xe) { i4_luma_nbr_flags |= 0xf; } break; } } } } return i4_luma_nbr_flags; } /*! ****************************************************************************** * \if Function name : ihevce_get_only_nbr_flag \endif * * \brief * This function sets the neighbour availability flags of given unit * based on the position, unit width and unit height * * \date * 18/09/2012 * * \author * Ittiam * * \return * none * ****************************************************************************** */ void ihevce_get_only_nbr_flag( nbr_avail_flags_t *ps_cu_nbr, UWORD8 *pu1_nbr_map, WORD32 nbr_map_strd, WORD32 unit_4x4_pos_x, WORD32 unit_4x4_pos_y, WORD32 unit_4x4_size_hz, WORD32 unit_4x4_size_vt) { /* map is stored at 4x4 level increment to point to current cu 4x4 */ pu1_nbr_map += (unit_4x4_pos_x); pu1_nbr_map += (unit_4x4_pos_y)*nbr_map_strd; /* Top flag */ ps_cu_nbr->u1_top_avail = *(pu1_nbr_map - nbr_map_strd); /* left flag */ ps_cu_nbr->u1_left_avail = *(pu1_nbr_map - 1); /* top left flag */ ps_cu_nbr->u1_top_lt_avail = *(pu1_nbr_map - nbr_map_strd - 1); /* top right flag */ { UWORD8 *pu1_top_rt_map; /* use map to get top right availablility */ pu1_top_rt_map = pu1_nbr_map - nbr_map_strd; pu1_top_rt_map += unit_4x4_size_hz; /* store the availbility */ ps_cu_nbr->u1_top_rt_avail = *pu1_top_rt_map; } /* bottom left flag */ { UWORD8 *pu1_bot_lt_map; /* use map to get bot left availablility */ pu1_bot_lt_map = pu1_nbr_map - 1; pu1_bot_lt_map += unit_4x4_size_vt * nbr_map_strd; /* store the availbility */ ps_cu_nbr->u1_bot_lt_avail = *pu1_bot_lt_map; } return; } /*! ****************************************************************************** * \if Function name : ihevce_set_nbr_map \endif * * \brief * This function sets the neighbour availability flags of given value * based on the position and size * * \date * 18/09/2012 * * \author * Ittiam * * \return * none * ****************************************************************************** */ void ihevce_set_nbr_map( UWORD8 *pu1_nbr_map, WORD32 nbr_map_strd, WORD32 unit_4x4_pos_x, WORD32 unit_4x4_pos_y, WORD32 unit_4x4_size, WORD32 val) { WORD32 i; /* map is stored at 4x4 level increment to point to current cu 4x4 */ pu1_nbr_map += (unit_4x4_pos_x); pu1_nbr_map += (unit_4x4_pos_y)*nbr_map_strd; /* loops to set the flags for given size */ for(i = 0; i < unit_4x4_size; i++) { memset(pu1_nbr_map, val, sizeof(UWORD8) * unit_4x4_size); /* row level updates */ pu1_nbr_map += nbr_map_strd; } return; } /*! ****************************************************************************** * \if Function name : ihevce_set_inter_nbr_map \endif * * \brief * This function sets the neighbour availability flags of given value * based on the position and horizontal width and vertical height * * \date * 18/09/2012 * * \author * Ittiam * * \return * none * List of Functions * * ****************************************************************************** */ void ihevce_set_inter_nbr_map( UWORD8 *pu1_nbr_map, WORD32 nbr_map_strd, WORD32 unit_4x4_pos_x, WORD32 unit_4x4_pos_y, WORD32 unit_4x4_size_hz, WORD32 unit_4x4_size_vt, WORD32 val) { WORD32 i; /* map is stored at 4x4 level increment to point to current cu 4x4 */ pu1_nbr_map += (unit_4x4_pos_x); pu1_nbr_map += (unit_4x4_pos_y)*nbr_map_strd; { /* loops to set the flags for given size */ for(i = 0; i < unit_4x4_size_vt; i++) { memset(pu1_nbr_map, val, sizeof(UWORD8) * unit_4x4_size_hz); /* row level updates */ pu1_nbr_map += nbr_map_strd; } } return; }