--- trunk/xvidcore/src/bitstream/mbcoding.c 2002/04/04 13:58:18 100 +++ trunk/xvidcore/src/bitstream/mbcoding.c 2002/04/25 06:55:00 136 @@ -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) { @@ -59,7 +107,7 @@ uint32_t run = l; if((abs(level) <= max_level_ptr[run]) && - (run <= max_run_ptr[abs(level)])) { // level < max_level and run < max_run + (run <= (uint32_t)max_run_ptr[abs(level)])) { // level < max_level and run < max_run vlc[intra]->code = 0; vlc[intra]->len = 0; @@ -72,7 +120,7 @@ level += max_level_ptr[run]; if((abs(level) <= max_level_ptr[run]) && - (run <= max_run_ptr[abs(level)])) { + (run <= (uint32_t) max_run_ptr[abs(level)])) { vlc[intra]->code = 0x06; vlc[intra]->len = 8; @@ -87,7 +135,7 @@ run -= max_run_ptr[abs(level)] + 1; // and change run if((abs(level) <= max_level_ptr[run]) && - (run <= max_run_ptr[abs(level)])) { + (run <= (uint32_t) max_run_ptr[abs(level)])) { vlc[intra]->code = 0x0e; vlc[intra]->len = 9; @@ -145,8 +193,8 @@ } static __inline void CodeVector(Bitstream *bs, - int16_t value, - int16_t f_code, + int32_t value, + int32_t f_code, Statistics *pStat) { @@ -198,7 +246,7 @@ static __inline void CodeCoeff(Bitstream *bs, - int16_t qcoeff[64], + const int16_t qcoeff[64], VLC *table, const uint16_t *zigzag, uint16_t intra) @@ -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) { @@ -232,7 +280,7 @@ } -static void CodeBlockIntra(const MBParam * pParam, +static void CodeBlockIntra(const FRAMEINFO * frame, const MACROBLOCK *pMB, int16_t qcoeff[6*64], Bitstream * bs, @@ -244,7 +292,7 @@ cbpy = pMB->cbp >> 2; // write mcbpc - if(pParam->coding_type == I_VOP) { + if(frame->coding_type == I_VOP) { mcbpc = ((pMB->mode >> 1) & 3) | ((pMB->cbp & 3) << 2); BitstreamPutBits(bs, mcbpc_intra_tab[mcbpc].code, mcbpc_intra_tab[mcbpc].len); } @@ -267,7 +315,7 @@ BitstreamPutBits(bs, pMB->dquant, 2); // write interlacing - if (pParam->global_flags & XVID_INTERLACING) + if (frame->global_flags & XVID_INTERLACING) { BitstreamPutBit(bs, pMB->field_dct); } @@ -302,7 +350,7 @@ } -static void CodeBlockInter(const MBParam * pParam, +static void CodeBlockInter(const FRAMEINFO * frame, const MACROBLOCK *pMB, int16_t qcoeff[6*64], Bitstream * bs, @@ -326,7 +374,7 @@ BitstreamPutBits(bs, pMB->dquant, 2); // interlacing - if (pParam->global_flags & XVID_INTERLACING) + if (frame->global_flags & XVID_INTERLACING) { BitstreamPutBit(bs, pMB->field_dct); DEBUG1("codep: field_dct: ", pMB->field_dct); @@ -349,8 +397,8 @@ // code motion vector(s) for(i = 0; i < (pMB->mode == MODE_INTER4V ? 4 : 1); i++) { - CodeVector(bs, pMB->pmvs[i].x, pParam->fixed_code, pStat); - CodeVector(bs, pMB->pmvs[i].y, pParam->fixed_code, pStat); + CodeVector(bs, pMB->pmvs[i].x, frame->fcode, pStat); + CodeVector(bs, pMB->pmvs[i].y, frame->fcode, pStat); } bits = BitstreamPos(bs); @@ -366,7 +414,7 @@ } -void MBCoding(const MBParam * pParam, +void MBCoding(const FRAMEINFO * frame, MACROBLOCK *pMB, int16_t qcoeff[6*64], Bitstream * bs, @@ -375,7 +423,7 @@ int intra = (pMB->mode == MODE_INTRA || pMB->mode == MODE_INTRA_Q); - if(pParam->coding_type == P_VOP) { + if(frame->coding_type == P_VOP) { if(pMB->cbp == 0 && pMB->mode == MODE_INTER && pMB->mvs[0].x == 0 && pMB->mvs[0].y == 0) { @@ -387,13 +435,151 @@ } if(intra) - CodeBlockIntra(pParam, pMB, qcoeff, bs, pStat); + CodeBlockIntra(frame, pMB, qcoeff, bs, pStat); + else + CodeBlockInter(frame, pMB, qcoeff, bs, pStat); + +} + +/*************************************************************** + * 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 - CodeBlockInter(pParam, pMB, qcoeff, bs, pStat); + { + 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) {