[svn] / branches / dev-api-3 / xvidcore / src / decoder.c Repository:
ViewVC logotype

Annotation of /branches/dev-api-3/xvidcore/src/decoder.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 702 - (view) (download)

1 : Isibaar 3 /**************************************************************************
2 :     *
3 : edgomez 146 * XVID MPEG-4 VIDEO CODEC
4 :     * - Decoder main module -
5 : Isibaar 3 *
6 : edgomez 146 * This program is an implementation of a part of one or more MPEG-4
7 :     * Video tools as specified in ISO/IEC 14496-2 standard. Those intending
8 :     * to use this software module in hardware or software products are
9 :     * advised that its use may infringe existing patents or copyrights, and
10 :     * any such use would be at such party's own risk. The original
11 :     * developer of this software module and his/her company, and subsequent
12 :     * editors and their companies, will have no liability for use of this
13 :     * software or modifications or derivatives thereof.
14 : Isibaar 3 *
15 : edgomez 146 * This program is free software; you can redistribute it and/or modify
16 :     * it under the terms of the GNU General Public License as published by
17 :     * the Free Software Foundation; either version 2 of the License, or
18 :     * (at your option) any later version.
19 : Isibaar 3 *
20 : edgomez 146 * This program is distributed in the hope that it will be useful,
21 :     * but WITHOUT ANY WARRANTY; without even the implied warranty of
22 :     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
23 :     * GNU General Public License for more details.
24 : Isibaar 3 *
25 : edgomez 146 * You should have received a copy of the GNU General Public License
26 :     * along with this program; if not, write to the Free Software
27 :     * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
28 : Isibaar 3 *
29 :     *************************************************************************/
30 :    
31 :     /**************************************************************************
32 :     *
33 : edgomez 146 * History:
34 : Isibaar 3 *
35 : chenm001 297 * 15.07.2002 fix a bug in B-frame decode at DIRECT mode
36 :     * MinChen <chenm001@163.com>
37 : chenm001 290 * 10.07.2002 added BFRAMES_DEC_DEBUG support
38 :     * Fix a little bug for low_delay flage
39 :     * MinChen <chenm001@163.com>
40 : chenm001 272 * 28.06.2002 added basic resync support to iframe/pframe_decode()
41 : chenm001 290 * 22.06.2002 added primative N_VOP support
42 : chenm001 297 * #define BFRAMES_DEC now enables Minchen's bframe decoder
43 : chenm001 168 * 08.05.2002 add low_delay support for B_VOP decode
44 :     * MinChen <chenm001@163.com>
45 :     * 05.05.2002 fix some B-frame decode problem
46 : chenm001 156 * 02.05.2002 add B-frame decode support(have some problem);
47 :     * MinChen <chenm001@163.com>
48 : chenm001 133 * 22.04.2002 add some B-frame decode support; chenm001 <chenm001@163.com>
49 : h 81 * 29.03.2002 interlacing fix - compensated block wasn't being used when
50 :     * reconstructing blocks, thus artifacts
51 :     * interlacing speedup - used transfers to re-interlace
52 :     * interlaced decoding should be as fast as progressive now
53 : h 69 * 26.03.2002 interlacing support - moved transfers outside decode loop
54 : edgomez 146 * 26.12.2001 decoder_mbinter: dequant/idct moved within if(coded) block
55 :     * 22.12.2001 lock based interpolation
56 :     * 01.12.2001 inital version; (c)2001 peter ross <pross@cs.rmit.edu.au>
57 : Isibaar 3 *
58 : suxen_drol 702 * $Id: decoder.c,v 1.37.2.18 2002-12-10 11:13:50 suxen_drol Exp $
59 : edgomez 146 *
60 : Isibaar 3 *************************************************************************/
61 :    
62 :     #include <stdlib.h>
63 : edgomez 146 #include <string.h>
64 : Isibaar 3
65 : chenm001 290 #ifdef BFRAMES_DEC_DEBUG
66 :     #define BFRAMES_DEC
67 :     #endif
68 :    
69 : Isibaar 3 #include "xvid.h"
70 :     #include "portab.h"
71 :    
72 :     #include "decoder.h"
73 :     #include "bitstream/bitstream.h"
74 :     #include "bitstream/mbcoding.h"
75 :    
76 :     #include "quant/quant_h263.h"
77 :     #include "quant/quant_mpeg4.h"
78 :     #include "dct/idct.h"
79 :     #include "dct/fdct.h"
80 :     #include "utils/mem_transfer.h"
81 :     #include "image/interpolate8x8.h"
82 : suxen_drol 695 #include "image/reduced.h"
83 : Isibaar 3
84 :     #include "bitstream/mbcoding.h"
85 :     #include "prediction/mbprediction.h"
86 :     #include "utils/timer.h"
87 :     #include "utils/emms.h"
88 : Isibaar 616 #include "motion/motion.h"
89 : Isibaar 3
90 :     #include "image/image.h"
91 :     #include "image/colorspace.h"
92 : Isibaar 41 #include "utils/mem_align.h"
93 : Isibaar 3
94 : edgomez 195 int
95 : suxen_drol 631 decoder_resize(DECODER * dec)
96 : Isibaar 3 {
97 : suxen_drol 631 /* free existing */
98 : Isibaar 3
99 : suxen_drol 631 image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
100 :     image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
101 :     image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
102 :     image_destroy(&dec->refh, dec->edged_width, dec->edged_height);
103 :     image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
104 : Isibaar 3
105 : suxen_drol 631 if (dec->last_mbs)
106 :     xvid_free(dec->last_mbs);
107 :     if (dec->mbs)
108 :     xvid_free(dec->mbs);
109 : Isibaar 3
110 : suxen_drol 631 /* realloc */
111 :    
112 : Isibaar 3 dec->mb_width = (dec->width + 15) / 16;
113 :     dec->mb_height = (dec->height + 15) / 16;
114 :    
115 :     dec->edged_width = 16 * dec->mb_width + 2 * EDGE_SIZE;
116 :     dec->edged_height = 16 * dec->mb_height + 2 * EDGE_SIZE;
117 : edgomez 195
118 :     if (image_create(&dec->cur, dec->edged_width, dec->edged_height)) {
119 : Isibaar 41 xvid_free(dec);
120 : Isibaar 3 return XVID_ERR_MEMORY;
121 :     }
122 :    
123 : edgomez 195 if (image_create(&dec->refn[0], dec->edged_width, dec->edged_height)) {
124 : Isibaar 3 image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
125 : Isibaar 41 xvid_free(dec);
126 : Isibaar 3 return XVID_ERR_MEMORY;
127 :     }
128 : suxen_drol 631
129 : chenm001 133 // add by chenm001 <chenm001@163.com>
130 :     // for support B-frame to reference last 2 frame
131 : edgomez 195 if (image_create(&dec->refn[1], dec->edged_width, dec->edged_height)) {
132 : chenm001 133 image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
133 :     image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
134 :     xvid_free(dec);
135 :     return XVID_ERR_MEMORY;
136 :     }
137 : edgomez 195 if (image_create(&dec->refn[2], dec->edged_width, dec->edged_height)) {
138 : chenm001 156 image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
139 :     image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
140 :     image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
141 :     xvid_free(dec);
142 :     return XVID_ERR_MEMORY;
143 :     }
144 : Isibaar 3
145 : Isibaar 579 if (image_create(&dec->refh, dec->edged_width, dec->edged_height)) {
146 :     image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
147 :     image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
148 :     image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
149 :     image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
150 :     xvid_free(dec);
151 :     return XVID_ERR_MEMORY;
152 :     }
153 :    
154 : edgomez 195 dec->mbs =
155 :     xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
156 :     CACHE_LINE);
157 :     if (dec->mbs == NULL) {
158 : Isibaar 3 image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
159 : chenm001 156 image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
160 :     image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
161 :     image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
162 : Isibaar 579 image_destroy(&dec->refh, dec->edged_width, dec->edged_height);
163 : Isibaar 41 xvid_free(dec);
164 : Isibaar 3 return XVID_ERR_MEMORY;
165 :     }
166 : Isibaar 208 memset(dec->mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);
167 :    
168 : chenm001 156 // add by chenm001 <chenm001@163.com>
169 :     // for skip MB flag
170 : edgomez 195 dec->last_mbs =
171 :     xvid_malloc(sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height,
172 :     CACHE_LINE);
173 :     if (dec->last_mbs == NULL) {
174 : chenm001 156 xvid_free(dec->mbs);
175 :     image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
176 :     image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
177 :     image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
178 :     image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
179 : Isibaar 579 image_destroy(&dec->refh, dec->edged_width, dec->edged_height);
180 : chenm001 156 xvid_free(dec);
181 :     return XVID_ERR_MEMORY;
182 :     }
183 : Isibaar 3
184 : Isibaar 208 memset(dec->last_mbs, 0, sizeof(MACROBLOCK) * dec->mb_width * dec->mb_height);
185 :    
186 : suxen_drol 631 return XVID_ERR_OK;
187 :     }
188 :    
189 :    
190 :     int
191 :     decoder_create(XVID_DEC_PARAM * param)
192 :     {
193 :     DECODER *dec;
194 :    
195 :     dec = xvid_malloc(sizeof(DECODER), CACHE_LINE);
196 :     if (dec == NULL) {
197 :     return XVID_ERR_MEMORY;
198 :     }
199 :     memset(dec, 0, sizeof(DECODER));
200 :    
201 :     param->handle = dec;
202 :    
203 :     dec->width = param->width;
204 :     dec->height = param->height;
205 :    
206 :     image_null(&dec->cur);
207 :     image_null(&dec->refn[0]);
208 :     image_null(&dec->refn[1]);
209 :     image_null(&dec->refn[2]);
210 :     image_null(&dec->refh);
211 :    
212 :     dec->mbs = NULL;
213 :     dec->last_mbs = NULL;
214 :    
215 : Isibaar 3 init_timer();
216 :    
217 : chenm001 156 // add by chenm001 <chenm001@163.com>
218 :     // for support B-frame to save reference frame's time
219 :     dec->frames = -1;
220 :     dec->time = dec->time_base = dec->last_time_base = 0;
221 : suxen_drol 631 dec->low_delay = 0;
222 : suxen_drol 695 dec->packed_mode = 0;
223 : edgomez 195
224 : suxen_drol 631 dec->fixed_dimensions = (dec->width > 0 && dec->height > 0);
225 :    
226 :     if (dec->fixed_dimensions)
227 :     return decoder_resize(dec);
228 :     else
229 :     return XVID_ERR_OK;
230 : Isibaar 3 }
231 :    
232 :    
233 : edgomez 195 int
234 :     decoder_destroy(DECODER * dec)
235 : Isibaar 3 {
236 : chenm001 156 xvid_free(dec->last_mbs);
237 : Isibaar 41 xvid_free(dec->mbs);
238 : chenm001 133 image_destroy(&dec->refn[0], dec->edged_width, dec->edged_height);
239 : chenm001 156 image_destroy(&dec->refn[1], dec->edged_width, dec->edged_height);
240 :     image_destroy(&dec->refn[2], dec->edged_width, dec->edged_height);
241 : Isibaar 579 image_destroy(&dec->refh, dec->edged_width, dec->edged_height);
242 : Isibaar 3 image_destroy(&dec->cur, dec->edged_width, dec->edged_height);
243 : Isibaar 41 xvid_free(dec);
244 : Isibaar 3
245 :     write_timer();
246 :     return XVID_ERR_OK;
247 :     }
248 :    
249 :    
250 :    
251 : edgomez 195 static const int32_t dquant_table[4] = {
252 : Isibaar 3 -1, -2, 1, 2
253 :     };
254 :    
255 :    
256 : chenm001 156
257 :    
258 : Isibaar 3 // decode an intra macroblock
259 :    
260 : edgomez 195 void
261 :     decoder_mbintra(DECODER * dec,
262 :     MACROBLOCK * pMB,
263 :     const uint32_t x_pos,
264 :     const uint32_t y_pos,
265 :     const uint32_t acpred_flag,
266 :     const uint32_t cbp,
267 :     Bitstream * bs,
268 :     const uint32_t quant,
269 : chenm001 272 const uint32_t intra_dc_threshold,
270 : suxen_drol 695 const unsigned int bound,
271 :     const int reduced_resolution)
272 : Isibaar 3 {
273 : edgomez 78
274 :     DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
275 : edgomez 195 DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
276 : edgomez 78
277 : h 81 uint32_t stride = dec->edged_width;
278 :     uint32_t stride2 = stride / 2;
279 :     uint32_t next_block = stride * 8;
280 : h 69 uint32_t i;
281 :     uint32_t iQuant = pMB->quant;
282 :     uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
283 :    
284 : suxen_drol 695 if (reduced_resolution) {
285 :     pY_Cur = dec->cur.y + (y_pos << 5) * stride + (x_pos << 5);
286 :     pU_Cur = dec->cur.u + (y_pos << 4) * stride2 + (x_pos << 4);
287 :     pV_Cur = dec->cur.v + (y_pos << 4) * stride2 + (x_pos << 4);
288 :     }else{
289 :     pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
290 :     pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
291 :     pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
292 :     }
293 : h 69
294 : edgomez 195 memset(block, 0, 6 * 64 * sizeof(int16_t)); // clear
295 : h 69
296 : edgomez 195 for (i = 0; i < 6; i++) {
297 : h 69 uint32_t iDcScaler = get_dc_scaler(iQuant, i < 4);
298 : Isibaar 3 int16_t predictors[8];
299 :     int start_coeff;
300 :    
301 :     start_timer();
302 : edgomez 195 predict_acdc(dec->mbs, x_pos, y_pos, dec->mb_width, i, &block[i * 64],
303 : chenm001 272 iQuant, iDcScaler, predictors, bound);
304 : edgomez 195 if (!acpred_flag) {
305 : h 69 pMB->acpred_directions[i] = 0;
306 : Isibaar 3 }
307 :     stop_prediction_timer();
308 :    
309 : edgomez 195 if (quant < intra_dc_threshold) {
310 : Isibaar 3 int dc_size;
311 :     int dc_dif;
312 :    
313 : edgomez 195 dc_size = i < 4 ? get_dc_size_lum(bs) : get_dc_size_chrom(bs);
314 :     dc_dif = dc_size ? get_dc_dif(bs, dc_size) : 0;
315 : Isibaar 3
316 : edgomez 195 if (dc_size > 8) {
317 :     BitstreamSkip(bs, 1); // marker
318 : Isibaar 3 }
319 : edgomez 195
320 :     block[i * 64 + 0] = dc_dif;
321 : Isibaar 3 start_coeff = 1;
322 : chenm001 272
323 :     DPRINTF(DPRINTF_COEFF,"block[0] %i", dc_dif);
324 : edgomez 195 } else {
325 : Isibaar 3 start_coeff = 0;
326 :     }
327 :    
328 :     start_timer();
329 : edgomez 195 if (cbp & (1 << (5 - i))) // coded
330 : Isibaar 3 {
331 : h 543 int direction = dec->alternate_vertical_scan ?
332 :     2 : pMB->acpred_directions[i];
333 :    
334 :     get_intra_block(bs, &block[i * 64], direction, start_coeff);
335 : Isibaar 3 }
336 :     stop_coding_timer();
337 :    
338 :     start_timer();
339 : edgomez 195 add_acdc(pMB, i, &block[i * 64], iDcScaler, predictors);
340 : Isibaar 3 stop_prediction_timer();
341 :    
342 :     start_timer();
343 : edgomez 195 if (dec->quant_type == 0) {
344 :     dequant_intra(&data[i * 64], &block[i * 64], iQuant, iDcScaler);
345 :     } else {
346 :     dequant4_intra(&data[i * 64], &block[i * 64], iQuant, iDcScaler);
347 : Isibaar 3 }
348 :     stop_iquant_timer();
349 :    
350 :     start_timer();
351 : edgomez 195 idct(&data[i * 64]);
352 : Isibaar 3 stop_idct_timer();
353 : suxen_drol 695
354 : h 69 }
355 : Isibaar 3
356 : edgomez 195 if (dec->interlacing && pMB->field_dct) {
357 : h 81 next_block = stride;
358 :     stride *= 2;
359 : Isibaar 3 }
360 : h 69
361 :     start_timer();
362 : suxen_drol 695
363 :     if (reduced_resolution)
364 :     {
365 :     next_block*=2;
366 :     copy_upsampled_8x8_16to8(pY_Cur, &data[0 * 64], stride);
367 :     copy_upsampled_8x8_16to8(pY_Cur + 16, &data[1 * 64], stride);
368 :     copy_upsampled_8x8_16to8(pY_Cur + next_block, &data[2 * 64], stride);
369 :     copy_upsampled_8x8_16to8(pY_Cur + 16 + next_block, &data[3 * 64], stride);
370 :     copy_upsampled_8x8_16to8(pU_Cur, &data[4 * 64], stride2);
371 :     copy_upsampled_8x8_16to8(pV_Cur, &data[5 * 64], stride2);
372 :     }else{
373 :     transfer_16to8copy(pY_Cur, &data[0 * 64], stride);
374 :     transfer_16to8copy(pY_Cur + 8, &data[1 * 64], stride);
375 :     transfer_16to8copy(pY_Cur + next_block, &data[2 * 64], stride);
376 :     transfer_16to8copy(pY_Cur + 8 + next_block, &data[3 * 64], stride);
377 :     transfer_16to8copy(pU_Cur, &data[4 * 64], stride2);
378 :     transfer_16to8copy(pV_Cur, &data[5 * 64], stride2);
379 :     }
380 : h 69 stop_transfer_timer();
381 : Isibaar 3 }
382 :    
383 :    
384 :    
385 :    
386 :    
387 :     #define SIGN(X) (((X)>0)?1:-1)
388 :     #define ABS(X) (((X)>0)?(X):-(X))
389 :    
390 :     // decode an inter macroblock
391 :    
392 : edgomez 195 void
393 :     decoder_mbinter(DECODER * dec,
394 :     const MACROBLOCK * pMB,
395 :     const uint32_t x_pos,
396 :     const uint32_t y_pos,
397 :     const uint32_t acpred_flag,
398 :     const uint32_t cbp,
399 :     Bitstream * bs,
400 :     const uint32_t quant,
401 : suxen_drol 695 const uint32_t rounding,
402 :     const int reduced_resolution)
403 : Isibaar 3 {
404 : h 69
405 : edgomez 195 DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
406 : edgomez 78 DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
407 :    
408 : h 81 uint32_t stride = dec->edged_width;
409 :     uint32_t stride2 = stride / 2;
410 : suxen_drol 695 uint32_t next_block = stride * (reduced_resolution ? 16 : 8);
411 : edgomez 78 uint32_t i;
412 :     uint32_t iQuant = pMB->quant;
413 : h 69 uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
414 : suxen_drol 695
415 : Isibaar 3 int uv_dx, uv_dy;
416 : suxen_drol 702 VECTOR mv[4]; /* local copy of mvs */
417 : Isibaar 3
418 : suxen_drol 695 if (reduced_resolution) {
419 :     pY_Cur = dec->cur.y + (y_pos << 5) * stride + (x_pos << 5);
420 :     pU_Cur = dec->cur.u + (y_pos << 4) * stride2 + (x_pos << 4);
421 :     pV_Cur = dec->cur.v + (y_pos << 4) * stride2 + (x_pos << 4);
422 : suxen_drol 702 for (i = 0; i < 4; i++) {
423 :     mv[i].x = RRV_MV_SCALEUP(pMB->mvs[i].x);
424 :     mv[i].y = RRV_MV_SCALEUP(pMB->mvs[i].y);
425 :     }
426 : suxen_drol 695 }else{
427 :     pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
428 :     pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
429 :     pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
430 : suxen_drol 702 for (i = 0; i < 4; i++)
431 :     mv[i] = pMB->mvs[i];
432 : suxen_drol 695 }
433 :    
434 : edgomez 195 if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q) {
435 : suxen_drol 695 uv_dx = mv[0].x;
436 :     uv_dy = mv[0].y;
437 : Isibaar 3
438 : Isibaar 333 if (dec->quarterpel)
439 :     {
440 : Isibaar 617 uv_dx /= 2;
441 :     uv_dy /= 2;
442 : Isibaar 333 }
443 :    
444 : Isibaar 617 uv_dx = (uv_dx >> 1) + roundtab_79[uv_dx & 0x3];
445 :     uv_dy = (uv_dy >> 1) + roundtab_79[uv_dy & 0x3];
446 :    
447 : Isibaar 588 start_timer();
448 : suxen_drol 695 if (reduced_resolution)
449 :     {
450 :     interpolate32x32_switch(dec->cur.y, dec->refn[0].y, 32*x_pos, 32*y_pos,
451 :     mv[0].x, mv[0].y, stride, rounding);
452 :     interpolate16x16_switch(dec->cur.u, dec->refn[0].u, 16 * x_pos, 16 * y_pos,
453 :     uv_dx, uv_dy, stride2, rounding);
454 :     interpolate16x16_switch(dec->cur.v, dec->refn[0].v, 16 * x_pos, 16 * y_pos,
455 :     uv_dx, uv_dy, stride2, rounding);
456 :    
457 : Isibaar 588 }
458 : suxen_drol 695 else
459 :     {
460 :     if(dec->quarterpel) {
461 :     interpolate16x16_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
462 :     dec->refh.y + 128, 16*x_pos, 16*y_pos,
463 :     mv[0].x, mv[0].y, stride, rounding);
464 :     }
465 :     else {
466 :     interpolate16x16_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos,
467 :     mv[0].x, mv[0].y, stride, rounding);
468 :     }
469 :    
470 :     interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8 * x_pos, 8 * y_pos,
471 :     uv_dx, uv_dy, stride2, rounding);
472 :     interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8 * x_pos, 8 * y_pos,
473 :     uv_dx, uv_dy, stride2, rounding);
474 : Isibaar 588 }
475 :     stop_comp_timer();
476 :    
477 : suxen_drol 695 } else { /* MODE_INTER4V */
478 : Isibaar 3 int sum;
479 : Isibaar 616
480 :     if(dec->quarterpel)
481 : suxen_drol 695 sum = (mv[0].x / 2) + (mv[1].x / 2) + (mv[2].x / 2) + (mv[3].x / 2);
482 : Isibaar 616 else
483 : suxen_drol 695 sum = mv[0].x + mv[1].x + mv[2].x + mv[3].x;
484 : Isibaar 3
485 : Isibaar 616 uv_dx = (sum >> 3) + roundtab_76[sum & 0xf];
486 : Isibaar 333
487 : Isibaar 616 if(dec->quarterpel)
488 : suxen_drol 695 sum = (mv[0].y / 2) + (mv[1].y / 2) + (mv[2].y / 2) + (mv[3].y / 2);
489 : Isibaar 616 else
490 : suxen_drol 695 sum = mv[0].y + mv[1].y + mv[2].y + mv[3].y;
491 : Isibaar 333
492 : Isibaar 616 uv_dy = (sum >> 3) + roundtab_76[sum & 0xf];
493 : Isibaar 333
494 : Isibaar 588 start_timer();
495 : suxen_drol 695 if (reduced_resolution)
496 :     {
497 :     interpolate16x16_switch(dec->cur.y, dec->refn[0].y, 32*x_pos, 32*y_pos,
498 :     mv[0].x, mv[0].y, stride, rounding);
499 :     interpolate16x16_switch(dec->cur.y, dec->refn[0].y, 32*x_pos + 16, 32*y_pos,
500 :     mv[1].x, mv[1].y, stride, rounding);
501 :     interpolate16x16_switch(dec->cur.y, dec->refn[0].y, 32*x_pos, 32*y_pos + 16,
502 :     mv[2].x, mv[2].y, stride, rounding);
503 :     interpolate16x16_switch(dec->cur.y, dec->refn[0].y, 32*x_pos + 16, 32*y_pos + 16,
504 :     mv[3].x, mv[3].y, stride, rounding);
505 :     interpolate16x16_switch(dec->cur.u, dec->refn[0].u, 16 * x_pos, 16 * y_pos,
506 :     uv_dx, uv_dy, stride2, rounding);
507 :     interpolate16x16_switch(dec->cur.v, dec->refn[0].v, 16 * x_pos, 16 * y_pos,
508 :     uv_dx, uv_dy, stride2, rounding);
509 :    
510 :     // set_block(pY_Cur, stride, 32, 32, 127);
511 :     }
512 :     else
513 :     {
514 :     if(dec->quarterpel) {
515 :     interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
516 :     dec->refh.y + 128, 16*x_pos, 16*y_pos,
517 :     mv[0].x, mv[0].y, stride, rounding);
518 :     interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
519 :     dec->refh.y + 128, 16*x_pos + 8, 16*y_pos,
520 :     mv[1].x, mv[1].y, stride, rounding);
521 :     interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
522 :     dec->refh.y + 128, 16*x_pos, 16*y_pos + 8,
523 :     mv[2].x, mv[2].y, stride, rounding);
524 :     interpolate8x8_quarterpel(dec->cur.y, dec->refn[0].y, dec->refh.y, dec->refh.y + 64,
525 :     dec->refh.y + 128, 16*x_pos + 8, 16*y_pos + 8,
526 :     mv[3].x, mv[3].y, stride, rounding);
527 :     }
528 :     else {
529 :     interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos,
530 :     mv[0].x, mv[0].y, stride, rounding);
531 :     interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos,
532 :     mv[1].x, mv[1].y, stride, rounding);
533 :     interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos, 16*y_pos + 8,
534 :     mv[2].x, mv[2].y, stride, rounding);
535 :     interpolate8x8_switch(dec->cur.y, dec->refn[0].y, 16*x_pos + 8, 16*y_pos + 8,
536 :     mv[3].x, mv[3].y, stride, rounding);
537 :     }
538 :    
539 :     interpolate8x8_switch(dec->cur.u, dec->refn[0].u, 8 * x_pos, 8 * y_pos,
540 :     uv_dx, uv_dy, stride2, rounding);
541 :     interpolate8x8_switch(dec->cur.v, dec->refn[0].v, 8 * x_pos, 8 * y_pos,
542 :     uv_dx, uv_dy, stride2, rounding);
543 : Isibaar 588 }
544 :     stop_comp_timer();
545 : Isibaar 333 }
546 :    
547 : edgomez 195 for (i = 0; i < 6; i++) {
548 : h 543 int direction = dec->alternate_vertical_scan ? 2 : 0;
549 :    
550 : edgomez 195 if (cbp & (1 << (5 - i))) // coded
551 : Isibaar 3 {
552 : edgomez 195 memset(&block[i * 64], 0, 64 * sizeof(int16_t)); // clear
553 : Isibaar 3
554 :     start_timer();
555 : h 543 get_inter_block(bs, &block[i * 64], direction);
556 : Isibaar 3 stop_coding_timer();
557 :    
558 :     start_timer();
559 : edgomez 195 if (dec->quant_type == 0) {
560 :     dequant_inter(&data[i * 64], &block[i * 64], iQuant);
561 :     } else {
562 :     dequant4_inter(&data[i * 64], &block[i * 64], iQuant);
563 : Isibaar 3 }
564 :     stop_iquant_timer();
565 :    
566 :     start_timer();
567 : edgomez 195 idct(&data[i * 64]);
568 : Isibaar 3 stop_idct_timer();
569 :     }
570 :     }
571 : h 69
572 : edgomez 195 if (dec->interlacing && pMB->field_dct) {
573 : h 81 next_block = stride;
574 :     stride *= 2;
575 : h 69 }
576 :    
577 :     start_timer();
578 : suxen_drol 695 if (reduced_resolution)
579 :     {
580 :     if (cbp & 32)
581 :     add_upsampled_8x8_16to8(pY_Cur, &data[0 * 64], stride);
582 :     if (cbp & 16)
583 :     add_upsampled_8x8_16to8(pY_Cur + 16, &data[1 * 64], stride);
584 :     if (cbp & 8)
585 :     add_upsampled_8x8_16to8(pY_Cur + next_block, &data[2 * 64], stride);
586 :     if (cbp & 4)
587 :     add_upsampled_8x8_16to8(pY_Cur + 16 + next_block, &data[3 * 64], stride);
588 :     if (cbp & 2)
589 :     add_upsampled_8x8_16to8(pU_Cur, &data[4 * 64], stride2);
590 :     if (cbp & 1)
591 :     add_upsampled_8x8_16to8(pV_Cur, &data[5 * 64], stride2);
592 :     }
593 :     else
594 :     {
595 :     if (cbp & 32)
596 :     transfer_16to8add(pY_Cur, &data[0 * 64], stride);
597 :     if (cbp & 16)
598 :     transfer_16to8add(pY_Cur + 8, &data[1 * 64], stride);
599 :     if (cbp & 8)
600 :     transfer_16to8add(pY_Cur + next_block, &data[2 * 64], stride);
601 :     if (cbp & 4)
602 :     transfer_16to8add(pY_Cur + 8 + next_block, &data[3 * 64], stride);
603 :     if (cbp & 2)
604 :     transfer_16to8add(pU_Cur, &data[4 * 64], stride2);
605 :     if (cbp & 1)
606 :     transfer_16to8add(pV_Cur, &data[5 * 64], stride2);
607 :     }
608 : h 69 stop_transfer_timer();
609 : Isibaar 3 }
610 :    
611 :    
612 : edgomez 195 void
613 :     decoder_iframe(DECODER * dec,
614 :     Bitstream * bs,
615 : suxen_drol 631 int reduced_resolution,
616 : edgomez 195 int quant,
617 :     int intra_dc_threshold)
618 : Isibaar 3 {
619 : chenm001 272 uint32_t bound;
620 : Isibaar 3 uint32_t x, y;
621 : suxen_drol 631 int mb_width = dec->mb_width;
622 :     int mb_height = dec->mb_height;
623 :    
624 :     if (reduced_resolution)
625 :     {
626 : suxen_drol 701 mb_width = (dec->width + 31) / 32;
627 :     mb_height = (dec->height + 31) / 32;
628 : suxen_drol 631 }
629 : Isibaar 3
630 : chenm001 272 bound = 0;
631 :    
632 : suxen_drol 631 for (y = 0; y < mb_height; y++) {
633 :     for (x = 0; x < mb_width; x++) {
634 : chenm001 272 MACROBLOCK *mb;
635 : Isibaar 3 uint32_t mcbpc;
636 :     uint32_t cbpc;
637 :     uint32_t acpred_flag;
638 :     uint32_t cbpy;
639 :     uint32_t cbp;
640 :    
641 : chenm001 272 while (BitstreamShowBits(bs, 9) == 1)
642 :     BitstreamSkip(bs, 9);
643 :    
644 :     if (check_resync_marker(bs, 0))
645 :     {
646 : suxen_drol 631 bound = read_video_packet_header(bs, dec, 0,
647 :     &quant, NULL, NULL, &intra_dc_threshold);
648 : suxen_drol 695 x = bound % mb_width;
649 :     y = bound / mb_width;
650 : chenm001 272 }
651 :     mb = &dec->mbs[y * dec->mb_width + x];
652 :    
653 :     DPRINTF(DPRINTF_MB, "macroblock (%i,%i) %08x", x, y, BitstreamShowBits(bs, 32));
654 :    
655 : Isibaar 3 mcbpc = get_mcbpc_intra(bs);
656 :     mb->mode = mcbpc & 7;
657 :     cbpc = (mcbpc >> 4);
658 :    
659 :     acpred_flag = BitstreamGetBit(bs);
660 :    
661 :     cbpy = get_cbpy(bs, 1);
662 :     cbp = (cbpy << 2) | cbpc;
663 :    
664 : edgomez 195 if (mb->mode == MODE_INTRA_Q) {
665 :     quant += dquant_table[BitstreamGetBits(bs, 2)];
666 :     if (quant > 31) {
667 : Isibaar 3 quant = 31;
668 : edgomez 195 } else if (quant < 1) {
669 : Isibaar 3 quant = 1;
670 :     }
671 :     }
672 :     mb->quant = quant;
673 : chenm001 297 mb->mvs[0].x = mb->mvs[0].y =
674 :     mb->mvs[1].x = mb->mvs[1].y =
675 :     mb->mvs[2].x = mb->mvs[2].y =
676 :     mb->mvs[3].x = mb->mvs[3].y =0;
677 : Isibaar 3
678 : edgomez 195 if (dec->interlacing) {
679 : h 69 mb->field_dct = BitstreamGetBit(bs);
680 :     DEBUG1("deci: field_dct: ", mb->field_dct);
681 :     }
682 :    
683 : edgomez 195 decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant,
684 : suxen_drol 695 intra_dc_threshold, bound, reduced_resolution);
685 : suxen_drol 631
686 : Isibaar 3 }
687 : albeu 315 if(dec->out_frm)
688 : suxen_drol 695 output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,0,y,mb_width);
689 : Isibaar 3 }
690 : edgomez 78
691 : Isibaar 3 }
692 :    
693 :    
694 : edgomez 195 void
695 :     get_motion_vector(DECODER * dec,
696 :     Bitstream * bs,
697 :     int x,
698 :     int y,
699 :     int k,
700 : suxen_drol 695 VECTOR * ret_mv,
701 : chenm001 272 int fcode,
702 :     const int bound)
703 : Isibaar 3 {
704 : edgomez 78
705 : Isibaar 3 int scale_fac = 1 << (fcode - 1);
706 :     int high = (32 * scale_fac) - 1;
707 :     int low = ((-32) * scale_fac);
708 :     int range = (64 * scale_fac);
709 :    
710 : chenm001 272 VECTOR pmv;
711 : suxen_drol 695 VECTOR mv;
712 : Isibaar 3
713 : chenm001 272 pmv = get_pmv2(dec->mbs, dec->mb_width, bound, x, y, k);
714 : Isibaar 3
715 : suxen_drol 695 mv.x = get_mv(bs, fcode);
716 :     mv.y = get_mv(bs, fcode);
717 : edgomez 195
718 : suxen_drol 695 DPRINTF(DPRINTF_MV,"mv_diff (%i,%i) pred (%i,%i)", mv.x, mv.y, pmv.x, pmv.y);
719 : suxen_drol 252
720 : suxen_drol 695 mv.x += pmv.x;
721 :     mv.y += pmv.y;
722 : chenm001 272
723 : suxen_drol 695 if (mv.x < low) {
724 :     mv.x += range;
725 :     } else if (mv.x > high) {
726 :     mv.x -= range;
727 : Isibaar 3 }
728 :    
729 : suxen_drol 695 if (mv.y < low) {
730 :     mv.y += range;
731 :     } else if (mv.y > high) {
732 :     mv.y -= range;
733 : Isibaar 3 }
734 :    
735 : suxen_drol 695 ret_mv->x = mv.x;
736 :     ret_mv->y = mv.y;
737 : suxen_drol 631 }
738 : Isibaar 3
739 : suxen_drol 631
740 :    
741 :     static __inline int gmc_sanitize(int value, int quarterpel, int fcode)
742 :     {
743 :     int length = 1 << (fcode+4);
744 :    
745 :     if (quarterpel) value *= 2;
746 :    
747 :     if (value < -length)
748 :     return -length;
749 :     else if (value >= length)
750 :     return length-1;
751 :     else return value;
752 : Isibaar 3 }
753 :    
754 :    
755 : suxen_drol 631 /* for P_VOP set gmc_mv to NULL */
756 : edgomez 195 void
757 :     decoder_pframe(DECODER * dec,
758 :     Bitstream * bs,
759 :     int rounding,
760 : suxen_drol 631 int reduced_resolution,
761 : edgomez 195 int quant,
762 :     int fcode,
763 : suxen_drol 631 int intra_dc_threshold,
764 :     VECTOR * gmc_mv)
765 : Isibaar 3 {
766 : edgomez 78
767 : Isibaar 3 uint32_t x, y;
768 : chenm001 272 uint32_t bound;
769 : albeu 315 int cp_mb, st_mb;
770 : suxen_drol 695 int mb_width = dec->mb_width;
771 :     int mb_height = dec->mb_height;
772 :    
773 :     if (reduced_resolution)
774 :     {
775 : suxen_drol 701 mb_width = (dec->width + 31) / 32;
776 :     mb_height = (dec->height + 31) / 32;
777 : suxen_drol 695 }
778 : Isibaar 3
779 :     start_timer();
780 : edgomez 195 image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height,
781 : h 543 dec->width, dec->height);
782 : Isibaar 3 stop_edges_timer();
783 :    
784 : chenm001 272 bound = 0;
785 :    
786 : suxen_drol 695 for (y = 0; y < mb_height; y++) {
787 : albeu 315 cp_mb = st_mb = 0;
788 : suxen_drol 695 for (x = 0; x < mb_width; x++) {
789 : chenm001 272 MACROBLOCK *mb;
790 : Isibaar 3
791 : chenm001 272 // skip stuffing
792 :     while (BitstreamShowBits(bs, 10) == 1)
793 :     BitstreamSkip(bs, 10);
794 :    
795 :     if (check_resync_marker(bs, fcode - 1))
796 :     {
797 : suxen_drol 631 bound = read_video_packet_header(bs, dec, fcode - 1,
798 :     &quant, &fcode, NULL, &intra_dc_threshold);
799 : suxen_drol 695 x = bound % mb_width;
800 :     y = bound / mb_width;
801 : chenm001 272 }
802 :     mb = &dec->mbs[y * dec->mb_width + x];
803 :    
804 :     DPRINTF(DPRINTF_MB, "macroblock (%i,%i) %08x", x, y, BitstreamShowBits(bs, 32));
805 :    
806 : edgomez 195 //if (!(dec->mb_skip[y*dec->mb_width + x]=BitstreamGetBit(bs))) // not_coded
807 :     if (!(BitstreamGetBit(bs))) // not_coded
808 : Isibaar 3 {
809 :     uint32_t mcbpc;
810 :     uint32_t cbpc;
811 :     uint32_t acpred_flag;
812 :     uint32_t cbpy;
813 :     uint32_t cbp;
814 :     uint32_t intra;
815 : suxen_drol 631 int mcsel = 0; // mcsel: '0'=local motion, '1'=GMC
816 : Isibaar 3
817 : albeu 315 cp_mb++;
818 : Isibaar 3 mcbpc = get_mcbpc_inter(bs);
819 :     mb->mode = mcbpc & 7;
820 :     cbpc = (mcbpc >> 4);
821 : chenm001 272
822 :     DPRINTF(DPRINTF_MB, "mode %i", mb->mode);
823 :     DPRINTF(DPRINTF_MB, "cbpc %i", cbpc);
824 : edgomez 12 acpred_flag = 0;
825 : Isibaar 3
826 :     intra = (mb->mode == MODE_INTRA || mb->mode == MODE_INTRA_Q);
827 : edgomez 195
828 :     if (intra) {
829 : Isibaar 3 acpred_flag = BitstreamGetBit(bs);
830 :     }
831 :    
832 : suxen_drol 631 if (gmc_mv && (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q))
833 :     {
834 :     mcsel = BitstreamGetBit(bs);
835 :     }
836 :    
837 : Isibaar 3 cbpy = get_cbpy(bs, intra);
838 : chenm001 272 DPRINTF(DPRINTF_MB, "cbpy %i", cbpy);
839 :    
840 : Isibaar 3 cbp = (cbpy << 2) | cbpc;
841 :    
842 : edgomez 195 if (mb->mode == MODE_INTER_Q || mb->mode == MODE_INTRA_Q) {
843 : chenm001 272 int dquant = dquant_table[BitstreamGetBits(bs, 2)];
844 :     DPRINTF(DPRINTF_MB, "dquant %i", dquant);
845 :     quant += dquant;
846 : edgomez 195 if (quant > 31) {
847 : Isibaar 3 quant = 31;
848 : chenm001 272 } else if (quant < 1) {
849 : Isibaar 3 quant = 1;
850 :     }
851 : chenm001 272 DPRINTF(DPRINTF_MB, "quant %i", quant);
852 : Isibaar 3 }
853 :     mb->quant = quant;
854 : h 69
855 : edgomez 195 if (dec->interlacing) {
856 : h 388 if (cbp || intra) {
857 :     mb->field_dct = BitstreamGetBit(bs);
858 :     DEBUG1("decp: field_dct: ", mb->field_dct);
859 :     }
860 : Isibaar 3
861 : edgomez 195 if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
862 : h 69 mb->field_pred = BitstreamGetBit(bs);
863 :     DEBUG1("decp: field_pred: ", mb->field_pred);
864 :    
865 : edgomez 195 if (mb->field_pred) {
866 : h 69 mb->field_for_top = BitstreamGetBit(bs);
867 :     DEBUG1("decp: field_for_top: ", mb->field_for_top);
868 :     mb->field_for_bot = BitstreamGetBit(bs);
869 :     DEBUG1("decp: field_for_bot: ", mb->field_for_bot);
870 :     }
871 :     }
872 : Isibaar 3 }
873 : h 69
874 : edgomez 195 if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) {
875 : suxen_drol 631
876 :     if (mcsel)
877 :     {
878 :     mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = gmc_sanitize(gmc_mv[0].x, dec->quarterpel, fcode);
879 :     mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = gmc_sanitize(gmc_mv[0].y, dec->quarterpel, fcode);
880 :    
881 :     } else if (dec->interlacing && mb->field_pred) {
882 : edgomez 195 get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0],
883 : chenm001 272 fcode, bound);
884 : edgomez 195 get_motion_vector(dec, bs, x, y, 0, &mb->mvs[1],
885 : chenm001 272 fcode, bound);
886 : edgomez 195 } else {
887 :     get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0],
888 : chenm001 272 fcode, bound);
889 : edgomez 195 mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
890 :     mb->mvs[0].x;
891 :     mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
892 :     mb->mvs[0].y;
893 : h 69 }
894 : chl 339 } else if (mb->mode == MODE_INTER4V ) {
895 :    
896 : chenm001 272 get_motion_vector(dec, bs, x, y, 0, &mb->mvs[0], fcode, bound);
897 :     get_motion_vector(dec, bs, x, y, 1, &mb->mvs[1], fcode, bound);
898 :     get_motion_vector(dec, bs, x, y, 2, &mb->mvs[2], fcode, bound);
899 :     get_motion_vector(dec, bs, x, y, 3, &mb->mvs[3], fcode, bound);
900 : edgomez 195 } else // MODE_INTRA, MODE_INTRA_Q
901 : Isibaar 3 {
902 : edgomez 195 mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x =
903 :     0;
904 :     mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y =
905 :     0;
906 :     decoder_mbintra(dec, mb, x, y, acpred_flag, cbp, bs, quant,
907 : suxen_drol 695 intra_dc_threshold, bound, reduced_resolution);
908 : Isibaar 3 continue;
909 :     }
910 :    
911 : edgomez 195 decoder_mbinter(dec, mb, x, y, acpred_flag, cbp, bs, quant,
912 : suxen_drol 695 rounding, reduced_resolution);
913 : suxen_drol 631
914 :     }
915 :     else if (gmc_mv) /* not coded S_VOP macroblock */
916 : Isibaar 3 {
917 : chenm001 156 mb->mode = MODE_NOT_CODED;
918 : suxen_drol 631 mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = gmc_sanitize(gmc_mv[0].x, dec->quarterpel, fcode);
919 :     mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = gmc_sanitize(gmc_mv[0].y, dec->quarterpel, fcode);
920 : suxen_drol 695 decoder_mbinter(dec, mb, x, y, 0, 0, bs, quant, rounding, reduced_resolution);
921 : suxen_drol 631 }
922 :     else /* not coded P_VOP macroblock */
923 :     {
924 :     mb->mode = MODE_NOT_CODED;
925 : suxen_drol 695
926 : Isibaar 3 mb->mvs[0].x = mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = 0;
927 :     mb->mvs[0].y = mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = 0;
928 :     // copy macroblock directly from ref to cur
929 :    
930 :     start_timer();
931 :    
932 : suxen_drol 695 if (reduced_resolution)
933 :     {
934 :     transfer32x32_copy(dec->cur.y + (32*y)*dec->edged_width + (32*x),
935 :     dec->refn[0].y + (32*y)*dec->edged_width + (32*x),
936 :     dec->edged_width);
937 : Isibaar 3
938 : suxen_drol 695 transfer16x16_copy(dec->cur.u + (16*y)*dec->edged_width/2 + (16*x),
939 :     dec->refn[0].u + (16*y)*dec->edged_width/2 + (16*x),
940 :     dec->edged_width/2);
941 : Isibaar 3
942 : suxen_drol 695 transfer16x16_copy(dec->cur.v + (16*y)*dec->edged_width/2 + (16*x),
943 :     dec->refn[0].v + (16*y)*dec->edged_width/2 + (16*x),
944 :     dec->edged_width/2);
945 :     }
946 :     else
947 :     {
948 :     transfer16x16_copy(dec->cur.y + (16*y)*dec->edged_width + (16*x),
949 :     dec->refn[0].y + (16*y)*dec->edged_width + (16*x),
950 :     dec->edged_width);
951 : Isibaar 3
952 : suxen_drol 695 transfer8x8_copy(dec->cur.u + (8*y)*dec->edged_width/2 + (8*x),
953 :     dec->refn[0].u + (8*y)*dec->edged_width/2 + (8*x),
954 :     dec->edged_width/2);
955 : Isibaar 3
956 : suxen_drol 695 transfer8x8_copy(dec->cur.v + (8*y)*dec->edged_width/2 + (8*x),
957 :     dec->refn[0].v + (8*y)*dec->edged_width/2 + (8*x),
958 :     dec->edged_width/2);
959 :     }
960 : suxen_drol 631
961 : Isibaar 3 stop_transfer_timer();
962 : suxen_drol 631
963 : albeu 315 if(dec->out_frm && cp_mb > 0) {
964 :     output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,st_mb,y,cp_mb);
965 :     cp_mb = 0;
966 :     }
967 :     st_mb = x+1;
968 : Isibaar 3 }
969 :     }
970 : albeu 315 if(dec->out_frm && cp_mb > 0)
971 :     output_slice(&dec->cur, dec->edged_width,dec->width,dec->out_frm,st_mb,y,cp_mb);
972 : Isibaar 3 }
973 :     }
974 :    
975 : chenm001 156
976 :     // add by MinChen <chenm001@163.com>
977 :     // decode B-frame motion vector
978 : edgomez 195 void
979 :     get_b_motion_vector(DECODER * dec,
980 :     Bitstream * bs,
981 :     int x,
982 :     int y,
983 :     VECTOR * mv,
984 :     int fcode,
985 :     const VECTOR pmv)
986 : chenm001 156 {
987 :     int scale_fac = 1 << (fcode - 1);
988 :     int high = (32 * scale_fac) - 1;
989 :     int low = ((-32) * scale_fac);
990 :     int range = (64 * scale_fac);
991 :    
992 :     int mv_x, mv_y;
993 :     int pmv_x, pmv_y;
994 :    
995 :     pmv_x = pmv.x;
996 :     pmv_y = pmv.y;
997 :    
998 :     mv_x = get_mv(bs, fcode);
999 :     mv_y = get_mv(bs, fcode);
1000 : edgomez 195
1001 : chenm001 156 mv_x += pmv_x;
1002 :     mv_y += pmv_y;
1003 :    
1004 : edgomez 195 if (mv_x < low) {
1005 : chenm001 156 mv_x += range;
1006 : edgomez 195 } else if (mv_x > high) {
1007 : chenm001 156 mv_x -= range;
1008 :     }
1009 :    
1010 : edgomez 195 if (mv_y < low) {
1011 : chenm001 156 mv_y += range;
1012 : edgomez 195 } else if (mv_y > high) {
1013 : chenm001 156 mv_y -= range;
1014 :     }
1015 :    
1016 :     mv->x = mv_x;
1017 :     mv->y = mv_y;
1018 :     }
1019 :    
1020 :    
1021 :     // add by MinChen <chenm001@163.com>
1022 :     // decode an B-frame forward & backward inter macroblock
1023 : edgomez 195 void
1024 :     decoder_bf_mbinter(DECODER * dec,
1025 :     const MACROBLOCK * pMB,
1026 :     const uint32_t x_pos,
1027 :     const uint32_t y_pos,
1028 :     const uint32_t cbp,
1029 :     Bitstream * bs,
1030 :     const uint32_t quant,
1031 :     const uint8_t ref)
1032 : chenm001 156 {
1033 :    
1034 : edgomez 195 DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
1035 : chenm001 156 DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
1036 :    
1037 :     uint32_t stride = dec->edged_width;
1038 :     uint32_t stride2 = stride / 2;
1039 :     uint32_t next_block = stride * 8;
1040 :     uint32_t i;
1041 :     uint32_t iQuant = pMB->quant;
1042 :     uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
1043 :     int uv_dx, uv_dy;
1044 :    
1045 :     pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
1046 :     pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
1047 :     pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
1048 :    
1049 : chenm001 297
1050 : edgomez 195 if (!(pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) {
1051 : chenm001 156 uv_dx = pMB->mvs[0].x;
1052 :     uv_dy = pMB->mvs[0].y;
1053 :    
1054 : Isibaar 664 if (dec->quarterpel)
1055 :     {
1056 :     uv_dx /= 2;
1057 :     uv_dy /= 2;
1058 :     }
1059 :    
1060 :     uv_dx = (uv_dx >> 1) + roundtab_79[uv_dx & 0x3];
1061 :     uv_dy = (uv_dy >> 1) + roundtab_79[uv_dy & 0x3];
1062 : edgomez 195 } else {
1063 : chenm001 156 int sum;
1064 : edgomez 195
1065 : Isibaar 664 if(dec->quarterpel)
1066 :     sum = (pMB->mvs[0].x / 2) + (pMB->mvs[1].x / 2) + (pMB->mvs[2].x / 2) + (pMB->mvs[3].x / 2);
1067 :     else
1068 :     sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
1069 : chenm001 156
1070 : Isibaar 664 uv_dx = (sum >> 3) + roundtab_76[sum & 0xf];
1071 :    
1072 :     if(dec->quarterpel)
1073 :     sum = (pMB->mvs[0].y / 2) + (pMB->mvs[1].y / 2) + (pMB->mvs[2].y / 2) + (pMB->mvs[3].y / 2);
1074 :     else
1075 :     sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
1076 :    
1077 :     uv_dy = (sum >> 3) + roundtab_76[sum & 0xf];
1078 : chenm001 156 }
1079 :    
1080 :     start_timer();
1081 : Isibaar 664 if(dec->quarterpel) {
1082 :     interpolate16x16_quarterpel(dec->cur.y, dec->refn[ref].y, dec->refh.y, dec->refh.y + 64,
1083 :     dec->refh.y + 128, 16*x_pos, 16*y_pos,
1084 :     pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1085 :     }
1086 :     else {
1087 :     interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos, 16*y_pos,
1088 :     pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1089 :     interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos + 8, 16*y_pos,
1090 :     pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
1091 :     interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos, 16*y_pos + 8,
1092 :     pMB->mvs[2].x, pMB->mvs[2].y, stride, 0);
1093 :     interpolate8x8_switch(dec->cur.y, dec->refn[ref].y, 16*x_pos + 8, 16*y_pos + 8,
1094 :     pMB->mvs[3].x, pMB->mvs[3].y, stride, 0);
1095 :     }
1096 :    
1097 : edgomez 195 interpolate8x8_switch(dec->cur.u, dec->refn[ref].u, 8 * x_pos, 8 * y_pos,
1098 :     uv_dx, uv_dy, stride2, 0);
1099 :     interpolate8x8_switch(dec->cur.v, dec->refn[ref].v, 8 * x_pos, 8 * y_pos,
1100 :     uv_dx, uv_dy, stride2, 0);
1101 : chenm001 156 stop_comp_timer();
1102 :    
1103 : edgomez 195 for (i = 0; i < 6; i++) {
1104 : h 543 int direction = dec->alternate_vertical_scan ? 2 : 0;
1105 :    
1106 : edgomez 195 if (cbp & (1 << (5 - i))) // coded
1107 : chenm001 156 {
1108 : edgomez 195 memset(&block[i * 64], 0, 64 * sizeof(int16_t)); // clear
1109 : chenm001 156
1110 :     start_timer();
1111 : h 543 get_inter_block(bs, &block[i * 64], direction);
1112 : chenm001 156 stop_coding_timer();
1113 :    
1114 :     start_timer();
1115 : edgomez 195 if (dec->quant_type == 0) {
1116 :     dequant_inter(&data[i * 64], &block[i * 64], iQuant);
1117 :     } else {
1118 :     dequant4_inter(&data[i * 64], &block[i * 64], iQuant);
1119 : chenm001 156 }
1120 :     stop_iquant_timer();
1121 :    
1122 :     start_timer();
1123 : edgomez 195 idct(&data[i * 64]);
1124 : chenm001 156 stop_idct_timer();
1125 :     }
1126 :     }
1127 :    
1128 : edgomez 195 if (dec->interlacing && pMB->field_dct) {
1129 : chenm001 156 next_block = stride;
1130 :     stride *= 2;
1131 :     }
1132 :    
1133 :     start_timer();
1134 :     if (cbp & 32)
1135 : edgomez 195 transfer_16to8add(pY_Cur, &data[0 * 64], stride);
1136 : chenm001 156 if (cbp & 16)
1137 : edgomez 195 transfer_16to8add(pY_Cur + 8, &data[1 * 64], stride);
1138 : chenm001 156 if (cbp & 8)
1139 : edgomez 195 transfer_16to8add(pY_Cur + next_block, &data[2 * 64], stride);
1140 : chenm001 156 if (cbp & 4)
1141 : edgomez 195 transfer_16to8add(pY_Cur + 8 + next_block, &data[3 * 64], stride);
1142 : chenm001 156 if (cbp & 2)
1143 : edgomez 195 transfer_16to8add(pU_Cur, &data[4 * 64], stride2);
1144 : chenm001 156 if (cbp & 1)
1145 : edgomez 195 transfer_16to8add(pV_Cur, &data[5 * 64], stride2);
1146 : chenm001 156 stop_transfer_timer();
1147 :     }
1148 :    
1149 :     // add by MinChen <chenm001@163.com>
1150 :     // decode an B-frame direct & inter macroblock
1151 : edgomez 195 void
1152 :     decoder_bf_interpolate_mbinter(DECODER * dec,
1153 :     IMAGE forward,
1154 :     IMAGE backward,
1155 :     const MACROBLOCK * pMB,
1156 :     const uint32_t x_pos,
1157 :     const uint32_t y_pos,
1158 :     Bitstream * bs)
1159 : chenm001 156 {
1160 :    
1161 : edgomez 195 DECLARE_ALIGNED_MATRIX(block, 6, 64, int16_t, CACHE_LINE);
1162 : chenm001 156 DECLARE_ALIGNED_MATRIX(data, 6, 64, int16_t, CACHE_LINE);
1163 :    
1164 : edgomez 195 uint32_t stride = dec->edged_width;
1165 :     uint32_t stride2 = stride / 2;
1166 :     uint32_t next_block = stride * 8;
1167 :     uint32_t iQuant = pMB->quant;
1168 :     int uv_dx, uv_dy;
1169 :     int b_uv_dx, b_uv_dy;
1170 :     uint32_t i;
1171 :     uint8_t *pY_Cur, *pU_Cur, *pV_Cur;
1172 : chenm001 297 const uint32_t cbp = pMB->cbp;
1173 : chenm001 156
1174 : chenm001 161 pY_Cur = dec->cur.y + (y_pos << 4) * stride + (x_pos << 4);
1175 :     pU_Cur = dec->cur.u + (y_pos << 3) * stride2 + (x_pos << 3);
1176 :     pV_Cur = dec->cur.v + (y_pos << 3) * stride2 + (x_pos << 3);
1177 :    
1178 : chenm001 297
1179 : edgomez 195 if ((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q)) {
1180 : chenm001 156 uv_dx = pMB->mvs[0].x;
1181 :     uv_dy = pMB->mvs[0].y;
1182 :    
1183 :     b_uv_dx = pMB->b_mvs[0].x;
1184 :     b_uv_dy = pMB->b_mvs[0].y;
1185 :    
1186 : Isibaar 664 if (dec->quarterpel)
1187 :     {
1188 :     uv_dx /= 2;
1189 :     uv_dy /= 2;
1190 :    
1191 :     b_uv_dx /= 2;
1192 :     b_uv_dy /= 2;
1193 :     }
1194 :    
1195 :     uv_dx = (uv_dx >> 1) + roundtab_79[uv_dx & 0x3];
1196 :     uv_dy = (uv_dy >> 1) + roundtab_79[uv_dy & 0x3];
1197 :    
1198 :     b_uv_dx = (b_uv_dx >> 1) + roundtab_79[b_uv_dx & 0x3];
1199 :     b_uv_dy = (b_uv_dy >> 1) + roundtab_79[b_uv_dy & 0x3];
1200 : edgomez 195 } else {
1201 : chenm001 156 int sum;
1202 : edgomez 195
1203 : Isibaar 664 if(dec->quarterpel)
1204 :     sum = (pMB->mvs[0].x / 2) + (pMB->mvs[1].x / 2) + (pMB->mvs[2].x / 2) + (pMB->mvs[3].x / 2);
1205 :     else
1206 :     sum = pMB->mvs[0].x + pMB->mvs[1].x + pMB->mvs[2].x + pMB->mvs[3].x;
1207 : chenm001 156
1208 : Isibaar 664 uv_dx = (sum >> 3) + roundtab_76[sum & 0xf];
1209 : chenm001 156
1210 : Isibaar 664 if(dec->quarterpel)
1211 :     sum = (pMB->mvs[0].y / 2) + (pMB->mvs[1].y / 2) + (pMB->mvs[2].y / 2) + (pMB->mvs[3].y / 2);
1212 :     else
1213 :     sum = pMB->mvs[0].y + pMB->mvs[1].y + pMB->mvs[2].y + pMB->mvs[3].y;
1214 : chenm001 156
1215 : Isibaar 664 uv_dy = (sum >> 3) + roundtab_76[sum & 0xf];
1216 :    
1217 :    
1218 :     if(dec->quarterpel)
1219 :     sum = (pMB->b_mvs[0].x / 2) + (pMB->b_mvs[1].x / 2) + (pMB->b_mvs[2].x / 2) + (pMB->b_mvs[3].x / 2);
1220 :     else
1221 :     sum = pMB->b_mvs[0].x + pMB->b_mvs[1].x + pMB->b_mvs[2].x + pMB->b_mvs[3].x;
1222 :    
1223 :     b_uv_dx = (sum >> 3) + roundtab_76[sum & 0xf];
1224 :    
1225 :     if(dec->quarterpel)
1226 :     sum = (pMB->b_mvs[0].y / 2) + (pMB->b_mvs[1].y / 2) + (pMB->b_mvs[2].y / 2) + (pMB->b_mvs[3].y / 2);
1227 :     else
1228 :     sum = pMB->b_mvs[0].y + pMB->b_mvs[1].y + pMB->b_mvs[2].y + pMB->b_mvs[3].y;
1229 :    
1230 :     b_uv_dy = (sum >> 3) + roundtab_76[sum & 0xf];
1231 : chenm001 156 }
1232 :    
1233 : edgomez 195
1234 : chenm001 156 start_timer();
1235 : Isibaar 664 if(dec->quarterpel) {
1236 :     if((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q))
1237 :     interpolate16x16_quarterpel(dec->cur.y, forward.y, dec->refh.y, dec->refh.y + 64,
1238 :     dec->refh.y + 128, 16*x_pos, 16*y_pos,
1239 :     pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1240 :     else {
1241 :     interpolate8x8_quarterpel(dec->cur.y, forward.y, dec->refh.y, dec->refh.y + 64,
1242 :     dec->refh.y + 128, 16*x_pos, 16*y_pos,
1243 :     pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1244 :     interpolate8x8_quarterpel(dec->cur.y, forward.y, dec->refh.y, dec->refh.y + 64,
1245 :     dec->refh.y + 128, 16*x_pos + 8, 16*y_pos,
1246 :     pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
1247 :     interpolate8x8_quarterpel(dec->cur.y, forward.y, dec->refh.y, dec->refh.y + 64,
1248 :     dec->refh.y + 128, 16*x_pos, 16*y_pos + 8,
1249 :     pMB->mvs[2].x, pMB->mvs[2].y, stride, 0);
1250 :     interpolate8x8_quarterpel(dec->cur.y, forward.y, dec->refh.y, dec->refh.y + 64,
1251 :     dec->refh.y + 128, 16*x_pos + 8, 16*y_pos + 8,
1252 :     pMB->mvs[3].x, pMB->mvs[3].y, stride, 0);
1253 :     }
1254 :     }
1255 :     else {
1256 :     interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos,
1257 :     pMB->mvs[0].x, pMB->mvs[0].y, stride, 0);
1258 :     interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8, 16 * y_pos,
1259 :     pMB->mvs[1].x, pMB->mvs[1].y, stride, 0);
1260 :     interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos, 16 * y_pos + 8,
1261 :     pMB->mvs[2].x, pMB->mvs[2].y, stride, 0);
1262 :     interpolate8x8_switch(dec->cur.y, forward.y, 16 * x_pos + 8,
1263 :     16 * y_pos + 8, pMB->mvs[3].x, pMB->mvs[3].y, stride,
1264 :     0);
1265 :     }
1266 :    
1267 : edgomez 195 interpolate8x8_switch(dec->cur.u, forward.u, 8 * x_pos, 8 * y_pos, uv_dx,
1268 :     uv_dy, stride2, 0);
1269 :     interpolate8x8_switch(dec->cur.v, forward.v, 8 * x_pos, 8 * y_pos, uv_dx,
1270 :     uv_dy, stride2, 0);
1271 : chenm001 156
1272 : chenm001 161
1273 : Isibaar 664 if(dec->quarterpel) {
1274 :     if((pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q))
1275 :     interpolate16x16_quarterpel(dec->refn[2].y, backward.y, dec->refh.y, dec->refh.y + 64,
1276 :     dec->refh.y + 128, 16*x_pos, 16*y_pos,
1277 :     pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride, 0);
1278 :     else {
1279 :     interpolate8x8_quarterpel(dec->refn[2].y, backward.y, dec->refh.y, dec->refh.y + 64,
1280 :     dec->refh.y + 128, 16*x_pos, 16*y_pos,
1281 :     pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride, 0);
1282 :     interpolate8x8_quarterpel(dec->refn[2].y, backward.y, dec->refh.y, dec->refh.y + 64,
1283 :     dec->refh.y + 128, 16*x_pos + 8, 16*y_pos,
1284 :     pMB->b_mvs[1].x, pMB->b_mvs[1].y, stride, 0);
1285 :     interpolate8x8_quarterpel(dec->refn[2].y, backward.y, dec->refh.y, dec->refh.y + 64,
1286 :     dec->refh.y + 128, 16*x_pos, 16*y_pos + 8,
1287 :     pMB->b_mvs[2].x, pMB->b_mvs[2].y, stride, 0);
1288 :     interpolate8x8_quarterpel(dec->refn[2].y, backward.y, dec->refh.y, dec->refh.y + 64,
1289 :     dec->refh.y + 128, 16*x_pos + 8, 16*y_pos + 8,
1290 :     pMB->b_mvs[3].x, pMB->b_mvs[3].y, stride, 0);
1291 :     }
1292 :     }
1293 :     else {
1294 :     interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos, 16 * y_pos,
1295 :     pMB->b_mvs[0].x, pMB->b_mvs[0].y, stride, 0);
1296 :     interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos + 8,
1297 :     16 * y_pos, pMB->b_mvs[1].x, pMB->b_mvs[1].y, stride,
1298 :     0);
1299 :     interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos,
1300 :     16 * y_pos + 8, pMB->b_mvs[2].x, pMB->b_mvs[2].y,
1301 :     stride, 0);
1302 :     interpolate8x8_switch(dec->refn[2].y, backward.y, 16 * x_pos + 8,
1303 :     16 * y_pos + 8, pMB->b_mvs[3].x, pMB->b_mvs[3].y,
1304 :     stride, 0);
1305 :     }
1306 :    
1307 : edgomez 195 interpolate8x8_switch(dec->refn[2].u, backward.u, 8 * x_pos, 8 * y_pos,
1308 :     b_uv_dx, b_uv_dy, stride2, 0);
1309 :     interpolate8x8_switch(dec->refn[2].v, backward.v, 8 * x_pos, 8 * y_pos,
1310 :     b_uv_dx, b_uv_dy, stride2, 0);
1311 : chenm001 156
1312 : Isibaar 579 interpolate8x8_avg2(dec->cur.y + (16 * y_pos * stride) + 16 * x_pos,
1313 :     dec->cur.y + (16 * y_pos * stride) + 16 * x_pos,
1314 :     dec->refn[2].y + (16 * y_pos * stride) + 16 * x_pos,
1315 : Isibaar 667 stride, 1, 8);
1316 : Isibaar 579
1317 :     interpolate8x8_avg2(dec->cur.y + (16 * y_pos * stride) + 16 * x_pos + 8,
1318 :     dec->cur.y + (16 * y_pos * stride) + 16 * x_pos + 8,
1319 :     dec->refn[2].y + (16 * y_pos * stride) + 16 * x_pos + 8,
1320 : Isibaar 667 stride, 1, 8);
1321 : Isibaar 579
1322 : Isibaar 589 interpolate8x8_avg2(dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos,
1323 :     dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos,
1324 :     dec->refn[2].y + ((16 * y_pos + 8) * stride) + 16 * x_pos,
1325 : Isibaar 667 stride, 1, 8);
1326 : Isibaar 579
1327 : Isibaar 589 interpolate8x8_avg2(dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos + 8,
1328 :     dec->cur.y + ((16 * y_pos + 8) * stride) + 16 * x_pos + 8,
1329 :     dec->refn[2].y + ((16 * y_pos + 8) * stride) + 16 * x_pos + 8,
1330 : Isibaar 667 stride, 1, 8);
1331 : Isibaar 579
1332 : Isibaar 589 interpolate8x8_avg2(dec->cur.u + (8 * y_pos * stride2) + 8 * x_pos,
1333 :     dec->cur.u + (8 * y_pos * stride2) + 8 * x_pos,
1334 :     dec->refn[2].u + (8 * y_pos * stride2) + 8 * x_pos,
1335 : Isibaar 667 stride2, 1, 8);
1336 : Isibaar 579
1337 : Isibaar 589 interpolate8x8_avg2(dec->cur.v + (8 * y_pos * stride2) + 8 * x_pos,
1338 :     dec->cur.v + (8 * y_pos * stride2) + 8 * x_pos,
1339 :     dec->refn[2].v + (8 * y_pos * stride2) + 8 * x_pos,
1340 : Isibaar 667 stride2, 1, 8);
1341 : Isibaar 579
1342 : chenm001 156 stop_comp_timer();
1343 : edgomez 195
1344 :     for (i = 0; i < 6; i++) {
1345 : h 543 int direction = dec->alternate_vertical_scan ? 2 : 0;
1346 :    
1347 : edgomez 195 if (cbp & (1 << (5 - i))) // coded
1348 : chenm001 161 {
1349 : edgomez 195 memset(&block[i * 64], 0, 64 * sizeof(int16_t)); // clear
1350 : chenm001 156
1351 : chenm001 161 start_timer();
1352 : h 543 get_inter_block(bs, &block[i * 64], direction);
1353 : chenm001 161 stop_coding_timer();
1354 :    
1355 :     start_timer();
1356 : edgomez 195 if (dec->quant_type == 0) {
1357 :     dequant_inter(&data[i * 64], &block[i * 64], iQuant);
1358 :     } else {
1359 :     dequant4_inter(&data[i * 64], &block[i * 64], iQuant);
1360 : chenm001 161 }
1361 :     stop_iquant_timer();
1362 :    
1363 :     start_timer();
1364 : edgomez 195 idct(&data[i * 64]);
1365 : chenm001 161 stop_idct_timer();
1366 :     }
1367 :     }
1368 :    
1369 : edgomez 195 if (dec->interlacing && pMB->field_dct) {
1370 : chenm001 161 next_block = stride;
1371 :     stride *= 2;
1372 :     }
1373 :    
1374 :     start_timer();
1375 :     if (cbp & 32)
1376 : edgomez 195 transfer_16to8add(pY_Cur, &data[0 * 64], stride);
1377 : chenm001 161 if (cbp & 16)
1378 : edgomez 195 transfer_16to8add(pY_Cur + 8, &data[1 * 64], stride);
1379 : chenm001 161 if (cbp & 8)
1380 : edgomez 195 transfer_16to8add(pY_Cur + next_block, &data[2 * 64], stride);
1381 : chenm001 161 if (cbp & 4)
1382 : edgomez 195 transfer_16to8add(pY_Cur + 8 + next_block, &data[3 * 64], stride);
1383 : chenm001 161 if (cbp & 2)
1384 : edgomez 195 transfer_16to8add(pU_Cur, &data[4 * 64], stride2);
1385 : chenm001 161 if (cbp & 1)
1386 : edgomez 195 transfer_16to8add(pV_Cur, &data[5 * 64], stride2);
1387 : chenm001 161 stop_transfer_timer();
1388 : chenm001 156 }
1389 :    
1390 :    
1391 :     // add by MinChen <chenm001@163.com>
1392 :     // for decode B-frame dbquant
1393 : edgomez 195 int32_t __inline
1394 :     get_dbquant(Bitstream * bs)
1395 : chenm001 156 {
1396 : edgomez 195 if (!BitstreamGetBit(bs)) // '0'
1397 :     return (0);
1398 : chenm001 156 else if (!BitstreamGetBit(bs)) // '10'
1399 : edgomez 195 return (-2);
1400 : chenm001 156 else
1401 : edgomez 195 return (2); // '11'
1402 : chenm001 156 }
1403 :    
1404 :     // add by MinChen <chenm001@163.com>
1405 :     // for decode B-frame mb_type
1406 :     // bit ret_value
1407 :     // 1 0
1408 :     // 01 1
1409 :     // 001 2
1410 :     // 0001 3
1411 : edgomez 195 int32_t __inline
1412 :     get_mbtype(Bitstream * bs)
1413 : chenm001 156 {
1414 : chenm001 161 int32_t mb_type;
1415 : edgomez 195
1416 :     for (mb_type = 0; mb_type <= 3; mb_type++) {
1417 :     if (BitstreamGetBit(bs))
1418 : chenm001 161 break;
1419 :     }
1420 :    
1421 : edgomez 195 if (mb_type <= 3)
1422 :     return (mb_type);
1423 : chenm001 156 else
1424 : edgomez 195 return (-1);
1425 : chenm001 156 }
1426 :    
1427 : edgomez 195 void
1428 :     decoder_bframe(DECODER * dec,
1429 :     Bitstream * bs,
1430 :     int quant,
1431 :     int fcode_forward,
1432 :     int fcode_backward)
1433 : chenm001 156 {
1434 : edgomez 195 uint32_t x, y;
1435 : chenm001 297 VECTOR mv;
1436 :     const VECTOR zeromv = {0,0};
1437 : chenm001 290 #ifdef BFRAMES_DEC_DEBUG
1438 :     FILE *fp;
1439 :     static char first=0;
1440 :     #define BFRAME_DEBUG if (!first && fp){ \
1441 :     fprintf(fp,"Y=%3d X=%3d MB=%2d CBP=%02X\n",y,x,mb->mb_type,mb->cbp); \
1442 :     }
1443 :     #endif
1444 : chenm001 156
1445 :     start_timer();
1446 : edgomez 195 image_setedges(&dec->refn[0], dec->edged_width, dec->edged_height,
1447 : h 543 dec->width, dec->height);
1448 : chenm001 297 image_setedges(&dec->refn[1], dec->edged_width, dec->edged_height,
1449 : h 543 dec->width, dec->height);
1450 : chenm001 156 stop_edges_timer();
1451 :    
1452 : chenm001 290 #ifdef BFRAMES_DEC_DEBUG
1453 :     if (!first){
1454 :     fp=fopen("C:\\XVIDDBG.TXT","w");
1455 :     }
1456 :     #endif
1457 : chenm001 161
1458 : edgomez 195 for (y = 0; y < dec->mb_height; y++) {
1459 : chenm001 161 // Initialize Pred Motion Vector
1460 : chenm001 297 dec->p_fmv = dec->p_bmv = zeromv;
1461 : edgomez 195 for (x = 0; x < dec->mb_width; x++) {
1462 :     MACROBLOCK *mb = &dec->mbs[y * dec->mb_width + x];
1463 :     MACROBLOCK *last_mb = &dec->last_mbs[y * dec->mb_width + x];
1464 : chenm001 156
1465 : chenm001 297 mv =
1466 :     mb->b_mvs[0] = mb->b_mvs[1] = mb->b_mvs[2] = mb->b_mvs[3] =
1467 :     mb->mvs[0] = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = zeromv;
1468 : chenm001 161
1469 : chenm001 156 // the last P_VOP is skip macroblock ?
1470 : edgomez 195 if (last_mb->mode == MODE_NOT_CODED) {
1471 : chenm001 161 //DEBUG2("Skip MB in B-frame at (X,Y)=!",x,y);
1472 : chenm001 290 mb->cbp = 0;
1473 :     #ifdef BFRAMES_DEC_DEBUG
1474 : chenm001 296 mb->mb_type = MODE_NOT_CODED;
1475 : chenm001 290 BFRAME_DEBUG
1476 :     #endif
1477 : chenm001 161 mb->mb_type = MODE_FORWARD;
1478 : chenm001 297 mb->quant = last_mb->quant;
1479 :     //mb->mvs[1].x = mb->mvs[2].x = mb->mvs[3].x = mb->mvs[0].x;
1480 :     //mb->mvs[1].y = mb->mvs[2].y = mb->mvs[3].y = mb->mvs[0].y;
1481 : chenm001 156
1482 : chenm001 297 decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, mb->quant, 1);
1483 : chenm001 156 continue;
1484 :     }
1485 : chenm001 161
1486 : edgomez 195 if (!BitstreamGetBit(bs)) { // modb=='0'
1487 :     const uint8_t modb2 = BitstreamGetBit(bs);
1488 : chenm001 156
1489 :     mb->mb_type = get_mbtype(bs);
1490 : edgomez 195
1491 :     if (!modb2) { // modb=='00'
1492 :     mb->cbp = BitstreamGetBits(bs, 6);
1493 : chenm001 156 } else {
1494 :     mb->cbp = 0;
1495 :     }
1496 : edgomez 195 if (mb->mb_type && mb->cbp) {
1497 : chenm001 156 quant += get_dbquant(bs);
1498 :    
1499 : edgomez 195 if (quant > 31) {
1500 : chenm001 156 quant = 31;
1501 : chenm001 272 } else if (quant < 1) {
1502 : chenm001 156 quant = 1;
1503 :     }
1504 :     }
1505 :     } else {
1506 :     mb->mb_type = MODE_DIRECT_NONE_MV;
1507 : edgomez 195 mb->cbp = 0;
1508 : chenm001 156 }
1509 :    
1510 : chenm001 297 mb->quant = quant;
1511 :     mb->mode = MODE_INTER4V;
1512 : chenm001 156 //DEBUG1("Switch bm_type=",mb->mb_type);
1513 :    
1514 : chenm001 290 #ifdef BFRAMES_DEC_DEBUG
1515 :     BFRAME_DEBUG
1516 :     #endif
1517 : chl 326
1518 : edgomez 195 switch (mb->mb_type) {
1519 : chenm001 156 case MODE_DIRECT:
1520 : chenm001 297 get_b_motion_vector(dec, bs, x, y, &mv, 1, zeromv);
1521 : chenm001 156
1522 :     case MODE_DIRECT_NONE_MV:
1523 : chl 326 {
1524 : chenm001 297 const int64_t TRB = dec->time_pp - dec->time_bp, TRD = dec->time_pp;
1525 : chenm001 156 int i;
1526 : edgomez 195
1527 :     for (i = 0; i < 4; i++) {
1528 : chenm001 297 mb->mvs[i].x = (int32_t) ((TRB * last_mb->mvs[i].x)
1529 :     / TRD + mv.x);
1530 :     mb->b_mvs[i].x = (int32_t) ((mv.x == 0)
1531 :     ? ((TRB - TRD) * last_mb->mvs[i].x)
1532 :     / TRD
1533 :     : mb->mvs[i].x - last_mb->mvs[i].x);
1534 :     mb->mvs[i].y = (int32_t) ((TRB * last_mb->mvs[i].y)
1535 :     / TRD + mv.y);
1536 :     mb->b_mvs[i].y = (int32_t) ((mv.y == 0)
1537 :     ? ((TRB - TRD) * last_mb->mvs[i].y)
1538 :     / TRD
1539 :     : mb->mvs[i].y - last_mb->mvs[i].y);
1540 : chenm001 156 }
1541 :     //DEBUG("B-frame Direct!\n");
1542 :     }
1543 : edgomez 195 decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1544 : chenm001 297 mb, x, y, bs);
1545 : chenm001 156 break;
1546 :    
1547 :     case MODE_INTERPOLATE:
1548 : edgomez 195 get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward,
1549 :     dec->p_fmv);
1550 : chenm001 297 dec->p_fmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = mb->mvs[0];
1551 : chenm001 156
1552 : edgomez 195 get_b_motion_vector(dec, bs, x, y, &mb->b_mvs[0],
1553 :     fcode_backward, dec->p_bmv);
1554 : chenm001 297 dec->p_bmv = mb->b_mvs[1] = mb->b_mvs[2] =
1555 :     mb->b_mvs[3] = mb->b_mvs[0];
1556 : chenm001 156
1557 : edgomez 195 decoder_bf_interpolate_mbinter(dec, dec->refn[1], dec->refn[0],
1558 : chenm001 297 mb, x, y, bs);
1559 : chenm001 156 //DEBUG("B-frame Bidir!\n");
1560 :     break;
1561 :    
1562 :     case MODE_BACKWARD:
1563 : edgomez 195 get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_backward,
1564 :     dec->p_bmv);
1565 : chenm001 297 dec->p_bmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = mb->mvs[0];
1566 : chenm001 161
1567 : chenm001 297 mb->mode = MODE_INTER;
1568 : chenm001 156 decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 0);
1569 :     //DEBUG("B-frame Backward!\n");
1570 :     break;
1571 :    
1572 :     case MODE_FORWARD:
1573 : edgomez 195 get_b_motion_vector(dec, bs, x, y, &mb->mvs[0], fcode_forward,
1574 :     dec->p_fmv);
1575 : chenm001 297 dec->p_fmv = mb->mvs[1] = mb->mvs[2] = mb->mvs[3] = mb->mvs[0];
1576 : chenm001 161
1577 : chenm001 297 mb->mode = MODE_INTER;
1578 : chenm001 156 decoder_bf_mbinter(dec, mb, x, y, mb->cbp, bs, quant, 1);
1579 : chenm001 161 //DEBUG("B-frame Forward!\n");
1580 : chenm001 156 break;
1581 :    
1582 :     default:
1583 : chl 326 DEBUG1("Not support B-frame mb_type =", mb->mb_type);
1584 : chenm001 156 }
1585 :    
1586 : edgomez 195 } // end of FOR
1587 : chenm001 156 }
1588 : chenm001 290 #ifdef BFRAMES_DEC_DEBUG
1589 :     if (!first){
1590 :     first=1;
1591 :     if (fp)
1592 :     fclose(fp);
1593 :     }
1594 :     #endif
1595 : chenm001 156 }
1596 :    
1597 :     // swap two MACROBLOCK array
1598 : edgomez 195 void
1599 :     mb_swap(MACROBLOCK ** mb1,
1600 :     MACROBLOCK ** mb2)
1601 : chenm001 156 {
1602 : edgomez 195 MACROBLOCK *temp = *mb1;
1603 :    
1604 :     *mb1 = *mb2;
1605 :     *mb2 = temp;
1606 : chenm001 156 }
1607 :    
1608 : edgomez 195 int
1609 :     decoder_decode(DECODER * dec,
1610 : suxen_drol 631 XVID_DEC_FRAME * frame, XVID_DEC_STATS * stats)
1611 : Isibaar 3 {
1612 : edgomez 78
1613 : Isibaar 3 Bitstream bs;
1614 :     uint32_t rounding;
1615 : suxen_drol 631 uint32_t reduced_resolution;
1616 : Isibaar 3 uint32_t quant;
1617 : chenm001 156 uint32_t fcode_forward;
1618 :     uint32_t fcode_backward;
1619 : Isibaar 3 uint32_t intra_dc_threshold;
1620 : suxen_drol 631 VECTOR gmc_mv[5];
1621 : chenm001 133 uint32_t vop_type;
1622 : suxen_drol 659 int success = 0;
1623 : Isibaar 3
1624 :     start_global_timer();
1625 : edgomez 195
1626 : albeu 315 dec->out_frm = (frame->colorspace == XVID_CSP_EXTERN) ? frame->image : NULL;
1627 :    
1628 : Isibaar 3 BitstreamInit(&bs, frame->bitstream, frame->length);
1629 :    
1630 : suxen_drol 631 // XXX: 0x7f is only valid whilst decoding vfw xvid/divx5 avi's
1631 : suxen_drol 636 if(frame->length == 1 && BitstreamShowBits(&bs, 8) == 0x7f)
1632 : suxen_drol 631 {
1633 :     if (stats)
1634 :     stats->notify = XVID_DEC_VOP;
1635 :     frame->length = 1;
1636 : Isibaar 641 image_output(&dec->refn[0], dec->width, dec->height, dec->edged_width,
1637 : suxen_drol 631 frame->image, frame->stride, frame->colorspace, dec->interlacing);
1638 : Isibaar 641 emms();
1639 : Isibaar 590 return XVID_ERR_OK;
1640 : suxen_drol 631 }
1641 : Isibaar 589
1642 : Isibaar 645 start:
1643 : chenm001 133 // add by chenm001 <chenm001@163.com>
1644 :     // for support B-frame to reference last 2 frame
1645 : Isibaar 179 dec->frames++;
1646 : suxen_drol 631
1647 :     xxx:
1648 : edgomez 195 vop_type =
1649 : suxen_drol 631 BitstreamReadHeaders(&bs, dec, &rounding, &reduced_resolution,
1650 :     &quant, &fcode_forward, &fcode_backward, &intra_dc_threshold, gmc_mv);
1651 : Isibaar 179
1652 : suxen_drol 695 //DPRINTF(DPRINTF_HEADER, "vop_type=%i", vop_type);
1653 : suxen_drol 631
1654 : suxen_drol 659 if (vop_type == -1 && success)
1655 :     goto done;
1656 :    
1657 : suxen_drol 631 if (vop_type == -2 || vop_type == -3)
1658 :     {
1659 :     if (vop_type == -3)
1660 :     decoder_resize(dec);
1661 :    
1662 :     if (stats)
1663 :     {
1664 :     stats->notify = XVID_DEC_VOL;
1665 :     stats->data.vol.general = 0;
1666 :     if (dec->interlacing)
1667 :     stats->data.vol.general |= XVID_INTERLACING;
1668 :     stats->data.vol.width = dec->width;
1669 :     stats->data.vol.height = dec->height;
1670 :     stats->data.vol.aspect_ratio = dec->aspect_ratio;
1671 :     stats->data.vol.par_width = dec->par_width;
1672 :     stats->data.vol.par_height = dec->par_height;
1673 :     frame->length = BitstreamPos(&bs) / 8;
1674 :     return XVID_ERR_OK;
1675 :     }
1676 :     goto xxx;
1677 :     }
1678 :    
1679 : edgomez 195 dec->p_bmv.x = dec->p_bmv.y = dec->p_fmv.y = dec->p_fmv.y = 0; // init pred vector to 0
1680 :    
1681 :     switch (vop_type) {
1682 :     case P_VOP:
1683 : suxen_drol 631 decoder_pframe(dec, &bs, rounding, reduced_resolution, quant,
1684 :     fcode_forward, intra_dc_threshold, NULL);
1685 : chenm001 272 #ifdef BFRAMES_DEC
1686 : edgomez 195 DEBUG1("P_VOP Time=", dec->time);
1687 : chenm001 272 #endif
1688 : Isibaar 3 break;
1689 :    
1690 : edgomez 195 case I_VOP:
1691 : suxen_drol 631 decoder_iframe(dec, &bs, reduced_resolution, quant, intra_dc_threshold);
1692 : chenm001 272 #ifdef BFRAMES_DEC
1693 : edgomez 195 DEBUG1("I_VOP Time=", dec->time);
1694 : chenm001 272 #endif
1695 : Isibaar 3 break;
1696 :    
1697 : edgomez 195 case B_VOP:
1698 : suxen_drol 233 #ifdef BFRAMES_DEC
1699 : edgomez 195 if (dec->time_pp > dec->time_bp) {
1700 :     DEBUG1("B_VOP Time=", dec->time);
1701 : chenm001 156 decoder_bframe(dec, &bs, quant, fcode_forward, fcode_backward);
1702 :     } else {
1703 :     DEBUG("broken B-frame!");
1704 :     }
1705 : suxen_drol 233 #else
1706 :     image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height);
1707 : Isibaar 179 #endif
1708 : Isibaar 3 break;
1709 : edgomez 195
1710 : suxen_drol 631 case S_VOP :
1711 :     decoder_pframe(dec, &bs, rounding, reduced_resolution, quant,
1712 :     fcode_forward, intra_dc_threshold, gmc_mv);
1713 :     break;
1714 :    
1715 : edgomez 195 case N_VOP: // vop not coded
1716 : suxen_drol 233 // when low_delay==0, N_VOP's should interpolate between the past and future frames
1717 :     image_copy(&dec->cur, &dec->refn[0], dec->edged_width, dec->height);
1718 : Isibaar 589 #ifdef BFRAMES_DEC
1719 :     DEBUG1("N_VOP Time=", dec->time);
1720 :     #endif
1721 : Isibaar 3 break;
1722 :    
1723 : edgomez 195 default:
1724 : suxen_drol 631 if (stats)
1725 :     stats->notify = 0;
1726 : suxen_drol 659
1727 :     emms();
1728 : Isibaar 3 return XVID_ERR_FAIL;
1729 :     }
1730 :    
1731 : suxen_drol 695
1732 :     if (reduced_resolution)
1733 :     {
1734 : suxen_drol 702 image_deblock_rrv(&dec->cur, dec->edged_width, dec->mbs,
1735 :     (dec->width + 31) / 32, (dec->height + 31) / 32, dec->mb_width);
1736 : suxen_drol 695 }
1737 :    
1738 : Isibaar 645 BitstreamByteAlign(&bs);
1739 : Isibaar 3
1740 : suxen_drol 233 #ifdef BFRAMES_DEC
1741 : chenm001 168 // test if no B_VOP
1742 : Isibaar 645 if (dec->low_delay || dec->frames == 0 || ((dec->packed_mode) && !(frame->length > BitstreamPos(&bs) / 8))) {
1743 : Isibaar 179 #endif
1744 : suxen_drol 631 image_output(&dec->cur, dec->width, dec->height, dec->edged_width,
1745 :     frame->image, frame->stride, frame->colorspace, dec->interlacing);
1746 : chenm001 290
1747 : suxen_drol 233 #ifdef BFRAMES_DEC
1748 : chenm001 168 } else {
1749 : Isibaar 645 if (dec->frames >= 1 && !(dec->packed_mode)) {
1750 : chenm001 168 start_timer();
1751 : suxen_drol 631 if ((vop_type == I_VOP || vop_type == P_VOP || vop_type == S_VOP)) {
1752 : edgomez 195 image_output(&dec->refn[0], dec->width, dec->height,
1753 :     dec->edged_width, frame->image, frame->stride,
1754 : suxen_drol 631 frame->colorspace, dec->interlacing);
1755 : chenm001 168 } else if (vop_type == B_VOP) {
1756 : edgomez 195 image_output(&dec->cur, dec->width, dec->height,
1757 :     dec->edged_width, frame->image, frame->stride,
1758 : suxen_drol 631 frame->colorspace, dec->interlacing);
1759 : chenm001 168 }
1760 :     stop_conv_timer();
1761 : chenm001 156 }
1762 :     }
1763 : Isibaar 179 #endif
1764 :    
1765 : suxen_drol 631 if (vop_type == I_VOP || vop_type == P_VOP || vop_type == S_VOP) {
1766 : chenm001 156 image_swap(&dec->refn[0], &dec->refn[1]);
1767 :     image_swap(&dec->cur, &dec->refn[0]);
1768 : chenm001 306
1769 : chenm001 156 // swap MACROBLOCK
1770 : chenm001 306 // the Divx will not set the low_delay flage some times
1771 :     // so follow code will wrong to not swap at that time
1772 :     // this will broken bitstream! so I'm change it,
1773 :     // But that is not the best way! can anyone tell me how
1774 :     // to do another way?
1775 :     // 18-07-2002 MinChen<chenm001@163.com>
1776 :     //if (!dec->low_delay && vop_type == P_VOP)
1777 :     if (vop_type == P_VOP)
1778 : chenm001 161 mb_swap(&dec->mbs, &dec->last_mbs);
1779 : chenm001 156 }
1780 :    
1781 : suxen_drol 659
1782 : suxen_drol 695 if (success == 0 && dec->packed_mode)
1783 :     {
1784 :     success = 1;
1785 :     // if (frame->length > BitstreamPos(&bs) / 8) // multiple vops packed together
1786 : Isibaar 645 goto start;
1787 : suxen_drol 695 }
1788 : Isibaar 645
1789 : suxen_drol 659 done :
1790 :    
1791 : Isibaar 645 frame->length = BitstreamPos(&bs) / 8;
1792 :    
1793 : suxen_drol 631 if (stats)
1794 :     {
1795 :     stats->notify = XVID_DEC_VOP;
1796 :     stats->data.vop.time_base = (int)dec->time_base;
1797 :     stats->data.vop.time_increment = 0; //XXX: todo
1798 :     }
1799 :    
1800 : Isibaar 3 emms();
1801 :    
1802 :     stop_global_timer();
1803 :    
1804 :     return XVID_ERR_OK;
1805 :     }

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