HEVC-SCC中打印CU的IBC和Palette的分布

对于屏幕内容图像编码，帧内预测包括普通帧内预测模式Intra、帧内块复制IntraBlockCopy、调色板Palette模式。这里介绍CTU划分为CU和标记CU的最佳模式。
xCompressCU( m_ppcBestCU[0], m_ppcTempCU[0], 0 DEBUG_STRING_PASS_INTO(sDebug) );函数是对一个CU进行编码，编码完之后最佳模式的信息保存在m_ppcBestCU[0]。所以在这个函数之后统计每个CU的深度信息。
这里使用的编码器是HM-16.18+SCM-8.7。

Void TEncCu::compressCtu( TComDataCU* pCtu, UChar* lastPaletteSize, Pel lastPalette[][MAX_PALETTE_PRED_SIZE] )
{
   ......
   xCompressCU( m_ppcBestCU[0], m_ppcTempCU[0], 0 DEBUG_STRING_PASS_INTO(sDebug) );
   ofstream  CUInfo,IBCInfo,PLTInfo;
   TComDataCU*  IBCCu = m_ppcBestCU[0];
   CUInfo.open("cudepth.txt", ios::app);//CU的深度信息保存在cudepth.txt文件中
   IBCInfo.open("ibcdepth.txt", ios::app);//使用IBC的CU的深度信息保存在ibcdepth.txt文件中
   PLTInfo.open("pltdepth.txt", ios::app);//使用了Palette的CU的深度信息保存在pltdepth.txt文件中
   for (UInt iPartitionNum = 0; iPartitionNum < IBCCu->getTotalNumPart(); iPartitionNum++)
   {
		CUInfo <<IBCCu->getDepth()[iPartitionNum];
		if(IBCCu->isIntraBC(iPartitionNum))
			IBCInfo << (int)IBCCu->getDepth()[iPartitionNum];
		else
		{
			IBCInfo <<(UChar) 0;
		}
		if (IBCCu->getPaletteModeFlag(iPartitionNum))
		   PLTInfo << IBCCu->getDepth()[iPartitionNum];
	    else
		   PLTInfo << (UChar)0;
	}
	CUInfo.close();
	IBCInfo.close();
	PLTInfo.close();
}

编码器运行完之后就会生成以上三个文件，将码流str.bin解码出来dec.yuv视频序列。用三个深度信息文件和dec.yuv文件，在MATLAB中对图像进行操作。MATLAB的代码如下：
function LCU_View_IBC(YUV_FileName,LCU_IBC_FileName,LCU_FileName,Width, Height, NumFrame)
这个函数表示标记CU的IBC分布。
function LCU_View_PLT(YUV_FileName,LCU_IBC_FileName,LCU_PLT_FileName,LCU_FileName,Width, Height, NumFrame)
这个函数表示标记CU的IBC和Palette分布。
具体的代码下载地址为：https://download.csdn.net/download/weixin_42147726/11990211
https://download.csdn.net/download/weixin_42147726/11990193
MATLAB运行的结果如图：

从这里可以看到CU的划分，黑色框框标记的是一个CTU，蓝色标记的是使用普通帧内的CU，绿色标记的是使用Palette的CU，红色标记的是使用IBC模式的CU。