--- mbcoding.c 2003/01/11 14:59:24 1.25.2.10 +++ mbcoding.c 2002/09/10 21:29:03 1.29 @@ -1,58 +1,40 @@ - /****************************************************************************** - * * - * 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 * - * * - ******************************************************************************/ - - /****************************************************************************** - * * - * mbcoding.c * - * * - * Copyright (C) 2002 - Michael Militzer * - * * - * For more information visit the XviD homepage: http://www.xvid.org * - * * - ******************************************************************************/ - - /****************************************************************************** - * * - * Revision history: * - * * - * 04.01.2003 GMC support - gruel * - * 28.06.2002 added check_resync_marker() * - * 14.04.2002 bframe encoding * - * 08.03.2002 initial version; isibaar * - * * - ******************************************************************************/ +/***************************************************************************** + * + * XVID MPEG-4 VIDEO CODEC + * - Macro Block coding functions - + * + * Copyright(C) 2002 Michael Militzer + * + * + * This program 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. + * + * This program 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. + * + * This program 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 + * + * $Id: mbcoding.c,v 1.29 2002/09/10 21:29:03 edgomez Exp $ + * + ****************************************************************************/ - -#include #include #include "../portab.h" -#include "../global.h" #include "bitstream.h" #include "zigzag.h" #include "vlc_codes.h" @@ -60,40 +42,22 @@ #include "../utils/mbfunctions.h" -VLC intra_table[4*2048*64]; -VLC inter_table[4*2048*64]; - -VLC DCT3Dintra[4096]; -VLC DCT3Dinter[4096]; - -/* not really MB related, but VLCs are only available here */ -void bs_put_spritetrajectory(Bitstream * bs, const int val) -{ - const int code = sprite_trajectory_code[val+16384].code; - const int len = sprite_trajectory_code[val+16384].len; - const int code2 = sprite_trajectory_len[len].code; - const int len2 = sprite_trajectory_len[len].len; - -// printf("GMC=%d Code/Len = %d / %d ",val, code,len); -// printf("Code2 / Len2 = %d / %d \n",code2,len2); +#define ABS(X) (((X)>0)?(X):-(X)) +#define CLIP(X,A) (X > A) ? (A) : (X) - BitstreamPutBits(bs, code2, len2); - if (len) BitstreamPutBits(bs, code, len); -} - -int bs_get_spritetrajectory(Bitstream * bs) -{ - int i; - for (i = 0; i < 12; i++) - { - if (BitstreamShowBits(bs, sprite_trajectory_len[i].len) == sprite_trajectory_len[i].code) - { - BitstreamSkip(bs, sprite_trajectory_len[i].len); - return i; - } - } - return -1; -} +/***************************************************************************** + * Local data + ****************************************************************************/ + +static VLC intra_table[524032]; +static VLC inter_table[524032]; + +static VLC DCT3Dintra[4096]; +static VLC DCT3Dinter[4096]; + +/***************************************************************************** + * Vector Length Coding Initialization + ****************************************************************************/ void init_vlc_tables(void) @@ -101,7 +65,7 @@ int32_t k, l, i, intra, last; VLC *vlc[2]; - VLC **coeff_ptr; + VLC const **coeff_ptr; VLC *vlc1, *vlc2; vlc1 = DCT3Dintra; @@ -110,8 +74,11 @@ vlc[0] = intra_table; vlc[1] = inter_table; - // generate encoding vlc lookup tables - // the lookup table idea is taken from the excellent fame project by Vivien Chapellier + /* + * Generate encoding vlc lookup tables + * the lookup table idea is taken from the excellent fame project + * by Vivien Chapellier + */ for (i = 0; i < 4; i++) { intra = i % 2; last = i / 2; @@ -119,8 +86,8 @@ coeff_ptr = coeff_vlc[last + 2 * intra]; for (k = -2047; k < 2048; k++) { // level - int8_t *max_level_ptr = max_level[last + 2 * intra]; - int8_t *max_run_ptr = max_run[last + 2 * intra]; + int8_t const *max_level_ptr = max_level[last + 2 * intra]; + int8_t const *max_run_ptr = max_run[last + 2 * intra]; for (l = 0; l < 64; l++) { // run int32_t level = k; @@ -211,30 +178,12 @@ DCT3D[0] = DCT3Dinter; DCT3D[1] = DCT3Dintra; - -/* init sprite_trajectory tables */ -/* even if GMC is not specified (it might be used later...) */ - - sprite_trajectory_code[0+16384].code = 0; - sprite_trajectory_code[0+16384].len = 0; - for (k=0;k<14;k++) - { - int limit = (1<cbp & (1 << (5 - i))) { - const uint16_t *scan_table = - frame->global_flags & XVID_ALTERNATESCAN ? - scan_tables[2] : scan_tables[pMB->acpred_directions[i]]; - bits = BitstreamPos(bs); - CodeCoeff(bs, &qcoeff[i * 64], intra_table, scan_table, 1); + CodeCoeff(bs, &qcoeff[i * 64], intra_table, + scan_tables[pMB->acpred_directions[i]], 1); bits = BitstreamPos(bs) - bits; pStat->iTextBits += bits; @@ -397,7 +345,7 @@ static void -CodeBlockInter(const FRAMEINFO * const frame, +CodeBlockInter(const FRAMEINFO * frame, const MACROBLOCK * pMB, int16_t qcoeff[6 * 64], Bitstream * bs, @@ -414,12 +362,6 @@ BitstreamPutBits(bs, mcbpc_inter_tab[mcbpc].code, mcbpc_inter_tab[mcbpc].len); - if ( (frame->coding_type == S_VOP) && (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q) ) - { - /* decision on GMC is done in encoder.c now */ - BitstreamPutBit(bs, pMB->mcsel); // mcsel: '0'=local motion, '1'=GMC - } - // write cbpy BitstreamPutBits(bs, cbpy_tab[cbpy].code, cbpy_tab[cbpy].len); @@ -431,13 +373,13 @@ if (frame->global_flags & XVID_INTERLACING) { if (pMB->cbp) { BitstreamPutBit(bs, pMB->field_dct); - DPRINTF(DPRINTF_MB,"codep: field_dct: %i", pMB->field_dct); + DEBUG1("codep: field_dct: ", pMB->field_dct); } // if inter block, write field ME flag if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q) { BitstreamPutBit(bs, pMB->field_pred); - DPRINTF(DPRINTF_MB,"codep: field_pred: %i", pMB->field_pred); + DEBUG1("codep: field_pred: ", pMB->field_pred); // write field prediction references if (pMB->field_pred) { @@ -446,41 +388,40 @@ } } } - // code motion vector(s) if motion is local - if (!pMB->mcsel) - for (i = 0; i < (pMB->mode == MODE_INTER4V ? 4 : 1); i++) { - CodeVector(bs, pMB->pmvs[i].x, frame->fcode, pStat); - CodeVector(bs, pMB->pmvs[i].y, frame->fcode, pStat); - } + // code motion vector(s) + for (i = 0; i < (pMB->mode == MODE_INTER4V ? 4 : 1); i++) { + CodeVector(bs, pMB->pmvs[i].x, frame->fcode, pStat); + CodeVector(bs, pMB->pmvs[i].y, frame->fcode, pStat); + } bits = BitstreamPos(bs); // code block coeffs for (i = 0; i < 6; i++) if (pMB->cbp & (1 << (5 - i))) - { - const uint16_t *scan_table = - frame->global_flags & XVID_ALTERNATESCAN ? - scan_tables[2] : scan_tables[0]; - - CodeCoeff(bs, &qcoeff[i * 64], inter_table, scan_table, 0); - } + CodeCoeff(bs, &qcoeff[i * 64], inter_table, scan_tables[0], 0); bits = BitstreamPos(bs) - bits; pStat->iTextBits += bits; + } +/***************************************************************************** + * Macro Block bitstream encoding functions + ****************************************************************************/ void -MBCoding(const FRAMEINFO * const frame, +MBCoding(const FRAMEINFO * frame, MACROBLOCK * pMB, int16_t qcoeff[6 * 64], Bitstream * bs, Statistics * pStat) { - if (frame->coding_type != I_VOP) - BitstreamPutBit(bs, 0); // not_coded - + + if (frame->coding_type == P_VOP) { + BitstreamPutBit(bs, 0); // coded + } + if (pMB->mode == MODE_INTRA || pMB->mode == MODE_INTRA_Q) CodeBlockIntra(frame, pMB, qcoeff, bs, pStat); else @@ -488,15 +429,15 @@ } -/* -// moved to mbcoding.h so that in can be 'static __inline' + void MBSkip(Bitstream * bs) { BitstreamPutBit(bs, 1); // not coded + return; } -*/ +#if 0 /*************************************************************** * bframe encoding start ***************************************************************/ @@ -509,22 +450,37 @@ 3 0001b forward mc+q dbquant, mvdf */ -static __inline void +void put_bvop_mbtype(Bitstream * bs, int value) { switch (value) { - case MODE_FORWARD: - BitstreamPutBit(bs, 0); - case MODE_BACKWARD: - BitstreamPutBit(bs, 0); - case MODE_INTERPOLATE: - BitstreamPutBit(bs, 0); - case MODE_DIRECT: - BitstreamPutBit(bs, 1); - default: - break; + 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! + } + } /* @@ -534,7 +490,7 @@ +2 11b */ -static __inline void +void put_bvop_dbquant(Bitstream * bs, int value) { @@ -557,19 +513,15 @@ } } - - void MBCodingBVOP(const MACROBLOCK * mb, const int16_t qcoeff[6 * 64], const int32_t fcode, const int32_t bcode, Bitstream * bs, - Statistics * pStat, - int direction) + Statistics * pStat) { - int vcode = fcode; - unsigned int i; + int i; /* ------------------------------------------------------------------ when a block is skipped it is decoded DIRECT(0,0) @@ -599,40 +551,41 @@ put_bvop_dbquant(bs, 0); // todo: mb->dquant = 0 } - switch (mb->mode) { - case MODE_INTERPOLATE: - CodeVector(bs, mb->pmvs[1].x, vcode, pStat); //forward vector of interpolate mode - CodeVector(bs, mb->pmvs[1].y, vcode, pStat); - case MODE_BACKWARD: - vcode = bcode; - case MODE_FORWARD: - CodeVector(bs, mb->pmvs[0].x, vcode, pStat); - CodeVector(bs, mb->pmvs[0].y, vcode, pStat); - break; - case MODE_DIRECT: - CodeVector(bs, mb->pmvs[3].x, 1, pStat); // fcode is always 1 for delta vector - CodeVector(bs, mb->pmvs[3].y, 1, pStat); // prediction is always (0,0) - default: break; + 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[direction], 0); + CodeCoeff(bs, &qcoeff[i * 64], inter_table, scan_tables[0], 0); } } } +#endif +/***************************************************************************** + * decoding stuff starts here + ****************************************************************************/ -/*************************************************************** - * decoding stuff starts here * - ***************************************************************/ - +/* + * For IVOP addbits == 0 + * For PVOP addbits == fcode - 1 + * For BVOP addbits == max(fcode,bcode) - 1 + * returns true or false + */ -// for IVOP addbits == 0 -// for PVOP addbits == fcode - 1 -// for BVOP addbits == max(fcode,bcode) - 1 -// returns true or false int check_resync_marker(Bitstream * bs, int addbits) { @@ -674,7 +627,7 @@ uint32_t index; - index = MIN(BitstreamShowBits(bs, 9), 256); + index = CLIP(BitstreamShowBits(bs, 9), 256); BitstreamSkip(bs, mcbpc_inter_table[index].len); @@ -700,7 +653,7 @@ } -static __inline int +int get_mv_data(Bitstream * bs) { @@ -809,6 +762,105 @@ } +/***************************************************************************** + * Local inlined function to "decode" written vlc codes + ****************************************************************************/ + +static __inline 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; + +} + +/***************************************************************************** + * MB reading functions + ****************************************************************************/ + void get_intra_block(Bitstream * bs, int16_t * block, @@ -817,12 +869,14 @@ { const uint16_t *scan = scan_tables[direction]; - int level, run, last; + int level; + int run; + int last; do { level = get_coeff(bs, &run, &last, 1, 0); if (run == -1) { - DPRINTF(DPRINTF_ERROR,"fatal: invalid run"); + DEBUG("fatal: invalid run"); break; } coeff += run; @@ -831,8 +885,8 @@ DPRINTF(DPRINTF_COEFF,"block[%i] %i", scan[coeff], level); //DPRINTF(DPRINTF_COEFF,"block[%i] %i %08x", scan[coeff], level, BitstreamShowBits(bs, 32)); - if (level < -2047 || level > 2047) { - DPRINTF(DPRINTF_ERROR,"warning: intra_overflow %i", level); + if (level < -127 || level > 127) { + DEBUG1("warning: intra_overflow", level); } coeff++; } while (!last); @@ -841,11 +895,10 @@ void get_inter_block(Bitstream * bs, - int16_t * block, - int direction) + int16_t * block) { - const uint16_t *scan = scan_tables[direction]; + const uint16_t *scan = scan_tables[0]; int p; int level; int run; @@ -855,7 +908,7 @@ do { level = get_coeff(bs, &run, &last, 0, 0); if (run == -1) { - DPRINTF(DPRINTF_ERROR,"fatal: invalid run"); + DEBUG("fatal: invalid run"); break; } p += run; @@ -865,8 +918,8 @@ DPRINTF(DPRINTF_COEFF,"block[%i] %i", scan[p], level); // DPRINTF(DPRINTF_COEFF,"block[%i] %i %08x", scan[p], level, BitstreamShowBits(bs, 32)); - if (level < -2047 || level > 2047) { - DPRINTF(DPRINTF_ERROR,"warning: inter overflow %i", level); + if (level < -127 || level > 127) { + DEBUG1("warning: inter_overflow", level); } p++; } while (!last);