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

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