wrapper/libhevc-mirror
1
0
Fork 0
mirror of https://github.com/ittiam-systems/libhevc.git synced 2026-08-13 22:32:34 +07:00
Code Issues Projects Releases Packages Wiki Activity Actions 1
libhevc-mirror/decoder/ihevcd_process_slice.c
Harish Mahendrakar 18fc8f6b52 Added support for 420 10-bit decoding. 
This is the initial version and doesn't enable SIMD functions yet.
Has not been tested extensively yet.
This is not ready to be merged into main branch yet.
2026-08-07 03:21:32 +00:00

1735 lines
74 KiB
C
Raw Blame History

/******************************************************************************
*
* 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_process_slice.c
*
* @brief
* Contains functions for processing slice data
*
* @author
* Harish
*
* @par List of Functions:
*
* @remarks
* None
*
*******************************************************************************
*/
/*****************************************************************************/
/* File Includes */
/*****************************************************************************/
#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_padding.h"
#include "ihevc_iquant_itrans_recon.h"
#include "ihevc_chroma_iquant_itrans_recon.h"
#include "ihevc_recon.h"
#include "ihevc_chroma_recon.h"
#include "ihevc_iquant_recon.h"
#include "ihevc_chroma_iquant_recon.h"
#include "ihevc_intra_pred.h"
#include "ihevc_error.h"
#include "ihevc_common_tables.h"
#include "ihevc_quant_tables.h"
#include "ihevcd_common_tables.h"
#include "ihevcd_profile.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 "ihevcd_debug.h"
#include "ihevcd_get_mv.h"
#include "ihevcd_inter_pred.h"
#include "ihevcd_iquant_itrans_recon_ctb.h"
#include "ihevcd_boundary_strength.h"
#include "ihevcd_deblk.h"
#include "ihevcd_fmt_conv.h"
#include "ihevcd_sao.h"
#include "ihevcd_profile.h"
IHEVCD_ERROR_T ihevcd_fmt_conv(codec_t *ps_codec,
process_ctxt_t *ps_proc,
UWORD8 *pu1_y_dst,
UWORD8 *pu1_u_dst,
UWORD8 *pu1_v_dst,
WORD32 cur_row,
WORD32 num_rows);
typedef enum
{
PROC_ALL,
PROC_INTER_PRED,
PROC_RECON,
PROC_DEBLK,
PROC_SAO
}proc_type_t;
void ihevcd_proc_map_check(process_ctxt_t *ps_proc, proc_type_t proc_type, WORD32 nctb)
{
tile_t *ps_tile = ps_proc->ps_tile;
sps_t *ps_sps = ps_proc->ps_sps;
pps_t *ps_pps = ps_proc->ps_pps;
codec_t *ps_codec = ps_proc->ps_codec;
WORD32 idx;
WORD32 nop_cnt;
WORD32 bit_pos = proc_type;
WORD32 bit_mask = (1 << bit_pos);
if(ps_proc->i4_check_proc_status)
{
nop_cnt = PROC_NOP_CNT;
while(1)
{
volatile UWORD8 *pu1_buf;
volatile WORD32 status;
status = 1;
/* Check if all dependencies for the next nCTBs are met */
{
WORD32 x_pos;
{
/* Check if the top right of next nCTBs are processed */
if(ps_proc->i4_ctb_y > 0)
{
x_pos = (ps_proc->i4_ctb_tile_x + nctb);
idx = MIN(x_pos, (ps_tile->u2_wd - 1));
/* Check if top-right CTB for the last CTB in nCTB is within the tile */
{
idx += ps_tile->u1_pos_x;
idx += ((ps_proc->i4_ctb_y - 1)
* ps_sps->i2_pic_wd_in_ctb);
pu1_buf = (ps_codec->pu1_proc_map + idx);
status = *pu1_buf & bit_mask;
}
}
}
/* If tiles are enabled, then test left and top-left as well */
ps_pps = ps_proc->ps_pps;
if(ps_pps->i1_tiles_enabled_flag)
{
/*Check if left ctb is processed*/
if((ps_proc->i4_ctb_x > 0) && ((0 != status)))
{
x_pos = ps_tile->u1_pos_x + ps_proc->i4_ctb_tile_x - 1;
idx = x_pos + (ps_proc->i4_ctb_y * ps_sps->i2_pic_wd_in_ctb);
pu1_buf = (ps_codec->pu1_proc_map + idx);
status = *pu1_buf & bit_mask;
}
/*Check if top left ctb is processed*/
if((ps_proc->i4_ctb_x > 0) && (0 != status) && (ps_proc->i4_ctb_y > 0))
{
x_pos = ps_tile->u1_pos_x + ps_proc->i4_ctb_tile_x - 1;
idx = x_pos + ((ps_proc->i4_ctb_y - 1) * ps_sps->i2_pic_wd_in_ctb);
pu1_buf = (ps_codec->pu1_proc_map + idx);
status = *pu1_buf & bit_mask;
}
}
}
if(status)
break;
/* if dependencies are not met, then wait for few cycles.
* Even after few iterations, if the dependencies are not met then yield
*/
if(nop_cnt > 0)
{
NOP(128);
nop_cnt -= 128;
}
else
{
nop_cnt = PROC_NOP_CNT;
ithread_yield();
//NOP(128 * 16);
}
}
DATA_SYNC();
}
}
void ihevcd_proc_map_update(process_ctxt_t *ps_proc, proc_type_t proc_type, WORD32 nctb)
{
codec_t *ps_codec = ps_proc->ps_codec;
WORD32 i, idx;
WORD32 bit_pos = proc_type;
WORD32 bit_mask = (1 << bit_pos);
/* Update the current CTBs processing status */
if(ps_proc->i4_check_proc_status)
{
DATA_SYNC();
for(i = 0; i < nctb; i++)
{
sps_t *ps_sps = ps_proc->ps_sps;
UWORD8 *pu1_buf;
idx = (ps_proc->i4_ctb_x + i);
idx += ((ps_proc->i4_ctb_y) * ps_sps->i2_pic_wd_in_ctb);
pu1_buf = (ps_codec->pu1_proc_map + idx);
*pu1_buf = *pu1_buf | bit_mask;
}
}
}
void ihevcd_slice_hdr_update(process_ctxt_t *ps_proc)
{
/* Slice x and y are initialized in proc_init. But initialize slice x and y count here
* if a new slice begins at the middle of a row since proc_init is invoked only at the beginning of each row */
if(!((ps_proc->i4_ctb_x == 0) && (ps_proc->i4_ctb_y == 0)))
{
slice_header_t *ps_slice_hdr_next = ps_proc->ps_codec->ps_slice_hdr_base + ((ps_proc->i4_cur_slice_idx + 1) & (MAX_SLICE_HDR_CNT - 1));
if((ps_slice_hdr_next->i2_ctb_x == ps_proc->i4_ctb_x)
&& (ps_slice_hdr_next->i2_ctb_y == ps_proc->i4_ctb_y))
{
if(0 == ps_slice_hdr_next->i1_dependent_slice_flag)
{
ps_proc->i4_ctb_slice_x = 0;
ps_proc->i4_ctb_slice_y = 0;
}
ps_proc->i4_cur_slice_idx++;
ps_proc->ps_slice_hdr = ps_slice_hdr_next;
}
}
}
void ihevcd_ctb_pos_update(process_ctxt_t *ps_proc, WORD32 nctb)
{
WORD32 tile_start_ctb_idx, slice_start_ctb_idx;
slice_header_t *ps_slice_hdr = ps_proc->ps_slice_hdr;
tile_t *ps_tile = ps_proc->ps_tile;
sps_t *ps_sps = ps_proc->ps_sps;
/* Update x and y positions */
ps_proc->i4_ctb_tile_x += nctb;
ps_proc->i4_ctb_x += nctb;
ps_proc->i4_ctb_slice_x += nctb;
/*If tile are enabled, then handle the tile & slice counters differently*/
if(ps_proc->ps_pps->i1_tiles_enabled_flag)
{
/* Update slice counters*/
slice_start_ctb_idx = ps_slice_hdr->i2_ctb_x + (ps_slice_hdr->i2_ctb_y * ps_sps->i2_pic_wd_in_ctb);
tile_start_ctb_idx = ps_tile->u1_pos_x + (ps_tile->u1_pos_y * ps_sps->i2_pic_wd_in_ctb);
/*
* There can be 2 cases where slice counters must be handled differently.
* 1 - Multiple tiles span across a single/one of the many slice.
* 2 - Multiple slices span across a single/one of the many tiles.
*/
/*Case 1 */
if(slice_start_ctb_idx < tile_start_ctb_idx)
{
/*End of tile row*/
if(ps_proc->i4_ctb_x > ps_slice_hdr->i2_ctb_x)
{
if(ps_proc->i4_ctb_slice_x >= (ps_tile->u2_wd + ps_tile->u1_pos_x))
{
ps_proc->i4_ctb_slice_y++;
ps_proc->i4_ctb_slice_x = ps_proc->i4_ctb_slice_x
- ps_tile->u2_wd;
}
}
else
{
WORD32 temp_stride = (ps_sps->i2_pic_wd_in_ctb - ps_slice_hdr->i2_ctb_x);
if(ps_proc->i4_ctb_slice_x >= (temp_stride + ps_tile->u2_wd + ps_tile->u1_pos_x))
{
ps_proc->i4_ctb_slice_y++;
ps_proc->i4_ctb_slice_x = ps_proc->i4_ctb_slice_x
- ps_tile->u2_wd;
}
}
}
/*Case 2*/
else if(ps_proc->i4_ctb_slice_x >= (ps_tile->u2_wd))
{
/*End of tile row*/
ps_proc->i4_ctb_slice_y++;
ps_proc->i4_ctb_slice_x = 0;
}
}
else
{
if(ps_proc->i4_ctb_slice_x >= ps_tile->u2_wd)
{
ps_proc->i4_ctb_slice_y++;
ps_proc->i4_ctb_slice_x = ps_proc->i4_ctb_slice_x
- ps_tile->u2_wd;
}
}
}
void ihevcd_ctb_avail_update(process_ctxt_t *ps_proc)
{
slice_header_t *ps_slice_hdr = ps_proc->ps_slice_hdr;
sps_t *ps_sps = ps_proc->ps_sps;
tile_t *ps_tile_prev;
tile_t *ps_tile = ps_proc->ps_tile;
WORD32 cur_pu_idx;
WORD32 tile_start_ctb_idx, slice_start_ctb_idx;
WORD16 i2_wd_in_ctb;
WORD32 continuous_tiles = 0;
WORD32 cur_ctb_idx;
WORD32 check_tile_wd;
if((0 != ps_tile->u1_pos_x) && (0 != ps_tile->u1_pos_y))
{
ps_tile_prev = ps_tile - 1;
}
else
{
ps_tile_prev = ps_tile;
}
check_tile_wd = ps_slice_hdr->i2_ctb_x + ps_tile_prev->u2_wd;
if(!(((check_tile_wd >= ps_sps->i2_pic_wd_in_ctb) && (check_tile_wd % ps_sps->i2_pic_wd_in_ctb == ps_tile->u1_pos_x))
|| ((ps_slice_hdr->i2_ctb_x == ps_tile->u1_pos_x))))
{
continuous_tiles = 1;
}
slice_start_ctb_idx = ps_slice_hdr->i2_ctb_x + (ps_slice_hdr->i2_ctb_y * ps_sps->i2_pic_wd_in_ctb);
tile_start_ctb_idx = ps_tile->u1_pos_x + (ps_tile->u1_pos_y * ps_sps->i2_pic_wd_in_ctb);
if((slice_start_ctb_idx < tile_start_ctb_idx) && (continuous_tiles))
{
//Slices span across multiple tiles.
i2_wd_in_ctb = ps_sps->i2_pic_wd_in_ctb;
}
else
{
i2_wd_in_ctb = ps_tile->u2_wd;
}
cur_ctb_idx = ps_proc->i4_ctb_x
+ ps_proc->i4_ctb_y * (ps_sps->i2_pic_wd_in_ctb);
/* Ctb level availability */
/* Bottom left will not be available at a CTB level, no need to pass this */
ps_proc->u1_top_ctb_avail = 1;
ps_proc->u1_left_ctb_avail = 1;
ps_proc->u1_top_lt_ctb_avail = 1;
ps_proc->u1_top_rt_ctb_avail = 1;
/* slice and tile boundaries */
if((0 == ps_proc->i4_ctb_y) || (0 == ps_proc->i4_ctb_tile_y))
{
ps_proc->u1_top_ctb_avail = 0;
ps_proc->u1_top_lt_ctb_avail = 0;
ps_proc->u1_top_rt_ctb_avail = 0;
}
if((0 == ps_proc->i4_ctb_x) || (0 == ps_proc->i4_ctb_tile_x))
{
ps_proc->u1_left_ctb_avail = 0;
ps_proc->u1_top_lt_ctb_avail = 0;
if((0 == ps_proc->i4_ctb_slice_y) || (0 == ps_proc->i4_ctb_tile_y))
{
ps_proc->u1_top_ctb_avail = 0;
if((i2_wd_in_ctb - 1) != ps_proc->i4_ctb_slice_x)
{
ps_proc->u1_top_rt_ctb_avail = 0;
}
}
}
/*For slices not beginning at start of a ctb row*/
else if(ps_proc->i4_ctb_x > 0)
{
if((0 == ps_proc->i4_ctb_slice_y) || (0 == ps_proc->i4_ctb_tile_y))
{
ps_proc->u1_top_ctb_avail = 0;
ps_proc->u1_top_lt_ctb_avail = 0;
if(0 == ps_proc->i4_ctb_slice_x)
{
ps_proc->u1_left_ctb_avail = 0;
}
if((i2_wd_in_ctb - 1) != ps_proc->i4_ctb_slice_x)
{
ps_proc->u1_top_rt_ctb_avail = 0;
}
}
else if((1 == ps_proc->i4_ctb_slice_y) && (0 == ps_proc->i4_ctb_slice_x))
{
ps_proc->u1_top_lt_ctb_avail = 0;
}
}
if((ps_proc->i4_ctb_x == (ps_sps->i2_pic_wd_in_ctb - 1)) || ((ps_tile->u2_wd - 1) == ps_proc->i4_ctb_tile_x))
{
ps_proc->u1_top_rt_ctb_avail = 0;
}
{
WORD32 next_ctb_idx;
next_ctb_idx = cur_ctb_idx + 1;
if(ps_tile->u2_wd == (ps_proc->i4_ctb_tile_x + 1))
{
if((ps_proc->i4_ctb_tile_y + 1) == ps_tile->u2_ht)
{
//Last tile
if(((ps_proc->i4_ctb_tile_y + 1 + ps_tile->u1_pos_y) == ps_sps->i2_pic_ht_in_ctb) && ((ps_proc->i4_ctb_tile_x + 1 + ps_tile->u1_pos_x) == ps_sps->i2_pic_wd_in_ctb))
{
next_ctb_idx = cur_ctb_idx + 1;
}
else //Not last tile, but new tile
{
tile_t *ps_tile_next = ps_proc->ps_tile + 1;
next_ctb_idx = ps_tile_next->u1_pos_x + (ps_tile_next->u1_pos_y * ps_sps->i2_pic_wd_in_ctb);
}
}
else //End of each tile row
{
next_ctb_idx = ((ps_tile->u1_pos_y + ps_proc->i4_ctb_tile_y + 1) * ps_sps->i2_pic_wd_in_ctb) + ps_tile->u1_pos_x;
}
}
ps_proc->i4_next_pu_ctb_cnt = next_ctb_idx;
ps_proc->i4_ctb_pu_cnt =
ps_proc->pu4_pic_pu_idx[next_ctb_idx]
- ps_proc->pu4_pic_pu_idx[cur_ctb_idx];
cur_pu_idx = ps_proc->pu4_pic_pu_idx[cur_ctb_idx];
ps_proc->i4_ctb_start_pu_idx = cur_pu_idx;
ps_proc->ps_pu = &ps_proc->ps_pic_pu[cur_pu_idx];
}
}
void ihevcd_update_ctb_tu_cnt(process_ctxt_t *ps_proc)
{
sps_t *ps_sps = ps_proc->ps_sps;
codec_t *ps_codec = ps_proc->ps_codec;
WORD32 cur_ctb_idx;
cur_ctb_idx = ps_proc->i4_ctb_x
+ ps_proc->i4_ctb_y * (ps_sps->i2_pic_wd_in_ctb);
{
tile_t *ps_tile;
WORD32 next_ctb_tu_idx;
ps_tile = ps_proc->ps_tile;
if(1 == ps_codec->i4_num_cores)
{
next_ctb_tu_idx = cur_ctb_idx % RESET_TU_BUF_NCTB + 1;
if(ps_tile->u2_wd == (ps_proc->i4_ctb_tile_x + 1))
{
if((ps_proc->i4_ctb_tile_y + 1) == ps_tile->u2_ht)
{
//Last tile
if(((ps_proc->i4_ctb_tile_y + 1 + ps_tile->u1_pos_y) == ps_sps->i2_pic_ht_in_ctb) && ((ps_proc->i4_ctb_tile_x + 1 + ps_tile->u1_pos_x) == ps_sps->i2_pic_wd_in_ctb))
{
next_ctb_tu_idx = (cur_ctb_idx % RESET_TU_BUF_NCTB) + 1;
}
else //Not last tile, but new tile
{
tile_t *ps_tile_next = ps_proc->ps_tile + 1;
next_ctb_tu_idx = ps_tile_next->u1_pos_x + (ps_tile_next->u1_pos_y * ps_sps->i2_pic_wd_in_ctb);
}
}
else //End of each tile row
{
next_ctb_tu_idx = ((ps_tile->u1_pos_y + ps_proc->i4_ctb_tile_y + 1) * ps_sps->i2_pic_wd_in_ctb) + ps_tile->u1_pos_x;
}
}
ps_proc->i4_next_tu_ctb_cnt = next_ctb_tu_idx;
ps_proc->i4_ctb_tu_cnt = ps_proc->pu4_pic_tu_idx[next_ctb_tu_idx] - ps_proc->pu4_pic_tu_idx[cur_ctb_idx % RESET_TU_BUF_NCTB];
}
else
{
next_ctb_tu_idx = cur_ctb_idx + 1;
if(ps_tile->u2_wd == (ps_proc->i4_ctb_tile_x + 1))
{
if((ps_proc->i4_ctb_tile_y + 1) == ps_tile->u2_ht)
{
//Last tile
if(((ps_proc->i4_ctb_tile_y + 1 + ps_tile->u1_pos_y) == ps_sps->i2_pic_ht_in_ctb) && ((ps_proc->i4_ctb_tile_x + 1 + ps_tile->u1_pos_x) == ps_sps->i2_pic_wd_in_ctb))
{
next_ctb_tu_idx = (cur_ctb_idx % RESET_TU_BUF_NCTB) + 1;
}
else //Not last tile, but new tile
{
tile_t *ps_tile_next = ps_proc->ps_tile + 1;
next_ctb_tu_idx = ps_tile_next->u1_pos_x + (ps_tile_next->u1_pos_y * ps_sps->i2_pic_wd_in_ctb);
}
}
else //End of each tile row
{
next_ctb_tu_idx = ((ps_tile->u1_pos_y + ps_proc->i4_ctb_tile_y + 1) * ps_sps->i2_pic_wd_in_ctb) + ps_tile->u1_pos_x;
}
}
ps_proc->i4_next_tu_ctb_cnt = next_ctb_tu_idx;
ps_proc->i4_ctb_tu_cnt = ps_proc->pu4_pic_tu_idx[next_ctb_tu_idx] -
ps_proc->pu4_pic_tu_idx[cur_ctb_idx];
}
}
}
IHEVCD_ERROR_T ihevcd_process(process_ctxt_t *ps_proc)
{
IHEVCD_ERROR_T ret = (IHEVCD_ERROR_T)IHEVCD_SUCCESS;
codec_t *ps_codec;
sps_t *ps_sps = ps_proc->ps_sps;
WORD32 nctb;
WORD32 i;
WORD32 idx;
WORD32 nop_cnt;
WORD32 num_minpu_in_ctb;
WORD32 cur_slice_idx, cur_ctb_tile_x, cur_ctb_slice_x, cur_ctb_tile_y, cur_ctb_slice_y;
WORD32 nxt_ctb_slice_y, nxt_ctb_slice_x;
tu_t *ps_tu_cur, *ps_tu_nxt;
UWORD8 *pu1_pu_map_cur, *pu1_pu_map_nxt;
WORD32 num_ctb, num_ctb_tmp;
proc_type_t proc_type;
ft_iquant_itrans_recon_ctb *pf_iquant_itrans_recon_ctb;
WORD32 ctb_size = 1 << ps_sps->i1_log2_ctb_size;
WORD32 h_samp_factor, v_samp_factor;
WORD32 chroma_row_strd;
WORD32 chroma_pixel_strd = 2;
h_samp_factor = (CHROMA_FMT_IDC_YUV444 == ps_sps->i1_chroma_format_idc) ? 1 : 2;
v_samp_factor = (CHROMA_FMT_IDC_YUV420 == ps_sps->i1_chroma_format_idc) ? 2 : 1;
PROFILE_DISABLE_PROCESS_CTB();
ps_codec = ps_proc->ps_codec;
pf_iquant_itrans_recon_ctb = (ft_iquant_itrans_recon_ctb *)(ps_codec->pv_iquant_itrans_recon_ctb);
num_minpu_in_ctb = (ctb_size / MIN_PU_SIZE) * (ctb_size / MIN_PU_SIZE);
nctb = MIN(ps_codec->i4_proc_nctb, ps_proc->i4_ctb_cnt);
nctb = MIN(nctb, (ps_proc->ps_tile->u2_wd - ps_proc->i4_ctb_tile_x));
if(ps_proc->i4_cur_slice_idx > (MAX_SLICE_HDR_CNT - 2 * ps_sps->i2_pic_wd_in_ctb))
{
num_ctb = 1;
}
else
{
num_ctb = ps_proc->i4_nctb;
}
nxt_ctb_slice_y = ps_proc->i4_ctb_slice_y;
nxt_ctb_slice_x = ps_proc->i4_ctb_slice_x;
pu1_pu_map_nxt = ps_proc->pu1_pu_map;
ps_tu_nxt = ps_proc->ps_tu;
while(ps_proc->i4_ctb_cnt)
{
ps_proc->i4_ctb_slice_y = nxt_ctb_slice_y;
ps_proc->i4_ctb_slice_x = nxt_ctb_slice_x;
ps_proc->pu1_pu_map = pu1_pu_map_nxt;
ps_proc->ps_tu = ps_tu_nxt;
cur_ctb_tile_x = ps_proc->i4_ctb_tile_x;
cur_ctb_tile_y = ps_proc->i4_ctb_tile_y;
cur_ctb_slice_x = ps_proc->i4_ctb_slice_x;
cur_ctb_slice_y = ps_proc->i4_ctb_slice_y;
cur_slice_idx = ps_proc->i4_cur_slice_idx;
ps_tu_cur = ps_proc->ps_tu;
pu1_pu_map_cur = ps_proc->pu1_pu_map;
proc_type = PROC_INTER_PRED;
if(ps_proc->i4_ctb_cnt < num_ctb)
{
num_ctb = ps_proc->i4_ctb_cnt;
}
num_ctb_tmp = num_ctb;
while(num_ctb_tmp)
{
slice_header_t *ps_slice_hdr;
tile_t *ps_tile = ps_proc->ps_tile;
/* Waiting for Parsing to be done*/
{
nop_cnt = PROC_NOP_CNT;
if(ps_proc->i4_check_parse_status || ps_proc->i4_check_proc_status)
{
while(1)
{
volatile UWORD8 *pu1_buf;
volatile WORD32 status;
status = 1;
/* Check if all dependencies for the next nCTBs are met */
/* Check if the next nCTBs are parsed */
if(ps_proc->i4_check_parse_status)
{
idx = (ps_proc->i4_ctb_x + nctb - 1);
idx += (ps_proc->i4_ctb_y * ps_sps->i2_pic_wd_in_ctb);
pu1_buf = (ps_codec->pu1_parse_map + idx);
status = *pu1_buf;
}
if(status)
break;
/* if dependencies are not met, then wait for few cycles.
* Even after few iterations, if the dependencies are not met then yield
*/
if(nop_cnt > 0)
{
NOP(128);
nop_cnt -= 128;
}
else
{
nop_cnt = PROC_NOP_CNT;
ithread_yield();
}
}
}
}
/* Check proc map to ensure dependencies for recon are met */
ihevcd_proc_map_check(ps_proc, proc_type, nctb);
ihevcd_slice_hdr_update(ps_proc);
ps_slice_hdr = ps_proc->ps_slice_hdr;
//ihevcd_mv_prediction();
//ihevcd_lvl_unpack();
//ihevcd_inter_iq_it_recon();
//Following does prediction, iq, it and recon on a TU by TU basis for intra TUs
//ihevcd_intra_process();
//ihevcd_ctb_boundary_strength_islice(ps_proc, ctb_size);
//ihevcd_deblk_ctb(ps_proc);
/* iq,it recon of Intra TU */
{
UWORD32 *pu4_ctb_top_pu_idx, *pu4_ctb_left_pu_idx, *pu4_ctb_top_left_pu_idx;
WORD32 cur_ctb_idx;
ihevcd_ctb_avail_update(ps_proc);
#if DEBUG_DUMP_FRAME_BUFFERS_INFO
au1_pic_avail_ctb_flags[ps_proc->i4_ctb_x + ps_proc->i4_ctb_y * ps_sps->i2_pic_wd_in_ctb] =
((ps_proc->u1_top_ctb_avail << 3) | (ps_proc->u1_left_ctb_avail << 2) | (ps_proc->u1_top_lt_ctb_avail << 1) | (ps_proc->u1_top_rt_ctb_avail));
au4_pic_ctb_slice_xy[ps_proc->i4_ctb_x + ps_proc->i4_ctb_y * ps_sps->i2_pic_wd_in_ctb] =
(((UWORD16)ps_proc->i4_ctb_slice_x << 16) | ((UWORD16)ps_proc->i4_ctb_slice_y << 16));
#endif
/*************************************************/
/**************** MV pred **********************/
/*************************************************/
if(PSLICE == ps_slice_hdr->i1_slice_type
|| BSLICE == ps_slice_hdr->i1_slice_type)
{
mv_ctxt_t s_mv_ctxt;
pu4_ctb_top_pu_idx = ps_proc->pu4_pic_pu_idx_top
+ (ps_proc->i4_ctb_x * ctb_size / MIN_PU_SIZE);
pu4_ctb_left_pu_idx = ps_proc->pu4_pic_pu_idx_left;
pu4_ctb_top_left_pu_idx = &ps_proc->u4_ctb_top_left_pu_idx;
/* Initializing s_mv_ctxt */
if(ps_codec->i4_num_cores > MV_PRED_NUM_CORES_THRESHOLD)
{
s_mv_ctxt.ps_pps = ps_proc->ps_pps;
s_mv_ctxt.ps_sps = ps_proc->ps_sps;
s_mv_ctxt.ps_slice_hdr = ps_proc->ps_slice_hdr;
s_mv_ctxt.i4_ctb_x = ps_proc->i4_ctb_x;
s_mv_ctxt.i4_ctb_y = ps_proc->i4_ctb_y;
s_mv_ctxt.ps_pu = ps_proc->ps_pu;
s_mv_ctxt.ps_pic_pu = ps_proc->ps_pic_pu;
s_mv_ctxt.ps_tile = ps_tile;
s_mv_ctxt.pu4_pic_pu_idx_map = ps_proc->pu4_pic_pu_idx_map;
s_mv_ctxt.pu4_pic_pu_idx = ps_proc->pu4_pic_pu_idx;
s_mv_ctxt.pu1_pic_pu_map = ps_proc->pu1_pic_pu_map;
s_mv_ctxt.i4_ctb_pu_cnt = ps_proc->i4_ctb_pu_cnt;
s_mv_ctxt.i4_ctb_start_pu_idx = ps_proc->i4_ctb_start_pu_idx;
s_mv_ctxt.u1_top_ctb_avail = ps_proc->u1_top_ctb_avail;
s_mv_ctxt.u1_top_rt_ctb_avail = ps_proc->u1_top_rt_ctb_avail;
s_mv_ctxt.u1_top_lt_ctb_avail = ps_proc->u1_top_lt_ctb_avail;
s_mv_ctxt.u1_left_ctb_avail = ps_proc->u1_left_ctb_avail;
ihevcd_get_mv_ctb(&s_mv_ctxt, pu4_ctb_top_pu_idx,
pu4_ctb_left_pu_idx, pu4_ctb_top_left_pu_idx);
}
ihevcd_inter_pred_ctb(ps_proc);
}
else if(ps_codec->i4_num_cores > MV_PRED_NUM_CORES_THRESHOLD)
{
WORD32 next_ctb_idx, num_pu_per_ctb, ctb_start_pu_idx, pu_cnt;
pu_t *ps_pu;
WORD32 num_minpu_in_ctb = (ctb_size / MIN_PU_SIZE) * (ctb_size / MIN_PU_SIZE);
UWORD8 *pu1_pic_pu_map_ctb = ps_proc->pu1_pic_pu_map +
(ps_proc->i4_ctb_x + ps_proc->i4_ctb_y * ps_sps->i2_pic_wd_in_ctb) * num_minpu_in_ctb;
WORD32 row, col;
UWORD32 *pu4_nbr_pu_idx = ps_proc->pu4_pic_pu_idx_map;
WORD32 nbr_pu_idx_strd = MAX_CTB_SIZE / MIN_PU_SIZE + 2;
WORD32 ctb_size_in_min_pu = (ctb_size / MIN_PU_SIZE);
/* Neighbor PU idx update inside CTB */
/* 1byte per 4x4. Indicates the PU idx that 4x4 block belongs to */
cur_ctb_idx = ps_proc->i4_ctb_x
+ ps_proc->i4_ctb_y * (ps_sps->i2_pic_wd_in_ctb);
next_ctb_idx = ps_proc->i4_next_pu_ctb_cnt;
num_pu_per_ctb = ps_proc->pu4_pic_pu_idx[next_ctb_idx]
- ps_proc->pu4_pic_pu_idx[cur_ctb_idx];
ctb_start_pu_idx = ps_proc->pu4_pic_pu_idx[cur_ctb_idx];
ps_pu = &ps_proc->ps_pic_pu[ctb_start_pu_idx];
for(pu_cnt = 0; pu_cnt < num_pu_per_ctb; pu_cnt++, ps_pu++)
{
UWORD32 cur_pu_idx;
WORD32 pu_ht = (ps_pu->b4_ht + 1) << 2;
WORD32 pu_wd = (ps_pu->b4_wd + 1) << 2;
cur_pu_idx = ctb_start_pu_idx + pu_cnt;
for(row = 0; row < pu_ht / MIN_PU_SIZE; row++)
for(col = 0; col < pu_wd / MIN_PU_SIZE; col++)
pu4_nbr_pu_idx[(1 + ps_pu->b4_pos_x + col)
+ (1 + ps_pu->b4_pos_y + row)
* nbr_pu_idx_strd] =
cur_pu_idx;
}
/* Updating Top and Left pointers */
{
WORD32 rows_remaining = ps_sps->i2_pic_height_in_luma_samples
- (ps_proc->i4_ctb_y << ps_sps->i1_log2_ctb_size);
WORD32 ctb_size_left = MIN(ctb_size, rows_remaining);
/* Top Left */
/* saving top left before updating top ptr, as updating top ptr will overwrite the top left for the next ctb */
ps_proc->u4_ctb_top_left_pu_idx = ps_proc->pu4_pic_pu_idx_top[((ps_proc->i4_ctb_x + 1) * ctb_size / MIN_PU_SIZE) - 1];
for(i = 0; i < ctb_size / MIN_PU_SIZE; i++)
{
/* Left */
/* Last column of au4_nbr_pu_idx */
ps_proc->pu4_pic_pu_idx_left[i] =
pu4_nbr_pu_idx[(ctb_size / MIN_PU_SIZE) + (i + 1) * nbr_pu_idx_strd];
/* Top */
/* Last row of au4_nbr_pu_idx */
ps_proc->pu4_pic_pu_idx_top[(ps_proc->i4_ctb_x * ctb_size / MIN_PU_SIZE) + i] =
pu4_nbr_pu_idx[(ctb_size_left / MIN_PU_SIZE) * nbr_pu_idx_strd + i + 1];
}
/* Updating the CTB level PU idx (Used for collocated MV pred)*/
{
WORD32 ctb_row, ctb_col, index_pic_map, index_nbr_map;
WORD32 first_pu_of_ctb;
first_pu_of_ctb = pu4_nbr_pu_idx[1 + nbr_pu_idx_strd];
UWORD32 cur_ctb_ht_in_min_pu = MIN(((ps_sps->i2_pic_height_in_luma_samples
- (ps_proc->i4_ctb_y << ps_sps->i1_log2_ctb_size)) / MIN_PU_SIZE), ctb_size_in_min_pu);
UWORD32 cur_ctb_wd_in_min_pu = MIN(((ps_sps->i2_pic_width_in_luma_samples
- (ps_proc->i4_ctb_x << ps_sps->i1_log2_ctb_size)) / MIN_PU_SIZE), ctb_size_in_min_pu);
index_pic_map = 0 * ctb_size_in_min_pu + 0;
index_nbr_map = (0 + 1) * nbr_pu_idx_strd + (0 + 1);
for(ctb_row = 0; ctb_row < cur_ctb_ht_in_min_pu; ctb_row++)
{
for(ctb_col = 0; ctb_col < cur_ctb_wd_in_min_pu; ctb_col++)
{
pu1_pic_pu_map_ctb[index_pic_map + ctb_col] = pu4_nbr_pu_idx[index_nbr_map + ctb_col]
- first_pu_of_ctb;
}
index_pic_map += ctb_size_in_min_pu;
index_nbr_map += nbr_pu_idx_strd;
}
}
}
}
}
if(ps_proc->ps_pps->i1_tiles_enabled_flag)
{
/*Update the tile index buffer with tile information for the current ctb*/
UWORD16 *pu1_tile_idx = ps_proc->pu1_tile_idx;
pu1_tile_idx[(ps_proc->i4_ctb_x + (ps_proc->i4_ctb_y * ps_sps->i2_pic_wd_in_ctb))]
= ps_proc->i4_cur_tile_idx;
}
/*************************************************/
/*********** BS, QP and Deblocking **************/
/*************************************************/
/* Boundary strength call has to be after IQ IT recon since QP population needs ps_proc->i4_qp_const_inc_ctb flag */
{
slice_header_t *ps_slice_hdr;
ps_slice_hdr = ps_proc->ps_slice_hdr;
/* Check if deblock is disabled for the current slice or if it is disabled for the current picture
* because of disable deblock api
*/
if(0 == ps_codec->i4_disable_deblk_pic)
{
if(ps_codec->i4_num_cores > MV_PRED_NUM_CORES_THRESHOLD)
{
/* Boundary strength calculation is done irrespective of whether deblocking is disabled
* in the slice or not, to handle deblocking slice boundaries */
if((0 == ps_codec->i4_slice_error))
{
ihevcd_update_ctb_tu_cnt(ps_proc);
ps_proc->s_bs_ctxt.ps_pps = ps_proc->ps_pps;
ps_proc->s_bs_ctxt.ps_sps = ps_proc->ps_sps;
ps_proc->s_bs_ctxt.ps_codec = ps_proc->ps_codec;
ps_proc->s_bs_ctxt.i4_ctb_tu_cnt = ps_proc->i4_ctb_tu_cnt;
ps_proc->s_bs_ctxt.i4_ctb_x = ps_proc->i4_ctb_x;
ps_proc->s_bs_ctxt.i4_ctb_y = ps_proc->i4_ctb_y;
ps_proc->s_bs_ctxt.i4_ctb_tile_x = ps_proc->i4_ctb_tile_x;
ps_proc->s_bs_ctxt.i4_ctb_tile_y = ps_proc->i4_ctb_tile_y;
ps_proc->s_bs_ctxt.i4_ctb_slice_x = ps_proc->i4_ctb_slice_x;
ps_proc->s_bs_ctxt.i4_ctb_slice_y = ps_proc->i4_ctb_slice_y;
ps_proc->s_bs_ctxt.ps_tu = ps_proc->ps_tu;
ps_proc->s_bs_ctxt.ps_pu = ps_proc->ps_pu;
ps_proc->s_bs_ctxt.pu4_pic_pu_idx_map = ps_proc->pu4_pic_pu_idx_map;
ps_proc->s_bs_ctxt.i4_next_pu_ctb_cnt = ps_proc->i4_next_pu_ctb_cnt;
ps_proc->s_bs_ctxt.i4_next_tu_ctb_cnt = ps_proc->i4_next_tu_ctb_cnt;
ps_proc->s_bs_ctxt.pu1_slice_idx = ps_proc->pu1_slice_idx;
ps_proc->s_bs_ctxt.ps_slice_hdr = ps_proc->ps_slice_hdr;
ps_proc->s_bs_ctxt.ps_tile = ps_proc->ps_tile;
if(ISLICE == ps_slice_hdr->i1_slice_type)
{
ihevcd_ctb_boundary_strength_islice(&ps_proc->s_bs_ctxt);
}
else
{
ihevcd_ctb_boundary_strength_pbslice(&ps_proc->s_bs_ctxt);
}
}
/* Boundary strength is set to zero if deblocking is disabled for the current slice */
if((0 != ps_slice_hdr->i1_slice_disable_deblocking_filter_flag))
{
WORD32 bs_strd = (ps_sps->i2_pic_wd_in_ctb + 1) * (ctb_size * ctb_size / 8 / 16);
UWORD32 *pu4_vert_bs = (UWORD32 *)((UWORD8 *)ps_proc->s_bs_ctxt.pu4_pic_vert_bs +
ps_proc->i4_ctb_x * (ctb_size * ctb_size / 8 / 16) +
ps_proc->i4_ctb_y * bs_strd);
UWORD32 *pu4_horz_bs = (UWORD32 *)((UWORD8 *)ps_proc->s_bs_ctxt.pu4_pic_horz_bs +
ps_proc->i4_ctb_x * (ctb_size * ctb_size / 8 / 16) +
ps_proc->i4_ctb_y * bs_strd);
memset(pu4_vert_bs, 0, (ctb_size / 8) * (ctb_size / 4) / 8 * 2);
memset(pu4_horz_bs, 0, (ctb_size / 8) * (ctb_size / 4) / 8 * 2);
}
}
}
}
/* Per CTB update the following */
{
WORD32 cur_ctb_idx = ps_proc->i4_ctb_x
+ ps_proc->i4_ctb_y * (ps_sps->i2_pic_wd_in_ctb);
cur_ctb_idx++;
ps_proc->pu1_pu_map += nctb * num_minpu_in_ctb;
ps_proc->ps_tu += ps_proc->i4_ctb_tu_cnt;
if((1 == ps_codec->i4_num_cores) &&
(0 == cur_ctb_idx % RESET_TU_BUF_NCTB))
{
ps_proc->ps_tu = ps_proc->ps_pic_tu;
}
ps_proc->ps_pu += ps_proc->i4_ctb_pu_cnt;
}
/* Update proc map for recon*/
ihevcd_proc_map_update(ps_proc, proc_type, nctb);
num_ctb_tmp -= nctb;
ihevcd_ctb_pos_update(ps_proc, nctb);
}
if(cur_slice_idx != ps_proc->i4_cur_slice_idx)
{
ps_proc->ps_slice_hdr = ps_codec->ps_slice_hdr_base + ((cur_slice_idx)&(MAX_SLICE_HDR_CNT - 1));
ps_proc->i4_cur_slice_idx = cur_slice_idx;
}
/* Restore the saved variables */
num_ctb_tmp = num_ctb;
ps_proc->i4_ctb_x -= num_ctb;
ps_proc->i4_ctb_tile_x = cur_ctb_tile_x;
ps_proc->i4_ctb_slice_x = cur_ctb_slice_x;
ps_proc->i4_ctb_tile_y = cur_ctb_tile_y;
ps_proc->i4_ctb_slice_y = cur_ctb_slice_y;
ps_proc->pu1_pu_map = pu1_pu_map_cur;
ps_proc->ps_tu = ps_tu_cur;
proc_type = PROC_RECON;
while(num_ctb_tmp)
{
/* Check proc map to ensure dependencies for recon are met */
ihevcd_proc_map_check(ps_proc, proc_type, nctb);
ihevcd_slice_hdr_update(ps_proc);
{
ihevcd_ctb_avail_update(ps_proc);
/*************************************************/
/**************** IQ IT RECON *******************/
/*************************************************/
ihevcd_update_ctb_tu_cnt(ps_proc);
/* When scaling matrix is not to be used(scaling_list_enable_flag is zero in SPS),
* default value of 16 has to be used. Since the value is same for all sizes,
* same table is used for all cases.
*/
if(0 == ps_sps->i1_scaling_list_enable_flag)
{
ps_proc->api2_dequant_intra_matrix[0] =
(WORD16 *)gi2_flat_scale_mat_32x32;
ps_proc->api2_dequant_intra_matrix[1] =
(WORD16 *)gi2_flat_scale_mat_32x32;
ps_proc->api2_dequant_intra_matrix[2] =
(WORD16 *)gi2_flat_scale_mat_32x32;
ps_proc->api2_dequant_intra_matrix[3] =
(WORD16 *)gi2_flat_scale_mat_32x32;
ps_proc->api2_dequant_inter_matrix[0] =
(WORD16 *)gi2_flat_scale_mat_32x32;
ps_proc->api2_dequant_inter_matrix[1] =
(WORD16 *)gi2_flat_scale_mat_32x32;
ps_proc->api2_dequant_inter_matrix[2] =
(WORD16 *)gi2_flat_scale_mat_32x32;
ps_proc->api2_dequant_inter_matrix[3] =
(WORD16 *)gi2_flat_scale_mat_32x32;
}
else
{
if(0 == ps_sps->i1_sps_scaling_list_data_present_flag)
{
ps_proc->api2_dequant_intra_matrix[0] =
(WORD16 *)gi2_flat_scale_mat_32x32;
ps_proc->api2_dequant_intra_matrix[1] =
(WORD16 *)gi2_intra_default_scale_mat_8x8;
ps_proc->api2_dequant_intra_matrix[2] =
(WORD16 *)gi2_intra_default_scale_mat_16x16;
ps_proc->api2_dequant_intra_matrix[3] =
(WORD16 *)gi2_intra_default_scale_mat_32x32;
ps_proc->api2_dequant_inter_matrix[0] =
(WORD16 *)gi2_flat_scale_mat_32x32;
ps_proc->api2_dequant_inter_matrix[1] =
(WORD16 *)gi2_inter_default_scale_mat_8x8;
ps_proc->api2_dequant_inter_matrix[2] =
(WORD16 *)gi2_inter_default_scale_mat_16x16;
ps_proc->api2_dequant_inter_matrix[3] =
(WORD16 *)gi2_inter_default_scale_mat_32x32;
}
/*TODO: Add support for custom scaling matrices */
}
/* CTB Level pointers */
ps_proc->pu1_cur_ctb_luma = ps_proc->pu1_cur_pic_luma
+ (ps_proc->i4_ctb_x * ctb_size
+ ps_proc->i4_ctb_y * ctb_size
* ps_codec->i4_strd) * ps_codec->i4_pixel_size_y;
if(CHROMA_FMT_IDC_MONOCHROME != ps_sps->i1_chroma_format_idc)
{
ps_proc->pu1_cur_ctb_chroma = ps_proc->pu1_cur_pic_chroma
+ ((ps_proc->i4_ctb_x * ctb_size * chroma_pixel_strd / h_samp_factor)
+ (ps_proc->i4_ctb_y * ctb_size * ps_codec->i4_strd * chroma_pixel_strd / (h_samp_factor * v_samp_factor))) * ps_codec->i4_pixel_size_uv;
}
pf_iquant_itrans_recon_ctb(ps_proc);
}
/* Per CTB update the following */
{
WORD32 cur_ctb_idx = ps_proc->i4_ctb_x
+ ps_proc->i4_ctb_y * (ps_sps->i2_pic_wd_in_ctb);
cur_ctb_idx++;
ps_proc->pu1_pu_map += nctb * num_minpu_in_ctb;
ps_proc->ps_tu += ps_proc->i4_ctb_tu_cnt;
if((1 == ps_codec->i4_num_cores) &&
(0 == cur_ctb_idx % RESET_TU_BUF_NCTB))
{
ps_proc->ps_tu = ps_proc->ps_pic_tu;
}
ps_proc->ps_pu += ps_proc->i4_ctb_pu_cnt;
}
/* Update proc map for recon*/
ihevcd_proc_map_update(ps_proc, proc_type, nctb);
num_ctb_tmp -= nctb;
ihevcd_ctb_pos_update(ps_proc, nctb);
}
if(cur_slice_idx != ps_proc->i4_cur_slice_idx)
{
ps_proc->ps_slice_hdr = ps_codec->ps_slice_hdr_base + ((cur_slice_idx)&(MAX_SLICE_HDR_CNT - 1));
ps_proc->i4_cur_slice_idx = cur_slice_idx;
}
/* Restore the saved variables */
num_ctb_tmp = num_ctb;
ps_proc->i4_ctb_x -= num_ctb;
ps_proc->i4_ctb_tile_x = cur_ctb_tile_x;
ps_proc->i4_ctb_slice_x = cur_ctb_slice_x;
ps_proc->i4_ctb_tile_y = cur_ctb_tile_y;
ps_proc->i4_ctb_slice_y = cur_ctb_slice_y;
pu1_pu_map_nxt = ps_proc->pu1_pu_map;
ps_tu_nxt = ps_proc->ps_tu;
ps_proc->pu1_pu_map = pu1_pu_map_cur;
ps_proc->ps_tu = ps_tu_cur;
proc_type = PROC_DEBLK;
while(num_ctb_tmp)
{
/* Check proc map to ensure dependencies for deblk are met */
ihevcd_proc_map_check(ps_proc, proc_type, nctb);
ihevcd_slice_hdr_update(ps_proc);
if(0 == ps_codec->i4_disable_deblk_pic)
{
WORD32 i4_is_last_ctb_x = 0;
WORD32 i4_is_last_ctb_y = 0;
/* Deblocking is done irrespective of whether it is disabled in the slice or not,
* to handle deblocking the slice boundaries */
{
ps_proc->s_deblk_ctxt.ps_pps = ps_proc->ps_pps;
ps_proc->s_deblk_ctxt.ps_sps = ps_proc->ps_sps;
ps_proc->s_deblk_ctxt.ps_codec = ps_proc->ps_codec;
ps_proc->s_deblk_ctxt.ps_slice_hdr = ps_proc->ps_slice_hdr;
ps_proc->s_deblk_ctxt.i4_ctb_x = ps_proc->i4_ctb_x;
ps_proc->s_deblk_ctxt.i4_ctb_y = ps_proc->i4_ctb_y;
ps_proc->s_deblk_ctxt.pu1_slice_idx = ps_proc->pu1_slice_idx;
ps_proc->s_deblk_ctxt.is_chroma_yuv420sp_vu = (ps_codec->e_ref_chroma_fmt == IV_YUV_420SP_VU);
/* Populating Current CTB's no_loop_filter flags */
{
WORD32 row;
WORD32 log2_ctb_size = ps_sps->i1_log2_ctb_size;
/* Loop filter strd in units of num bits */
WORD32 loop_filter_strd = ((ps_sps->i2_pic_width_in_luma_samples + 63) >> 6) << 3;
/* Bit position is the current 8x8 bit offset wrt pic_no_loop_filter
* bit_pos has to be a WOR32 so that when it is negative, the downshift still retains it to be a negative value */
WORD32 bit_pos = ((ps_proc->i4_ctb_y << (log2_ctb_size - 3)) - 1) * loop_filter_strd + (ps_proc->i4_ctb_x << (log2_ctb_size - 3)) - 1;
for(row = 0; row < (ctb_size >> 3) + 1; row++)
{
/* Go to the corresponding byte - read 32 bits and downshift */
ps_proc->s_deblk_ctxt.au2_ctb_no_loop_filter_flag[row] = (*(UWORD32 *)(ps_proc->pu1_pic_no_loop_filter_flag + (bit_pos >> 3))) >> (bit_pos & 7);
bit_pos += loop_filter_strd;
}
}
ihevcd_deblk_ctb(&ps_proc->s_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(ps_proc->i4_ctb_x == ps_sps->i2_pic_wd_in_ctb - 1)
{
WORD32 i4_is_last_ctb_x = 1;
WORD32 i4_is_last_ctb_y = 0;
WORD32 last_x_pos;
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_proc->s_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(ps_proc->i4_ctb_y == ps_sps->i2_pic_ht_in_ctb - 1)
{
WORD32 i4_is_last_ctb_x = 0;
WORD32 i4_is_last_ctb_y = 1;
WORD32 last_y_pos;
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_proc->s_deblk_ctxt, i4_is_last_ctb_x, i4_is_last_ctb_y);
}
}
}
}
/* Update proc map for deblk*/
ihevcd_proc_map_update(ps_proc, proc_type, nctb);
num_ctb_tmp -= nctb;
ihevcd_ctb_pos_update(ps_proc, nctb);
}
if(cur_slice_idx != ps_proc->i4_cur_slice_idx)
{
ps_proc->ps_slice_hdr = ps_codec->ps_slice_hdr_base + ((cur_slice_idx)&(MAX_SLICE_HDR_CNT - 1));
ps_proc->i4_cur_slice_idx = cur_slice_idx;
}
/* Restore the saved variables */
num_ctb_tmp = num_ctb;
ps_proc->i4_ctb_x -= num_ctb;
ps_proc->i4_ctb_tile_x = cur_ctb_tile_x;
ps_proc->i4_ctb_tile_y = cur_ctb_tile_y;
ps_proc->pu1_pu_map = pu1_pu_map_cur;
ps_proc->ps_tu = ps_tu_cur;
nxt_ctb_slice_y = ps_proc->i4_ctb_slice_y;
nxt_ctb_slice_x = ps_proc->i4_ctb_slice_x;
ps_proc->i4_ctb_slice_y = cur_ctb_slice_y;
ps_proc->i4_ctb_slice_x = cur_ctb_slice_x;
proc_type = PROC_SAO;
while(num_ctb_tmp)
{
/* Check proc map to ensure dependencies for SAO are met */
ihevcd_proc_map_check(ps_proc, proc_type, nctb);
ihevcd_slice_hdr_update(ps_proc);
{
/* SAO is done even when it is disabled in the current slice, because
* it is performed on a shifted CTB and the neighbor CTBs can belong
* to different slices with SAO enabled */
if(0 == ps_codec->i4_disable_sao_pic)
{
ftype_sao_shift_ctb *pf_sao_shift_ctb;
ps_proc->s_sao_ctxt.ps_pps = ps_proc->ps_pps;
ps_proc->s_sao_ctxt.ps_sps = ps_proc->ps_sps;
ps_proc->s_sao_ctxt.ps_tile = ps_proc->ps_tile;
ps_proc->s_sao_ctxt.ps_codec = ps_proc->ps_codec;
ps_proc->s_sao_ctxt.ps_slice_hdr = ps_proc->ps_slice_hdr;
ps_proc->s_sao_ctxt.i4_cur_slice_idx = ps_proc->i4_cur_slice_idx;
pf_sao_shift_ctb = (ftype_sao_shift_ctb *)(ps_proc->s_sao_ctxt.pf_sao_shift_ctb);
ps_proc->s_sao_ctxt.i4_ctb_x = ps_proc->i4_ctb_x;
ps_proc->s_sao_ctxt.i4_ctb_y = ps_proc->i4_ctb_y;
ps_proc->s_sao_ctxt.is_chroma_yuv420sp_vu = (ps_codec->e_ref_chroma_fmt == IV_YUV_420SP_VU);
pf_sao_shift_ctb(&ps_proc->s_sao_ctxt);
}
/* Call padding if required */
{
chroma_row_strd = ps_codec->i4_strd * chroma_pixel_strd / h_samp_factor;
if(0 == ps_proc->i4_ctb_x)
{
WORD32 pad_ht_luma;
WORD32 pad_ht_chroma;
ps_proc->pu1_cur_ctb_luma = ps_proc->pu1_cur_pic_luma
+ (ps_proc->i4_ctb_x * ctb_size
+ ps_proc->i4_ctb_y * ctb_size
* ps_codec->i4_strd) * ps_codec->i4_pixel_size_y;
pad_ht_luma = ctb_size;
pad_ht_luma += (ps_sps->i2_pic_ht_in_ctb - 1) == ps_proc->i4_ctb_y ? 8 : 0; /* Pad left after 1st CTB is processed */
if (1 == ps_codec->i4_pixel_size_y)
{
ps_codec->s_func_selector.ihevc_pad_left_luma_fptr(ps_proc->pu1_cur_ctb_luma - 8 * ps_codec->i4_strd,
ps_codec->i4_strd, pad_ht_luma, PAD_LEFT);
}
else
{
ps_codec->s_func_selector.ihevc_hbd_pad_left_luma_fptr((UWORD16 *)ps_proc->pu1_cur_ctb_luma - 8 * ps_codec->i4_strd,
ps_codec->i4_strd, pad_ht_luma, PAD_LEFT);
}
if(CHROMA_FMT_IDC_MONOCHROME != ps_sps->i1_chroma_format_idc)
{
ps_proc->pu1_cur_ctb_chroma = ps_proc->pu1_cur_pic_chroma
+ ((ps_proc->i4_ctb_x * ctb_size * chroma_pixel_strd / h_samp_factor)
+ (ps_proc->i4_ctb_y * ctb_size * chroma_row_strd / v_samp_factor)) * ps_codec->i4_pixel_size_uv;
pad_ht_chroma = ctb_size / v_samp_factor;
if (1 == ps_codec->i4_pixel_size_uv)
{
ps_codec->s_func_selector.ihevc_pad_left_chroma_fptr(
ps_proc->pu1_cur_ctb_chroma - (8 * v_samp_factor) * chroma_row_strd,
chroma_row_strd, pad_ht_chroma, PAD_LEFT * chroma_pixel_strd / h_samp_factor);
}
else
{
ps_codec->s_func_selector.ihevc_hbd_pad_left_chroma_fptr(
(UWORD16 *)ps_proc->pu1_cur_ctb_chroma - (8 * v_samp_factor) * chroma_row_strd,
chroma_row_strd, pad_ht_chroma, PAD_LEFT * chroma_pixel_strd / h_samp_factor);
}
}
}
if((ps_sps->i2_pic_wd_in_ctb - 1) == ps_proc->i4_ctb_x)
{
WORD32 pad_ht_luma;
WORD32 pad_ht_chroma;
WORD32 cols_remaining = ps_sps->i2_pic_width_in_luma_samples - (ps_proc->i4_ctb_x << ps_sps->i1_log2_ctb_size);
ps_proc->pu1_cur_ctb_luma = ps_proc->pu1_cur_pic_luma
+ (ps_proc->i4_ctb_x * ctb_size
+ ps_proc->i4_ctb_y * ctb_size
* ps_codec->i4_strd) * ps_codec->i4_pixel_size_y;
if(CHROMA_FMT_IDC_MONOCHROME != ps_sps->i1_chroma_format_idc)
{
ps_proc->pu1_cur_ctb_chroma = ps_proc->pu1_cur_pic_chroma
+ ((ps_proc->i4_ctb_x * ctb_size * chroma_pixel_strd / h_samp_factor)
+ (ps_proc->i4_ctb_y * ctb_size * chroma_row_strd / v_samp_factor)) * ps_codec->i4_pixel_size_uv;
pad_ht_chroma = ctb_size / v_samp_factor;
}
pad_ht_luma = ctb_size;
if((ps_sps->i2_pic_ht_in_ctb - 1) == ps_proc->i4_ctb_y)
{
pad_ht_luma += 8;
if (CHROMA_FMT_IDC_MONOCHROME != ps_sps->i1_chroma_format_idc)
{
pad_ht_chroma += (8 * v_samp_factor);
if (1 == ps_codec->i4_pixel_size_uv)
{
ps_codec->s_func_selector.ihevc_pad_left_chroma_fptr(
ps_proc->pu1_cur_pic_chroma + ((ps_sps->i2_pic_height_in_luma_samples / v_samp_factor) - (8 * v_samp_factor)) * chroma_row_strd,
chroma_row_strd, (8 * v_samp_factor),
PAD_LEFT * chroma_pixel_strd / h_samp_factor);
}
else
{
ps_codec->s_func_selector.ihevc_hbd_pad_left_chroma_fptr(
(UWORD16 *)ps_proc->pu1_cur_pic_chroma + ((ps_sps->i2_pic_height_in_luma_samples / v_samp_factor) - (8 * v_samp_factor)) * chroma_row_strd,
chroma_row_strd, (8 * v_samp_factor),
PAD_LEFT * chroma_pixel_strd / h_samp_factor);
}
}
}
/* Pad right after last CTB in the current row is processed */
if (1 == ps_codec->i4_pixel_size_y)
{
ps_codec->s_func_selector.ihevc_pad_right_luma_fptr(
ps_proc->pu1_cur_ctb_luma + cols_remaining - 8 * ps_codec->i4_strd,
ps_codec->i4_strd, pad_ht_luma, PAD_RIGHT);
}
else
{
ps_codec->s_func_selector.ihevc_hbd_pad_right_luma_fptr((UWORD16 *)ps_proc->pu1_cur_ctb_luma + cols_remaining - 8 * ps_codec->i4_strd,
ps_codec->i4_strd, pad_ht_luma, PAD_RIGHT);
}
if(CHROMA_FMT_IDC_MONOCHROME != ps_sps->i1_chroma_format_idc)
{
if (1 == ps_codec->i4_pixel_size_uv)
{
ps_codec->s_func_selector.ihevc_pad_right_chroma_fptr(
ps_proc->pu1_cur_ctb_chroma + (cols_remaining * chroma_pixel_strd / h_samp_factor) - (8 * v_samp_factor) * chroma_row_strd,
chroma_row_strd, pad_ht_chroma,
PAD_RIGHT * chroma_pixel_strd / h_samp_factor);
}
else
{
ihevc_hbd_pad_right_chroma(
(UWORD16 *)ps_proc->pu1_cur_ctb_chroma + (cols_remaining * chroma_pixel_strd / h_samp_factor) - (8 * v_samp_factor) * chroma_row_strd,
chroma_row_strd, pad_ht_chroma,
PAD_RIGHT * chroma_pixel_strd / h_samp_factor);
}
}
if((ps_sps->i2_pic_ht_in_ctb - 1) == ps_proc->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 */
if (1 == ps_codec->i4_pixel_size_y)
{
ihevc_pad_top(ps_proc->pu1_cur_pic_luma - PAD_LEFT, ps_codec->i4_strd,
ps_sps->i2_pic_width_in_luma_samples + PAD_WD, PAD_TOP);
}
else
{
ihevc_hbd_pad_top((UWORD16 *)ps_proc->pu1_cur_pic_luma - PAD_LEFT, ps_codec->i4_strd,
ps_sps->i2_pic_width_in_luma_samples + PAD_WD, PAD_TOP);
}
if(CHROMA_FMT_IDC_MONOCHROME != ps_sps->i1_chroma_format_idc)
{
if (1 == ps_codec->i4_pixel_size_uv)
{
ihevc_pad_top(ps_proc->pu1_cur_pic_chroma - PAD_LEFT * (chroma_pixel_strd / h_samp_factor),
chroma_row_strd,
(ps_sps->i2_pic_width_in_luma_samples + PAD_WD) * (chroma_pixel_strd / h_samp_factor),
PAD_TOP / v_samp_factor);
}
else
{
ihevc_hbd_pad_top((UWORD16 *)ps_proc->pu1_cur_pic_chroma - PAD_LEFT * (chroma_pixel_strd / h_samp_factor),
chroma_row_strd,
(ps_sps->i2_pic_width_in_luma_samples + PAD_WD) * (chroma_pixel_strd / h_samp_factor),
PAD_TOP / v_samp_factor);
}
}
/* Pad bottom after padding left and right for current rows after processing 1st CTB row */
pu1_buf = ps_proc->pu1_cur_pic_luma +
(ps_codec->i4_strd * ps_sps->i2_pic_height_in_luma_samples - PAD_LEFT) * ps_codec->i4_pixel_size_y;
if (1 == ps_codec->i4_pixel_size_y)
{
ihevc_pad_bottom(pu1_buf, ps_codec->i4_strd,
ps_sps->i2_pic_width_in_luma_samples + PAD_WD, PAD_BOT);
}
else
{
ihevc_hbd_pad_bottom((UWORD16 *)pu1_buf, ps_codec->i4_strd,
ps_sps->i2_pic_width_in_luma_samples + PAD_WD, PAD_BOT);
}
if(CHROMA_FMT_IDC_MONOCHROME != ps_sps->i1_chroma_format_idc)
{
pu1_buf = ps_proc->pu1_cur_pic_chroma
+ (chroma_row_strd * (ps_sps->i2_pic_height_in_luma_samples / v_samp_factor)
- (PAD_LEFT * chroma_pixel_strd / h_samp_factor)) * ps_codec->i4_pixel_size_uv;
if (1 == ps_codec->i4_pixel_size_uv)
{
ihevc_pad_bottom(pu1_buf,
chroma_row_strd,
(ps_sps->i2_pic_width_in_luma_samples + PAD_WD) * (chroma_pixel_strd / h_samp_factor),
PAD_BOT / v_samp_factor);
}
else
{
ihevc_hbd_pad_bottom((UWORD16 *)pu1_buf,
chroma_row_strd,
(ps_sps->i2_pic_width_in_luma_samples + PAD_WD) * (chroma_pixel_strd / h_samp_factor),
PAD_BOT / v_samp_factor);
}
}
}
} }
}
/* Update proc map for SAO*/
ihevcd_proc_map_update(ps_proc, proc_type, nctb);
/* Update proc map for Completion of CTB*/
ihevcd_proc_map_update(ps_proc, PROC_ALL, nctb);
{
tile_t *ps_tile;
ps_tile = ps_proc->ps_tile;
num_ctb_tmp -= nctb;
ps_proc->i4_ctb_tile_x += nctb;
ps_proc->i4_ctb_x += nctb;
ps_proc->i4_ctb_slice_x += nctb;
/* Update tile counters */
if(ps_proc->i4_ctb_tile_x >= (ps_tile->u2_wd))
{
/*End of tile row*/
ps_proc->i4_ctb_tile_x = 0;
ps_proc->i4_ctb_x = ps_tile->u1_pos_x;
ps_proc->i4_ctb_tile_y++;
ps_proc->i4_ctb_y++;
if(ps_proc->i4_ctb_tile_y == ps_tile->u2_ht)
{
/* Reached End of Tile */
ps_proc->i4_ctb_tile_y = 0;
ps_proc->i4_ctb_tile_x = 0;
ps_proc->ps_tile++;
//End of picture
if(!((ps_tile->u2_ht + ps_tile->u1_pos_y == ps_sps->i2_pic_ht_in_ctb) && (ps_tile->u2_wd + ps_tile->u1_pos_x == ps_sps->i2_pic_wd_in_ctb)))
{
ps_tile = ps_proc->ps_tile;
ps_proc->i4_ctb_x = ps_tile->u1_pos_x;
ps_proc->i4_ctb_y = ps_tile->u1_pos_y;
}
}
}
}
}
ps_proc->i4_ctb_cnt -= num_ctb;
}
return ret;
}
void ihevcd_init_proc_ctxt(process_ctxt_t *ps_proc, WORD32 tu_coeff_data_ofst)
{
codec_t *ps_codec;
slice_header_t *ps_slice_hdr;
pps_t *ps_pps;
sps_t *ps_sps;
tile_t *ps_tile, *ps_tile_prev;
WORD32 tile_idx;
WORD32 ctb_size;
WORD32 num_minpu_in_ctb;
WORD32 num_ctb_in_row;
WORD32 ctb_addr;
WORD32 i4_wd_in_ctb;
WORD32 tile_start_ctb_idx;
WORD32 slice_start_ctb_idx;
WORD32 check_tile_wd;
WORD32 continuous_tiles = 0; //Refers to tiles that are continuous, within a slice, horizontally
ps_codec = ps_proc->ps_codec;
ps_slice_hdr = ps_codec->ps_slice_hdr_base + ((ps_proc->i4_cur_slice_idx) & (MAX_SLICE_HDR_CNT - 1));
ps_proc->ps_slice_hdr = ps_slice_hdr;
ps_proc->ps_pps = ps_codec->ps_pps_base + ps_slice_hdr->i1_pps_id;
ps_pps = ps_proc->ps_pps;
ps_proc->ps_sps = ps_codec->ps_sps_base + ps_pps->i1_sps_id;
ps_sps = ps_proc->ps_sps;
ps_proc->i4_init_done = 1;
ctb_size = 1 << ps_sps->i1_log2_ctb_size;
num_minpu_in_ctb = (ctb_size / MIN_PU_SIZE) * (ctb_size / MIN_PU_SIZE);
num_ctb_in_row = ps_sps->i2_pic_wd_in_ctb;
ps_proc->s_sao_ctxt.pu1_slice_idx = ps_proc->pu1_slice_idx;
ihevcd_get_tile_pos(ps_pps, ps_sps, ps_proc->i4_ctb_x, ps_proc->i4_ctb_y,
&ps_proc->i4_ctb_tile_x, &ps_proc->i4_ctb_tile_y,
&tile_idx);
ps_proc->ps_tile = ps_pps->ps_tile + tile_idx;
ps_proc->i4_cur_tile_idx = tile_idx;
ps_tile = ps_proc->ps_tile;
if(ps_pps->i1_tiles_enabled_flag)
{
if(tile_idx)
ps_tile_prev = ps_tile - 1;
else
ps_tile_prev = ps_tile;
slice_start_ctb_idx = ps_slice_hdr->i2_ctb_x + (ps_slice_hdr->i2_ctb_y * ps_sps->i2_pic_wd_in_ctb);
tile_start_ctb_idx = ps_tile->u1_pos_x + (ps_tile->u1_pos_y * ps_sps->i2_pic_wd_in_ctb);
/*Check if
* 1. Last tile that ends in frame boundary and 1st tile in next row belongs to same slice
* 1.1. If it does, check if the slice that has these tiles spans across the frame row.
* 2. Vertical tiles are present within a slice */
if(((ps_slice_hdr->i2_ctb_x == ps_tile->u1_pos_x) && (ps_slice_hdr->i2_ctb_y != ps_tile->u1_pos_y)))
{
continuous_tiles = 1;
}
else
{
check_tile_wd = ps_slice_hdr->i2_ctb_x + ps_tile_prev->u2_wd;
if(!(((check_tile_wd >= ps_sps->i2_pic_wd_in_ctb) && (check_tile_wd % ps_sps->i2_pic_wd_in_ctb == ps_tile->u1_pos_x))
|| ((ps_slice_hdr->i2_ctb_x == ps_tile->u1_pos_x))))
{
continuous_tiles = 1;
}
}
{
WORD32 i2_independent_ctb_x = ps_slice_hdr->i2_independent_ctb_x;
WORD32 i2_independent_ctb_y = ps_slice_hdr->i2_independent_ctb_y;
/* Handles cases where
* 1. Slices begin at the start of each tile
* 2. Tiles lie in the same slice row.i.e, starting tile_x > slice_x, but tile_y == slice_y
* */
if(ps_proc->i4_ctb_x >= i2_independent_ctb_x)
{
ps_proc->i4_ctb_slice_x = ps_proc->i4_ctb_x - i2_independent_ctb_x;
}
else
{
/* Indicates multiple tiles in a slice case where
* The new tile belongs to an older slice that started in the previous rows-not the present row
* & (tile_y > slice_y and tile_x < slice_x)
*/
if((slice_start_ctb_idx < tile_start_ctb_idx) && (continuous_tiles))
{
i4_wd_in_ctb = ps_sps->i2_pic_wd_in_ctb;
}
/* Indicates many-tiles-in-one-slice case, for slices that end without spanning the frame width*/
else
{
i4_wd_in_ctb = ps_tile->u2_wd;
}
if(continuous_tiles)
{
ps_proc->i4_ctb_slice_x = i4_wd_in_ctb
- (i2_independent_ctb_x - ps_proc->i4_ctb_x);
}
else
{
ps_proc->i4_ctb_slice_x = ps_proc->i4_ctb_x - ps_tile->u1_pos_x;
}
}
/* Initialize ctb slice y to zero and at the start of slice row initialize it
to difference between ctb_y and slice's start ctb y */
ps_proc->i4_ctb_slice_y = ps_proc->i4_ctb_y - i2_independent_ctb_y;
/*If beginning of tile, check if slice counters are set correctly*/
if((0 == ps_proc->i4_ctb_tile_x) && (0 == ps_proc->i4_ctb_tile_y))
{
if(ps_slice_hdr->i1_dependent_slice_flag)
{
ps_proc->i4_ctb_slice_x = 0;
ps_proc->i4_ctb_slice_y = 0;
}
/*For slices that span across multiple tiles*/
else if(slice_start_ctb_idx < tile_start_ctb_idx)
{
ps_proc->i4_ctb_slice_y = ps_tile->u1_pos_y - i2_independent_ctb_y;
/* Two Cases
* 1 - slice spans across frame-width- but dose not start from 1st column
* 2 - Slice spans across multiple tiles anywhere is a frame
*/
/*TODO:In a multiple slice clip, if an independent slice span across more than 2 tiles in a row, it is not supported*/
if(continuous_tiles) //Case 2-implemented for slices that span not more than 2 tiles
{
if(i2_independent_ctb_y <= ps_tile->u1_pos_y)
{
//Check if ctb x is before or after
if(i2_independent_ctb_x > ps_tile->u1_pos_x)
{
ps_proc->i4_ctb_slice_y -= 1;
}
}
}
}
}
//Slice starts from a column which is not the starting tile-column, but is within the tile
if(((i2_independent_ctb_x - ps_tile->u1_pos_x) != 0) && ((ps_proc->i4_ctb_slice_y != 0))
&& ((i2_independent_ctb_x >= ps_tile->u1_pos_x) && (i2_independent_ctb_x < ps_tile->u1_pos_x + ps_tile->u2_wd)))
{
ps_proc->i4_ctb_slice_y -= 1;
}
}
}
else
{
WORD32 i2_independent_ctb_x = ps_slice_hdr->i2_independent_ctb_x;
WORD32 i2_independent_ctb_y = ps_slice_hdr->i2_independent_ctb_y;
{
ps_proc->i4_ctb_slice_x = ps_proc->i4_ctb_x - i2_independent_ctb_x;
ps_proc->i4_ctb_slice_y = ps_proc->i4_ctb_y - i2_independent_ctb_y;
if(ps_proc->i4_ctb_slice_x < 0)
{
ps_proc->i4_ctb_slice_x += ps_sps->i2_pic_wd_in_ctb;
ps_proc->i4_ctb_slice_y -= 1;
}
/* Initialize ctb slice y to zero and at the start of slice row initialize it
to difference between ctb_y and slice's start ctb y */
}
}
/* Compute TU offset for the current CTB set */
{
WORD32 ctb_luma_min_tu_cnt;
WORD32 ctb_addr;
ctb_addr = ps_proc->i4_ctb_y * num_ctb_in_row + ps_proc->i4_ctb_x;
ctb_luma_min_tu_cnt = (1 << ps_sps->i1_log2_ctb_size) / MIN_TU_SIZE;
ctb_luma_min_tu_cnt *= ctb_luma_min_tu_cnt;
ps_proc->pu1_tu_map = ps_proc->pu1_pic_tu_map
+ ctb_luma_min_tu_cnt * ctb_addr;
if(1 == ps_codec->i4_num_cores)
{
ps_proc->ps_tu = ps_proc->ps_pic_tu + ps_proc->pu4_pic_tu_idx[ctb_addr % RESET_TU_BUF_NCTB];
}
else
{
ps_proc->ps_tu = ps_proc->ps_pic_tu + ps_proc->pu4_pic_tu_idx[ctb_addr];
}
ps_proc->pv_tu_coeff_data = (UWORD8 *)ps_proc->pv_pic_tu_coeff_data
+ tu_coeff_data_ofst;
}
/* Compute PU related elements for the current CTB set */
{
WORD32 pu_idx;
ctb_addr = ps_proc->i4_ctb_y * num_ctb_in_row + ps_proc->i4_ctb_x;
pu_idx = ps_proc->pu4_pic_pu_idx[ctb_addr];
ps_proc->pu1_pu_map = ps_proc->pu1_pic_pu_map
+ ctb_addr * num_minpu_in_ctb;
ps_proc->ps_pu = ps_proc->ps_pic_pu + pu_idx;
}
/* Number of ctbs processed in one loop of process function */
{
ps_proc->i4_nctb = MIN(ps_codec->u4_nctb, ps_tile->u2_wd);
}
}
void ihevcd_process_thread(process_ctxt_t *ps_proc)
{
IHEVCD_ERROR_T ret = (IHEVCD_ERROR_T)IHEVCD_SUCCESS;
{
ithread_set_affinity(ps_proc->i4_id + 1);
}
while(1)
{
codec_t *ps_dec = ps_proc->ps_codec;
if(ps_proc->ps_codec->i4_threads_active)
{
DEBUG("In ihevcd_process_thread \n");
ret = ithread_mutex_lock(ps_dec->apv_proc_start_mutex[ps_proc->i4_id]);
if((IHEVCD_ERROR_T)IHEVCD_SUCCESS != ret)
break;
while(!ps_dec->ai4_process_start[ps_proc->i4_id])
{
ithread_cond_wait(ps_dec->apv_proc_start_condition[ps_proc->i4_id],
ps_dec->apv_proc_start_mutex[ps_proc->i4_id]);
}
ps_dec->ai4_process_start[ps_proc->i4_id] = 0;
ret = ithread_mutex_unlock(ps_dec->apv_proc_start_mutex[ps_proc->i4_id]);
if((IHEVCD_ERROR_T)IHEVCD_SUCCESS != ret)
break;
DEBUG(" Got control at ihevcd_process_thread \n");
if(ps_dec->i4_break_threads == 1)
break;
}
while(1)
{
proc_job_t s_job;
ret = ihevcd_jobq_dequeue((jobq_t *)ps_proc->pv_proc_jobq, &s_job,
sizeof(proc_job_t), 1);
if((IHEVCD_ERROR_T)IHEVCD_SUCCESS != ret)
break;
ps_proc->i4_ctb_cnt = s_job.i2_ctb_cnt;
ps_proc->i4_ctb_x = s_job.i2_ctb_x;
ps_proc->i4_ctb_y = s_job.i2_ctb_y;
ps_proc->i4_cur_slice_idx = s_job.i2_slice_idx;
if(CMD_PROCESS == s_job.i4_cmd)
{
ihevcd_init_proc_ctxt(ps_proc, s_job.i4_tu_coeff_data_ofst);
ihevcd_process(ps_proc);
}
else if(CMD_FMTCONV == s_job.i4_cmd)
{
sps_t *ps_sps;
codec_t *ps_codec;
ivd_out_bufdesc_t *ps_out_buffer;
WORD32 num_rows;
if(0 == ps_proc->i4_init_done)
{
ihevcd_init_proc_ctxt(ps_proc, 0);
}
ps_sps = ps_proc->ps_sps;
ps_codec = ps_proc->ps_codec;
ps_out_buffer = ps_proc->ps_out_buffer;
num_rows = 1 << ps_sps->i1_log2_ctb_size;
num_rows = MIN(num_rows,
(ps_codec->i4_disp_ht - (s_job.i2_ctb_y << ps_sps->i1_log2_ctb_size))
);
if(num_rows < 0)
num_rows = 0;
ihevcd_fmt_conv(ps_proc->ps_codec, ps_proc, ps_out_buffer->pu1_bufs[0],
ps_out_buffer->pu1_bufs[1], ps_out_buffer->pu1_bufs[2],
s_job.i2_ctb_y << ps_sps->i1_log2_ctb_size, num_rows);
}
}
if(ps_proc->ps_codec->i4_threads_active)
{
ret = ithread_mutex_lock(ps_dec->apv_proc_done_mutex[ps_proc->i4_id]);
if((IHEVCD_ERROR_T)IHEVCD_SUCCESS != ret)
break;
ps_dec->ai4_process_done[ps_proc->i4_id] = 1;
ithread_cond_signal(ps_dec->apv_proc_done_condition[ps_proc->i4_id]);
ret = ithread_mutex_unlock(ps_dec->apv_proc_done_mutex[ps_proc->i4_id]);
if((IHEVCD_ERROR_T)IHEVCD_SUCCESS != ret)
break;
}
else
{
break;
}
}
//ithread_exit(0);
return;
}
Reference in a new issue View git blame Copy permalink