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

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