--- branches/dev-api-4/xvidcore/src/motion/motion_est.h	2003/04/05 16:47:44	974
+++ branches/dev-api-4/xvidcore/src/motion/motion_est.h	2003/05/14 20:23:02	1023
@@ -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.3.2.4 2003-04-05 16:47:44 edgomez Exp $
+ *  $Id: motion_est.h,v 1.3.2.7 2003-05-14 20:23:02 edgomez 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
 
@@ -114,12 +116,7 @@
 	VECTOR * currentMV;
 	VECTOR * currentQMV;
 	int32_t * iMinSAD;
-	const uint8_t * Ref;
-	const uint8_t * RefH;
-	const uint8_t * RefV;
-	const uint8_t * RefHV;
-	const uint8_t * RefCU;
-	const uint8_t * RefCV;
+	const uint8_t * RefP[6]; // N, V, H, HV, cU, cV
 	const uint8_t * CurU;
 	const uint8_t * CurV;
 	uint8_t * RefQ;
@@ -133,21 +130,17 @@
 	int chroma;
 	int rrv;
 //fields for interpolate and direct modes
-	const uint8_t * bRef;
-	const uint8_t * bRefH;
-	const uint8_t * bRefV;
-	const uint8_t * bRefHV;
-	const uint8_t * b_RefCU;
-	const uint8_t * b_RefCV;
-
+	const uint8_t * b_RefP[6]; // N, V, H, HV, cU, cV
 	VECTOR bpredMV;
 	uint32_t bFcode;
 // fields for direct mode
 	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;
 
@@ -228,16 +221,14 @@
 		const int x,
 		const int y,
 		const uint32_t MotionFlags,
-		const uint32_t GlobalFlags,
-		const uint32_t iQuant,
+		const uint32_t VopFlags,
+		const uint32_t VolFlags,
 		SearchData * const Data,
 		const MBParam * const pParam,
 		const MACROBLOCK * const pMBs,
 		const MACROBLOCK * const prevMBs,
-		int inter4v,
 		MACROBLOCK * const pMB);
 
-
 static WARPPOINTS
 GlobalMotionEst(const MACROBLOCK * const pMBs, 
 				const MBParam * const pParam,
@@ -292,6 +283,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,
@@ -312,4 +304,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 > 3);
+	int b_dc;
+
+	fdct(data);
+	data[0] -= 1024;
+
+	if (quant_type == 0) quant_intra(coeff, data, quant, iDcScaler);
+	else quant4_intra(coeff, data, quant, iDcScaler);
+
+	b_dc = coeff[0];
+	if (block < 4) {
+		coeff[0] -= *dcpred;
+		*dcpred = b_dc;
+	}
+
+	*cbp |= 1 << (5 - block);
+	bits = BITS_MULT*CodeCoeffIntra_CalcBits(coeff, scan_tables[0]);
+	bits += BITS_MULT*dcy_tab[coeff[0] + 255].len;
+	if (bits != 0) *cbp |= 1 << (5 - block);
+
+	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_ */