📄 errenc.cpp
字号:
/*************************************************************************
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.h>
#include "typeapi.h"
#include "codehead.h"
#include "entropy/bitstrm.hpp"
#include "entropy/entropy.hpp"
#include "entropy/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_FILE
static char BASED_CODE THIS_FILE[] = __FILE__;
#endif
#define new DEBUG_NEW
#endif // __MFC_
// inline functions
inline 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 shape
Int 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;
}
#else
Int 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;
}
#endif
UInt CVideoObjectEncoder::encodeMVVP (const CMotionVector* pmv, const CMBMode* pmbmd, Int iMBX, Int iMBY)
{
Int iMBnum = iMBY * m_iNumMBX + iMBX;
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -