--- trunk/xvidcore/src/bitstream/mbcoding.c 2002/04/11 10:48:03 116 +++ 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" @@ -57,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; @@ -70,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; @@ -85,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; @@ -143,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) { @@ -196,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) @@ -230,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, @@ -242,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); } @@ -265,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); } @@ -300,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, @@ -324,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); @@ -347,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); @@ -364,7 +414,7 @@ } -void MBCoding(const MBParam * pParam, +void MBCoding(const FRAMEINFO * frame, MACROBLOCK *pMB, int16_t qcoeff[6*64], Bitstream * bs, @@ -373,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) { @@ -385,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 * ***************************************************************/