--- trunk/xvidcore/src/motion/motion_est.h	2003/04/08 11:12:07	976
+++ trunk/xvidcore/src/motion/motion_est.h	2003/06/19 10:06:40	1072
@@ -26,7 +26,7 @@
  *  along with this program; if not, write to the Free Software
  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  *
- *  $Id: motion_est.h,v 1.6 2003-04-08 11:12:07 syskin Exp $
+ *  $Id: motion_est.h,v 1.9 2003-06-19 10:06:40 syskin Exp $
  *
  ***************************************************************************/
 
@@ -50,6 +50,8 @@
 #define NEIGH_TEND_8X8		40.0
 #define NEIGH_8X8_BIAS		30
 
+#define BITS_MULT			16
+
 /* Parameters which control inter/inter4v decision */
 #define IMV16X16			2
 
@@ -135,8 +137,10 @@
 	VECTOR directmvF[4];
 	VECTOR directmvB[4];
 	const VECTOR * referencemv;
-// _BITS stuff
+// BITS/R-D stuff
 	int16_t * dctSpace;
+	uint32_t iQuant;
+	uint32_t quant_type;
 
 } SearchData;
 
@@ -218,12 +222,10 @@
 		const int y,
 		const uint32_t MotionFlags,
 		const uint32_t GlobalFlags,
-		const uint32_t iQuant,
 		SearchData * const Data,
 		const MBParam * const pParam,
 		const MACROBLOCK * const pMBs,
 		const MACROBLOCK * const prevMBs,
-		int inter4v,
 		MACROBLOCK * const pMB);
 
 
@@ -263,6 +265,9 @@
 	if (Flags & QUARTERPELREFINE8_BITS)
 		Flags &= ~PMV_QUARTERPELREFINE8;
 
+	if (Flags & QUARTERPELREFINE16_BITS)
+		Flags &= ~PMV_QUARTERPELREFINE16;
+
 	if (!(GlobalFlags & XVID_QUARTERPEL))
 		Flags &= ~(PMV_QUARTERPELREFINE16+PMV_QUARTERPELREFINE8+QUARTERPELREFINE16_BITS+QUARTERPELREFINE8_BITS);
 
@@ -281,6 +286,7 @@
 #include "../quant/quant_mpeg4.h"
 #include "../quant/quant_h263.h"
 #include "../bitstream/vlc_codes.h"
+#include "../dct/fdct.h"
 
 static int
 CountMBBitsInter(SearchData * const Data,
@@ -301,4 +307,85 @@
 int CodeCoeffIntra_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag);
 int CodeCoeffInter_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag);
 
+#define LAMBDA		( (int)(1.0*BITS_MULT) )
+
+static __inline unsigned int
+Block_CalcBits(	int16_t * const coeff, 
+				int16_t * const data,
+				const uint32_t quant, const int quant_type,
+				uint32_t * cbp,
+				const int block)
+{
+	int sum;
+	int bits;
+	const int lambda = LAMBDA*quant*quant;
+	int distortion = 0;
+	int i;
+
+	fdct(data);
+
+	if (quant_type == 0) sum = quant_inter(coeff, data, quant);
+	else sum = quant4_inter(coeff, data, quant);
+
+	if (sum > 0) {
+		*cbp |= 1 << (5 - block);
+		bits = BITS_MULT * CodeCoeffInter_CalcBits(coeff, scan_tables[0]);
+	} else bits = 0;
+
+	if (quant_type == 0) dequant_inter(coeff, coeff, quant);
+	else dequant4_inter(coeff, coeff, quant);
+
+	for (i = 0; i < 64; i++) {
+		distortion += (data[i] - coeff[i])*(data[i] - coeff[i]);
+	}
+	bits += (BITS_MULT*BITS_MULT*distortion)/lambda;
+
+	return bits;
+}
+
+static __inline unsigned int
+Block_CalcBitsIntra(int16_t * const coeff, 
+					int16_t * const data,
+					const uint32_t quant, const int quant_type,
+					uint32_t * cbp,
+					const int block,
+					int * dcpred)
+{
+	int bits, i;
+	const int lambda = LAMBDA*quant*quant;
+	int distortion = 0;
+	uint32_t iDcScaler = get_dc_scaler(quant, block < 4);
+	int b_dc;
+
+	fdct(data);
+	data[0] -= 1024;
+
+	if (quant_type == 0) quant_intra_c(coeff, data, quant, iDcScaler); // MUST BE _C
+	else quant4_intra_c(coeff, data, quant, iDcScaler); // MUST BE _C
+
+	b_dc = coeff[0];
+	if (block < 4) {
+		coeff[0] -= *dcpred;
+		*dcpred = b_dc;
+	}
+
+	bits = BITS_MULT*CodeCoeffIntra_CalcBits(coeff, scan_tables[0]);
+	if (bits != 0) *cbp |= 1 << (5 - block);
+
+	if (block < 4) bits += BITS_MULT*dcy_tab[coeff[0] + 255].len;
+	else bits += BITS_MULT*dcc_tab[coeff[0] + 255].len;
+
+	coeff[0] = b_dc;
+	if (quant_type == 0) dequant_intra(coeff, coeff, quant, iDcScaler);
+	else dequant4_intra(coeff, coeff, quant, iDcScaler);
+
+	for (i = 0; i < 64; i++) {
+		distortion += (data[i] - coeff[i])*(data[i] - coeff[i]);
+	}
+
+	bits += (BITS_MULT*BITS_MULT*distortion)/lambda;
+
+	return bits;
+}
+
 #endif							/* _MOTION_EST_H_ */