相信只要是做算法改进的,首先都会遇到这么一个问题:CU,PU及TU这几个在HM中该如何打印出它们最终的划分情况呢?也经常有人来问我这个问题,一般来说,因为问我的时候我一般手头都没有现成的代码可以提供,另一方面,也没空把vs打开写上一段,所以只能是告诉他们大概的思路,其实我很早之前的一篇博客已经进行了介绍,只是当时没有把代码附上,后来也找不到了,就不了了之了。本文算是把这个遗憾给补上吧。直接附上代码,由于功能简单,就不注释了~
printf(" ---------------CTU Address(Raster Scan): %d----------------- ", pcCU->getAddr()); printf(" ---------------CU starts-------------- "); int iCount = 0; int iWidthInPart = g_uiMaxCUWidth >> 2; for (int i = 0; i < pcCU->getTotalNumPart(); i++) { if ( (iCount & (iWidthInPart - 1)) == 0) printf(" "); printf("%d ", pcCU->getDepth(g_auiRasterToZscan[i])); iCount++; } printf(" ---------------CU ends-------------- "); printf(" ---------------TU starts-------------- "); iCount = 0; for (int i = 0; i < pcCU->getTotalNumPart(); i++) { if ( (iCount & (iWidthInPart - 1)) == 0) printf(" "); printf("%d ", pcCU->getTransformIdx(g_auiRasterToZscan[i]) + pcCU->getDepth(g_auiRasterToZscan[i])); iCount++; } printf(" ---------------TU ends-------------- ");
上面这一段程序,以4x4块为单位进行打印,数字范围0~4, 0表示不划分,1表示处于深度1,依此类推。其实,只要认真研究下TComDataCU这个类,基本上你需要的那些统计信息都可以通过类似以上方法进行统计。如果想做成类似H264visa这样的分析工具,那么基于这些信息,剩下的就是图形化界面的工作了。