--- estimation.h 2003/09/10 22:18:59 1.1 +++ estimation.h 2003/12/18 02:02:08 1.1.2.8 @@ -0,0 +1,187 @@ +/***************************************************************************** + * + * XVID MPEG-4 VIDEO CODEC + * - Motion Estimation related header - + * + * Copyright(C) 2002 Christoph Lampert + * 2002 Michael Militzer + * 2002-2003 Radoslaw Czyz + * + * 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: estimation.h,v 1.1.2.8 2003/12/18 02:02:08 Isibaar Exp $ + * + ****************************************************************************/ + +#ifndef _ESTIMATION_H_ +#define _ESTIMATION_H_ + +#include "../portab.h" +#include "../global.h" +#include "../image/reduced.h" + +/* hard coded motion search parameters */ + +/* very large value */ +#define MV_MAX_ERROR (4096 * 256) + +/* INTER bias for INTER/INTRA decision; mpeg4 spec suggests 2*nb */ +#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_8X8_BIAS 40 + +#define BITS_MULT 16 + +#define INITIAL_SKIP_THRESH 10 +#define FINAL_SKIP_THRESH 50 +#define MAX_SAD00_FOR_SKIP 20 +#define MAX_CHROMA_SAD_FOR_SKIP 22 + +/* Parameters which control inter/inter4v decision */ +#define IMV16X16 2 + +extern const int xvid_me_lambda_vec16[32]; + +#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) + +#define MVequal(A,B) ( ((A).x)==((B).x) && ((A).y)==((B).y) ) + +static const VECTOR zeroMV = { 0, 0 }; + +typedef struct +{ + /* data modified by CheckCandidates */ + int32_t iMinSAD[5]; /* smallest SADs found so far */ + VECTOR currentMV[5]; /* best vectors found so far */ + VECTOR currentQMV[5]; /* as above, but used during qpel search */ + 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 */ + const uint8_t * Cur; /* current picture */ + const uint8_t *CurU, *CurV; /* current picture - chroma planes */ + + uint8_t * RefQ; /* temporary space for interpolations */ + uint32_t lambda16; /* how much vector bits weight */ + 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 */ + + /* fields for direct mode */ + VECTOR directmvF[4]; /* scaled reference vectors */ + VECTOR directmvB[4]; /* as above */ + 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 */ + const uint16_t * scan_table; /* current scan table */ + const uint16_t * mpeg_quant_matrices; /* current MPEG quantization matrices */ + +} SearchData; + +typedef void(CheckFunc)(const int x, const int y, + SearchData * const Data, + const unsigned int Direction); + +CheckFunc CheckCandidate16no4v; /* shared between p-vop and b-vop search */ + +uint8_t * +xvid_me_interpolate8x8qpel(const int x, const int y, const uint32_t block, + const uint32_t dir, const SearchData * const data); + +uint8_t * +xvid_me_interpolate16x16qpel(const int x, const int y, const uint32_t dir, + const SearchData * const data); + +int32_t +xvid_me_ChromaSAD(const int dx, const int dy, SearchData * const data); + +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); + +#define iDiamondSize 2 +typedef void +MainSearchFunc(int x, int y, SearchData * const Data, + int bDirection, CheckFunc * const CheckCandidate); + +MainSearchFunc xvid_me_DiamondSearch, xvid_me_AdvDiamondSearch, xvid_me_SquareSearch; + +void +xvid_me_SubpelRefine(SearchData * const data, CheckFunc * const CheckCandidate); + +void +SubpelRefine_Fast(SearchData * data, CheckFunc * CheckCandidate); + +void +xvid_me_ModeDecision_RD(SearchData * const Data, + MACROBLOCK * const pMB, + const MACROBLOCK * const pMBs, + const int x, const int y, + const MBParam * const pParam, + const uint32_t MotionFlags, + const uint32_t VopFlags, + const uint32_t VolFlags, + const IMAGE * const pCurrent, + const IMAGE * const pRef, + const IMAGE * const vGMC, + const int coding_type); + +void +xvid_me_ModeDecision_Fast(SearchData * const Data, + MACROBLOCK * const pMB, + const MACROBLOCK * const pMBs, + const int x, const int y, + const MBParam * const pParam, + const uint32_t MotionFlags, + const uint32_t VopFlags, + const uint32_t VolFlags, + const IMAGE * const pCurrent, + const IMAGE * const pRef, + const IMAGE * const vGMC, + const int coding_type); + + +#endif /* _ESTIMATION_H_ */