📄 motioncompensation.cpp
字号:
for( y = sY; y < eY; y++ )
{
for( x = sX; x< eX; x++)
{
pDes[x] = pSrc[x];
}
pDes += iDesStride;
pSrc += iSrcStride;
}
pSrc = pcMbBufSrc->getMbCbAddr();
pDes = pcMbBufDes->getMbCbAddr();
iDesStride = pcMbBufDes->getCStride();
iSrcStride = pcMbBufSrc->getCStride();
pSrc += sY/2*iSrcStride;
pDes += sY/2*iDesStride;
for( y = sY/2; y < eY/2; y++ )
{
for( x = sX/2; x< eX/2; x++)
pDes[x] = pSrc[x];
pDes += iDesStride;
pSrc += iSrcStride;
}
pSrc = pcMbBufSrc->getMbCrAddr();
pDes = pcMbBufDes->getMbCrAddr();
pSrc += sY/2*iSrcStride;
pDes += sY/2*iDesStride;
for( y = sY/2; y < eY/2; y++ )
{
for( x = sX/2; x< eX/2; x++)
pDes[x] = pSrc[x];
pDes += iDesStride;
pSrc += iSrcStride;
}
return Err::m_nOK;
}
ErrVal MotionCompensation::compensateMb( MbDataAccess& rcMbDataAccess,
RefFrameList& rcRefFrameList0,
RefFrameList& rcRefFrameList1,
YuvMbBuffer* pcRecBuffer,
Bool bCalcMv )
{
MbMode eMbMode = rcMbDataAccess.getMbData().getMbMode();
switch( eMbMode )
{
case MODE_16x16:
{
if( bCalcMv )
{
xCalc16x16( rcMbDataAccess, NULL );
}
MC8x8 cMC8x8D( B_8x8_0 );
Int iRefIdx0 = rcMbDataAccess.getMbMotionData( LIST_0 ).getRefIdx();
Int iRefIdx1 = rcMbDataAccess.getMbMotionData( LIST_1 ).getRefIdx();
Frame* pcRefFrame0 = ( iRefIdx0 > 0 ? rcRefFrameList0[ iRefIdx0 ] : NULL );
Frame* pcRefFrame1 = ( iRefIdx1 > 0 ? rcRefFrameList1[ iRefIdx1 ] : NULL );
xGetMbPredData( rcMbDataAccess, pcRefFrame0, pcRefFrame1, cMC8x8D );
xPredLuma( pcRecBuffer, 16, 16, cMC8x8D );
xPredChroma( pcRecBuffer, 8, 8, cMC8x8D );
}
break;
case MODE_16x8:
{
if( bCalcMv )
{
xCalc16x8( rcMbDataAccess, NULL );
}
MC8x8 cMC8x8D0( B_8x8_0 );
MC8x8 cMC8x8D1( B_8x8_2 );
Int iRefIdx0 = rcMbDataAccess.getMbMotionData( LIST_0 ).getRefIdx( PART_16x8_0 );
Int iRefIdx1 = rcMbDataAccess.getMbMotionData( LIST_1 ).getRefIdx( PART_16x8_0 );
Frame* pcRefFrame0 = ( iRefIdx0 > 0 ? rcRefFrameList0[ iRefIdx0 ] : NULL );
Frame* pcRefFrame1 = ( iRefIdx1 > 0 ? rcRefFrameList1[ iRefIdx1 ] : NULL );
xGetMbPredData( rcMbDataAccess, pcRefFrame0, pcRefFrame1, cMC8x8D0 );
iRefIdx0 = rcMbDataAccess.getMbMotionData( LIST_0 ).getRefIdx( PART_16x8_1 );
iRefIdx1 = rcMbDataAccess.getMbMotionData( LIST_1 ).getRefIdx( PART_16x8_1 );
pcRefFrame0 = ( iRefIdx0 > 0 ? rcRefFrameList0[ iRefIdx0 ] : NULL );
pcRefFrame1 = ( iRefIdx1 > 0 ? rcRefFrameList1[ iRefIdx1 ] : NULL );
xGetMbPredData( rcMbDataAccess, pcRefFrame0, pcRefFrame1, cMC8x8D1 );
xPredLuma( pcRecBuffer, 16, 8, cMC8x8D0 );
xPredLuma( pcRecBuffer, 16, 8, cMC8x8D1 );
xPredChroma( pcRecBuffer, 8, 4, cMC8x8D0 );
xPredChroma( pcRecBuffer, 8, 4, cMC8x8D1 );
}
break;
case MODE_8x16:
{
if( bCalcMv )
{
xCalc8x16( rcMbDataAccess, NULL );
}
MC8x8 cMC8x8D0( B_8x8_0 );
MC8x8 cMC8x8D1( B_8x8_1 );
Int iRefIdx0 = rcMbDataAccess.getMbMotionData( LIST_0 ).getRefIdx( PART_8x16_0 );
Int iRefIdx1 = rcMbDataAccess.getMbMotionData( LIST_1 ).getRefIdx( PART_8x16_0 );
Frame* pcRefFrame0 = ( iRefIdx0 > 0 ? rcRefFrameList0[ iRefIdx0 ] : NULL );
Frame* pcRefFrame1 = ( iRefIdx1 > 0 ? rcRefFrameList1[ iRefIdx1 ] : NULL );
xGetMbPredData( rcMbDataAccess, pcRefFrame0, pcRefFrame1, cMC8x8D0 );
iRefIdx0 = rcMbDataAccess.getMbMotionData( LIST_0 ).getRefIdx( PART_8x16_1 );
iRefIdx1 = rcMbDataAccess.getMbMotionData( LIST_1 ).getRefIdx( PART_8x16_1 );
pcRefFrame0 = ( iRefIdx0 > 0 ? rcRefFrameList0[ iRefIdx0 ] : NULL );
pcRefFrame1 = ( iRefIdx1 > 0 ? rcRefFrameList1[ iRefIdx1 ] : NULL );
xGetMbPredData( rcMbDataAccess, pcRefFrame0, pcRefFrame1, cMC8x8D1 );
xPredLuma( pcRecBuffer, 8, 16, cMC8x8D0 );
xPredLuma( pcRecBuffer, 8, 16, cMC8x8D1 );
xPredChroma( pcRecBuffer, 4, 8, cMC8x8D0 );
xPredChroma( pcRecBuffer, 4, 8, cMC8x8D1 );
}
break;
case MODE_SKIP:
{
if( rcMbDataAccess.getSH().isBSlice() )
{
if( bCalcMv )
{
xCalcSDirect( rcMbDataAccess, NULL );
}
B8x8Idx c8x8Idx;
RNOK( compensateDirectBlock( rcMbDataAccess, pcRecBuffer, c8x8Idx, rcRefFrameList0, rcRefFrameList1 ) ); c8x8Idx++;
RNOK( compensateDirectBlock( rcMbDataAccess, pcRecBuffer, c8x8Idx, rcRefFrameList0, rcRefFrameList1 ) ); c8x8Idx++;
RNOK( compensateDirectBlock( rcMbDataAccess, pcRecBuffer, c8x8Idx, rcRefFrameList0, rcRefFrameList1 ) ); c8x8Idx++;
RNOK( compensateDirectBlock( rcMbDataAccess, pcRecBuffer, c8x8Idx, rcRefFrameList0, rcRefFrameList1 ) ); ;
}
else
{
// TMM_EC {{
if ( rcMbDataAccess.getSH().getErrorConcealMode() == EC_TEMPORAL_DIRECT)
bCalcMv = true;
// TMM_EC }}
if( bCalcMv )
{
Mv cMvPred;
rcMbDataAccess.getMvPredictorSkipMode ( cMvPred );
rcMbDataAccess.getMbMotionData( LIST_0 ).setRefIdx ( 1 );
rcMbDataAccess.getMbMotionData( LIST_0 ).setAllMv ( cMvPred );
rcMbDataAccess.getMbMotionData( LIST_1 ).setRefIdx ( 0 );
rcMbDataAccess.getMbMotionData( LIST_1 ).setAllMv ( Mv::ZeroMv() );
}
MC8x8 cMC8x8D( B_8x8_0 );
Int iRefIdx0 = rcMbDataAccess.getMbMotionData( LIST_0 ).getRefIdx( PART_8x16_0 );
Int iRefIdx1 = rcMbDataAccess.getMbMotionData( LIST_1 ).getRefIdx( PART_8x16_0 );
Frame* pcRefFrame0 = ( iRefIdx0 > 0 ? rcRefFrameList0[ iRefIdx0 ] : NULL );
Frame* pcRefFrame1 = ( iRefIdx1 > 0 ? rcRefFrameList1[ iRefIdx1 ] : NULL );
xGetMbPredData( rcMbDataAccess, pcRefFrame0, pcRefFrame1, cMC8x8D );
xPredLuma( pcRecBuffer, 16, 16, cMC8x8D );
xPredChroma( pcRecBuffer, 8, 8, cMC8x8D );
return Err::m_nOK;
}
}
break;
case MODE_8x8:
case MODE_8x8ref0:
printf("function not defined for the case\n");
RERR();
break;
default:
break;
}
return Err::m_nOK;
}
__inline Void MotionCompensation::xGetMbPredData( MbDataAccess& rcMbDataAccess,
const Frame* pcRefFrame0,
const Frame* pcRefFrame1,
MC8x8& rcMC8x8D )
{
rcMC8x8D.clear();
Int iPredCount = 0;
const SliceHeader& rcSH = rcMbDataAccess.getSH();
const Frame* apcFrame[2];
for (Int n = 0; n < 2; n++)
{
const Frame* pcRefFrame = n ? pcRefFrame1 : pcRefFrame0;
ListIdx eLstIdx = ListIdx(n);
const MbMvData& rcMbMvdData = rcMbDataAccess.getMbMvdData( eLstIdx );
Mv3D& rcMv3D = rcMC8x8D.m_aacMv[eLstIdx][0];
rcMbDataAccess.getMbMotionData( eLstIdx ).getMv3D( rcMv3D, rcMC8x8D.m_cIdx );
rcMC8x8D.m_aacMvd[eLstIdx][SPART_4x4_0] = rcMbMvdData.getMv( rcMC8x8D.m_cIdx + SPART_4x4_0 );
if( pcRefFrame != NULL )
{
iPredCount++;
rcMv3D.limitComponents( m_cMin, m_cMax );
rcMC8x8D.m_sChromaOffset[eLstIdx] = xCorrectChromaMv( rcMbDataAccess, pcRefFrame->getPicType() );
rcMC8x8D.m_apcRefBuffer [eLstIdx] = const_cast<Frame*>(pcRefFrame)->getFullPelYuvBuffer();
apcFrame[n] = pcRefFrame;
rcMC8x8D.m_apcPW[eLstIdx] = &rcSH.getPredWeight( eLstIdx, rcMv3D.getRef(), rcMbDataAccess.getMbData().getFieldFlag() ); // TMM_INTERLACE
}
}
if( ( 2 == iPredCount ) && ( 2 == rcSH.getPPS().getWeightedBiPredIdc() ) )
{
Int iScale = rcSH.getDistScaleFactorWP( apcFrame[0], apcFrame[1] );
rcMC8x8D.m_acPW[LIST_1].scaleL1Weight( iScale );
rcMC8x8D.m_acPW[LIST_0].scaleL0Weight( rcMC8x8D.m_acPW[LIST_1] );
rcMC8x8D.m_apcPW[LIST_1] = &rcMC8x8D.m_acPW[LIST_1];
rcMC8x8D.m_apcPW[LIST_0] = &rcMC8x8D.m_acPW[LIST_0];
}
}
__inline Void MotionCompensation::xGetBlkPredData( MbDataAccess& rcMbDataAccess,
const Frame* pcRefFrame0,
const Frame* pcRefFrame1,
MC8x8& rcMC8x8D,
BlkMode eBlkMode )
{
rcMC8x8D.clear();
Int iPredCount = 0;
const SliceHeader& rcSH = rcMbDataAccess.getSH();
const Frame* apcFrame[2];
for( Int n = 0; n < 2; n++)
{
const Frame* pcRefFrame = n ? pcRefFrame1 : pcRefFrame0;
ListIdx eLstIdx = ListIdx( n );
Mv3D& rcMv3D = rcMC8x8D.m_aacMv[eLstIdx][SPART_4x4_0];
const MbMotionData& rcMbMotionData = rcMbDataAccess.getMbMotionData( eLstIdx );
const MbMvData& rcMbMvdData = rcMbDataAccess.getMbMvdData( eLstIdx );
rcMbMotionData.getMv3D( rcMv3D, rcMC8x8D.m_cIdx + SPART_4x4_0 );
rcMv3D.limitComponents( m_cMin, m_cMax );
rcMC8x8D.m_aacMvd[eLstIdx][SPART_4x4_0] = rcMbMvdData.getMv( rcMC8x8D.m_cIdx + SPART_4x4_0 );
if( pcRefFrame != NULL )
{
iPredCount++;
rcMC8x8D.m_sChromaOffset[eLstIdx] = xCorrectChromaMv( rcMbDataAccess, pcRefFrame->getPicType() );
rcMC8x8D.m_apcRefBuffer[eLstIdx] = const_cast<Frame*>(pcRefFrame)->getFullPelYuvBuffer();
apcFrame[n] = pcRefFrame;
rcMC8x8D.m_apcPW[eLstIdx] = &rcSH.getPredWeight( eLstIdx, rcMv3D.getRef(), rcMbDataAccess.getMbData().getFieldFlag() ); // TMM_INTERLACE
switch( eBlkMode )
{
case BLK_8x8:
break;
case BLK_8x4:
{
rcMC8x8D.m_aacMv[eLstIdx][SPART_4x4_2] = rcMbMotionData.getMv( rcMC8x8D.m_cIdx + SPART_4x4_2 );
rcMC8x8D.m_aacMv[eLstIdx][SPART_4x4_2].limitComponents( m_cMin, m_cMax );
rcMC8x8D.m_aacMvd[eLstIdx][SPART_4x4_2] = rcMbMvdData.getMv( rcMC8x8D.m_cIdx + SPART_4x4_2 );
}
break;
case BLK_4x8:
{
rcMC8x8D.m_aacMv[eLstIdx][SPART_4x4_1] = rcMbMotionData.getMv( rcMC8x8D.m_cIdx + SPART_4x4_1 );
rcMC8x8D.m_aacMv[eLstIdx][SPART_4x4_1].limitComponents( m_cMin, m_cMax );
rcMC8x8D.m_aacMvd[eLstIdx][SPART_4x4_1] = rcMbMvdData.getMv( rcMC8x8D.m_cIdx + SPART_4x4_1 );
}
break;
case BLK_SKIP:
case BLK_4x4:
{
rcMC8x8D.m_aacMv[eLstIdx][SPART_4x4_1] = rcMbMotionData.getMv( rcMC8x8D.m_cIdx + SPART_4x4_1 );
rcMC8x8D.m_aacMv[eLstIdx][SPART_4x4_1].limitComponents( m_cMin, m_cMax );
rcMC8x8D.m_aacMvd[eLstIdx][SPART_4x4_1] = rcMbMvdData.getMv( rcMC8x8D.m_cIdx + SPART_4x4_1 );
rcMC8x8D.m_aacMv[eLstIdx][SPART_4x4_2] = rcMbMotionData.getMv( rcMC8x8D.m_cIdx + SPART_4x4_2 );
rcMC8x8D.m_aacMv[eLstIdx][SPART_4x4_2].limitComponents( m_cMin, m_cMax );
rcMC8x8D.m_aacMvd[eLstIdx][SPART_4x4_2] = rcMbMvdData.getMv( rcMC8x8D.m_cIdx + SPART_4x4_2 );
rcMC8x8D.m_aacMv[eLstIdx][SPART_4x4_3] = rcMbMotionData.getMv( rcMC8x8D.m_cIdx + SPART_4x4_3 );
rcMC8x8D.m_aacMv[eLstIdx][SPART_4x4_3].limitComponents( m_cMin, m_cMax );
rcMC8x8D.m_aacMvd[eLstIdx][SPART_4x4_3] = rcMbMvdData.getMv( rcMC8x8D.m_cIdx + SPART_4x4_3 );
}
break;
default:
{
AF();
}
}
}
}
if( ( 2 == iPredCount ) && ( 2 == rcSH.getPPS().getWeightedBiPredIdc() ) )
{
Int iScale = rcSH.getDistScaleFactorWP( apcFrame[0], apcFrame[1] );
rcMC8x8D.m_acPW[LIST_1].scaleL1Weight( iScale );
rcMC8x8D.m_acPW[LIST_0].scaleL0Weight( rcMC8x8D.m_acPW[LIST_1] );
rcMC8x8D.m_apcPW[LIST_1] = &rcMC8x8D.m_acPW[LIST_1];
rcMC8x8D.m_apcPW[LIST_0] = &rcMC8x8D.m_acPW[LIST_0];
}
}
// scaled by 10-bits
TCoeff aiNormMatrix4x4[16]=
{
512, 410, 512, 410,
410, 328, 410, 328,
512, 410, 512, 410,
410, 328, 410, 328,
};
TCoeff aiNormMatrix8x8[64] =
{
512, 454, 819, 454, 512, 454, 819, 454,
454, 402, 726, 402, 454, 402, 726, 402,
819, 726, 1311, 726, 819, 726, 1311, 726,
454, 402, 726, 402, 454, 402, 726, 402,
512, 454, 819, 454, 512, 454, 819, 454,
454, 402, 726, 402, 454, 402, 726, 402,
819, 726, 1311, 726, 819, 726, 1311, 726,
454, 402, 726, 402, 454, 402, 726, 402,
};
ErrVal
MotionCompensation::xCompensateMbAllModes(MbDataAccess& rcMbDataAccess,
RefFrameList& rcRefFrameList0,
RefFrameList& rcRefFrameList1,
YuvMbBuffer* pcYuvMbBuffer )
{
if( rcMbDataAccess.getMbData().getMbMode() == MODE_8x8 )
{
for( B8x8Idx c8x8Idx; c8x8Idx.isLegal(); c8x8Idx++ )
{
RNOK( compensateSubMb( c8x8Idx, rcMbDataAccess, rcRefFrameList0, rcRefFrameList1, pcYuvMbBuffer, false, false ) );
}
}
else
{
RNOK( compensateMb( rcMbDataAccess, rcRefFrameList0, rcRefFrameList1, pcYuvMbBuffer, false ) );
}
return Err::m_nOK;
}
H264AVC_NAMESPACE_END
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -