--- trunk/xvidcore/src/bitstream/mbcoding.c 2002/04/04 13:58:18 100 +++ trunk/xvidcore/src/bitstream/mbcoding.c 2002/04/13 16:30:02 118 @@ -1,3 +1,53 @@ + /****************************************************************************** + * * + * This file is part of XviD, a free MPEG-4 video encoder/decoder * + * * + * XviD is an implementation of a part of one or more MPEG-4 Video tools * + * as specified in ISO/IEC 14496-2 standard. Those intending to use this * + * software module in hardware or software products are advised that its * + * use may infringe existing patents or copyrights, and any such use * + * would be at such party's own risk. The original developer of this * + * software module and his/her company, and subsequent editors and their * + * companies, will have no liability for use of this software or * + * modifications or derivatives thereof. * + * * + * XviD is free software; you can redistribute it and/or modify it * + * under the terms of the GNU General Public License as published by * + * the Free Software Foundation; either version 2 of the License, or * + * (at your option) any later version. * + * * + * XviD is distributed in the hope that it will be useful, but * + * WITHOUT ANY WARRANTY; without even the implied warranty of * + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * + * GNU General Public License for more details. * + * * + * You should have received a copy of the GNU General Public License * + * along with this program; if not, write to the Free Software * + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA * + * * + ******************************************************************************/ + + /****************************************************************************** + * * + * bitstream.c * + * * + * Copyright (C) 2001 - Peter Ross * + * * + * For more information visit the XviD homepage: http://www.xvid.org * + * * + ******************************************************************************/ + + /****************************************************************************** + * * + * Revision history: * + * * + * 14.04.2002 bframe encoding + * 08.03.2002 initial version; isibaar * + * * + ******************************************************************************/ + + + #include #include "../portab.h" #include "bitstream.h" @@ -7,18 +57,16 @@ #include "../utils/mbfunctions.h" -#define ESCAPE 7167 #define ABS(X) (((X)>0)?(X):-(X)) #define CLIP(X,A) (X > A) ? (A) : (X) -VLC DCT3Dintra[4096]; -VLC DCT3Dinter[4096]; -static VLC *DCT3D[2]; - VLC intra_table[65536]; VLC inter_table[65536]; -static short clip_table[4096]; +VLC DCT3Dintra[4096]; +VLC DCT3Dinter[4096]; + +static int16_t clip_table[4096]; void init_vlc_tables(void) { @@ -209,15 +257,15 @@ VLC *vlc; j = intra; - last = 1 + intra; + last = intra; - while((v = qcoeff[zigzag[j++]]) == 0); + while((v = qcoeff[zigzag[j]]) == 0) j++; do { // count zeroes vlc = table + 64*255 + (clip_table[2048+v] << 6) + j - last; - last = j + 1; - while(j < 64 && (v = qcoeff[zigzag[j++]]) == 0); + last = ++j; + while(j < 64 && (v = qcoeff[zigzag[j]]) == 0) j++; // write code if(j != 64) { @@ -393,6 +441,144 @@ } +/*************************************************************** + * 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! + + } + +} + +/* + dbquant + -2 10b + 0 0b + +2 11b +*/ + +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 int16_t fcode, + const int16_t bcode, + Bitstream * bs, + Statistics * pStat) +{ + int i; + +/* ------------------------------------------------------------------ + when a block is skipped it is decoded DIRECT(0,) + hence are interpolated from forward & backward frames + ------------------------------------------------------------------ */ + + if (mb->mode == 5) + { + 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) + { + // TODO: direct + } + + 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 * @@ -546,100 +732,6 @@ } -int get_coeff(Bitstream * bs, int *run, int *last, int intra, int short_video_header) -{ - - uint32_t mode; - const VLC *tab; - int32_t level; - - if(short_video_header) // inter-VLCs will be used for both intra and inter blocks - intra = 0; - - tab = &DCT3D[intra][BitstreamShowBits(bs, 12)]; - - if(tab->code == -1) - goto error; - - BitstreamSkip(bs, tab->len); - - if(tab->code != ESCAPE) { - if(!intra) - { - *run = (tab->code >> 4) & 255; - level = tab->code & 15; - *last = (tab->code >> 12) & 1; - } - else - { - *run = (tab->code >> 8) & 255; - level = tab->code & 255; - *last = (tab->code >> 16) & 1; - } - return BitstreamGetBit(bs) ? -level : level; - } - - if(short_video_header) - { - // escape mode 4 - H.263 type, only used if short_video_header = 1 - *last = BitstreamGetBit(bs); - *run = BitstreamGetBits(bs, 6); - level = BitstreamGetBits(bs, 8); - - if (level == 0 || level == 128) - DEBUG1("Illegal LEVEL for ESCAPE mode 4:", level); - - return (level >= 128 ? -(256 - level) : level); - } - - mode = BitstreamShowBits(bs, 2); - - if(mode < 3) { - BitstreamSkip(bs, (mode == 2) ? 2 : 1); - - tab = &DCT3D[intra][BitstreamShowBits(bs, 12)]; - if (tab->code == -1) - goto error; - - BitstreamSkip(bs, tab->len); - - if (!intra) { - *run = (tab->code >> 4) & 255; - level = tab->code & 15; - *last = (tab->code >> 12) & 1; - } - else - { - *run = (tab->code >> 8) & 255; - level = tab->code & 255; - *last = (tab->code >> 16) & 1; - } - - if(mode < 2) // first escape mode, level is offset - level += max_level[*last + (!intra<<1)][*run]; // need to add back the max level - else if(mode == 2) // second escape mode, run is offset - *run += max_run[*last + (!intra<<1)][level] + 1; - - return BitstreamGetBit(bs) ? -level : level; - } - - // third escape mode - fixed length codes - BitstreamSkip(bs, 2); - *last = BitstreamGetBits(bs, 1); - *run = BitstreamGetBits(bs, 6); - BitstreamSkip(bs, 1); // marker - level = BitstreamGetBits(bs, 12); - BitstreamSkip(bs, 1); // marker - - return (level & 0x800) ? (level | (-1 ^ 0xfff)) : level; - - error: - *run = VLC_ERROR; - return 0; - -} - - void get_intra_block(Bitstream * bs, int16_t * block, int direction, int coeff) {