📄 mbencoder.cpp
字号:
m_pcIntPicBuffer->getYuvBufferCtrl().initMb();
UInt p=1;
UInt q=100;
if(bInter)
{
for(UInt i=0;i<=m_uiLayerID;i++)
{
//Bug_Fix JVT-R057 0806{
p=p*(100-m_auiPLR[i]);
//Bug_Fix JVT-R057 0806}
}
//q=(UInt)pow(100,(m_uiLayerID+1));
q=(UInt)pow(100.0,(int)(m_uiLayerID+1));
}
else
{
p=100-m_auiPLR[m_uiLayerID];
}
if(mode!=INTRA_BL)
{
for(y=blockY;y<(blockY+4);y++)
for(x=blockX;x<(blockX+4);x++)
{
if(m_pcFrameEcEp)
ec_ep=m_pcFrameEcEp->getChannelDistortion()[y*KBlock+x];
else
ec_ep=0;
m_pcIntPicBuffer->getYuvBufferCtrl().initMb();
if(m_pcFrameEcEp)
ec_rec=GetEC_REC(m_pcIntPicBuffer,m_pcFrameEcEp->getFullPelYuvBuffer(),x,y);
else
ec_rec=0;
pcFrame->getChannelDistortion()[y*KBlock+x]=(q-p)*(ec_rec+ec_ep)/q;
}
}
else
{
Int ep_base;
Int blockIndex;
Int xx,yy;
for(y=blockY;y<(blockY+4);y++)
for(x=blockX;x<(blockX+4);x++)
{
if(m_pcFrameEcEp)
ec_ep=m_pcFrameEcEp->getChannelDistortion()[y*KBlock+x];
else
ec_ep=0;
m_pcIntPicBuffer->getYuvBufferCtrl().initMb();
if(m_pcFrameEcEp)
ec_rec=GetEC_REC(m_pcIntPicBuffer,m_pcFrameEcEp->getFullPelYuvBuffer(),x,y);
else
ec_rec=0;
xx=(Int)(x/m_aadRatio[m_uiLayerID][0]);
yy=(Int)(y/m_aadRatio[m_uiLayerID][1]);
blockIndex=yy*(Int)(KBlock/m_aadRatio[m_uiLayerID][0])+xx;
ep_base=pcBaseLayer->getChannelDistortion()[blockIndex];
pcFrame->getChannelDistortion()[y*KBlock+x]=(p*ep_base+(q-p)*(ec_rec+ec_ep))/q;
}
}
}
//JVT-R057 LA-RDO}
rdCost = m_pcIntMbBestData->rdCost();
m_pcIntMbBestData ->uninit();
m_pcIntMbTempData ->uninit();
m_pcIntMbBest8x8Data->uninit();
m_pcIntMbTemp8x8Data->uninit();
return Err::m_nOK;
}
ErrVal
MbEncoder::xCheckInterMbMode8x8( IntMbTempData*& rpcMbTempData,
IntMbTempData*& rpcMbBestData,
IntMbTempData* pcMbRefData,
RefFrameList& rcRefFrameList0,
RefFrameList& rcRefFrameList1,
MbDataAccess* pcMbDataAccessBaseMotion,
IntFrame* pcBaseLayerRec
)
{
ROTRS( ! rpcMbTempData->getSH().getPPS().getTransform8x8ModeFlag(), Err::m_nOK );
ROTRS( ! pcMbRefData->is8x8TrafoFlagPresent( rpcMbTempData->getSH().getSPS().getDirect8x8InferenceFlag() ), Err::m_nOK );
if( pcMbRefData == rpcMbTempData )
{
rpcMbTempData->clearCost ();
rpcMbTempData->getMbTCoeffs ().clear();
rpcMbTempData->setTransformSize8x8( true );
}
else
{
rpcMbTempData->clear ();
rpcMbTempData->setMbMode ( pcMbRefData->getMbMode () );
rpcMbTempData->setBLSkipFlag ( pcMbRefData->getBLSkipFlag () );
rpcMbTempData->setResidualPredFlags ( pcMbRefData->getResidualPredFlags () );
rpcMbTempData->setBlkMode ( B_8x8_0, pcMbRefData->getBlkMode ( B_8x8_0 ) );
rpcMbTempData->setBlkMode ( B_8x8_1, pcMbRefData->getBlkMode ( B_8x8_1 ) );
rpcMbTempData->setBlkMode ( B_8x8_2, pcMbRefData->getBlkMode ( B_8x8_2 ) );
rpcMbTempData->setBlkMode ( B_8x8_3, pcMbRefData->getBlkMode ( B_8x8_3 ) );
rpcMbTempData->setTransformSize8x8 ( true );
rpcMbTempData->getMbMotionData( LIST_0 ).copyFrom( pcMbRefData->getMbMotionData( LIST_0 ) );
rpcMbTempData->getMbMotionData( LIST_1 ).copyFrom( pcMbRefData->getMbMotionData( LIST_1 ) );
rpcMbTempData->getMbMvdData ( LIST_0 ).copyFrom( pcMbRefData->getMbMvdData ( LIST_0 ) );
rpcMbTempData->getMbMvdData ( LIST_1 ).copyFrom( pcMbRefData->getMbMvdData ( LIST_1 ) );
}
RNOK( xSetRdCost8x8InterMb( *rpcMbTempData, pcMbDataAccessBaseMotion, rcRefFrameList0, rcRefFrameList1,
false, 0, false, pcBaseLayerRec ) );
//JVT-R057 LA-RDO{
if(m_bLARDOEnable)
rpcMbTempData->rdCost()+=getEpRef();
//JVT-R057 LA-RDO}
RNOK( xCheckBestEstimation( rpcMbTempData, rpcMbBestData ) );
return Err::m_nOK;
}
ErrVal
MbEncoder::encodeMacroblock( MbDataAccess& rcMbDataAccess,
IntFrame* pcFrame,
RefFrameList& rcList0,
RefFrameList& rcList1,
UInt uiNumMaxIter,
UInt uiIterSearchRange,
Double dLambda )
{
ROF( m_bInitDone );
UInt uiQp = rcMbDataAccess.getMbData().getQp();
RNOK( m_pcRateDistortionIf->setMbQpLambda( rcMbDataAccess, uiQp, dLambda ) );
m_pcIntMbBestIntraChroma = NULL;
m_pcIntMbBestData ->init( rcMbDataAccess );
m_pcIntMbTempData ->init( rcMbDataAccess );
m_pcIntMbBest8x8Data->init( rcMbDataAccess );
m_pcIntMbTemp8x8Data->init( rcMbDataAccess );
m_pcIntPicBuffer = pcFrame->getFullPelYuvBuffer();
m_pcXDistortion->loadOrgMbPelData( m_pcIntPicBuffer, m_pcIntOrgMbPelData );
m_pcTransform->setQp( rcMbDataAccess, rcMbDataAccess.getSH().getUseBasePredictionFlag() );
//====== evaluate macroblock modes ======
if( rcMbDataAccess.getSH().isInterP() )
{
RNOK( xEstimateMbSkip ( m_pcIntMbTempData, m_pcIntMbBestData, rcList0, rcList1 ) );
}
if( rcMbDataAccess.getSH().isInterB() )
{
RNOK( xEstimateMbDirect ( m_pcIntMbTempData, m_pcIntMbBestData, rcList0, rcList1, NULL, false ) );
}
if( rcMbDataAccess.getSH().isInterP() || rcMbDataAccess.getSH().isInterB() )
{
RNOK( xEstimateMb16x16 ( m_pcIntMbTempData, m_pcIntMbBestData, rcList0, rcList1, false, uiNumMaxIter, uiIterSearchRange, false, NULL, false ) );
RNOK( xEstimateMb16x8 ( m_pcIntMbTempData, m_pcIntMbBestData, rcList0, rcList1, false, uiNumMaxIter, uiIterSearchRange, false, NULL, false ) );
RNOK( xEstimateMb8x16 ( m_pcIntMbTempData, m_pcIntMbBestData, rcList0, rcList1, false, uiNumMaxIter, uiIterSearchRange, false, NULL, false ) );
RNOK( xEstimateMb8x8 ( m_pcIntMbTempData, m_pcIntMbBestData, rcList0, rcList1, false, uiNumMaxIter, uiIterSearchRange, false, NULL, false ) );
RNOK( xEstimateMb8x8Frext ( m_pcIntMbTempData, m_pcIntMbBestData, rcList0, rcList1, false, uiNumMaxIter, uiIterSearchRange, false, NULL, false ) );
}
RNOK( xEstimateMbIntra16 ( m_pcIntMbTempData, m_pcIntMbBestData, rcMbDataAccess.getSH().isInterB() ) );
RNOK( xEstimateMbIntra8 ( m_pcIntMbTempData, m_pcIntMbBestData, rcMbDataAccess.getSH().isInterB() ) );
RNOK( xEstimateMbIntra4 ( m_pcIntMbTempData, m_pcIntMbBestData, rcMbDataAccess.getSH().isInterB() ) );
RNOK( xEstimateMbPCM ( m_pcIntMbTempData, m_pcIntMbBestData, rcMbDataAccess.getSH().isInterB() ) );
//===== fix estimation =====
RNOK( m_pcRateDistortionIf->fixMacroblockQP( *m_pcIntMbBestData ) );
xStoreEstimation( rcMbDataAccess, *m_pcIntMbBestData, NULL, NULL, false, NULL );
//===== uninit =====
m_pcIntMbBestData ->uninit();
m_pcIntMbTempData ->uninit();
m_pcIntMbBest8x8Data->uninit();
m_pcIntMbTemp8x8Data->uninit();
return Err::m_nOK;
}
ErrVal
MbEncoder::encodeInterP( MbDataAccess& rcMbDataAccess,
MbDataAccess* pcMbDataAccessBase,
Int iSpatialScalabilityType,
IntFrame* pcOrgFrame,
IntFrame* pcFrame,
IntFrame* pcRecSubband,
IntFrame* pcPredSignal,
IntFrame* pcBaseLayerRec,
IntFrame* pcBaseLayerSbb,
RefFrameList& rcRefFrameList0,
RefFrameList* pcRefFrameList0Base,
Double dLambda,
Double& rdCost,
Bool bSkipModeAllowed )
{
ROF( bInitDone );
UInt uiQp = rcMbDataAccess.getMbData().getQp();
if (rcMbDataAccess.getSH().getAVCRewriteFlag())
{
uiQp = rcMbDataAccess.getLastQp(); // to avoid QP get changed by fixMacroblockQP
rcMbDataAccess.getMbData().setQp( uiQp );
}
RNOK( m_pcRateDistortionIf->setMbQpLambda( rcMbDataAccess, uiQp, dLambda ) )
rcMbDataAccess.setMbDataAccessBase(pcMbDataAccessBase);
m_pcIntMbBestIntraChroma = NULL;
m_pcIntMbBestData ->init( rcMbDataAccess );
m_pcIntMbTempData ->init( rcMbDataAccess );
m_pcIntMbBest8x8Data->init( rcMbDataAccess );
m_pcIntMbTemp8x8Data->init( rcMbDataAccess );
m_pcIntPicBuffer = pcFrame->getFullPelYuvBuffer();
m_pcXDistortion->loadOrgMbPelData( pcOrgFrame->getFullPelYuvBuffer(), m_pcIntOrgMbPelData );
m_pcTransform->setQp ( rcMbDataAccess, true );
m_pcTransform->setClipMode( false );
RefFrameList cRefFrameList1;
IntYuvMbBuffer cBaseLayerBuffer;
Bool bBaseLayerAvailable = (NULL != pcMbDataAccessBase) && (rcMbDataAccess.getSH().getBaseLayerId() != MSYS_UINT_MAX);
Bool bIntraEnable = ! rcMbDataAccess.isFieldMbInMbaffFrame() || rcMbDataAccess.isTopMb() || rcMbDataAccess.getMbDataComplementary().isIntra();
Bool bInterEnable = ! rcMbDataAccess.isFieldMbInMbaffFrame() || rcMbDataAccess.isTopMb() || ! rcMbDataAccess.getMbDataComplementary().isIntra();
//-- JVT-T037
Bool b = false;
if ( rcMbDataAccess.isFieldMbInMbaffFrame() && rcMbDataAccess.getMbPicType() == BOT_FIELD && rcMbDataAccess.getMbDataComplementary().isIntra() )
{
if ( pcMbDataAccessBase )
{
if ( !pcMbDataAccessBase->getMbData().isIntra() || !pcMbDataAccessBase->getMbDataComplementary().isIntra() )
{
b = true;
}
}
}
bInterEnable |= b;
//--
Bool bDefaultResPredFlag = false;
Bool avcRewriteFlag = rcMbDataAccess.getSH().getAVCRewriteFlag();
if( rcMbDataAccess.getSH().getPPS().getEntropyCodingModeFlag() &&
rcMbDataAccess.getSH().getBaseLayerId() != MSYS_UINT_MAX )
{
ROF( pcBaseLayerSbb );
cBaseLayerBuffer.loadBuffer ( const_cast<IntFrame*>(pcBaseLayerSbb)->getFullPelYuvBuffer() );
if( !avcRewriteFlag)
bDefaultResPredFlag = cBaseLayerBuffer.isZero();
}
//===== residual prediction =====
if( bBaseLayerAvailable && bInterEnable )
{
ROF( pcBaseLayerSbb );
cBaseLayerBuffer .loadBuffer ( pcBaseLayerSbb->getFullPelYuvBuffer() );
{
if( !avcRewriteFlag)
m_pcIntOrgMbPelData->subtract ( cBaseLayerBuffer );
if( ! pcMbDataAccessBase->getMbData().isIntra() && rcRefFrameList0.getActive() ) // JVT-Q065 EIDR
{
//--- only if base layer is in inter mode ---
// TMM_ESS
if ( pcMbDataAccessBase->getMbData().getInCropWindowFlag()&& ( !cBaseLayerBuffer.isZero() || avcRewriteFlag || !rcMbDataAccess.getSH().getAdaptiveResPredictionFlag() ) )
RNOK( xEstimateMbBLSkip ( m_pcIntMbTempData, m_pcIntMbBestData, rcRefFrameList0, cRefFrameList1, pcBaseLayerRec, false, iSpatialScalabilityType, pcMbDataAccessBase, rcMbDataAccess, true ) );
if ( !avcRewriteFlag && pcMbDataAccessBase->getMbData().getInCropWindowFlag() && rcMbDataAccess.isConstrainedInterLayerPred( ) &&
pcRefFrameList0Base == 0 && rcMbDataAccess.getSH().getAdaptiveResPredictionFlag() && rcMbDataAccess.useSmoothedRef() )
{
RNOK( xEstimateMbSR ( m_pcIntMbTempData, m_pcIntMbBestData, rcRefFrameList0, cRefFrameList1, pcBaseLayerSbb,
pcMbDataAccessBase, true, pcBaseLayerRec ) );
}
}
if( rcMbDataAccess.getSH().getAdaptivePredictionFlag() && rcRefFrameList0.getActive() && !avcRewriteFlag) // JVT-Q065 EIDR
{
//--- only if adaptive inter-layer prediction ---
RNOK( xEstimateMb16x16 ( m_pcIntMbTempData, m_pcIntMbBestData, rcRefFrameList0, cRefFrameList1, false, 0, 0, false, pcMbDataAccessBase, true ) );
RNOK( xEstimateMb16x8 ( m_pcIntMbTempData, m_pcIntMbBestData, rcRefFrameList0, cRefFrameList1, false, 0, 0, false, pcMbDataAccessBase, true ) );
RNOK( xEstimateMb8x16 ( m_pcIntMbTempData, m_pcIntMbBestData, rcRefFrameList0, cRefFrameList1, false, 0, 0, false, pcMbDataAccessBase, true ) );
RNOK( xEstimateMb8x8 ( m_pcIntMbTempData, m_pcIntMbBestData, rcRefFrameList0, cRefFrameList1, false, 0, 0, false, pcMbDataAccessBase, true ) );
RNOK( xEstimateMb8x8Frext ( m_pcIntMbTempData, m_pcIntMbBestData, rcRefFrameList0, cRefFrameList1, false, 0, 0, false, pcMbDataAccessBase, true ) );
}
if( !avcRewriteFlag )
m_pcIntOrgMbPelData->add( cBaseLayerBuffer );
}
}
//===== intra base layer mode =====
if( bBaseLayerAvailable && bIntraEnable )
{
//===== only when intra BL is allowed =====
if( ( avcRewriteFlag && pcMbDataAccessBase->getMbData().isIntra() )
|| (!avcRewriteFlag && (pcMbDataAccessBase->getMbData().isIntra() || ! rcMbDataAccess.isConstrainedInterLayerPred() )) )
{
if ( pcMbDataAccessBase->getMbData().getInCropWindowFlag() ) // TMM_ESS
RNOK( xEstimateMbIntraBL( m_pcIntMbTempData, m_pcIntMbBestData, pcBaseLayerRec, false, pcMbDataAccessBase ) );
}
}
if( bInterEnable )
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -