[svn] / trunk / xvidcore / src / decoder.c Repository:
ViewVC logotype

Annotation of /trunk/xvidcore/src/decoder.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 161 - (view) (download)

1 : Isibaar 3 /**************************************************************************
2 :     *
3 : edgomez 146 * XVID MPEG-4 VIDEO CODEC
4 :     * - Decoder main module -
5 : Isibaar 3 *
6 : edgomez 146 * This program is an implementation of a part of one or more MPEG-4
7 :     * Video tools as specified in ISO/IEC 14496-2 standard. Those intending
8 :     * to use this software module in hardware or software products are
9 :     * advised that its use may infringe existing patents or copyrights, and
10 :     * any such use would be at such party's own risk. The original
11 :     * developer of this software module and his/her company, and subsequent
12 :     * editors and their companies, will have no liability for use of this
13 :     * software or modifications or derivatives thereof.
14 : Isibaar 3 *
15 : edgomez 146 * This program is free software; you can redistribute it and/or modify
16 :     * it under the terms of the GNU General Public License as published by
17 :     * the Free Software Foundation; either version 2 of the License, or
18 :     * (at your option) any later version.
19 : Isibaar 3 *
20 : edgomez 146 * This program is distributed in the hope that it will be useful,
21 :     * but WITHOUT ANY WARRANTY; without even the implied warranty of
22 :     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
23 :     * GNU General Public License for more details.
24 : Isibaar 3 *
25 : edgomez 146 * You should have received a copy of the GNU General Public License
26 :     * along with this program; if not, write to the Free Software
27 :     * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
28 : Isibaar 3 *
29 :     *************************************************************************/
30 :    
31 :     /**************************************************************************
32 :     *
33 : edgomez 146 * History:
34 : Isibaar 3 *
35 : chenm001 156 * 02.05.2002 add B-frame decode support(have some problem);
36 :     * MinChen <chenm001@163.com>
37 : chenm001 133 * 22.04.2002 add some B-frame decode support; chenm001 <chenm001@163.com>
38 : h 81 * 29.03.2002 interlacing fix - compensated block wasn't being used when
39 :     * reconstructing blocks, thus artifacts
40 :     * interlacing speedup - used transfers to re-interlace
41 :     * interlaced decoding should be as fast as progressive now
42 : h 69 * 26.03.2002 interlacing support - moved transfers outside decode loop
43 : edgomez 146 * 26.12.2001 decoder_mbinter: dequant/idct moved within if(coded) block
44 :     * 22.12.2001 lock based interpolation
45 :     * 01.12.2001 inital version; (c)2001 peter ross <pross@cs.rmit.edu.au>
46 : Isibaar 3 *
47 : chenm001 161 * $Id: decoder.c,v 1.15 2002-05-06 03:51:43 chenm001 Exp $
48 : edgomez 146 *
49 : Isibaar 3 *************************************************************************/
50 :    
51 :     #include <stdlib.h>
52 : edgomez 146 #include <string.h>
53 : Isibaar 3
54 :     #include "xvid.h"
55 :     #include "portab.h"
56 :    
57 :     #include "decoder.h"
58 :     #include "bitstream/bitstream.h"
59 :     #include "bitstream/mbcoding.h"
60 :    
61 :     #include "quant/quant_h263.h"
62 :     #include "quant/quant_mpeg4.h"
63 :     #include "dct/idct.h"
64 :     #include "dct/fdct.h"
65 :     #include "utils/mem_transfer.h"
66 :     #include "image/interpolate8x8.h"
67 :    
68 :     #include "bitstream/mbcoding.h"
69 :     #include "prediction/mbprediction.h"
70 :     #include "utils/timer.h"
71 :     #include "utils/emms.h"
72 :    
73 :     #include "image/image.h"
74 :     #include "image/colorspace.h"
75 : Isibaar 41 #include "utils/mem_align.h"
76 : Isibaar 3
77 :     int decoder_create(XVID_DEC_PARAM * param)
78 :     {
79 :     DECODER * dec;
80 :    
81 : Isibaar 42 dec = xvid_malloc(sizeof(DECODER), CACHE_LINE);
82 : Isibaar 3 if (dec == NULL)
83 :     {
84 :     return XVID_ERR_MEMORY;
85 :     }
86 :     param->handle = dec;
87 :    
88 :     dec->width = param->width;
89 :     dec->height = param->height;
90 :    
91 :     dec->mb_width = (dec->width + 15) / 16;
92 :     dec->mb_height = (dec->height + 15) / 16;
93 :    
94 :     dec->edged_width = 16 * dec->mb_width + 2 * EDGE_SIZE;
95 :     dec->edged_height = 16 * dec->mb_height + 2 * EDGE_SIZE;
96 :    
97 :     if (image_create(&dec->cur, dec->edged_width, dec->edged_height))
98 :     {
99 : Isibaar 41 xvid_free(dec);
100 : Isibaar 3 return XVID_ERR_MEMORY;
101 :     }
102 :    
103 : chenm001 133 if (image_create(&dec->refn[0], dec->edged_width, dec->edged_height))
104 : Isibaar 3 {
105 :     image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
106 : Isibaar 41 xvid_free(dec);
107 : Isibaar 3 return XVID_ERR_MEMORY;
108 :     }
109 : chenm001 133 // add by chenm001 <chenm001@163.com>
110 :     // for support B-frame to reference last 2 frame
111 :     if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height))
112 :     {
113 :     image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
114 :     image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
115 :     xvid_free(dec);
116 :     return XVID_ERR_MEMORY;
117 :     }
118 : chenm001 156 if (image_create(&dec->refn[2], dec->edged_width, dec->edged_height))
119 :     {
120 :     image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
121 :     image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
122 :     image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
123 :     xvid_free(dec);
124 :     return XVID_ERR_MEMORY;
125 :     }
126 : Isibaar 3
127 : Isibaar 42 dec->mbs = xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height, CACHE_LINE);
128 : Isibaar 3 if (dec->mbs == NULL)
129 :     {
130 :     image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
131 : chenm001 156 image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
132 :     image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
133 :     image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
134 : Isibaar 41 xvid_free(dec);
135 : Isibaar 3 return XVID_ERR_MEMORY;
136 :     }
137 : chenm001 156 // add by chenm001 <chenm001@163.com>
138 :     // for skip MB flag
139 :     dec->last_mbs = xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height, CACHE_LINE);
140 :     if (dec->last_mbs == NULL)
141 :     {
142 :     xvid_free(dec->mbs);
143 :     image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
144 :     image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
145 :     image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
146 :     image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
147 :     xvid_free(dec);
148 :     return XVID_ERR_MEMORY;
149 :     }
150 : Isibaar 3
151 :     init_timer();
152 :    
153 : chenm001 156 // add by chenm001 <chenm001@163.com>
154 :     // for support B-frame to save reference frame's time
155 :     dec->frames = -1;
156 :     dec->time = dec->time_base = dec->last_time_base = 0;
157 :    
158 : Isibaar 3 return XVID_ERR_OK;
159 :     }
160 :    
161 :    
162 :     int decoder_destroy(DECODER * dec)
163 :     {
164 : chenm001 156 xvid_free(dec->last_mbs);
165 : Isibaar 41 xvid_free(dec->mbs);
166 : chenm001 133 image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
167 : chenm001 156 image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
168 :     image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
169 : Isibaar 3 image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
170 : Isibaar 41 xvid_free(dec);
171 : Isibaar 3
172 :     write_timer();
173 :     return XVID_ERR_OK;
174 :     }
175 :    
176 :    
177 :    
178 :     static const int32_t dquant_table[4] =
179 :     {
180 :     -1, -2, 1, 2
181 :     };
182 :    
183 :    
184 : chenm001 156
185 :    
186 : Isibaar 3 // decode an intra macroblock
187 :    
188 : h 69 void decoder_mbintra(DECODER * dec,
189 : edgomez 78 MACROBLOCK * pMB,
190 :     const uint32_t x_pos,
191 :     const uint32_t y_pos,
192 :     const uint32_t acpred_flag,
193 :     const uint32_t cbp,
194 :     Bitstream * bs,
195 :     const uint32_t quant,
196 :     const uint32_t intra_dc_threshold)
197 : Isibaar 3 {
198 : edgomez 78
199 :     DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
200 :     DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
201 :    
202 : h 81 uint32_t stride = dec->edged_width;
203 :     uint32_t stride2 = stride / 2;
204 :     uint32_t next_block = stride * 8;
205 : h 69 uint32_t i;
206 :     uint32_t iQuant = pMB->quant;
207 :     uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
208 :    
209 : edgomez 78 pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
210 : h 81 pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
211 :     pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
212 : h 69
213 : edgomez 78 memset(block, 0, 6*64*sizeof(int16_t)); // clear
214 : h 69
215 :     for (i = 0; i < 6; i++)
216 : Isibaar 3 {
217 : h 69 uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);
218 : Isibaar 3 int16_t predictors[8];
219 :     int start_coeff;
220 :    
221 :     start_timer();
222 : edgomez 78 predict_acdc(dec->mbs, x_pos, y_pos, dec->mb_width, i, &block[i*64], iQuant, iDcScaler, predictors);
223 : Isibaar 3 if (!acpred_flag)
224 :     {
225 : h 69 pMB->acpred_directions[i] = 0;
226 : Isibaar 3 }
227 :     stop_prediction_timer();
228 :    
229 :     if (quant < intra_dc_threshold)
230 :     {
231 :     int dc_size;
232 :     int dc_dif;
233 :    
234 : h 69 dc_size = i < 4 ? get_dc_size_lum(bs) : get_dc_size_chrom(bs);
235 : Isibaar 3 dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0 ;
236 :    
237 :     if (dc_size > 8)
238 :     {
239 :     BitstreamSkip(bs, 1); // marker
240 :     }
241 :    
242 : edgomez 78 block[i*64 + 0] = dc_dif;
243 : Isibaar 3 start_coeff = 1;
244 :     }
245 :     else
246 :     {
247 :     start_coeff = 0;
248 :     }
249 :    
250 :     start_timer();
251 : h 69 if (cbp & (1 << (5-i))) // coded
252 : Isibaar 3 {
253 : edgomez 78 get_intra_block(bs, &block[i*64], pMB->acpred_directions[i], start_coeff);
254 : Isibaar 3 }
255 :     stop_coding_timer();
256 :    
257 :     start_timer();
258 : edgomez 78 add_acdc(pMB, i, &block[i*64], iDcScaler, predictors);
259 : Isibaar 3 stop_prediction_timer();
260 :    
261 :     start_timer();
262 :     if (dec->quant_type == 0)
263 :     {
264 : edgomez 78 dequant_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);
265 : Isibaar 3 }
266 :     else
267 :     {
268 : edgomez 78 dequant4_intra(&data[i*64], &block[i*64], iQuant, iDcScaler);
269 : Isibaar 3 }
270 :     stop_iquant_timer();
271 :    
272 :     start_timer();
273 : edgomez 78 idct(&data[i*64]);
274 : Isibaar 3 stop_idct_timer();
275 : h 69 }
276 : Isibaar 3
277 : Isibaar 110 if (dec->interlacing && pMB->field_dct)
278 : h 69 {
279 : h 81 next_block = stride;
280 :     stride *= 2;
281 : Isibaar 3 }
282 : h 69
283 :     start_timer();
284 : edgomez 78 transfer_16to8copy(pY_Cur, &data[0*64], stride);
285 :     transfer_16to8copy(pY_Cur + 8, &data[1*64], stride);
286 : h 81 transfer_16to8copy(pY_Cur + next_block, &data[2*64], stride);
287 :     transfer_16to8copy(pY_Cur + 8 + next_block, &data[3*64], stride);
288 :     transfer_16to8copy(pU_Cur, &data[4*64], stride2);
289 :     transfer_16to8copy(pV_Cur, &data[5*64], stride2);
290 : h 69 stop_transfer_timer();
291 : Isibaar 3 }
292 :    
293 :    
294 :    
295 :    
296 :    
297 :     #define SIGN(X) (((X)>0)?1:-1)
298 :     #define ABS(X) (((X)>0)?(X):-(X))
299 :     static const uint32_t roundtab[16] =
300 : edgomez 78 { 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2 };
301 : Isibaar 3
302 :    
303 :     // decode an inter macroblock
304 :    
305 : h 69 void decoder_mbinter(DECODER * dec,
306 : edgomez 78 const MACROBLOCK * pMB,
307 :     const uint32_t x_pos,
308 :     const uint32_t y_pos,
309 :     const uint32_t acpred_flag,
310 :     const uint32_t cbp,
311 :     Bitstream * bs,
312 :     const uint32_t quant,
313 :     const uint32_t rounding)
314 : Isibaar 3 {
315 : h 69
316 : edgomez 78 DECLARE_ALIGNED_MATRIX(block,6, 64, int16_t, CACHE_LINE);
317 :     DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
318 :    
319 : h 81 uint32_t stride = dec->edged_width;
320 :     uint32_t stride2 = stride / 2;
321 :     uint32_t next_block = stride * 8;
322 : edgomez 78 uint32_t i;
323 :     uint32_t iQuant = pMB->quant;
324 : h 69 uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
325 : Isibaar 3 int uv_dx, uv_dy;
326 :    
327 : edgomez 78 pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
328 : h 81 pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
329 :     pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
330 : h 69
331 :     if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)
332 : Isibaar 3 {
333 : h 69 uv_dx = pMB->mvs[0].x;
334 :     uv_dy = pMB->mvs[0].y;
335 : Isibaar 3
336 :     uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
337 :     uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
338 :     }
339 :     else
340 :     {
341 :     int sum;
342 : h 69 sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
343 : Isibaar 3 uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );
344 :    
345 : h 69 sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
346 : Isibaar 3 uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );
347 :     }
348 :    
349 :     start_timer();
350 : chenm001 133 interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos , pMB->mvs[0].x, pMB->mvs[0].y, stride, rounding);
351 :     interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos , pMB->mvs[1].x, pMB->mvs[1].y, stride, rounding);
352 :     interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos + 8, pMB->mvs[2].x, pMB->mvs[2].y, stride, rounding);
353 :     interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride, rounding);
354 :     interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8*x_pos, 8*y_pos, uv_dx, uv_dy, stride2, rounding);
355 :     interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8*x_pos, 8*y_pos, uv_dx, uv_dy, stride2, rounding);
356 : Isibaar 3 stop_comp_timer();
357 :    
358 : h 69 for (i = 0; i < 6; i++)
359 : Isibaar 3 {
360 : h 69 if (cbp & (1 << (5-i))) // coded
361 : Isibaar 3 {
362 : edgomez 78 memset(&block[i*64], 0, 64 * sizeof(int16_t)); // clear
363 : Isibaar 3
364 :     start_timer();
365 : edgomez 78 get_inter_block(bs, &block[i*64]);
366 : Isibaar 3 stop_coding_timer();
367 :    
368 :     start_timer();
369 :     if (dec->quant_type == 0)
370 :     {
371 : edgomez 78 dequant_inter(&data[i*64], &block[i*64], iQuant);
372 : Isibaar 3 }
373 :     else
374 :     {
375 : edgomez 78 dequant4_inter(&data[i*64], &block[i*64], iQuant);
376 : Isibaar 3 }
377 :     stop_iquant_timer();
378 :    
379 :     start_timer();
380 : edgomez 78 idct(&data[i*64]);
381 : Isibaar 3 stop_idct_timer();
382 :     }
383 :     }
384 : h 69
385 : Isibaar 110 if (dec->interlacing && pMB->field_dct)
386 : h 69 {
387 : h 81 next_block = stride;
388 :     stride *= 2;
389 : h 69 }
390 :    
391 :     start_timer();
392 :     if (cbp & 32)
393 : edgomez 78 transfer_16to8add(pY_Cur, &data[0*64], stride);
394 : h 69 if (cbp & 16)
395 : edgomez 78 transfer_16to8add(pY_Cur + 8, &data[1*64], stride);
396 : h 69 if (cbp & 8)
397 : h 81 transfer_16to8add(pY_Cur + next_block, &data[2*64], stride);
398 : h 69 if (cbp & 4)
399 : h 81 transfer_16to8add(pY_Cur + 8 + next_block, &data[3*64], stride);
400 : h 69 if (cbp & 2)
401 : h 81 transfer_16to8add(pU_Cur, &data[4*64], stride2);
402 : h 69 if (cbp & 1)
403 : h 81 transfer_16to8add(pV_Cur, &data[5*64], stride2);
404 : h 69 stop_transfer_timer();
405 : Isibaar 3 }
406 :    
407 :    
408 :     void decoder_iframe(DECODER * dec, Bitstream * bs, int quant, int intra_dc_threshold)
409 :     {
410 : edgomez 78
411 : Isibaar 3 uint32_t x, y;
412 :    
413 :     for (y = 0; y < dec->mb_height; y++)
414 :     {
415 :     for (x = 0; x < dec->mb_width; x++)
416 :     {
417 :     MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];
418 :    
419 :     uint32_t mcbpc;
420 :     uint32_t cbpc;
421 :     uint32_t acpred_flag;
422 :     uint32_t cbpy;
423 :     uint32_t cbp;
424 :    
425 :     mcbpc = get_mcbpc_intra(bs);
426 :     mb->mode = mcbpc & 7;
427 :     cbpc = (mcbpc >> 4);
428 :    
429 :     acpred_flag = BitstreamGetBit(bs);
430 :    
431 :     if (mb->mode == MODE_STUFFING)
432 :     {
433 :     DEBUG("-- STUFFING ?");
434 :     continue;
435 :     }
436 :    
437 :     cbpy = get_cbpy(bs, 1);
438 :     cbp = (cbpy << 2) | cbpc;
439 :    
440 :     if (mb->mode == MODE_INTRA_Q)
441 :     {
442 :     quant += dquant_table[BitstreamGetBits(bs,2)];
443 :     if (quant > 31)
444 :     {
445 :     quant = 31;
446 :     }
447 :     else if (quant < 1)
448 :     {
449 :     quant = 1;
450 :     }
451 :     }
452 :     mb->quant = quant;
453 :    
454 : h 69 if (dec->interlacing)
455 :     {
456 :     mb->field_dct = BitstreamGetBit(bs);
457 :     DEBUG1("deci: field_dct: ", mb->field_dct);
458 :     }
459 :    
460 : Isibaar 3 decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant, intra_dc_threshold);
461 :     }
462 :     }
463 : edgomez 78
464 : Isibaar 3 }
465 :    
466 :    
467 :     void get_motion_vector(DECODER *dec, Bitstream *bs, int x, int y, int k, VECTOR * mv, int fcode)
468 :     {
469 : edgomez 78
470 : Isibaar 3 int scale_fac = 1 << (fcode - 1);
471 :     int high = (32 * scale_fac) - 1;
472 :     int low = ((-32) * scale_fac);
473 :     int range = (64 * scale_fac);
474 :    
475 :     VECTOR pmv[4];
476 : edgomez 148 int32_t psad[4];
477 : Isibaar 3
478 :     int mv_x, mv_y;
479 :     int pmv_x, pmv_y;
480 :    
481 :    
482 :     get_pmvdata(dec->mbs, x, y, dec->mb_width, k, pmv, psad);
483 :    
484 :     pmv_x = pmv[0].x;
485 :     pmv_y = pmv[0].y;
486 :    
487 :     mv_x = get_mv(bs, fcode);
488 :     mv_y = get_mv(bs, fcode);
489 :    
490 :     mv_x += pmv_x;
491 :     mv_y += pmv_y;
492 :    
493 :     if (mv_x < low)
494 :     {
495 :     mv_x += range;
496 :     }
497 :     else if (mv_x > high)
498 :     {
499 :     mv_x -= range;
500 :     }
501 :    
502 :     if (mv_y < low)
503 :     {
504 :     mv_y += range;
505 :     }
506 :     else if (mv_y > high)
507 :     {
508 :     mv_y -= range;
509 :     }
510 :    
511 :     mv->x = mv_x;
512 :     mv->y = mv_y;
513 :    
514 :     }
515 :    
516 :    
517 :     void decoder_pframe(DECODER * dec, Bitstream * bs, int rounding, int quant, int fcode, int intra_dc_threshold)
518 :     {
519 : edgomez 78
520 : Isibaar 3 uint32_t x, y;
521 :    
522 :     start_timer();
523 : chenm001 133 image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height, dec->width, dec->height, dec->interlacing);
524 : Isibaar 3 stop_edges_timer();
525 :    
526 :     for (y = 0; y < dec->mb_height; y++)
527 :     {
528 :     for (x = 0; x < dec->mb_width; x++)
529 :     {
530 :     MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];
531 :    
532 : chenm001 156 //if (!(dec->mb_skip[y*dec->mb_width + x]=BitstreamGetBit(bs))) // not_coded
533 :     if (!(BitstreamGetBit(bs))) // not_coded
534 : Isibaar 3 {
535 :     uint32_t mcbpc;
536 :     uint32_t cbpc;
537 :     uint32_t acpred_flag;
538 :     uint32_t cbpy;
539 :     uint32_t cbp;
540 :     uint32_t intra;
541 :    
542 :     mcbpc = get_mcbpc_inter(bs);
543 :     mb->mode = mcbpc & 7;
544 :     cbpc = (mcbpc >> 4);
545 : edgomez 12 acpred_flag = 0;
546 : Isibaar 3
547 :     intra = (mb->mode == MODE_INTRA || mb->mode == MODE_INTRA_Q);
548 :    
549 :     if (intra)
550 :     {
551 :     acpred_flag = BitstreamGetBit(bs);
552 :     }
553 :    
554 :     if (mb->mode == MODE_STUFFING)
555 :     {
556 :     DEBUG("-- STUFFING ?");
557 :     continue;
558 :     }
559 :    
560 :     cbpy = get_cbpy(bs, intra);
561 :     cbp = (cbpy << 2) | cbpc;
562 :    
563 :     if (mb->mode == MODE_INTER_Q || mb->mode == MODE_INTRA_Q)
564 :     {
565 :     quant += dquant_table[BitstreamGetBits(bs,2)];
566 :     if (quant > 31)
567 :     {
568 :     quant = 31;
569 :     }
570 :     else if (mb->quant < 1)
571 :     {
572 :     quant = 1;
573 :     }
574 :     }
575 :     mb->quant = quant;
576 : h 69
577 :     if (dec->interlacing)
578 : Isibaar 3 {
579 : h 69 mb->field_dct = BitstreamGetBit(bs);
580 :     DEBUG1("decp: field_dct: ", mb->field_dct);
581 : Isibaar 3
582 : h 69 if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q)
583 :     {
584 :     mb->field_pred = BitstreamGetBit(bs);
585 :     DEBUG1("decp: field_pred: ", mb->field_pred);
586 :    
587 :     if (mb->field_pred)
588 :     {
589 :     mb->field_for_top = BitstreamGetBit(bs);
590 :     DEBUG1("decp: field_for_top: ", mb->field_for_top);
591 :     mb->field_for_bot = BitstreamGetBit(bs);
592 :     DEBUG1("decp: field_for_bot: ", mb->field_for_bot);
593 :     }
594 :     }
595 : Isibaar 3 }
596 : h 69
597 :     if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q)
598 :     {
599 :     if (dec->interlacing && mb->field_pred)
600 :     {
601 :     get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);
602 :     get_motion_vector(dec, bs, x, y, 0, &mb->mvs[1], fcode);
603 :     }
604 :     else
605 :     {
606 :     get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);
607 :     mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;
608 :     mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;
609 :     }
610 :     }
611 : Isibaar 3 else if (mb->mode == MODE_INTER4V /* || mb->mode == MODE_INTER4V_Q */)
612 :     {
613 :     get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode);
614 :     get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode);
615 :     get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode);
616 :     get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode);
617 :     }
618 :     else // MODE_INTRA, MODE_INTRA_Q
619 :     {
620 :     mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;
621 :     mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;
622 :     decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant, intra_dc_threshold);
623 :     continue;
624 :     }
625 :    
626 :     decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant, rounding);
627 :     }
628 :     else // not coded
629 :     {
630 : chenm001 161 //DEBUG2("P-frame MB at (X,Y)=",x,y);
631 : chenm001 156 mb->mode = MODE_NOT_CODED;
632 : Isibaar 3 mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;
633 :     mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;
634 :    
635 :     // copy macroblock directly from ref to cur
636 :    
637 :     start_timer();
638 :    
639 :     transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x),
640 : chenm001 133 dec->refn[0].y + (16*y)*dec->edged_width + (16*x),
641 : edgomez 78 dec->edged_width);
642 : Isibaar 3
643 :     transfer8x8_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x+8),
644 : chenm001 133 dec->refn[0].y + (16*y)*dec->edged_width + (16*x+8),
645 : edgomez 78 dec->edged_width);
646 : Isibaar 3
647 :     transfer8x8_copy(dec->cur.y + (16*y+8)*dec->edged_width + (16*x),
648 : chenm001 133 dec->refn[0].y + (16*y+8)*dec->edged_width + (16*x),
649 : edgomez 78 dec->edged_width);
650 : Isibaar 3
651 :     transfer8x8_copy(dec->cur.y + (16*y+8)*dec->edged_width + (16*x+8),
652 : chenm001 133 dec->refn[0].y + (16*y+8)*dec->edged_width + (16*x+8),
653 : edgomez 78 dec->edged_width);
654 : Isibaar 3
655 :     transfer8x8_copy(dec->cur.u + (8*y)*dec->edged_width/2 + (8*x),
656 : chenm001 133 dec->refn[0].u + (8*y)*dec->edged_width/2 + (8*x),
657 : edgomez 78 dec->edged_width/2);
658 : Isibaar 3
659 :     transfer8x8_copy(dec->cur.v + (8*y)*dec->edged_width/2 + (8*x),
660 : chenm001 133 dec->refn[0].v + (8*y)*dec->edged_width/2 + (8*x),
661 : edgomez 78 dec->edged_width/2);
662 : Isibaar 3
663 :     stop_transfer_timer();
664 :     }
665 :     }
666 :     }
667 :     }
668 :    
669 : chenm001 156
670 :     // add by MinChen <chenm001@163.com>
671 :     // decode B-frame motion vector
672 :     void get_b_motion_vector(DECODER *dec, Bitstream *bs, int x, int y, VECTOR * mv, int fcode, const VECTOR pmv)
673 :     {
674 :     int scale_fac = 1 << (fcode - 1);
675 :     int high = (32 * scale_fac) - 1;
676 :     int low = ((-32) * scale_fac);
677 :     int range = (64 * scale_fac);
678 :    
679 :     int mv_x, mv_y;
680 :     int pmv_x, pmv_y;
681 :    
682 :     pmv_x = pmv.x;
683 :     pmv_y = pmv.y;
684 :    
685 :     mv_x = get_mv(bs, fcode);
686 :     mv_y = get_mv(bs, fcode);
687 :    
688 :     mv_x += pmv_x;
689 :     mv_y += pmv_y;
690 :    
691 :     if (mv_x < low)
692 :     {
693 :     mv_x += range;
694 :     }
695 :     else if (mv_x > high)
696 :     {
697 :     mv_x -= range;
698 :     }
699 :    
700 :     if (mv_y < low)
701 :     {
702 :     mv_y += range;
703 :     }
704 :     else if (mv_y > high)
705 :     {
706 :     mv_y -= range;
707 :     }
708 :    
709 :     mv->x = mv_x;
710 :     mv->y = mv_y;
711 :     }
712 :    
713 :    
714 :     // add by MinChen <chenm001@163.com>
715 :     // decode an B-frame forward & backward inter macroblock
716 :     void decoder_bf_mbinter(DECODER * dec,
717 :     const MACROBLOCK * pMB,
718 :     const uint32_t x_pos,
719 :     const uint32_t y_pos,
720 :     const uint32_t cbp,
721 :     Bitstream * bs,
722 :     const uint32_t quant,
723 :     const uint8_t ref)
724 :     {
725 :    
726 :     DECLARE_ALIGNED_MATRIX(block,6, 64, int16_t, CACHE_LINE);
727 :     DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
728 :    
729 :     uint32_t stride = dec->edged_width;
730 :     uint32_t stride2 = stride / 2;
731 :     uint32_t next_block = stride * 8;
732 :     uint32_t i;
733 :     uint32_t iQuant = pMB->quant;
734 :     uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
735 :     int uv_dx, uv_dy;
736 :    
737 :     pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
738 :     pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
739 :     pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
740 :    
741 :     if (!(pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q))
742 :     {
743 :     uv_dx = pMB->mvs[0].x;
744 :     uv_dy = pMB->mvs[0].y;
745 :    
746 :     uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
747 :     uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
748 :     }
749 :     else
750 :     {
751 :     int sum;
752 :     sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
753 :     uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );
754 :    
755 :     sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
756 :     uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );
757 :     }
758 :    
759 :     start_timer();
760 :     interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos, 16*y_pos , pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
761 :     interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos + 8, 16*y_pos , pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
762 :     interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos, 16*y_pos + 8, pMB->mvs[2].x, pMB->mvs[2].y, stride, 0);
763 :     interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos + 8, 16*y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride, 0);
764 :     interpolate8x8_switch(dec->cur.u, dec->refn[ref].u, 8*x_pos, 8*y_pos, uv_dx, uv_dy, stride2, 0);
765 :     interpolate8x8_switch(dec->cur.v, dec->refn[ref].v, 8*x_pos, 8*y_pos, uv_dx, uv_dy, stride2, 0);
766 :     stop_comp_timer();
767 :    
768 :     for (i = 0; i < 6; i++)
769 :     {
770 :     if (cbp & (1 << (5-i))) // coded
771 :     {
772 :     memset(&block[i*64], 0, 64 * sizeof(int16_t)); // clear
773 :    
774 :     start_timer();
775 :     get_inter_block(bs, &block[i*64]);
776 :     stop_coding_timer();
777 :    
778 :     start_timer();
779 :     if (dec->quant_type == 0)
780 :     {
781 :     dequant_inter(&data[i*64], &block[i*64], iQuant);
782 :     }
783 :     else
784 :     {
785 :     dequant4_inter(&data[i*64], &block[i*64], iQuant);
786 :     }
787 :     stop_iquant_timer();
788 :    
789 : chenm001 161 data[0]=10;
790 : chenm001 156 start_timer();
791 :     idct(&data[i*64]);
792 :     stop_idct_timer();
793 :     }
794 :     }
795 :    
796 :     if (dec->interlacing && pMB->field_dct)
797 :     {
798 :     next_block = stride;
799 :     stride *= 2;
800 :     }
801 :    
802 :     start_timer();
803 :     if (cbp & 32)
804 :     transfer_16to8add(pY_Cur, &data[0*64], stride);
805 :     if (cbp & 16)
806 :     transfer_16to8add(pY_Cur + 8, &data[1*64], stride);
807 :     if (cbp & 8)
808 :     transfer_16to8add(pY_Cur + next_block, &data[2*64], stride);
809 :     if (cbp & 4)
810 :     transfer_16to8add(pY_Cur + 8 + next_block, &data[3*64], stride);
811 :     if (cbp & 2)
812 :     transfer_16to8add(pU_Cur, &data[4*64], stride2);
813 :     if (cbp & 1)
814 :     transfer_16to8add(pV_Cur, &data[5*64], stride2);
815 :     stop_transfer_timer();
816 :     }
817 :    
818 :    
819 :     // add by MinChen <chenm001@163.com>
820 :     // decode an B-frame direct & inter macroblock
821 :     void decoder_bf_interpolate_mbinter(DECODER * dec,
822 :     IMAGE forward, IMAGE backward,
823 :     const MACROBLOCK * pMB,
824 :     const uint32_t x_pos,
825 :     const uint32_t y_pos,
826 : chenm001 161 const uint32_t cbp,
827 :     Bitstream * bs )
828 : chenm001 156 {
829 :    
830 :     DECLARE_ALIGNED_MATRIX(block,6, 64, int16_t, CACHE_LINE);
831 :     DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
832 :    
833 : chenm001 161 uint32_t stride = dec->edged_width;
834 :     uint32_t stride2 = stride / 2;
835 :     uint32_t next_block = stride * 8;
836 :     uint32_t iQuant = pMB->quant;
837 :     int uv_dx, uv_dy;
838 :     int b_uv_dx, b_uv_dy;
839 :     uint32_t i;
840 :     uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
841 : chenm001 156
842 : chenm001 161 pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
843 :     pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
844 :     pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
845 :    
846 : chenm001 156 if ((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q))
847 :     {
848 :     uv_dx = pMB->mvs[0].x;
849 :     uv_dy = pMB->mvs[0].y;
850 :    
851 :     uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
852 :     uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
853 :    
854 :     b_uv_dx = pMB->b_mvs[0].x;
855 :     b_uv_dy = pMB->b_mvs[0].y;
856 :    
857 :     b_uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
858 :     b_uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
859 :     }
860 :     else
861 :     {
862 :     int sum;
863 :     sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
864 :     uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );
865 :    
866 :     sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
867 :     uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );
868 :    
869 :     sum = pMB->b_mvs[0].x + pMB->b_mvs[1].x + pMB->b_mvs[2].x + pMB->b_mvs[3].x;
870 :     b_uv_dx = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );
871 :    
872 :     sum = pMB->b_mvs[0].y + pMB->b_mvs[1].y + pMB->b_mvs[2].y + pMB->b_mvs[3].y;
873 :     b_uv_dy = (sum == 0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) );
874 :     }
875 :    
876 :    
877 :     start_timer();
878 :     interpolate8x8_switch(dec->cur.y, forward.y, 16*x_pos, 16*y_pos , pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
879 :     interpolate8x8_switch(dec->cur.y, forward.y, 16*x_pos + 8, 16*y_pos , pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
880 :     interpolate8x8_switch(dec->cur.y, forward.y, 16*x_pos, 16*y_pos + 8, pMB->mvs[2].x, pMB->mvs[2].y, stride, 0);
881 :     interpolate8x8_switch(dec->cur.y, forward.y, 16*x_pos + 8, 16*y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride, 0);
882 :     interpolate8x8_switch(dec->cur.u, forward.u, 8*x_pos, 8*y_pos, uv_dx, uv_dy, stride2, 0);
883 :     interpolate8x8_switch(dec->cur.v, forward.v, 8*x_pos, 8*y_pos, uv_dx, uv_dy, stride2, 0);
884 :    
885 : chenm001 161
886 : chenm001 156 interpolate8x8_switch(dec->refn[2].y, backward.y, 16*x_pos, 16*y_pos , pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride, 0);
887 :     interpolate8x8_switch(dec->refn[2].y, backward.y, 16*x_pos + 8, 16*y_pos , pMB->b_mvs[1].x, pMB->b_mvs[1].y, stride, 0);
888 :     interpolate8x8_switch(dec->refn[2].y, backward.y, 16*x_pos, 16*y_pos + 8, pMB->b_mvs[2].x, pMB->b_mvs[2].y, stride, 0);
889 :     interpolate8x8_switch(dec->refn[2].y, backward.y, 16*x_pos + 8, 16*y_pos + 8, pMB->b_mvs[3].x, pMB->b_mvs[3].y, stride, 0);
890 :     interpolate8x8_switch(dec->refn[2].u, backward.u, 8*x_pos, 8*y_pos, b_uv_dx, b_uv_dy, stride2, 0);
891 :     interpolate8x8_switch(dec->refn[2].v, backward.v, 8*x_pos, 8*y_pos, b_uv_dx, b_uv_dy, stride2, 0);
892 :    
893 :     interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos, 16*y_pos , stride);
894 :     interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos + 8, 16*y_pos , stride);
895 :     interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos, 16*y_pos + 8, stride);
896 :     interpolate8x8_c(dec->cur.y, dec->refn[2].y, 16*x_pos + 8, 16*y_pos + 8, stride);
897 :     interpolate8x8_c(dec->cur.u, dec->refn[2].u, 8*x_pos, 8*y_pos, stride2);
898 :     interpolate8x8_c(dec->cur.v, dec->refn[2].v, 8*x_pos, 8*y_pos, stride2);
899 :    
900 :     stop_comp_timer();
901 :    
902 : chenm001 161 for (i = 0; i < 6; i++)
903 :     {
904 :     if (cbp & (1 << (5-i))) // coded
905 :     {
906 :     memset(&block[i*64], 0, 64 * sizeof(int16_t)); // clear
907 : chenm001 156
908 : chenm001 161 start_timer();
909 :     get_inter_block(bs, &block[i*64]);
910 :     stop_coding_timer();
911 :    
912 :     start_timer();
913 :     if (dec->quant_type == 0)
914 :     {
915 :     dequant_inter(&data[i*64], &block[i*64], iQuant);
916 :     }
917 :     else
918 :     {
919 :     dequant4_inter(&data[i*64], &block[i*64], iQuant);
920 :     }
921 :     stop_iquant_timer();
922 :    
923 :     start_timer();
924 :     idct(&data[i*64]);
925 :     stop_idct_timer();
926 :     }
927 :     }
928 :    
929 :     if (dec->interlacing && pMB->field_dct)
930 :     {
931 :     next_block = stride;
932 :     stride *= 2;
933 :     }
934 :    
935 :     start_timer();
936 :     if (cbp & 32)
937 :     transfer_16to8add(pY_Cur, &data[0*64], stride);
938 :     if (cbp & 16)
939 :     transfer_16to8add(pY_Cur + 8, &data[1*64], stride);
940 :     if (cbp & 8)
941 :     transfer_16to8add(pY_Cur + next_block, &data[2*64], stride);
942 :     if (cbp & 4)
943 :     transfer_16to8add(pY_Cur + 8 + next_block, &data[3*64], stride);
944 :     if (cbp & 2)
945 :     transfer_16to8add(pU_Cur, &data[4*64], stride2);
946 :     if (cbp & 1)
947 :     transfer_16to8add(pV_Cur, &data[5*64], stride2);
948 :     stop_transfer_timer();
949 : chenm001 156 }
950 :    
951 :    
952 :     // add by MinChen <chenm001@163.com>
953 :     // for decode B-frame dbquant
954 :     int32_t __inline get_dbquant(Bitstream * bs)
955 :     {
956 :     if (!BitstreamGetBit(bs)) // '0'
957 :     return(0);
958 :     else if (!BitstreamGetBit(bs)) // '10'
959 :     return(-2);
960 :     else
961 :     return(2); // '11'
962 :     }
963 :    
964 :     // add by MinChen <chenm001@163.com>
965 :     // for decode B-frame mb_type
966 :     // bit ret_value
967 :     // 1 0
968 :     // 01 1
969 :     // 001 2
970 :     // 0001 3
971 :     int32_t __inline get_mbtype(Bitstream * bs)
972 :     {
973 : chenm001 161 int32_t mb_type;
974 :     for(mb_type=0;mb_type<=3;mb_type++){
975 :     if (BitstreamGetBit(bs))
976 :     break;
977 :     }
978 :    
979 : chenm001 156 if (mb_type<=3)
980 :     return(mb_type);
981 :     else
982 :     return(-1);
983 :     }
984 :    
985 :     void decoder_bframe(DECODER * dec, Bitstream * bs, int quant, int fcode_forward, int fcode_backward)
986 :     {
987 :    
988 :     uint32_t x, y;
989 :     VECTOR mv, zeromv;
990 :    
991 :     start_timer();
992 :     image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height, dec->width, dec->height, dec->interlacing);
993 :     //image_setedges(&dec->refn[1], dec->edged_width, dec->edged_height, dec->width, dec->height, dec->interlacing);
994 :     stop_edges_timer();
995 :    
996 : chenm001 161
997 : chenm001 156 for (y = 0; y < dec->mb_height; y++)
998 :     {
999 : chenm001 161 // Initialize Pred Motion Vector
1000 :     dec->p_fmv.x = dec->p_fmv.y = dec->p_bmv.x = dec->p_bmv.y = 0;
1001 : chenm001 156 for (x = 0; x < dec->mb_width; x++)
1002 :     {
1003 :     MACROBLOCK * mb = &dec->mbs[y*dec->mb_width + x];
1004 :     MACROBLOCK * last_mb = &dec->last_mbs[y*dec->mb_width + x];
1005 :    
1006 : chenm001 161 mb->mvs[0].x=mb->mvs[0].y=zeromv.x = zeromv.y = mv.x = mv.y = 0;
1007 :    
1008 : chenm001 156 // the last P_VOP is skip macroblock ?
1009 :     if (last_mb->mode == MODE_NOT_CODED){
1010 : chenm001 161 //DEBUG2("Skip MB in B-frame at (X,Y)=!",x,y);
1011 :     mb->mb_type = MODE_FORWARD;
1012 :     mb->cbp = 0;
1013 :     mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;
1014 :     mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;
1015 :     mb->quant = 8;
1016 : chenm001 156
1017 : chenm001 161 decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);
1018 : chenm001 156 continue;
1019 :     }
1020 :    
1021 : chenm001 161 //t=BitstreamShowBits(bs,32);
1022 :    
1023 : chenm001 156 if (!BitstreamGetBit(bs)){ // modb=='0'
1024 : chenm001 161 const uint8_t modb2=BitstreamGetBit(bs);
1025 : chenm001 156
1026 :     mb->mb_type = get_mbtype(bs);
1027 :    
1028 :     if (!modb2){ // modb=='00'
1029 :     mb->cbp = BitstreamGetBits(bs,6);
1030 :     } else {
1031 :     mb->cbp = 0;
1032 :     }
1033 :     if (mb->mb_type && mb->cbp){
1034 :     quant += get_dbquant(bs);
1035 :    
1036 :     if (quant > 31)
1037 :     {
1038 :     quant = 31;
1039 :     }
1040 :     else if (mb->quant < 1)
1041 :     {
1042 :     quant = 1;
1043 :     }
1044 : chenm001 161 } else {
1045 :     quant = 8;
1046 : chenm001 156 }
1047 :     mb->quant = quant;
1048 :     } else {
1049 :     mb->mb_type = MODE_DIRECT_NONE_MV;
1050 :     mb->cbp = 0;
1051 :     }
1052 :    
1053 : chenm001 161 mb->mode = MODE_INTER;
1054 : chenm001 156 //DEBUG1("Switch bm_type=",mb->mb_type);
1055 :    
1056 :     switch(mb->mb_type)
1057 :     {
1058 :     case MODE_DIRECT:
1059 :     get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], 1, zeromv);
1060 :    
1061 :     case MODE_DIRECT_NONE_MV:
1062 :     { // Because this file is a C file not C++ so I use '{' to define var
1063 : chenm001 161 const int64_t TRB=dec->time_pp-dec->time_bp,
1064 : chenm001 156 TRD=dec->time_pp;
1065 :     int i;
1066 :     for(i=0;i<4;i++){
1067 : chenm001 161 mb->mvs[i].x = (int32_t)((TRB * last_mb->mvs[i].x)/TRD+mb->mvs[0].x);
1068 :     mb->b_mvs[i].x = (int32_t)((mb->mvs[0].x==0)?((TRB-TRD)*last_mb->mvs[i].x)/TRD:mb->mvs[i].x-last_mb->mvs[i].x);
1069 :     mb->mvs[i].y = (int32_t)((TRB * last_mb->mvs[i].y)/TRD+mb->mvs[0].y);
1070 :     mb->b_mvs[i].y = (int32_t)((mb->mvs[0].y==0)?((TRB-TRD)*last_mb->mvs[i].y)/TRD:mb->mvs[i].y-last_mb->mvs[i].y);
1071 : chenm001 156 }
1072 :     //DEBUG("B-frame Direct!\n");
1073 :     }
1074 :     mb->mode = MODE_INTER4V;
1075 : chenm001 161 decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0], mb, x, y, mb->cbp, bs);
1076 : chenm001 156 break;
1077 :    
1078 :     case MODE_INTERPOLATE:
1079 :     get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward, dec->p_fmv);
1080 :     dec->p_fmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;
1081 :     dec->p_fmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;
1082 :    
1083 :     get_b_motion_vector(dec, bs, x, y, &mb->b_mvs[0], fcode_backward, dec->p_bmv);
1084 :     dec->p_bmv.x = mb->b_mvs[1].x = mb->b_mvs[2].x = mb->b_mvs[3].x = mb->b_mvs[0].x;
1085 :     dec->p_bmv.y = mb->b_mvs[1].y = mb->b_mvs[2].y = mb->b_mvs[3].y = mb->b_mvs[0].y;
1086 :    
1087 : chenm001 161 decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0], mb, x, y, mb->cbp, bs);
1088 : chenm001 156 //DEBUG("B-frame Bidir!\n");
1089 :     break;
1090 :    
1091 :     case MODE_BACKWARD:
1092 :     get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_backward, dec->p_bmv);
1093 :     dec->p_bmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;
1094 :     dec->p_bmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;
1095 : chenm001 161
1096 : chenm001 156 decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 0);
1097 :     //DEBUG("B-frame Backward!\n");
1098 :     break;
1099 :    
1100 :     case MODE_FORWARD:
1101 :     get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward, dec->p_fmv);
1102 :     dec->p_fmv.x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;
1103 :     dec->p_fmv.y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;
1104 : chenm001 161
1105 : chenm001 156 decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);
1106 : chenm001 161 //DEBUG("B-frame Forward!\n");
1107 : chenm001 156 break;
1108 :    
1109 :     default:
1110 :     DEBUG1("Not support B-frame mb_type =",mb->mb_type);
1111 :     }
1112 :    
1113 :     } // end of FOR
1114 :     }
1115 :     }
1116 :    
1117 :     // swap two MACROBLOCK array
1118 : chenm001 161 void mb_swap(MACROBLOCK **mb1, MACROBLOCK **mb2)
1119 : chenm001 156 {
1120 : chenm001 161 MACROBLOCK *temp=*mb1;
1121 :     *mb1=*mb2;
1122 :     *mb2=temp;
1123 : chenm001 156 }
1124 :    
1125 : Isibaar 3 int decoder_decode(DECODER * dec, XVID_DEC_FRAME * frame)
1126 :     {
1127 : edgomez 78
1128 : Isibaar 3 Bitstream bs;
1129 :     uint32_t rounding;
1130 :     uint32_t quant;
1131 : chenm001 156 uint32_t fcode_forward;
1132 :     uint32_t fcode_backward;
1133 : Isibaar 3 uint32_t intra_dc_threshold;
1134 : chenm001 133 uint32_t vop_type;
1135 : Isibaar 3
1136 :     start_global_timer();
1137 :    
1138 :     BitstreamInit(&bs, frame->bitstream, frame->length);
1139 :    
1140 : chenm001 133 // add by chenm001 <chenm001@163.com>
1141 :     // for support B-frame to reference last 2 frame
1142 : chenm001 156 dec->frames ++;
1143 :     vop_type=BitstreamReadHeaders(&bs, dec, &rounding, &quant, &fcode_forward, &fcode_backward, &intra_dc_threshold);
1144 :    
1145 :     dec->p_bmv.x=dec->p_bmv.y=dec->p_fmv.y=dec->p_fmv.y=0; // init pred vector to 0
1146 : chenm001 133 switch (vop_type)
1147 : Isibaar 3 {
1148 :     case P_VOP :
1149 : chenm001 156 decoder_pframe(dec, &bs, rounding, quant, fcode_forward, intra_dc_threshold);
1150 :     DEBUG1("P_VOP Time=",dec->time);
1151 : Isibaar 3 break;
1152 :    
1153 :     case I_VOP :
1154 :     decoder_iframe(dec, &bs, quant, intra_dc_threshold);
1155 : chenm001 156 DEBUG1("I_VOP Time=",dec->time);
1156 : Isibaar 3 break;
1157 :    
1158 : chenm001 156 case B_VOP :
1159 :     if (dec->time_pp > dec->time_bp){
1160 : chenm001 161 DEBUG1("B_VOP Time=",dec->time);
1161 : chenm001 156 decoder_bframe(dec, &bs, quant, fcode_forward, fcode_backward);
1162 :     } else {
1163 :     DEBUG("broken B-frame!");
1164 :     }
1165 : Isibaar 3 break;
1166 :    
1167 :     case N_VOP : // vop not coded
1168 :     break;
1169 :    
1170 :     default :
1171 :     return XVID_ERR_FAIL;
1172 :     }
1173 :    
1174 :     frame->length = BitstreamPos(&bs) / 8;
1175 :    
1176 : chenm001 156 if (dec->frames >= 1){
1177 :     start_timer();
1178 :     if ((vop_type == I_VOP || vop_type == P_VOP))
1179 :     {
1180 : chenm001 161 image_output(&dec->refn[0], dec->width, dec->height, dec->edged_width,
1181 : chenm001 156 frame->image, frame->stride, frame->colorspace);
1182 :     } else if (vop_type == B_VOP) {
1183 :     image_output(&dec->cur, dec->width, dec->height, dec->edged_width,
1184 :     frame->image, frame->stride, frame->colorspace);
1185 :     }
1186 :     stop_conv_timer();
1187 :     }
1188 :     if (vop_type==I_VOP || vop_type==P_VOP){
1189 :     image_swap(&dec->refn[0], &dec->refn[1]);
1190 :     image_swap(&dec->cur, &dec->refn[0]);
1191 :     // swap MACROBLOCK
1192 : chenm001 161 if (vop_type==P_VOP)
1193 :     mb_swap(&dec->mbs, &dec->last_mbs);
1194 : chenm001 156 }
1195 :    
1196 : Isibaar 3 emms();
1197 :    
1198 :     stop_global_timer();
1199 :    
1200 :     return XVID_ERR_OK;
1201 :     }

No admin address has been configured
ViewVC Help
Powered by ViewVC 1.0.4