欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

HEVC多次进行熵编码的原因

程序员文章站 2024-01-11 15:07:34
...

HEVC多次进行熵编码的原因



    在HM中有个让人很疑惑的地方,就是熵编码会被多次调用
    1、compressSlice中有两次
        (1)第一次是compressCU中,使用熵编码来进行RDO优化,用来选择最优的编码参数
        (2)第二次是紧接着compressCU后面调用了encodeCU
    2、encodeSlice中有一次
        (1)调用encodeCU


原因解释


    后来有大神解释了一下:
    1、在compressCU中调用熵编码,是为了进行RDO优化,这个比较好理解
    2、compressCU后面调用encodeCU,是为了保持CABAC的状态,因为熵编码是以slice为单位进行的,在一个slice中要保证这种状态的连续性,RDO的时候,在开始编码第二个CTU的时候的CABAC的状态得保证和第一个CTU熵编码完了以后的CABAC状态是一样的,因此要调用encodeCU
    3、在encodeSlice中调用encodeCU的目的是真正的熵编码
    4、另外需要注意的是,第一、二次调用熵编码操作不会写入比特流中,只有第三次调用才会写入比特流中。

HEVC多次进行熵编码的原因

HEVC多次进行熵编码的原因



在代码中的实现


Void TEncSlice::compressSlice( TComPic*& rpcPic )
{
	//***省略

	TEncTop* pcEncTop = (TEncTop*) m_pcCfg;


	TEncSbac**** ppppcRDSbacCoders    = pcEncTop->getRDSbacCoders();

	// 比特计数器
	TComBitCounter* pcBitCounters     = pcEncTop->getBitCounters();

	
	// ***省略

	for( uiEncCUOrder = uiStartCUAddr/rpcPic->getNumPartInCU();
		uiEncCUOrder < (uiBoundingCUAddr+(rpcPic->getNumPartInCU()-1))/rpcPic->getNumPartInCU();
		uiCUAddr = rpcPic->getPicSym()->getCUOrderMap(++uiEncCUOrder) )
	{
		// ***省略

		// set go-on entropy coder
		m_pcEntropyCoder->setEntropyCoder ( m_pcRDGoOnSbacCoder, pcSlice );
		// 注意这里,pcBitCounters是比特计数器,它和比特流类有相同的接口,但是只是用于比特计数,不会写数据到比特流中
		m_pcEntropyCoder->setBitstream( &pcBitCounters[uiSubStrm] );

		// 启用比特计数的功能
		((TEncBinCABAC*)m_pcRDGoOnSbacCoder->getEncBinIf())->setBinCountingEnableFlag(true);

		// ***
		
		// 选择最优的编码模式
		m_pcCuEncoder->compressCU( pcCU );

		// restore entropy coder to an initial stage
		m_pcEntropyCoder->setEntropyCoder ( m_pppcRDSbacCoder[0][CI_CURR_BEST], pcSlice );

		// 同样还是比特计数器
		m_pcEntropyCoder->setBitstream( &pcBitCounters[uiSubStrm] );
		m_pcCuEncoder->setBitCounter( &pcBitCounters[uiSubStrm] );

		m_pcBitCounter = &pcBitCounters[uiSubStrm];
		pppcRDSbacCoder->setBinCountingEnableFlag( true );
		m_pcBitCounter->resetBits();
		pppcRDSbacCoder->setBinsCoded( 0 );

		// 为了在一个slice保证CABAC状态的连续性,因为CABAC是以slice为单位的
		m_pcCuEncoder->encodeCU( pcCU );

		
		// *** 省略
	} // for end


	// ***省略
}


Void TEncSlice::encodeSlice   ( TComPic*& rpcPic, TComOutputBitstream* pcSubstreams )
{
	//*** 省略

	// 遍历条带中的每一个CU
	for( uiEncCUOrder = uiStartCUAddr /rpcPic->getNumPartInCU();
		uiEncCUOrder < (uiBoundingCUAddr+rpcPic->getNumPartInCU()-1)/rpcPic->getNumPartInCU();
		uiCUAddr = rpcPic->getPicSym()->getCUOrderMap(++uiEncCUOrder) )
	{
		// ***省略

		// 注意这里,pcSubstreams是比特流,不是比特计数器,因此,熵编码后会将数据写入比特流中
		m_pcEntropyCoder->setBitstream( &pcSubstreams[uiSubStrm] );
		
		// *** 省略
		if ( (m_pcCfg->getSliceMode()!=0 || m_pcCfg->getSliceSegmentMode()!=0) &&
			uiCUAddr == rpcPic->getPicSym()->getCUOrderMap((uiBoundingCUAddr+rpcPic->getNumPartInCU()-1)/rpcPic->getNumPartInCU()-1) )
		{
			// 和compressSlice一样调用了encodeCU函数,但是实现的功能已经不一样了
			// 因为compressSlice主要是进行帧内(帧间)最优模式的选择,还有变换、量化等功能
			// 而在这里,由于前面的几个步骤都已经进行完毕,所以,这里是单纯的进行熵编码
			m_pcCuEncoder->encodeCU( pcCU );
		}
		else
		{
			m_pcCuEncoder->encodeCU( pcCU );
		}

		// ***省略
	}
	// ***省略
}