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

Annotation of /xvidcore/src/bitstream/mbcoding.c

Parent Directory Parent Directory | Revision Log Revision Log

Revision 1.25.2.14 - (view) (download)

1 : suxen_drol 1.11 /******************************************************************************
2 :     * *
3 :     * This file is part of XviD, a free MPEG-4 video encoder/decoder *
4 :     * *
5 :     * XviD is an implementation of a part of one or more MPEG-4 Video tools *
6 :     * as specified in ISO/IEC 14496-2 standard. Those intending to use this *
7 :     * software module in hardware or software products are advised that its *
8 :     * use may infringe existing patents or copyrights, and any such use *
9 :     * would be at such party's own risk. The original developer of this *
10 :     * software module and his/her company, and subsequent editors and their *
11 :     * companies, will have no liability for use of this software or *
12 :     * modifications or derivatives thereof. *
13 :     * *
14 :     * XviD is free software; you can redistribute it and/or modify it *
15 :     * under the terms of the GNU General Public License as published by *
16 :     * the Free Software Foundation; either version 2 of the License, or *
17 :     * (at your option) any later version. *
18 :     * *
19 :     * XviD is distributed in the hope that it will be useful, but *
20 :     * WITHOUT ANY WARRANTY; without even the implied warranty of *
21 :     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
22 :     * GNU General Public License for more details. *
23 :     * *
24 :     * You should have received a copy of the GNU General Public License *
25 :     * along with this program; if not, write to the Free Software *
26 :     * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA *
27 :     * *
28 :     ******************************************************************************/
29 :    
30 :     /******************************************************************************
31 :     * *
32 : Isibaar 1.15 * mbcoding.c *
33 : suxen_drol 1.11 * *
34 : Isibaar 1.15 * Copyright (C) 2002 - Michael Militzer <isibaar@xvid.org> *
35 : suxen_drol 1.11 * *
36 :     * For more information visit the XviD homepage: http://www.xvid.org *
37 :     * *
38 :     ******************************************************************************/
39 :    
40 :     /******************************************************************************
41 :     * *
42 :     * Revision history: *
43 :     * *
44 : edgomez 1.25.2.11 * 28.10.2002 GMC support - gruel *
45 : suxen_drol 1.19 * 28.06.2002 added check_resync_marker() *
46 : Isibaar 1.15 * 14.04.2002 bframe encoding *
47 : suxen_drol 1.11 * 08.03.2002 initial version; isibaar *
48 :     * *
49 :     ******************************************************************************/
50 :    
51 :    
52 : chl 1.25.2.10 #include <stdio.h>
53 : Isibaar 1.8 #include <stdlib.h>
54 : Isibaar 1.1 #include "../portab.h"
55 : suxen_drol 1.25.2.9 #include "../global.h"
56 : Isibaar 1.1 #include "bitstream.h"
57 :     #include "zigzag.h"
58 :     #include "vlc_codes.h"
59 : Isibaar 1.8 #include "mbcoding.h"
60 : Isibaar 1.1
61 :     #include "../utils/mbfunctions.h"
62 :    
63 : edgomez 1.25.2.11 /* #define BIGLUT */
64 : Isibaar 1.9
65 : edgomez 1.25.2.11 #ifdef BIGLUT
66 :     #define LEVELOFFSET 2048
67 :     #else
68 :     #define LEVELOFFSET 32
69 :     #endif
70 :    
71 :     static REVERSE_EVENT DCT3D[2][4096];
72 :    
73 :     #ifdef BIGLUT
74 :     static VLC coeff_VLC[2][2][4096][64];
75 : suxen_drol 1.25.2.14 VLC *intra_table;
76 :     static VLC *inter_table;
77 : edgomez 1.25.2.11 #else
78 :     static VLC coeff_VLC[2][2][64][64];
79 :     #endif
80 : Isibaar 1.1
81 : chl 1.25.2.3 /* not really MB related, but VLCs are only available here */
82 : suxen_drol 1.25.2.4 void bs_put_spritetrajectory(Bitstream * bs, const int val)
83 : chl 1.25.2.3 {
84 :     const int code = sprite_trajectory_code[val+16384].code;
85 :     const int len = sprite_trajectory_code[val+16384].len;
86 :     const int code2 = sprite_trajectory_len[len].code;
87 :     const int len2 = sprite_trajectory_len[len].len;
88 :    
89 :     // printf("GMC=%d Code/Len = %d / %d ",val, code,len);
90 :     // printf("Code2 / Len2 = %d / %d \n",code2,len2);
91 :    
92 :     BitstreamPutBits(bs, code2, len2);
93 :     if (len) BitstreamPutBits(bs, code, len);
94 :     }
95 :    
96 : suxen_drol 1.25.2.5 int bs_get_spritetrajectory(Bitstream * bs)
97 :     {
98 :     int i;
99 :     for (i = 0; i < 12; i++)
100 :     {
101 :     if (BitstreamShowBits(bs, sprite_trajectory_len[i].len) == sprite_trajectory_len[i].code)
102 :     {
103 :     BitstreamSkip(bs, sprite_trajectory_len[i].len);
104 :     return i;
105 :     }
106 :     }
107 :     return -1;
108 :     }
109 : chl 1.25.2.3
110 : edgomez 1.17 void
111 :     init_vlc_tables(void)
112 : Isibaar 1.1 {
113 : suxen_drol 1.25.2.13 ptr_t i, j, k, intra, last, run, run_esc, level, level_esc, escape, escape_len, offset;
114 : edgomez 1.25.2.11 int32_t l;
115 : edgomez 1.7
116 : edgomez 1.25.2.11 #ifdef BIGLUT
117 :     intra_table = coeff_VLC[1];
118 :     inter_table = coeff_VLC[0];
119 :     #endif
120 : edgomez 1.17
121 : Isibaar 1.8
122 : edgomez 1.25.2.11 for (intra = 0; intra < 2; intra++)
123 :     for (i = 0; i < 4096; i++)
124 :     DCT3D[intra][i].event.level = 0;
125 : Isibaar 1.8
126 : edgomez 1.25.2.11 for (intra = 0; intra < 2; intra++)
127 :     for (last = 0; last < 2; last++)
128 :     {
129 :     for (run = 0; run < 63 + last; run++)
130 :     for (level = 0; level < 32 << intra; level++)
131 :     {
132 :     #ifdef BIGLUT
133 :     offset = LEVELOFFSET;
134 :     #else
135 :     offset = !intra * LEVELOFFSET;
136 :     #endif
137 :     coeff_VLC[intra][last][level + offset][run].len = 128;
138 : Isibaar 1.8 }
139 : edgomez 1.25.2.11 }
140 : Isibaar 1.8
141 : edgomez 1.25.2.11 for (intra = 0; intra < 2; intra++)
142 :     for (i = 0; i < 102; i++)
143 :     {
144 :     #ifdef BIGLUT
145 :     offset = LEVELOFFSET;
146 :     #else
147 :     offset = !intra * LEVELOFFSET;
148 :     #endif
149 :     for (j = 0; j < 1 << (12 - coeff_tab[intra][i].vlc.len); j++)
150 :     {
151 :     DCT3D[intra][(coeff_tab[intra][i].vlc.code << (12 - coeff_tab[intra][i].vlc.len)) | j].len = coeff_tab[intra][i].vlc.len;
152 :     DCT3D[intra][(coeff_tab[intra][i].vlc.code << (12 - coeff_tab[intra][i].vlc.len)) | j].event = coeff_tab[intra][i].event;
153 : Isibaar 1.1 }
154 : edgomez 1.7
155 : edgomez 1.25.2.11 coeff_VLC[intra][coeff_tab[intra][i].event.last][coeff_tab[intra][i].event.level + offset][coeff_tab[intra][i].event.run].code
156 :     = coeff_tab[intra][i].vlc.code << 1;
157 :     coeff_VLC[intra][coeff_tab[intra][i].event.last][coeff_tab[intra][i].event.level + offset][coeff_tab[intra][i].event.run].len
158 :     = coeff_tab[intra][i].vlc.len + 1;
159 :     #ifndef BIGLUT
160 :     if (!intra)
161 :     #endif
162 :     {
163 :     coeff_VLC[intra][coeff_tab[intra][i].event.last][offset - coeff_tab[intra][i].event.level][coeff_tab[intra][i].event.run].code
164 :     = (coeff_tab[intra][i].vlc.code << 1) | 1;
165 :     coeff_VLC[intra][coeff_tab[intra][i].event.last][offset - coeff_tab[intra][i].event.level][coeff_tab[intra][i].event.run].len
166 :     = coeff_tab[intra][i].vlc.len + 1;
167 :     }
168 : Isibaar 1.1 }
169 :    
170 : edgomez 1.25.2.11 for (intra = 0; intra < 2; intra++)
171 :     for (last = 0; last < 2; last++)
172 :     for (run = 0; run < 63 + last; run++)
173 :     {
174 :     for (level = 1; level < 32 << intra; level++)
175 :     {
176 :     if (level <= max_level[intra][last][run] && run <= max_run[intra][last][level])
177 :     continue;
178 :    
179 :     #ifdef BIGLUT
180 :     offset = LEVELOFFSET;
181 :     #else
182 :     offset = !intra * LEVELOFFSET;
183 :     #endif
184 :     level_esc = level - max_level[intra][last][run];
185 :     run_esc = run - 1 - max_run[intra][last][level];
186 :     /*use this test to use shorter esc2 codes when possible
187 :     if (level_esc <= max_level[intra][last][run] && run <= max_run[intra][last][level_esc]
188 :     && !(coeff_VLC[intra][last][level_esc + offset][run].len + 7 + 1
189 :     > coeff_VLC[intra][last][level + offset][run_esc].code + 7 + 2))*/
190 :    
191 :     if (level_esc <= max_level[intra][last][run] && run <= max_run[intra][last][level_esc])
192 :     {
193 :     escape = ESCAPE1;
194 :     escape_len = 7 + 1;
195 :     run_esc = run;
196 :     }
197 :     else
198 :     {
199 :     if (level <= max_level[intra][last][run_esc] && run_esc <= max_run[intra][last][level])
200 :     {
201 :     escape = ESCAPE2;
202 :     escape_len = 7 + 2;
203 :     level_esc = level;
204 :     }
205 :     else
206 :     {
207 :     #ifndef BIGLUT
208 :     if (!intra)
209 :     #endif
210 :     {
211 :     coeff_VLC[intra][last][level + offset][run].code
212 :     = (ESCAPE3 << 21) | (last << 20) | (run << 14) | (1 << 13) | ((level & 0xfff) << 1) | 1;
213 :     coeff_VLC[intra][last][level + offset][run].len = 30;
214 :     coeff_VLC[intra][last][offset - level][run].code
215 :     = (ESCAPE3 << 21) | (last << 20) | (run << 14) | (1 << 13) | ((-level & 0xfff) << 1) | 1;
216 :     coeff_VLC[intra][last][offset - level][run].len = 30;
217 :     }
218 :     continue;
219 :     }
220 :     }
221 : Isibaar 1.1
222 : edgomez 1.25.2.11 coeff_VLC[intra][last][level + offset][run].code
223 :     = (escape << coeff_VLC[intra][last][level_esc + offset][run_esc].len)
224 :     | coeff_VLC[intra][last][level_esc + offset][run_esc].code;
225 :     coeff_VLC[intra][last][level + offset][run].len
226 :     = coeff_VLC[intra][last][level_esc + offset][run_esc].len + escape_len;
227 :     #ifndef BIGLUT
228 :     if (!intra)
229 :     #endif
230 :     {
231 :     coeff_VLC[intra][last][offset - level][run].code
232 :     = (escape << coeff_VLC[intra][last][level_esc + offset][run_esc].len)
233 :     | coeff_VLC[intra][last][level_esc + offset][run_esc].code | 1;
234 :     coeff_VLC[intra][last][offset - level][run].len
235 :     = coeff_VLC[intra][last][level_esc + offset][run_esc].len + escape_len;
236 :     }
237 :     }
238 : chl 1.25.2.3
239 : edgomez 1.25.2.11 #ifdef BIGLUT
240 :     for (level = 32 << intra; level < 2048; level++)
241 :     {
242 :     coeff_VLC[intra][last][level + offset][run].code
243 :     = (ESCAPE3 << 21) | (last << 20) | (run << 14) | (1 << 13) | ((level & 0xfff) << 1) | 1;
244 :     coeff_VLC[intra][last][level + offset][run].len = 30;
245 :    
246 :     coeff_VLC[intra][last][offset - level][run].code
247 :     = (ESCAPE3 << 21) | (last << 20) | (run << 14) | (1 << 13) | ((-level & 0xfff) << 1) | 1;
248 :     coeff_VLC[intra][last][offset - level][run].len = 30;
249 :     }
250 :     #else
251 :     if (!intra)
252 :     {
253 :     coeff_VLC[intra][last][0][run].code
254 :     = (ESCAPE3 << 21) | (last << 20) | (run << 14) | (1 << 13) | ((-32 & 0xfff) << 1) | 1;
255 :     coeff_VLC[intra][last][0][run].len = 30;
256 :     }
257 :     #endif
258 :     }
259 : chl 1.25.2.3 /* init sprite_trajectory tables */
260 :     /* even if GMC is not specified (it might be used later...) */
261 :    
262 :     sprite_trajectory_code[0+16384].code = 0;
263 :     sprite_trajectory_code[0+16384].len = 0;
264 :     for (k=0;k<14;k++)
265 :     {
266 :     int limit = (1<<k);
267 :    
268 : edgomez 1.25.2.11 for (l=-(2*limit-1); l <= -limit; l++)
269 : chl 1.25.2.3 {
270 : chl 1.25.2.12 sprite_trajectory_code[l+16384].code = (2*limit-1)+l;
271 :     sprite_trajectory_code[l+16384].len = k+1;
272 : chl 1.25.2.3 }
273 :    
274 : edgomez 1.25.2.11 for (l=limit; l<= 2*limit-1; l++)
275 : chl 1.25.2.3 {
276 : chl 1.25.2.12 sprite_trajectory_code[l+16384].code = l;
277 :     sprite_trajectory_code[l+16384].len = k+1;
278 : chl 1.25.2.3 }
279 :     }
280 : Isibaar 1.1 }
281 :    
282 : edgomez 1.17 static __inline void
283 :     CodeVector(Bitstream * bs,
284 :     int32_t value,
285 :     int32_t f_code,
286 :     Statistics * pStat)
287 : Isibaar 1.1 {
288 : edgomez 1.7
289 : Isibaar 1.1 const int scale_factor = 1 << (f_code - 1);
290 :     const int cmp = scale_factor << 5;
291 :    
292 : edgomez 1.17 if (value < (-1 * cmp))
293 : Isibaar 1.1 value += 64 * scale_factor;
294 : edgomez 1.17
295 :     if (value > (cmp - 1))
296 : Isibaar 1.1 value -= 64 * scale_factor;
297 :    
298 : edgomez 1.7 pStat->iMvSum += value * value;
299 :     pStat->iMvCount++;
300 : Isibaar 1.1
301 : edgomez 1.7 if (value == 0) {
302 : edgomez 1.17 BitstreamPutBits(bs, mb_motion_table[32].code,
303 :     mb_motion_table[32].len);
304 : edgomez 1.7 } else {
305 : Isibaar 1.1 uint16_t length, code, mv_res, sign;
306 : edgomez 1.17
307 : Isibaar 1.1 length = 16 << f_code;
308 :     f_code--;
309 : edgomez 1.17
310 : Isibaar 1.1 sign = (value < 0);
311 :    
312 : edgomez 1.17 if (value >= length)
313 : Isibaar 1.1 value -= 2 * length;
314 : edgomez 1.17 else if (value < -length)
315 : Isibaar 1.1 value += 2 * length;
316 :    
317 : edgomez 1.17 if (sign)
318 : Isibaar 1.1 value = -value;
319 :    
320 :     value--;
321 :     mv_res = value & ((1 << f_code) - 1);
322 :     code = ((value - mv_res) >> f_code) + 1;
323 :    
324 : edgomez 1.17 if (sign)
325 : Isibaar 1.1 code = -code;
326 :    
327 :     code += 32;
328 : edgomez 1.17 BitstreamPutBits(bs, mb_motion_table[code].code,
329 :     mb_motion_table[code].len);
330 :    
331 :     if (f_code)
332 : Isibaar 1.1 BitstreamPutBits(bs, mv_res, f_code);
333 : edgomez 1.7 }
334 :    
335 : Isibaar 1.1 }
336 :    
337 : edgomez 1.25.2.11 #ifdef BIGLUT
338 : Isibaar 1.1
339 : edgomez 1.17 static __inline void
340 :     CodeCoeff(Bitstream * bs,
341 :     const int16_t qcoeff[64],
342 :     VLC * table,
343 :     const uint16_t * zigzag,
344 :     uint16_t intra)
345 : edgomez 1.7 {
346 :    
347 : Isibaar 1.1 uint32_t j, last;
348 :     short v;
349 :     VLC *vlc;
350 : edgomez 1.17
351 : Isibaar 1.1 j = intra;
352 : Isibaar 1.10 last = intra;
353 : Isibaar 1.1
354 : edgomez 1.17 while (j < 64 && (v = qcoeff[zigzag[j]]) == 0)
355 :     j++;
356 :    
357 : Isibaar 1.1 do {
358 : edgomez 1.25.2.11 vlc = table + 64 * 2048 + (v << 6) + j - last;
359 : Isibaar 1.15 last = ++j;
360 :    
361 : edgomez 1.25.2.11 /* count zeroes */
362 : edgomez 1.17 while (j < 64 && (v = qcoeff[zigzag[j]]) == 0)
363 :     j++;
364 :    
365 : edgomez 1.25.2.11 /* write code */
366 : edgomez 1.17 if (j != 64) {
367 : Isibaar 1.1 BitstreamPutBits(bs, vlc->code, vlc->len);
368 :     } else {
369 : edgomez 1.25.2.11 vlc += 64 * 4096;
370 : Isibaar 1.1 BitstreamPutBits(bs, vlc->code, vlc->len);
371 :     break;
372 :     }
373 : edgomez 1.17 } while (1);
374 : edgomez 1.7
375 : Isibaar 1.1 }
376 :    
377 : suxen_drol 1.25.2.14
378 :    
379 :     /* returns the number of bits required to encode qcoeff */
380 :     int
381 :     CodeCoeff_CalcBits(const int16_t qcoeff[64],
382 :     VLC * table,
383 :     const uint16_t * zigzag,
384 :     uint16_t intra)
385 :     {
386 :     int bits = 0;
387 :     uint32_t j, last;
388 :     short v;
389 :     VLC *vlc;
390 :    
391 :     j = intra;
392 :     last = intra;
393 :    
394 :     while (j < 64 && (v = qcoeff[zigzag[j]]) == 0)
395 :     j++;
396 :    
397 :     if (j >= 64) return 0; /* empty block */
398 :    
399 :     do {
400 :     vlc = table + 64 * 2048 + (v << 6) + j - last;
401 :     last = ++j;
402 :    
403 :     /* count zeroes */
404 :     while (j < 64 && (v = qcoeff[zigzag[j]]) == 0)
405 :     j++;
406 :    
407 :     /* write code */
408 :     if (j != 64) {
409 :     bits += vlc->len;
410 :     } else {
411 :     vlc += 64 * 4096;
412 :     bits += vlc->len;
413 :     break;
414 :     }
415 :     } while (1);
416 :    
417 :     return bits;
418 :     }
419 :    
420 :    
421 : edgomez 1.25.2.11 #else
422 :    
423 :     static __inline void
424 :     CodeCoeffInter(Bitstream * bs,
425 :     const int16_t qcoeff[64],
426 :     const uint16_t * zigzag)
427 :     {
428 :     uint32_t i, run, prev_run, code, len;
429 :     int32_t level, prev_level, level_shifted;
430 :    
431 :     i = 0;
432 :     run = 0;
433 :    
434 :     while (!(level = qcoeff[zigzag[i++]]))
435 :     run++;
436 :    
437 :     prev_level = level;
438 :     prev_run = run;
439 :     run = 0;
440 :    
441 :     while (i < 64)
442 :     {
443 :     if ((level = qcoeff[zigzag[i++]]) != 0)
444 :     {
445 :     level_shifted = prev_level + 32;
446 :     if (!(level_shifted & -64))
447 :     {
448 :     code = coeff_VLC[0][0][level_shifted][prev_run].code;
449 :     len = coeff_VLC[0][0][level_shifted][prev_run].len;
450 :     }
451 :     else
452 :     {
453 :     code = (ESCAPE3 << 21) | (prev_run << 14) | (1 << 13) | ((prev_level & 0xfff) << 1) | 1;
454 :     len = 30;
455 :     }
456 :     BitstreamPutBits(bs, code, len);
457 :     prev_level = level;
458 :     prev_run = run;
459 :     run = 0;
460 :     }
461 :     else
462 :     run++;
463 :     }
464 :    
465 :     level_shifted = prev_level + 32;
466 :     if (!(level_shifted & -64))
467 :     {
468 :     code = coeff_VLC[0][1][level_shifted][prev_run].code;
469 :     len = coeff_VLC[0][1][level_shifted][prev_run].len;
470 :     }
471 :     else
472 :     {
473 :     code = (ESCAPE3 << 21) | (1 << 20) | (prev_run << 14) | (1 << 13) | ((prev_level & 0xfff) << 1) | 1;
474 :     len = 30;
475 :     }
476 :     BitstreamPutBits(bs, code, len);
477 :     }
478 :    
479 :     static __inline void
480 :     CodeCoeffIntra(Bitstream * bs,
481 :     const int16_t qcoeff[64],
482 :     const uint16_t * zigzag)
483 :     {
484 :     uint32_t i, abs_level, run, prev_run, code, len;
485 :     int32_t level, prev_level;
486 :    
487 :     i = 1;
488 :     run = 0;
489 :    
490 : suxen_drol 1.25.2.14 while (i<64 && !(level = qcoeff[zigzag[i++]]))
491 : edgomez 1.25.2.11 run++;
492 :    
493 :     prev_level = level;
494 :     prev_run = run;
495 :     run = 0;
496 :    
497 :     while (i < 64)
498 :     {
499 :     if ((level = qcoeff[zigzag[i++]]) != 0)
500 :     {
501 :     abs_level = ABS(prev_level);
502 :     abs_level = abs_level < 64 ? abs_level : 0;
503 :     code = coeff_VLC[1][0][abs_level][prev_run].code;
504 :     len = coeff_VLC[1][0][abs_level][prev_run].len;
505 :     if (len != 128)
506 :     code |= (prev_level < 0);
507 :     else
508 :     {
509 :     code = (ESCAPE3 << 21) | (prev_run << 14) | (1 << 13) | ((prev_level & 0xfff) << 1) | 1;
510 :     len = 30;
511 :     }
512 :     BitstreamPutBits(bs, code, len);
513 :     prev_level = level;
514 :     prev_run = run;
515 :     run = 0;
516 :     }
517 :     else
518 :     run++;
519 :     }
520 :    
521 :     abs_level = ABS(prev_level);
522 :     abs_level = abs_level < 64 ? abs_level : 0;
523 :     code = coeff_VLC[1][1][abs_level][prev_run].code;
524 :     len = coeff_VLC[1][1][abs_level][prev_run].len;
525 :     if (len != 128)
526 :     code |= (prev_level < 0);
527 :     else
528 :     {
529 :     code = (ESCAPE3 << 21) | (1 << 20) | (prev_run << 14) | (1 << 13) | ((prev_level & 0xfff) << 1) | 1;
530 :     len = 30;
531 :     }
532 :     BitstreamPutBits(bs, code, len);
533 :     }
534 : suxen_drol 1.25.2.14
535 :    
536 :    
537 :     /* returns the number of bits required to encode qcoeff */
538 :    
539 :     int
540 :     CodeCoeffIntra_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag)
541 :     {
542 :     int bits = 0;
543 :     uint32_t i, abs_level, run, prev_run, len;
544 :     int32_t level, prev_level;
545 :    
546 :     i = 1;
547 :     run = 0;
548 :    
549 :     while (i<64 && !(level = qcoeff[zigzag[i++]]))
550 :     run++;
551 :    
552 :     if (i >= 64) return 0; /* empty block */
553 :    
554 :     prev_level = level;
555 :     prev_run = run;
556 :     run = 0;
557 :    
558 :     while (i < 64)
559 :     {
560 :     if ((level = qcoeff[zigzag[i++]]) != 0)
561 :     {
562 :     abs_level = ABS(prev_level);
563 :     abs_level = abs_level < 64 ? abs_level : 0;
564 :     len = coeff_VLC[1][0][abs_level][prev_run].len;
565 :     bits += len!=128 ? len : 30;
566 :    
567 :     prev_level = level;
568 :     prev_run = run;
569 :     run = 0;
570 :     }
571 :     else
572 :     run++;
573 :     }
574 :    
575 :     abs_level = ABS(prev_level);
576 :     abs_level = abs_level < 64 ? abs_level : 0;
577 :     len = coeff_VLC[1][1][abs_level][prev_run].len;
578 :     bits += len!=128 ? len : 30;
579 :    
580 :     return bits;
581 :     }
582 :    
583 : edgomez 1.25.2.11
584 :     #endif
585 : Isibaar 1.1
586 : chl 1.25.2.1 static __inline void
587 : chl 1.25.2.3 CodeBlockIntra(const FRAMEINFO * const frame,
588 : edgomez 1.17 const MACROBLOCK * pMB,
589 :     int16_t qcoeff[6 * 64],
590 : edgomez 1.7 Bitstream * bs,
591 :     Statistics * pStat)
592 : Isibaar 1.1 {
593 : edgomez 1.7
594 : Isibaar 1.1 uint32_t i, mcbpc, cbpy, bits;
595 :    
596 :     cbpy = pMB->cbp >> 2;
597 :    
598 : edgomez 1.7 // write mcbpc
599 : edgomez 1.17 if (frame->coding_type == I_VOP) {
600 : edgomez 1.7 mcbpc = ((pMB->mode >> 1) & 3) | ((pMB->cbp & 3) << 2);
601 : edgomez 1.17 BitstreamPutBits(bs, mcbpc_intra_tab[mcbpc].code,
602 :     mcbpc_intra_tab[mcbpc].len);
603 :     } else {
604 : edgomez 1.7 mcbpc = (pMB->mode & 7) | ((pMB->cbp & 3) << 3);
605 : edgomez 1.17 BitstreamPutBits(bs, mcbpc_inter_tab[mcbpc].code,
606 :     mcbpc_inter_tab[mcbpc].len);
607 : Isibaar 1.4 }
608 : Isibaar 1.1
609 :     // ac prediction flag
610 : edgomez 1.17 if (pMB->acpred_directions[0])
611 : edgomez 1.7 BitstreamPutBits(bs, 1, 1);
612 : Isibaar 1.1 else
613 : edgomez 1.7 BitstreamPutBits(bs, 0, 1);
614 : Isibaar 1.1
615 : edgomez 1.7 // write cbpy
616 : edgomez 1.17 BitstreamPutBits(bs, cbpy_tab[cbpy].code, cbpy_tab[cbpy].len);
617 : Isibaar 1.1
618 :     // write dquant
619 : edgomez 1.17 if (pMB->mode == MODE_INTRA_Q)
620 : Isibaar 1.1 BitstreamPutBits(bs, pMB->dquant, 2);
621 :    
622 : h 1.6 // write interlacing
623 : edgomez 1.17 if (frame->global_flags & XVID_INTERLACING) {
624 : h 1.6 BitstreamPutBit(bs, pMB->field_dct);
625 :     }
626 : Isibaar 1.1 // code block coeffs
627 : edgomez 1.17 for (i = 0; i < 6; i++) {
628 :     if (i < 4)
629 :     BitstreamPutBits(bs, dcy_tab[qcoeff[i * 64 + 0] + 255].code,
630 :     dcy_tab[qcoeff[i * 64 + 0] + 255].len);
631 : Isibaar 1.1 else
632 : edgomez 1.17 BitstreamPutBits(bs, dcc_tab[qcoeff[i * 64 + 0] + 255].code,
633 :     dcc_tab[qcoeff[i * 64 + 0] + 255].len);
634 :    
635 :     if (pMB->cbp & (1 << (5 - i))) {
636 : h 1.25.2.2 const uint16_t *scan_table =
637 :     frame->global_flags & XVID_ALTERNATESCAN ?
638 :     scan_tables[2] : scan_tables[pMB->acpred_directions[i]];
639 :    
640 : Isibaar 1.1 bits = BitstreamPos(bs);
641 :    
642 : edgomez 1.25.2.11 #ifdef BIGLUT
643 : h 1.25.2.2 CodeCoeff(bs, &qcoeff[i * 64], intra_table, scan_table, 1);
644 : edgomez 1.25.2.11 #else
645 :     CodeCoeffIntra(bs, &qcoeff[i * 64], scan_table);
646 :     #endif
647 : Isibaar 1.1
648 :     bits = BitstreamPos(bs) - bits;
649 :     pStat->iTextBits += bits;
650 :     }
651 :     }
652 : edgomez 1.7
653 : Isibaar 1.1 }
654 :    
655 :    
656 : edgomez 1.17 static void
657 : chl 1.25.2.3 CodeBlockInter(const FRAMEINFO * const frame,
658 : edgomez 1.17 const MACROBLOCK * pMB,
659 :     int16_t qcoeff[6 * 64],
660 : edgomez 1.7 Bitstream * bs,
661 :     Statistics * pStat)
662 : Isibaar 1.1 {
663 : edgomez 1.7
664 : Isibaar 1.1 int32_t i;
665 :     uint32_t bits, mcbpc, cbpy;
666 :    
667 : edgomez 1.7 mcbpc = (pMB->mode & 7) | ((pMB->cbp & 3) << 3);
668 : Isibaar 1.1 cbpy = 15 - (pMB->cbp >> 2);
669 :    
670 :     // write mcbpc
671 : edgomez 1.17 BitstreamPutBits(bs, mcbpc_inter_tab[mcbpc].code,
672 :     mcbpc_inter_tab[mcbpc].len);
673 : Isibaar 1.1
674 : chl 1.25.2.3 if ( (frame->coding_type == S_VOP) && (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q) )
675 : chl 1.25.2.10 BitstreamPutBit(bs, pMB->mcsel); // mcsel: '0'=local motion, '1'=GMC
676 : chl 1.25.2.3
677 : Isibaar 1.1 // write cbpy
678 :     BitstreamPutBits(bs, cbpy_tab[cbpy].code, cbpy_tab[cbpy].len);
679 :    
680 :     // write dquant
681 : edgomez 1.17 if (pMB->mode == MODE_INTER_Q)
682 : Isibaar 1.1 BitstreamPutBits(bs, pMB->dquant, 2);
683 : edgomez 1.17
684 : h 1.6 // interlacing
685 : edgomez 1.17 if (frame->global_flags & XVID_INTERLACING) {
686 : h 1.25 if (pMB->cbp) {
687 :     BitstreamPutBit(bs, pMB->field_dct);
688 : suxen_drol 1.25.2.6 DPRINTF(DPRINTF_MB,"codep: field_dct: %i", pMB->field_dct);
689 : h 1.25 }
690 : h 1.6
691 :     // if inter block, write field ME flag
692 : edgomez 1.17 if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q) {
693 : h 1.6 BitstreamPutBit(bs, pMB->field_pred);
694 : suxen_drol 1.25.2.6 DPRINTF(DPRINTF_MB,"codep: field_pred: %i", pMB->field_pred);
695 : h 1.6
696 :     // write field prediction references
697 : edgomez 1.17 if (pMB->field_pred) {
698 : h 1.6 BitstreamPutBit(bs, pMB->field_for_top);
699 :     BitstreamPutBit(bs, pMB->field_for_bot);
700 :     }
701 :     }
702 :     }
703 : chl 1.25.2.3 // code motion vector(s) if motion is local
704 : chl 1.25.2.10 if (!pMB->mcsel)
705 : chl 1.25.2.3 for (i = 0; i < (pMB->mode == MODE_INTER4V ? 4 : 1); i++) {
706 :     CodeVector(bs, pMB->pmvs[i].x, frame->fcode, pStat);
707 :     CodeVector(bs, pMB->pmvs[i].y, frame->fcode, pStat);
708 :     }
709 : Isibaar 1.1
710 :     bits = BitstreamPos(bs);
711 : edgomez 1.17
712 : Isibaar 1.1 // code block coeffs
713 : edgomez 1.17 for (i = 0; i < 6; i++)
714 :     if (pMB->cbp & (1 << (5 - i)))
715 : h 1.25.2.2 {
716 :     const uint16_t *scan_table =
717 :     frame->global_flags & XVID_ALTERNATESCAN ?
718 :     scan_tables[2] : scan_tables[0];
719 :    
720 : edgomez 1.25.2.11 #ifdef BIGLUT
721 : h 1.25.2.2 CodeCoeff(bs, &qcoeff[i * 64], inter_table, scan_table, 0);
722 : edgomez 1.25.2.11 #else
723 :     CodeCoeffInter(bs, &qcoeff[i * 64], scan_table);
724 :     #endif
725 : h 1.25.2.2 }
726 : Isibaar 1.1
727 :     bits = BitstreamPos(bs) - bits;
728 :     pStat->iTextBits += bits;
729 :     }
730 :    
731 :    
732 : edgomez 1.17 void
733 : chl 1.25.2.3 MBCoding(const FRAMEINFO * const frame,
734 : edgomez 1.17 MACROBLOCK * pMB,
735 :     int16_t qcoeff[6 * 64],
736 :     Bitstream * bs,
737 :     Statistics * pStat)
738 : Isibaar 1.1 {
739 : chl 1.25.2.3 if (frame->coding_type != I_VOP)
740 :     BitstreamPutBit(bs, 0); // not_coded
741 :    
742 : chl 1.22 if (pMB->mode == MODE_INTRA || pMB->mode == MODE_INTRA_Q)
743 : suxen_drol 1.13 CodeBlockIntra(frame, pMB, qcoeff, bs, pStat);
744 : Isibaar 1.1 else
745 : suxen_drol 1.13 CodeBlockInter(frame, pMB, qcoeff, bs, pStat);
746 : edgomez 1.7
747 : Isibaar 1.1 }
748 : chl 1.24
749 : chl 1.25.2.1 /*
750 :     // moved to mbcoding.h so that in can be 'static __inline'
751 : chl 1.24 void
752 :     MBSkip(Bitstream * bs)
753 :     {
754 :     BitstreamPutBit(bs, 1); // not coded
755 :     }
756 : chl 1.25.2.1 */
757 : suxen_drol 1.11
758 :     /***************************************************************
759 :     * bframe encoding start
760 :     ***************************************************************/
761 :    
762 :     /*
763 :     mbtype
764 :     0 1b direct(h263) mvdb
765 :     1 01b interpolate mc+q dbquant, mvdf, mvdb
766 :     2 001b backward mc+q dbquant, mvdb
767 :     3 0001b forward mc+q dbquant, mvdf
768 :     */
769 :    
770 : chl 1.25.2.1 static __inline void
771 : edgomez 1.17 put_bvop_mbtype(Bitstream * bs,
772 :     int value)
773 :     {
774 :     switch (value) {
775 : chl 1.25.2.1 case MODE_FORWARD:
776 :     BitstreamPutBit(bs, 0);
777 :     case MODE_BACKWARD:
778 :     BitstreamPutBit(bs, 0);
779 :     case MODE_INTERPOLATE:
780 :     BitstreamPutBit(bs, 0);
781 :     case MODE_DIRECT:
782 :     BitstreamPutBit(bs, 1);
783 :     default:
784 :     break;
785 : suxen_drol 1.11 }
786 :     }
787 :    
788 :     /*
789 :     dbquant
790 :     -2 10b
791 :     0 0b
792 :     +2 11b
793 :     */
794 :    
795 : chl 1.25.2.1 static __inline void
796 : edgomez 1.17 put_bvop_dbquant(Bitstream * bs,
797 :     int value)
798 :     {
799 :     switch (value) {
800 :     case 0:
801 :     BitstreamPutBit(bs, 0);
802 :     return;
803 :    
804 :     case -2:
805 :     BitstreamPutBit(bs, 1);
806 :     BitstreamPutBit(bs, 0);
807 :     return;
808 : suxen_drol 1.11
809 : edgomez 1.17 case 2:
810 :     BitstreamPutBit(bs, 1);
811 :     BitstreamPutBit(bs, 1);
812 :     return;
813 : suxen_drol 1.11
814 : edgomez 1.17 default:; // invalid
815 : suxen_drol 1.11 }
816 :     }
817 :    
818 :    
819 :    
820 : edgomez 1.17 void
821 :     MBCodingBVOP(const MACROBLOCK * mb,
822 :     const int16_t qcoeff[6 * 64],
823 :     const int32_t fcode,
824 :     const int32_t bcode,
825 :     Bitstream * bs,
826 : h 1.25.2.2 Statistics * pStat,
827 :     int direction)
828 : suxen_drol 1.11 {
829 : chl 1.25.2.1 int vcode = fcode;
830 :     unsigned int i;
831 : suxen_drol 1.11
832 :     /* ------------------------------------------------------------------
833 : chl 1.21 when a block is skipped it is decoded DIRECT(0,0)
834 :     hence is interpolated from forward & backward frames
835 : suxen_drol 1.11 ------------------------------------------------------------------ */
836 :    
837 : chl 1.21 if (mb->mode == MODE_DIRECT_NONE_MV) {
838 : edgomez 1.17 BitstreamPutBit(bs, 1); // skipped
839 : suxen_drol 1.11 return;
840 :     }
841 :    
842 :     BitstreamPutBit(bs, 0); // not skipped
843 :    
844 : edgomez 1.17 if (mb->cbp == 0) {
845 :     BitstreamPutBit(bs, 1); // cbp == 0
846 :     } else {
847 :     BitstreamPutBit(bs, 0); // cbp == xxx
848 : suxen_drol 1.11 }
849 :    
850 :     put_bvop_mbtype(bs, mb->mode);
851 :    
852 : edgomez 1.17 if (mb->cbp) {
853 : suxen_drol 1.11 BitstreamPutBits(bs, mb->cbp, 6);
854 :     }
855 :    
856 : edgomez 1.17 if (mb->mode != MODE_DIRECT && mb->cbp != 0) {
857 :     put_bvop_dbquant(bs, 0); // todo: mb->dquant = 0
858 : suxen_drol 1.11 }
859 :    
860 : chl 1.25.2.1 switch (mb->mode) {
861 :     case MODE_INTERPOLATE:
862 :     CodeVector(bs, mb->pmvs[1].x, vcode, pStat); //forward vector of interpolate mode
863 :     CodeVector(bs, mb->pmvs[1].y, vcode, pStat);
864 :     case MODE_BACKWARD:
865 :     vcode = bcode;
866 :     case MODE_FORWARD:
867 :     CodeVector(bs, mb->pmvs[0].x, vcode, pStat);
868 :     CodeVector(bs, mb->pmvs[0].y, vcode, pStat);
869 :     break;
870 :     case MODE_DIRECT:
871 :     CodeVector(bs, mb->pmvs[3].x, 1, pStat); // fcode is always 1 for delta vector
872 :     CodeVector(bs, mb->pmvs[3].y, 1, pStat); // prediction is always (0,0)
873 :     default: break;
874 : suxen_drol 1.11 }
875 : edgomez 1.17
876 :     for (i = 0; i < 6; i++) {
877 :     if (mb->cbp & (1 << (5 - i))) {
878 : edgomez 1.25.2.11 #ifdef BIGLUT
879 :     CodeCoeff(bs, &qcoeff[i * 64], inter_table, scan_tables[0], 0);
880 :     #else
881 :     CodeCoeffInter(bs, &qcoeff[i * 64], scan_tables[0]);
882 :     #endif
883 : suxen_drol 1.11 }
884 : edgomez 1.17 }
885 : suxen_drol 1.11 }
886 :    
887 : Isibaar 1.1
888 :    
889 :     /***************************************************************
890 : edgomez 1.2 * decoding stuff starts here *
891 :     ***************************************************************/
892 : Isibaar 1.1
893 : suxen_drol 1.19
894 :     // for IVOP addbits == 0
895 :     // for PVOP addbits == fcode - 1
896 :     // for BVOP addbits == max(fcode,bcode) - 1
897 :     // returns true or false
898 :     int
899 :     check_resync_marker(Bitstream * bs, int addbits)
900 :     {
901 :     uint32_t nbits;
902 :     uint32_t code;
903 :     uint32_t nbitsresyncmarker = NUMBITS_VP_RESYNC_MARKER + addbits;
904 :    
905 :     nbits = BitstreamNumBitsToByteAlign(bs);
906 :     code = BitstreamShowBits(bs, nbits);
907 :    
908 :     if (code == (((uint32_t)1 << (nbits - 1)) - 1))
909 :     {
910 :     return BitstreamShowBitsFromByteAlign(bs, nbitsresyncmarker) == RESYNC_MARKER;
911 :     }
912 :    
913 :     return 0;
914 :     }
915 :    
916 :    
917 :    
918 : edgomez 1.17 int
919 :     get_mcbpc_intra(Bitstream * bs)
920 : Isibaar 1.1 {
921 : edgomez 1.7
922 : Isibaar 1.1 uint32_t index;
923 : edgomez 1.17
924 : suxen_drol 1.19 index = BitstreamShowBits(bs, 9);
925 : Isibaar 1.1 index >>= 3;
926 :    
927 :     BitstreamSkip(bs, mcbpc_intra_table[index].len);
928 : edgomez 1.7
929 : Isibaar 1.1 return mcbpc_intra_table[index].code;
930 : edgomez 1.7
931 : Isibaar 1.1 }
932 :    
933 : edgomez 1.17 int
934 :     get_mcbpc_inter(Bitstream * bs)
935 : Isibaar 1.1 {
936 : edgomez 1.7
937 : Isibaar 1.1 uint32_t index;
938 : suxen_drol 1.19
939 : suxen_drol 1.25.2.9 index = MIN(BitstreamShowBits(bs, 9), 256);
940 : Isibaar 1.1
941 : edgomez 1.17 BitstreamSkip(bs, mcbpc_inter_table[index].len);
942 : edgomez 1.7
943 : Isibaar 1.1 return mcbpc_inter_table[index].code;
944 : edgomez 1.7
945 : Isibaar 1.1 }
946 :    
947 : edgomez 1.17 int
948 :     get_cbpy(Bitstream * bs,
949 :     int intra)
950 : Isibaar 1.1 {
951 : edgomez 1.7
952 : Isibaar 1.1 int cbpy;
953 :     uint32_t index = BitstreamShowBits(bs, 6);
954 :    
955 :     BitstreamSkip(bs, cbpy_table[index].len);
956 :     cbpy = cbpy_table[index].code;
957 :    
958 : edgomez 1.17 if (!intra)
959 : Isibaar 1.1 cbpy = 15 - cbpy;
960 :    
961 :     return cbpy;
962 : edgomez 1.7
963 : Isibaar 1.1 }
964 :    
965 : chl 1.25.2.1 static __inline int
966 : edgomez 1.17 get_mv_data(Bitstream * bs)
967 : Isibaar 1.1 {
968 : edgomez 1.7
969 : Isibaar 1.1 uint32_t index;
970 :    
971 : edgomez 1.17 if (BitstreamGetBit(bs))
972 : Isibaar 1.1 return 0;
973 : edgomez 1.17
974 : Isibaar 1.1 index = BitstreamShowBits(bs, 12);
975 :    
976 : edgomez 1.17 if (index >= 512) {
977 : Isibaar 1.1 index = (index >> 8) - 2;
978 :     BitstreamSkip(bs, TMNMVtab0[index].len);
979 :     return TMNMVtab0[index].code;
980 :     }
981 : edgomez 1.17
982 :     if (index >= 128) {
983 : Isibaar 1.1 index = (index >> 2) - 32;
984 :     BitstreamSkip(bs, TMNMVtab1[index].len);
985 :     return TMNMVtab1[index].code;
986 :     }
987 :    
988 : edgomez 1.17 index -= 4;
989 : Isibaar 1.1
990 :     BitstreamSkip(bs, TMNMVtab2[index].len);
991 :     return TMNMVtab2[index].code;
992 : edgomez 1.7
993 : Isibaar 1.1 }
994 :    
995 : edgomez 1.17 int
996 :     get_mv(Bitstream * bs,
997 :     int fcode)
998 : Isibaar 1.1 {
999 : edgomez 1.7
1000 : Isibaar 1.1 int data;
1001 :     int res;
1002 :     int mv;
1003 :     int scale_fac = 1 << (fcode - 1);
1004 :    
1005 :     data = get_mv_data(bs);
1006 : edgomez 1.17
1007 :     if (scale_fac == 1 || data == 0)
1008 : Isibaar 1.1 return data;
1009 :    
1010 :     res = BitstreamGetBits(bs, fcode - 1);
1011 :     mv = ((ABS(data) - 1) * scale_fac) + res + 1;
1012 : edgomez 1.17
1013 : Isibaar 1.1 return data < 0 ? -mv : mv;
1014 : edgomez 1.7
1015 : Isibaar 1.1 }
1016 :    
1017 : edgomez 1.17 int
1018 :     get_dc_dif(Bitstream * bs,
1019 :     uint32_t dc_size)
1020 : Isibaar 1.1 {
1021 : edgomez 1.7
1022 : Isibaar 1.1 int code = BitstreamGetBits(bs, dc_size);
1023 :     int msb = code >> (dc_size - 1);
1024 :    
1025 : edgomez 1.17 if (msb == 0)
1026 :     return (-1 * (code ^ ((1 << dc_size) - 1)));
1027 : Isibaar 1.1
1028 :     return code;
1029 : edgomez 1.7
1030 : Isibaar 1.1 }
1031 :    
1032 : edgomez 1.17 int
1033 :     get_dc_size_lum(Bitstream * bs)
1034 : Isibaar 1.1 {
1035 : edgomez 1.7
1036 : Isibaar 1.1 int code, i;
1037 : edgomez 1.17
1038 : Isibaar 1.1 code = BitstreamShowBits(bs, 11);
1039 :    
1040 : edgomez 1.17 for (i = 11; i > 3; i--) {
1041 :     if (code == 1) {
1042 : Isibaar 1.1 BitstreamSkip(bs, i);
1043 :     return i + 1;
1044 :     }
1045 :     code >>= 1;
1046 :     }
1047 :    
1048 :     BitstreamSkip(bs, dc_lum_tab[code].len);
1049 :     return dc_lum_tab[code].code;
1050 : edgomez 1.7
1051 : Isibaar 1.1 }
1052 :    
1053 :    
1054 : edgomez 1.17 int
1055 :     get_dc_size_chrom(Bitstream * bs)
1056 : Isibaar 1.1 {
1057 : edgomez 1.7
1058 : Isibaar 1.1 uint32_t code, i;
1059 : edgomez 1.17
1060 : Isibaar 1.1 code = BitstreamShowBits(bs, 12);
1061 :    
1062 : edgomez 1.17 for (i = 12; i > 2; i--) {
1063 :     if (code == 1) {
1064 : Isibaar 1.1 BitstreamSkip(bs, i);
1065 :     return i;
1066 :     }
1067 :     code >>= 1;
1068 :     }
1069 :    
1070 :     return 3 - BitstreamGetBits(bs, 2);
1071 : edgomez 1.7
1072 : edgomez 1.25.2.11 }
1073 :    
1074 :     static __inline int
1075 :     get_coeff(Bitstream * bs,
1076 :     int *run,
1077 :     int *last,
1078 :     int intra,
1079 :     int short_video_header)
1080 :     {
1081 :    
1082 :     uint32_t mode;
1083 :     int32_t level;
1084 :     REVERSE_EVENT *reverse_event;
1085 :    
1086 :     if (short_video_header) /* inter-VLCs will be used for both intra and inter blocks */
1087 :     intra = 0;
1088 :    
1089 :     if (BitstreamShowBits(bs, 7) != ESCAPE) {
1090 :     reverse_event = &DCT3D[intra][BitstreamShowBits(bs, 12)];
1091 :    
1092 :     if ((level = reverse_event->event.level) == 0)
1093 :     goto error;
1094 :    
1095 :     *last = reverse_event->event.last;
1096 :     *run = reverse_event->event.run;
1097 :    
1098 :     BitstreamSkip(bs, reverse_event->len);
1099 :    
1100 :     return BitstreamGetBits(bs, 1) ? -level : level;
1101 :     }
1102 :    
1103 :     BitstreamSkip(bs, 7);
1104 :    
1105 :     if (short_video_header) {
1106 :     /* escape mode 4 - H.263 type, only used if short_video_header = 1 */
1107 :     *last = BitstreamGetBit(bs);
1108 :     *run = BitstreamGetBits(bs, 6);
1109 :     level = BitstreamGetBits(bs, 8);
1110 :    
1111 :     if (level == 0 || level == 128)
1112 :     DPRINTF(DPRINTF_ERROR, "Illegal LEVEL for ESCAPE mode 4: %d", level);
1113 :    
1114 :     return (level << 24) >> 24;
1115 :     }
1116 :    
1117 :     mode = BitstreamShowBits(bs, 2);
1118 :    
1119 :     if (mode < 3) {
1120 :     BitstreamSkip(bs, (mode == 2) ? 2 : 1);
1121 :    
1122 :     reverse_event = &DCT3D[intra][BitstreamShowBits(bs, 12)];
1123 :    
1124 :     if ((level = reverse_event->event.level) == 0)
1125 :     goto error;
1126 :    
1127 :     *last = reverse_event->event.last;
1128 :     *run = reverse_event->event.run;
1129 :    
1130 :     BitstreamSkip(bs, reverse_event->len);
1131 :    
1132 :     if (mode < 2) /* first escape mode, level is offset */
1133 :     level += max_level[intra][*last][*run];
1134 :     else /* second escape mode, run is offset */
1135 :     *run += max_run[intra][*last][level] + 1;
1136 :    
1137 :     return BitstreamGetBits(bs, 1) ? -level : level;
1138 :     }
1139 :    
1140 :     /* third escape mode - fixed length codes */
1141 :     BitstreamSkip(bs, 2);
1142 :     *last = BitstreamGetBits(bs, 1);
1143 :     *run = BitstreamGetBits(bs, 6);
1144 :     BitstreamSkip(bs, 1); /* marker */
1145 :     level = BitstreamGetBits(bs, 12);
1146 :     BitstreamSkip(bs, 1); /* marker */
1147 :    
1148 :     return (level << 20) >> 20;
1149 :    
1150 :     error:
1151 :     *run = VLC_ERROR;
1152 :     return 0;
1153 : Isibaar 1.1 }
1154 :    
1155 : edgomez 1.17 void
1156 :     get_intra_block(Bitstream * bs,
1157 :     int16_t * block,
1158 :     int direction,
1159 :     int coeff)
1160 : Isibaar 1.1 {
1161 : edgomez 1.7
1162 : edgomez 1.17 const uint16_t *scan = scan_tables[direction];
1163 : chl 1.25.2.1 int level, run, last;
1164 : Isibaar 1.1
1165 : edgomez 1.17 do {
1166 : Isibaar 1.1 level = get_coeff(bs, &run, &last, 1, 0);
1167 : edgomez 1.17 if (run == -1) {
1168 : suxen_drol 1.25.2.6 DPRINTF(DPRINTF_ERROR,"fatal: invalid run");
1169 : Isibaar 1.1 break;
1170 :     }
1171 :     coeff += run;
1172 : edgomez 1.17 block[scan[coeff]] = level;
1173 : suxen_drol 1.20
1174 :     DPRINTF(DPRINTF_COEFF,"block[%i] %i", scan[coeff], level);
1175 :     //DPRINTF(DPRINTF_COEFF,"block[%i] %i %08x", scan[coeff], level, BitstreamShowBits(bs, 32));
1176 :    
1177 : suxen_drol 1.25.2.8 if (level < -2047 || level > 2047) {
1178 : suxen_drol 1.25.2.6 DPRINTF(DPRINTF_ERROR,"warning: intra_overflow %i", level);
1179 : Isibaar 1.1 }
1180 :     coeff++;
1181 :     } while (!last);
1182 : edgomez 1.7
1183 : Isibaar 1.1 }
1184 :    
1185 : edgomez 1.17 void
1186 :     get_inter_block(Bitstream * bs,
1187 : h 1.25.2.2 int16_t * block,
1188 :     int direction)
1189 : Isibaar 1.1 {
1190 : edgomez 1.7
1191 : h 1.25.2.2 const uint16_t *scan = scan_tables[direction];
1192 : Isibaar 1.1 int p;
1193 :     int level;
1194 :     int run;
1195 :     int last;
1196 :    
1197 :     p = 0;
1198 : edgomez 1.17 do {
1199 : Isibaar 1.1 level = get_coeff(bs, &run, &last, 0, 0);
1200 : edgomez 1.17 if (run == -1) {
1201 : suxen_drol 1.25.2.6 DPRINTF(DPRINTF_ERROR,"fatal: invalid run");
1202 : Isibaar 1.1 break;
1203 :     }
1204 :     p += run;
1205 : chenm001 1.16
1206 : edgomez 1.17 block[scan[p]] = level;
1207 : suxen_drol 1.20
1208 :     DPRINTF(DPRINTF_COEFF,"block[%i] %i", scan[p], level);
1209 :     // DPRINTF(DPRINTF_COEFF,"block[%i] %i %08x", scan[p], level, BitstreamShowBits(bs, 32));
1210 :    
1211 : suxen_drol 1.25.2.8 if (level < -2047 || level > 2047) {
1212 : suxen_drol 1.25.2.6 DPRINTF(DPRINTF_ERROR,"warning: inter overflow %i", level);
1213 : Isibaar 1.1 }
1214 :     p++;
1215 :     } while (!last);
1216 : edgomez 1.7
1217 : Isibaar 1.1 }
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4