[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 1532 - (view) (download)

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

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