/*************************************************************************This software module was originally developed by 	Yoshihiro Kikuchi (TOSHIBA CORPORATION)	Takeshi Nagai (TOSHIBA CORPORATION)    and edited by:	Toshiaki Watanabe (TOSHIBA CORPORATION)	Noboru Yamaguchi (TOSHIBA CORPORATION)	and also edited by 	Xuemin Chen (xchen@nlvl.com) Next Level System, Inc.	Bob Eifrig (beifrig@nlvl.com) Next Level Systems, Inc.    and also edited by 	Yoshinori Suzuki (Hitachi, Ltd.)    and also edited by 	Hideaki Kimata (NTT)    and also edited by    Fujitsu Laboratories Ltd. (contact: Eishi Morimatsu)  in the course of development of the <MPEG-4 Video(ISO/IEC 14496-2)>. This  software module is an implementation of a part of one or more <MPEG-4 Video  (ISO/IEC 14496-2)> tools as specified by the <MPEG-4 Video(ISO/IEC 14496-2)  >. ISO/IEC gives users of the <MPEG-4 Video(ISO/IEC 14496-2)> free license  to this software module or modifications thereof for use in hardware or  software products claiming conformance to the <MPEG-4 Video(ISO/IEC 14496-2  )>. Those intending to use this software module in hardware or software  products are advised that its use may infringe existing patents. The  original developer of this software module and his/her company, the  subsequent editors and their companies, and ISO/IEC have no liability for  use of this software module or modifications thereof in an implementation.  Copyright is not released for non <MPEG-4 Video(ISO/IEC 14496-2)>  conforming products. TOSHIBA CORPORATION retains full right to use the code  for his/her own purpose, assign or donate the code to a third party and to  inhibit third parties from using the code for non <MPEG-4 Video(ISO/IEC  14496-2)> conforming products. This copyright notice must be included in  all copies or derivative works.  Copyright (c)1997.Revision History:	May 9, 1999:	tm5 rate control by DemoGraFX, duhoff@mediaone.net	Aug.23, 1999:   NEWPRED added by Hideaki Kimata (NTT) 	Sep.06	1999 : RRV added by Eishi Morimatsu (Fujitsu Laboratories Ltd.) *************************************************************************/#include <stdio.h>#include <math.h>#include <stdlib.h>#include <iostream>#include "typeapi.h"#include "codehead.h"#include "bitstrm.hpp"#include "entropy.hpp"#include "huffman.hpp"#include "mode.hpp"#include "global.hpp"#include "vopses.hpp"#include "vopseenc.hpp"#include "tm5rc.hpp"    // added by mwi // NEWPRED#include "newpred.hpp"// ~NEWPRED// RRV insertion#include "rrv.hpp"// ~RRV#ifdef __MFC_#ifdef _DEBUG#undef THIS_FILEstatic char BASED_CODE THIS_FILE[] = __FILE__;#endif#define new DEBUG_NEW				   #endif // __MFC_// inline functionsinline Int clog2(Int a)	{return ( ((a)<=0) ? 0 : (int) ceil( log10( (double) (a) ) / log10(2.0) ) );}Void CVideoObjectEncoder::VideoPacketResetVOP(){	m_iVPCounter = m_pbitstrmOut -> getCounter();		// reset coding bits counter	assert(m_iNumMBX>0 && m_iNumMBY>0);// RRV modification	m_numBitsVPMBnum = clog2 (m_iNumMBX * m_iNumMBY *m_iRRVScale *m_iRRVScale);	// number of bits for macroblock_number//	m_numBitsVPMBnum = clog2 (m_iNumMBX * m_iNumMBY);	// number of bits for macroblock_number// ~RRV	m_numVideoPacket = 1;								// reset number of video packet in a VOP	m_iVPMBnum = 0;									// start MB in the 1st VP}Void CVideoObjectEncoder::codeVideoPacketHeader (	Int iMBX, Int iMBY, Int QuantScale){	m_iVPMBnum = VPMBnum(iMBX, iMBY);	m_statsVOP.nBitsHead += codeVideoPacketHeader(QuantScale);}#if 0 // revised HEC for shapeInt CVideoObjectEncoder::codeVideoPacketHeader (Int QuantScale){#ifdef __TRACE_AND_STATS_	m_pbitstrmOut->trace ("Video_Packet_Header\n");#endif // __TRACE_AND_STATS_	UInt nBits = 0;	assert(m_iVPMBnum > 0 && m_iVPMBnum < m_iNumMBX * m_iNumMBY);	assert(QuantScale > 0 && QuantScale < 32);	m_pbitstrmOut->flush ();	Int nBitsResyncMarker = NUMBITS_VP_RESYNC_MARKER;	if(m_volmd.bShapeOnly==FALSE)	{		if(m_vopmd.vopPredType == PVOP || (m_uiSprite == 2 && m_vopmd.vopPredType == SPRITE)) // GMC			nBitsResyncMarker += (m_vopmd.mvInfoForward.uiFCode - 1);		else if(m_vopmd.vopPredType == BVOP)			nBitsResyncMarker += max(m_vopmd.mvInfoForward.uiFCode, m_vopmd.mvInfoBackward.uiFCode) - 1;	}	m_pbitstrmOut -> putBits (RESYNC_MARKER, nBitsResyncMarker, "resync_marker");	m_pbitstrmOut -> putBits (m_iVPMBnum, m_numBitsVPMBnum, "VPH_macroblock_number");	if(m_volmd.bShapeOnly==FALSE) {		m_pbitstrmOut -> putBits (QuantScale, NUMBITS_VP_QUANTIZER, "VPH_quant_scale");	}	Bool	HEC = ((++m_numVideoPacket) == 2);	// use HEC in 2nd VP	m_pbitstrmOut -> putBits (HEC, NUMBITS_VP_HEC, "VPH_header_extension_code");#ifdef __TRACE_AND_STATS_	nBits += nBitsResyncMarker + m_numBitsVPMBnum + NUMBITS_VP_QUANTIZER + NUMBITS_VP_HEC;#endif // __TRACE_AND_STATS_	if( HEC == TRUE ) {		m_pbitstrmOut -> putBits ((Int) 0xFFFFFFFE, (UInt) m_nBitsModuloBase + 1, "VPH_Modulo_Time_Base");		m_pbitstrmOut->putBits ((Int) 1, 1, "Marker"); // marker bit		if(m_iNumBitsTimeIncr!=0)			m_pbitstrmOut -> putBits (m_iVopTimeIncr, m_iNumBitsTimeIncr, "VPH_Time_Incr");		m_pbitstrmOut->putBits ((Int) 1, 1, "Marker"); // marker bit		m_pbitstrmOut -> putBits (m_vopmd.vopPredType, NUMBITS_VP_PRED_TYPE, "VPH_Pred_Type");#ifdef __TRACE_AND_STATS_		nBits += m_nBitsModuloBase+1 + m_iNumBitsTimeIncr +2 + NUMBITS_VP_PRED_TYPE + NUMBITS_VP_INTRA_DC_SWITCH_THR;#endif // __TRACE_AND_STATS_		if(m_volmd.bShapeOnly==FALSE) {			m_pbitstrmOut -> putBits (m_vopmd.iIntraDcSwitchThr, NUMBITS_VP_INTRA_DC_SWITCH_THR, "VOP_DC_VLC_Switch");// GMC 			if ((m_uiSprite == 2) && (m_vopmd.vopPredType == SPRITE) && (m_iNumOfPnts>0))#ifdef __TRACE_AND_STATS_ 				nBits += codeWarpPoints ();#else 				codeWarpPoints ();#endif// ~GMC// RRV insertion			if((m_volmd.breduced_resolution_vop_enable == 1)&&(m_volmd.fAUsage == RECTANGLE)&&((m_vopmd.vopPredType == PVOP)||(m_vopmd.vopPredType == IVOP)))			{				m_pbitstrmOut -> putBits (m_vopmd.RRVmode.iRRVOnOff, 1, "VOP_Reduced_Resolution");#ifdef __TRACE_AND_STATS_				nBits ++;#endif // __TRACE_AND_STATS_			}	// ~RRV			if(m_vopmd.vopPredType != IVOP) {				m_pbitstrmOut -> putBits (m_vopmd.mvInfoForward.uiFCode, NUMBITS_VOP_FCODE, "VPH_Fcode_Forward");#ifdef __TRACE_AND_STATS_				nBits += NUMBITS_VOP_FCODE;#endif // __TRACE_AND_STATS_				if (m_vopmd.vopPredType == BVOP) {					m_pbitstrmOut -> putBits (m_vopmd.mvInfoBackward.uiFCode, NUMBITS_VOP_FCODE, "VOP_Fcode_Backward");#ifdef __TRACE_AND_STATS_					nBits += NUMBITS_VOP_FCODE;#endif // __TRACE_AND_STATS_				}			}		}	}// NEWPRED	if (m_volmd.bNewpredEnable) {		g_pNewPredEnc->SetVPData(				NP_VP_HEADER,				&m_vopmd.m_iVopID,					&m_vopmd.m_iNumBitsVopID,				&m_vopmd.m_iVopID4Prediction_Indication,				&m_vopmd.m_iVopID4Prediction		);		m_pbitstrmOut -> putBits(m_vopmd.m_iVopID, m_vopmd.m_iNumBitsVopID, "VOP_Id");		m_pbitstrmOut->putBits(m_vopmd.m_iVopID4Prediction_Indication,					NUMBITS_VOP_ID_FOR_PREDICTION_INDICATION, "VOP_Id_For_Prediction_Indication");		if( m_vopmd.m_iVopID4Prediction_Indication )		{			m_pbitstrmOut->putBits(m_vopmd.m_iVopID4Prediction, m_vopmd.m_iNumBitsVopID, "VOP_Id_For_Prediction");#ifdef __TRACE_AND_STATS_			nBits += m_vopmd.m_iNumBitsVopID;#endif // __TRACE_AND_STATS_		}		m_pbitstrmOut->putBits(MARKER_BIT, MARKER_BIT, "Marker_Bit");#ifdef __TRACE_AND_STATS_		nBits += m_vopmd.m_iNumBitsVopID;		nBits += NUMBITS_VOP_ID_FOR_PREDICTION_INDICATION;		nBits += MARKER_BIT;#endif // __TRACE_AND_STATS_		}// ~NEWPRED	return nBits;}#elseInt CVideoObjectEncoder::codeVideoPacketHeader (Int QuantScale){#ifdef __TRACE_AND_STATS_	m_pbitstrmOut->trace ("Video_Packet_Header\n");#endif // __TRACE_AND_STATS_	UInt nBits = 0;	assert(m_iVPMBnum > 0 && m_iVPMBnum < m_iNumMBX * m_iNumMBY);	assert(QuantScale > 0 && QuantScale < 32);	m_pbitstrmOut->flush ();	Int nBitsResyncMarker = NUMBITS_VP_RESYNC_MARKER;	Bool	HEC = ((++m_numVideoPacket) == 2);	// use HEC in 2nd VP	if(m_volmd.bShapeOnly==FALSE)	{		if(m_vopmd.vopPredType == PVOP || (m_uiSprite == 2 && m_vopmd.vopPredType == SPRITE)) // GMC_V2			nBitsResyncMarker += (m_vopmd.mvInfoForward.uiFCode - 1);		else if(m_vopmd.vopPredType == BVOP)			nBitsResyncMarker += max(m_vopmd.mvInfoForward.uiFCode, m_vopmd.mvInfoBackward.uiFCode) - 1;	}	m_pbitstrmOut -> putBits (RESYNC_MARKER, nBitsResyncMarker, "resync_marker");	if(m_volmd.fAUsage != RECTANGLE) {		m_pbitstrmOut -> putBits (HEC, NUMBITS_VP_HEC, "VPH_header_extension_code");		if (HEC == TRUE && !(m_uiSprite == 1 && m_vopmd.vopPredType == IVOP)) {			m_pbitstrmOut->putBits (m_rctCurrVOPY.width, NUMBITS_VOP_WIDTH, "VPH_VOP_Width");			m_pbitstrmOut->putBits ((Int) 1, 1, "Marker"); // marker bit			m_pbitstrmOut->putBits (m_rctCurrVOPY.height (), NUMBITS_VOP_HEIGHT, "VPH_VOP_Height");			m_pbitstrmOut->putBits ((Int) 1, 1, "Marker"); // marker bit#ifdef __TRACE_AND_STATS_			nBits += NUMBITS_VOP_WIDTH + NUMBITS_VOP_HEIGHT + 2;#endif // __TRACE_AND_STATS_			if (m_rctCurrVOPY.left >= 0) {				m_pbitstrmOut -> putBits (m_rctCurrVOPY.left, NUMBITS_VOP_HORIZONTAL_SPA_REF, "VPH_VOP_Hori_Ref");			} else {				m_pbitstrmOut -> putBits (m_rctCurrVOPY.left & 0x00001FFFF, NUMBITS_VOP_HORIZONTAL_SPA_REF, "VPH_VOP_Hori_Ref");			}			m_pbitstrmOut->putBits ((Int) 1, 1, "Marker"); // marker bit			if (m_rctCurrVOPY.top >= 0) {				m_pbitstrmOut -> putBits (m_rctCurrVOPY.top, NUMBITS_VOP_VERTICAL_SPA_REF, "VPH_VOP_Vert_Ref");			} else {				m_pbitstrmOut -> putBits (m_rctCurrVOPY.top & 0x00001FFFF, NUMBITS_VOP_VERTICAL_SPA_REF, "VPH_VOP_Vert_Ref");			}			m_pbitstrmOut->putBits ((Int) 1, 1, "Marker"); // marker bit#ifdef __TRACE_AND_STATS_			nBits += NUMBITS_VOP_HORIZONTAL_SPA_REF + NUMBITS_VOP_VERTICAL_SPA_REF + 2;#endif // __TRACE_AND_STATS_		}	}	m_pbitstrmOut -> putBits (m_iVPMBnum, m_numBitsVPMBnum, "VPH_macroblock_number");	if(m_volmd.bShapeOnly==FALSE) {		m_pbitstrmOut -> putBits (QuantScale, NUMBITS_VP_QUANTIZER, "VPH_quant_scale");	}	if(m_volmd.fAUsage == RECTANGLE)		m_pbitstrmOut -> putBits (HEC, NUMBITS_VP_HEC, "VPH_header_extension_code");#ifdef __TRACE_AND_STATS_	nBits += nBitsResyncMarker + m_numBitsVPMBnum + NUMBITS_VP_QUANTIZER + NUMBITS_VP_HEC;#endif // __TRACE_AND_STATS_	if( HEC == TRUE ) {		m_pbitstrmOut -> putBits ((Int) 0xFFFFFFFE, (UInt) m_nBitsModuloBase + 1, "VPH_Modulo_Time_Base");		m_pbitstrmOut->putBits ((Int) 1, 1, "Marker"); // marker bit		if(m_iNumBitsTimeIncr!=0)			m_pbitstrmOut -> putBits (m_iVopTimeIncr, m_iNumBitsTimeIncr, "VPH_Time_Incr");		m_pbitstrmOut->putBits ((Int) 1, 1, "Marker"); // marker bit		m_pbitstrmOut -> putBits (m_vopmd.vopPredType, NUMBITS_VP_PRED_TYPE, "VPH_Pred_Type");#ifdef __TRACE_AND_STATS_		nBits += m_nBitsModuloBase+1 + m_iNumBitsTimeIncr +2 + NUMBITS_VP_PRED_TYPE + NUMBITS_VP_INTRA_DC_SWITCH_THR;#endif // __TRACE_AND_STATS_		if(m_volmd.fAUsage != RECTANGLE) {			m_pbitstrmOut->putBits (m_volmd.bNoCrChange, 1, "VPH_VOP_CR_Change_Disable");#ifdef __TRACE_AND_STATS_			nBits ++;#endif // __TRACE_AND_STATS_			if (m_volmd.bShapeOnly==FALSE && m_vopmd.vopPredType != IVOP)  			{  				m_pbitstrmOut -> putBits (m_vopmd.bShapeCodingType, 1, "VPH_shape_coding_type");#ifdef __TRACE_AND_STATS_				nBits ++;#endif // __TRACE_AND_STATS_			}		}		if(m_volmd.bShapeOnly==FALSE) {			m_pbitstrmOut -> putBits (m_vopmd.iIntraDcSwitchThr, NUMBITS_VP_INTRA_DC_SWITCH_THR, "VOP_DC_VLC_Switch");// GMC_V2			if ((m_uiSprite == 2) && (m_vopmd.vopPredType == SPRITE) && (m_iNumOfPnts>0))#ifdef __TRACE_AND_STATS_				nBits += codeWarpPoints ();#else				codeWarpPoints ();#endif// ~GMC_V2// RRV insertion			if((m_volmd.breduced_resolution_vop_enable == 1)&&(m_volmd.fAUsage == RECTANGLE)&&((m_vopmd.vopPredType == PVOP)||(m_vopmd.vopPredType == IVOP)))			{				m_pbitstrmOut -> putBits (m_vopmd.RRVmode.iRRVOnOff, 1, "VOP_Reduced_Resolution");#ifdef __TRACE_AND_STATS_				nBits ++;#endif // __TRACE_AND_STATS_			}	// ~RRV			if(m_vopmd.vopPredType != IVOP) {				m_pbitstrmOut -> putBits (m_vopmd.mvInfoForward.uiFCode, NUMBITS_VOP_FCODE, "VPH_Fcode_Forward");#ifdef __TRACE_AND_STATS_				nBits += NUMBITS_VOP_FCODE;#endif // __TRACE_AND_STATS_				if (m_vopmd.vopPredType == BVOP) {					m_pbitstrmOut -> putBits (m_vopmd.mvInfoBackward.uiFCode, NUMBITS_VOP_FCODE, "VOP_Fcode_Backward");#ifdef __TRACE_AND_STATS_					nBits += NUMBITS_VOP_FCODE;#endif // __TRACE_AND_STATS_				}			}		}	}// NEWPRED	if (m_volmd.bNewpredEnable) {		g_pNewPredEnc->SetVPData(				NP_VP_HEADER,				&m_vopmd.m_iVopID,					&m_vopmd.m_iNumBitsVopID,				&m_vopmd.m_iVopID4Prediction_Indication,				&m_vopmd.m_iVopID4Prediction		);		m_pbitstrmOut -> putBits(m_vopmd.m_iVopID, m_vopmd.m_iNumBitsVopID, "VOP_Id");		m_pbitstrmOut->putBits(m_vopmd.m_iVopID4Prediction_Indication,					NUMBITS_VOP_ID_FOR_PREDICTION_INDICATION, "VOP_Id_For_Prediction_Indication");		if( m_vopmd.m_iVopID4Prediction_Indication )		{			m_pbitstrmOut->putBits(m_vopmd.m_iVopID4Prediction, m_vopmd.m_iNumBitsVopID, "VOP_Id_For_Prediction");#ifdef __TRACE_AND_STATS_			nBits += m_vopmd.m_iNumBitsVopID;#endif // __TRACE_AND_STATS_		}		m_pbitstrmOut->putBits(MARKER_BIT, MARKER_BIT, "Marker_Bit");#ifdef __TRACE_AND_STATS_		nBits += m_vopmd.m_iNumBitsVopID;		nBits += NUMBITS_VOP_ID_FOR_PREDICTION_INDICATION;		nBits += MARKER_BIT;#endif // __TRACE_AND_STATS_	}// ~NEWPRED	return nBits;}#endifUInt CVideoObjectEncoder::encodeMVVP (const CMotionVector* pmv, const CMBMode* pmbmd, Int iMBX, Int iMBY){	Int iMBnum = iMBY * m_iNumMBX + iMBX;	assert(iMBnum >= m_iVPMBnum);	return encodeMV (pmv, pmbmd, bVPNoLeft(iMBnum, iMBX), bVPNoRightTop(iMBnum, iMBX), bVPNoTop(iMBnum),		iMBX, iMBY);}//////////////////////////////////////////////////////////////////////  The following functions are for Data partitioning mode  //////////////////////////////////////////////////////////////////////Void CVideoObjectEncoder::encodeNSForIVOP_DP ()	{	assert( m_volmd.bDataPartitioning );	assert( m_vopmd.vopPredType==IVOP );	memset (m_rgmv, 0, m_iNumMB * PVOP_MV_PER_REF_PER_MB * sizeof (CMotionVector));	CMBMode* pmbmd = m_rgmbmd;