--- motion.h 2002/09/07 09:12:21 1.15 +++ motion.h 2009/02/18 15:10:27 1.25 @@ -1,34 +1,27 @@ /***************************************************************************** * * XVID MPEG-4 VIDEO CODEC - * - Motion Estimation header - + * - Motion module header - * - * Copyright(C) 2002 Christoph Lampert + * Copyright(C) 2002-2003 Radoslaw Czyz + * 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 + * 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 + * 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 + * 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 + * along with this program ; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA * - * $Id: motion.h,v 1.15 2002/09/07 09:12:21 chl Exp $ + * $Id: motion.h,v 1.25 2009/02/18 15:10:27 Isibaar Exp $ * ***************************************************************************/ @@ -38,349 +31,130 @@ #include "../portab.h" #include "../global.h" -/* hard coded motion search parameters for motion_est and smp_motion_est */ - -// very large value -#define MV_MAX_ERROR (4096 * 256) - -// stop search if sdelta < THRESHOLD -#define MV16_THRESHOLD 192 -#define MV8_THRESHOLD 56 - -#define NEIGH_MOVE_THRESH 0 -// how much a block's MV must differ from his neighbour -// to be search for INTER4V. The more, the faster... - -/* sad16(0,0) bias; mpeg4 spec suggests nb/2+1 */ -/* nb = vop pixels * 2^(bpp-8) */ -#define MV16_00_BIAS (128+1) -#define MV8_00_BIAS (0) - -/* INTER bias for INTER/INTRA decision; mpeg4 spec suggests 2*nb */ -#define MV16_INTER_BIAS 512 - -/* Parameters which control inter/inter4v decision */ -#define IMV16X16 5 - -/* vector map (vlc delta size) smoother parameters */ -#define NEIGH_TEND_16X16 2 -#define NEIGH_TEND_8X8 2 - -// fast ((A)/2)*2 -#define EVEN(A) (((A)<0?(A)+1:(A)) & ~1) - -#define MVzero(A) ( ((A).x)==(0) && ((A).y)==(0) ) -#define MVequal(A,B) ( ((A).x)==((B).x) && ((A).y)==((B).y) ) - -/* default methods of Search, will be changed to function variable */ - -#ifndef SEARCH16 -#define SEARCH16 PMVfastSearch16 -//#define SEARCH16 FullSearch16 -//#define SEARCH16 EPZSSearch16 -#endif - -#ifndef SEARCH8 -#define SEARCH8 PMVfastSearch8 -//#define SEARCH8 EPZSSearch8 -#endif - - -/* - * Calculate the min/max range (in halfpixels) - * relative to the _MACROBLOCK_ position - */ - -static void __inline -get_range(int32_t * const min_dx, - int32_t * const max_dx, - int32_t * const min_dy, - int32_t * const max_dy, - const uint32_t x, - const uint32_t y, - const uint32_t block_sz, /* block dimension, 8 or 16 */ - - const uint32_t width, - const uint32_t height, - const uint32_t fcode) -{ - - const int search_range = 32 << (fcode - 1); - const int high = search_range - 1; - const int low = -search_range; - - /* convert full-pixel measurements to half pixel */ - const int hp_width = 2 * width; - const int hp_height = 2 * height; - const int hp_edge = 2 * block_sz; - - /* we need _right end_ of block, not x-coordinate */ - const int hp_x = 2 * (x) * block_sz; - - /* same for _bottom end_ */ - const int hp_y = 2 * (y) * block_sz; - - *max_dx = MIN(high, hp_width - hp_x); - *max_dy = MIN(high, hp_height - hp_y); - *min_dx = MAX(low, -(hp_edge + hp_x)); - *min_dy = MAX(low, -(hp_edge + hp_y)); - -} - - -/* - * getref: calculate reference image pointer - * the decision to use interpolation h/v/hv or the normal image is - * based on dx & dy. - */ - -static __inline const uint8_t * -get_ref(const uint8_t * const refn, - const uint8_t * const refh, - const uint8_t * const refv, - const uint8_t * const refhv, - const uint32_t x, - const uint32_t y, - const uint32_t block, /* block dimension, 8 or 16 */ - - const int32_t dx, - const int32_t dy, - const uint32_t stride) -{ - - - switch (((dx & 1) << 1) + (dy & 1)) { /* ((dx%2)?2:0)+((dy%2)?1:0) */ - case 0: - return refn + (int) ((x * block + dx / 2) + (y * block + dy / 2) * stride); - case 1: - return refv + (int) ((x * block + dx / 2) + (y * block + - (dy - 1) / 2) * stride); - case 2: - return refh + (int) ((x * block + (dx - 1) / 2) + (y * block + - dy / 2) * stride); - default: - case 3: - return refhv + (int) ((x * block + (dx - 1) / 2) + (y * block + - (dy - 1) / 2) * stride); - } - -} - - -/* This is somehow a copy of get_ref, but with MV instead of X,Y */ - -static __inline const uint8_t * -get_ref_mv(const uint8_t * const refn, - const uint8_t * const refh, - const uint8_t * const refv, - const uint8_t * const refhv, - const uint32_t x, - const uint32_t y, - const uint32_t block, /* block dimension, 8 or 16 */ - - const VECTOR * mv, /* measured in half-pel! */ - - const uint32_t stride) -{ - - switch ((((mv->x) & 1) << 1) + ((mv->y) & 1)) { - case 0: - return refn + (int) ((x * block + (mv->x) / 2) + (y * block + - (mv->y) / 2) * stride); - case 1: - return refv + (int) ((x * block + (mv->x) / 2) + (y * block + - ((mv->y) - 1) / 2) * stride); - case 2: - return refh + (int) ((x * block + ((mv->x) - 1) / 2) + (y * block + - (mv->y) / 2) * stride); - default: - case 3: - return refhv + (int) ((x * block + ((mv->x) - 1) / 2) + (y * block + - ((mv->y) - - 1) / 2) * stride); - } - -} - - -static __inline const uint8_t * -get_iref(const uint8_t * const ref, - const uint32_t x, - const uint32_t y, - const uint32_t block, /* block dimension, 8 or 16 */ - - const int32_t dx, - const int32_t dy, - const uint32_t stride) -{ - return ref + (int) ((x * block + dx / 2) + (y * block + dy / 2) * stride); -} - -static __inline const uint8_t * -get_iref_mv(const uint8_t * const ref, - const uint32_t x, - const uint32_t y, - const uint32_t block, /* block dimension, 8 or 16 */ - - const VECTOR * mv, /* as usual measured in half-pel */ - - const uint32_t stride) -{ - return ref + (int) ((x * block + (mv->x) / 2) + (y * block + (mv->y) / 2) * stride); -} - - -/* prototypes for MainSearch functions, i.e. Diamondsearch, FullSearch or whatever */ - -typedef int32_t(MainSearch16Func) (const uint8_t * const pRef, - const uint8_t * const pRefH, - const uint8_t * const pRefV, - const uint8_t * const pRefHV, - const uint8_t * const cur, - const int x, - const int y, - const int start_x, - const int start_y, - int iMinSAD, - VECTOR * const currMV, - const int center_x, - const int center_y, - const int32_t min_dx, - const int32_t max_dx, - const int32_t min_dy, - const int32_t max_dy, - const int32_t iEdgedWidth, - const int32_t iDiamondSize, - const int32_t iFcode, - const int32_t iQuant, - int iFound); - -typedef MainSearch16Func *MainSearch16FuncPtr; - - -typedef int32_t(MainSearch8Func) (const uint8_t * const pRef, - const uint8_t * const pRefH, - const uint8_t * const pRefV, - const uint8_t * const pRefHV, - const uint8_t * const cur, - const int x, - const int y, - const int start_x, - const int start_y, - int iMinSAD, - VECTOR * const currMV, - const int center_x, - const int center_y, - const int32_t min_dx, - const int32_t max_dx, - const int32_t min_dy, - const int32_t max_dy, - const int32_t iEdgedWidth, - const int32_t iDiamondSize, - const int32_t iFcode, - const int32_t iQuant, - int iFound); - -typedef MainSearch8Func *MainSearch8FuncPtr; - - -/* prototypes for MotionEstimation functions, i.e. PMVfast, EPZS or whatever */ - -typedef int32_t(Search16Func) ( const uint8_t * const pRef, - const uint8_t * const pRefH, - const uint8_t * const pRefV, - const uint8_t * const pRefHV, - const IMAGE * const pCur, - const int x, - const int y, - const int start_x, - const int start_y, - const int center_x, - const int center_y, - const uint32_t MotionFlags, - const uint32_t iQuant, - const uint32_t iFcode, - const MBParam * const pParam, - const MACROBLOCK * const pMBs, - const MACROBLOCK * const prevMBs, - VECTOR * const currMV, - VECTOR * const currPMV); - -typedef Search16Func *Search16FuncPtr; - -typedef int32_t(Search8Func) ( const uint8_t * const pRef, - const uint8_t * const pRefH, - const uint8_t * const pRefV, - const uint8_t * const pRefHV, - const IMAGE * const pCur, - const int x, - const int y, - const int start_x, - const int start_y, - const int center_x, - const int center_y, - const uint32_t MotionFlags, - const uint32_t iQuant, - const uint32_t iFcode, - const MBParam * const pParam, - const MACROBLOCK * const pMBs, - const MACROBLOCK * const prevMBs, - VECTOR * const currMV, - VECTOR * const currPMV); - -typedef Search8Func *Search8FuncPtr; - -Search16Func PMVfastSearch16; -Search16Func EPZSSearch16; -Search16Func PMVfastIntSearch16; - -Search8Func PMVfastSearch8; -Search8Func EPZSSearch8; - - -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); - -#ifdef _SMP -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); -#endif - -typedef int32_t(Halfpel8_RefineFunc) (const uint8_t * const pRef, - const uint8_t * const pRefH, - const uint8_t * const pRefV, - const uint8_t * const pRefHV, - const uint8_t * const cur, - const int x, - const int y, - VECTOR * const currMV, - int32_t iMinSAD, - const int center_x, - const int center_y, - const int32_t min_dx, - const int32_t max_dx, - const int32_t min_dy, - const int32_t max_dy, - const int32_t iFcode, - const int32_t iQuant, - const int32_t iEdgedWidth); - -typedef Halfpel8_RefineFunc *Halfpel8_RefineFuncPtr; -extern Halfpel8_RefineFuncPtr Halfpel8_Refine; -Halfpel8_RefineFunc Halfpel8_Refine_c; -Halfpel8_RefineFunc Halfpel8_Refine_ia64; - +/***************************************************************************** + * Modified rounding tables -- defined in estimation_common.c + * Original tables see ISO spec tables 7-6 -> 7-9 + ****************************************************************************/ + +extern const uint32_t roundtab[16]; +/* K = 4 */ +extern const uint32_t roundtab_76[16]; +/* K = 2 */ +extern const uint32_t roundtab_78[8]; +/* K = 1 */ +extern const uint32_t roundtab_79[4]; + + +/** MotionEstimation **/ + +void MotionEstimation(MBParam * const pParam, + FRAMEINFO * const current, + FRAMEINFO * const reference, + const IMAGE * const pRefH, + const IMAGE * const pRefV, + const IMAGE * const pRefHV, + const IMAGE * const pGMC, + const uint32_t iLimit); + +void +MotionEstimationBVOP(MBParam * const pParam, + FRAMEINFO * const frame, + const int32_t time_bp, + const int32_t time_pp, + const MACROBLOCK * const f_mbs, + const IMAGE * const f_ref, + const IMAGE * const f_refH, + const IMAGE * const f_refV, + const IMAGE * const f_refHV, + const FRAMEINFO * const b_reference, + const IMAGE * const b_ref, + const IMAGE * const b_refH, + const IMAGE * const b_refV, + const IMAGE * const b_refHV); + +void +GMEanalysis(const MBParam * const pParam, + const FRAMEINFO * const current, + const FRAMEINFO * const reference, + const IMAGE * const pRefH, + const IMAGE * const pRefV, + const IMAGE * const pRefHV); + +WARPPOINTS +GlobalMotionEst(MACROBLOCK * const pMBs, + const MBParam * const pParam, + const FRAMEINFO * const current, + const FRAMEINFO * const reference, + const IMAGE * const pRefH, + const IMAGE * const pRefV, + const IMAGE * const pRefHV); + +int +GlobalMotionEstRefine( + WARPPOINTS *const startwp, + MACROBLOCK * const pMBs, + const MBParam * const pParam, + const FRAMEINFO * const current, + const FRAMEINFO * const reference, + const IMAGE * const pCurr, + const IMAGE * const pRef, + const IMAGE * const pRefH, + const IMAGE * const pRefV, + const IMAGE * const pRefHV); + +int +globalSAD(const WARPPOINTS *const wp, + const MBParam * const pParam, + const MACROBLOCK * const pMBs, + const FRAMEINFO * const current, + const IMAGE * const pRef, + const IMAGE * const pCurr, + uint8_t *const GMCblock); + + +int +MEanalysis( const IMAGE * const pRef, + const FRAMEINFO * const Current, + const MBParam * const pParam, + const int maxIntra, + const int intraCount, + const int bCount, + const int b_thresh, + const MACROBLOCK * const prev_mbs); + +/** MotionCompensation **/ + +void +MBMotionCompensation(MACROBLOCK * const mb, + const uint32_t i, + const uint32_t j, + const IMAGE * const ref, + const IMAGE * const refh, + const IMAGE * const refv, + const IMAGE * const refhv, + const IMAGE * const refGMC, + IMAGE * const cur, + int16_t * dct_codes, + const uint32_t width, + const uint32_t height, + const uint32_t edged_width, + const int32_t quarterpel, + const int32_t rounding); + +void +MBMotionCompensationBVOP(MBParam * pParam, + MACROBLOCK * const mb, + const uint32_t i, + const uint32_t j, + IMAGE * const cur, + const IMAGE * const f_ref, + const IMAGE * const f_refh, + const IMAGE * const f_refv, + const IMAGE * const f_refhv, + const IMAGE * const b_ref, + const IMAGE * const b_refh, + const IMAGE * const b_refv, + const IMAGE * const b_refhv, + int16_t * dct_codes); #endif /* _MOTION_H_ */