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

Annotation of /xvidcore/src/decoder.c

Parent Directory Parent Directory | Revision Log Revision Log

Revision 1.37.2.2 - (view) (download)

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