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.
324 lines
11 KiB
324 lines
11 KiB
4 months ago
|
/******************************************************************************
|
||
|
|
*
|
||
|
|
* Copyright (C) 2012 Ittiam Systems Pvt Ltd, Bangalore
|
||
|
|
*
|
||
|
|
* 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.
|
||
|
|
*
|
||
|
|
******************************************************************************/
|
||
|
|
/**
|
||
|
|
*******************************************************************************
|
||
|
|
* @file
|
||
|
|
* ihevcd_intra_pred_mode_prediction.c.c
|
||
|
|
*
|
||
|
|
* @brief
|
||
|
|
* Contains functions for intra pred mode prediction
|
||
|
|
*
|
||
|
|
* @author
|
||
|
|
* Ittiam
|
||
|
|
*
|
||
|
|
* @par List of Functions:
|
||
|
|
* - ihevcd_intra_pred_mode_prediction()
|
||
|
|
*
|
||
|
|
* @remarks
|
||
|
|
* None
|
||
|
|
*
|
||
|
|
*******************************************************************************
|
||
|
|
*/
|
||
|
|
/*****************************************************************************/
|
||
|
|
/* File Includes */
|
||
|
|
/*****************************************************************************/
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stddef.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
|
||
|
|
#include "ihevc_typedefs.h"
|
||
|
|
#include "iv.h"
|
||
|
|
#include "ivd.h"
|
||
|
|
#include "ihevcd_cxa.h"
|
||
|
|
|
||
|
|
#include "ihevc_defs.h"
|
||
|
|
#include "ihevc_debug.h"
|
||
|
|
#include "ihevc_structs.h"
|
||
|
|
#include "ihevc_macros.h"
|
||
|
|
#include "ihevc_mem_fns.h"
|
||
|
|
#include "ihevc_platform_macros.h"
|
||
|
|
|
||
|
|
#include "ihevcd_defs.h"
|
||
|
|
#include "ihevc_cabac_tables.h"
|
||
|
|
#include "ihevcd_function_selector.h"
|
||
|
|
#include "ihevcd_structs.h"
|
||
|
|
#include "ihevcd_error.h"
|
||
|
|
|
||
|
|
#include "ihevcd_bitstream.h"
|
||
|
|
|
||
|
|
|
||
|
|
/*****************************************************************************/
|
||
|
|
/* Function Prototypes */
|
||
|
|
/*****************************************************************************/
|
||
|
|
|
||
|
|
/*****************************************************************************/
|
||
|
|
/* Availability check is not done inside the function */
|
||
|
|
/* Whenever the top and left are not available, it is assumed that Intra DC */
|
||
|
|
/* mode will initialized in place of non available */
|
||
|
|
/* neighbors */
|
||
|
|
/*****************************************************************************/
|
||
|
|
|
||
|
|
/**
|
||
|
|
*******************************************************************************
|
||
|
|
*
|
||
|
|
* @brief Computes intra prediction mode for a CU
|
||
|
|
*
|
||
|
|
* @par Description
|
||
|
|
* Computes intra prediction mode for a CU
|
||
|
|
*
|
||
|
|
* @param[in,out] ps_cu
|
||
|
|
* Codic unit context
|
||
|
|
*
|
||
|
|
* @param[in] ps_parse
|
||
|
|
* parse context
|
||
|
|
*
|
||
|
|
* @param[in] ps_codec
|
||
|
|
* codec context
|
||
|
|
*
|
||
|
|
* @param[in] log2_cb_size
|
||
|
|
* log of cb size base 2
|
||
|
|
*
|
||
|
|
* @returns none
|
||
|
|
*
|
||
|
|
* @remarks
|
||
|
|
* Availability check is moved to CTB level. If the neighbors are
|
||
|
|
* not available or if the pred mode of neighbor is not MODE_INTRA,
|
||
|
|
* INTRA_DC mode will be updated in top and left buffers.
|
||
|
|
*******************************************************************************
|
||
|
|
*/
|
||
|
|
void ihevcd_intra_pred_mode_prediction(codec_t *ps_codec,
|
||
|
|
WORD32 log2_cb_size,
|
||
|
|
WORD32 x0,
|
||
|
|
WORD32 y0)
|
||
|
|
{
|
||
|
|
WORD32 i, j, num_pred_blocks;
|
||
|
|
WORD32 available_l, available_t;
|
||
|
|
WORD32 cand_intra_pred_mode_l, cand_intra_pred_mode_t;
|
||
|
|
WORD32 cand_mode_list[3];
|
||
|
|
WORD32 cb_size, block_offset_in_min_pu;
|
||
|
|
UWORD8 *pu1_luma_intra_pred_mode_top;
|
||
|
|
UWORD8 *pu1_luma_intra_pred_mode_left;
|
||
|
|
|
||
|
|
parse_ctxt_t *ps_parse = &ps_codec->s_parse;
|
||
|
|
parse_cu_t *ps_cu = &ps_codec->s_parse.s_cu;
|
||
|
|
sps_t *ps_sps = ps_parse->ps_sps;
|
||
|
|
|
||
|
|
|
||
|
|
available_t = 1;
|
||
|
|
available_l = 1;
|
||
|
|
/* i4_pos_x and i4_pos_y are in minCu units (8x8), convert them to 4x4 units by multiplying by 2 */
|
||
|
|
pu1_luma_intra_pred_mode_top = ps_parse->pu1_luma_intra_pred_mode_top
|
||
|
|
+ (ps_cu->i4_pos_x * 2);
|
||
|
|
|
||
|
|
pu1_luma_intra_pred_mode_left = ps_parse->pu1_luma_intra_pred_mode_left
|
||
|
|
+ (ps_cu->i4_pos_y * 2);
|
||
|
|
|
||
|
|
/*
|
||
|
|
if(0 == ps_cu->i4_pos_y)
|
||
|
|
{
|
||
|
|
memset(pu1_luma_intra_pred_mode_top, INTRA_DC, 16);
|
||
|
|
}
|
||
|
|
|
||
|
|
if(0 == ps_cu->i4_pos_x)
|
||
|
|
{
|
||
|
|
memset(pu1_luma_intra_pred_mode_left, INTRA_DC, 16);
|
||
|
|
}
|
||
|
|
*/
|
||
|
|
if(ps_cu->i4_pos_y)
|
||
|
|
{
|
||
|
|
UWORD8 *pu1_pic_intra_flag = ps_codec->s_parse.pu1_pic_intra_flag;
|
||
|
|
WORD32 top_intra_flag;
|
||
|
|
|
||
|
|
WORD32 numbytes_row = (ps_sps->i2_pic_width_in_luma_samples + 63) / 64;
|
||
|
|
pu1_pic_intra_flag += ((y0 - 8) / 8) * numbytes_row;
|
||
|
|
pu1_pic_intra_flag += (x0 / 64);
|
||
|
|
top_intra_flag = *pu1_pic_intra_flag;
|
||
|
|
top_intra_flag &= (1 << ((x0 / 8) % 8));
|
||
|
|
|
||
|
|
if(0 == top_intra_flag)
|
||
|
|
{
|
||
|
|
available_t = 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
available_t = 0;
|
||
|
|
|
||
|
|
|
||
|
|
if((0 == ps_cu->i4_pos_x) && (((0 == ps_codec->s_parse.i4_ctb_slice_x) && (0 == ps_codec->s_parse.i4_ctb_slice_y)) ||
|
||
|
|
(0 == ps_codec->s_parse.i4_ctb_tile_x)))
|
||
|
|
{
|
||
|
|
available_l = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
if(available_l)
|
||
|
|
{
|
||
|
|
UWORD8 *pu1_pic_intra_flag = ps_codec->s_parse.pu1_pic_intra_flag;
|
||
|
|
WORD32 left_intra_flag;
|
||
|
|
WORD32 numbytes_row = (ps_sps->i2_pic_width_in_luma_samples + 63) / 64;
|
||
|
|
pu1_pic_intra_flag += (y0 / 8) * numbytes_row;
|
||
|
|
pu1_pic_intra_flag += ((x0 - 8) / 64);
|
||
|
|
left_intra_flag = *pu1_pic_intra_flag;
|
||
|
|
left_intra_flag &= (1 << (((x0 - 8) / 8) % 8));
|
||
|
|
|
||
|
|
if(0 == left_intra_flag)
|
||
|
|
{
|
||
|
|
available_l = 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
cb_size = (1 << log2_cb_size);
|
||
|
|
|
||
|
|
block_offset_in_min_pu = (cb_size / 2) / MIN_PU_SIZE;
|
||
|
|
|
||
|
|
num_pred_blocks = (ps_cu->i4_part_mode == PART_NxN) ? 2 : 1;
|
||
|
|
|
||
|
|
for(i = 0; i < num_pred_blocks; i++)
|
||
|
|
{
|
||
|
|
WORD32 available_l_tmp;
|
||
|
|
available_l_tmp = available_l;
|
||
|
|
for(j = 0; j < num_pred_blocks; j++)
|
||
|
|
{
|
||
|
|
/* Computing Candidate intra pred mode left */
|
||
|
|
{
|
||
|
|
WORD32 block_offset;
|
||
|
|
|
||
|
|
block_offset = i * block_offset_in_min_pu;
|
||
|
|
cand_intra_pred_mode_l = INTRA_DC;
|
||
|
|
if(available_l_tmp)
|
||
|
|
{
|
||
|
|
cand_intra_pred_mode_l =
|
||
|
|
pu1_luma_intra_pred_mode_left[block_offset];
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
{
|
||
|
|
WORD32 block_offset;
|
||
|
|
block_offset = j * block_offset_in_min_pu;
|
||
|
|
cand_intra_pred_mode_t = INTRA_DC;
|
||
|
|
if(available_t)
|
||
|
|
{
|
||
|
|
cand_intra_pred_mode_t =
|
||
|
|
pu1_luma_intra_pred_mode_top[block_offset];
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Computing Candidate mode list */
|
||
|
|
if(cand_intra_pred_mode_l == cand_intra_pred_mode_t)
|
||
|
|
{
|
||
|
|
if(cand_intra_pred_mode_l < 2)
|
||
|
|
{
|
||
|
|
cand_mode_list[0] = INTRA_PLANAR;
|
||
|
|
cand_mode_list[1] = INTRA_DC;
|
||
|
|
cand_mode_list[2] = INTRA_ANGULAR(26); /* angular 26 = Vertical */
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
cand_mode_list[0] = cand_intra_pred_mode_l;
|
||
|
|
cand_mode_list[1] = 2
|
||
|
|
+ ((cand_intra_pred_mode_l + 29) % 32);
|
||
|
|
cand_mode_list[2] = 2
|
||
|
|
+ ((cand_intra_pred_mode_l - 2 + 1) % 32);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
cand_mode_list[0] = cand_intra_pred_mode_l;
|
||
|
|
cand_mode_list[1] = cand_intra_pred_mode_t;
|
||
|
|
|
||
|
|
if((cand_intra_pred_mode_l != INTRA_PLANAR)
|
||
|
|
&& (cand_intra_pred_mode_t != INTRA_PLANAR))
|
||
|
|
{
|
||
|
|
cand_mode_list[2] = INTRA_PLANAR;
|
||
|
|
}
|
||
|
|
else if((cand_intra_pred_mode_l != INTRA_DC)
|
||
|
|
&& (cand_intra_pred_mode_t != INTRA_DC))
|
||
|
|
{
|
||
|
|
cand_mode_list[2] = INTRA_DC;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
cand_mode_list[2] = INTRA_ANGULAR(26);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/* Computing Intra pred mode */
|
||
|
|
if(ps_cu->ai4_prev_intra_luma_pred_flag[2 * i + j] == 1)
|
||
|
|
{
|
||
|
|
ps_cu->ai4_intra_luma_pred_mode[2 * i + j] =
|
||
|
|
cand_mode_list[ps_cu->ai4_mpm_idx[2 * i + j]];
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
WORD32 intra_pred_mode;
|
||
|
|
/* Arranging cand_mode_list in increasing order */
|
||
|
|
if(cand_mode_list[0] > cand_mode_list[1])
|
||
|
|
{
|
||
|
|
SWAP(cand_mode_list[0], cand_mode_list[1]);
|
||
|
|
}
|
||
|
|
if(cand_mode_list[0] > cand_mode_list[2])
|
||
|
|
{
|
||
|
|
SWAP(cand_mode_list[0], cand_mode_list[2]);
|
||
|
|
}
|
||
|
|
if(cand_mode_list[1] > cand_mode_list[2])
|
||
|
|
{
|
||
|
|
SWAP(cand_mode_list[1], cand_mode_list[2]);
|
||
|
|
}
|
||
|
|
|
||
|
|
intra_pred_mode = ps_cu->ai4_rem_intra_luma_pred_mode[2 * i + j];
|
||
|
|
|
||
|
|
if(intra_pred_mode >= cand_mode_list[0])
|
||
|
|
intra_pred_mode++;
|
||
|
|
|
||
|
|
if(intra_pred_mode >= cand_mode_list[1])
|
||
|
|
intra_pred_mode++;
|
||
|
|
|
||
|
|
if(intra_pred_mode >= cand_mode_list[2])
|
||
|
|
intra_pred_mode++;
|
||
|
|
|
||
|
|
ps_cu->ai4_intra_luma_pred_mode[2 * i + j] = intra_pred_mode;
|
||
|
|
}
|
||
|
|
/* Update Top and Left intra pred mode */
|
||
|
|
{
|
||
|
|
WORD32 intra_pred_mode;
|
||
|
|
|
||
|
|
intra_pred_mode = ps_cu->ai4_intra_luma_pred_mode[2 * i + j];
|
||
|
|
|
||
|
|
ps_codec->s_func_selector.ihevc_memset_fptr(pu1_luma_intra_pred_mode_left + i * block_offset_in_min_pu, intra_pred_mode, (cb_size / num_pred_blocks) / MIN_PU_SIZE);
|
||
|
|
ps_codec->s_func_selector.ihevc_memset_fptr(pu1_luma_intra_pred_mode_top + j * block_offset_in_min_pu, intra_pred_mode, (cb_size / num_pred_blocks) / MIN_PU_SIZE);
|
||
|
|
|
||
|
|
}
|
||
|
|
/* If partition is PART_NxN, then left is available for second column always */
|
||
|
|
available_l_tmp = 1;
|
||
|
|
|
||
|
|
}
|
||
|
|
/* If partition is PART_NxN, then top is available for bottom row always */
|
||
|
|
available_t = 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* In case it is PART_2Nx2N partition, replicate intra pred mode in other three entries */
|
||
|
|
if(ps_cu->i4_part_mode == PART_2Nx2N)
|
||
|
|
{
|
||
|
|
ps_cu->ai4_intra_luma_pred_mode[1] = ps_cu->ai4_intra_luma_pred_mode[0];
|
||
|
|
ps_cu->ai4_intra_luma_pred_mode[2] = ps_cu->ai4_intra_luma_pred_mode[0];
|
||
|
|
ps_cu->ai4_intra_luma_pred_mode[3] = ps_cu->ai4_intra_luma_pred_mode[0];
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|