--- estimation.h 2003/12/18 02:02:08 1.1.2.8 +++ estimation.h 2010/12/24 13:21:35 1.16 @@ -4,7 +4,7 @@ * - Motion Estimation related header - * * Copyright(C) 2002 Christoph Lampert - * 2002 Michael Militzer + * 2002-2010 Michael Militzer * 2002-2003 Radoslaw Czyz * * This program is free software ; you can redistribute it and/or modify @@ -21,7 +21,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: estimation.h,v 1.1.2.8 2003/12/18 02:02:08 Isibaar Exp $ + * $Id: estimation.h,v 1.16 2010/12/24 13:21:35 Isibaar Exp $ * ****************************************************************************/ @@ -30,7 +30,7 @@ #include "../portab.h" #include "../global.h" -#include "../image/reduced.h" +#include "sad.h" /* hard coded motion search parameters */ @@ -41,13 +41,14 @@ #define MV16_INTER_BIAS 450 /* vector map (vlc delta size) smoother parameters ! float !*/ -#define NEIGH_TEND_16X16 10.5 -#define NEIGH_TEND_8X8 40.0 +#define NEIGH_TEND_16X16 0.6 +#define NEIGH_TEND_8X8 1.5 + #define NEIGH_8X8_BIAS 40 #define BITS_MULT 16 -#define INITIAL_SKIP_THRESH 10 +#define INITIAL_SKIP_THRESH 6 #define FINAL_SKIP_THRESH 50 #define MAX_SAD00_FOR_SKIP 20 #define MAX_CHROMA_SAD_FOR_SKIP 22 @@ -60,8 +61,6 @@ #define CHECK_CANDIDATE(X,Y,D) { \ CheckCandidate((X),(Y), data, (D) ); } -#define RRV_MV_SCALEDOWN(a) ( (a)>=0 ? (a+1)/2 : (a-1)/2 ) - /* fast ((A)/2)*2 */ #define EVEN(A) (((A)<0?(A)+1:(A)) & ~1) @@ -71,6 +70,8 @@ typedef struct { + int max_dx, min_dx, max_dy, min_dy; /* maximum search range */ + /* data modified by CheckCandidates */ int32_t iMinSAD[5]; /* smallest SADs found so far */ VECTOR currentMV[5]; /* best vectors found so far */ @@ -78,11 +79,8 @@ int temp[4]; /* temporary space */ unsigned int dir; /* 'direction', set when better vector is found */ int chromaX, chromaY, chromaSAD; /* info to make ChromaSAD faster */ - VECTOR currentQMV2; /* extra vector for SubpelRefine_fast */ - int32_t iMinSAD2; /* extra SAD value for SubpelRefine_fast */ /* general fields */ - int max_dx, min_dx, max_dy, min_dy; /* maximum range */ uint32_t rounding; /* rounding type in use */ VECTOR predMV; /* vector which predicts current vector */ const uint8_t * RefP[6]; /* reference pictures - N, V, H, HV, cU, cV */ @@ -94,32 +92,44 @@ uint32_t lambda8; /* as above - for inter4v mode */ uint32_t iEdgedWidth; /* picture's stride */ uint32_t iFcode; /* current fcode */ - + int qpel; /* if we're coding in qpel mode */ int qpel_precision; /* if X and Y are in qpel precision (refinement probably) */ int chroma; /* should we include chroma SAD? */ - int rrv; /* are we using reduced resolution? */ /* fields for interpolate and direct modes */ const uint8_t * b_RefP[6]; /* backward reference pictures - N, V, H, HV, cU, cV */ - VECTOR bpredMV; /* backward prediction - used interpolate mode only */ - uint32_t bFcode; /* backward fcode - used as above */ + VECTOR bpredMV; /* backward prediction - used in Interpolate-mode search only */ + uint32_t bFcode; /* backward fcode - used in Interpolate-mode search only */ + int b_chromaX, b_chromaY; /* fields for direct mode */ VECTOR directmvF[4]; /* scaled reference vectors */ - VECTOR directmvB[4]; /* as above */ + VECTOR directmvB[4]; const VECTOR * referencemv; /* pointer to not-scaled reference vectors */ /* BITS/R-D stuff */ int16_t * dctSpace; /* temporary space for dct */ uint32_t iQuant; /* current quant */ uint32_t quant_type; /* current quant type */ - unsigned int cbp[2]; /* CBP of the best vector found so far + cbp for inter4v search */ + unsigned int cbp[2]; /* CBP of the best vector found so far + cbp for inter4v search */ const uint16_t * scan_table; /* current scan table */ const uint16_t * mpeg_quant_matrices; /* current MPEG quantization matrices */ - + int lambda[6]; /* R-D lambdas for all 6 blocks */ + unsigned int quant_sq; /* quant squared - saves many multiplications in VHQ */ + uint32_t rel_var8[6]; /* relative variances for all 6 sub-blocks */ + int metric; /* distortion metric for R-D optimizations, currently: PSNR=0, PSNRHVSM=1 */ } SearchData; +static __inline uint32_t +masked_sseh8_16bit(int16_t * const orig, + int16_t * const rec, + const uint32_t rel_var8) +{ + uint16_t mask = ((isqrt(2*coeff8_energy(orig)*rel_var8) + 48) >> 6); + return (5*sseh8_16bit(orig, rec, (uint16_t) mask)) >> 7; +} + typedef void(CheckFunc)(const int x, const int y, SearchData * const Data, const unsigned int Direction); @@ -140,7 +150,7 @@ int xvid_me_SkipDecisionP(const IMAGE * current, const IMAGE * reference, const int x, const int y, - const uint32_t stride, const uint32_t iQuant, int rrv); + const uint32_t stride, const uint32_t iQuant); #define iDiamondSize 2 typedef void @@ -150,10 +160,10 @@ MainSearchFunc xvid_me_DiamondSearch, xvid_me_AdvDiamondSearch, xvid_me_SquareSearch; void -xvid_me_SubpelRefine(SearchData * const data, CheckFunc * const CheckCandidate); +xvid_me_SubpelRefine(VECTOR centerMV, SearchData * const data, CheckFunc * const CheckCandidate, int dir); -void -SubpelRefine_Fast(SearchData * data, CheckFunc * CheckCandidate); +void +FullRefine_Fast(SearchData * data, CheckFunc * CheckCandidate, int direction); void xvid_me_ModeDecision_RD(SearchData * const Data, @@ -167,7 +177,8 @@ const IMAGE * const pCurrent, const IMAGE * const pRef, const IMAGE * const vGMC, - const int coding_type); + const int coding_type, + const int bound); void xvid_me_ModeDecision_Fast(SearchData * const Data, @@ -181,7 +192,26 @@ const IMAGE * const pCurrent, const IMAGE * const pRef, const IMAGE * const vGMC, - const int coding_type); + const int coding_type, + const int bound); + +void +ModeDecision_BVOP_RD(SearchData * const Data_d, + SearchData * const Data_b, + SearchData * const Data_f, + SearchData * const Data_i, + MACROBLOCK * const pMB, + const MACROBLOCK * const b_mb, + VECTOR * f_predMV, + VECTOR * b_predMV, + const uint32_t MotionFlags, + const uint32_t VopFlags, + const MBParam * const pParam, + int x, int y, + int best_sad, + int force_direct); +unsigned int +getMinFcode(const int MVmax); #endif /* _ESTIMATION_H_ */