1*495ae853SAndroid Build Coastguard Worker /******************************************************************************
2*495ae853SAndroid Build Coastguard Worker *
3*495ae853SAndroid Build Coastguard Worker * Copyright (C) 2015 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 * \file ih264d_inter_pred.c
23*495ae853SAndroid Build Coastguard Worker *
24*495ae853SAndroid Build Coastguard Worker * \brief
25*495ae853SAndroid Build Coastguard Worker * This file contains routines to perform MotionCompensation tasks
26*495ae853SAndroid Build Coastguard Worker *
27*495ae853SAndroid Build Coastguard Worker * Detailed_description
28*495ae853SAndroid Build Coastguard Worker *
29*495ae853SAndroid Build Coastguard Worker * \date
30*495ae853SAndroid Build Coastguard Worker * 20/11/2002
31*495ae853SAndroid Build Coastguard Worker *
32*495ae853SAndroid Build Coastguard Worker * \author Arvind Raman
33*495ae853SAndroid Build Coastguard Worker **************************************************************************
34*495ae853SAndroid Build Coastguard Worker */
35*495ae853SAndroid Build Coastguard Worker
36*495ae853SAndroid Build Coastguard Worker #include <string.h>
37*495ae853SAndroid Build Coastguard Worker #include "ih264d_defs.h"
38*495ae853SAndroid Build Coastguard Worker #include "ih264d_mvpred.h"
39*495ae853SAndroid Build Coastguard Worker #include "ih264d_error_handler.h"
40*495ae853SAndroid Build Coastguard Worker #include "ih264d_structs.h"
41*495ae853SAndroid Build Coastguard Worker #include "ih264d_defs.h"
42*495ae853SAndroid Build Coastguard Worker #include "ih264d_inter_pred.h"
43*495ae853SAndroid Build Coastguard Worker #include "ih264_typedefs.h"
44*495ae853SAndroid Build Coastguard Worker #include "ih264_macros.h"
45*495ae853SAndroid Build Coastguard Worker #include "ih264_platform_macros.h"
46*495ae853SAndroid Build Coastguard Worker #include "ih264d_debug.h"
47*495ae853SAndroid Build Coastguard Worker #include "ih264d_tables.h"
48*495ae853SAndroid Build Coastguard Worker #include "ih264d_mb_utils.h"
49*495ae853SAndroid Build Coastguard Worker
50*495ae853SAndroid Build Coastguard Worker
51*495ae853SAndroid Build Coastguard Worker void ih264d_pad_on_demand(pred_info_t *ps_pred, UWORD8 lum_chrom_blk);
52*495ae853SAndroid Build Coastguard Worker
53*495ae853SAndroid Build Coastguard Worker
54*495ae853SAndroid Build Coastguard Worker
ih264d_copy_multiplex_data(UWORD8 * puc_Source,UWORD8 * puc_To,UWORD32 uc_w,UWORD32 uc_h,UWORD32 ui16_sourceWidth,UWORD32 ui16_toWidth)55*495ae853SAndroid Build Coastguard Worker void ih264d_copy_multiplex_data(UWORD8 *puc_Source,
56*495ae853SAndroid Build Coastguard Worker UWORD8 *puc_To,
57*495ae853SAndroid Build Coastguard Worker UWORD32 uc_w,
58*495ae853SAndroid Build Coastguard Worker UWORD32 uc_h,
59*495ae853SAndroid Build Coastguard Worker UWORD32 ui16_sourceWidth,
60*495ae853SAndroid Build Coastguard Worker UWORD32 ui16_toWidth)
61*495ae853SAndroid Build Coastguard Worker {
62*495ae853SAndroid Build Coastguard Worker UWORD8 uc_i, uc_j;
63*495ae853SAndroid Build Coastguard Worker
64*495ae853SAndroid Build Coastguard Worker for(uc_i = 0; uc_i < uc_h; uc_i++)
65*495ae853SAndroid Build Coastguard Worker {
66*495ae853SAndroid Build Coastguard Worker memcpy(puc_To, puc_Source, uc_w);
67*495ae853SAndroid Build Coastguard Worker puc_To += ui16_toWidth;
68*495ae853SAndroid Build Coastguard Worker puc_Source += ui16_sourceWidth;
69*495ae853SAndroid Build Coastguard Worker }
70*495ae853SAndroid Build Coastguard Worker }
71*495ae853SAndroid Build Coastguard Worker
72*495ae853SAndroid Build Coastguard Worker
73*495ae853SAndroid Build Coastguard Worker /*!
74*495ae853SAndroid Build Coastguard Worker **************************************************************************
75*495ae853SAndroid Build Coastguard Worker * \if Function name : dma_2d1d \endif
76*495ae853SAndroid Build Coastguard Worker *
77*495ae853SAndroid Build Coastguard Worker * \brief
78*495ae853SAndroid Build Coastguard Worker * 2D -> 1D linear DMA into the reference buffers
79*495ae853SAndroid Build Coastguard Worker *
80*495ae853SAndroid Build Coastguard Worker * \return
81*495ae853SAndroid Build Coastguard Worker * None
82*495ae853SAndroid Build Coastguard Worker **************************************************************************
83*495ae853SAndroid Build Coastguard Worker */
ih264d_copy_2d1d(UWORD8 * puc_src,UWORD8 * puc_dest,UWORD16 ui16_srcWidth,UWORD16 ui16_widthToFill,UWORD16 ui16_heightToFill)84*495ae853SAndroid Build Coastguard Worker void ih264d_copy_2d1d(UWORD8 *puc_src,
85*495ae853SAndroid Build Coastguard Worker UWORD8 *puc_dest,
86*495ae853SAndroid Build Coastguard Worker UWORD16 ui16_srcWidth,
87*495ae853SAndroid Build Coastguard Worker UWORD16 ui16_widthToFill,
88*495ae853SAndroid Build Coastguard Worker UWORD16 ui16_heightToFill)
89*495ae853SAndroid Build Coastguard Worker {
90*495ae853SAndroid Build Coastguard Worker UWORD32 uc_w, uc_h;
91*495ae853SAndroid Build Coastguard Worker for(uc_h = ui16_heightToFill; uc_h != 0; uc_h--)
92*495ae853SAndroid Build Coastguard Worker {
93*495ae853SAndroid Build Coastguard Worker memcpy(puc_dest, puc_src, ui16_widthToFill);
94*495ae853SAndroid Build Coastguard Worker puc_dest += ui16_widthToFill;
95*495ae853SAndroid Build Coastguard Worker puc_src += ui16_srcWidth;
96*495ae853SAndroid Build Coastguard Worker }
97*495ae853SAndroid Build Coastguard Worker }
98*495ae853SAndroid Build Coastguard Worker
99*495ae853SAndroid Build Coastguard Worker /*!
100*495ae853SAndroid Build Coastguard Worker **************************************************************************
101*495ae853SAndroid Build Coastguard Worker * \if Function name : ih264d_fill_pred_info \endif
102*495ae853SAndroid Build Coastguard Worker *
103*495ae853SAndroid Build Coastguard Worker * \brief
104*495ae853SAndroid Build Coastguard Worker * Fills inter prediction related info
105*495ae853SAndroid Build Coastguard Worker *
106*495ae853SAndroid Build Coastguard Worker * \return
107*495ae853SAndroid Build Coastguard Worker * None
108*495ae853SAndroid Build Coastguard Worker **************************************************************************
109*495ae853SAndroid Build Coastguard Worker */
ih264d_fill_pred_info(WORD16 * pi2_mv,WORD32 part_width,WORD32 part_height,WORD32 sub_mb_num,WORD32 pred_dir,pred_info_pkd_t * ps_pred_pkd,WORD8 i1_buf_id,WORD8 i1_ref_idx,UWORD32 * pu4_wt_offset,UWORD8 u1_pic_type)110*495ae853SAndroid Build Coastguard Worker void ih264d_fill_pred_info(WORD16 *pi2_mv,WORD32 part_width,WORD32 part_height, WORD32 sub_mb_num,
111*495ae853SAndroid Build Coastguard Worker WORD32 pred_dir,pred_info_pkd_t *ps_pred_pkd,WORD8 i1_buf_id,
112*495ae853SAndroid Build Coastguard Worker WORD8 i1_ref_idx,UWORD32 *pu4_wt_offset,UWORD8 u1_pic_type)
113*495ae853SAndroid Build Coastguard Worker {
114*495ae853SAndroid Build Coastguard Worker WORD32 insert_bits;
115*495ae853SAndroid Build Coastguard Worker
116*495ae853SAndroid Build Coastguard Worker ps_pred_pkd->i2_mv[0] = pi2_mv[0];
117*495ae853SAndroid Build Coastguard Worker ps_pred_pkd->i2_mv[1] = pi2_mv[1];
118*495ae853SAndroid Build Coastguard Worker
119*495ae853SAndroid Build Coastguard Worker insert_bits = sub_mb_num & 3; /*sub mb x*/
120*495ae853SAndroid Build Coastguard Worker ps_pred_pkd->i1_size_pos_info = insert_bits;
121*495ae853SAndroid Build Coastguard Worker insert_bits = sub_mb_num >> 2;/*sub mb y*/
122*495ae853SAndroid Build Coastguard Worker ps_pred_pkd->i1_size_pos_info |= insert_bits << 2;
123*495ae853SAndroid Build Coastguard Worker insert_bits = part_width >> 1;
124*495ae853SAndroid Build Coastguard Worker ps_pred_pkd->i1_size_pos_info |= insert_bits << 4;
125*495ae853SAndroid Build Coastguard Worker insert_bits = part_height >> 1;
126*495ae853SAndroid Build Coastguard Worker ps_pred_pkd->i1_size_pos_info |= insert_bits << 6;
127*495ae853SAndroid Build Coastguard Worker
128*495ae853SAndroid Build Coastguard Worker ps_pred_pkd->i1_ref_idx_info = i1_ref_idx;
129*495ae853SAndroid Build Coastguard Worker ps_pred_pkd->i1_ref_idx_info |= (pred_dir << 6);
130*495ae853SAndroid Build Coastguard Worker ps_pred_pkd->i1_buf_id = i1_buf_id;
131*495ae853SAndroid Build Coastguard Worker ps_pred_pkd->pu4_wt_offst = pu4_wt_offset;
132*495ae853SAndroid Build Coastguard Worker ps_pred_pkd->u1_pic_type = u1_pic_type;
133*495ae853SAndroid Build Coastguard Worker
134*495ae853SAndroid Build Coastguard Worker
135*495ae853SAndroid Build Coastguard Worker }
136*495ae853SAndroid Build Coastguard Worker
137*495ae853SAndroid Build Coastguard Worker
138*495ae853SAndroid Build Coastguard Worker
139*495ae853SAndroid Build Coastguard Worker
140*495ae853SAndroid Build Coastguard Worker
141*495ae853SAndroid Build Coastguard Worker
142*495ae853SAndroid Build Coastguard Worker
143*495ae853SAndroid Build Coastguard Worker /*****************************************************************************/
144*495ae853SAndroid Build Coastguard Worker /* \if Function name : formMbPartInfo \endif */
145*495ae853SAndroid Build Coastguard Worker /* */
146*495ae853SAndroid Build Coastguard Worker /* \brief */
147*495ae853SAndroid Build Coastguard Worker /* Form the Mb partition information structure, to be used by the MC */
148*495ae853SAndroid Build Coastguard Worker /* routine */
149*495ae853SAndroid Build Coastguard Worker /* */
150*495ae853SAndroid Build Coastguard Worker /* \return */
151*495ae853SAndroid Build Coastguard Worker /* None */
152*495ae853SAndroid Build Coastguard Worker /* \note */
153*495ae853SAndroid Build Coastguard Worker /* c_bufx is used to select PredBuffer, */
154*495ae853SAndroid Build Coastguard Worker /* if it's only Forward/Backward prediction always buffer used is */
155*495ae853SAndroid Build Coastguard Worker /* puc_MbLumaPredBuffer[0 to X1],pu1_mb_cb_pred_buffer[0 to X1] and */
156*495ae853SAndroid Build Coastguard Worker /* pu1_mb_cr_pred_buffer[0 to X1] */
157*495ae853SAndroid Build Coastguard Worker /* */
158*495ae853SAndroid Build Coastguard Worker /* if it's bidirect for forward ..PredBuffer[0 to X1] buffer is used and */
159*495ae853SAndroid Build Coastguard Worker /* ..PredBuffer[X2 to X3] for backward prediction. and */
160*495ae853SAndroid Build Coastguard Worker /* */
161*495ae853SAndroid Build Coastguard Worker /* Final predicted samples values are the average of ..PredBuffer[0 to X1]*/
162*495ae853SAndroid Build Coastguard Worker /* and ..PredBuffer[X2 to X3] */
163*495ae853SAndroid Build Coastguard Worker /* */
164*495ae853SAndroid Build Coastguard Worker /* X1 is 255 for Luma and 63 for Chroma */
165*495ae853SAndroid Build Coastguard Worker /* X2 is 256 for Luma and 64 for Chroma */
166*495ae853SAndroid Build Coastguard Worker /* X3 is 511 for Luma and 127 for Chroma */
167*495ae853SAndroid Build Coastguard Worker /* */
168*495ae853SAndroid Build Coastguard Worker /* DD MM YYYY Author(s) Changes (Describe the changes made) */
169*495ae853SAndroid Build Coastguard Worker /* 11 05 2005 SWRN Modified to handle pod */
170*495ae853SAndroid Build Coastguard Worker /*****************************************************************************/
171*495ae853SAndroid Build Coastguard Worker
ih264d_form_mb_part_info_bp(pred_info_pkd_t * ps_pred_pkd,dec_struct_t * ps_dec,UWORD16 u2_mb_x,UWORD16 u2_mb_y,WORD32 mb_index,dec_mb_info_t * ps_cur_mb_info)172*495ae853SAndroid Build Coastguard Worker WORD32 ih264d_form_mb_part_info_bp(pred_info_pkd_t *ps_pred_pkd,
173*495ae853SAndroid Build Coastguard Worker dec_struct_t * ps_dec,
174*495ae853SAndroid Build Coastguard Worker UWORD16 u2_mb_x,
175*495ae853SAndroid Build Coastguard Worker UWORD16 u2_mb_y,
176*495ae853SAndroid Build Coastguard Worker WORD32 mb_index,
177*495ae853SAndroid Build Coastguard Worker dec_mb_info_t *ps_cur_mb_info)
178*495ae853SAndroid Build Coastguard Worker {
179*495ae853SAndroid Build Coastguard Worker /* The reference buffer pointer */
180*495ae853SAndroid Build Coastguard Worker WORD32 i2_frm_x, i2_frm_y;
181*495ae853SAndroid Build Coastguard Worker WORD32 i2_tmp_mv_x, i2_tmp_mv_y;
182*495ae853SAndroid Build Coastguard Worker WORD32 i2_rec_x, i2_rec_y;
183*495ae853SAndroid Build Coastguard Worker
184*495ae853SAndroid Build Coastguard Worker WORD32 u2_pic_ht;
185*495ae853SAndroid Build Coastguard Worker WORD32 u2_frm_wd;
186*495ae853SAndroid Build Coastguard Worker WORD32 u2_rec_wd;
187*495ae853SAndroid Build Coastguard Worker UWORD8 u1_sub_x = 0,u1_sub_y=0 ;
188*495ae853SAndroid Build Coastguard Worker UWORD8 u1_part_wd = 0,u1_part_ht = 0;
189*495ae853SAndroid Build Coastguard Worker WORD16 i2_mv_x,i2_mv_y;
190*495ae853SAndroid Build Coastguard Worker
191*495ae853SAndroid Build Coastguard Worker /********************************************/
192*495ae853SAndroid Build Coastguard Worker /* i1_mc_wd width reqd for mcomp */
193*495ae853SAndroid Build Coastguard Worker /* u1_dma_ht height reqd for mcomp */
194*495ae853SAndroid Build Coastguard Worker /* u1_dma_wd width aligned to 4 bytes */
195*495ae853SAndroid Build Coastguard Worker /* u1_dx fractional part of width */
196*495ae853SAndroid Build Coastguard Worker /* u1_dx fractional part of height */
197*495ae853SAndroid Build Coastguard Worker /********************************************/
198*495ae853SAndroid Build Coastguard Worker UWORD32 i1_mc_wd;
199*495ae853SAndroid Build Coastguard Worker
200*495ae853SAndroid Build Coastguard Worker WORD32 u1_dma_ht;
201*495ae853SAndroid Build Coastguard Worker
202*495ae853SAndroid Build Coastguard Worker UWORD32 u1_dma_wd;
203*495ae853SAndroid Build Coastguard Worker UWORD32 u1_dx;
204*495ae853SAndroid Build Coastguard Worker UWORD32 u1_dy;
205*495ae853SAndroid Build Coastguard Worker pred_info_t * ps_pred = ps_dec->ps_pred + ps_dec->u4_pred_info_idx;
206*495ae853SAndroid Build Coastguard Worker dec_slice_params_t * const ps_cur_slice = ps_dec->ps_cur_slice;
207*495ae853SAndroid Build Coastguard Worker tfr_ctxt_t *ps_frame_buf;
208*495ae853SAndroid Build Coastguard Worker struct pic_buffer_t *ps_ref_frm;
209*495ae853SAndroid Build Coastguard Worker UWORD8 u1_scale_ref,u1_mbaff,u1_field;
210*495ae853SAndroid Build Coastguard Worker pic_buffer_t **pps_ref_frame;
211*495ae853SAndroid Build Coastguard Worker WORD8 i1_size_pos_info,i1_buf_id;
212*495ae853SAndroid Build Coastguard Worker
213*495ae853SAndroid Build Coastguard Worker PROFILE_DISABLE_MB_PART_INFO()
214*495ae853SAndroid Build Coastguard Worker
215*495ae853SAndroid Build Coastguard Worker UNUSED(ps_cur_mb_info);
216*495ae853SAndroid Build Coastguard Worker i1_size_pos_info = ps_pred_pkd->i1_size_pos_info;
217*495ae853SAndroid Build Coastguard Worker GET_XPOS_PRED(u1_sub_x,i1_size_pos_info);
218*495ae853SAndroid Build Coastguard Worker GET_YPOS_PRED(u1_sub_y,i1_size_pos_info);
219*495ae853SAndroid Build Coastguard Worker GET_WIDTH_PRED(u1_part_wd,i1_size_pos_info);
220*495ae853SAndroid Build Coastguard Worker GET_HEIGHT_PRED(u1_part_ht,i1_size_pos_info);
221*495ae853SAndroid Build Coastguard Worker i2_mv_x = ps_pred_pkd->i2_mv[0];
222*495ae853SAndroid Build Coastguard Worker i2_mv_y = ps_pred_pkd->i2_mv[1];
223*495ae853SAndroid Build Coastguard Worker i1_buf_id = ps_pred_pkd->i1_buf_id;
224*495ae853SAndroid Build Coastguard Worker
225*495ae853SAndroid Build Coastguard Worker
226*495ae853SAndroid Build Coastguard Worker ps_ref_frm = ps_dec->apv_buf_id_pic_buf_map[i1_buf_id];
227*495ae853SAndroid Build Coastguard Worker
228*495ae853SAndroid Build Coastguard Worker
229*495ae853SAndroid Build Coastguard Worker {
230*495ae853SAndroid Build Coastguard Worker ps_frame_buf = &ps_dec->s_tran_addrecon;
231*495ae853SAndroid Build Coastguard Worker }
232*495ae853SAndroid Build Coastguard Worker
233*495ae853SAndroid Build Coastguard Worker
234*495ae853SAndroid Build Coastguard Worker /* Transfer Setup Y */
235*495ae853SAndroid Build Coastguard Worker {
236*495ae853SAndroid Build Coastguard Worker UWORD8 *pu1_pred, *pu1_rec;
237*495ae853SAndroid Build Coastguard Worker
238*495ae853SAndroid Build Coastguard Worker /* calculating rounded motion vectors and fractional components */
239*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_x = i2_mv_x;
240*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_y = i2_mv_y;
241*495ae853SAndroid Build Coastguard Worker u1_dx = i2_tmp_mv_x & 0x3;
242*495ae853SAndroid Build Coastguard Worker u1_dy = i2_tmp_mv_y & 0x3;
243*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_x >>= 2;
244*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_y >>= 2;
245*495ae853SAndroid Build Coastguard Worker i1_mc_wd = u1_part_wd << 2;
246*495ae853SAndroid Build Coastguard Worker u1_dma_ht = u1_part_ht << 2;
247*495ae853SAndroid Build Coastguard Worker if(u1_dx)
248*495ae853SAndroid Build Coastguard Worker {
249*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_x -= 2;
250*495ae853SAndroid Build Coastguard Worker i1_mc_wd += 5;
251*495ae853SAndroid Build Coastguard Worker }
252*495ae853SAndroid Build Coastguard Worker if(u1_dy)
253*495ae853SAndroid Build Coastguard Worker {
254*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_y -= 2;
255*495ae853SAndroid Build Coastguard Worker u1_dma_ht += 5;
256*495ae853SAndroid Build Coastguard Worker }
257*495ae853SAndroid Build Coastguard Worker
258*495ae853SAndroid Build Coastguard Worker /********************************************************************/
259*495ae853SAndroid Build Coastguard Worker /* Calulating the horizontal and the vertical u4_ofst from top left */
260*495ae853SAndroid Build Coastguard Worker /* edge of the reference frame, and subsequent clipping */
261*495ae853SAndroid Build Coastguard Worker /********************************************************************/
262*495ae853SAndroid Build Coastguard Worker u2_pic_ht = ps_dec->u2_pic_ht;
263*495ae853SAndroid Build Coastguard Worker u2_frm_wd = ps_dec->u2_frm_wd_y;
264*495ae853SAndroid Build Coastguard Worker i2_rec_x = u1_sub_x << 2;
265*495ae853SAndroid Build Coastguard Worker i2_rec_y = u1_sub_y << 2;
266*495ae853SAndroid Build Coastguard Worker
267*495ae853SAndroid Build Coastguard Worker i2_frm_x = (u2_mb_x << 4) + i2_rec_x + i2_tmp_mv_x;
268*495ae853SAndroid Build Coastguard Worker i2_frm_y = (u2_mb_y << 4) + i2_rec_y + i2_tmp_mv_y;
269*495ae853SAndroid Build Coastguard Worker
270*495ae853SAndroid Build Coastguard Worker i2_frm_x = CLIP3(MAX_OFFSET_OUTSIDE_X_FRM, (ps_dec->u2_pic_wd - 1),
271*495ae853SAndroid Build Coastguard Worker i2_frm_x);
272*495ae853SAndroid Build Coastguard Worker i2_frm_y = CLIP3(((1 - u1_dma_ht)), (u2_pic_ht - (1)), i2_frm_y);
273*495ae853SAndroid Build Coastguard Worker
274*495ae853SAndroid Build Coastguard Worker pu1_pred = ps_ref_frm->pu1_buf1 + i2_frm_y * u2_frm_wd + i2_frm_x;
275*495ae853SAndroid Build Coastguard Worker
276*495ae853SAndroid Build Coastguard Worker u1_dma_wd = (i1_mc_wd + 3) & 0xFC;
277*495ae853SAndroid Build Coastguard Worker
278*495ae853SAndroid Build Coastguard Worker /********************************************************************/
279*495ae853SAndroid Build Coastguard Worker /* Calulating the horizontal and the vertical u4_ofst from top left */
280*495ae853SAndroid Build Coastguard Worker /* edge of the recon buffer */
281*495ae853SAndroid Build Coastguard Worker /********************************************************************/
282*495ae853SAndroid Build Coastguard Worker u2_rec_wd = MB_SIZE;
283*495ae853SAndroid Build Coastguard Worker {
284*495ae853SAndroid Build Coastguard Worker u2_rec_wd = ps_dec->u2_frm_wd_y;
285*495ae853SAndroid Build Coastguard Worker i2_rec_x += (mb_index << 4);
286*495ae853SAndroid Build Coastguard Worker pu1_rec = ps_frame_buf->pu1_dest_y + i2_rec_y * u2_rec_wd
287*495ae853SAndroid Build Coastguard Worker + i2_rec_x;
288*495ae853SAndroid Build Coastguard Worker }
289*495ae853SAndroid Build Coastguard Worker
290*495ae853SAndroid Build Coastguard Worker /* filling the pred and dma structures for Y */
291*495ae853SAndroid Build Coastguard Worker u2_frm_wd = ps_dec->u2_frm_wd_y;
292*495ae853SAndroid Build Coastguard Worker
293*495ae853SAndroid Build Coastguard Worker ps_pred->u2_u1_ref_buf_wd = u1_dma_wd;
294*495ae853SAndroid Build Coastguard Worker ps_pred->i1_dma_ht = u1_dma_ht;
295*495ae853SAndroid Build Coastguard Worker ps_pred->i1_mc_wd = i1_mc_wd;
296*495ae853SAndroid Build Coastguard Worker ps_pred->u2_frm_wd = u2_frm_wd;
297*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_rec_y_u = pu1_rec;
298*495ae853SAndroid Build Coastguard Worker ps_pred->u2_dst_stride = u2_rec_wd;
299*495ae853SAndroid Build Coastguard Worker
300*495ae853SAndroid Build Coastguard Worker ps_pred->i1_mb_partwidth = u1_part_wd << 2;
301*495ae853SAndroid Build Coastguard Worker ps_pred->i1_mb_partheight = u1_part_ht << 2;
302*495ae853SAndroid Build Coastguard Worker ps_pred->u1_dydx = (u1_dy << 2) + u1_dx;
303*495ae853SAndroid Build Coastguard Worker
304*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_y_ref = pu1_pred;
305*495ae853SAndroid Build Coastguard Worker
306*495ae853SAndroid Build Coastguard Worker }
307*495ae853SAndroid Build Coastguard Worker
308*495ae853SAndroid Build Coastguard Worker /* Increment ps_pred index */
309*495ae853SAndroid Build Coastguard Worker ps_pred++;
310*495ae853SAndroid Build Coastguard Worker
311*495ae853SAndroid Build Coastguard Worker /* Transfer Setup U & V */
312*495ae853SAndroid Build Coastguard Worker {
313*495ae853SAndroid Build Coastguard Worker WORD32 i4_ref_offset, i4_rec_offset;
314*495ae853SAndroid Build Coastguard Worker UWORD8 *pu1_pred_u, *pu1_pred_v;
315*495ae853SAndroid Build Coastguard Worker
316*495ae853SAndroid Build Coastguard Worker
317*495ae853SAndroid Build Coastguard Worker /* calculating rounded motion vectors and fractional components */
318*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_x = i2_mv_x;
319*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_y = i2_mv_y;
320*495ae853SAndroid Build Coastguard Worker
321*495ae853SAndroid Build Coastguard Worker /************************************************************************/
322*495ae853SAndroid Build Coastguard Worker /* Table 8-9: Derivation of the vertical component of the chroma vector */
323*495ae853SAndroid Build Coastguard Worker /* in field coding mode */
324*495ae853SAndroid Build Coastguard Worker /************************************************************************/
325*495ae853SAndroid Build Coastguard Worker
326*495ae853SAndroid Build Coastguard Worker /* Eighth sample of the chroma MV */
327*495ae853SAndroid Build Coastguard Worker u1_dx = i2_tmp_mv_x & 0x7;
328*495ae853SAndroid Build Coastguard Worker u1_dy = i2_tmp_mv_y & 0x7;
329*495ae853SAndroid Build Coastguard Worker
330*495ae853SAndroid Build Coastguard Worker /********************************************************************/
331*495ae853SAndroid Build Coastguard Worker /* Calculating the full pel MV for chroma which is 1/2 of the Luma */
332*495ae853SAndroid Build Coastguard Worker /* MV in full pel units */
333*495ae853SAndroid Build Coastguard Worker /********************************************************************/
334*495ae853SAndroid Build Coastguard Worker i2_mv_x = i2_tmp_mv_x;
335*495ae853SAndroid Build Coastguard Worker i2_mv_y = i2_tmp_mv_y;
336*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_x = SIGN_POW2_DIV(i2_tmp_mv_x, 3);
337*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_y = SIGN_POW2_DIV(i2_tmp_mv_y, 3);
338*495ae853SAndroid Build Coastguard Worker i1_mc_wd = u1_part_wd << 1;
339*495ae853SAndroid Build Coastguard Worker u1_dma_ht = u1_part_ht << 1;
340*495ae853SAndroid Build Coastguard Worker if(u1_dx)
341*495ae853SAndroid Build Coastguard Worker {
342*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_x -= (i2_mv_x < 0);
343*495ae853SAndroid Build Coastguard Worker i1_mc_wd++;
344*495ae853SAndroid Build Coastguard Worker }
345*495ae853SAndroid Build Coastguard Worker if(u1_dy != 0)
346*495ae853SAndroid Build Coastguard Worker {
347*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_y -= (i2_mv_y < 0);
348*495ae853SAndroid Build Coastguard Worker u1_dma_ht++;
349*495ae853SAndroid Build Coastguard Worker }
350*495ae853SAndroid Build Coastguard Worker
351*495ae853SAndroid Build Coastguard Worker /********************************************************************/
352*495ae853SAndroid Build Coastguard Worker /* Calulating the horizontal and the vertical u4_ofst from top left */
353*495ae853SAndroid Build Coastguard Worker /* edge of the reference frame, and subsequent clipping */
354*495ae853SAndroid Build Coastguard Worker /********************************************************************/
355*495ae853SAndroid Build Coastguard Worker u2_pic_ht >>= 1;
356*495ae853SAndroid Build Coastguard Worker u2_frm_wd = ps_dec->u2_frm_wd_uv;
357*495ae853SAndroid Build Coastguard Worker i2_rec_x = u1_sub_x << 1;
358*495ae853SAndroid Build Coastguard Worker i2_rec_y = u1_sub_y << 1;
359*495ae853SAndroid Build Coastguard Worker
360*495ae853SAndroid Build Coastguard Worker i2_frm_x = (u2_mb_x << 3) + i2_rec_x + i2_tmp_mv_x;
361*495ae853SAndroid Build Coastguard Worker i2_frm_y = (u2_mb_y << 3) + i2_rec_y + i2_tmp_mv_y;
362*495ae853SAndroid Build Coastguard Worker
363*495ae853SAndroid Build Coastguard Worker i2_frm_x = CLIP3(MAX_OFFSET_OUTSIDE_UV_FRM,
364*495ae853SAndroid Build Coastguard Worker ((ps_dec->u2_pic_wd >> 1) - 1), i2_frm_x);
365*495ae853SAndroid Build Coastguard Worker i2_frm_y = CLIP3(((1 - u1_dma_ht)), (u2_pic_ht - (1)), i2_frm_y);
366*495ae853SAndroid Build Coastguard Worker
367*495ae853SAndroid Build Coastguard Worker i4_ref_offset = i2_frm_y * u2_frm_wd + i2_frm_x * YUV420SP_FACTOR;
368*495ae853SAndroid Build Coastguard Worker u1_dma_wd = (i1_mc_wd + 3) & 0xFC;
369*495ae853SAndroid Build Coastguard Worker
370*495ae853SAndroid Build Coastguard Worker /********************************************************************/
371*495ae853SAndroid Build Coastguard Worker /* Calulating the horizontal and the vertical u4_ofst from top left */
372*495ae853SAndroid Build Coastguard Worker /* edge of the recon buffer */
373*495ae853SAndroid Build Coastguard Worker /********************************************************************/
374*495ae853SAndroid Build Coastguard Worker /* CHANGED CODE */
375*495ae853SAndroid Build Coastguard Worker u2_rec_wd = BLK8x8SIZE * YUV420SP_FACTOR;
376*495ae853SAndroid Build Coastguard Worker i4_rec_offset = i2_rec_y * u2_rec_wd + i2_rec_x * YUV420SP_FACTOR;
377*495ae853SAndroid Build Coastguard Worker
378*495ae853SAndroid Build Coastguard Worker {
379*495ae853SAndroid Build Coastguard Worker u2_rec_wd = ps_dec->u2_frm_wd_uv;
380*495ae853SAndroid Build Coastguard Worker i2_rec_x += (mb_index << 3);
381*495ae853SAndroid Build Coastguard Worker i4_rec_offset = i2_rec_y * u2_rec_wd + i2_rec_x * YUV420SP_FACTOR;
382*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_rec_y_u = ps_frame_buf->pu1_dest_u + i4_rec_offset;
383*495ae853SAndroid Build Coastguard Worker ps_pred->u1_pi1_wt_ofst_rec_v = ps_frame_buf->pu1_dest_v
384*495ae853SAndroid Build Coastguard Worker + i4_rec_offset;
385*495ae853SAndroid Build Coastguard Worker }
386*495ae853SAndroid Build Coastguard Worker
387*495ae853SAndroid Build Coastguard Worker /* CHANGED CODE */
388*495ae853SAndroid Build Coastguard Worker
389*495ae853SAndroid Build Coastguard Worker /* filling the common pred structures for U */
390*495ae853SAndroid Build Coastguard Worker u2_frm_wd = ps_dec->u2_frm_wd_uv;
391*495ae853SAndroid Build Coastguard Worker
392*495ae853SAndroid Build Coastguard Worker ps_pred->u2_u1_ref_buf_wd = u1_dma_wd;
393*495ae853SAndroid Build Coastguard Worker ps_pred->i1_dma_ht = u1_dma_ht;
394*495ae853SAndroid Build Coastguard Worker ps_pred->i1_mc_wd = i1_mc_wd;
395*495ae853SAndroid Build Coastguard Worker
396*495ae853SAndroid Build Coastguard Worker ps_pred->u2_frm_wd = u2_frm_wd;
397*495ae853SAndroid Build Coastguard Worker ps_pred->u2_dst_stride = u2_rec_wd;
398*495ae853SAndroid Build Coastguard Worker
399*495ae853SAndroid Build Coastguard Worker ps_pred->i1_mb_partwidth = u1_part_wd << 1;
400*495ae853SAndroid Build Coastguard Worker ps_pred->i1_mb_partheight = u1_part_ht << 1;
401*495ae853SAndroid Build Coastguard Worker ps_pred->u1_dydx = (u1_dy << 3) + u1_dx;
402*495ae853SAndroid Build Coastguard Worker
403*495ae853SAndroid Build Coastguard Worker pu1_pred_u = ps_ref_frm->pu1_buf2 + i4_ref_offset;
404*495ae853SAndroid Build Coastguard Worker pu1_pred_v = ps_ref_frm->pu1_buf3 + i4_ref_offset;
405*495ae853SAndroid Build Coastguard Worker
406*495ae853SAndroid Build Coastguard Worker /* Copy U & V partitions */
407*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_u_ref = pu1_pred_u;
408*495ae853SAndroid Build Coastguard Worker
409*495ae853SAndroid Build Coastguard Worker /* Increment the reference buffer Index */
410*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_v_ref = pu1_pred_v;
411*495ae853SAndroid Build Coastguard Worker }
412*495ae853SAndroid Build Coastguard Worker
413*495ae853SAndroid Build Coastguard Worker /* Increment ps_pred index */
414*495ae853SAndroid Build Coastguard Worker ps_dec->u4_pred_info_idx += 2;
415*495ae853SAndroid Build Coastguard Worker
416*495ae853SAndroid Build Coastguard Worker return OK;
417*495ae853SAndroid Build Coastguard Worker
418*495ae853SAndroid Build Coastguard Worker }
419*495ae853SAndroid Build Coastguard Worker
420*495ae853SAndroid Build Coastguard Worker
421*495ae853SAndroid Build Coastguard Worker /*****************************************************************************/
422*495ae853SAndroid Build Coastguard Worker /* \if Function name : formMbPartInfo \endif */
423*495ae853SAndroid Build Coastguard Worker /* */
424*495ae853SAndroid Build Coastguard Worker /* \brief */
425*495ae853SAndroid Build Coastguard Worker /* Form the Mb partition information structure, to be used by the MC */
426*495ae853SAndroid Build Coastguard Worker /* routine */
427*495ae853SAndroid Build Coastguard Worker /* */
428*495ae853SAndroid Build Coastguard Worker /* \return */
429*495ae853SAndroid Build Coastguard Worker /* None */
430*495ae853SAndroid Build Coastguard Worker /* \note */
431*495ae853SAndroid Build Coastguard Worker /* c_bufx is used to select PredBuffer, */
432*495ae853SAndroid Build Coastguard Worker /* if it's only Forward/Backward prediction always buffer used is */
433*495ae853SAndroid Build Coastguard Worker /* puc_MbLumaPredBuffer[0 to X1],pu1_mb_cb_pred_buffer[0 to X1] and */
434*495ae853SAndroid Build Coastguard Worker /* pu1_mb_cr_pred_buffer[0 to X1] */
435*495ae853SAndroid Build Coastguard Worker /* */
436*495ae853SAndroid Build Coastguard Worker /* if it's bidirect for forward ..PredBuffer[0 to X1] buffer is used and */
437*495ae853SAndroid Build Coastguard Worker /* ..PredBuffer[X2 to X3] for backward prediction. and */
438*495ae853SAndroid Build Coastguard Worker /* */
439*495ae853SAndroid Build Coastguard Worker /* Final predicted samples values are the average of ..PredBuffer[0 to X1]*/
440*495ae853SAndroid Build Coastguard Worker /* and ..PredBuffer[X2 to X3] */
441*495ae853SAndroid Build Coastguard Worker /* */
442*495ae853SAndroid Build Coastguard Worker /* X1 is 255 for Luma and 63 for Chroma */
443*495ae853SAndroid Build Coastguard Worker /* X2 is 256 for Luma and 64 for Chroma */
444*495ae853SAndroid Build Coastguard Worker /* X3 is 511 for Luma and 127 for Chroma */
445*495ae853SAndroid Build Coastguard Worker /* */
446*495ae853SAndroid Build Coastguard Worker /* DD MM YYYY Author(s) Changes (Describe the changes made) */
447*495ae853SAndroid Build Coastguard Worker /* 11 05 2005 SWRN Modified to handle pod */
448*495ae853SAndroid Build Coastguard Worker /*****************************************************************************/
ih264d_form_mb_part_info_mp(pred_info_pkd_t * ps_pred_pkd,dec_struct_t * ps_dec,UWORD16 u2_mb_x,UWORD16 u2_mb_y,WORD32 mb_index,dec_mb_info_t * ps_cur_mb_info)449*495ae853SAndroid Build Coastguard Worker WORD32 ih264d_form_mb_part_info_mp(pred_info_pkd_t *ps_pred_pkd,
450*495ae853SAndroid Build Coastguard Worker dec_struct_t * ps_dec,
451*495ae853SAndroid Build Coastguard Worker UWORD16 u2_mb_x,
452*495ae853SAndroid Build Coastguard Worker UWORD16 u2_mb_y,
453*495ae853SAndroid Build Coastguard Worker WORD32 mb_index,
454*495ae853SAndroid Build Coastguard Worker dec_mb_info_t *ps_cur_mb_info)
455*495ae853SAndroid Build Coastguard Worker {
456*495ae853SAndroid Build Coastguard Worker /* The reference buffer pointer */
457*495ae853SAndroid Build Coastguard Worker UWORD8 *pu1_ref_buf;
458*495ae853SAndroid Build Coastguard Worker WORD16 i2_frm_x, i2_frm_y, i2_tmp_mv_x, i2_tmp_mv_y, i2_pod_ht;
459*495ae853SAndroid Build Coastguard Worker WORD16 i2_rec_x, i2_rec_y;
460*495ae853SAndroid Build Coastguard Worker UWORD16 u2_pic_ht, u2_frm_wd, u2_rec_wd;
461*495ae853SAndroid Build Coastguard Worker UWORD8 u1_wght_pred_type, u1_wted_bipred_idc;
462*495ae853SAndroid Build Coastguard Worker UWORD16 u2_tot_ref_scratch_size;
463*495ae853SAndroid Build Coastguard Worker UWORD8 u1_sub_x = 0;
464*495ae853SAndroid Build Coastguard Worker UWORD8 u1_sub_y = 0;
465*495ae853SAndroid Build Coastguard Worker UWORD8 u1_is_bi_dir = 0;
466*495ae853SAndroid Build Coastguard Worker
467*495ae853SAndroid Build Coastguard Worker /********************************************/
468*495ae853SAndroid Build Coastguard Worker /* i1_mc_wd width reqd for mcomp */
469*495ae853SAndroid Build Coastguard Worker /* u1_dma_ht height reqd for mcomp */
470*495ae853SAndroid Build Coastguard Worker /* u1_dma_wd width aligned to 4 bytes */
471*495ae853SAndroid Build Coastguard Worker /* u1_dx fractional part of width */
472*495ae853SAndroid Build Coastguard Worker /* u1_dx fractional part of height */
473*495ae853SAndroid Build Coastguard Worker /********************************************/
474*495ae853SAndroid Build Coastguard Worker UWORD8 i1_mc_wd, u1_dma_ht, u1_dma_wd, u1_dx, u1_dy;
475*495ae853SAndroid Build Coastguard Worker pred_info_t * ps_pred ;
476*495ae853SAndroid Build Coastguard Worker dec_slice_params_t * const ps_cur_slice = ps_dec->ps_cur_slice;
477*495ae853SAndroid Build Coastguard Worker const UWORD8 u1_slice_type = ps_dec->ps_decode_cur_slice->slice_type;
478*495ae853SAndroid Build Coastguard Worker UWORD8 u1_pod_bot, u1_pod_top;
479*495ae853SAndroid Build Coastguard Worker
480*495ae853SAndroid Build Coastguard Worker /* load the pictype for pod u4_flag & chroma motion vector derivation */
481*495ae853SAndroid Build Coastguard Worker UWORD8 u1_ref_pic_type ;
482*495ae853SAndroid Build Coastguard Worker
483*495ae853SAndroid Build Coastguard Worker /* set default value to flags specifying field nature of picture & mb */
484*495ae853SAndroid Build Coastguard Worker UWORD32 u1_mb_fld = 0, u1_mb_or_pic_fld;
485*495ae853SAndroid Build Coastguard Worker UWORD32 u1_mb_bot = 0, u1_pic_bot = 0, u1_mb_or_pic_bot;
486*495ae853SAndroid Build Coastguard Worker tfr_ctxt_t *ps_frame_buf;
487*495ae853SAndroid Build Coastguard Worker /* calculate flags specifying field nature of picture & mb */
488*495ae853SAndroid Build Coastguard Worker const UWORD32 u1_pic_fld = ps_cur_slice->u1_field_pic_flag;
489*495ae853SAndroid Build Coastguard Worker WORD8 i1_pred;
490*495ae853SAndroid Build Coastguard Worker WORD8 i1_size_pos_info,i1_buf_id,i1_ref_idx;
491*495ae853SAndroid Build Coastguard Worker UWORD8 u1_part_wd,u1_part_ht;
492*495ae853SAndroid Build Coastguard Worker WORD16 i2_mv_x,i2_mv_y;
493*495ae853SAndroid Build Coastguard Worker struct pic_buffer_t *ps_ref_frm;
494*495ae853SAndroid Build Coastguard Worker UWORD32 *pu4_wt_offset;
495*495ae853SAndroid Build Coastguard Worker UWORD8 *pu1_buf1,*pu1_buf2,*pu1_buf3;
496*495ae853SAndroid Build Coastguard Worker
497*495ae853SAndroid Build Coastguard Worker
498*495ae853SAndroid Build Coastguard Worker PROFILE_DISABLE_MB_PART_INFO()
499*495ae853SAndroid Build Coastguard Worker
500*495ae853SAndroid Build Coastguard Worker ps_pred = ps_dec->ps_pred + ps_dec->u4_pred_info_idx;
501*495ae853SAndroid Build Coastguard Worker
502*495ae853SAndroid Build Coastguard Worker
503*495ae853SAndroid Build Coastguard Worker i1_size_pos_info = ps_pred_pkd->i1_size_pos_info;
504*495ae853SAndroid Build Coastguard Worker GET_XPOS_PRED(u1_sub_x,i1_size_pos_info);
505*495ae853SAndroid Build Coastguard Worker GET_YPOS_PRED(u1_sub_y,i1_size_pos_info);
506*495ae853SAndroid Build Coastguard Worker GET_WIDTH_PRED(u1_part_wd,i1_size_pos_info);
507*495ae853SAndroid Build Coastguard Worker GET_HEIGHT_PRED(u1_part_ht,i1_size_pos_info);
508*495ae853SAndroid Build Coastguard Worker i2_mv_x = ps_pred_pkd->i2_mv[0];
509*495ae853SAndroid Build Coastguard Worker i2_mv_y = ps_pred_pkd->i2_mv[1];
510*495ae853SAndroid Build Coastguard Worker i1_ref_idx = ps_pred_pkd->i1_ref_idx_info & 0x3f;
511*495ae853SAndroid Build Coastguard Worker i1_buf_id = ps_pred_pkd->i1_buf_id;
512*495ae853SAndroid Build Coastguard Worker ps_ref_frm = ps_dec->apv_buf_id_pic_buf_map[i1_buf_id];
513*495ae853SAndroid Build Coastguard Worker
514*495ae853SAndroid Build Coastguard Worker i1_pred = (ps_pred_pkd->i1_ref_idx_info & 0xC0) >> 6;
515*495ae853SAndroid Build Coastguard Worker u1_is_bi_dir = (i1_pred == BI_PRED);
516*495ae853SAndroid Build Coastguard Worker
517*495ae853SAndroid Build Coastguard Worker
518*495ae853SAndroid Build Coastguard Worker u1_ref_pic_type = ps_pred_pkd->u1_pic_type & PIC_MASK;
519*495ae853SAndroid Build Coastguard Worker
520*495ae853SAndroid Build Coastguard Worker pu1_buf1 = ps_ref_frm->pu1_buf1;
521*495ae853SAndroid Build Coastguard Worker pu1_buf2 = ps_ref_frm->pu1_buf2;
522*495ae853SAndroid Build Coastguard Worker pu1_buf3 = ps_ref_frm->pu1_buf3;
523*495ae853SAndroid Build Coastguard Worker
524*495ae853SAndroid Build Coastguard Worker if(u1_ref_pic_type == BOT_FLD)
525*495ae853SAndroid Build Coastguard Worker {
526*495ae853SAndroid Build Coastguard Worker pu1_buf1 += ps_ref_frm->u2_frm_wd_y;
527*495ae853SAndroid Build Coastguard Worker pu1_buf2 += ps_ref_frm->u2_frm_wd_uv;
528*495ae853SAndroid Build Coastguard Worker pu1_buf3 += ps_ref_frm->u2_frm_wd_uv;
529*495ae853SAndroid Build Coastguard Worker
530*495ae853SAndroid Build Coastguard Worker }
531*495ae853SAndroid Build Coastguard Worker
532*495ae853SAndroid Build Coastguard Worker
533*495ae853SAndroid Build Coastguard Worker
534*495ae853SAndroid Build Coastguard Worker if(ps_dec->ps_cur_pps->u1_wted_pred_flag)
535*495ae853SAndroid Build Coastguard Worker {
536*495ae853SAndroid Build Coastguard Worker pu4_wt_offset = (UWORD32*)&ps_dec->pu4_wt_ofsts[2
537*495ae853SAndroid Build Coastguard Worker * X3(i1_ref_idx)];
538*495ae853SAndroid Build Coastguard Worker }
539*495ae853SAndroid Build Coastguard Worker
540*495ae853SAndroid Build Coastguard Worker
541*495ae853SAndroid Build Coastguard Worker pu4_wt_offset = ps_pred_pkd->pu4_wt_offst;
542*495ae853SAndroid Build Coastguard Worker
543*495ae853SAndroid Build Coastguard Worker
544*495ae853SAndroid Build Coastguard Worker /* Pointer to the frame buffer */
545*495ae853SAndroid Build Coastguard Worker {
546*495ae853SAndroid Build Coastguard Worker ps_frame_buf = &ps_dec->s_tran_addrecon;
547*495ae853SAndroid Build Coastguard Worker /* CHANGED CODE */
548*495ae853SAndroid Build Coastguard Worker }
549*495ae853SAndroid Build Coastguard Worker
550*495ae853SAndroid Build Coastguard Worker if(!u1_pic_fld)
551*495ae853SAndroid Build Coastguard Worker {
552*495ae853SAndroid Build Coastguard Worker u1_mb_fld = ps_cur_mb_info->u1_mb_field_decodingflag;
553*495ae853SAndroid Build Coastguard Worker u1_mb_bot = 1 - ps_cur_mb_info->u1_topmb;
554*495ae853SAndroid Build Coastguard Worker }
555*495ae853SAndroid Build Coastguard Worker else
556*495ae853SAndroid Build Coastguard Worker u1_pic_bot = ps_cur_slice->u1_bottom_field_flag;
557*495ae853SAndroid Build Coastguard Worker
558*495ae853SAndroid Build Coastguard Worker /****************************************************************/
559*495ae853SAndroid Build Coastguard Worker /* calculating the flags the tell whether to use frame-padding */
560*495ae853SAndroid Build Coastguard Worker /* or use software pad-on-demand */
561*495ae853SAndroid Build Coastguard Worker /****************************************************************/
562*495ae853SAndroid Build Coastguard Worker u1_mb_or_pic_bot = u1_mb_bot | u1_pic_bot;
563*495ae853SAndroid Build Coastguard Worker u1_mb_or_pic_fld = u1_mb_fld | u1_pic_fld;
564*495ae853SAndroid Build Coastguard Worker u1_pod_bot = u1_mb_or_pic_fld && (u1_ref_pic_type == TOP_FLD);
565*495ae853SAndroid Build Coastguard Worker u1_pod_top = u1_mb_or_pic_fld && (u1_ref_pic_type == BOT_FLD);
566*495ae853SAndroid Build Coastguard Worker
567*495ae853SAndroid Build Coastguard Worker /* Weighted Pred additions */
568*495ae853SAndroid Build Coastguard Worker u1_wted_bipred_idc = ps_dec->ps_cur_pps->u1_wted_bipred_idc;
569*495ae853SAndroid Build Coastguard Worker
570*495ae853SAndroid Build Coastguard Worker if((u1_slice_type == P_SLICE) || (u1_slice_type == SP_SLICE))
571*495ae853SAndroid Build Coastguard Worker {
572*495ae853SAndroid Build Coastguard Worker /* P Slice only */
573*495ae853SAndroid Build Coastguard Worker u1_wght_pred_type = ps_dec->ps_cur_pps->u1_wted_pred_flag;
574*495ae853SAndroid Build Coastguard Worker
575*495ae853SAndroid Build Coastguard Worker }
576*495ae853SAndroid Build Coastguard Worker else
577*495ae853SAndroid Build Coastguard Worker {
578*495ae853SAndroid Build Coastguard Worker /* B Slice only */
579*495ae853SAndroid Build Coastguard Worker u1_wght_pred_type = 1 + u1_is_bi_dir;
580*495ae853SAndroid Build Coastguard Worker if(u1_wted_bipred_idc == 0)
581*495ae853SAndroid Build Coastguard Worker u1_wght_pred_type = 0;
582*495ae853SAndroid Build Coastguard Worker if((u1_wted_bipred_idc == 2) && (!u1_is_bi_dir))
583*495ae853SAndroid Build Coastguard Worker u1_wght_pred_type = 0;
584*495ae853SAndroid Build Coastguard Worker }
585*495ae853SAndroid Build Coastguard Worker /* load the scratch reference buffer index */
586*495ae853SAndroid Build Coastguard Worker pu1_ref_buf = ps_dec->pu1_ref_buff + ps_dec->u4_dma_buf_idx;
587*495ae853SAndroid Build Coastguard Worker u2_tot_ref_scratch_size = 0;
588*495ae853SAndroid Build Coastguard Worker
589*495ae853SAndroid Build Coastguard Worker
590*495ae853SAndroid Build Coastguard Worker /* Transfer Setup Y */
591*495ae853SAndroid Build Coastguard Worker {
592*495ae853SAndroid Build Coastguard Worker UWORD8 *pu1_pred, *pu1_rec;
593*495ae853SAndroid Build Coastguard Worker /* calculating rounded motion vectors and fractional components */
594*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_x = i2_mv_x;
595*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_y = i2_mv_y;
596*495ae853SAndroid Build Coastguard Worker
597*495ae853SAndroid Build Coastguard Worker u1_dx = i2_tmp_mv_x & 0x3;
598*495ae853SAndroid Build Coastguard Worker u1_dy = i2_tmp_mv_y & 0x3;
599*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_x >>= 2;
600*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_y >>= 2;
601*495ae853SAndroid Build Coastguard Worker i1_mc_wd = u1_part_wd << 2;
602*495ae853SAndroid Build Coastguard Worker u1_dma_ht = u1_part_ht << 2;
603*495ae853SAndroid Build Coastguard Worker if(u1_dx)
604*495ae853SAndroid Build Coastguard Worker {
605*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_x -= 2;
606*495ae853SAndroid Build Coastguard Worker i1_mc_wd += 5;
607*495ae853SAndroid Build Coastguard Worker }
608*495ae853SAndroid Build Coastguard Worker if(u1_dy)
609*495ae853SAndroid Build Coastguard Worker {
610*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_y -= 2;
611*495ae853SAndroid Build Coastguard Worker u1_dma_ht += 5;
612*495ae853SAndroid Build Coastguard Worker }
613*495ae853SAndroid Build Coastguard Worker
614*495ae853SAndroid Build Coastguard Worker /********************************************************************/
615*495ae853SAndroid Build Coastguard Worker /* Calulating the horizontal and the vertical u4_ofst from top left */
616*495ae853SAndroid Build Coastguard Worker /* edge of the reference frame, and subsequent clipping */
617*495ae853SAndroid Build Coastguard Worker /********************************************************************/
618*495ae853SAndroid Build Coastguard Worker u2_pic_ht = ps_dec->u2_pic_ht >> u1_pic_fld;
619*495ae853SAndroid Build Coastguard Worker u2_frm_wd = ps_dec->u2_frm_wd_y << u1_pic_fld;
620*495ae853SAndroid Build Coastguard Worker i2_frm_x = (u2_mb_x << 4) + (u1_sub_x << 2) + i2_tmp_mv_x;
621*495ae853SAndroid Build Coastguard Worker i2_frm_y = ((u2_mb_y + (u1_mb_bot && !u1_mb_fld)) << 4)
622*495ae853SAndroid Build Coastguard Worker + (((u1_sub_y << 2) + i2_tmp_mv_y) << u1_mb_fld);
623*495ae853SAndroid Build Coastguard Worker
624*495ae853SAndroid Build Coastguard Worker i2_frm_x = CLIP3(MAX_OFFSET_OUTSIDE_X_FRM, (ps_dec->u2_pic_wd - 1),
625*495ae853SAndroid Build Coastguard Worker i2_frm_x);
626*495ae853SAndroid Build Coastguard Worker i2_frm_y = CLIP3(((1 - u1_dma_ht) << u1_mb_fld),
627*495ae853SAndroid Build Coastguard Worker (u2_pic_ht - (1 << u1_mb_fld)), i2_frm_y);
628*495ae853SAndroid Build Coastguard Worker
629*495ae853SAndroid Build Coastguard Worker pu1_pred = pu1_buf1 + i2_frm_y * u2_frm_wd + i2_frm_x;
630*495ae853SAndroid Build Coastguard Worker u1_dma_wd = (i1_mc_wd + 3) & 0xFC;
631*495ae853SAndroid Build Coastguard Worker /********************************************************************/
632*495ae853SAndroid Build Coastguard Worker /* Calulating the horizontal and the vertical u4_ofst from top left */
633*495ae853SAndroid Build Coastguard Worker /* edge of the recon buffer */
634*495ae853SAndroid Build Coastguard Worker /********************************************************************/
635*495ae853SAndroid Build Coastguard Worker /* CHANGED CODE */
636*495ae853SAndroid Build Coastguard Worker u2_rec_wd = MB_SIZE;
637*495ae853SAndroid Build Coastguard Worker i2_rec_x = u1_sub_x << 2;
638*495ae853SAndroid Build Coastguard Worker i2_rec_y = u1_sub_y << 2;
639*495ae853SAndroid Build Coastguard Worker {
640*495ae853SAndroid Build Coastguard Worker u2_rec_wd = ps_dec->u2_frm_wd_y << u1_mb_or_pic_fld;
641*495ae853SAndroid Build Coastguard Worker i2_rec_x += (mb_index << 4);
642*495ae853SAndroid Build Coastguard Worker pu1_rec = ps_frame_buf->pu1_dest_y + i2_rec_y * u2_rec_wd
643*495ae853SAndroid Build Coastguard Worker + i2_rec_x;
644*495ae853SAndroid Build Coastguard Worker if(u1_mb_bot)
645*495ae853SAndroid Build Coastguard Worker pu1_rec += ps_dec->u2_frm_wd_y << ((u1_mb_fld) ? 0 : 4);
646*495ae853SAndroid Build Coastguard Worker }
647*495ae853SAndroid Build Coastguard Worker
648*495ae853SAndroid Build Coastguard Worker /* CHANGED CODE */
649*495ae853SAndroid Build Coastguard Worker
650*495ae853SAndroid Build Coastguard Worker /* filling the pred and dma structures for Y */
651*495ae853SAndroid Build Coastguard Worker u2_frm_wd = ps_dec->u2_frm_wd_y << u1_mb_or_pic_fld;
652*495ae853SAndroid Build Coastguard Worker
653*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_dma_dest_addr = pu1_ref_buf;
654*495ae853SAndroid Build Coastguard Worker ps_pred->u2_u1_ref_buf_wd = u1_dma_wd;
655*495ae853SAndroid Build Coastguard Worker ps_pred->u2_frm_wd = u2_frm_wd;
656*495ae853SAndroid Build Coastguard Worker ps_pred->i1_dma_ht = u1_dma_ht;
657*495ae853SAndroid Build Coastguard Worker ps_pred->i1_mc_wd = i1_mc_wd;
658*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_rec_y_u = pu1_rec;
659*495ae853SAndroid Build Coastguard Worker ps_pred->u2_dst_stride = u2_rec_wd;
660*495ae853SAndroid Build Coastguard Worker
661*495ae853SAndroid Build Coastguard Worker ps_pred->i1_mb_partwidth = u1_part_wd << 2;
662*495ae853SAndroid Build Coastguard Worker ps_pred->i1_mb_partheight = u1_part_ht << 2;
663*495ae853SAndroid Build Coastguard Worker ps_pred->u1_dydx = (u1_dy << 2) + u1_dx;
664*495ae853SAndroid Build Coastguard Worker ps_pred->u1_is_bi_direct = u1_is_bi_dir;
665*495ae853SAndroid Build Coastguard Worker ps_pred->u1_pi1_wt_ofst_rec_v = (UWORD8 *)pu4_wt_offset;
666*495ae853SAndroid Build Coastguard Worker ps_pred->u1_wght_pred_type = u1_wght_pred_type;
667*495ae853SAndroid Build Coastguard Worker ps_pred->i1_pod_ht = 0;
668*495ae853SAndroid Build Coastguard Worker
669*495ae853SAndroid Build Coastguard Worker /* Increment the Reference buffer Indices */
670*495ae853SAndroid Build Coastguard Worker pu1_ref_buf += u1_dma_wd * u1_dma_ht;
671*495ae853SAndroid Build Coastguard Worker u2_tot_ref_scratch_size += u1_dma_wd * u1_dma_ht;
672*495ae853SAndroid Build Coastguard Worker
673*495ae853SAndroid Build Coastguard Worker /* unrestricted field motion comp for top region outside frame */
674*495ae853SAndroid Build Coastguard Worker i2_pod_ht = (-i2_frm_y) >> u1_mb_fld;
675*495ae853SAndroid Build Coastguard Worker if((i2_pod_ht > 0) && u1_pod_top)
676*495ae853SAndroid Build Coastguard Worker {
677*495ae853SAndroid Build Coastguard Worker ps_pred->i1_pod_ht = (WORD8)(-i2_pod_ht);
678*495ae853SAndroid Build Coastguard Worker u1_dma_ht -= i2_pod_ht;
679*495ae853SAndroid Build Coastguard Worker pu1_pred += i2_pod_ht * u2_frm_wd;
680*495ae853SAndroid Build Coastguard Worker }
681*495ae853SAndroid Build Coastguard Worker /* unrestricted field motion comp for bottom region outside frame */
682*495ae853SAndroid Build Coastguard Worker else if(u1_pod_bot)
683*495ae853SAndroid Build Coastguard Worker {
684*495ae853SAndroid Build Coastguard Worker i2_pod_ht = u1_dma_ht + ((i2_frm_y - u2_pic_ht) >> u1_mb_fld);
685*495ae853SAndroid Build Coastguard Worker if(i2_pod_ht > 0)
686*495ae853SAndroid Build Coastguard Worker {
687*495ae853SAndroid Build Coastguard Worker u1_dma_ht -= i2_pod_ht;
688*495ae853SAndroid Build Coastguard Worker ps_pred->i1_pod_ht = (WORD8)i2_pod_ht;
689*495ae853SAndroid Build Coastguard Worker }
690*495ae853SAndroid Build Coastguard Worker }
691*495ae853SAndroid Build Coastguard Worker
692*495ae853SAndroid Build Coastguard Worker /* Copy Y partition */
693*495ae853SAndroid Build Coastguard Worker
694*495ae853SAndroid Build Coastguard Worker /*
695*495ae853SAndroid Build Coastguard Worker * ps_pred->i1_pod_ht is non zero when MBAFF is present. In case of MBAFF the reference data
696*495ae853SAndroid Build Coastguard Worker * is copied in the Scrath buffer so that the padding_on_demand doesnot corrupt the frame data
697*495ae853SAndroid Build Coastguard Worker */
698*495ae853SAndroid Build Coastguard Worker if(ps_pred->i1_pod_ht)
699*495ae853SAndroid Build Coastguard Worker {
700*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_pred = pu1_pred;
701*495ae853SAndroid Build Coastguard Worker ps_pred->u1_dma_ht_y = u1_dma_ht;
702*495ae853SAndroid Build Coastguard Worker ps_pred->u1_dma_wd_y = u1_dma_wd;
703*495ae853SAndroid Build Coastguard Worker }
704*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_y_ref = pu1_pred;
705*495ae853SAndroid Build Coastguard Worker }
706*495ae853SAndroid Build Coastguard Worker
707*495ae853SAndroid Build Coastguard Worker
708*495ae853SAndroid Build Coastguard Worker
709*495ae853SAndroid Build Coastguard Worker /* Increment ps_pred index */
710*495ae853SAndroid Build Coastguard Worker ps_pred++;
711*495ae853SAndroid Build Coastguard Worker
712*495ae853SAndroid Build Coastguard Worker /* Transfer Setup U & V */
713*495ae853SAndroid Build Coastguard Worker {
714*495ae853SAndroid Build Coastguard Worker WORD32 i4_ref_offset, i4_rec_offset;
715*495ae853SAndroid Build Coastguard Worker UWORD8 *pu1_pred_u, *pu1_pred_v, u1_tmp_dma_ht;
716*495ae853SAndroid Build Coastguard Worker /* CHANGED CODE */
717*495ae853SAndroid Build Coastguard Worker UWORD8 u1_chroma_cbp = (UWORD8)(ps_cur_mb_info->u1_cbp >> 4);
718*495ae853SAndroid Build Coastguard Worker /* CHANGED CODE */
719*495ae853SAndroid Build Coastguard Worker
720*495ae853SAndroid Build Coastguard Worker /* calculating rounded motion vectors and fractional components */
721*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_x = i2_mv_x;
722*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_y = i2_mv_y;
723*495ae853SAndroid Build Coastguard Worker
724*495ae853SAndroid Build Coastguard Worker /************************************************************************/
725*495ae853SAndroid Build Coastguard Worker /* Table 8-9: Derivation of the vertical component of the chroma vector */
726*495ae853SAndroid Build Coastguard Worker /* in field coding mode */
727*495ae853SAndroid Build Coastguard Worker /************************************************************************/
728*495ae853SAndroid Build Coastguard Worker if(u1_pod_bot && u1_mb_or_pic_bot)
729*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_y += 2;
730*495ae853SAndroid Build Coastguard Worker if(u1_pod_top && !u1_mb_or_pic_bot)
731*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_y -= 2;
732*495ae853SAndroid Build Coastguard Worker
733*495ae853SAndroid Build Coastguard Worker /* Eighth sample of the chroma MV */
734*495ae853SAndroid Build Coastguard Worker u1_dx = i2_tmp_mv_x & 0x7;
735*495ae853SAndroid Build Coastguard Worker u1_dy = i2_tmp_mv_y & 0x7;
736*495ae853SAndroid Build Coastguard Worker
737*495ae853SAndroid Build Coastguard Worker /********************************************************************/
738*495ae853SAndroid Build Coastguard Worker /* Calculating the full pel MV for chroma which is 1/2 of the Luma */
739*495ae853SAndroid Build Coastguard Worker /* MV in full pel units */
740*495ae853SAndroid Build Coastguard Worker /********************************************************************/
741*495ae853SAndroid Build Coastguard Worker i2_mv_x = i2_tmp_mv_x;
742*495ae853SAndroid Build Coastguard Worker i2_mv_y = i2_tmp_mv_y;
743*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_x = SIGN_POW2_DIV(i2_tmp_mv_x, 3);
744*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_y = SIGN_POW2_DIV(i2_tmp_mv_y, 3);
745*495ae853SAndroid Build Coastguard Worker i1_mc_wd = u1_part_wd << 1;
746*495ae853SAndroid Build Coastguard Worker u1_dma_ht = u1_part_ht << 1;
747*495ae853SAndroid Build Coastguard Worker if(u1_dx)
748*495ae853SAndroid Build Coastguard Worker {
749*495ae853SAndroid Build Coastguard Worker if(i2_mv_x < 0)
750*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_x -= 1;
751*495ae853SAndroid Build Coastguard Worker i1_mc_wd++;
752*495ae853SAndroid Build Coastguard Worker }
753*495ae853SAndroid Build Coastguard Worker if(u1_dy != 0)
754*495ae853SAndroid Build Coastguard Worker {
755*495ae853SAndroid Build Coastguard Worker if(i2_mv_y < 0)
756*495ae853SAndroid Build Coastguard Worker i2_tmp_mv_y -= 1;
757*495ae853SAndroid Build Coastguard Worker u1_dma_ht++;
758*495ae853SAndroid Build Coastguard Worker }
759*495ae853SAndroid Build Coastguard Worker
760*495ae853SAndroid Build Coastguard Worker /********************************************************************/
761*495ae853SAndroid Build Coastguard Worker /* Calulating the horizontal and the vertical u4_ofst from top left */
762*495ae853SAndroid Build Coastguard Worker /* edge of the reference frame, and subsequent clipping */
763*495ae853SAndroid Build Coastguard Worker /********************************************************************/
764*495ae853SAndroid Build Coastguard Worker u2_pic_ht >>= 1;
765*495ae853SAndroid Build Coastguard Worker u2_frm_wd = ps_dec->u2_frm_wd_uv << u1_pic_fld;
766*495ae853SAndroid Build Coastguard Worker i2_frm_x = (u2_mb_x << 3) + (u1_sub_x << 1) + i2_tmp_mv_x;
767*495ae853SAndroid Build Coastguard Worker i2_frm_y = ((u2_mb_y + (u1_mb_bot && !u1_mb_fld)) << 3)
768*495ae853SAndroid Build Coastguard Worker + (((u1_sub_y << 1) + i2_tmp_mv_y) << u1_mb_fld);
769*495ae853SAndroid Build Coastguard Worker
770*495ae853SAndroid Build Coastguard Worker i2_frm_x = CLIP3(MAX_OFFSET_OUTSIDE_UV_FRM,
771*495ae853SAndroid Build Coastguard Worker ((ps_dec->u2_pic_wd >> 1) - 1), i2_frm_x);
772*495ae853SAndroid Build Coastguard Worker i2_frm_y = CLIP3(((1 - u1_dma_ht) << u1_mb_fld),
773*495ae853SAndroid Build Coastguard Worker (u2_pic_ht - (1 << u1_mb_fld)), i2_frm_y);
774*495ae853SAndroid Build Coastguard Worker
775*495ae853SAndroid Build Coastguard Worker i4_ref_offset = i2_frm_y * u2_frm_wd + i2_frm_x * YUV420SP_FACTOR;
776*495ae853SAndroid Build Coastguard Worker u1_dma_wd = (i1_mc_wd + 3) & 0xFC;
777*495ae853SAndroid Build Coastguard Worker
778*495ae853SAndroid Build Coastguard Worker /********************************************************************/
779*495ae853SAndroid Build Coastguard Worker /* Calulating the horizontal and the vertical u4_ofst from top left */
780*495ae853SAndroid Build Coastguard Worker /* edge of the recon buffer */
781*495ae853SAndroid Build Coastguard Worker /********************************************************************/
782*495ae853SAndroid Build Coastguard Worker /* CHANGED CODE */
783*495ae853SAndroid Build Coastguard Worker u2_rec_wd = BLK8x8SIZE * YUV420SP_FACTOR;
784*495ae853SAndroid Build Coastguard Worker i2_rec_x = u1_sub_x << 1;
785*495ae853SAndroid Build Coastguard Worker i2_rec_y = u1_sub_y << 1;
786*495ae853SAndroid Build Coastguard Worker i4_rec_offset = i2_rec_y * u2_rec_wd + i2_rec_x * YUV420SP_FACTOR;
787*495ae853SAndroid Build Coastguard Worker {
788*495ae853SAndroid Build Coastguard Worker u2_rec_wd = ps_dec->u2_frm_wd_uv << u1_mb_or_pic_fld;
789*495ae853SAndroid Build Coastguard Worker
790*495ae853SAndroid Build Coastguard Worker i2_rec_x += (mb_index << 3);
791*495ae853SAndroid Build Coastguard Worker i4_rec_offset = i2_rec_y * u2_rec_wd + i2_rec_x * YUV420SP_FACTOR;
792*495ae853SAndroid Build Coastguard Worker if(u1_mb_bot)
793*495ae853SAndroid Build Coastguard Worker i4_rec_offset += ps_dec->u2_frm_wd_uv << ((u1_mb_fld) ? 0 : 3);
794*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_rec_y_u = ps_frame_buf->pu1_dest_u + i4_rec_offset;
795*495ae853SAndroid Build Coastguard Worker ps_pred->u1_pi1_wt_ofst_rec_v = ps_frame_buf->pu1_dest_v
796*495ae853SAndroid Build Coastguard Worker + i4_rec_offset;
797*495ae853SAndroid Build Coastguard Worker
798*495ae853SAndroid Build Coastguard Worker }
799*495ae853SAndroid Build Coastguard Worker
800*495ae853SAndroid Build Coastguard Worker /* CHANGED CODE */
801*495ae853SAndroid Build Coastguard Worker
802*495ae853SAndroid Build Coastguard Worker /* filling the common pred structures for U */
803*495ae853SAndroid Build Coastguard Worker u2_frm_wd = ps_dec->u2_frm_wd_uv << u1_mb_or_pic_fld;
804*495ae853SAndroid Build Coastguard Worker u1_tmp_dma_ht = u1_dma_ht;
805*495ae853SAndroid Build Coastguard Worker ps_pred->u2_u1_ref_buf_wd = u1_dma_wd;
806*495ae853SAndroid Build Coastguard Worker ps_pred->u2_frm_wd = u2_frm_wd;
807*495ae853SAndroid Build Coastguard Worker ps_pred->i1_dma_ht = u1_dma_ht;
808*495ae853SAndroid Build Coastguard Worker ps_pred->i1_mc_wd = i1_mc_wd;
809*495ae853SAndroid Build Coastguard Worker ps_pred->u2_dst_stride = u2_rec_wd;
810*495ae853SAndroid Build Coastguard Worker
811*495ae853SAndroid Build Coastguard Worker ps_pred->i1_mb_partwidth = u1_part_wd << 1;
812*495ae853SAndroid Build Coastguard Worker ps_pred->i1_mb_partheight = u1_part_ht << 1;
813*495ae853SAndroid Build Coastguard Worker ps_pred->u1_dydx = (u1_dy << 3) + u1_dx;
814*495ae853SAndroid Build Coastguard Worker ps_pred->u1_is_bi_direct = u1_is_bi_dir;
815*495ae853SAndroid Build Coastguard Worker ps_pred->u1_wght_pred_type = u1_wght_pred_type;
816*495ae853SAndroid Build Coastguard Worker ps_pred->i1_pod_ht = 0;
817*495ae853SAndroid Build Coastguard Worker
818*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_dma_dest_addr = pu1_ref_buf;
819*495ae853SAndroid Build Coastguard Worker
820*495ae853SAndroid Build Coastguard Worker /* unrestricted field motion comp for top region outside frame */
821*495ae853SAndroid Build Coastguard Worker i2_pod_ht = (-i2_frm_y) >> u1_mb_fld;
822*495ae853SAndroid Build Coastguard Worker if((i2_pod_ht > 0) && u1_pod_top)
823*495ae853SAndroid Build Coastguard Worker {
824*495ae853SAndroid Build Coastguard Worker i4_ref_offset += i2_pod_ht * u2_frm_wd;
825*495ae853SAndroid Build Coastguard Worker u1_dma_ht -= i2_pod_ht;
826*495ae853SAndroid Build Coastguard Worker ps_pred->i1_pod_ht = (WORD8)(-i2_pod_ht);
827*495ae853SAndroid Build Coastguard Worker }
828*495ae853SAndroid Build Coastguard Worker /* unrestricted field motion comp for bottom region outside frame */
829*495ae853SAndroid Build Coastguard Worker else if(u1_pod_bot)
830*495ae853SAndroid Build Coastguard Worker {
831*495ae853SAndroid Build Coastguard Worker i2_pod_ht = u1_dma_ht + ((i2_frm_y - u2_pic_ht) >> u1_mb_fld);
832*495ae853SAndroid Build Coastguard Worker if(i2_pod_ht > 0)
833*495ae853SAndroid Build Coastguard Worker {
834*495ae853SAndroid Build Coastguard Worker u1_dma_ht -= i2_pod_ht;
835*495ae853SAndroid Build Coastguard Worker ps_pred->i1_pod_ht = (WORD8)i2_pod_ht;
836*495ae853SAndroid Build Coastguard Worker }
837*495ae853SAndroid Build Coastguard Worker }
838*495ae853SAndroid Build Coastguard Worker
839*495ae853SAndroid Build Coastguard Worker pu1_pred_u = pu1_buf2 + i4_ref_offset;
840*495ae853SAndroid Build Coastguard Worker pu1_pred_v = pu1_buf3 + i4_ref_offset;
841*495ae853SAndroid Build Coastguard Worker
842*495ae853SAndroid Build Coastguard Worker /* Copy U & V partitions */
843*495ae853SAndroid Build Coastguard Worker if(ps_pred->i1_pod_ht)
844*495ae853SAndroid Build Coastguard Worker {
845*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_pred_u = pu1_pred_u;
846*495ae853SAndroid Build Coastguard Worker ps_pred->u1_dma_ht_uv = u1_dma_ht;
847*495ae853SAndroid Build Coastguard Worker ps_pred->u1_dma_wd_uv = u1_dma_wd;
848*495ae853SAndroid Build Coastguard Worker
849*495ae853SAndroid Build Coastguard Worker }
850*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_u_ref = pu1_pred_u;
851*495ae853SAndroid Build Coastguard Worker
852*495ae853SAndroid Build Coastguard Worker /* Increment the reference buffer Index */
853*495ae853SAndroid Build Coastguard Worker u2_tot_ref_scratch_size += (u1_dma_wd * u1_tmp_dma_ht) << 1;
854*495ae853SAndroid Build Coastguard Worker
855*495ae853SAndroid Build Coastguard Worker if(ps_pred->i1_pod_ht)
856*495ae853SAndroid Build Coastguard Worker {
857*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_pred_v = pu1_pred_v;
858*495ae853SAndroid Build Coastguard Worker ps_pred->u1_dma_ht_uv = u1_dma_ht;
859*495ae853SAndroid Build Coastguard Worker ps_pred->u1_dma_wd_uv = u1_dma_wd;
860*495ae853SAndroid Build Coastguard Worker }
861*495ae853SAndroid Build Coastguard Worker
862*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_v_ref = pu1_pred_v;
863*495ae853SAndroid Build Coastguard Worker }
864*495ae853SAndroid Build Coastguard Worker
865*495ae853SAndroid Build Coastguard Worker /* Increment ps_pred index */
866*495ae853SAndroid Build Coastguard Worker ps_dec->u4_pred_info_idx += 2;
867*495ae853SAndroid Build Coastguard Worker
868*495ae853SAndroid Build Coastguard Worker
869*495ae853SAndroid Build Coastguard Worker /* Increment the reference buffer Index */
870*495ae853SAndroid Build Coastguard Worker ps_dec->u4_dma_buf_idx += u2_tot_ref_scratch_size;
871*495ae853SAndroid Build Coastguard Worker
872*495ae853SAndroid Build Coastguard Worker if(ps_dec->u4_dma_buf_idx > MAX_REF_BUF_SIZE)
873*495ae853SAndroid Build Coastguard Worker return ERROR_NUM_MV;
874*495ae853SAndroid Build Coastguard Worker
875*495ae853SAndroid Build Coastguard Worker return OK;
876*495ae853SAndroid Build Coastguard Worker
877*495ae853SAndroid Build Coastguard Worker
878*495ae853SAndroid Build Coastguard Worker
879*495ae853SAndroid Build Coastguard Worker }
880*495ae853SAndroid Build Coastguard Worker
881*495ae853SAndroid Build Coastguard Worker
882*495ae853SAndroid Build Coastguard Worker /*!
883*495ae853SAndroid Build Coastguard Worker **************************************************************************
884*495ae853SAndroid Build Coastguard Worker * \if Function name : MotionCompensate \endif
885*495ae853SAndroid Build Coastguard Worker *
886*495ae853SAndroid Build Coastguard Worker * \brief
887*495ae853SAndroid Build Coastguard Worker * The routine forms predictor blocks for the entire MB and stores it in
888*495ae853SAndroid Build Coastguard Worker * predictor buffers.This function works only for BASELINE profile
889*495ae853SAndroid Build Coastguard Worker *
890*495ae853SAndroid Build Coastguard Worker * \param ps_dec: Pointer to the structure decStruct. This is used to get
891*495ae853SAndroid Build Coastguard Worker * pointers to the current and the reference frame and to the MbParams
892*495ae853SAndroid Build Coastguard Worker * structure.
893*495ae853SAndroid Build Coastguard Worker *
894*495ae853SAndroid Build Coastguard Worker * \return
895*495ae853SAndroid Build Coastguard Worker * None
896*495ae853SAndroid Build Coastguard Worker *
897*495ae853SAndroid Build Coastguard Worker * \note
898*495ae853SAndroid Build Coastguard Worker * The routine forms predictors for all the luma and the chroma MB
899*495ae853SAndroid Build Coastguard Worker * partitions.
900*495ae853SAndroid Build Coastguard Worker **************************************************************************
901*495ae853SAndroid Build Coastguard Worker */
902*495ae853SAndroid Build Coastguard Worker
ih264d_motion_compensate_bp(dec_struct_t * ps_dec,dec_mb_info_t * ps_cur_mb_info)903*495ae853SAndroid Build Coastguard Worker void ih264d_motion_compensate_bp(dec_struct_t * ps_dec, dec_mb_info_t *ps_cur_mb_info)
904*495ae853SAndroid Build Coastguard Worker {
905*495ae853SAndroid Build Coastguard Worker pred_info_t *ps_pred ;
906*495ae853SAndroid Build Coastguard Worker UWORD8 *puc_ref, *pu1_dest_y;
907*495ae853SAndroid Build Coastguard Worker UWORD8 *pu1_dest_u;
908*495ae853SAndroid Build Coastguard Worker UWORD32 u2_num_pels, u2_ref_wd_y, u2_ref_wd_uv, u2_dst_wd;
909*495ae853SAndroid Build Coastguard Worker
910*495ae853SAndroid Build Coastguard Worker UWORD32 u4_wd_y, u4_ht_y, u4_wd_uv;
911*495ae853SAndroid Build Coastguard Worker UWORD32 u4_ht_uv;
912*495ae853SAndroid Build Coastguard Worker UWORD8 *puc_pred0 = (UWORD8 *)(ps_dec->pi2_pred1);
913*495ae853SAndroid Build Coastguard Worker
914*495ae853SAndroid Build Coastguard Worker
915*495ae853SAndroid Build Coastguard Worker PROFILE_DISABLE_INTER_PRED()
916*495ae853SAndroid Build Coastguard Worker UNUSED(ps_cur_mb_info);
917*495ae853SAndroid Build Coastguard Worker ps_pred = ps_dec->ps_pred ;
918*495ae853SAndroid Build Coastguard Worker
919*495ae853SAndroid Build Coastguard Worker for(u2_num_pels = 0; u2_num_pels < 256;)
920*495ae853SAndroid Build Coastguard Worker {
921*495ae853SAndroid Build Coastguard Worker UWORD32 uc_dx, uc_dy;
922*495ae853SAndroid Build Coastguard Worker /* Pointer to the destination buffer. If the CBPs of all 8x8 blocks in
923*495ae853SAndroid Build Coastguard Worker the MB partition are zero then it would be better to copy the
924*495ae853SAndroid Build Coastguard Worker predictor valus directly to the current frame buffer */
925*495ae853SAndroid Build Coastguard Worker /*
926*495ae853SAndroid Build Coastguard Worker * ps_pred->i1_pod_ht is non zero when MBAFF is present. In case of MBAFF the reference data
927*495ae853SAndroid Build Coastguard Worker * is copied in the Scrath buffer so that the padding_on_demand doesnot corrupt the frame data
928*495ae853SAndroid Build Coastguard Worker */
929*495ae853SAndroid Build Coastguard Worker
930*495ae853SAndroid Build Coastguard Worker u2_ref_wd_y = ps_pred->u2_frm_wd;
931*495ae853SAndroid Build Coastguard Worker puc_ref = ps_pred->pu1_y_ref;
932*495ae853SAndroid Build Coastguard Worker if(ps_pred->u1_dydx & 0x3)
933*495ae853SAndroid Build Coastguard Worker puc_ref += 2;
934*495ae853SAndroid Build Coastguard Worker if(ps_pred->u1_dydx >> 2)
935*495ae853SAndroid Build Coastguard Worker puc_ref += 2 * u2_ref_wd_y;
936*495ae853SAndroid Build Coastguard Worker
937*495ae853SAndroid Build Coastguard Worker u4_wd_y = ps_pred->i1_mb_partwidth;
938*495ae853SAndroid Build Coastguard Worker u4_ht_y = ps_pred->i1_mb_partheight;
939*495ae853SAndroid Build Coastguard Worker uc_dx = ps_pred->u1_dydx;
940*495ae853SAndroid Build Coastguard Worker uc_dy = uc_dx >> 2;
941*495ae853SAndroid Build Coastguard Worker uc_dx &= 0x3;
942*495ae853SAndroid Build Coastguard Worker
943*495ae853SAndroid Build Coastguard Worker pu1_dest_y = ps_pred->pu1_rec_y_u;
944*495ae853SAndroid Build Coastguard Worker u2_dst_wd = ps_pred->u2_dst_stride;
945*495ae853SAndroid Build Coastguard Worker
946*495ae853SAndroid Build Coastguard Worker ps_dec->apf_inter_pred_luma[ps_pred->u1_dydx](puc_ref, pu1_dest_y,
947*495ae853SAndroid Build Coastguard Worker u2_ref_wd_y,
948*495ae853SAndroid Build Coastguard Worker u2_dst_wd,
949*495ae853SAndroid Build Coastguard Worker u4_ht_y,
950*495ae853SAndroid Build Coastguard Worker u4_wd_y, puc_pred0,
951*495ae853SAndroid Build Coastguard Worker ps_pred->u1_dydx);
952*495ae853SAndroid Build Coastguard Worker
953*495ae853SAndroid Build Coastguard Worker ps_pred++;
954*495ae853SAndroid Build Coastguard Worker
955*495ae853SAndroid Build Coastguard Worker /* Interpolate samples for the chroma components */
956*495ae853SAndroid Build Coastguard Worker {
957*495ae853SAndroid Build Coastguard Worker UWORD8 *pu1_ref_u;
958*495ae853SAndroid Build Coastguard Worker
959*495ae853SAndroid Build Coastguard Worker u2_ref_wd_uv = ps_pred->u2_frm_wd;
960*495ae853SAndroid Build Coastguard Worker pu1_ref_u = ps_pred->pu1_u_ref;
961*495ae853SAndroid Build Coastguard Worker
962*495ae853SAndroid Build Coastguard Worker u4_wd_uv = ps_pred->i1_mb_partwidth;
963*495ae853SAndroid Build Coastguard Worker u4_ht_uv = ps_pred->i1_mb_partheight;
964*495ae853SAndroid Build Coastguard Worker uc_dx = ps_pred->u1_dydx; /* 8*dy + dx */
965*495ae853SAndroid Build Coastguard Worker uc_dy = uc_dx >> 3;
966*495ae853SAndroid Build Coastguard Worker uc_dx &= 0x7;
967*495ae853SAndroid Build Coastguard Worker
968*495ae853SAndroid Build Coastguard Worker pu1_dest_u = ps_pred->pu1_rec_y_u;
969*495ae853SAndroid Build Coastguard Worker u2_dst_wd = ps_pred->u2_dst_stride;
970*495ae853SAndroid Build Coastguard Worker
971*495ae853SAndroid Build Coastguard Worker ps_pred++;
972*495ae853SAndroid Build Coastguard Worker ps_dec->pf_inter_pred_chroma(pu1_ref_u, pu1_dest_u, u2_ref_wd_uv,
973*495ae853SAndroid Build Coastguard Worker u2_dst_wd, uc_dx, uc_dy,
974*495ae853SAndroid Build Coastguard Worker u4_ht_uv, u4_wd_uv);
975*495ae853SAndroid Build Coastguard Worker
976*495ae853SAndroid Build Coastguard Worker }
977*495ae853SAndroid Build Coastguard Worker
978*495ae853SAndroid Build Coastguard Worker u2_num_pels += (UWORD8)u4_wd_y * (UWORD8)u4_ht_y;
979*495ae853SAndroid Build Coastguard Worker
980*495ae853SAndroid Build Coastguard Worker }
981*495ae853SAndroid Build Coastguard Worker }
982*495ae853SAndroid Build Coastguard Worker
983*495ae853SAndroid Build Coastguard Worker
984*495ae853SAndroid Build Coastguard Worker /*
985*495ae853SAndroid Build Coastguard Worker **************************************************************************
986*495ae853SAndroid Build Coastguard Worker * \if Function name : MotionCompensateB \endif
987*495ae853SAndroid Build Coastguard Worker *
988*495ae853SAndroid Build Coastguard Worker * \brief
989*495ae853SAndroid Build Coastguard Worker * The routine forms predictor blocks for the entire MB and stores it in
990*495ae853SAndroid Build Coastguard Worker * predictor buffers.
991*495ae853SAndroid Build Coastguard Worker *
992*495ae853SAndroid Build Coastguard Worker * \param ps_dec: Pointer to the structure decStruct. This is used to get
993*495ae853SAndroid Build Coastguard Worker * pointers to the current and the reference frame and to the MbParams
994*495ae853SAndroid Build Coastguard Worker * structure.
995*495ae853SAndroid Build Coastguard Worker *
996*495ae853SAndroid Build Coastguard Worker * \return
997*495ae853SAndroid Build Coastguard Worker * None
998*495ae853SAndroid Build Coastguard Worker *
999*495ae853SAndroid Build Coastguard Worker * \note
1000*495ae853SAndroid Build Coastguard Worker * The routine forms predictors for all the luma and the chroma MB
1001*495ae853SAndroid Build Coastguard Worker * partitions.
1002*495ae853SAndroid Build Coastguard Worker **************************************************************************
1003*495ae853SAndroid Build Coastguard Worker */
1004*495ae853SAndroid Build Coastguard Worker
ih264d_motion_compensate_mp(dec_struct_t * ps_dec,dec_mb_info_t * ps_cur_mb_info)1005*495ae853SAndroid Build Coastguard Worker void ih264d_motion_compensate_mp(dec_struct_t * ps_dec, dec_mb_info_t *ps_cur_mb_info)
1006*495ae853SAndroid Build Coastguard Worker {
1007*495ae853SAndroid Build Coastguard Worker pred_info_t *ps_pred ;
1008*495ae853SAndroid Build Coastguard Worker pred_info_t *ps_pred_y_forw, *ps_pred_y_back, *ps_pred_cr_forw;
1009*495ae853SAndroid Build Coastguard Worker UWORD8 *puc_ref, *pu1_dest_y, *puc_pred0, *puc_pred1;
1010*495ae853SAndroid Build Coastguard Worker UWORD8 *pu1_dest_u, *pu1_dest_v;
1011*495ae853SAndroid Build Coastguard Worker WORD16 *pi16_intm;
1012*495ae853SAndroid Build Coastguard Worker UWORD32 u2_num_pels, u2_ref_wd_y, u2_ref_wd_uv, u2_dst_wd;
1013*495ae853SAndroid Build Coastguard Worker UWORD32 u2_dest_wd_y, u2_dest_wd_uv;
1014*495ae853SAndroid Build Coastguard Worker UWORD32 u2_row_buf_wd_y = 0;
1015*495ae853SAndroid Build Coastguard Worker UWORD32 u2_row_buf_wd_uv = 0;
1016*495ae853SAndroid Build Coastguard Worker UWORD32 u2_log2Y_crwd;
1017*495ae853SAndroid Build Coastguard Worker UWORD32 u4_wd_y, u4_ht_y, u1_dir, u4_wd_uv;
1018*495ae853SAndroid Build Coastguard Worker UWORD32 u4_ht_uv;
1019*495ae853SAndroid Build Coastguard Worker UWORD8 *pu1_temp_mc_buffer = ps_dec->pu1_temp_mc_buffer;
1020*495ae853SAndroid Build Coastguard Worker WORD32 i2_pod_ht;
1021*495ae853SAndroid Build Coastguard Worker UWORD32 u2_pic_ht, u2_frm_wd, u2_rec_wd;
1022*495ae853SAndroid Build Coastguard Worker UWORD32 u1_pod_bot, u1_pod_top;
1023*495ae853SAndroid Build Coastguard Worker UWORD8 *pu1_pred, *pu1_dma_dst;
1024*495ae853SAndroid Build Coastguard Worker UWORD32 u1_dma_wd, u1_dma_ht;
1025*495ae853SAndroid Build Coastguard Worker
1026*495ae853SAndroid Build Coastguard Worker dec_slice_params_t * const ps_cur_slice = ps_dec->ps_cur_slice;
1027*495ae853SAndroid Build Coastguard Worker
1028*495ae853SAndroid Build Coastguard Worker /* set default value to flags specifying field nature of picture & mb */
1029*495ae853SAndroid Build Coastguard Worker UWORD32 u1_mb_fld = 0, u1_mb_or_pic_fld;
1030*495ae853SAndroid Build Coastguard Worker UWORD32 u1_mb_or_pic_bot;
1031*495ae853SAndroid Build Coastguard Worker /* calculate flags specifying field nature of picture & mb */
1032*495ae853SAndroid Build Coastguard Worker const UWORD8 u1_pic_fld = ps_cur_slice->u1_field_pic_flag;
1033*495ae853SAndroid Build Coastguard Worker
1034*495ae853SAndroid Build Coastguard Worker PROFILE_DISABLE_INTER_PRED()
1035*495ae853SAndroid Build Coastguard Worker ps_pred = ps_dec->ps_pred ;
1036*495ae853SAndroid Build Coastguard Worker /* Initialize both ps_pred_y_forw, ps_pred_cr_forw and ps_pred_y_back
1037*495ae853SAndroid Build Coastguard Worker * to avoid static analysis warnings */
1038*495ae853SAndroid Build Coastguard Worker ps_pred_y_forw = ps_pred;
1039*495ae853SAndroid Build Coastguard Worker ps_pred_y_back = ps_pred;
1040*495ae853SAndroid Build Coastguard Worker ps_pred_cr_forw = ps_pred;
1041*495ae853SAndroid Build Coastguard Worker
1042*495ae853SAndroid Build Coastguard Worker u2_log2Y_crwd = ps_dec->ps_decode_cur_slice->u2_log2Y_crwd;
1043*495ae853SAndroid Build Coastguard Worker
1044*495ae853SAndroid Build Coastguard Worker if(!u1_pic_fld)
1045*495ae853SAndroid Build Coastguard Worker {
1046*495ae853SAndroid Build Coastguard Worker u1_mb_fld = ps_cur_mb_info->u1_mb_field_decodingflag;
1047*495ae853SAndroid Build Coastguard Worker }
1048*495ae853SAndroid Build Coastguard Worker
1049*495ae853SAndroid Build Coastguard Worker u1_mb_or_pic_fld = u1_mb_fld | u1_pic_fld;
1050*495ae853SAndroid Build Coastguard Worker
1051*495ae853SAndroid Build Coastguard Worker pi16_intm = ps_dec->pi2_pred1;
1052*495ae853SAndroid Build Coastguard Worker puc_pred0 = (UWORD8 *)pi16_intm;
1053*495ae853SAndroid Build Coastguard Worker puc_pred1 = puc_pred0 + PRED_BUFFER_WIDTH * PRED_BUFFER_HEIGHT * sizeof(WORD16);
1054*495ae853SAndroid Build Coastguard Worker
1055*495ae853SAndroid Build Coastguard Worker for(u2_num_pels = 0; u2_num_pels < 256;)
1056*495ae853SAndroid Build Coastguard Worker {
1057*495ae853SAndroid Build Coastguard Worker UWORD8 uc_dx, uc_dy;
1058*495ae853SAndroid Build Coastguard Worker const UWORD8 u1_is_bi_direct = ps_pred->u1_is_bi_direct;
1059*495ae853SAndroid Build Coastguard Worker for(u1_dir = 0; u1_dir <= u1_is_bi_direct; u1_dir++)
1060*495ae853SAndroid Build Coastguard Worker {
1061*495ae853SAndroid Build Coastguard Worker /* Pointer to the destination buffer. If the CBPs of all 8x8 blocks in
1062*495ae853SAndroid Build Coastguard Worker the MB partition are zero then it would be better to copy the
1063*495ae853SAndroid Build Coastguard Worker predictor valus directly to the current frame buffer */
1064*495ae853SAndroid Build Coastguard Worker /*
1065*495ae853SAndroid Build Coastguard Worker * ps_pred->i1_pod_ht is non zero when MBAFF is present. In case of MBAFF the reference data
1066*495ae853SAndroid Build Coastguard Worker * is copied in the Scrath buffer so that the padding_on_demand doesnot corrupt the frame data
1067*495ae853SAndroid Build Coastguard Worker */
1068*495ae853SAndroid Build Coastguard Worker
1069*495ae853SAndroid Build Coastguard Worker if(ps_pred->i1_pod_ht)
1070*495ae853SAndroid Build Coastguard Worker {
1071*495ae853SAndroid Build Coastguard Worker u2_ref_wd_y = ps_pred->u2_u1_ref_buf_wd;
1072*495ae853SAndroid Build Coastguard Worker puc_ref = ps_pred->pu1_dma_dest_addr;
1073*495ae853SAndroid Build Coastguard Worker }
1074*495ae853SAndroid Build Coastguard Worker else
1075*495ae853SAndroid Build Coastguard Worker {
1076*495ae853SAndroid Build Coastguard Worker u2_ref_wd_y = ps_pred->u2_frm_wd;
1077*495ae853SAndroid Build Coastguard Worker puc_ref = ps_pred->pu1_y_ref;
1078*495ae853SAndroid Build Coastguard Worker
1079*495ae853SAndroid Build Coastguard Worker }
1080*495ae853SAndroid Build Coastguard Worker
1081*495ae853SAndroid Build Coastguard Worker if(ps_pred->u1_dydx & 0x3)
1082*495ae853SAndroid Build Coastguard Worker puc_ref += 2;
1083*495ae853SAndroid Build Coastguard Worker if(ps_pred->u1_dydx >> 2)
1084*495ae853SAndroid Build Coastguard Worker puc_ref += 2 * u2_ref_wd_y;
1085*495ae853SAndroid Build Coastguard Worker u4_wd_y = ps_pred->i1_mb_partwidth;
1086*495ae853SAndroid Build Coastguard Worker u4_ht_y = ps_pred->i1_mb_partheight;
1087*495ae853SAndroid Build Coastguard Worker
1088*495ae853SAndroid Build Coastguard Worker uc_dx = ps_pred->u1_dydx;
1089*495ae853SAndroid Build Coastguard Worker uc_dy = uc_dx >> 2;
1090*495ae853SAndroid Build Coastguard Worker uc_dx &= 0x3;
1091*495ae853SAndroid Build Coastguard Worker if(u1_dir == 0)
1092*495ae853SAndroid Build Coastguard Worker {
1093*495ae853SAndroid Build Coastguard Worker pu1_dest_y = ps_pred->pu1_rec_y_u;
1094*495ae853SAndroid Build Coastguard Worker u2_row_buf_wd_y = ps_pred->u2_dst_stride;
1095*495ae853SAndroid Build Coastguard Worker u2_dst_wd = ps_pred->u2_dst_stride;
1096*495ae853SAndroid Build Coastguard Worker u2_dest_wd_y = u2_dst_wd;
1097*495ae853SAndroid Build Coastguard Worker ps_pred_y_forw = ps_pred;
1098*495ae853SAndroid Build Coastguard Worker }
1099*495ae853SAndroid Build Coastguard Worker else
1100*495ae853SAndroid Build Coastguard Worker {
1101*495ae853SAndroid Build Coastguard Worker pu1_dest_y = pu1_temp_mc_buffer;
1102*495ae853SAndroid Build Coastguard Worker u2_dst_wd = MB_SIZE;
1103*495ae853SAndroid Build Coastguard Worker u2_dest_wd_y = u2_dst_wd;
1104*495ae853SAndroid Build Coastguard Worker ps_pred_y_back = ps_pred;
1105*495ae853SAndroid Build Coastguard Worker ps_pred_y_back->pu1_rec_y_u = pu1_dest_y;
1106*495ae853SAndroid Build Coastguard Worker }
1107*495ae853SAndroid Build Coastguard Worker
1108*495ae853SAndroid Build Coastguard Worker /* padding on demand (POD) for y done here */
1109*495ae853SAndroid Build Coastguard Worker
1110*495ae853SAndroid Build Coastguard Worker if(ps_pred->i1_pod_ht)
1111*495ae853SAndroid Build Coastguard Worker {
1112*495ae853SAndroid Build Coastguard Worker pu1_pred = ps_pred->pu1_pred;
1113*495ae853SAndroid Build Coastguard Worker pu1_dma_dst = ps_pred->pu1_dma_dest_addr;
1114*495ae853SAndroid Build Coastguard Worker u1_dma_wd = ps_pred->u1_dma_wd_y;
1115*495ae853SAndroid Build Coastguard Worker u1_dma_ht = ps_pred->u1_dma_ht_y;
1116*495ae853SAndroid Build Coastguard Worker u2_frm_wd = ps_dec->u2_frm_wd_y << u1_mb_or_pic_fld;
1117*495ae853SAndroid Build Coastguard Worker if(ps_pred->i1_pod_ht < 0)
1118*495ae853SAndroid Build Coastguard Worker {
1119*495ae853SAndroid Build Coastguard Worker pu1_dma_dst = pu1_dma_dst - (ps_pred->i1_pod_ht * ps_pred->u2_u1_ref_buf_wd);
1120*495ae853SAndroid Build Coastguard Worker }
1121*495ae853SAndroid Build Coastguard Worker ih264d_copy_2d1d(pu1_pred, pu1_dma_dst, u2_frm_wd, u1_dma_wd,
1122*495ae853SAndroid Build Coastguard Worker u1_dma_ht);
1123*495ae853SAndroid Build Coastguard Worker ih264d_pad_on_demand(ps_pred, LUM_BLK);
1124*495ae853SAndroid Build Coastguard Worker }
1125*495ae853SAndroid Build Coastguard Worker ps_dec->apf_inter_pred_luma[ps_pred->u1_dydx](puc_ref, pu1_dest_y,
1126*495ae853SAndroid Build Coastguard Worker u2_ref_wd_y,
1127*495ae853SAndroid Build Coastguard Worker u2_dst_wd,
1128*495ae853SAndroid Build Coastguard Worker u4_ht_y,
1129*495ae853SAndroid Build Coastguard Worker u4_wd_y,
1130*495ae853SAndroid Build Coastguard Worker puc_pred0,
1131*495ae853SAndroid Build Coastguard Worker ps_pred->u1_dydx);
1132*495ae853SAndroid Build Coastguard Worker ps_pred++;
1133*495ae853SAndroid Build Coastguard Worker
1134*495ae853SAndroid Build Coastguard Worker /* Interpolate samples for the chroma components */
1135*495ae853SAndroid Build Coastguard Worker {
1136*495ae853SAndroid Build Coastguard Worker UWORD8 *pu1_ref_u;
1137*495ae853SAndroid Build Coastguard Worker UWORD32 u1_dma_ht;
1138*495ae853SAndroid Build Coastguard Worker
1139*495ae853SAndroid Build Coastguard Worker /* padding on demand (POD) for U and V done here */
1140*495ae853SAndroid Build Coastguard Worker u1_dma_ht = ps_pred->i1_dma_ht;
1141*495ae853SAndroid Build Coastguard Worker
1142*495ae853SAndroid Build Coastguard Worker if(ps_pred->i1_pod_ht)
1143*495ae853SAndroid Build Coastguard Worker {
1144*495ae853SAndroid Build Coastguard Worker pu1_pred = ps_pred->pu1_pred_u;
1145*495ae853SAndroid Build Coastguard Worker pu1_dma_dst = ps_pred->pu1_dma_dest_addr;
1146*495ae853SAndroid Build Coastguard Worker u1_dma_ht = ps_pred->u1_dma_ht_uv;
1147*495ae853SAndroid Build Coastguard Worker u1_dma_wd = ps_pred->u1_dma_wd_uv * YUV420SP_FACTOR;
1148*495ae853SAndroid Build Coastguard Worker u2_frm_wd = ps_dec->u2_frm_wd_uv << u1_mb_or_pic_fld;
1149*495ae853SAndroid Build Coastguard Worker if(ps_pred->i1_pod_ht < 0)
1150*495ae853SAndroid Build Coastguard Worker {
1151*495ae853SAndroid Build Coastguard Worker /*Top POD*/
1152*495ae853SAndroid Build Coastguard Worker pu1_dma_dst -= (ps_pred->i1_pod_ht
1153*495ae853SAndroid Build Coastguard Worker * ps_pred->u2_u1_ref_buf_wd
1154*495ae853SAndroid Build Coastguard Worker * YUV420SP_FACTOR);
1155*495ae853SAndroid Build Coastguard Worker }
1156*495ae853SAndroid Build Coastguard Worker
1157*495ae853SAndroid Build Coastguard Worker ih264d_copy_2d1d(pu1_pred, pu1_dma_dst, u2_frm_wd,
1158*495ae853SAndroid Build Coastguard Worker u1_dma_wd, u1_dma_ht);
1159*495ae853SAndroid Build Coastguard Worker
1160*495ae853SAndroid Build Coastguard Worker pu1_dma_dst += (ps_pred->i1_dma_ht
1161*495ae853SAndroid Build Coastguard Worker * ps_pred->u2_u1_ref_buf_wd);
1162*495ae853SAndroid Build Coastguard Worker pu1_pred = ps_pred->pu1_pred_v;
1163*495ae853SAndroid Build Coastguard Worker
1164*495ae853SAndroid Build Coastguard Worker ih264d_pad_on_demand(ps_pred, CHROM_BLK);
1165*495ae853SAndroid Build Coastguard Worker }
1166*495ae853SAndroid Build Coastguard Worker
1167*495ae853SAndroid Build Coastguard Worker if(ps_pred->i1_pod_ht)
1168*495ae853SAndroid Build Coastguard Worker {
1169*495ae853SAndroid Build Coastguard Worker pu1_ref_u = ps_pred->pu1_dma_dest_addr;
1170*495ae853SAndroid Build Coastguard Worker
1171*495ae853SAndroid Build Coastguard Worker u2_ref_wd_uv = ps_pred->u2_u1_ref_buf_wd
1172*495ae853SAndroid Build Coastguard Worker * YUV420SP_FACTOR;
1173*495ae853SAndroid Build Coastguard Worker }
1174*495ae853SAndroid Build Coastguard Worker else
1175*495ae853SAndroid Build Coastguard Worker {
1176*495ae853SAndroid Build Coastguard Worker u2_ref_wd_uv = ps_pred->u2_frm_wd;
1177*495ae853SAndroid Build Coastguard Worker pu1_ref_u = ps_pred->pu1_u_ref;
1178*495ae853SAndroid Build Coastguard Worker
1179*495ae853SAndroid Build Coastguard Worker }
1180*495ae853SAndroid Build Coastguard Worker
1181*495ae853SAndroid Build Coastguard Worker u4_wd_uv = ps_pred->i1_mb_partwidth;
1182*495ae853SAndroid Build Coastguard Worker u4_ht_uv = ps_pred->i1_mb_partheight;
1183*495ae853SAndroid Build Coastguard Worker uc_dx = ps_pred->u1_dydx; /* 8*dy + dx */
1184*495ae853SAndroid Build Coastguard Worker uc_dy = uc_dx >> 3;
1185*495ae853SAndroid Build Coastguard Worker uc_dx &= 0x7;
1186*495ae853SAndroid Build Coastguard Worker if(u1_dir == 0)
1187*495ae853SAndroid Build Coastguard Worker {
1188*495ae853SAndroid Build Coastguard Worker pu1_dest_u = ps_pred->pu1_rec_y_u;
1189*495ae853SAndroid Build Coastguard Worker
1190*495ae853SAndroid Build Coastguard Worker pu1_dest_v = ps_pred->u1_pi1_wt_ofst_rec_v;
1191*495ae853SAndroid Build Coastguard Worker u2_row_buf_wd_uv = ps_pred->u2_dst_stride;
1192*495ae853SAndroid Build Coastguard Worker u2_dst_wd = ps_pred->u2_dst_stride;
1193*495ae853SAndroid Build Coastguard Worker u2_dest_wd_uv = u2_dst_wd;
1194*495ae853SAndroid Build Coastguard Worker ps_pred_cr_forw = ps_pred;
1195*495ae853SAndroid Build Coastguard Worker }
1196*495ae853SAndroid Build Coastguard Worker else
1197*495ae853SAndroid Build Coastguard Worker {
1198*495ae853SAndroid Build Coastguard Worker pu1_dest_u = puc_pred0;
1199*495ae853SAndroid Build Coastguard Worker
1200*495ae853SAndroid Build Coastguard Worker pu1_dest_v = puc_pred1;
1201*495ae853SAndroid Build Coastguard Worker u2_dest_wd_uv = BUFFER_WIDTH;
1202*495ae853SAndroid Build Coastguard Worker u2_dst_wd = BUFFER_WIDTH;
1203*495ae853SAndroid Build Coastguard Worker ps_pred->pu1_rec_y_u = pu1_dest_u;
1204*495ae853SAndroid Build Coastguard Worker ps_pred->u1_pi1_wt_ofst_rec_v = pu1_dest_v;
1205*495ae853SAndroid Build Coastguard Worker }
1206*495ae853SAndroid Build Coastguard Worker
1207*495ae853SAndroid Build Coastguard Worker ps_pred++;
1208*495ae853SAndroid Build Coastguard Worker ps_dec->pf_inter_pred_chroma(pu1_ref_u, pu1_dest_u,
1209*495ae853SAndroid Build Coastguard Worker u2_ref_wd_uv, u2_dst_wd,
1210*495ae853SAndroid Build Coastguard Worker uc_dx, uc_dy, u4_ht_uv,
1211*495ae853SAndroid Build Coastguard Worker u4_wd_uv);
1212*495ae853SAndroid Build Coastguard Worker
1213*495ae853SAndroid Build Coastguard Worker if(ps_cur_mb_info->u1_Mux == 1)
1214*495ae853SAndroid Build Coastguard Worker {
1215*495ae853SAndroid Build Coastguard Worker /******************************************************************/
1216*495ae853SAndroid Build Coastguard Worker /* padding on demand (POD) for U and V done here */
1217*495ae853SAndroid Build Coastguard Worker /* ps_pred now points to the Y entry of the 0,0 component */
1218*495ae853SAndroid Build Coastguard Worker /* Y need not be checked for POD because Y lies within */
1219*495ae853SAndroid Build Coastguard Worker /* the picture((0,0) mv for Y doesnot get changed. But (0,0) for */
1220*495ae853SAndroid Build Coastguard Worker /* U and V can need POD beacause of cross-field mv adjustments */
1221*495ae853SAndroid Build Coastguard Worker /* (Table 8-9 of standard) */
1222*495ae853SAndroid Build Coastguard Worker /******************************************************************/
1223*495ae853SAndroid Build Coastguard Worker if((ps_pred + 1)->i1_pod_ht)
1224*495ae853SAndroid Build Coastguard Worker {
1225*495ae853SAndroid Build Coastguard Worker pu1_pred = (ps_pred + 1)->pu1_pred_u;
1226*495ae853SAndroid Build Coastguard Worker pu1_dma_dst = (ps_pred + 1)->pu1_dma_dest_addr;
1227*495ae853SAndroid Build Coastguard Worker u1_dma_ht = (ps_pred + 1)->u1_dma_ht_uv;
1228*495ae853SAndroid Build Coastguard Worker u1_dma_wd = (ps_pred + 1)->u1_dma_wd_uv
1229*495ae853SAndroid Build Coastguard Worker * YUV420SP_FACTOR;
1230*495ae853SAndroid Build Coastguard Worker u2_frm_wd = ps_dec->u2_frm_wd_uv << u1_mb_or_pic_fld;
1231*495ae853SAndroid Build Coastguard Worker if((ps_pred + 1)->i1_pod_ht < 0)
1232*495ae853SAndroid Build Coastguard Worker {
1233*495ae853SAndroid Build Coastguard Worker /*Top POD*/
1234*495ae853SAndroid Build Coastguard Worker pu1_dma_dst -= ((ps_pred + 1)->i1_pod_ht
1235*495ae853SAndroid Build Coastguard Worker * (ps_pred + 1)->u2_u1_ref_buf_wd
1236*495ae853SAndroid Build Coastguard Worker * YUV420SP_FACTOR);
1237*495ae853SAndroid Build Coastguard Worker }
1238*495ae853SAndroid Build Coastguard Worker ih264d_copy_2d1d(pu1_pred, pu1_dma_dst, u2_frm_wd,
1239*495ae853SAndroid Build Coastguard Worker u1_dma_wd, u1_dma_ht);
1240*495ae853SAndroid Build Coastguard Worker pu1_dma_dst += ((ps_pred + 1)->i1_dma_ht
1241*495ae853SAndroid Build Coastguard Worker * (ps_pred + 1)->u2_u1_ref_buf_wd); //(u1_dma_ht * u1_dma_wd);//
1242*495ae853SAndroid Build Coastguard Worker pu1_pred = (ps_pred + 1)->pu1_pred_v;
1243*495ae853SAndroid Build Coastguard Worker ih264d_pad_on_demand(ps_pred + 1, CHROM_BLK);
1244*495ae853SAndroid Build Coastguard Worker
1245*495ae853SAndroid Build Coastguard Worker }
1246*495ae853SAndroid Build Coastguard Worker
1247*495ae853SAndroid Build Coastguard Worker ih264d_multiplex_ref_data(ps_dec, ps_pred, pu1_dest_y,
1248*495ae853SAndroid Build Coastguard Worker pu1_dest_u, ps_cur_mb_info,
1249*495ae853SAndroid Build Coastguard Worker u2_dest_wd_y, u2_dest_wd_uv,
1250*495ae853SAndroid Build Coastguard Worker u1_dir);
1251*495ae853SAndroid Build Coastguard Worker ps_pred += 2;
1252*495ae853SAndroid Build Coastguard Worker }
1253*495ae853SAndroid Build Coastguard Worker }
1254*495ae853SAndroid Build Coastguard Worker }
1255*495ae853SAndroid Build Coastguard Worker if(u1_dir != 0)
1256*495ae853SAndroid Build Coastguard Worker u2_ref_wd_y = MB_SIZE;
1257*495ae853SAndroid Build Coastguard Worker
1258*495ae853SAndroid Build Coastguard Worker u2_num_pels += u4_wd_y * u4_ht_y;
1259*495ae853SAndroid Build Coastguard Worker /* if BI_DIRECT, average the two pred's, and put in ..PredBuffer[0] */
1260*495ae853SAndroid Build Coastguard Worker if((u1_is_bi_direct != 0) || (ps_pred_y_forw->u1_wght_pred_type != 0))
1261*495ae853SAndroid Build Coastguard Worker {
1262*495ae853SAndroid Build Coastguard Worker
1263*495ae853SAndroid Build Coastguard Worker switch(ps_pred_y_forw->u1_wght_pred_type)
1264*495ae853SAndroid Build Coastguard Worker {
1265*495ae853SAndroid Build Coastguard Worker case 0:
1266*495ae853SAndroid Build Coastguard Worker ps_dec->pf_default_weighted_pred_luma(
1267*495ae853SAndroid Build Coastguard Worker ps_pred_y_forw->pu1_rec_y_u, pu1_dest_y,
1268*495ae853SAndroid Build Coastguard Worker ps_pred_y_forw->pu1_rec_y_u,
1269*495ae853SAndroid Build Coastguard Worker u2_row_buf_wd_y, u2_ref_wd_y,
1270*495ae853SAndroid Build Coastguard Worker u2_row_buf_wd_y, u4_ht_uv * 2,
1271*495ae853SAndroid Build Coastguard Worker u4_wd_uv * 2);
1272*495ae853SAndroid Build Coastguard Worker
1273*495ae853SAndroid Build Coastguard Worker ps_dec->pf_default_weighted_pred_chroma(
1274*495ae853SAndroid Build Coastguard Worker ps_pred_cr_forw->pu1_rec_y_u, pu1_dest_u,
1275*495ae853SAndroid Build Coastguard Worker ps_pred_cr_forw->pu1_rec_y_u,
1276*495ae853SAndroid Build Coastguard Worker u2_row_buf_wd_uv, u2_dst_wd,
1277*495ae853SAndroid Build Coastguard Worker u2_row_buf_wd_uv, u4_ht_uv,
1278*495ae853SAndroid Build Coastguard Worker u4_wd_uv);
1279*495ae853SAndroid Build Coastguard Worker
1280*495ae853SAndroid Build Coastguard Worker break;
1281*495ae853SAndroid Build Coastguard Worker case 1:
1282*495ae853SAndroid Build Coastguard Worker {
1283*495ae853SAndroid Build Coastguard Worker UWORD32 *pu4_weight_ofst =
1284*495ae853SAndroid Build Coastguard Worker (UWORD32*)ps_pred_y_forw->u1_pi1_wt_ofst_rec_v;
1285*495ae853SAndroid Build Coastguard Worker UWORD32 u4_wt_ofst_u, u4_wt_ofst_v;
1286*495ae853SAndroid Build Coastguard Worker UWORD32 u4_wt_ofst_y =
1287*495ae853SAndroid Build Coastguard Worker (UWORD32)(pu4_weight_ofst[0]);
1288*495ae853SAndroid Build Coastguard Worker WORD32 weight = (WORD16)(u4_wt_ofst_y & 0xffff);
1289*495ae853SAndroid Build Coastguard Worker WORD32 ofst = (WORD8)(u4_wt_ofst_y >> 16);
1290*495ae853SAndroid Build Coastguard Worker
1291*495ae853SAndroid Build Coastguard Worker ps_dec->pf_weighted_pred_luma(ps_pred_y_forw->pu1_rec_y_u,
1292*495ae853SAndroid Build Coastguard Worker ps_pred_y_forw->pu1_rec_y_u,
1293*495ae853SAndroid Build Coastguard Worker u2_row_buf_wd_y,
1294*495ae853SAndroid Build Coastguard Worker u2_row_buf_wd_y,
1295*495ae853SAndroid Build Coastguard Worker (u2_log2Y_crwd & 0x0ff),
1296*495ae853SAndroid Build Coastguard Worker weight, ofst, u4_ht_y,
1297*495ae853SAndroid Build Coastguard Worker u4_wd_y);
1298*495ae853SAndroid Build Coastguard Worker
1299*495ae853SAndroid Build Coastguard Worker u4_wt_ofst_u = (UWORD32)(pu4_weight_ofst[2]);
1300*495ae853SAndroid Build Coastguard Worker u4_wt_ofst_v = (UWORD32)(pu4_weight_ofst[4]);
1301*495ae853SAndroid Build Coastguard Worker weight = ((u4_wt_ofst_v & 0xffff) << 16)
1302*495ae853SAndroid Build Coastguard Worker | (u4_wt_ofst_u & 0xffff);
1303*495ae853SAndroid Build Coastguard Worker ofst = ((u4_wt_ofst_v >> 16) << 8)
1304*495ae853SAndroid Build Coastguard Worker | ((u4_wt_ofst_u >> 16) & 0xFF);
1305*495ae853SAndroid Build Coastguard Worker
1306*495ae853SAndroid Build Coastguard Worker ps_dec->pf_weighted_pred_chroma(
1307*495ae853SAndroid Build Coastguard Worker ps_pred_cr_forw->pu1_rec_y_u,
1308*495ae853SAndroid Build Coastguard Worker ps_pred_cr_forw->pu1_rec_y_u,
1309*495ae853SAndroid Build Coastguard Worker u2_row_buf_wd_uv, u2_row_buf_wd_uv,
1310*495ae853SAndroid Build Coastguard Worker (u2_log2Y_crwd >> 8), weight, ofst,
1311*495ae853SAndroid Build Coastguard Worker u4_ht_y >> 1, u4_wd_y >> 1);
1312*495ae853SAndroid Build Coastguard Worker }
1313*495ae853SAndroid Build Coastguard Worker
1314*495ae853SAndroid Build Coastguard Worker break;
1315*495ae853SAndroid Build Coastguard Worker case 2:
1316*495ae853SAndroid Build Coastguard Worker {
1317*495ae853SAndroid Build Coastguard Worker UWORD32 *pu4_weight_ofst =
1318*495ae853SAndroid Build Coastguard Worker (UWORD32*)ps_pred_y_forw->u1_pi1_wt_ofst_rec_v;
1319*495ae853SAndroid Build Coastguard Worker UWORD32 u4_wt_ofst_u, u4_wt_ofst_v;
1320*495ae853SAndroid Build Coastguard Worker UWORD32 u4_wt_ofst_y;
1321*495ae853SAndroid Build Coastguard Worker WORD32 weight1, weight2;
1322*495ae853SAndroid Build Coastguard Worker WORD32 ofst1, ofst2;
1323*495ae853SAndroid Build Coastguard Worker
1324*495ae853SAndroid Build Coastguard Worker u4_wt_ofst_y = (UWORD32)(pu4_weight_ofst[0]);
1325*495ae853SAndroid Build Coastguard Worker
1326*495ae853SAndroid Build Coastguard Worker weight1 = (WORD16)(u4_wt_ofst_y & 0xffff);
1327*495ae853SAndroid Build Coastguard Worker ofst1 = (WORD8)(u4_wt_ofst_y >> 16);
1328*495ae853SAndroid Build Coastguard Worker
1329*495ae853SAndroid Build Coastguard Worker u4_wt_ofst_y = (UWORD32)(pu4_weight_ofst[1]);
1330*495ae853SAndroid Build Coastguard Worker weight2 = (WORD16)(u4_wt_ofst_y & 0xffff);
1331*495ae853SAndroid Build Coastguard Worker ofst2 = (WORD8)(u4_wt_ofst_y >> 16);
1332*495ae853SAndroid Build Coastguard Worker
1333*495ae853SAndroid Build Coastguard Worker ps_dec->pf_weighted_bi_pred_luma(ps_pred_y_forw->pu1_rec_y_u,
1334*495ae853SAndroid Build Coastguard Worker ps_pred_y_back->pu1_rec_y_u,
1335*495ae853SAndroid Build Coastguard Worker ps_pred_y_forw->pu1_rec_y_u,
1336*495ae853SAndroid Build Coastguard Worker u2_row_buf_wd_y,
1337*495ae853SAndroid Build Coastguard Worker u2_ref_wd_y,
1338*495ae853SAndroid Build Coastguard Worker u2_row_buf_wd_y,
1339*495ae853SAndroid Build Coastguard Worker (u2_log2Y_crwd & 0x0ff),
1340*495ae853SAndroid Build Coastguard Worker weight1, weight2, ofst1,
1341*495ae853SAndroid Build Coastguard Worker ofst2, u4_ht_y,
1342*495ae853SAndroid Build Coastguard Worker u4_wd_y);
1343*495ae853SAndroid Build Coastguard Worker
1344*495ae853SAndroid Build Coastguard Worker u4_wt_ofst_u = (UWORD32)(pu4_weight_ofst[2]);
1345*495ae853SAndroid Build Coastguard Worker u4_wt_ofst_v = (UWORD32)(pu4_weight_ofst[4]);
1346*495ae853SAndroid Build Coastguard Worker weight1 = ((u4_wt_ofst_v & 0xffff) << 16)
1347*495ae853SAndroid Build Coastguard Worker | (u4_wt_ofst_u & 0xffff);
1348*495ae853SAndroid Build Coastguard Worker ofst1 = ((u4_wt_ofst_v >> 16) << 8)
1349*495ae853SAndroid Build Coastguard Worker | ((u4_wt_ofst_u >> 16) & 0xFF);
1350*495ae853SAndroid Build Coastguard Worker
1351*495ae853SAndroid Build Coastguard Worker u4_wt_ofst_u = (UWORD32)(pu4_weight_ofst[3]);
1352*495ae853SAndroid Build Coastguard Worker u4_wt_ofst_v = (UWORD32)(pu4_weight_ofst[5]);
1353*495ae853SAndroid Build Coastguard Worker weight2 = ((u4_wt_ofst_v & 0xffff) << 16)
1354*495ae853SAndroid Build Coastguard Worker | (u4_wt_ofst_u & 0xffff);
1355*495ae853SAndroid Build Coastguard Worker ofst2 = ((u4_wt_ofst_v >> 16) << 8)
1356*495ae853SAndroid Build Coastguard Worker | ((u4_wt_ofst_u >> 16) & 0xFF);
1357*495ae853SAndroid Build Coastguard Worker
1358*495ae853SAndroid Build Coastguard Worker ps_dec->pf_weighted_bi_pred_chroma(
1359*495ae853SAndroid Build Coastguard Worker (ps_pred_y_forw + 1)->pu1_rec_y_u,
1360*495ae853SAndroid Build Coastguard Worker (ps_pred_y_back + 1)->pu1_rec_y_u,
1361*495ae853SAndroid Build Coastguard Worker (ps_pred_y_forw + 1)->pu1_rec_y_u,
1362*495ae853SAndroid Build Coastguard Worker u2_row_buf_wd_uv, u2_dst_wd,
1363*495ae853SAndroid Build Coastguard Worker u2_row_buf_wd_uv, (u2_log2Y_crwd >> 8),
1364*495ae853SAndroid Build Coastguard Worker weight1, weight2, ofst1, ofst2,
1365*495ae853SAndroid Build Coastguard Worker u4_ht_y >> 1, u4_wd_y >> 1);
1366*495ae853SAndroid Build Coastguard Worker }
1367*495ae853SAndroid Build Coastguard Worker
1368*495ae853SAndroid Build Coastguard Worker break;
1369*495ae853SAndroid Build Coastguard Worker }
1370*495ae853SAndroid Build Coastguard Worker
1371*495ae853SAndroid Build Coastguard Worker }
1372*495ae853SAndroid Build Coastguard Worker }
1373*495ae853SAndroid Build Coastguard Worker }
1374*495ae853SAndroid Build Coastguard Worker
1375*495ae853SAndroid Build Coastguard Worker
1376*495ae853SAndroid Build Coastguard Worker /*!
1377*495ae853SAndroid Build Coastguard Worker **************************************************************************
1378*495ae853SAndroid Build Coastguard Worker * \if Function name : ih264d_multiplex_ref_data \endif
1379*495ae853SAndroid Build Coastguard Worker *
1380*495ae853SAndroid Build Coastguard Worker * \brief
1381*495ae853SAndroid Build Coastguard Worker * Initializes forward and backward refernce lists for B slice decoding.
1382*495ae853SAndroid Build Coastguard Worker *
1383*495ae853SAndroid Build Coastguard Worker *
1384*495ae853SAndroid Build Coastguard Worker * \return
1385*495ae853SAndroid Build Coastguard Worker * 0 on Success and Error code otherwise
1386*495ae853SAndroid Build Coastguard Worker **************************************************************************
1387*495ae853SAndroid Build Coastguard Worker */
1388*495ae853SAndroid Build Coastguard Worker
ih264d_multiplex_ref_data(dec_struct_t * ps_dec,pred_info_t * ps_pred,UWORD8 * pu1_dest_y,UWORD8 * pu1_dest_u,dec_mb_info_t * ps_cur_mb_info,UWORD16 u2_dest_wd_y,UWORD16 u2_dest_wd_uv,UWORD8 u1_dir)1389*495ae853SAndroid Build Coastguard Worker void ih264d_multiplex_ref_data(dec_struct_t * ps_dec,
1390*495ae853SAndroid Build Coastguard Worker pred_info_t *ps_pred,
1391*495ae853SAndroid Build Coastguard Worker UWORD8* pu1_dest_y,
1392*495ae853SAndroid Build Coastguard Worker UWORD8* pu1_dest_u,
1393*495ae853SAndroid Build Coastguard Worker dec_mb_info_t *ps_cur_mb_info,
1394*495ae853SAndroid Build Coastguard Worker UWORD16 u2_dest_wd_y,
1395*495ae853SAndroid Build Coastguard Worker UWORD16 u2_dest_wd_uv,
1396*495ae853SAndroid Build Coastguard Worker UWORD8 u1_dir)
1397*495ae853SAndroid Build Coastguard Worker {
1398*495ae853SAndroid Build Coastguard Worker UWORD16 u2_mask = ps_cur_mb_info->u2_mask[u1_dir];
1399*495ae853SAndroid Build Coastguard Worker UWORD8 *pu1_ref_y, *pu1_ref_u;
1400*495ae853SAndroid Build Coastguard Worker UWORD8 uc_cond, i, j, u1_dydx;
1401*495ae853SAndroid Build Coastguard Worker UWORD16 u2_ref_wd_y, u2_ref_wd_uv;
1402*495ae853SAndroid Build Coastguard Worker
1403*495ae853SAndroid Build Coastguard Worker PROFILE_DISABLE_INTER_PRED()
1404*495ae853SAndroid Build Coastguard Worker
1405*495ae853SAndroid Build Coastguard Worker if(ps_pred->i1_pod_ht)
1406*495ae853SAndroid Build Coastguard Worker {
1407*495ae853SAndroid Build Coastguard Worker pu1_ref_y = ps_pred->pu1_dma_dest_addr;
1408*495ae853SAndroid Build Coastguard Worker
1409*495ae853SAndroid Build Coastguard Worker u2_ref_wd_y = ps_pred->u2_u1_ref_buf_wd;
1410*495ae853SAndroid Build Coastguard Worker }
1411*495ae853SAndroid Build Coastguard Worker else
1412*495ae853SAndroid Build Coastguard Worker {
1413*495ae853SAndroid Build Coastguard Worker pu1_ref_y = ps_pred->pu1_y_ref;
1414*495ae853SAndroid Build Coastguard Worker u2_ref_wd_y = ps_pred->u2_frm_wd;
1415*495ae853SAndroid Build Coastguard Worker }
1416*495ae853SAndroid Build Coastguard Worker
1417*495ae853SAndroid Build Coastguard Worker ps_pred++;
1418*495ae853SAndroid Build Coastguard Worker if(ps_pred->i1_pod_ht)
1419*495ae853SAndroid Build Coastguard Worker {
1420*495ae853SAndroid Build Coastguard Worker pu1_ref_u = ps_pred->pu1_dma_dest_addr;
1421*495ae853SAndroid Build Coastguard Worker u2_ref_wd_uv = ps_pred->u2_u1_ref_buf_wd * YUV420SP_FACTOR;
1422*495ae853SAndroid Build Coastguard Worker
1423*495ae853SAndroid Build Coastguard Worker }
1424*495ae853SAndroid Build Coastguard Worker else
1425*495ae853SAndroid Build Coastguard Worker {
1426*495ae853SAndroid Build Coastguard Worker pu1_ref_u = ps_pred->pu1_u_ref;
1427*495ae853SAndroid Build Coastguard Worker u2_ref_wd_uv = ps_pred->u2_frm_wd;
1428*495ae853SAndroid Build Coastguard Worker
1429*495ae853SAndroid Build Coastguard Worker }
1430*495ae853SAndroid Build Coastguard Worker
1431*495ae853SAndroid Build Coastguard Worker u1_dydx = ps_pred->u1_dydx;
1432*495ae853SAndroid Build Coastguard Worker
1433*495ae853SAndroid Build Coastguard Worker {
1434*495ae853SAndroid Build Coastguard Worker UWORD8 uc_dx, uc_dy;
1435*495ae853SAndroid Build Coastguard Worker UWORD8 *pu1_scratch_u;
1436*495ae853SAndroid Build Coastguard Worker
1437*495ae853SAndroid Build Coastguard Worker uc_dx = u1_dydx & 0x3;
1438*495ae853SAndroid Build Coastguard Worker uc_dy = u1_dydx >> 3;
1439*495ae853SAndroid Build Coastguard Worker if(u1_dydx != 0)
1440*495ae853SAndroid Build Coastguard Worker {
1441*495ae853SAndroid Build Coastguard Worker pred_info_t * ps_prv_pred = ps_pred - 2;
1442*495ae853SAndroid Build Coastguard Worker pu1_scratch_u = ps_prv_pred->pu1_dma_dest_addr;
1443*495ae853SAndroid Build Coastguard Worker ps_dec->pf_inter_pred_chroma(pu1_ref_u, pu1_scratch_u,
1444*495ae853SAndroid Build Coastguard Worker u2_ref_wd_uv, 16, uc_dx, uc_dy, 8,
1445*495ae853SAndroid Build Coastguard Worker 8);
1446*495ae853SAndroid Build Coastguard Worker
1447*495ae853SAndroid Build Coastguard Worker /* Modify ref pointer and refWidth to point to scratch */
1448*495ae853SAndroid Build Coastguard Worker /* buffer to be used below in ih264d_copy_multiplex_data functions */
1449*495ae853SAndroid Build Coastguard Worker /* CHANGED CODE */
1450*495ae853SAndroid Build Coastguard Worker pu1_ref_u = pu1_scratch_u;
1451*495ae853SAndroid Build Coastguard Worker u2_ref_wd_uv = 8 * YUV420SP_FACTOR;
1452*495ae853SAndroid Build Coastguard Worker }
1453*495ae853SAndroid Build Coastguard Worker }
1454*495ae853SAndroid Build Coastguard Worker {
1455*495ae853SAndroid Build Coastguard Worker for(i = 0; i < 4; i++)
1456*495ae853SAndroid Build Coastguard Worker {
1457*495ae853SAndroid Build Coastguard Worker for(j = 0; j < 4; j++)
1458*495ae853SAndroid Build Coastguard Worker {
1459*495ae853SAndroid Build Coastguard Worker uc_cond = u2_mask & 1;
1460*495ae853SAndroid Build Coastguard Worker u2_mask >>= 1;
1461*495ae853SAndroid Build Coastguard Worker if(uc_cond)
1462*495ae853SAndroid Build Coastguard Worker {
1463*495ae853SAndroid Build Coastguard Worker *(UWORD32 *)(pu1_dest_y + u2_dest_wd_y) =
1464*495ae853SAndroid Build Coastguard Worker *(UWORD32 *)(pu1_ref_y + u2_ref_wd_y);
1465*495ae853SAndroid Build Coastguard Worker *(UWORD32 *)(pu1_dest_y + 2 * u2_dest_wd_y) =
1466*495ae853SAndroid Build Coastguard Worker *(UWORD32 *)(pu1_ref_y + 2 * u2_ref_wd_y);
1467*495ae853SAndroid Build Coastguard Worker *(UWORD32 *)(pu1_dest_y + 3 * u2_dest_wd_y) =
1468*495ae853SAndroid Build Coastguard Worker *(UWORD32 *)(pu1_ref_y + 3 * u2_ref_wd_y);
1469*495ae853SAndroid Build Coastguard Worker {
1470*495ae853SAndroid Build Coastguard Worker UWORD32 *dst, *src;
1471*495ae853SAndroid Build Coastguard Worker dst = (UWORD32 *)pu1_dest_y;
1472*495ae853SAndroid Build Coastguard Worker src = (UWORD32 *)pu1_ref_y;
1473*495ae853SAndroid Build Coastguard Worker *dst = *src;
1474*495ae853SAndroid Build Coastguard Worker dst++;
1475*495ae853SAndroid Build Coastguard Worker src++;
1476*495ae853SAndroid Build Coastguard Worker pu1_dest_y = (UWORD8 *)dst;
1477*495ae853SAndroid Build Coastguard Worker pu1_ref_y = (UWORD8 *)src;
1478*495ae853SAndroid Build Coastguard Worker }
1479*495ae853SAndroid Build Coastguard Worker *(UWORD32 *)(pu1_dest_u + u2_dest_wd_uv) =
1480*495ae853SAndroid Build Coastguard Worker *(UWORD32 *)(pu1_ref_u + u2_ref_wd_uv);
1481*495ae853SAndroid Build Coastguard Worker {
1482*495ae853SAndroid Build Coastguard Worker UWORD32 *dst, *src;
1483*495ae853SAndroid Build Coastguard Worker dst = (UWORD32 *)pu1_dest_u;
1484*495ae853SAndroid Build Coastguard Worker src = (UWORD32 *)pu1_ref_u;
1485*495ae853SAndroid Build Coastguard Worker *dst = *src;
1486*495ae853SAndroid Build Coastguard Worker dst++;
1487*495ae853SAndroid Build Coastguard Worker src++;
1488*495ae853SAndroid Build Coastguard Worker pu1_dest_u = (UWORD8 *)dst;
1489*495ae853SAndroid Build Coastguard Worker pu1_ref_u = (UWORD8 *)src;
1490*495ae853SAndroid Build Coastguard Worker }
1491*495ae853SAndroid Build Coastguard Worker
1492*495ae853SAndroid Build Coastguard Worker }
1493*495ae853SAndroid Build Coastguard Worker else
1494*495ae853SAndroid Build Coastguard Worker {
1495*495ae853SAndroid Build Coastguard Worker pu1_dest_y += 4;
1496*495ae853SAndroid Build Coastguard Worker pu1_ref_y += 4;
1497*495ae853SAndroid Build Coastguard Worker pu1_dest_u += 2 * YUV420SP_FACTOR;
1498*495ae853SAndroid Build Coastguard Worker pu1_ref_u += 2 * YUV420SP_FACTOR;
1499*495ae853SAndroid Build Coastguard Worker }
1500*495ae853SAndroid Build Coastguard Worker }
1501*495ae853SAndroid Build Coastguard Worker pu1_ref_y += 4 * (u2_ref_wd_y - 4);
1502*495ae853SAndroid Build Coastguard Worker pu1_ref_u += 2 * (u2_ref_wd_uv - 4 * YUV420SP_FACTOR);
1503*495ae853SAndroid Build Coastguard Worker pu1_dest_y += 4 * (u2_dest_wd_y - 4);
1504*495ae853SAndroid Build Coastguard Worker pu1_dest_u += 2 * (u2_dest_wd_uv - 4 * YUV420SP_FACTOR);
1505*495ae853SAndroid Build Coastguard Worker }
1506*495ae853SAndroid Build Coastguard Worker }
1507*495ae853SAndroid Build Coastguard Worker }
1508*495ae853SAndroid Build Coastguard Worker
ih264d_pad_on_demand(pred_info_t * ps_pred,UWORD8 lum_chrom_blk)1509*495ae853SAndroid Build Coastguard Worker void ih264d_pad_on_demand(pred_info_t *ps_pred, UWORD8 lum_chrom_blk)
1510*495ae853SAndroid Build Coastguard Worker {
1511*495ae853SAndroid Build Coastguard Worker if(CHROM_BLK == lum_chrom_blk)
1512*495ae853SAndroid Build Coastguard Worker {
1513*495ae853SAndroid Build Coastguard Worker UWORD32 *pu4_pod_src_u, *pu4_pod_dst_u;
1514*495ae853SAndroid Build Coastguard Worker UWORD32 *pu4_pod_src_v, *pu4_pod_dst_v;
1515*495ae853SAndroid Build Coastguard Worker WORD32 j, u1_wd_stride;
1516*495ae853SAndroid Build Coastguard Worker WORD32 i, u1_dma_ht, i1_ht;
1517*495ae853SAndroid Build Coastguard Worker UWORD32 u2_dma_size;
1518*495ae853SAndroid Build Coastguard Worker u1_wd_stride = (ps_pred->u2_u1_ref_buf_wd >> 2) * YUV420SP_FACTOR;
1519*495ae853SAndroid Build Coastguard Worker u1_dma_ht = ps_pred->i1_dma_ht;
1520*495ae853SAndroid Build Coastguard Worker u2_dma_size = u1_wd_stride * u1_dma_ht;
1521*495ae853SAndroid Build Coastguard Worker pu4_pod_src_u = (UWORD32 *)ps_pred->pu1_dma_dest_addr;
1522*495ae853SAndroid Build Coastguard Worker pu4_pod_dst_u = pu4_pod_src_u;
1523*495ae853SAndroid Build Coastguard Worker
1524*495ae853SAndroid Build Coastguard Worker pu4_pod_src_v = pu4_pod_src_u + u2_dma_size;
1525*495ae853SAndroid Build Coastguard Worker pu4_pod_dst_v = pu4_pod_src_v;
1526*495ae853SAndroid Build Coastguard Worker
1527*495ae853SAndroid Build Coastguard Worker i1_ht = ps_pred->i1_pod_ht;
1528*495ae853SAndroid Build Coastguard Worker pu4_pod_src_u -= u1_wd_stride * i1_ht;
1529*495ae853SAndroid Build Coastguard Worker pu4_pod_src_v -= u1_wd_stride * i1_ht;
1530*495ae853SAndroid Build Coastguard Worker if(i1_ht < 0)
1531*495ae853SAndroid Build Coastguard Worker /* Top POD */
1532*495ae853SAndroid Build Coastguard Worker i1_ht = -i1_ht;
1533*495ae853SAndroid Build Coastguard Worker else
1534*495ae853SAndroid Build Coastguard Worker {
1535*495ae853SAndroid Build Coastguard Worker /* Bottom POD */
1536*495ae853SAndroid Build Coastguard Worker pu4_pod_src_u += (u1_dma_ht - 1) * u1_wd_stride;
1537*495ae853SAndroid Build Coastguard Worker pu4_pod_dst_u += (u1_dma_ht - i1_ht) * u1_wd_stride;
1538*495ae853SAndroid Build Coastguard Worker pu4_pod_src_v += (u1_dma_ht - 1) * u1_wd_stride;
1539*495ae853SAndroid Build Coastguard Worker pu4_pod_dst_v += (u1_dma_ht - i1_ht) * u1_wd_stride;
1540*495ae853SAndroid Build Coastguard Worker }
1541*495ae853SAndroid Build Coastguard Worker
1542*495ae853SAndroid Build Coastguard Worker for(i = 0; i < i1_ht; i++)
1543*495ae853SAndroid Build Coastguard Worker for(j = 0; j < u1_wd_stride; j++)
1544*495ae853SAndroid Build Coastguard Worker {
1545*495ae853SAndroid Build Coastguard Worker *pu4_pod_dst_u++ = *(pu4_pod_src_u + j);
1546*495ae853SAndroid Build Coastguard Worker
1547*495ae853SAndroid Build Coastguard Worker }
1548*495ae853SAndroid Build Coastguard Worker }
1549*495ae853SAndroid Build Coastguard Worker else
1550*495ae853SAndroid Build Coastguard Worker {
1551*495ae853SAndroid Build Coastguard Worker UWORD32 *pu4_pod_src, *pu4_pod_dst;
1552*495ae853SAndroid Build Coastguard Worker WORD32 j, u1_wd_stride;
1553*495ae853SAndroid Build Coastguard Worker WORD32 i, i1_ht;
1554*495ae853SAndroid Build Coastguard Worker pu4_pod_src = (UWORD32 *)ps_pred->pu1_dma_dest_addr;
1555*495ae853SAndroid Build Coastguard Worker pu4_pod_dst = pu4_pod_src;
1556*495ae853SAndroid Build Coastguard Worker u1_wd_stride = ps_pred->u2_u1_ref_buf_wd >> 2;
1557*495ae853SAndroid Build Coastguard Worker i1_ht = ps_pred->i1_pod_ht;
1558*495ae853SAndroid Build Coastguard Worker pu4_pod_src -= u1_wd_stride * i1_ht;
1559*495ae853SAndroid Build Coastguard Worker if(i1_ht < 0)
1560*495ae853SAndroid Build Coastguard Worker /* Top POD */
1561*495ae853SAndroid Build Coastguard Worker i1_ht = -i1_ht;
1562*495ae853SAndroid Build Coastguard Worker else
1563*495ae853SAndroid Build Coastguard Worker {
1564*495ae853SAndroid Build Coastguard Worker /* Bottom POD */
1565*495ae853SAndroid Build Coastguard Worker pu4_pod_src += (ps_pred->i1_dma_ht - 1) * u1_wd_stride;
1566*495ae853SAndroid Build Coastguard Worker pu4_pod_dst += (ps_pred->i1_dma_ht - i1_ht) * u1_wd_stride;
1567*495ae853SAndroid Build Coastguard Worker }
1568*495ae853SAndroid Build Coastguard Worker
1569*495ae853SAndroid Build Coastguard Worker for(i = 0; i < i1_ht; i++)
1570*495ae853SAndroid Build Coastguard Worker for(j = 0; j < u1_wd_stride; j++)
1571*495ae853SAndroid Build Coastguard Worker *pu4_pod_dst++ = *(pu4_pod_src + j);
1572*495ae853SAndroid Build Coastguard Worker }
1573*495ae853SAndroid Build Coastguard Worker }
1574*495ae853SAndroid Build Coastguard Worker
1575