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