121 |
uint32_t iEdgedWidth; |
uint32_t iEdgedWidth; |
122 |
uint32_t iFcode; |
uint32_t iFcode; |
123 |
int * temp; |
int * temp; |
124 |
|
int * dir; |
125 |
int qpel, qpel_precision; |
int qpel, qpel_precision; |
126 |
int chroma; |
int chroma; |
127 |
int rrv; |
int rrv; |
146 |
|
|
147 |
|
|
148 |
typedef void(CheckFunc)(const int x, const int y, |
typedef void(CheckFunc)(const int x, const int y, |
149 |
const int Direction, int * const dir, |
const SearchData * const Data, |
150 |
const SearchData * const Data); |
const int Direction); |
|
CheckFunc *CheckCandidate; |
|
151 |
|
|
152 |
/* |
/* |
153 |
* Calculate the min/max range |
* Calculate the min/max range |
160 |
int32_t * const max_dy, |
int32_t * const max_dy, |
161 |
const uint32_t x, |
const uint32_t x, |
162 |
const uint32_t y, |
const uint32_t y, |
163 |
uint32_t block_sz, /* block dimension, 8 or 16 */ |
uint32_t block_sz, /* block dimension, 3(8) or 4(16) */ |
164 |
const uint32_t width, |
const uint32_t width, |
165 |
const uint32_t height, |
const uint32_t height, |
166 |
const uint32_t fcode, |
const uint32_t fcode, |
167 |
const int qpel, /* 1 if the resulting range should be in qpel precision; otherwise 0 */ |
const int precision, /* 2 for qpel, 1 for halfpel */ |
168 |
const int rrv) |
const int rrv) |
169 |
{ |
{ |
170 |
int k, m = qpel ? 4 : 2; |
int k; |
171 |
const int search_range = 32 << (fcode - 1); |
const int search_range = 16 << fcode; |
172 |
int high = search_range - 1; |
int high = search_range - 1; |
173 |
int low = -search_range; |
int low = -search_range; |
174 |
|
|
175 |
if (rrv) { |
if (rrv) { |
176 |
high = RRV_MV_SCALEUP(high); |
high = RRV_MV_SCALEUP(high); |
177 |
low = RRV_MV_SCALEUP(low); |
low = RRV_MV_SCALEUP(low); |
178 |
block_sz *= 2; |
block_sz++; |
179 |
} |
} |
180 |
|
|
181 |
k = m * (int)(width - x * block_sz); |
k = (int)(width - (x<<block_sz))<<precision; |
182 |
*max_dx = MIN(high, k); |
*max_dx = MIN(high, k); |
183 |
k = m * (int)(height - y * block_sz); |
k = (int)(height - (y<<block_sz))<<precision; |
184 |
*max_dy = MIN(high, k); |
*max_dy = MIN(high, k); |
185 |
|
|
186 |
k = -m * (int)((x+1) * block_sz); |
k = (-(int)((x+1)<<block_sz))<<precision; |
187 |
*min_dx = MAX(low, k); |
*min_dx = MAX(low, k); |
188 |
k = -m * (int)((y+1) * block_sz); |
k = (-(int)((y+1)<<block_sz))<<precision; |
189 |
*min_dy = MAX(low, k); |
*min_dy = MAX(low, k); |
190 |
} |
} |
191 |
|
|
192 |
typedef void MainSearchFunc(int x, int y, const SearchData * const Data, int bDirection); |
typedef void |
193 |
|
MainSearchFunc(int x, int y, const SearchData * const Data, |
194 |
|
int bDirection, CheckFunc * const CheckCandidate); |
195 |
|
|
196 |
static MainSearchFunc DiamondSearch, AdvDiamondSearch, SquareSearch; |
static MainSearchFunc DiamondSearch, AdvDiamondSearch, SquareSearch; |
197 |
|
|
335 |
#include "../dct/fdct.h" |
#include "../dct/fdct.h" |
336 |
|
|
337 |
static int |
static int |
338 |
CountMBBitsInter(SearchData * const Data, |
findRDinter(SearchData * const Data, |
339 |
const MACROBLOCK * const pMBs, const int x, const int y, |
const MACROBLOCK * const pMBs, const int x, const int y, |
340 |
const MBParam * const pParam, |
const MBParam * const pParam, |
341 |
const uint32_t MotionFlags); |
const uint32_t MotionFlags); |
342 |
|
|
343 |
static int |
static int |
344 |
CountMBBitsInter4v(const SearchData * const Data, |
findRDinter4v(const SearchData * const Data, |
345 |
MACROBLOCK * const pMB, const MACROBLOCK * const pMBs, |
MACROBLOCK * const pMB, const MACROBLOCK * const pMBs, |
346 |
const int x, const int y, |
const int x, const int y, |
347 |
const MBParam * const pParam, const uint32_t MotionFlags, |
const MBParam * const pParam, const uint32_t MotionFlags, |
348 |
const VECTOR * const backup); |
const VECTOR * const backup); |
349 |
|
|
350 |
static int |
static int |
351 |
CountMBBitsIntra(const SearchData * const Data); |
findRDintra(const SearchData * const Data); |
352 |
|
|
353 |
static int |
static int |
354 |
CountMBBitsGMC(const SearchData * const Data, const IMAGE * const vGMC, const int x, const int y); |
findRDgmc(const SearchData * const Data, const IMAGE * const vGMC, const int x, const int y); |
355 |
|
|
356 |
int CodeCoeffIntra_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag); |
int CodeCoeffIntra_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag); |
357 |
int CodeCoeffInter_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag); |
int CodeCoeffInter_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag); |
393 |
distortion += data[i]*data[i]; |
distortion += data[i]*data[i]; |
394 |
} |
} |
395 |
|
|
396 |
|
return bits + (LAMBDA*distortion)/(quant*quant); |
|
bits += (LAMBDA*distortion)/(quant*quant); |
|
|
|
|
|
return bits; |
|
397 |
} |
} |
398 |
|
|
399 |
static __inline unsigned int |
static __inline unsigned int |
432 |
if (quant_type) dequant_intra(dqcoeff, coeff, quant, iDcScaler); |
if (quant_type) dequant_intra(dqcoeff, coeff, quant, iDcScaler); |
433 |
else dequant4_intra(dqcoeff, coeff, quant, iDcScaler); |
else dequant4_intra(dqcoeff, coeff, quant, iDcScaler); |
434 |
|
|
435 |
for (i = 0; i < 64; i++) { |
for (i = 0; i < 64; i++) |
436 |
distortion += (data[i] - dqcoeff[i])*(data[i] - dqcoeff[i]); |
distortion += (data[i] - dqcoeff[i])*(data[i] - dqcoeff[i]); |
|
} |
|
|
|
|
|
bits += (LAMBDA*distortion)/(quant*quant); |
|
437 |
|
|
438 |
return bits; |
return bits + (LAMBDA*distortion)/(quant*quant); |
439 |
} |
} |
440 |
|
|
441 |
#endif /* _MOTION_EST_H_ */ |
#endif /* _MOTION_EST_H_ */ |