HM中CU，TU的划分

转载于这位大神的博客，为了更方便的学习：http://blog.csdn.net/hevc_cjl/article/details/9839331

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

1. printf("\n---------------CTU Address(Raster Scan): %d-----------------\n", pcCU->getAddr());  
2. printf("\n---------------CU starts--------------\n");  
3. int iCount = 0;  
4. int iWidthInPart = g_uiMaxCUWidth >> 2;  
5. for (int i = 0; i < pcCU->getTotalNumPart(); i++)  
6. {  
7.  if ( (iCount & (iWidthInPart - 1)) == 0)  
8.   printf("\n");  
9.   
10.  printf("%d ", pcCU->getDepth(g_auiRasterToZscan[i]));  
11.  iCount++;  
12. }  
13. printf("\n---------------CU ends--------------\n");  
14. printf("\n---------------TU starts--------------\n");  
15. iCount = 0;  
16. for (int i = 0; i < pcCU->getTotalNumPart(); i++)  
17. {  
18.  if ( (iCount & (iWidthInPart - 1)) == 0)  
19.   printf("\n");  
20.   
21.  printf("%d ", pcCU->getTransformIdx(g_auiRasterToZscan[i]) + pcCU->getDepth(g_auiRasterToZscan[i]));  
22.  iCount++;  
23. }  
24. printf("\n---------------TU ends--------------\n"); 
    

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