?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
typo fixes
[mplayer/greg.git] / libmpeg2 / slice.c
blobdf3038885c8e64ec5da42dd3c47324fe734cf068
1 /*
2 * slice.c
3 * Copyright (C) 2000-2003 Michel Lespinasse <walken@zoy.org>
4 * Copyright (C) 2003 Peter Gubanov <peter@elecard.net.ru>
5 * Copyright (C) 1999-2000 Aaron Holtzman <aholtzma@ess.engr.uvic.ca>
6 *
7 * This file is part of mpeg2dec, a free MPEG-2 video stream decoder.
8 * See http://libmpeg2.sourceforge.net/ for updates.
9 *
10 * mpeg2dec is free software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
14 *
15 * mpeg2dec 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 * Modified for use with MPlayer, see libmpeg-0.4.0.diff for the exact changes.
25 * detailed CVS changelog at http://www.mplayerhq.hu/cgi-bin/cvsweb.cgi/main/
26 * $Id$
27 */
28
29 #include "config.h"
30
31 #include <inttypes.h>
32
33 #include "mpeg2.h"
34 #include "attributes.h"
35 #include "mpeg2_internal.h"
36
37 extern mpeg2_mc_t mpeg2_mc;
38 extern void (* mpeg2_idct_copy) (int16_t * block, uint8_t * dest, int stride);
39 extern void (* mpeg2_idct_add) (int last, int16_t * block,
40 uint8_t * dest, int stride);
41 extern void (* mpeg2_cpu_state_save) (cpu_state_t * state);
42 extern void (* mpeg2_cpu_state_restore) (cpu_state_t * state);
43
44 #include "vlc.h"
45
46 static inline int get_macroblock_modes (mpeg2_decoder_t * const decoder)
47 {
48 #define bit_buf (decoder->bitstream_buf)
49 #define bits (decoder->bitstream_bits)
50 #define bit_ptr (decoder->bitstream_ptr)
51 int macroblock_modes;
52 const MBtab * tab;
53
54 switch (decoder->coding_type) {
55 case I_TYPE:
56
57 tab = MB_I + UBITS (bit_buf, 1);
58 DUMPBITS (bit_buf, bits, tab->len);
59 macroblock_modes = tab->modes;
60
61 if ((! (decoder->frame_pred_frame_dct)) &&
62 (decoder->picture_structure == FRAME_PICTURE)) {
63 macroblock_modes |= UBITS (bit_buf, 1) * DCT_TYPE_INTERLACED;
64 DUMPBITS (bit_buf, bits, 1);
65 }
66
67 return macroblock_modes;
68
69 case P_TYPE:
70
71 tab = MB_P + UBITS (bit_buf, 5);
72 DUMPBITS (bit_buf, bits, tab->len);
73 macroblock_modes = tab->modes;
74
75 if (decoder->picture_structure != FRAME_PICTURE) {
76 if (macroblock_modes & MACROBLOCK_MOTION_FORWARD) {
77 macroblock_modes |= UBITS (bit_buf, 2) << MOTION_TYPE_SHIFT;
78 DUMPBITS (bit_buf, bits, 2);
79 }
80 return macroblock_modes | MACROBLOCK_MOTION_FORWARD;
81 } else if (decoder->frame_pred_frame_dct) {
82 if (macroblock_modes & MACROBLOCK_MOTION_FORWARD)
83 macroblock_modes |= MC_FRAME << MOTION_TYPE_SHIFT;
84 return macroblock_modes | MACROBLOCK_MOTION_FORWARD;
85 } else {
86 if (macroblock_modes & MACROBLOCK_MOTION_FORWARD) {
87 macroblock_modes |= UBITS (bit_buf, 2) << MOTION_TYPE_SHIFT;
88 DUMPBITS (bit_buf, bits, 2);
89 }
90 if (macroblock_modes & (MACROBLOCK_INTRA | MACROBLOCK_PATTERN)) {
91 macroblock_modes |= UBITS (bit_buf, 1) * DCT_TYPE_INTERLACED;
92 DUMPBITS (bit_buf, bits, 1);
93 }
94 return macroblock_modes | MACROBLOCK_MOTION_FORWARD;
95 }
96
97 case B_TYPE:
98
99 tab = MB_B + UBITS (bit_buf, 6);
100 DUMPBITS (bit_buf, bits, tab->len);
101 macroblock_modes = tab->modes;
102
103 if (decoder->picture_structure != FRAME_PICTURE) {
104 if (! (macroblock_modes & MACROBLOCK_INTRA)) {
105 macroblock_modes |= UBITS (bit_buf, 2) << MOTION_TYPE_SHIFT;
106 DUMPBITS (bit_buf, bits, 2);
107 }
108 return macroblock_modes;
109 } else if (decoder->frame_pred_frame_dct) {
110 /* if (! (macroblock_modes & MACROBLOCK_INTRA)) */
111 macroblock_modes |= MC_FRAME << MOTION_TYPE_SHIFT;
112 return macroblock_modes;
113 } else {
114 if (macroblock_modes & MACROBLOCK_INTRA)
115 goto intra;
116 macroblock_modes |= UBITS (bit_buf, 2) << MOTION_TYPE_SHIFT;
117 DUMPBITS (bit_buf, bits, 2);
118 if (macroblock_modes & (MACROBLOCK_INTRA | MACROBLOCK_PATTERN)) {
119 intra:
120 macroblock_modes |= UBITS (bit_buf, 1) * DCT_TYPE_INTERLACED;
121 DUMPBITS (bit_buf, bits, 1);
122 }
123 return macroblock_modes;
124 }
125
126 case D_TYPE:
127
128 DUMPBITS (bit_buf, bits, 1);
129 return MACROBLOCK_INTRA;
130
131 default:
132 return 0;
133 }
134 #undef bit_buf
135 #undef bits
136 #undef bit_ptr
137 }
138
139 static inline void get_quantizer_scale (mpeg2_decoder_t * const decoder)
140 {
141 #define bit_buf (decoder->bitstream_buf)
142 #define bits (decoder->bitstream_bits)
143 #define bit_ptr (decoder->bitstream_ptr)
144
145 int quantizer_scale_code;
146
147 quantizer_scale_code = UBITS (bit_buf, 5);
148 DUMPBITS (bit_buf, bits, 5);
149 decoder->quantizer_scale = decoder->quantizer_scales[quantizer_scale_code];
150
151 decoder->quantizer_matrix[0] =
152 decoder->quantizer_prescale[0][quantizer_scale_code];
153 decoder->quantizer_matrix[1] =
154 decoder->quantizer_prescale[1][quantizer_scale_code];
155 decoder->quantizer_matrix[2] =
156 decoder->chroma_quantizer[0][quantizer_scale_code];
157 decoder->quantizer_matrix[3] =
158 decoder->chroma_quantizer[1][quantizer_scale_code];
159 #undef bit_buf
160 #undef bits
161 #undef bit_ptr
162 }
163
164 static inline int get_motion_delta (mpeg2_decoder_t * const decoder,
165 const int f_code)
166 {
167 #define bit_buf (decoder->bitstream_buf)
168 #define bits (decoder->bitstream_bits)
169 #define bit_ptr (decoder->bitstream_ptr)
170
171 int delta;
172 int sign;
173 const MVtab * tab;
174
175 if (bit_buf & 0x80000000) {
176 DUMPBITS (bit_buf, bits, 1);
177 return 0;
178 } else if (bit_buf >= 0x0c000000) {
179
180 tab = MV_4 + UBITS (bit_buf, 4);
181 delta = (tab->delta << f_code) + 1;
182 bits += tab->len + f_code + 1;
183 bit_buf <<= tab->len;
184
185 sign = SBITS (bit_buf, 1);
186 bit_buf <<= 1;
187
188 if (f_code)
189 delta += UBITS (bit_buf, f_code);
190 bit_buf <<= f_code;
191
192 return (delta ^ sign) - sign;
193
194 } else {
195
196 tab = MV_10 + UBITS (bit_buf, 10);
197 delta = (tab->delta << f_code) + 1;
198 bits += tab->len + 1;
199 bit_buf <<= tab->len;
200
201 sign = SBITS (bit_buf, 1);
202 bit_buf <<= 1;
203
204 if (f_code) {
205 NEEDBITS (bit_buf, bits, bit_ptr);
206 delta += UBITS (bit_buf, f_code);
207 DUMPBITS (bit_buf, bits, f_code);
208 }
209
210 return (delta ^ sign) - sign;
211
212 }
213 #undef bit_buf
214 #undef bits
215 #undef bit_ptr
216 }
217
218 static inline int bound_motion_vector (const int vector, const int f_code)
219 {
220 return ((int32_t)vector << (27 - f_code)) >> (27 - f_code);
221 }
222
223 static inline int get_dmv (mpeg2_decoder_t * const decoder)
224 {
225 #define bit_buf (decoder->bitstream_buf)
226 #define bits (decoder->bitstream_bits)
227 #define bit_ptr (decoder->bitstream_ptr)
228
229 const DMVtab * tab;
230
231 tab = DMV_2 + UBITS (bit_buf, 2);
232 DUMPBITS (bit_buf, bits, tab->len);
233 return tab->dmv;
234 #undef bit_buf
235 #undef bits
236 #undef bit_ptr
237 }
238
239 static inline int get_coded_block_pattern (mpeg2_decoder_t * const decoder)
240 {
241 #define bit_buf (decoder->bitstream_buf)
242 #define bits (decoder->bitstream_bits)
243 #define bit_ptr (decoder->bitstream_ptr)
244
245 const CBPtab * tab;
246
247 NEEDBITS (bit_buf, bits, bit_ptr);
248
249 if (bit_buf >= 0x20000000) {
250
251 tab = CBP_7 + (UBITS (bit_buf, 7) - 16);
252 DUMPBITS (bit_buf, bits, tab->len);
253 return tab->cbp;
254
255 } else {
256
257 tab = CBP_9 + UBITS (bit_buf, 9);
258 DUMPBITS (bit_buf, bits, tab->len);
259 return tab->cbp;
260 }
261
262 #undef bit_buf
263 #undef bits
264 #undef bit_ptr
265 }
266
267 static inline int get_luma_dc_dct_diff (mpeg2_decoder_t * const decoder)
268 {
269 #define bit_buf (decoder->bitstream_buf)
270 #define bits (decoder->bitstream_bits)
271 #define bit_ptr (decoder->bitstream_ptr)
272 const DCtab * tab;
273 int size;
274 int dc_diff;
275
276 if (bit_buf < 0xf8000000) {
277 tab = DC_lum_5 + UBITS (bit_buf, 5);
278 size = tab->size;
279 if (size) {
280 bits += tab->len + size;
281 bit_buf <<= tab->len;
282 dc_diff =
283 UBITS (bit_buf, size) - UBITS (SBITS (~bit_buf, 1), size);
284 bit_buf <<= size;
285 return dc_diff << decoder->intra_dc_precision;
286 } else {
287 DUMPBITS (bit_buf, bits, 3);
288 return 0;
289 }
290 } else {
291 tab = DC_long + (UBITS (bit_buf, 9) - 0x1e0);
292 size = tab->size;
293 DUMPBITS (bit_buf, bits, tab->len);
294 NEEDBITS (bit_buf, bits, bit_ptr);
295 dc_diff = UBITS (bit_buf, size) - UBITS (SBITS (~bit_buf, 1), size);
296 DUMPBITS (bit_buf, bits, size);
297 return dc_diff << decoder->intra_dc_precision;
298 }
299 #undef bit_buf
300 #undef bits
301 #undef bit_ptr
302 }
303
304 static inline int get_chroma_dc_dct_diff (mpeg2_decoder_t * const decoder)
305 {
306 #define bit_buf (decoder->bitstream_buf)
307 #define bits (decoder->bitstream_bits)
308 #define bit_ptr (decoder->bitstream_ptr)
309 const DCtab * tab;
310 int size;
311 int dc_diff;
312
313 if (bit_buf < 0xf8000000) {
314 tab = DC_chrom_5 + UBITS (bit_buf, 5);
315 size = tab->size;
316 if (size) {
317 bits += tab->len + size;
318 bit_buf <<= tab->len;
319 dc_diff =
320 UBITS (bit_buf, size) - UBITS (SBITS (~bit_buf, 1), size);
321 bit_buf <<= size;
322 return dc_diff << decoder->intra_dc_precision;
323 } else {
324 DUMPBITS (bit_buf, bits, 2);
325 return 0;
326 }
327 } else {
328 tab = DC_long + (UBITS (bit_buf, 10) - 0x3e0);
329 size = tab->size;
330 DUMPBITS (bit_buf, bits, tab->len + 1);
331 NEEDBITS (bit_buf, bits, bit_ptr);
332 dc_diff = UBITS (bit_buf, size) - UBITS (SBITS (~bit_buf, 1), size);
333 DUMPBITS (bit_buf, bits, size);
334 return dc_diff << decoder->intra_dc_precision;
335 }
336 #undef bit_buf
337 #undef bits
338 #undef bit_ptr
339 }
340
341 #define SATURATE(val) \
342 do { \
343 val <<= 4; \
344 if (unlikely (val != (int16_t) val)) \
345 val = (SBITS (val, 1) ^ 2047) << 4; \
346 } while (0)
347
348 static void get_intra_block_B14 (mpeg2_decoder_t * const decoder,
349 const uint16_t * const quant_matrix)
350 {
351 int i;
352 int j;
353 int val;
354 const uint8_t * const scan = decoder->scan;
355 int mismatch;
356 const DCTtab * tab;
357 uint32_t bit_buf;
358 int bits;
359 const uint8_t * bit_ptr;
360 int16_t * const dest = decoder->DCTblock;
361
362 i = 0;
363 mismatch = ~dest[0];
364
365 bit_buf = decoder->bitstream_buf;
366 bits = decoder->bitstream_bits;
367 bit_ptr = decoder->bitstream_ptr;
368
369 NEEDBITS (bit_buf, bits, bit_ptr);
370
371 while (1) {
372 if (bit_buf >= 0x28000000) {
373
374 tab = DCT_B14AC_5 + (UBITS (bit_buf, 5) - 5);
375
376 i += tab->run;
377 if (i >= 64)
378 break; /* end of block */
379
380 normal_code:
381 j = scan[i];
382 bit_buf <<= tab->len;
383 bits += tab->len + 1;
384 val = (tab->level * quant_matrix[j]) >> 4;
385
386 /* if (bitstream_get (1)) val = -val; */
387 val = (val ^ SBITS (bit_buf, 1)) - SBITS (bit_buf, 1);
388
389 SATURATE (val);
390 dest[j] = val;
391 mismatch ^= val;
392
393 bit_buf <<= 1;
394 NEEDBITS (bit_buf, bits, bit_ptr);
395
396 continue;
397
398 } else if (bit_buf >= 0x04000000) {
399
400 tab = DCT_B14_8 + (UBITS (bit_buf, 8) - 4);
401
402 i += tab->run;
403 if (i < 64)
404 goto normal_code;
405
406 /* escape code */
407
408 i += UBITS (bit_buf << 6, 6) - 64;
409 if (i >= 64)
410 break; /* illegal, check needed to avoid buffer overflow */
411
412 j = scan[i];
413
414 DUMPBITS (bit_buf, bits, 12);
415 NEEDBITS (bit_buf, bits, bit_ptr);
416 val = (SBITS (bit_buf, 12) * quant_matrix[j]) / 16;
417
418 SATURATE (val);
419 dest[j] = val;
420 mismatch ^= val;
421
422 DUMPBITS (bit_buf, bits, 12);
423 NEEDBITS (bit_buf, bits, bit_ptr);
424
425 continue;
426
427 } else if (bit_buf >= 0x02000000) {
428 tab = DCT_B14_10 + (UBITS (bit_buf, 10) - 8);
429 i += tab->run;
430 if (i < 64)
431 goto normal_code;
432 } else if (bit_buf >= 0x00800000) {
433 tab = DCT_13 + (UBITS (bit_buf, 13) - 16);
434 i += tab->run;
435 if (i < 64)
436 goto normal_code;
437 } else if (bit_buf >= 0x00200000) {
438 tab = DCT_15 + (UBITS (bit_buf, 15) - 16);
439 i += tab->run;
440 if (i < 64)
441 goto normal_code;
442 } else {
443 tab = DCT_16 + UBITS (bit_buf, 16);
444 bit_buf <<= 16;
445 GETWORD (bit_buf, bits + 16, bit_ptr);
446 i += tab->run;
447 if (i < 64)
448 goto normal_code;
449 }
450 break; /* illegal, check needed to avoid buffer overflow */
451 }
452 dest[63] ^= mismatch & 16;
453 DUMPBITS (bit_buf, bits, 2); /* dump end of block code */
454 decoder->bitstream_buf = bit_buf;
455 decoder->bitstream_bits = bits;
456 decoder->bitstream_ptr = bit_ptr;
457 }
458
459 static void get_intra_block_B15 (mpeg2_decoder_t * const decoder,
460 const uint16_t * const quant_matrix)
461 {
462 int i;
463 int j;
464 int val;
465 const uint8_t * const scan = decoder->scan;
466 int mismatch;
467 const DCTtab * tab;
468 uint32_t bit_buf;
469 int bits;
470 const uint8_t * bit_ptr;
471 int16_t * const dest = decoder->DCTblock;
472
473 i = 0;
474 mismatch = ~dest[0];
475
476 bit_buf = decoder->bitstream_buf;
477 bits = decoder->bitstream_bits;
478 bit_ptr = decoder->bitstream_ptr;
479
480 NEEDBITS (bit_buf, bits, bit_ptr);
481
482 while (1) {
483 if (bit_buf >= 0x04000000) {
484
485 tab = DCT_B15_8 + (UBITS (bit_buf, 8) - 4);
486
487 i += tab->run;
488 if (i < 64) {
489
490 normal_code:
491 j = scan[i];
492 bit_buf <<= tab->len;
493 bits += tab->len + 1;
494 val = (tab->level * quant_matrix[j]) >> 4;
495
496 /* if (bitstream_get (1)) val = -val; */
497 val = (val ^ SBITS (bit_buf, 1)) - SBITS (bit_buf, 1);
498
499 SATURATE (val);
500 dest[j] = val;
501 mismatch ^= val;
502
503 bit_buf <<= 1;
504 NEEDBITS (bit_buf, bits, bit_ptr);
505
506 continue;
507
508 } else {
509
510 /* end of block. I commented out this code because if we */
511 /* dont exit here we will still exit at the later test :) */
512
513 /* if (i >= 128) break; */ /* end of block */
514
515 /* escape code */
516
517 i += UBITS (bit_buf << 6, 6) - 64;
518 if (i >= 64)
519 break; /* illegal, check against buffer overflow */
520
521 j = scan[i];
522
523 DUMPBITS (bit_buf, bits, 12);
524 NEEDBITS (bit_buf, bits, bit_ptr);
525 val = (SBITS (bit_buf, 12) * quant_matrix[j]) / 16;
526
527 SATURATE (val);
528 dest[j] = val;
529 mismatch ^= val;
530
531 DUMPBITS (bit_buf, bits, 12);
532 NEEDBITS (bit_buf, bits, bit_ptr);
533
534 continue;
535
536 }
537 } else if (bit_buf >= 0x02000000) {
538 tab = DCT_B15_10 + (UBITS (bit_buf, 10) - 8);
539 i += tab->run;
540 if (i < 64)
541 goto normal_code;
542 } else if (bit_buf >= 0x00800000) {
543 tab = DCT_13 + (UBITS (bit_buf, 13) - 16);
544 i += tab->run;
545 if (i < 64)
546 goto normal_code;
547 } else if (bit_buf >= 0x00200000) {
548 tab = DCT_15 + (UBITS (bit_buf, 15) - 16);
549 i += tab->run;
550 if (i < 64)
551 goto normal_code;
552 } else {
553 tab = DCT_16 + UBITS (bit_buf, 16);
554 bit_buf <<= 16;
555 GETWORD (bit_buf, bits + 16, bit_ptr);
556 i += tab->run;
557 if (i < 64)
558 goto normal_code;
559 }
560 break; /* illegal, check needed to avoid buffer overflow */
561 }
562 dest[63] ^= mismatch & 16;
563 DUMPBITS (bit_buf, bits, 4); /* dump end of block code */
564 decoder->bitstream_buf = bit_buf;
565 decoder->bitstream_bits = bits;
566 decoder->bitstream_ptr = bit_ptr;
567 }
568
569 static int get_non_intra_block (mpeg2_decoder_t * const decoder,
570 const uint16_t * const quant_matrix)
571 {
572 int i;
573 int j;
574 int val;
575 const uint8_t * const scan = decoder->scan;
576 int mismatch;
577 const DCTtab * tab;
578 uint32_t bit_buf;
579 int bits;
580 const uint8_t * bit_ptr;
581 int16_t * const dest = decoder->DCTblock;
582
583 i = -1;
584 mismatch = -1;
585
586 bit_buf = decoder->bitstream_buf;
587 bits = decoder->bitstream_bits;
588 bit_ptr = decoder->bitstream_ptr;
589
590 NEEDBITS (bit_buf, bits, bit_ptr);
591 if (bit_buf >= 0x28000000) {
592 tab = DCT_B14DC_5 + (UBITS (bit_buf, 5) - 5);
593 goto entry_1;
594 } else
595 goto entry_2;
596
597 while (1) {
598 if (bit_buf >= 0x28000000) {
599
600 tab = DCT_B14AC_5 + (UBITS (bit_buf, 5) - 5);
601
602 entry_1:
603 i += tab->run;
604 if (i >= 64)
605 break; /* end of block */
606
607 normal_code:
608 j = scan[i];
609 bit_buf <<= tab->len;
610 bits += tab->len + 1;
611 val = ((2 * tab->level + 1) * quant_matrix[j]) >> 5;
612
613 /* if (bitstream_get (1)) val = -val; */
614 val = (val ^ SBITS (bit_buf, 1)) - SBITS (bit_buf, 1);
615
616 SATURATE (val);
617 dest[j] = val;
618 mismatch ^= val;
619
620 bit_buf <<= 1;
621 NEEDBITS (bit_buf, bits, bit_ptr);
622
623 continue;
624
625 }
626
627 entry_2:
628 if (bit_buf >= 0x04000000) {
629
630 tab = DCT_B14_8 + (UBITS (bit_buf, 8) - 4);
631
632 i += tab->run;
633 if (i < 64)
634 goto normal_code;
635
636 /* escape code */
637
638 i += UBITS (bit_buf << 6, 6) - 64;
639 if (i >= 64)
640 break; /* illegal, check needed to avoid buffer overflow */
641
642 j = scan[i];
643
644 DUMPBITS (bit_buf, bits, 12);
645 NEEDBITS (bit_buf, bits, bit_ptr);
646 val = 2 * (SBITS (bit_buf, 12) + SBITS (bit_buf, 1)) + 1;
647 val = (val * quant_matrix[j]) / 32;
648
649 SATURATE (val);
650 dest[j] = val;
651 mismatch ^= val;
652
653 DUMPBITS (bit_buf, bits, 12);
654 NEEDBITS (bit_buf, bits, bit_ptr);
655
656 continue;
657
658 } else if (bit_buf >= 0x02000000) {
659 tab = DCT_B14_10 + (UBITS (bit_buf, 10) - 8);
660 i += tab->run;
661 if (i < 64)
662 goto normal_code;
663 } else if (bit_buf >= 0x00800000) {
664 tab = DCT_13 + (UBITS (bit_buf, 13) - 16);
665 i += tab->run;
666 if (i < 64)
667 goto normal_code;
668 } else if (bit_buf >= 0x00200000) {
669 tab = DCT_15 + (UBITS (bit_buf, 15) - 16);
670 i += tab->run;
671 if (i < 64)
672 goto normal_code;
673 } else {
674 tab = DCT_16 + UBITS (bit_buf, 16);
675 bit_buf <<= 16;
676 GETWORD (bit_buf, bits + 16, bit_ptr);
677 i += tab->run;
678 if (i < 64)
679 goto normal_code;
680 }
681 break; /* illegal, check needed to avoid buffer overflow */
682 }
683 dest[63] ^= mismatch & 16;
684 DUMPBITS (bit_buf, bits, 2); /* dump end of block code */
685 decoder->bitstream_buf = bit_buf;
686 decoder->bitstream_bits = bits;
687 decoder->bitstream_ptr = bit_ptr;
688 return i;
689 }
690
691 static void get_mpeg1_intra_block (mpeg2_decoder_t * const decoder)
692 {
693 int i;
694 int j;
695 int val;
696 const uint8_t * const scan = decoder->scan;
697 const uint16_t * const quant_matrix = decoder->quantizer_matrix[0];
698 const DCTtab * tab;
699 uint32_t bit_buf;
700 int bits;
701 const uint8_t * bit_ptr;
702 int16_t * const dest = decoder->DCTblock;
703
704 i = 0;
705
706 bit_buf = decoder->bitstream_buf;
707 bits = decoder->bitstream_bits;
708 bit_ptr = decoder->bitstream_ptr;
709
710 NEEDBITS (bit_buf, bits, bit_ptr);
711
712 while (1) {
713 if (bit_buf >= 0x28000000) {
714
715 tab = DCT_B14AC_5 + (UBITS (bit_buf, 5) - 5);
716
717 i += tab->run;
718 if (i >= 64)
719 break; /* end of block */
720
721 normal_code:
722 j = scan[i];
723 bit_buf <<= tab->len;
724 bits += tab->len + 1;
725 val = (tab->level * quant_matrix[j]) >> 4;
726
727 /* oddification */
728 val = (val - 1) | 1;
729
730 /* if (bitstream_get (1)) val = -val; */
731 val = (val ^ SBITS (bit_buf, 1)) - SBITS (bit_buf, 1);
732
733 SATURATE (val);
734 dest[j] = val;
735
736 bit_buf <<= 1;
737 NEEDBITS (bit_buf, bits, bit_ptr);
738
739 continue;
740
741 } else if (bit_buf >= 0x04000000) {
742
743 tab = DCT_B14_8 + (UBITS (bit_buf, 8) - 4);
744
745 i += tab->run;
746 if (i < 64)
747 goto normal_code;
748
749 /* escape code */
750
751 i += UBITS (bit_buf << 6, 6) - 64;
752 if (i >= 64)
753 break; /* illegal, check needed to avoid buffer overflow */
754
755 j = scan[i];
756
757 DUMPBITS (bit_buf, bits, 12);
758 NEEDBITS (bit_buf, bits, bit_ptr);
759 val = SBITS (bit_buf, 8);
760 if (! (val & 0x7f)) {
761 DUMPBITS (bit_buf, bits, 8);
762 val = UBITS (bit_buf, 8) + 2 * val;
763 }
764 val = (val * quant_matrix[j]) / 16;
765
766 /* oddification */
767 val = (val + ~SBITS (val, 1)) | 1;
768
769 SATURATE (val);
770 dest[j] = val;
771
772 DUMPBITS (bit_buf, bits, 8);
773 NEEDBITS (bit_buf, bits, bit_ptr);
774
775 continue;
776
777 } else if (bit_buf >= 0x02000000) {
778 tab = DCT_B14_10 + (UBITS (bit_buf, 10) - 8);
779 i += tab->run;
780 if (i < 64)
781 goto normal_code;
782 } else if (bit_buf >= 0x00800000) {
783 tab = DCT_13 + (UBITS (bit_buf, 13) - 16);
784 i += tab->run;
785 if (i < 64)
786 goto normal_code;
787 } else if (bit_buf >= 0x00200000) {
788 tab = DCT_15 + (UBITS (bit_buf, 15) - 16);
789 i += tab->run;
790 if (i < 64)
791 goto normal_code;
792 } else {
793 tab = DCT_16 + UBITS (bit_buf, 16);
794 bit_buf <<= 16;
795 GETWORD (bit_buf, bits + 16, bit_ptr);
796 i += tab->run;
797 if (i < 64)
798 goto normal_code;
799 }
800 break; /* illegal, check needed to avoid buffer overflow */
801 }
802 DUMPBITS (bit_buf, bits, 2); /* dump end of block code */
803 decoder->bitstream_buf = bit_buf;
804 decoder->bitstream_bits = bits;
805 decoder->bitstream_ptr = bit_ptr;
806 }
807
808 static int get_mpeg1_non_intra_block (mpeg2_decoder_t * const decoder)
809 {
810 int i;
811 int j;
812 int val;
813 const uint8_t * const scan = decoder->scan;
814 const uint16_t * const quant_matrix = decoder->quantizer_matrix[1];
815 const DCTtab * tab;
816 uint32_t bit_buf;
817 int bits;
818 const uint8_t * bit_ptr;
819 int16_t * const dest = decoder->DCTblock;
820
821 i = -1;
822
823 bit_buf = decoder->bitstream_buf;
824 bits = decoder->bitstream_bits;
825 bit_ptr = decoder->bitstream_ptr;
826
827 NEEDBITS (bit_buf, bits, bit_ptr);
828 if (bit_buf >= 0x28000000) {
829 tab = DCT_B14DC_5 + (UBITS (bit_buf, 5) - 5);
830 goto entry_1;
831 } else
832 goto entry_2;
833
834 while (1) {
835 if (bit_buf >= 0x28000000) {
836
837 tab = DCT_B14AC_5 + (UBITS (bit_buf, 5) - 5);
838
839 entry_1:
840 i += tab->run;
841 if (i >= 64)
842 break; /* end of block */
843
844 normal_code:
845 j = scan[i];
846 bit_buf <<= tab->len;
847 bits += tab->len + 1;
848 val = ((2 * tab->level + 1) * quant_matrix[j]) >> 5;
849
850 /* oddification */
851 val = (val - 1) | 1;
852
853 /* if (bitstream_get (1)) val = -val; */
854 val = (val ^ SBITS (bit_buf, 1)) - SBITS (bit_buf, 1);
855
856 SATURATE (val);
857 dest[j] = val;
858
859 bit_buf <<= 1;
860 NEEDBITS (bit_buf, bits, bit_ptr);
861
862 continue;
863
864 }
865
866 entry_2:
867 if (bit_buf >= 0x04000000) {
868
869 tab = DCT_B14_8 + (UBITS (bit_buf, 8) - 4);
870
871 i += tab->run;
872 if (i < 64)
873 goto normal_code;
874
875 /* escape code */
876
877 i += UBITS (bit_buf << 6, 6) - 64;
878 if (i >= 64)
879 break; /* illegal, check needed to avoid buffer overflow */
880
881 j = scan[i];
882
883 DUMPBITS (bit_buf, bits, 12);
884 NEEDBITS (bit_buf, bits, bit_ptr);
885 val = SBITS (bit_buf, 8);
886 if (! (val & 0x7f)) {
887 DUMPBITS (bit_buf, bits, 8);
888 val = UBITS (bit_buf, 8) + 2 * val;
889 }
890 val = 2 * (val + SBITS (val, 1)) + 1;
891 val = (val * quant_matrix[j]) / 32;
892
893 /* oddification */
894 val = (val + ~SBITS (val, 1)) | 1;
895
896 SATURATE (val);
897 dest[j] = val;
898
899 DUMPBITS (bit_buf, bits, 8);
900 NEEDBITS (bit_buf, bits, bit_ptr);
901
902 continue;
903
904 } else if (bit_buf >= 0x02000000) {
905 tab = DCT_B14_10 + (UBITS (bit_buf, 10) - 8);
906 i += tab->run;
907 if (i < 64)
908 goto normal_code;
909 } else if (bit_buf >= 0x00800000) {
910 tab = DCT_13 + (UBITS (bit_buf, 13) - 16);
911 i += tab->run;
912 if (i < 64)
913 goto normal_code;
914 } else if (bit_buf >= 0x00200000) {
915 tab = DCT_15 + (UBITS (bit_buf, 15) - 16);
916 i += tab->run;
917 if (i < 64)
918 goto normal_code;
919 } else {
920 tab = DCT_16 + UBITS (bit_buf, 16);
921 bit_buf <<= 16;
922 GETWORD (bit_buf, bits + 16, bit_ptr);
923 i += tab->run;
924 if (i < 64)
925 goto normal_code;
926 }
927 break; /* illegal, check needed to avoid buffer overflow */
928 }
929 DUMPBITS (bit_buf, bits, 2); /* dump end of block code */
930 decoder->bitstream_buf = bit_buf;
931 decoder->bitstream_bits = bits;
932 decoder->bitstream_ptr = bit_ptr;
933 return i;
934 }
935
936 static inline void slice_intra_DCT (mpeg2_decoder_t * const decoder,
937 const int cc,
938 uint8_t * const dest, const int stride)
939 {
940 #define bit_buf (decoder->bitstream_buf)
941 #define bits (decoder->bitstream_bits)
942 #define bit_ptr (decoder->bitstream_ptr)
943 NEEDBITS (bit_buf, bits, bit_ptr);
944 /* Get the intra DC coefficient and inverse quantize it */
945 if (cc == 0)
946 decoder->DCTblock[0] =
947 decoder->dc_dct_pred[0] += get_luma_dc_dct_diff (decoder);
948 else
949 decoder->DCTblock[0] =
950 decoder->dc_dct_pred[cc] += get_chroma_dc_dct_diff (decoder);
951
952 if (decoder->mpeg1) {
953 if (decoder->coding_type != D_TYPE)
954 get_mpeg1_intra_block (decoder);
955 } else if (decoder->intra_vlc_format)
956 get_intra_block_B15 (decoder, decoder->quantizer_matrix[cc ? 2 : 0]);
957 else
958 get_intra_block_B14 (decoder, decoder->quantizer_matrix[cc ? 2 : 0]);
959 mpeg2_idct_copy (decoder->DCTblock, dest, stride);
960 #undef bit_buf
961 #undef bits
962 #undef bit_ptr
963 }
964
965 static inline void slice_non_intra_DCT (mpeg2_decoder_t * const decoder,
966 const int cc,
967 uint8_t * const dest, const int stride)
968 {
969 int last;
970
971 if (decoder->mpeg1)
972 last = get_mpeg1_non_intra_block (decoder);
973 else
974 last = get_non_intra_block (decoder,
975 decoder->quantizer_matrix[cc ? 3 : 1]);
976 mpeg2_idct_add (last, decoder->DCTblock, dest, stride);
977 }
978
979 #define MOTION_420(table,ref,motion_x,motion_y,size,y) \
980 pos_x = 2 * decoder->offset + motion_x; \
981 pos_y = 2 * decoder->v_offset + motion_y + 2 * y; \
982 if (unlikely (pos_x > decoder->limit_x)) { \
983 pos_x = ((int)pos_x < 0) ? 0 : decoder->limit_x; \
984 motion_x = pos_x - 2 * decoder->offset; \
985 } \
986 if (unlikely (pos_y > decoder->limit_y_ ## size)) { \
987 pos_y = ((int)pos_y < 0) ? 0 : decoder->limit_y_ ## size; \
988 motion_y = pos_y - 2 * decoder->v_offset - 2 * y; \
989 } \
990 xy_half = ((pos_y & 1) << 1) | (pos_x & 1); \
991 table[xy_half] (decoder->dest[0] + y * decoder->stride + decoder->offset, \
992 ref[0] + (pos_x >> 1) + (pos_y >> 1) * decoder->stride, \
993 decoder->stride, size); \
994 motion_x /= 2; motion_y /= 2; \
995 xy_half = ((motion_y & 1) << 1) | (motion_x & 1); \
996 offset = (((decoder->offset + motion_x) >> 1) + \
997 ((((decoder->v_offset + motion_y) >> 1) + y/2) * \
998 decoder->uv_stride)); \
999 table[4+xy_half] (decoder->dest[1] + y/2 * decoder->uv_stride + \
1000 (decoder->offset >> 1), ref[1] + offset, \
1001 decoder->uv_stride, size/2); \
1002 table[4+xy_half] (decoder->dest[2] + y/2 * decoder->uv_stride + \
1003 (decoder->offset >> 1), ref[2] + offset, \
1004 decoder->uv_stride, size/2)
1005
1006 #define MOTION_FIELD_420(table,ref,motion_x,motion_y,dest_field,op,src_field) \
1007 pos_x = 2 * decoder->offset + motion_x; \
1008 pos_y = decoder->v_offset + motion_y; \
1009 if (unlikely (pos_x > decoder->limit_x)) { \
1010 pos_x = ((int)pos_x < 0) ? 0 : decoder->limit_x; \
1011 motion_x = pos_x - 2 * decoder->offset; \
1012 } \
1013 if (unlikely (pos_y > decoder->limit_y)) { \
1014 pos_y = ((int)pos_y < 0) ? 0 : decoder->limit_y; \
1015 motion_y = pos_y - decoder->v_offset; \
1016 } \
1017 xy_half = ((pos_y & 1) << 1) | (pos_x & 1); \
1018 table[xy_half] (decoder->dest[0] + dest_field * decoder->stride + \
1019 decoder->offset, \
1020 (ref[0] + (pos_x >> 1) + \
1021 ((pos_y op) + src_field) * decoder->stride), \
1022 2 * decoder->stride, 8); \
1023 motion_x /= 2; motion_y /= 2; \
1024 xy_half = ((motion_y & 1) << 1) | (motion_x & 1); \
1025 offset = (((decoder->offset + motion_x) >> 1) + \
1026 (((decoder->v_offset >> 1) + (motion_y op) + src_field) * \
1027 decoder->uv_stride)); \
1028 table[4+xy_half] (decoder->dest[1] + dest_field * decoder->uv_stride + \
1029 (decoder->offset >> 1), ref[1] + offset, \
1030 2 * decoder->uv_stride, 4); \
1031 table[4+xy_half] (decoder->dest[2] + dest_field * decoder->uv_stride + \
1032 (decoder->offset >> 1), ref[2] + offset, \
1033 2 * decoder->uv_stride, 4)
1034
1035 #define MOTION_DMV_420(table,ref,motion_x,motion_y) \
1036 pos_x = 2 * decoder->offset + motion_x; \
1037 pos_y = decoder->v_offset + motion_y; \
1038 if (unlikely (pos_x > decoder->limit_x)) { \
1039 pos_x = ((int)pos_x < 0) ? 0 : decoder->limit_x; \
1040 motion_x = pos_x - 2 * decoder->offset; \
1041 } \
1042 if (unlikely (pos_y > decoder->limit_y)) { \
1043 pos_y = ((int)pos_y < 0) ? 0 : decoder->limit_y; \
1044 motion_y = pos_y - decoder->v_offset; \
1045 } \
1046 xy_half = ((pos_y & 1) << 1) | (pos_x & 1); \
1047 offset = (pos_x >> 1) + (pos_y & ~1) * decoder->stride; \
1048 table[xy_half] (decoder->dest[0] + decoder->offset, \
1049 ref[0] + offset, 2 * decoder->stride, 8); \
1050 table[xy_half] (decoder->dest[0] + decoder->stride + decoder->offset, \
1051 ref[0] + decoder->stride + offset, \
1052 2 * decoder->stride, 8); \
1053 motion_x /= 2; motion_y /= 2; \
1054 xy_half = ((motion_y & 1) << 1) | (motion_x & 1); \
1055 offset = (((decoder->offset + motion_x) >> 1) + \
1056 (((decoder->v_offset >> 1) + (motion_y & ~1)) * \
1057 decoder->uv_stride)); \
1058 table[4+xy_half] (decoder->dest[1] + (decoder->offset >> 1), \
1059 ref[1] + offset, 2 * decoder->uv_stride, 4); \
1060 table[4+xy_half] (decoder->dest[1] + decoder->uv_stride + \
1061 (decoder->offset >> 1), \
1062 ref[1] + decoder->uv_stride + offset, \
1063 2 * decoder->uv_stride, 4); \
1064 table[4+xy_half] (decoder->dest[2] + (decoder->offset >> 1), \
1065 ref[2] + offset, 2 * decoder->uv_stride, 4); \
1066 table[4+xy_half] (decoder->dest[2] + decoder->uv_stride + \
1067 (decoder->offset >> 1), \
1068 ref[2] + decoder->uv_stride + offset, \
1069 2 * decoder->uv_stride, 4)
1070
1071 #define MOTION_ZERO_420(table,ref) \
1072 table[0] (decoder->dest[0] + decoder->offset, \
1073 (ref[0] + decoder->offset + \
1074 decoder->v_offset * decoder->stride), decoder->stride, 16); \
1075 offset = ((decoder->offset >> 1) + \
1076 (decoder->v_offset >> 1) * decoder->uv_stride); \
1077 table[4] (decoder->dest[1] + (decoder->offset >> 1), \
1078 ref[1] + offset, decoder->uv_stride, 8); \
1079 table[4] (decoder->dest[2] + (decoder->offset >> 1), \
1080 ref[2] + offset, decoder->uv_stride, 8)
1081
1082 #define MOTION_422(table,ref,motion_x,motion_y,size,y) \
1083 pos_x = 2 * decoder->offset + motion_x; \
1084 pos_y = 2 * decoder->v_offset + motion_y + 2 * y; \
1085 if (unlikely (pos_x > decoder->limit_x)) { \
1086 pos_x = ((int)pos_x < 0) ? 0 : decoder->limit_x; \
1087 motion_x = pos_x - 2 * decoder->offset; \
1088 } \
1089 if (unlikely (pos_y > decoder->limit_y_ ## size)) { \
1090 pos_y = ((int)pos_y < 0) ? 0 : decoder->limit_y_ ## size; \
1091 motion_y = pos_y - 2 * decoder->v_offset - 2 * y; \
1092 } \
1093 xy_half = ((pos_y & 1) << 1) | (pos_x & 1); \
1094 offset = (pos_x >> 1) + (pos_y >> 1) * decoder->stride; \
1095 table[xy_half] (decoder->dest[0] + y * decoder->stride + decoder->offset, \
1096 ref[0] + offset, decoder->stride, size); \
1097 offset = (offset + (motion_x & (motion_x < 0))) >> 1; \
1098 motion_x /= 2; \
1099 xy_half = ((pos_y & 1) << 1) | (motion_x & 1); \
1100 table[4+xy_half] (decoder->dest[1] + y * decoder->uv_stride + \
1101 (decoder->offset >> 1), ref[1] + offset, \
1102 decoder->uv_stride, size); \
1103 table[4+xy_half] (decoder->dest[2] + y * decoder->uv_stride + \
1104 (decoder->offset >> 1), ref[2] + offset, \
1105 decoder->uv_stride, size)
1106
1107 #define MOTION_FIELD_422(table,ref,motion_x,motion_y,dest_field,op,src_field) \
1108 pos_x = 2 * decoder->offset + motion_x; \
1109 pos_y = decoder->v_offset + motion_y; \
1110 if (unlikely (pos_x > decoder->limit_x)) { \
1111 pos_x = ((int)pos_x < 0) ? 0 : decoder->limit_x; \
1112 motion_x = pos_x - 2 * decoder->offset; \
1113 } \
1114 if (unlikely (pos_y > decoder->limit_y)) { \
1115 pos_y = ((int)pos_y < 0) ? 0 : decoder->limit_y; \
1116 motion_y = pos_y - decoder->v_offset; \
1117 } \
1118 xy_half = ((pos_y & 1) << 1) | (pos_x & 1); \
1119 offset = (pos_x >> 1) + ((pos_y op) + src_field) * decoder->stride; \
1120 table[xy_half] (decoder->dest[0] + dest_field * decoder->stride + \
1121 decoder->offset, ref[0] + offset, \
1122 2 * decoder->stride, 8); \
1123 offset = (offset + (motion_x & (motion_x < 0))) >> 1; \
1124 motion_x /= 2; \
1125 xy_half = ((pos_y & 1) << 1) | (motion_x & 1); \
1126 table[4+xy_half] (decoder->dest[1] + dest_field * decoder->uv_stride + \
1127 (decoder->offset >> 1), ref[1] + offset, \
1128 2 * decoder->uv_stride, 8); \
1129 table[4+xy_half] (decoder->dest[2] + dest_field * decoder->uv_stride + \
1130 (decoder->offset >> 1), ref[2] + offset, \
1131 2 * decoder->uv_stride, 8)
1132
1133 #define MOTION_DMV_422(table,ref,motion_x,motion_y) \
1134 pos_x = 2 * decoder->offset + motion_x; \
1135 pos_y = decoder->v_offset + motion_y; \
1136 if (unlikely (pos_x > decoder->limit_x)) { \
1137 pos_x = ((int)pos_x < 0) ? 0 : decoder->limit_x; \
1138 motion_x = pos_x - 2 * decoder->offset; \
1139 } \
1140 if (unlikely (pos_y > decoder->limit_y)) { \
1141 pos_y = ((int)pos_y < 0) ? 0 : decoder->limit_y; \
1142 motion_y = pos_y - decoder->v_offset; \
1143 } \
1144 xy_half = ((pos_y & 1) << 1) | (pos_x & 1); \
1145 offset = (pos_x >> 1) + (pos_y & ~1) * decoder->stride; \
1146 table[xy_half] (decoder->dest[0] + decoder->offset, \
1147 ref[0] + offset, 2 * decoder->stride, 8); \
1148 table[xy_half] (decoder->dest[0] + decoder->stride + decoder->offset, \
1149 ref[0] + decoder->stride + offset, \
1150 2 * decoder->stride, 8); \
1151 offset = (offset + (motion_x & (motion_x < 0))) >> 1; \
1152 motion_x /= 2; \
1153 xy_half = ((pos_y & 1) << 1) | (motion_x & 1); \
1154 table[4+xy_half] (decoder->dest[1] + (decoder->offset >> 1), \
1155 ref[1] + offset, 2 * decoder->uv_stride, 8); \
1156 table[4+xy_half] (decoder->dest[1] + decoder->uv_stride + \
1157 (decoder->offset >> 1), \
1158 ref[1] + decoder->uv_stride + offset, \
1159 2 * decoder->uv_stride, 8); \
1160 table[4+xy_half] (decoder->dest[2] + (decoder->offset >> 1), \
1161 ref[2] + offset, 2 * decoder->uv_stride, 8); \
1162 table[4+xy_half] (decoder->dest[2] + decoder->uv_stride + \
1163 (decoder->offset >> 1), \
1164 ref[2] + decoder->uv_stride + offset, \
1165 2 * decoder->uv_stride, 8)
1166
1167 #define MOTION_ZERO_422(table,ref) \
1168 offset = decoder->offset + decoder->v_offset * decoder->stride; \
1169 table[0] (decoder->dest[0] + decoder->offset, \
1170 ref[0] + offset, decoder->stride, 16); \
1171 offset >>= 1; \
1172 table[4] (decoder->dest[1] + (decoder->offset >> 1), \
1173 ref[1] + offset, decoder->uv_stride, 16); \
1174 table[4] (decoder->dest[2] + (decoder->offset >> 1), \
1175 ref[2] + offset, decoder->uv_stride, 16)
1176
1177 #define MOTION_444(table,ref,motion_x,motion_y,size,y) \
1178 pos_x = 2 * decoder->offset + motion_x; \
1179 pos_y = 2 * decoder->v_offset + motion_y + 2 * y; \
1180 if (unlikely (pos_x > decoder->limit_x)) { \
1181 pos_x = ((int)pos_x < 0) ? 0 : decoder->limit_x; \
1182 motion_x = pos_x - 2 * decoder->offset; \
1183 } \
1184 if (unlikely (pos_y > decoder->limit_y_ ## size)) { \
1185 pos_y = ((int)pos_y < 0) ? 0 : decoder->limit_y_ ## size; \
1186 motion_y = pos_y - 2 * decoder->v_offset - 2 * y; \
1187 } \
1188 xy_half = ((pos_y & 1) << 1) | (pos_x & 1); \
1189 offset = (pos_x >> 1) + (pos_y >> 1) * decoder->stride; \
1190 table[xy_half] (decoder->dest[0] + y * decoder->stride + decoder->offset, \
1191 ref[0] + offset, decoder->stride, size); \
1192 table[xy_half] (decoder->dest[1] + y * decoder->stride + decoder->offset, \
1193 ref[1] + offset, decoder->stride, size); \
1194 table[xy_half] (decoder->dest[2] + y * decoder->stride + decoder->offset, \
1195 ref[2] + offset, decoder->stride, size)
1196
1197 #define MOTION_FIELD_444(table,ref,motion_x,motion_y,dest_field,op,src_field) \
1198 pos_x = 2 * decoder->offset + motion_x; \
1199 pos_y = decoder->v_offset + motion_y; \
1200 if (unlikely (pos_x > decoder->limit_x)) { \
1201 pos_x = ((int)pos_x < 0) ? 0 : decoder->limit_x; \
1202 motion_x = pos_x - 2 * decoder->offset; \
1203 } \
1204 if (unlikely (pos_y > decoder->limit_y)) { \
1205 pos_y = ((int)pos_y < 0) ? 0 : decoder->limit_y; \
1206 motion_y = pos_y - decoder->v_offset; \
1207 } \
1208 xy_half = ((pos_y & 1) << 1) | (pos_x & 1); \
1209 offset = (pos_x >> 1) + ((pos_y op) + src_field) * decoder->stride; \
1210 table[xy_half] (decoder->dest[0] + dest_field * decoder->stride + \
1211 decoder->offset, ref[0] + offset, \
1212 2 * decoder->stride, 8); \
1213 table[xy_half] (decoder->dest[1] + dest_field * decoder->stride + \
1214 decoder->offset, ref[1] + offset, \
1215 2 * decoder->stride, 8); \
1216 table[xy_half] (decoder->dest[2] + dest_field * decoder->stride + \
1217 decoder->offset, ref[2] + offset, \
1218 2 * decoder->stride, 8)
1219
1220 #define MOTION_DMV_444(table,ref,motion_x,motion_y) \
1221 pos_x = 2 * decoder->offset + motion_x; \
1222 pos_y = decoder->v_offset + motion_y; \
1223 if (unlikely (pos_x > decoder->limit_x)) { \
1224 pos_x = ((int)pos_x < 0) ? 0 : decoder->limit_x; \
1225 motion_x = pos_x - 2 * decoder->offset; \
1226 } \
1227 if (unlikely (pos_y > decoder->limit_y)) { \
1228 pos_y = ((int)pos_y < 0) ? 0 : decoder->limit_y; \
1229 motion_y = pos_y - decoder->v_offset; \
1230 } \
1231 xy_half = ((pos_y & 1) << 1) | (pos_x & 1); \
1232 offset = (pos_x >> 1) + (pos_y & ~1) * decoder->stride; \
1233 table[xy_half] (decoder->dest[0] + decoder->offset, \
1234 ref[0] + offset, 2 * decoder->stride, 8); \
1235 table[xy_half] (decoder->dest[0] + decoder->stride + decoder->offset, \
1236 ref[0] + decoder->stride + offset, \
1237 2 * decoder->stride, 8); \
1238 table[xy_half] (decoder->dest[1] + decoder->offset, \
1239 ref[1] + offset, 2 * decoder->stride, 8); \
1240 table[xy_half] (decoder->dest[1] + decoder->stride + decoder->offset, \
1241 ref[1] + decoder->stride + offset, \
1242 2 * decoder->stride, 8); \
1243 table[xy_half] (decoder->dest[2] + decoder->offset, \
1244 ref[2] + offset, 2 * decoder->stride, 8); \
1245 table[xy_half] (decoder->dest[2] + decoder->stride + decoder->offset, \
1246 ref[2] + decoder->stride + offset, \
1247 2 * decoder->stride, 8)
1248
1249 #define MOTION_ZERO_444(table,ref) \
1250 offset = decoder->offset + decoder->v_offset * decoder->stride; \
1251 table[0] (decoder->dest[0] + decoder->offset, \
1252 ref[0] + offset, decoder->stride, 16); \
1253 table[4] (decoder->dest[1] + decoder->offset, \
1254 ref[1] + offset, decoder->stride, 16); \
1255 table[4] (decoder->dest[2] + (decoder->offset >> 1), \
1256 ref[2] + offset, decoder->stride, 16)
1257
1258 #define bit_buf (decoder->bitstream_buf)
1259 #define bits (decoder->bitstream_bits)
1260 #define bit_ptr (decoder->bitstream_ptr)
1261
1262 static void motion_mp1 (mpeg2_decoder_t * const decoder,
1263 motion_t * const motion,
1264 mpeg2_mc_fct * const * const table)
1265 {
1266 int motion_x, motion_y;
1267 unsigned int pos_x, pos_y, xy_half, offset;
1268
1269 NEEDBITS (bit_buf, bits, bit_ptr);
1270 motion_x = (motion->pmv[0][0] +
1271 (get_motion_delta (decoder,
1272 motion->f_code[0]) << motion->f_code[1]));
1273 motion_x = bound_motion_vector (motion_x,
1274 motion->f_code[0] + motion->f_code[1]);
1275 motion->pmv[0][0] = motion_x;
1276
1277 NEEDBITS (bit_buf, bits, bit_ptr);
1278 motion_y = (motion->pmv[0][1] +
1279 (get_motion_delta (decoder,
1280 motion->f_code[0]) << motion->f_code[1]));
1281 motion_y = bound_motion_vector (motion_y,
1282 motion->f_code[0] + motion->f_code[1]);
1283 motion->pmv[0][1] = motion_y;
1284
1285 MOTION_420 (table, motion->ref[0], motion_x, motion_y, 16, 0);
1286 }
1287
1288 #define MOTION_FUNCTIONS(FORMAT,MOTION,MOTION_FIELD,MOTION_DMV,MOTION_ZERO) \
1289 \
1290 static void motion_fr_frame_##FORMAT (mpeg2_decoder_t * const decoder, \
1291 motion_t * const motion, \
1292 mpeg2_mc_fct * const * const table) \
1293 { \
1294 int motion_x, motion_y; \
1295 unsigned int pos_x, pos_y, xy_half, offset; \
1296 \
1297 NEEDBITS (bit_buf, bits, bit_ptr); \
1298 motion_x = motion->pmv[0][0] + get_motion_delta (decoder, \
1299 motion->f_code[0]); \
1300 motion_x = bound_motion_vector (motion_x, motion->f_code[0]); \
1301 motion->pmv[1][0] = motion->pmv[0][0] = motion_x; \
1302 \
1303 NEEDBITS (bit_buf, bits, bit_ptr); \
1304 motion_y = motion->pmv[0][1] + get_motion_delta (decoder, \
1305 motion->f_code[1]); \
1306 motion_y = bound_motion_vector (motion_y, motion->f_code[1]); \
1307 motion->pmv[1][1] = motion->pmv[0][1] = motion_y; \
1308 \
1309 MOTION (table, motion->ref[0], motion_x, motion_y, 16, 0); \
1310 } \
1311 \
1312 static void motion_fr_field_##FORMAT (mpeg2_decoder_t * const decoder, \
1313 motion_t * const motion, \
1314 mpeg2_mc_fct * const * const table) \
1315 { \
1316 int motion_x, motion_y, field; \
1317 unsigned int pos_x, pos_y, xy_half, offset; \
1318 \
1319 NEEDBITS (bit_buf, bits, bit_ptr); \
1320 field = UBITS (bit_buf, 1); \
1321 DUMPBITS (bit_buf, bits, 1); \
1322 \
1323 motion_x = motion->pmv[0][0] + get_motion_delta (decoder, \
1324 motion->f_code[0]); \
1325 motion_x = bound_motion_vector (motion_x, motion->f_code[0]); \
1326 motion->pmv[0][0] = motion_x; \
1327 \
1328 NEEDBITS (bit_buf, bits, bit_ptr); \
1329 motion_y = ((motion->pmv[0][1] >> 1) + \
1330 get_motion_delta (decoder, motion->f_code[1])); \
1331 /* motion_y = bound_motion_vector (motion_y, motion->f_code[1]); */ \
1332 motion->pmv[0][1] = motion_y << 1; \
1333 \
1334 MOTION_FIELD (table, motion->ref[0], motion_x, motion_y, 0, & ~1, field); \
1335 \
1336 NEEDBITS (bit_buf, bits, bit_ptr); \
1337 field = UBITS (bit_buf, 1); \
1338 DUMPBITS (bit_buf, bits, 1); \
1339 \
1340 motion_x = motion->pmv[1][0] + get_motion_delta (decoder, \
1341 motion->f_code[0]); \
1342 motion_x = bound_motion_vector (motion_x, motion->f_code[0]); \
1343 motion->pmv[1][0] = motion_x; \
1344 \
1345 NEEDBITS (bit_buf, bits, bit_ptr); \
1346 motion_y = ((motion->pmv[1][1] >> 1) + \
1347 get_motion_delta (decoder, motion->f_code[1])); \
1348 /* motion_y = bound_motion_vector (motion_y, motion->f_code[1]); */ \
1349 motion->pmv[1][1] = motion_y << 1; \
1350 \
1351 MOTION_FIELD (table, motion->ref[0], motion_x, motion_y, 1, & ~1, field); \
1352 } \
1353 \
1354 static void motion_fr_dmv_##FORMAT (mpeg2_decoder_t * const decoder, \
1355 motion_t * const motion, \
1356 mpeg2_mc_fct * const * const table) \
1357 { \
1358 int motion_x, motion_y, dmv_x, dmv_y, m, other_x, other_y; \
1359 unsigned int pos_x, pos_y, xy_half, offset; \
1360 \
1361 NEEDBITS (bit_buf, bits, bit_ptr); \
1362 motion_x = motion->pmv[0][0] + get_motion_delta (decoder, \
1363 motion->f_code[0]); \
1364 motion_x = bound_motion_vector (motion_x, motion->f_code[0]); \
1365 motion->pmv[1][0] = motion->pmv[0][0] = motion_x; \
1366 NEEDBITS (bit_buf, bits, bit_ptr); \
1367 dmv_x = get_dmv (decoder); \
1368 \
1369 motion_y = ((motion->pmv[0][1] >> 1) + \
1370 get_motion_delta (decoder, motion->f_code[1])); \
1371 /* motion_y = bound_motion_vector (motion_y, motion->f_code[1]); */ \
1372 motion->pmv[1][1] = motion->pmv[0][1] = motion_y << 1; \
1373 dmv_y = get_dmv (decoder); \
1374 \
1375 m = decoder->top_field_first ? 1 : 3; \
1376 other_x = ((motion_x * m + (motion_x > 0)) >> 1) + dmv_x; \
1377 other_y = ((motion_y * m + (motion_y > 0)) >> 1) + dmv_y - 1; \
1378 MOTION_FIELD (mpeg2_mc.put, motion->ref[0], other_x, other_y, 0, | 1, 0); \
1379 \
1380 m = decoder->top_field_first ? 3 : 1; \
1381 other_x = ((motion_x * m + (motion_x > 0)) >> 1) + dmv_x; \
1382 other_y = ((motion_y * m + (motion_y > 0)) >> 1) + dmv_y + 1; \
1383 MOTION_FIELD (mpeg2_mc.put, motion->ref[0], other_x, other_y, 1, & ~1, 0);\
1384 \
1385 MOTION_DMV (mpeg2_mc.avg, motion->ref[0], motion_x, motion_y); \
1386 } \
1387 \
1388 static void motion_reuse_##FORMAT (mpeg2_decoder_t * const decoder, \
1389 motion_t * const motion, \
1390 mpeg2_mc_fct * const * const table) \
1391 { \
1392 int motion_x, motion_y; \
1393 unsigned int pos_x, pos_y, xy_half, offset; \
1394 \
1395 motion_x = motion->pmv[0][0]; \
1396 motion_y = motion->pmv[0][1]; \
1397 \
1398 MOTION (table, motion->ref[0], motion_x, motion_y, 16, 0); \
1399 } \
1400 \
1401 static void motion_zero_##FORMAT (mpeg2_decoder_t * const decoder, \
1402 motion_t * const motion, \
1403 mpeg2_mc_fct * const * const table) \
1404 { \
1405 unsigned int offset; \
1406 \
1407 motion->pmv[0][0] = motion->pmv[0][1] = 0; \
1408 motion->pmv[1][0] = motion->pmv[1][1] = 0; \
1409 \
1410 MOTION_ZERO (table, motion->ref[0]); \
1411 } \
1412 \
1413 static void motion_fi_field_##FORMAT (mpeg2_decoder_t * const decoder, \
1414 motion_t * const motion, \
1415 mpeg2_mc_fct * const * const table) \
1416 { \
1417 int motion_x, motion_y; \
1418 uint8_t ** ref_field; \
1419 unsigned int pos_x, pos_y, xy_half, offset; \
1420 \
1421 NEEDBITS (bit_buf, bits, bit_ptr); \
1422 ref_field = motion->ref2[UBITS (bit_buf, 1)]; \
1423 DUMPBITS (bit_buf, bits, 1); \
1424 \
1425 motion_x = motion->pmv[0][0] + get_motion_delta (decoder, \
1426 motion->f_code[0]); \
1427 motion_x = bound_motion_vector (motion_x, motion->f_code[0]); \
1428 motion->pmv[1][0] = motion->pmv[0][0] = motion_x; \
1429 \
1430 NEEDBITS (bit_buf, bits, bit_ptr); \
1431 motion_y = motion->pmv[0][1] + get_motion_delta (decoder, \
1432 motion->f_code[1]); \
1433 motion_y = bound_motion_vector (motion_y, motion->f_code[1]); \
1434 motion->pmv[1][1] = motion->pmv[0][1] = motion_y; \
1435 \
1436 MOTION (table, ref_field, motion_x, motion_y, 16, 0); \
1437 } \
1438 \
1439 static void motion_fi_16x8_##FORMAT (mpeg2_decoder_t * const decoder, \
1440 motion_t * const motion, \
1441 mpeg2_mc_fct * const * const table) \
1442 { \
1443 int motion_x, motion_y; \
1444 uint8_t ** ref_field; \
1445 unsigned int pos_x, pos_y, xy_half, offset; \
1446 \
1447 NEEDBITS (bit_buf, bits, bit_ptr); \
1448 ref_field = motion->ref2[UBITS (bit_buf, 1)]; \
1449 DUMPBITS (bit_buf, bits, 1); \
1450 \
1451 motion_x = motion->pmv[0][0] + get_motion_delta (decoder, \
1452 motion->f_code[0]); \
1453 motion_x = bound_motion_vector (motion_x, motion->f_code[0]); \
1454 motion->pmv[0][0] = motion_x; \
1455 \
1456 NEEDBITS (bit_buf, bits, bit_ptr); \
1457 motion_y = motion->pmv[0][1] + get_motion_delta (decoder, \
1458 motion->f_code[1]); \
1459 motion_y = bound_motion_vector (motion_y, motion->f_code[1]); \
1460 motion->pmv[0][1] = motion_y; \
1461 \
1462 MOTION (table, ref_field, motion_x, motion_y, 8, 0); \
1463 \
1464 NEEDBITS (bit_buf, bits, bit_ptr); \
1465 ref_field = motion->ref2[UBITS (bit_buf, 1)]; \
1466 DUMPBITS (bit_buf, bits, 1); \
1467 \
1468 motion_x = motion->pmv[1][0] + get_motion_delta (decoder, \
1469 motion->f_code[0]); \
1470 motion_x = bound_motion_vector (motion_x, motion->f_code[0]); \
1471 motion->pmv[1][0] = motion_x; \
1472 \
1473 NEEDBITS (bit_buf, bits, bit_ptr); \
1474 motion_y = motion->pmv[1][1] + get_motion_delta (decoder, \
1475 motion->f_code[1]); \
1476 motion_y = bound_motion_vector (motion_y, motion->f_code[1]); \
1477 motion->pmv[1][1] = motion_y; \
1478 \
1479 MOTION (table, ref_field, motion_x, motion_y, 8, 8); \
1480 } \
1481 \
1482 static void motion_fi_dmv_##FORMAT (mpeg2_decoder_t * const decoder, \
1483 motion_t * const motion, \
1484 mpeg2_mc_fct * const * const table) \
1485 { \
1486 int motion_x, motion_y, other_x, other_y; \
1487 unsigned int pos_x, pos_y, xy_half, offset; \
1488 \
1489 NEEDBITS (bit_buf, bits, bit_ptr); \
1490 motion_x = motion->pmv[0][0] + get_motion_delta (decoder, \
1491 motion->f_code[0]); \
1492 motion_x = bound_motion_vector (motion_x, motion->f_code[0]); \
1493 motion->pmv[1][0] = motion->pmv[0][0] = motion_x; \
1494 NEEDBITS (bit_buf, bits, bit_ptr); \
1495 other_x = ((motion_x + (motion_x > 0)) >> 1) + get_dmv (decoder); \
1496 \
1497 motion_y = motion->pmv[0][1] + get_motion_delta (decoder, \
1498 motion->f_code[1]); \
1499 motion_y = bound_motion_vector (motion_y, motion->f_code[1]); \
1500 motion->pmv[1][1] = motion->pmv[0][1] = motion_y; \
1501 other_y = (((motion_y + (motion_y > 0)) >> 1) + get_dmv (decoder) + \
1502 decoder->dmv_offset); \
1503 \
1504 MOTION (mpeg2_mc.put, motion->ref[0], motion_x, motion_y, 16, 0); \
1505 MOTION (mpeg2_mc.avg, motion->ref[1], other_x, other_y, 16, 0); \
1506 } \
1507
1508 MOTION_FUNCTIONS (420, MOTION_420, MOTION_FIELD_420, MOTION_DMV_420,
1509 MOTION_ZERO_420)
1510 MOTION_FUNCTIONS (422, MOTION_422, MOTION_FIELD_422, MOTION_DMV_422,
1511 MOTION_ZERO_422)
1512 MOTION_FUNCTIONS (444, MOTION_444, MOTION_FIELD_444, MOTION_DMV_444,
1513 MOTION_ZERO_444)
1514
1515 /* like motion_frame, but parsing without actual motion compensation */
1516 static void motion_fr_conceal (mpeg2_decoder_t * const decoder)
1517 {
1518 int tmp;
1519
1520 NEEDBITS (bit_buf, bits, bit_ptr);
1521 tmp = (decoder->f_motion.pmv[0][0] +
1522 get_motion_delta (decoder, decoder->f_motion.f_code[0]));
1523 tmp = bound_motion_vector (tmp, decoder->f_motion.f_code[0]);
1524 decoder->f_motion.pmv[1][0] = decoder->f_motion.pmv[0][0] = tmp;
1525
1526 NEEDBITS (bit_buf, bits, bit_ptr);
1527 tmp = (decoder->f_motion.pmv[0][1] +
1528 get_motion_delta (decoder, decoder->f_motion.f_code[1]));
1529 tmp = bound_motion_vector (tmp, decoder->f_motion.f_code[1]);
1530 decoder->f_motion.pmv[1][1] = decoder->f_motion.pmv[0][1] = tmp;
1531
1532 DUMPBITS (bit_buf, bits, 1); /* remove marker_bit */
1533 }
1534
1535 static void motion_fi_conceal (mpeg2_decoder_t * const decoder)
1536 {
1537 int tmp;
1538
1539 NEEDBITS (bit_buf, bits, bit_ptr);
1540 DUMPBITS (bit_buf, bits, 1); /* remove field_select */
1541
1542 tmp = (decoder->f_motion.pmv[0][0] +
1543 get_motion_delta (decoder, decoder->f_motion.f_code[0]));
1544 tmp = bound_motion_vector (tmp, decoder->f_motion.f_code[0]);
1545 decoder->f_motion.pmv[1][0] = decoder->f_motion.pmv[0][0] = tmp;
1546
1547 NEEDBITS (bit_buf, bits, bit_ptr);
1548 tmp = (decoder->f_motion.pmv[0][1] +
1549 get_motion_delta (decoder, decoder->f_motion.f_code[1]));
1550 tmp = bound_motion_vector (tmp, decoder->f_motion.f_code[1]);
1551 decoder->f_motion.pmv[1][1] = decoder->f_motion.pmv[0][1] = tmp;
1552
1553 DUMPBITS (bit_buf, bits, 1); /* remove marker_bit */
1554 }
1555
1556 #undef bit_buf
1557 #undef bits
1558 #undef bit_ptr
1559
1560 #define MOTION_CALL(routine,direction) \
1561 do { \
1562 if ((direction) & MACROBLOCK_MOTION_FORWARD) \
1563 routine (decoder, &(decoder->f_motion), mpeg2_mc.put); \
1564 if ((direction) & MACROBLOCK_MOTION_BACKWARD) \
1565 routine (decoder, &(decoder->b_motion), \
1566 ((direction) & MACROBLOCK_MOTION_FORWARD ? \
1567 mpeg2_mc.avg : mpeg2_mc.put)); \
1568 } while (0)
1569
1570 #define NEXT_MACROBLOCK \
1571 do { \
1572 if(decoder->quant_store) \
1573 decoder->quant_store[decoder->quant_stride*(decoder->v_offset>>4) \
1574 +(decoder->offset>>4)] = decoder->quantizer_scale; \
1575 decoder->offset += 16; \
1576 if (decoder->offset == decoder->width) { \
1577 do { /* just so we can use the break statement */ \
1578 if (decoder->convert) { \
1579 decoder->convert (decoder->convert_id, decoder->dest, \
1580 decoder->v_offset); \
1581 if (decoder->coding_type == B_TYPE) \
1582 break; \
1583 } \
1584 decoder->dest[0] += decoder->slice_stride; \
1585 decoder->dest[1] += decoder->slice_uv_stride; \
1586 decoder->dest[2] += decoder->slice_uv_stride; \
1587 } while (0); \
1588 decoder->v_offset += 16; \
1589 if (decoder->v_offset > decoder->limit_y) { \
1590 if (mpeg2_cpu_state_restore) \
1591 mpeg2_cpu_state_restore (&cpu_state); \
1592 return; \
1593 } \
1594 decoder->offset = 0; \
1595 } \
1596 } while (0)
1597
1598 void mpeg2_init_fbuf (mpeg2_decoder_t * decoder, uint8_t * current_fbuf[3],
1599 uint8_t * forward_fbuf[3], uint8_t * backward_fbuf[3])
1600 {
1601 int offset, stride, height, bottom_field;
1602
1603 stride = decoder->stride_frame;
1604 bottom_field = (decoder->picture_structure == BOTTOM_FIELD);
1605 offset = bottom_field ? stride : 0;
1606 height = decoder->height;
1607
1608 decoder->picture_dest[0] = current_fbuf[0] + offset;
1609 decoder->picture_dest[1] = current_fbuf[1] + (offset >> 1);
1610 decoder->picture_dest[2] = current_fbuf[2] + (offset >> 1);
1611
1612 decoder->f_motion.ref[0][0] = forward_fbuf[0] + offset;
1613 decoder->f_motion.ref[0][1] = forward_fbuf[1] + (offset >> 1);
1614 decoder->f_motion.ref[0][2] = forward_fbuf[2] + (offset >> 1);
1615
1616 decoder->b_motion.ref[0][0] = backward_fbuf[0] + offset;
1617 decoder->b_motion.ref[0][1] = backward_fbuf[1] + (offset >> 1);
1618 decoder->b_motion.ref[0][2] = backward_fbuf[2] + (offset >> 1);
1619
1620 if (decoder->picture_structure != FRAME_PICTURE) {
1621 decoder->dmv_offset = bottom_field ? 1 : -1;
1622 decoder->f_motion.ref2[0] = decoder->f_motion.ref[bottom_field];
1623 decoder->f_motion.ref2[1] = decoder->f_motion.ref[!bottom_field];
1624 decoder->b_motion.ref2[0] = decoder->b_motion.ref[bottom_field];
1625 decoder->b_motion.ref2[1] = decoder->b_motion.ref[!bottom_field];
1626 offset = stride - offset;
1627
1628 if (decoder->second_field && (decoder->coding_type != B_TYPE))
1629 forward_fbuf = current_fbuf;
1630
1631 decoder->f_motion.ref[1][0] = forward_fbuf[0] + offset;
1632 decoder->f_motion.ref[1][1] = forward_fbuf[1] + (offset >> 1);
1633 decoder->f_motion.ref[1][2] = forward_fbuf[2] + (offset >> 1);
1634
1635 decoder->b_motion.ref[1][0] = backward_fbuf[0] + offset;
1636 decoder->b_motion.ref[1][1] = backward_fbuf[1] + (offset >> 1);
1637 decoder->b_motion.ref[1][2] = backward_fbuf[2] + (offset >> 1);
1638
1639 stride <<= 1;
1640 height >>= 1;
1641 }
1642
1643 decoder->stride = stride;
1644 decoder->uv_stride = stride >> 1;
1645 decoder->slice_stride = 16 * stride;
1646 decoder->slice_uv_stride =
1647 decoder->slice_stride >> (2 - decoder->chroma_format);
1648 decoder->limit_x = 2 * decoder->width - 32;
1649 decoder->limit_y_16 = 2 * height - 32;
1650 decoder->limit_y_8 = 2 * height - 16;
1651 decoder->limit_y = height - 16;
1652
1653 if (decoder->mpeg1) {
1654 decoder->motion_parser[0] = motion_zero_420;
1655 decoder->motion_parser[MC_FRAME] = motion_mp1;
1656 decoder->motion_parser[4] = motion_reuse_420;
1657 } else if (decoder->picture_structure == FRAME_PICTURE) {
1658 if (decoder->chroma_format == 0) {
1659 decoder->motion_parser[0] = motion_zero_420;
1660 decoder->motion_parser[MC_FIELD] = motion_fr_field_420;
1661 decoder->motion_parser[MC_FRAME] = motion_fr_frame_420;
1662 decoder->motion_parser[MC_DMV] = motion_fr_dmv_420;
1663 decoder->motion_parser[4] = motion_reuse_420;
1664 } else if (decoder->chroma_format == 1) {
1665 decoder->motion_parser[0] = motion_zero_422;
1666 decoder->motion_parser[MC_FIELD] = motion_fr_field_422;
1667 decoder->motion_parser[MC_FRAME] = motion_fr_frame_422;
1668 decoder->motion_parser[MC_DMV] = motion_fr_dmv_422;
1669 decoder->motion_parser[4] = motion_reuse_422;
1670 } else {
1671 decoder->motion_parser[0] = motion_zero_444;
1672 decoder->motion_parser[MC_FIELD] = motion_fr_field_444;
1673 decoder->motion_parser[MC_FRAME] = motion_fr_frame_444;
1674 decoder->motion_parser[MC_DMV] = motion_fr_dmv_444;
1675 decoder->motion_parser[4] = motion_reuse_444;
1676 }
1677 } else {
1678 if (decoder->chroma_format == 0) {
1679 decoder->motion_parser[0] = motion_zero_420;
1680 decoder->motion_parser[MC_FIELD] = motion_fi_field_420;
1681 decoder->motion_parser[MC_16X8] = motion_fi_16x8_420;
1682 decoder->motion_parser[MC_DMV] = motion_fi_dmv_420;
1683 decoder->motion_parser[4] = motion_reuse_420;
1684 } else if (decoder->chroma_format == 1) {
1685 decoder->motion_parser[0] = motion_zero_422;
1686 decoder->motion_parser[MC_FIELD] = motion_fi_field_422;
1687 decoder->motion_parser[MC_16X8] = motion_fi_16x8_422;
1688 decoder->motion_parser[MC_DMV] = motion_fi_dmv_422;
1689 decoder->motion_parser[4] = motion_reuse_422;
1690 } else {
1691 decoder->motion_parser[0] = motion_zero_444;
1692 decoder->motion_parser[MC_FIELD] = motion_fi_field_444;
1693 decoder->motion_parser[MC_16X8] = motion_fi_16x8_444;
1694 decoder->motion_parser[MC_DMV] = motion_fi_dmv_444;
1695 decoder->motion_parser[4] = motion_reuse_444;
1696 }
1697 }
1698 }
1699
1700 static inline int slice_init (mpeg2_decoder_t * const decoder, int code)
1701 {
1702 #define bit_buf (decoder->bitstream_buf)
1703 #define bits (decoder->bitstream_bits)
1704 #define bit_ptr (decoder->bitstream_ptr)
1705 int offset;
1706 const MBAtab * mba;
1707
1708 decoder->dc_dct_pred[0] = decoder->dc_dct_pred[1] =
1709 decoder->dc_dct_pred[2] = 16384;
1710
1711 decoder->f_motion.pmv[0][0] = decoder->f_motion.pmv[0][1] = 0;
1712 decoder->f_motion.pmv[1][0] = decoder->f_motion.pmv[1][1] = 0;
1713 decoder->b_motion.pmv[0][0] = decoder->b_motion.pmv[0][1] = 0;
1714 decoder->b_motion.pmv[1][0] = decoder->b_motion.pmv[1][1] = 0;
1715
1716 if (decoder->vertical_position_extension) {
1717 code += UBITS (bit_buf, 3) << 7;
1718 DUMPBITS (bit_buf, bits, 3);
1719 }
1720 decoder->v_offset = (code - 1) * 16;
1721 offset = 0;
1722 if (!(decoder->convert) || decoder->coding_type != B_TYPE)
1723 offset = (code - 1) * decoder->slice_stride;
1724
1725 decoder->dest[0] = decoder->picture_dest[0] + offset;
1726 offset >>= (2 - decoder->chroma_format);
1727 decoder->dest[1] = decoder->picture_dest[1] + offset;
1728 decoder->dest[2] = decoder->picture_dest[2] + offset;
1729
1730 get_quantizer_scale (decoder);
1731
1732 /* ignore intra_slice and all the extra data */
1733 while (bit_buf & 0x80000000) {
1734 DUMPBITS (bit_buf, bits, 9);
1735 NEEDBITS (bit_buf, bits, bit_ptr);
1736 }
1737
1738 /* decode initial macroblock address increment */
1739 offset = 0;
1740 while (1) {
1741 if (bit_buf >= 0x08000000) {
1742 mba = MBA_5 + (UBITS (bit_buf, 6) - 2);
1743 break;
1744 } else if (bit_buf >= 0x01800000) {
1745 mba = MBA_11 + (UBITS (bit_buf, 12) - 24);
1746 break;
1747 } else switch (UBITS (bit_buf, 12)) {
1748 case 8: /* macroblock_escape */
1749 offset += 33;
1750 DUMPBITS (bit_buf, bits, 11);
1751 NEEDBITS (bit_buf, bits, bit_ptr);
1752 continue;
1753 case 15: /* macroblock_stuffing (MPEG1 only) */
1754 bit_buf &= 0xfffff;
1755 DUMPBITS (bit_buf, bits, 11);
1756 NEEDBITS (bit_buf, bits, bit_ptr);
1757 continue;
1758 default: /* error */
1759 return 1;
1760 }
1761 }
1762 DUMPBITS (bit_buf, bits, mba->len + 1);
1763 decoder->offset = (offset + mba->mba) << 4;
1764
1765 while (decoder->offset - decoder->width >= 0) {
1766 decoder->offset -= decoder->width;
1767 if (!(decoder->convert) || decoder->coding_type != B_TYPE) {
1768 decoder->dest[0] += decoder->slice_stride;
1769 decoder->dest[1] += decoder->slice_uv_stride;
1770 decoder->dest[2] += decoder->slice_uv_stride;
1771 }
1772 decoder->v_offset += 16;
1773 }
1774 if (decoder->v_offset > decoder->limit_y)
1775 return 1;
1776
1777 return 0;
1778 #undef bit_buf
1779 #undef bits
1780 #undef bit_ptr
1781 }
1782
1783 void mpeg2_slice (mpeg2_decoder_t * const decoder, const int code,
1784 const uint8_t * const buffer)
1785 {
1786 #define bit_buf (decoder->bitstream_buf)
1787 #define bits (decoder->bitstream_bits)
1788 #define bit_ptr (decoder->bitstream_ptr)
1789 cpu_state_t cpu_state;
1790
1791 bitstream_init (decoder, buffer);
1792
1793 if (slice_init (decoder, code))
1794 return;
1795
1796 if (mpeg2_cpu_state_save)
1797 mpeg2_cpu_state_save (&cpu_state);
1798
1799 while (1) {
1800 int macroblock_modes;
1801 int mba_inc;
1802 const MBAtab * mba;
1803
1804 NEEDBITS (bit_buf, bits, bit_ptr);
1805
1806 macroblock_modes = get_macroblock_modes (decoder);
1807
1808 /* maybe integrate MACROBLOCK_QUANT test into get_macroblock_modes ? */
1809 if (macroblock_modes & MACROBLOCK_QUANT)
1810 get_quantizer_scale (decoder);
1811
1812 if (macroblock_modes & MACROBLOCK_INTRA) {
1813
1814 int DCT_offset, DCT_stride;
1815 int offset;
1816 uint8_t * dest_y;
1817
1818 if (decoder->concealment_motion_vectors) {
1819 if (decoder->picture_structure == FRAME_PICTURE)
1820 motion_fr_conceal (decoder);
1821 else
1822 motion_fi_conceal (decoder);
1823 } else {
1824 decoder->f_motion.pmv[0][0] = decoder->f_motion.pmv[0][1] = 0;
1825 decoder->f_motion.pmv[1][0] = decoder->f_motion.pmv[1][1] = 0;
1826 decoder->b_motion.pmv[0][0] = decoder->b_motion.pmv[0][1] = 0;
1827 decoder->b_motion.pmv[1][0] = decoder->b_motion.pmv[1][1] = 0;
1828 }
1829
1830 if (macroblock_modes & DCT_TYPE_INTERLACED) {
1831 DCT_offset = decoder->stride;
1832 DCT_stride = decoder->stride * 2;
1833 } else {
1834 DCT_offset = decoder->stride * 8;
1835 DCT_stride = decoder->stride;
1836 }
1837
1838 offset = decoder->offset;
1839 dest_y = decoder->dest[0] + offset;
1840 slice_intra_DCT (decoder, 0, dest_y, DCT_stride);
1841 slice_intra_DCT (decoder, 0, dest_y + 8, DCT_stride);
1842 slice_intra_DCT (decoder, 0, dest_y + DCT_offset, DCT_stride);
1843 slice_intra_DCT (decoder, 0, dest_y + DCT_offset + 8, DCT_stride);
1844 if (likely (decoder->chroma_format == 0)) {
1845 slice_intra_DCT (decoder, 1, decoder->dest[1] + (offset >> 1),
1846 decoder->uv_stride);
1847 slice_intra_DCT (decoder, 2, decoder->dest[2] + (offset >> 1),
1848 decoder->uv_stride);
1849 if (decoder->coding_type == D_TYPE) {
1850 NEEDBITS (bit_buf, bits, bit_ptr);
1851 DUMPBITS (bit_buf, bits, 1);
1852 }
1853 } else if (likely (decoder->chroma_format == 1)) {
1854 uint8_t * dest_u = decoder->dest[1] + (offset >> 1);
1855 uint8_t * dest_v = decoder->dest[2] + (offset >> 1);
1856 DCT_stride >>= 1;
1857 DCT_offset >>= 1;
1858 slice_intra_DCT (decoder, 1, dest_u, DCT_stride);
1859 slice_intra_DCT (decoder, 2, dest_v, DCT_stride);
1860 slice_intra_DCT (decoder, 1, dest_u + DCT_offset, DCT_stride);
1861 slice_intra_DCT (decoder, 2, dest_v + DCT_offset, DCT_stride);
1862 } else {
1863 uint8_t * dest_u = decoder->dest[1] + offset;
1864 uint8_t * dest_v = decoder->dest[2] + offset;
1865 slice_intra_DCT (decoder, 1, dest_u, DCT_stride);
1866 slice_intra_DCT (decoder, 2, dest_v, DCT_stride);
1867 slice_intra_DCT (decoder, 1, dest_u + DCT_offset, DCT_stride);
1868 slice_intra_DCT (decoder, 2, dest_v + DCT_offset, DCT_stride);
1869 slice_intra_DCT (decoder, 1, dest_u + 8, DCT_stride);
1870 slice_intra_DCT (decoder, 2, dest_v + 8, DCT_stride);
1871 slice_intra_DCT (decoder, 1, dest_u + DCT_offset + 8,
1872 DCT_stride);
1873 slice_intra_DCT (decoder, 2, dest_v + DCT_offset + 8,
1874 DCT_stride);
1875 }
1876 } else {
1877
1878 motion_parser_t * parser;
1879
1880 parser =
1881 decoder->motion_parser[macroblock_modes >> MOTION_TYPE_SHIFT];
1882 MOTION_CALL (parser, macroblock_modes);
1883
1884 if (macroblock_modes & MACROBLOCK_PATTERN) {
1885 int coded_block_pattern;
1886 int DCT_offset, DCT_stride;
1887
1888 if (macroblock_modes & DCT_TYPE_INTERLACED) {
1889 DCT_offset = decoder->stride;
1890 DCT_stride = decoder->stride * 2;
1891 } else {
1892 DCT_offset = decoder->stride * 8;
1893 DCT_stride = decoder->stride;
1894 }
1895
1896 coded_block_pattern = get_coded_block_pattern (decoder);
1897
1898 if (likely (decoder->chroma_format == 0)) {
1899 int offset = decoder->offset;
1900 uint8_t * dest_y = decoder->dest[0] + offset;
1901 if (coded_block_pattern & 1)
1902 slice_non_intra_DCT (decoder, 0, dest_y, DCT_stride);
1903 if (coded_block_pattern & 2)
1904 slice_non_intra_DCT (decoder, 0, dest_y + 8,
1905 DCT_stride);
1906 if (coded_block_pattern & 4)
1907 slice_non_intra_DCT (decoder, 0, dest_y + DCT_offset,
1908 DCT_stride);
1909 if (coded_block_pattern & 8)
1910 slice_non_intra_DCT (decoder, 0,
1911 dest_y + DCT_offset + 8,
1912 DCT_stride);
1913 if (coded_block_pattern & 16)
1914 slice_non_intra_DCT (decoder, 1,
1915 decoder->dest[1] + (offset >> 1),
1916 decoder->uv_stride);
1917 if (coded_block_pattern & 32)
1918 slice_non_intra_DCT (decoder, 2,
1919 decoder->dest[2] + (offset >> 1),
1920 decoder->uv_stride);
1921 } else if (likely (decoder->chroma_format == 1)) {
1922 int offset;
1923 uint8_t * dest_y;
1924
1925 coded_block_pattern |= bit_buf & (3 << 30);
1926 DUMPBITS (bit_buf, bits, 2);
1927
1928 offset = decoder->offset;
1929 dest_y = decoder->dest[0] + offset;
1930 if (coded_block_pattern & 1)
1931 slice_non_intra_DCT (decoder, 0, dest_y, DCT_stride);
1932 if (coded_block_pattern & 2)
1933 slice_non_intra_DCT (decoder, 0, dest_y + 8,
1934 DCT_stride);
1935 if (coded_block_pattern & 4)
1936 slice_non_intra_DCT (decoder, 0, dest_y + DCT_offset,
1937 DCT_stride);
1938 if (coded_block_pattern & 8)
1939 slice_non_intra_DCT (decoder, 0,
1940 dest_y + DCT_offset + 8,
1941 DCT_stride);
1942
1943 DCT_stride >>= 1;
1944 DCT_offset = (DCT_offset + offset) >> 1;
1945 if (coded_block_pattern & 16)
1946 slice_non_intra_DCT (decoder, 1,
1947 decoder->dest[1] + (offset >> 1),
1948 DCT_stride);
1949 if (coded_block_pattern & 32)
1950 slice_non_intra_DCT (decoder, 2,
1951 decoder->dest[2] + (offset >> 1),
1952 DCT_stride);
1953 if (coded_block_pattern & (2 << 30))
1954 slice_non_intra_DCT (decoder, 1,
1955 decoder->dest[1] + DCT_offset,
1956 DCT_stride);
1957 if (coded_block_pattern & (1 << 30))
1958 slice_non_intra_DCT (decoder, 2,
1959 decoder->dest[2] + DCT_offset,
1960 DCT_stride);
1961 } else {
1962 int offset;
1963 uint8_t * dest_y, * dest_u, * dest_v;
1964
1965 coded_block_pattern |= bit_buf & (63 << 26);
1966 DUMPBITS (bit_buf, bits, 6);
1967
1968 offset = decoder->offset;
1969 dest_y = decoder->dest[0] + offset;
1970 dest_u = decoder->dest[1] + offset;
1971 dest_v = decoder->dest[2] + offset;
1972
1973 if (coded_block_pattern & 1)
1974 slice_non_intra_DCT (decoder, 0, dest_y, DCT_stride);
1975 if (coded_block_pattern & 2)
1976 slice_non_intra_DCT (decoder, 0, dest_y + 8,
1977 DCT_stride);
1978 if (coded_block_pattern & 4)
1979 slice_non_intra_DCT (decoder, 0, dest_y + DCT_offset,
1980 DCT_stride);
1981 if (coded_block_pattern & 8)
1982 slice_non_intra_DCT (decoder, 0,
1983 dest_y + DCT_offset + 8,
1984 DCT_stride);
1985
1986 if (coded_block_pattern & 16)
1987 slice_non_intra_DCT (decoder, 1, dest_u, DCT_stride);
1988 if (coded_block_pattern & 32)
1989 slice_non_intra_DCT (decoder, 2, dest_v, DCT_stride);
1990 if (coded_block_pattern & (32 << 26))
1991 slice_non_intra_DCT (decoder, 1, dest_u + DCT_offset,
1992 DCT_stride);
1993 if (coded_block_pattern & (16 << 26))
1994 slice_non_intra_DCT (decoder, 2, dest_v + DCT_offset,
1995 DCT_stride);
1996 if (coded_block_pattern & (8 << 26))
1997 slice_non_intra_DCT (decoder, 1, dest_u + 8,
1998 DCT_stride);
1999 if (coded_block_pattern & (4 << 26))
2000 slice_non_intra_DCT (decoder, 2, dest_v + 8,
2001 DCT_stride);
2002 if (coded_block_pattern & (2 << 26))
2003 slice_non_intra_DCT (decoder, 1,
2004 dest_u + DCT_offset + 8,
2005 DCT_stride);
2006 if (coded_block_pattern & (1 << 26))
2007 slice_non_intra_DCT (decoder, 2,
2008 dest_v + DCT_offset + 8,
2009 DCT_stride);
2010 }
2011 }
2012
2013 decoder->dc_dct_pred[0] = decoder->dc_dct_pred[1] =
2014 decoder->dc_dct_pred[2] = 16384;
2015 }
2016
2017 NEXT_MACROBLOCK;
2018
2019 NEEDBITS (bit_buf, bits, bit_ptr);
2020 mba_inc = 0;
2021 while (1) {
2022 if (bit_buf >= 0x10000000) {
2023 mba = MBA_5 + (UBITS (bit_buf, 5) - 2);
2024 break;
2025 } else if (bit_buf >= 0x03000000) {
2026 mba = MBA_11 + (UBITS (bit_buf, 11) - 24);
2027 break;
2028 } else switch (UBITS (bit_buf, 11)) {
2029 case 8: /* macroblock_escape */
2030 mba_inc += 33;
2031 /* pass through */
2032 case 15: /* macroblock_stuffing (MPEG1 only) */
2033 DUMPBITS (bit_buf, bits, 11);
2034 NEEDBITS (bit_buf, bits, bit_ptr);
2035 continue;
2036 default: /* end of slice, or error */
2037 if (mpeg2_cpu_state_restore)
2038 mpeg2_cpu_state_restore (&cpu_state);
2039 return;
2040 }
2041 }
2042 DUMPBITS (bit_buf, bits, mba->len);
2043 mba_inc += mba->mba;
2044
2045 if (mba_inc) {
2046 decoder->dc_dct_pred[0] = decoder->dc_dct_pred[1] =
2047 decoder->dc_dct_pred[2] = 16384;
2048
2049 if (decoder->coding_type == P_TYPE) {
2050 do {
2051 MOTION_CALL (decoder->motion_parser[0],
2052 MACROBLOCK_MOTION_FORWARD);
2053 NEXT_MACROBLOCK;
2054 } while (--mba_inc);
2055 } else {
2056 do {
2057 MOTION_CALL (decoder->motion_parser[4], macroblock_modes);
2058 NEXT_MACROBLOCK;
2059 } while (--mba_inc);
2060 }
2061 }
2062 }
2063 #undef bit_buf
2064 #undef bits
2065 #undef bit_ptr
2066 }