Parent Directory
| 
Revision Log
Revision 1016 - (view) (download)
| 1 : | edgomez | 851 | /************************************************************************** |
| 2 : | * | ||
| 3 : | * XVID MPEG-4 VIDEO CODEC | ||
| 4 : | * - Motion estimation header - | ||
| 5 : | * | ||
| 6 : | * This program is an implementation of a part of one or more MPEG-4 | ||
| 7 : | * Video tools as specified in ISO/IEC 14496-2 standard. Those intending | ||
| 8 : | * to use this software module in hardware or software products are | ||
| 9 : | * advised that its use may infringe existing patents or copyrights, and | ||
| 10 : | * any such use would be at such party's own risk. The original | ||
| 11 : | * developer of this software module and his/her company, and subsequent | ||
| 12 : | * editors and their companies, will have no liability for use of this | ||
| 13 : | * software or modifications or derivatives thereof. | ||
| 14 : | * | ||
| 15 : | * This program is free software; you can redistribute it and/or modify | ||
| 16 : | * it under the terms of the GNU General Public License as published by | ||
| 17 : | * the Free Software Foundation; either version 2 of the License, or | ||
| 18 : | * (at your option) any later version. | ||
| 19 : | * | ||
| 20 : | * This program is distributed in the hope that it will be useful, | ||
| 21 : | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
| 22 : | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
| 23 : | * GNU General Public License for more details. | ||
| 24 : | * | ||
| 25 : | * You should have received a copy of the GNU General Public License | ||
| 26 : | * along with this program; if not, write to the Free Software | ||
| 27 : | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | ||
| 28 : | * | ||
| 29 : | syskin | 1016 | * $Id: motion_est.h,v 1.7 2003-05-13 12:48:20 syskin Exp $ |
| 30 : | edgomez | 851 | * |
| 31 : | ***************************************************************************/ | ||
| 32 : | |||
| 33 : | #ifndef _MOTION_EST_H_ | ||
| 34 : | #define _MOTION_EST_H_ | ||
| 35 : | |||
| 36 : | #include "../portab.h" | ||
| 37 : | #include "../global.h" | ||
| 38 : | #include "../image/reduced.h" | ||
| 39 : | |||
| 40 : | /* hard coded motion search parameters for motion_est and smp_motion_est */ | ||
| 41 : | |||
| 42 : | // very large value | ||
| 43 : | #define MV_MAX_ERROR (4096 * 256) | ||
| 44 : | |||
| 45 : | /* INTER bias for INTER/INTRA decision; mpeg4 spec suggests 2*nb */ | ||
| 46 : | #define MV16_INTER_BIAS 512 | ||
| 47 : | |||
| 48 : | /* vector map (vlc delta size) smoother parameters ! float !*/ | ||
| 49 : | #define NEIGH_TEND_16X16 10.5 | ||
| 50 : | #define NEIGH_TEND_8X8 40.0 | ||
| 51 : | #define NEIGH_8X8_BIAS 30 | ||
| 52 : | |||
| 53 : | /* Parameters which control inter/inter4v decision */ | ||
| 54 : | #define IMV16X16 2 | ||
| 55 : | |||
| 56 : | static const int lambda_vec16[32] = | ||
| 57 : | { 0 ,(int)(1.00235 * NEIGH_TEND_16X16 + 0.5), | ||
| 58 : | (int)(1.15582*NEIGH_TEND_16X16 + 0.5), (int)(1.31976*NEIGH_TEND_16X16 + 0.5), | ||
| 59 : | (int)(1.49591*NEIGH_TEND_16X16 + 0.5), (int)(1.68601*NEIGH_TEND_16X16 + 0.5), | ||
| 60 : | (int)(1.89187*NEIGH_TEND_16X16 + 0.5), (int)(2.11542*NEIGH_TEND_16X16 + 0.5), | ||
| 61 : | (int)(2.35878*NEIGH_TEND_16X16 + 0.5), (int)(2.62429*NEIGH_TEND_16X16 + 0.5), | ||
| 62 : | (int)(2.91455*NEIGH_TEND_16X16 + 0.5), (int)(3.23253*NEIGH_TEND_16X16 + 0.5), | ||
| 63 : | (int)(3.58158*NEIGH_TEND_16X16 + 0.5), (int)(3.96555*NEIGH_TEND_16X16 + 0.5), | ||
| 64 : | (int)(4.38887*NEIGH_TEND_16X16 + 0.5), (int)(4.85673*NEIGH_TEND_16X16 + 0.5), | ||
| 65 : | (int)(5.37519*NEIGH_TEND_16X16 + 0.5), (int)(5.95144*NEIGH_TEND_16X16 + 0.5), | ||
| 66 : | (int)(6.59408*NEIGH_TEND_16X16 + 0.5), (int)(7.31349*NEIGH_TEND_16X16 + 0.5), | ||
| 67 : | (int)(8.12242*NEIGH_TEND_16X16 + 0.5), (int)(9.03669*NEIGH_TEND_16X16 + 0.5), | ||
| 68 : | (int)(10.0763*NEIGH_TEND_16X16 + 0.5), (int)(11.2669*NEIGH_TEND_16X16 + 0.5), | ||
| 69 : | (int)(12.6426*NEIGH_TEND_16X16 + 0.5), (int)(14.2493*NEIGH_TEND_16X16 + 0.5), | ||
| 70 : | (int)(16.1512*NEIGH_TEND_16X16 + 0.5), (int)(18.442*NEIGH_TEND_16X16 + 0.5), | ||
| 71 : | (int)(21.2656*NEIGH_TEND_16X16 + 0.5), (int)(24.8580*NEIGH_TEND_16X16 + 0.5), | ||
| 72 : | (int)(29.6436*NEIGH_TEND_16X16 + 0.5), (int)(36.4949*NEIGH_TEND_16X16 + 0.5) }; | ||
| 73 : | |||
| 74 : | static const int lambda_vec8[32] = | ||
| 75 : | { 0 ,(int)(1.00235 * NEIGH_TEND_8X8 + 0.5), | ||
| 76 : | (int)(1.15582 + NEIGH_TEND_8X8 + 0.5), (int)(1.31976*NEIGH_TEND_8X8 + 0.5), | ||
| 77 : | (int)(1.49591*NEIGH_TEND_8X8 + 0.5), (int)(1.68601*NEIGH_TEND_8X8 + 0.5), | ||
| 78 : | (int)(1.89187*NEIGH_TEND_8X8 + 0.5), (int)(2.11542*NEIGH_TEND_8X8 + 0.5), | ||
| 79 : | (int)(2.35878*NEIGH_TEND_8X8 + 0.5), (int)(2.62429*NEIGH_TEND_8X8 + 0.5), | ||
| 80 : | (int)(2.91455*NEIGH_TEND_8X8 + 0.5), (int)(3.23253*NEIGH_TEND_8X8 + 0.5), | ||
| 81 : | (int)(3.58158*NEIGH_TEND_8X8 + 0.5), (int)(3.96555*NEIGH_TEND_8X8 + 0.5), | ||
| 82 : | (int)(4.38887*NEIGH_TEND_8X8 + 0.5), (int)(4.85673*NEIGH_TEND_8X8 + 0.5), | ||
| 83 : | (int)(5.37519*NEIGH_TEND_8X8 + 0.5), (int)(5.95144*NEIGH_TEND_8X8 + 0.5), | ||
| 84 : | (int)(6.59408*NEIGH_TEND_8X8 + 0.5), (int)(7.31349*NEIGH_TEND_8X8 + 0.5), | ||
| 85 : | (int)(8.12242*NEIGH_TEND_8X8 + 0.5), (int)(9.03669*NEIGH_TEND_8X8 + 0.5), | ||
| 86 : | (int)(10.0763*NEIGH_TEND_8X8 + 0.5), (int)(11.2669*NEIGH_TEND_8X8 + 0.5), | ||
| 87 : | (int)(12.6426*NEIGH_TEND_8X8 + 0.5), (int)(14.2493*NEIGH_TEND_8X8 + 0.5), | ||
| 88 : | (int)(16.1512*NEIGH_TEND_8X8 + 0.5), (int)(18.442*NEIGH_TEND_8X8 + 0.5), | ||
| 89 : | (int)(21.2656*NEIGH_TEND_8X8 + 0.5), (int)(24.8580*NEIGH_TEND_8X8 + 0.5), | ||
| 90 : | (int)(29.6436*NEIGH_TEND_8X8 + 0.5), (int)(36.4949*NEIGH_TEND_8X8 + 0.5) }; | ||
| 91 : | |||
| 92 : | // mv.length table | ||
| 93 : | syskin | 973 | static const int mvtab[64] = { |
| 94 : | 1, 2, 3, 4, 6, 7, 7, 7, | ||
| 95 : | 9, 9, 9, 10, 10, 10, 10, 10, | ||
| 96 : | 10, 10, 10, 10, 10, 10, 10, 10, | ||
| 97 : | 10, 11, 11, 11, 11, 11, 11, 12, | ||
| 98 : | 12, 12, 12, 12, 12, 12, 12, 12, | ||
| 99 : | 12, 12, 12, 12, 12, 12, 12, 12, | ||
| 100 : | 12, 12, 12, 12, 12, 12, 12, 12, 12 }; | ||
| 101 : | edgomez | 851 | |
| 102 : | static const int DQtab[4] = { | ||
| 103 : | -1, -2, 1, 2 | ||
| 104 : | }; | ||
| 105 : | |||
| 106 : | #define RRV_MV_SCALEDOWN(a) ( (a)>=0 ? (a+1)/2 : (a-1)/2 ) | ||
| 107 : | |||
| 108 : | typedef struct | ||
| 109 : | { | ||
| 110 : | // general fields | ||
| 111 : | int max_dx, min_dx, max_dy, min_dy; | ||
| 112 : | uint32_t rounding; | ||
| 113 : | VECTOR predMV; | ||
| 114 : | VECTOR * currentMV; | ||
| 115 : | VECTOR * currentQMV; | ||
| 116 : | int32_t * iMinSAD; | ||
| 117 : | syskin | 976 | const uint8_t * RefP[6]; // N, V, H, HV, cU, cV |
| 118 : | edgomez | 851 | const uint8_t * CurU; |
| 119 : | const uint8_t * CurV; | ||
| 120 : | uint8_t * RefQ; | ||
| 121 : | const uint8_t * Cur; | ||
| 122 : | uint32_t lambda16; | ||
| 123 : | uint32_t lambda8; | ||
| 124 : | uint32_t iEdgedWidth; | ||
| 125 : | uint32_t iFcode; | ||
| 126 : | int * temp; | ||
| 127 : | int qpel, qpel_precision; | ||
| 128 : | int chroma; | ||
| 129 : | int rrv; | ||
| 130 : | //fields for interpolate and direct modes | ||
| 131 : | syskin | 976 | const uint8_t * b_RefP[6]; // N, V, H, HV, cU, cV |
| 132 : | edgomez | 851 | VECTOR bpredMV; |
| 133 : | uint32_t bFcode; | ||
| 134 : | // fields for direct mode | ||
| 135 : | VECTOR directmvF[4]; | ||
| 136 : | VECTOR directmvB[4]; | ||
| 137 : | const VECTOR * referencemv; | ||
| 138 : | edgomez | 960 | // _BITS stuff |
| 139 : | int16_t * dctSpace; | ||
| 140 : | syskin | 1016 | uint32_t iQuant; |
| 141 : | uint32_t quant_type; | ||
| 142 : | edgomez | 851 | |
| 143 : | } SearchData; | ||
| 144 : | |||
| 145 : | |||
| 146 : | typedef void(CheckFunc)(const int x, const int y, | ||
| 147 : | const int Direction, int * const dir, | ||
| 148 : | const SearchData * const Data); | ||
| 149 : | CheckFunc *CheckCandidate; | ||
| 150 : | |||
| 151 : | /* | ||
| 152 : | * Calculate the min/max range | ||
| 153 : | * relative to the _MACROBLOCK_ position | ||
| 154 : | */ | ||
| 155 : | static void __inline | ||
| 156 : | get_range(int32_t * const min_dx, | ||
| 157 : | int32_t * const max_dx, | ||
| 158 : | int32_t * const min_dy, | ||
| 159 : | int32_t * const max_dy, | ||
| 160 : | const uint32_t x, | ||
| 161 : | const uint32_t y, | ||
| 162 : | uint32_t block_sz, /* block dimension, 8 or 16 */ | ||
| 163 : | const uint32_t width, | ||
| 164 : | const uint32_t height, | ||
| 165 : | const uint32_t fcode, | ||
| 166 : | const int qpel, /* 1 if the resulting range should be in qpel precision; otherwise 0 */ | ||
| 167 : | const int rrv) | ||
| 168 : | { | ||
| 169 : | int k, m = qpel ? 4 : 2; | ||
| 170 : | const int search_range = 32 << (fcode - 1); | ||
| 171 : | int high = search_range - 1; | ||
| 172 : | int low = -search_range; | ||
| 173 : | |||
| 174 : | if (rrv) { | ||
| 175 : | high = RRV_MV_SCALEUP(high); | ||
| 176 : | low = RRV_MV_SCALEUP(low); | ||
| 177 : | block_sz *= 2; | ||
| 178 : | } | ||
| 179 : | |||
| 180 : | k = m * (int)(width - x * block_sz); | ||
| 181 : | *max_dx = MIN(high, k); | ||
| 182 : | k = m * (int)(height - y * block_sz); | ||
| 183 : | *max_dy = MIN(high, k); | ||
| 184 : | |||
| 185 : | k = -m * (int)((x+1) * block_sz); | ||
| 186 : | *min_dx = MAX(low, k); | ||
| 187 : | k = -m * (int)((y+1) * block_sz); | ||
| 188 : | *min_dy = MAX(low, k); | ||
| 189 : | } | ||
| 190 : | |||
| 191 : | typedef void MainSearchFunc(int x, int y, const SearchData * const Data, int bDirection); | ||
| 192 : | |||
| 193 : | static MainSearchFunc DiamondSearch, AdvDiamondSearch, SquareSearch; | ||
| 194 : | |||
| 195 : | static void Search8(const SearchData * const OldData, | ||
| 196 : | const int x, const int y, | ||
| 197 : | const uint32_t MotionFlags, | ||
| 198 : | const MBParam * const pParam, | ||
| 199 : | MACROBLOCK * const pMB, | ||
| 200 : | const MACROBLOCK * const pMBs, | ||
| 201 : | const int block, | ||
| 202 : | SearchData * const Data); | ||
| 203 : | |||
| 204 : | bool | ||
| 205 : | MotionEstimation(MBParam * const pParam, | ||
| 206 : | syskin | 884 | FRAMEINFO * const current, |
| 207 : | FRAMEINFO * const reference, | ||
| 208 : | const IMAGE * const pRefH, | ||
| 209 : | const IMAGE * const pRefV, | ||
| 210 : | const IMAGE * const pRefHV, | ||
| 211 : | const uint32_t iLimit); | ||
| 212 : | edgomez | 851 | |
| 213 : | static void | ||
| 214 : | SearchP(const IMAGE * const pRef, | ||
| 215 : | const uint8_t * const pRefH, | ||
| 216 : | const uint8_t * const pRefV, | ||
| 217 : | const uint8_t * const pRefHV, | ||
| 218 : | const IMAGE * const pCur, | ||
| 219 : | const int x, | ||
| 220 : | const int y, | ||
| 221 : | const uint32_t MotionFlags, | ||
| 222 : | const uint32_t GlobalFlags, | ||
| 223 : | SearchData * const Data, | ||
| 224 : | const MBParam * const pParam, | ||
| 225 : | const MACROBLOCK * const pMBs, | ||
| 226 : | const MACROBLOCK * const prevMBs, | ||
| 227 : | MACROBLOCK * const pMB); | ||
| 228 : | |||
| 229 : | |||
| 230 : | static WARPPOINTS | ||
| 231 : | GlobalMotionEst(const MACROBLOCK * const pMBs, | ||
| 232 : | const MBParam * const pParam, | ||
| 233 : | const FRAMEINFO * const current, | ||
| 234 : | const FRAMEINFO * const reference, | ||
| 235 : | const IMAGE * const pRefH, | ||
| 236 : | const IMAGE * const pRefV, | ||
| 237 : | const IMAGE * const pRefHV ); | ||
| 238 : | |||
| 239 : | #define iDiamondSize 2 | ||
| 240 : | |||
| 241 : | static __inline uint32_t | ||
| 242 : | MakeGoodMotionFlags(const uint32_t MotionFlags, const uint32_t GlobalFlags) | ||
| 243 : | { | ||
| 244 : | uint32_t Flags = MotionFlags; | ||
| 245 : | |||
| 246 : | if (!(GlobalFlags & XVID_MODEDECISION_BITS)) | ||
| 247 : | Flags &= ~(QUARTERPELREFINE16_BITS+QUARTERPELREFINE8_BITS+HALFPELREFINE16_BITS+HALFPELREFINE8_BITS+EXTSEARCH_BITS); | ||
| 248 : | |||
| 249 : | if (Flags & EXTSEARCH_BITS) | ||
| 250 : | Flags |= HALFPELREFINE16_BITS; | ||
| 251 : | |||
| 252 : | if (Flags & EXTSEARCH_BITS && MotionFlags & PMV_EXTSEARCH8) | ||
| 253 : | Flags |= HALFPELREFINE8_BITS; | ||
| 254 : | |||
| 255 : | if (Flags & HALFPELREFINE16_BITS) | ||
| 256 : | Flags |= QUARTERPELREFINE16_BITS; | ||
| 257 : | |||
| 258 : | if (Flags & HALFPELREFINE8_BITS) { | ||
| 259 : | Flags |= QUARTERPELREFINE8_BITS; | ||
| 260 : | Flags &= ~PMV_HALFPELREFINE8; | ||
| 261 : | } | ||
| 262 : | |||
| 263 : | if (Flags & QUARTERPELREFINE8_BITS) | ||
| 264 : | Flags &= ~PMV_QUARTERPELREFINE8; | ||
| 265 : | |||
| 266 : | if (!(GlobalFlags & XVID_QUARTERPEL)) | ||
| 267 : | Flags &= ~(PMV_QUARTERPELREFINE16+PMV_QUARTERPELREFINE8+QUARTERPELREFINE16_BITS+QUARTERPELREFINE8_BITS); | ||
| 268 : | |||
| 269 : | if (!(GlobalFlags & XVID_HALFPEL)) | ||
| 270 : | Flags &= ~(PMV_EXTSEARCH16+PMV_HALFPELREFINE16+PMV_HALFPELREFINE8+HALFPELREFINE16_BITS+HALFPELREFINE8_BITS); | ||
| 271 : | |||
| 272 : | if (GlobalFlags & (XVID_GREYSCALE + XVID_REDUCED)) | ||
| 273 : | Flags &= ~(PMV_CHROMA16 + PMV_CHROMA8); | ||
| 274 : | |||
| 275 : | return Flags; | ||
| 276 : | } | ||
| 277 : | |||
| 278 : | /* BITS mode decision and search */ | ||
| 279 : | |||
| 280 : | #include "../bitstream/zigzag.h" | ||
| 281 : | #include "../quant/quant_mpeg4.h" | ||
| 282 : | #include "../quant/quant_h263.h" | ||
| 283 : | #include "../bitstream/vlc_codes.h" | ||
| 284 : | syskin | 1016 | #include "../dct/fdct.h" |
| 285 : | edgomez | 851 | |
| 286 : | static int | ||
| 287 : | CountMBBitsInter(SearchData * const Data, | ||
| 288 : | const MACROBLOCK * const pMBs, const int x, const int y, | ||
| 289 : | const MBParam * const pParam, | ||
| 290 : | const uint32_t MotionFlags); | ||
| 291 : | |||
| 292 : | static int | ||
| 293 : | CountMBBitsInter4v(const SearchData * const Data, | ||
| 294 : | MACROBLOCK * const pMB, const MACROBLOCK * const pMBs, | ||
| 295 : | const int x, const int y, | ||
| 296 : | const MBParam * const pParam, const uint32_t MotionFlags, | ||
| 297 : | const VECTOR * const backup); | ||
| 298 : | |||
| 299 : | static int | ||
| 300 : | CountMBBitsIntra(const SearchData * const Data); | ||
| 301 : | |||
| 302 : | int CodeCoeffIntra_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag); | ||
| 303 : | int CodeCoeffInter_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag); | ||
| 304 : | syskin | 1016 | /* |
| 305 : | static int | ||
| 306 : | CountDistortionSkip(const SearchData * const Data); | ||
| 307 : | */ | ||
| 308 : | static __inline unsigned int | ||
| 309 : | Block_CalcBits(uint16_t * const coeff, | ||
| 310 : | uint16_t * const data, | ||
| 311 : | const uint32_t quant, const int quant_type, | ||
| 312 : | uint32_t * cbp, | ||
| 313 : | const int block, | ||
| 314 : | const int RD) | ||
| 315 : | { | ||
| 316 : | int sum; | ||
| 317 : | edgomez | 851 | |
| 318 : | syskin | 1016 | fdct(data); |
| 319 : | |||
| 320 : | if (quant_type == 0) sum = quant_inter(coeff, data, quant); | ||
| 321 : | else sum = quant4_inter(coeff, data, quant); | ||
| 322 : | |||
| 323 : | if (sum > 0) { | ||
| 324 : | *cbp |= 1 << (5 - block); | ||
| 325 : | return CodeCoeffInter_CalcBits(coeff, scan_tables[0]); | ||
| 326 : | } else return 0; | ||
| 327 : | } | ||
| 328 : | |||
| 329 : | /* RD experiment. ignore. | ||
| 330 : | static __inline unsigned int | ||
| 331 : | Block_CalcBits( int16_t * const coeff, | ||
| 332 : | int16_t * const data, | ||
| 333 : | const uint32_t quant, const int quant_type, | ||
| 334 : | uint32_t * cbp, | ||
| 335 : | const int block, | ||
| 336 : | const int RD) | ||
| 337 : | { | ||
| 338 : | int sum; | ||
| 339 : | int bits; | ||
| 340 : | const int lambda = quant*quant/2; | ||
| 341 : | int distortion = 0; | ||
| 342 : | |||
| 343 : | fdct(data); | ||
| 344 : | |||
| 345 : | if (quant_type == 0) sum = quant_inter(coeff, data, quant); | ||
| 346 : | else sum = quant4_inter(coeff, data, quant); | ||
| 347 : | |||
| 348 : | if (sum > 0) { | ||
| 349 : | *cbp |= 1 << (5 - block); | ||
| 350 : | bits = CodeCoeffInter_CalcBits(coeff, scan_tables[0]); | ||
| 351 : | } else bits = 0; | ||
| 352 : | |||
| 353 : | if (RD) { | ||
| 354 : | int i; | ||
| 355 : | if (quant_type == 0) dequant_inter_c(coeff, coeff, quant); | ||
| 356 : | else dequant4_inter_c(coeff, coeff, quant); | ||
| 357 : | |||
| 358 : | for (i = 0; i < 64; i++) { | ||
| 359 : | distortion += (data[i] - coeff[i])*(data[i] - coeff[i]); | ||
| 360 : | } | ||
| 361 : | } | ||
| 362 : | |||
| 363 : | bits += distortion/lambda; | ||
| 364 : | |||
| 365 : | return bits; | ||
| 366 : | } | ||
| 367 : | */ | ||
| 368 : | |||
| 369 : | edgomez | 851 | #endif /* _MOTION_EST_H_ */ |
| No admin address has been configured | ViewVC Help |
| Powered by ViewVC 1.0.4 |