/** decode motion information for every PU block.
* \param pcCU
* \param uiAbsPartIdx
* \param uiDepth
* \param pcSubCU
* \returns Void
*/
Void TDecEntropy::decodePUWise( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, TComDataCU* pcSubCU )
{
PartSize ePartSize = pcCU->getPartitionSize( uiAbsPartIdx );
UInt uiNumPU = ( ePartSize == SIZE_2Nx2N ? 1 : ( ePartSize == SIZE_NxN ? 4 : 2 ) );
UInt uiPUOffset = ( g_auiPUOffset[UInt( ePartSize )] << ( ( pcCU->getSlice()->getSPS()->getMaxCUDepth() - uiDepth ) << 1 ) ) >> 4;
#if H_3D_IV_MERGE
TComMvField cMvFieldNeighbours[MRG_MAX_NUM_CANDS_MEM << 1]; // double length for mv of both lists
UChar uhInterDirNeighbours[MRG_MAX_NUM_CANDS_MEM];
#else
TComMvField cMvFieldNeighbours[MRG_MAX_NUM_CANDS << 1]; // double length for mv of both lists
UChar uhInterDirNeighbours[MRG_MAX_NUM_CANDS];
#endif
#if H_3D_SPIVMP
Bool bSPIVMPFlag[MRG_MAX_NUM_CANDS_MEM];
TComMvField* pcMvFieldSP;
UChar* puhInterDirSP;
pcMvFieldSP = new TComMvField[pcCU->getPic()->getPicSym()->getNumPartition()*2];
puhInterDirSP = new UChar[pcCU->getPic()->getPicSym()->getNumPartition()];
#endif
for ( UInt ui = 0; ui < pcCU->getSlice()->getMaxNumMergeCand(); ui++ )
{
uhInterDirNeighbours[ui] = 0;
}
Int numValidMergeCand = 0;
Bool isMerged = false;
pcSubCU->copyInterPredInfoFrom( pcCU, uiAbsPartIdx, REF_PIC_LIST_0 );
pcSubCU->copyInterPredInfoFrom( pcCU, uiAbsPartIdx, REF_PIC_LIST_1 );
#if H_3D_IV_MERGE
pcSubCU->copyDVInfoFrom( pcCU, uiAbsPartIdx);
#endif
for ( UInt uiPartIdx = 0, uiSubPartIdx = uiAbsPartIdx; uiPartIdx < uiNumPU; uiPartIdx++, uiSubPartIdx += uiPUOffset )
{
#if H_MV_ENC_DEC_TRAC
DTRACE_PU_S("=========== prediction_unit ===========\n")
// ToDo:
//DTRACE_PU("x0", uiLPelX)
//DTRACE_PU("x1", uiTPelY)
#endif
decodeMergeFlag( pcCU, uiSubPartIdx, uiDepth, uiPartIdx );
if ( pcCU->getMergeFlag( uiSubPartIdx ) )
{
decodeMergeIndex( pcCU, uiPartIdx, uiSubPartIdx, uiDepth );
UInt uiMergeIndex = pcCU->getMergeIndex(uiSubPartIdx);
#if H_3D_IC
decodeICFlag( pcCU, uiAbsPartIdx, uiDepth );
#endif
#if H_3D_ARP
decodeARPW ( pcCU, uiAbsPartIdx, uiDepth );
#endif
if ( pcCU->getSlice()->getPPS()->getLog2ParallelMergeLevelMinus2() && ePartSize != SIZE_2Nx2N && pcSubCU->getWidth( 0 ) <= 8 )
{
pcSubCU->setPartSizeSubParts( SIZE_2Nx2N, 0, uiDepth );
if ( !isMerged )
{
#if H_3D_VSP
Int vspFlag[MRG_MAX_NUM_CANDS_MEM];
memset(vspFlag, 0, sizeof(Int)*MRG_MAX_NUM_CANDS_MEM);
InheritedVSPDisInfo inheritedVSPDisInfo[MRG_MAX_NUM_CANDS_MEM];
#if H_3D_SPIVMP
memset(bSPIVMPFlag, false, sizeof(Bool)*MRG_MAX_NUM_CANDS_MEM);
#endif
pcSubCU->initAvailableFlags();
pcSubCU->getInterMergeCandidates( 0, 0, cMvFieldNeighbours, uhInterDirNeighbours, numValidMergeCand);
pcSubCU->xGetInterMergeCandidates( 0, 0, cMvFieldNeighbours, uhInterDirNeighbours, vspFlag, inheritedVSPDisInfo
#if H_3D_SPIVMP
, bSPIVMPFlag, pcMvFieldSP, puhInterDirSP
#endif
, numValidMergeCand );
pcCU->setVSPFlagSubParts( vspFlag[uiMergeIndex], uiSubPartIdx, uiPartIdx, uiDepth );
if(vspFlag[uiMergeIndex])
{
pcCU->setDvInfoSubParts(inheritedVSPDisInfo[uiMergeIndex].m_acDvInfo, uiSubPartIdx, uiPartIdx, uiDepth);
}
#else
#if H_3D
pcSubCU->initAvailableFlags();
pcSubCU->getInterMergeCandidates( 0, 0, cMvFieldNeighbours, uhInterDirNeighbours, numValidMergeCand);
pcSubCU->xGetInterMergeCandidates( 0, 0, cMvFieldNeighbours, uhInterDirNeighbours, numValidMergeCand );
#else
pcSubCU->getInterMergeCandidates( 0, 0, cMvFieldNeighbours, uhInterDirNeighbours, numValidMergeCand );
#endif
#endif
isMerged = true;
}
pcSubCU->setPartSizeSubParts( ePartSize, 0, uiDepth );
}
else
{
uiMergeIndex = pcCU->getMergeIndex(uiSubPartIdx);
#if H_3D_VSP
Int vspFlag[MRG_MAX_NUM_CANDS_MEM];
memset(vspFlag, 0, sizeof(Int)*MRG_MAX_NUM_CANDS_MEM);
InheritedVSPDisInfo inheritedVSPDisInfo[MRG_MAX_NUM_CANDS_MEM];
#if H_3D_SPIVMP
memset(bSPIVMPFlag, false, sizeof(Bool)*MRG_MAX_NUM_CANDS_MEM);
#endif
pcSubCU->initAvailableFlags();
pcSubCU->getInterMergeCandidates( uiSubPartIdx-uiAbsPartIdx, uiPartIdx, cMvFieldNeighbours, uhInterDirNeighbours, numValidMergeCand, uiMergeIndex );
pcSubCU->xGetInterMergeCandidates( uiSubPartIdx-uiAbsPartIdx, uiPartIdx, cMvFieldNeighbours, uhInterDirNeighbours, vspFlag, inheritedVSPDisInfo
#if H_3D_SPIVMP
, bSPIVMPFlag, pcMvFieldSP, puhInterDirSP
#endif
,numValidMergeCand, uiMergeIndex );
pcCU->setVSPFlagSubParts( vspFlag[uiMergeIndex], uiSubPartIdx, uiPartIdx, uiDepth );
if(vspFlag[uiMergeIndex])
{
pcCU->setDvInfoSubParts(inheritedVSPDisInfo[uiMergeIndex].m_acDvInfo, uiSubPartIdx, uiPartIdx, uiDepth);
}
#else
#if H_3D
pcSubCU->initAvailableFlags();
pcSubCU->getInterMergeCandidates( uiSubPartIdx-uiAbsPartIdx, uiPartIdx, cMvFieldNeighbours, uhInterDirNeighbours, numValidMergeCand, uiMergeIndex );
pcSubCU->xGetInterMergeCandidates( uiSubPartIdx-uiAbsPartIdx, uiPartIdx, cMvFieldNeighbours, uhInterDirNeighbours, numValidMergeCand, uiMergeIndex );
#else
pcSubCU->getInterMergeCandidates( uiSubPartIdx-uiAbsPartIdx, uiPartIdx, cMvFieldNeighbours, uhInterDirNeighbours, numValidMergeCand, uiMergeIndex );
#endif
#endif
}
pcCU->setInterDirSubParts( uhInterDirNeighbours[uiMergeIndex], uiSubPartIdx, uiPartIdx, uiDepth );
TComMv cTmpMv( 0, 0 );
for ( UInt uiRefListIdx = 0; uiRefListIdx < 2; uiRefListIdx++ )
{
if ( pcCU->getSlice()->getNumRefIdx( RefPicList( uiRefListIdx ) ) > 0 )
{
pcCU->setMVPIdxSubParts( 0, RefPicList( uiRefListIdx ), uiSubPartIdx, uiPartIdx, uiDepth);
pcCU->setMVPNumSubParts( 0, RefPicList( uiRefListIdx ), uiSubPartIdx, uiPartIdx, uiDepth);
pcCU->getCUMvField( RefPicList( uiRefListIdx ) )->setAllMvd( cTmpMv, ePartSize, uiSubPartIdx, uiDepth, uiPartIdx );
pcCU->getCUMvField( RefPicList( uiRefListIdx ) )->setAllMvField( cMvFieldNeighbours[ 2*uiMergeIndex + uiRefListIdx ], ePartSize, uiSubPartIdx, uiDepth, uiPartIdx );
}
}
#if H_3D_SPIVMP
pcCU->setSPIVMPFlagSubParts(bSPIVMPFlag[uiMergeIndex], uiSubPartIdx, uiPartIdx, uiDepth );
if (bSPIVMPFlag[uiMergeIndex] != 0)
{
Int iWidth, iHeight;
UInt uiIdx;
pcCU->getPartIndexAndSize( uiPartIdx, uiIdx, iWidth, iHeight, uiSubPartIdx, true );
UInt uiSPAddr;
Int iNumSPInOneLine, iNumSP, iSPWidth, iSPHeight;
pcCU->getSPPara(iWidth, iHeight, iNumSP, iNumSPInOneLine, iSPWidth, iSPHeight);
for (Int iPartitionIdx = 0; iPartitionIdx < iNumSP; iPartitionIdx++)
{
pcCU->getSPAbsPartIdx(uiSubPartIdx, iSPWidth, iSPHeight, iPartitionIdx, iNumSPInOneLine, uiSPAddr);
pcCU->setInterDirSP(puhInterDirSP[iPartitionIdx], uiSPAddr, iSPWidth, iSPHeight);
pcCU->getCUMvField( REF_PIC_LIST_0 )->setMvFieldSP(pcCU, uiSPAddr, pcMvFieldSP[2*iPartitionIdx], iSPWidth, iSPHeight);
pcCU->getCUMvField( REF_PIC_LIST_1 )->setMvFieldSP(pcCU, uiSPAddr, pcMvFieldSP[2*iPartitionIdx + 1], iSPWidth, iSPHeight);
}
}
#endif
}
else
{
decodeInterDirPU( pcCU, uiSubPartIdx, uiDepth, uiPartIdx );
for ( UInt uiRefListIdx = 0; uiRefListIdx < 2; uiRefListIdx++ )
{
if ( pcCU->getSlice()->getNumRefIdx( RefPicList( uiRefListIdx ) ) > 0 )
{
decodeRefFrmIdxPU( pcCU, uiSubPartIdx, uiDepth, uiPartIdx, RefPicList( uiRefListIdx ) );
decodeMvdPU ( pcCU, uiSubPartIdx, uiDepth, uiPartIdx, RefPicList( uiRefListIdx ) );
decodeMVPIdxPU ( pcSubCU, uiSubPartIdx-uiAbsPartIdx, uiDepth, uiPartIdx, RefPicList( uiRefListIdx ) );
}
}
#if H_3D_IC
decodeICFlag( pcCU, uiAbsPartIdx, uiDepth );
#endif
#if H_3D_ARP
decodeARPW ( pcCU, uiAbsPartIdx, uiDepth );
#endif
}
#if H_3D_VSP
if ( (pcCU->getInterDir(uiSubPartIdx) == 3) && pcSubCU->isBipredRestriction(uiPartIdx) && (pcCU->getVSPFlag(uiSubPartIdx) == false))
#else
if ( (pcCU->getInterDir(uiSubPartIdx) == 3) && pcSubCU->isBipredRestriction(uiPartIdx) )
#endif
{
pcCU->getCUMvField( REF_PIC_LIST_1 )->setAllMv( TComMv(0,0), ePartSize, uiSubPartIdx, uiDepth, uiPartIdx);
pcCU->getCUMvField( REF_PIC_LIST_1 )->setAllRefIdx( -1, ePartSize, uiSubPartIdx, uiDepth, uiPartIdx);
pcCU->setInterDirSubParts( 1, uiSubPartIdx, uiPartIdx, uiDepth);
}
}
#if H_3D_SPIVMP
delete[] pcMvFieldSP;
delete[] puhInterDirSP;
#endif
return;
}
/** decode motion information for every PU block.
* \param pcCU
* \param uiAbsPartIdx
* \param uiDepth
* \param pcSubCU
* \returns Void
*/
Void TDecEntropy::decodePUWise( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, TComDataCU* pcSubCU )
{
PartSize ePartSize = pcCU->getPartitionSize( uiAbsPartIdx );
UInt uiNumPU = ( ePartSize == SIZE_2Nx2N ? 1 : ( ePartSize == SIZE_NxN ? 4 : 2 ) );
UInt uiPUOffset = ( g_auiPUOffset[UInt( ePartSize )] << ( ( pcCU->getSlice()->getSPS()->getMaxCUDepth() - uiDepth ) << 1 ) ) >> 4;
TComMvField cMvFieldNeighbours[MRG_MAX_NUM_CANDS << 1]; // double length for mv of both lists
UChar uhInterDirNeighbours[MRG_MAX_NUM_CANDS];
for ( UInt ui = 0; ui < MRG_MAX_NUM_CANDS; ui++ )
{
uhInterDirNeighbours[ui] = 0;
}
Int numValidMergeCand = 0;
bool isMerged = false;
pcSubCU->copyInterPredInfoFrom( pcCU, uiAbsPartIdx, REF_PIC_LIST_0 );
pcSubCU->copyInterPredInfoFrom( pcCU, uiAbsPartIdx, REF_PIC_LIST_1 );
for ( UInt uiPartIdx = 0, uiSubPartIdx = uiAbsPartIdx; uiPartIdx < uiNumPU; uiPartIdx++, uiSubPartIdx += uiPUOffset )
{
decodeMergeFlag( pcCU, uiSubPartIdx, uiDepth, uiPartIdx );
if ( pcCU->getMergeFlag( uiSubPartIdx ) )
{
decodeMergeIndex( pcCU, uiPartIdx, uiSubPartIdx, ePartSize, uhInterDirNeighbours, cMvFieldNeighbours, uiDepth );
UInt uiMergeIndex = pcCU->getMergeIndex(uiSubPartIdx);
if ( pcCU->getSlice()->getPPS()->getLog2ParallelMergeLevelMinus2() && ePartSize != SIZE_2Nx2N && pcSubCU->getWidth( 0 ) <= 8 )
{
pcSubCU->setPartSizeSubParts( SIZE_2Nx2N, 0, uiDepth );
if ( !isMerged )
{
pcSubCU->getInterMergeCandidates( 0, 0, uiDepth, cMvFieldNeighbours, uhInterDirNeighbours, numValidMergeCand );
isMerged = true;
}
pcSubCU->setPartSizeSubParts( ePartSize, 0, uiDepth );
}
else
{
uiMergeIndex = pcCU->getMergeIndex(uiSubPartIdx);
pcSubCU->getInterMergeCandidates( uiSubPartIdx-uiAbsPartIdx, uiPartIdx, uiDepth, cMvFieldNeighbours, uhInterDirNeighbours, numValidMergeCand, uiMergeIndex );
}
pcCU->setInterDirSubParts( uhInterDirNeighbours[uiMergeIndex], uiSubPartIdx, uiPartIdx, uiDepth );
TComMv cTmpMv( 0, 0 );
for ( UInt uiRefListIdx = 0; uiRefListIdx < 2; uiRefListIdx++ )
{
if ( pcCU->getSlice()->getNumRefIdx( RefPicList( uiRefListIdx ) ) > 0 )
{
pcCU->setMVPIdxSubParts( 0, RefPicList( uiRefListIdx ), uiSubPartIdx, uiPartIdx, uiDepth);
pcCU->setMVPNumSubParts( 0, RefPicList( uiRefListIdx ), uiSubPartIdx, uiPartIdx, uiDepth);
pcCU->getCUMvField( RefPicList( uiRefListIdx ) )->setAllMvd( cTmpMv, ePartSize, uiSubPartIdx, uiDepth, uiPartIdx );
pcCU->getCUMvField( RefPicList( uiRefListIdx ) )->setAllMvField( cMvFieldNeighbours[ 2*uiMergeIndex + uiRefListIdx ], ePartSize, uiSubPartIdx, uiDepth, uiPartIdx );
}
}
}
else
{
decodeInterDirPU( pcCU, uiSubPartIdx, uiDepth, uiPartIdx );
for ( UInt uiRefListIdx = 0; uiRefListIdx < 2; uiRefListIdx++ )
{
if ( pcCU->getSlice()->getNumRefIdx( RefPicList( uiRefListIdx ) ) > 0 )
{
decodeRefFrmIdxPU( pcCU, uiSubPartIdx, uiDepth, uiPartIdx, RefPicList( uiRefListIdx ) );
decodeMvdPU ( pcCU, uiSubPartIdx, uiDepth, uiPartIdx, RefPicList( uiRefListIdx ) );
decodeMVPIdxPU ( pcSubCU, uiSubPartIdx-uiAbsPartIdx, uiDepth, uiPartIdx, RefPicList( uiRefListIdx ) );
}
}
}
if ( (pcCU->getInterDir(uiSubPartIdx) == 3) && pcSubCU->isBipredRestriction(uiPartIdx) )
{
pcCU->getCUMvField( REF_PIC_LIST_1 )->setAllMv( TComMv(0,0), ePartSize, uiSubPartIdx, uiDepth, uiPartIdx);
pcCU->getCUMvField( REF_PIC_LIST_1 )->setAllRefIdx( -1, ePartSize, uiSubPartIdx, uiDepth, uiPartIdx);
pcCU->setInterDirSubParts( 1, uiSubPartIdx, uiPartIdx, uiDepth);
}
}
return;
}
