有了前面几篇的铺垫,本文就可以把整像素部分的运动估计给结束掉了。到目前为止,只剩下xTZSearch这个函数没分析了,在开始这个函数的代码解释之前,让我们共同来理一下TZSearch的基本流程:
1. 搜索预测得到的mv所指向的点:中值预测mv,当前PU的左,上及右上PU的mv,还有零运动矢量(0,0)
2. 在步骤1中找到匹配误差最小的点作为接下来搜索的起始点
3. 步长从1开始,以2的指数递增,进行8点钻石搜索,该步骤中可以设置搜索的最大次数(以某个步长遍历一遍就算1次)
4. 如果步骤3搜索得到的最佳步长为1,则需要以该最佳点为起点做1次两点钻石搜索,因为前面8点搜索的时候,这个最佳点的8个邻点会有两个没有搜索到
5. 如果步骤3搜索得到的最佳步长大于某个阈值(iRaster),则以步骤2得到的点作为起点,做步长为iRaster的光栅扫描(即在运动搜索的范围内遍历所有点)
6. 最后,在经过前面1~5歩之后,以得到的最佳点为起点,再次重复步骤3和4
7. 保存最佳mv和SAD。
总结完毕,我只是把主线给列了出来,当中会有些不同,主要是我只考虑HM的默认配置,可以按照这个主线看下面代码及注释了。
Void TEncSearch::xTZSearch( TComDataCU* pcCU, TComPattern* pcPatternKey, Pel* piRefY, Int iRefStride, TComMv* pcMvSrchRngLT, TComMv* pcMvSrchRngRB, TComMv& rcMv, UInt& ruiSAD )
{//!< 确定运动估计搜索范围的边界
Int iSrchRngHorLeft = pcMvSrchRngLT->getHor();
Int iSrchRngHorRight = pcMvSrchRngRB->getHor();
Int iSrchRngVerTop = pcMvSrchRngLT->getVer();
Int iSrchRngVerBottom = pcMvSrchRngRB->getV