[svn] / branches / dev-api-3 / xvidcore / src / decoder.c Repository:
ViewVC logotype

Annotation of /branches/dev-api-3/xvidcore/src/decoder.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 645 - (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 297 * 15.07.2002 fix a bug in B-frame decode at DIRECT mode
36 :     * MinChen <chenm001@163.com>
37 : chenm001 290 * 10.07.2002 added BFRAMES_DEC_DEBUG support
38 :     * Fix a little bug for low_delay flage
39 :     * MinChen <chenm001@163.com>
40 : chenm001 272 * 28.06.2002 added basic resync support to iframe/pframe_decode()
41 : chenm001 290 * 22.06.2002 added primative N_VOP support
42 : chenm001 297 * #define BFRAMES_DEC now enables Minchen's bframe decoder
43 : chenm001 168 * 08.05.2002 add low_delay support for B_VOP decode
44 :     * MinChen <chenm001@163.com>
45 :     * 05.05.2002 fix some B-frame decode problem
46 : chenm001 156 * 02.05.2002 add B-frame decode support(have some problem);
47 :     * MinChen <chenm001@163.com>
48 : chenm001 133 * 22.04.2002 add some B-frame decode support; chenm001 <chenm001@163.com>
49 : h 81 * 29.03.2002 interlacing fix - compensated block wasn't being used when
50 :     * reconstructing blocks, thus artifacts
51 :     * interlacing speedup - used transfers to re-interlace
52 :     * interlaced decoding should be as fast as progressive now
53 : h 69 * 26.03.2002 interlacing support - moved transfers outside decode loop
54 : edgomez 146 * 26.12.2001 decoder_mbinter: dequant/idct moved within if(coded) block
55 :     * 22.12.2001 lock based interpolation
56 :     * 01.12.2001 inital version; (c)2001 peter ross <pross@cs.rmit.edu.au>
57 : Isibaar 3 *
58 : Isibaar 645 * $Id: decoder.c,v 1.37.2.11 2002-11-12 15:53:36 Isibaar Exp $
59 : edgomez 146 *
60 : Isibaar 3 *************************************************************************/
61 :    
62 :     #include <stdlib.h>
63 : edgomez 146 #include <string.h>
64 : Isibaar 3
65 : chenm001 290 #ifdef BFRAMES_DEC_DEBUG
66 :     #define BFRAMES_DEC
67 :     #endif
68 :    
69 : Isibaar 3 #include "xvid.h"
70 :     #include "portab.h"
71 :    
72 :     #include "decoder.h"
73 :     #include "bitstream/bitstream.h"
74 :     #include "bitstream/mbcoding.h"
75 :    
76 :     #include "quant/quant_h263.h"
77 :     #include "quant/quant_mpeg4.h"
78 :     #include "dct/idct.h"
79 :     #include "dct/fdct.h"
80 :     #include "utils/mem_transfer.h"
81 :     #include "image/interpolate8x8.h"
82 :    
83 :     #include "bitstream/mbcoding.h"
84 :     #include "prediction/mbprediction.h"
85 :     #include "utils/timer.h"
86 :     #include "utils/emms.h"
87 : Isibaar 616 #include "motion/motion.h"
88 : Isibaar 3
89 :     #include "image/image.h"
90 :     #include "image/colorspace.h"
91 : Isibaar 41 #include "utils/mem_align.h"
92 : Isibaar 3
93 : edgomez 195 int
94 : suxen_drol 631 decoder_resize(DECODER * dec)
95 : Isibaar 3 {
96 : suxen_drol 631 /* free existing */
97 : Isibaar 3
98 : suxen_drol 631 image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
99 :     image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
100 :     image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
101 :     image_destroy(&dec->refh, dec->edged_width, dec->edged_height);
102 :     image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
103 : Isibaar 3
104 : suxen_drol 631 if (dec->last_mbs)
105 :     xvid_free(dec->last_mbs);
106 :     if (dec->mbs)
107 :     xvid_free(dec->mbs);
108 : Isibaar 3
109 : suxen_drol 631 /* realloc */
110 :    
111 : Isibaar 3 dec->mb_width = (dec->width + 15) / 16;
112 :     dec->mb_height = (dec->height + 15) / 16;
113 :    
114 :     dec->edged_width = 16 * dec->mb_width + 2 * EDGE_SIZE;
115 :     dec->edged_height = 16 * dec->mb_height + 2 * EDGE_SIZE;
116 : edgomez 195
117 :     if (image_create(&dec->cur, dec->edged_width, dec->edged_height)) {
118 : Isibaar 41 xvid_free(dec);
119 : Isibaar 3 return XVID_ERR_MEMORY;
120 :     }
121 :    
122 : edgomez 195 if (image_create(&dec->refn[0], dec->edged_width, dec->edged_height)) {
123 : Isibaar 3 image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
124 : Isibaar 41 xvid_free(dec);
125 : Isibaar 3 return XVID_ERR_MEMORY;
126 :     }
127 : suxen_drol 631
128 : chenm001 133 // add by chenm001 <chenm001@163.com>
129 :     // for support B-frame to reference last 2 frame
130 : edgomez 195 if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height)) {
131 : chenm001 133 image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
132 :     image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
133 :     xvid_free(dec);
134 :     return XVID_ERR_MEMORY;
135 :     }
136 : edgomez 195 if (image_create(&dec->refn[2], dec->edged_width, dec->edged_height)) {
137 : chenm001 156 image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
138 :     image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
139 :     image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
140 :     xvid_free(dec);
141 :     return XVID_ERR_MEMORY;
142 :     }
143 : Isibaar 3
144 : Isibaar 579 if (image_create(&dec->refh, dec->edged_width, dec->edged_height)) {
145 :     image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
146 :     image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
147 :     image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
148 :     image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
149 :     xvid_free(dec);
150 :     return XVID_ERR_MEMORY;
151 :     }
152 :    
153 : edgomez 195 dec->mbs =
154 :     xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
155 :     CACHE_LINE);
156 :     if (dec->mbs == NULL) {
157 : Isibaar 3 image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
158 : chenm001 156 image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
159 :     image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
160 :     image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
161 : Isibaar 579 image_destroy(&dec->refh, dec->edged_width, dec->edged_height);
162 : Isibaar 41 xvid_free(dec);
163 : Isibaar 3 return XVID_ERR_MEMORY;
164 :     }
165 : Isibaar 208 memset(dec->mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);
166 :    
167 : chenm001 156 // add by chenm001 <chenm001@163.com>
168 :     // for skip MB flag
169 : edgomez 195 dec->last_mbs =
170 :     xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
171 :     CACHE_LINE);
172 :     if (dec->last_mbs == NULL) {
173 : chenm001 156 xvid_free(dec->mbs);
174 :     image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
175 :     image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
176 :     image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
177 :     image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
178 : Isibaar 579 image_destroy(&dec->refh, dec->edged_width, dec->edged_height);
179 : chenm001 156 xvid_free(dec);
180 :     return XVID_ERR_MEMORY;
181 :     }
182 : Isibaar 3
183 : Isibaar 208 memset(dec->last_mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);
184 :    
185 : suxen_drol 631 return XVID_ERR_OK;
186 :     }
187 :    
188 :    
189 :     int
190 :     decoder_create(XVID_DEC_PARAM * param)
191 :     {
192 :     DECODER *dec;
193 :    
194 :     dec = xvid_malloc(sizeof(DECODER), CACHE_LINE);
195 :     if (dec == NULL) {
196 :     return XVID_ERR_MEMORY;
197 :     }
198 :     memset(dec, 0, sizeof(DECODER));
199 :    
200 :     param->handle = dec;
201 :    
202 :     dec->width = param->width;
203 :     dec->height = param->height;
204 :    
205 :     image_null(&dec->cur);
206 :     image_null(&dec->refn[0]);
207 :     image_null(&dec->refn[1]);
208 :     image_null(&dec->refn[2]);
209 :     image_null(&dec->refh);
210 :    
211 :     dec->mbs = NULL;
212 :     dec->last_mbs = NULL;
213 :    
214 : Isibaar 3 init_timer();
215 :    
216 : chenm001 156 // add by chenm001 <chenm001@163.com>
217 :     // for support B-frame to save reference frame's time
218 :     dec->frames = -1;
219 :     dec->time = dec->time_base = dec->last_time_base = 0;
220 : suxen_drol 631 dec->low_delay = 0;
221 : edgomez 195
222 : suxen_drol 631 dec->fixed_dimensions = (dec->width > 0 && dec->height > 0);
223 :    
224 :     if (dec->fixed_dimensions)
225 :     return decoder_resize(dec);
226 :     else
227 :     return XVID_ERR_OK;
228 : Isibaar 3 }
229 :    
230 :    
231 : edgomez 195 int
232 :     decoder_destroy(DECODER * dec)
233 : Isibaar 3 {
234 : chenm001 156 xvid_free(dec->last_mbs);
235 : Isibaar 41 xvid_free(dec->mbs);
236 : chenm001 133 image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
237 : chenm001 156 image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
238 :     image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
239 : Isibaar 579 image_destroy(&dec->refh, dec->edged_width, dec->edged_height);
240 : Isibaar 3 image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
241 : Isibaar 41 xvid_free(dec);
242 : Isibaar 3
243 :     write_timer();
244 :     return XVID_ERR_OK;
245 :     }
246 :    
247 :    
248 :    
249 : edgomez 195 static const int32_t dquant_table[4] = {
250 : Isibaar 3 -1, -2, 1, 2
251 :     };
252 :    
253 :    
254 : chenm001 156
255 :    
256 : Isibaar 3 // decode an intra macroblock
257 :    
258 : edgomez 195 void
259 :     decoder_mbintra(DECODER * dec,
260 :     MACROBLOCK * pMB,
261 :     const uint32_t x_pos,
262 :     const uint32_t y_pos,
263 :     const uint32_t acpred_flag,
264 :     const uint32_t cbp,
265 :     Bitstream * bs,
266 :     const uint32_t quant,
267 : chenm001 272 const uint32_t intra_dc_threshold,
268 :     const unsigned int bound)
269 : Isibaar 3 {
270 : edgomez 78
271 :     DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
272 : edgomez 195 DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
273 : edgomez 78
274 : h 81 uint32_t stride = dec->edged_width;
275 :     uint32_t stride2 = stride / 2;
276 :     uint32_t next_block = stride * 8;
277 : h 69 uint32_t i;
278 :     uint32_t iQuant = pMB->quant;
279 :     uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
280 :    
281 : edgomez 78 pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
282 : h 81 pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
283 :     pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
284 : h 69
285 : edgomez 195 memset(block, 0, 6 * 64 * sizeof(int16_t)); // clear
286 : h 69
287 : edgomez 195 for (i = 0; i < 6; i++) {
288 : h 69 uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);
289 : Isibaar 3 int16_t predictors[8];
290 :     int start_coeff;
291 :    
292 :     start_timer();
293 : edgomez 195 predict_acdc(dec->mbs, x_pos, y_pos, dec->mb_width, i, &block[i * 64],
294 : chenm001 272 iQuant, iDcScaler, predictors, bound);
295 : edgomez 195 if (!acpred_flag) {
296 : h 69 pMB->acpred_directions[i] = 0;
297 : Isibaar 3 }
298 :     stop_prediction_timer();
299 :    
300 : edgomez 195 if (quant < intra_dc_threshold) {
301 : Isibaar 3 int dc_size;
302 :     int dc_dif;
303 :    
304 : edgomez 195 dc_size = i < 4 ? get_dc_size_lum(bs) : get_dc_size_chrom(bs);
305 :     dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0;
306 : Isibaar 3
307 : edgomez 195 if (dc_size > 8) {
308 :     BitstreamSkip(bs, 1); // marker
309 : Isibaar 3 }
310 : edgomez 195
311 :     block[i * 64 + 0] = dc_dif;
312 : Isibaar 3 start_coeff = 1;
313 : chenm001 272
314 :     DPRINTF(DPRINTF_COEFF,"block[0] %i", dc_dif);
315 : edgomez 195 } else {
316 : Isibaar 3 start_coeff = 0;
317 :     }
318 :    
319 :     start_timer();
320 : edgomez 195 if (cbp & (1 << (5 - i))) // coded
321 : Isibaar 3 {
322 : h 543 int direction = dec->alternate_vertical_scan ?
323 :     2 : pMB->acpred_directions[i];
324 :    
325 :     get_intra_block(bs, &block[i * 64], direction, start_coeff);
326 : Isibaar 3 }
327 :     stop_coding_timer();
328 :    
329 :     start_timer();
330 : edgomez 195 add_acdc(pMB, i, &block[i * 64], iDcScaler, predictors);
331 : Isibaar 3 stop_prediction_timer();
332 :    
333 :     start_timer();
334 : edgomez 195 if (dec->quant_type == 0) {
335 :     dequant_intra(&data[i * 64], &block[i * 64], iQuant, iDcScaler);
336 :     } else {
337 :     dequant4_intra(&data[i * 64], &block[i * 64], iQuant, iDcScaler);
338 : Isibaar 3 }
339 :     stop_iquant_timer();
340 :    
341 :     start_timer();
342 : edgomez 195 idct(&data[i * 64]);
343 : Isibaar 3 stop_idct_timer();
344 : h 69 }
345 : Isibaar 3
346 : edgomez 195 if (dec->interlacing && pMB->field_dct) {
347 : h 81 next_block = stride;
348 :     stride *= 2;
349 : Isibaar 3 }
350 : h 69
351 :     start_timer();
352 : edgomez 195 transfer_16to8copy(pY_Cur, &data[0 * 64], stride);
353 :     transfer_16to8copy(pY_Cur + 8, &data[1 * 64], stride);
354 :     transfer_16to8copy(pY_Cur + next_block, &data[2 * 64], stride);
355 :     transfer_16to8copy(pY_Cur + 8 + next_block, &data[3 * 64], stride);
356 :     transfer_16to8copy(pU_Cur, &data[4 * 64], stride2);
357 :     transfer_16to8copy(pV_Cur, &data[5 * 64], stride2);
358 : h 69 stop_transfer_timer();
359 : Isibaar 3 }
360 :    
361 :    
362 :    
363 :    
364 :    
365 :     #define SIGN(X) (((X)>0)?1:-1)
366 :     #define ABS(X) (((X)>0)?(X):-(X))
367 :    
368 :     // decode an inter macroblock
369 :    
370 : edgomez 195 void
371 :     decoder_mbinter(DECODER * dec,
372 :     const MACROBLOCK * pMB,
373 :     const uint32_t x_pos,
374 :     const uint32_t y_pos,
375 :     const uint32_t acpred_flag,
376 :     const uint32_t cbp,
377 :     Bitstream * bs,
378 :     const uint32_t quant,
379 :     const uint32_t rounding)
380 : Isibaar 3 {
381 : h 69
382 : edgomez 195 DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
383 : edgomez 78 DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
384 :    
385 : h 81 uint32_t stride = dec->edged_width;
386 :     uint32_t stride2 = stride / 2;
387 :     uint32_t next_block = stride * 8;
388 : edgomez 78 uint32_t i;
389 :     uint32_t iQuant = pMB->quant;
390 : h 69 uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
391 : Isibaar 3 int uv_dx, uv_dy;
392 :    
393 : edgomez 78 pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
394 : h 81 pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
395 :     pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
396 : h 69
397 : edgomez 195 if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q) {
398 : h 69 uv_dx = pMB->mvs[0].x;
399 :     uv_dy = pMB->mvs[0].y;
400 : Isibaar 3
401 : Isibaar 333 if (dec->quarterpel)
402 :     {
403 : Isibaar 617 uv_dx /= 2;
404 :     uv_dy /= 2;
405 : Isibaar 333 }
406 :    
407 : Isibaar 617 uv_dx = (uv_dx >> 1) + roundtab_79[uv_dx & 0x3];
408 :     uv_dy = (uv_dy >> 1) + roundtab_79[uv_dy & 0x3];
409 :    
410 : Isibaar 588 start_timer();
411 :     if(dec->quarterpel) {
412 :     interpolate16x16_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
413 :     dec->refh.y + 128, 16*x_pos, 16*y_pos,
414 :     pMB->mvs[0].x, pMB->mvs[0].y, stride, rounding);
415 :     }
416 :     else {
417 :     interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos,
418 :     pMB->mvs[0].x, pMB->mvs[0].y, stride, rounding);
419 :     interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos,
420 :     pMB->mvs[1].x, pMB->mvs[1].y, stride, rounding);
421 :     interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos + 8,
422 :     pMB->mvs[2].x, pMB->mvs[2].y, stride, rounding);
423 :     interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos + 8,
424 :     pMB->mvs[3].x, pMB->mvs[3].y, stride, rounding);
425 :     }
426 :    
427 :     interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8 * x_pos, 8 * y_pos,
428 :     uv_dx, uv_dy, stride2, rounding);
429 :     interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8 * x_pos, 8 * y_pos,
430 :     uv_dx, uv_dy, stride2, rounding);
431 :     stop_comp_timer();
432 :    
433 : edgomez 195 } else {
434 : Isibaar 3 int sum;
435 : Isibaar 616
436 :     if(dec->quarterpel)
437 :     sum = (pMB->mvs[0].x / 2) + (pMB->mvs[1].x / 2) + (pMB->mvs[2].x / 2) + (pMB->mvs[3].x / 2);
438 :     else
439 :     sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
440 : Isibaar 3
441 : Isibaar 616 uv_dx = (sum >> 3) + roundtab_76[sum & 0xf];
442 : Isibaar 333
443 : Isibaar 616 if(dec->quarterpel)
444 :     sum = (pMB->mvs[0].y / 2) + (pMB->mvs[1].y / 2) + (pMB->mvs[2].y / 2) + (pMB->mvs[3].y / 2);
445 :     else
446 :     sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
447 : Isibaar 333
448 : Isibaar 616 uv_dy = (sum >> 3) + roundtab_76[sum & 0xf];
449 : Isibaar 333
450 : Isibaar 588 start_timer();
451 :     if(dec->quarterpel) {
452 :     interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
453 :     dec->refh.y + 128, 16*x_pos, 16*y_pos,
454 :     pMB->mvs[0].x, pMB->mvs[0].y, stride, rounding);
455 :     interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
456 :     dec->refh.y + 128, 16*x_pos + 8, 16*y_pos,
457 :     pMB->mvs[1].x, pMB->mvs[1].y, stride, rounding);
458 :     interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
459 :     dec->refh.y + 128, 16*x_pos, 16*y_pos + 8,
460 :     pMB->mvs[2].x, pMB->mvs[2].y, stride, rounding);
461 :     interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
462 :     dec->refh.y + 128, 16*x_pos + 8, 16*y_pos + 8,
463 :     pMB->mvs[3].x, pMB->mvs[3].y, stride, rounding);
464 :     }
465 :     else {
466 :     interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos,
467 : Isibaar 333 pMB->mvs[0].x, pMB->mvs[0].y, stride, rounding);
468 : Isibaar 588 interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos,
469 : Isibaar 333 pMB->mvs[1].x, pMB->mvs[1].y, stride, rounding);
470 : Isibaar 588 interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos + 8,
471 : Isibaar 333 pMB->mvs[2].x, pMB->mvs[2].y, stride, rounding);
472 : Isibaar 588 interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos + 8,
473 : Isibaar 333 pMB->mvs[3].x, pMB->mvs[3].y, stride, rounding);
474 : Isibaar 588 }
475 :    
476 :     interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8 * x_pos, 8 * y_pos,
477 :     uv_dx, uv_dy, stride2, rounding);
478 :     interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8 * x_pos, 8 * y_pos,
479 :     uv_dx, uv_dy, stride2, rounding);
480 :     stop_comp_timer();
481 : Isibaar 333 }
482 :    
483 : edgomez 195 for (i = 0; i < 6; i++) {
484 : h 543 int direction = dec->alternate_vertical_scan ? 2 : 0;
485 :    
486 : edgomez 195 if (cbp & (1 << (5 - i))) // coded
487 : Isibaar 3 {
488 : edgomez 195 memset(&block[i * 64], 0, 64 * sizeof(int16_t)); // clear
489 : Isibaar 3
490 :     start_timer();
491 : h 543 get_inter_block(bs, &block[i * 64], direction);
492 : Isibaar 3 stop_coding_timer();
493 :    
494 :     start_timer();
495 : edgomez 195 if (dec->quant_type == 0) {
496 :     dequant_inter(&data[i * 64], &block[i * 64], iQuant);
497 :     } else {
498 :     dequant4_inter(&data[i * 64], &block[i * 64], iQuant);
499 : Isibaar 3 }
500 :     stop_iquant_timer();
501 :    
502 :     start_timer();
503 : edgomez 195 idct(&data[i * 64]);
504 : Isibaar 3 stop_idct_timer();
505 :     }
506 :     }
507 : h 69
508 : edgomez 195 if (dec->interlacing && pMB->field_dct) {
509 : h 81 next_block = stride;
510 :     stride *= 2;
511 : h 69 }
512 :    
513 :     start_timer();
514 :     if (cbp & 32)
515 : edgomez 195 transfer_16to8add(pY_Cur, &data[0 * 64], stride);
516 : h 69 if (cbp & 16)
517 : edgomez 195 transfer_16to8add(pY_Cur + 8, &data[1 * 64], stride);
518 : h 69 if (cbp & 8)
519 : edgomez 195 transfer_16to8add(pY_Cur + next_block, &data[2 * 64], stride);
520 : h 69 if (cbp & 4)
521 : edgomez 195 transfer_16to8add(pY_Cur + 8 + next_block, &data[3 * 64], stride);
522 : h 69 if (cbp & 2)
523 : edgomez 195 transfer_16to8add(pU_Cur, &data[4 * 64], stride2);
524 : h 69 if (cbp & 1)
525 : edgomez 195 transfer_16to8add(pV_Cur, &data[5 * 64], stride2);
526 : h 69 stop_transfer_timer();
527 : Isibaar 3 }
528 :    
529 :    
530 : edgomez 195 void
531 :     decoder_iframe(DECODER * dec,
532 :     Bitstream * bs,
533 : suxen_drol 631 int reduced_resolution,
534 : edgomez 195 int quant,
535 :     int intra_dc_threshold)
536 : Isibaar 3 {
537 : chenm001 272 uint32_t bound;
538 : Isibaar 3 uint32_t x, y;
539 : suxen_drol 631 int mb_width = dec->mb_width;
540 :     int mb_height = dec->mb_height;
541 :    
542 :     if (reduced_resolution)
543 :     {
544 :     mb_width /= 2;
545 :     mb_height /= 2;
546 :     }
547 : Isibaar 3
548 : chenm001 272 bound = 0;
549 :    
550 : suxen_drol 631 for (y = 0; y < mb_height; y++) {
551 :     for (x = 0; x < mb_width; x++) {
552 : chenm001 272 MACROBLOCK *mb;
553 : Isibaar 3 uint32_t mcbpc;
554 :     uint32_t cbpc;
555 :     uint32_t acpred_flag;
556 :     uint32_t cbpy;
557 :     uint32_t cbp;
558 :    
559 : chenm001 272 while (BitstreamShowBits(bs, 9) == 1)
560 :     BitstreamSkip(bs, 9);
561 :    
562 :     if (check_resync_marker(bs, 0))
563 :     {
564 : suxen_drol 631 bound = read_video_packet_header(bs, dec, 0,
565 :     &quant, NULL, NULL, &intra_dc_threshold);
566 : chenm001 272 x = bound % dec->mb_width;
567 :     y = bound / dec->mb_width;
568 :     }
569 :     mb = &dec->mbs[y * dec->mb_width + x];
570 :    
571 :     DPRINTF(DPRINTF_MB, "macroblock (%i,%i) %08x", x, y, BitstreamShowBits(bs, 32));
572 :    
573 : Isibaar 3 mcbpc = get_mcbpc_intra(bs);
574 :     mb->mode = mcbpc & 7;
575 :     cbpc = (mcbpc >> 4);
576 :    
577 :     acpred_flag = BitstreamGetBit(bs);
578 :    
579 :     cbpy = get_cbpy(bs, 1);
580 :     cbp = (cbpy << 2) | cbpc;
581 :    
582 : edgomez 195 if (mb->mode == MODE_INTRA_Q) {
583 :     quant += dquant_table[BitstreamGetBits(bs, 2)];
584 :     if (quant > 31) {
585 : Isibaar 3 quant = 31;
586 : edgomez 195 } else if (quant < 1) {
587 : Isibaar 3 quant = 1;
588 :     }
589 :     }
590 :     mb->quant = quant;
591 : chenm001 297 mb->mvs[0].x = mb->mvs[0].y =
592 :     mb->mvs[1].x = mb->mvs[1].y =
593 :     mb->mvs[2].x = mb->mvs[2].y =
594 :     mb->mvs[3].x = mb->mvs[3].y =0;
595 : Isibaar 3
596 : edgomez 195 if (dec->interlacing) {
597 : h 69 mb->field_dct = BitstreamGetBit(bs);
598 :     DEBUG1("deci: field_dct: ", mb->field_dct);
599 :     }
600 :    
601 : edgomez 195 decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant,
602 : chenm001 272 intra_dc_threshold, bound);
603 : suxen_drol 631
604 : Isibaar 3 }
605 : albeu 315 if(dec->out_frm)
606 :     output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,0,y,dec->mb_width);
607 : Isibaar 3 }
608 : edgomez 78
609 : Isibaar 3 }
610 :    
611 :    
612 : edgomez 195 void
613 :     get_motion_vector(DECODER * dec,
614 :     Bitstream * bs,
615 :     int x,
616 :     int y,
617 :     int k,
618 :     VECTOR * mv,
619 : chenm001 272 int fcode,
620 :     const int bound)
621 : Isibaar 3 {
622 : edgomez 78
623 : Isibaar 3 int scale_fac = 1 << (fcode - 1);
624 :     int high = (32 * scale_fac) - 1;
625 :     int low = ((-32) * scale_fac);
626 :     int range = (64 * scale_fac);
627 :    
628 : chenm001 272 VECTOR pmv;
629 : Isibaar 3 int mv_x, mv_y;
630 :    
631 : chenm001 272 pmv = get_pmv2(dec->mbs, dec->mb_width, bound, x, y, k);
632 : Isibaar 3
633 :     mv_x = get_mv(bs, fcode);
634 :     mv_y = get_mv(bs, fcode);
635 : edgomez 195
636 : chenm001 272 DPRINTF(DPRINTF_MV,"mv_diff (%i,%i) pred (%i,%i)", mv_x, mv_y, pmv.x, pmv.y);
637 : suxen_drol 252
638 : chenm001 272 mv_x += pmv.x;
639 :     mv_y += pmv.y;
640 :    
641 : edgomez 195 if (mv_x < low) {
642 : Isibaar 3 mv_x += range;
643 : edgomez 195 } else if (mv_x > high) {
644 : Isibaar 3 mv_x -= range;
645 :     }
646 :    
647 : edgomez 195 if (mv_y < low) {
648 : Isibaar 3 mv_y += range;
649 : edgomez 195 } else if (mv_y > high) {
650 : Isibaar 3 mv_y -= range;
651 :     }
652 :    
653 :     mv->x = mv_x;
654 :     mv->y = mv_y;
655 : suxen_drol 631 }
656 : Isibaar 3
657 : suxen_drol 631
658 :    
659 :     static __inline int gmc_sanitize(int value, int quarterpel, int fcode)
660 :     {
661 :     int length = 1 << (fcode+4);
662 :    
663 :     if (quarterpel) value *= 2;
664 :    
665 :     if (value < -length)
666 :     return -length;
667 :     else if (value >= length)
668 :     return length-1;
669 :     else return value;
670 : Isibaar 3 }
671 :    
672 :    
673 : suxen_drol 631 /* for P_VOP set gmc_mv to NULL */
674 : edgomez 195 void
675 :     decoder_pframe(DECODER * dec,
676 :     Bitstream * bs,
677 :     int rounding,
678 : suxen_drol 631 int reduced_resolution,
679 : edgomez 195 int quant,
680 :     int fcode,
681 : suxen_drol 631 int intra_dc_threshold,
682 :     VECTOR * gmc_mv)
683 : Isibaar 3 {
684 : edgomez 78
685 : Isibaar 3 uint32_t x, y;
686 : chenm001 272 uint32_t bound;
687 : albeu 315 int cp_mb, st_mb;
688 : Isibaar 3
689 :     start_timer();
690 : edgomez 195 image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height,
691 : h 543 dec->width, dec->height);
692 : Isibaar 3 stop_edges_timer();
693 :    
694 : chenm001 272 bound = 0;
695 :    
696 : edgomez 195 for (y = 0; y < dec->mb_height; y++) {
697 : albeu 315 cp_mb = st_mb = 0;
698 : edgomez 195 for (x = 0; x < dec->mb_width; x++) {
699 : chenm001 272 MACROBLOCK *mb;
700 : Isibaar 3
701 : chenm001 272 // skip stuffing
702 :     while (BitstreamShowBits(bs, 10) == 1)
703 :     BitstreamSkip(bs, 10);
704 :    
705 :     if (check_resync_marker(bs, fcode - 1))
706 :     {
707 : suxen_drol 631 bound = read_video_packet_header(bs, dec, fcode - 1,
708 :     &quant, &fcode, NULL, &intra_dc_threshold);
709 : chenm001 272 x = bound % dec->mb_width;
710 :     y = bound / dec->mb_width;
711 :     }
712 :     mb = &dec->mbs[y * dec->mb_width + x];
713 :    
714 :     DPRINTF(DPRINTF_MB, "macroblock (%i,%i) %08x", x, y, BitstreamShowBits(bs, 32));
715 :    
716 : edgomez 195 //if (!(dec->mb_skip[y*dec->mb_width + x]=BitstreamGetBit(bs))) // not_coded
717 :     if (!(BitstreamGetBit(bs))) // not_coded
718 : Isibaar 3 {
719 :     uint32_t mcbpc;
720 :     uint32_t cbpc;
721 :     uint32_t acpred_flag;
722 :     uint32_t cbpy;
723 :     uint32_t cbp;
724 :     uint32_t intra;
725 : suxen_drol 631 int mcsel = 0; // mcsel: '0'=local motion, '1'=GMC
726 : Isibaar 3
727 : albeu 315 cp_mb++;
728 : Isibaar 3 mcbpc = get_mcbpc_inter(bs);
729 :     mb->mode = mcbpc & 7;
730 :     cbpc = (mcbpc >> 4);
731 : chenm001 272
732 :     DPRINTF(DPRINTF_MB, "mode %i", mb->mode);
733 :     DPRINTF(DPRINTF_MB, "cbpc %i", cbpc);
734 : edgomez 12 acpred_flag = 0;
735 : Isibaar 3
736 :     intra = (mb->mode == MODE_INTRA || mb->mode == MODE_INTRA_Q);
737 : edgomez 195
738 :     if (intra) {
739 : Isibaar 3 acpred_flag = BitstreamGetBit(bs);
740 :     }
741 :    
742 : suxen_drol 631 if (gmc_mv && (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q))
743 :     {
744 :     mcsel = BitstreamGetBit(bs);
745 :     }
746 :    
747 : Isibaar 3 cbpy = get_cbpy(bs, intra);
748 : chenm001 272 DPRINTF(DPRINTF_MB, "cbpy %i", cbpy);
749 :    
750 : Isibaar 3 cbp = (cbpy << 2) | cbpc;
751 :    
752 : edgomez 195 if (mb->mode == MODE_INTER_Q || mb->mode == MODE_INTRA_Q) {
753 : chenm001 272 int dquant = dquant_table[BitstreamGetBits(bs, 2)];
754 :     DPRINTF(DPRINTF_MB, "dquant %i", dquant);
755 :     quant += dquant;
756 : edgomez 195 if (quant > 31) {
757 : Isibaar 3 quant = 31;
758 : chenm001 272 } else if (quant < 1) {
759 : Isibaar 3 quant = 1;
760 :     }
761 : chenm001 272 DPRINTF(DPRINTF_MB, "quant %i", quant);
762 : Isibaar 3 }
763 :     mb->quant = quant;
764 : h 69
765 : edgomez 195 if (dec->interlacing) {
766 : h 388 if (cbp || intra) {
767 :     mb->field_dct = BitstreamGetBit(bs);
768 :     DEBUG1("decp: field_dct: ", mb->field_dct);
769 :     }
770 : Isibaar 3
771 : edgomez 195 if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
772 : h 69 mb->field_pred = BitstreamGetBit(bs);
773 :     DEBUG1("decp: field_pred: ", mb->field_pred);
774 :    
775 : edgomez 195 if (mb->field_pred) {
776 : h 69 mb->field_for_top = BitstreamGetBit(bs);
777 :     DEBUG1("decp: field_for_top: ", mb->field_for_top);
778 :     mb->field_for_bot = BitstreamGetBit(bs);
779 :     DEBUG1("decp: field_for_bot: ", mb->field_for_bot);
780 :     }
781 :     }
782 : Isibaar 3 }
783 : h 69
784 : edgomez 195 if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
785 : suxen_drol 631
786 :     if (mcsel)
787 :     {
788 :     mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = gmc_sanitize(gmc_mv[0].x, dec->quarterpel, fcode);
789 :     mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = gmc_sanitize(gmc_mv[0].y, dec->quarterpel, fcode);
790 :    
791 :     } else if (dec->interlacing && mb->field_pred) {
792 : edgomez 195 get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0],
793 : chenm001 272 fcode, bound);
794 : edgomez 195 get_motion_vector(dec, bs, x, y, 0, &mb->mvs[1],
795 : chenm001 272 fcode, bound);
796 : edgomez 195 } else {
797 :     get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0],
798 : chenm001 272 fcode, bound);
799 : edgomez 195 mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
800 :     mb->mvs[0].x;
801 :     mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
802 :     mb->mvs[0].y;
803 : h 69 }
804 : chl 339 } else if (mb->mode == MODE_INTER4V ) {
805 :    
806 : chenm001 272 get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode, bound);
807 :     get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode, bound);
808 :     get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode, bound);
809 :     get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode, bound);
810 : edgomez 195 } else // MODE_INTRA, MODE_INTRA_Q
811 : Isibaar 3 {
812 : edgomez 195 mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
813 :     0;
814 :     mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
815 :     0;
816 :     decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant,
817 : chenm001 272 intra_dc_threshold, bound);
818 : Isibaar 3 continue;
819 :     }
820 :    
821 : edgomez 195 decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant,
822 :     rounding);
823 : suxen_drol 631
824 :     }
825 :     else if (gmc_mv) /* not coded S_VOP macroblock */
826 : Isibaar 3 {
827 : chenm001 156 mb->mode = MODE_NOT_CODED;
828 : suxen_drol 631 mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = gmc_sanitize(gmc_mv[0].x, dec->quarterpel, fcode);
829 :     mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = gmc_sanitize(gmc_mv[0].y, dec->quarterpel, fcode);
830 :     decoder_mbinter(dec, mb, x, y, 0, 0, bs, quant, rounding);
831 :     }
832 :     else /* not coded P_VOP macroblock */
833 :     {
834 :     mb->mode = MODE_NOT_CODED;
835 : Isibaar 3 mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;
836 :     mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;
837 :     // copy macroblock directly from ref to cur
838 :    
839 :     start_timer();
840 :    
841 : edgomez 195 transfer8x8_copy(dec->cur.y + (16 * y) * dec->edged_width +
842 :     (16 * x),
843 :     dec->refn[0].y + (16 * y) * dec->edged_width +
844 :     (16 * x), dec->edged_width);
845 : Isibaar 3
846 : edgomez 195 transfer8x8_copy(dec->cur.y + (16 * y) * dec->edged_width +
847 :     (16 * x + 8),
848 :     dec->refn[0].y + (16 * y) * dec->edged_width +
849 :     (16 * x + 8), dec->edged_width);
850 : Isibaar 3
851 : edgomez 195 transfer8x8_copy(dec->cur.y + (16 * y + 8) * dec->edged_width +
852 :     (16 * x),
853 :     dec->refn[0].y + (16 * y +
854 :     8) * dec->edged_width +
855 :     (16 * x), dec->edged_width);
856 : Isibaar 3
857 : edgomez 195 transfer8x8_copy(dec->cur.y + (16 * y + 8) * dec->edged_width +
858 :     (16 * x + 8),
859 :     dec->refn[0].y + (16 * y +
860 :     8) * dec->edged_width +
861 :     (16 * x + 8), dec->edged_width);
862 : Isibaar 3
863 : edgomez 195 transfer8x8_copy(dec->cur.u + (8 * y) * dec->edged_width / 2 +
864 :     (8 * x),
865 :     dec->refn[0].u +
866 :     (8 * y) * dec->edged_width / 2 + (8 * x),
867 :     dec->edged_width / 2);
868 : Isibaar 3
869 : edgomez 195 transfer8x8_copy(dec->cur.v + (8 * y) * dec->edged_width / 2 +
870 :     (8 * x),
871 :     dec->refn[0].v +
872 :     (8 * y) * dec->edged_width / 2 + (8 * x),
873 :     dec->edged_width / 2);
874 : suxen_drol 631
875 : Isibaar 3 stop_transfer_timer();
876 : suxen_drol 631
877 : albeu 315 if(dec->out_frm && cp_mb > 0) {
878 :     output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,st_mb,y,cp_mb);
879 :     cp_mb = 0;
880 :     }
881 :     st_mb = x+1;
882 : Isibaar 3 }
883 :     }
884 : albeu 315 if(dec->out_frm && cp_mb > 0)
885 :     output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,st_mb,y,cp_mb);
886 : Isibaar 3 }
887 :     }
888 :    
889 : chenm001 156
890 :     // add by MinChen <chenm001@163.com>
891 :     // decode B-frame motion vector
892 : edgomez 195 void
893 :     get_b_motion_vector(DECODER * dec,
894 :     Bitstream * bs,
895 :     int x,
896 :     int y,
897 :     VECTOR * mv,
898 :     int fcode,
899 :     const VECTOR pmv)
900 : chenm001 156 {
901 :     int scale_fac = 1 << (fcode - 1);
902 :     int high = (32 * scale_fac) - 1;
903 :     int low = ((-32) * scale_fac);
904 :     int range = (64 * scale_fac);
905 :    
906 :     int mv_x, mv_y;
907 :     int pmv_x, pmv_y;
908 :    
909 :     pmv_x = pmv.x;
910 :     pmv_y = pmv.y;
911 :    
912 :     mv_x = get_mv(bs, fcode);
913 :     mv_y = get_mv(bs, fcode);
914 : edgomez 195
915 : chenm001 156 mv_x += pmv_x;
916 :     mv_y += pmv_y;
917 :    
918 : edgomez 195 if (mv_x < low) {
919 : chenm001 156 mv_x += range;
920 : edgomez 195 } else if (mv_x > high) {
921 : chenm001 156 mv_x -= range;
922 :     }
923 :    
924 : edgomez 195 if (mv_y < low) {
925 : chenm001 156 mv_y += range;
926 : edgomez 195 } else if (mv_y > high) {
927 : chenm001 156 mv_y -= range;
928 :     }
929 :    
930 :     mv->x = mv_x;
931 :     mv->y = mv_y;
932 :     }
933 :    
934 :    
935 :     // add by MinChen <chenm001@163.com>
936 :     // decode an B-frame forward & backward inter macroblock
937 : edgomez 195 void
938 :     decoder_bf_mbinter(DECODER * dec,
939 :     const MACROBLOCK * pMB,
940 :     const uint32_t x_pos,
941 :     const uint32_t y_pos,
942 :     const uint32_t cbp,
943 :     Bitstream * bs,
944 :     const uint32_t quant,
945 :     const uint8_t ref)
946 : chenm001 156 {
947 :    
948 : edgomez 195 DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
949 : chenm001 156 DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
950 :    
951 :     uint32_t stride = dec->edged_width;
952 :     uint32_t stride2 = stride / 2;
953 :     uint32_t next_block = stride * 8;
954 :     uint32_t i;
955 :     uint32_t iQuant = pMB->quant;
956 :     uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
957 :     int uv_dx, uv_dy;
958 :    
959 :     pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
960 :     pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
961 :     pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
962 :    
963 : chenm001 297
964 : edgomez 195 if (!(pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) {
965 : chenm001 156 uv_dx = pMB->mvs[0].x;
966 :     uv_dy = pMB->mvs[0].y;
967 :    
968 :     uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
969 :     uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
970 : edgomez 195 } else {
971 : chenm001 156 int sum;
972 : edgomez 195
973 : chenm001 156 sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
974 : edgomez 195 uv_dx =
975 :     (sum ==
976 :     0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
977 :     (ABS(sum) / 16) * 2));
978 : chenm001 156
979 :     sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
980 : edgomez 195 uv_dy =
981 :     (sum ==
982 :     0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
983 :     (ABS(sum) / 16) * 2));
984 : chenm001 156 }
985 :    
986 :     start_timer();
987 : edgomez 195 interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos, 16 * y_pos,
988 :     pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
989 :     interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos + 8,
990 :     16 * y_pos, pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
991 :     interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos,
992 :     16 * y_pos + 8, pMB->mvs[2].x, pMB->mvs[2].y, stride,
993 :     0);
994 :     interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16 * x_pos + 8,
995 :     16 * y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,
996 :     0);
997 :     interpolate8x8_switch(dec->cur.u, dec->refn[ref].u, 8 * x_pos, 8 * y_pos,
998 :     uv_dx, uv_dy, stride2, 0);
999 :     interpolate8x8_switch(dec->cur.v, dec->refn[ref].v, 8 * x_pos, 8 * y_pos,
1000 :     uv_dx, uv_dy, stride2, 0);
1001 : chenm001 156 stop_comp_timer();
1002 :    
1003 : edgomez 195 for (i = 0; i < 6; i++) {
1004 : h 543 int direction = dec->alternate_vertical_scan ? 2 : 0;
1005 :    
1006 : edgomez 195 if (cbp & (1 << (5 - i))) // coded
1007 : chenm001 156 {
1008 : edgomez 195 memset(&block[i * 64], 0, 64 * sizeof(int16_t)); // clear
1009 : chenm001 156
1010 :     start_timer();
1011 : h 543 get_inter_block(bs, &block[i * 64], direction);
1012 : chenm001 156 stop_coding_timer();
1013 :    
1014 :     start_timer();
1015 : edgomez 195 if (dec->quant_type == 0) {
1016 :     dequant_inter(&data[i * 64], &block[i * 64], iQuant);
1017 :     } else {
1018 :     dequant4_inter(&data[i * 64], &block[i * 64], iQuant);
1019 : chenm001 156 }
1020 :     stop_iquant_timer();
1021 :    
1022 :     start_timer();
1023 : edgomez 195 idct(&data[i * 64]);
1024 : chenm001 156 stop_idct_timer();
1025 :     }
1026 :     }
1027 :    
1028 : edgomez 195 if (dec->interlacing && pMB->field_dct) {
1029 : chenm001 156 next_block = stride;
1030 :     stride *= 2;
1031 :     }
1032 :    
1033 :     start_timer();
1034 :     if (cbp & 32)
1035 : edgomez 195 transfer_16to8add(pY_Cur, &data[0 * 64], stride);
1036 : chenm001 156 if (cbp & 16)
1037 : edgomez 195 transfer_16to8add(pY_Cur + 8, &data[1 * 64], stride);
1038 : chenm001 156 if (cbp & 8)
1039 : edgomez 195 transfer_16to8add(pY_Cur + next_block, &data[2 * 64], stride);
1040 : chenm001 156 if (cbp & 4)
1041 : edgomez 195 transfer_16to8add(pY_Cur + 8 + next_block, &data[3 * 64], stride);
1042 : chenm001 156 if (cbp & 2)
1043 : edgomez 195 transfer_16to8add(pU_Cur, &data[4 * 64], stride2);
1044 : chenm001 156 if (cbp & 1)
1045 : edgomez 195 transfer_16to8add(pV_Cur, &data[5 * 64], stride2);
1046 : chenm001 156 stop_transfer_timer();
1047 :     }
1048 :    
1049 :     // add by MinChen <chenm001@163.com>
1050 :     // decode an B-frame direct & inter macroblock
1051 : edgomez 195 void
1052 :     decoder_bf_interpolate_mbinter(DECODER * dec,
1053 :     IMAGE forward,
1054 :     IMAGE backward,
1055 :     const MACROBLOCK * pMB,
1056 :     const uint32_t x_pos,
1057 :     const uint32_t y_pos,
1058 :     Bitstream * bs)
1059 : chenm001 156 {
1060 :    
1061 : edgomez 195 DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
1062 : chenm001 156 DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
1063 :    
1064 : edgomez 195 uint32_t stride = dec->edged_width;
1065 :     uint32_t stride2 = stride / 2;
1066 :     uint32_t next_block = stride * 8;
1067 :     uint32_t iQuant = pMB->quant;
1068 :     int uv_dx, uv_dy;
1069 :     int b_uv_dx, b_uv_dy;
1070 :     uint32_t i;
1071 :     uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
1072 : chenm001 297 const uint32_t cbp = pMB->cbp;
1073 : chenm001 156
1074 : chenm001 161 pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
1075 :     pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
1076 :     pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
1077 :    
1078 : chenm001 297
1079 : edgomez 195 if ((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) {
1080 : chenm001 156 uv_dx = pMB->mvs[0].x;
1081 :     uv_dy = pMB->mvs[0].y;
1082 :    
1083 :     uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
1084 :     uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
1085 :    
1086 :     b_uv_dx = pMB->b_mvs[0].x;
1087 :     b_uv_dy = pMB->b_mvs[0].y;
1088 :    
1089 :     b_uv_dx = (uv_dx & 3) ? (uv_dx >> 1) | 1 : uv_dx / 2;
1090 :     b_uv_dy = (uv_dy & 3) ? (uv_dy >> 1) | 1 : uv_dy / 2;
1091 : edgomez 195 } else {
1092 : chenm001 156 int sum;
1093 : edgomez 195
1094 : chenm001 156 sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
1095 : edgomez 195 uv_dx =
1096 :     (sum ==
1097 :     0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
1098 :     (ABS(sum) / 16) * 2));
1099 : chenm001 156
1100 :     sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
1101 : edgomez 195 uv_dy =
1102 :     (sum ==
1103 :     0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
1104 :     (ABS(sum) / 16) * 2));
1105 : chenm001 156
1106 : edgomez 195 sum =
1107 :     pMB->b_mvs[0].x + pMB->b_mvs[1].x + pMB->b_mvs[2].x +
1108 :     pMB->b_mvs[3].x;
1109 :     b_uv_dx =
1110 :     (sum ==
1111 :     0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
1112 :     (ABS(sum) / 16) * 2));
1113 : chenm001 156
1114 : edgomez 195 sum =
1115 :     pMB->b_mvs[0].y + pMB->b_mvs[1].y + pMB->b_mvs[2].y +
1116 :     pMB->b_mvs[3].y;
1117 :     b_uv_dy =
1118 :     (sum ==
1119 :     0 ? 0 : SIGN(sum) * (roundtab[ABS(sum) % 16] +
1120 :     (ABS(sum) / 16) * 2));
1121 : chenm001 156 }
1122 :    
1123 : edgomez 195
1124 : chenm001 156 start_timer();
1125 : edgomez 195 interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos,
1126 :     pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1127 :     interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8, 16 * y_pos,
1128 :     pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
1129 :     interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos + 8,
1130 :     pMB->mvs[2].x, pMB->mvs[2].y, stride, 0);
1131 :     interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8,
1132 :     16 * y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,
1133 :     0);
1134 :     interpolate8x8_switch(dec->cur.u, forward.u, 8 * x_pos, 8 * y_pos, uv_dx,
1135 :     uv_dy, stride2, 0);
1136 :     interpolate8x8_switch(dec->cur.v, forward.v, 8 * x_pos, 8 * y_pos, uv_dx,
1137 :     uv_dy, stride2, 0);
1138 : chenm001 156
1139 : chenm001 161
1140 : edgomez 195 interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos, 16 * y_pos,
1141 :     pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride, 0);
1142 :     interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos + 8,
1143 :     16 * y_pos, pMB->b_mvs[1].x, pMB->b_mvs[1].y, stride,
1144 :     0);
1145 :     interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos,
1146 :     16 * y_pos + 8, pMB->b_mvs[2].x, pMB->b_mvs[2].y,
1147 :     stride, 0);
1148 :     interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos + 8,
1149 :     16 * y_pos + 8, pMB->b_mvs[3].x, pMB->b_mvs[3].y,
1150 :     stride, 0);
1151 :     interpolate8x8_switch(dec->refn[2].u, backward.u, 8 * x_pos, 8 * y_pos,
1152 :     b_uv_dx, b_uv_dy, stride2, 0);
1153 :     interpolate8x8_switch(dec->refn[2].v, backward.v, 8 * x_pos, 8 * y_pos,
1154 :     b_uv_dx, b_uv_dy, stride2, 0);
1155 : chenm001 156
1156 : Isibaar 579 interpolate8x8_avg2(dec->cur.y + (16 * y_pos * stride) + 16 * x_pos,
1157 :     dec->cur.y + (16 * y_pos * stride) + 16 * x_pos,
1158 :     dec->refn[2].y + (16 * y_pos * stride) + 16 * x_pos,
1159 :     stride, 0);
1160 :    
1161 :     interpolate8x8_avg2(dec->cur.y + (16 * y_pos * stride) + 16 * x_pos + 8,
1162 :     dec->cur.y + (16 * y_pos * stride) + 16 * x_pos + 8,
1163 :     dec->refn[2].y + (16 * y_pos * stride) + 16 * x_pos + 8,
1164 :     stride, 0);
1165 :    
1166 : Isibaar 589 interpolate8x8_avg2(dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos,
1167 :     dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos,
1168 :     dec->refn[2].y + ((16 * y_pos + 8) * stride) + 16 * x_pos,
1169 : Isibaar 579 stride, 0);
1170 :    
1171 : Isibaar 589 interpolate8x8_avg2(dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos + 8,
1172 :     dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos + 8,
1173 :     dec->refn[2].y + ((16 * y_pos + 8) * stride) + 16 * x_pos + 8,
1174 : Isibaar 579 stride, 0);
1175 :    
1176 : Isibaar 589 interpolate8x8_avg2(dec->cur.u + (8 * y_pos * stride2) + 8 * x_pos,
1177 :     dec->cur.u + (8 * y_pos * stride2) + 8 * x_pos,
1178 :     dec->refn[2].u + (8 * y_pos * stride2) + 8 * x_pos,
1179 : Isibaar 579 stride2, 0);
1180 :    
1181 : Isibaar 589 interpolate8x8_avg2(dec->cur.v + (8 * y_pos * stride2) + 8 * x_pos,
1182 :     dec->cur.v + (8 * y_pos * stride2) + 8 * x_pos,
1183 :     dec->refn[2].v + (8 * y_pos * stride2) + 8 * x_pos,
1184 : Isibaar 579 stride2, 0);
1185 :    
1186 : chenm001 156 stop_comp_timer();
1187 : edgomez 195
1188 :     for (i = 0; i < 6; i++) {
1189 : h 543 int direction = dec->alternate_vertical_scan ? 2 : 0;
1190 :    
1191 : edgomez 195 if (cbp & (1 << (5 - i))) // coded
1192 : chenm001 161 {
1193 : edgomez 195 memset(&block[i * 64], 0, 64 * sizeof(int16_t)); // clear
1194 : chenm001 156
1195 : chenm001 161 start_timer();
1196 : h 543 get_inter_block(bs, &block[i * 64], direction);
1197 : chenm001 161 stop_coding_timer();
1198 :    
1199 :     start_timer();
1200 : edgomez 195 if (dec->quant_type == 0) {
1201 :     dequant_inter(&data[i * 64], &block[i * 64], iQuant);
1202 :     } else {
1203 :     dequant4_inter(&data[i * 64], &block[i * 64], iQuant);
1204 : chenm001 161 }
1205 :     stop_iquant_timer();
1206 :    
1207 :     start_timer();
1208 : edgomez 195 idct(&data[i * 64]);
1209 : chenm001 161 stop_idct_timer();
1210 :     }
1211 :     }
1212 :    
1213 : edgomez 195 if (dec->interlacing && pMB->field_dct) {
1214 : chenm001 161 next_block = stride;
1215 :     stride *= 2;
1216 :     }
1217 :    
1218 :     start_timer();
1219 :     if (cbp & 32)
1220 : edgomez 195 transfer_16to8add(pY_Cur, &data[0 * 64], stride);
1221 : chenm001 161 if (cbp & 16)
1222 : edgomez 195 transfer_16to8add(pY_Cur + 8, &data[1 * 64], stride);
1223 : chenm001 161 if (cbp & 8)
1224 : edgomez 195 transfer_16to8add(pY_Cur + next_block, &data[2 * 64], stride);
1225 : chenm001 161 if (cbp & 4)
1226 : edgomez 195 transfer_16to8add(pY_Cur + 8 + next_block, &data[3 * 64], stride);
1227 : chenm001 161 if (cbp & 2)
1228 : edgomez 195 transfer_16to8add(pU_Cur, &data[4 * 64], stride2);
1229 : chenm001 161 if (cbp & 1)
1230 : edgomez 195 transfer_16to8add(pV_Cur, &data[5 * 64], stride2);
1231 : chenm001 161 stop_transfer_timer();
1232 : chenm001 156 }
1233 :    
1234 :    
1235 :     // add by MinChen <chenm001@163.com>
1236 :     // for decode B-frame dbquant
1237 : edgomez 195 int32_t __inline
1238 :     get_dbquant(Bitstream * bs)
1239 : chenm001 156 {
1240 : edgomez 195 if (!BitstreamGetBit(bs)) // '0'
1241 :     return (0);
1242 : chenm001 156 else if (!BitstreamGetBit(bs)) // '10'
1243 : edgomez 195 return (-2);
1244 : chenm001 156 else
1245 : edgomez 195 return (2); // '11'
1246 : chenm001 156 }
1247 :    
1248 :     // add by MinChen <chenm001@163.com>
1249 :     // for decode B-frame mb_type
1250 :     // bit ret_value
1251 :     // 1 0
1252 :     // 01 1
1253 :     // 001 2
1254 :     // 0001 3
1255 : edgomez 195 int32_t __inline
1256 :     get_mbtype(Bitstream * bs)
1257 : chenm001 156 {
1258 : chenm001 161 int32_t mb_type;
1259 : edgomez 195
1260 :     for (mb_type = 0; mb_type <= 3; mb_type++) {
1261 :     if (BitstreamGetBit(bs))
1262 : chenm001 161 break;
1263 :     }
1264 :    
1265 : edgomez 195 if (mb_type <= 3)
1266 :     return (mb_type);
1267 : chenm001 156 else
1268 : edgomez 195 return (-1);
1269 : chenm001 156 }
1270 :    
1271 : edgomez 195 void
1272 :     decoder_bframe(DECODER * dec,
1273 :     Bitstream * bs,
1274 :     int quant,
1275 :     int fcode_forward,
1276 :     int fcode_backward)
1277 : chenm001 156 {
1278 : edgomez 195 uint32_t x, y;
1279 : chenm001 297 VECTOR mv;
1280 :     const VECTOR zeromv = {0,0};
1281 : chenm001 290 #ifdef BFRAMES_DEC_DEBUG
1282 :     FILE *fp;
1283 :     static char first=0;
1284 :     #define BFRAME_DEBUG if (!first && fp){ \
1285 :     fprintf(fp,"Y=%3d X=%3d MB=%2d CBP=%02X\n",y,x,mb->mb_type,mb->cbp); \
1286 :     }
1287 :     #endif
1288 : chenm001 156
1289 :     start_timer();
1290 : edgomez 195 image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height,
1291 : h 543 dec->width, dec->height);
1292 : chenm001 297 image_setedges(&dec->refn[1], dec->edged_width, dec->edged_height,
1293 : h 543 dec->width, dec->height);
1294 : chenm001 156 stop_edges_timer();
1295 :    
1296 : chenm001 290 #ifdef BFRAMES_DEC_DEBUG
1297 :     if (!first){
1298 :     fp=fopen("C:\\XVIDDBG.TXT","w");
1299 :     }
1300 :     #endif
1301 : chenm001 161
1302 : edgomez 195 for (y = 0; y < dec->mb_height; y++) {
1303 : chenm001 161 // Initialize Pred Motion Vector
1304 : chenm001 297 dec->p_fmv = dec->p_bmv = zeromv;
1305 : edgomez 195 for (x = 0; x < dec->mb_width; x++) {
1306 :     MACROBLOCK *mb = &dec->mbs[y * dec->mb_width + x];
1307 :     MACROBLOCK *last_mb = &dec->last_mbs[y * dec->mb_width + x];
1308 : chenm001 156
1309 : chenm001 297 mv =
1310 :     mb->b_mvs[0] = mb->b_mvs[1] = mb->b_mvs[2] = mb->b_mvs[3] =
1311 :     mb->mvs[0] = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = zeromv;
1312 : chenm001 161
1313 : chenm001 156 // the last P_VOP is skip macroblock ?
1314 : edgomez 195 if (last_mb->mode == MODE_NOT_CODED) {
1315 : chenm001 161 //DEBUG2("Skip MB in B-frame at (X,Y)=!",x,y);
1316 : chenm001 290 mb->cbp = 0;
1317 :     #ifdef BFRAMES_DEC_DEBUG
1318 : chenm001 296 mb->mb_type = MODE_NOT_CODED;
1319 : chenm001 290 BFRAME_DEBUG
1320 :     #endif
1321 : chenm001 161 mb->mb_type = MODE_FORWARD;
1322 : chenm001 297 mb->quant = last_mb->quant;
1323 :     //mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;
1324 :     //mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;
1325 : chenm001 156
1326 : chenm001 297 decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, mb->quant, 1);
1327 : chenm001 156 continue;
1328 :     }
1329 : chenm001 161
1330 : edgomez 195 if (!BitstreamGetBit(bs)) { // modb=='0'
1331 :     const uint8_t modb2 = BitstreamGetBit(bs);
1332 : chenm001 156
1333 :     mb->mb_type = get_mbtype(bs);
1334 : edgomez 195
1335 :     if (!modb2) { // modb=='00'
1336 :     mb->cbp = BitstreamGetBits(bs, 6);
1337 : chenm001 156 } else {
1338 :     mb->cbp = 0;
1339 :     }
1340 : edgomez 195 if (mb->mb_type && mb->cbp) {
1341 : chenm001 156 quant += get_dbquant(bs);
1342 :    
1343 : edgomez 195 if (quant > 31) {
1344 : chenm001 156 quant = 31;
1345 : chenm001 272 } else if (quant < 1) {
1346 : chenm001 156 quant = 1;
1347 :     }
1348 :     }
1349 :     } else {
1350 :     mb->mb_type = MODE_DIRECT_NONE_MV;
1351 : edgomez 195 mb->cbp = 0;
1352 : chenm001 156 }
1353 :    
1354 : chenm001 297 mb->quant = quant;
1355 :     mb->mode = MODE_INTER4V;
1356 : chenm001 156 //DEBUG1("Switch bm_type=",mb->mb_type);
1357 :    
1358 : chenm001 290 #ifdef BFRAMES_DEC_DEBUG
1359 :     BFRAME_DEBUG
1360 :     #endif
1361 : chl 326
1362 : edgomez 195 switch (mb->mb_type) {
1363 : chenm001 156 case MODE_DIRECT:
1364 : chenm001 297 get_b_motion_vector(dec, bs, x, y, &mv, 1, zeromv);
1365 : chenm001 156
1366 :     case MODE_DIRECT_NONE_MV:
1367 : chl 326 {
1368 : chenm001 297 const int64_t TRB = dec->time_pp - dec->time_bp, TRD = dec->time_pp;
1369 : chenm001 156 int i;
1370 : edgomez 195
1371 :     for (i = 0; i < 4; i++) {
1372 : chenm001 297 mb->mvs[i].x = (int32_t) ((TRB * last_mb->mvs[i].x)
1373 :     / TRD + mv.x);
1374 :     mb->b_mvs[i].x = (int32_t) ((mv.x == 0)
1375 :     ? ((TRB - TRD) * last_mb->mvs[i].x)
1376 :     / TRD
1377 :     : mb->mvs[i].x - last_mb->mvs[i].x);
1378 :     mb->mvs[i].y = (int32_t) ((TRB * last_mb->mvs[i].y)
1379 :     / TRD + mv.y);
1380 :     mb->b_mvs[i].y = (int32_t) ((mv.y == 0)
1381 :     ? ((TRB - TRD) * last_mb->mvs[i].y)
1382 :     / TRD
1383 :     : mb->mvs[i].y - last_mb->mvs[i].y);
1384 : chenm001 156 }
1385 :     //DEBUG("B-frame Direct!\n");
1386 :     }
1387 : edgomez 195 decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1388 : chenm001 297 mb, x, y, bs);
1389 : chenm001 156 break;
1390 :    
1391 :     case MODE_INTERPOLATE:
1392 : edgomez 195 get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward,
1393 :     dec->p_fmv);
1394 : chenm001 297 dec->p_fmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = mb->mvs[0];
1395 : chenm001 156
1396 : edgomez 195 get_b_motion_vector(dec, bs, x, y, &mb->b_mvs[0],
1397 :     fcode_backward, dec->p_bmv);
1398 : chenm001 297 dec->p_bmv = mb->b_mvs[1] = mb->b_mvs[2] =
1399 :     mb->b_mvs[3] = mb->b_mvs[0];
1400 : chenm001 156
1401 : edgomez 195 decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1402 : chenm001 297 mb, x, y, bs);
1403 : chenm001 156 //DEBUG("B-frame Bidir!\n");
1404 :     break;
1405 :    
1406 :     case MODE_BACKWARD:
1407 : edgomez 195 get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_backward,
1408 :     dec->p_bmv);
1409 : chenm001 297 dec->p_bmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = mb->mvs[0];
1410 : chenm001 161
1411 : chenm001 297 mb->mode = MODE_INTER;
1412 : chenm001 156 decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 0);
1413 :     //DEBUG("B-frame Backward!\n");
1414 :     break;
1415 :    
1416 :     case MODE_FORWARD:
1417 : edgomez 195 get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward,
1418 :     dec->p_fmv);
1419 : chenm001 297 dec->p_fmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = mb->mvs[0];
1420 : chenm001 161
1421 : chenm001 297 mb->mode = MODE_INTER;
1422 : chenm001 156 decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);
1423 : chenm001 161 //DEBUG("B-frame Forward!\n");
1424 : chenm001 156 break;
1425 :    
1426 :     default:
1427 : chl 326 DEBUG1("Not support B-frame mb_type =", mb->mb_type);
1428 : chenm001 156 }
1429 :    
1430 : edgomez 195 } // end of FOR
1431 : chenm001 156 }
1432 : chenm001 290 #ifdef BFRAMES_DEC_DEBUG
1433 :     if (!first){
1434 :     first=1;
1435 :     if (fp)
1436 :     fclose(fp);
1437 :     }
1438 :     #endif
1439 : chenm001 156 }
1440 :    
1441 :     // swap two MACROBLOCK array
1442 : edgomez 195 void
1443 :     mb_swap(MACROBLOCK ** mb1,
1444 :     MACROBLOCK ** mb2)
1445 : chenm001 156 {
1446 : edgomez 195 MACROBLOCK *temp = *mb1;
1447 :    
1448 :     *mb1 = *mb2;
1449 :     *mb2 = temp;
1450 : chenm001 156 }
1451 :    
1452 : edgomez 195 int
1453 :     decoder_decode(DECODER * dec,
1454 : suxen_drol 631 XVID_DEC_FRAME * frame, XVID_DEC_STATS * stats)
1455 : Isibaar 3 {
1456 : edgomez 78
1457 : Isibaar 3 Bitstream bs;
1458 :     uint32_t rounding;
1459 : suxen_drol 631 uint32_t reduced_resolution;
1460 : Isibaar 3 uint32_t quant;
1461 : chenm001 156 uint32_t fcode_forward;
1462 :     uint32_t fcode_backward;
1463 : Isibaar 3 uint32_t intra_dc_threshold;
1464 : suxen_drol 631 VECTOR gmc_mv[5];
1465 : chenm001 133 uint32_t vop_type;
1466 : Isibaar 3
1467 :     start_global_timer();
1468 : edgomez 195
1469 : albeu 315 dec->out_frm = (frame->colorspace == XVID_CSP_EXTERN) ? frame->image : NULL;
1470 :    
1471 : Isibaar 3 BitstreamInit(&bs, frame->bitstream, frame->length);
1472 :    
1473 : suxen_drol 631 // XXX: 0x7f is only valid whilst decoding vfw xvid/divx5 avi's
1474 : suxen_drol 636 if(frame->length == 1 && BitstreamShowBits(&bs, 8) == 0x7f)
1475 : suxen_drol 631 {
1476 :     if (stats)
1477 :     stats->notify = XVID_DEC_VOP;
1478 :     frame->length = 1;
1479 : Isibaar 641 image_output(&dec->refn[0], dec->width, dec->height, dec->edged_width,
1480 : suxen_drol 631 frame->image, frame->stride, frame->colorspace, dec->interlacing);
1481 : Isibaar 641 emms();
1482 : Isibaar 590 return XVID_ERR_OK;
1483 : suxen_drol 631 }
1484 : Isibaar 589
1485 : Isibaar 645 start:
1486 : chenm001 133 // add by chenm001 <chenm001@163.com>
1487 :     // for support B-frame to reference last 2 frame
1488 : Isibaar 179 dec->frames++;
1489 : suxen_drol 631
1490 :     xxx:
1491 : edgomez 195 vop_type =
1492 : suxen_drol 631 BitstreamReadHeaders(&bs, dec, &rounding, &reduced_resolution,
1493 :     &quant, &fcode_forward, &fcode_backward, &intra_dc_threshold, gmc_mv);
1494 : Isibaar 179
1495 : suxen_drol 631 DPRINTF(DPRINTF_HEADER, "vop_type=%i", vop_type);
1496 :    
1497 :     if (vop_type == -2 || vop_type == -3)
1498 :     {
1499 :     if (vop_type == -3)
1500 :     decoder_resize(dec);
1501 :    
1502 :     if (stats)
1503 :     {
1504 :     stats->notify = XVID_DEC_VOL;
1505 :     stats->data.vol.general = 0;
1506 :     if (dec->interlacing)
1507 :     stats->data.vol.general |= XVID_INTERLACING;
1508 :     stats->data.vol.width = dec->width;
1509 :     stats->data.vol.height = dec->height;
1510 :     stats->data.vol.aspect_ratio = dec->aspect_ratio;
1511 :     stats->data.vol.par_width = dec->par_width;
1512 :     stats->data.vol.par_height = dec->par_height;
1513 :     frame->length = BitstreamPos(&bs) / 8;
1514 :     return XVID_ERR_OK;
1515 :     }
1516 :     goto xxx;
1517 :     }
1518 :    
1519 : edgomez 195 dec->p_bmv.x = dec->p_bmv.y = dec->p_fmv.y = dec->p_fmv.y = 0; // init pred vector to 0
1520 :    
1521 :     switch (vop_type) {
1522 :     case P_VOP:
1523 : suxen_drol 631 decoder_pframe(dec, &bs, rounding, reduced_resolution, quant,
1524 :     fcode_forward, intra_dc_threshold, NULL);
1525 : chenm001 272 #ifdef BFRAMES_DEC
1526 : edgomez 195 DEBUG1("P_VOP Time=", dec->time);
1527 : chenm001 272 #endif
1528 : Isibaar 3 break;
1529 :    
1530 : edgomez 195 case I_VOP:
1531 : suxen_drol 631 decoder_iframe(dec, &bs, reduced_resolution, quant, intra_dc_threshold);
1532 : chenm001 272 #ifdef BFRAMES_DEC
1533 : edgomez 195 DEBUG1("I_VOP Time=", dec->time);
1534 : chenm001 272 #endif
1535 : Isibaar 3 break;
1536 :    
1537 : edgomez 195 case B_VOP:
1538 : suxen_drol 233 #ifdef BFRAMES_DEC
1539 : edgomez 195 if (dec->time_pp > dec->time_bp) {
1540 :     DEBUG1("B_VOP Time=", dec->time);
1541 : chenm001 156 decoder_bframe(dec, &bs, quant, fcode_forward, fcode_backward);
1542 :     } else {
1543 :     DEBUG("broken B-frame!");
1544 :     }
1545 : suxen_drol 233 #else
1546 :     image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height);
1547 : Isibaar 179 #endif
1548 : Isibaar 3 break;
1549 : edgomez 195
1550 : suxen_drol 631 case S_VOP :
1551 :     decoder_pframe(dec, &bs, rounding, reduced_resolution, quant,
1552 :     fcode_forward, intra_dc_threshold, gmc_mv);
1553 :     break;
1554 :    
1555 : edgomez 195 case N_VOP: // vop not coded
1556 : suxen_drol 233 // when low_delay==0, N_VOP's should interpolate between the past and future frames
1557 :     image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height);
1558 : Isibaar 589 #ifdef BFRAMES_DEC
1559 :     DEBUG1("N_VOP Time=", dec->time);
1560 :     #endif
1561 : Isibaar 3 break;
1562 :    
1563 : edgomez 195 default:
1564 : suxen_drol 631 if (stats)
1565 :     stats->notify = 0;
1566 : Isibaar 3 return XVID_ERR_FAIL;
1567 :     }
1568 :    
1569 : Isibaar 645 BitstreamByteAlign(&bs);
1570 : Isibaar 3
1571 : suxen_drol 233 #ifdef BFRAMES_DEC
1572 : chenm001 168 // test if no B_VOP
1573 : Isibaar 645 if (dec->low_delay || dec->frames == 0 || ((dec->packed_mode) && !(frame->length > BitstreamPos(&bs) / 8))) {
1574 : Isibaar 179 #endif
1575 : suxen_drol 631 image_output(&dec->cur, dec->width, dec->height, dec->edged_width,
1576 :     frame->image, frame->stride, frame->colorspace, dec->interlacing);
1577 : chenm001 290
1578 : suxen_drol 233 #ifdef BFRAMES_DEC
1579 : chenm001 168 } else {
1580 : Isibaar 645 if (dec->frames >= 1 && !(dec->packed_mode)) {
1581 : chenm001 168 start_timer();
1582 : suxen_drol 631 if ((vop_type == I_VOP || vop_type == P_VOP || vop_type == S_VOP)) {
1583 : edgomez 195 image_output(&dec->refn[0], dec->width, dec->height,
1584 :     dec->edged_width, frame->image, frame->stride,
1585 : suxen_drol 631 frame->colorspace, dec->interlacing);
1586 : chenm001 168 } else if (vop_type == B_VOP) {
1587 : edgomez 195 image_output(&dec->cur, dec->width, dec->height,
1588 :     dec->edged_width, frame->image, frame->stride,
1589 : suxen_drol 631 frame->colorspace, dec->interlacing);
1590 : chenm001 168 }
1591 :     stop_conv_timer();
1592 : chenm001 156 }
1593 :     }
1594 : Isibaar 179 #endif
1595 :    
1596 : suxen_drol 631 if (vop_type == I_VOP || vop_type == P_VOP || vop_type == S_VOP) {
1597 : chenm001 156 image_swap(&dec->refn[0], &dec->refn[1]);
1598 :     image_swap(&dec->cur, &dec->refn[0]);
1599 : chenm001 306
1600 : chenm001 156 // swap MACROBLOCK
1601 : chenm001 306 // the Divx will not set the low_delay flage some times
1602 :     // so follow code will wrong to not swap at that time
1603 :     // this will broken bitstream! so I'm change it,
1604 :     // But that is not the best way! can anyone tell me how
1605 :     // to do another way?
1606 :     // 18-07-2002 MinChen<chenm001@163.com>
1607 :     //if (!dec->low_delay && vop_type == P_VOP)
1608 :     if (vop_type == P_VOP)
1609 : chenm001 161 mb_swap(&dec->mbs, &dec->last_mbs);
1610 : chenm001 156 }
1611 :    
1612 : Isibaar 645 if (frame->length > BitstreamPos(&bs) / 8) // multiple vops packed together
1613 :     goto start;
1614 :    
1615 :     frame->length = BitstreamPos(&bs) / 8;
1616 :    
1617 : suxen_drol 631 if (stats)
1618 :     {
1619 :     stats->notify = XVID_DEC_VOP;
1620 :     stats->data.vop.time_base = (int)dec->time_base;
1621 :     stats->data.vop.time_increment = 0; //XXX: todo
1622 :     }
1623 :    
1624 : Isibaar 3 emms();
1625 :    
1626 :     stop_global_timer();
1627 :    
1628 :     return XVID_ERR_OK;
1629 :     }

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