HM中CU，TU的划分

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

 

	  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 < 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 < 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这样的分析工具，那么基于这些信息，剩下的就是图形化界面的工作了。