1 /******************************************************************************
2 *
3 * Copyright (C) 2015 The Android Open Source Project
4 *
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at:
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16 *
17 *****************************************************************************
18 * Originally developed and contributed by Ittiam Systems Pvt. Ltd, Bangalore
19 */
20 #include <stdio.h>
21 #include <string.h>
22
23 #include "iv_datatypedef.h"
24 #include "iv.h"
25
26 #include "impeg2_buf_mgr.h"
27 #include "impeg2_disp_mgr.h"
28 #include "impeg2_defs.h"
29 #include "impeg2_platform_macros.h"
30 #include "impeg2_inter_pred.h"
31 #include "impeg2_idct.h"
32 #include "impeg2_globals.h"
33 #include "impeg2_mem_func.h"
34 #include "impeg2_format_conv.h"
35 #include "impeg2_macros.h"
36
37 #include "ivd.h"
38 #include "impeg2d.h"
39 #include "impeg2d_bitstream.h"
40 #include "impeg2d_structs.h"
41 #include "impeg2d_globals.h"
42 #include "impeg2d_vld_tables.h"
43 #include "impeg2d_vld.h"
44 #include "impeg2d_pic_proc.h"
45 #include "impeg2d_debug.h"
46
47 void impeg2d_init_function_ptr(void *pv_codec);
impeg2d_format_convert(dec_state_t * ps_dec,pic_buf_t * ps_src_pic,iv_yuv_buf_t * ps_disp_frm_buf,UWORD32 u4_start_row,UWORD32 u4_num_rows)48 void impeg2d_format_convert(dec_state_t *ps_dec,
49 pic_buf_t *ps_src_pic,
50 iv_yuv_buf_t *ps_disp_frm_buf,
51 UWORD32 u4_start_row, UWORD32 u4_num_rows)
52 {
53 UWORD8 *pu1_src_y,*pu1_src_u,*pu1_src_v;
54 UWORD8 *pu1_dst_y,*pu1_dst_u,*pu1_dst_v;
55
56
57
58 if((NULL == ps_src_pic) || (NULL == ps_src_pic->pu1_y) || (0 == u4_num_rows))
59 return;
60
61 pu1_src_y = ps_src_pic->pu1_y + (u4_start_row * ps_dec->u2_frame_width);
62 pu1_src_u = ps_src_pic->pu1_u + ((u4_start_row >> 1) * (ps_dec->u2_frame_width >> 1));
63 pu1_src_v = ps_src_pic->pu1_v + ((u4_start_row >> 1) *(ps_dec->u2_frame_width >> 1));
64
65 pu1_dst_y = (UWORD8 *)ps_disp_frm_buf->pv_y_buf + (u4_start_row * ps_dec->u4_frm_buf_stride);
66 pu1_dst_u = (UWORD8 *)ps_disp_frm_buf->pv_u_buf +((u4_start_row >> 1)*(ps_dec->u4_frm_buf_stride >> 1));
67 pu1_dst_v = (UWORD8 *)ps_disp_frm_buf->pv_v_buf +((u4_start_row >> 1)*(ps_dec->u4_frm_buf_stride >> 1));
68
69 if (IV_YUV_420P == ps_dec->i4_chromaFormat)
70 {
71 ps_dec->pf_copy_yuv420p_buf(pu1_src_y, pu1_src_u, pu1_src_v, pu1_dst_y,
72 pu1_dst_u, pu1_dst_v,
73 ps_dec->u2_horizontal_size,
74 u4_num_rows,
75 ps_dec->u2_frame_width,
76 (ps_dec->u2_frame_width >> 1),
77 (ps_dec->u2_frame_width >> 1),
78 ps_dec->u4_frm_buf_stride,
79 (ps_dec->u4_frm_buf_stride >> 1),
80 (ps_dec->u4_frm_buf_stride >> 1));
81 }
82 else if (IV_YUV_422ILE == ps_dec->i4_chromaFormat)
83 {
84 void *pv_yuv422i;
85 UWORD32 u2_height,u2_width,u2_stride_y,u2_stride_u,u2_stride_v;
86 UWORD32 u2_stride_yuv422i;
87
88
89 pv_yuv422i = (UWORD8 *)ps_disp_frm_buf->pv_y_buf + ((ps_dec->u2_vertical_size)*(ps_dec->u4_frm_buf_stride));
90 u2_height = u4_num_rows;
91 u2_width = ps_dec->u2_horizontal_size;
92 u2_stride_y = ps_dec->u2_frame_width;
93 u2_stride_u = u2_stride_y >> 1;
94 u2_stride_v = u2_stride_u;
95 u2_stride_yuv422i = (0 == ps_dec->u4_frm_buf_stride) ? ps_dec->u2_horizontal_size : ps_dec->u4_frm_buf_stride;
96
97 ps_dec->pf_fmt_conv_yuv420p_to_yuv422ile(pu1_src_y,
98 pu1_src_u,
99 pu1_src_v,
100 pv_yuv422i,
101 u2_width,
102 u2_height,
103 u2_stride_y,
104 u2_stride_u,
105 u2_stride_v,
106 u2_stride_yuv422i);
107
108 }
109 else if((ps_dec->i4_chromaFormat == IV_YUV_420SP_UV) ||
110 (ps_dec->i4_chromaFormat == IV_YUV_420SP_VU))
111 {
112
113 UWORD32 dest_inc_Y=0,dest_inc_UV=0;
114 WORD32 convert_uv_only;
115
116 pu1_dst_u = (UWORD8 *)ps_disp_frm_buf->pv_u_buf +((u4_start_row >> 1)*(ps_dec->u4_frm_buf_stride));
117 dest_inc_Y = ps_dec->u4_frm_buf_stride;
118 dest_inc_UV = ((ps_dec->u4_frm_buf_stride + 1) >> 1) << 1;
119 convert_uv_only = 0;
120
121 if(1 == ps_dec->u4_share_disp_buf)
122 convert_uv_only = 1;
123
124 if(pu1_src_y == pu1_dst_y)
125 convert_uv_only = 1;
126
127 if(ps_dec->i4_chromaFormat == IV_YUV_420SP_UV)
128 {
129 ps_dec->pf_fmt_conv_yuv420p_to_yuv420sp_uv(pu1_src_y,
130 pu1_src_u,
131 pu1_src_v,
132 pu1_dst_y,
133 pu1_dst_u,
134 u4_num_rows,
135 ps_dec->u2_horizontal_size,
136 ps_dec->u2_frame_width,
137 ps_dec->u2_frame_width >> 1,
138 ps_dec->u2_frame_width >> 1,
139 dest_inc_Y,
140 dest_inc_UV,
141 convert_uv_only);
142 }
143 else
144 {
145 ps_dec->pf_fmt_conv_yuv420p_to_yuv420sp_vu(pu1_src_y,
146 pu1_src_u,
147 pu1_src_v,
148 pu1_dst_y,
149 pu1_dst_u,
150 u4_num_rows,
151 ps_dec->u2_horizontal_size,
152 ps_dec->u2_frame_width,
153 ps_dec->u2_frame_width >> 1,
154 ps_dec->u2_frame_width >> 1,
155 dest_inc_Y,
156 dest_inc_UV,
157 convert_uv_only);
158 }
159
160
161
162 }
163
164 }
165
166
167 /*******************************************************************************
168 *
169 * Function Name : impeg2d_get_frm_buf
170 *
171 * Description : Gets YUV component buffers for the frame
172 *
173 * Arguments :
174 * frm_buf : YUV buffer
175 * frm : Reference frame
176 * width : Width of the frame
177 * Height : Height of the frame
178 *
179 * Values Returned : None
180 *******************************************************************************/
impeg2d_get_frm_buf(yuv_buf_t * ps_frm_buf,UWORD8 * pu1_frm,UWORD32 u4_width,UWORD32 u4_height)181 void impeg2d_get_frm_buf(yuv_buf_t *ps_frm_buf,UWORD8 *pu1_frm,UWORD32 u4_width,UWORD32 u4_height)
182 {
183 UWORD32 u4_luma_size = u4_width * u4_height;
184 UWORD32 u4_chroma_size = (u4_width * u4_height)>>2;
185
186 ps_frm_buf->pu1_y = pu1_frm;
187 ps_frm_buf->pu1_u = pu1_frm + u4_luma_size;
188 ps_frm_buf->pu1_v = pu1_frm + u4_luma_size + u4_chroma_size;
189
190 }
191 /*******************************************************************************
192 *
193 * Function Name : impeg2d_get_bottom_field_buf
194 *
195 * Description : Gets YUV component buffers for bottom field of the frame
196 *
197 * Arguments :
198 * frm_buf : YUV buffer
199 * frm : Reference frame
200 * width : Width of the frame
201 * Height : Height of the frame
202 *
203 * Values Returned : None
204 *******************************************************************************/
impeg2d_get_bottom_field_buf(yuv_buf_t * ps_src_buf,yuv_buf_t * ps_dst_buf,UWORD32 u4_width)205 void impeg2d_get_bottom_field_buf(yuv_buf_t *ps_src_buf,yuv_buf_t *ps_dst_buf,
206 UWORD32 u4_width)
207 {
208 ps_dst_buf->pu1_y = ps_src_buf->pu1_y + u4_width;
209 ps_dst_buf->pu1_u = ps_src_buf->pu1_u + (u4_width>>1);
210 ps_dst_buf->pu1_v = ps_src_buf->pu1_v + (u4_width>>1);
211
212 }
213 /*******************************************************************************
214 * Function Name : impeg2d_get_mb_addr_incr
215 *
216 * Description : Decodes the Macroblock address increment
217 *
218 * Arguments :
219 * stream : Bitstream
220 *
221 * Values Returned : Macroblock address increment
222 *******************************************************************************/
impeg2d_get_mb_addr_incr(stream_t * ps_stream)223 UWORD16 impeg2d_get_mb_addr_incr(stream_t *ps_stream)
224 {
225 UWORD16 u2_mb_addr_incr = 0;
226 while (impeg2d_bit_stream_nxt(ps_stream,MB_ESCAPE_CODE_LEN) == MB_ESCAPE_CODE &&
227 ps_stream->u4_offset < ps_stream->u4_max_offset)
228 {
229 impeg2d_bit_stream_flush(ps_stream,MB_ESCAPE_CODE_LEN);
230 u2_mb_addr_incr += 33;
231 }
232 u2_mb_addr_incr += impeg2d_dec_vld_symbol(ps_stream,gai2_impeg2d_mb_addr_incr,MB_ADDR_INCR_LEN) +
233 MB_ADDR_INCR_OFFSET;
234 return(u2_mb_addr_incr);
235 }
236
237 /*******************************************************************************
238 *
239 * Function Name : impeg2d_init_video_state
240 *
241 * Description : Initializes the Video decoder state
242 *
243 * Arguments :
244 * dec : Decoder context
245 * videoType : MPEG_2_Video / MPEG_1_Video
246 *
247 * Values Returned : None
248 *******************************************************************************/
impeg2d_init_video_state(dec_state_t * ps_dec,e_video_type_t e_video_type)249 IMPEG2D_ERROR_CODES_T impeg2d_init_video_state(dec_state_t *ps_dec, e_video_type_t e_video_type)
250 {
251 /*-----------------------------------------------------------------------*/
252 /* Bit Stream that conforms to MPEG-1 <ISO/IEC 11172-2> standard */
253 /*-----------------------------------------------------------------------*/
254 if(e_video_type == MPEG_1_VIDEO)
255 {
256 ps_dec->u2_is_mpeg2 = 0;
257
258 /*-------------------------------------------------------------------*/
259 /* force MPEG-1 parameters for proper decoder behavior */
260 /* see ISO/IEC 13818-2 section D.9.14 */
261 /*-------------------------------------------------------------------*/
262 ps_dec->u2_progressive_sequence = 1;
263 ps_dec->u2_intra_dc_precision = 0;
264 ps_dec->u2_picture_structure = FRAME_PICTURE;
265 ps_dec->u2_frame_pred_frame_dct = 1;
266 ps_dec->u2_concealment_motion_vectors = 0;
267 ps_dec->u2_q_scale_type = 0;
268 ps_dec->u2_intra_vlc_format = 0;
269 ps_dec->u2_alternate_scan = 0;
270 ps_dec->u2_repeat_first_field = 0;
271 ps_dec->u2_progressive_frame = 1;
272 ps_dec->u2_frame_rate_extension_n = 0;
273 ps_dec->u2_frame_rate_extension_d = 0;
274
275 ps_dec->pf_vld_inv_quant = impeg2d_vld_inv_quant_mpeg1;
276 /*-------------------------------------------------------------------*/
277 /* Setting of parameters other than those mentioned in MPEG2 standard*/
278 /* but used in decoding process. */
279 /*-------------------------------------------------------------------*/
280 }
281 /*-----------------------------------------------------------------------*/
282 /* Bit Stream that conforms to MPEG-2 */
283 /*-----------------------------------------------------------------------*/
284 else
285 {
286 ps_dec->u2_is_mpeg2 = 1;
287 ps_dec->u2_full_pel_forw_vector = 0;
288 ps_dec->u2_forw_f_code = 7;
289 ps_dec->u2_full_pel_back_vector = 0;
290 ps_dec->u2_back_f_code = 7;
291 ps_dec->pf_vld_inv_quant = impeg2d_vld_inv_quant_mpeg2;
292
293
294 }
295
296
297 impeg2d_init_function_ptr(ps_dec);
298
299 /* Set the frame Width and frame Height */
300 ps_dec->u2_frame_height = ALIGN16(ps_dec->u2_vertical_size);
301 ps_dec->u2_frame_width = ALIGN16(ps_dec->u2_horizontal_size);
302 ps_dec->u2_num_horiz_mb = (ps_dec->u2_horizontal_size + 15) >> 4;
303 // dec->u4_frm_buf_stride = dec->frameWidth;
304 if (ps_dec->u2_frame_height > ps_dec->u2_create_max_height || ps_dec->u2_frame_width > ps_dec->u2_create_max_width)
305 {
306 return IMPEG2D_PIC_SIZE_NOT_SUPPORTED;
307 }
308
309 ps_dec->u2_num_flds_decoded = 0;
310
311 /* Calculate the frame period */
312 {
313 UWORD32 numer;
314 UWORD32 denom;
315 numer = (UWORD32)gau2_impeg2_frm_rate_code[ps_dec->u2_frame_rate_code][1] *
316 (UWORD32)(ps_dec->u2_frame_rate_extension_d + 1);
317
318 denom = (UWORD32)gau2_impeg2_frm_rate_code[ps_dec->u2_frame_rate_code][0] *
319 (UWORD32)(ps_dec->u2_frame_rate_extension_n + 1);
320 ps_dec->u2_framePeriod = (numer * 1000 * 100) / denom;
321 }
322
323
324 if(VERTICAL_SCAN == ps_dec->u2_alternate_scan)
325 {
326 ps_dec->pu1_inv_scan_matrix = (UWORD8 *)gau1_impeg2_inv_scan_vertical;
327 }
328 else
329 {
330 ps_dec->pu1_inv_scan_matrix = (UWORD8 *)gau1_impeg2_inv_scan_zig_zag;
331 }
332 return (IMPEG2D_ERROR_CODES_T)IVD_ERROR_NONE;
333 }
334 /*******************************************************************************
335 *
336 * Function Name : impeg2d_pre_pic_dec_proc
337 *
338 * Description : Does the processing neccessary before picture decoding
339 *
340 * Arguments :
341 * dec : Decoder context
342 *
343 * Values Returned : None
344 *******************************************************************************/
impeg2d_pre_pic_dec_proc(dec_state_t * ps_dec)345 IMPEG2D_ERROR_CODES_T impeg2d_pre_pic_dec_proc(dec_state_t *ps_dec)
346 {
347 WORD32 u4_get_disp;
348 pic_buf_t *ps_disp_pic;
349 IMPEG2D_ERROR_CODES_T e_error = (IMPEG2D_ERROR_CODES_T)IVD_ERROR_NONE;
350
351 u4_get_disp = 0;
352 ps_disp_pic = NULL;
353
354 /* Field Picture */
355 if(ps_dec->u2_picture_structure != FRAME_PICTURE)
356 {
357 ps_dec->u2_num_vert_mb = (ps_dec->u2_vertical_size + 31) >> 5;
358
359 if(ps_dec->u2_num_flds_decoded == 0)
360 {
361 pic_buf_t *ps_pic_buf;
362 u4_get_disp = 1;
363
364 ps_pic_buf = impeg2_buf_mgr_get_next_free(ps_dec->pv_pic_buf_mg, &ps_dec->i4_cur_buf_id);
365
366 if (NULL == ps_pic_buf)
367 {
368 return IMPEG2D_NO_FREE_BUF_ERR;
369 }
370
371 impeg2_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mg, ps_dec->i4_cur_buf_id, BUF_MGR_DISP);
372 impeg2_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mg, ps_dec->i4_cur_buf_id, BUF_MGR_REF);
373 if(ps_dec->u4_deinterlace)
374 impeg2_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mg, ps_dec->i4_cur_buf_id, MPEG2_BUF_MGR_DEINT);
375
376 ps_pic_buf->u4_ts = ps_dec->u4_inp_ts;
377 ps_pic_buf->e_pic_type = ps_dec->e_pic_type;
378 ps_dec->ps_cur_pic = ps_pic_buf;
379 ps_dec->s_cur_frm_buf.pu1_y = ps_pic_buf->pu1_y;
380 ps_dec->s_cur_frm_buf.pu1_u = ps_pic_buf->pu1_u;
381 ps_dec->s_cur_frm_buf.pu1_v = ps_pic_buf->pu1_v;
382 }
383
384 if(ps_dec->u2_picture_structure == TOP_FIELD)
385 {
386 ps_dec->u2_fld_parity = TOP;
387 }
388 else
389 {
390 ps_dec->u2_fld_parity = BOTTOM;
391 }
392 ps_dec->u2_field_dct = 0;
393 ps_dec->u2_read_dct_type = 0;
394 ps_dec->u2_read_motion_type = 1;
395 ps_dec->u2_fld_pic = 1;
396 ps_dec->u2_frm_pic = 0;
397 ps_dec->ps_func_forw_or_back = gas_impeg2d_func_fld_fw_or_bk;
398 ps_dec->ps_func_bi_direct = gas_impeg2d_func_fld_bi_direct;
399 }
400 /* Frame Picture */
401 else
402 {
403 pic_buf_t *ps_pic_buf;
404
405
406 ps_dec->u2_num_vert_mb = (ps_dec->u2_vertical_size + 15) >> 4;
407 u4_get_disp = 1;
408 ps_pic_buf = impeg2_buf_mgr_get_next_free(ps_dec->pv_pic_buf_mg, &ps_dec->i4_cur_buf_id);
409
410 if (NULL == ps_pic_buf)
411 {
412 return IMPEG2D_NO_FREE_BUF_ERR;
413 }
414 impeg2_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mg, ps_dec->i4_cur_buf_id, BUF_MGR_DISP);
415 impeg2_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mg, ps_dec->i4_cur_buf_id, BUF_MGR_REF);
416 if(ps_dec->u4_deinterlace)
417 impeg2_buf_mgr_set_status((buf_mgr_t *)ps_dec->pv_pic_buf_mg, ps_dec->i4_cur_buf_id, MPEG2_BUF_MGR_DEINT);
418
419 ps_pic_buf->u4_ts = ps_dec->u4_inp_ts;
420 ps_pic_buf->e_pic_type = ps_dec->e_pic_type;
421 ps_dec->ps_cur_pic = ps_pic_buf;
422 ps_dec->s_cur_frm_buf.pu1_y = ps_pic_buf->pu1_y;
423 ps_dec->s_cur_frm_buf.pu1_u = ps_pic_buf->pu1_u;
424 ps_dec->s_cur_frm_buf.pu1_v = ps_pic_buf->pu1_v;
425
426
427 if(ps_dec->u2_frame_pred_frame_dct == 0)
428 {
429 ps_dec->u2_read_dct_type = 1;
430 ps_dec->u2_read_motion_type = 1;
431 }
432 else
433 {
434 ps_dec->u2_read_dct_type = 0;
435 ps_dec->u2_read_motion_type = 0;
436 ps_dec->u2_motion_type = 2;
437 ps_dec->u2_field_dct = 0;
438 }
439
440 ps_dec->u2_fld_parity = TOP;
441 ps_dec->u2_fld_pic = 0;
442 ps_dec->u2_frm_pic = 1;
443 ps_dec->ps_func_forw_or_back = gas_impeg2d_func_frm_fw_or_bk;
444 ps_dec->ps_func_bi_direct = gas_impeg2d_func_frm_bi_direct;
445 }
446 ps_dec->u2_def_dc_pred[Y_LUMA] = 128 << ps_dec->u2_intra_dc_precision;
447 ps_dec->u2_def_dc_pred[U_CHROMA] = 128 << ps_dec->u2_intra_dc_precision;
448 ps_dec->u2_def_dc_pred[V_CHROMA] = 128 << ps_dec->u2_intra_dc_precision;
449 ps_dec->u2_num_mbs_left = ps_dec->u2_num_horiz_mb * ps_dec->u2_num_vert_mb;
450 if(u4_get_disp)
451 {
452 if(ps_dec->u4_num_frames_decoded > 1)
453 {
454 ps_disp_pic = impeg2_disp_mgr_get(&ps_dec->s_disp_mgr, &ps_dec->i4_disp_buf_id);
455 }
456 ps_dec->ps_disp_pic = ps_disp_pic;
457 if(ps_disp_pic)
458 {
459 if(1 == ps_dec->u4_share_disp_buf)
460 {
461 ps_dec->ps_disp_frm_buf->pv_y_buf = ps_disp_pic->pu1_y;
462 if(IV_YUV_420P == ps_dec->i4_chromaFormat)
463 {
464 ps_dec->ps_disp_frm_buf->pv_u_buf = ps_disp_pic->pu1_u;
465 ps_dec->ps_disp_frm_buf->pv_v_buf = ps_disp_pic->pu1_v;
466 }
467 else
468 {
469 UWORD8 *pu1_buf;
470
471 pu1_buf = ps_dec->as_disp_buffers[ps_disp_pic->i4_buf_id].pu1_bufs[1];
472 ps_dec->ps_disp_frm_buf->pv_u_buf = pu1_buf;
473
474 pu1_buf = ps_dec->as_disp_buffers[ps_disp_pic->i4_buf_id].pu1_bufs[2];
475 ps_dec->ps_disp_frm_buf->pv_v_buf = pu1_buf;
476 }
477 }
478 }
479 }
480
481
482 switch(ps_dec->e_pic_type)
483 {
484 case I_PIC:
485 {
486 ps_dec->pf_decode_slice = impeg2d_dec_i_slice;
487 break;
488 }
489 case D_PIC:
490 {
491 ps_dec->pf_decode_slice = impeg2d_dec_d_slice;
492 break;
493 }
494 case P_PIC:
495 {
496 ps_dec->pf_decode_slice = impeg2d_dec_p_b_slice;
497 ps_dec->pu2_mb_type = gau2_impeg2d_p_mb_type;
498 break;
499 }
500 case B_PIC:
501 {
502 ps_dec->pf_decode_slice = impeg2d_dec_p_b_slice;
503 ps_dec->pu2_mb_type = gau2_impeg2d_b_mb_type;
504 break;
505 }
506 default:
507 return IMPEG2D_INVALID_PIC_TYPE;
508 }
509
510 /*************************************************************************/
511 /* Set the reference pictures */
512 /*************************************************************************/
513
514 /* Error resilience: If forward and backward pictures are going to be NULL*/
515 /* then assign both to the current */
516 /* if one of them NULL then we will assign the non null to the NULL one */
517
518 if(ps_dec->e_pic_type == P_PIC)
519 {
520 if (NULL == ps_dec->as_recent_fld[1][0].pu1_y)
521 {
522 ps_dec->as_recent_fld[1][0] = ps_dec->s_cur_frm_buf;
523 }
524 if (NULL == ps_dec->as_recent_fld[1][1].pu1_y)
525 {
526 impeg2d_get_bottom_field_buf(&ps_dec->s_cur_frm_buf, &ps_dec->as_recent_fld[1][1],
527 ps_dec->u2_frame_width);
528 }
529
530 ps_dec->as_ref_buf[FORW][TOP] = ps_dec->as_recent_fld[1][0];
531 ps_dec->as_ref_buf[FORW][BOTTOM] = ps_dec->as_recent_fld[1][1];
532
533
534 }
535 else if(ps_dec->e_pic_type == B_PIC)
536 {
537 if((NULL == ps_dec->as_recent_fld[1][0].pu1_y) && (NULL == ps_dec->as_recent_fld[0][0].pu1_y))
538 {
539 // assign the current picture to both
540 ps_dec->as_recent_fld[1][0] = ps_dec->s_cur_frm_buf;
541 impeg2d_get_bottom_field_buf(&ps_dec->s_cur_frm_buf, &ps_dec->as_recent_fld[1][1],
542 ps_dec->u2_frame_width);
543 ps_dec->as_recent_fld[0][0] = ps_dec->s_cur_frm_buf;
544 ps_dec->as_recent_fld[0][1] = ps_dec->as_recent_fld[1][1];
545 }
546 //Assign the non-null picture to the null picture
547 else if ((NULL != ps_dec->as_recent_fld[1][0].pu1_y) && (NULL == ps_dec->as_recent_fld[0][0].pu1_y))
548 {
549 ps_dec->as_recent_fld[0][0] = ps_dec->as_recent_fld[1][0];
550 ps_dec->as_recent_fld[0][1] = ps_dec->as_recent_fld[1][1];
551 }
552 else if ((NULL == ps_dec->as_recent_fld[1][0].pu1_y) && (NULL != ps_dec->as_recent_fld[0][0].pu1_y))
553 {
554 ps_dec->as_recent_fld[1][0] = ps_dec->as_recent_fld[0][0];
555 ps_dec->as_recent_fld[1][1] = ps_dec->as_recent_fld[0][1];
556 }
557
558 ps_dec->as_ref_buf[FORW][TOP] = ps_dec->as_recent_fld[0][0];
559 ps_dec->as_ref_buf[FORW][BOTTOM] = ps_dec->as_recent_fld[0][1];
560 ps_dec->as_ref_buf[BACK][TOP] = ps_dec->as_recent_fld[1][0];
561 ps_dec->as_ref_buf[BACK][BOTTOM] = ps_dec->as_recent_fld[1][1];
562
563
564 }
565
566 return e_error;
567 }
568
569 /*******************************************************************************
570 *
571 * Function Name : impeg2d_post_pic_dec_proc
572 *
573 * Description : Performs processing that is needed at the end of picture
574 * decode
575 *
576 * Arguments :
577 * dec : Decoder context
578 *
579 * Values Returned : None
580 *******************************************************************************/
impeg2d_post_pic_dec_proc(dec_state_t * ps_dec)581 void impeg2d_post_pic_dec_proc(dec_state_t *ps_dec)
582 {
583
584 WORD32 u4_update_pic_buf = 0;
585 /*************************************************************************/
586 /* Processing at the end of picture */
587 /*************************************************************************/
588 if(ps_dec->u2_picture_structure != FRAME_PICTURE)
589 {
590 ps_dec->u2_num_vert_mb = (ps_dec->u2_vertical_size + 31) >> 5;
591
592 if(ps_dec->u2_num_flds_decoded == 1)
593 {
594 ps_dec->u2_num_flds_decoded = 0;
595 u4_update_pic_buf = 1;
596 }
597 else
598 {
599 ps_dec->u2_num_flds_decoded = 1;
600 }
601 }
602 else
603 {
604 u4_update_pic_buf = 1;
605 }
606
607 if(u4_update_pic_buf)
608 {
609 ps_dec->i4_frame_decoded = 1;
610 if(ps_dec->e_pic_type != B_PIC)
611 {
612 /* In any sequence first two pictures have to be reference pictures */
613 /* Adding of first picture in the sequence */
614 if(ps_dec->aps_ref_pics[0] == NULL)
615 {
616 ps_dec->aps_ref_pics[0] = ps_dec->ps_cur_pic;
617 }
618
619 /* Adding of second picture in the sequence */
620 else if(ps_dec->aps_ref_pics[1] == NULL)
621 {
622 ps_dec->aps_ref_pics[1] = ps_dec->ps_cur_pic;
623 impeg2_disp_mgr_add(&ps_dec->s_disp_mgr, ps_dec->aps_ref_pics[0], ps_dec->aps_ref_pics[0]->i4_buf_id);
624 }
625 else
626 {
627
628 impeg2_disp_mgr_add(&ps_dec->s_disp_mgr, ps_dec->aps_ref_pics[1], ps_dec->aps_ref_pics[1]->i4_buf_id);
629 impeg2_buf_mgr_release(ps_dec->pv_pic_buf_mg, ps_dec->aps_ref_pics[0]->i4_buf_id, BUF_MGR_REF);
630 ps_dec->aps_ref_pics[0] = ps_dec->aps_ref_pics[1];
631 ps_dec->aps_ref_pics[1] = ps_dec->ps_cur_pic;
632
633 }
634 }
635 else
636 {
637 impeg2_disp_mgr_add(&ps_dec->s_disp_mgr, ps_dec->ps_cur_pic, ps_dec->ps_cur_pic->i4_buf_id);
638
639 impeg2_buf_mgr_release(ps_dec->pv_pic_buf_mg, ps_dec->ps_cur_pic->i4_buf_id, BUF_MGR_REF);
640 }
641
642 }
643 /*************************************************************************/
644 /* Update the list of recent reference pictures */
645 /*************************************************************************/
646 if(ps_dec->e_pic_type != B_PIC)
647 {
648 switch(ps_dec->u2_picture_structure)
649 {
650 case FRAME_PICTURE:
651 {
652 ps_dec->as_recent_fld[0][0] = ps_dec->as_recent_fld[1][0];
653 ps_dec->as_recent_fld[0][1] = ps_dec->as_recent_fld[1][1];
654
655 ps_dec->as_recent_fld[1][0] = ps_dec->s_cur_frm_buf;
656 impeg2d_get_bottom_field_buf(&ps_dec->s_cur_frm_buf, &ps_dec->as_recent_fld[1][1],
657 ps_dec->u2_frame_width);
658 break;
659 }
660 case TOP_FIELD:
661 {
662 ps_dec->as_recent_fld[0][0] = ps_dec->as_recent_fld[1][0];
663 ps_dec->as_recent_fld[1][0] = ps_dec->s_cur_frm_buf;
664 break;
665 }
666 case BOTTOM_FIELD:
667 {
668 ps_dec->as_recent_fld[0][1] = ps_dec->as_recent_fld[1][1];
669 impeg2d_get_bottom_field_buf(&ps_dec->s_cur_frm_buf, &ps_dec->as_recent_fld[1][1],
670 ps_dec->u2_frame_width);
671 break;
672 }
673 }
674 }
675 }
676