xref: /aosp_15_r20/external/libavc/encoder/svc/isvce_residual_pred.c (revision 495ae853bb871d1e5a258cb02c2cc13cde8ddb9a)
1*495ae853SAndroid Build Coastguard Worker /******************************************************************************
2*495ae853SAndroid Build Coastguard Worker  *
3*495ae853SAndroid Build Coastguard Worker  * Copyright (C) 2022 The Android Open Source Project
4*495ae853SAndroid Build Coastguard Worker  *
5*495ae853SAndroid Build Coastguard Worker  * Licensed under the Apache License, Version 2.0 (the "License");
6*495ae853SAndroid Build Coastguard Worker  * you may not use this file except in compliance with the License.
7*495ae853SAndroid Build Coastguard Worker  * You may obtain a copy of the License at:
8*495ae853SAndroid Build Coastguard Worker  *
9*495ae853SAndroid Build Coastguard Worker  * http://www.apache.org/licenses/LICENSE-2.0
10*495ae853SAndroid Build Coastguard Worker  *
11*495ae853SAndroid Build Coastguard Worker  * Unless required by applicable law or agreed to in writing, software
12*495ae853SAndroid Build Coastguard Worker  * distributed under the License is distributed on an "AS IS" BASIS,
13*495ae853SAndroid Build Coastguard Worker  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14*495ae853SAndroid Build Coastguard Worker  * See the License for the specific language governing permissions and
15*495ae853SAndroid Build Coastguard Worker  * limitations under the License.
16*495ae853SAndroid Build Coastguard Worker  *
17*495ae853SAndroid Build Coastguard Worker  *****************************************************************************
18*495ae853SAndroid Build Coastguard Worker  * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
19*495ae853SAndroid Build Coastguard Worker  */
20*495ae853SAndroid Build Coastguard Worker 
21*495ae853SAndroid Build Coastguard Worker /**
22*495ae853SAndroid Build Coastguard Worker *******************************************************************************
23*495ae853SAndroid Build Coastguard Worker * @file
24*495ae853SAndroid Build Coastguard Worker *  isvce_residual_pred.c
25*495ae853SAndroid Build Coastguard Worker *
26*495ae853SAndroid Build Coastguard Worker * @brief
27*495ae853SAndroid Build Coastguard Worker *  Contains functions used for SVC residual prediction
28*495ae853SAndroid Build Coastguard Worker *
29*495ae853SAndroid Build Coastguard Worker *******************************************************************************
30*495ae853SAndroid Build Coastguard Worker */
31*495ae853SAndroid Build Coastguard Worker #include <stdint.h>
32*495ae853SAndroid Build Coastguard Worker #include <math.h>
33*495ae853SAndroid Build Coastguard Worker 
34*495ae853SAndroid Build Coastguard Worker #include "ih264_typedefs.h"
35*495ae853SAndroid Build Coastguard Worker #include "iv2.h"
36*495ae853SAndroid Build Coastguard Worker #include "isvc_macros.h"
37*495ae853SAndroid Build Coastguard Worker #include "ih264_debug.h"
38*495ae853SAndroid Build Coastguard Worker #include "isvc_defs.h"
39*495ae853SAndroid Build Coastguard Worker #include "isvc_structs.h"
40*495ae853SAndroid Build Coastguard Worker #include "isvce_defs.h"
41*495ae853SAndroid Build Coastguard Worker #include "isvce_structs.h"
42*495ae853SAndroid Build Coastguard Worker #include "isvce_res_pred_private_defs.h"
43*495ae853SAndroid Build Coastguard Worker #include "isvce_residual_pred.h"
44*495ae853SAndroid Build Coastguard Worker #include "isvce_utils.h"
45*495ae853SAndroid Build Coastguard Worker #include "isvc_defs.h"
46*495ae853SAndroid Build Coastguard Worker 
isvce_chroma_residual_sampler_2x(coordinates_t * ps_ref_array_positions,coordinates_t * ps_ref_array_phases,buffer_container_t * ps_inp,buffer_container_t * ps_out,buffer_container_t * ps_scratch,UWORD32 u4_ref_nnz,UWORD8 u1_ref_tx_size)47*495ae853SAndroid Build Coastguard Worker void isvce_chroma_residual_sampler_2x(coordinates_t *ps_ref_array_positions,
48*495ae853SAndroid Build Coastguard Worker                                       coordinates_t *ps_ref_array_phases,
49*495ae853SAndroid Build Coastguard Worker                                       buffer_container_t *ps_inp, buffer_container_t *ps_out,
50*495ae853SAndroid Build Coastguard Worker                                       buffer_container_t *ps_scratch, UWORD32 u4_ref_nnz,
51*495ae853SAndroid Build Coastguard Worker                                       UWORD8 u1_ref_tx_size)
52*495ae853SAndroid Build Coastguard Worker {
53*495ae853SAndroid Build Coastguard Worker     WORD32 i4_i;
54*495ae853SAndroid Build Coastguard Worker     WORD16 *pi2_ref_data_byte;
55*495ae853SAndroid Build Coastguard Worker     WORD32 *pi4_ref_array;
56*495ae853SAndroid Build Coastguard Worker     WORD32 i4_phase1, i4_phase2;
57*495ae853SAndroid Build Coastguard Worker 
58*495ae853SAndroid Build Coastguard Worker     WORD16 *pi2_inp_data = ps_inp->pv_data;
59*495ae853SAndroid Build Coastguard Worker     WORD16 *pi2_out_res = ps_out->pv_data;
60*495ae853SAndroid Build Coastguard Worker     WORD32 i4_inp_data_stride = ps_inp->i4_data_stride;
61*495ae853SAndroid Build Coastguard Worker     WORD32 i4_out_res_stride = ps_out->i4_data_stride;
62*495ae853SAndroid Build Coastguard Worker 
63*495ae853SAndroid Build Coastguard Worker     UNUSED(u4_ref_nnz);
64*495ae853SAndroid Build Coastguard Worker 
65*495ae853SAndroid Build Coastguard Worker     UNUSED(ps_ref_array_positions);
66*495ae853SAndroid Build Coastguard Worker     UNUSED(u1_ref_tx_size);
67*495ae853SAndroid Build Coastguard Worker 
68*495ae853SAndroid Build Coastguard Worker     /* For 2x scaling, offsets always point to TL pixel outside MB */
69*495ae853SAndroid Build Coastguard Worker     /* Hence, refTransBlkIdc will be different and since phase */
70*495ae853SAndroid Build Coastguard Worker     /* for first refArray pos for horiz filtering samples > 8, */
71*495ae853SAndroid Build Coastguard Worker     /* first row and first column from the refArray is never used */
72*495ae853SAndroid Build Coastguard Worker     pi2_inp_data += 2 + i4_inp_data_stride;
73*495ae853SAndroid Build Coastguard Worker 
74*495ae853SAndroid Build Coastguard Worker     pi2_ref_data_byte = pi2_inp_data;
75*495ae853SAndroid Build Coastguard Worker 
76*495ae853SAndroid Build Coastguard Worker     i4_phase1 = ps_ref_array_phases[0].i4_abscissa;
77*495ae853SAndroid Build Coastguard Worker     i4_phase2 = ps_ref_array_phases[1].i4_abscissa;
78*495ae853SAndroid Build Coastguard Worker 
79*495ae853SAndroid Build Coastguard Worker     ASSERT(i4_phase1 >= 8);
80*495ae853SAndroid Build Coastguard Worker 
81*495ae853SAndroid Build Coastguard Worker     pi4_ref_array = (WORD32 *) ps_scratch->pv_data;
82*495ae853SAndroid Build Coastguard Worker 
83*495ae853SAndroid Build Coastguard Worker     for(i4_i = 0; i4_i < BLK_SIZE; i4_i++)
84*495ae853SAndroid Build Coastguard Worker     {
85*495ae853SAndroid Build Coastguard Worker         WORD16 i2_coeff1, i2_coeff2;
86*495ae853SAndroid Build Coastguard Worker 
87*495ae853SAndroid Build Coastguard Worker         i2_coeff1 = (WORD16) (pi2_ref_data_byte[0]);
88*495ae853SAndroid Build Coastguard Worker 
89*495ae853SAndroid Build Coastguard Worker         /* populate the first inter sample */
90*495ae853SAndroid Build Coastguard Worker         *pi4_ref_array++ = i2_coeff1 << 4;
91*495ae853SAndroid Build Coastguard Worker 
92*495ae853SAndroid Build Coastguard Worker         {
93*495ae853SAndroid Build Coastguard Worker             /* unroll count 1 */
94*495ae853SAndroid Build Coastguard Worker             i2_coeff2 = (WORD16) (pi2_ref_data_byte[2]);
95*495ae853SAndroid Build Coastguard Worker 
96*495ae853SAndroid Build Coastguard Worker             /* populate 2 samples based on current coeffs */
97*495ae853SAndroid Build Coastguard Worker             *pi4_ref_array++ = ((16 - i4_phase2) * i2_coeff1 + i4_phase2 * i2_coeff2);
98*495ae853SAndroid Build Coastguard Worker 
99*495ae853SAndroid Build Coastguard Worker             /* unroll count 2 */
100*495ae853SAndroid Build Coastguard Worker             *pi4_ref_array++ = ((16 - i4_phase1) * i2_coeff1 + i4_phase1 * i2_coeff2);
101*495ae853SAndroid Build Coastguard Worker 
102*495ae853SAndroid Build Coastguard Worker             /* unroll count 3 */
103*495ae853SAndroid Build Coastguard Worker             i2_coeff1 = (WORD16) (pi2_ref_data_byte[4]);
104*495ae853SAndroid Build Coastguard Worker 
105*495ae853SAndroid Build Coastguard Worker             /* populate 2 samples based on current coeffs */
106*495ae853SAndroid Build Coastguard Worker             *pi4_ref_array++ = ((16 - i4_phase2) * i2_coeff2 + i4_phase2 * i2_coeff1);
107*495ae853SAndroid Build Coastguard Worker 
108*495ae853SAndroid Build Coastguard Worker             /* unroll count 4 */
109*495ae853SAndroid Build Coastguard Worker             *pi4_ref_array++ = ((16 - i4_phase1) * i2_coeff2 + i4_phase1 * i2_coeff1);
110*495ae853SAndroid Build Coastguard Worker 
111*495ae853SAndroid Build Coastguard Worker             /* unroll count 5 */
112*495ae853SAndroid Build Coastguard Worker             i2_coeff2 = (WORD16) (pi2_ref_data_byte[6]);
113*495ae853SAndroid Build Coastguard Worker 
114*495ae853SAndroid Build Coastguard Worker             /* populate 2 samples based on current coeffs */
115*495ae853SAndroid Build Coastguard Worker             *pi4_ref_array++ = ((16 - i4_phase2) * i2_coeff1 + i4_phase2 * i2_coeff2);
116*495ae853SAndroid Build Coastguard Worker 
117*495ae853SAndroid Build Coastguard Worker             /* unroll count 6 */
118*495ae853SAndroid Build Coastguard Worker             *pi4_ref_array++ = ((16 - i4_phase1) * i2_coeff1 + i4_phase1 * i2_coeff2);
119*495ae853SAndroid Build Coastguard Worker         }
120*495ae853SAndroid Build Coastguard Worker 
121*495ae853SAndroid Build Coastguard Worker         /* populate the last inter sample */
122*495ae853SAndroid Build Coastguard Worker         *pi4_ref_array++ = i2_coeff2 << 4;
123*495ae853SAndroid Build Coastguard Worker 
124*495ae853SAndroid Build Coastguard Worker         /* vertical loop uopdates */
125*495ae853SAndroid Build Coastguard Worker         pi2_ref_data_byte = pi2_inp_data + ((i4_i + 1) * i4_inp_data_stride);
126*495ae853SAndroid Build Coastguard Worker     }
127*495ae853SAndroid Build Coastguard Worker 
128*495ae853SAndroid Build Coastguard Worker     /* ----------- Vertical Interpolation ---------------- */
129*495ae853SAndroid Build Coastguard Worker     pi4_ref_array = (WORD32 *) ps_scratch->pv_data;
130*495ae853SAndroid Build Coastguard Worker 
131*495ae853SAndroid Build Coastguard Worker     i4_phase1 = ps_ref_array_phases[0].i4_ordinate;
132*495ae853SAndroid Build Coastguard Worker     i4_phase2 = ps_ref_array_phases[2].i4_ordinate;
133*495ae853SAndroid Build Coastguard Worker 
134*495ae853SAndroid Build Coastguard Worker     for(i4_i = 0; i4_i < BLK8x8SIZE; i4_i++)
135*495ae853SAndroid Build Coastguard Worker     {
136*495ae853SAndroid Build Coastguard Worker         WORD16 *pi2_out;
137*495ae853SAndroid Build Coastguard Worker         WORD32 *pi4_ref_array_temp;
138*495ae853SAndroid Build Coastguard Worker         WORD32 i4_horz_samp_1, i4_horz_samp_2;
139*495ae853SAndroid Build Coastguard Worker         pi2_out = pi2_out_res;
140*495ae853SAndroid Build Coastguard Worker         pi4_ref_array_temp = pi4_ref_array;
141*495ae853SAndroid Build Coastguard Worker 
142*495ae853SAndroid Build Coastguard Worker         /* populate the first inter sample */
143*495ae853SAndroid Build Coastguard Worker         i4_horz_samp_1 = *pi4_ref_array_temp;
144*495ae853SAndroid Build Coastguard Worker         pi4_ref_array_temp += BLK8x8SIZE;
145*495ae853SAndroid Build Coastguard Worker         *pi2_out = (i4_horz_samp_1 + 8) >> 4;
146*495ae853SAndroid Build Coastguard Worker         pi2_out += i4_out_res_stride;
147*495ae853SAndroid Build Coastguard Worker 
148*495ae853SAndroid Build Coastguard Worker         {
149*495ae853SAndroid Build Coastguard Worker             /* unroll count 1 */
150*495ae853SAndroid Build Coastguard Worker             i4_horz_samp_2 = *pi4_ref_array_temp;
151*495ae853SAndroid Build Coastguard Worker             pi4_ref_array_temp += BLK8x8SIZE;
152*495ae853SAndroid Build Coastguard Worker 
153*495ae853SAndroid Build Coastguard Worker             /* populate 2 samples based on current coeffs */
154*495ae853SAndroid Build Coastguard Worker             *pi2_out = ((16 - i4_phase2) * i4_horz_samp_1 + i4_phase2 * i4_horz_samp_2 + 128) >> 8;
155*495ae853SAndroid Build Coastguard Worker             pi2_out += i4_out_res_stride;
156*495ae853SAndroid Build Coastguard Worker 
157*495ae853SAndroid Build Coastguard Worker             /* unroll count 2 */
158*495ae853SAndroid Build Coastguard Worker             *pi2_out = ((16 - i4_phase1) * i4_horz_samp_1 + i4_phase1 * i4_horz_samp_2 + 128) >> 8;
159*495ae853SAndroid Build Coastguard Worker             pi2_out += i4_out_res_stride;
160*495ae853SAndroid Build Coastguard Worker 
161*495ae853SAndroid Build Coastguard Worker             /* unroll count 3 */
162*495ae853SAndroid Build Coastguard Worker             i4_horz_samp_1 = *pi4_ref_array_temp;
163*495ae853SAndroid Build Coastguard Worker             pi4_ref_array_temp += BLK8x8SIZE;
164*495ae853SAndroid Build Coastguard Worker 
165*495ae853SAndroid Build Coastguard Worker             /* populate 2 samples based on current coeffs */
166*495ae853SAndroid Build Coastguard Worker             *pi2_out = ((16 - i4_phase2) * i4_horz_samp_2 + i4_phase2 * i4_horz_samp_1 + 128) >> 8;
167*495ae853SAndroid Build Coastguard Worker             pi2_out += i4_out_res_stride;
168*495ae853SAndroid Build Coastguard Worker 
169*495ae853SAndroid Build Coastguard Worker             /* unroll count 4 */
170*495ae853SAndroid Build Coastguard Worker             *pi2_out = ((16 - i4_phase1) * i4_horz_samp_2 + i4_phase1 * i4_horz_samp_1 + 128) >> 8;
171*495ae853SAndroid Build Coastguard Worker             pi2_out += i4_out_res_stride;
172*495ae853SAndroid Build Coastguard Worker 
173*495ae853SAndroid Build Coastguard Worker             /* unroll count 5 */
174*495ae853SAndroid Build Coastguard Worker             i4_horz_samp_2 = *pi4_ref_array_temp;
175*495ae853SAndroid Build Coastguard Worker 
176*495ae853SAndroid Build Coastguard Worker             /* populate 2 samples based on current coeffs */
177*495ae853SAndroid Build Coastguard Worker             *pi2_out = ((16 - i4_phase2) * i4_horz_samp_1 + i4_phase2 * i4_horz_samp_2 + 128) >> 8;
178*495ae853SAndroid Build Coastguard Worker             pi2_out += i4_out_res_stride;
179*495ae853SAndroid Build Coastguard Worker 
180*495ae853SAndroid Build Coastguard Worker             /* unroll count 6 */
181*495ae853SAndroid Build Coastguard Worker             *pi2_out = ((16 - i4_phase1) * i4_horz_samp_1 + i4_phase1 * i4_horz_samp_2 + 128) >> 8;
182*495ae853SAndroid Build Coastguard Worker             pi2_out += i4_out_res_stride;
183*495ae853SAndroid Build Coastguard Worker         }
184*495ae853SAndroid Build Coastguard Worker 
185*495ae853SAndroid Build Coastguard Worker         /* populate the last inter sample */
186*495ae853SAndroid Build Coastguard Worker         *pi2_out = (i4_horz_samp_2 + 8) >> 4;
187*495ae853SAndroid Build Coastguard Worker 
188*495ae853SAndroid Build Coastguard Worker         /* horizontal loop updates */
189*495ae853SAndroid Build Coastguard Worker         pi4_ref_array++;
190*495ae853SAndroid Build Coastguard Worker         pi2_out_res += 2;
191*495ae853SAndroid Build Coastguard Worker     }
192*495ae853SAndroid Build Coastguard Worker }
193*495ae853SAndroid Build Coastguard Worker 
isvce_luma_residual_sampler_2x(coordinates_t * ps_ref_array_positions,coordinates_t * ps_ref_array_phases,buffer_container_t * ps_inp,buffer_container_t * ps_out,buffer_container_t * ps_scratch,UWORD32 u4_ref_nnz,UWORD8 u1_ref_tx_size)194*495ae853SAndroid Build Coastguard Worker void isvce_luma_residual_sampler_2x(coordinates_t *ps_ref_array_positions,
195*495ae853SAndroid Build Coastguard Worker                                     coordinates_t *ps_ref_array_phases, buffer_container_t *ps_inp,
196*495ae853SAndroid Build Coastguard Worker                                     buffer_container_t *ps_out, buffer_container_t *ps_scratch,
197*495ae853SAndroid Build Coastguard Worker                                     UWORD32 u4_ref_nnz, UWORD8 u1_ref_tx_size)
198*495ae853SAndroid Build Coastguard Worker {
199*495ae853SAndroid Build Coastguard Worker     WORD16 *pi2_inp_data = ps_inp->pv_data;
200*495ae853SAndroid Build Coastguard Worker     WORD16 *pi2_out_res = ps_out->pv_data;
201*495ae853SAndroid Build Coastguard Worker     WORD32 i4_inp_data_stride = ps_inp->i4_data_stride;
202*495ae853SAndroid Build Coastguard Worker     WORD32 i4_out_res_stride = ps_out->i4_data_stride;
203*495ae853SAndroid Build Coastguard Worker     WORD16 *pi2_refarray_buffer = ps_scratch->pv_data;
204*495ae853SAndroid Build Coastguard Worker     WORD32 i4_blk_ctr;
205*495ae853SAndroid Build Coastguard Worker 
206*495ae853SAndroid Build Coastguard Worker     UNUSED(ps_ref_array_positions);
207*495ae853SAndroid Build Coastguard Worker     UNUSED(ps_ref_array_phases);
208*495ae853SAndroid Build Coastguard Worker 
209*495ae853SAndroid Build Coastguard Worker     /* For 2x scaling, offsets always point to TL pixel outside MB */
210*495ae853SAndroid Build Coastguard Worker     /* Hence, refTransBlkIdc will be different and since phase */
211*495ae853SAndroid Build Coastguard Worker     /* for first refArray pos for horiz filtering samples > 8, */
212*495ae853SAndroid Build Coastguard Worker     /* first row and first column from the refArray is never used */
213*495ae853SAndroid Build Coastguard Worker     pi2_inp_data += 1 + i4_inp_data_stride;
214*495ae853SAndroid Build Coastguard Worker 
215*495ae853SAndroid Build Coastguard Worker     if((u1_ref_tx_size) && (0 != u4_ref_nnz))
216*495ae853SAndroid Build Coastguard Worker     {
217*495ae853SAndroid Build Coastguard Worker         WORD16 *pi2_ref_data_byte;
218*495ae853SAndroid Build Coastguard Worker         WORD32 *pi4_ref_array;
219*495ae853SAndroid Build Coastguard Worker         WORD32 i4_i, i4_j;
220*495ae853SAndroid Build Coastguard Worker 
221*495ae853SAndroid Build Coastguard Worker         pi2_ref_data_byte = pi2_inp_data;
222*495ae853SAndroid Build Coastguard Worker 
223*495ae853SAndroid Build Coastguard Worker         /* ----------- Horizontal Interpolation ---------------- */
224*495ae853SAndroid Build Coastguard Worker         pi4_ref_array = (WORD32 *) pi2_refarray_buffer;
225*495ae853SAndroid Build Coastguard Worker 
226*495ae853SAndroid Build Coastguard Worker         for(i4_i = 0; i4_i < BLK8x8SIZE; i4_i++)
227*495ae853SAndroid Build Coastguard Worker         {
228*495ae853SAndroid Build Coastguard Worker             WORD16 i2_coeff1, i2_coeff2;
229*495ae853SAndroid Build Coastguard Worker 
230*495ae853SAndroid Build Coastguard Worker             i2_coeff1 = (WORD16) (*pi2_ref_data_byte++);
231*495ae853SAndroid Build Coastguard Worker 
232*495ae853SAndroid Build Coastguard Worker             /* populate the first inter sample */
233*495ae853SAndroid Build Coastguard Worker             *pi4_ref_array++ = i2_coeff1 << 2;
234*495ae853SAndroid Build Coastguard Worker 
235*495ae853SAndroid Build Coastguard Worker             for(i4_j = 0; i4_j < 14; i4_j += 2)
236*495ae853SAndroid Build Coastguard Worker             {
237*495ae853SAndroid Build Coastguard Worker                 i2_coeff2 = (WORD16) (*pi2_ref_data_byte++);
238*495ae853SAndroid Build Coastguard Worker 
239*495ae853SAndroid Build Coastguard Worker                 /* populate 2 samples based on current coeffs */
240*495ae853SAndroid Build Coastguard Worker                 *pi4_ref_array++ = ((i2_coeff1 << 1) + (i2_coeff1) + (i2_coeff2));
241*495ae853SAndroid Build Coastguard Worker 
242*495ae853SAndroid Build Coastguard Worker                 *pi4_ref_array++ = ((i2_coeff2 << 1) + (i2_coeff2) + (i2_coeff1));
243*495ae853SAndroid Build Coastguard Worker 
244*495ae853SAndroid Build Coastguard Worker                 /* store the coeff 2 to coeff 1 */
245*495ae853SAndroid Build Coastguard Worker                 /* (used in next iteration)     */
246*495ae853SAndroid Build Coastguard Worker                 i2_coeff1 = i2_coeff2;
247*495ae853SAndroid Build Coastguard Worker             }
248*495ae853SAndroid Build Coastguard Worker 
249*495ae853SAndroid Build Coastguard Worker             /* populate the last inter sample */
250*495ae853SAndroid Build Coastguard Worker             *pi4_ref_array++ = i2_coeff1 << 2;
251*495ae853SAndroid Build Coastguard Worker 
252*495ae853SAndroid Build Coastguard Worker             /* vertical loop uopdates */
253*495ae853SAndroid Build Coastguard Worker             pi2_ref_data_byte = pi2_inp_data + ((i4_i + 1) * i4_inp_data_stride);
254*495ae853SAndroid Build Coastguard Worker         }
255*495ae853SAndroid Build Coastguard Worker 
256*495ae853SAndroid Build Coastguard Worker         /* ----------- Vertical Interpolation ---------------- */
257*495ae853SAndroid Build Coastguard Worker         pi4_ref_array = (WORD32 *) pi2_refarray_buffer;
258*495ae853SAndroid Build Coastguard Worker 
259*495ae853SAndroid Build Coastguard Worker         for(i4_i = 0; i4_i < MB_SIZE; i4_i++)
260*495ae853SAndroid Build Coastguard Worker         {
261*495ae853SAndroid Build Coastguard Worker             WORD32 *pi4_ref_array_temp;
262*495ae853SAndroid Build Coastguard Worker             WORD16 *pi2_out;
263*495ae853SAndroid Build Coastguard Worker             WORD32 i4_horz_samp_1, i4_horz_samp_2;
264*495ae853SAndroid Build Coastguard Worker 
265*495ae853SAndroid Build Coastguard Worker             pi4_ref_array_temp = pi4_ref_array;
266*495ae853SAndroid Build Coastguard Worker             pi2_out = pi2_out_res;
267*495ae853SAndroid Build Coastguard Worker             i4_horz_samp_1 = *pi4_ref_array_temp;
268*495ae853SAndroid Build Coastguard Worker 
269*495ae853SAndroid Build Coastguard Worker             /* populate the first inter sample */
270*495ae853SAndroid Build Coastguard Worker             *pi2_out = (i4_horz_samp_1 + 2) >> 2;
271*495ae853SAndroid Build Coastguard Worker             pi2_out += i4_out_res_stride;
272*495ae853SAndroid Build Coastguard Worker 
273*495ae853SAndroid Build Coastguard Worker             for(i4_j = 0; i4_j < 14; i4_j += 2)
274*495ae853SAndroid Build Coastguard Worker             {
275*495ae853SAndroid Build Coastguard Worker                 pi4_ref_array_temp += MB_SIZE;
276*495ae853SAndroid Build Coastguard Worker                 i4_horz_samp_2 = *pi4_ref_array_temp;
277*495ae853SAndroid Build Coastguard Worker 
278*495ae853SAndroid Build Coastguard Worker                 /* populate 2 samples based on current coeffs */
279*495ae853SAndroid Build Coastguard Worker                 *pi2_out = ((i4_horz_samp_1 << 1) + (i4_horz_samp_1) + (i4_horz_samp_2) + 8) >> 4;
280*495ae853SAndroid Build Coastguard Worker                 pi2_out += i4_out_res_stride;
281*495ae853SAndroid Build Coastguard Worker 
282*495ae853SAndroid Build Coastguard Worker                 *pi2_out = ((i4_horz_samp_2 << 1) + (i4_horz_samp_2) + (i4_horz_samp_1) + 8) >> 4;
283*495ae853SAndroid Build Coastguard Worker                 pi2_out += i4_out_res_stride;
284*495ae853SAndroid Build Coastguard Worker 
285*495ae853SAndroid Build Coastguard Worker                 /* store the coeff 2 to coeff 1 */
286*495ae853SAndroid Build Coastguard Worker                 /* (used in next iteration)     */
287*495ae853SAndroid Build Coastguard Worker                 i4_horz_samp_1 = i4_horz_samp_2;
288*495ae853SAndroid Build Coastguard Worker             }
289*495ae853SAndroid Build Coastguard Worker 
290*495ae853SAndroid Build Coastguard Worker             /* populate the first inter sample */
291*495ae853SAndroid Build Coastguard Worker             *pi2_out = (i4_horz_samp_1 + 2) >> 2;
292*495ae853SAndroid Build Coastguard Worker 
293*495ae853SAndroid Build Coastguard Worker             /* horizontal loop updates */
294*495ae853SAndroid Build Coastguard Worker             pi4_ref_array++;
295*495ae853SAndroid Build Coastguard Worker             pi2_out_res++;
296*495ae853SAndroid Build Coastguard Worker         }
297*495ae853SAndroid Build Coastguard Worker     }
298*495ae853SAndroid Build Coastguard Worker     else
299*495ae853SAndroid Build Coastguard Worker     {
300*495ae853SAndroid Build Coastguard Worker         /* ----------------------------------------------------------------- */
301*495ae853SAndroid Build Coastguard Worker         /* LOOP over number of blocks                                        */
302*495ae853SAndroid Build Coastguard Worker         /* ----------------------------------------------------------------- */
303*495ae853SAndroid Build Coastguard Worker         for(i4_blk_ctr = 0; i4_blk_ctr < BLK_SIZE; i4_blk_ctr++)
304*495ae853SAndroid Build Coastguard Worker         {
305*495ae853SAndroid Build Coastguard Worker             WORD16 *pi2_ref_data_byte;
306*495ae853SAndroid Build Coastguard Worker             WORD32 *pi4_ref_array;
307*495ae853SAndroid Build Coastguard Worker             WORD32 i4_i;
308*495ae853SAndroid Build Coastguard Worker 
309*495ae853SAndroid Build Coastguard Worker             /* if reference layer is not coded then no processing */
310*495ae853SAndroid Build Coastguard Worker             if(0 != (u4_ref_nnz & 0x1))
311*495ae853SAndroid Build Coastguard Worker             {
312*495ae853SAndroid Build Coastguard Worker                 pi2_ref_data_byte = pi2_inp_data;
313*495ae853SAndroid Build Coastguard Worker 
314*495ae853SAndroid Build Coastguard Worker                 /* ----------- Horizontal Interpolation ---------------- */
315*495ae853SAndroid Build Coastguard Worker                 pi4_ref_array = (WORD32 *) pi2_refarray_buffer;
316*495ae853SAndroid Build Coastguard Worker 
317*495ae853SAndroid Build Coastguard Worker                 for(i4_i = 0; i4_i < BLK_SIZE; i4_i++)
318*495ae853SAndroid Build Coastguard Worker                 {
319*495ae853SAndroid Build Coastguard Worker                     WORD16 i2_coeff1, i2_coeff2;
320*495ae853SAndroid Build Coastguard Worker 
321*495ae853SAndroid Build Coastguard Worker                     i2_coeff1 = (WORD16) (*pi2_ref_data_byte++);
322*495ae853SAndroid Build Coastguard Worker 
323*495ae853SAndroid Build Coastguard Worker                     /* populate the first inter sample */
324*495ae853SAndroid Build Coastguard Worker                     *pi4_ref_array++ = i2_coeff1 << 2;
325*495ae853SAndroid Build Coastguard Worker 
326*495ae853SAndroid Build Coastguard Worker                     {
327*495ae853SAndroid Build Coastguard Worker                         i2_coeff2 = (WORD16) (*pi2_ref_data_byte++);
328*495ae853SAndroid Build Coastguard Worker 
329*495ae853SAndroid Build Coastguard Worker                         /* populate 2 samples based on current coeffs */
330*495ae853SAndroid Build Coastguard Worker                         *pi4_ref_array++ = ((i2_coeff1 << 1) + (i2_coeff1) + (i2_coeff2));
331*495ae853SAndroid Build Coastguard Worker 
332*495ae853SAndroid Build Coastguard Worker                         *pi4_ref_array++ = ((i2_coeff2 << 1) + (i2_coeff2) + (i2_coeff1));
333*495ae853SAndroid Build Coastguard Worker 
334*495ae853SAndroid Build Coastguard Worker                         i2_coeff1 = (WORD16) (*pi2_ref_data_byte++);
335*495ae853SAndroid Build Coastguard Worker 
336*495ae853SAndroid Build Coastguard Worker                         /* populate 2 samples based on current coeffs */
337*495ae853SAndroid Build Coastguard Worker                         *pi4_ref_array++ = ((i2_coeff2 << 1) + (i2_coeff2) + (i2_coeff1));
338*495ae853SAndroid Build Coastguard Worker 
339*495ae853SAndroid Build Coastguard Worker                         *pi4_ref_array++ = ((i2_coeff1 << 1) + (i2_coeff1) + (i2_coeff2));
340*495ae853SAndroid Build Coastguard Worker 
341*495ae853SAndroid Build Coastguard Worker                         i2_coeff2 = (WORD16) (*pi2_ref_data_byte++);
342*495ae853SAndroid Build Coastguard Worker 
343*495ae853SAndroid Build Coastguard Worker                         /* populate 2 samples based on current coeffs */
344*495ae853SAndroid Build Coastguard Worker                         *pi4_ref_array++ = ((i2_coeff1 << 1) + (i2_coeff1) + (i2_coeff2));
345*495ae853SAndroid Build Coastguard Worker 
346*495ae853SAndroid Build Coastguard Worker                         *pi4_ref_array++ = ((i2_coeff2 << 1) + (i2_coeff2) + (i2_coeff1));
347*495ae853SAndroid Build Coastguard Worker                     }
348*495ae853SAndroid Build Coastguard Worker 
349*495ae853SAndroid Build Coastguard Worker                     /* populate the last inter sample */
350*495ae853SAndroid Build Coastguard Worker                     *pi4_ref_array++ = i2_coeff2 << 2;
351*495ae853SAndroid Build Coastguard Worker 
352*495ae853SAndroid Build Coastguard Worker                     /* vertical loop uopdates */
353*495ae853SAndroid Build Coastguard Worker                     pi2_ref_data_byte = pi2_inp_data + ((i4_i + 1) * i4_inp_data_stride);
354*495ae853SAndroid Build Coastguard Worker                 }
355*495ae853SAndroid Build Coastguard Worker 
356*495ae853SAndroid Build Coastguard Worker                 /* ----------- Vertical Interpolation ---------------- */
357*495ae853SAndroid Build Coastguard Worker                 pi4_ref_array = (WORD32 *) pi2_refarray_buffer;
358*495ae853SAndroid Build Coastguard Worker 
359*495ae853SAndroid Build Coastguard Worker                 for(i4_i = 0; i4_i < BLK8x8SIZE; i4_i++)
360*495ae853SAndroid Build Coastguard Worker                 {
361*495ae853SAndroid Build Coastguard Worker                     WORD32 *pi4_ref_array_temp;
362*495ae853SAndroid Build Coastguard Worker                     WORD16 *pi2_out;
363*495ae853SAndroid Build Coastguard Worker                     WORD32 i4_horz_samp_1, i4_horz_samp_2;
364*495ae853SAndroid Build Coastguard Worker 
365*495ae853SAndroid Build Coastguard Worker                     pi4_ref_array_temp = pi4_ref_array;
366*495ae853SAndroid Build Coastguard Worker                     pi2_out = pi2_out_res;
367*495ae853SAndroid Build Coastguard Worker                     i4_horz_samp_1 = *pi4_ref_array_temp;
368*495ae853SAndroid Build Coastguard Worker 
369*495ae853SAndroid Build Coastguard Worker                     /* populate the first inter sample */
370*495ae853SAndroid Build Coastguard Worker                     *pi2_out = (i4_horz_samp_1 + 2) >> 2;
371*495ae853SAndroid Build Coastguard Worker                     pi2_out += i4_out_res_stride;
372*495ae853SAndroid Build Coastguard Worker 
373*495ae853SAndroid Build Coastguard Worker                     {
374*495ae853SAndroid Build Coastguard Worker                         /* unroll loop count 1 */
375*495ae853SAndroid Build Coastguard Worker                         pi4_ref_array_temp += BLK8x8SIZE;
376*495ae853SAndroid Build Coastguard Worker                         i4_horz_samp_2 = *pi4_ref_array_temp;
377*495ae853SAndroid Build Coastguard Worker 
378*495ae853SAndroid Build Coastguard Worker                         /* populate 2 samples based on current coeffs */
379*495ae853SAndroid Build Coastguard Worker                         *pi2_out =
380*495ae853SAndroid Build Coastguard Worker                             ((i4_horz_samp_1 << 1) + (i4_horz_samp_1) + (i4_horz_samp_2) + 8) >> 4;
381*495ae853SAndroid Build Coastguard Worker                         pi2_out += i4_out_res_stride;
382*495ae853SAndroid Build Coastguard Worker 
383*495ae853SAndroid Build Coastguard Worker                         *pi2_out =
384*495ae853SAndroid Build Coastguard Worker                             ((i4_horz_samp_2 << 1) + (i4_horz_samp_2) + (i4_horz_samp_1) + 8) >> 4;
385*495ae853SAndroid Build Coastguard Worker                         pi2_out += i4_out_res_stride;
386*495ae853SAndroid Build Coastguard Worker 
387*495ae853SAndroid Build Coastguard Worker                         /* unroll loop count 2 */
388*495ae853SAndroid Build Coastguard Worker                         pi4_ref_array_temp += BLK8x8SIZE;
389*495ae853SAndroid Build Coastguard Worker                         i4_horz_samp_1 = *pi4_ref_array_temp;
390*495ae853SAndroid Build Coastguard Worker 
391*495ae853SAndroid Build Coastguard Worker                         /* populate 2 samples based on current coeffs */
392*495ae853SAndroid Build Coastguard Worker                         *pi2_out =
393*495ae853SAndroid Build Coastguard Worker                             ((i4_horz_samp_2 << 1) + (i4_horz_samp_2) + (i4_horz_samp_1) + 8) >> 4;
394*495ae853SAndroid Build Coastguard Worker                         pi2_out += i4_out_res_stride;
395*495ae853SAndroid Build Coastguard Worker 
396*495ae853SAndroid Build Coastguard Worker                         *pi2_out =
397*495ae853SAndroid Build Coastguard Worker                             ((i4_horz_samp_1 << 1) + (i4_horz_samp_1) + (i4_horz_samp_2) + 8) >> 4;
398*495ae853SAndroid Build Coastguard Worker                         pi2_out += i4_out_res_stride;
399*495ae853SAndroid Build Coastguard Worker 
400*495ae853SAndroid Build Coastguard Worker                         /* unroll loop count 3 */
401*495ae853SAndroid Build Coastguard Worker                         pi4_ref_array_temp += BLK8x8SIZE;
402*495ae853SAndroid Build Coastguard Worker                         i4_horz_samp_2 = *pi4_ref_array_temp;
403*495ae853SAndroid Build Coastguard Worker 
404*495ae853SAndroid Build Coastguard Worker                         /* populate 2 samples based on current coeffs */
405*495ae853SAndroid Build Coastguard Worker                         *pi2_out =
406*495ae853SAndroid Build Coastguard Worker                             ((i4_horz_samp_1 << 1) + (i4_horz_samp_1) + (i4_horz_samp_2) + 8) >> 4;
407*495ae853SAndroid Build Coastguard Worker                         pi2_out += i4_out_res_stride;
408*495ae853SAndroid Build Coastguard Worker 
409*495ae853SAndroid Build Coastguard Worker                         *pi2_out =
410*495ae853SAndroid Build Coastguard Worker                             ((i4_horz_samp_2 << 1) + (i4_horz_samp_2) + (i4_horz_samp_1) + 8) >> 4;
411*495ae853SAndroid Build Coastguard Worker                         pi2_out += i4_out_res_stride;
412*495ae853SAndroid Build Coastguard Worker                     }
413*495ae853SAndroid Build Coastguard Worker 
414*495ae853SAndroid Build Coastguard Worker                     /* populate the last inter sample */
415*495ae853SAndroid Build Coastguard Worker                     *pi2_out = (i4_horz_samp_2 + 2) >> 2;
416*495ae853SAndroid Build Coastguard Worker 
417*495ae853SAndroid Build Coastguard Worker                     /* horizontal loop updates */
418*495ae853SAndroid Build Coastguard Worker                     pi4_ref_array++;
419*495ae853SAndroid Build Coastguard Worker                     pi2_out_res++;
420*495ae853SAndroid Build Coastguard Worker                 }
421*495ae853SAndroid Build Coastguard Worker             }
422*495ae853SAndroid Build Coastguard Worker             else
423*495ae853SAndroid Build Coastguard Worker             {
424*495ae853SAndroid Build Coastguard Worker                 pi2_out_res += BLK8x8SIZE;
425*495ae853SAndroid Build Coastguard Worker             }
426*495ae853SAndroid Build Coastguard Worker 
427*495ae853SAndroid Build Coastguard Worker             if(1 == i4_blk_ctr)
428*495ae853SAndroid Build Coastguard Worker             {
429*495ae853SAndroid Build Coastguard Worker                 pi2_inp_data -= BLK_SIZE;
430*495ae853SAndroid Build Coastguard Worker                 pi2_inp_data += (i4_inp_data_stride * BLK_SIZE);
431*495ae853SAndroid Build Coastguard Worker                 pi2_out_res -= MB_SIZE;
432*495ae853SAndroid Build Coastguard Worker                 pi2_out_res += (i4_out_res_stride * BLK8x8SIZE);
433*495ae853SAndroid Build Coastguard Worker                 u4_ref_nnz >>= 2;
434*495ae853SAndroid Build Coastguard Worker             }
435*495ae853SAndroid Build Coastguard Worker             else
436*495ae853SAndroid Build Coastguard Worker             {
437*495ae853SAndroid Build Coastguard Worker                 pi2_inp_data += BLK_SIZE;
438*495ae853SAndroid Build Coastguard Worker             }
439*495ae853SAndroid Build Coastguard Worker 
440*495ae853SAndroid Build Coastguard Worker             u4_ref_nnz >>= 1;
441*495ae853SAndroid Build Coastguard Worker         }
442*495ae853SAndroid Build Coastguard Worker     }
443*495ae853SAndroid Build Coastguard Worker }
444*495ae853SAndroid Build Coastguard Worker 
445*495ae853SAndroid Build Coastguard Worker /**
446*495ae853SAndroid Build Coastguard Worker *******************************************************************************
447*495ae853SAndroid Build Coastguard Worker *
448*495ae853SAndroid Build Coastguard Worker * @brief
449*495ae853SAndroid Build Coastguard Worker *  Returns size of buffers for storing residual pred ctxt
450*495ae853SAndroid Build Coastguard Worker *
451*495ae853SAndroid Build Coastguard Worker * @param[in] u1_num_spatial_layers
452*495ae853SAndroid Build Coastguard Worker *  Num Spatial Layers
453*495ae853SAndroid Build Coastguard Worker *
454*495ae853SAndroid Build Coastguard Worker * @param[in] d_spatial_res_ratio
455*495ae853SAndroid Build Coastguard Worker *  Resolution Ratio b/w spatial layers
456*495ae853SAndroid Build Coastguard Worker *
457*495ae853SAndroid Build Coastguard Worker * @param[in] u4_wd
458*495ae853SAndroid Build Coastguard Worker *  Input Width
459*495ae853SAndroid Build Coastguard Worker *
460*495ae853SAndroid Build Coastguard Worker * @param[in] u4_ht
461*495ae853SAndroid Build Coastguard Worker *  Input Height
462*495ae853SAndroid Build Coastguard Worker *
463*495ae853SAndroid Build Coastguard Worker * @returns  Size of buffers
464*495ae853SAndroid Build Coastguard Worker *
465*495ae853SAndroid Build Coastguard Worker *******************************************************************************
466*495ae853SAndroid Build Coastguard Worker */
isvce_get_svc_res_pred_ctxt_size(UWORD8 u1_num_spatial_layers,DOUBLE d_spatial_res_ratio,UWORD32 u4_wd,UWORD32 u4_ht)467*495ae853SAndroid Build Coastguard Worker UWORD32 isvce_get_svc_res_pred_ctxt_size(UWORD8 u1_num_spatial_layers, DOUBLE d_spatial_res_ratio,
468*495ae853SAndroid Build Coastguard Worker                                          UWORD32 u4_wd, UWORD32 u4_ht)
469*495ae853SAndroid Build Coastguard Worker {
470*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_size = 0;
471*495ae853SAndroid Build Coastguard Worker 
472*495ae853SAndroid Build Coastguard Worker     if(u1_num_spatial_layers > 1)
473*495ae853SAndroid Build Coastguard Worker     {
474*495ae853SAndroid Build Coastguard Worker         WORD32 i;
475*495ae853SAndroid Build Coastguard Worker 
476*495ae853SAndroid Build Coastguard Worker         u4_size += MAX_PROCESS_CTXT * sizeof(svc_res_pred_ctxt_t);
477*495ae853SAndroid Build Coastguard Worker         u4_size += MAX_PROCESS_CTXT * sizeof(res_pred_state_t);
478*495ae853SAndroid Build Coastguard Worker 
479*495ae853SAndroid Build Coastguard Worker         /* Mem for storing pred */
480*495ae853SAndroid Build Coastguard Worker         u4_size += MAX_PROCESS_CTXT * MB_SIZE * MB_SIZE * sizeof(WORD16);
481*495ae853SAndroid Build Coastguard Worker         u4_size += MAX_PROCESS_CTXT * MB_SIZE * (MB_SIZE / 2) * sizeof(WORD16);
482*495ae853SAndroid Build Coastguard Worker 
483*495ae853SAndroid Build Coastguard Worker         /* Mem for storing intermediates */
484*495ae853SAndroid Build Coastguard Worker         u4_size += MAX_PROCESS_CTXT * REF_ARRAY_MAX_WIDTH * REF_ARRAY_MAX_HEIGHT * sizeof(WORD16);
485*495ae853SAndroid Build Coastguard Worker 
486*495ae853SAndroid Build Coastguard Worker         /* Mem for pu1_ref_x_ptr_incr and pu1_ref_y_ptr_incr*/
487*495ae853SAndroid Build Coastguard Worker         u4_size +=
488*495ae853SAndroid Build Coastguard Worker             2 * MAX_PROCESS_CTXT * REF_ARRAY_MAX_WIDTH * REF_ARRAY_MAX_HEIGHT * sizeof(UWORD8);
489*495ae853SAndroid Build Coastguard Worker 
490*495ae853SAndroid Build Coastguard Worker         u4_size += MAX_PROCESS_CTXT * u1_num_spatial_layers * sizeof(res_pred_layer_state_t);
491*495ae853SAndroid Build Coastguard Worker 
492*495ae853SAndroid Build Coastguard Worker         for(i = u1_num_spatial_layers - 1; i >= 1; i--)
493*495ae853SAndroid Build Coastguard Worker         {
494*495ae853SAndroid Build Coastguard Worker             WORD32 i4_layer_luma_wd =
495*495ae853SAndroid Build Coastguard Worker                 (WORD32) ((DOUBLE) u4_wd /
496*495ae853SAndroid Build Coastguard Worker                           pow(d_spatial_res_ratio, u1_num_spatial_layers - 1 - i)) +
497*495ae853SAndroid Build Coastguard Worker                 0.99;
498*495ae853SAndroid Build Coastguard Worker             WORD32 i4_layer_luma_ht =
499*495ae853SAndroid Build Coastguard Worker                 ((DOUBLE) u4_ht / pow(d_spatial_res_ratio, u1_num_spatial_layers - 1 - i)) + 0.99;
500*495ae853SAndroid Build Coastguard Worker             WORD32 i4_layer_luma_mbs = (i4_layer_luma_wd / MB_SIZE) * (i4_layer_luma_ht / MB_SIZE);
501*495ae853SAndroid Build Coastguard Worker             WORD32 i4_layer_u_wd = i4_layer_luma_wd / 2.0 + 0.99;
502*495ae853SAndroid Build Coastguard Worker             WORD32 i4_layer_u_ht = i4_layer_luma_ht / 2.0 + 0.99;
503*495ae853SAndroid Build Coastguard Worker             WORD32 i4_layer_u_mbs =
504*495ae853SAndroid Build Coastguard Worker                 (i4_layer_u_wd / (MB_SIZE / 2)) * (i4_layer_u_ht / (MB_SIZE / 2));
505*495ae853SAndroid Build Coastguard Worker 
506*495ae853SAndroid Build Coastguard Worker             /* ps_luma_mb_states */
507*495ae853SAndroid Build Coastguard Worker             {
508*495ae853SAndroid Build Coastguard Worker                 u4_size += i4_layer_luma_mbs * sizeof(res_pred_mb_state_t);
509*495ae853SAndroid Build Coastguard Worker 
510*495ae853SAndroid Build Coastguard Worker                 /* ps_ref_array_positions */
511*495ae853SAndroid Build Coastguard Worker                 u4_size +=
512*495ae853SAndroid Build Coastguard Worker                     ((1.5 == d_spatial_res_ratio) ? (i4_layer_luma_mbs * MB_SIZE * MB_SIZE) : 0) *
513*495ae853SAndroid Build Coastguard Worker                     sizeof(coordinates_t);
514*495ae853SAndroid Build Coastguard Worker 
515*495ae853SAndroid Build Coastguard Worker                 /* ps_ref_array_phases */
516*495ae853SAndroid Build Coastguard Worker                 u4_size += ((1.5 == d_spatial_res_ratio) ? (i4_layer_luma_mbs * 5) : 0) *
517*495ae853SAndroid Build Coastguard Worker                            sizeof(coordinates_t);
518*495ae853SAndroid Build Coastguard Worker             }
519*495ae853SAndroid Build Coastguard Worker 
520*495ae853SAndroid Build Coastguard Worker             /* ps_chroma_mb_states */
521*495ae853SAndroid Build Coastguard Worker             {
522*495ae853SAndroid Build Coastguard Worker                 u4_size += i4_layer_u_mbs * sizeof(res_pred_mb_state_t);
523*495ae853SAndroid Build Coastguard Worker 
524*495ae853SAndroid Build Coastguard Worker                 /* ps_ref_array_positions */
525*495ae853SAndroid Build Coastguard Worker                 u4_size +=
526*495ae853SAndroid Build Coastguard Worker                     ((1.5 == d_spatial_res_ratio) ? (i4_layer_u_mbs * (MB_SIZE / 2) * (MB_SIZE / 2))
527*495ae853SAndroid Build Coastguard Worker                                                   : 0) *
528*495ae853SAndroid Build Coastguard Worker                     sizeof(coordinates_t);
529*495ae853SAndroid Build Coastguard Worker 
530*495ae853SAndroid Build Coastguard Worker                 /* ps_ref_array_phases */
531*495ae853SAndroid Build Coastguard Worker                 u4_size += ((1.5 == d_spatial_res_ratio) ? (i4_layer_u_mbs * 5) : 3) *
532*495ae853SAndroid Build Coastguard Worker                            sizeof(coordinates_t);
533*495ae853SAndroid Build Coastguard Worker             }
534*495ae853SAndroid Build Coastguard Worker         }
535*495ae853SAndroid Build Coastguard Worker 
536*495ae853SAndroid Build Coastguard Worker         for(i = u1_num_spatial_layers - 1; i >= 0; i--)
537*495ae853SAndroid Build Coastguard Worker         {
538*495ae853SAndroid Build Coastguard Worker             WORD32 i4_layer_luma_wd =
539*495ae853SAndroid Build Coastguard Worker                 (WORD32) ((DOUBLE) u4_wd /
540*495ae853SAndroid Build Coastguard Worker                           pow(d_spatial_res_ratio, u1_num_spatial_layers - 1 - i)) +
541*495ae853SAndroid Build Coastguard Worker                 0.99;
542*495ae853SAndroid Build Coastguard Worker             WORD32 i4_layer_luma_ht =
543*495ae853SAndroid Build Coastguard Worker                 ((DOUBLE) u4_ht / pow(d_spatial_res_ratio, u1_num_spatial_layers - 1 - i)) + 0.99;
544*495ae853SAndroid Build Coastguard Worker             WORD32 i4_layer_luma_mbs =
545*495ae853SAndroid Build Coastguard Worker                 ((i4_layer_luma_wd / MB_SIZE) + 2) * ((i4_layer_luma_ht / MB_SIZE) + 2);
546*495ae853SAndroid Build Coastguard Worker 
547*495ae853SAndroid Build Coastguard Worker             /* pi1_mb_mode */
548*495ae853SAndroid Build Coastguard Worker             u4_size += i4_layer_luma_mbs * sizeof(WORD8);
549*495ae853SAndroid Build Coastguard Worker         }
550*495ae853SAndroid Build Coastguard Worker     }
551*495ae853SAndroid Build Coastguard Worker     else
552*495ae853SAndroid Build Coastguard Worker     {
553*495ae853SAndroid Build Coastguard Worker         u4_size += MAX_PROCESS_CTXT * sizeof(yuv_buf_props_t);
554*495ae853SAndroid Build Coastguard Worker 
555*495ae853SAndroid Build Coastguard Worker         /* Mem for storing pred */
556*495ae853SAndroid Build Coastguard Worker         u4_size += MAX_PROCESS_CTXT * MB_SIZE * MB_SIZE * sizeof(WORD16);
557*495ae853SAndroid Build Coastguard Worker         u4_size += MAX_PROCESS_CTXT * MB_SIZE * (MB_SIZE / 2) * sizeof(WORD16);
558*495ae853SAndroid Build Coastguard Worker     }
559*495ae853SAndroid Build Coastguard Worker 
560*495ae853SAndroid Build Coastguard Worker     return u4_size;
561*495ae853SAndroid Build Coastguard Worker }
562*495ae853SAndroid Build Coastguard Worker 
isvce_get_scaled_pixel_pos(layer_resampler_props_t * ps_layer_props,WORD32 i4_pixel_pos,UWORD8 u1_dim_id)563*495ae853SAndroid Build Coastguard Worker static FORCEINLINE WORD32 isvce_get_scaled_pixel_pos(layer_resampler_props_t *ps_layer_props,
564*495ae853SAndroid Build Coastguard Worker                                                      WORD32 i4_pixel_pos, UWORD8 u1_dim_id)
565*495ae853SAndroid Build Coastguard Worker {
566*495ae853SAndroid Build Coastguard Worker     if(1 == u1_dim_id)
567*495ae853SAndroid Build Coastguard Worker     {
568*495ae853SAndroid Build Coastguard Worker         return (((i4_pixel_pos - ps_layer_props->i4_offset_y) *
569*495ae853SAndroid Build Coastguard Worker                      ((WORD64) ps_layer_props->u4_scale_y) +
570*495ae853SAndroid Build Coastguard Worker                  ps_layer_props->i4_add_y) >>
571*495ae853SAndroid Build Coastguard Worker                 (ps_layer_props->u4_shift_y - 4)) -
572*495ae853SAndroid Build Coastguard Worker                ps_layer_props->i4_delta_y;
573*495ae853SAndroid Build Coastguard Worker     }
574*495ae853SAndroid Build Coastguard Worker     else
575*495ae853SAndroid Build Coastguard Worker     {
576*495ae853SAndroid Build Coastguard Worker         return (((i4_pixel_pos - ps_layer_props->i4_offset_x) *
577*495ae853SAndroid Build Coastguard Worker                      ((WORD64) ps_layer_props->u4_scale_x) +
578*495ae853SAndroid Build Coastguard Worker                  ps_layer_props->i4_add_x) >>
579*495ae853SAndroid Build Coastguard Worker                 (ps_layer_props->u4_shift_x - 4)) -
580*495ae853SAndroid Build Coastguard Worker                ps_layer_props->i4_delta_x;
581*495ae853SAndroid Build Coastguard Worker     }
582*495ae853SAndroid Build Coastguard Worker }
583*495ae853SAndroid Build Coastguard Worker 
isvce_ref_array_pos_and_phase_init_dyadic(layer_resampler_props_t * ps_layer_props,res_pred_mb_state_t * ps_mb_state,coordinates_t * ps_mb_pos,UWORD8 u1_frame_mbs_only_flag,UWORD8 u1_field_mb_flag,UWORD8 u1_ref_layer_frame_mbs_only_flag)584*495ae853SAndroid Build Coastguard Worker static FORCEINLINE void isvce_ref_array_pos_and_phase_init_dyadic(
585*495ae853SAndroid Build Coastguard Worker     layer_resampler_props_t *ps_layer_props, res_pred_mb_state_t *ps_mb_state,
586*495ae853SAndroid Build Coastguard Worker     coordinates_t *ps_mb_pos, UWORD8 u1_frame_mbs_only_flag, UWORD8 u1_field_mb_flag,
587*495ae853SAndroid Build Coastguard Worker     UWORD8 u1_ref_layer_frame_mbs_only_flag)
588*495ae853SAndroid Build Coastguard Worker {
589*495ae853SAndroid Build Coastguard Worker     UWORD32 i, j;
590*495ae853SAndroid Build Coastguard Worker 
591*495ae853SAndroid Build Coastguard Worker     coordinates_t *ps_ref_array_phases = ps_mb_state->ps_ref_array_phases;
592*495ae853SAndroid Build Coastguard Worker 
593*495ae853SAndroid Build Coastguard Worker     WORD32 i4_x_offset = ps_mb_state->s_offsets.i4_abscissa;
594*495ae853SAndroid Build Coastguard Worker     WORD32 i4_y_offset = ps_mb_state->s_offsets.i4_ordinate;
595*495ae853SAndroid Build Coastguard Worker 
596*495ae853SAndroid Build Coastguard Worker     for(i = 0; i < 2; i++)
597*495ae853SAndroid Build Coastguard Worker     {
598*495ae853SAndroid Build Coastguard Worker         WORD32 i4_y_ref16;
599*495ae853SAndroid Build Coastguard Worker 
600*495ae853SAndroid Build Coastguard Worker         WORD32 i4_yc = ps_mb_pos->i4_ordinate * ps_layer_props->u4_mb_ht + i;
601*495ae853SAndroid Build Coastguard Worker 
602*495ae853SAndroid Build Coastguard Worker         if((0 == u1_frame_mbs_only_flag) || (0 == u1_ref_layer_frame_mbs_only_flag))
603*495ae853SAndroid Build Coastguard Worker         {
604*495ae853SAndroid Build Coastguard Worker             i4_yc = i4_yc >> (1 - u1_field_mb_flag);
605*495ae853SAndroid Build Coastguard Worker         }
606*495ae853SAndroid Build Coastguard Worker 
607*495ae853SAndroid Build Coastguard Worker         i4_y_ref16 = isvce_get_scaled_pixel_pos(ps_layer_props, i4_yc, 1);
608*495ae853SAndroid Build Coastguard Worker 
609*495ae853SAndroid Build Coastguard Worker         for(j = 0; j < ((0 == i) ? 2 : 1); j++)
610*495ae853SAndroid Build Coastguard Worker         {
611*495ae853SAndroid Build Coastguard Worker             WORD32 i4_x_ref16;
612*495ae853SAndroid Build Coastguard Worker 
613*495ae853SAndroid Build Coastguard Worker             WORD32 i4_xc = ps_mb_pos->i4_abscissa * ps_layer_props->u4_mb_wd + j;
614*495ae853SAndroid Build Coastguard Worker 
615*495ae853SAndroid Build Coastguard Worker             i4_x_ref16 = isvce_get_scaled_pixel_pos(ps_layer_props, i4_xc, 0);
616*495ae853SAndroid Build Coastguard Worker 
617*495ae853SAndroid Build Coastguard Worker             ps_ref_array_phases[j + i * 2].i4_abscissa = (i4_x_ref16 - (16 * i4_x_offset)) & 15;
618*495ae853SAndroid Build Coastguard Worker             ps_ref_array_phases[j + i * 2].i4_ordinate = (i4_y_ref16 - (16 * i4_y_offset)) & 15;
619*495ae853SAndroid Build Coastguard Worker         }
620*495ae853SAndroid Build Coastguard Worker     }
621*495ae853SAndroid Build Coastguard Worker }
622*495ae853SAndroid Build Coastguard Worker 
isvce_ref_array_pos_and_phase_init(layer_resampler_props_t * ps_layer_props,res_pred_mb_state_t * ps_mb_state,coordinates_t * ps_mb_pos,UWORD8 u1_frame_mbs_only_flag,UWORD8 u1_field_mb_flag,UWORD8 u1_ref_layer_frame_mbs_only_flag)623*495ae853SAndroid Build Coastguard Worker static FORCEINLINE void isvce_ref_array_pos_and_phase_init(layer_resampler_props_t *ps_layer_props,
624*495ae853SAndroid Build Coastguard Worker                                                            res_pred_mb_state_t *ps_mb_state,
625*495ae853SAndroid Build Coastguard Worker                                                            coordinates_t *ps_mb_pos,
626*495ae853SAndroid Build Coastguard Worker                                                            UWORD8 u1_frame_mbs_only_flag,
627*495ae853SAndroid Build Coastguard Worker                                                            UWORD8 u1_field_mb_flag,
628*495ae853SAndroid Build Coastguard Worker                                                            UWORD8 u1_ref_layer_frame_mbs_only_flag)
629*495ae853SAndroid Build Coastguard Worker {
630*495ae853SAndroid Build Coastguard Worker     UWORD32 i, j;
631*495ae853SAndroid Build Coastguard Worker 
632*495ae853SAndroid Build Coastguard Worker     coordinates_t *ps_ref_array_positions = ps_mb_state->ps_ref_array_positions;
633*495ae853SAndroid Build Coastguard Worker     coordinates_t *ps_ref_array_phases = ps_mb_state->ps_ref_array_phases;
634*495ae853SAndroid Build Coastguard Worker 
635*495ae853SAndroid Build Coastguard Worker     WORD32 i4_x_offset = ps_mb_state->s_offsets.i4_abscissa;
636*495ae853SAndroid Build Coastguard Worker     WORD32 i4_y_offset = ps_mb_state->s_offsets.i4_ordinate;
637*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_phase_array_idx = 0;
638*495ae853SAndroid Build Coastguard Worker 
639*495ae853SAndroid Build Coastguard Worker     for(i = 0; i < ps_layer_props->u4_mb_ht; i++)
640*495ae853SAndroid Build Coastguard Worker     {
641*495ae853SAndroid Build Coastguard Worker         WORD32 i4_y_ref16;
642*495ae853SAndroid Build Coastguard Worker 
643*495ae853SAndroid Build Coastguard Worker         WORD32 i4_yc = ps_mb_pos->i4_ordinate * ps_layer_props->u4_mb_ht + i;
644*495ae853SAndroid Build Coastguard Worker 
645*495ae853SAndroid Build Coastguard Worker         if((0 == u1_frame_mbs_only_flag) || (0 == u1_ref_layer_frame_mbs_only_flag))
646*495ae853SAndroid Build Coastguard Worker         {
647*495ae853SAndroid Build Coastguard Worker             i4_yc = i4_yc >> (1 - u1_field_mb_flag);
648*495ae853SAndroid Build Coastguard Worker         }
649*495ae853SAndroid Build Coastguard Worker 
650*495ae853SAndroid Build Coastguard Worker         i4_y_ref16 = isvce_get_scaled_pixel_pos(ps_layer_props, i4_yc, 1);
651*495ae853SAndroid Build Coastguard Worker 
652*495ae853SAndroid Build Coastguard Worker         for(j = 0; j < ps_layer_props->u4_mb_wd; j++)
653*495ae853SAndroid Build Coastguard Worker         {
654*495ae853SAndroid Build Coastguard Worker             WORD32 i4_x_ref16;
655*495ae853SAndroid Build Coastguard Worker 
656*495ae853SAndroid Build Coastguard Worker             WORD32 i4_xc = ps_mb_pos->i4_abscissa * ps_layer_props->u4_mb_wd + j;
657*495ae853SAndroid Build Coastguard Worker 
658*495ae853SAndroid Build Coastguard Worker             i4_x_ref16 = isvce_get_scaled_pixel_pos(ps_layer_props, i4_xc, 0);
659*495ae853SAndroid Build Coastguard Worker 
660*495ae853SAndroid Build Coastguard Worker             ps_ref_array_positions[j + i * ps_layer_props->u4_mb_wd].i4_abscissa =
661*495ae853SAndroid Build Coastguard Worker                 (i4_x_ref16 >> 4) - i4_x_offset;
662*495ae853SAndroid Build Coastguard Worker             ps_ref_array_positions[j + i * ps_layer_props->u4_mb_wd].i4_ordinate =
663*495ae853SAndroid Build Coastguard Worker                 (i4_y_ref16 >> 4) - i4_y_offset;
664*495ae853SAndroid Build Coastguard Worker 
665*495ae853SAndroid Build Coastguard Worker             if(((0 == i) && (j < 3)) || ((0 == j) && (i < 3)))
666*495ae853SAndroid Build Coastguard Worker             {
667*495ae853SAndroid Build Coastguard Worker                 ps_ref_array_phases[u4_phase_array_idx].i4_abscissa =
668*495ae853SAndroid Build Coastguard Worker                     (i4_x_ref16 - (16 * i4_x_offset)) & 15;
669*495ae853SAndroid Build Coastguard Worker                 ps_ref_array_phases[u4_phase_array_idx].i4_ordinate =
670*495ae853SAndroid Build Coastguard Worker                     (i4_y_ref16 - (16 * i4_y_offset)) & 15;
671*495ae853SAndroid Build Coastguard Worker 
672*495ae853SAndroid Build Coastguard Worker                 u4_phase_array_idx++;
673*495ae853SAndroid Build Coastguard Worker             }
674*495ae853SAndroid Build Coastguard Worker         }
675*495ae853SAndroid Build Coastguard Worker     }
676*495ae853SAndroid Build Coastguard Worker }
677*495ae853SAndroid Build Coastguard Worker 
isvce_res_pred_layer_state_init(res_pred_layer_state_t * ps_layer_state,DOUBLE d_spatial_res_ratio,UWORD32 u4_wd,UWORD32 u4_ht,IV_COLOR_FORMAT_T e_color_format)678*495ae853SAndroid Build Coastguard Worker static void isvce_res_pred_layer_state_init(res_pred_layer_state_t *ps_layer_state,
679*495ae853SAndroid Build Coastguard Worker                                             DOUBLE d_spatial_res_ratio, UWORD32 u4_wd,
680*495ae853SAndroid Build Coastguard Worker                                             UWORD32 u4_ht, IV_COLOR_FORMAT_T e_color_format)
681*495ae853SAndroid Build Coastguard Worker {
682*495ae853SAndroid Build Coastguard Worker     UWORD32 i, j, k;
683*495ae853SAndroid Build Coastguard Worker 
684*495ae853SAndroid Build Coastguard Worker     const UWORD8 u1_ref_layer_field_pic_flag = 0;
685*495ae853SAndroid Build Coastguard Worker     const UWORD8 u1_field_pic_flag = 0;
686*495ae853SAndroid Build Coastguard Worker     const UWORD8 u1_frame_mbs_only_flag = 1;
687*495ae853SAndroid Build Coastguard Worker     const UWORD8 u1_ref_layer_frame_mbs_only_flag = 1;
688*495ae853SAndroid Build Coastguard Worker     const UWORD8 u1_field_mb_flag = 0;
689*495ae853SAndroid Build Coastguard Worker 
690*495ae853SAndroid Build Coastguard Worker     ASSERT((IV_YUV_420P == e_color_format) || (IV_YUV_420SP_UV == e_color_format));
691*495ae853SAndroid Build Coastguard Worker 
692*495ae853SAndroid Build Coastguard Worker     UNUSED(e_color_format);
693*495ae853SAndroid Build Coastguard Worker 
694*495ae853SAndroid Build Coastguard Worker     for(i = 0; i < 2; i++)
695*495ae853SAndroid Build Coastguard Worker     {
696*495ae853SAndroid Build Coastguard Worker         res_pred_mb_state_t *ps_mb_states;
697*495ae853SAndroid Build Coastguard Worker         layer_resampler_props_t *ps_layer_props;
698*495ae853SAndroid Build Coastguard Worker 
699*495ae853SAndroid Build Coastguard Worker         UWORD32 u4_wd_in_mbs;
700*495ae853SAndroid Build Coastguard Worker         UWORD32 u4_ht_in_mbs;
701*495ae853SAndroid Build Coastguard Worker 
702*495ae853SAndroid Build Coastguard Worker         UWORD8 u1_is_chroma = (Y != ((COMPONENT_TYPE) i));
703*495ae853SAndroid Build Coastguard Worker         UWORD32 u4_ref_wd = (u4_wd / d_spatial_res_ratio);
704*495ae853SAndroid Build Coastguard Worker         UWORD32 u4_ref_ht = (u4_ht / d_spatial_res_ratio) * (1 + u1_ref_layer_field_pic_flag);
705*495ae853SAndroid Build Coastguard Worker         UWORD32 u4_scaled_wd = u4_wd;
706*495ae853SAndroid Build Coastguard Worker         UWORD32 u4_scaled_ht = u4_ht * (1 + u1_field_pic_flag);
707*495ae853SAndroid Build Coastguard Worker 
708*495ae853SAndroid Build Coastguard Worker         ps_mb_states =
709*495ae853SAndroid Build Coastguard Worker             u1_is_chroma ? ps_layer_state->ps_chroma_mb_states : ps_layer_state->ps_luma_mb_states;
710*495ae853SAndroid Build Coastguard Worker         ps_layer_props =
711*495ae853SAndroid Build Coastguard Worker             u1_is_chroma ? ps_layer_state->ps_chroma_props : ps_layer_state->ps_luma_props;
712*495ae853SAndroid Build Coastguard Worker 
713*495ae853SAndroid Build Coastguard Worker         u4_ref_wd = u4_ref_wd >> u1_is_chroma;
714*495ae853SAndroid Build Coastguard Worker         u4_ref_ht = u4_ref_ht >> u1_is_chroma;
715*495ae853SAndroid Build Coastguard Worker         u4_scaled_wd = u4_scaled_wd >> u1_is_chroma;
716*495ae853SAndroid Build Coastguard Worker         u4_scaled_ht = u4_scaled_ht >> u1_is_chroma;
717*495ae853SAndroid Build Coastguard Worker 
718*495ae853SAndroid Build Coastguard Worker         u4_wd_in_mbs = u4_scaled_wd / ps_layer_props->u4_mb_wd;
719*495ae853SAndroid Build Coastguard Worker         u4_ht_in_mbs = u4_scaled_ht / ps_layer_props->u4_mb_ht;
720*495ae853SAndroid Build Coastguard Worker 
721*495ae853SAndroid Build Coastguard Worker         for(j = 0; j < u4_ht_in_mbs; j++)
722*495ae853SAndroid Build Coastguard Worker         {
723*495ae853SAndroid Build Coastguard Worker             WORD32 i4_y_refmin16;
724*495ae853SAndroid Build Coastguard Worker             WORD32 i4_y_refmax16;
725*495ae853SAndroid Build Coastguard Worker             WORD32 i4_y_offset;
726*495ae853SAndroid Build Coastguard Worker 
727*495ae853SAndroid Build Coastguard Worker             i4_y_refmin16 =
728*495ae853SAndroid Build Coastguard Worker                 isvce_get_scaled_pixel_pos(ps_layer_props, j * ps_layer_props->u4_mb_ht, 1);
729*495ae853SAndroid Build Coastguard Worker             i4_y_refmax16 = isvce_get_scaled_pixel_pos(
730*495ae853SAndroid Build Coastguard Worker                 ps_layer_props, j * ps_layer_props->u4_mb_ht + ps_layer_props->u4_mb_ht - 1, 1);
731*495ae853SAndroid Build Coastguard Worker             i4_y_offset = i4_y_refmin16 >> 4;
732*495ae853SAndroid Build Coastguard Worker 
733*495ae853SAndroid Build Coastguard Worker             for(k = 0; k < u4_wd_in_mbs; k++)
734*495ae853SAndroid Build Coastguard Worker             {
735*495ae853SAndroid Build Coastguard Worker                 WORD32 i4_x_refmin16;
736*495ae853SAndroid Build Coastguard Worker                 WORD32 i4_x_refmax16;
737*495ae853SAndroid Build Coastguard Worker                 WORD32 i4_x_offset;
738*495ae853SAndroid Build Coastguard Worker 
739*495ae853SAndroid Build Coastguard Worker                 coordinates_t s_mb_pos = {k, j};
740*495ae853SAndroid Build Coastguard Worker 
741*495ae853SAndroid Build Coastguard Worker                 i4_x_refmin16 =
742*495ae853SAndroid Build Coastguard Worker                     isvce_get_scaled_pixel_pos(ps_layer_props, k * ps_layer_props->u4_mb_wd, 0);
743*495ae853SAndroid Build Coastguard Worker                 i4_x_refmax16 = isvce_get_scaled_pixel_pos(
744*495ae853SAndroid Build Coastguard Worker                     ps_layer_props, k * ps_layer_props->u4_mb_wd + ps_layer_props->u4_mb_wd - 1, 0);
745*495ae853SAndroid Build Coastguard Worker                 i4_x_offset = i4_x_refmin16 >> 4;
746*495ae853SAndroid Build Coastguard Worker 
747*495ae853SAndroid Build Coastguard Worker                 ps_mb_states[k + j * u4_wd_in_mbs].s_offsets.i4_abscissa = i4_x_offset;
748*495ae853SAndroid Build Coastguard Worker                 ps_mb_states[k + j * u4_wd_in_mbs].s_offsets.i4_ordinate = i4_y_offset;
749*495ae853SAndroid Build Coastguard Worker                 ps_mb_states[k + j * u4_wd_in_mbs].s_ref_array_dims.i4_abscissa =
750*495ae853SAndroid Build Coastguard Worker                     (i4_x_refmax16 >> 4) - i4_x_offset + 2;
751*495ae853SAndroid Build Coastguard Worker                 ps_mb_states[k + j * u4_wd_in_mbs].s_ref_array_dims.i4_ordinate =
752*495ae853SAndroid Build Coastguard Worker                     (i4_y_refmax16 >> 4) - i4_y_offset + 2;
753*495ae853SAndroid Build Coastguard Worker 
754*495ae853SAndroid Build Coastguard Worker                 if((0 == k) && (0 == j) && (2 == d_spatial_res_ratio) && u1_is_chroma)
755*495ae853SAndroid Build Coastguard Worker                 {
756*495ae853SAndroid Build Coastguard Worker                     isvce_ref_array_pos_and_phase_init_dyadic(
757*495ae853SAndroid Build Coastguard Worker                         ps_layer_props, &ps_mb_states[k + j * u4_wd_in_mbs], &s_mb_pos,
758*495ae853SAndroid Build Coastguard Worker                         u1_frame_mbs_only_flag, u1_field_mb_flag, u1_ref_layer_frame_mbs_only_flag);
759*495ae853SAndroid Build Coastguard Worker                 }
760*495ae853SAndroid Build Coastguard Worker                 else if(1.5 == d_spatial_res_ratio)
761*495ae853SAndroid Build Coastguard Worker                 {
762*495ae853SAndroid Build Coastguard Worker                     isvce_ref_array_pos_and_phase_init(
763*495ae853SAndroid Build Coastguard Worker                         ps_layer_props, &ps_mb_states[k + j * u4_wd_in_mbs], &s_mb_pos,
764*495ae853SAndroid Build Coastguard Worker                         u1_frame_mbs_only_flag, u1_field_mb_flag, u1_ref_layer_frame_mbs_only_flag);
765*495ae853SAndroid Build Coastguard Worker                 }
766*495ae853SAndroid Build Coastguard Worker             }
767*495ae853SAndroid Build Coastguard Worker         }
768*495ae853SAndroid Build Coastguard Worker     }
769*495ae853SAndroid Build Coastguard Worker }
770*495ae853SAndroid Build Coastguard Worker 
isvce_svc_residual_sampling_function_selector(res_pred_state_t * ps_res_pred_state,DOUBLE d_spatial_res_ratio,IV_ARCH_T e_arch)771*495ae853SAndroid Build Coastguard Worker void isvce_svc_residual_sampling_function_selector(res_pred_state_t *ps_res_pred_state,
772*495ae853SAndroid Build Coastguard Worker                                                    DOUBLE d_spatial_res_ratio, IV_ARCH_T e_arch)
773*495ae853SAndroid Build Coastguard Worker {
774*495ae853SAndroid Build Coastguard Worker     if(2. == d_spatial_res_ratio)
775*495ae853SAndroid Build Coastguard Worker     {
776*495ae853SAndroid Build Coastguard Worker         ps_res_pred_state->apf_residual_samplers[U] = isvce_chroma_residual_sampler_2x;
777*495ae853SAndroid Build Coastguard Worker         ps_res_pred_state->apf_residual_samplers[V] = isvce_chroma_residual_sampler_2x;
778*495ae853SAndroid Build Coastguard Worker 
779*495ae853SAndroid Build Coastguard Worker         switch(e_arch)
780*495ae853SAndroid Build Coastguard Worker         {
781*495ae853SAndroid Build Coastguard Worker #if defined(X86)
782*495ae853SAndroid Build Coastguard Worker             case ARCH_X86_SSE42:
783*495ae853SAndroid Build Coastguard Worker             {
784*495ae853SAndroid Build Coastguard Worker                 ps_res_pred_state->apf_residual_samplers[Y] = isvce_luma_residual_sampler_2x_sse42;
785*495ae853SAndroid Build Coastguard Worker 
786*495ae853SAndroid Build Coastguard Worker                 break;
787*495ae853SAndroid Build Coastguard Worker             }
788*495ae853SAndroid Build Coastguard Worker #elif defined(ARMV8)
789*495ae853SAndroid Build Coastguard Worker             case ARCH_ARM_A53:
790*495ae853SAndroid Build Coastguard Worker             case ARCH_ARM_A57:
791*495ae853SAndroid Build Coastguard Worker             case ARCH_ARM_V8_NEON:
792*495ae853SAndroid Build Coastguard Worker             {
793*495ae853SAndroid Build Coastguard Worker                 ps_res_pred_state->apf_residual_samplers[Y] = isvce_luma_residual_sampler_2x_neon;
794*495ae853SAndroid Build Coastguard Worker 
795*495ae853SAndroid Build Coastguard Worker                 break;
796*495ae853SAndroid Build Coastguard Worker             }
797*495ae853SAndroid Build Coastguard Worker #elif defined(ARM) && !defined(DISABLE_NEON)
798*495ae853SAndroid Build Coastguard Worker             case ARCH_ARM_A9Q:
799*495ae853SAndroid Build Coastguard Worker             case ARCH_ARM_A9A:
800*495ae853SAndroid Build Coastguard Worker             case ARCH_ARM_A9:
801*495ae853SAndroid Build Coastguard Worker             case ARCH_ARM_A7:
802*495ae853SAndroid Build Coastguard Worker             case ARCH_ARM_A5:
803*495ae853SAndroid Build Coastguard Worker             case ARCH_ARM_A15:
804*495ae853SAndroid Build Coastguard Worker             {
805*495ae853SAndroid Build Coastguard Worker                 ps_res_pred_state->apf_residual_samplers[Y] = isvce_luma_residual_sampler_2x_neon;
806*495ae853SAndroid Build Coastguard Worker 
807*495ae853SAndroid Build Coastguard Worker                 break;
808*495ae853SAndroid Build Coastguard Worker             }
809*495ae853SAndroid Build Coastguard Worker #endif
810*495ae853SAndroid Build Coastguard Worker             default:
811*495ae853SAndroid Build Coastguard Worker             {
812*495ae853SAndroid Build Coastguard Worker                 ps_res_pred_state->apf_residual_samplers[Y] = isvce_luma_residual_sampler_2x;
813*495ae853SAndroid Build Coastguard Worker 
814*495ae853SAndroid Build Coastguard Worker                 break;
815*495ae853SAndroid Build Coastguard Worker             }
816*495ae853SAndroid Build Coastguard Worker         }
817*495ae853SAndroid Build Coastguard Worker     }
818*495ae853SAndroid Build Coastguard Worker 
819*495ae853SAndroid Build Coastguard Worker     switch(e_arch)
820*495ae853SAndroid Build Coastguard Worker     {
821*495ae853SAndroid Build Coastguard Worker #if defined(X86)
822*495ae853SAndroid Build Coastguard Worker         case ARCH_X86_SSE42:
823*495ae853SAndroid Build Coastguard Worker         {
824*495ae853SAndroid Build Coastguard Worker             ps_res_pred_state->pf_get_sad_with_residual_pred =
825*495ae853SAndroid Build Coastguard Worker                 isvce_get_sad_with_residual_pred_sse42;
826*495ae853SAndroid Build Coastguard Worker 
827*495ae853SAndroid Build Coastguard Worker             break;
828*495ae853SAndroid Build Coastguard Worker     }
829*495ae853SAndroid Build Coastguard Worker #elif defined(ARMV8)
830*495ae853SAndroid Build Coastguard Worker         case ARCH_ARM_A53:
831*495ae853SAndroid Build Coastguard Worker         case ARCH_ARM_A57:
832*495ae853SAndroid Build Coastguard Worker         case ARCH_ARM_V8_NEON:
833*495ae853SAndroid Build Coastguard Worker         {
834*495ae853SAndroid Build Coastguard Worker             ps_res_pred_state->pf_get_sad_with_residual_pred =
835*495ae853SAndroid Build Coastguard Worker                 isvce_get_sad_with_residual_pred_neon;
836*495ae853SAndroid Build Coastguard Worker 
837*495ae853SAndroid Build Coastguard Worker             break;
838*495ae853SAndroid Build Coastguard Worker     }
839*495ae853SAndroid Build Coastguard Worker #elif defined(ARM) && !defined(DISABLE_NEON)
840*495ae853SAndroid Build Coastguard Worker         case ARCH_ARM_A9Q:
841*495ae853SAndroid Build Coastguard Worker         case ARCH_ARM_A9A:
842*495ae853SAndroid Build Coastguard Worker         case ARCH_ARM_A9:
843*495ae853SAndroid Build Coastguard Worker         case ARCH_ARM_A7:
844*495ae853SAndroid Build Coastguard Worker         case ARCH_ARM_A5:
845*495ae853SAndroid Build Coastguard Worker         case ARCH_ARM_A15:
846*495ae853SAndroid Build Coastguard Worker         {
847*495ae853SAndroid Build Coastguard Worker             ps_res_pred_state->pf_get_sad_with_residual_pred =
848*495ae853SAndroid Build Coastguard Worker                 isvce_get_sad_with_residual_pred_neon;
849*495ae853SAndroid Build Coastguard Worker 
850*495ae853SAndroid Build Coastguard Worker             break;
851*495ae853SAndroid Build Coastguard Worker     }
852*495ae853SAndroid Build Coastguard Worker #endif
853*495ae853SAndroid Build Coastguard Worker     default:
854*495ae853SAndroid Build Coastguard Worker     {
855*495ae853SAndroid Build Coastguard Worker             ps_res_pred_state->pf_get_sad_with_residual_pred = isvce_get_sad_with_residual_pred;
856*495ae853SAndroid Build Coastguard Worker 
857*495ae853SAndroid Build Coastguard Worker             break;
858*495ae853SAndroid Build Coastguard Worker     }
859*495ae853SAndroid Build Coastguard Worker     }
860*495ae853SAndroid Build Coastguard Worker }
861*495ae853SAndroid Build Coastguard Worker 
862*495ae853SAndroid Build Coastguard Worker /**
863*495ae853SAndroid Build Coastguard Worker *******************************************************************************
864*495ae853SAndroid Build Coastguard Worker *
865*495ae853SAndroid Build Coastguard Worker * @brief
866*495ae853SAndroid Build Coastguard Worker *  Function to initialize svc ilp buffers
867*495ae853SAndroid Build Coastguard Worker *
868*495ae853SAndroid Build Coastguard Worker * @param[in] ps_codec
869*495ae853SAndroid Build Coastguard Worker *  Pointer to codec context
870*495ae853SAndroid Build Coastguard Worker *
871*495ae853SAndroid Build Coastguard Worker * @param[in] ps_mem_rec
872*495ae853SAndroid Build Coastguard Worker *  Pointer to memory allocated for input buffers
873*495ae853SAndroid Build Coastguard Worker *
874*495ae853SAndroid Build Coastguard Worker *******************************************************************************
875*495ae853SAndroid Build Coastguard Worker */
isvce_svc_res_pred_ctxt_init(isvce_codec_t * ps_codec,iv_mem_rec_t * ps_mem_rec)876*495ae853SAndroid Build Coastguard Worker void isvce_svc_res_pred_ctxt_init(isvce_codec_t *ps_codec, iv_mem_rec_t *ps_mem_rec)
877*495ae853SAndroid Build Coastguard Worker {
878*495ae853SAndroid Build Coastguard Worker     WORD32 i, j, k;
879*495ae853SAndroid Build Coastguard Worker 
880*495ae853SAndroid Build Coastguard Worker     const WORD32 i4_num_proc_ctxts = sizeof(ps_codec->as_process) / sizeof(ps_codec->as_process[0]);
881*495ae853SAndroid Build Coastguard Worker     UWORD8 u1_num_spatial_layers = ps_codec->s_cfg.s_svc_params.u1_num_spatial_layers;
882*495ae853SAndroid Build Coastguard Worker     DOUBLE d_spatial_res_ratio = ps_codec->s_cfg.s_svc_params.d_spatial_res_ratio;
883*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_wd = ps_codec->s_cfg.u4_wd;
884*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_ht = ps_codec->s_cfg.u4_ht;
885*495ae853SAndroid Build Coastguard Worker     UWORD8 *pu1_buf = ps_mem_rec->pv_base;
886*495ae853SAndroid Build Coastguard Worker     WORD64 i8_alloc_mem_size =
887*495ae853SAndroid Build Coastguard Worker         isvce_get_svc_res_pred_ctxt_size(u1_num_spatial_layers, d_spatial_res_ratio, u4_wd, u4_ht);
888*495ae853SAndroid Build Coastguard Worker 
889*495ae853SAndroid Build Coastguard Worker     if(u1_num_spatial_layers > 1)
890*495ae853SAndroid Build Coastguard Worker     {
891*495ae853SAndroid Build Coastguard Worker         res_pred_mb_state_t *aps_luma_mb_states[MAX_NUM_SPATIAL_LAYERS];
892*495ae853SAndroid Build Coastguard Worker         res_pred_mb_state_t *aps_chroma_mb_states[MAX_NUM_SPATIAL_LAYERS];
893*495ae853SAndroid Build Coastguard Worker 
894*495ae853SAndroid Build Coastguard Worker         WORD8 *api1_mb_mode[MAX_NUM_SPATIAL_LAYERS];
895*495ae853SAndroid Build Coastguard Worker         WORD32 ai4_mb_mode_stride[MAX_NUM_SPATIAL_LAYERS];
896*495ae853SAndroid Build Coastguard Worker 
897*495ae853SAndroid Build Coastguard Worker         WORD32 i4_size;
898*495ae853SAndroid Build Coastguard Worker 
899*495ae853SAndroid Build Coastguard Worker         for(i = 0; i < i4_num_proc_ctxts; i++)
900*495ae853SAndroid Build Coastguard Worker         {
901*495ae853SAndroid Build Coastguard Worker             res_pred_state_t *ps_res_pred_state;
902*495ae853SAndroid Build Coastguard Worker             svc_res_pred_ctxt_t *ps_res_pred_ctxt;
903*495ae853SAndroid Build Coastguard Worker             yuv_buf_props_t *ps_mb_res_buf;
904*495ae853SAndroid Build Coastguard Worker             res_pred_mem_store_t *ps_mem_store;
905*495ae853SAndroid Build Coastguard Worker 
906*495ae853SAndroid Build Coastguard Worker             isvce_process_ctxt_t *ps_proc = ps_codec->as_process + i;
907*495ae853SAndroid Build Coastguard Worker 
908*495ae853SAndroid Build Coastguard Worker             ps_res_pred_ctxt = ps_proc->ps_res_pred_ctxt = (svc_res_pred_ctxt_t *) pu1_buf;
909*495ae853SAndroid Build Coastguard Worker             pu1_buf += sizeof(svc_res_pred_ctxt_t);
910*495ae853SAndroid Build Coastguard Worker             i8_alloc_mem_size -= sizeof(svc_res_pred_ctxt_t);
911*495ae853SAndroid Build Coastguard Worker 
912*495ae853SAndroid Build Coastguard Worker             ps_res_pred_ctxt->s_res_pred_constants.pv_state = pu1_buf;
913*495ae853SAndroid Build Coastguard Worker             ps_res_pred_state = (res_pred_state_t *) pu1_buf;
914*495ae853SAndroid Build Coastguard Worker             pu1_buf += sizeof(res_pred_state_t);
915*495ae853SAndroid Build Coastguard Worker             i8_alloc_mem_size -= sizeof(res_pred_state_t);
916*495ae853SAndroid Build Coastguard Worker 
917*495ae853SAndroid Build Coastguard Worker             ps_res_pred_state->ps_layer_state = (res_pred_layer_state_t *) pu1_buf;
918*495ae853SAndroid Build Coastguard Worker             pu1_buf += u1_num_spatial_layers * sizeof(ps_res_pred_state->ps_layer_state[0]);
919*495ae853SAndroid Build Coastguard Worker             i8_alloc_mem_size -=
920*495ae853SAndroid Build Coastguard Worker                 u1_num_spatial_layers * sizeof(ps_res_pred_state->ps_layer_state[0]);
921*495ae853SAndroid Build Coastguard Worker 
922*495ae853SAndroid Build Coastguard Worker             i4_size = REF_ARRAY_MAX_WIDTH * REF_ARRAY_MAX_HEIGHT * sizeof(UWORD8);
923*495ae853SAndroid Build Coastguard Worker             ps_res_pred_state->pu1_ref_x_ptr_incr = (UWORD8 *) pu1_buf;
924*495ae853SAndroid Build Coastguard Worker             pu1_buf += i4_size;
925*495ae853SAndroid Build Coastguard Worker             ps_res_pred_state->pu1_ref_y_ptr_incr = (UWORD8 *) pu1_buf;
926*495ae853SAndroid Build Coastguard Worker             pu1_buf += i4_size;
927*495ae853SAndroid Build Coastguard Worker 
928*495ae853SAndroid Build Coastguard Worker             ASSERT(i8_alloc_mem_size >= 0);
929*495ae853SAndroid Build Coastguard Worker 
930*495ae853SAndroid Build Coastguard Worker             if(0 == i)
931*495ae853SAndroid Build Coastguard Worker             {
932*495ae853SAndroid Build Coastguard Worker                 UWORD32 au4_ref_pos_array_size[NUM_SP_COMPONENTS];
933*495ae853SAndroid Build Coastguard Worker                 UWORD32 au4_ref_phase_array_size[NUM_SP_COMPONENTS];
934*495ae853SAndroid Build Coastguard Worker 
935*495ae853SAndroid Build Coastguard Worker                 if(1.5 == d_spatial_res_ratio)
936*495ae853SAndroid Build Coastguard Worker                 {
937*495ae853SAndroid Build Coastguard Worker                     au4_ref_pos_array_size[Y] = MB_SIZE * MB_SIZE;
938*495ae853SAndroid Build Coastguard Worker                     au4_ref_phase_array_size[Y] = 5;
939*495ae853SAndroid Build Coastguard Worker                     au4_ref_pos_array_size[U] = (MB_SIZE / 2) * (MB_SIZE / 2);
940*495ae853SAndroid Build Coastguard Worker                     au4_ref_phase_array_size[U] = 5;
941*495ae853SAndroid Build Coastguard Worker                 }
942*495ae853SAndroid Build Coastguard Worker                 else
943*495ae853SAndroid Build Coastguard Worker                 {
944*495ae853SAndroid Build Coastguard Worker                     au4_ref_pos_array_size[Y] = au4_ref_pos_array_size[U] = 0;
945*495ae853SAndroid Build Coastguard Worker                     au4_ref_phase_array_size[Y] = 0;
946*495ae853SAndroid Build Coastguard Worker                     au4_ref_phase_array_size[U] = 3;
947*495ae853SAndroid Build Coastguard Worker                 }
948*495ae853SAndroid Build Coastguard Worker 
949*495ae853SAndroid Build Coastguard Worker                 for(j = u1_num_spatial_layers - 1; j >= 1; j--)
950*495ae853SAndroid Build Coastguard Worker                 {
951*495ae853SAndroid Build Coastguard Worker                     res_pred_layer_state_t *ps_layer = &ps_res_pred_state->ps_layer_state[j];
952*495ae853SAndroid Build Coastguard Worker 
953*495ae853SAndroid Build Coastguard Worker                     WORD32 i4_layer_luma_wd =
954*495ae853SAndroid Build Coastguard Worker                         ((DOUBLE) u4_wd / pow(d_spatial_res_ratio, u1_num_spatial_layers - 1 - j)) +
955*495ae853SAndroid Build Coastguard Worker                         0.99;
956*495ae853SAndroid Build Coastguard Worker                     WORD32 i4_layer_luma_ht =
957*495ae853SAndroid Build Coastguard Worker                         ((DOUBLE) u4_ht / pow(d_spatial_res_ratio, u1_num_spatial_layers - 1 - j)) +
958*495ae853SAndroid Build Coastguard Worker                         0.99;
959*495ae853SAndroid Build Coastguard Worker                     WORD32 i4_layer_luma_mbs =
960*495ae853SAndroid Build Coastguard Worker                         (i4_layer_luma_wd / MB_SIZE) * (i4_layer_luma_ht / MB_SIZE);
961*495ae853SAndroid Build Coastguard Worker                     WORD32 i4_layer_u_wd = i4_layer_luma_wd / 2.0 + 0.99;
962*495ae853SAndroid Build Coastguard Worker                     WORD32 i4_layer_u_ht = i4_layer_luma_ht / 2.0 + 0.99;
963*495ae853SAndroid Build Coastguard Worker                     WORD32 i4_layer_u_mbs =
964*495ae853SAndroid Build Coastguard Worker                         (i4_layer_u_wd / (MB_SIZE / 2)) * (i4_layer_u_ht / (MB_SIZE / 2));
965*495ae853SAndroid Build Coastguard Worker 
966*495ae853SAndroid Build Coastguard Worker                     ps_layer->ps_luma_mb_states = (res_pred_mb_state_t *) pu1_buf;
967*495ae853SAndroid Build Coastguard Worker                     aps_luma_mb_states[j] = ps_layer->ps_luma_mb_states;
968*495ae853SAndroid Build Coastguard Worker                     pu1_buf += i4_layer_luma_mbs * sizeof(ps_layer->ps_luma_mb_states[0]);
969*495ae853SAndroid Build Coastguard Worker                     i8_alloc_mem_size -=
970*495ae853SAndroid Build Coastguard Worker                         u1_num_spatial_layers * sizeof(ps_layer->ps_luma_mb_states[0]);
971*495ae853SAndroid Build Coastguard Worker 
972*495ae853SAndroid Build Coastguard Worker                     ps_layer->ps_chroma_mb_states = (res_pred_mb_state_t *) pu1_buf;
973*495ae853SAndroid Build Coastguard Worker                     aps_chroma_mb_states[j] = ps_layer->ps_chroma_mb_states;
974*495ae853SAndroid Build Coastguard Worker                     pu1_buf += i4_layer_u_mbs * sizeof(ps_layer->ps_chroma_mb_states[0]);
975*495ae853SAndroid Build Coastguard Worker                     i8_alloc_mem_size -= i4_layer_u_mbs * sizeof(ps_layer->ps_chroma_mb_states[0]);
976*495ae853SAndroid Build Coastguard Worker 
977*495ae853SAndroid Build Coastguard Worker                     if(1.5 == d_spatial_res_ratio)
978*495ae853SAndroid Build Coastguard Worker                     {
979*495ae853SAndroid Build Coastguard Worker                         coordinates_t *ps_ref_array_pos = (coordinates_t *) pu1_buf;
980*495ae853SAndroid Build Coastguard Worker                         coordinates_t *ps_ref_array_phases =
981*495ae853SAndroid Build Coastguard Worker                             ps_ref_array_pos + i4_layer_luma_mbs * au4_ref_pos_array_size[Y];
982*495ae853SAndroid Build Coastguard Worker 
983*495ae853SAndroid Build Coastguard Worker                         for(k = 0; k < i4_layer_luma_mbs; k++)
984*495ae853SAndroid Build Coastguard Worker                         {
985*495ae853SAndroid Build Coastguard Worker                             ps_layer->ps_luma_mb_states[k].ps_ref_array_positions =
986*495ae853SAndroid Build Coastguard Worker                                 ps_ref_array_pos + k * au4_ref_pos_array_size[Y];
987*495ae853SAndroid Build Coastguard Worker                             ps_layer->ps_luma_mb_states[k].ps_ref_array_phases =
988*495ae853SAndroid Build Coastguard Worker                                 ps_ref_array_phases + k * au4_ref_phase_array_size[Y];
989*495ae853SAndroid Build Coastguard Worker                             pu1_buf += au4_ref_pos_array_size[Y] * sizeof(ps_ref_array_pos[0]);
990*495ae853SAndroid Build Coastguard Worker                             i8_alloc_mem_size -=
991*495ae853SAndroid Build Coastguard Worker                                 au4_ref_pos_array_size[Y] * sizeof(ps_ref_array_pos[0]);
992*495ae853SAndroid Build Coastguard Worker                             pu1_buf += au4_ref_phase_array_size[Y] * sizeof(ps_ref_array_phases[0]);
993*495ae853SAndroid Build Coastguard Worker                             i8_alloc_mem_size -=
994*495ae853SAndroid Build Coastguard Worker                                 au4_ref_phase_array_size[Y] * sizeof(ps_ref_array_phases[0]);
995*495ae853SAndroid Build Coastguard Worker                         }
996*495ae853SAndroid Build Coastguard Worker 
997*495ae853SAndroid Build Coastguard Worker                         ps_ref_array_pos = (coordinates_t *) pu1_buf;
998*495ae853SAndroid Build Coastguard Worker                         ps_ref_array_phases =
999*495ae853SAndroid Build Coastguard Worker                             ps_ref_array_pos + i4_layer_u_mbs * au4_ref_pos_array_size[U];
1000*495ae853SAndroid Build Coastguard Worker 
1001*495ae853SAndroid Build Coastguard Worker                         for(k = 0; k < i4_layer_u_mbs; k++)
1002*495ae853SAndroid Build Coastguard Worker                         {
1003*495ae853SAndroid Build Coastguard Worker                             ps_layer->ps_chroma_mb_states[k].ps_ref_array_positions =
1004*495ae853SAndroid Build Coastguard Worker                                 ps_ref_array_pos + k * au4_ref_pos_array_size[U];
1005*495ae853SAndroid Build Coastguard Worker                             ps_layer->ps_chroma_mb_states[k].ps_ref_array_phases =
1006*495ae853SAndroid Build Coastguard Worker                                 ps_ref_array_phases + k * au4_ref_phase_array_size[U];
1007*495ae853SAndroid Build Coastguard Worker                             pu1_buf += au4_ref_pos_array_size[U] * sizeof(ps_ref_array_pos[0]);
1008*495ae853SAndroid Build Coastguard Worker                             i8_alloc_mem_size -=
1009*495ae853SAndroid Build Coastguard Worker                                 au4_ref_pos_array_size[U] * sizeof(ps_ref_array_pos[0]);
1010*495ae853SAndroid Build Coastguard Worker                             pu1_buf += au4_ref_phase_array_size[U] * sizeof(ps_ref_array_phases[0]);
1011*495ae853SAndroid Build Coastguard Worker                             i8_alloc_mem_size -=
1012*495ae853SAndroid Build Coastguard Worker                                 au4_ref_phase_array_size[U] * sizeof(ps_ref_array_phases[0]);
1013*495ae853SAndroid Build Coastguard Worker                         }
1014*495ae853SAndroid Build Coastguard Worker                     }
1015*495ae853SAndroid Build Coastguard Worker                     else
1016*495ae853SAndroid Build Coastguard Worker                     {
1017*495ae853SAndroid Build Coastguard Worker                         coordinates_t *ps_ref_array_pos = NULL;
1018*495ae853SAndroid Build Coastguard Worker                         coordinates_t *ps_ref_array_phases = NULL;
1019*495ae853SAndroid Build Coastguard Worker 
1020*495ae853SAndroid Build Coastguard Worker                         for(k = 0; k < i4_layer_luma_mbs; k++)
1021*495ae853SAndroid Build Coastguard Worker                         {
1022*495ae853SAndroid Build Coastguard Worker                             ps_layer->ps_luma_mb_states[k].ps_ref_array_positions =
1023*495ae853SAndroid Build Coastguard Worker                                 ps_ref_array_pos;
1024*495ae853SAndroid Build Coastguard Worker                             ps_layer->ps_luma_mb_states[k].ps_ref_array_phases =
1025*495ae853SAndroid Build Coastguard Worker                                 ps_ref_array_phases;
1026*495ae853SAndroid Build Coastguard Worker                         }
1027*495ae853SAndroid Build Coastguard Worker 
1028*495ae853SAndroid Build Coastguard Worker                         ps_ref_array_pos = NULL;
1029*495ae853SAndroid Build Coastguard Worker                         ps_ref_array_phases = (coordinates_t *) pu1_buf;
1030*495ae853SAndroid Build Coastguard Worker 
1031*495ae853SAndroid Build Coastguard Worker                         for(k = 0; k < i4_layer_u_mbs; k++)
1032*495ae853SAndroid Build Coastguard Worker                         {
1033*495ae853SAndroid Build Coastguard Worker                             ps_layer->ps_chroma_mb_states[k].ps_ref_array_positions =
1034*495ae853SAndroid Build Coastguard Worker                                 ps_ref_array_pos;
1035*495ae853SAndroid Build Coastguard Worker                             ps_layer->ps_chroma_mb_states[k].ps_ref_array_phases =
1036*495ae853SAndroid Build Coastguard Worker                                 ps_ref_array_phases;
1037*495ae853SAndroid Build Coastguard Worker                         }
1038*495ae853SAndroid Build Coastguard Worker 
1039*495ae853SAndroid Build Coastguard Worker                         pu1_buf += au4_ref_phase_array_size[U] * sizeof(ps_ref_array_pos[0]);
1040*495ae853SAndroid Build Coastguard Worker                         i8_alloc_mem_size -=
1041*495ae853SAndroid Build Coastguard Worker                             au4_ref_phase_array_size[U] * sizeof(ps_ref_array_phases[0]);
1042*495ae853SAndroid Build Coastguard Worker                     }
1043*495ae853SAndroid Build Coastguard Worker 
1044*495ae853SAndroid Build Coastguard Worker                     ASSERT(i8_alloc_mem_size >= 0);
1045*495ae853SAndroid Build Coastguard Worker                     /* Asserts below verify that
1046*495ae853SAndroid Build Coastguard Worker                      * 'ps_codec->s_svc_ilp_data.aps_layer_resampler_props' is initialised
1047*495ae853SAndroid Build Coastguard Worker                      */
1048*495ae853SAndroid Build Coastguard Worker                     ASSERT(ps_codec->s_svc_ilp_data.aps_layer_resampler_props[Y][j].u4_mb_wd ==
1049*495ae853SAndroid Build Coastguard Worker                            MB_SIZE);
1050*495ae853SAndroid Build Coastguard Worker                     ASSERT(ps_codec->s_svc_ilp_data.aps_layer_resampler_props[UV][j].u4_mb_wd ==
1051*495ae853SAndroid Build Coastguard Worker                            (MB_SIZE / 2));
1052*495ae853SAndroid Build Coastguard Worker 
1053*495ae853SAndroid Build Coastguard Worker                     ps_layer->ps_luma_props =
1054*495ae853SAndroid Build Coastguard Worker                         &ps_codec->s_svc_ilp_data.aps_layer_resampler_props[Y][j];
1055*495ae853SAndroid Build Coastguard Worker                     ps_layer->ps_chroma_props =
1056*495ae853SAndroid Build Coastguard Worker                         &ps_codec->s_svc_ilp_data.aps_layer_resampler_props[UV][j];
1057*495ae853SAndroid Build Coastguard Worker 
1058*495ae853SAndroid Build Coastguard Worker                     isvce_res_pred_layer_state_init(ps_layer, d_spatial_res_ratio, i4_layer_luma_wd,
1059*495ae853SAndroid Build Coastguard Worker                                                     i4_layer_luma_ht,
1060*495ae853SAndroid Build Coastguard Worker                                                     ps_codec->s_cfg.e_inp_color_fmt);
1061*495ae853SAndroid Build Coastguard Worker                 }
1062*495ae853SAndroid Build Coastguard Worker 
1063*495ae853SAndroid Build Coastguard Worker                 for(j = u1_num_spatial_layers - 1; j >= 0; j--)
1064*495ae853SAndroid Build Coastguard Worker                 {
1065*495ae853SAndroid Build Coastguard Worker                     res_pred_layer_state_t *ps_layer = &ps_res_pred_state->ps_layer_state[j];
1066*495ae853SAndroid Build Coastguard Worker 
1067*495ae853SAndroid Build Coastguard Worker                     WORD32 i4_layer_luma_wd =
1068*495ae853SAndroid Build Coastguard Worker                         ((DOUBLE) u4_wd / pow(d_spatial_res_ratio, u1_num_spatial_layers - 1 - j)) +
1069*495ae853SAndroid Build Coastguard Worker                         0.99;
1070*495ae853SAndroid Build Coastguard Worker                     WORD32 i4_layer_luma_ht =
1071*495ae853SAndroid Build Coastguard Worker                         ((DOUBLE) u4_ht / pow(d_spatial_res_ratio, u1_num_spatial_layers - 1 - j)) +
1072*495ae853SAndroid Build Coastguard Worker                         0.99;
1073*495ae853SAndroid Build Coastguard Worker                     WORD32 i4_layer_luma_mbs =
1074*495ae853SAndroid Build Coastguard Worker                         ((i4_layer_luma_wd / MB_SIZE) + 2) * ((i4_layer_luma_ht / MB_SIZE) + 2);
1075*495ae853SAndroid Build Coastguard Worker 
1076*495ae853SAndroid Build Coastguard Worker                     ps_layer->pi1_mb_mode = (WORD8 *) pu1_buf;
1077*495ae853SAndroid Build Coastguard Worker                     pu1_buf += i4_layer_luma_mbs * sizeof(WORD8);
1078*495ae853SAndroid Build Coastguard Worker                     memset(ps_layer->pi1_mb_mode, -1, i4_layer_luma_mbs);
1079*495ae853SAndroid Build Coastguard Worker 
1080*495ae853SAndroid Build Coastguard Worker                     ps_layer->i4_mb_mode_stride = ai4_mb_mode_stride[j] =
1081*495ae853SAndroid Build Coastguard Worker                         (i4_layer_luma_wd / MB_SIZE) + 2;
1082*495ae853SAndroid Build Coastguard Worker                     ps_layer->pi1_mb_mode += 1 + ps_layer->i4_mb_mode_stride;
1083*495ae853SAndroid Build Coastguard Worker                     api1_mb_mode[j] = ps_layer->pi1_mb_mode;
1084*495ae853SAndroid Build Coastguard Worker                 }
1085*495ae853SAndroid Build Coastguard Worker             }
1086*495ae853SAndroid Build Coastguard Worker             else
1087*495ae853SAndroid Build Coastguard Worker             {
1088*495ae853SAndroid Build Coastguard Worker                 for(j = u1_num_spatial_layers - 1; j >= 1; j--)
1089*495ae853SAndroid Build Coastguard Worker                 {
1090*495ae853SAndroid Build Coastguard Worker                     res_pred_layer_state_t *ps_layer = &ps_res_pred_state->ps_layer_state[j];
1091*495ae853SAndroid Build Coastguard Worker 
1092*495ae853SAndroid Build Coastguard Worker                     ps_layer->ps_luma_mb_states = aps_luma_mb_states[j];
1093*495ae853SAndroid Build Coastguard Worker                     ps_layer->ps_chroma_mb_states = aps_chroma_mb_states[j];
1094*495ae853SAndroid Build Coastguard Worker 
1095*495ae853SAndroid Build Coastguard Worker                     ps_layer->ps_luma_props =
1096*495ae853SAndroid Build Coastguard Worker                         &ps_codec->s_svc_ilp_data.aps_layer_resampler_props[Y][j];
1097*495ae853SAndroid Build Coastguard Worker                     ps_layer->ps_chroma_props =
1098*495ae853SAndroid Build Coastguard Worker                         &ps_codec->s_svc_ilp_data.aps_layer_resampler_props[UV][j];
1099*495ae853SAndroid Build Coastguard Worker                 }
1100*495ae853SAndroid Build Coastguard Worker                 for(j = u1_num_spatial_layers - 1; j >= 0; j--)
1101*495ae853SAndroid Build Coastguard Worker                 {
1102*495ae853SAndroid Build Coastguard Worker                     res_pred_layer_state_t *ps_layer = &ps_res_pred_state->ps_layer_state[j];
1103*495ae853SAndroid Build Coastguard Worker 
1104*495ae853SAndroid Build Coastguard Worker                     ps_layer->pi1_mb_mode = api1_mb_mode[j];
1105*495ae853SAndroid Build Coastguard Worker                     ps_layer->i4_mb_mode_stride = ai4_mb_mode_stride[j];
1106*495ae853SAndroid Build Coastguard Worker                 }
1107*495ae853SAndroid Build Coastguard Worker             }
1108*495ae853SAndroid Build Coastguard Worker 
1109*495ae853SAndroid Build Coastguard Worker             ps_mb_res_buf = &ps_res_pred_ctxt->s_res_pred_outputs.s_res_pred;
1110*495ae853SAndroid Build Coastguard Worker             ps_mem_store = &ps_res_pred_state->s_mem_store;
1111*495ae853SAndroid Build Coastguard Worker             ps_proc->ps_mb_res_buf = ps_mb_res_buf;
1112*495ae853SAndroid Build Coastguard Worker 
1113*495ae853SAndroid Build Coastguard Worker             for(j = 0; j < NUM_SP_COMPONENTS; j++)
1114*495ae853SAndroid Build Coastguard Worker             {
1115*495ae853SAndroid Build Coastguard Worker                 buffer_container_t *ps_comp_buf = &ps_mb_res_buf->as_component_bufs[j];
1116*495ae853SAndroid Build Coastguard Worker 
1117*495ae853SAndroid Build Coastguard Worker                 UWORD8 u1_is_chroma = (Y != ((COMPONENT_TYPE) j));
1118*495ae853SAndroid Build Coastguard Worker 
1119*495ae853SAndroid Build Coastguard Worker                 ps_comp_buf->pv_data = pu1_buf;
1120*495ae853SAndroid Build Coastguard Worker                 ps_comp_buf->i4_data_stride = MB_SIZE;
1121*495ae853SAndroid Build Coastguard Worker                 pu1_buf += MB_SIZE * (MB_SIZE >> u1_is_chroma) * sizeof(WORD16);
1122*495ae853SAndroid Build Coastguard Worker                 i8_alloc_mem_size -= MB_SIZE * (MB_SIZE >> u1_is_chroma) * sizeof(WORD16);
1123*495ae853SAndroid Build Coastguard Worker             }
1124*495ae853SAndroid Build Coastguard Worker 
1125*495ae853SAndroid Build Coastguard Worker             ps_mem_store->s_scratch.pv_data = pu1_buf;
1126*495ae853SAndroid Build Coastguard Worker             ps_mem_store->s_scratch.i4_data_stride = REF_ARRAY_MAX_WIDTH;
1127*495ae853SAndroid Build Coastguard Worker             pu1_buf += REF_ARRAY_MAX_WIDTH * REF_ARRAY_MAX_HEIGHT * sizeof(WORD16);
1128*495ae853SAndroid Build Coastguard Worker             i8_alloc_mem_size -= REF_ARRAY_MAX_WIDTH * REF_ARRAY_MAX_HEIGHT * sizeof(WORD16);
1129*495ae853SAndroid Build Coastguard Worker 
1130*495ae853SAndroid Build Coastguard Worker             ASSERT(i8_alloc_mem_size >= 0);
1131*495ae853SAndroid Build Coastguard Worker 
1132*495ae853SAndroid Build Coastguard Worker             ps_mb_res_buf->as_component_bufs[V].pv_data = NULL;
1133*495ae853SAndroid Build Coastguard Worker             ps_mb_res_buf->e_color_format = IV_YUV_420SP_UV;
1134*495ae853SAndroid Build Coastguard Worker             ps_mb_res_buf->u1_bit_depth = 10;
1135*495ae853SAndroid Build Coastguard Worker             ps_mb_res_buf->u4_width = MB_SIZE;
1136*495ae853SAndroid Build Coastguard Worker             ps_mb_res_buf->u4_height = MB_SIZE;
1137*495ae853SAndroid Build Coastguard Worker 
1138*495ae853SAndroid Build Coastguard Worker             isvce_svc_residual_sampling_function_selector(ps_res_pred_state, d_spatial_res_ratio,
1139*495ae853SAndroid Build Coastguard Worker                                                           ps_codec->s_cfg.e_arch);
1140*495ae853SAndroid Build Coastguard Worker         }
1141*495ae853SAndroid Build Coastguard Worker     }
1142*495ae853SAndroid Build Coastguard Worker     else
1143*495ae853SAndroid Build Coastguard Worker     {
1144*495ae853SAndroid Build Coastguard Worker         for(i = 0; i < i4_num_proc_ctxts; i++)
1145*495ae853SAndroid Build Coastguard Worker         {
1146*495ae853SAndroid Build Coastguard Worker             isvce_process_ctxt_t *ps_proc = ps_codec->as_process + i;
1147*495ae853SAndroid Build Coastguard Worker 
1148*495ae853SAndroid Build Coastguard Worker             ps_proc->ps_res_pred_ctxt = NULL;
1149*495ae853SAndroid Build Coastguard Worker 
1150*495ae853SAndroid Build Coastguard Worker             ps_proc->ps_mb_res_buf = (yuv_buf_props_t *) pu1_buf;
1151*495ae853SAndroid Build Coastguard Worker             pu1_buf += sizeof(yuv_buf_props_t);
1152*495ae853SAndroid Build Coastguard Worker             i8_alloc_mem_size -= sizeof(yuv_buf_props_t);
1153*495ae853SAndroid Build Coastguard Worker 
1154*495ae853SAndroid Build Coastguard Worker             for(j = 0; j < NUM_SP_COMPONENTS; j++)
1155*495ae853SAndroid Build Coastguard Worker             {
1156*495ae853SAndroid Build Coastguard Worker                 buffer_container_t *ps_comp_buf = &ps_proc->ps_mb_res_buf->as_component_bufs[j];
1157*495ae853SAndroid Build Coastguard Worker 
1158*495ae853SAndroid Build Coastguard Worker                 UWORD8 u1_is_chroma = (Y != ((COMPONENT_TYPE) j));
1159*495ae853SAndroid Build Coastguard Worker 
1160*495ae853SAndroid Build Coastguard Worker                 ps_comp_buf->pv_data = pu1_buf;
1161*495ae853SAndroid Build Coastguard Worker                 ps_comp_buf->i4_data_stride = MB_SIZE;
1162*495ae853SAndroid Build Coastguard Worker                 pu1_buf += MB_SIZE * (MB_SIZE >> u1_is_chroma) * sizeof(WORD16);
1163*495ae853SAndroid Build Coastguard Worker                 i8_alloc_mem_size -= MB_SIZE * (MB_SIZE >> u1_is_chroma) * sizeof(WORD16);
1164*495ae853SAndroid Build Coastguard Worker             }
1165*495ae853SAndroid Build Coastguard Worker 
1166*495ae853SAndroid Build Coastguard Worker             ASSERT(i8_alloc_mem_size >= 0);
1167*495ae853SAndroid Build Coastguard Worker         }
1168*495ae853SAndroid Build Coastguard Worker     }
1169*495ae853SAndroid Build Coastguard Worker }
1170*495ae853SAndroid Build Coastguard Worker 
isvce_get_mb_residual_pred(svc_res_pred_ctxt_t * ps_res_pred_ctxt)1171*495ae853SAndroid Build Coastguard Worker void isvce_get_mb_residual_pred(svc_res_pred_ctxt_t *ps_res_pred_ctxt)
1172*495ae853SAndroid Build Coastguard Worker {
1173*495ae853SAndroid Build Coastguard Worker     buffer_container_t s_inp;
1174*495ae853SAndroid Build Coastguard Worker     buffer_container_t s_out;
1175*495ae853SAndroid Build Coastguard Worker     coordinates_t s_frame_dims;
1176*495ae853SAndroid Build Coastguard Worker     coordinates_t s_frame_dims_in_mbs;
1177*495ae853SAndroid Build Coastguard Worker     coordinates_t s_ref_array_offsets;
1178*495ae853SAndroid Build Coastguard Worker     svc_layer_data_t *ps_ref_layer_data;
1179*495ae853SAndroid Build Coastguard Worker     res_pred_layer_state_t *ps_layer_state;
1180*495ae853SAndroid Build Coastguard Worker     yuv_buf_props_t *ps_ref_residual_buf;
1181*495ae853SAndroid Build Coastguard Worker     res_pred_mb_state_t *ps_luma_mb_state;
1182*495ae853SAndroid Build Coastguard Worker     res_pred_mb_state_t *ps_chroma_mb_state;
1183*495ae853SAndroid Build Coastguard Worker     isvce_mb_info_t *ps_ref_mb;
1184*495ae853SAndroid Build Coastguard Worker 
1185*495ae853SAndroid Build Coastguard Worker     WORD32 i;
1186*495ae853SAndroid Build Coastguard Worker 
1187*495ae853SAndroid Build Coastguard Worker     res_pred_constants_t *ps_res_pred_constants = &ps_res_pred_ctxt->s_res_pred_constants;
1188*495ae853SAndroid Build Coastguard Worker     res_pred_variables_t *ps_res_pred_variables = &ps_res_pred_ctxt->s_res_pred_variables;
1189*495ae853SAndroid Build Coastguard Worker     res_pred_outputs_t *ps_res_pred_outputs = &ps_res_pred_ctxt->s_res_pred_outputs;
1190*495ae853SAndroid Build Coastguard Worker     res_pred_state_t *ps_res_pred_state = (res_pred_state_t *) ps_res_pred_constants->pv_state;
1191*495ae853SAndroid Build Coastguard Worker     res_pred_mem_store_t *ps_mem_store = &ps_res_pred_state->s_mem_store;
1192*495ae853SAndroid Build Coastguard Worker     svc_ilp_data_t *ps_svc_ilp_data = ps_res_pred_variables->ps_svc_ilp_data;
1193*495ae853SAndroid Build Coastguard Worker     coordinates_t *ps_mb_pos = &ps_res_pred_variables->s_mb_pos;
1194*495ae853SAndroid Build Coastguard Worker 
1195*495ae853SAndroid Build Coastguard Worker     UWORD8 u1_spatial_layer_id = ps_res_pred_variables->u1_spatial_layer_id;
1196*495ae853SAndroid Build Coastguard Worker 
1197*495ae853SAndroid Build Coastguard Worker     ASSERT(u1_spatial_layer_id > 0);
1198*495ae853SAndroid Build Coastguard Worker 
1199*495ae853SAndroid Build Coastguard Worker     s_frame_dims.i4_abscissa = ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id].u4_width;
1200*495ae853SAndroid Build Coastguard Worker     s_frame_dims.i4_ordinate = ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id].u4_height;
1201*495ae853SAndroid Build Coastguard Worker     s_frame_dims_in_mbs.i4_abscissa = s_frame_dims.i4_abscissa / MB_SIZE;
1202*495ae853SAndroid Build Coastguard Worker     s_frame_dims_in_mbs.i4_ordinate = s_frame_dims.i4_ordinate / MB_SIZE;
1203*495ae853SAndroid Build Coastguard Worker 
1204*495ae853SAndroid Build Coastguard Worker     ps_ref_layer_data =
1205*495ae853SAndroid Build Coastguard Worker         &ps_svc_ilp_data->ps_svc_au_data->ps_svc_layer_data[u1_spatial_layer_id - 1];
1206*495ae853SAndroid Build Coastguard Worker     ps_layer_state = &ps_res_pred_state->ps_layer_state[u1_spatial_layer_id];
1207*495ae853SAndroid Build Coastguard Worker     ps_ref_residual_buf = &ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id - 1];
1208*495ae853SAndroid Build Coastguard Worker     ps_luma_mb_state = ps_layer_state->ps_luma_mb_states + ps_mb_pos->i4_abscissa +
1209*495ae853SAndroid Build Coastguard Worker                        ps_mb_pos->i4_ordinate * s_frame_dims_in_mbs.i4_abscissa;
1210*495ae853SAndroid Build Coastguard Worker     ps_chroma_mb_state = ps_layer_state->ps_chroma_mb_states + ps_mb_pos->i4_abscissa +
1211*495ae853SAndroid Build Coastguard Worker                          ps_mb_pos->i4_ordinate * s_frame_dims_in_mbs.i4_abscissa;
1212*495ae853SAndroid Build Coastguard Worker 
1213*495ae853SAndroid Build Coastguard Worker     for(i = 0; i < NUM_COMPONENTS; i++)
1214*495ae853SAndroid Build Coastguard Worker     {
1215*495ae853SAndroid Build Coastguard Worker         res_pred_mb_state_t *ps_mb_state;
1216*495ae853SAndroid Build Coastguard Worker         layer_resampler_props_t *ps_layer_props;
1217*495ae853SAndroid Build Coastguard Worker 
1218*495ae853SAndroid Build Coastguard Worker         UWORD8 u1_is_chroma = (Y != ((COMPONENT_TYPE) i));
1219*495ae853SAndroid Build Coastguard Worker 
1220*495ae853SAndroid Build Coastguard Worker         ps_mb_state = u1_is_chroma ? ps_chroma_mb_state : ps_luma_mb_state;
1221*495ae853SAndroid Build Coastguard Worker         ps_layer_props =
1222*495ae853SAndroid Build Coastguard Worker             u1_is_chroma ? ps_layer_state->ps_chroma_props : ps_layer_state->ps_luma_props;
1223*495ae853SAndroid Build Coastguard Worker 
1224*495ae853SAndroid Build Coastguard Worker         /* Presence of appropriate padding is assumed */
1225*495ae853SAndroid Build Coastguard Worker         s_ref_array_offsets = ps_mb_state->s_offsets;
1226*495ae853SAndroid Build Coastguard Worker 
1227*495ae853SAndroid Build Coastguard Worker         s_inp = ps_ref_residual_buf->as_component_bufs[u1_is_chroma ? UV : Y];
1228*495ae853SAndroid Build Coastguard Worker         s_inp.pv_data = ((WORD16 *) s_inp.pv_data) + (V == ((COMPONENT_TYPE) i)) +
1229*495ae853SAndroid Build Coastguard Worker                         (s_ref_array_offsets.i4_abscissa << u1_is_chroma) +
1230*495ae853SAndroid Build Coastguard Worker                         s_ref_array_offsets.i4_ordinate * s_inp.i4_data_stride;
1231*495ae853SAndroid Build Coastguard Worker 
1232*495ae853SAndroid Build Coastguard Worker         s_out = ps_res_pred_outputs->s_res_pred.as_component_bufs[u1_is_chroma ? UV : Y];
1233*495ae853SAndroid Build Coastguard Worker         s_out.pv_data = ((WORD16 *) s_out.pv_data) + (V == ((COMPONENT_TYPE) i));
1234*495ae853SAndroid Build Coastguard Worker 
1235*495ae853SAndroid Build Coastguard Worker         ps_ref_mb =
1236*495ae853SAndroid Build Coastguard Worker             ps_ref_layer_data->ps_mb_info +
1237*495ae853SAndroid Build Coastguard Worker             ((s_ref_array_offsets.i4_abscissa + (ps_mb_state->s_ref_array_dims.i4_abscissa / 2)) /
1238*495ae853SAndroid Build Coastguard Worker              ps_layer_props->u4_mb_wd) +
1239*495ae853SAndroid Build Coastguard Worker             ((s_ref_array_offsets.i4_ordinate + (ps_mb_state->s_ref_array_dims.i4_ordinate / 2)) /
1240*495ae853SAndroid Build Coastguard Worker              ps_layer_props->u4_mb_ht) *
1241*495ae853SAndroid Build Coastguard Worker                 (s_frame_dims_in_mbs.i4_abscissa / 2);
1242*495ae853SAndroid Build Coastguard Worker 
1243*495ae853SAndroid Build Coastguard Worker         ps_res_pred_state->apf_residual_samplers[i](
1244*495ae853SAndroid Build Coastguard Worker             ps_mb_state->ps_ref_array_positions, ps_mb_state->ps_ref_array_phases, &s_inp, &s_out,
1245*495ae853SAndroid Build Coastguard Worker             &ps_mem_store->s_scratch, UINT32_MAX, ps_ref_mb->u1_tx_size == 8);
1246*495ae853SAndroid Build Coastguard Worker     }
1247*495ae853SAndroid Build Coastguard Worker }
1248*495ae853SAndroid Build Coastguard Worker 
isvce_get_ref_layer_mbtype_tx_size(WORD8 * pi1_ref_mb_modes,WORD32 i4_ref_mode_stride,WORD32 i4_element_size,WORD32 i4_x_ref,WORD32 i4_y_ref,WORD32 * pi4_mb_type,WORD32 * pi4_tx_size,WORD32 i4_chroma_flag)1249*495ae853SAndroid Build Coastguard Worker void isvce_get_ref_layer_mbtype_tx_size(WORD8 *pi1_ref_mb_modes, WORD32 i4_ref_mode_stride,
1250*495ae853SAndroid Build Coastguard Worker                                         WORD32 i4_element_size, WORD32 i4_x_ref, WORD32 i4_y_ref,
1251*495ae853SAndroid Build Coastguard Worker                                         WORD32 *pi4_mb_type, WORD32 *pi4_tx_size,
1252*495ae853SAndroid Build Coastguard Worker                                         WORD32 i4_chroma_flag)
1253*495ae853SAndroid Build Coastguard Worker {
1254*495ae853SAndroid Build Coastguard Worker     WORD32 i4_mb_wd_sft, i4_mb_ht_sft;
1255*495ae853SAndroid Build Coastguard Worker     WORD32 i4_mb_x, i4_mb_y;
1256*495ae853SAndroid Build Coastguard Worker     WORD8 i1_mb_mode;
1257*495ae853SAndroid Build Coastguard Worker 
1258*495ae853SAndroid Build Coastguard Worker     if(i4_x_ref < 0)
1259*495ae853SAndroid Build Coastguard Worker     {
1260*495ae853SAndroid Build Coastguard Worker         i4_x_ref = 0;
1261*495ae853SAndroid Build Coastguard Worker     }
1262*495ae853SAndroid Build Coastguard Worker     if(i4_y_ref < 0)
1263*495ae853SAndroid Build Coastguard Worker     {
1264*495ae853SAndroid Build Coastguard Worker         i4_y_ref = 0;
1265*495ae853SAndroid Build Coastguard Worker     }
1266*495ae853SAndroid Build Coastguard Worker 
1267*495ae853SAndroid Build Coastguard Worker     i4_mb_wd_sft = (MB_WIDTH_SHIFT - i4_chroma_flag);
1268*495ae853SAndroid Build Coastguard Worker     i4_mb_ht_sft = (MB_HEIGHT_SHIFT - i4_chroma_flag);
1269*495ae853SAndroid Build Coastguard Worker     i4_mb_x = (i4_x_ref >> i4_mb_wd_sft);
1270*495ae853SAndroid Build Coastguard Worker     i4_mb_y = (i4_y_ref >> i4_mb_ht_sft);
1271*495ae853SAndroid Build Coastguard Worker 
1272*495ae853SAndroid Build Coastguard Worker     pi1_ref_mb_modes += (i4_mb_y * i4_ref_mode_stride * i4_element_size);
1273*495ae853SAndroid Build Coastguard Worker     pi1_ref_mb_modes += (i4_mb_x * i4_element_size);
1274*495ae853SAndroid Build Coastguard Worker     i1_mb_mode = *pi1_ref_mb_modes;
1275*495ae853SAndroid Build Coastguard Worker     i1_mb_mode = (i1_mb_mode < 0) ? i1_mb_mode : SVC_EXTRACT_MB_MODE(*pi1_ref_mb_modes);
1276*495ae853SAndroid Build Coastguard Worker 
1277*495ae853SAndroid Build Coastguard Worker     if(i1_mb_mode <= SVC_INTER_MB)
1278*495ae853SAndroid Build Coastguard Worker     {
1279*495ae853SAndroid Build Coastguard Worker         *pi4_mb_type = SVC_INTER_MB;
1280*495ae853SAndroid Build Coastguard Worker         *pi4_tx_size = GET_BIT_TX_SIZE(*pi1_ref_mb_modes, 1);
1281*495ae853SAndroid Build Coastguard Worker     }
1282*495ae853SAndroid Build Coastguard Worker     else
1283*495ae853SAndroid Build Coastguard Worker     {
1284*495ae853SAndroid Build Coastguard Worker         *pi4_mb_type = SVC_INTRA_MB;
1285*495ae853SAndroid Build Coastguard Worker         *pi4_tx_size = 1;
1286*495ae853SAndroid Build Coastguard Worker     }
1287*495ae853SAndroid Build Coastguard Worker }
1288*495ae853SAndroid Build Coastguard Worker 
isvce_ref_layer_ptr_incr(WORD8 * pi1_ref_mb_modes,WORD32 i4_ref_mode_stride,WORD32 i4_element_size,WORD32 i4_x_offset,WORD32 i4_y_offset,WORD32 i4_refary_wd,WORD32 i4_refary_ht,UWORD8 * pu1_ref_x_ptr_incr,UWORD8 * pu1_ref_y_ptr_incr,WORD32 i4_chroma_flag)1289*495ae853SAndroid Build Coastguard Worker void isvce_ref_layer_ptr_incr(WORD8 *pi1_ref_mb_modes, WORD32 i4_ref_mode_stride,
1290*495ae853SAndroid Build Coastguard Worker                               WORD32 i4_element_size, WORD32 i4_x_offset, WORD32 i4_y_offset,
1291*495ae853SAndroid Build Coastguard Worker                               WORD32 i4_refary_wd, WORD32 i4_refary_ht, UWORD8 *pu1_ref_x_ptr_incr,
1292*495ae853SAndroid Build Coastguard Worker                               UWORD8 *pu1_ref_y_ptr_incr, WORD32 i4_chroma_flag)
1293*495ae853SAndroid Build Coastguard Worker {
1294*495ae853SAndroid Build Coastguard Worker     WORD32 i4_x, i4_y;
1295*495ae853SAndroid Build Coastguard Worker     WORD32 i4_x_idx, i4_y_idx;
1296*495ae853SAndroid Build Coastguard Worker     WORD32 i4_prev_x, i4_prev_y;
1297*495ae853SAndroid Build Coastguard Worker     WORD32 i4_const_val;
1298*495ae853SAndroid Build Coastguard Worker     WORD32 i4_pos_x, i4_pos_y;
1299*495ae853SAndroid Build Coastguard Worker     WORD32 i4_trans_size;
1300*495ae853SAndroid Build Coastguard Worker     WORD32 i4_mb_type, i4_tx_size;
1301*495ae853SAndroid Build Coastguard Worker     WORD32 i4_act_ary_wd, i4_act_ary_ht;
1302*495ae853SAndroid Build Coastguard Worker     WORD32 i4_and_const;
1303*495ae853SAndroid Build Coastguard Worker     UWORD8 *pu1_incr_x, *pu1_incr_y;
1304*495ae853SAndroid Build Coastguard Worker 
1305*495ae853SAndroid Build Coastguard Worker     memset(pu1_ref_x_ptr_incr, 1, (i4_refary_wd * i4_refary_ht));
1306*495ae853SAndroid Build Coastguard Worker     memset(pu1_ref_y_ptr_incr, 1, (i4_refary_wd * i4_refary_ht));
1307*495ae853SAndroid Build Coastguard Worker 
1308*495ae853SAndroid Build Coastguard Worker     i4_act_ary_wd = i4_refary_wd;
1309*495ae853SAndroid Build Coastguard Worker     i4_act_ary_ht = i4_refary_ht;
1310*495ae853SAndroid Build Coastguard Worker 
1311*495ae853SAndroid Build Coastguard Worker     i4_x = 0;
1312*495ae853SAndroid Build Coastguard Worker     i4_y = 0;
1313*495ae853SAndroid Build Coastguard Worker     i4_prev_y = 0;
1314*495ae853SAndroid Build Coastguard Worker 
1315*495ae853SAndroid Build Coastguard Worker     if(0 == i4_chroma_flag)
1316*495ae853SAndroid Build Coastguard Worker     {
1317*495ae853SAndroid Build Coastguard Worker         do
1318*495ae853SAndroid Build Coastguard Worker         {
1319*495ae853SAndroid Build Coastguard Worker             WORD32 i4_x_ref, i4_y_ref;
1320*495ae853SAndroid Build Coastguard Worker             WORD32 i4_idx;
1321*495ae853SAndroid Build Coastguard Worker             WORD32 i4_wd, i4_ht;
1322*495ae853SAndroid Build Coastguard Worker             WORD32 i4_max_pos_x, i4_max_pos_y;
1323*495ae853SAndroid Build Coastguard Worker 
1324*495ae853SAndroid Build Coastguard Worker             i4_prev_x = i4_x;
1325*495ae853SAndroid Build Coastguard Worker 
1326*495ae853SAndroid Build Coastguard Worker             i4_x_ref = i4_x_offset + i4_x;
1327*495ae853SAndroid Build Coastguard Worker             i4_y_ref = i4_y_offset + i4_y;
1328*495ae853SAndroid Build Coastguard Worker 
1329*495ae853SAndroid Build Coastguard Worker             isvce_get_ref_layer_mbtype_tx_size(pi1_ref_mb_modes, i4_ref_mode_stride,
1330*495ae853SAndroid Build Coastguard Worker                                                i4_element_size, i4_x_ref, i4_y_ref, &i4_mb_type,
1331*495ae853SAndroid Build Coastguard Worker                                                &i4_tx_size, i4_chroma_flag);
1332*495ae853SAndroid Build Coastguard Worker 
1333*495ae853SAndroid Build Coastguard Worker             i4_trans_size = ((i4_tx_size + 1) << 2);
1334*495ae853SAndroid Build Coastguard Worker             i4_const_val = i4_trans_size - 1;
1335*495ae853SAndroid Build Coastguard Worker             i4_and_const = i4_const_val;
1336*495ae853SAndroid Build Coastguard Worker 
1337*495ae853SAndroid Build Coastguard Worker             /* Fill horizontal tx block edges of current reference mb with 0 */
1338*495ae853SAndroid Build Coastguard Worker             pu1_incr_x = pu1_ref_x_ptr_incr + i4_x;
1339*495ae853SAndroid Build Coastguard Worker             pu1_incr_x += (i4_y * i4_refary_wd);
1340*495ae853SAndroid Build Coastguard Worker 
1341*495ae853SAndroid Build Coastguard Worker             i4_ht = (16 - (i4_y_ref & 0xF));
1342*495ae853SAndroid Build Coastguard Worker             i4_ht = MIN((i4_act_ary_ht - i4_y), i4_ht);
1343*495ae853SAndroid Build Coastguard Worker 
1344*495ae853SAndroid Build Coastguard Worker             i4_x_idx = i4_x;
1345*495ae853SAndroid Build Coastguard Worker 
1346*495ae853SAndroid Build Coastguard Worker             i4_pos_x = i4_x_ref & 0xF;
1347*495ae853SAndroid Build Coastguard Worker 
1348*495ae853SAndroid Build Coastguard Worker             i4_max_pos_x = 16;
1349*495ae853SAndroid Build Coastguard Worker             i4_x += (16 - i4_pos_x);
1350*495ae853SAndroid Build Coastguard Worker 
1351*495ae853SAndroid Build Coastguard Worker             /* Get the transform block edge pos */
1352*495ae853SAndroid Build Coastguard Worker             i4_idx = (i4_const_val - (i4_pos_x & i4_and_const));
1353*495ae853SAndroid Build Coastguard Worker 
1354*495ae853SAndroid Build Coastguard Worker             i4_x_idx += i4_idx;
1355*495ae853SAndroid Build Coastguard Worker 
1356*495ae853SAndroid Build Coastguard Worker             while((i4_pos_x < i4_max_pos_x) && (i4_x_idx < i4_act_ary_wd))
1357*495ae853SAndroid Build Coastguard Worker             {
1358*495ae853SAndroid Build Coastguard Worker                 WORD32 i4_i;
1359*495ae853SAndroid Build Coastguard Worker                 UWORD8 *pu1_incr;
1360*495ae853SAndroid Build Coastguard Worker 
1361*495ae853SAndroid Build Coastguard Worker                 pu1_incr = pu1_incr_x + i4_idx;
1362*495ae853SAndroid Build Coastguard Worker 
1363*495ae853SAndroid Build Coastguard Worker                 for(i4_i = 0; i4_i < i4_ht; i4_i++)
1364*495ae853SAndroid Build Coastguard Worker                 { /* Fill the block edge with 0s */
1365*495ae853SAndroid Build Coastguard Worker                     *pu1_incr = 0;
1366*495ae853SAndroid Build Coastguard Worker                     pu1_incr += i4_refary_wd;
1367*495ae853SAndroid Build Coastguard Worker                 }
1368*495ae853SAndroid Build Coastguard Worker 
1369*495ae853SAndroid Build Coastguard Worker                 i4_pos_x += i4_trans_size;
1370*495ae853SAndroid Build Coastguard Worker                 pu1_incr_x += i4_trans_size;
1371*495ae853SAndroid Build Coastguard Worker                 i4_x_idx += MIN(i4_trans_size, (i4_act_ary_wd - i4_x_idx));
1372*495ae853SAndroid Build Coastguard Worker             }
1373*495ae853SAndroid Build Coastguard Worker 
1374*495ae853SAndroid Build Coastguard Worker             /* Fill vertical tx block edges of current reference mb with 0 */
1375*495ae853SAndroid Build Coastguard Worker             pu1_incr_y = pu1_ref_y_ptr_incr + i4_prev_x;
1376*495ae853SAndroid Build Coastguard Worker             pu1_incr_y += (i4_y * i4_refary_wd);
1377*495ae853SAndroid Build Coastguard Worker 
1378*495ae853SAndroid Build Coastguard Worker             i4_wd = (16 - (i4_x_ref & 0xF));
1379*495ae853SAndroid Build Coastguard Worker             i4_wd = MIN((i4_act_ary_wd - i4_prev_x), i4_wd);
1380*495ae853SAndroid Build Coastguard Worker 
1381*495ae853SAndroid Build Coastguard Worker             i4_y_idx = i4_y;
1382*495ae853SAndroid Build Coastguard Worker 
1383*495ae853SAndroid Build Coastguard Worker             i4_pos_y = i4_y_ref & 0xF;
1384*495ae853SAndroid Build Coastguard Worker 
1385*495ae853SAndroid Build Coastguard Worker             i4_max_pos_y = 16;
1386*495ae853SAndroid Build Coastguard Worker             i4_y += (16 - i4_pos_y);
1387*495ae853SAndroid Build Coastguard Worker 
1388*495ae853SAndroid Build Coastguard Worker             /* Get the transform block edge pos */
1389*495ae853SAndroid Build Coastguard Worker             i4_idx = (i4_const_val - (i4_pos_y & i4_and_const));
1390*495ae853SAndroid Build Coastguard Worker 
1391*495ae853SAndroid Build Coastguard Worker             i4_y_idx += i4_idx;
1392*495ae853SAndroid Build Coastguard Worker 
1393*495ae853SAndroid Build Coastguard Worker             while((i4_pos_y < i4_max_pos_y) && (i4_y_idx < i4_act_ary_ht))
1394*495ae853SAndroid Build Coastguard Worker             {
1395*495ae853SAndroid Build Coastguard Worker                 WORD32 i4_i;
1396*495ae853SAndroid Build Coastguard Worker                 UWORD8 *pu1_incr;
1397*495ae853SAndroid Build Coastguard Worker 
1398*495ae853SAndroid Build Coastguard Worker                 pu1_incr = pu1_incr_y + i4_idx * i4_refary_wd;
1399*495ae853SAndroid Build Coastguard Worker 
1400*495ae853SAndroid Build Coastguard Worker                 for(i4_i = 0; i4_i < i4_wd; i4_i++)
1401*495ae853SAndroid Build Coastguard Worker                 { /* Fill the block edge with 0s */
1402*495ae853SAndroid Build Coastguard Worker                     *pu1_incr = 0;
1403*495ae853SAndroid Build Coastguard Worker                     pu1_incr++;
1404*495ae853SAndroid Build Coastguard Worker                 }
1405*495ae853SAndroid Build Coastguard Worker 
1406*495ae853SAndroid Build Coastguard Worker                 i4_pos_y += i4_trans_size;
1407*495ae853SAndroid Build Coastguard Worker                 pu1_incr_y += i4_trans_size * i4_refary_wd;
1408*495ae853SAndroid Build Coastguard Worker                 i4_y_idx += MIN(i4_trans_size, (i4_act_ary_ht - i4_y_idx));
1409*495ae853SAndroid Build Coastguard Worker             }
1410*495ae853SAndroid Build Coastguard Worker 
1411*495ae853SAndroid Build Coastguard Worker             if(i4_x < i4_act_ary_wd)
1412*495ae853SAndroid Build Coastguard Worker             {
1413*495ae853SAndroid Build Coastguard Worker                 i4_y = i4_prev_y;
1414*495ae853SAndroid Build Coastguard Worker             }
1415*495ae853SAndroid Build Coastguard Worker             else if(i4_y < i4_act_ary_ht)
1416*495ae853SAndroid Build Coastguard Worker             {
1417*495ae853SAndroid Build Coastguard Worker                 i4_prev_y = i4_y;
1418*495ae853SAndroid Build Coastguard Worker                 i4_x = 0;
1419*495ae853SAndroid Build Coastguard Worker             }
1420*495ae853SAndroid Build Coastguard Worker         } while((i4_y < i4_act_ary_ht) || (i4_x < i4_act_ary_wd));
1421*495ae853SAndroid Build Coastguard Worker     }
1422*495ae853SAndroid Build Coastguard Worker     else
1423*495ae853SAndroid Build Coastguard Worker     {
1424*495ae853SAndroid Build Coastguard Worker         i4_trans_size = 4;
1425*495ae853SAndroid Build Coastguard Worker         i4_const_val = 3;
1426*495ae853SAndroid Build Coastguard Worker 
1427*495ae853SAndroid Build Coastguard Worker         do
1428*495ae853SAndroid Build Coastguard Worker         {
1429*495ae853SAndroid Build Coastguard Worker             WORD32 i4_x_ref, i4_y_ref;
1430*495ae853SAndroid Build Coastguard Worker             WORD32 i4_idx;
1431*495ae853SAndroid Build Coastguard Worker             WORD32 i4_wd, i4_ht;
1432*495ae853SAndroid Build Coastguard Worker             WORD32 i4_max_pos_x, i4_max_pos_y;
1433*495ae853SAndroid Build Coastguard Worker 
1434*495ae853SAndroid Build Coastguard Worker             i4_prev_x = i4_x;
1435*495ae853SAndroid Build Coastguard Worker 
1436*495ae853SAndroid Build Coastguard Worker             i4_x_ref = i4_x_offset + i4_x;
1437*495ae853SAndroid Build Coastguard Worker             i4_y_ref = i4_y_offset + i4_y;
1438*495ae853SAndroid Build Coastguard Worker 
1439*495ae853SAndroid Build Coastguard Worker             /* Fill horizontal tx block edges of current reference mb with 0 */
1440*495ae853SAndroid Build Coastguard Worker             pu1_incr_x = pu1_ref_x_ptr_incr + i4_x;
1441*495ae853SAndroid Build Coastguard Worker             pu1_incr_x += (i4_y * i4_refary_wd);
1442*495ae853SAndroid Build Coastguard Worker 
1443*495ae853SAndroid Build Coastguard Worker             i4_ht = (8 - (i4_y_ref & 0x7));
1444*495ae853SAndroid Build Coastguard Worker             i4_ht = MIN((i4_act_ary_ht - i4_y), i4_ht);
1445*495ae853SAndroid Build Coastguard Worker 
1446*495ae853SAndroid Build Coastguard Worker             i4_x_idx = i4_x;
1447*495ae853SAndroid Build Coastguard Worker 
1448*495ae853SAndroid Build Coastguard Worker             i4_pos_x = i4_x_ref & 0x7;
1449*495ae853SAndroid Build Coastguard Worker 
1450*495ae853SAndroid Build Coastguard Worker             i4_max_pos_x = 8;
1451*495ae853SAndroid Build Coastguard Worker             i4_x += (8 - i4_pos_x);
1452*495ae853SAndroid Build Coastguard Worker 
1453*495ae853SAndroid Build Coastguard Worker             /* Get the transform block edge pos */
1454*495ae853SAndroid Build Coastguard Worker             i4_idx = (i4_const_val - (i4_pos_x & 0x3));
1455*495ae853SAndroid Build Coastguard Worker 
1456*495ae853SAndroid Build Coastguard Worker             i4_x_idx += i4_idx;
1457*495ae853SAndroid Build Coastguard Worker 
1458*495ae853SAndroid Build Coastguard Worker             while((i4_pos_x < i4_max_pos_x) && (i4_x_idx < i4_act_ary_wd))
1459*495ae853SAndroid Build Coastguard Worker             {
1460*495ae853SAndroid Build Coastguard Worker                 WORD32 i4_i;
1461*495ae853SAndroid Build Coastguard Worker                 UWORD8 *pu1_incr;
1462*495ae853SAndroid Build Coastguard Worker 
1463*495ae853SAndroid Build Coastguard Worker                 pu1_incr = pu1_incr_x + i4_idx;
1464*495ae853SAndroid Build Coastguard Worker 
1465*495ae853SAndroid Build Coastguard Worker                 for(i4_i = 0; i4_i < i4_ht; i4_i++)
1466*495ae853SAndroid Build Coastguard Worker                 { /* Fill the block edge with 0s */
1467*495ae853SAndroid Build Coastguard Worker                     *pu1_incr = 0;
1468*495ae853SAndroid Build Coastguard Worker                     pu1_incr += i4_refary_wd;
1469*495ae853SAndroid Build Coastguard Worker                 }
1470*495ae853SAndroid Build Coastguard Worker 
1471*495ae853SAndroid Build Coastguard Worker                 i4_pos_x += i4_trans_size;
1472*495ae853SAndroid Build Coastguard Worker                 pu1_incr_x += i4_trans_size;
1473*495ae853SAndroid Build Coastguard Worker                 i4_x_idx += MIN(i4_trans_size, (i4_act_ary_wd - i4_x_idx));
1474*495ae853SAndroid Build Coastguard Worker             }
1475*495ae853SAndroid Build Coastguard Worker 
1476*495ae853SAndroid Build Coastguard Worker             /* Fill vertical tx block edges of current reference mb with 0 */
1477*495ae853SAndroid Build Coastguard Worker             pu1_incr_y = pu1_ref_y_ptr_incr + i4_prev_x;
1478*495ae853SAndroid Build Coastguard Worker             pu1_incr_y += (i4_y * i4_refary_wd);
1479*495ae853SAndroid Build Coastguard Worker 
1480*495ae853SAndroid Build Coastguard Worker             i4_wd = (8 - (i4_x_ref & 0x7));
1481*495ae853SAndroid Build Coastguard Worker             i4_wd = MIN((i4_act_ary_wd - i4_prev_x), i4_wd);
1482*495ae853SAndroid Build Coastguard Worker 
1483*495ae853SAndroid Build Coastguard Worker             i4_y_idx = i4_y;
1484*495ae853SAndroid Build Coastguard Worker 
1485*495ae853SAndroid Build Coastguard Worker             i4_pos_y = i4_y_ref & 0x7;
1486*495ae853SAndroid Build Coastguard Worker 
1487*495ae853SAndroid Build Coastguard Worker             i4_max_pos_y = 8;
1488*495ae853SAndroid Build Coastguard Worker             i4_y += (8 - i4_pos_y);
1489*495ae853SAndroid Build Coastguard Worker 
1490*495ae853SAndroid Build Coastguard Worker             /* Get the transform block edge pos */
1491*495ae853SAndroid Build Coastguard Worker             i4_idx = (i4_const_val - (i4_pos_y & 0x3));
1492*495ae853SAndroid Build Coastguard Worker 
1493*495ae853SAndroid Build Coastguard Worker             i4_y_idx += i4_idx;
1494*495ae853SAndroid Build Coastguard Worker 
1495*495ae853SAndroid Build Coastguard Worker             while((i4_pos_y < i4_max_pos_y) && (i4_y_idx < i4_act_ary_ht))
1496*495ae853SAndroid Build Coastguard Worker             {
1497*495ae853SAndroid Build Coastguard Worker                 WORD32 i4_i;
1498*495ae853SAndroid Build Coastguard Worker                 UWORD8 *pu1_incr;
1499*495ae853SAndroid Build Coastguard Worker 
1500*495ae853SAndroid Build Coastguard Worker                 pu1_incr = pu1_incr_y + i4_idx * i4_refary_wd;
1501*495ae853SAndroid Build Coastguard Worker 
1502*495ae853SAndroid Build Coastguard Worker                 for(i4_i = 0; i4_i < i4_wd; i4_i++)
1503*495ae853SAndroid Build Coastguard Worker                 { /* Fill the block edge with 0s */
1504*495ae853SAndroid Build Coastguard Worker                     *pu1_incr = 0;
1505*495ae853SAndroid Build Coastguard Worker                     pu1_incr++;
1506*495ae853SAndroid Build Coastguard Worker                 }
1507*495ae853SAndroid Build Coastguard Worker 
1508*495ae853SAndroid Build Coastguard Worker                 i4_pos_y += i4_trans_size;
1509*495ae853SAndroid Build Coastguard Worker                 pu1_incr_y += i4_trans_size * i4_refary_wd;
1510*495ae853SAndroid Build Coastguard Worker                 i4_y_idx += MIN(i4_trans_size, (i4_act_ary_ht - i4_y_idx));
1511*495ae853SAndroid Build Coastguard Worker             }
1512*495ae853SAndroid Build Coastguard Worker 
1513*495ae853SAndroid Build Coastguard Worker             if(i4_x < i4_act_ary_wd)
1514*495ae853SAndroid Build Coastguard Worker             {
1515*495ae853SAndroid Build Coastguard Worker                 i4_y = i4_prev_y;
1516*495ae853SAndroid Build Coastguard Worker             }
1517*495ae853SAndroid Build Coastguard Worker             else if(i4_y < i4_act_ary_ht)
1518*495ae853SAndroid Build Coastguard Worker             {
1519*495ae853SAndroid Build Coastguard Worker                 i4_prev_y = i4_y;
1520*495ae853SAndroid Build Coastguard Worker                 i4_x = 0;
1521*495ae853SAndroid Build Coastguard Worker             }
1522*495ae853SAndroid Build Coastguard Worker         } while((i4_y < i4_act_ary_ht) || (i4_x < i4_act_ary_wd));
1523*495ae853SAndroid Build Coastguard Worker     }
1524*495ae853SAndroid Build Coastguard Worker }
1525*495ae853SAndroid Build Coastguard Worker 
isvce_residual_reflayer_const(svc_res_pred_ctxt_t * ps_res_pred_ctxt,WORD16 * pi2_inp_data,WORD32 i4_inp_data_stride,WORD8 * ps_ref_mb_mode,WORD32 i4_ref_mb_mode_stride,WORD32 * pi4_refarr_wd,WORD32 i4_chroma_flag)1526*495ae853SAndroid Build Coastguard Worker void isvce_residual_reflayer_const(svc_res_pred_ctxt_t *ps_res_pred_ctxt, WORD16 *pi2_inp_data,
1527*495ae853SAndroid Build Coastguard Worker                                    WORD32 i4_inp_data_stride, WORD8 *ps_ref_mb_mode,
1528*495ae853SAndroid Build Coastguard Worker                                    WORD32 i4_ref_mb_mode_stride, WORD32 *pi4_refarr_wd,
1529*495ae853SAndroid Build Coastguard Worker                                    WORD32 i4_chroma_flag)
1530*495ae853SAndroid Build Coastguard Worker {
1531*495ae853SAndroid Build Coastguard Worker     WORD8 *pi1_ref_mb_modes;
1532*495ae853SAndroid Build Coastguard Worker     WORD32 i4_ref_mode_stride;
1533*495ae853SAndroid Build Coastguard Worker 
1534*495ae853SAndroid Build Coastguard Worker     WORD32 i4_x, i4_y;
1535*495ae853SAndroid Build Coastguard Worker     WORD32 i4_ref_wd;
1536*495ae853SAndroid Build Coastguard Worker     WORD32 i4_ref_ht;
1537*495ae853SAndroid Build Coastguard Worker     WORD32 i4_x_offset;
1538*495ae853SAndroid Build Coastguard Worker     WORD32 i4_y_offset;
1539*495ae853SAndroid Build Coastguard Worker     WORD32 i4_refarray_wd;
1540*495ae853SAndroid Build Coastguard Worker     WORD32 i4_refarray_ht;
1541*495ae853SAndroid Build Coastguard Worker 
1542*495ae853SAndroid Build Coastguard Worker     WORD16 *pi2_ref_array;
1543*495ae853SAndroid Build Coastguard Worker 
1544*495ae853SAndroid Build Coastguard Worker     res_pred_mb_state_t *ps_mb_states;
1545*495ae853SAndroid Build Coastguard Worker     res_pred_layer_state_t *ps_layer_state;
1546*495ae853SAndroid Build Coastguard Worker 
1547*495ae853SAndroid Build Coastguard Worker     res_pred_constants_t *ps_res_pred_constants = &ps_res_pred_ctxt->s_res_pred_constants;
1548*495ae853SAndroid Build Coastguard Worker     res_pred_variables_t *ps_res_pred_variables = &ps_res_pred_ctxt->s_res_pred_variables;
1549*495ae853SAndroid Build Coastguard Worker     res_pred_state_t *ps_res_pred_state = (res_pred_state_t *) ps_res_pred_constants->pv_state;
1550*495ae853SAndroid Build Coastguard Worker     res_pred_mem_store_t *ps_mem_store = &ps_res_pred_state->s_mem_store;
1551*495ae853SAndroid Build Coastguard Worker     svc_ilp_data_t *ps_svc_ilp_data = ps_res_pred_variables->ps_svc_ilp_data;
1552*495ae853SAndroid Build Coastguard Worker     coordinates_t *ps_mb_pos = &ps_res_pred_variables->s_mb_pos;
1553*495ae853SAndroid Build Coastguard Worker 
1554*495ae853SAndroid Build Coastguard Worker     UWORD8 u1_spatial_layer_id = ps_res_pred_variables->u1_spatial_layer_id;
1555*495ae853SAndroid Build Coastguard Worker 
1556*495ae853SAndroid Build Coastguard Worker     ps_layer_state = &ps_res_pred_state->ps_layer_state[u1_spatial_layer_id];
1557*495ae853SAndroid Build Coastguard Worker     pi2_ref_array = (WORD16 *) ps_mem_store->s_scratch.pv_data;
1558*495ae853SAndroid Build Coastguard Worker 
1559*495ae853SAndroid Build Coastguard Worker     pi1_ref_mb_modes = (WORD8 *) ps_ref_mb_mode;
1560*495ae853SAndroid Build Coastguard Worker     i4_ref_mode_stride = i4_ref_mb_mode_stride;
1561*495ae853SAndroid Build Coastguard Worker 
1562*495ae853SAndroid Build Coastguard Worker     ASSERT(NULL != pi1_ref_mb_modes);
1563*495ae853SAndroid Build Coastguard Worker 
1564*495ae853SAndroid Build Coastguard Worker     {
1565*495ae853SAndroid Build Coastguard Worker         WORD32 i4_base_width;
1566*495ae853SAndroid Build Coastguard Worker         WORD32 i4_base_height;
1567*495ae853SAndroid Build Coastguard Worker 
1568*495ae853SAndroid Build Coastguard Worker         coordinates_t s_frame_dims, s_frame_dims_in_mbs;
1569*495ae853SAndroid Build Coastguard Worker 
1570*495ae853SAndroid Build Coastguard Worker         s_frame_dims.i4_abscissa = ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id].u4_width;
1571*495ae853SAndroid Build Coastguard Worker         s_frame_dims.i4_ordinate = ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id].u4_height;
1572*495ae853SAndroid Build Coastguard Worker         s_frame_dims_in_mbs.i4_abscissa = s_frame_dims.i4_abscissa / MB_SIZE;
1573*495ae853SAndroid Build Coastguard Worker         s_frame_dims_in_mbs.i4_ordinate = s_frame_dims.i4_ordinate / MB_SIZE;
1574*495ae853SAndroid Build Coastguard Worker 
1575*495ae853SAndroid Build Coastguard Worker         ps_mb_states = i4_chroma_flag ? ps_layer_state->ps_chroma_mb_states
1576*495ae853SAndroid Build Coastguard Worker                                       : ps_layer_state->ps_luma_mb_states;
1577*495ae853SAndroid Build Coastguard Worker 
1578*495ae853SAndroid Build Coastguard Worker         ps_mb_states +=
1579*495ae853SAndroid Build Coastguard Worker             ps_mb_pos->i4_abscissa + ps_mb_pos->i4_ordinate * s_frame_dims_in_mbs.i4_abscissa;
1580*495ae853SAndroid Build Coastguard Worker 
1581*495ae853SAndroid Build Coastguard Worker         i4_base_width = ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id - 1].u4_width;
1582*495ae853SAndroid Build Coastguard Worker         i4_base_height = ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id - 1].u4_height;
1583*495ae853SAndroid Build Coastguard Worker 
1584*495ae853SAndroid Build Coastguard Worker         i4_ref_wd = i4_base_width >> i4_chroma_flag;
1585*495ae853SAndroid Build Coastguard Worker         i4_ref_ht = i4_base_height >> i4_chroma_flag;
1586*495ae853SAndroid Build Coastguard Worker 
1587*495ae853SAndroid Build Coastguard Worker         i4_x_offset = ps_mb_states->s_offsets.i4_abscissa;
1588*495ae853SAndroid Build Coastguard Worker         i4_y_offset = ps_mb_states->s_offsets.i4_ordinate;
1589*495ae853SAndroid Build Coastguard Worker         i4_refarray_wd = ps_mb_states->s_ref_array_dims.i4_abscissa;
1590*495ae853SAndroid Build Coastguard Worker         i4_refarray_ht = ps_mb_states->s_ref_array_dims.i4_ordinate;
1591*495ae853SAndroid Build Coastguard Worker     }
1592*495ae853SAndroid Build Coastguard Worker 
1593*495ae853SAndroid Build Coastguard Worker     {
1594*495ae853SAndroid Build Coastguard Worker         isvce_ref_layer_ptr_incr(pi1_ref_mb_modes, i4_ref_mode_stride, 1, i4_x_offset, i4_y_offset,
1595*495ae853SAndroid Build Coastguard Worker                                  i4_refarray_wd, i4_refarray_ht,
1596*495ae853SAndroid Build Coastguard Worker                                  ps_res_pred_state->pu1_ref_x_ptr_incr,
1597*495ae853SAndroid Build Coastguard Worker                                  ps_res_pred_state->pu1_ref_y_ptr_incr, i4_chroma_flag);
1598*495ae853SAndroid Build Coastguard Worker     }
1599*495ae853SAndroid Build Coastguard Worker 
1600*495ae853SAndroid Build Coastguard Worker     for(i4_y = 0; i4_y < i4_refarray_ht; i4_y++)
1601*495ae853SAndroid Build Coastguard Worker     {
1602*495ae853SAndroid Build Coastguard Worker         for(i4_x = 0; i4_x < i4_refarray_wd; i4_x++)
1603*495ae853SAndroid Build Coastguard Worker         {
1604*495ae853SAndroid Build Coastguard Worker             WORD32 i4_x_ref;
1605*495ae853SAndroid Build Coastguard Worker             WORD32 i4_y_ref;
1606*495ae853SAndroid Build Coastguard Worker             WORD32 i4_ref_mb_type, i4_ref_tx_size;
1607*495ae853SAndroid Build Coastguard Worker             WORD16 *pi2_ref_data_byte;
1608*495ae853SAndroid Build Coastguard Worker             WORD16 *pi2_ref_array_temp;
1609*495ae853SAndroid Build Coastguard Worker 
1610*495ae853SAndroid Build Coastguard Worker             i4_x_ref = MAX(0, MIN(i4_ref_wd - 1, i4_x + i4_x_offset));
1611*495ae853SAndroid Build Coastguard Worker             i4_y_ref = MAX(0, MIN(i4_ref_ht - 1, i4_y + i4_y_offset));
1612*495ae853SAndroid Build Coastguard Worker 
1613*495ae853SAndroid Build Coastguard Worker             isvce_get_ref_layer_mbtype_tx_size(pi1_ref_mb_modes, i4_ref_mode_stride, 1, i4_x_ref,
1614*495ae853SAndroid Build Coastguard Worker                                                i4_y_ref, &i4_ref_mb_type, &i4_ref_tx_size,
1615*495ae853SAndroid Build Coastguard Worker                                                i4_chroma_flag);
1616*495ae853SAndroid Build Coastguard Worker 
1617*495ae853SAndroid Build Coastguard Worker             if(0 <= i4_x_offset)
1618*495ae853SAndroid Build Coastguard Worker             {
1619*495ae853SAndroid Build Coastguard Worker                 i4_x_ref = i4_x_ref - i4_x_offset;
1620*495ae853SAndroid Build Coastguard Worker             }
1621*495ae853SAndroid Build Coastguard Worker 
1622*495ae853SAndroid Build Coastguard Worker             if(0 <= i4_y_offset)
1623*495ae853SAndroid Build Coastguard Worker             {
1624*495ae853SAndroid Build Coastguard Worker                 i4_y_ref = i4_y_ref - i4_y_offset;
1625*495ae853SAndroid Build Coastguard Worker             }
1626*495ae853SAndroid Build Coastguard Worker 
1627*495ae853SAndroid Build Coastguard Worker             pi2_ref_array_temp = pi2_ref_array + i4_x;
1628*495ae853SAndroid Build Coastguard Worker             pi2_ref_array_temp += i4_y * i4_refarray_wd;
1629*495ae853SAndroid Build Coastguard Worker 
1630*495ae853SAndroid Build Coastguard Worker             if(SVC_INTER_MB == i4_ref_mb_type)
1631*495ae853SAndroid Build Coastguard Worker             {
1632*495ae853SAndroid Build Coastguard Worker                 pi2_ref_data_byte = pi2_inp_data + (i4_x_ref << i4_chroma_flag);
1633*495ae853SAndroid Build Coastguard Worker                 pi2_ref_data_byte += i4_y_ref * i4_inp_data_stride;
1634*495ae853SAndroid Build Coastguard Worker 
1635*495ae853SAndroid Build Coastguard Worker                 *pi2_ref_array_temp = (WORD16) (*pi2_ref_data_byte);
1636*495ae853SAndroid Build Coastguard Worker             }
1637*495ae853SAndroid Build Coastguard Worker             else
1638*495ae853SAndroid Build Coastguard Worker             {
1639*495ae853SAndroid Build Coastguard Worker                 *pi2_ref_array_temp = 0;
1640*495ae853SAndroid Build Coastguard Worker             }
1641*495ae853SAndroid Build Coastguard Worker         }
1642*495ae853SAndroid Build Coastguard Worker     }
1643*495ae853SAndroid Build Coastguard Worker     *pi4_refarr_wd = i4_refarray_wd;
1644*495ae853SAndroid Build Coastguard Worker }
1645*495ae853SAndroid Build Coastguard Worker 
isvce_interpolate_residual(svc_res_pred_ctxt_t * ps_res_pred_ctxt,WORD16 * pi2_out,WORD32 i4_out_stride,WORD32 i4_refarray_wd,WORD32 i4_chroma_flag,coordinates_t * ps_mb_pos)1646*495ae853SAndroid Build Coastguard Worker void isvce_interpolate_residual(svc_res_pred_ctxt_t *ps_res_pred_ctxt, WORD16 *pi2_out,
1647*495ae853SAndroid Build Coastguard Worker                                 WORD32 i4_out_stride, WORD32 i4_refarray_wd, WORD32 i4_chroma_flag,
1648*495ae853SAndroid Build Coastguard Worker                                 coordinates_t *ps_mb_pos)
1649*495ae853SAndroid Build Coastguard Worker {
1650*495ae853SAndroid Build Coastguard Worker     res_pred_constants_t *ps_res_pred_constants = &ps_res_pred_ctxt->s_res_pred_constants;
1651*495ae853SAndroid Build Coastguard Worker     res_pred_variables_t *ps_res_pred_variables = &ps_res_pred_ctxt->s_res_pred_variables;
1652*495ae853SAndroid Build Coastguard Worker     res_pred_state_t *ps_res_pred_state = (res_pred_state_t *) ps_res_pred_constants->pv_state;
1653*495ae853SAndroid Build Coastguard Worker     res_pred_mem_store_t *ps_mem_store = &ps_res_pred_state->s_mem_store;
1654*495ae853SAndroid Build Coastguard Worker 
1655*495ae853SAndroid Build Coastguard Worker     WORD32 i4_x, i4_y;
1656*495ae853SAndroid Build Coastguard Worker     WORD32 i4_temp_array_ht;
1657*495ae853SAndroid Build Coastguard Worker     WORD32 i4_mb_wd;
1658*495ae853SAndroid Build Coastguard Worker     WORD32 i4_mb_ht;
1659*495ae853SAndroid Build Coastguard Worker     WORD16 *pi2_ref_array;
1660*495ae853SAndroid Build Coastguard Worker     UWORD8 *pu1_ref_x_ptr_incr, *pu1_ref_y_ptr_incr;
1661*495ae853SAndroid Build Coastguard Worker 
1662*495ae853SAndroid Build Coastguard Worker     coordinates_t *ps_phase;
1663*495ae853SAndroid Build Coastguard Worker     coordinates_t *ps_pos;
1664*495ae853SAndroid Build Coastguard Worker     res_pred_mb_state_t *ps_mb_states;
1665*495ae853SAndroid Build Coastguard Worker 
1666*495ae853SAndroid Build Coastguard Worker     coordinates_t s_frame_dims;
1667*495ae853SAndroid Build Coastguard Worker     coordinates_t s_frame_dims_in_mbs;
1668*495ae853SAndroid Build Coastguard Worker 
1669*495ae853SAndroid Build Coastguard Worker     UWORD8 u1_spatial_layer_id = ps_res_pred_variables->u1_spatial_layer_id;
1670*495ae853SAndroid Build Coastguard Worker 
1671*495ae853SAndroid Build Coastguard Worker     svc_ilp_data_t *ps_svc_ilp_data = ps_res_pred_variables->ps_svc_ilp_data;
1672*495ae853SAndroid Build Coastguard Worker 
1673*495ae853SAndroid Build Coastguard Worker     res_pred_mb_state_t *ps_mb_state;
1674*495ae853SAndroid Build Coastguard Worker 
1675*495ae853SAndroid Build Coastguard Worker     s_frame_dims.i4_abscissa = ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id].u4_width;
1676*495ae853SAndroid Build Coastguard Worker     s_frame_dims.i4_ordinate = ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id].u4_height;
1677*495ae853SAndroid Build Coastguard Worker     s_frame_dims_in_mbs.i4_abscissa = s_frame_dims.i4_abscissa / MB_SIZE;
1678*495ae853SAndroid Build Coastguard Worker     s_frame_dims_in_mbs.i4_ordinate = s_frame_dims.i4_ordinate / MB_SIZE;
1679*495ae853SAndroid Build Coastguard Worker 
1680*495ae853SAndroid Build Coastguard Worker     pu1_ref_x_ptr_incr = ps_res_pred_state->pu1_ref_x_ptr_incr;
1681*495ae853SAndroid Build Coastguard Worker     pu1_ref_y_ptr_incr = ps_res_pred_state->pu1_ref_y_ptr_incr;
1682*495ae853SAndroid Build Coastguard Worker 
1683*495ae853SAndroid Build Coastguard Worker     ps_mb_states = i4_chroma_flag
1684*495ae853SAndroid Build Coastguard Worker                        ? ps_res_pred_state->ps_layer_state[u1_spatial_layer_id].ps_chroma_mb_states
1685*495ae853SAndroid Build Coastguard Worker                        : ps_res_pred_state->ps_layer_state[u1_spatial_layer_id].ps_luma_mb_states;
1686*495ae853SAndroid Build Coastguard Worker 
1687*495ae853SAndroid Build Coastguard Worker     i4_mb_wd = MB_SIZE >> i4_chroma_flag;
1688*495ae853SAndroid Build Coastguard Worker     i4_mb_ht = MB_SIZE >> i4_chroma_flag;
1689*495ae853SAndroid Build Coastguard Worker 
1690*495ae853SAndroid Build Coastguard Worker     ps_mb_state = &ps_mb_states[ps_mb_pos->i4_abscissa +
1691*495ae853SAndroid Build Coastguard Worker                                 (ps_mb_pos->i4_ordinate * s_frame_dims_in_mbs.i4_abscissa)];
1692*495ae853SAndroid Build Coastguard Worker 
1693*495ae853SAndroid Build Coastguard Worker     ps_phase = ps_mb_state->ps_ref_array_phases;
1694*495ae853SAndroid Build Coastguard Worker     ps_pos = ps_mb_state->ps_ref_array_positions;
1695*495ae853SAndroid Build Coastguard Worker 
1696*495ae853SAndroid Build Coastguard Worker     i4_temp_array_ht = i4_mb_ht;
1697*495ae853SAndroid Build Coastguard Worker 
1698*495ae853SAndroid Build Coastguard Worker     pi2_ref_array = (WORD16 *) ps_mem_store->s_scratch.pv_data;
1699*495ae853SAndroid Build Coastguard Worker 
1700*495ae853SAndroid Build Coastguard Worker     for(i4_y = 0; i4_y < i4_temp_array_ht; i4_y++)
1701*495ae853SAndroid Build Coastguard Worker     {
1702*495ae853SAndroid Build Coastguard Worker         for(i4_x = 0; i4_x < i4_mb_wd; i4_x++)
1703*495ae853SAndroid Build Coastguard Worker         {
1704*495ae853SAndroid Build Coastguard Worker             WORD32 i4_i;
1705*495ae853SAndroid Build Coastguard Worker             WORD32 i4_y_ref;
1706*495ae853SAndroid Build Coastguard Worker             WORD32 i4_y_phase;
1707*495ae853SAndroid Build Coastguard Worker             WORD32 i4_x_ref;
1708*495ae853SAndroid Build Coastguard Worker             WORD32 i4_x_phase;
1709*495ae853SAndroid Build Coastguard Worker             WORD32 i4_x_ref_round;
1710*495ae853SAndroid Build Coastguard Worker             WORD16 *pi2_out_curr;
1711*495ae853SAndroid Build Coastguard Worker             WORD32 ai4_temp_pred[2];
1712*495ae853SAndroid Build Coastguard Worker             UWORD8 *pu1_ref_y_ptr_incr_temp;
1713*495ae853SAndroid Build Coastguard Worker             WORD32 *pi4_temp_pred;
1714*495ae853SAndroid Build Coastguard Worker             UWORD8 u1_incr_y;
1715*495ae853SAndroid Build Coastguard Worker             WORD16 i2_res;
1716*495ae853SAndroid Build Coastguard Worker 
1717*495ae853SAndroid Build Coastguard Worker             pi2_out_curr = pi2_out + (i4_x << i4_chroma_flag) + (i4_y * i4_out_stride);
1718*495ae853SAndroid Build Coastguard Worker 
1719*495ae853SAndroid Build Coastguard Worker             i4_y_ref = ps_pos[(i4_mb_wd * i4_y) + i4_x].i4_ordinate;
1720*495ae853SAndroid Build Coastguard Worker             i4_y_phase = ps_phase[((i4_y % 3) > 0) * 2 + (i4_y % 3)].i4_ordinate;
1721*495ae853SAndroid Build Coastguard Worker 
1722*495ae853SAndroid Build Coastguard Worker             i4_x_ref = ps_pos[(i4_mb_wd * i4_y) + i4_x].i4_abscissa;
1723*495ae853SAndroid Build Coastguard Worker             i4_x_phase = ps_phase[i4_x % 3].i4_abscissa;
1724*495ae853SAndroid Build Coastguard Worker 
1725*495ae853SAndroid Build Coastguard Worker             /* horizontal processing*/
1726*495ae853SAndroid Build Coastguard Worker             for(i4_i = 0; i4_i < 2; i4_i++)
1727*495ae853SAndroid Build Coastguard Worker             {
1728*495ae853SAndroid Build Coastguard Worker                 UWORD8 *pu1_ref_x_ptr_incr_temp;
1729*495ae853SAndroid Build Coastguard Worker                 UWORD8 u1_incr;
1730*495ae853SAndroid Build Coastguard Worker                 WORD16 *pi2_ref_array_1, *pi2_ref_array_2;
1731*495ae853SAndroid Build Coastguard Worker 
1732*495ae853SAndroid Build Coastguard Worker                 pu1_ref_x_ptr_incr_temp = pu1_ref_x_ptr_incr + i4_x_ref;
1733*495ae853SAndroid Build Coastguard Worker                 pu1_ref_x_ptr_incr_temp += ((i4_y_ref + i4_i) * i4_refarray_wd);
1734*495ae853SAndroid Build Coastguard Worker                 u1_incr = *pu1_ref_x_ptr_incr_temp;
1735*495ae853SAndroid Build Coastguard Worker 
1736*495ae853SAndroid Build Coastguard Worker                 pi2_ref_array_1 = pi2_ref_array + i4_x_ref;
1737*495ae853SAndroid Build Coastguard Worker                 pi2_ref_array_1 += ((i4_y_ref + i4_i) * i4_refarray_wd);
1738*495ae853SAndroid Build Coastguard Worker 
1739*495ae853SAndroid Build Coastguard Worker                 if(!u1_incr)
1740*495ae853SAndroid Build Coastguard Worker                 {
1741*495ae853SAndroid Build Coastguard Worker                     pi2_ref_array_1 += (i4_x_phase >> 3);
1742*495ae853SAndroid Build Coastguard Worker                 }
1743*495ae853SAndroid Build Coastguard Worker 
1744*495ae853SAndroid Build Coastguard Worker                 pi2_ref_array_2 = pi2_ref_array_1 + u1_incr;
1745*495ae853SAndroid Build Coastguard Worker 
1746*495ae853SAndroid Build Coastguard Worker                 ai4_temp_pred[i4_i] =
1747*495ae853SAndroid Build Coastguard Worker                     (16 - i4_x_phase) * (*pi2_ref_array_1) + i4_x_phase * (*pi2_ref_array_2);
1748*495ae853SAndroid Build Coastguard Worker             }
1749*495ae853SAndroid Build Coastguard Worker 
1750*495ae853SAndroid Build Coastguard Worker             /* vertical processing */
1751*495ae853SAndroid Build Coastguard Worker             i4_x_ref_round = (i4_x_ref + (i4_x_phase >> 3));
1752*495ae853SAndroid Build Coastguard Worker 
1753*495ae853SAndroid Build Coastguard Worker             pu1_ref_y_ptr_incr_temp =
1754*495ae853SAndroid Build Coastguard Worker                 pu1_ref_y_ptr_incr + i4_x_ref_round + (i4_y_ref * i4_refarray_wd);
1755*495ae853SAndroid Build Coastguard Worker             u1_incr_y = *pu1_ref_y_ptr_incr_temp;
1756*495ae853SAndroid Build Coastguard Worker 
1757*495ae853SAndroid Build Coastguard Worker             pi4_temp_pred = &ai4_temp_pred[0];
1758*495ae853SAndroid Build Coastguard Worker             if(!u1_incr_y)
1759*495ae853SAndroid Build Coastguard Worker             {
1760*495ae853SAndroid Build Coastguard Worker                 pi4_temp_pred += (i4_y_phase >> 3);
1761*495ae853SAndroid Build Coastguard Worker             }
1762*495ae853SAndroid Build Coastguard Worker             i2_res = (((16 - i4_y_phase) * pi4_temp_pred[0] +
1763*495ae853SAndroid Build Coastguard Worker                        i4_y_phase * pi4_temp_pred[u1_incr_y] + 128) >>
1764*495ae853SAndroid Build Coastguard Worker                       8);
1765*495ae853SAndroid Build Coastguard Worker             *pi2_out_curr = i2_res;
1766*495ae853SAndroid Build Coastguard Worker         }
1767*495ae853SAndroid Build Coastguard Worker     }
1768*495ae853SAndroid Build Coastguard Worker }
1769*495ae853SAndroid Build Coastguard Worker 
isvce_get_mb_residual_pred_non_dyadic(svc_res_pred_ctxt_t * ps_res_pred_ctxt)1770*495ae853SAndroid Build Coastguard Worker void isvce_get_mb_residual_pred_non_dyadic(svc_res_pred_ctxt_t *ps_res_pred_ctxt)
1771*495ae853SAndroid Build Coastguard Worker {
1772*495ae853SAndroid Build Coastguard Worker     buffer_container_t s_inp;
1773*495ae853SAndroid Build Coastguard Worker     buffer_container_t s_out;
1774*495ae853SAndroid Build Coastguard Worker     coordinates_t s_frame_dims;
1775*495ae853SAndroid Build Coastguard Worker     coordinates_t s_frame_dims_in_mbs;
1776*495ae853SAndroid Build Coastguard Worker     coordinates_t s_ref_array_offsets;
1777*495ae853SAndroid Build Coastguard Worker     res_pred_layer_state_t *ps_layer_state, *ps_ref_layer_state;
1778*495ae853SAndroid Build Coastguard Worker     yuv_buf_props_t *ps_ref_residual_buf;
1779*495ae853SAndroid Build Coastguard Worker     res_pred_mb_state_t *ps_luma_mb_state;
1780*495ae853SAndroid Build Coastguard Worker     res_pred_mb_state_t *ps_chroma_mb_state;
1781*495ae853SAndroid Build Coastguard Worker 
1782*495ae853SAndroid Build Coastguard Worker     WORD16 *pi2_inp, *pi2_out;
1783*495ae853SAndroid Build Coastguard Worker     WORD32 i4_inp_stride, i4_out_stride;
1784*495ae853SAndroid Build Coastguard Worker 
1785*495ae853SAndroid Build Coastguard Worker     res_pred_constants_t *ps_res_pred_constants = &ps_res_pred_ctxt->s_res_pred_constants;
1786*495ae853SAndroid Build Coastguard Worker     res_pred_variables_t *ps_res_pred_variables = &ps_res_pred_ctxt->s_res_pred_variables;
1787*495ae853SAndroid Build Coastguard Worker     res_pred_outputs_t *ps_res_pred_outputs = &ps_res_pred_ctxt->s_res_pred_outputs;
1788*495ae853SAndroid Build Coastguard Worker     res_pred_state_t *ps_res_pred_state = (res_pred_state_t *) ps_res_pred_constants->pv_state;
1789*495ae853SAndroid Build Coastguard Worker     svc_ilp_data_t *ps_svc_ilp_data = ps_res_pred_variables->ps_svc_ilp_data;
1790*495ae853SAndroid Build Coastguard Worker     coordinates_t *ps_mb_pos = &ps_res_pred_variables->s_mb_pos;
1791*495ae853SAndroid Build Coastguard Worker 
1792*495ae853SAndroid Build Coastguard Worker     UWORD8 u1_spatial_layer_id = ps_res_pred_variables->u1_spatial_layer_id;
1793*495ae853SAndroid Build Coastguard Worker 
1794*495ae853SAndroid Build Coastguard Worker     WORD32 i4_refarray_wd;
1795*495ae853SAndroid Build Coastguard Worker 
1796*495ae853SAndroid Build Coastguard Worker     WORD32 i;
1797*495ae853SAndroid Build Coastguard Worker 
1798*495ae853SAndroid Build Coastguard Worker     ASSERT(u1_spatial_layer_id > 0);
1799*495ae853SAndroid Build Coastguard Worker 
1800*495ae853SAndroid Build Coastguard Worker     s_frame_dims.i4_abscissa = ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id].u4_width;
1801*495ae853SAndroid Build Coastguard Worker     s_frame_dims.i4_ordinate = ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id].u4_height;
1802*495ae853SAndroid Build Coastguard Worker     s_frame_dims_in_mbs.i4_abscissa = s_frame_dims.i4_abscissa / MB_SIZE;
1803*495ae853SAndroid Build Coastguard Worker     s_frame_dims_in_mbs.i4_ordinate = s_frame_dims.i4_ordinate / MB_SIZE;
1804*495ae853SAndroid Build Coastguard Worker 
1805*495ae853SAndroid Build Coastguard Worker     ps_layer_state = &ps_res_pred_state->ps_layer_state[u1_spatial_layer_id];
1806*495ae853SAndroid Build Coastguard Worker     ps_ref_layer_state = &ps_res_pred_state->ps_layer_state[u1_spatial_layer_id - 1];
1807*495ae853SAndroid Build Coastguard Worker     ps_ref_residual_buf = &ps_svc_ilp_data->ps_residual_bufs[u1_spatial_layer_id - 1];
1808*495ae853SAndroid Build Coastguard Worker     ps_luma_mb_state = ps_layer_state->ps_luma_mb_states + ps_mb_pos->i4_abscissa +
1809*495ae853SAndroid Build Coastguard Worker                        ps_mb_pos->i4_ordinate * s_frame_dims_in_mbs.i4_abscissa;
1810*495ae853SAndroid Build Coastguard Worker     ps_chroma_mb_state = ps_layer_state->ps_chroma_mb_states + ps_mb_pos->i4_abscissa +
1811*495ae853SAndroid Build Coastguard Worker                          ps_mb_pos->i4_ordinate * s_frame_dims_in_mbs.i4_abscissa;
1812*495ae853SAndroid Build Coastguard Worker 
1813*495ae853SAndroid Build Coastguard Worker     for(i = 0; i < NUM_COMPONENTS; i++)
1814*495ae853SAndroid Build Coastguard Worker     {
1815*495ae853SAndroid Build Coastguard Worker         res_pred_mb_state_t *ps_mb_state;
1816*495ae853SAndroid Build Coastguard Worker 
1817*495ae853SAndroid Build Coastguard Worker         UWORD8 u1_is_chroma = (Y != ((COMPONENT_TYPE) i));
1818*495ae853SAndroid Build Coastguard Worker 
1819*495ae853SAndroid Build Coastguard Worker         ps_mb_state = u1_is_chroma ? ps_chroma_mb_state : ps_luma_mb_state;
1820*495ae853SAndroid Build Coastguard Worker 
1821*495ae853SAndroid Build Coastguard Worker         s_ref_array_offsets.i4_abscissa =
1822*495ae853SAndroid Build Coastguard Worker             MAX(0, MIN(ps_mb_state->s_offsets.i4_abscissa,
1823*495ae853SAndroid Build Coastguard Worker                        (s_frame_dims.i4_abscissa >> u1_is_chroma) - 1));
1824*495ae853SAndroid Build Coastguard Worker         s_ref_array_offsets.i4_ordinate =
1825*495ae853SAndroid Build Coastguard Worker             MAX(0, MIN(ps_mb_state->s_offsets.i4_ordinate,
1826*495ae853SAndroid Build Coastguard Worker                        (s_frame_dims.i4_ordinate >> u1_is_chroma) - 1));
1827*495ae853SAndroid Build Coastguard Worker 
1828*495ae853SAndroid Build Coastguard Worker         s_inp = ps_ref_residual_buf->as_component_bufs[u1_is_chroma ? UV : Y];
1829*495ae853SAndroid Build Coastguard Worker         s_inp.pv_data = ((WORD16 *) s_inp.pv_data) + (V == ((COMPONENT_TYPE) i)) +
1830*495ae853SAndroid Build Coastguard Worker                         (s_ref_array_offsets.i4_abscissa << u1_is_chroma) +
1831*495ae853SAndroid Build Coastguard Worker                         s_ref_array_offsets.i4_ordinate * s_inp.i4_data_stride;
1832*495ae853SAndroid Build Coastguard Worker 
1833*495ae853SAndroid Build Coastguard Worker         s_out = ps_res_pred_outputs->s_res_pred.as_component_bufs[u1_is_chroma ? UV : Y];
1834*495ae853SAndroid Build Coastguard Worker         s_out.pv_data = ((WORD16 *) s_out.pv_data) + (V == ((COMPONENT_TYPE) i));
1835*495ae853SAndroid Build Coastguard Worker 
1836*495ae853SAndroid Build Coastguard Worker         pi2_inp = (WORD16 *) s_inp.pv_data;
1837*495ae853SAndroid Build Coastguard Worker         pi2_out = (WORD16 *) s_out.pv_data;
1838*495ae853SAndroid Build Coastguard Worker 
1839*495ae853SAndroid Build Coastguard Worker         i4_inp_stride = s_inp.i4_data_stride;
1840*495ae853SAndroid Build Coastguard Worker         i4_out_stride = s_out.i4_data_stride;
1841*495ae853SAndroid Build Coastguard Worker 
1842*495ae853SAndroid Build Coastguard Worker         /* ------- Constructing refSampleArray ----------------------- */
1843*495ae853SAndroid Build Coastguard Worker         isvce_residual_reflayer_const(
1844*495ae853SAndroid Build Coastguard Worker             ps_res_pred_ctxt, pi2_inp, i4_inp_stride, ps_ref_layer_state->pi1_mb_mode,
1845*495ae853SAndroid Build Coastguard Worker             ps_ref_layer_state->i4_mb_mode_stride, &i4_refarray_wd, u1_is_chroma);
1846*495ae853SAndroid Build Coastguard Worker 
1847*495ae853SAndroid Build Coastguard Worker         /* ---- Interpolation process for Residual prediction	 ------ */
1848*495ae853SAndroid Build Coastguard Worker         isvce_interpolate_residual(ps_res_pred_ctxt, pi2_out, i4_out_stride, i4_refarray_wd,
1849*495ae853SAndroid Build Coastguard Worker                                    u1_is_chroma, ps_mb_pos);
1850*495ae853SAndroid Build Coastguard Worker     }
1851*495ae853SAndroid Build Coastguard Worker }
1852*495ae853SAndroid Build Coastguard Worker 
isvce_get_sad_with_residual_pred(buffer_container_t * ps_src,buffer_container_t * ps_pred,buffer_container_t * ps_res,UWORD32 u4_mb_wd,UWORD32 u4_mb_ht)1853*495ae853SAndroid Build Coastguard Worker UWORD32 isvce_get_sad_with_residual_pred(buffer_container_t *ps_src, buffer_container_t *ps_pred,
1854*495ae853SAndroid Build Coastguard Worker                                          buffer_container_t *ps_res, UWORD32 u4_mb_wd,
1855*495ae853SAndroid Build Coastguard Worker                                          UWORD32 u4_mb_ht)
1856*495ae853SAndroid Build Coastguard Worker {
1857*495ae853SAndroid Build Coastguard Worker     UWORD32 i, j;
1858*495ae853SAndroid Build Coastguard Worker 
1859*495ae853SAndroid Build Coastguard Worker     UWORD32 u4_sad = 0;
1860*495ae853SAndroid Build Coastguard Worker 
1861*495ae853SAndroid Build Coastguard Worker     for(i = 0; i < u4_mb_ht; i++)
1862*495ae853SAndroid Build Coastguard Worker     {
1863*495ae853SAndroid Build Coastguard Worker         for(j = 0; j < u4_mb_wd; j++)
1864*495ae853SAndroid Build Coastguard Worker         {
1865*495ae853SAndroid Build Coastguard Worker             WORD16 i2_src = ((UWORD8 *) ps_src->pv_data)[j + i * ps_src->i4_data_stride];
1866*495ae853SAndroid Build Coastguard Worker             WORD16 i2_pred = ((UWORD8 *) ps_pred->pv_data)[j + i * ps_pred->i4_data_stride];
1867*495ae853SAndroid Build Coastguard Worker             WORD16 i2_res = ((WORD16 *) ps_res->pv_data)[j + i * ps_res->i4_data_stride];
1868*495ae853SAndroid Build Coastguard Worker 
1869*495ae853SAndroid Build Coastguard Worker             u4_sad += ABS(i2_src - i2_pred - i2_res);
1870*495ae853SAndroid Build Coastguard Worker         }
1871*495ae853SAndroid Build Coastguard Worker     }
1872*495ae853SAndroid Build Coastguard Worker     return u4_sad;
1873*495ae853SAndroid Build Coastguard Worker }
1874*495ae853SAndroid Build Coastguard Worker 
1875*495ae853SAndroid Build Coastguard Worker /**
1876*495ae853SAndroid Build Coastguard Worker *******************************************************************************
1877*495ae853SAndroid Build Coastguard Worker *
1878*495ae853SAndroid Build Coastguard Worker * @brief
1879*495ae853SAndroid Build Coastguard Worker *  Function to evaluate residual_prediction_flag
1880*495ae853SAndroid Build Coastguard Worker *
1881*495ae853SAndroid Build Coastguard Worker * @param[in] ps_src
1882*495ae853SAndroid Build Coastguard Worker *  Pointer to MB src buffers
1883*495ae853SAndroid Build Coastguard Worker *
1884*495ae853SAndroid Build Coastguard Worker * @param[in] ps_pred
1885*495ae853SAndroid Build Coastguard Worker *  Pointer to MB pred buffers
1886*495ae853SAndroid Build Coastguard Worker *
1887*495ae853SAndroid Build Coastguard Worker * @param[in] ps_res
1888*495ae853SAndroid Build Coastguard Worker *  Pointer to MB res buffers
1889*495ae853SAndroid Build Coastguard Worker *
1890*495ae853SAndroid Build Coastguard Worker * @param[out] pu4_res_pred_sad
1891*495ae853SAndroid Build Coastguard Worker *  Output variable for SAD
1892*495ae853SAndroid Build Coastguard Worker *
1893*495ae853SAndroid Build Coastguard Worker * @param[out] pu1_residual_prediction_flag
1894*495ae853SAndroid Build Coastguard Worker *  Output variable for residual_prediction_flag
1895*495ae853SAndroid Build Coastguard Worker *
1896*495ae853SAndroid Build Coastguard Worker * @param[in] u4_winning_sad
1897*495ae853SAndroid Build Coastguard Worker *  Winning mode's SAD
1898*495ae853SAndroid Build Coastguard Worker *
1899*495ae853SAndroid Build Coastguard Worker * @notes The algorithm currently uses only luma for evaluating
1900*495ae853SAndroid Build Coastguard Worker *        residual_prediction_flag.
1901*495ae853SAndroid Build Coastguard Worker *
1902*495ae853SAndroid Build Coastguard Worker *******************************************************************************
1903*495ae853SAndroid Build Coastguard Worker */
isvce_residual_pred_eval(svc_res_pred_ctxt_t * ps_res_pred_ctxt,yuv_buf_props_t * ps_src,yuv_buf_props_t * ps_pred,yuv_buf_props_t * ps_res,UWORD32 * pu4_res_pred_sad,UWORD8 * pu1_residual_prediction_flag,UWORD32 u4_winning_sad)1904*495ae853SAndroid Build Coastguard Worker void isvce_residual_pred_eval(svc_res_pred_ctxt_t *ps_res_pred_ctxt, yuv_buf_props_t *ps_src,
1905*495ae853SAndroid Build Coastguard Worker                               yuv_buf_props_t *ps_pred, yuv_buf_props_t *ps_res,
1906*495ae853SAndroid Build Coastguard Worker                               UWORD32 *pu4_res_pred_sad, UWORD8 *pu1_residual_prediction_flag,
1907*495ae853SAndroid Build Coastguard Worker                               UWORD32 u4_winning_sad)
1908*495ae853SAndroid Build Coastguard Worker {
1909*495ae853SAndroid Build Coastguard Worker     res_pred_constants_t *ps_res_pred_constants = &ps_res_pred_ctxt->s_res_pred_constants;
1910*495ae853SAndroid Build Coastguard Worker     res_pred_state_t *ps_res_pred_state = (res_pred_state_t *) ps_res_pred_constants->pv_state;
1911*495ae853SAndroid Build Coastguard Worker     pu4_res_pred_sad[0] = ps_res_pred_state->pf_get_sad_with_residual_pred(
1912*495ae853SAndroid Build Coastguard Worker         &ps_src->as_component_bufs[Y], &ps_pred->as_component_bufs[Y],
1913*495ae853SAndroid Build Coastguard Worker         &ps_res->as_component_bufs[Y], MB_SIZE, MB_SIZE);
1914*495ae853SAndroid Build Coastguard Worker 
1915*495ae853SAndroid Build Coastguard Worker     pu1_residual_prediction_flag[0] = pu4_res_pred_sad[0] < u4_winning_sad;
1916*495ae853SAndroid Build Coastguard Worker }
1917*495ae853SAndroid Build Coastguard Worker 
isvce_update_res_pred_info(isvce_process_ctxt_t * ps_proc)1918*495ae853SAndroid Build Coastguard Worker void isvce_update_res_pred_info(isvce_process_ctxt_t *ps_proc)
1919*495ae853SAndroid Build Coastguard Worker {
1920*495ae853SAndroid Build Coastguard Worker     if(ps_proc->s_svc_params.u1_num_spatial_layers > 1)
1921*495ae853SAndroid Build Coastguard Worker     {
1922*495ae853SAndroid Build Coastguard Worker         svc_res_pred_ctxt_t *ps_res_pred_ctxt = ps_proc->ps_res_pred_ctxt;
1923*495ae853SAndroid Build Coastguard Worker         res_pred_constants_t *ps_res_pred_constants = &ps_res_pred_ctxt->s_res_pred_constants;
1924*495ae853SAndroid Build Coastguard Worker         res_pred_state_t *ps_res_pred_state = (res_pred_state_t *) ps_res_pred_constants->pv_state;
1925*495ae853SAndroid Build Coastguard Worker         res_pred_layer_state_t *ps_layer_state =
1926*495ae853SAndroid Build Coastguard Worker             &ps_res_pred_state->ps_layer_state[ps_proc->u1_spatial_layer_id];
1927*495ae853SAndroid Build Coastguard Worker 
1928*495ae853SAndroid Build Coastguard Worker         WORD8 i1_is_intra = ps_proc->ps_mb_info->u1_is_intra;
1929*495ae853SAndroid Build Coastguard Worker 
1930*495ae853SAndroid Build Coastguard Worker         WORD8 *pi1_mb_mode =
1931*495ae853SAndroid Build Coastguard Worker             &ps_layer_state->pi1_mb_mode[ps_proc->i4_mb_x +
1932*495ae853SAndroid Build Coastguard Worker                                          (ps_proc->i4_mb_y * (ps_layer_state->i4_mb_mode_stride))];
1933*495ae853SAndroid Build Coastguard Worker 
1934*495ae853SAndroid Build Coastguard Worker         if(ps_proc->ps_mb_info->u1_base_mode_flag == 1 && i1_is_intra)
1935*495ae853SAndroid Build Coastguard Worker         {
1936*495ae853SAndroid Build Coastguard Worker             *pi1_mb_mode = SVC_IBL_MB;
1937*495ae853SAndroid Build Coastguard Worker         }
1938*495ae853SAndroid Build Coastguard Worker         else
1939*495ae853SAndroid Build Coastguard Worker         {
1940*495ae853SAndroid Build Coastguard Worker             if(i1_is_intra)
1941*495ae853SAndroid Build Coastguard Worker             {
1942*495ae853SAndroid Build Coastguard Worker                 *pi1_mb_mode = SVC_INTRA_MB;
1943*495ae853SAndroid Build Coastguard Worker             }
1944*495ae853SAndroid Build Coastguard Worker             else
1945*495ae853SAndroid Build Coastguard Worker             {
1946*495ae853SAndroid Build Coastguard Worker                 *pi1_mb_mode = SVC_INTER_MB;
1947*495ae853SAndroid Build Coastguard Worker             }
1948*495ae853SAndroid Build Coastguard Worker         }
1949*495ae853SAndroid Build Coastguard Worker     }
1950*495ae853SAndroid Build Coastguard Worker }
1951