xref: /aosp_15_r20/external/libhevc/decoder/ihevcd_process_slice.c (revision c83a76b084498d55f252f48b2e3786804cdf24b7)
1*c83a76b0SSuyog Pawar /******************************************************************************
2*c83a76b0SSuyog Pawar *
3*c83a76b0SSuyog Pawar * Copyright (C) 2012 Ittiam Systems Pvt Ltd, Bangalore
4*c83a76b0SSuyog Pawar *
5*c83a76b0SSuyog Pawar * Licensed under the Apache License, Version 2.0 (the "License");
6*c83a76b0SSuyog Pawar * you may not use this file except in compliance with the License.
7*c83a76b0SSuyog Pawar * You may obtain a copy of the License at:
8*c83a76b0SSuyog Pawar *
9*c83a76b0SSuyog Pawar * http://www.apache.org/licenses/LICENSE-2.0
10*c83a76b0SSuyog Pawar *
11*c83a76b0SSuyog Pawar * Unless required by applicable law or agreed to in writing, software
12*c83a76b0SSuyog Pawar * distributed under the License is distributed on an "AS IS" BASIS,
13*c83a76b0SSuyog Pawar * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14*c83a76b0SSuyog Pawar * See the License for the specific language governing permissions and
15*c83a76b0SSuyog Pawar * limitations under the License.
16*c83a76b0SSuyog Pawar *
17*c83a76b0SSuyog Pawar ******************************************************************************/
18*c83a76b0SSuyog Pawar /**
19*c83a76b0SSuyog Pawar  *******************************************************************************
20*c83a76b0SSuyog Pawar  * @file
21*c83a76b0SSuyog Pawar  *  ihevcd_process_slice.c
22*c83a76b0SSuyog Pawar  *
23*c83a76b0SSuyog Pawar  * @brief
24*c83a76b0SSuyog Pawar  *  Contains functions for processing slice data
25*c83a76b0SSuyog Pawar  *
26*c83a76b0SSuyog Pawar  * @author
27*c83a76b0SSuyog Pawar  *  Harish
28*c83a76b0SSuyog Pawar  *
29*c83a76b0SSuyog Pawar  * @par List of Functions:
30*c83a76b0SSuyog Pawar  *
31*c83a76b0SSuyog Pawar  * @remarks
32*c83a76b0SSuyog Pawar  *  None
33*c83a76b0SSuyog Pawar  *
34*c83a76b0SSuyog Pawar  *******************************************************************************
35*c83a76b0SSuyog Pawar  */
36*c83a76b0SSuyog Pawar /*****************************************************************************/
37*c83a76b0SSuyog Pawar /* File Includes                                                             */
38*c83a76b0SSuyog Pawar /*****************************************************************************/
39*c83a76b0SSuyog Pawar #include <stdio.h>
40*c83a76b0SSuyog Pawar #include <stddef.h>
41*c83a76b0SSuyog Pawar #include <stdlib.h>
42*c83a76b0SSuyog Pawar #include <string.h>
43*c83a76b0SSuyog Pawar #include <assert.h>
44*c83a76b0SSuyog Pawar 
45*c83a76b0SSuyog Pawar #include "ihevc_typedefs.h"
46*c83a76b0SSuyog Pawar #include "iv.h"
47*c83a76b0SSuyog Pawar #include "ivd.h"
48*c83a76b0SSuyog Pawar #include "ihevcd_cxa.h"
49*c83a76b0SSuyog Pawar #include "ithread.h"
50*c83a76b0SSuyog Pawar 
51*c83a76b0SSuyog Pawar #include "ihevc_defs.h"
52*c83a76b0SSuyog Pawar #include "ihevc_debug.h"
53*c83a76b0SSuyog Pawar #include "ihevc_defs.h"
54*c83a76b0SSuyog Pawar #include "ihevc_structs.h"
55*c83a76b0SSuyog Pawar #include "ihevc_macros.h"
56*c83a76b0SSuyog Pawar #include "ihevc_platform_macros.h"
57*c83a76b0SSuyog Pawar #include "ihevc_cabac_tables.h"
58*c83a76b0SSuyog Pawar #include "ihevc_padding.h"
59*c83a76b0SSuyog Pawar #include "ihevc_iquant_itrans_recon.h"
60*c83a76b0SSuyog Pawar #include "ihevc_chroma_iquant_itrans_recon.h"
61*c83a76b0SSuyog Pawar #include "ihevc_recon.h"
62*c83a76b0SSuyog Pawar #include "ihevc_chroma_recon.h"
63*c83a76b0SSuyog Pawar #include "ihevc_iquant_recon.h"
64*c83a76b0SSuyog Pawar #include "ihevc_chroma_iquant_recon.h"
65*c83a76b0SSuyog Pawar #include "ihevc_intra_pred.h"
66*c83a76b0SSuyog Pawar 
67*c83a76b0SSuyog Pawar #include "ihevc_error.h"
68*c83a76b0SSuyog Pawar #include "ihevc_common_tables.h"
69*c83a76b0SSuyog Pawar #include "ihevc_quant_tables.h"
70*c83a76b0SSuyog Pawar #include "ihevcd_common_tables.h"
71*c83a76b0SSuyog Pawar 
72*c83a76b0SSuyog Pawar #include "ihevcd_profile.h"
73*c83a76b0SSuyog Pawar #include "ihevcd_trace.h"
74*c83a76b0SSuyog Pawar #include "ihevcd_defs.h"
75*c83a76b0SSuyog Pawar #include "ihevcd_function_selector.h"
76*c83a76b0SSuyog Pawar #include "ihevcd_structs.h"
77*c83a76b0SSuyog Pawar #include "ihevcd_error.h"
78*c83a76b0SSuyog Pawar #include "ihevcd_nal.h"
79*c83a76b0SSuyog Pawar #include "ihevcd_bitstream.h"
80*c83a76b0SSuyog Pawar #include "ihevcd_job_queue.h"
81*c83a76b0SSuyog Pawar #include "ihevcd_utils.h"
82*c83a76b0SSuyog Pawar #include "ihevcd_debug.h"
83*c83a76b0SSuyog Pawar #include "ihevcd_get_mv.h"
84*c83a76b0SSuyog Pawar #include "ihevcd_inter_pred.h"
85*c83a76b0SSuyog Pawar #include "ihevcd_iquant_itrans_recon_ctb.h"
86*c83a76b0SSuyog Pawar #include "ihevcd_boundary_strength.h"
87*c83a76b0SSuyog Pawar #include "ihevcd_deblk.h"
88*c83a76b0SSuyog Pawar #include "ihevcd_fmt_conv.h"
89*c83a76b0SSuyog Pawar #include "ihevcd_sao.h"
90*c83a76b0SSuyog Pawar #include "ihevcd_profile.h"
91*c83a76b0SSuyog Pawar 
92*c83a76b0SSuyog Pawar IHEVCD_ERROR_T ihevcd_fmt_conv(codec_t *ps_codec,
93*c83a76b0SSuyog Pawar                                process_ctxt_t *ps_proc,
94*c83a76b0SSuyog Pawar                                UWORD8 *pu1_y_dst,
95*c83a76b0SSuyog Pawar                                UWORD8 *pu1_u_dst,
96*c83a76b0SSuyog Pawar                                UWORD8 *pu1_v_dst,
97*c83a76b0SSuyog Pawar                                WORD32 cur_row,
98*c83a76b0SSuyog Pawar                                WORD32 num_rows);
99*c83a76b0SSuyog Pawar 
100*c83a76b0SSuyog Pawar typedef enum
101*c83a76b0SSuyog Pawar {
102*c83a76b0SSuyog Pawar     PROC_ALL,
103*c83a76b0SSuyog Pawar     PROC_INTER_PRED,
104*c83a76b0SSuyog Pawar     PROC_RECON,
105*c83a76b0SSuyog Pawar     PROC_DEBLK,
106*c83a76b0SSuyog Pawar     PROC_SAO
107*c83a76b0SSuyog Pawar }proc_type_t;
108*c83a76b0SSuyog Pawar 
ihevcd_proc_map_check(process_ctxt_t * ps_proc,proc_type_t proc_type,WORD32 nctb)109*c83a76b0SSuyog Pawar void ihevcd_proc_map_check(process_ctxt_t *ps_proc, proc_type_t proc_type, WORD32 nctb)
110*c83a76b0SSuyog Pawar {
111*c83a76b0SSuyog Pawar     tile_t *ps_tile = ps_proc->ps_tile;
112*c83a76b0SSuyog Pawar     sps_t *ps_sps = ps_proc->ps_sps;
113*c83a76b0SSuyog Pawar     pps_t *ps_pps = ps_proc->ps_pps;
114*c83a76b0SSuyog Pawar     codec_t *ps_codec = ps_proc->ps_codec;
115*c83a76b0SSuyog Pawar     WORD32 idx;
116*c83a76b0SSuyog Pawar     WORD32 nop_cnt;
117*c83a76b0SSuyog Pawar     WORD32 bit_pos = proc_type;
118*c83a76b0SSuyog Pawar     WORD32 bit_mask = (1 << bit_pos);
119*c83a76b0SSuyog Pawar 
120*c83a76b0SSuyog Pawar     if(ps_proc->i4_check_proc_status)
121*c83a76b0SSuyog Pawar     {
122*c83a76b0SSuyog Pawar         nop_cnt = PROC_NOP_CNT;
123*c83a76b0SSuyog Pawar         while(1)
124*c83a76b0SSuyog Pawar         {
125*c83a76b0SSuyog Pawar             volatile UWORD8 *pu1_buf;
126*c83a76b0SSuyog Pawar             volatile WORD32 status;
127*c83a76b0SSuyog Pawar             status = 1;
128*c83a76b0SSuyog Pawar             /* Check if all dependencies for the next nCTBs are met */
129*c83a76b0SSuyog Pawar             {
130*c83a76b0SSuyog Pawar                 WORD32 x_pos;
131*c83a76b0SSuyog Pawar 
132*c83a76b0SSuyog Pawar                 {
133*c83a76b0SSuyog Pawar                     /* Check if the top right of next nCTBs are processed */
134*c83a76b0SSuyog Pawar                     if(ps_proc->i4_ctb_y > 0)
135*c83a76b0SSuyog Pawar                     {
136*c83a76b0SSuyog Pawar                         x_pos = (ps_proc->i4_ctb_tile_x + nctb);
137*c83a76b0SSuyog Pawar                         idx = MIN(x_pos, (ps_tile->u2_wd - 1));
138*c83a76b0SSuyog Pawar 
139*c83a76b0SSuyog Pawar                         /* Check if top-right CTB for the last CTB in nCTB is within the tile */
140*c83a76b0SSuyog Pawar                         {
141*c83a76b0SSuyog Pawar                             idx += ps_tile->u1_pos_x;
142*c83a76b0SSuyog Pawar                             idx += ((ps_proc->i4_ctb_y - 1)
143*c83a76b0SSuyog Pawar                                             * ps_sps->i2_pic_wd_in_ctb);
144*c83a76b0SSuyog Pawar                             pu1_buf = (ps_codec->pu1_proc_map + idx);
145*c83a76b0SSuyog Pawar                             status = *pu1_buf & bit_mask;
146*c83a76b0SSuyog Pawar                         }
147*c83a76b0SSuyog Pawar                     }
148*c83a76b0SSuyog Pawar                 }
149*c83a76b0SSuyog Pawar 
150*c83a76b0SSuyog Pawar                 /* If tiles are enabled, then test left and top-left as well */
151*c83a76b0SSuyog Pawar                 ps_pps = ps_proc->ps_pps;
152*c83a76b0SSuyog Pawar                 if(ps_pps->i1_tiles_enabled_flag)
153*c83a76b0SSuyog Pawar                 {
154*c83a76b0SSuyog Pawar                     /*Check if left ctb is processed*/
155*c83a76b0SSuyog Pawar                     if((ps_proc->i4_ctb_x > 0) && ((0 != status)))
156*c83a76b0SSuyog Pawar                     {
157*c83a76b0SSuyog Pawar                         x_pos   = ps_tile->u1_pos_x + ps_proc->i4_ctb_tile_x - 1;
158*c83a76b0SSuyog Pawar                         idx     = x_pos + (ps_proc->i4_ctb_y * ps_sps->i2_pic_wd_in_ctb);
159*c83a76b0SSuyog Pawar                         pu1_buf = (ps_codec->pu1_proc_map + idx);
160*c83a76b0SSuyog Pawar                         status  = *pu1_buf & bit_mask;
161*c83a76b0SSuyog Pawar                     }
162*c83a76b0SSuyog Pawar 
163*c83a76b0SSuyog Pawar                     /*Check if top left ctb is processed*/
164*c83a76b0SSuyog Pawar                     if((ps_proc->i4_ctb_x > 0) && (0 != status) && (ps_proc->i4_ctb_y > 0))
165*c83a76b0SSuyog Pawar                     {
166*c83a76b0SSuyog Pawar                         x_pos   = ps_tile->u1_pos_x + ps_proc->i4_ctb_tile_x - 1;
167*c83a76b0SSuyog Pawar                         idx     = x_pos + ((ps_proc->i4_ctb_y - 1) * ps_sps->i2_pic_wd_in_ctb);
168*c83a76b0SSuyog Pawar                         pu1_buf = (ps_codec->pu1_proc_map + idx);
169*c83a76b0SSuyog Pawar                         status  = *pu1_buf & bit_mask;
170*c83a76b0SSuyog Pawar                     }
171*c83a76b0SSuyog Pawar                 }
172*c83a76b0SSuyog Pawar             }
173*c83a76b0SSuyog Pawar 
174*c83a76b0SSuyog Pawar             if(status)
175*c83a76b0SSuyog Pawar                 break;
176*c83a76b0SSuyog Pawar 
177*c83a76b0SSuyog Pawar             /* if dependencies are not met, then wait for few cycles.
178*c83a76b0SSuyog Pawar              * Even after few iterations, if the dependencies are not met then yield
179*c83a76b0SSuyog Pawar              */
180*c83a76b0SSuyog Pawar             if(nop_cnt > 0)
181*c83a76b0SSuyog Pawar             {
182*c83a76b0SSuyog Pawar                 NOP(128);
183*c83a76b0SSuyog Pawar                 nop_cnt -= 128;
184*c83a76b0SSuyog Pawar             }
185*c83a76b0SSuyog Pawar             else
186*c83a76b0SSuyog Pawar             {
187*c83a76b0SSuyog Pawar                 nop_cnt = PROC_NOP_CNT;
188*c83a76b0SSuyog Pawar                 ithread_yield();
189*c83a76b0SSuyog Pawar                 //NOP(128 * 16);
190*c83a76b0SSuyog Pawar             }
191*c83a76b0SSuyog Pawar         }
192*c83a76b0SSuyog Pawar         DATA_SYNC();
193*c83a76b0SSuyog Pawar     }
194*c83a76b0SSuyog Pawar }
195*c83a76b0SSuyog Pawar 
ihevcd_proc_map_update(process_ctxt_t * ps_proc,proc_type_t proc_type,WORD32 nctb)196*c83a76b0SSuyog Pawar void ihevcd_proc_map_update(process_ctxt_t *ps_proc, proc_type_t proc_type, WORD32 nctb)
197*c83a76b0SSuyog Pawar {
198*c83a76b0SSuyog Pawar     codec_t *ps_codec = ps_proc->ps_codec;
199*c83a76b0SSuyog Pawar     WORD32 i, idx;
200*c83a76b0SSuyog Pawar     WORD32 bit_pos = proc_type;
201*c83a76b0SSuyog Pawar     WORD32 bit_mask = (1 << bit_pos);
202*c83a76b0SSuyog Pawar 
203*c83a76b0SSuyog Pawar     /* Update the current CTBs processing status */
204*c83a76b0SSuyog Pawar     if(ps_proc->i4_check_proc_status)
205*c83a76b0SSuyog Pawar     {
206*c83a76b0SSuyog Pawar         DATA_SYNC();
207*c83a76b0SSuyog Pawar         for(i = 0; i < nctb; i++)
208*c83a76b0SSuyog Pawar         {
209*c83a76b0SSuyog Pawar             sps_t *ps_sps = ps_proc->ps_sps;
210*c83a76b0SSuyog Pawar             UWORD8 *pu1_buf;
211*c83a76b0SSuyog Pawar             idx = (ps_proc->i4_ctb_x + i);
212*c83a76b0SSuyog Pawar             idx += ((ps_proc->i4_ctb_y) * ps_sps->i2_pic_wd_in_ctb);
213*c83a76b0SSuyog Pawar             pu1_buf = (ps_codec->pu1_proc_map + idx);
214*c83a76b0SSuyog Pawar             *pu1_buf = *pu1_buf | bit_mask;
215*c83a76b0SSuyog Pawar         }
216*c83a76b0SSuyog Pawar     }
217*c83a76b0SSuyog Pawar }
218*c83a76b0SSuyog Pawar 
219*c83a76b0SSuyog Pawar 
ihevcd_slice_hdr_update(process_ctxt_t * ps_proc)220*c83a76b0SSuyog Pawar void ihevcd_slice_hdr_update(process_ctxt_t *ps_proc)
221*c83a76b0SSuyog Pawar {
222*c83a76b0SSuyog Pawar 
223*c83a76b0SSuyog Pawar     /* Slice x and y are initialized in proc_init. But initialize slice x and y count here
224*c83a76b0SSuyog Pawar      *  if a new slice begins at the middle of a row since proc_init is invoked only at the beginning of each row */
225*c83a76b0SSuyog Pawar     if(!((ps_proc->i4_ctb_x == 0) && (ps_proc->i4_ctb_y == 0)))
226*c83a76b0SSuyog Pawar     {
227*c83a76b0SSuyog Pawar         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));
228*c83a76b0SSuyog Pawar 
229*c83a76b0SSuyog Pawar         if((ps_slice_hdr_next->i2_ctb_x == ps_proc->i4_ctb_x)
230*c83a76b0SSuyog Pawar                         && (ps_slice_hdr_next->i2_ctb_y == ps_proc->i4_ctb_y))
231*c83a76b0SSuyog Pawar         {
232*c83a76b0SSuyog Pawar             if(0 == ps_slice_hdr_next->i1_dependent_slice_flag)
233*c83a76b0SSuyog Pawar             {
234*c83a76b0SSuyog Pawar                 ps_proc->i4_ctb_slice_x = 0;
235*c83a76b0SSuyog Pawar                 ps_proc->i4_ctb_slice_y = 0;
236*c83a76b0SSuyog Pawar             }
237*c83a76b0SSuyog Pawar 
238*c83a76b0SSuyog Pawar             ps_proc->i4_cur_slice_idx++;
239*c83a76b0SSuyog Pawar             ps_proc->ps_slice_hdr = ps_slice_hdr_next;
240*c83a76b0SSuyog Pawar         }
241*c83a76b0SSuyog Pawar 
242*c83a76b0SSuyog Pawar     }
243*c83a76b0SSuyog Pawar }
244*c83a76b0SSuyog Pawar 
ihevcd_ctb_pos_update(process_ctxt_t * ps_proc,WORD32 nctb)245*c83a76b0SSuyog Pawar void ihevcd_ctb_pos_update(process_ctxt_t *ps_proc, WORD32 nctb)
246*c83a76b0SSuyog Pawar {
247*c83a76b0SSuyog Pawar     WORD32 tile_start_ctb_idx, slice_start_ctb_idx;
248*c83a76b0SSuyog Pawar     slice_header_t *ps_slice_hdr = ps_proc->ps_slice_hdr;
249*c83a76b0SSuyog Pawar     tile_t *ps_tile = ps_proc->ps_tile;
250*c83a76b0SSuyog Pawar     sps_t *ps_sps = ps_proc->ps_sps;
251*c83a76b0SSuyog Pawar 
252*c83a76b0SSuyog Pawar     /* Update x and y positions */
253*c83a76b0SSuyog Pawar     ps_proc->i4_ctb_tile_x += nctb;
254*c83a76b0SSuyog Pawar     ps_proc->i4_ctb_x += nctb;
255*c83a76b0SSuyog Pawar 
256*c83a76b0SSuyog Pawar     ps_proc->i4_ctb_slice_x += nctb;
257*c83a76b0SSuyog Pawar     /*If tile are enabled, then handle the tile & slice counters differently*/
258*c83a76b0SSuyog Pawar     if(ps_proc->ps_pps->i1_tiles_enabled_flag)
259*c83a76b0SSuyog Pawar     {
260*c83a76b0SSuyog Pawar         /* Update slice counters*/
261*c83a76b0SSuyog Pawar         slice_start_ctb_idx = ps_slice_hdr->i2_ctb_x + (ps_slice_hdr->i2_ctb_y * ps_sps->i2_pic_wd_in_ctb);
262*c83a76b0SSuyog Pawar         tile_start_ctb_idx = ps_tile->u1_pos_x + (ps_tile->u1_pos_y * ps_sps->i2_pic_wd_in_ctb);
263*c83a76b0SSuyog Pawar         /*
264*c83a76b0SSuyog Pawar          * There can be 2 cases where slice counters must be handled differently.
265*c83a76b0SSuyog Pawar          * 1 - Multiple tiles span across a single/one of the many slice.
266*c83a76b0SSuyog Pawar          * 2 - Multiple slices span across a single/one of the many tiles.
267*c83a76b0SSuyog Pawar          */
268*c83a76b0SSuyog Pawar 
269*c83a76b0SSuyog Pawar         /*Case 1 */
270*c83a76b0SSuyog Pawar         if(slice_start_ctb_idx < tile_start_ctb_idx)
271*c83a76b0SSuyog Pawar         {
272*c83a76b0SSuyog Pawar             /*End of tile row*/
273*c83a76b0SSuyog Pawar             if(ps_proc->i4_ctb_x > ps_slice_hdr->i2_ctb_x)
274*c83a76b0SSuyog Pawar             {
275*c83a76b0SSuyog Pawar                 if(ps_proc->i4_ctb_slice_x >= (ps_tile->u2_wd + ps_tile->u1_pos_x))
276*c83a76b0SSuyog Pawar                 {
277*c83a76b0SSuyog Pawar                     ps_proc->i4_ctb_slice_y++;
278*c83a76b0SSuyog Pawar                     ps_proc->i4_ctb_slice_x = ps_proc->i4_ctb_slice_x
279*c83a76b0SSuyog Pawar                                     - ps_tile->u2_wd;
280*c83a76b0SSuyog Pawar                 }
281*c83a76b0SSuyog Pawar             }
282*c83a76b0SSuyog Pawar             else
283*c83a76b0SSuyog Pawar             {
284*c83a76b0SSuyog Pawar                 WORD32 temp_stride = (ps_sps->i2_pic_wd_in_ctb - ps_slice_hdr->i2_ctb_x);
285*c83a76b0SSuyog Pawar                 if(ps_proc->i4_ctb_slice_x >= (temp_stride + ps_tile->u2_wd + ps_tile->u1_pos_x))
286*c83a76b0SSuyog Pawar                 {
287*c83a76b0SSuyog Pawar                     ps_proc->i4_ctb_slice_y++;
288*c83a76b0SSuyog Pawar                     ps_proc->i4_ctb_slice_x = ps_proc->i4_ctb_slice_x
289*c83a76b0SSuyog Pawar                                     - ps_tile->u2_wd;
290*c83a76b0SSuyog Pawar                 }
291*c83a76b0SSuyog Pawar             }
292*c83a76b0SSuyog Pawar         }
293*c83a76b0SSuyog Pawar         /*Case 2*/
294*c83a76b0SSuyog Pawar         else if(ps_proc->i4_ctb_slice_x >= (ps_tile->u2_wd))
295*c83a76b0SSuyog Pawar         {
296*c83a76b0SSuyog Pawar             /*End of tile row*/
297*c83a76b0SSuyog Pawar             ps_proc->i4_ctb_slice_y++;
298*c83a76b0SSuyog Pawar             ps_proc->i4_ctb_slice_x = 0;
299*c83a76b0SSuyog Pawar         }
300*c83a76b0SSuyog Pawar     }
301*c83a76b0SSuyog Pawar     else
302*c83a76b0SSuyog Pawar     {
303*c83a76b0SSuyog Pawar         if(ps_proc->i4_ctb_slice_x >= ps_tile->u2_wd)
304*c83a76b0SSuyog Pawar         {
305*c83a76b0SSuyog Pawar             ps_proc->i4_ctb_slice_y++;
306*c83a76b0SSuyog Pawar             ps_proc->i4_ctb_slice_x = ps_proc->i4_ctb_slice_x
307*c83a76b0SSuyog Pawar                             - ps_tile->u2_wd;
308*c83a76b0SSuyog Pawar         }
309*c83a76b0SSuyog Pawar     }
310*c83a76b0SSuyog Pawar }
311*c83a76b0SSuyog Pawar 
ihevcd_ctb_avail_update(process_ctxt_t * ps_proc)312*c83a76b0SSuyog Pawar void ihevcd_ctb_avail_update(process_ctxt_t *ps_proc)
313*c83a76b0SSuyog Pawar {
314*c83a76b0SSuyog Pawar     slice_header_t *ps_slice_hdr = ps_proc->ps_slice_hdr;
315*c83a76b0SSuyog Pawar     sps_t *ps_sps = ps_proc->ps_sps;
316*c83a76b0SSuyog Pawar     tile_t *ps_tile_prev;
317*c83a76b0SSuyog Pawar     tile_t *ps_tile = ps_proc->ps_tile;
318*c83a76b0SSuyog Pawar     WORD32 cur_pu_idx;
319*c83a76b0SSuyog Pawar     WORD32 tile_start_ctb_idx, slice_start_ctb_idx;
320*c83a76b0SSuyog Pawar     WORD16 i2_wd_in_ctb;
321*c83a76b0SSuyog Pawar     WORD32 continuous_tiles = 0;
322*c83a76b0SSuyog Pawar     WORD32 cur_ctb_idx;
323*c83a76b0SSuyog Pawar     WORD32 check_tile_wd;
324*c83a76b0SSuyog Pawar 
325*c83a76b0SSuyog Pawar     if((0 != ps_tile->u1_pos_x) && (0 != ps_tile->u1_pos_y))
326*c83a76b0SSuyog Pawar     {
327*c83a76b0SSuyog Pawar         ps_tile_prev = ps_tile - 1;
328*c83a76b0SSuyog Pawar     }
329*c83a76b0SSuyog Pawar     else
330*c83a76b0SSuyog Pawar     {
331*c83a76b0SSuyog Pawar         ps_tile_prev = ps_tile;
332*c83a76b0SSuyog Pawar     }
333*c83a76b0SSuyog Pawar 
334*c83a76b0SSuyog Pawar 
335*c83a76b0SSuyog Pawar     check_tile_wd = ps_slice_hdr->i2_ctb_x + ps_tile_prev->u2_wd;
336*c83a76b0SSuyog Pawar     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))
337*c83a76b0SSuyog Pawar                                     || ((ps_slice_hdr->i2_ctb_x == ps_tile->u1_pos_x))))
338*c83a76b0SSuyog Pawar     {
339*c83a76b0SSuyog Pawar         continuous_tiles = 1;
340*c83a76b0SSuyog Pawar     }
341*c83a76b0SSuyog Pawar 
342*c83a76b0SSuyog Pawar     slice_start_ctb_idx = ps_slice_hdr->i2_ctb_x + (ps_slice_hdr->i2_ctb_y * ps_sps->i2_pic_wd_in_ctb);
343*c83a76b0SSuyog Pawar     tile_start_ctb_idx = ps_tile->u1_pos_x + (ps_tile->u1_pos_y * ps_sps->i2_pic_wd_in_ctb);
344*c83a76b0SSuyog Pawar 
345*c83a76b0SSuyog Pawar     if((slice_start_ctb_idx < tile_start_ctb_idx) && (continuous_tiles))
346*c83a76b0SSuyog Pawar     {
347*c83a76b0SSuyog Pawar         //Slices span across multiple tiles.
348*c83a76b0SSuyog Pawar         i2_wd_in_ctb = ps_sps->i2_pic_wd_in_ctb;
349*c83a76b0SSuyog Pawar     }
350*c83a76b0SSuyog Pawar     else
351*c83a76b0SSuyog Pawar     {
352*c83a76b0SSuyog Pawar         i2_wd_in_ctb = ps_tile->u2_wd;
353*c83a76b0SSuyog Pawar     }
354*c83a76b0SSuyog Pawar     cur_ctb_idx = ps_proc->i4_ctb_x
355*c83a76b0SSuyog Pawar                     + ps_proc->i4_ctb_y * (ps_sps->i2_pic_wd_in_ctb);
356*c83a76b0SSuyog Pawar 
357*c83a76b0SSuyog Pawar     /* Ctb level availability */
358*c83a76b0SSuyog Pawar     /* Bottom left will not be available at a CTB level, no need to pass this */
359*c83a76b0SSuyog Pawar     ps_proc->u1_top_ctb_avail = 1;
360*c83a76b0SSuyog Pawar     ps_proc->u1_left_ctb_avail = 1;
361*c83a76b0SSuyog Pawar     ps_proc->u1_top_lt_ctb_avail = 1;
362*c83a76b0SSuyog Pawar     ps_proc->u1_top_rt_ctb_avail = 1;
363*c83a76b0SSuyog Pawar     /* slice and tile boundaries */
364*c83a76b0SSuyog Pawar 
365*c83a76b0SSuyog Pawar     if((0 == ps_proc->i4_ctb_y) || (0 == ps_proc->i4_ctb_tile_y))
366*c83a76b0SSuyog Pawar     {
367*c83a76b0SSuyog Pawar         ps_proc->u1_top_ctb_avail = 0;
368*c83a76b0SSuyog Pawar         ps_proc->u1_top_lt_ctb_avail = 0;
369*c83a76b0SSuyog Pawar         ps_proc->u1_top_rt_ctb_avail = 0;
370*c83a76b0SSuyog Pawar     }
371*c83a76b0SSuyog Pawar 
372*c83a76b0SSuyog Pawar     if((0 == ps_proc->i4_ctb_x) || (0 == ps_proc->i4_ctb_tile_x))
373*c83a76b0SSuyog Pawar     {
374*c83a76b0SSuyog Pawar         ps_proc->u1_left_ctb_avail = 0;
375*c83a76b0SSuyog Pawar         ps_proc->u1_top_lt_ctb_avail = 0;
376*c83a76b0SSuyog Pawar         if((0 == ps_proc->i4_ctb_slice_y) || (0 == ps_proc->i4_ctb_tile_y))
377*c83a76b0SSuyog Pawar         {
378*c83a76b0SSuyog Pawar             ps_proc->u1_top_ctb_avail = 0;
379*c83a76b0SSuyog Pawar             if((i2_wd_in_ctb - 1) != ps_proc->i4_ctb_slice_x)
380*c83a76b0SSuyog Pawar             {
381*c83a76b0SSuyog Pawar                 ps_proc->u1_top_rt_ctb_avail = 0;
382*c83a76b0SSuyog Pawar             }
383*c83a76b0SSuyog Pawar         }
384*c83a76b0SSuyog Pawar     }
385*c83a76b0SSuyog Pawar     /*For slices not beginning at start of a ctb row*/
386*c83a76b0SSuyog Pawar     else if(ps_proc->i4_ctb_x > 0)
387*c83a76b0SSuyog Pawar     {
388*c83a76b0SSuyog Pawar         if((0 == ps_proc->i4_ctb_slice_y) || (0 == ps_proc->i4_ctb_tile_y))
389*c83a76b0SSuyog Pawar         {
390*c83a76b0SSuyog Pawar             ps_proc->u1_top_ctb_avail = 0;
391*c83a76b0SSuyog Pawar             ps_proc->u1_top_lt_ctb_avail = 0;
392*c83a76b0SSuyog Pawar             if(0 == ps_proc->i4_ctb_slice_x)
393*c83a76b0SSuyog Pawar             {
394*c83a76b0SSuyog Pawar                 ps_proc->u1_left_ctb_avail = 0;
395*c83a76b0SSuyog Pawar             }
396*c83a76b0SSuyog Pawar             if((i2_wd_in_ctb - 1) != ps_proc->i4_ctb_slice_x)
397*c83a76b0SSuyog Pawar             {
398*c83a76b0SSuyog Pawar                 ps_proc->u1_top_rt_ctb_avail = 0;
399*c83a76b0SSuyog Pawar             }
400*c83a76b0SSuyog Pawar         }
401*c83a76b0SSuyog Pawar         else if((1 == ps_proc->i4_ctb_slice_y) && (0 == ps_proc->i4_ctb_slice_x))
402*c83a76b0SSuyog Pawar         {
403*c83a76b0SSuyog Pawar             ps_proc->u1_top_lt_ctb_avail = 0;
404*c83a76b0SSuyog Pawar         }
405*c83a76b0SSuyog Pawar     }
406*c83a76b0SSuyog Pawar 
407*c83a76b0SSuyog Pawar     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))
408*c83a76b0SSuyog Pawar     {
409*c83a76b0SSuyog Pawar         ps_proc->u1_top_rt_ctb_avail = 0;
410*c83a76b0SSuyog Pawar     }
411*c83a76b0SSuyog Pawar 
412*c83a76b0SSuyog Pawar 
413*c83a76b0SSuyog Pawar     {
414*c83a76b0SSuyog Pawar         WORD32 next_ctb_idx;
415*c83a76b0SSuyog Pawar         next_ctb_idx = cur_ctb_idx + 1;
416*c83a76b0SSuyog Pawar 
417*c83a76b0SSuyog Pawar         if(ps_tile->u2_wd == (ps_proc->i4_ctb_tile_x + 1))
418*c83a76b0SSuyog Pawar         {
419*c83a76b0SSuyog Pawar             if((ps_proc->i4_ctb_tile_y + 1) == ps_tile->u2_ht)
420*c83a76b0SSuyog Pawar             {
421*c83a76b0SSuyog Pawar                 //Last tile
422*c83a76b0SSuyog Pawar                 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))
423*c83a76b0SSuyog Pawar                 {
424*c83a76b0SSuyog Pawar                     next_ctb_idx = cur_ctb_idx + 1;
425*c83a76b0SSuyog Pawar                 }
426*c83a76b0SSuyog Pawar                 else //Not last tile, but new tile
427*c83a76b0SSuyog Pawar                 {
428*c83a76b0SSuyog Pawar                     tile_t *ps_tile_next = ps_proc->ps_tile + 1;
429*c83a76b0SSuyog Pawar                     next_ctb_idx = ps_tile_next->u1_pos_x + (ps_tile_next->u1_pos_y * ps_sps->i2_pic_wd_in_ctb);
430*c83a76b0SSuyog Pawar                 }
431*c83a76b0SSuyog Pawar             }
432*c83a76b0SSuyog Pawar             else //End of each tile row
433*c83a76b0SSuyog Pawar             {
434*c83a76b0SSuyog Pawar                 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;
435*c83a76b0SSuyog Pawar             }
436*c83a76b0SSuyog Pawar         }
437*c83a76b0SSuyog Pawar         ps_proc->i4_next_pu_ctb_cnt = next_ctb_idx;
438*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_pu_cnt =
439*c83a76b0SSuyog Pawar                         ps_proc->pu4_pic_pu_idx[next_ctb_idx]
440*c83a76b0SSuyog Pawar                         - ps_proc->pu4_pic_pu_idx[cur_ctb_idx];
441*c83a76b0SSuyog Pawar         cur_pu_idx = ps_proc->pu4_pic_pu_idx[cur_ctb_idx];
442*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_start_pu_idx = cur_pu_idx;
443*c83a76b0SSuyog Pawar         ps_proc->ps_pu = &ps_proc->ps_pic_pu[cur_pu_idx];
444*c83a76b0SSuyog Pawar     }
445*c83a76b0SSuyog Pawar }
446*c83a76b0SSuyog Pawar 
ihevcd_update_ctb_tu_cnt(process_ctxt_t * ps_proc)447*c83a76b0SSuyog Pawar void ihevcd_update_ctb_tu_cnt(process_ctxt_t *ps_proc)
448*c83a76b0SSuyog Pawar {
449*c83a76b0SSuyog Pawar     sps_t *ps_sps = ps_proc->ps_sps;
450*c83a76b0SSuyog Pawar     codec_t *ps_codec = ps_proc->ps_codec;
451*c83a76b0SSuyog Pawar     WORD32 cur_ctb_idx;
452*c83a76b0SSuyog Pawar 
453*c83a76b0SSuyog Pawar     cur_ctb_idx = ps_proc->i4_ctb_x
454*c83a76b0SSuyog Pawar                     + ps_proc->i4_ctb_y * (ps_sps->i2_pic_wd_in_ctb);
455*c83a76b0SSuyog Pawar 
456*c83a76b0SSuyog Pawar     {
457*c83a76b0SSuyog Pawar         tile_t *ps_tile;
458*c83a76b0SSuyog Pawar         WORD32 next_ctb_tu_idx;
459*c83a76b0SSuyog Pawar         ps_tile = ps_proc->ps_tile;
460*c83a76b0SSuyog Pawar 
461*c83a76b0SSuyog Pawar 
462*c83a76b0SSuyog Pawar         if(1 == ps_codec->i4_num_cores)
463*c83a76b0SSuyog Pawar         {
464*c83a76b0SSuyog Pawar             next_ctb_tu_idx = cur_ctb_idx % RESET_TU_BUF_NCTB + 1;
465*c83a76b0SSuyog Pawar             if(ps_tile->u2_wd == (ps_proc->i4_ctb_tile_x + 1))
466*c83a76b0SSuyog Pawar             {
467*c83a76b0SSuyog Pawar                 if((ps_proc->i4_ctb_tile_y + 1) == ps_tile->u2_ht)
468*c83a76b0SSuyog Pawar                 {
469*c83a76b0SSuyog Pawar                     //Last tile
470*c83a76b0SSuyog Pawar                     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))
471*c83a76b0SSuyog Pawar                     {
472*c83a76b0SSuyog Pawar                         next_ctb_tu_idx = (cur_ctb_idx % RESET_TU_BUF_NCTB) + 1;
473*c83a76b0SSuyog Pawar                     }
474*c83a76b0SSuyog Pawar                     else //Not last tile, but new tile
475*c83a76b0SSuyog Pawar                     {
476*c83a76b0SSuyog Pawar                         tile_t *ps_tile_next = ps_proc->ps_tile + 1;
477*c83a76b0SSuyog Pawar                         next_ctb_tu_idx = ps_tile_next->u1_pos_x + (ps_tile_next->u1_pos_y * ps_sps->i2_pic_wd_in_ctb);
478*c83a76b0SSuyog Pawar                     }
479*c83a76b0SSuyog Pawar                 }
480*c83a76b0SSuyog Pawar                 else //End of each tile row
481*c83a76b0SSuyog Pawar                 {
482*c83a76b0SSuyog Pawar                     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;
483*c83a76b0SSuyog Pawar                 }
484*c83a76b0SSuyog Pawar             }
485*c83a76b0SSuyog Pawar             ps_proc->i4_next_tu_ctb_cnt = next_ctb_tu_idx;
486*c83a76b0SSuyog Pawar             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];
487*c83a76b0SSuyog Pawar         }
488*c83a76b0SSuyog Pawar         else
489*c83a76b0SSuyog Pawar         {
490*c83a76b0SSuyog Pawar             next_ctb_tu_idx = cur_ctb_idx + 1;
491*c83a76b0SSuyog Pawar             if(ps_tile->u2_wd == (ps_proc->i4_ctb_tile_x + 1))
492*c83a76b0SSuyog Pawar             {
493*c83a76b0SSuyog Pawar                 if((ps_proc->i4_ctb_tile_y + 1) == ps_tile->u2_ht)
494*c83a76b0SSuyog Pawar                 {
495*c83a76b0SSuyog Pawar                     //Last tile
496*c83a76b0SSuyog Pawar                     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))
497*c83a76b0SSuyog Pawar                     {
498*c83a76b0SSuyog Pawar                         next_ctb_tu_idx = (cur_ctb_idx % RESET_TU_BUF_NCTB) + 1;
499*c83a76b0SSuyog Pawar                     }
500*c83a76b0SSuyog Pawar                     else //Not last tile, but new tile
501*c83a76b0SSuyog Pawar                     {
502*c83a76b0SSuyog Pawar                         tile_t *ps_tile_next = ps_proc->ps_tile + 1;
503*c83a76b0SSuyog Pawar                         next_ctb_tu_idx = ps_tile_next->u1_pos_x + (ps_tile_next->u1_pos_y * ps_sps->i2_pic_wd_in_ctb);
504*c83a76b0SSuyog Pawar                     }
505*c83a76b0SSuyog Pawar                 }
506*c83a76b0SSuyog Pawar                 else //End of each tile row
507*c83a76b0SSuyog Pawar                 {
508*c83a76b0SSuyog Pawar                     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;
509*c83a76b0SSuyog Pawar                 }
510*c83a76b0SSuyog Pawar             }
511*c83a76b0SSuyog Pawar             ps_proc->i4_next_tu_ctb_cnt = next_ctb_tu_idx;
512*c83a76b0SSuyog Pawar             ps_proc->i4_ctb_tu_cnt = ps_proc->pu4_pic_tu_idx[next_ctb_tu_idx] -
513*c83a76b0SSuyog Pawar                             ps_proc->pu4_pic_tu_idx[cur_ctb_idx];
514*c83a76b0SSuyog Pawar         }
515*c83a76b0SSuyog Pawar     }
516*c83a76b0SSuyog Pawar }
517*c83a76b0SSuyog Pawar 
ihevcd_process(process_ctxt_t * ps_proc)518*c83a76b0SSuyog Pawar IHEVCD_ERROR_T ihevcd_process(process_ctxt_t *ps_proc)
519*c83a76b0SSuyog Pawar {
520*c83a76b0SSuyog Pawar     IHEVCD_ERROR_T ret = (IHEVCD_ERROR_T)IHEVCD_SUCCESS;
521*c83a76b0SSuyog Pawar     codec_t *ps_codec;
522*c83a76b0SSuyog Pawar     sps_t *ps_sps = ps_proc->ps_sps;
523*c83a76b0SSuyog Pawar 
524*c83a76b0SSuyog Pawar     WORD32 nctb;
525*c83a76b0SSuyog Pawar     WORD32 i;
526*c83a76b0SSuyog Pawar     WORD32 idx;
527*c83a76b0SSuyog Pawar     WORD32 nop_cnt;
528*c83a76b0SSuyog Pawar     WORD32 num_minpu_in_ctb;
529*c83a76b0SSuyog Pawar     WORD32 cur_slice_idx, cur_ctb_tile_x, cur_ctb_slice_x, cur_ctb_tile_y, cur_ctb_slice_y;
530*c83a76b0SSuyog Pawar     WORD32 nxt_ctb_slice_y, nxt_ctb_slice_x;
531*c83a76b0SSuyog Pawar     tu_t *ps_tu_cur, *ps_tu_nxt;
532*c83a76b0SSuyog Pawar     UWORD8 *pu1_pu_map_cur, *pu1_pu_map_nxt;
533*c83a76b0SSuyog Pawar     WORD32 num_ctb, num_ctb_tmp;
534*c83a76b0SSuyog Pawar     proc_type_t proc_type;
535*c83a76b0SSuyog Pawar 
536*c83a76b0SSuyog Pawar 
537*c83a76b0SSuyog Pawar     WORD32 ctb_size = 1 << ps_sps->i1_log2_ctb_size;
538*c83a76b0SSuyog Pawar 
539*c83a76b0SSuyog Pawar     PROFILE_DISABLE_PROCESS_CTB();
540*c83a76b0SSuyog Pawar 
541*c83a76b0SSuyog Pawar     ps_codec = ps_proc->ps_codec;
542*c83a76b0SSuyog Pawar     num_minpu_in_ctb = (ctb_size / MIN_PU_SIZE) * (ctb_size / MIN_PU_SIZE);
543*c83a76b0SSuyog Pawar 
544*c83a76b0SSuyog Pawar     nctb = MIN(ps_codec->i4_proc_nctb, ps_proc->i4_ctb_cnt);
545*c83a76b0SSuyog Pawar     nctb = MIN(nctb, (ps_proc->ps_tile->u2_wd - ps_proc->i4_ctb_tile_x));
546*c83a76b0SSuyog Pawar 
547*c83a76b0SSuyog Pawar     if(ps_proc->i4_cur_slice_idx > (MAX_SLICE_HDR_CNT - 2 * ps_sps->i2_pic_wd_in_ctb))
548*c83a76b0SSuyog Pawar     {
549*c83a76b0SSuyog Pawar         num_ctb = 1;
550*c83a76b0SSuyog Pawar     }
551*c83a76b0SSuyog Pawar     else
552*c83a76b0SSuyog Pawar     {
553*c83a76b0SSuyog Pawar         num_ctb = ps_proc->i4_nctb;
554*c83a76b0SSuyog Pawar     }
555*c83a76b0SSuyog Pawar     nxt_ctb_slice_y = ps_proc->i4_ctb_slice_y;
556*c83a76b0SSuyog Pawar     nxt_ctb_slice_x = ps_proc->i4_ctb_slice_x;
557*c83a76b0SSuyog Pawar     pu1_pu_map_nxt = ps_proc->pu1_pu_map;
558*c83a76b0SSuyog Pawar     ps_tu_nxt = ps_proc->ps_tu;
559*c83a76b0SSuyog Pawar 
560*c83a76b0SSuyog Pawar     while(ps_proc->i4_ctb_cnt)
561*c83a76b0SSuyog Pawar     {
562*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_slice_y = nxt_ctb_slice_y;
563*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_slice_x = nxt_ctb_slice_x;
564*c83a76b0SSuyog Pawar         ps_proc->pu1_pu_map = pu1_pu_map_nxt;
565*c83a76b0SSuyog Pawar         ps_proc->ps_tu = ps_tu_nxt;
566*c83a76b0SSuyog Pawar 
567*c83a76b0SSuyog Pawar         cur_ctb_tile_x = ps_proc->i4_ctb_tile_x;
568*c83a76b0SSuyog Pawar         cur_ctb_tile_y = ps_proc->i4_ctb_tile_y;
569*c83a76b0SSuyog Pawar         cur_ctb_slice_x = ps_proc->i4_ctb_slice_x;
570*c83a76b0SSuyog Pawar         cur_ctb_slice_y = ps_proc->i4_ctb_slice_y;
571*c83a76b0SSuyog Pawar         cur_slice_idx = ps_proc->i4_cur_slice_idx;
572*c83a76b0SSuyog Pawar         ps_tu_cur = ps_proc->ps_tu;
573*c83a76b0SSuyog Pawar         pu1_pu_map_cur = ps_proc->pu1_pu_map;
574*c83a76b0SSuyog Pawar         proc_type = PROC_INTER_PRED;
575*c83a76b0SSuyog Pawar 
576*c83a76b0SSuyog Pawar         if(ps_proc->i4_ctb_cnt < num_ctb)
577*c83a76b0SSuyog Pawar         {
578*c83a76b0SSuyog Pawar             num_ctb = ps_proc->i4_ctb_cnt;
579*c83a76b0SSuyog Pawar         }
580*c83a76b0SSuyog Pawar         num_ctb_tmp = num_ctb;
581*c83a76b0SSuyog Pawar 
582*c83a76b0SSuyog Pawar         while(num_ctb_tmp)
583*c83a76b0SSuyog Pawar         {
584*c83a76b0SSuyog Pawar             slice_header_t *ps_slice_hdr;
585*c83a76b0SSuyog Pawar             tile_t *ps_tile = ps_proc->ps_tile;
586*c83a76b0SSuyog Pawar 
587*c83a76b0SSuyog Pawar             /* Waiting for Parsing to be done*/
588*c83a76b0SSuyog Pawar             {
589*c83a76b0SSuyog Pawar 
590*c83a76b0SSuyog Pawar 
591*c83a76b0SSuyog Pawar                 nop_cnt = PROC_NOP_CNT;
592*c83a76b0SSuyog Pawar                 if(ps_proc->i4_check_parse_status || ps_proc->i4_check_proc_status)
593*c83a76b0SSuyog Pawar                 {
594*c83a76b0SSuyog Pawar                     while(1)
595*c83a76b0SSuyog Pawar                     {
596*c83a76b0SSuyog Pawar                         volatile UWORD8 *pu1_buf;
597*c83a76b0SSuyog Pawar                         volatile WORD32 status;
598*c83a76b0SSuyog Pawar                         status = 1;
599*c83a76b0SSuyog Pawar                         /* Check if all dependencies for the next nCTBs are met */
600*c83a76b0SSuyog Pawar                         /* Check if the next nCTBs are parsed */
601*c83a76b0SSuyog Pawar                         if(ps_proc->i4_check_parse_status)
602*c83a76b0SSuyog Pawar                         {
603*c83a76b0SSuyog Pawar                             idx = (ps_proc->i4_ctb_x + nctb - 1);
604*c83a76b0SSuyog Pawar                             idx += (ps_proc->i4_ctb_y * ps_sps->i2_pic_wd_in_ctb);
605*c83a76b0SSuyog Pawar                             pu1_buf = (ps_codec->pu1_parse_map + idx);
606*c83a76b0SSuyog Pawar                             status = *pu1_buf;
607*c83a76b0SSuyog Pawar                         }
608*c83a76b0SSuyog Pawar 
609*c83a76b0SSuyog Pawar                         if(status)
610*c83a76b0SSuyog Pawar                             break;
611*c83a76b0SSuyog Pawar 
612*c83a76b0SSuyog Pawar                         /* if dependencies are not met, then wait for few cycles.
613*c83a76b0SSuyog Pawar                          * Even after few iterations, if the dependencies are not met then yield
614*c83a76b0SSuyog Pawar                          */
615*c83a76b0SSuyog Pawar                         if(nop_cnt > 0)
616*c83a76b0SSuyog Pawar                         {
617*c83a76b0SSuyog Pawar                             NOP(128);
618*c83a76b0SSuyog Pawar                             nop_cnt -= 128;
619*c83a76b0SSuyog Pawar                         }
620*c83a76b0SSuyog Pawar                         else
621*c83a76b0SSuyog Pawar                         {
622*c83a76b0SSuyog Pawar                             nop_cnt = PROC_NOP_CNT;
623*c83a76b0SSuyog Pawar                             ithread_yield();
624*c83a76b0SSuyog Pawar                         }
625*c83a76b0SSuyog Pawar                     }
626*c83a76b0SSuyog Pawar                 }
627*c83a76b0SSuyog Pawar             }
628*c83a76b0SSuyog Pawar 
629*c83a76b0SSuyog Pawar             /* Check proc map to ensure dependencies for recon are met */
630*c83a76b0SSuyog Pawar             ihevcd_proc_map_check(ps_proc, proc_type, nctb);
631*c83a76b0SSuyog Pawar 
632*c83a76b0SSuyog Pawar             ihevcd_slice_hdr_update(ps_proc);
633*c83a76b0SSuyog Pawar             ps_slice_hdr = ps_proc->ps_slice_hdr;
634*c83a76b0SSuyog Pawar 
635*c83a76b0SSuyog Pawar             //ihevcd_mv_prediction();
636*c83a76b0SSuyog Pawar             //ihevcd_lvl_unpack();
637*c83a76b0SSuyog Pawar             //ihevcd_inter_iq_it_recon();
638*c83a76b0SSuyog Pawar             //Following does prediction, iq, it and recon on a TU by TU basis for intra TUs
639*c83a76b0SSuyog Pawar             //ihevcd_intra_process();
640*c83a76b0SSuyog Pawar             //ihevcd_ctb_boundary_strength_islice(ps_proc, ctb_size);
641*c83a76b0SSuyog Pawar             //ihevcd_deblk_ctb(ps_proc);
642*c83a76b0SSuyog Pawar 
643*c83a76b0SSuyog Pawar             /* iq,it recon of Intra TU */
644*c83a76b0SSuyog Pawar             {
645*c83a76b0SSuyog Pawar                 UWORD32 *pu4_ctb_top_pu_idx, *pu4_ctb_left_pu_idx, *pu4_ctb_top_left_pu_idx;
646*c83a76b0SSuyog Pawar                 WORD32 cur_ctb_idx;
647*c83a76b0SSuyog Pawar 
648*c83a76b0SSuyog Pawar                 ihevcd_ctb_avail_update(ps_proc);
649*c83a76b0SSuyog Pawar 
650*c83a76b0SSuyog Pawar #if DEBUG_DUMP_FRAME_BUFFERS_INFO
651*c83a76b0SSuyog Pawar                 au1_pic_avail_ctb_flags[ps_proc->i4_ctb_x + ps_proc->i4_ctb_y * ps_sps->i2_pic_wd_in_ctb] =
652*c83a76b0SSuyog Pawar                                 ((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));
653*c83a76b0SSuyog Pawar                 au4_pic_ctb_slice_xy[ps_proc->i4_ctb_x + ps_proc->i4_ctb_y * ps_sps->i2_pic_wd_in_ctb] =
654*c83a76b0SSuyog Pawar                                 (((UWORD16)ps_proc->i4_ctb_slice_x << 16) | ((UWORD16)ps_proc->i4_ctb_slice_y << 16));
655*c83a76b0SSuyog Pawar #endif
656*c83a76b0SSuyog Pawar 
657*c83a76b0SSuyog Pawar                 /*************************************************/
658*c83a76b0SSuyog Pawar                 /****************   MV pred **********************/
659*c83a76b0SSuyog Pawar                 /*************************************************/
660*c83a76b0SSuyog Pawar                 if(PSLICE == ps_slice_hdr->i1_slice_type
661*c83a76b0SSuyog Pawar                                 || BSLICE == ps_slice_hdr->i1_slice_type)
662*c83a76b0SSuyog Pawar                 {
663*c83a76b0SSuyog Pawar                     mv_ctxt_t s_mv_ctxt;
664*c83a76b0SSuyog Pawar 
665*c83a76b0SSuyog Pawar                     pu4_ctb_top_pu_idx = ps_proc->pu4_pic_pu_idx_top
666*c83a76b0SSuyog Pawar                                     + (ps_proc->i4_ctb_x * ctb_size / MIN_PU_SIZE);
667*c83a76b0SSuyog Pawar                     pu4_ctb_left_pu_idx = ps_proc->pu4_pic_pu_idx_left;
668*c83a76b0SSuyog Pawar                     pu4_ctb_top_left_pu_idx = &ps_proc->u4_ctb_top_left_pu_idx;
669*c83a76b0SSuyog Pawar 
670*c83a76b0SSuyog Pawar                     /* Initializing s_mv_ctxt */
671*c83a76b0SSuyog Pawar                     if(ps_codec->i4_num_cores > MV_PRED_NUM_CORES_THRESHOLD)
672*c83a76b0SSuyog Pawar                     {
673*c83a76b0SSuyog Pawar                         s_mv_ctxt.ps_pps = ps_proc->ps_pps;
674*c83a76b0SSuyog Pawar                         s_mv_ctxt.ps_sps = ps_proc->ps_sps;
675*c83a76b0SSuyog Pawar                         s_mv_ctxt.ps_slice_hdr = ps_proc->ps_slice_hdr;
676*c83a76b0SSuyog Pawar                         s_mv_ctxt.i4_ctb_x = ps_proc->i4_ctb_x;
677*c83a76b0SSuyog Pawar                         s_mv_ctxt.i4_ctb_y = ps_proc->i4_ctb_y;
678*c83a76b0SSuyog Pawar                         s_mv_ctxt.ps_pu = ps_proc->ps_pu;
679*c83a76b0SSuyog Pawar                         s_mv_ctxt.ps_pic_pu = ps_proc->ps_pic_pu;
680*c83a76b0SSuyog Pawar                         s_mv_ctxt.ps_tile = ps_tile;
681*c83a76b0SSuyog Pawar                         s_mv_ctxt.pu4_pic_pu_idx_map = ps_proc->pu4_pic_pu_idx_map;
682*c83a76b0SSuyog Pawar                         s_mv_ctxt.pu4_pic_pu_idx = ps_proc->pu4_pic_pu_idx;
683*c83a76b0SSuyog Pawar                         s_mv_ctxt.pu1_pic_pu_map = ps_proc->pu1_pic_pu_map;
684*c83a76b0SSuyog Pawar                         s_mv_ctxt.i4_ctb_pu_cnt = ps_proc->i4_ctb_pu_cnt;
685*c83a76b0SSuyog Pawar                         s_mv_ctxt.i4_ctb_start_pu_idx = ps_proc->i4_ctb_start_pu_idx;
686*c83a76b0SSuyog Pawar                         s_mv_ctxt.u1_top_ctb_avail = ps_proc->u1_top_ctb_avail;
687*c83a76b0SSuyog Pawar                         s_mv_ctxt.u1_top_rt_ctb_avail = ps_proc->u1_top_rt_ctb_avail;
688*c83a76b0SSuyog Pawar                         s_mv_ctxt.u1_top_lt_ctb_avail = ps_proc->u1_top_lt_ctb_avail;
689*c83a76b0SSuyog Pawar                         s_mv_ctxt.u1_left_ctb_avail = ps_proc->u1_left_ctb_avail;
690*c83a76b0SSuyog Pawar 
691*c83a76b0SSuyog Pawar                         ihevcd_get_mv_ctb(&s_mv_ctxt, pu4_ctb_top_pu_idx,
692*c83a76b0SSuyog Pawar                                           pu4_ctb_left_pu_idx, pu4_ctb_top_left_pu_idx);
693*c83a76b0SSuyog Pawar                     }
694*c83a76b0SSuyog Pawar 
695*c83a76b0SSuyog Pawar                     ihevcd_inter_pred_ctb(ps_proc);
696*c83a76b0SSuyog Pawar                 }
697*c83a76b0SSuyog Pawar                 else if(ps_codec->i4_num_cores > MV_PRED_NUM_CORES_THRESHOLD)
698*c83a76b0SSuyog Pawar                 {
699*c83a76b0SSuyog Pawar                     WORD32 next_ctb_idx, num_pu_per_ctb, ctb_start_pu_idx, pu_cnt;
700*c83a76b0SSuyog Pawar                     pu_t *ps_pu;
701*c83a76b0SSuyog Pawar                     WORD32 num_minpu_in_ctb = (ctb_size / MIN_PU_SIZE) * (ctb_size / MIN_PU_SIZE);
702*c83a76b0SSuyog Pawar                     UWORD8 *pu1_pic_pu_map_ctb = ps_proc->pu1_pic_pu_map +
703*c83a76b0SSuyog Pawar                                     (ps_proc->i4_ctb_x + ps_proc->i4_ctb_y * ps_sps->i2_pic_wd_in_ctb) * num_minpu_in_ctb;
704*c83a76b0SSuyog Pawar                     WORD32 row, col;
705*c83a76b0SSuyog Pawar                     UWORD32 *pu4_nbr_pu_idx = ps_proc->pu4_pic_pu_idx_map;
706*c83a76b0SSuyog Pawar                     WORD32 nbr_pu_idx_strd = MAX_CTB_SIZE / MIN_PU_SIZE + 2;
707*c83a76b0SSuyog Pawar                     WORD32 ctb_size_in_min_pu = (ctb_size / MIN_PU_SIZE);
708*c83a76b0SSuyog Pawar 
709*c83a76b0SSuyog Pawar                     /* Neighbor PU idx update inside CTB */
710*c83a76b0SSuyog Pawar                     /* 1byte per 4x4. Indicates the PU idx that 4x4 block belongs to */
711*c83a76b0SSuyog Pawar 
712*c83a76b0SSuyog Pawar                     cur_ctb_idx = ps_proc->i4_ctb_x
713*c83a76b0SSuyog Pawar                                     + ps_proc->i4_ctb_y * (ps_sps->i2_pic_wd_in_ctb);
714*c83a76b0SSuyog Pawar                     next_ctb_idx = ps_proc->i4_next_pu_ctb_cnt;
715*c83a76b0SSuyog Pawar                     num_pu_per_ctb = ps_proc->pu4_pic_pu_idx[next_ctb_idx]
716*c83a76b0SSuyog Pawar                                     - ps_proc->pu4_pic_pu_idx[cur_ctb_idx];
717*c83a76b0SSuyog Pawar                     ctb_start_pu_idx = ps_proc->pu4_pic_pu_idx[cur_ctb_idx];
718*c83a76b0SSuyog Pawar                     ps_pu = &ps_proc->ps_pic_pu[ctb_start_pu_idx];
719*c83a76b0SSuyog Pawar 
720*c83a76b0SSuyog Pawar                     for(pu_cnt = 0; pu_cnt < num_pu_per_ctb; pu_cnt++, ps_pu++)
721*c83a76b0SSuyog Pawar                     {
722*c83a76b0SSuyog Pawar                         UWORD32 cur_pu_idx;
723*c83a76b0SSuyog Pawar                         WORD32 pu_ht = (ps_pu->b4_ht + 1) << 2;
724*c83a76b0SSuyog Pawar                         WORD32 pu_wd = (ps_pu->b4_wd + 1) << 2;
725*c83a76b0SSuyog Pawar 
726*c83a76b0SSuyog Pawar                         cur_pu_idx = ctb_start_pu_idx + pu_cnt;
727*c83a76b0SSuyog Pawar 
728*c83a76b0SSuyog Pawar                         for(row = 0; row < pu_ht / MIN_PU_SIZE; row++)
729*c83a76b0SSuyog Pawar                             for(col = 0; col < pu_wd / MIN_PU_SIZE; col++)
730*c83a76b0SSuyog Pawar                                 pu4_nbr_pu_idx[(1 + ps_pu->b4_pos_x + col)
731*c83a76b0SSuyog Pawar                                                 + (1 + ps_pu->b4_pos_y + row)
732*c83a76b0SSuyog Pawar                                                 * nbr_pu_idx_strd] =
733*c83a76b0SSuyog Pawar                                                 cur_pu_idx;
734*c83a76b0SSuyog Pawar                     }
735*c83a76b0SSuyog Pawar 
736*c83a76b0SSuyog Pawar                     /* Updating Top and Left pointers */
737*c83a76b0SSuyog Pawar                     {
738*c83a76b0SSuyog Pawar                         WORD32 rows_remaining = ps_sps->i2_pic_height_in_luma_samples
739*c83a76b0SSuyog Pawar                                         - (ps_proc->i4_ctb_y << ps_sps->i1_log2_ctb_size);
740*c83a76b0SSuyog Pawar                         WORD32 ctb_size_left = MIN(ctb_size, rows_remaining);
741*c83a76b0SSuyog Pawar 
742*c83a76b0SSuyog Pawar                         /* Top Left */
743*c83a76b0SSuyog Pawar                         /* saving top left before updating top ptr, as updating top ptr will overwrite the top left for the next ctb */
744*c83a76b0SSuyog Pawar                         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];
745*c83a76b0SSuyog Pawar                         for(i = 0; i < ctb_size / MIN_PU_SIZE; i++)
746*c83a76b0SSuyog Pawar                         {
747*c83a76b0SSuyog Pawar                             /* Left */
748*c83a76b0SSuyog Pawar                             /* Last column of au4_nbr_pu_idx */
749*c83a76b0SSuyog Pawar                             ps_proc->pu4_pic_pu_idx_left[i] =
750*c83a76b0SSuyog Pawar                                             pu4_nbr_pu_idx[(ctb_size / MIN_PU_SIZE) + (i + 1) * nbr_pu_idx_strd];
751*c83a76b0SSuyog Pawar                             /* Top */
752*c83a76b0SSuyog Pawar                             /* Last row of au4_nbr_pu_idx */
753*c83a76b0SSuyog Pawar                             ps_proc->pu4_pic_pu_idx_top[(ps_proc->i4_ctb_x * ctb_size / MIN_PU_SIZE) + i] =
754*c83a76b0SSuyog Pawar                                             pu4_nbr_pu_idx[(ctb_size_left / MIN_PU_SIZE) * nbr_pu_idx_strd + i + 1];
755*c83a76b0SSuyog Pawar 
756*c83a76b0SSuyog Pawar                         }
757*c83a76b0SSuyog Pawar 
758*c83a76b0SSuyog Pawar                         /* Updating the CTB level PU idx (Used for collocated MV pred)*/
759*c83a76b0SSuyog Pawar                         {
760*c83a76b0SSuyog Pawar                             WORD32 ctb_row, ctb_col, index_pic_map, index_nbr_map;
761*c83a76b0SSuyog Pawar                             WORD32 first_pu_of_ctb;
762*c83a76b0SSuyog Pawar                             first_pu_of_ctb = pu4_nbr_pu_idx[1 + nbr_pu_idx_strd];
763*c83a76b0SSuyog Pawar                             UWORD32 cur_ctb_ht_in_min_pu = MIN(((ps_sps->i2_pic_height_in_luma_samples
764*c83a76b0SSuyog Pawar                                         - (ps_proc->i4_ctb_y << ps_sps->i1_log2_ctb_size)) / MIN_PU_SIZE), ctb_size_in_min_pu);
765*c83a76b0SSuyog Pawar                             UWORD32 cur_ctb_wd_in_min_pu = MIN(((ps_sps->i2_pic_width_in_luma_samples
766*c83a76b0SSuyog Pawar                                         - (ps_proc->i4_ctb_x << ps_sps->i1_log2_ctb_size)) / MIN_PU_SIZE), ctb_size_in_min_pu);
767*c83a76b0SSuyog Pawar 
768*c83a76b0SSuyog Pawar                             index_pic_map = 0 * ctb_size_in_min_pu + 0;
769*c83a76b0SSuyog Pawar                             index_nbr_map = (0 + 1) * nbr_pu_idx_strd + (0 + 1);
770*c83a76b0SSuyog Pawar 
771*c83a76b0SSuyog Pawar                             for(ctb_row = 0; ctb_row < cur_ctb_ht_in_min_pu; ctb_row++)
772*c83a76b0SSuyog Pawar                             {
773*c83a76b0SSuyog Pawar                                 for(ctb_col = 0; ctb_col < cur_ctb_wd_in_min_pu; ctb_col++)
774*c83a76b0SSuyog Pawar                                 {
775*c83a76b0SSuyog Pawar                                     pu1_pic_pu_map_ctb[index_pic_map + ctb_col] = pu4_nbr_pu_idx[index_nbr_map + ctb_col]
776*c83a76b0SSuyog Pawar                                                     - first_pu_of_ctb;
777*c83a76b0SSuyog Pawar                                 }
778*c83a76b0SSuyog Pawar                                 index_pic_map += ctb_size_in_min_pu;
779*c83a76b0SSuyog Pawar                                 index_nbr_map += nbr_pu_idx_strd;
780*c83a76b0SSuyog Pawar                             }
781*c83a76b0SSuyog Pawar                         }
782*c83a76b0SSuyog Pawar                     }
783*c83a76b0SSuyog Pawar                 }
784*c83a76b0SSuyog Pawar             }
785*c83a76b0SSuyog Pawar 
786*c83a76b0SSuyog Pawar             if(ps_proc->ps_pps->i1_tiles_enabled_flag)
787*c83a76b0SSuyog Pawar             {
788*c83a76b0SSuyog Pawar                 /*Update the tile index buffer with tile information for the current ctb*/
789*c83a76b0SSuyog Pawar                 UWORD16 *pu1_tile_idx = ps_proc->pu1_tile_idx;
790*c83a76b0SSuyog Pawar                 pu1_tile_idx[(ps_proc->i4_ctb_x + (ps_proc->i4_ctb_y * ps_sps->i2_pic_wd_in_ctb))]
791*c83a76b0SSuyog Pawar                                 = ps_proc->i4_cur_tile_idx;
792*c83a76b0SSuyog Pawar             }
793*c83a76b0SSuyog Pawar 
794*c83a76b0SSuyog Pawar             /*************************************************/
795*c83a76b0SSuyog Pawar             /*********** BS, QP and Deblocking  **************/
796*c83a76b0SSuyog Pawar             /*************************************************/
797*c83a76b0SSuyog Pawar             /* Boundary strength call has to be after IQ IT recon since QP population needs ps_proc->i4_qp_const_inc_ctb flag */
798*c83a76b0SSuyog Pawar 
799*c83a76b0SSuyog Pawar             {
800*c83a76b0SSuyog Pawar                 slice_header_t *ps_slice_hdr;
801*c83a76b0SSuyog Pawar                 ps_slice_hdr = ps_proc->ps_slice_hdr;
802*c83a76b0SSuyog Pawar 
803*c83a76b0SSuyog Pawar 
804*c83a76b0SSuyog Pawar                 /* Check if deblock is disabled for the current slice or if it is disabled for the current picture
805*c83a76b0SSuyog Pawar                  * because of disable deblock api
806*c83a76b0SSuyog Pawar                  */
807*c83a76b0SSuyog Pawar                 if(0 == ps_codec->i4_disable_deblk_pic)
808*c83a76b0SSuyog Pawar                 {
809*c83a76b0SSuyog Pawar                     if(ps_codec->i4_num_cores > MV_PRED_NUM_CORES_THRESHOLD)
810*c83a76b0SSuyog Pawar                     {
811*c83a76b0SSuyog Pawar                         /* Boundary strength calculation is done irrespective of whether deblocking is disabled
812*c83a76b0SSuyog Pawar                          * in the slice or not, to handle deblocking slice boundaries */
813*c83a76b0SSuyog Pawar                         if((0 == ps_codec->i4_slice_error))
814*c83a76b0SSuyog Pawar                         {
815*c83a76b0SSuyog Pawar                             ihevcd_update_ctb_tu_cnt(ps_proc);
816*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.ps_pps = ps_proc->ps_pps;
817*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.ps_sps = ps_proc->ps_sps;
818*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.ps_codec = ps_proc->ps_codec;
819*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.i4_ctb_tu_cnt = ps_proc->i4_ctb_tu_cnt;
820*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.i4_ctb_x = ps_proc->i4_ctb_x;
821*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.i4_ctb_y = ps_proc->i4_ctb_y;
822*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.i4_ctb_tile_x = ps_proc->i4_ctb_tile_x;
823*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.i4_ctb_tile_y = ps_proc->i4_ctb_tile_y;
824*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.i4_ctb_slice_x = ps_proc->i4_ctb_slice_x;
825*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.i4_ctb_slice_y = ps_proc->i4_ctb_slice_y;
826*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.ps_tu = ps_proc->ps_tu;
827*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.ps_pu = ps_proc->ps_pu;
828*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.pu4_pic_pu_idx_map = ps_proc->pu4_pic_pu_idx_map;
829*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.i4_next_pu_ctb_cnt = ps_proc->i4_next_pu_ctb_cnt;
830*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.i4_next_tu_ctb_cnt = ps_proc->i4_next_tu_ctb_cnt;
831*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.pu1_slice_idx = ps_proc->pu1_slice_idx;
832*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.ps_slice_hdr = ps_proc->ps_slice_hdr;
833*c83a76b0SSuyog Pawar                             ps_proc->s_bs_ctxt.ps_tile = ps_proc->ps_tile;
834*c83a76b0SSuyog Pawar 
835*c83a76b0SSuyog Pawar                             if(ISLICE == ps_slice_hdr->i1_slice_type)
836*c83a76b0SSuyog Pawar                             {
837*c83a76b0SSuyog Pawar                                 ihevcd_ctb_boundary_strength_islice(&ps_proc->s_bs_ctxt);
838*c83a76b0SSuyog Pawar                             }
839*c83a76b0SSuyog Pawar                             else
840*c83a76b0SSuyog Pawar                             {
841*c83a76b0SSuyog Pawar                                 ihevcd_ctb_boundary_strength_pbslice(&ps_proc->s_bs_ctxt);
842*c83a76b0SSuyog Pawar                             }
843*c83a76b0SSuyog Pawar                         }
844*c83a76b0SSuyog Pawar 
845*c83a76b0SSuyog Pawar                         /* Boundary strength is set to zero if deblocking is disabled for the current slice */
846*c83a76b0SSuyog Pawar                         if((0 != ps_slice_hdr->i1_slice_disable_deblocking_filter_flag))
847*c83a76b0SSuyog Pawar                         {
848*c83a76b0SSuyog Pawar                             WORD32 bs_strd = (ps_sps->i2_pic_wd_in_ctb + 1) * (ctb_size * ctb_size / 8 / 16);
849*c83a76b0SSuyog Pawar 
850*c83a76b0SSuyog Pawar                             UWORD32 *pu4_vert_bs = (UWORD32 *)((UWORD8 *)ps_proc->s_bs_ctxt.pu4_pic_vert_bs +
851*c83a76b0SSuyog Pawar                                             ps_proc->i4_ctb_x * (ctb_size * ctb_size / 8 / 16) +
852*c83a76b0SSuyog Pawar                                             ps_proc->i4_ctb_y * bs_strd);
853*c83a76b0SSuyog Pawar                             UWORD32 *pu4_horz_bs = (UWORD32 *)((UWORD8 *)ps_proc->s_bs_ctxt.pu4_pic_horz_bs +
854*c83a76b0SSuyog Pawar                                             ps_proc->i4_ctb_x * (ctb_size * ctb_size / 8 / 16) +
855*c83a76b0SSuyog Pawar                                             ps_proc->i4_ctb_y * bs_strd);
856*c83a76b0SSuyog Pawar 
857*c83a76b0SSuyog Pawar                             memset(pu4_vert_bs, 0, (ctb_size / 8) * (ctb_size / 4) / 8 * 2);
858*c83a76b0SSuyog Pawar                             memset(pu4_horz_bs, 0, (ctb_size / 8) * (ctb_size / 4) / 8 * 2);
859*c83a76b0SSuyog Pawar                         }
860*c83a76b0SSuyog Pawar                     }
861*c83a76b0SSuyog Pawar                 }
862*c83a76b0SSuyog Pawar             }
863*c83a76b0SSuyog Pawar 
864*c83a76b0SSuyog Pawar             /* Per CTB update the following */
865*c83a76b0SSuyog Pawar             {
866*c83a76b0SSuyog Pawar                 WORD32 cur_ctb_idx = ps_proc->i4_ctb_x
867*c83a76b0SSuyog Pawar                                 + ps_proc->i4_ctb_y * (ps_sps->i2_pic_wd_in_ctb);
868*c83a76b0SSuyog Pawar                 cur_ctb_idx++;
869*c83a76b0SSuyog Pawar 
870*c83a76b0SSuyog Pawar                 ps_proc->pu1_pu_map += nctb * num_minpu_in_ctb;
871*c83a76b0SSuyog Pawar                 ps_proc->ps_tu += ps_proc->i4_ctb_tu_cnt;
872*c83a76b0SSuyog Pawar                 if((1 == ps_codec->i4_num_cores) &&
873*c83a76b0SSuyog Pawar                                 (0 == cur_ctb_idx % RESET_TU_BUF_NCTB))
874*c83a76b0SSuyog Pawar                 {
875*c83a76b0SSuyog Pawar                     ps_proc->ps_tu = ps_proc->ps_pic_tu;
876*c83a76b0SSuyog Pawar                 }
877*c83a76b0SSuyog Pawar                 ps_proc->ps_pu += ps_proc->i4_ctb_pu_cnt;
878*c83a76b0SSuyog Pawar             }
879*c83a76b0SSuyog Pawar 
880*c83a76b0SSuyog Pawar             /* Update proc map for recon*/
881*c83a76b0SSuyog Pawar             ihevcd_proc_map_update(ps_proc, proc_type, nctb);
882*c83a76b0SSuyog Pawar 
883*c83a76b0SSuyog Pawar             num_ctb_tmp -= nctb;
884*c83a76b0SSuyog Pawar             ihevcd_ctb_pos_update(ps_proc, nctb);
885*c83a76b0SSuyog Pawar 
886*c83a76b0SSuyog Pawar         }
887*c83a76b0SSuyog Pawar 
888*c83a76b0SSuyog Pawar         if(cur_slice_idx != ps_proc->i4_cur_slice_idx)
889*c83a76b0SSuyog Pawar         {
890*c83a76b0SSuyog Pawar             ps_proc->ps_slice_hdr = ps_codec->ps_slice_hdr_base + ((cur_slice_idx)&(MAX_SLICE_HDR_CNT - 1));
891*c83a76b0SSuyog Pawar             ps_proc->i4_cur_slice_idx = cur_slice_idx;
892*c83a76b0SSuyog Pawar         }
893*c83a76b0SSuyog Pawar         /* Restore the saved variables  */
894*c83a76b0SSuyog Pawar         num_ctb_tmp = num_ctb;
895*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_x -= num_ctb;
896*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_tile_x = cur_ctb_tile_x;
897*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_slice_x = cur_ctb_slice_x;
898*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_tile_y = cur_ctb_tile_y;
899*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_slice_y = cur_ctb_slice_y;
900*c83a76b0SSuyog Pawar         ps_proc->pu1_pu_map = pu1_pu_map_cur;
901*c83a76b0SSuyog Pawar         ps_proc->ps_tu = ps_tu_cur;
902*c83a76b0SSuyog Pawar         proc_type = PROC_RECON;
903*c83a76b0SSuyog Pawar 
904*c83a76b0SSuyog Pawar         while(num_ctb_tmp)
905*c83a76b0SSuyog Pawar         {
906*c83a76b0SSuyog Pawar 
907*c83a76b0SSuyog Pawar             /* Check proc map to ensure dependencies for recon are met */
908*c83a76b0SSuyog Pawar             ihevcd_proc_map_check(ps_proc, proc_type, nctb);
909*c83a76b0SSuyog Pawar 
910*c83a76b0SSuyog Pawar             ihevcd_slice_hdr_update(ps_proc);
911*c83a76b0SSuyog Pawar 
912*c83a76b0SSuyog Pawar             {
913*c83a76b0SSuyog Pawar 
914*c83a76b0SSuyog Pawar                 ihevcd_ctb_avail_update(ps_proc);
915*c83a76b0SSuyog Pawar 
916*c83a76b0SSuyog Pawar                 /*************************************************/
917*c83a76b0SSuyog Pawar                 /**************** IQ IT RECON  *******************/
918*c83a76b0SSuyog Pawar                 /*************************************************/
919*c83a76b0SSuyog Pawar 
920*c83a76b0SSuyog Pawar                 ihevcd_update_ctb_tu_cnt(ps_proc);
921*c83a76b0SSuyog Pawar 
922*c83a76b0SSuyog Pawar                 /* When scaling matrix is not to be used(scaling_list_enable_flag is zero in SPS),
923*c83a76b0SSuyog Pawar                  * default value of 16 has to be used. Since the value is same for all sizes,
924*c83a76b0SSuyog Pawar                  * same table is used for all cases.
925*c83a76b0SSuyog Pawar                  */
926*c83a76b0SSuyog Pawar                 if(0 == ps_sps->i1_scaling_list_enable_flag)
927*c83a76b0SSuyog Pawar                 {
928*c83a76b0SSuyog Pawar                     ps_proc->api2_dequant_intra_matrix[0] =
929*c83a76b0SSuyog Pawar                                     (WORD16 *)gi2_flat_scale_mat_32x32;
930*c83a76b0SSuyog Pawar                     ps_proc->api2_dequant_intra_matrix[1] =
931*c83a76b0SSuyog Pawar                                     (WORD16 *)gi2_flat_scale_mat_32x32;
932*c83a76b0SSuyog Pawar                     ps_proc->api2_dequant_intra_matrix[2] =
933*c83a76b0SSuyog Pawar                                     (WORD16 *)gi2_flat_scale_mat_32x32;
934*c83a76b0SSuyog Pawar                     ps_proc->api2_dequant_intra_matrix[3] =
935*c83a76b0SSuyog Pawar                                     (WORD16 *)gi2_flat_scale_mat_32x32;
936*c83a76b0SSuyog Pawar 
937*c83a76b0SSuyog Pawar                     ps_proc->api2_dequant_inter_matrix[0] =
938*c83a76b0SSuyog Pawar                                     (WORD16 *)gi2_flat_scale_mat_32x32;
939*c83a76b0SSuyog Pawar                     ps_proc->api2_dequant_inter_matrix[1] =
940*c83a76b0SSuyog Pawar                                     (WORD16 *)gi2_flat_scale_mat_32x32;
941*c83a76b0SSuyog Pawar                     ps_proc->api2_dequant_inter_matrix[2] =
942*c83a76b0SSuyog Pawar                                     (WORD16 *)gi2_flat_scale_mat_32x32;
943*c83a76b0SSuyog Pawar                     ps_proc->api2_dequant_inter_matrix[3] =
944*c83a76b0SSuyog Pawar                                     (WORD16 *)gi2_flat_scale_mat_32x32;
945*c83a76b0SSuyog Pawar                 }
946*c83a76b0SSuyog Pawar                 else
947*c83a76b0SSuyog Pawar                 {
948*c83a76b0SSuyog Pawar                     if(0 == ps_sps->i1_sps_scaling_list_data_present_flag)
949*c83a76b0SSuyog Pawar                     {
950*c83a76b0SSuyog Pawar                         ps_proc->api2_dequant_intra_matrix[0] =
951*c83a76b0SSuyog Pawar                                         (WORD16 *)gi2_flat_scale_mat_32x32;
952*c83a76b0SSuyog Pawar                         ps_proc->api2_dequant_intra_matrix[1] =
953*c83a76b0SSuyog Pawar                                         (WORD16 *)gi2_intra_default_scale_mat_8x8;
954*c83a76b0SSuyog Pawar                         ps_proc->api2_dequant_intra_matrix[2] =
955*c83a76b0SSuyog Pawar                                         (WORD16 *)gi2_intra_default_scale_mat_16x16;
956*c83a76b0SSuyog Pawar                         ps_proc->api2_dequant_intra_matrix[3] =
957*c83a76b0SSuyog Pawar                                         (WORD16 *)gi2_intra_default_scale_mat_32x32;
958*c83a76b0SSuyog Pawar 
959*c83a76b0SSuyog Pawar                         ps_proc->api2_dequant_inter_matrix[0] =
960*c83a76b0SSuyog Pawar                                         (WORD16 *)gi2_flat_scale_mat_32x32;
961*c83a76b0SSuyog Pawar                         ps_proc->api2_dequant_inter_matrix[1] =
962*c83a76b0SSuyog Pawar                                         (WORD16 *)gi2_inter_default_scale_mat_8x8;
963*c83a76b0SSuyog Pawar                         ps_proc->api2_dequant_inter_matrix[2] =
964*c83a76b0SSuyog Pawar                                         (WORD16 *)gi2_inter_default_scale_mat_16x16;
965*c83a76b0SSuyog Pawar                         ps_proc->api2_dequant_inter_matrix[3] =
966*c83a76b0SSuyog Pawar                                         (WORD16 *)gi2_inter_default_scale_mat_32x32;
967*c83a76b0SSuyog Pawar                     }
968*c83a76b0SSuyog Pawar                     /*TODO: Add support for custom scaling matrices */
969*c83a76b0SSuyog Pawar                 }
970*c83a76b0SSuyog Pawar 
971*c83a76b0SSuyog Pawar 
972*c83a76b0SSuyog Pawar                 /* CTB Level pointers */
973*c83a76b0SSuyog Pawar                 ps_proc->pu1_cur_ctb_luma = ps_proc->pu1_cur_pic_luma
974*c83a76b0SSuyog Pawar                                 + (ps_proc->i4_ctb_x * ctb_size
975*c83a76b0SSuyog Pawar                                 + ps_proc->i4_ctb_y * ctb_size
976*c83a76b0SSuyog Pawar                                 * ps_codec->i4_strd);
977*c83a76b0SSuyog Pawar                 ps_proc->pu1_cur_ctb_chroma = ps_proc->pu1_cur_pic_chroma
978*c83a76b0SSuyog Pawar                                 + ps_proc->i4_ctb_x * ctb_size
979*c83a76b0SSuyog Pawar                                 + (ps_proc->i4_ctb_y * ctb_size * ps_codec->i4_strd / 2);
980*c83a76b0SSuyog Pawar 
981*c83a76b0SSuyog Pawar                 ihevcd_iquant_itrans_recon_ctb(ps_proc);
982*c83a76b0SSuyog Pawar             }
983*c83a76b0SSuyog Pawar 
984*c83a76b0SSuyog Pawar             /* Per CTB update the following */
985*c83a76b0SSuyog Pawar             {
986*c83a76b0SSuyog Pawar                 WORD32 cur_ctb_idx = ps_proc->i4_ctb_x
987*c83a76b0SSuyog Pawar                                 + ps_proc->i4_ctb_y * (ps_sps->i2_pic_wd_in_ctb);
988*c83a76b0SSuyog Pawar                 cur_ctb_idx++;
989*c83a76b0SSuyog Pawar 
990*c83a76b0SSuyog Pawar                 ps_proc->pu1_pu_map += nctb * num_minpu_in_ctb;
991*c83a76b0SSuyog Pawar                 ps_proc->ps_tu += ps_proc->i4_ctb_tu_cnt;
992*c83a76b0SSuyog Pawar                 if((1 == ps_codec->i4_num_cores) &&
993*c83a76b0SSuyog Pawar                                 (0 == cur_ctb_idx % RESET_TU_BUF_NCTB))
994*c83a76b0SSuyog Pawar                 {
995*c83a76b0SSuyog Pawar                     ps_proc->ps_tu = ps_proc->ps_pic_tu;
996*c83a76b0SSuyog Pawar                 }
997*c83a76b0SSuyog Pawar                 ps_proc->ps_pu += ps_proc->i4_ctb_pu_cnt;
998*c83a76b0SSuyog Pawar             }
999*c83a76b0SSuyog Pawar 
1000*c83a76b0SSuyog Pawar 
1001*c83a76b0SSuyog Pawar             /* Update proc map for recon*/
1002*c83a76b0SSuyog Pawar             ihevcd_proc_map_update(ps_proc, proc_type, nctb);
1003*c83a76b0SSuyog Pawar 
1004*c83a76b0SSuyog Pawar             num_ctb_tmp -= nctb;
1005*c83a76b0SSuyog Pawar             ihevcd_ctb_pos_update(ps_proc, nctb);
1006*c83a76b0SSuyog Pawar         }
1007*c83a76b0SSuyog Pawar 
1008*c83a76b0SSuyog Pawar         if(cur_slice_idx != ps_proc->i4_cur_slice_idx)
1009*c83a76b0SSuyog Pawar         {
1010*c83a76b0SSuyog Pawar             ps_proc->ps_slice_hdr = ps_codec->ps_slice_hdr_base + ((cur_slice_idx)&(MAX_SLICE_HDR_CNT - 1));
1011*c83a76b0SSuyog Pawar             ps_proc->i4_cur_slice_idx = cur_slice_idx;
1012*c83a76b0SSuyog Pawar         }
1013*c83a76b0SSuyog Pawar         /* Restore the saved variables  */
1014*c83a76b0SSuyog Pawar         num_ctb_tmp = num_ctb;
1015*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_x -= num_ctb;
1016*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_tile_x = cur_ctb_tile_x;
1017*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_slice_x = cur_ctb_slice_x;
1018*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_tile_y = cur_ctb_tile_y;
1019*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_slice_y = cur_ctb_slice_y;
1020*c83a76b0SSuyog Pawar         pu1_pu_map_nxt = ps_proc->pu1_pu_map;
1021*c83a76b0SSuyog Pawar         ps_tu_nxt = ps_proc->ps_tu;
1022*c83a76b0SSuyog Pawar         ps_proc->pu1_pu_map = pu1_pu_map_cur;
1023*c83a76b0SSuyog Pawar         ps_proc->ps_tu = ps_tu_cur;
1024*c83a76b0SSuyog Pawar         proc_type = PROC_DEBLK;
1025*c83a76b0SSuyog Pawar 
1026*c83a76b0SSuyog Pawar         while(num_ctb_tmp)
1027*c83a76b0SSuyog Pawar         {
1028*c83a76b0SSuyog Pawar 
1029*c83a76b0SSuyog Pawar 
1030*c83a76b0SSuyog Pawar             /* Check proc map to ensure dependencies for deblk are met */
1031*c83a76b0SSuyog Pawar             ihevcd_proc_map_check(ps_proc, proc_type, nctb);
1032*c83a76b0SSuyog Pawar 
1033*c83a76b0SSuyog Pawar             ihevcd_slice_hdr_update(ps_proc);
1034*c83a76b0SSuyog Pawar 
1035*c83a76b0SSuyog Pawar 
1036*c83a76b0SSuyog Pawar             if(((0 == FRAME_ILF_PAD || ps_codec->i4_num_cores != 1)) &&
1037*c83a76b0SSuyog Pawar                (0 == ps_codec->i4_disable_deblk_pic))
1038*c83a76b0SSuyog Pawar             {
1039*c83a76b0SSuyog Pawar                 WORD32 i4_is_last_ctb_x = 0;
1040*c83a76b0SSuyog Pawar                 WORD32 i4_is_last_ctb_y = 0;
1041*c83a76b0SSuyog Pawar 
1042*c83a76b0SSuyog Pawar 
1043*c83a76b0SSuyog Pawar                 /* Deblocking is done irrespective of whether it is disabled in the slice or not,
1044*c83a76b0SSuyog Pawar                  * to handle deblocking the slice boundaries */
1045*c83a76b0SSuyog Pawar                 {
1046*c83a76b0SSuyog Pawar                     ps_proc->s_deblk_ctxt.ps_pps = ps_proc->ps_pps;
1047*c83a76b0SSuyog Pawar                     ps_proc->s_deblk_ctxt.ps_sps = ps_proc->ps_sps;
1048*c83a76b0SSuyog Pawar                     ps_proc->s_deblk_ctxt.ps_codec = ps_proc->ps_codec;
1049*c83a76b0SSuyog Pawar                     ps_proc->s_deblk_ctxt.ps_slice_hdr = ps_proc->ps_slice_hdr;
1050*c83a76b0SSuyog Pawar                     ps_proc->s_deblk_ctxt.i4_ctb_x = ps_proc->i4_ctb_x;
1051*c83a76b0SSuyog Pawar                     ps_proc->s_deblk_ctxt.i4_ctb_y = ps_proc->i4_ctb_y;
1052*c83a76b0SSuyog Pawar                     ps_proc->s_deblk_ctxt.pu1_slice_idx = ps_proc->pu1_slice_idx;
1053*c83a76b0SSuyog Pawar                     ps_proc->s_deblk_ctxt.is_chroma_yuv420sp_vu = (ps_codec->e_ref_chroma_fmt == IV_YUV_420SP_VU);
1054*c83a76b0SSuyog Pawar 
1055*c83a76b0SSuyog Pawar                     /* Populating Current CTB's no_loop_filter flags */
1056*c83a76b0SSuyog Pawar                     {
1057*c83a76b0SSuyog Pawar                         WORD32 row;
1058*c83a76b0SSuyog Pawar                         WORD32 log2_ctb_size = ps_sps->i1_log2_ctb_size;
1059*c83a76b0SSuyog Pawar 
1060*c83a76b0SSuyog Pawar                         /* Loop filter strd in units of num bits */
1061*c83a76b0SSuyog Pawar                         WORD32 loop_filter_strd = ((ps_sps->i2_pic_width_in_luma_samples + 63) >> 6) << 3;
1062*c83a76b0SSuyog Pawar                         /* Bit position is the current 8x8 bit offset wrt pic_no_loop_filter
1063*c83a76b0SSuyog Pawar                          * bit_pos has to be a WOR32 so that when it is negative, the downshift still retains it to be a negative value */
1064*c83a76b0SSuyog Pawar                         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;
1065*c83a76b0SSuyog Pawar 
1066*c83a76b0SSuyog Pawar                         for(row = 0; row < (ctb_size >> 3) + 1; row++)
1067*c83a76b0SSuyog Pawar                         {
1068*c83a76b0SSuyog Pawar                             /* Go to the corresponding byte - read 32 bits and downshift */
1069*c83a76b0SSuyog Pawar                             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);
1070*c83a76b0SSuyog Pawar                             bit_pos += loop_filter_strd;
1071*c83a76b0SSuyog Pawar                         }
1072*c83a76b0SSuyog Pawar                     }
1073*c83a76b0SSuyog Pawar 
1074*c83a76b0SSuyog Pawar                     ihevcd_deblk_ctb(&ps_proc->s_deblk_ctxt, i4_is_last_ctb_x, i4_is_last_ctb_y);
1075*c83a76b0SSuyog Pawar 
1076*c83a76b0SSuyog Pawar                     /* If the last CTB in the row was a complete CTB then deblocking has to be called from remaining pixels, since deblocking
1077*c83a76b0SSuyog Pawar                      * is applied on a shifted CTB structure
1078*c83a76b0SSuyog Pawar                      */
1079*c83a76b0SSuyog Pawar                     if(ps_proc->i4_ctb_x == ps_sps->i2_pic_wd_in_ctb - 1)
1080*c83a76b0SSuyog Pawar                     {
1081*c83a76b0SSuyog Pawar                         WORD32 i4_is_last_ctb_x = 1;
1082*c83a76b0SSuyog Pawar                         WORD32 i4_is_last_ctb_y = 0;
1083*c83a76b0SSuyog Pawar 
1084*c83a76b0SSuyog Pawar                         WORD32 last_x_pos;
1085*c83a76b0SSuyog Pawar                         last_x_pos = (ps_sps->i2_pic_wd_in_ctb << ps_sps->i1_log2_ctb_size);
1086*c83a76b0SSuyog Pawar                         if(last_x_pos  ==  ps_sps->i2_pic_width_in_luma_samples)
1087*c83a76b0SSuyog Pawar                         {
1088*c83a76b0SSuyog Pawar                             ihevcd_deblk_ctb(&ps_proc->s_deblk_ctxt, i4_is_last_ctb_x, i4_is_last_ctb_y);
1089*c83a76b0SSuyog Pawar                         }
1090*c83a76b0SSuyog Pawar                     }
1091*c83a76b0SSuyog Pawar 
1092*c83a76b0SSuyog Pawar 
1093*c83a76b0SSuyog Pawar                     /* If the last CTB in the column was a complete CTB then deblocking has to be called from remaining pixels, since deblocking
1094*c83a76b0SSuyog Pawar                      * is applied on a shifted CTB structure
1095*c83a76b0SSuyog Pawar                      */
1096*c83a76b0SSuyog Pawar                     if(ps_proc->i4_ctb_y == ps_sps->i2_pic_ht_in_ctb - 1)
1097*c83a76b0SSuyog Pawar                     {
1098*c83a76b0SSuyog Pawar                         WORD32 i4_is_last_ctb_x = 0;
1099*c83a76b0SSuyog Pawar                         WORD32 i4_is_last_ctb_y = 1;
1100*c83a76b0SSuyog Pawar                         WORD32 last_y_pos;
1101*c83a76b0SSuyog Pawar                         last_y_pos = (ps_sps->i2_pic_ht_in_ctb << ps_sps->i1_log2_ctb_size);
1102*c83a76b0SSuyog Pawar                         if(last_y_pos == ps_sps->i2_pic_height_in_luma_samples)
1103*c83a76b0SSuyog Pawar                         {
1104*c83a76b0SSuyog Pawar                             ihevcd_deblk_ctb(&ps_proc->s_deblk_ctxt, i4_is_last_ctb_x, i4_is_last_ctb_y);
1105*c83a76b0SSuyog Pawar                         }
1106*c83a76b0SSuyog Pawar                     }
1107*c83a76b0SSuyog Pawar                 }
1108*c83a76b0SSuyog Pawar             }
1109*c83a76b0SSuyog Pawar 
1110*c83a76b0SSuyog Pawar             /* Update proc map for deblk*/
1111*c83a76b0SSuyog Pawar             ihevcd_proc_map_update(ps_proc, proc_type, nctb);
1112*c83a76b0SSuyog Pawar 
1113*c83a76b0SSuyog Pawar             num_ctb_tmp -= nctb;
1114*c83a76b0SSuyog Pawar             ihevcd_ctb_pos_update(ps_proc, nctb);
1115*c83a76b0SSuyog Pawar         }
1116*c83a76b0SSuyog Pawar 
1117*c83a76b0SSuyog Pawar         if(cur_slice_idx != ps_proc->i4_cur_slice_idx)
1118*c83a76b0SSuyog Pawar         {
1119*c83a76b0SSuyog Pawar             ps_proc->ps_slice_hdr = ps_codec->ps_slice_hdr_base + ((cur_slice_idx)&(MAX_SLICE_HDR_CNT - 1));
1120*c83a76b0SSuyog Pawar             ps_proc->i4_cur_slice_idx = cur_slice_idx;
1121*c83a76b0SSuyog Pawar         }
1122*c83a76b0SSuyog Pawar         /* Restore the saved variables  */
1123*c83a76b0SSuyog Pawar         num_ctb_tmp = num_ctb;
1124*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_x -= num_ctb;
1125*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_tile_x = cur_ctb_tile_x;
1126*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_tile_y = cur_ctb_tile_y;
1127*c83a76b0SSuyog Pawar         ps_proc->pu1_pu_map = pu1_pu_map_cur;
1128*c83a76b0SSuyog Pawar         ps_proc->ps_tu = ps_tu_cur;
1129*c83a76b0SSuyog Pawar         nxt_ctb_slice_y = ps_proc->i4_ctb_slice_y;
1130*c83a76b0SSuyog Pawar         nxt_ctb_slice_x = ps_proc->i4_ctb_slice_x;
1131*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_slice_y = cur_ctb_slice_y;
1132*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_slice_x = cur_ctb_slice_x;
1133*c83a76b0SSuyog Pawar         proc_type = PROC_SAO;
1134*c83a76b0SSuyog Pawar 
1135*c83a76b0SSuyog Pawar         while(num_ctb_tmp)
1136*c83a76b0SSuyog Pawar         {
1137*c83a76b0SSuyog Pawar 
1138*c83a76b0SSuyog Pawar 
1139*c83a76b0SSuyog Pawar             /* Check proc map to ensure dependencies for SAO are met */
1140*c83a76b0SSuyog Pawar             ihevcd_proc_map_check(ps_proc, proc_type, nctb);
1141*c83a76b0SSuyog Pawar 
1142*c83a76b0SSuyog Pawar             ihevcd_slice_hdr_update(ps_proc);
1143*c83a76b0SSuyog Pawar 
1144*c83a76b0SSuyog Pawar 
1145*c83a76b0SSuyog Pawar             if(0 == FRAME_ILF_PAD || ps_codec->i4_num_cores != 1)
1146*c83a76b0SSuyog Pawar             {
1147*c83a76b0SSuyog Pawar                 /* SAO is done even when it is disabled in the current slice, because
1148*c83a76b0SSuyog Pawar                  * it is performed on a shifted CTB and the neighbor CTBs can belong
1149*c83a76b0SSuyog Pawar                  * to different slices with SAO enabled */
1150*c83a76b0SSuyog Pawar                 if(0 == ps_codec->i4_disable_sao_pic)
1151*c83a76b0SSuyog Pawar                 {
1152*c83a76b0SSuyog Pawar                     ps_proc->s_sao_ctxt.ps_pps = ps_proc->ps_pps;
1153*c83a76b0SSuyog Pawar                     ps_proc->s_sao_ctxt.ps_sps = ps_proc->ps_sps;
1154*c83a76b0SSuyog Pawar                     ps_proc->s_sao_ctxt.ps_tile = ps_proc->ps_tile;
1155*c83a76b0SSuyog Pawar                     ps_proc->s_sao_ctxt.ps_codec = ps_proc->ps_codec;
1156*c83a76b0SSuyog Pawar                     ps_proc->s_sao_ctxt.ps_slice_hdr = ps_proc->ps_slice_hdr;
1157*c83a76b0SSuyog Pawar                     ps_proc->s_sao_ctxt.i4_cur_slice_idx = ps_proc->i4_cur_slice_idx;
1158*c83a76b0SSuyog Pawar 
1159*c83a76b0SSuyog Pawar 
1160*c83a76b0SSuyog Pawar #if SAO_PROCESS_SHIFT_CTB
1161*c83a76b0SSuyog Pawar                     ps_proc->s_sao_ctxt.i4_ctb_x = ps_proc->i4_ctb_x;
1162*c83a76b0SSuyog Pawar                     ps_proc->s_sao_ctxt.i4_ctb_y = ps_proc->i4_ctb_y;
1163*c83a76b0SSuyog Pawar                     ps_proc->s_sao_ctxt.is_chroma_yuv420sp_vu = (ps_codec->e_ref_chroma_fmt == IV_YUV_420SP_VU);
1164*c83a76b0SSuyog Pawar 
1165*c83a76b0SSuyog Pawar                     ihevcd_sao_shift_ctb(&ps_proc->s_sao_ctxt);
1166*c83a76b0SSuyog Pawar #else
1167*c83a76b0SSuyog Pawar                     if(ps_proc->i4_ctb_x > 1 && ps_proc->i4_ctb_y > 0)
1168*c83a76b0SSuyog Pawar                     {
1169*c83a76b0SSuyog Pawar                         ps_proc->s_sao_ctxt.i4_ctb_x = ps_proc->i4_ctb_x - 2;
1170*c83a76b0SSuyog Pawar                         ps_proc->s_sao_ctxt.i4_ctb_y = ps_proc->i4_ctb_y - 1;
1171*c83a76b0SSuyog Pawar 
1172*c83a76b0SSuyog Pawar                         ihevcd_sao_ctb(&ps_proc->s_sao_ctxt);
1173*c83a76b0SSuyog Pawar                     }
1174*c83a76b0SSuyog Pawar 
1175*c83a76b0SSuyog Pawar                     if(ps_sps->i2_pic_wd_in_ctb - 1 == ps_proc->i4_ctb_x && ps_proc->i4_ctb_y > 0)
1176*c83a76b0SSuyog Pawar                     {
1177*c83a76b0SSuyog Pawar                         ps_proc->s_sao_ctxt.i4_ctb_x = ps_proc->i4_ctb_x - 1;
1178*c83a76b0SSuyog Pawar                         ps_proc->s_sao_ctxt.i4_ctb_y = ps_proc->i4_ctb_y - 1;
1179*c83a76b0SSuyog Pawar 
1180*c83a76b0SSuyog Pawar                         ihevcd_sao_ctb(&ps_proc->s_sao_ctxt);
1181*c83a76b0SSuyog Pawar 
1182*c83a76b0SSuyog Pawar                         ps_proc->s_sao_ctxt.i4_ctb_x = ps_proc->i4_ctb_x;
1183*c83a76b0SSuyog Pawar                         ps_proc->s_sao_ctxt.i4_ctb_y = ps_proc->i4_ctb_y - 1;
1184*c83a76b0SSuyog Pawar 
1185*c83a76b0SSuyog Pawar                         ihevcd_sao_ctb(&ps_proc->s_sao_ctxt);
1186*c83a76b0SSuyog Pawar 
1187*c83a76b0SSuyog Pawar                         if(ps_sps->i2_pic_ht_in_ctb - 1 == ps_proc->i4_ctb_y)
1188*c83a76b0SSuyog Pawar                         {
1189*c83a76b0SSuyog Pawar                             WORD32 i4_ctb_x;
1190*c83a76b0SSuyog Pawar                             ps_proc->s_sao_ctxt.i4_ctb_y = ps_proc->i4_ctb_y;
1191*c83a76b0SSuyog Pawar                             for(i4_ctb_x = 0; i4_ctb_x < ps_sps->i2_pic_wd_in_ctb; i4_ctb_x++)
1192*c83a76b0SSuyog Pawar                             {
1193*c83a76b0SSuyog Pawar                                 ps_proc->s_sao_ctxt.i4_ctb_x = i4_ctb_x;
1194*c83a76b0SSuyog Pawar                                 ihevcd_sao_ctb(&ps_proc->s_sao_ctxt);
1195*c83a76b0SSuyog Pawar                             }
1196*c83a76b0SSuyog Pawar                         }
1197*c83a76b0SSuyog Pawar                     }
1198*c83a76b0SSuyog Pawar #endif
1199*c83a76b0SSuyog Pawar                 }
1200*c83a76b0SSuyog Pawar 
1201*c83a76b0SSuyog Pawar 
1202*c83a76b0SSuyog Pawar                 /* Call padding if required */
1203*c83a76b0SSuyog Pawar                 {
1204*c83a76b0SSuyog Pawar #if SAO_PROCESS_SHIFT_CTB
1205*c83a76b0SSuyog Pawar 
1206*c83a76b0SSuyog Pawar                     if(0 == ps_proc->i4_ctb_x)
1207*c83a76b0SSuyog Pawar                     {
1208*c83a76b0SSuyog Pawar                         WORD32 pad_ht_luma;
1209*c83a76b0SSuyog Pawar                         WORD32 pad_ht_chroma;
1210*c83a76b0SSuyog Pawar 
1211*c83a76b0SSuyog Pawar                         ps_proc->pu1_cur_ctb_luma = ps_proc->pu1_cur_pic_luma
1212*c83a76b0SSuyog Pawar                                         + (ps_proc->i4_ctb_x * ctb_size
1213*c83a76b0SSuyog Pawar                                         + ps_proc->i4_ctb_y * ctb_size
1214*c83a76b0SSuyog Pawar                                         * ps_codec->i4_strd);
1215*c83a76b0SSuyog Pawar                         ps_proc->pu1_cur_ctb_chroma = ps_proc->pu1_cur_pic_chroma
1216*c83a76b0SSuyog Pawar                                         + ps_proc->i4_ctb_x * ctb_size
1217*c83a76b0SSuyog Pawar                                         + (ps_proc->i4_ctb_y * ctb_size * ps_codec->i4_strd / 2);
1218*c83a76b0SSuyog Pawar 
1219*c83a76b0SSuyog Pawar                         pad_ht_luma = ctb_size;
1220*c83a76b0SSuyog Pawar                         pad_ht_luma += (ps_sps->i2_pic_ht_in_ctb - 1) == ps_proc->i4_ctb_y ? 8 : 0;
1221*c83a76b0SSuyog Pawar                         pad_ht_chroma = ctb_size / 2;
1222*c83a76b0SSuyog Pawar                         /* Pad left after 1st CTB is processed */
1223*c83a76b0SSuyog Pawar                         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);
1224*c83a76b0SSuyog Pawar                         ps_codec->s_func_selector.ihevc_pad_left_chroma_fptr(ps_proc->pu1_cur_ctb_chroma - 16 * ps_codec->i4_strd, ps_codec->i4_strd, pad_ht_chroma, PAD_LEFT);
1225*c83a76b0SSuyog Pawar                     }
1226*c83a76b0SSuyog Pawar 
1227*c83a76b0SSuyog Pawar                     if((ps_sps->i2_pic_wd_in_ctb - 1) == ps_proc->i4_ctb_x)
1228*c83a76b0SSuyog Pawar                     {
1229*c83a76b0SSuyog Pawar                         WORD32 pad_ht_luma;
1230*c83a76b0SSuyog Pawar                         WORD32 pad_ht_chroma;
1231*c83a76b0SSuyog Pawar                         WORD32 cols_remaining = ps_sps->i2_pic_width_in_luma_samples - (ps_proc->i4_ctb_x << ps_sps->i1_log2_ctb_size);
1232*c83a76b0SSuyog Pawar 
1233*c83a76b0SSuyog Pawar                         ps_proc->pu1_cur_ctb_luma = ps_proc->pu1_cur_pic_luma
1234*c83a76b0SSuyog Pawar                                         + (ps_proc->i4_ctb_x * ctb_size
1235*c83a76b0SSuyog Pawar                                         + ps_proc->i4_ctb_y * ctb_size
1236*c83a76b0SSuyog Pawar                                         * ps_codec->i4_strd);
1237*c83a76b0SSuyog Pawar                         ps_proc->pu1_cur_ctb_chroma = ps_proc->pu1_cur_pic_chroma
1238*c83a76b0SSuyog Pawar                                         + ps_proc->i4_ctb_x * ctb_size
1239*c83a76b0SSuyog Pawar                                         + (ps_proc->i4_ctb_y * ctb_size * ps_codec->i4_strd / 2);
1240*c83a76b0SSuyog Pawar 
1241*c83a76b0SSuyog Pawar                         pad_ht_luma = ctb_size;
1242*c83a76b0SSuyog Pawar                         pad_ht_chroma = ctb_size / 2;
1243*c83a76b0SSuyog Pawar                         if((ps_sps->i2_pic_ht_in_ctb - 1) == ps_proc->i4_ctb_y)
1244*c83a76b0SSuyog Pawar                         {
1245*c83a76b0SSuyog Pawar                             pad_ht_luma += 8;
1246*c83a76b0SSuyog Pawar                             pad_ht_chroma += 16;
1247*c83a76b0SSuyog Pawar                             ps_codec->s_func_selector.ihevc_pad_left_chroma_fptr(ps_proc->pu1_cur_pic_chroma + (ps_sps->i2_pic_height_in_luma_samples / 2 - 16) * ps_codec->i4_strd,
1248*c83a76b0SSuyog Pawar                                                                                  ps_codec->i4_strd, 16, PAD_LEFT);
1249*c83a76b0SSuyog Pawar                         }
1250*c83a76b0SSuyog Pawar                         /* Pad right after last CTB in the current row is processed */
1251*c83a76b0SSuyog Pawar                         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);
1252*c83a76b0SSuyog Pawar                         ps_codec->s_func_selector.ihevc_pad_right_chroma_fptr(ps_proc->pu1_cur_ctb_chroma + cols_remaining - 16 * ps_codec->i4_strd, ps_codec->i4_strd, pad_ht_chroma, PAD_RIGHT);
1253*c83a76b0SSuyog Pawar 
1254*c83a76b0SSuyog Pawar                         if((ps_sps->i2_pic_ht_in_ctb - 1) == ps_proc->i4_ctb_y)
1255*c83a76b0SSuyog Pawar                         {
1256*c83a76b0SSuyog Pawar                             UWORD8 *pu1_buf;
1257*c83a76b0SSuyog Pawar                             /* Since SAO is shifted by 8x8, chroma padding can not be done till second row is processed */
1258*c83a76b0SSuyog Pawar                             /* Hence moving top padding to to end of frame, Moving it to second row also results in problems when there is only one row */
1259*c83a76b0SSuyog Pawar                             /* Pad top after padding left and right for current rows after processing 1st CTB row */
1260*c83a76b0SSuyog Pawar                             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);
1261*c83a76b0SSuyog Pawar                             ihevc_pad_top(ps_proc->pu1_cur_pic_chroma - PAD_LEFT, ps_codec->i4_strd, ps_sps->i2_pic_width_in_luma_samples + PAD_WD, PAD_TOP / 2);
1262*c83a76b0SSuyog Pawar 
1263*c83a76b0SSuyog Pawar                             pu1_buf = ps_proc->pu1_cur_pic_luma + ps_codec->i4_strd * ps_sps->i2_pic_height_in_luma_samples - PAD_LEFT;
1264*c83a76b0SSuyog Pawar                             /* Pad top after padding left and right for current rows after processing 1st CTB row */
1265*c83a76b0SSuyog Pawar                             ihevc_pad_bottom(pu1_buf, ps_codec->i4_strd, ps_sps->i2_pic_width_in_luma_samples + PAD_WD, PAD_BOT);
1266*c83a76b0SSuyog Pawar 
1267*c83a76b0SSuyog Pawar                             pu1_buf = ps_proc->pu1_cur_pic_chroma + ps_codec->i4_strd * (ps_sps->i2_pic_height_in_luma_samples / 2) - PAD_LEFT;
1268*c83a76b0SSuyog Pawar                             ihevc_pad_bottom(pu1_buf, ps_codec->i4_strd, ps_sps->i2_pic_width_in_luma_samples + PAD_WD, PAD_BOT / 2);
1269*c83a76b0SSuyog Pawar                         }
1270*c83a76b0SSuyog Pawar                     }
1271*c83a76b0SSuyog Pawar #else
1272*c83a76b0SSuyog Pawar                     if(ps_proc->i4_ctb_y > 1)
1273*c83a76b0SSuyog Pawar                     {
1274*c83a76b0SSuyog Pawar                         if(0 == ps_proc->i4_ctb_x)
1275*c83a76b0SSuyog Pawar                         {
1276*c83a76b0SSuyog Pawar                             WORD32 pad_ht_luma;
1277*c83a76b0SSuyog Pawar                             WORD32 pad_ht_chroma;
1278*c83a76b0SSuyog Pawar 
1279*c83a76b0SSuyog Pawar                             pad_ht_luma = ctb_size;
1280*c83a76b0SSuyog Pawar                             pad_ht_chroma = ctb_size / 2;
1281*c83a76b0SSuyog Pawar                             /* Pad left after 1st CTB is processed */
1282*c83a76b0SSuyog Pawar                             ps_codec->s_func_selector.ihevc_pad_left_luma_fptr(ps_proc->pu1_cur_ctb_luma - 2 * ctb_size * ps_codec->i4_strd, ps_codec->i4_strd, pad_ht_luma, PAD_LEFT);
1283*c83a76b0SSuyog Pawar                             ps_codec->s_func_selector.ihevc_pad_left_chroma_fptr(ps_proc->pu1_cur_ctb_chroma - ctb_size * ps_codec->i4_strd, ps_codec->i4_strd, pad_ht_chroma, PAD_LEFT);
1284*c83a76b0SSuyog Pawar                         }
1285*c83a76b0SSuyog Pawar                         else if((ps_sps->i2_pic_wd_in_ctb - 1) == ps_proc->i4_ctb_x)
1286*c83a76b0SSuyog Pawar                         {
1287*c83a76b0SSuyog Pawar                             WORD32 pad_ht_luma;
1288*c83a76b0SSuyog Pawar                             WORD32 pad_ht_chroma;
1289*c83a76b0SSuyog Pawar                             WORD32 cols_remaining = ps_sps->i2_pic_width_in_luma_samples - (ps_proc->i4_ctb_x << ps_sps->i1_log2_ctb_size);
1290*c83a76b0SSuyog Pawar 
1291*c83a76b0SSuyog Pawar                             pad_ht_luma = ((ps_sps->i2_pic_ht_in_ctb - 1) == ps_proc->i4_ctb_y) ? 3 * ctb_size : ctb_size;
1292*c83a76b0SSuyog Pawar                             pad_ht_chroma = ((ps_sps->i2_pic_ht_in_ctb - 1) == ps_proc->i4_ctb_y) ? 3 * ctb_size / 2 : ctb_size / 2;
1293*c83a76b0SSuyog Pawar                             /* Pad right after last CTB in the current row is processed */
1294*c83a76b0SSuyog Pawar                             ps_codec->s_func_selector.ihevc_pad_right_luma_fptr(ps_proc->pu1_cur_ctb_luma + cols_remaining - 2 * ctb_size * ps_codec->i4_strd, ps_codec->i4_strd, pad_ht_luma, PAD_RIGHT);
1295*c83a76b0SSuyog Pawar                             ps_codec->s_func_selector.ihevc_pad_right_chroma_fptr(ps_proc->pu1_cur_ctb_chroma + cols_remaining - ctb_size * ps_codec->i4_strd, ps_codec->i4_strd, pad_ht_chroma, PAD_RIGHT);
1296*c83a76b0SSuyog Pawar 
1297*c83a76b0SSuyog Pawar                             if((ps_sps->i2_pic_ht_in_ctb - 1) == ps_proc->i4_ctb_y)
1298*c83a76b0SSuyog Pawar                             {
1299*c83a76b0SSuyog Pawar                                 UWORD8 *pu1_buf;
1300*c83a76b0SSuyog Pawar                                 WORD32 pad_ht_luma;
1301*c83a76b0SSuyog Pawar                                 WORD32 pad_ht_chroma;
1302*c83a76b0SSuyog Pawar 
1303*c83a76b0SSuyog Pawar                                 pad_ht_luma = 2 * ctb_size;
1304*c83a76b0SSuyog Pawar                                 pad_ht_chroma = ctb_size;
1305*c83a76b0SSuyog Pawar 
1306*c83a76b0SSuyog Pawar                                 ps_codec->s_func_selector.ihevc_pad_left_luma_fptr(ps_proc->pu1_cur_pic_luma + ps_codec->i4_strd * (ps_sps->i2_pic_height_in_luma_samples - 2 * ctb_size),
1307*c83a76b0SSuyog Pawar                                                                                    ps_codec->i4_strd, pad_ht_luma, PAD_LEFT);
1308*c83a76b0SSuyog Pawar                                 ps_codec->s_func_selector.ihevc_pad_left_chroma_fptr(ps_proc->pu1_cur_pic_chroma + ps_codec->i4_strd * (ps_sps->i2_pic_height_in_luma_samples / 2 - ctb_size),
1309*c83a76b0SSuyog Pawar                                                                                      ps_codec->i4_strd, pad_ht_chroma, PAD_LEFT);
1310*c83a76b0SSuyog Pawar 
1311*c83a76b0SSuyog Pawar                                 /* Since SAO is shifted by 8x8, chroma padding can not be done till second row is processed */
1312*c83a76b0SSuyog Pawar                                 /* Hence moving top padding to to end of frame, Moving it to second row also results in problems when there is only one row */
1313*c83a76b0SSuyog Pawar                                 /* Pad top after padding left and right for current rows after processing 1st CTB row */
1314*c83a76b0SSuyog Pawar                                 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);
1315*c83a76b0SSuyog Pawar                                 ihevc_pad_top(ps_proc->pu1_cur_pic_chroma - PAD_LEFT, ps_codec->i4_strd, ps_sps->i2_pic_width_in_luma_samples + PAD_WD, PAD_TOP / 2);
1316*c83a76b0SSuyog Pawar 
1317*c83a76b0SSuyog Pawar                                 pu1_buf = ps_proc->pu1_cur_pic_luma + ps_codec->i4_strd * ps_sps->i2_pic_height_in_luma_samples - PAD_LEFT;
1318*c83a76b0SSuyog Pawar                                 /* Pad top after padding left and right for current rows after processing 1st CTB row */
1319*c83a76b0SSuyog Pawar                                 ihevc_pad_bottom(pu1_buf, ps_codec->i4_strd, ps_sps->i2_pic_width_in_luma_samples + PAD_WD, PAD_BOT);
1320*c83a76b0SSuyog Pawar 
1321*c83a76b0SSuyog Pawar                                 pu1_buf = ps_proc->pu1_cur_pic_chroma + ps_codec->i4_strd * (ps_sps->i2_pic_height_in_luma_samples / 2) - PAD_LEFT;
1322*c83a76b0SSuyog Pawar                                 ihevc_pad_bottom(pu1_buf, ps_codec->i4_strd, ps_sps->i2_pic_width_in_luma_samples + PAD_WD, PAD_BOT / 2);
1323*c83a76b0SSuyog Pawar                             }
1324*c83a76b0SSuyog Pawar                         }
1325*c83a76b0SSuyog Pawar                     }
1326*c83a76b0SSuyog Pawar #endif
1327*c83a76b0SSuyog Pawar                 }
1328*c83a76b0SSuyog Pawar             }
1329*c83a76b0SSuyog Pawar 
1330*c83a76b0SSuyog Pawar 
1331*c83a76b0SSuyog Pawar             /* Update proc map for SAO*/
1332*c83a76b0SSuyog Pawar             ihevcd_proc_map_update(ps_proc, proc_type, nctb);
1333*c83a76b0SSuyog Pawar             /* Update proc map for Completion of CTB*/
1334*c83a76b0SSuyog Pawar             ihevcd_proc_map_update(ps_proc, PROC_ALL, nctb);
1335*c83a76b0SSuyog Pawar             {
1336*c83a76b0SSuyog Pawar                 tile_t *ps_tile;
1337*c83a76b0SSuyog Pawar 
1338*c83a76b0SSuyog Pawar                 ps_tile = ps_proc->ps_tile;
1339*c83a76b0SSuyog Pawar                 num_ctb_tmp -= nctb;
1340*c83a76b0SSuyog Pawar 
1341*c83a76b0SSuyog Pawar                 ps_proc->i4_ctb_tile_x += nctb;
1342*c83a76b0SSuyog Pawar                 ps_proc->i4_ctb_x += nctb;
1343*c83a76b0SSuyog Pawar 
1344*c83a76b0SSuyog Pawar                 ps_proc->i4_ctb_slice_x += nctb;
1345*c83a76b0SSuyog Pawar 
1346*c83a76b0SSuyog Pawar 
1347*c83a76b0SSuyog Pawar                 /* Update tile counters */
1348*c83a76b0SSuyog Pawar                 if(ps_proc->i4_ctb_tile_x >= (ps_tile->u2_wd))
1349*c83a76b0SSuyog Pawar                 {
1350*c83a76b0SSuyog Pawar                     /*End of tile row*/
1351*c83a76b0SSuyog Pawar                     ps_proc->i4_ctb_tile_x = 0;
1352*c83a76b0SSuyog Pawar                     ps_proc->i4_ctb_x = ps_tile->u1_pos_x;
1353*c83a76b0SSuyog Pawar 
1354*c83a76b0SSuyog Pawar                     ps_proc->i4_ctb_tile_y++;
1355*c83a76b0SSuyog Pawar                     ps_proc->i4_ctb_y++;
1356*c83a76b0SSuyog Pawar                     if(ps_proc->i4_ctb_tile_y == ps_tile->u2_ht)
1357*c83a76b0SSuyog Pawar                     {
1358*c83a76b0SSuyog Pawar                         /* Reached End of Tile */
1359*c83a76b0SSuyog Pawar                         ps_proc->i4_ctb_tile_y = 0;
1360*c83a76b0SSuyog Pawar                         ps_proc->i4_ctb_tile_x = 0;
1361*c83a76b0SSuyog Pawar                         ps_proc->ps_tile++;
1362*c83a76b0SSuyog Pawar                         //End of picture
1363*c83a76b0SSuyog Pawar                         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)))
1364*c83a76b0SSuyog Pawar                         {
1365*c83a76b0SSuyog Pawar                             ps_tile = ps_proc->ps_tile;
1366*c83a76b0SSuyog Pawar                             ps_proc->i4_ctb_x = ps_tile->u1_pos_x;
1367*c83a76b0SSuyog Pawar                             ps_proc->i4_ctb_y = ps_tile->u1_pos_y;
1368*c83a76b0SSuyog Pawar 
1369*c83a76b0SSuyog Pawar                         }
1370*c83a76b0SSuyog Pawar                     }
1371*c83a76b0SSuyog Pawar                 }
1372*c83a76b0SSuyog Pawar             }
1373*c83a76b0SSuyog Pawar         }
1374*c83a76b0SSuyog Pawar 
1375*c83a76b0SSuyog Pawar         ps_proc->i4_ctb_cnt -= num_ctb;
1376*c83a76b0SSuyog Pawar     }
1377*c83a76b0SSuyog Pawar     return ret;
1378*c83a76b0SSuyog Pawar }
1379*c83a76b0SSuyog Pawar 
ihevcd_init_proc_ctxt(process_ctxt_t * ps_proc,WORD32 tu_coeff_data_ofst)1380*c83a76b0SSuyog Pawar void ihevcd_init_proc_ctxt(process_ctxt_t *ps_proc, WORD32 tu_coeff_data_ofst)
1381*c83a76b0SSuyog Pawar {
1382*c83a76b0SSuyog Pawar     codec_t *ps_codec;
1383*c83a76b0SSuyog Pawar     slice_header_t *ps_slice_hdr;
1384*c83a76b0SSuyog Pawar     pps_t *ps_pps;
1385*c83a76b0SSuyog Pawar     sps_t *ps_sps;
1386*c83a76b0SSuyog Pawar     tile_t *ps_tile, *ps_tile_prev;
1387*c83a76b0SSuyog Pawar     WORD32 tile_idx;
1388*c83a76b0SSuyog Pawar     WORD32 ctb_size;
1389*c83a76b0SSuyog Pawar     WORD32 num_minpu_in_ctb;
1390*c83a76b0SSuyog Pawar     WORD32 num_ctb_in_row;
1391*c83a76b0SSuyog Pawar     WORD32 ctb_addr;
1392*c83a76b0SSuyog Pawar     WORD32 i4_wd_in_ctb;
1393*c83a76b0SSuyog Pawar     WORD32 tile_start_ctb_idx;
1394*c83a76b0SSuyog Pawar     WORD32 slice_start_ctb_idx;
1395*c83a76b0SSuyog Pawar     WORD32 check_tile_wd;
1396*c83a76b0SSuyog Pawar     WORD32 continuous_tiles = 0; //Refers to tiles that are continuous, within a slice, horizontally
1397*c83a76b0SSuyog Pawar 
1398*c83a76b0SSuyog Pawar     ps_codec = ps_proc->ps_codec;
1399*c83a76b0SSuyog Pawar 
1400*c83a76b0SSuyog Pawar     ps_slice_hdr = ps_codec->ps_slice_hdr_base + ((ps_proc->i4_cur_slice_idx) & (MAX_SLICE_HDR_CNT - 1));
1401*c83a76b0SSuyog Pawar     ps_proc->ps_slice_hdr = ps_slice_hdr;
1402*c83a76b0SSuyog Pawar     ps_proc->ps_pps = ps_codec->ps_pps_base + ps_slice_hdr->i1_pps_id;
1403*c83a76b0SSuyog Pawar     ps_pps = ps_proc->ps_pps;
1404*c83a76b0SSuyog Pawar     ps_proc->ps_sps = ps_codec->ps_sps_base + ps_pps->i1_sps_id;
1405*c83a76b0SSuyog Pawar     ps_sps = ps_proc->ps_sps;
1406*c83a76b0SSuyog Pawar     ps_proc->i4_init_done = 1;
1407*c83a76b0SSuyog Pawar     ctb_size = 1 << ps_sps->i1_log2_ctb_size;
1408*c83a76b0SSuyog Pawar     num_minpu_in_ctb = (ctb_size / MIN_PU_SIZE) * (ctb_size / MIN_PU_SIZE);
1409*c83a76b0SSuyog Pawar     num_ctb_in_row = ps_sps->i2_pic_wd_in_ctb;
1410*c83a76b0SSuyog Pawar 
1411*c83a76b0SSuyog Pawar     ps_proc->s_sao_ctxt.pu1_slice_idx = ps_proc->pu1_slice_idx;
1412*c83a76b0SSuyog Pawar 
1413*c83a76b0SSuyog Pawar     ihevcd_get_tile_pos(ps_pps, ps_sps, ps_proc->i4_ctb_x, ps_proc->i4_ctb_y,
1414*c83a76b0SSuyog Pawar                         &ps_proc->i4_ctb_tile_x, &ps_proc->i4_ctb_tile_y,
1415*c83a76b0SSuyog Pawar                         &tile_idx);
1416*c83a76b0SSuyog Pawar 
1417*c83a76b0SSuyog Pawar     ps_proc->ps_tile = ps_pps->ps_tile + tile_idx;
1418*c83a76b0SSuyog Pawar     ps_proc->i4_cur_tile_idx = tile_idx;
1419*c83a76b0SSuyog Pawar     ps_tile = ps_proc->ps_tile;
1420*c83a76b0SSuyog Pawar 
1421*c83a76b0SSuyog Pawar     if(ps_pps->i1_tiles_enabled_flag)
1422*c83a76b0SSuyog Pawar     {
1423*c83a76b0SSuyog Pawar         if(tile_idx)
1424*c83a76b0SSuyog Pawar             ps_tile_prev = ps_tile - 1;
1425*c83a76b0SSuyog Pawar         else
1426*c83a76b0SSuyog Pawar             ps_tile_prev = ps_tile;
1427*c83a76b0SSuyog Pawar 
1428*c83a76b0SSuyog Pawar         slice_start_ctb_idx = ps_slice_hdr->i2_ctb_x + (ps_slice_hdr->i2_ctb_y * ps_sps->i2_pic_wd_in_ctb);
1429*c83a76b0SSuyog Pawar         tile_start_ctb_idx = ps_tile->u1_pos_x + (ps_tile->u1_pos_y * ps_sps->i2_pic_wd_in_ctb);
1430*c83a76b0SSuyog Pawar 
1431*c83a76b0SSuyog Pawar         /*Check if
1432*c83a76b0SSuyog Pawar          * 1. Last tile that ends in frame boundary and 1st tile in next row belongs to same slice
1433*c83a76b0SSuyog Pawar          * 1.1. If it does, check if the slice that has these tiles spans across the frame row.
1434*c83a76b0SSuyog Pawar          * 2. Vertical tiles are present within a slice */
1435*c83a76b0SSuyog Pawar         if(((ps_slice_hdr->i2_ctb_x == ps_tile->u1_pos_x) && (ps_slice_hdr->i2_ctb_y != ps_tile->u1_pos_y)))
1436*c83a76b0SSuyog Pawar         {
1437*c83a76b0SSuyog Pawar             continuous_tiles = 1;
1438*c83a76b0SSuyog Pawar         }
1439*c83a76b0SSuyog Pawar         else
1440*c83a76b0SSuyog Pawar         {
1441*c83a76b0SSuyog Pawar             check_tile_wd = ps_slice_hdr->i2_ctb_x + ps_tile_prev->u2_wd;
1442*c83a76b0SSuyog Pawar             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))
1443*c83a76b0SSuyog Pawar                                             || ((ps_slice_hdr->i2_ctb_x == ps_tile->u1_pos_x))))
1444*c83a76b0SSuyog Pawar             {
1445*c83a76b0SSuyog Pawar                 continuous_tiles = 1;
1446*c83a76b0SSuyog Pawar             }
1447*c83a76b0SSuyog Pawar         }
1448*c83a76b0SSuyog Pawar 
1449*c83a76b0SSuyog Pawar         {
1450*c83a76b0SSuyog Pawar             WORD32 i2_independent_ctb_x = ps_slice_hdr->i2_independent_ctb_x;
1451*c83a76b0SSuyog Pawar             WORD32 i2_independent_ctb_y = ps_slice_hdr->i2_independent_ctb_y;
1452*c83a76b0SSuyog Pawar 
1453*c83a76b0SSuyog Pawar             /* Handles cases where
1454*c83a76b0SSuyog Pawar              * 1. Slices begin at the start of each tile
1455*c83a76b0SSuyog Pawar              * 2. Tiles lie in the same slice row.i.e, starting tile_x > slice_x, but tile_y == slice_y
1456*c83a76b0SSuyog Pawar              * */
1457*c83a76b0SSuyog Pawar             if(ps_proc->i4_ctb_x >= i2_independent_ctb_x)
1458*c83a76b0SSuyog Pawar             {
1459*c83a76b0SSuyog Pawar                 ps_proc->i4_ctb_slice_x = ps_proc->i4_ctb_x - i2_independent_ctb_x;
1460*c83a76b0SSuyog Pawar             }
1461*c83a76b0SSuyog Pawar             else
1462*c83a76b0SSuyog Pawar             {
1463*c83a76b0SSuyog Pawar                 /* Indicates multiple tiles in a slice case where
1464*c83a76b0SSuyog Pawar                  * The new tile belongs to an older slice that started in the previous rows-not the present row
1465*c83a76b0SSuyog Pawar                  * & (tile_y > slice_y and tile_x < slice_x)
1466*c83a76b0SSuyog Pawar                  */
1467*c83a76b0SSuyog Pawar                 if((slice_start_ctb_idx < tile_start_ctb_idx) && (continuous_tiles))
1468*c83a76b0SSuyog Pawar                 {
1469*c83a76b0SSuyog Pawar                     i4_wd_in_ctb = ps_sps->i2_pic_wd_in_ctb;
1470*c83a76b0SSuyog Pawar                 }
1471*c83a76b0SSuyog Pawar                 /* Indicates many-tiles-in-one-slice case, for slices that end without spanning the frame width*/
1472*c83a76b0SSuyog Pawar                 else
1473*c83a76b0SSuyog Pawar                 {
1474*c83a76b0SSuyog Pawar                     i4_wd_in_ctb = ps_tile->u2_wd;
1475*c83a76b0SSuyog Pawar                 }
1476*c83a76b0SSuyog Pawar 
1477*c83a76b0SSuyog Pawar                 if(continuous_tiles)
1478*c83a76b0SSuyog Pawar                 {
1479*c83a76b0SSuyog Pawar                     ps_proc->i4_ctb_slice_x = i4_wd_in_ctb
1480*c83a76b0SSuyog Pawar                                     - (i2_independent_ctb_x - ps_proc->i4_ctb_x);
1481*c83a76b0SSuyog Pawar                 }
1482*c83a76b0SSuyog Pawar                 else
1483*c83a76b0SSuyog Pawar                 {
1484*c83a76b0SSuyog Pawar                     ps_proc->i4_ctb_slice_x = ps_proc->i4_ctb_x - ps_tile->u1_pos_x;
1485*c83a76b0SSuyog Pawar                 }
1486*c83a76b0SSuyog Pawar             }
1487*c83a76b0SSuyog Pawar             /* Initialize ctb slice y to zero and at the start of slice row initialize it
1488*c83a76b0SSuyog Pawar         to difference between ctb_y and slice's start ctb y */
1489*c83a76b0SSuyog Pawar 
1490*c83a76b0SSuyog Pawar             ps_proc->i4_ctb_slice_y = ps_proc->i4_ctb_y - i2_independent_ctb_y;
1491*c83a76b0SSuyog Pawar 
1492*c83a76b0SSuyog Pawar             /*If beginning of tile, check if slice counters are set correctly*/
1493*c83a76b0SSuyog Pawar             if((0 == ps_proc->i4_ctb_tile_x) && (0 == ps_proc->i4_ctb_tile_y))
1494*c83a76b0SSuyog Pawar             {
1495*c83a76b0SSuyog Pawar                 if(ps_slice_hdr->i1_dependent_slice_flag)
1496*c83a76b0SSuyog Pawar                 {
1497*c83a76b0SSuyog Pawar                     ps_proc->i4_ctb_slice_x = 0;
1498*c83a76b0SSuyog Pawar                     ps_proc->i4_ctb_slice_y = 0;
1499*c83a76b0SSuyog Pawar                 }
1500*c83a76b0SSuyog Pawar                 /*For slices that span across multiple tiles*/
1501*c83a76b0SSuyog Pawar                 else if(slice_start_ctb_idx < tile_start_ctb_idx)
1502*c83a76b0SSuyog Pawar                 {
1503*c83a76b0SSuyog Pawar                     ps_proc->i4_ctb_slice_y = ps_tile->u1_pos_y - i2_independent_ctb_y;
1504*c83a76b0SSuyog Pawar                     /* Two Cases
1505*c83a76b0SSuyog Pawar                      * 1 - slice spans across frame-width- but dose not start from 1st column
1506*c83a76b0SSuyog Pawar                      * 2 - Slice spans across multiple tiles anywhere is a frame
1507*c83a76b0SSuyog Pawar                      */
1508*c83a76b0SSuyog Pawar                     /*TODO:In a multiple slice clip,  if an independent slice span across more than 2 tiles in a row, it is not supported*/
1509*c83a76b0SSuyog Pawar                     if(continuous_tiles) //Case 2-implemented for slices that span not more than 2 tiles
1510*c83a76b0SSuyog Pawar                     {
1511*c83a76b0SSuyog Pawar                         if(i2_independent_ctb_y <= ps_tile->u1_pos_y)
1512*c83a76b0SSuyog Pawar                         {
1513*c83a76b0SSuyog Pawar                             //Check if ctb x is before or after
1514*c83a76b0SSuyog Pawar                             if(i2_independent_ctb_x > ps_tile->u1_pos_x)
1515*c83a76b0SSuyog Pawar                             {
1516*c83a76b0SSuyog Pawar                                 ps_proc->i4_ctb_slice_y -= 1;
1517*c83a76b0SSuyog Pawar                             }
1518*c83a76b0SSuyog Pawar                         }
1519*c83a76b0SSuyog Pawar                     }
1520*c83a76b0SSuyog Pawar                 }
1521*c83a76b0SSuyog Pawar             }
1522*c83a76b0SSuyog Pawar             //Slice starts from a column which is not the starting tile-column, but is within the tile
1523*c83a76b0SSuyog Pawar             if(((i2_independent_ctb_x - ps_tile->u1_pos_x) != 0) && ((ps_proc->i4_ctb_slice_y != 0))
1524*c83a76b0SSuyog Pawar                             && ((i2_independent_ctb_x >= ps_tile->u1_pos_x) && (i2_independent_ctb_x < ps_tile->u1_pos_x + ps_tile->u2_wd)))
1525*c83a76b0SSuyog Pawar             {
1526*c83a76b0SSuyog Pawar                 ps_proc->i4_ctb_slice_y -= 1;
1527*c83a76b0SSuyog Pawar             }
1528*c83a76b0SSuyog Pawar         }
1529*c83a76b0SSuyog Pawar     }
1530*c83a76b0SSuyog Pawar     else
1531*c83a76b0SSuyog Pawar     {
1532*c83a76b0SSuyog Pawar         WORD32 i2_independent_ctb_x = ps_slice_hdr->i2_independent_ctb_x;
1533*c83a76b0SSuyog Pawar         WORD32 i2_independent_ctb_y = ps_slice_hdr->i2_independent_ctb_y;
1534*c83a76b0SSuyog Pawar 
1535*c83a76b0SSuyog Pawar 
1536*c83a76b0SSuyog Pawar         {
1537*c83a76b0SSuyog Pawar             ps_proc->i4_ctb_slice_x = ps_proc->i4_ctb_x - i2_independent_ctb_x;
1538*c83a76b0SSuyog Pawar             ps_proc->i4_ctb_slice_y = ps_proc->i4_ctb_y - i2_independent_ctb_y;
1539*c83a76b0SSuyog Pawar             if(ps_proc->i4_ctb_slice_x < 0)
1540*c83a76b0SSuyog Pawar             {
1541*c83a76b0SSuyog Pawar                 ps_proc->i4_ctb_slice_x += ps_sps->i2_pic_wd_in_ctb;
1542*c83a76b0SSuyog Pawar                 ps_proc->i4_ctb_slice_y -= 1;
1543*c83a76b0SSuyog Pawar             }
1544*c83a76b0SSuyog Pawar 
1545*c83a76b0SSuyog Pawar             /* Initialize ctb slice y to zero and at the start of slice row initialize it
1546*c83a76b0SSuyog Pawar             to difference between ctb_y and slice's start ctb y */
1547*c83a76b0SSuyog Pawar         }
1548*c83a76b0SSuyog Pawar     }
1549*c83a76b0SSuyog Pawar 
1550*c83a76b0SSuyog Pawar     /* Compute TU offset for the current CTB set */
1551*c83a76b0SSuyog Pawar     {
1552*c83a76b0SSuyog Pawar 
1553*c83a76b0SSuyog Pawar         WORD32 ctb_luma_min_tu_cnt;
1554*c83a76b0SSuyog Pawar         WORD32 ctb_addr;
1555*c83a76b0SSuyog Pawar 
1556*c83a76b0SSuyog Pawar         ctb_addr = ps_proc->i4_ctb_y * num_ctb_in_row + ps_proc->i4_ctb_x;
1557*c83a76b0SSuyog Pawar 
1558*c83a76b0SSuyog Pawar         ctb_luma_min_tu_cnt = (1 << ps_sps->i1_log2_ctb_size) / MIN_TU_SIZE;
1559*c83a76b0SSuyog Pawar         ctb_luma_min_tu_cnt *= ctb_luma_min_tu_cnt;
1560*c83a76b0SSuyog Pawar 
1561*c83a76b0SSuyog Pawar         ps_proc->pu1_tu_map = ps_proc->pu1_pic_tu_map
1562*c83a76b0SSuyog Pawar                         + ctb_luma_min_tu_cnt * ctb_addr;
1563*c83a76b0SSuyog Pawar         if(1 == ps_codec->i4_num_cores)
1564*c83a76b0SSuyog Pawar         {
1565*c83a76b0SSuyog Pawar             ps_proc->ps_tu = ps_proc->ps_pic_tu + ps_proc->pu4_pic_tu_idx[ctb_addr % RESET_TU_BUF_NCTB];
1566*c83a76b0SSuyog Pawar         }
1567*c83a76b0SSuyog Pawar         else
1568*c83a76b0SSuyog Pawar         {
1569*c83a76b0SSuyog Pawar             ps_proc->ps_tu = ps_proc->ps_pic_tu + ps_proc->pu4_pic_tu_idx[ctb_addr];
1570*c83a76b0SSuyog Pawar         }
1571*c83a76b0SSuyog Pawar         ps_proc->pv_tu_coeff_data = (UWORD8 *)ps_proc->pv_pic_tu_coeff_data
1572*c83a76b0SSuyog Pawar                         + tu_coeff_data_ofst;
1573*c83a76b0SSuyog Pawar 
1574*c83a76b0SSuyog Pawar     }
1575*c83a76b0SSuyog Pawar 
1576*c83a76b0SSuyog Pawar     /* Compute PU related elements for the current CTB set */
1577*c83a76b0SSuyog Pawar     {
1578*c83a76b0SSuyog Pawar         WORD32 pu_idx;
1579*c83a76b0SSuyog Pawar         ctb_addr = ps_proc->i4_ctb_y * num_ctb_in_row + ps_proc->i4_ctb_x;
1580*c83a76b0SSuyog Pawar         pu_idx = ps_proc->pu4_pic_pu_idx[ctb_addr];
1581*c83a76b0SSuyog Pawar         ps_proc->pu1_pu_map = ps_proc->pu1_pic_pu_map
1582*c83a76b0SSuyog Pawar                         + ctb_addr * num_minpu_in_ctb;
1583*c83a76b0SSuyog Pawar         ps_proc->ps_pu = ps_proc->ps_pic_pu + pu_idx;
1584*c83a76b0SSuyog Pawar     }
1585*c83a76b0SSuyog Pawar 
1586*c83a76b0SSuyog Pawar     /* Number of ctbs processed in one loop of process function */
1587*c83a76b0SSuyog Pawar     {
1588*c83a76b0SSuyog Pawar         ps_proc->i4_nctb = MIN(ps_codec->u4_nctb, ps_tile->u2_wd);
1589*c83a76b0SSuyog Pawar     }
1590*c83a76b0SSuyog Pawar 
1591*c83a76b0SSuyog Pawar }
ihevcd_process_thread(process_ctxt_t * ps_proc)1592*c83a76b0SSuyog Pawar void ihevcd_process_thread(process_ctxt_t *ps_proc)
1593*c83a76b0SSuyog Pawar {
1594*c83a76b0SSuyog Pawar     IHEVCD_ERROR_T ret = (IHEVCD_ERROR_T)IHEVCD_SUCCESS;
1595*c83a76b0SSuyog Pawar     {
1596*c83a76b0SSuyog Pawar         ithread_set_affinity(ps_proc->i4_id + 1);
1597*c83a76b0SSuyog Pawar     }
1598*c83a76b0SSuyog Pawar 
1599*c83a76b0SSuyog Pawar     while(1)
1600*c83a76b0SSuyog Pawar     {
1601*c83a76b0SSuyog Pawar         codec_t *ps_dec = ps_proc->ps_codec;
1602*c83a76b0SSuyog Pawar         if(ps_proc->ps_codec->i4_threads_active)
1603*c83a76b0SSuyog Pawar         {
1604*c83a76b0SSuyog Pawar             DEBUG("In ihevcd_process_thread \n");
1605*c83a76b0SSuyog Pawar 
1606*c83a76b0SSuyog Pawar             ret = ithread_mutex_lock(ps_dec->apv_proc_start_mutex[ps_proc->i4_id]);
1607*c83a76b0SSuyog Pawar             if((IHEVCD_ERROR_T)IHEVCD_SUCCESS != ret)
1608*c83a76b0SSuyog Pawar                 break;
1609*c83a76b0SSuyog Pawar 
1610*c83a76b0SSuyog Pawar             while(!ps_dec->ai4_process_start[ps_proc->i4_id])
1611*c83a76b0SSuyog Pawar             {
1612*c83a76b0SSuyog Pawar                 ithread_cond_wait(ps_dec->apv_proc_start_condition[ps_proc->i4_id],
1613*c83a76b0SSuyog Pawar                                   ps_dec->apv_proc_start_mutex[ps_proc->i4_id]);
1614*c83a76b0SSuyog Pawar             }
1615*c83a76b0SSuyog Pawar             ps_dec->ai4_process_start[ps_proc->i4_id] = 0;
1616*c83a76b0SSuyog Pawar             ret = ithread_mutex_unlock(ps_dec->apv_proc_start_mutex[ps_proc->i4_id]);
1617*c83a76b0SSuyog Pawar             if((IHEVCD_ERROR_T)IHEVCD_SUCCESS != ret)
1618*c83a76b0SSuyog Pawar                 break;
1619*c83a76b0SSuyog Pawar 
1620*c83a76b0SSuyog Pawar             DEBUG(" Got control at ihevcd_process_thread \n");
1621*c83a76b0SSuyog Pawar 
1622*c83a76b0SSuyog Pawar             if(ps_dec->i4_break_threads == 1)
1623*c83a76b0SSuyog Pawar                 break;
1624*c83a76b0SSuyog Pawar         }
1625*c83a76b0SSuyog Pawar         while(1)
1626*c83a76b0SSuyog Pawar         {
1627*c83a76b0SSuyog Pawar             proc_job_t s_job;
1628*c83a76b0SSuyog Pawar 
1629*c83a76b0SSuyog Pawar             ret = ihevcd_jobq_dequeue((jobq_t *)ps_proc->pv_proc_jobq, &s_job,
1630*c83a76b0SSuyog Pawar                                     sizeof(proc_job_t), 1);
1631*c83a76b0SSuyog Pawar             if((IHEVCD_ERROR_T)IHEVCD_SUCCESS != ret)
1632*c83a76b0SSuyog Pawar                 break;
1633*c83a76b0SSuyog Pawar 
1634*c83a76b0SSuyog Pawar             ps_proc->i4_ctb_cnt = s_job.i2_ctb_cnt;
1635*c83a76b0SSuyog Pawar             ps_proc->i4_ctb_x = s_job.i2_ctb_x;
1636*c83a76b0SSuyog Pawar             ps_proc->i4_ctb_y = s_job.i2_ctb_y;
1637*c83a76b0SSuyog Pawar             ps_proc->i4_cur_slice_idx = s_job.i2_slice_idx;
1638*c83a76b0SSuyog Pawar 
1639*c83a76b0SSuyog Pawar 
1640*c83a76b0SSuyog Pawar 
1641*c83a76b0SSuyog Pawar             if(CMD_PROCESS == s_job.i4_cmd)
1642*c83a76b0SSuyog Pawar             {
1643*c83a76b0SSuyog Pawar                 ihevcd_init_proc_ctxt(ps_proc, s_job.i4_tu_coeff_data_ofst);
1644*c83a76b0SSuyog Pawar                 ihevcd_process(ps_proc);
1645*c83a76b0SSuyog Pawar             }
1646*c83a76b0SSuyog Pawar             else if(CMD_FMTCONV == s_job.i4_cmd)
1647*c83a76b0SSuyog Pawar             {
1648*c83a76b0SSuyog Pawar                 sps_t *ps_sps;
1649*c83a76b0SSuyog Pawar                 codec_t *ps_codec;
1650*c83a76b0SSuyog Pawar                 ivd_out_bufdesc_t *ps_out_buffer;
1651*c83a76b0SSuyog Pawar                 WORD32 num_rows;
1652*c83a76b0SSuyog Pawar 
1653*c83a76b0SSuyog Pawar                 if(0 == ps_proc->i4_init_done)
1654*c83a76b0SSuyog Pawar                 {
1655*c83a76b0SSuyog Pawar                     ihevcd_init_proc_ctxt(ps_proc, 0);
1656*c83a76b0SSuyog Pawar                 }
1657*c83a76b0SSuyog Pawar                 ps_sps = ps_proc->ps_sps;
1658*c83a76b0SSuyog Pawar                 ps_codec = ps_proc->ps_codec;
1659*c83a76b0SSuyog Pawar                 ps_out_buffer = ps_proc->ps_out_buffer;
1660*c83a76b0SSuyog Pawar                 num_rows = 1 << ps_sps->i1_log2_ctb_size;
1661*c83a76b0SSuyog Pawar 
1662*c83a76b0SSuyog Pawar                 num_rows = MIN(num_rows,
1663*c83a76b0SSuyog Pawar                                (ps_codec->i4_disp_ht - (s_job.i2_ctb_y << ps_sps->i1_log2_ctb_size))
1664*c83a76b0SSuyog Pawar                               );
1665*c83a76b0SSuyog Pawar 
1666*c83a76b0SSuyog Pawar                 if(num_rows < 0)
1667*c83a76b0SSuyog Pawar                     num_rows = 0;
1668*c83a76b0SSuyog Pawar 
1669*c83a76b0SSuyog Pawar                 ihevcd_fmt_conv(ps_proc->ps_codec, ps_proc, ps_out_buffer->pu1_bufs[0],
1670*c83a76b0SSuyog Pawar                                 ps_out_buffer->pu1_bufs[1], ps_out_buffer->pu1_bufs[2],
1671*c83a76b0SSuyog Pawar                                 s_job.i2_ctb_y << ps_sps->i1_log2_ctb_size, num_rows);
1672*c83a76b0SSuyog Pawar             }
1673*c83a76b0SSuyog Pawar         }
1674*c83a76b0SSuyog Pawar         if(ps_proc->ps_codec->i4_threads_active)
1675*c83a76b0SSuyog Pawar         {
1676*c83a76b0SSuyog Pawar             ret = ithread_mutex_lock(ps_dec->apv_proc_done_mutex[ps_proc->i4_id]);
1677*c83a76b0SSuyog Pawar             if((IHEVCD_ERROR_T)IHEVCD_SUCCESS != ret)
1678*c83a76b0SSuyog Pawar                 break;
1679*c83a76b0SSuyog Pawar 
1680*c83a76b0SSuyog Pawar             ps_dec->ai4_process_done[ps_proc->i4_id] = 1;
1681*c83a76b0SSuyog Pawar             ithread_cond_signal(ps_dec->apv_proc_done_condition[ps_proc->i4_id]);
1682*c83a76b0SSuyog Pawar 
1683*c83a76b0SSuyog Pawar             ret = ithread_mutex_unlock(ps_dec->apv_proc_done_mutex[ps_proc->i4_id]);
1684*c83a76b0SSuyog Pawar             if((IHEVCD_ERROR_T)IHEVCD_SUCCESS != ret)
1685*c83a76b0SSuyog Pawar                 break;
1686*c83a76b0SSuyog Pawar         }
1687*c83a76b0SSuyog Pawar         else
1688*c83a76b0SSuyog Pawar         {
1689*c83a76b0SSuyog Pawar             break;
1690*c83a76b0SSuyog Pawar         }
1691*c83a76b0SSuyog Pawar     }
1692*c83a76b0SSuyog Pawar     //ithread_exit(0);
1693*c83a76b0SSuyog Pawar     return;
1694*c83a76b0SSuyog Pawar }
1695*c83a76b0SSuyog Pawar 
1696