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

Annotation of /xvidcore/src/decoder.c

Parent Directory Parent Directory | Revision Log Revision Log

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