1 |
/***************************************************************************** |
/***************************************************************************** |
2 |
* |
* |
3 |
* XVID MPEG-4 VIDEO CODEC |
* XVID MPEG-4 VIDEO CODEC |
4 |
* - Vector Length Coding tables - |
* - Macro Block coding functions - |
5 |
* |
* |
6 |
* Copyright(C) 2002 Michael Militzer <isibaar@xvid.org> |
* Copyright(C) 2002 Michael Militzer <isibaar@xvid.org> |
7 |
* |
* |
49 |
* Local data |
* Local data |
50 |
****************************************************************************/ |
****************************************************************************/ |
51 |
|
|
52 |
static VLC intra_table[524032]; |
/* msvc sp5+pp gets confused if they globals are made static */ |
53 |
static VLC inter_table[524032]; |
VLC intra_table[524032]; |
54 |
|
VLC inter_table[524032]; |
55 |
|
|
56 |
static VLC DCT3Dintra[4096]; |
static VLC DCT3Dintra[4096]; |
57 |
static VLC DCT3Dinter[4096]; |
static VLC DCT3Dinter[4096]; |
58 |
|
|
59 |
/***************************************************************************** |
/***************************************************************************** |
60 |
* Functions |
* Vector Length Coding Initialization |
61 |
****************************************************************************/ |
****************************************************************************/ |
62 |
|
|
63 |
void |
void |
66 |
|
|
67 |
int32_t k, l, i, intra, last; |
int32_t k, l, i, intra, last; |
68 |
VLC *vlc[2]; |
VLC *vlc[2]; |
69 |
VLC **coeff_ptr; |
VLC const **coeff_ptr; |
70 |
VLC *vlc1, *vlc2; |
VLC *vlc1, *vlc2; |
71 |
|
|
72 |
vlc1 = DCT3Dintra; |
vlc1 = DCT3Dintra; |
87 |
coeff_ptr = coeff_vlc[last + 2 * intra]; |
coeff_ptr = coeff_vlc[last + 2 * intra]; |
88 |
|
|
89 |
for (k = -2047; k < 2048; k++) { // level |
for (k = -2047; k < 2048; k++) { // level |
90 |
int8_t *max_level_ptr = max_level[last + 2 * intra]; |
int8_t const *max_level_ptr = max_level[last + 2 * intra]; |
91 |
int8_t *max_run_ptr = max_run[last + 2 * intra]; |
int8_t const *max_run_ptr = max_run[last + 2 * intra]; |
92 |
|
|
93 |
for (l = 0; l < 64; l++) { // run |
for (l = 0; l < 64; l++) { // run |
94 |
int32_t level = k; |
int32_t level = k; |
181 |
|
|
182 |
} |
} |
183 |
|
|
184 |
|
/***************************************************************************** |
185 |
|
* Local inlined functions for MB coding |
186 |
|
****************************************************************************/ |
187 |
|
|
188 |
static __inline void |
static __inline void |
189 |
CodeVector(Bitstream * bs, |
CodeVector(Bitstream * bs, |
190 |
int32_t value, |
int32_t value, |
240 |
|
|
241 |
} |
} |
242 |
|
|
|
|
|
243 |
static __inline void |
static __inline void |
244 |
CodeCoeff(Bitstream * bs, |
CodeCoeff(Bitstream * bs, |
245 |
const int16_t qcoeff[64], |
const int16_t qcoeff[64], |
278 |
|
|
279 |
} |
} |
280 |
|
|
281 |
|
/***************************************************************************** |
282 |
|
* Local functions |
283 |
|
****************************************************************************/ |
284 |
|
|
285 |
static void |
static void |
286 |
CodeBlockIntra(const FRAMEINFO * frame, |
CodeBlockIntra(const FRAMEINFO * frame, |
407 |
|
|
408 |
} |
} |
409 |
|
|
410 |
|
/***************************************************************************** |
411 |
|
* Macro Block bitstream encoding functions |
412 |
|
****************************************************************************/ |
413 |
|
|
414 |
void |
void |
415 |
MBCoding(const FRAMEINFO * frame, |
MBCoding(const FRAMEINFO * frame, |
438 |
return; |
return; |
439 |
} |
} |
440 |
|
|
441 |
|
/***************************************************************************** |
442 |
/*************************************************************** |
* decoding stuff starts here |
443 |
* bframe encoding start |
****************************************************************************/ |
|
***************************************************************/ |
|
|
|
|
|
/* |
|
|
mbtype |
|
|
0 1b direct(h263) mvdb |
|
|
1 01b interpolate mc+q dbquant, mvdf, mvdb |
|
|
2 001b backward mc+q dbquant, mvdb |
|
|
3 0001b forward mc+q dbquant, mvdf |
|
|
*/ |
|
|
|
|
|
void |
|
|
put_bvop_mbtype(Bitstream * bs, |
|
|
int value) |
|
|
{ |
|
|
switch (value) { |
|
|
case 0: |
|
|
BitstreamPutBit(bs, 1); |
|
|
return; |
|
|
|
|
|
case 1: |
|
|
BitstreamPutBit(bs, 0); |
|
|
BitstreamPutBit(bs, 1); |
|
|
return; |
|
|
|
|
|
case 2: |
|
|
BitstreamPutBit(bs, 0); |
|
|
BitstreamPutBit(bs, 0); |
|
|
BitstreamPutBit(bs, 1); |
|
|
return; |
|
|
|
|
|
case 3: |
|
|
BitstreamPutBit(bs, 0); |
|
|
BitstreamPutBit(bs, 0); |
|
|
BitstreamPutBit(bs, 0); |
|
|
BitstreamPutBit(bs, 1); |
|
|
return; |
|
|
|
|
|
default:; // invalid! |
|
|
|
|
|
} |
|
|
|
|
|
} |
|
444 |
|
|
445 |
/* |
/* |
446 |
dbquant |
* For IVOP addbits == 0 |
447 |
-2 10b |
* For PVOP addbits == fcode - 1 |
448 |
0 0b |
* For BVOP addbits == max(fcode,bcode) - 1 |
449 |
+2 11b |
* returns true or false |
450 |
*/ |
*/ |
451 |
|
|
|
void |
|
|
put_bvop_dbquant(Bitstream * bs, |
|
|
int value) |
|
|
{ |
|
|
switch (value) { |
|
|
case 0: |
|
|
BitstreamPutBit(bs, 0); |
|
|
return; |
|
|
|
|
|
case -2: |
|
|
BitstreamPutBit(bs, 1); |
|
|
BitstreamPutBit(bs, 0); |
|
|
return; |
|
|
|
|
|
case 2: |
|
|
BitstreamPutBit(bs, 1); |
|
|
BitstreamPutBit(bs, 1); |
|
|
return; |
|
|
|
|
|
default:; // invalid |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
void |
|
|
MBCodingBVOP(const MACROBLOCK * mb, |
|
|
const int16_t qcoeff[6 * 64], |
|
|
const int32_t fcode, |
|
|
const int32_t bcode, |
|
|
Bitstream * bs, |
|
|
Statistics * pStat) |
|
|
{ |
|
|
int i; |
|
|
|
|
|
/* ------------------------------------------------------------------ |
|
|
when a block is skipped it is decoded DIRECT(0,0) |
|
|
hence is interpolated from forward & backward frames |
|
|
------------------------------------------------------------------ */ |
|
|
|
|
|
if (mb->mode == MODE_DIRECT_NONE_MV) { |
|
|
BitstreamPutBit(bs, 1); // skipped |
|
|
return; |
|
|
} |
|
|
|
|
|
BitstreamPutBit(bs, 0); // not skipped |
|
|
|
|
|
if (mb->cbp == 0) { |
|
|
BitstreamPutBit(bs, 1); // cbp == 0 |
|
|
} else { |
|
|
BitstreamPutBit(bs, 0); // cbp == xxx |
|
|
} |
|
|
|
|
|
put_bvop_mbtype(bs, mb->mode); |
|
|
|
|
|
if (mb->cbp) { |
|
|
BitstreamPutBits(bs, mb->cbp, 6); |
|
|
} |
|
|
|
|
|
if (mb->mode != MODE_DIRECT && mb->cbp != 0) { |
|
|
put_bvop_dbquant(bs, 0); // todo: mb->dquant = 0 |
|
|
} |
|
|
|
|
|
if (mb->mode == MODE_INTERPOLATE || mb->mode == MODE_FORWARD) { |
|
|
CodeVector(bs, mb->pmvs[0].x, fcode, pStat); |
|
|
CodeVector(bs, mb->pmvs[0].y, fcode, pStat); |
|
|
} |
|
|
|
|
|
if (mb->mode == MODE_INTERPOLATE || mb->mode == MODE_BACKWARD) { |
|
|
CodeVector(bs, mb->b_pmvs[0].x, bcode, pStat); |
|
|
CodeVector(bs, mb->b_pmvs[0].y, bcode, pStat); |
|
|
} |
|
|
|
|
|
if (mb->mode == MODE_DIRECT) { |
|
|
CodeVector(bs, mb->deltamv.x, 1, pStat); /* fcode is always 1 for delta vector */ |
|
|
CodeVector(bs, mb->deltamv.y, 1, pStat); /* prediction is always (0,0) */ |
|
|
} |
|
|
|
|
|
for (i = 0; i < 6; i++) { |
|
|
if (mb->cbp & (1 << (5 - i))) { |
|
|
CodeCoeff(bs, &qcoeff[i * 64], inter_table, scan_tables[0], 0); |
|
|
} |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
/*************************************************************** |
|
|
* decoding stuff starts here * |
|
|
***************************************************************/ |
|
|
|
|
|
|
|
|
// for IVOP addbits == 0 |
|
|
// for PVOP addbits == fcode - 1 |
|
|
// for BVOP addbits == max(fcode,bcode) - 1 |
|
|
// returns true or false |
|
452 |
int |
int |
453 |
check_resync_marker(Bitstream * bs, int addbits) |
check_resync_marker(Bitstream * bs, int addbits) |
454 |
{ |
{ |
625 |
|
|
626 |
} |
} |
627 |
|
|
628 |
|
/***************************************************************************** |
629 |
|
* Local inlined function to "decode" written vlc codes |
630 |
|
****************************************************************************/ |
631 |
|
|
632 |
|
static __inline int |
633 |
|
get_coeff(Bitstream * bs, |
634 |
|
int *run, |
635 |
|
int *last, |
636 |
|
int intra, |
637 |
|
int short_video_header) |
638 |
|
{ |
639 |
|
|
640 |
|
uint32_t mode; |
641 |
|
const VLC *tab; |
642 |
|
int32_t level; |
643 |
|
|
644 |
|
if (short_video_header) // inter-VLCs will be used for both intra and inter blocks |
645 |
|
intra = 0; |
646 |
|
|
647 |
|
tab = &DCT3D[intra][BitstreamShowBits(bs, 12)]; |
648 |
|
|
649 |
|
if (tab->code == -1) |
650 |
|
goto error; |
651 |
|
|
652 |
|
BitstreamSkip(bs, tab->len); |
653 |
|
|
654 |
|
if (tab->code != ESCAPE) { |
655 |
|
if (!intra) { |
656 |
|
*run = (tab->code >> 4) & 255; |
657 |
|
level = tab->code & 15; |
658 |
|
*last = (tab->code >> 12) & 1; |
659 |
|
} else { |
660 |
|
*run = (tab->code >> 8) & 255; |
661 |
|
level = tab->code & 255; |
662 |
|
*last = (tab->code >> 16) & 1; |
663 |
|
} |
664 |
|
return BitstreamGetBit(bs) ? -level : level; |
665 |
|
} |
666 |
|
|
667 |
|
if (short_video_header) { |
668 |
|
// escape mode 4 - H.263 type, only used if short_video_header = 1 |
669 |
|
*last = BitstreamGetBit(bs); |
670 |
|
*run = BitstreamGetBits(bs, 6); |
671 |
|
level = BitstreamGetBits(bs, 8); |
672 |
|
|
673 |
|
if (level == 0 || level == 128) |
674 |
|
DEBUG1("Illegal LEVEL for ESCAPE mode 4:", level); |
675 |
|
|
676 |
|
return (level >= 128 ? -(256 - level) : level); |
677 |
|
} |
678 |
|
|
679 |
|
mode = BitstreamShowBits(bs, 2); |
680 |
|
|
681 |
|
if (mode < 3) { |
682 |
|
BitstreamSkip(bs, (mode == 2) ? 2 : 1); |
683 |
|
|
684 |
|
tab = &DCT3D[intra][BitstreamShowBits(bs, 12)]; |
685 |
|
if (tab->code == -1) |
686 |
|
goto error; |
687 |
|
|
688 |
|
BitstreamSkip(bs, tab->len); |
689 |
|
|
690 |
|
if (!intra) { |
691 |
|
*run = (tab->code >> 4) & 255; |
692 |
|
level = tab->code & 15; |
693 |
|
*last = (tab->code >> 12) & 1; |
694 |
|
} else { |
695 |
|
*run = (tab->code >> 8) & 255; |
696 |
|
level = tab->code & 255; |
697 |
|
*last = (tab->code >> 16) & 1; |
698 |
|
} |
699 |
|
|
700 |
|
if (mode < 2) // first escape mode, level is offset |
701 |
|
level += max_level[*last + (!intra << 1)][*run]; // need to add back the max level |
702 |
|
else if (mode == 2) // second escape mode, run is offset |
703 |
|
*run += max_run[*last + (!intra << 1)][level] + 1; |
704 |
|
|
705 |
|
return BitstreamGetBit(bs) ? -level : level; |
706 |
|
} |
707 |
|
// third escape mode - fixed length codes |
708 |
|
BitstreamSkip(bs, 2); |
709 |
|
*last = BitstreamGetBits(bs, 1); |
710 |
|
*run = BitstreamGetBits(bs, 6); |
711 |
|
BitstreamSkip(bs, 1); // marker |
712 |
|
level = BitstreamGetBits(bs, 12); |
713 |
|
BitstreamSkip(bs, 1); // marker |
714 |
|
|
715 |
|
return (level & 0x800) ? (level | (-1 ^ 0xfff)) : level; |
716 |
|
|
717 |
|
error: |
718 |
|
*run = VLC_ERROR; |
719 |
|
return 0; |
720 |
|
|
721 |
|
} |
722 |
|
|
723 |
|
/***************************************************************************** |
724 |
|
* MB reading functions |
725 |
|
****************************************************************************/ |
726 |
|
|
727 |
void |
void |
728 |
get_intra_block(Bitstream * bs, |
get_intra_block(Bitstream * bs, |
729 |
int16_t * block, |
int16_t * block, |