Parent Directory
| 
Revision Log
Revision 1988 - (view) (download)
| 1 : | edgomez | 1382 | /***************************************************************************** |
| 2 : | * | ||
| 3 : | * XVID MPEG-4 VIDEO CODEC | ||
| 4 : | * - Motion Estimation related header - | ||
| 5 : | * | ||
| 6 : | * Copyright(C) 2002 Christoph Lampert <gruel@web.de> | ||
| 7 : | Isibaar | 1909 | * 2002-2010 Michael Militzer <michael@xvid.org> |
| 8 : | edgomez | 1382 | * 2002-2003 Radoslaw Czyz <xvid@syskin.cjb.net> |
| 9 : | * | ||
| 10 : | * This program is free software ; you can redistribute it and/or modify | ||
| 11 : | * it under the terms of the GNU General Public License as published by | ||
| 12 : | * the Free Software Foundation ; either version 2 of the License, or | ||
| 13 : | * (at your option) any later version. | ||
| 14 : | * | ||
| 15 : | * This program is distributed in the hope that it will be useful, | ||
| 16 : | * but WITHOUT ANY WARRANTY ; without even the implied warranty of | ||
| 17 : | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
| 18 : | * GNU General Public License for more details. | ||
| 19 : | * | ||
| 20 : | * You should have received a copy of the GNU General Public License | ||
| 21 : | * along with this program ; if not, write to the Free Software | ||
| 22 : | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | ||
| 23 : | * | ||
| 24 : | Isibaar | 1988 | * $Id$ |
| 25 : | edgomez | 1382 | * |
| 26 : | ****************************************************************************/ | ||
| 27 : | |||
| 28 : | #ifndef _ESTIMATION_H_ | ||
| 29 : | #define _ESTIMATION_H_ | ||
| 30 : | |||
| 31 : | #include "../portab.h" | ||
| 32 : | #include "../global.h" | ||
| 33 : | Isibaar | 1909 | #include "sad.h" |
| 34 : | edgomez | 1382 | |
| 35 : | /* hard coded motion search parameters */ | ||
| 36 : | |||
| 37 : | /* very large value */ | ||
| 38 : | #define MV_MAX_ERROR (4096 * 256) | ||
| 39 : | |||
| 40 : | /* INTER bias for INTER/INTRA decision; mpeg4 spec suggests 2*nb */ | ||
| 41 : | #define MV16_INTER_BIAS 450 | ||
| 42 : | |||
| 43 : | /* vector map (vlc delta size) smoother parameters ! float !*/ | ||
| 44 : | Isibaar | 1604 | #define NEIGH_TEND_16X16 0.6 |
| 45 : | syskin | 1659 | #define NEIGH_TEND_8X8 1.5 |
| 46 : | |||
| 47 : | edgomez | 1382 | #define NEIGH_8X8_BIAS 40 |
| 48 : | |||
| 49 : | #define BITS_MULT 16 | ||
| 50 : | |||
| 51 : | syskin | 1443 | #define INITIAL_SKIP_THRESH 6 |
| 52 : | edgomez | 1382 | #define FINAL_SKIP_THRESH 50 |
| 53 : | #define MAX_SAD00_FOR_SKIP 20 | ||
| 54 : | #define MAX_CHROMA_SAD_FOR_SKIP 22 | ||
| 55 : | |||
| 56 : | /* Parameters which control inter/inter4v decision */ | ||
| 57 : | #define IMV16X16 2 | ||
| 58 : | |||
| 59 : | extern const int xvid_me_lambda_vec16[32]; | ||
| 60 : | |||
| 61 : | #define CHECK_CANDIDATE(X,Y,D) { \ | ||
| 62 : | CheckCandidate((X),(Y), data, (D) ); } | ||
| 63 : | |||
| 64 : | /* fast ((A)/2)*2 */ | ||
| 65 : | #define EVEN(A) (((A)<0?(A)+1:(A)) & ~1) | ||
| 66 : | |||
| 67 : | #define MVequal(A,B) ( ((A).x)==((B).x) && ((A).y)==((B).y) ) | ||
| 68 : | |||
| 69 : | static const VECTOR zeroMV = { 0, 0 }; | ||
| 70 : | |||
| 71 : | typedef struct | ||
| 72 : | { | ||
| 73 : | syskin | 1567 | int max_dx, min_dx, max_dy, min_dy; /* maximum search range */ |
| 74 : | |||
| 75 : | edgomez | 1382 | /* data modified by CheckCandidates */ |
| 76 : | int32_t iMinSAD[5]; /* smallest SADs found so far */ | ||
| 77 : | VECTOR currentMV[5]; /* best vectors found so far */ | ||
| 78 : | VECTOR currentQMV[5]; /* as above, but used during qpel search */ | ||
| 79 : | int temp[4]; /* temporary space */ | ||
| 80 : | unsigned int dir; /* 'direction', set when better vector is found */ | ||
| 81 : | int chromaX, chromaY, chromaSAD; /* info to make ChromaSAD faster */ | ||
| 82 : | |||
| 83 : | /* general fields */ | ||
| 84 : | uint32_t rounding; /* rounding type in use */ | ||
| 85 : | VECTOR predMV; /* vector which predicts current vector */ | ||
| 86 : | const uint8_t * RefP[6]; /* reference pictures - N, V, H, HV, cU, cV */ | ||
| 87 : | const uint8_t * Cur; /* current picture */ | ||
| 88 : | const uint8_t *CurU, *CurV; /* current picture - chroma planes */ | ||
| 89 : | |||
| 90 : | uint8_t * RefQ; /* temporary space for interpolations */ | ||
| 91 : | uint32_t lambda16; /* how much vector bits weight */ | ||
| 92 : | uint32_t lambda8; /* as above - for inter4v mode */ | ||
| 93 : | uint32_t iEdgedWidth; /* picture's stride */ | ||
| 94 : | uint32_t iFcode; /* current fcode */ | ||
| 95 : | syskin | 1441 | |
| 96 : | edgomez | 1382 | int qpel; /* if we're coding in qpel mode */ |
| 97 : | int qpel_precision; /* if X and Y are in qpel precision (refinement probably) */ | ||
| 98 : | int chroma; /* should we include chroma SAD? */ | ||
| 99 : | |||
| 100 : | /* fields for interpolate and direct modes */ | ||
| 101 : | const uint8_t * b_RefP[6]; /* backward reference pictures - N, V, H, HV, cU, cV */ | ||
| 102 : | syskin | 1564 | VECTOR bpredMV; /* backward prediction - used in Interpolate-mode search only */ |
| 103 : | uint32_t bFcode; /* backward fcode - used in Interpolate-mode search only */ | ||
| 104 : | syskin | 1443 | int b_chromaX, b_chromaY; |
| 105 : | edgomez | 1382 | |
| 106 : | /* fields for direct mode */ | ||
| 107 : | VECTOR directmvF[4]; /* scaled reference vectors */ | ||
| 108 : | syskin | 1564 | VECTOR directmvB[4]; |
| 109 : | edgomez | 1382 | const VECTOR * referencemv; /* pointer to not-scaled reference vectors */ |
| 110 : | |||
| 111 : | /* BITS/R-D stuff */ | ||
| 112 : | int16_t * dctSpace; /* temporary space for dct */ | ||
| 113 : | uint32_t iQuant; /* current quant */ | ||
| 114 : | uint32_t quant_type; /* current quant type */ | ||
| 115 : | syskin | 1506 | unsigned int cbp[2]; /* CBP of the best vector found so far + cbp for inter4v search */ |
| 116 : | edgomez | 1382 | const uint16_t * scan_table; /* current scan table */ |
| 117 : | const uint16_t * mpeg_quant_matrices; /* current MPEG quantization matrices */ | ||
| 118 : | syskin | 1506 | int lambda[6]; /* R-D lambdas for all 6 blocks */ |
| 119 : | syskin | 1569 | unsigned int quant_sq; /* quant squared - saves many multiplications in VHQ */ |
| 120 : | Isibaar | 1909 | uint32_t rel_var8[6]; /* relative variances for all 6 sub-blocks */ |
| 121 : | int metric; /* distortion metric for R-D optimizations, currently: PSNR=0, PSNRHVSM=1 */ | ||
| 122 : | edgomez | 1382 | } SearchData; |
| 123 : | |||
| 124 : | Isibaar | 1909 | static __inline uint32_t |
| 125 : | masked_sseh8_16bit(int16_t * const orig, | ||
| 126 : | int16_t * const rec, | ||
| 127 : | const uint32_t rel_var8) | ||
| 128 : | { | ||
| 129 : | uint16_t mask = ((isqrt(2*coeff8_energy(orig)*rel_var8) + 48) >> 6); | ||
| 130 : | return (5*sseh8_16bit(orig, rec, (uint16_t) mask)) >> 7; | ||
| 131 : | } | ||
| 132 : | |||
| 133 : | edgomez | 1382 | typedef void(CheckFunc)(const int x, const int y, |
| 134 : | SearchData * const Data, | ||
| 135 : | const unsigned int Direction); | ||
| 136 : | |||
| 137 : | CheckFunc CheckCandidate16no4v; /* shared between p-vop and b-vop search */ | ||
| 138 : | |||
| 139 : | uint8_t * | ||
| 140 : | xvid_me_interpolate8x8qpel(const int x, const int y, const uint32_t block, | ||
| 141 : | const uint32_t dir, const SearchData * const data); | ||
| 142 : | |||
| 143 : | uint8_t * | ||
| 144 : | xvid_me_interpolate16x16qpel(const int x, const int y, const uint32_t dir, | ||
| 145 : | const SearchData * const data); | ||
| 146 : | |||
| 147 : | int32_t | ||
| 148 : | xvid_me_ChromaSAD(const int dx, const int dy, SearchData * const data); | ||
| 149 : | |||
| 150 : | int | ||
| 151 : | xvid_me_SkipDecisionP(const IMAGE * current, const IMAGE * reference, | ||
| 152 : | const int x, const int y, | ||
| 153 : | syskin | 1564 | const uint32_t stride, const uint32_t iQuant); |
| 154 : | edgomez | 1382 | |
| 155 : | #define iDiamondSize 2 | ||
| 156 : | typedef void | ||
| 157 : | MainSearchFunc(int x, int y, SearchData * const Data, | ||
| 158 : | int bDirection, CheckFunc * const CheckCandidate); | ||
| 159 : | |||
| 160 : | MainSearchFunc xvid_me_DiamondSearch, xvid_me_AdvDiamondSearch, xvid_me_SquareSearch; | ||
| 161 : | |||
| 162 : | void | ||
| 163 : | syskin | 1478 | xvid_me_SubpelRefine(VECTOR centerMV, SearchData * const data, CheckFunc * const CheckCandidate, int dir); |
| 164 : | edgomez | 1382 | |
| 165 : | syskin | 1441 | void |
| 166 : | FullRefine_Fast(SearchData * data, CheckFunc * CheckCandidate, int direction); | ||
| 167 : | edgomez | 1382 | |
| 168 : | void | ||
| 169 : | xvid_me_ModeDecision_RD(SearchData * const Data, | ||
| 170 : | MACROBLOCK * const pMB, | ||
| 171 : | const MACROBLOCK * const pMBs, | ||
| 172 : | const int x, const int y, | ||
| 173 : | const MBParam * const pParam, | ||
| 174 : | const uint32_t MotionFlags, | ||
| 175 : | const uint32_t VopFlags, | ||
| 176 : | const uint32_t VolFlags, | ||
| 177 : | const IMAGE * const pCurrent, | ||
| 178 : | const IMAGE * const pRef, | ||
| 179 : | const IMAGE * const vGMC, | ||
| 180 : | Isibaar | 1913 | const int coding_type, |
| 181 : | const int bound); | ||
| 182 : | edgomez | 1382 | |
| 183 : | void | ||
| 184 : | xvid_me_ModeDecision_Fast(SearchData * const Data, | ||
| 185 : | MACROBLOCK * const pMB, | ||
| 186 : | const MACROBLOCK * const pMBs, | ||
| 187 : | const int x, const int y, | ||
| 188 : | const MBParam * const pParam, | ||
| 189 : | const uint32_t MotionFlags, | ||
| 190 : | const uint32_t VopFlags, | ||
| 191 : | const uint32_t VolFlags, | ||
| 192 : | const IMAGE * const pCurrent, | ||
| 193 : | const IMAGE * const pRef, | ||
| 194 : | const IMAGE * const vGMC, | ||
| 195 : | Isibaar | 1913 | const int coding_type, |
| 196 : | const int bound); | ||
| 197 : | edgomez | 1382 | |
| 198 : | syskin | 1506 | void |
| 199 : | ModeDecision_BVOP_RD(SearchData * const Data_d, | ||
| 200 : | SearchData * const Data_b, | ||
| 201 : | SearchData * const Data_f, | ||
| 202 : | SearchData * const Data_i, | ||
| 203 : | MACROBLOCK * const pMB, | ||
| 204 : | const MACROBLOCK * const b_mb, | ||
| 205 : | VECTOR * f_predMV, | ||
| 206 : | VECTOR * b_predMV, | ||
| 207 : | const uint32_t MotionFlags, | ||
| 208 : | Isibaar | 1909 | const uint32_t VopFlags, |
| 209 : | syskin | 1506 | const MBParam * const pParam, |
| 210 : | syskin | 1577 | int x, int y, |
| 211 : | Isibaar | 1919 | int best_sad, |
| 212 : | int force_direct); | ||
| 213 : | edgomez | 1382 | |
| 214 : | syskin | 1567 | unsigned int |
| 215 : | getMinFcode(const int MVmax); | ||
| 216 : | syskin | 1506 | |
| 217 : | edgomez | 1382 | #endif /* _ESTIMATION_H_ */ |
| No admin address has been configured | ViewVC Help |
| Powered by ViewVC 1.0.4 |