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

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