--- motion_est.h 2002/09/23 20:36:02 1.1.2.1 +++ motion_est.h 2003/01/24 11:44:24 1.1.2.14 @@ -26,7 +26,7 @@ * 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_est.h,v 1.1.2.1 2002/09/23 20:36:02 chl Exp $ + * $Id: motion_est.h,v 1.1.2.14 2003/01/24 11:44:24 syskin Exp $ * ***************************************************************************/ @@ -35,6 +35,7 @@ #include "../portab.h" #include "../global.h" +#include "../image/reduced.h" /* hard coded motion search parameters for motion_est and smp_motion_est */ @@ -44,13 +45,14 @@ /* INTER bias for INTER/INTRA decision; mpeg4 spec suggests 2*nb */ #define MV16_INTER_BIAS 512 +/* vector map (vlc delta size) smoother parameters ! float !*/ +#define NEIGH_TEND_16X16 10.5 +#define NEIGH_TEND_8X8 40.0 +#define NEIGH_8X8_BIAS 30 + /* Parameters which control inter/inter4v decision */ #define IMV16X16 2 -/* vector map (vlc delta size) smoother parameters ! float !*/ -#define NEIGH_TEND_16X16 4.0 -#define NEIGH_TEND_8X8 7.0 - static const int lambda_vec16[32] = { 0 ,(int)(1.00235 * NEIGH_TEND_16X16 + 0.5), (int)(1.15582*NEIGH_TEND_16X16 + 0.5), (int)(1.31976*NEIGH_TEND_16X16 + 0.5), @@ -69,7 +71,7 @@ (int)(21.2656*NEIGH_TEND_16X16 + 0.5), (int)(24.8580*NEIGH_TEND_16X16 + 0.5), (int)(29.6436*NEIGH_TEND_16X16 + 0.5), (int)(36.4949*NEIGH_TEND_16X16 + 0.5) }; -static const int lambda_vec8[32] = +static const int lambda_vec8[32] = { 0 ,(int)(1.00235 * NEIGH_TEND_8X8 + 0.5), (int)(1.15582 + NEIGH_TEND_8X8 + 0.5), (int)(1.31976*NEIGH_TEND_8X8 + 0.5), (int)(1.49591*NEIGH_TEND_8X8 + 0.5), (int)(1.68601*NEIGH_TEND_8X8 + 0.5), @@ -95,48 +97,66 @@ 10, 11, 11, 11, 11, 11, 11, 12, 12 }; +static const int DQtab[4] = { + -1, -2, 1, 2 +}; + +#define RRV_MV_SCALEDOWN(a) ( (a)>=0 ? (a+1)/2 : (a-1)/2 ) + +static const VECTOR zeroMV = {0,0}; + typedef struct - { +{ // general fields - int max_dx, min_dx, max_dy, min_dy; - VECTOR predMV; - VECTOR *currentMV; - int32_t *iMinSAD; - const uint8_t * Ref; - const uint8_t * RefH; - const uint8_t * RefV; - const uint8_t * RefHV; - const uint8_t * Cur; - uint32_t iQuant; - uint32_t iEdgedWidth; - uint32_t iFcode; - int * temp; -//fields for interpolate and direct mode - const uint8_t *bRef; - const uint8_t *bRefH; - const uint8_t *bRefV; - const uint8_t *bRefHV; - VECTOR bpredMV; - uint32_t bFcode; + int max_dx, min_dx, max_dy, min_dy; + uint32_t rounding; + VECTOR predMV; + VECTOR * currentMV; + VECTOR * currentQMV; + int32_t * iMinSAD; + const uint8_t * Ref; + const uint8_t * RefH; + const uint8_t * RefV; + const uint8_t * RefHV; + const uint8_t * RefCU; + const uint8_t * RefCV; + const uint8_t * CurU; + const uint8_t * CurV; + uint8_t * RefQ; + const uint8_t * Cur; + uint32_t lambda16; + uint32_t lambda8; + uint32_t iEdgedWidth; + uint32_t iFcode; + int * temp; + int qpel, qpel_precision; + int chroma; + int rrv; +//fields for interpolate and direct modes + const uint8_t * bRef; + const uint8_t * bRefH; + const uint8_t * bRefV; + const uint8_t * bRefHV; + const uint8_t * b_RefCU; + const uint8_t * b_RefCV; + + VECTOR bpredMV; + uint32_t bFcode; // fields for direct mode - VECTOR directmvF[4]; - VECTOR directmvB[4]; - const VECTOR * referencemv; - } - SearchData; + VECTOR directmvF[4]; + VECTOR directmvB[4]; + const VECTOR * referencemv; + +} SearchData; typedef void(CheckFunc)(const int x, const int y, const int Direction, int * const dir, const SearchData * const Data); - -static CheckFunc CheckCandidate16, CheckCandidate16no4v, CheckCandidateInt, - CheckCandidateDirect, CheckCandidateDirectno4v, - CheckCandidate8; CheckFunc *CheckCandidate; /* - * Calculate the min/max range (in halfpixels) + * Calculate the min/max range * relative to the _MACROBLOCK_ position */ static void __inline @@ -146,29 +166,35 @@ int32_t * const max_dy, const uint32_t x, const uint32_t y, - const uint32_t block_sz, /* block dimension, 8 or 16 */ + uint32_t block_sz, /* block dimension, 8 or 16 */ const uint32_t width, const uint32_t height, - const uint32_t fcode) + const uint32_t fcode, + const int qpel, /* 1 if the resulting range should be in qpel precision; otherwise 0 */ + const int rrv) { + int k, m = qpel ? 4 : 2; + const int search_range = 32 << (fcode - 1); + int high = search_range - 1; + int low = -search_range; + + if (rrv) { + high = RRV_MV_SCALEUP(high); + low = RRV_MV_SCALEUP(low); + block_sz *= 2; + } - int k; - int high = (32 << (fcode - 1)) - 1; - - k = 2 * (int)(width - x*block_sz); + k = m * (int)(width - x * block_sz); *max_dx = MIN(high, k); - k = 2 * (int)(height - y*block_sz); + k = m * (int)(height - y * block_sz); *max_dy = MIN(high, k); - high = -(32 << (fcode - 1)); - k = -2 * (int)((x+1) * block_sz); - *min_dx = MAX(high, k); - k = -2 * (int)((y+1) * block_sz); - *min_dy = MAX(high, k); - + k = -m * (int)((x+1) * block_sz); + *min_dx = MAX(low, k); + k = -m * (int)((y+1) * block_sz); + *min_dy = MAX(low, k); } - typedef void MainSearchFunc(int x, int y, const SearchData * const Data, int bDirection); static MainSearchFunc DiamondSearch, AdvDiamondSearch, SquareSearch; @@ -179,7 +205,8 @@ const MBParam * const pParam, MACROBLOCK * const pMB, const MACROBLOCK * const pMBs, - const int block); + const int block, + SearchData * const Data); bool MotionEstimation(MBParam * const pParam, @@ -191,7 +218,7 @@ const uint32_t iLimit); static void -SearchP(const uint8_t * const pRef, +SearchP(const IMAGE * const pRef, const uint8_t * const pRefH, const uint8_t * const pRefV, const uint8_t * const pRefHV, @@ -200,7 +227,7 @@ const int y, const uint32_t MotionFlags, const uint32_t iQuant, - const uint32_t iFcode, + SearchData * const Data, const MBParam * const pParam, const MACROBLOCK * const pMBs, const MACROBLOCK * const prevMBs, @@ -208,15 +235,15 @@ MACROBLOCK * const pMB); -#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 +static WARPPOINTS +GlobalMotionEst(const 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 ); + +#define iDiamondSize 2 #endif /* _MOTION_EST_H_ */