21 |
* along with this program ; if not, write to the Free Software |
* along with this program ; if not, write to the Free Software |
22 |
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
23 |
* |
* |
24 |
* $Id: motion_est.h,v 1.3.2.14 2003-08-02 15:08:45 edgomez Exp $ |
* $Id: motion_est.h,v 1.3.2.17 2003-08-26 09:25:25 edgomez Exp $ |
25 |
* |
* |
26 |
****************************************************************************/ |
****************************************************************************/ |
27 |
|
|
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 |
|
|
333 |
#include "../dct/fdct.h" |
#include "../dct/fdct.h" |
334 |
|
|
335 |
static int |
static int |
336 |
CountMBBitsInter(SearchData * const Data, |
findRDinter(SearchData * const Data, |
337 |
const MACROBLOCK * const pMBs, const int x, const int y, |
const MACROBLOCK * const pMBs, const int x, const int y, |
338 |
const MBParam * const pParam, |
const MBParam * const pParam, |
339 |
const uint32_t MotionFlags); |
const uint32_t MotionFlags); |
340 |
|
|
341 |
static int |
static int |
342 |
CountMBBitsInter4v(const SearchData * const Data, |
findRDinter4v(const SearchData * const Data, |
343 |
MACROBLOCK * const pMB, const MACROBLOCK * const pMBs, |
MACROBLOCK * const pMB, const MACROBLOCK * const pMBs, |
344 |
const int x, const int y, |
const int x, const int y, |
345 |
const MBParam * const pParam, const uint32_t MotionFlags, |
const MBParam * const pParam, const uint32_t MotionFlags, |
346 |
const VECTOR * const backup); |
const VECTOR * const backup); |
347 |
|
|
348 |
static int |
static int |
349 |
CountMBBitsIntra(const SearchData * const Data); |
findRDintra(const SearchData * const Data); |
350 |
|
|
351 |
static int |
static int |
352 |
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); |
353 |
|
|
354 |
int CodeCoeffIntra_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag); |
int CodeCoeffIntra_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag); |
355 |
int CodeCoeffInter_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag); |
int CodeCoeffInter_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag); |
391 |
distortion += data[i]*data[i]; |
distortion += data[i]*data[i]; |
392 |
} |
} |
393 |
|
|
394 |
|
return bits + (LAMBDA*distortion)/(quant*quant); |
|
bits += (LAMBDA*distortion)/(quant*quant); |
|
|
|
|
|
return bits; |
|
395 |
} |
} |
396 |
|
|
397 |
static __inline unsigned int |
static __inline unsigned int |
430 |
if (quant_type) dequant_intra(dqcoeff, coeff, quant, iDcScaler); |
if (quant_type) dequant_intra(dqcoeff, coeff, quant, iDcScaler); |
431 |
else dequant4_intra(dqcoeff, coeff, quant, iDcScaler); |
else dequant4_intra(dqcoeff, coeff, quant, iDcScaler); |
432 |
|
|
433 |
for (i = 0; i < 64; i++) { |
for (i = 0; i < 64; i++) |
434 |
distortion += (data[i] - dqcoeff[i])*(data[i] - dqcoeff[i]); |
distortion += (data[i] - dqcoeff[i])*(data[i] - dqcoeff[i]); |
|
} |
|
|
|
|
|
bits += (LAMBDA*distortion)/(quant*quant); |
|
435 |
|
|
436 |
return bits; |
return bits + (LAMBDA*distortion)/(quant*quant); |
437 |
} |
} |
438 |
|
|
439 |
#endif /* _MOTION_EST_H_ */ |
#endif /* _MOTION_EST_H_ */ |