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

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

Parent Directory Parent Directory | Revision Log Revision Log


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

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