[svn] / branches / dev-api-4 / xvidcore / src / bitstream / mbcoding.c Repository:
ViewVC logotype

Annotation of /branches/dev-api-4/xvidcore/src/bitstream/mbcoding.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 845 - (view) (download)
Original Path: trunk/xvidcore/src/bitstream/mbcoding.c

1 : edgomez 451 /*****************************************************************************
2 :     *
3 :     * XVID MPEG-4 VIDEO CODEC
4 : edgomez 453 * - Macro Block coding functions -
5 : edgomez 451 *
6 :     * Copyright(C) 2002 Michael Militzer <isibaar@xvid.org>
7 :     *
8 : edgomez 655 * This file is part of XviD, a free MPEG-4 video encoder/decoder
9 : edgomez 451 *
10 : edgomez 655 * XviD is free software; you can redistribute it and/or modify it
11 :     * under the terms of the GNU General Public License as published by
12 : edgomez 451 * the Free Software Foundation; either version 2 of the License, or
13 :     * (at your option) any later version.
14 :     *
15 :     * This program is distributed in the hope that it will be useful,
16 :     * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 :     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 :     * GNU General Public License for more details.
19 :     *
20 :     * You should have received a copy of the GNU General Public License
21 :     * along with this program; if not, write to the Free Software
22 :     * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
23 :     *
24 : edgomez 655 * Under section 8 of the GNU General Public License, the copyright
25 :     * holders of XVID explicitly forbid distribution in the following
26 :     * countries:
27 : edgomez 451 *
28 : edgomez 655 * - Japan
29 :     * - United States of America
30 :     *
31 :     * Linking XviD statically or dynamically with other modules is making a
32 :     * combined work based on XviD. Thus, the terms and conditions of the
33 :     * GNU General Public License cover the whole combination.
34 :     *
35 :     * As a special exception, the copyright holders of XviD give you
36 :     * permission to link XviD with independent modules that communicate with
37 :     * XviD solely through the VFW1.1 and DShow interfaces, regardless of the
38 :     * license terms of these independent modules, and to copy and distribute
39 :     * the resulting combined work under terms of your choice, provided that
40 :     * every copy of the combined work is accompanied by a complete copy of
41 :     * the source code of XviD (the version of XviD used to produce the
42 :     * combined work), being distributed under the terms of the GNU General
43 :     * Public License plus this exception. An independent module is a module
44 :     * which is not derived from or based on XviD.
45 :     *
46 :     * Note that people who make modified versions of XviD are not obligated
47 :     * to grant this special exception for their modified versions; it is
48 :     * their choice whether to do so. The GNU General Public License gives
49 :     * permission to release a modified version without this exception; this
50 :     * exception also makes it possible to release a modified version which
51 :     * carries forward this exception.
52 :     *
53 : edgomez 845 * $Id: mbcoding.c,v 1.41 2003-02-13 17:31:33 edgomez Exp $
54 : edgomez 655 *
55 : edgomez 451 ****************************************************************************/
56 : suxen_drol 118
57 : Isibaar 100 #include <stdlib.h>
58 : Isibaar 3 #include "../portab.h"
59 :     #include "bitstream.h"
60 :     #include "zigzag.h"
61 :     #include "vlc_codes.h"
62 : Isibaar 100 #include "mbcoding.h"
63 : Isibaar 3
64 :     #include "../utils/mbfunctions.h"
65 :    
66 :     #define ABS(X) (((X)>0)?(X):-(X))
67 :     #define CLIP(X,A) (X > A) ? (A) : (X)
68 :    
69 : edgomez 766 /* #define BIGLUT */
70 : edgomez 764
71 : edgomez 766 #ifdef BIGLUT
72 : edgomez 764 #define LEVELOFFSET 2048
73 :     #else
74 :     #define LEVELOFFSET 32
75 :     #endif
76 :    
77 : edgomez 451 /*****************************************************************************
78 :     * Local data
79 :     ****************************************************************************/
80 : Isibaar 100
81 : edgomez 764 static REVERSE_EVENT DCT3D[2][4096];
82 : Isibaar 3
83 : edgomez 766 #ifdef BIGLUT
84 : edgomez 764 static VLC coeff_VLC[2][2][4096][64];
85 : edgomez 766 static VLC *intra_table, *inter_table;
86 : edgomez 764 #else
87 :     static VLC coeff_VLC[2][2][64][64];
88 :     #endif
89 : edgomez 451
90 :     /*****************************************************************************
91 : edgomez 453 * Vector Length Coding Initialization
92 : edgomez 451 ****************************************************************************/
93 :    
94 : edgomez 195 void
95 :     init_vlc_tables(void)
96 : Isibaar 3 {
97 : edgomez 816 uint32_t i, j, intra, last, run, run_esc, level, level_esc, escape, escape_len, offset;
98 : edgomez 78
99 : edgomez 766 #ifdef BIGLUT
100 :     intra_table = (VLC*)coeff_VLC[1];
101 :     inter_table = (VLC*)coeff_VLC[0];
102 : edgomez 764 #endif
103 : Isibaar 3
104 : edgomez 195
105 : edgomez 764 for (intra = 0; intra < 2; intra++)
106 :     for (i = 0; i < 4096; i++)
107 :     DCT3D[intra][i].event.level = 0;
108 : edgomez 195
109 : edgomez 764 for (intra = 0; intra < 2; intra++)
110 :     for (last = 0; last < 2; last++)
111 :     {
112 :     for (run = 0; run < 63 + last; run++)
113 : edgomez 845 for (level = 0; level < (uint32_t)(32 << intra); level++)
114 : edgomez 764 {
115 : edgomez 766 #ifdef BIGLUT
116 : edgomez 764 offset = LEVELOFFSET;
117 :     #else
118 :     offset = !intra * LEVELOFFSET;
119 :     #endif
120 :     coeff_VLC[intra][last][level + offset][run].len = 128;
121 :     }
122 :     }
123 : Isibaar 3
124 : edgomez 764 for (intra = 0; intra < 2; intra++)
125 :     for (i = 0; i < 102; i++)
126 :     {
127 : edgomez 766 #ifdef BIGLUT
128 : edgomez 764 offset = LEVELOFFSET;
129 :     #else
130 :     offset = !intra * LEVELOFFSET;
131 :     #endif
132 : edgomez 816 for (j = 0; j < (uint32_t)(1 << (12 - coeff_tab[intra][i].vlc.len)); j++)
133 : edgomez 764 {
134 :     DCT3D[intra][(coeff_tab[intra][i].vlc.code << (12 - coeff_tab[intra][i].vlc.len)) | j].len = coeff_tab[intra][i].vlc.len;
135 :     DCT3D[intra][(coeff_tab[intra][i].vlc.code << (12 - coeff_tab[intra][i].vlc.len)) | j].event = coeff_tab[intra][i].event;
136 :     }
137 : edgomez 195
138 : edgomez 764 coeff_VLC[intra][coeff_tab[intra][i].event.last][coeff_tab[intra][i].event.level + offset][coeff_tab[intra][i].event.run].code
139 :     = coeff_tab[intra][i].vlc.code << 1;
140 :     coeff_VLC[intra][coeff_tab[intra][i].event.last][coeff_tab[intra][i].event.level + offset][coeff_tab[intra][i].event.run].len
141 :     = coeff_tab[intra][i].vlc.len + 1;
142 : edgomez 766 #ifndef BIGLUT
143 : edgomez 764 if (!intra)
144 :     #endif
145 :     {
146 :     coeff_VLC[intra][coeff_tab[intra][i].event.last][offset - coeff_tab[intra][i].event.level][coeff_tab[intra][i].event.run].code
147 :     = (coeff_tab[intra][i].vlc.code << 1) | 1;
148 :     coeff_VLC[intra][coeff_tab[intra][i].event.last][offset - coeff_tab[intra][i].event.level][coeff_tab[intra][i].event.run].len
149 :     = coeff_tab[intra][i].vlc.len + 1;
150 :     }
151 :     }
152 : edgomez 195
153 : edgomez 764 for (intra = 0; intra < 2; intra++)
154 :     for (last = 0; last < 2; last++)
155 :     for (run = 0; run < 63 + last; run++)
156 :     {
157 : edgomez 816 for (level = 1; level < (uint32_t)(32 << intra); level++)
158 : edgomez 764 {
159 : edgomez 766 if (level <= max_level[intra][last][run] && run <= max_run[intra][last][level])
160 :     continue;
161 :    
162 :     #ifdef BIGLUT
163 : edgomez 764 offset = LEVELOFFSET;
164 :     #else
165 :     offset = !intra * LEVELOFFSET;
166 :     #endif
167 : edgomez 766 level_esc = level - max_level[intra][last][run];
168 :     run_esc = run - 1 - max_run[intra][last][level];
169 :    
170 :     if (level_esc <= max_level[intra][last][run] && run <= max_run[intra][last][level_esc])
171 :     {
172 :     escape = ESCAPE1;
173 :     escape_len = 7 + 1;
174 :     run_esc = run;
175 :     }
176 :     else
177 :     {
178 : edgomez 816 if (run_esc <= max_run[intra][last][level] && level <= max_level[intra][last][run_esc])
179 : edgomez 764 {
180 : edgomez 766 escape = ESCAPE2;
181 :     escape_len = 7 + 2;
182 :     level_esc = level;
183 : edgomez 764 }
184 : edgomez 766 else
185 : edgomez 764 {
186 : edgomez 766 #ifndef BIGLUT
187 : edgomez 764 if (!intra)
188 :     #endif
189 :     {
190 : edgomez 766 coeff_VLC[intra][last][level + offset][run].code
191 :     = (ESCAPE3 << 21) | (last << 20) | (run << 14) | (1 << 13) | ((level & 0xfff) << 1) | 1;
192 :     coeff_VLC[intra][last][level + offset][run].len = 30;
193 :     coeff_VLC[intra][last][offset - level][run].code
194 :     = (ESCAPE3 << 21) | (last << 20) | (run << 14) | (1 << 13) | ((-level & 0xfff) << 1) | 1;
195 :     coeff_VLC[intra][last][offset - level][run].len = 30;
196 : edgomez 764 }
197 : edgomez 766 continue;
198 : edgomez 764 }
199 : edgomez 766 }
200 :    
201 :     coeff_VLC[intra][last][level + offset][run].code
202 :     = (escape << coeff_VLC[intra][last][level_esc + offset][run_esc].len)
203 :     | coeff_VLC[intra][last][level_esc + offset][run_esc].code;
204 :     coeff_VLC[intra][last][level + offset][run].len
205 :     = coeff_VLC[intra][last][level_esc + offset][run_esc].len + escape_len;
206 :     #ifndef BIGLUT
207 : edgomez 764 if (!intra)
208 :     #endif
209 : edgomez 766 {
210 :     coeff_VLC[intra][last][offset - level][run].code
211 :     = (escape << coeff_VLC[intra][last][level_esc + offset][run_esc].len)
212 :     | coeff_VLC[intra][last][level_esc + offset][run_esc].code | 1;
213 :     coeff_VLC[intra][last][offset - level][run].len
214 :     = coeff_VLC[intra][last][level_esc + offset][run_esc].len + escape_len;
215 :     }
216 : Isibaar 3 }
217 : edgomez 766
218 :     #ifdef BIGLUT
219 : edgomez 816 for (level = (uint32_t)(32 << intra); level < 2048; level++)
220 : edgomez 764 {
221 :     coeff_VLC[intra][last][level + offset][run].code
222 :     = (ESCAPE3 << 21) | (last << 20) | (run << 14) | (1 << 13) | ((level & 0xfff) << 1) | 1;
223 :     coeff_VLC[intra][last][level + offset][run].len = 30;
224 : Isibaar 100
225 : edgomez 764 coeff_VLC[intra][last][offset - level][run].code
226 :     = (ESCAPE3 << 21) | (last << 20) | (run << 14) | (1 << 13) | ((-level & 0xfff) << 1) | 1;
227 :     coeff_VLC[intra][last][offset - level][run].len = 30;
228 : Isibaar 100 }
229 : edgomez 764 #else
230 :     if (!intra)
231 :     {
232 :     coeff_VLC[intra][last][0][run].code
233 :     = (ESCAPE3 << 21) | (last << 20) | (run << 14) | (1 << 13) | ((-32 & 0xfff) << 1) | 1;
234 :     coeff_VLC[intra][last][0][run].len = 30;
235 :     }
236 :     #endif
237 : Isibaar 3 }
238 : edgomez 845
239 :     /* Shut up the compiler -- gcc 3.3 pre release */
240 :     i = dc_threshold[0];
241 :    
242 : Isibaar 3 }
243 :    
244 : edgomez 453 /*****************************************************************************
245 :     * Local inlined functions for MB coding
246 :     ****************************************************************************/
247 :    
248 : edgomez 195 static __inline void
249 :     CodeVector(Bitstream * bs,
250 :     int32_t value,
251 :     int32_t f_code,
252 :     Statistics * pStat)
253 : Isibaar 3 {
254 : edgomez 78
255 : Isibaar 3 const int scale_factor = 1 << (f_code - 1);
256 :     const int cmp = scale_factor << 5;
257 :    
258 : edgomez 195 if (value < (-1 * cmp))
259 : Isibaar 3 value += 64 * scale_factor;
260 : edgomez 195
261 :     if (value > (cmp - 1))
262 : Isibaar 3 value -= 64 * scale_factor;
263 :    
264 : edgomez 78 pStat->iMvSum += value * value;
265 :     pStat->iMvCount++;
266 : Isibaar 3
267 : edgomez 78 if (value == 0) {
268 : edgomez 195 BitstreamPutBits(bs, mb_motion_table[32].code,
269 :     mb_motion_table[32].len);
270 : edgomez 78 } else {
271 : Isibaar 3 uint16_t length, code, mv_res, sign;
272 : edgomez 195
273 : Isibaar 3 length = 16 << f_code;
274 :     f_code--;
275 : edgomez 195
276 : Isibaar 3 sign = (value < 0);
277 :    
278 : edgomez 195 if (value >= length)
279 : Isibaar 3 value -= 2 * length;
280 : edgomez 195 else if (value < -length)
281 : Isibaar 3 value += 2 * length;
282 :    
283 : edgomez 195 if (sign)
284 : Isibaar 3 value = -value;
285 :    
286 :     value--;
287 :     mv_res = value & ((1 << f_code) - 1);
288 :     code = ((value - mv_res) >> f_code) + 1;
289 :    
290 : edgomez 195 if (sign)
291 : Isibaar 3 code = -code;
292 :    
293 :     code += 32;
294 : edgomez 195 BitstreamPutBits(bs, mb_motion_table[code].code,
295 :     mb_motion_table[code].len);
296 :    
297 :     if (f_code)
298 : Isibaar 3 BitstreamPutBits(bs, mv_res, f_code);
299 : edgomez 78 }
300 :    
301 : Isibaar 3 }
302 :    
303 : edgomez 766 #ifdef BIGLUT
304 : edgomez 764
305 : edgomez 195 static __inline void
306 :     CodeCoeff(Bitstream * bs,
307 :     const int16_t qcoeff[64],
308 :     VLC * table,
309 :     const uint16_t * zigzag,
310 :     uint16_t intra)
311 : edgomez 78 {
312 :    
313 : Isibaar 3 uint32_t j, last;
314 :     short v;
315 :     VLC *vlc;
316 : edgomez 195
317 : Isibaar 3 j = intra;
318 : Isibaar 116 last = intra;
319 : Isibaar 3
320 : edgomez 195 while (j < 64 && (v = qcoeff[zigzag[j]]) == 0)
321 :     j++;
322 :    
323 : Isibaar 3 do {
324 : edgomez 764 vlc = table + 64 * 2048 + (v << 6) + j - last;
325 : Isibaar 116 last = ++j;
326 : Isibaar 153
327 : edgomez 677 /* count zeroes */
328 : edgomez 195 while (j < 64 && (v = qcoeff[zigzag[j]]) == 0)
329 :     j++;
330 :    
331 : edgomez 677 /* write code */
332 : edgomez 195 if (j != 64) {
333 : Isibaar 3 BitstreamPutBits(bs, vlc->code, vlc->len);
334 :     } else {
335 : edgomez 764 vlc += 64 * 4096;
336 : Isibaar 3 BitstreamPutBits(bs, vlc->code, vlc->len);
337 :     break;
338 :     }
339 : edgomez 195 } while (1);
340 : edgomez 78
341 : Isibaar 3 }
342 :    
343 : edgomez 764 #else
344 :    
345 :     static __inline void
346 :     CodeCoeffInter(Bitstream * bs,
347 :     const int16_t qcoeff[64],
348 :     const uint16_t * zigzag)
349 :     {
350 :     uint32_t i, run, prev_run, code, len;
351 :     int32_t level, prev_level, level_shifted;
352 :    
353 :     i = 0;
354 :     run = 0;
355 :    
356 :     while (!(level = qcoeff[zigzag[i++]]))
357 :     run++;
358 :    
359 :     prev_level = level;
360 :     prev_run = run;
361 :     run = 0;
362 :    
363 :     while (i < 64)
364 :     {
365 :     if ((level = qcoeff[zigzag[i++]]) != 0)
366 :     {
367 :     level_shifted = prev_level + 32;
368 :     if (!(level_shifted & -64))
369 :     {
370 :     code = coeff_VLC[0][0][level_shifted][prev_run].code;
371 :     len = coeff_VLC[0][0][level_shifted][prev_run].len;
372 :     }
373 :     else
374 :     {
375 :     code = (ESCAPE3 << 21) | (prev_run << 14) | (1 << 13) | ((prev_level & 0xfff) << 1) | 1;
376 :     len = 30;
377 :     }
378 :     BitstreamPutBits(bs, code, len);
379 :     prev_level = level;
380 :     prev_run = run;
381 :     run = 0;
382 :     }
383 :     else
384 :     run++;
385 :     }
386 :    
387 :     level_shifted = prev_level + 32;
388 :     if (!(level_shifted & -64))
389 :     {
390 :     code = coeff_VLC[0][1][level_shifted][prev_run].code;
391 :     len = coeff_VLC[0][1][level_shifted][prev_run].len;
392 :     }
393 :     else
394 :     {
395 :     code = (ESCAPE3 << 21) | (1 << 20) | (prev_run << 14) | (1 << 13) | ((prev_level & 0xfff) << 1) | 1;
396 :     len = 30;
397 :     }
398 :     BitstreamPutBits(bs, code, len);
399 :     }
400 :    
401 :     static __inline void
402 :     CodeCoeffIntra(Bitstream * bs,
403 :     const int16_t qcoeff[64],
404 :     const uint16_t * zigzag)
405 :     {
406 :     uint32_t i, abs_level, run, prev_run, code, len;
407 :     int32_t level, prev_level;
408 :    
409 :     i = 1;
410 :     run = 0;
411 :    
412 :     while (!(level = qcoeff[zigzag[i++]]))
413 :     run++;
414 :    
415 :     prev_level = level;
416 :     prev_run = run;
417 :     run = 0;
418 :    
419 :     while (i < 64)
420 :     {
421 :     if ((level = qcoeff[zigzag[i++]]) != 0)
422 :     {
423 :     abs_level = ABS(prev_level);
424 :     abs_level = abs_level < 64 ? abs_level : 0;
425 :     code = coeff_VLC[1][0][abs_level][prev_run].code;
426 :     len = coeff_VLC[1][0][abs_level][prev_run].len;
427 :     if (len != 128)
428 :     code |= (prev_level < 0);
429 :     else
430 :     {
431 :     code = (ESCAPE3 << 21) | (prev_run << 14) | (1 << 13) | ((prev_level & 0xfff) << 1) | 1;
432 :     len = 30;
433 :     }
434 :     BitstreamPutBits(bs, code, len);
435 :     prev_level = level;
436 :     prev_run = run;
437 :     run = 0;
438 :     }
439 :     else
440 :     run++;
441 :     }
442 :    
443 :     abs_level = ABS(prev_level);
444 :     abs_level = abs_level < 64 ? abs_level : 0;
445 :     code = coeff_VLC[1][1][abs_level][prev_run].code;
446 :     len = coeff_VLC[1][1][abs_level][prev_run].len;
447 :     if (len != 128)
448 :     code |= (prev_level < 0);
449 :     else
450 :     {
451 :     code = (ESCAPE3 << 21) | (1 << 20) | (prev_run << 14) | (1 << 13) | ((prev_level & 0xfff) << 1) | 1;
452 :     len = 30;
453 :     }
454 :     BitstreamPutBits(bs, code, len);
455 :     }
456 :    
457 :     #endif
458 :    
459 : edgomez 453 /*****************************************************************************
460 :     * Local functions
461 :     ****************************************************************************/
462 : Isibaar 3
463 : edgomez 195 static void
464 :     CodeBlockIntra(const FRAMEINFO * frame,
465 :     const MACROBLOCK * pMB,
466 :     int16_t qcoeff[6 * 64],
467 : edgomez 78 Bitstream * bs,
468 :     Statistics * pStat)
469 : Isibaar 3 {
470 : edgomez 78
471 : Isibaar 3 uint32_t i, mcbpc, cbpy, bits;
472 :    
473 :     cbpy = pMB->cbp >> 2;
474 :    
475 : edgomez 677 /* write mcbpc */
476 : edgomez 195 if (frame->coding_type == I_VOP) {
477 : edgomez 78 mcbpc = ((pMB->mode >> 1) & 3) | ((pMB->cbp & 3) << 2);
478 : edgomez 195 BitstreamPutBits(bs, mcbpc_intra_tab[mcbpc].code,
479 :     mcbpc_intra_tab[mcbpc].len);
480 :     } else {
481 : edgomez 78 mcbpc = (pMB->mode & 7) | ((pMB->cbp & 3) << 3);
482 : edgomez 195 BitstreamPutBits(bs, mcbpc_inter_tab[mcbpc].code,
483 :     mcbpc_inter_tab[mcbpc].len);
484 : Isibaar 28 }
485 : Isibaar 3
486 : edgomez 677 /* ac prediction flag */
487 : edgomez 195 if (pMB->acpred_directions[0])
488 : edgomez 78 BitstreamPutBits(bs, 1, 1);
489 : Isibaar 3 else
490 : edgomez 78 BitstreamPutBits(bs, 0, 1);
491 : Isibaar 3
492 : edgomez 677 /* write cbpy */
493 : edgomez 195 BitstreamPutBits(bs, cbpy_tab[cbpy].code, cbpy_tab[cbpy].len);
494 : Isibaar 3
495 : edgomez 677 /* write dquant */
496 : edgomez 195 if (pMB->mode == MODE_INTRA_Q)
497 : Isibaar 3 BitstreamPutBits(bs, pMB->dquant, 2);
498 :    
499 : edgomez 677 /* write interlacing */
500 : edgomez 195 if (frame->global_flags & XVID_INTERLACING) {
501 : h 69 BitstreamPutBit(bs, pMB->field_dct);
502 :     }
503 : edgomez 677 /* code block coeffs */
504 : edgomez 195 for (i = 0; i < 6; i++) {
505 :     if (i < 4)
506 :     BitstreamPutBits(bs, dcy_tab[qcoeff[i * 64 + 0] + 255].code,
507 :     dcy_tab[qcoeff[i * 64 + 0] + 255].len);
508 : Isibaar 3 else
509 : edgomez 195 BitstreamPutBits(bs, dcc_tab[qcoeff[i * 64 + 0] + 255].code,
510 :     dcc_tab[qcoeff[i * 64 + 0] + 255].len);
511 :    
512 :     if (pMB->cbp & (1 << (5 - i))) {
513 : Isibaar 3 bits = BitstreamPos(bs);
514 :    
515 : edgomez 766 #ifdef BIGLUT
516 : edgomez 195 CodeCoeff(bs, &qcoeff[i * 64], intra_table,
517 :     scan_tables[pMB->acpred_directions[i]], 1);
518 : edgomez 764 #else
519 :     CodeCoeffIntra(bs, &qcoeff[i * 64], scan_tables[pMB->acpred_directions[i]]);
520 :     #endif
521 : Isibaar 3 bits = BitstreamPos(bs) - bits;
522 :     pStat->iTextBits += bits;
523 :     }
524 :     }
525 : edgomez 78
526 : Isibaar 3 }
527 :    
528 :    
529 : edgomez 195 static void
530 :     CodeBlockInter(const FRAMEINFO * frame,
531 :     const MACROBLOCK * pMB,
532 :     int16_t qcoeff[6 * 64],
533 : edgomez 78 Bitstream * bs,
534 :     Statistics * pStat)
535 : Isibaar 3 {
536 : edgomez 78
537 : Isibaar 3 int32_t i;
538 :     uint32_t bits, mcbpc, cbpy;
539 :    
540 : edgomez 78 mcbpc = (pMB->mode & 7) | ((pMB->cbp & 3) << 3);
541 : Isibaar 3 cbpy = 15 - (pMB->cbp >> 2);
542 :    
543 : edgomez 677 /* write mcbpc */
544 : edgomez 195 BitstreamPutBits(bs, mcbpc_inter_tab[mcbpc].code,
545 :     mcbpc_inter_tab[mcbpc].len);
546 : Isibaar 3
547 : edgomez 677 /* write cbpy */
548 : Isibaar 3 BitstreamPutBits(bs, cbpy_tab[cbpy].code, cbpy_tab[cbpy].len);
549 :    
550 : edgomez 677 /* write dquant */
551 : edgomez 195 if (pMB->mode == MODE_INTER_Q)
552 : Isibaar 3 BitstreamPutBits(bs, pMB->dquant, 2);
553 : edgomez 195
554 : edgomez 677 /* interlacing */
555 : edgomez 195 if (frame->global_flags & XVID_INTERLACING) {
556 : h 388 if (pMB->cbp) {
557 :     BitstreamPutBit(bs, pMB->field_dct);
558 : edgomez 514 DPRINTF(DPRINTF_DEBUG, "codep: field_dct: %d", pMB->field_dct);
559 : h 388 }
560 : h 69
561 : edgomez 677 /* if inter block, write field ME flag */
562 : edgomez 195 if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q) {
563 : h 69 BitstreamPutBit(bs, pMB->field_pred);
564 : edgomez 514 DPRINTF(DPRINTF_DEBUG, "codep: field_pred: %d", pMB->field_pred);
565 : h 69
566 : edgomez 677 /* write field prediction references */
567 : edgomez 195 if (pMB->field_pred) {
568 : h 69 BitstreamPutBit(bs, pMB->field_for_top);
569 :     BitstreamPutBit(bs, pMB->field_for_bot);
570 :     }
571 :     }
572 :     }
573 : edgomez 677 /* code motion vector(s) */
574 : edgomez 195 for (i = 0; i < (pMB->mode == MODE_INTER4V ? 4 : 1); i++) {
575 : suxen_drol 136 CodeVector(bs, pMB->pmvs[i].x, frame->fcode, pStat);
576 :     CodeVector(bs, pMB->pmvs[i].y, frame->fcode, pStat);
577 : Isibaar 3 }
578 :    
579 :     bits = BitstreamPos(bs);
580 : edgomez 195
581 : edgomez 677 /* code block coeffs */
582 : edgomez 195 for (i = 0; i < 6; i++)
583 :     if (pMB->cbp & (1 << (5 - i)))
584 : edgomez 766 #ifdef BIGLUT
585 : edgomez 195 CodeCoeff(bs, &qcoeff[i * 64], inter_table, scan_tables[0], 0);
586 : edgomez 764 #else
587 :     CodeCoeffInter(bs, &qcoeff[i * 64], scan_tables[0]);
588 :     #endif
589 : Isibaar 3
590 :     bits = BitstreamPos(bs) - bits;
591 :     pStat->iTextBits += bits;
592 : edgomez 78
593 : Isibaar 3 }
594 :    
595 : edgomez 453 /*****************************************************************************
596 :     * Macro Block bitstream encoding functions
597 :     ****************************************************************************/
598 : Isibaar 3
599 : edgomez 195 void
600 :     MBCoding(const FRAMEINFO * frame,
601 :     MACROBLOCK * pMB,
602 :     int16_t qcoeff[6 * 64],
603 :     Bitstream * bs,
604 :     Statistics * pStat)
605 : Isibaar 3 {
606 : edgomez 78
607 : edgomez 195 if (frame->coding_type == P_VOP) {
608 : edgomez 677 BitstreamPutBit(bs, 0); /* coded */
609 : Isibaar 3 }
610 :    
611 : chl 335 if (pMB->mode == MODE_INTRA || pMB->mode == MODE_INTRA_Q)
612 : suxen_drol 136 CodeBlockIntra(frame, pMB, qcoeff, bs, pStat);
613 : Isibaar 3 else
614 : suxen_drol 136 CodeBlockInter(frame, pMB, qcoeff, bs, pStat);
615 : edgomez 78
616 : Isibaar 3 }
617 :    
618 : chl 347
619 :     void
620 :     MBSkip(Bitstream * bs)
621 :     {
622 : edgomez 677 BitstreamPutBit(bs, 1); /* not coded */
623 : chl 347 return;
624 :     }
625 :    
626 : edgomez 453 /*****************************************************************************
627 :     * decoding stuff starts here
628 :     ****************************************************************************/
629 : suxen_drol 118
630 : edgomez 453 /*
631 :     * For IVOP addbits == 0
632 :     * For PVOP addbits == fcode - 1
633 :     * For BVOP addbits == max(fcode,bcode) - 1
634 :     * returns true or false
635 :     */
636 : Isibaar 3
637 : suxen_drol 248 int
638 :     check_resync_marker(Bitstream * bs, int addbits)
639 :     {
640 :     uint32_t nbits;
641 :     uint32_t code;
642 :     uint32_t nbitsresyncmarker = NUMBITS_VP_RESYNC_MARKER + addbits;
643 :    
644 :     nbits = BitstreamNumBitsToByteAlign(bs);
645 :     code = BitstreamShowBits(bs, nbits);
646 :    
647 :     if (code == (((uint32_t)1 << (nbits - 1)) - 1))
648 :     {
649 :     return BitstreamShowBitsFromByteAlign(bs, nbitsresyncmarker) == RESYNC_MARKER;
650 :     }
651 :    
652 :     return 0;
653 :     }
654 :    
655 :    
656 :    
657 : edgomez 195 int
658 :     get_mcbpc_intra(Bitstream * bs)
659 : Isibaar 3 {
660 : edgomez 78
661 : Isibaar 3 uint32_t index;
662 : edgomez 195
663 : suxen_drol 248 index = BitstreamShowBits(bs, 9);
664 : Isibaar 3 index >>= 3;
665 :    
666 :     BitstreamSkip(bs, mcbpc_intra_table[index].len);
667 : edgomez 78
668 : Isibaar 3 return mcbpc_intra_table[index].code;
669 : edgomez 78
670 : Isibaar 3 }
671 :    
672 : edgomez 195 int
673 :     get_mcbpc_inter(Bitstream * bs)
674 : Isibaar 3 {
675 : edgomez 78
676 : Isibaar 3 uint32_t index;
677 : suxen_drol 248
678 :     index = CLIP(BitstreamShowBits(bs, 9), 256);
679 : edgomez 195
680 :     BitstreamSkip(bs, mcbpc_inter_table[index].len);
681 : edgomez 78
682 : Isibaar 3 return mcbpc_inter_table[index].code;
683 : edgomez 78
684 : Isibaar 3 }
685 :    
686 : edgomez 195 int
687 :     get_cbpy(Bitstream * bs,
688 :     int intra)
689 : Isibaar 3 {
690 : edgomez 78
691 : Isibaar 3 int cbpy;
692 :     uint32_t index = BitstreamShowBits(bs, 6);
693 :    
694 :     BitstreamSkip(bs, cbpy_table[index].len);
695 :     cbpy = cbpy_table[index].code;
696 :    
697 : edgomez 195 if (!intra)
698 : Isibaar 3 cbpy = 15 - cbpy;
699 :    
700 :     return cbpy;
701 : edgomez 78
702 : Isibaar 3 }
703 :    
704 : edgomez 195 int
705 :     get_mv_data(Bitstream * bs)
706 : Isibaar 3 {
707 : edgomez 78
708 : Isibaar 3 uint32_t index;
709 :    
710 : edgomez 195 if (BitstreamGetBit(bs))
711 : Isibaar 3 return 0;
712 : edgomez 195
713 : Isibaar 3 index = BitstreamShowBits(bs, 12);
714 :    
715 : edgomez 195 if (index >= 512) {
716 : Isibaar 3 index = (index >> 8) - 2;
717 :     BitstreamSkip(bs, TMNMVtab0[index].len);
718 :     return TMNMVtab0[index].code;
719 :     }
720 : edgomez 195
721 :     if (index >= 128) {
722 : Isibaar 3 index = (index >> 2) - 32;
723 :     BitstreamSkip(bs, TMNMVtab1[index].len);
724 :     return TMNMVtab1[index].code;
725 :     }
726 :    
727 : edgomez 195 index -= 4;
728 : Isibaar 3
729 :     BitstreamSkip(bs, TMNMVtab2[index].len);
730 :     return TMNMVtab2[index].code;
731 : edgomez 78
732 : Isibaar 3 }
733 :    
734 : edgomez 195 int
735 :     get_mv(Bitstream * bs,
736 :     int fcode)
737 : Isibaar 3 {
738 : edgomez 78
739 : Isibaar 3 int data;
740 :     int res;
741 :     int mv;
742 :     int scale_fac = 1 << (fcode - 1);
743 :    
744 :     data = get_mv_data(bs);
745 : edgomez 195
746 :     if (scale_fac == 1 || data == 0)
747 : Isibaar 3 return data;
748 :    
749 :     res = BitstreamGetBits(bs, fcode - 1);
750 :     mv = ((ABS(data) - 1) * scale_fac) + res + 1;
751 : edgomez 195
752 : Isibaar 3 return data < 0 ? -mv : mv;
753 : edgomez 78
754 : Isibaar 3 }
755 :    
756 : edgomez 195 int
757 :     get_dc_dif(Bitstream * bs,
758 :     uint32_t dc_size)
759 : Isibaar 3 {
760 : edgomez 78
761 : Isibaar 3 int code = BitstreamGetBits(bs, dc_size);
762 :     int msb = code >> (dc_size - 1);
763 :    
764 : edgomez 195 if (msb == 0)
765 :     return (-1 * (code ^ ((1 << dc_size) - 1)));
766 : Isibaar 3
767 :     return code;
768 : edgomez 78
769 : Isibaar 3 }
770 :    
771 : edgomez 195 int
772 :     get_dc_size_lum(Bitstream * bs)
773 : Isibaar 3 {
774 : edgomez 78
775 : Isibaar 3 int code, i;
776 : edgomez 195
777 : Isibaar 3 code = BitstreamShowBits(bs, 11);
778 :    
779 : edgomez 195 for (i = 11; i > 3; i--) {
780 :     if (code == 1) {
781 : Isibaar 3 BitstreamSkip(bs, i);
782 :     return i + 1;
783 :     }
784 :     code >>= 1;
785 :     }
786 :    
787 :     BitstreamSkip(bs, dc_lum_tab[code].len);
788 :     return dc_lum_tab[code].code;
789 : edgomez 78
790 : Isibaar 3 }
791 :    
792 :    
793 : edgomez 195 int
794 :     get_dc_size_chrom(Bitstream * bs)
795 : Isibaar 3 {
796 : edgomez 78
797 : Isibaar 3 uint32_t code, i;
798 : edgomez 195
799 : Isibaar 3 code = BitstreamShowBits(bs, 12);
800 :    
801 : edgomez 195 for (i = 12; i > 2; i--) {
802 :     if (code == 1) {
803 : Isibaar 3 BitstreamSkip(bs, i);
804 :     return i;
805 :     }
806 :     code >>= 1;
807 :     }
808 :    
809 :     return 3 - BitstreamGetBits(bs, 2);
810 : edgomez 78
811 : Isibaar 3 }
812 :    
813 : edgomez 453 /*****************************************************************************
814 :     * Local inlined function to "decode" written vlc codes
815 :     ****************************************************************************/
816 :    
817 :     static __inline int
818 :     get_coeff(Bitstream * bs,
819 :     int *run,
820 :     int *last,
821 :     int intra,
822 :     int short_video_header)
823 :     {
824 :    
825 :     uint32_t mode;
826 :     int32_t level;
827 : edgomez 764 REVERSE_EVENT *reverse_event;
828 : edgomez 453
829 : edgomez 677 if (short_video_header) /* inter-VLCs will be used for both intra and inter blocks */
830 : edgomez 453 intra = 0;
831 :    
832 : edgomez 764 if (BitstreamShowBits(bs, 7) != ESCAPE) {
833 :     reverse_event = &DCT3D[intra][BitstreamShowBits(bs, 12)];
834 : edgomez 453
835 : edgomez 764 if ((level = reverse_event->event.level) == 0)
836 :     goto error;
837 : edgomez 453
838 : edgomez 764 *last = reverse_event->event.last;
839 :     *run = reverse_event->event.run;
840 : edgomez 453
841 : edgomez 764 BitstreamSkip(bs, reverse_event->len);
842 :    
843 :     return BitstreamGetBits(bs, 1) ? -level : level;
844 : edgomez 453 }
845 :    
846 : edgomez 764 BitstreamSkip(bs, 7);
847 :    
848 : edgomez 453 if (short_video_header) {
849 : edgomez 677 /* escape mode 4 - H.263 type, only used if short_video_header = 1 */
850 : edgomez 453 *last = BitstreamGetBit(bs);
851 :     *run = BitstreamGetBits(bs, 6);
852 :     level = BitstreamGetBits(bs, 8);
853 :    
854 :     if (level == 0 || level == 128)
855 : edgomez 514 DPRINTF(DPRINTF_ERROR, "Illegal LEVEL for ESCAPE mode 4: %d", level);
856 : edgomez 453
857 : edgomez 764 return (level << 24) >> 24;
858 : edgomez 453 }
859 :    
860 :     mode = BitstreamShowBits(bs, 2);
861 :    
862 :     if (mode < 3) {
863 :     BitstreamSkip(bs, (mode == 2) ? 2 : 1);
864 :    
865 : edgomez 764 reverse_event = &DCT3D[intra][BitstreamShowBits(bs, 12)];
866 :    
867 :     if ((level = reverse_event->event.level) == 0)
868 : edgomez 453 goto error;
869 :    
870 : edgomez 764 *last = reverse_event->event.last;
871 :     *run = reverse_event->event.run;
872 : edgomez 453
873 : edgomez 764 BitstreamSkip(bs, reverse_event->len);
874 : edgomez 453
875 : edgomez 677 if (mode < 2) /* first escape mode, level is offset */
876 : edgomez 764 level += max_level[intra][*last][*run];
877 :     else /* second escape mode, run is offset */
878 :     *run += max_run[intra][*last][level] + 1;
879 : edgomez 453
880 : edgomez 764 return BitstreamGetBits(bs, 1) ? -level : level;
881 : edgomez 453 }
882 : edgomez 764
883 : edgomez 677 /* third escape mode - fixed length codes */
884 : edgomez 453 BitstreamSkip(bs, 2);
885 :     *last = BitstreamGetBits(bs, 1);
886 :     *run = BitstreamGetBits(bs, 6);
887 : edgomez 677 BitstreamSkip(bs, 1); /* marker */
888 : edgomez 453 level = BitstreamGetBits(bs, 12);
889 : edgomez 677 BitstreamSkip(bs, 1); /* marker */
890 : edgomez 453
891 : edgomez 764 return (level << 20) >> 20;
892 : edgomez 453
893 :     error:
894 :     *run = VLC_ERROR;
895 :     return 0;
896 :     }
897 :    
898 :     /*****************************************************************************
899 :     * MB reading functions
900 :     ****************************************************************************/
901 :    
902 : edgomez 195 void
903 :     get_intra_block(Bitstream * bs,
904 :     int16_t * block,
905 :     int direction,
906 :     int coeff)
907 : Isibaar 3 {
908 : edgomez 78
909 : edgomez 195 const uint16_t *scan = scan_tables[direction];
910 : Isibaar 3 int level;
911 :     int run;
912 :     int last;
913 :    
914 : edgomez 195 do {
915 : Isibaar 3 level = get_coeff(bs, &run, &last, 1, 0);
916 : edgomez 195 if (run == -1) {
917 : edgomez 514 DPRINTF(DPRINTF_DEBUG, "fatal: invalid run");
918 : Isibaar 3 break;
919 :     }
920 :     coeff += run;
921 : edgomez 195 block[scan[coeff]] = level;
922 : suxen_drol 252
923 :     DPRINTF(DPRINTF_COEFF,"block[%i] %i", scan[coeff], level);
924 : edgomez 677 /*DPRINTF(DPRINTF_COEFF,"block[%i] %i %08x", scan[coeff], level, BitstreamShowBits(bs, 32)); */
925 : suxen_drol 252
926 : suxen_drol 759 if (level < -2047 || level > 2047) {
927 : edgomez 514 DPRINTF(DPRINTF_DEBUG, "warning: intra_overflow: %d", level);
928 : Isibaar 3 }
929 :     coeff++;
930 :     } while (!last);
931 : edgomez 78
932 : Isibaar 3 }
933 :    
934 : edgomez 195 void
935 :     get_inter_block(Bitstream * bs,
936 :     int16_t * block)
937 : Isibaar 3 {
938 : edgomez 78
939 : edgomez 195 const uint16_t *scan = scan_tables[0];
940 : Isibaar 3 int p;
941 :     int level;
942 :     int run;
943 :     int last;
944 :    
945 :     p = 0;
946 : edgomez 195 do {
947 : Isibaar 3 level = get_coeff(bs, &run, &last, 0, 0);
948 : edgomez 195 if (run == -1) {
949 : edgomez 514 DPRINTF(DPRINTF_ERROR, "fatal: invalid run");
950 : Isibaar 3 break;
951 :     }
952 :     p += run;
953 : chenm001 161
954 : edgomez 195 block[scan[p]] = level;
955 : suxen_drol 252
956 :     DPRINTF(DPRINTF_COEFF,"block[%i] %i", scan[p], level);
957 :    
958 : suxen_drol 759 if (level < -2047 || level > 2047) {
959 : edgomez 514 DPRINTF(DPRINTF_DEBUG, "warning: inter_overflow: %d", level);
960 : Isibaar 3 }
961 :     p++;
962 :     } while (!last);
963 : edgomez 78
964 : Isibaar 3 }

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