gopencoder.cpp
来自「SVC最新更新代码」· C++ 代码 · 共 1,714 行 · 第 1/5 页
CPP
1,714 行
{
RNOK( m_pcRedundantCtrl->uninit() );
delete m_pcRedundantCtrl;
m_pcRedundantCtrl = 0 ;
}
if(m_pcRedundant1Ctrl)
{
RNOK( m_pcRedundant1Ctrl->uninit() );
delete m_pcRedundant1Ctrl;
m_pcRedundant1Ctrl = 0 ;
}
//RPIC bug fix
if( m_pcBaseLayerCtrl )
{
RNOK( m_pcBaseLayerCtrl->uninit() );
delete m_pcBaseLayerCtrl;
m_pcBaseLayerCtrl = 0;
}
if( m_pcBaseLayerCtrlField )
{
RNOK( m_pcBaseLayerCtrlField->uninit() );
delete m_pcBaseLayerCtrlField;
m_pcBaseLayerCtrlField = 0;
}
//========== DELETE UPDATE WEIGHTS ARRAY and WRITE BUFFER ==========
delete [] m_pucWriteBuffer;
m_pucWriteBuffer = 0;
m_uiWriteBufferSize = 0;
//S051{
delete[] m_auiFrameBits;
//S051}
return Err::m_nOK;
}
ErrVal
LayerEncoder::uninit()
{
m_bInitDone = false;
xDeleteData();
if( m_pMotionInfoFile )
{
::fclose( m_pMotionInfoFile );
}
if( m_pESSFile )
{
::fclose( m_pESSFile );
}
// JVT-S054 (ADD) ->
if( m_puiFirstMbInSlice )
{
free(m_puiFirstMbInSlice);
m_puiFirstMbInSlice = 0;
}
// JVT-S054 (ADD) <-
delete [] m_apabBaseModeFlagAllowedArrays[0];
delete [] m_apabBaseModeFlagAllowedArrays[1];
m_apabBaseModeFlagAllowedArrays[0] = 0;
m_apabBaseModeFlagAllowedArrays[1] = 0;
return Err::m_nOK;
}
ErrVal
LayerEncoder::xFillAndUpsampleFrame( Frame* pcFrame, PicType ePicType, Bool bFrameMbsOnlyFlag )
{
RNOK( m_pcYuvFullPelBufferCtrl->initMb() );
RNOK( m_pcYuvHalfPelBufferCtrl->initMb() );
if( ! pcFrame->isHalfPel() )
{
XPel* pHPData = NULL;
RNOK( pcFrame->initHalfPel( pHPData ) );
}
RNOK( pcFrame->extendFrame( m_pcQuarterPelFilter, ePicType, bFrameMbsOnlyFlag ) );
return Err::m_nOK;
}
ErrVal
LayerEncoder::xFillAndExtendFrame( Frame* pcFrame, PicType ePicType, Bool bFrameMbsOnlyFlag )
{
RNOK( m_pcYuvFullPelBufferCtrl->initMb() );
RNOK( pcFrame->extendFrame( NULL, ePicType, bFrameMbsOnlyFlag ) );
return Err::m_nOK;
}
ErrVal
LayerEncoder::xMotionEstimation( RefFrameList* pcRefFrameList0,
RefFrameList* pcRefFrameList1,
MbDataCtrl* pcMbDataCtrlCol,
Frame* pcOrigFrame,
Frame* pcIntraRecFrame,
ControlData& rcControlData,
UInt uiNumMaxIter,
UInt uiIterSearchRange,
UInt uiFrameIdInGOP,
PicType ePicType )
{
MbEncoder* pcMbEncoder = m_pcSliceEncoder->getMbEncoder ();
SliceHeader& rcSliceHeader = *rcControlData.getSliceHeader ( ePicType );
MbDataCtrl* pcMbDataCtrl = rcControlData.getMbDataCtrl ();
Frame* pcBaseLayerFrame = rcControlData.getBaseLayerRec ();
Frame* pcBaseLayerResidual = rcControlData.getBaseLayerSbb ();
MbDataCtrl* pcBaseLayerCtrl = rcControlData.getBaseLayerCtrl ();
Bool bEstimateBase = rcSliceHeader.getNoInterLayerPredFlag() && ! pcBaseLayerCtrl;
Bool bEstimateMotion = rcSliceHeader.getAdaptiveBaseModeFlag() || bEstimateBase;
UInt uiMaxMvPerMb = rcSliceHeader.getSPS().getMaxMVsPer2Mb () >> 1; // hard limit (don't take into account last macroblock)
// JVT-S054 (REPLACE) ->
//===== copy motion if non-adaptive prediction =====
if( ! bEstimateMotion )
{
ROF ( pcBaseLayerCtrl )
FMO* pcFMO = rcControlData.getSliceHeader( ePicType )->getFMO(); //TMM
for (UInt iSliceGroupID=0;!pcFMO->SliceGroupCompletelyCoded(iSliceGroupID);iSliceGroupID++)
{
rcSliceHeader.setFirstMbInSlice(pcFMO->getFirstMacroblockInSlice(iSliceGroupID));
rcSliceHeader.setLastMbInSlice(pcFMO->getLastMBInSliceGroup(iSliceGroupID));
// JVT-S054 (2) (ADD)
rcSliceHeader.setNumMbsInSlice(rcSliceHeader.getFMO()->getNumMbsInSlice(rcSliceHeader.getFirstMbInSlice(), rcSliceHeader.getLastMbInSlice()));
UInt uiMbAddress = rcSliceHeader.getFirstMbInSlice();
UInt uiLastMbAddress = rcSliceHeader.getLastMbInSlice();
UInt uiNumMBInSlice;
//===== initialization =====
RNOK( pcMbDataCtrl->initSlice( rcSliceHeader, ENCODE_PROCESS, false, pcMbDataCtrlCol ) );
//===== loop over macroblocks =====
for( uiNumMBInSlice = 0; uiMbAddress <= uiLastMbAddress; )
{
UInt uiMbY, uiMbX;
MbDataAccess* pcMbDataAccess = 0;
MbDataAccess* pcMbDataAccessBase = 0;
rcSliceHeader.getMbPositionFromAddress( uiMbY, uiMbX, uiMbAddress );
//===== init macroblock =====
RNOK ( pcMbDataCtrl ->initMb( pcMbDataAccess, uiMbY, uiMbX ) );
if ( pcBaseLayerCtrl )
{
RNOK( pcBaseLayerCtrl ->initMb( pcMbDataAccessBase, uiMbY, uiMbX ) );
}
RNOK( pcMbDataCtrl->getMbData(uiMbX, uiMbY).copyMotion( pcBaseLayerCtrl->getMbData(uiMbX, uiMbY), pcMbDataCtrl->getSliceId() ) );
RNOK( pcMbDataAccess->getMbMotionData( LIST_0 ).setRefPicIdcs( pcRefFrameList0 ) );
RNOK( pcMbDataAccess->getMbMotionData( LIST_1 ).setRefPicIdcs( pcRefFrameList1 ) );
// <<<< bug fix by heiko.schwarz@hhi.fhg.de
if( pcBaseLayerFrame ) // the motion data are not just copied, but inferred from the base layer
{
pcMbDataCtrl->getMbDataByIndex( uiMbAddress ).getMbMvdData( LIST_0 ).clear();
pcMbDataCtrl->getMbDataByIndex( uiMbAddress ).getMbMvdData( LIST_1 ).clear();
}
// >>>> bug fix by heiko.schwarz@hhi.fhg.de
uiMbAddress = rcSliceHeader.getFMO()->getNextMBNr(uiMbAddress);
uiNumMBInSlice++;
}
}
return Err::m_nOK;
}
if( ePicType!=FRAME )
{
if( pcOrigFrame ) RNOK( pcOrigFrame ->addFieldBuffer( ePicType ) );
if( pcIntraRecFrame ) RNOK( pcIntraRecFrame ->addFieldBuffer( ePicType ) );
if( pcBaseLayerFrame ) RNOK( pcBaseLayerFrame ->addFieldBuffer( ePicType ) );
if( pcBaseLayerResidual ) RNOK( pcBaseLayerResidual->addFieldBuffer( ePicType ) );
}
FMO* pcFMO = rcControlData.getSliceHeader( ePicType )->getFMO(); //TMM
for (UInt iSliceGroupID=0;!pcFMO->SliceGroupCompletelyCoded(iSliceGroupID);iSliceGroupID++)
{
rcSliceHeader.setFirstMbInSlice(pcFMO->getFirstMacroblockInSlice(iSliceGroupID));
rcSliceHeader.setLastMbInSlice(pcFMO->getLastMBInSliceGroup(iSliceGroupID));
// JVT-S054 (2) (ADD)
rcSliceHeader.setNumMbsInSlice(rcSliceHeader.getFMO()->getNumMbsInSlice(rcSliceHeader.getFirstMbInSlice(), rcSliceHeader.getLastMbInSlice()));
UInt uiMbAddress = rcSliceHeader.getFirstMbInSlice();
UInt uiLastMbAddress = rcSliceHeader.getLastMbInSlice();
UInt uiNumMBInSlice;
//===== initialization =====
RefFrameList rcRefFrameList1 = rcControlData.getPrdFrameList( LIST_1 );
if( rcSliceHeader.isMbaffFrame() )
{
RNOK( pcMbDataCtrl->initUsedField( rcSliceHeader, rcRefFrameList1 ) );
}
//===== initialization =====
RNOK( pcMbDataCtrl->initSlice( rcSliceHeader, ENCODE_PROCESS, false, pcMbDataCtrlCol ) );
if( ! m_bLoadMotionInfo )
{
RNOK( m_pcMotionEstimation->initSlice( rcSliceHeader ) );
RNOK( pcMbEncoder ->initSlice( rcSliceHeader ) );
RNOK( m_pcMotionEstimation->getSW()->initSlice( rcSliceHeader ) );
}
//===== loop over macroblocks =====
for( uiNumMBInSlice = 0; uiMbAddress <= uiLastMbAddress; )
{
MbDataAccess* pcMbDataAccess = 0;
MbDataAccess* pcMbDataAccessBase = 0;
UInt uiMbY, uiMbX;
Double dCost = 0;
rcSliceHeader.getMbPositionFromAddress( uiMbY, uiMbX, uiMbAddress );
//===== init macroblock =====
RNOK ( pcMbDataCtrl ->initMb( pcMbDataAccess, uiMbY, uiMbX ) );
if ( pcBaseLayerCtrl )
{
RNOK( pcBaseLayerCtrl ->initMb( pcMbDataAccessBase, uiMbY, uiMbX ) );
if (rcSliceHeader.getTCoeffLevelPredictionFlag())
{
pcMbDataAccess->setMbDataAccessBase( pcMbDataAccessBase );
}
}
if( ! m_bLoadMotionInfo )
{
//===== initialisation =====
RNOK( m_pcYuvFullPelBufferCtrl->initMb( uiMbY, uiMbX, false ) );
RNOK( m_pcYuvHalfPelBufferCtrl->initMb( uiMbY, uiMbX, false ) );
RNOK( m_pcMotionEstimation ->initMb( uiMbY, uiMbX, *pcMbDataAccess ) );
if( !rcSliceHeader.getNoInterLayerPredFlag() )
{
pcMbEncoder->setBaseModeAllowedFlag( m_apabBaseModeFlagAllowedArrays[0][uiMbAddress] );
}
RNOK( pcMbEncoder->estimatePrediction ( *pcMbDataAccess,
pcMbDataAccessBase,
*pcRefFrameList0,
*pcRefFrameList1,
pcBaseLayerFrame ? pcBaseLayerFrame ->getPic( ePicType ) : NULL,
pcBaseLayerResidual ? pcBaseLayerResidual->getPic( ePicType ) : NULL,
*pcOrigFrame ->getPic( ePicType ),
*pcIntraRecFrame ->getPic( ePicType ),
uiMaxMvPerMb,
m_bMCBlks8x8Disable,
m_bBiPred8x8Disable,
uiNumMaxIter,
uiIterSearchRange,
rcControlData.getLambda(),
dCost,
true ) );
// TMM_INTERLACE{
/* if( m_bSaveMotionInfo )
{
//===== save prediction data =====
// saveAll is displaced because the Mvs are Ok but the other data could be modified (mode,...).
// Do it after m_pcSliceEncoder->encodeHighPassPicture
// RNOK( pcMbDataAccess->getMbData().saveAll( m_pMotionInfoFile ) );
}*/
// TMM_INTERLACE}
}
else
{
//===== load prediction data =====
RNOK( pcMbDataAccess->getMbData().loadAll( m_pMotionInfoFile ) );
}
uiMbAddress = rcSliceHeader.getFMO()->getNextMBNr(uiMbAddress);
uiNumMBInSlice++;
}
}
if( ePicType!=FRAME )
{
if( pcOrigFrame ) RNOK( pcOrigFrame ->removeFieldBuffer( ePicType ) );
if( pcIntraRecFrame ) RNOK( pcIntraRecFrame ->removeFieldBuffer( ePicType ) );
if( pcBaseLayerFrame ) RNOK( pcBaseLayerFrame ->removeFieldBuffer( ePicType ) );
if( pcBaseLayerResidual ) RNOK( pcBaseLayerResidual->removeFieldBuffer( ePicType ) );
}
return Err::m_nOK;
}
ErrVal
LayerEncoder::xMotionCompensation( Frame* pcMCFrame,
RefFrameList* pcRefFrameList0,
RefFrameList* pcRefFrameList1,
MbDataCtrl* pcMbDataCtrl,
SliceHeader& rcSH )
{
Bool bCalcMv = false;
Bool bFaultTolerant = false;
MbEncoder* pcMbEncoder = m_pcSliceEncoder->getMbEncoder();
RNOK( pcMbDataCtrl ->initSlice( rcSH, PRE_PROCESS, false, NULL ) );
RNOK( m_pcMotionEstimation->initSlice( rcSH ) );
MbDataCtrl* pcBaseMbDataCtrl = getBaseMbDataCtrl();
RefFrameList* apcRefFrameList0[4] = { NULL, NULL, NULL, NULL };
RefFrameList* apcRefFrameList1[4] = { NULL, NULL, NULL, NULL };
const PicType ePicType = rcSH.getPicType();
const Bool bMbAff = rcSH.isMbaffFrame ();
Frame* apcBLFrame[3] = { 0, 0, m_pcBaseLayerFrame };
if( bMbAff )
{
RefFrameList acRefFrameList0[2];
RefFrameList acRefFrameList1[2];
RNOK( gSetFrameFieldLists( acRefFrameList0[0], acRefFrameList0[1], *pcRefFrameList0 ) );
RNOK( gSetFrameFieldLists( acRefFrameList1[0], acRefFrameList1[1], *pcRefFrameList1 ) );
apcRefFrameList0[ TOP_FIELD ] = ( NULL == pcRefFrameList0 ) ? NULL : &acRefFrameList0[0];
apcRefFrameList0[ BOT_FIELD ] = ( NULL == pcRefFrameList0 ) ? NULL : &acRefFrameList0[1];
⌨️ 快捷键说明
复制代码Ctrl + C
搜索代码Ctrl + F
全屏模式F11
增大字号Ctrl + =
减小字号Ctrl + -
显示快捷键?