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.
1256 lines
54 KiB
1256 lines
54 KiB
/******************************************************************************
|
|
*
|
|
* 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_sao.c
|
|
*
|
|
* @brief
|
|
* Contains definition for the ctb level sao function
|
|
*
|
|
* @author
|
|
* Ittiam
|
|
*
|
|
* @par List of Functions:
|
|
* ihevce_sao_set_avilability()
|
|
* ihevce_sao_ctb()
|
|
* ihevce_sao_analyse()
|
|
*
|
|
* @remarks
|
|
* None
|
|
*
|
|
*******************************************************************************
|
|
*/
|
|
|
|
/*****************************************************************************/
|
|
/* File Includes */
|
|
/*****************************************************************************/
|
|
/* System include files */
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <assert.h>
|
|
#include <stdarg.h>
|
|
#include <math.h>
|
|
|
|
/* 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_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_entropy_structs.h"
|
|
#include "ihevce_cmn_utils_instr_set_router.h"
|
|
#include "ihevce_enc_loop_structs.h"
|
|
#include "ihevce_cabac_rdo.h"
|
|
#include "ihevce_sao.h"
|
|
|
|
/*****************************************************************************/
|
|
/* Function Definitions */
|
|
/*****************************************************************************/
|
|
|
|
/**
|
|
*******************************************************************************
|
|
*
|
|
* @brief
|
|
* ihevce_sao_set_avilability
|
|
*
|
|
* @par Description:
|
|
* Sets the availability flag for SAO.
|
|
*
|
|
* @param[in]
|
|
* ps_sao_ctxt: Pointer to SAO context
|
|
* @returns
|
|
*
|
|
* @remarks
|
|
* None
|
|
*
|
|
*******************************************************************************
|
|
*/
|
|
void ihevce_sao_set_avilability(
|
|
UWORD8 *pu1_avail, sao_ctxt_t *ps_sao_ctxt, ihevce_tile_params_t *ps_tile_params)
|
|
{
|
|
WORD32 i;
|
|
|
|
WORD32 ctb_x_pos = ps_sao_ctxt->i4_ctb_x;
|
|
WORD32 ctb_y_pos = ps_sao_ctxt->i4_ctb_y;
|
|
|
|
for(i = 0; i < 8; i++)
|
|
{
|
|
pu1_avail[i] = 255;
|
|
}
|
|
|
|
/* SAO_note_01: If the CTB lies on a tile or a slice boundary and
|
|
in-loop filtering is enabled at tile and slice boundary, then SAO must
|
|
be performed at tile/slice boundaries also.
|
|
Hence the boundary checks should be based on frame position of CTB
|
|
rather than s_ctb_nbr_avail_flags.u1_left_avail flags.
|
|
Search for <SAO_note_01> in workspace to know more */
|
|
/* Availaibility flags for first col*/
|
|
if(ctb_x_pos == ps_tile_params->i4_first_ctb_x)
|
|
{
|
|
pu1_avail[0] = 0;
|
|
pu1_avail[4] = 0;
|
|
pu1_avail[6] = 0;
|
|
}
|
|
|
|
/* Availaibility flags for last col*/
|
|
if((ctb_x_pos + 1) ==
|
|
(ps_tile_params->i4_first_ctb_x + ps_tile_params->i4_curr_tile_wd_in_ctb_unit))
|
|
{
|
|
pu1_avail[1] = 0;
|
|
pu1_avail[5] = 0;
|
|
pu1_avail[7] = 0;
|
|
}
|
|
|
|
/* Availaibility flags for first row*/
|
|
if(ctb_y_pos == ps_tile_params->i4_first_ctb_y)
|
|
{
|
|
pu1_avail[2] = 0;
|
|
pu1_avail[4] = 0;
|
|
pu1_avail[5] = 0;
|
|
}
|
|
|
|
/* Availaibility flags for last row*/
|
|
if((ctb_y_pos + 1) ==
|
|
(ps_tile_params->i4_first_ctb_y + ps_tile_params->i4_curr_tile_ht_in_ctb_unit))
|
|
{
|
|
pu1_avail[3] = 0;
|
|
pu1_avail[6] = 0;
|
|
pu1_avail[7] = 0;
|
|
}
|
|
}
|
|
|
|
/**
|
|
*******************************************************************************
|
|
*
|
|
* @brief
|
|
* Sao CTB level function.
|
|
*
|
|
* @par Description:
|
|
* For a given CTB, sao is done. Both the luma and chroma
|
|
* blocks are processed
|
|
*
|
|
* @param[in]
|
|
* ps_sao_ctxt: Pointer to SAO context
|
|
*
|
|
* @returns
|
|
*
|
|
* @remarks
|
|
* None
|
|
*
|
|
*******************************************************************************
|
|
*/
|
|
void ihevce_sao_ctb(sao_ctxt_t *ps_sao_ctxt, ihevce_tile_params_t *ps_tile_params)
|
|
{
|
|
sao_enc_t *ps_sao;
|
|
UWORD8 u1_src_top_left_luma, u1_src_top_left_chroma[2];
|
|
UWORD8 *pu1_src_left_luma_buf, *pu1_src_top_luma_buf;
|
|
UWORD8 *pu1_src_left_chroma_buf, *pu1_src_top_chroma_buf;
|
|
UWORD8 *pu1_src_luma, *pu1_src_chroma;
|
|
WORD32 luma_src_stride, ctb_size;
|
|
WORD32 chroma_src_stride;
|
|
UWORD8 au1_avail_luma[8], au1_avail_chroma[8];
|
|
WORD32 sao_blk_wd, sao_blk_ht, sao_wd_chroma, sao_ht_chroma;
|
|
UWORD8 *pu1_top_left_luma, *pu1_top_left_chroma;
|
|
UWORD8 *pu1_src_bot_left_luma, *pu1_src_top_right_luma;
|
|
UWORD8 *pu1_src_bot_left_chroma, *pu1_src_top_right_chroma;
|
|
UWORD8 u1_is_422 = (ps_sao_ctxt->ps_sps->i1_chroma_format_idc == 2);
|
|
|
|
ps_sao = ps_sao_ctxt->ps_sao;
|
|
|
|
ASSERT(
|
|
(abs(ps_sao->u1_y_offset[1]) <= 7) && (abs(ps_sao->u1_y_offset[2]) <= 7) &&
|
|
(abs(ps_sao->u1_y_offset[3]) <= 7) && (abs(ps_sao->u1_y_offset[4]) <= 7));
|
|
ASSERT(
|
|
(abs(ps_sao->u1_cb_offset[1]) <= 7) && (abs(ps_sao->u1_cb_offset[2]) <= 7) &&
|
|
(abs(ps_sao->u1_cb_offset[3]) <= 7) && (abs(ps_sao->u1_cb_offset[4]) <= 7));
|
|
ASSERT(
|
|
(abs(ps_sao->u1_cr_offset[1]) <= 7) && (abs(ps_sao->u1_cr_offset[2]) <= 7) &&
|
|
(abs(ps_sao->u1_cr_offset[3]) <= 7) && (abs(ps_sao->u1_cr_offset[4]) <= 7));
|
|
ASSERT(
|
|
(ps_sao->b5_y_band_pos <= 28) && (ps_sao->b5_cb_band_pos <= 28) &&
|
|
(ps_sao->b5_cr_band_pos <= 28));
|
|
|
|
if(ps_sao_ctxt->i1_slice_sao_luma_flag)
|
|
{
|
|
/*initialize the src pointer to current row*/
|
|
luma_src_stride = ps_sao_ctxt->i4_cur_luma_recon_stride;
|
|
|
|
ctb_size = ps_sao_ctxt->i4_ctb_size;
|
|
|
|
/* 1 extra byte in top buf stride for top left of 1st ctb of every row*/
|
|
ps_sao->u1_y_offset[0] = 0; /* 0th element is not being used */
|
|
sao_blk_wd = ps_sao_ctxt->i4_sao_blk_wd;
|
|
sao_blk_ht = ps_sao_ctxt->i4_sao_blk_ht;
|
|
|
|
pu1_src_luma = ps_sao_ctxt->pu1_cur_luma_recon_buf;
|
|
/* Pointer to the top luma buffer corresponding to the current ctb row*/
|
|
pu1_src_top_luma_buf = ps_sao_ctxt->pu1_curr_sao_src_top_luma;
|
|
|
|
/* Pointer to left luma buffer corresponding to the current ctb row*/
|
|
pu1_src_left_luma_buf = ps_sao_ctxt->au1_left_luma_scratch;
|
|
|
|
/* Pointer to the top right luma buffer corresponding to the current ctb row*/
|
|
pu1_src_top_right_luma = pu1_src_top_luma_buf /*- top_buf_stide*/ + sao_blk_wd;
|
|
|
|
/* Pointer to the bottom left luma buffer corresponding to the current ctb row*/
|
|
pu1_src_bot_left_luma =
|
|
ps_sao_ctxt->pu1_frm_luma_recon_buf + ctb_size * ps_sao_ctxt->i4_frm_luma_recon_stride -
|
|
1 + (ps_sao_ctxt->i4_frm_luma_recon_stride * ps_sao_ctxt->i4_ctb_y * ctb_size) +
|
|
(ps_sao_ctxt->i4_ctb_x * ctb_size); /* Bottom left*/
|
|
|
|
/* Back up the top left pixel for (x+1, y+1)th ctb*/
|
|
u1_src_top_left_luma = *(pu1_src_top_luma_buf + sao_blk_wd - 1);
|
|
pu1_top_left_luma = pu1_src_top_luma_buf - 1;
|
|
|
|
if(SAO_BAND == ps_sao->b3_y_type_idx)
|
|
{
|
|
ihevc_sao_band_offset_luma(
|
|
pu1_src_luma,
|
|
luma_src_stride,
|
|
pu1_src_left_luma_buf, /* Pass the pointer to the left luma buffer backed up in the (x-1,y)th ctb */
|
|
pu1_src_top_luma_buf, /* Pass the ptr to the top luma buf backed up in the (x,y-1)th ctb */
|
|
pu1_src_top_luma_buf - 1, /* Top left*/
|
|
ps_sao->b5_y_band_pos,
|
|
ps_sao->u1_y_offset,
|
|
sao_blk_wd,
|
|
sao_blk_ht);
|
|
|
|
if((ps_sao_ctxt->i4_ctb_y > 0))
|
|
{
|
|
*(pu1_src_top_luma_buf + sao_blk_wd - 1) = u1_src_top_left_luma;
|
|
}
|
|
}
|
|
else if(ps_sao->b3_y_type_idx >= SAO_EDGE_0_DEG)
|
|
{
|
|
/*In case of edge offset, 1st and 2nd offsets are always inferred as offsets
|
|
* corresponding to EO category 1 and 2 which should be always positive
|
|
* And 3rd and 4th offsets are always inferred as offsets corresponding to
|
|
* EO category 3 and 4 which should be negative for all the EO classes(or EO typeidx)
|
|
*/
|
|
// clang-format off
|
|
ASSERT((ps_sao->u1_y_offset[1] >= 0) && (ps_sao->u1_y_offset[2] >= 0));
|
|
ASSERT((ps_sao->u1_y_offset[3] <= 0) && (ps_sao->u1_y_offset[4] <= 0));
|
|
// clang-format on
|
|
|
|
ihevce_sao_set_avilability(au1_avail_luma, ps_sao_ctxt, ps_tile_params);
|
|
|
|
ps_sao_ctxt->apf_sao_luma[ps_sao->b3_y_type_idx - 2](
|
|
pu1_src_luma,
|
|
luma_src_stride,
|
|
pu1_src_left_luma_buf, /* Pass the pointer to the left luma buffer backed up in the (x-1,y)th ctb */
|
|
pu1_src_top_luma_buf, /* Pass the ptr to the top luma buf backed up in the (x,y-1)th ctb */
|
|
pu1_top_left_luma, /* Top left*/
|
|
pu1_src_top_right_luma, /* Top right*/
|
|
pu1_src_bot_left_luma, /* Bottom left*/
|
|
au1_avail_luma,
|
|
ps_sao->u1_y_offset,
|
|
sao_blk_wd,
|
|
sao_blk_ht);
|
|
|
|
if((ps_sao_ctxt->i4_ctb_y > 0))
|
|
{
|
|
*(pu1_src_top_luma_buf + sao_blk_wd - 1) = u1_src_top_left_luma;
|
|
}
|
|
}
|
|
}
|
|
|
|
if(ps_sao_ctxt->i1_slice_sao_chroma_flag)
|
|
{
|
|
/*initialize the src pointer to current row*/
|
|
chroma_src_stride = ps_sao_ctxt->i4_cur_chroma_recon_stride;
|
|
ctb_size = ps_sao_ctxt->i4_ctb_size;
|
|
|
|
/* 1 extra byte in top buf stride for top left of 1st ctb of every row*/
|
|
//top_buf_stide = ps_sao_ctxt->u4_ctb_aligned_wd + 2;
|
|
ps_sao->u1_cb_offset[0] = 0; /* 0th element is not used */
|
|
ps_sao->u1_cr_offset[0] = 0;
|
|
sao_wd_chroma = ps_sao_ctxt->i4_sao_blk_wd;
|
|
sao_ht_chroma = ps_sao_ctxt->i4_sao_blk_ht / (!u1_is_422 + 1);
|
|
|
|
pu1_src_chroma = ps_sao_ctxt->pu1_cur_chroma_recon_buf;
|
|
/* Pointer to the top luma buffer corresponding to the current ctb row*/
|
|
pu1_src_top_chroma_buf = ps_sao_ctxt->pu1_curr_sao_src_top_chroma;
|
|
// clang-format off
|
|
/* Pointer to left luma buffer corresponding to the current ctb row*/
|
|
pu1_src_left_chroma_buf = ps_sao_ctxt->au1_left_chroma_scratch; //ps_sao_ctxt->au1_sao_src_left_chroma;
|
|
// clang-format on
|
|
/* Pointer to the top right chroma buffer corresponding to the current ctb row*/
|
|
pu1_src_top_right_chroma = pu1_src_top_chroma_buf /*- top_buf_stide*/ + sao_wd_chroma;
|
|
|
|
/* Pointer to the bottom left luma buffer corresponding to the current ctb row*/
|
|
pu1_src_bot_left_chroma =
|
|
ps_sao_ctxt->pu1_frm_chroma_recon_buf +
|
|
(ctb_size >> !u1_is_422) * ps_sao_ctxt->i4_frm_chroma_recon_stride - 2 +
|
|
(ps_sao_ctxt->i4_frm_chroma_recon_stride * ps_sao_ctxt->i4_ctb_y *
|
|
(ctb_size >> !u1_is_422)) +
|
|
(ps_sao_ctxt->i4_ctb_x * ctb_size); /* Bottom left*/
|
|
|
|
/* Back up the top left pixel for (x+1, y+1)th ctb*/
|
|
u1_src_top_left_chroma[0] = *(pu1_src_top_chroma_buf + sao_wd_chroma - 2);
|
|
u1_src_top_left_chroma[1] = *(pu1_src_top_chroma_buf + sao_wd_chroma - 1);
|
|
pu1_top_left_chroma = pu1_src_top_chroma_buf - 2;
|
|
|
|
if(SAO_BAND == ps_sao->b3_cb_type_idx)
|
|
{
|
|
ihevc_sao_band_offset_chroma(
|
|
pu1_src_chroma,
|
|
chroma_src_stride,
|
|
pu1_src_left_chroma_buf, /* Pass the pointer to the left luma buffer backed up in the (x-1,y)th ctb */
|
|
pu1_src_top_chroma_buf, /* Pass the ptr to the top luma buf backed up in the (x,y-1)th ctb */
|
|
pu1_top_left_chroma, /* Top left*/
|
|
ps_sao->b5_cb_band_pos,
|
|
ps_sao->b5_cr_band_pos,
|
|
ps_sao->u1_cb_offset,
|
|
ps_sao->u1_cr_offset,
|
|
sao_wd_chroma,
|
|
sao_ht_chroma);
|
|
|
|
if((ps_sao_ctxt->i4_ctb_y > 0))
|
|
{
|
|
*(pu1_src_top_chroma_buf + sao_wd_chroma - 2) = u1_src_top_left_chroma[0];
|
|
*(pu1_src_top_chroma_buf + sao_wd_chroma - 1) = u1_src_top_left_chroma[1];
|
|
}
|
|
}
|
|
else if(ps_sao->b3_cb_type_idx >= SAO_EDGE_0_DEG)
|
|
{
|
|
/*In case of edge offset, 1st and 2nd offsets are always inferred as offsets
|
|
* corresponding to EO category 1 and 2 which should be always positive
|
|
* And 3rd and 4th offsets are always inferred as offsets corresponding to
|
|
* EO category 3 and 4 which should be negative for all the EO classes(or EO typeidx)
|
|
*/
|
|
ASSERT((ps_sao->u1_cb_offset[1] >= 0) && (ps_sao->u1_cb_offset[2] >= 0));
|
|
ASSERT((ps_sao->u1_cb_offset[3] <= 0) && (ps_sao->u1_cb_offset[4] <= 0));
|
|
|
|
ASSERT((ps_sao->u1_cr_offset[1] >= 0) && (ps_sao->u1_cr_offset[2] >= 0));
|
|
ASSERT((ps_sao->u1_cr_offset[3] <= 0) && (ps_sao->u1_cr_offset[4] <= 0));
|
|
|
|
ihevce_sao_set_avilability(au1_avail_chroma, ps_sao_ctxt, ps_tile_params);
|
|
|
|
ps_sao_ctxt->apf_sao_chroma[ps_sao->b3_cb_type_idx - 2](
|
|
pu1_src_chroma,
|
|
chroma_src_stride,
|
|
pu1_src_left_chroma_buf, /* Pass the pointer to the left luma buffer backed up in the (x-1,y)th ctb */
|
|
pu1_src_top_chroma_buf, /* Pass the ptr to the top luma buf backed up in the (x,y-1)th ctb */
|
|
pu1_top_left_chroma, /* Top left*/
|
|
pu1_src_top_right_chroma, /* Top right*/
|
|
pu1_src_bot_left_chroma, /* Bottom left*/
|
|
au1_avail_chroma,
|
|
ps_sao->u1_cb_offset,
|
|
ps_sao->u1_cr_offset,
|
|
sao_wd_chroma,
|
|
sao_ht_chroma);
|
|
|
|
if((ps_sao_ctxt->i4_ctb_y > 0))
|
|
{
|
|
*(pu1_src_top_chroma_buf + sao_wd_chroma - 2) = u1_src_top_left_chroma[0];
|
|
*(pu1_src_top_chroma_buf + sao_wd_chroma - 1) = u1_src_top_left_chroma[1];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
*******************************************************************************
|
|
*
|
|
* @brief
|
|
* CTB level function to do SAO analysis.
|
|
*
|
|
* @par Description:
|
|
* For a given CTB, sao analysis is done for both luma and chroma.
|
|
*
|
|
*
|
|
* @param[in]
|
|
* ps_sao_ctxt: Pointer to SAO context
|
|
* ps_ctb_enc_loop_out : pointer to ctb level output structure from enc loop
|
|
*
|
|
* @returns
|
|
*
|
|
* @remarks
|
|
* None
|
|
*
|
|
* @Assumptions:
|
|
* 1) Initial Cabac state for current ctb to be sao'ed (i.e (x-1,y-1)th ctb) is assumed to be
|
|
* almost same as cabac state of (x,y)th ctb.
|
|
* 2) Distortion is calculated in spatial domain but lamda used to calculate the cost is
|
|
* in freq domain.
|
|
*******************************************************************************
|
|
*/
|
|
void ihevce_sao_analyse(
|
|
sao_ctxt_t *ps_sao_ctxt,
|
|
ctb_enc_loop_out_t *ps_ctb_enc_loop_out,
|
|
UWORD32 *pu4_frame_rdopt_header_bits,
|
|
ihevce_tile_params_t *ps_tile_params)
|
|
{
|
|
UWORD8 *pu1_luma_scratch_buf;
|
|
UWORD8 *pu1_chroma_scratch_buf;
|
|
UWORD8 *pu1_src_luma, *pu1_recon_luma;
|
|
UWORD8 *pu1_src_chroma, *pu1_recon_chroma;
|
|
WORD32 luma_src_stride, luma_recon_stride, ctb_size, ctb_wd, ctb_ht;
|
|
WORD32 chroma_src_stride, chroma_recon_stride;
|
|
WORD32 i4_luma_scratch_buf_stride;
|
|
WORD32 i4_chroma_scratch_buf_stride;
|
|
sao_ctxt_t s_sao_ctxt;
|
|
UWORD32 ctb_bits = 0, distortion = 0, curr_cost = 0, best_cost = 0;
|
|
LWORD64 i8_cl_ssd_lambda_qf, i8_cl_ssd_lambda_chroma_qf;
|
|
WORD32 rdo_cand, num_luma_rdo_cand = 0, num_rdo_cand = 0;
|
|
WORD32 curr_buf_idx, best_buf_idx, best_cand_idx;
|
|
WORD32 row;
|
|
WORD32 edgeidx;
|
|
WORD32 acc_error_category[5] = { 0, 0, 0, 0, 0 }, category_count[5] = { 0, 0, 0, 0, 0 };
|
|
sao_enc_t s_best_luma_chroma_cand;
|
|
WORD32 best_ctb_sao_bits = 0;
|
|
#if DISABLE_SAO_WHEN_NOISY && !defined(ENC_VER_v2)
|
|
UWORD8 u1_force_no_offset =
|
|
ps_sao_ctxt
|
|
->ps_ctb_data
|
|
[ps_sao_ctxt->i4_ctb_x + ps_sao_ctxt->i4_ctb_data_stride * ps_sao_ctxt->i4_ctb_y]
|
|
.s_ctb_noise_params.i4_noise_present;
|
|
#endif
|
|
UWORD8 u1_is_422 = (ps_sao_ctxt->ps_sps->i1_chroma_format_idc == 2);
|
|
|
|
*pu4_frame_rdopt_header_bits = 0;
|
|
|
|
ctb_size = ps_sao_ctxt->i4_ctb_size;
|
|
ctb_wd = ps_sao_ctxt->i4_sao_blk_wd;
|
|
ctb_ht = ps_sao_ctxt->i4_sao_blk_ht;
|
|
|
|
s_sao_ctxt = ps_sao_ctxt[0];
|
|
|
|
/* Memset the best luma_chroma_cand structure to avoid asserts in debug mode*/
|
|
memset(&s_best_luma_chroma_cand, 0, sizeof(sao_enc_t));
|
|
|
|
/* Initialize the pointer and strides for luma buffers*/
|
|
pu1_recon_luma = ps_sao_ctxt->pu1_cur_luma_recon_buf;
|
|
luma_recon_stride = ps_sao_ctxt->i4_cur_luma_recon_stride;
|
|
|
|
pu1_src_luma = ps_sao_ctxt->pu1_cur_luma_src_buf;
|
|
luma_src_stride = ps_sao_ctxt->i4_cur_luma_src_stride;
|
|
i4_luma_scratch_buf_stride = SCRATCH_BUF_STRIDE;
|
|
|
|
/* Initialize the pointer and strides for luma buffers*/
|
|
pu1_recon_chroma = ps_sao_ctxt->pu1_cur_chroma_recon_buf;
|
|
chroma_recon_stride = ps_sao_ctxt->i4_cur_chroma_recon_stride;
|
|
|
|
pu1_src_chroma = ps_sao_ctxt->pu1_cur_chroma_src_buf;
|
|
chroma_src_stride = ps_sao_ctxt->i4_cur_chroma_src_stride;
|
|
i4_chroma_scratch_buf_stride = SCRATCH_BUF_STRIDE;
|
|
|
|
i8_cl_ssd_lambda_qf = ps_sao_ctxt->i8_cl_ssd_lambda_qf;
|
|
i8_cl_ssd_lambda_chroma_qf = ps_sao_ctxt->i8_cl_ssd_lambda_chroma_qf;
|
|
|
|
/*****************************************************/
|
|
/********************RDO FOR LUMA CAND****************/
|
|
/*****************************************************/
|
|
|
|
#if !DISABLE_SAO_WHEN_NOISY
|
|
if(ps_sao_ctxt->ps_slice_hdr->i1_slice_sao_luma_flag)
|
|
#else
|
|
if(ps_sao_ctxt->ps_slice_hdr->i1_slice_sao_luma_flag && !u1_force_no_offset)
|
|
#endif
|
|
{
|
|
/* Candidate for Edge offset SAO*/
|
|
/* Following is the convention for curr pixel and
|
|
* two neighbouring pixels for 0 deg, 90 deg, 135 deg and 45 deg */
|
|
/*
|
|
* 0 deg : a c b 90 deg: a 135 deg: a 45 deg: a
|
|
* c c c
|
|
* b b b
|
|
*/
|
|
|
|
/* 0 deg SAO CAND*/
|
|
/* Reset the error and edge count*/
|
|
for(edgeidx = 0; edgeidx < 5; edgeidx++)
|
|
{
|
|
acc_error_category[edgeidx] = 0;
|
|
category_count[edgeidx] = 0;
|
|
}
|
|
|
|
/* Call the funciton to populate the EO parameter for this ctb for 0 deg EO class*/
|
|
// clang-format off
|
|
ps_sao_ctxt->ps_cmn_utils_optimised_function_list->pf_get_luma_eo_sao_params(ps_sao_ctxt, SAO_EDGE_0_DEG,
|
|
acc_error_category, category_count);
|
|
// clang-format on
|
|
// clang-format off
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b3_y_type_idx = SAO_EDGE_0_DEG;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[1] = category_count[0]
|
|
? (CLIP3(acc_error_category[0] / category_count[0], 0, 7))
|
|
: 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[2] = category_count[1]
|
|
? (CLIP3(acc_error_category[1] / category_count[1], 0, 7))
|
|
: 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[3] = category_count[3]
|
|
? (CLIP3(acc_error_category[3] / category_count[3], -7, 0))
|
|
: 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[4] =category_count[4]
|
|
? (CLIP3(acc_error_category[4] / category_count[4], -7, 0))
|
|
: 0;
|
|
// clang-format on
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b5_y_band_pos = 0;
|
|
// clang-format off
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b3_cb_type_idx = SAO_NONE;
|
|
// clang-format on
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[1] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[2] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[3] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[4] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b5_cb_band_pos = 0;
|
|
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b3_cr_type_idx = SAO_NONE;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[1] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[2] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[3] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[4] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b5_cr_band_pos = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b1_sao_merge_left_flag = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b1_sao_merge_up_flag = 0;
|
|
|
|
num_luma_rdo_cand++;
|
|
|
|
/* 90 degree SAO CAND*/
|
|
for(edgeidx = 0; edgeidx < 5; edgeidx++)
|
|
{
|
|
acc_error_category[edgeidx] = 0;
|
|
category_count[edgeidx] = 0;
|
|
}
|
|
|
|
/* Call the funciton to populate the EO parameter for this ctb for 90 deg EO class*/
|
|
// clang-format off
|
|
ps_sao_ctxt->ps_cmn_utils_optimised_function_list->pf_get_luma_eo_sao_params(ps_sao_ctxt, SAO_EDGE_90_DEG,
|
|
acc_error_category, category_count);
|
|
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b3_y_type_idx = SAO_EDGE_90_DEG;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[1] = category_count[0]
|
|
? (CLIP3(acc_error_category[0] / category_count[0], 0, 7))
|
|
: 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[2] = category_count[1]
|
|
? (CLIP3(acc_error_category[1] / category_count[1], 0, 7))
|
|
: 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[3] = category_count[3]
|
|
? (CLIP3(acc_error_category[3] / category_count[3], -7, 0))
|
|
: 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[4] = category_count[4]
|
|
? (CLIP3(acc_error_category[4] / category_count[4], -7, 0))
|
|
: 0;
|
|
// clang-format on
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b5_y_band_pos = 0;
|
|
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b3_cb_type_idx = SAO_NONE;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[1] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[2] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[3] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[4] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b5_cb_band_pos = 0;
|
|
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b3_cr_type_idx = SAO_NONE;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[1] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[2] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[3] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[4] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b5_cr_band_pos = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b1_sao_merge_left_flag = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b1_sao_merge_up_flag = 0;
|
|
|
|
num_luma_rdo_cand++;
|
|
|
|
/* 135 degree SAO CAND*/
|
|
for(edgeidx = 0; edgeidx < 5; edgeidx++)
|
|
{
|
|
acc_error_category[edgeidx] = 0;
|
|
category_count[edgeidx] = 0;
|
|
}
|
|
|
|
/* Call the funciton to populate the EO parameter for this ctb for 135 deg EO class*/
|
|
// clang-format off
|
|
ps_sao_ctxt->ps_cmn_utils_optimised_function_list->pf_get_luma_eo_sao_params(ps_sao_ctxt, SAO_EDGE_135_DEG,
|
|
acc_error_category, category_count);
|
|
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b3_y_type_idx = SAO_EDGE_135_DEG;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[1] = category_count[0]
|
|
? (CLIP3(acc_error_category[0] / category_count[0], 0, 7))
|
|
: 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[2] = category_count[1]
|
|
? (CLIP3(acc_error_category[1] / category_count[1], 0, 7))
|
|
: 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[3] = category_count[3]
|
|
? (CLIP3(acc_error_category[3] / category_count[3], -7, 0))
|
|
: 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[4] = category_count[4]
|
|
? (CLIP3(acc_error_category[4] / category_count[4], -7, 0))
|
|
: 0;
|
|
// clang-format on
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b5_y_band_pos = 0;
|
|
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b3_cb_type_idx = SAO_NONE;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[1] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[2] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[3] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[4] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b5_cb_band_pos = 0;
|
|
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b3_cr_type_idx = SAO_NONE;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[1] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[2] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[3] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[4] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b5_cr_band_pos = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b1_sao_merge_left_flag = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b1_sao_merge_up_flag = 0;
|
|
|
|
num_luma_rdo_cand++;
|
|
|
|
/* 45 degree SAO CAND*/
|
|
for(edgeidx = 0; edgeidx < 5; edgeidx++)
|
|
{
|
|
acc_error_category[edgeidx] = 0;
|
|
category_count[edgeidx] = 0;
|
|
}
|
|
|
|
/* Call the funciton to populate the EO parameter for this ctb for 45 deg EO class*/
|
|
// clang-format off
|
|
ps_sao_ctxt->ps_cmn_utils_optimised_function_list->pf_get_luma_eo_sao_params(ps_sao_ctxt, SAO_EDGE_45_DEG,
|
|
acc_error_category, category_count);
|
|
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b3_y_type_idx = SAO_EDGE_45_DEG;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[1] = category_count[0]
|
|
? (CLIP3(acc_error_category[0] / category_count[0], 0, 7))
|
|
: 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[2] = category_count[1]
|
|
? (CLIP3(acc_error_category[1] / category_count[1], 0, 7))
|
|
: 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[3] = category_count[3]
|
|
? (CLIP3(acc_error_category[3] / category_count[3], -7, 0))
|
|
: 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_y_offset[4] = category_count[4]
|
|
? (CLIP3(acc_error_category[4] / category_count[4], -7, 0))
|
|
: 0;
|
|
// clang-format on
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b5_y_band_pos = 0;
|
|
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b3_cb_type_idx = SAO_NONE;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[1] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[2] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[3] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cb_offset[4] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b5_cb_band_pos = 0;
|
|
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b3_cr_type_idx = SAO_NONE;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[1] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[2] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[3] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].u1_cr_offset[4] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b5_cr_band_pos = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b1_sao_merge_left_flag = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_luma_rdo_cand].b1_sao_merge_up_flag = 0;
|
|
|
|
num_luma_rdo_cand++;
|
|
|
|
/* First cand will be best cand after 1st iteration*/
|
|
curr_buf_idx = 0;
|
|
best_buf_idx = 1;
|
|
best_cost = 0xFFFFFFFF;
|
|
best_cand_idx = 0;
|
|
|
|
/*Back up the top pixels for (x,y+1)th ctb*/
|
|
if(!ps_sao_ctxt->i4_is_last_ctb_row)
|
|
{
|
|
memcpy(
|
|
ps_sao_ctxt->pu1_curr_sao_src_top_luma + ps_sao_ctxt->i4_frm_top_luma_buf_stride,
|
|
pu1_recon_luma + luma_recon_stride * (ctb_size - 1),
|
|
ps_sao_ctxt->i4_sao_blk_wd);
|
|
}
|
|
|
|
for(rdo_cand = 0; rdo_cand < num_luma_rdo_cand; rdo_cand++)
|
|
{
|
|
s_sao_ctxt.ps_sao = &ps_sao_ctxt->as_sao_rd_cand[rdo_cand];
|
|
|
|
/* This memcpy is required because cabac uses parameters from this structure
|
|
* to evaluate bits and this structure ptr is sent to cabac through
|
|
* "ihevce_cabac_rdo_encode_sao" function
|
|
*/
|
|
memcpy(&ps_ctb_enc_loop_out->s_sao, s_sao_ctxt.ps_sao, sizeof(sao_enc_t));
|
|
|
|
/* Copy the left pixels to the scratch buffer for evry rdo cand because its
|
|
overwritten by the sao leaf level function for next ctb*/
|
|
memcpy(
|
|
s_sao_ctxt.au1_left_luma_scratch,
|
|
ps_sao_ctxt->au1_sao_src_left_luma,
|
|
ps_sao_ctxt->i4_sao_blk_ht);
|
|
|
|
/* Copy the top and top left pixels to the scratch buffer for evry rdo cand because its
|
|
overwritten by the sao leaf level function for next ctb*/
|
|
memcpy(
|
|
s_sao_ctxt.au1_top_luma_scratch,
|
|
ps_sao_ctxt->pu1_curr_sao_src_top_luma - 1,
|
|
ps_sao_ctxt->i4_sao_blk_wd + 2);
|
|
s_sao_ctxt.pu1_curr_sao_src_top_luma = s_sao_ctxt.au1_top_luma_scratch + 1;
|
|
|
|
pu1_luma_scratch_buf = ps_sao_ctxt->au1_sao_luma_scratch[curr_buf_idx];
|
|
|
|
ASSERT(
|
|
(abs(s_sao_ctxt.ps_sao->u1_y_offset[1]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_y_offset[2]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_y_offset[3]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_y_offset[4]) <= 7));
|
|
ASSERT(
|
|
(abs(s_sao_ctxt.ps_sao->u1_cb_offset[1]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_cb_offset[2]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_cb_offset[3]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_cb_offset[4]) <= 7));
|
|
ASSERT(
|
|
(abs(s_sao_ctxt.ps_sao->u1_cr_offset[1]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_cr_offset[2]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_cr_offset[3]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_cr_offset[4]) <= 7));
|
|
ASSERT(
|
|
(s_sao_ctxt.ps_sao->b5_y_band_pos <= 28) &&
|
|
(s_sao_ctxt.ps_sao->b5_cb_band_pos <= 28) &&
|
|
(s_sao_ctxt.ps_sao->b5_cr_band_pos <= 28));
|
|
|
|
/* Copy the deblocked recon data to scratch buffer to do sao*/
|
|
|
|
ps_sao_ctxt->ps_cmn_utils_optimised_function_list->pf_copy_2d(
|
|
pu1_luma_scratch_buf,
|
|
i4_luma_scratch_buf_stride,
|
|
pu1_recon_luma,
|
|
luma_recon_stride,
|
|
SCRATCH_BUF_STRIDE,
|
|
ctb_ht + 1);
|
|
|
|
s_sao_ctxt.pu1_cur_luma_recon_buf = pu1_luma_scratch_buf;
|
|
s_sao_ctxt.i4_cur_luma_recon_stride = i4_luma_scratch_buf_stride;
|
|
|
|
s_sao_ctxt.i1_slice_sao_luma_flag = s_sao_ctxt.ps_slice_hdr->i1_slice_sao_luma_flag;
|
|
s_sao_ctxt.i1_slice_sao_chroma_flag = 0;
|
|
|
|
ihevce_sao_ctb(&s_sao_ctxt, ps_tile_params);
|
|
|
|
/* Calculate the distortion between sao'ed ctb and original src ctb*/
|
|
// clang-format off
|
|
distortion =
|
|
ps_sao_ctxt->ps_cmn_utils_optimised_function_list->pf_ssd_calculator(pu1_src_luma,
|
|
s_sao_ctxt.pu1_cur_luma_recon_buf, luma_src_stride,
|
|
s_sao_ctxt.i4_cur_luma_recon_stride, ctb_wd, ctb_ht, NULL_PLANE);
|
|
// clang-format on
|
|
|
|
ps_sao_ctxt->ps_rdopt_entropy_ctxt->i4_curr_buf_idx = curr_buf_idx;
|
|
ctb_bits = ihevce_cabac_rdo_encode_sao(
|
|
ps_sao_ctxt->ps_rdopt_entropy_ctxt, ps_ctb_enc_loop_out);
|
|
|
|
/* Calculate the cost as D+(lamda)*R */
|
|
curr_cost = distortion +
|
|
COMPUTE_RATE_COST_CLIP30(ctb_bits, i8_cl_ssd_lambda_qf, LAMBDA_Q_SHIFT);
|
|
|
|
if(curr_cost < best_cost)
|
|
{
|
|
best_cost = curr_cost;
|
|
best_buf_idx = ps_sao_ctxt->ps_rdopt_entropy_ctxt->i4_curr_buf_idx;
|
|
best_cand_idx = rdo_cand;
|
|
curr_buf_idx = !curr_buf_idx;
|
|
}
|
|
}
|
|
|
|
/* Copy the sao parameters of the best luma cand into the luma_chroma cnad structure for next stage of RDO
|
|
* between luma_chroma combined cand, NO SAO cand, LEFT and TOP merge cand
|
|
*/
|
|
s_best_luma_chroma_cand.b3_y_type_idx =
|
|
ps_sao_ctxt->as_sao_rd_cand[best_cand_idx].b3_y_type_idx;
|
|
s_best_luma_chroma_cand.u1_y_offset[1] =
|
|
ps_sao_ctxt->as_sao_rd_cand[best_cand_idx].u1_y_offset[1];
|
|
s_best_luma_chroma_cand.u1_y_offset[2] =
|
|
ps_sao_ctxt->as_sao_rd_cand[best_cand_idx].u1_y_offset[2];
|
|
s_best_luma_chroma_cand.u1_y_offset[3] =
|
|
ps_sao_ctxt->as_sao_rd_cand[best_cand_idx].u1_y_offset[3];
|
|
s_best_luma_chroma_cand.u1_y_offset[4] =
|
|
ps_sao_ctxt->as_sao_rd_cand[best_cand_idx].u1_y_offset[4];
|
|
s_best_luma_chroma_cand.b5_y_band_pos =
|
|
ps_sao_ctxt->as_sao_rd_cand[best_cand_idx].b5_y_band_pos;
|
|
}
|
|
else
|
|
{
|
|
/*Back up the top pixels for (x,y+1)th ctb*/
|
|
if(!ps_sao_ctxt->i4_is_last_ctb_row)
|
|
{
|
|
memcpy(
|
|
ps_sao_ctxt->pu1_curr_sao_src_top_luma + ps_sao_ctxt->i4_frm_top_luma_buf_stride,
|
|
pu1_recon_luma + luma_recon_stride * (ctb_size - 1),
|
|
ps_sao_ctxt->i4_sao_blk_wd);
|
|
}
|
|
|
|
s_best_luma_chroma_cand.b3_y_type_idx = SAO_NONE;
|
|
s_best_luma_chroma_cand.u1_y_offset[1] = 0;
|
|
s_best_luma_chroma_cand.u1_y_offset[2] = 0;
|
|
s_best_luma_chroma_cand.u1_y_offset[3] = 0;
|
|
s_best_luma_chroma_cand.u1_y_offset[4] = 0;
|
|
s_best_luma_chroma_cand.b5_y_band_pos = 0;
|
|
s_best_luma_chroma_cand.b1_sao_merge_left_flag = 0;
|
|
s_best_luma_chroma_cand.b1_sao_merge_up_flag = 0;
|
|
|
|
s_best_luma_chroma_cand.b3_cb_type_idx = SAO_NONE;
|
|
s_best_luma_chroma_cand.u1_cb_offset[1] = 0;
|
|
s_best_luma_chroma_cand.u1_cb_offset[2] = 0;
|
|
s_best_luma_chroma_cand.u1_cb_offset[3] = 0;
|
|
s_best_luma_chroma_cand.u1_cb_offset[4] = 0;
|
|
s_best_luma_chroma_cand.b5_cb_band_pos = 0;
|
|
|
|
s_best_luma_chroma_cand.b3_cr_type_idx = SAO_NONE;
|
|
s_best_luma_chroma_cand.u1_cr_offset[1] = 0;
|
|
s_best_luma_chroma_cand.u1_cr_offset[2] = 0;
|
|
s_best_luma_chroma_cand.u1_cr_offset[3] = 0;
|
|
s_best_luma_chroma_cand.u1_cr_offset[4] = 0;
|
|
s_best_luma_chroma_cand.b5_cr_band_pos = 0;
|
|
}
|
|
/*****************************************************/
|
|
/********************RDO FOR CHROMA CAND**************/
|
|
/*****************************************************/
|
|
#if !DISABLE_SAO_WHEN_NOISY
|
|
if(ps_sao_ctxt->ps_slice_hdr->i1_slice_sao_chroma_flag)
|
|
#else
|
|
if(ps_sao_ctxt->ps_slice_hdr->i1_slice_sao_chroma_flag && !u1_force_no_offset)
|
|
#endif
|
|
{
|
|
/*Back up the top pixels for (x,y+1)th ctb*/
|
|
if(!ps_sao_ctxt->i4_is_last_ctb_row)
|
|
{
|
|
memcpy(
|
|
ps_sao_ctxt->pu1_curr_sao_src_top_chroma +
|
|
ps_sao_ctxt->i4_frm_top_chroma_buf_stride,
|
|
pu1_recon_chroma + chroma_recon_stride * ((ctb_size >> !u1_is_422) - 1),
|
|
ps_sao_ctxt->i4_sao_blk_wd);
|
|
}
|
|
|
|
/* Reset the error and edge count*/
|
|
for(edgeidx = 0; edgeidx < 5; edgeidx++)
|
|
{
|
|
acc_error_category[edgeidx] = 0;
|
|
category_count[edgeidx] = 0;
|
|
}
|
|
// clang-format off
|
|
ps_sao_ctxt->ps_cmn_utils_optimised_function_list->pf_get_chroma_eo_sao_params(ps_sao_ctxt,
|
|
s_best_luma_chroma_cand.b3_y_type_idx, acc_error_category,
|
|
category_count);
|
|
// clang-format on
|
|
|
|
/* Copy the sao parameters of the best luma cand into the luma_chroma cnad structure for next stage of RDO
|
|
* between luma_chroma combined cand, NO SAO cand, LEFT and TOP merge cand
|
|
*/
|
|
// clang-format off
|
|
s_best_luma_chroma_cand.b3_cb_type_idx = s_best_luma_chroma_cand.b3_y_type_idx;
|
|
s_best_luma_chroma_cand.u1_cb_offset[1] = category_count[0]
|
|
? (CLIP3(acc_error_category[0] / category_count[0], 0, 7))
|
|
: 0;
|
|
s_best_luma_chroma_cand.u1_cb_offset[2] = category_count[1]
|
|
? (CLIP3(acc_error_category[1] / category_count[1], 0, 7))
|
|
: 0;
|
|
s_best_luma_chroma_cand.u1_cb_offset[3] = category_count[3]
|
|
? (CLIP3(acc_error_category[3] / category_count[3], -7, 0))
|
|
: 0;
|
|
s_best_luma_chroma_cand.u1_cb_offset[4] = category_count[4]
|
|
? (CLIP3(acc_error_category[4] / category_count[4], -7, 0))
|
|
: 0;
|
|
s_best_luma_chroma_cand.b5_cb_band_pos = 0;
|
|
|
|
s_best_luma_chroma_cand.b3_cr_type_idx = s_best_luma_chroma_cand.b3_y_type_idx;
|
|
s_best_luma_chroma_cand.u1_cr_offset[1] = category_count[0]
|
|
? (CLIP3(acc_error_category[0] / category_count[0], 0, 7))
|
|
: 0;
|
|
s_best_luma_chroma_cand.u1_cr_offset[2] = category_count[1]
|
|
? (CLIP3(acc_error_category[1] / category_count[1], 0, 7))
|
|
: 0;
|
|
s_best_luma_chroma_cand.u1_cr_offset[3] = category_count[3]
|
|
? (CLIP3(acc_error_category[3] / category_count[3], -7, 0))
|
|
: 0;
|
|
s_best_luma_chroma_cand.u1_cr_offset[4] = category_count[4]
|
|
? (CLIP3(acc_error_category[4] / category_count[4], -7, 0))
|
|
: 0;
|
|
// clang-format on
|
|
s_best_luma_chroma_cand.b5_cr_band_pos = 0;
|
|
}
|
|
else
|
|
{
|
|
/*Back up the top pixels for (x,y+1)th ctb*/
|
|
if(!ps_sao_ctxt->i4_is_last_ctb_row)
|
|
{
|
|
memcpy(
|
|
ps_sao_ctxt->pu1_curr_sao_src_top_chroma +
|
|
ps_sao_ctxt->i4_frm_top_chroma_buf_stride,
|
|
pu1_recon_chroma + chroma_recon_stride * ((ctb_size >> !u1_is_422) - 1),
|
|
ps_sao_ctxt->i4_sao_blk_wd);
|
|
}
|
|
|
|
s_best_luma_chroma_cand.b3_cb_type_idx = SAO_NONE;
|
|
s_best_luma_chroma_cand.u1_cb_offset[1] = 0;
|
|
s_best_luma_chroma_cand.u1_cb_offset[2] = 0;
|
|
s_best_luma_chroma_cand.u1_cb_offset[3] = 0;
|
|
s_best_luma_chroma_cand.u1_cb_offset[4] = 0;
|
|
s_best_luma_chroma_cand.b5_cb_band_pos = 0;
|
|
|
|
s_best_luma_chroma_cand.b3_cr_type_idx = SAO_NONE;
|
|
s_best_luma_chroma_cand.u1_cr_offset[1] = 0;
|
|
s_best_luma_chroma_cand.u1_cr_offset[2] = 0;
|
|
s_best_luma_chroma_cand.u1_cr_offset[3] = 0;
|
|
s_best_luma_chroma_cand.u1_cr_offset[4] = 0;
|
|
s_best_luma_chroma_cand.b5_cr_band_pos = 0;
|
|
|
|
s_best_luma_chroma_cand.b1_sao_merge_left_flag = 0;
|
|
s_best_luma_chroma_cand.b1_sao_merge_up_flag = 0;
|
|
}
|
|
|
|
s_best_luma_chroma_cand.b1_sao_merge_left_flag = 0;
|
|
s_best_luma_chroma_cand.b1_sao_merge_up_flag = 0;
|
|
|
|
/*****************************************************/
|
|
/**RDO for Best Luma - Chroma combined, No SAO,*******/
|
|
/*************Left merge and Top merge****************/
|
|
/*****************************************************/
|
|
|
|
/* No SAO cand*/
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].b1_sao_merge_left_flag = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].b1_sao_merge_up_flag = 0;
|
|
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].b3_y_type_idx = SAO_NONE;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].u1_y_offset[1] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].u1_y_offset[2] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].u1_y_offset[3] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].u1_y_offset[4] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].b5_y_band_pos = 0;
|
|
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].b3_cb_type_idx = SAO_NONE;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].u1_cb_offset[1] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].u1_cb_offset[2] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].u1_cb_offset[3] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].u1_cb_offset[4] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].b5_cb_band_pos = 0;
|
|
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].b3_cr_type_idx = SAO_NONE;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].u1_cr_offset[1] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].u1_cr_offset[2] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].u1_cr_offset[3] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].u1_cr_offset[4] = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].b5_cr_band_pos = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].b1_sao_merge_left_flag = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].b1_sao_merge_up_flag = 0;
|
|
|
|
num_rdo_cand++;
|
|
|
|
/* SAO_note_01: If the CTB lies on a tile or a slice boundary, then
|
|
the standard mandates that the merge candidates must be set to unavailable.
|
|
Hence, check for tile boundary condition by reading
|
|
s_ctb_nbr_avail_flags.u1_left_avail rather than frame position of CTB.
|
|
A special case: Merge-candidates should be available at dependent-slices boundaries.
|
|
Search for <SAO_note_01> in workspace to know more */
|
|
|
|
#if !DISABLE_SAO_WHEN_NOISY
|
|
if(1)
|
|
#else
|
|
if(!u1_force_no_offset)
|
|
#endif
|
|
{
|
|
/* Merge left cand*/
|
|
if(ps_ctb_enc_loop_out->s_ctb_nbr_avail_flags.u1_left_avail)
|
|
{
|
|
memcpy(
|
|
&ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand],
|
|
&ps_sao_ctxt->s_left_ctb_sao,
|
|
sizeof(sao_enc_t));
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].b1_sao_merge_left_flag = 1;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].b1_sao_merge_up_flag = 0;
|
|
num_rdo_cand++;
|
|
}
|
|
|
|
/* Merge top cand*/
|
|
if(ps_ctb_enc_loop_out->s_ctb_nbr_avail_flags.u1_top_avail)
|
|
{
|
|
memcpy(
|
|
&ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand],
|
|
(ps_sao_ctxt->ps_top_ctb_sao - ps_sao_ctxt->u4_num_ctbs_horz),
|
|
sizeof(sao_enc_t));
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].b1_sao_merge_left_flag = 0;
|
|
ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand].b1_sao_merge_up_flag = 1;
|
|
num_rdo_cand++;
|
|
}
|
|
|
|
/* Best luma-chroma candidate*/
|
|
memcpy(
|
|
&ps_sao_ctxt->as_sao_rd_cand[num_rdo_cand],
|
|
&s_best_luma_chroma_cand,
|
|
sizeof(sao_enc_t));
|
|
num_rdo_cand++;
|
|
}
|
|
|
|
{
|
|
UWORD32 luma_distortion = 0, chroma_distortion = 0;
|
|
/* First cand will be best cand after 1st iteration*/
|
|
curr_buf_idx = 0;
|
|
best_buf_idx = 1;
|
|
best_cost = 0xFFFFFFFF;
|
|
best_cand_idx = 0;
|
|
|
|
for(rdo_cand = 0; rdo_cand < num_rdo_cand; rdo_cand++)
|
|
{
|
|
s_sao_ctxt.ps_sao = &ps_sao_ctxt->as_sao_rd_cand[rdo_cand];
|
|
|
|
distortion = 0;
|
|
|
|
/* This memcpy is required because cabac uses parameters from this structure
|
|
* to evaluate bits and this structure ptr is sent to cabac through
|
|
* "ihevce_cabac_rdo_encode_sao" function
|
|
*/
|
|
memcpy(&ps_ctb_enc_loop_out->s_sao, s_sao_ctxt.ps_sao, sizeof(sao_enc_t));
|
|
|
|
if(ps_sao_ctxt->ps_slice_hdr->i1_slice_sao_luma_flag)
|
|
{
|
|
/* Copy the left pixels to the scratch buffer for evry rdo cand because its
|
|
overwritten by the sao leaf level function for next ctb*/
|
|
memcpy(
|
|
s_sao_ctxt.au1_left_luma_scratch,
|
|
ps_sao_ctxt->au1_sao_src_left_luma,
|
|
ps_sao_ctxt->i4_sao_blk_ht);
|
|
|
|
/* Copy the top and top left pixels to the scratch buffer for evry rdo cand because its
|
|
overwritten by the sao leaf level function for next ctb*/
|
|
memcpy(
|
|
s_sao_ctxt.au1_top_luma_scratch,
|
|
ps_sao_ctxt->pu1_curr_sao_src_top_luma - 1,
|
|
ps_sao_ctxt->i4_sao_blk_wd + 2);
|
|
s_sao_ctxt.pu1_curr_sao_src_top_luma = s_sao_ctxt.au1_top_luma_scratch + 1;
|
|
|
|
pu1_luma_scratch_buf = ps_sao_ctxt->au1_sao_luma_scratch[curr_buf_idx];
|
|
|
|
/* Copy the deblocked recon data to scratch buffer to do sao*/
|
|
|
|
ps_sao_ctxt->ps_cmn_utils_optimised_function_list->pf_copy_2d(
|
|
pu1_luma_scratch_buf,
|
|
i4_luma_scratch_buf_stride,
|
|
pu1_recon_luma,
|
|
luma_recon_stride,
|
|
SCRATCH_BUF_STRIDE,
|
|
ctb_ht + 1);
|
|
s_sao_ctxt.pu1_cur_luma_recon_buf = pu1_luma_scratch_buf;
|
|
s_sao_ctxt.i4_cur_luma_recon_stride = i4_luma_scratch_buf_stride;
|
|
|
|
ASSERT(
|
|
(abs(s_sao_ctxt.ps_sao->u1_y_offset[1]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_y_offset[2]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_y_offset[3]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_y_offset[4]) <= 7));
|
|
}
|
|
if(ps_sao_ctxt->ps_slice_hdr->i1_slice_sao_chroma_flag)
|
|
{
|
|
/* Copy the left pixels to the scratch buffer for evry rdo cand because its
|
|
overwritten by the sao leaf level function for next ctb*/
|
|
memcpy(
|
|
s_sao_ctxt.au1_left_chroma_scratch,
|
|
ps_sao_ctxt->au1_sao_src_left_chroma,
|
|
(ps_sao_ctxt->i4_sao_blk_ht >> !u1_is_422) * 2);
|
|
|
|
/* Copy the top and top left pixels to the scratch buffer for evry rdo cand because its
|
|
overwritten by the sao leaf level function for next ctb*/
|
|
memcpy(
|
|
s_sao_ctxt.au1_top_chroma_scratch,
|
|
ps_sao_ctxt->pu1_curr_sao_src_top_chroma - 2,
|
|
ps_sao_ctxt->i4_sao_blk_wd + 4);
|
|
|
|
s_sao_ctxt.pu1_curr_sao_src_top_chroma = s_sao_ctxt.au1_top_chroma_scratch + 2;
|
|
|
|
pu1_chroma_scratch_buf = ps_sao_ctxt->au1_sao_chroma_scratch[curr_buf_idx];
|
|
|
|
/* Copy the deblocked recon data to scratch buffer to do sao*/
|
|
|
|
ps_sao_ctxt->ps_cmn_utils_optimised_function_list->pf_copy_2d(
|
|
pu1_chroma_scratch_buf,
|
|
i4_chroma_scratch_buf_stride,
|
|
pu1_recon_chroma,
|
|
chroma_recon_stride,
|
|
SCRATCH_BUF_STRIDE,
|
|
(ctb_ht >> !u1_is_422) + 1);
|
|
|
|
s_sao_ctxt.pu1_cur_chroma_recon_buf = pu1_chroma_scratch_buf;
|
|
s_sao_ctxt.i4_cur_chroma_recon_stride = i4_chroma_scratch_buf_stride;
|
|
|
|
ASSERT(
|
|
(abs(s_sao_ctxt.ps_sao->u1_cb_offset[1]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_cb_offset[2]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_cb_offset[3]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_cb_offset[4]) <= 7));
|
|
ASSERT(
|
|
(abs(s_sao_ctxt.ps_sao->u1_cr_offset[1]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_cr_offset[2]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_cr_offset[3]) <= 7) &&
|
|
(abs(s_sao_ctxt.ps_sao->u1_cr_offset[4]) <= 7));
|
|
}
|
|
|
|
ASSERT(
|
|
(s_sao_ctxt.ps_sao->b5_y_band_pos <= 28) &&
|
|
(s_sao_ctxt.ps_sao->b5_cb_band_pos <= 28) &&
|
|
(s_sao_ctxt.ps_sao->b5_cr_band_pos <= 28));
|
|
|
|
s_sao_ctxt.i1_slice_sao_luma_flag = s_sao_ctxt.ps_slice_hdr->i1_slice_sao_luma_flag;
|
|
s_sao_ctxt.i1_slice_sao_chroma_flag = s_sao_ctxt.ps_slice_hdr->i1_slice_sao_chroma_flag;
|
|
|
|
ihevce_sao_ctb(&s_sao_ctxt, ps_tile_params);
|
|
|
|
if(ps_sao_ctxt->ps_slice_hdr->i1_slice_sao_luma_flag)
|
|
{ // clang-format off
|
|
luma_distortion =
|
|
ps_sao_ctxt->ps_cmn_utils_optimised_function_list->pf_ssd_calculator(pu1_src_luma,
|
|
s_sao_ctxt.pu1_cur_luma_recon_buf, luma_src_stride,
|
|
s_sao_ctxt.i4_cur_luma_recon_stride, ctb_wd,
|
|
ctb_ht,
|
|
NULL_PLANE);
|
|
} // clang-format on
|
|
|
|
if(ps_sao_ctxt->ps_slice_hdr->i1_slice_sao_chroma_flag)
|
|
{ // clang-format off
|
|
chroma_distortion =
|
|
ps_sao_ctxt->ps_cmn_utils_optimised_function_list->pf_ssd_calculator(pu1_src_chroma,
|
|
s_sao_ctxt.pu1_cur_chroma_recon_buf,
|
|
chroma_src_stride,
|
|
s_sao_ctxt.i4_cur_chroma_recon_stride, ctb_wd,
|
|
(ctb_ht >> !u1_is_422),
|
|
NULL_PLANE);
|
|
} // clang-format on
|
|
|
|
/*chroma distortion is added after correction because of lambda difference*/
|
|
distortion =
|
|
luma_distortion +
|
|
(UWORD32)(chroma_distortion * (i8_cl_ssd_lambda_qf / i8_cl_ssd_lambda_chroma_qf));
|
|
|
|
ps_sao_ctxt->ps_rdopt_entropy_ctxt->i4_curr_buf_idx = curr_buf_idx;
|
|
ctb_bits = ihevce_cabac_rdo_encode_sao(
|
|
ps_sao_ctxt->ps_rdopt_entropy_ctxt, ps_ctb_enc_loop_out);
|
|
|
|
/* Calculate the cost as D+(lamda)*R */
|
|
curr_cost = distortion +
|
|
COMPUTE_RATE_COST_CLIP30(ctb_bits, i8_cl_ssd_lambda_qf, LAMBDA_Q_SHIFT);
|
|
|
|
if(curr_cost < best_cost)
|
|
{
|
|
best_ctb_sao_bits = ctb_bits;
|
|
best_cost = curr_cost;
|
|
best_buf_idx = ps_sao_ctxt->ps_rdopt_entropy_ctxt->i4_curr_buf_idx;
|
|
best_cand_idx = rdo_cand;
|
|
curr_buf_idx = !curr_buf_idx;
|
|
}
|
|
}
|
|
/*Adding sao bits to header bits*/
|
|
*pu4_frame_rdopt_header_bits = best_ctb_sao_bits;
|
|
|
|
ihevce_update_best_sao_cabac_state(ps_sao_ctxt->ps_rdopt_entropy_ctxt, best_buf_idx);
|
|
|
|
/* store the sao parameters of curr ctb for top merge and left merge*/
|
|
memcpy(
|
|
ps_sao_ctxt->ps_top_ctb_sao,
|
|
&ps_sao_ctxt->as_sao_rd_cand[best_cand_idx],
|
|
sizeof(sao_enc_t));
|
|
memcpy(
|
|
&ps_sao_ctxt->s_left_ctb_sao,
|
|
&ps_sao_ctxt->as_sao_rd_cand[best_cand_idx],
|
|
sizeof(sao_enc_t));
|
|
|
|
/* Copy the sao parameters of winning candidate into the structure which will be sent to entropy thrd*/
|
|
memcpy(
|
|
&ps_ctb_enc_loop_out->s_sao,
|
|
&ps_sao_ctxt->as_sao_rd_cand[best_cand_idx],
|
|
sizeof(sao_enc_t));
|
|
|
|
if(!ps_sao_ctxt->i4_is_last_ctb_col)
|
|
{
|
|
/* Update left luma buffer for next ctb */
|
|
for(row = 0; row < ps_sao_ctxt->i4_sao_blk_ht; row++)
|
|
{
|
|
ps_sao_ctxt->au1_sao_src_left_luma[row] =
|
|
ps_sao_ctxt->pu1_cur_luma_recon_buf
|
|
[row * ps_sao_ctxt->i4_cur_luma_recon_stride +
|
|
(ps_sao_ctxt->i4_sao_blk_wd - 1)];
|
|
}
|
|
}
|
|
|
|
if(!ps_sao_ctxt->i4_is_last_ctb_col)
|
|
{
|
|
/* Update left chroma buffer for next ctb */
|
|
for(row = 0; row < (ps_sao_ctxt->i4_sao_blk_ht >> 1); row++)
|
|
{
|
|
*(UWORD16 *)(ps_sao_ctxt->au1_sao_src_left_chroma + row * 2) =
|
|
*(UWORD16 *)(ps_sao_ctxt->pu1_cur_chroma_recon_buf +
|
|
row * ps_sao_ctxt->i4_cur_chroma_recon_stride +
|
|
(ps_sao_ctxt->i4_sao_blk_wd - 2));
|
|
}
|
|
}
|
|
|
|
if(ps_sao_ctxt->ps_slice_hdr->i1_slice_sao_luma_flag)
|
|
{
|
|
/* Copy the sao'ed output of the best candidate to the recon buffer*/
|
|
|
|
ps_sao_ctxt->ps_cmn_utils_optimised_function_list->pf_copy_2d(
|
|
ps_sao_ctxt->pu1_cur_luma_recon_buf,
|
|
ps_sao_ctxt->i4_cur_luma_recon_stride,
|
|
ps_sao_ctxt->au1_sao_luma_scratch[best_buf_idx],
|
|
i4_luma_scratch_buf_stride,
|
|
ctb_wd,
|
|
ctb_ht);
|
|
}
|
|
if(ps_sao_ctxt->ps_slice_hdr->i1_slice_sao_chroma_flag)
|
|
{
|
|
/* Copy the sao'ed output of the best candidate to the chroma recon buffer*/
|
|
|
|
ps_sao_ctxt->ps_cmn_utils_optimised_function_list->pf_copy_2d(
|
|
ps_sao_ctxt->pu1_cur_chroma_recon_buf,
|
|
ps_sao_ctxt->i4_cur_chroma_recon_stride,
|
|
ps_sao_ctxt->au1_sao_chroma_scratch[best_buf_idx],
|
|
i4_chroma_scratch_buf_stride,
|
|
ctb_wd,
|
|
ctb_ht >> !u1_is_422);
|
|
}
|
|
}
|
|
}
|