hevc角度预测_HEVC帧间预测之6——运动估计（三）

HEVC帧间预测之六——运动估计(三)

本文分析xTZSearch调用了两个最为主要的函数：xTZ8PointDiamondSearch和xTZ2PointSearch，值得一提的是，HM中还提供了另外一个搜索函数xTZ8PointSquareSearch，但由于实际并没有使用这个函数，且它其实跟钻石搜索只是搜索点的选择略有不同，分析起来基本上也是一样的，这里就不重复啰嗦了。

__inline Void TEncSearch::xTZ2PointSearch( TComPattern* pcPatternKey, IntTZSearchStruct& rcStruct, TComMv* pcMvSrchRngLT, TComMv* pcMvSrchRngRB )

{

Int iSrchRngHorLeft = pcMvSrchRngLT->getHor();

Int iSrchRngHorRight = pcMvSrchRngRB->getHor();

Int iSrchRngVerTop = pcMvSrchRngLT->getVer();

Int iSrchRngVerBottom = pcMvSrchRngRB->getVer();

// 2 point search,// 1 2 3

// check only the 2 untested points// 4 0 5

// around the start point// 6 7 8

//!< 注意到，这里的1,3,6,8实际上是搜索步长iDist==2的时候由iDist>>1进行赋值的，实际距离以1计算，

//!< 在前面经过xTZ8PointSearch确定最佳步长为1后，会在这里对以最佳点为中心、周围没搜索过的点进行运动估计

Int iStartX = rcStruct.iBestX;

Int iStartY = rcStruct.iBestY;

switch( rcStruct.ucPointNr ) //!< 主要思想是根据该最佳点的位置处理未搜索过的点

{

case 1:

{

if ( (iStartX - 1) >= iSrchRngHorLeft )

{

xTZSearchHelp( pcPatternKey, rcStruct, iStartX - 1, iStartY, 0, 2 ); //!< 左

}

if ( (iStartY - 1) >= iSrchRngVerTop )

{

xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iStartY - 1, 0, 2 ); //!< 上

}

}

break;

case 2:

{

if ( (iStartY - 1) >= iSrchRngVerTop )

{

if ( (iStartX - 1) >= iSrchRngHorLeft )

{

xTZSearchHelp( pcPatternKey, rcStruct, iStartX - 1, iStartY - 1, 0, 2 ); //!< 左上

}

if ( (iStartX + 1) <= iSrchRngHorRight )

{

xTZSearchHelp( pcPatternKey, rcStruct, iStartX + 1, iStartY - 1, 0, 2 ); //!< 右上

}

}

}

break;

case 3:

{

if ( (iStartY - 1) >= iSrchRngVerTop )

{

xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iStartY - 1, 0, 2 ); //!< 上

}

if ( (iStartX + 1) <= iSrchRngHorRight )

{

xTZSearchHelp( pcPatternKey, rcStruct, iStartX + 1, iStartY, 0, 2 );//!< 右

}

}

break;

case 4:

{

if ( (iStartX - 1) >= iSrchRngHorLeft )

{

if ( (iStartY + 1) <= iSrchRngVerBottom )

{

x