HEVC里面CU与TU打印到屏幕及提取到txt文本

爱被打了一巴掌 2022-07-13 06:37 124阅读 0赞

测试模型为HM16.5

在解决方案资源管理器里面找到项目TlibEncoder，会看到里面有个源文件TencCU.cpp

（1）打印到屏幕

（在TencCU.cpp的xCompressCU函数（237行）后面添加，即可提取每个CU的相关信息并将其打印到屏幕）

printf("\\n---------------CTUAddress(Raster Scan): %d-----------------\\n",pCtu->getCtuRsAddr());

printf("\\n---------------CUstarts--------------\\n");

int iCount =0;

TComSlice \*constpcSlice = pCtu->getSlice();

constTComSPS   &sps=\*(pcSlice->getSPS());

intiWidthInPart = sps.getMaxCUWidth() >> 2;

for (int i = 0; i < pCtu->getTotalNumPart();i++)

if ( (iCount & (iWidthInPart - 1)) == 0)

printf("\\n");

printf("%d ",pCtu->getDepth(g\_auiRasterToZscan\[i\]));

iCount++;

printf("\\n---------------CUends--------------\\n");

printf("\\n---------------TUstarts--------------\\n");

iCount = 0;

for (int i = 0; i < pCtu->getTotalNumPart();i++)

if ( (iCount & (iWidthInPart - 1)) == 0)

printf("\\n");

printf("%d ",pCtu->getTransformIdx(g\_auiRasterToZscan\[i\]) +pCtu->getDepth(g\_auiRasterToZscan\[i\]));

iCount++;

printf("\\n---------------TUends--------------\\n");

（2） 将CU的划分方式提取出来并存储到txt文本

需要添加头文件include<stdlib.h>

然后定义一个文件指针 FILE \*fp;

创建一个txt文件，例如命名为BasketballDrill\_spilt\_CU.txt，注意后面"w"是打开文件方式,将原有内容清空写入新数据,如果文件不存在则新建,类似的还有"r"只读打开(不能写入);"a"向文件中追加内容,原有内容不变,等等

所以要用“a”的形式，以免循环后数值被覆盖

fp = fopen ("BasketballDrill\_spilt\_CU.txt", "a");

然后将打印的printf注释掉，在（1）中相同的位置添加如下代码即可。

注：要等项目运行完再打开创建的txt文本。

// 添加 开始：CU和TU划分  
  printf("\\n---------------CTU Address(Raster Scan): %d-----------------\\n", pCtu->getCtuRsAddr());    
  printf("\\n---------------CU starts--------------\\n");    
  FILE \*fp;  
  fp = fopen ("BasketballDrill\_spilt\_CU.txt", "a");  
  
  
  int iCount = 0;    
  
  
  TComSlice \*const pcSlice = pCtu->getSlice();  
  const TComSPS   &sps =\*(pcSlice->getSPS());  
  
  
  int iWidthInPart = sps.getMaxCUWidth() >> 2;    
  
  
  for (int i = 0; i < pCtu->getTotalNumPart(); i++)    
  \{    
 if ( (iCount & (iWidthInPart - 1)) == 0)    
 // printf("\\n");    
 fprintf(fp,"\\n");  
  
  
// printf("%d ", pCtu->getDepth(g\_auiRasterToZscan\[i\]));    
 iCount++;    
 fprintf(fp,"%d ",pCtu->getDepth(g\_auiRasterToZscan\[i\]));  
   \}    
    
   printf("\\n---------------CU ends--------------\\n");    
   printf("\\n---------------TU starts--------------\\n");    
  
  
   iCount = 0;    
  
  
   for (int i = 0; i < pCtu->getTotalNumPart(); i++)    
   \{   
 if ( (iCount & (iWidthInPart - 1)) == 0)    
 //printf("\\n");     
  
  
 //printf("%d ", pCtu->getTransformIdx(g\_auiRasterToZscan\[i\]) + pCtu->getDepth(g\_auiRasterToZscan\[i\]));    
 iCount++;    
    \}    
   // printf("\\n---------------TU ends--------------\\n");    
  
  
  // 添加 结束：CU和TU划分