--- mbfunctions.h 2002/03/08 02:44:58 1.1 +++ mbfunctions.h 2002/11/17 00:51:10 1.15 @@ -1,84 +1,224 @@ -/************************************************************************** +/***************************************************************************** * - * Modifications: + * XVID MPEG-4 VIDEO CODEC + * - Motion estimation fuctions header file - * - * 02.12.2001 motion estimation/compensation split - * 16.11.2001 const/uint32_t changes to MBMotionEstComp() - * 26.08.2001 added inter4v_mode parameter to MBMotionEstComp() + * Copyright(C) 2002 Michael Militzer * - * Michael Militzer + * This file is part of XviD, a free MPEG-4 video encoder/decoder * - **************************************************************************/ + * 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. + * + * 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 + * + * Under section 8 of the GNU General Public License, the copyright + * holders of XVID explicitly forbid distribution in the following + * countries: + * + * - Japan + * - United States of America + * + * Linking XviD statically or dynamically with other modules is making a + * combined work based on XviD. Thus, the terms and conditions of the + * GNU General Public License cover the whole combination. + * + * As a special exception, the copyright holders of XviD give you + * permission to link XviD with independent modules that communicate with + * XviD solely through the VFW1.1 and DShow interfaces, regardless of the + * license terms of these independent modules, and to copy and distribute + * the resulting combined work under terms of your choice, provided that + * every copy of the combined work is accompanied by a complete copy of + * the source code of XviD (the version of XviD used to produce the + * combined work), being distributed under the terms of the GNU General + * Public License plus this exception. An independent module is a module + * which is not derived from or based on XviD. + * + * Note that people who make modified versions of XviD are not obligated + * to grant this special exception for their modified versions; it is + * their choice whether to do so. The GNU General Public License gives + * permission to release a modified version without this exception; this + * exception also makes it possible to release a modified version which + * carries forward this exception. + * + * $Id: mbfunctions.h,v 1.15 2002/11/17 00:51:10 edgomez Exp $ + * + ****************************************************************************/ -#ifndef _ENCORE_BLOCK_H -#define _ENCORE_BLOCK_H +#ifndef _MBFUNCTIONS_H +#define _MBFUNCTIONS_H #include "../encoder.h" #include "../bitstream/bitstream.h" - - -/** MotionEstimation **/ - -bool MotionEstimation( - MACROBLOCK * const pMBs, - MBParam * const pParam, - const IMAGE * const pRef, - const IMAGE * const pRefH, - const IMAGE * const pRefV, - const IMAGE * const pRefHV, - IMAGE * const pCurrent, - const uint32_t iLimit); - - -/** MBMotionCompensation **/ -void MBMotionCompensation( - MACROBLOCK * const pMB, - const uint32_t j, - const uint32_t i, - const IMAGE * const pRef, - const IMAGE * const pRefH, - const IMAGE * const pRefV, - const IMAGE * const pRefHV, - IMAGE * const pCurrent, - int16_t dct_codes[][64], - const uint32_t width, - const uint32_t height, - const uint32_t edged_width, - const uint32_t rounding); - - -/** MBTransQuant.c **/ - - -void MBTransQuantIntra(const MBParam *pParam, - const uint32_t x_pos, /* <-- The x position of the MB to be searched */ - const uint32_t y_pos, /* <-- The y position of the MB to be searched */ - int16_t data[][64], /* <-> the data of the MB to be coded */ - int16_t qcoeff[][64], /* <-> the quantized DCT coefficients */ - IMAGE * const pCurrent /* <-> the reconstructed image ( function will update one - MB in it with data from data[] ) */ -); - - -uint8_t MBTransQuantInter(const MBParam *pParam, /* <-- the parameter for DCT transformation - and Quantization */ - const uint32_t x_pos, /* <-- The x position of the MB to be searched */ - const uint32_t y_pos, /* <-- The y position of the MB to be searched */ - int16_t data[][64], /* <-> the data of the MB to be coded */ - int16_t qcoeff[][64], /* <-> the quantized DCT coefficients */ - IMAGE * const pCurrent /* <-> the reconstructed image ( function will - update one MB in it with data from data[] ) */ -); - - -/** MBCoding.c **/ - -void MBCoding(const MBParam *pParam, /* <-- the parameter for coding of the bitstream */ - MACROBLOCK *pMB, /* <-- Info of the MB to be coded */ - int16_t qcoeff[][64], /* <-- the quantized DCT coefficients */ - Bitstream * bs, /* <-> the bitstream */ - Statistics * pStat /* <-> statistical data collected for current frame */ - ); +/***************************************************************************** + * Prototypes + ****************************************************************************/ + + +/* MotionEstimation */ + +bool MotionEstimation(MBParam * const pParam, + FRAMEINFO * const current, + FRAMEINFO * const reference, + const IMAGE * const pRefH, + const IMAGE * const pRefV, + const IMAGE * const pRefHV, + const uint32_t iLimit); + + +bool SMP_MotionEstimation(MBParam * const pParam, + FRAMEINFO * const current, + FRAMEINFO * const reference, + const IMAGE * const pRefH, + const IMAGE * const pRefV, + const IMAGE * const pRefHV, + const uint32_t iLimit); + + + +/* MBMotionCompensation */ + +void MBMotionCompensation(MACROBLOCK * const pMB, + const uint32_t j, + const uint32_t i, + const IMAGE * const pRef, + const IMAGE * const pRefH, + const IMAGE * const pRefV, + const IMAGE * const pRefHV, + IMAGE * const pCurrent, + int16_t dct_codes[6 * 64], + const uint32_t width, + const uint32_t height, + const uint32_t edged_width, + const uint32_t rounding); + + +/* MBTransQuant.c */ + +void MBTransQuantIntra(const MBParam * pParam, + FRAMEINFO * frame, + MACROBLOCK * pMB, + const uint32_t x_pos, /* <-- The x position of the MB to be searched */ + const uint32_t y_pos, /* <-- The y position of the MB to be searched */ + int16_t data[6 * 64], /* <-> the data of the MB to be coded */ + int16_t qcoeff[6 * 64]); /* <-> the quantized DCT coefficients */ + + + +void MBTransQuantIntra2(const MBParam * pParam, + FRAMEINFO * frame, + MACROBLOCK * pMB, + const uint32_t x_pos, /* <-- The x position of the MB to be searched */ + const uint32_t y_pos, /* <-- The y position of the MB to be searched */ + int16_t data[6 * 64], /* <-> the data of the MB to be coded */ + int16_t qcoeff[6 * 64]); /* <-> the quantized DCT coefficients */ + + + +uint8_t MBTransQuantInter(const MBParam * pParam, + FRAMEINFO * frame, + MACROBLOCK * pMB, + const uint32_t x_pos, + const uint32_t y_pos, + int16_t data[6 * 64], + int16_t qcoeff[6 * 64]); + + +uint8_t MBTransQuantInter2(const MBParam * pParam, + FRAMEINFO * frame, + MACROBLOCK * pMB, + const uint32_t x_pos, + const uint32_t y_pos, + int16_t data[6 * 64], + int16_t qcoeff[6 * 64]); + +uint8_t MBTransQuantInterBVOP(const MBParam * pParam, + FRAMEINFO * frame, + MACROBLOCK * pMB, + int16_t data[6 * 64], + int16_t qcoeff[6 * 64]); + +void MBTrans(const MBParam * pParam, + FRAMEINFO * frame, + MACROBLOCK * pMB, + const uint32_t x_pos, + const uint32_t y_pos, + int16_t data[6 * 64]); + +void MBfDCT(const MBParam * pParam, + FRAMEINFO * frame, + MACROBLOCK * pMB, + int16_t data[6 * 64]); + +uint8_t MBQuantInter(const MBParam * pParam, + const int iQuant, + int16_t data[6 * 64], + int16_t qcoeff[6 * 64]); + +void MBQuantDeQuantIntra(const MBParam * pParam, + FRAMEINFO * frame, + MACROBLOCK *pMB, + int16_t qcoeff[6 * 64], + int16_t data[6*64]); + +void MBQuantIntra(const MBParam * pParam, + FRAMEINFO * frame, + MACROBLOCK *pMB, + int16_t data[6*64], + int16_t qcoeff[6 * 64]); + +void MBDeQuantIntra(const MBParam * pParam, + const int iQuant, + int16_t qcoeff[6 * 64], + int16_t data[6*64]); + +void MBDeQuantInter(const MBParam * pParam, + const int iQuant, + int16_t data[6 * 64], + int16_t qcoeff[6 * 64], + const uint8_t cbp); + + +void MBiDCT(int16_t data[6 * 64], + const uint8_t cbp); + + +void MBTransAdd(const MBParam * pParam, + FRAMEINFO * frame, + MACROBLOCK * pMB, + const uint32_t x_pos, + const uint32_t y_pos, + int16_t data[6 * 64], + const uint8_t cbp); + + + +/* interlacing */ + +uint32_t MBDecideFieldDCT(int16_t data[6 * 64]); /* <- decide whether to use field-based DCT for interlacing */ + +void MBFrameToField(int16_t data[6 * 64]); /* de-interlace vertical Y blocks */ + + +/* MBCoding.c */ + +void MBSkip(Bitstream * bs); /* just the bitstream. Since MB is skipped, no info is needed */ + + +void MBCoding(const FRAMEINFO * frame, /* <-- the parameter for coding of the bitstream */ + MACROBLOCK * pMB, /* <-- Info of the MB to be coded */ + int16_t qcoeff[6 * 64], /* <-- the quantized DCT coefficients */ + Bitstream * bs, /* <-> the bitstream */ + Statistics * pStat); /* <-> statistical data collected for current frame */ #endif