[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 297 - (view) (download)
Original Path: trunk/xvidcore/src/decoder.c

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

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