【转载】CU、PU、TU划分显示代码

来源：互联网发布：vue main.js引入less 编辑：程序博客网时间：2024/04/26 13:27

转载自 http://blog.sina.com.cn/s/blog_7b7ddaf90101ftvl.html

相信只要是做算法改进的，首先都会遇到这么一个问题：CU，PU及TU这几个在HM中该如何打印出它们最终的划分情况呢？也经常有人来问我这个问题，一般来说，因为问我的时候我一般手头都没有现成的代码可以提供，另一方面，也没空把vs打开写上一段，所以只能是告诉他们大概的思路，其实我很早之前的一篇博客已经进行了介绍，只是当时没有把代码附上，后来也找不到了，就不了了之了。本文算是把这个遗憾给补上吧。直接附上代码，由于功能简单，就不注释了~

[html]view plaincopy
printf("\n---------------CTU Address(Raster Scan): %d-----------------\n", pcCU->getAddr());  
printf("\n---------------CU starts--------------\n");  
int iCount = 0;  
int iWidthInPart = g_uiMaxCUWidth >> 2;  
for (int i = 0; i <</span> pcCU->getTotalNumPart(); i++)  
{  
 if ( (iCount & (iWidthInPart - 1)) == 0)  
  printf("\n");  
  
 printf("%d ", pcCU->getDepth(g_auiRasterToZscan[i]));  
 iCount++;  
}  
printf("\n---------------CU ends--------------\n");  
printf("\n---------------TU starts--------------\n");  
iCount = 0;  
for (int i = 0; i <</span> pcCU->getTotalNumPart(); i++)  
{  
 if ( (iCount & (iWidthInPart - 1)) == 0)  
  printf("\n");  
  
 printf("%d ", pcCU->getTransformIdx(g_auiRasterToZscan[i]) + pcCU->getDepth(g_auiRasterToZscan[i]));  
 iCount++;  
}  
printf("\n---------------TU ends--------------\n");  

上面这一段程序，以4x4块为单位进行打印，数字范围0~4, 0表示不划分，1表示处于深度1，依此类推。其实，只要认真研究下TComDataCU这个类，基本上你需要的那些统计信息都可以通过类似以上方法进行统计。如果想做成类似H264visa这样的分析工具，那么基于这些信息，剩下的就是图形化界面的工作了。

0 0