fillMvpCand在xEstimateMvPredAMVP函数中被调用,用来构建AMVP列表,将AMVP候选存储于amvpInfo
AMVP列表的构建,添加顺序为:
1.左下、左pu的无缩放信息
2.左下、左pu的缩放信息
3.右上、上、左上pu无缩放信息
4.如果左下和左的pu都无效,那么添加右上、上、左上pu的缩放信息
5.添加TMVP
6.补0
其中,调用addMVPCandUnscaled函数添加相应位置处的pu运动信息,调用addMVPCandWithScaling函数则用来添加相应位置处pu运动信息经过scale之后的运动信息,scale方式就是和tmvp中一样的按照poc距离缩放
void PU::fillMvpCand(PredictionUnit &pu, const RefPicList &eRefPicList, const int &refIdx, AMVPInfo &amvpInfo)
{
CodingStructure &cs = *pu.cs;
AMVPInfo *pInfo = &amvpInfo; //AMVP列表
pInfo->numCand = 0; //AMVP候选的个数
if (refIdx < 0)
{
return;
}
//-- Get Spatial MV
Position posLT = pu.Y().topLeft();
Position posRT = pu.Y().topRight();
Position posLB = pu.Y().bottomLeft(); //pu的左上、右上、左下的pos
Bool isScaledFlagLX = false; /// variable name from specification; true when the PUs below left or left are available (availableA0 || availableA1).
{ //左下的pu信息
const PredictionUnit* tmpPU = cs.getPURestricted( posLB.offset( -1, 1 ), pu, pu.chType ); // getPUBelowLeft(idx, partIdxLB);
isScaledFlagLX = tmpPU != NULL && CU::isInter( *tmpPU->cu );
if( !isScaledFlagLX ) //左下pu不存在时,获取左侧pu信息
{
tmpPU = cs.getPURestricted( posLB.offset( -1, 0 ), pu, pu.chType );
isScaledFlagLX = tmpPU != NULL && CU::isInter( *tmpPU->cu );
}
}
// Left predictor search
if( isScaledFlagLX ) //左下、左侧pu其中之一存在,isScaledFlagLX为true
{
Bool bAdded = addMVPCandUnscaled( pu, eRefPicList, refIdx, posLB, MD_BELOW_LEFT, *pInfo ); //AMVP列表添加左下的pu信息(无缩放)
if( !bAdded )
{
bAdded = addMVPCandUnscaled( pu, eRefPicList, refIdx, posLB, MD_LEFT, *pInfo ); //左下pu无缩放不能添加,AMVP列表添加左侧的pu信息(无缩放)
if( !bAdded )
{ //左侧pu无缩放不能添加,AMVP列表添加左左下的pu信息(缩放)
bAdded = addMVPCandWithScaling( pu, eRefPicList, refIdx, posLB, MD_BELOW_LEFT, *pInfo );
if( !bAdded )
{ //左下pu信息缩放不能添加,AMVP列表添加左侧的pu信息(缩放)
addMVPCandWithScaling( pu, eRefPicList, refIdx, posLB, MD_LEFT, *pInfo );
}
}
}
}
// Above predictor search
{ //AMVP列表添加右上的pu信息(无缩放)
Bool bAdded = addMVPCandUnscaled( pu, eRefPicList, refIdx, posRT, MD_ABOVE_RIGHT, *pInfo );
if( !bAdded )
{ //右上pu无缩放不能添加,AMVP列表添加上侧的pu信息(无缩放)
bAdded = addMVPCandUnscaled( pu, eRefPicList, refIdx, posRT, MD_ABOVE, *pInfo );
if( !bAdded )
{ //上侧pu无缩放不能添加,AMVP列表添加左上的pu信息(无缩放)
addMVPCandUnscaled( pu, eRefPicList, refIdx, posLT, MD_ABOVE_LEFT, *pInfo );
}
}
}
if( !isScaledFlagLX ) //如果左侧和左下的pu都不可用
{ //AMVP列表添加右上的pu信息(缩放)
Bool bAdded = addMVPCandWithScaling( pu, eRefPicList, refIdx, posRT, MD_ABOVE_RIGHT, *pInfo );
if( !bAdded )
{ //右上pu缩放信息不能添加,AMVP列表添加上侧的pu信息(缩放)
bAdded = addMVPCandWithScaling( pu, eRefPicList, refIdx, posRT, MD_ABOVE, *pInfo );
if( !bAdded )
{ //上侧pu缩放信息不能添加,AMVP列表添加左上的pu信息(缩放)
addMVPCandWithScaling( pu, eRefPicList, refIdx, posLT, MD_ABOVE_LEFT, *pInfo );
}
}
}
if( pInfo->numCand == 2 )
{
if( pInfo->mvCand[0] == pInfo->mvCand[1] )
{
pInfo->numCand = 1; //AMVP列表冗余裁剪
}
}
if( cs.slice->getEnableTMVPFlag() ) //AMVP列表中添加TMVP的运动信息
{
// Get Temporal Motion Predictor
const int refIdx_Col = refIdx;
Position posRB = pu.Y().bottomRight().offset(-3, -3);
const PreCalcValues& pcv = *cs.pcv;
Position posC0;
bool C0Avail = false;
Position posC1 = pu.Y().center(); //C1为中心位置,C0右下位置
Mv cColMv;
if( ( ( posRB.x + pcv.minCUWidth ) < pcv.lumaWidth ) && ( ( posRB.y + pcv.minCUHeight ) < pcv.lumaHeight ) )
{
Position posInCtu( posRB.x & pcv.maxCUWidthMask, posRB.y & pcv.maxCUHeightMask );
if ((posInCtu.x + 4 < pcv.maxCUWidth) && // is not at the last column of CTU
(posInCtu.y + 4 < pcv.maxCUHeight)) // is not at the last row of CTU
{
posC0 = posRB.offset(4, 4);
C0Avail = true;
}
else if (posInCtu.x + 4 < pcv.maxCUWidth) // is not at the last column of CTU But is last row of CTU
{
// in the reference the CTU address is not set - thus probably resulting in no using this C0 possibility
posC0 = posRB.offset(4, 4);
}
else if (posInCtu.y + 4 < pcv.maxCUHeight) // is not at the last row of CTU But is last column of CTU
{
posC0 = posRB.offset(4, 4);
C0Avail = true;
}
else //is the right bottom corner of CTU
{
// same as for last column but not last row
posC0 = posRB.offset(4, 4);
}
}
//getColocatedMVP获取时域MVP
if ((C0Avail && getColocatedMVP(pu, eRefPicList, posC0, cColMv, refIdx_Col)) || getColocatedMVP(pu, eRefPicList, posC1, cColMv, refIdx_Col))
{
pInfo->mvCand[pInfo->numCand++] = cColMv; //获得到的TMVP添加到AMVP列表中
}
}
if (pInfo->numCand > AMVP_MAX_NUM_CANDS)
{
pInfo->numCand = AMVP_MAX_NUM_CANDS; //列表裁剪
}
while (pInfo->numCand < AMVP_MAX_NUM_CANDS)
{
pInfo->mvCand[pInfo->numCand] = Mv( 0, 0 ); //补0
pInfo->numCand++;
}
}
addMVPCandUnscaled函数用来向AMVP列表中添加相应位置处的pu运动信息
参数pos为pu的左下或右上或左上位置,eDir为pos对应位置处的位置偏移
bool PU::addMVPCandUnscaled( const PredictionUnit &pu, const RefPicList &eRefPicList, const int &iRefIdx, const Position &pos, const MvpDir &eDir, AMVPInfo &info )
{
CodingStructure &cs = *pu.cs;
const PredictionUnit *neibPU = NULL;
Position neibPos;
switch (eDir) //pos经过edir的偏移之后,得到需要添加的pu的位置
{
case MD_LEFT:
neibPos = pos.offset( -1, 0 );
break;
case MD_ABOVE:
neibPos = pos.offset( 0, -1 );
break;
case MD_ABOVE_RIGHT:
neibPos = pos.offset( 1, -1 ); //右上
break;
case MD_BELOW_LEFT:
neibPos = pos.offset( -1, 1 );
break;
case MD_ABOVE_LEFT:
neibPos = pos.offset( -1, -1 ); //左上
break;
default:
break;
}
neibPU = cs.getPURestricted( neibPos, pu, pu.chType ); //获取pu
if( neibPU == NULL || !CU::isInter( *neibPU->cu ) )
{
return false;
}
const MotionInfo& neibMi = neibPU->getMotionInfo( neibPos ); //pu的运动信息
const Int currRefPOC = cs.slice->getRefPic( eRefPicList, iRefIdx )->getPOC(); //当前图像的参考帧的poc
const RefPicList eRefPicList2nd = ( eRefPicList == REF_PIC_LIST_0 ) ? REF_PIC_LIST_1 : REF_PIC_LIST_0; //相反方向的参考列表
for( Int predictorSource = 0; predictorSource < 2; predictorSource++ ) // examine the indicated reference picture list, then if not available, examine the other list.
{ //两次循环,检查pu的两个参考方向的参考列表
const RefPicList eRefPicListIndex = ( predictorSource == 0 ) ? eRefPicList : eRefPicList2nd;
const Int neibRefIdx = neibMi.refIdx[eRefPicListIndex]; //pu对应inter方向的refIdx
if( neibRefIdx >= 0 && currRefPOC == cs.slice->getRefPOC( eRefPicListIndex, neibRefIdx ) )
{ //如果pu的参考帧存在,并且当前图像的参考帧和pu参考帧相同
{
info.mvCand[info.numCand++] = neibMi.mv[eRefPicListIndex]; //添加pu运动信息为AMVP候选
return true; //添加到一个候选就return
}
}
}
return false;
}