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.
215 lines
8.6 KiB
215 lines
8.6 KiB
/******************************************************************************
|
|
*
|
|
* 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
|
|
* ihevc_ilf_padding_frame.c
|
|
*
|
|
* @brief
|
|
* Does frame level loop filtering (deblocking and SAO) and padding
|
|
*
|
|
* @author
|
|
* Srinivas T
|
|
*
|
|
* @par List of Functions:
|
|
* - ihevc_ilf_pad_frame()
|
|
*
|
|
* @remarks
|
|
* None
|
|
*
|
|
*******************************************************************************
|
|
*/
|
|
|
|
|
|
#include <stdio.h>
|
|
#include <stddef.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <assert.h>
|
|
|
|
#include "ihevc_typedefs.h"
|
|
#include "iv.h"
|
|
#include "ivd.h"
|
|
#include "ihevcd_cxa.h"
|
|
#include "ithread.h"
|
|
|
|
#include "ihevc_defs.h"
|
|
#include "ihevc_debug.h"
|
|
#include "ihevc_defs.h"
|
|
#include "ihevc_structs.h"
|
|
#include "ihevc_macros.h"
|
|
#include "ihevc_platform_macros.h"
|
|
#include "ihevc_cabac_tables.h"
|
|
|
|
#include "ihevc_error.h"
|
|
#include "ihevc_common_tables.h"
|
|
|
|
#include "ihevcd_trace.h"
|
|
#include "ihevcd_defs.h"
|
|
#include "ihevcd_function_selector.h"
|
|
#include "ihevcd_structs.h"
|
|
#include "ihevcd_error.h"
|
|
#include "ihevcd_nal.h"
|
|
#include "ihevcd_bitstream.h"
|
|
#include "ihevcd_job_queue.h"
|
|
#include "ihevcd_utils.h"
|
|
|
|
#include "ihevc_deblk.h"
|
|
#include "ihevc_deblk_tables.h"
|
|
#include "ihevcd_profile.h"
|
|
#include "ihevcd_deblk.h"
|
|
#include "ihevcd_sao.h"
|
|
#include "ihevc_padding.h"
|
|
|
|
void ihevcd_ilf_pad_frame(deblk_ctxt_t *ps_deblk_ctxt, sao_ctxt_t *ps_sao_ctxt)
|
|
{
|
|
sps_t *ps_sps;
|
|
slice_header_t *ps_slice_hdr;
|
|
codec_t *ps_codec;
|
|
WORD32 i4_ctb_x, i4_ctb_y;
|
|
WORD32 ctb_size;
|
|
|
|
ps_sps = ps_deblk_ctxt->ps_sps;
|
|
ps_slice_hdr = ps_deblk_ctxt->ps_slice_hdr;
|
|
ps_codec = ps_deblk_ctxt->ps_codec;
|
|
ctb_size = (1 << ps_sps->i1_log2_ctb_size);
|
|
|
|
for(i4_ctb_y = 0; i4_ctb_y < ps_sps->i2_pic_ht_in_ctb; i4_ctb_y++)
|
|
{
|
|
for(i4_ctb_x = 0; i4_ctb_x < ps_sps->i2_pic_wd_in_ctb; i4_ctb_x++)
|
|
{
|
|
WORD32 i4_is_last_ctb_x = 0;
|
|
WORD32 i4_is_last_ctb_y = 0;
|
|
|
|
/*TODO:
|
|
* Slice header also has to be updated
|
|
* */
|
|
ps_deblk_ctxt->i4_ctb_x = i4_ctb_x;
|
|
ps_deblk_ctxt->i4_ctb_y = i4_ctb_y;
|
|
|
|
ps_sao_ctxt->i4_ctb_x = i4_ctb_x;
|
|
ps_sao_ctxt->i4_ctb_y = i4_ctb_y;
|
|
|
|
if((0 == ps_slice_hdr->i1_slice_disable_deblocking_filter_flag) &&
|
|
(0 == ps_codec->i4_disable_deblk_pic))
|
|
{
|
|
ihevcd_deblk_ctb(ps_deblk_ctxt, i4_is_last_ctb_x, i4_is_last_ctb_y);
|
|
|
|
/* If the last CTB in the row was a complete CTB then deblocking has to be called from remaining pixels, since deblocking
|
|
* is applied on a shifted CTB structure
|
|
*/
|
|
if(i4_ctb_x == ps_sps->i2_pic_wd_in_ctb - 1)
|
|
{
|
|
WORD32 last_x_pos;
|
|
i4_is_last_ctb_x = 1;
|
|
i4_is_last_ctb_y = 0;
|
|
|
|
|
|
last_x_pos = (ps_sps->i2_pic_wd_in_ctb << ps_sps->i1_log2_ctb_size);
|
|
if(last_x_pos == ps_sps->i2_pic_width_in_luma_samples)
|
|
{
|
|
ihevcd_deblk_ctb(ps_deblk_ctxt, i4_is_last_ctb_x, i4_is_last_ctb_y);
|
|
}
|
|
}
|
|
|
|
|
|
/* If the last CTB in the column was a complete CTB then deblocking has to be called from remaining pixels, since deblocking
|
|
* is applied on a shifted CTB structure
|
|
*/
|
|
if(i4_ctb_y == ps_sps->i2_pic_ht_in_ctb - 1)
|
|
{
|
|
WORD32 last_y_pos;
|
|
i4_is_last_ctb_y = 1;
|
|
i4_is_last_ctb_x = 0;
|
|
|
|
last_y_pos = (ps_sps->i2_pic_ht_in_ctb << ps_sps->i1_log2_ctb_size);
|
|
if(last_y_pos == ps_sps->i2_pic_height_in_luma_samples)
|
|
{
|
|
ihevcd_deblk_ctb(ps_deblk_ctxt, i4_is_last_ctb_x, i4_is_last_ctb_y);
|
|
}
|
|
}
|
|
}
|
|
|
|
if(ps_slice_hdr->i1_slice_sao_luma_flag || ps_slice_hdr->i1_slice_sao_chroma_flag)
|
|
{
|
|
ihevcd_sao_ctb(ps_sao_ctxt);
|
|
}
|
|
|
|
/* Call padding if required */
|
|
{
|
|
UWORD8 *pu1_cur_ctb_luma = ps_deblk_ctxt->pu1_cur_pic_luma
|
|
+ (i4_ctb_x * ctb_size
|
|
+ i4_ctb_y * ctb_size
|
|
* ps_codec->i4_strd);
|
|
UWORD8 *pu1_cur_ctb_chroma = ps_deblk_ctxt->pu1_cur_pic_chroma
|
|
+ i4_ctb_x * ctb_size
|
|
+ (i4_ctb_y * ctb_size * ps_codec->i4_strd / 2);
|
|
|
|
if(0 == i4_ctb_x)
|
|
{
|
|
WORD32 pad_ht_luma;
|
|
WORD32 pad_ht_chroma;
|
|
|
|
pad_ht_luma = ctb_size;
|
|
pad_ht_luma += (ps_sps->i2_pic_ht_in_ctb - 1) == i4_ctb_y ? 8 : 0;
|
|
pad_ht_chroma = ctb_size / 2;
|
|
pad_ht_chroma += (ps_sps->i2_pic_ht_in_ctb - 1) == i4_ctb_y ? 8 : 0;
|
|
/* Pad left after 1st CTB is processed */
|
|
ps_codec->s_func_selector.ihevc_pad_left_luma_fptr(pu1_cur_ctb_luma - 8 * ps_codec->i4_strd, ps_codec->i4_strd, pad_ht_luma, PAD_LEFT);
|
|
ps_codec->s_func_selector.ihevc_pad_left_chroma_fptr(pu1_cur_ctb_chroma - 8 * ps_codec->i4_strd, ps_codec->i4_strd, pad_ht_chroma, PAD_LEFT);
|
|
}
|
|
else if((ps_sps->i2_pic_wd_in_ctb - 1) == i4_ctb_x)
|
|
{
|
|
WORD32 pad_ht_luma;
|
|
WORD32 pad_ht_chroma;
|
|
WORD32 cols_remaining = ps_sps->i2_pic_width_in_luma_samples - (i4_ctb_x << ps_sps->i1_log2_ctb_size);
|
|
|
|
pad_ht_luma = ctb_size;
|
|
pad_ht_luma += (ps_sps->i2_pic_ht_in_ctb - 1) == i4_ctb_y ? 8 : 0;
|
|
pad_ht_chroma = ctb_size / 2;
|
|
pad_ht_chroma += (ps_sps->i2_pic_ht_in_ctb - 1) == i4_ctb_y ? 8 : 0;
|
|
/* Pad right after last CTB in the current row is processed */
|
|
ps_codec->s_func_selector.ihevc_pad_right_luma_fptr(pu1_cur_ctb_luma + cols_remaining - 8 * ps_codec->i4_strd, ps_codec->i4_strd, pad_ht_luma, PAD_RIGHT);
|
|
ps_codec->s_func_selector.ihevc_pad_right_chroma_fptr(pu1_cur_ctb_chroma + cols_remaining - 8 * ps_codec->i4_strd, ps_codec->i4_strd, pad_ht_chroma, PAD_RIGHT);
|
|
|
|
|
|
if((ps_sps->i2_pic_ht_in_ctb - 1) == i4_ctb_y)
|
|
{
|
|
UWORD8 *pu1_buf;
|
|
/* Since SAO is shifted by 8x8, chroma padding can not be done till second row is processed */
|
|
/* Hence moving top padding to to end of frame, Moving it to second row also results in problems when there is only one row */
|
|
/* Pad top after padding left and right for current rows after processing 1st CTB row */
|
|
ihevc_pad_top(ps_deblk_ctxt->pu1_cur_pic_luma - PAD_LEFT, ps_codec->i4_strd, ps_sps->i2_pic_width_in_luma_samples + PAD_WD, PAD_TOP);
|
|
ihevc_pad_top(ps_deblk_ctxt->pu1_cur_pic_chroma - PAD_LEFT, ps_codec->i4_strd, ps_sps->i2_pic_width_in_luma_samples + PAD_WD, PAD_TOP / 2);
|
|
|
|
pu1_buf = ps_deblk_ctxt->pu1_cur_pic_luma + ps_codec->i4_strd * ps_sps->i2_pic_height_in_luma_samples - PAD_LEFT;
|
|
/* Pad top after padding left and right for current rows after processing 1st CTB row */
|
|
ihevc_pad_bottom(pu1_buf, ps_codec->i4_strd, ps_sps->i2_pic_width_in_luma_samples + PAD_WD, PAD_BOT);
|
|
|
|
pu1_buf = ps_deblk_ctxt->pu1_cur_pic_chroma + ps_codec->i4_strd * (ps_sps->i2_pic_height_in_luma_samples / 2) - PAD_LEFT;
|
|
ihevc_pad_bottom(pu1_buf, ps_codec->i4_strd, ps_sps->i2_pic_width_in_luma_samples + PAD_WD, PAD_BOT / 2);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
}
|
|
}
|
|
|
|
}
|