分析HM代碼,首先必須把class TComDataCU完全搞明白!
爲了找到合適的模式或者合適的分割, 基本上都是採用兩個TComDataCU來操作, TComDataCU*& rpcBestCU, TComDataCU*& rpcTempCU.
相當於兩個指針對應兩個乒乓buffer, m_ppcBestCU[uiDepth], m_ppcTempCU[uiDepth], 當然每一個uiDepth都有一個對應的BestCU和TempCU的buffer.
class TComDataCU
{
private:
// -------------------------------------------------------
// class pointers
// -------------------------------------------------------
TComPic* m_pcPic; ///< picture class pointer
TComSlice* m_pcSlice; ///< slice header pointer
TComPattern* m_pcPattern; ///< neighbour access class pointer
// -------------------------------------------------------
// CU description
// -------------------------------------------------------
UInt m_uiCUAddr; ///< CU address in a slice
UInt m_uiAbsIdxInLCU; ///< absolute address in a CU. It's Z scan order
UInt m_uiCUPelX; ///< CU position in a pixel (X)
UInt m_uiCUPelY; ///< CU position in a pixel (Y)
UInt m_uiNumPartition; ///< total number of minimum partitions in a CU
UChar* m_puhWidth; ///< array of widths
UChar* m_puhHeight; ///< array of heights
UChar* m_puhDepth; ///< array of depths
Int m_unitSize; ///< size of a "minimum partition"
// -------------------------------------------------------
// CU data
// -------------------------------------------------------
Bool* m_skipFlag; ///< array of skip flags
Char* m_pePartSize; ///< array of partition sizes
Char* m_pePredMode; ///< array of prediction modes
Bool* m_CUTransquantBypass; ///< array of cu_transquant_bypass flags
Char* m_phQP; ///< array of QP values
UChar* m_puhTrIdx; ///< array of transform indices
UChar* m_puhTransformSkip[3];///< array of transform skipping flags
UChar* m_puhCbf[3]; ///< array of coded block flags (CBF)
TComCUMvField m_acCUMvField[2]; ///< array of motion vectors
TCoeff* m_pcTrCoeffY; ///< transformed coefficient buffer (Y)
TCoeff* m_pcTrCoeffCb; ///< transformed coefficient buffer (Cb)
TCoeff* m_pcTrCoeffCr; ///< transformed coefficient buffer (Cr)
#if ADAPTIVE_QP_SELECTION
Int* m_pcArlCoeffY; ///< ARL coefficient buffer (Y)
Int* m_pcArlCoeffCb; ///< ARL coefficient buffer (Cb)
Int* m_pcArlCoeffCr; ///< ARL coefficient buffer (Cr)
bool m_ArlCoeffIsAliasedAllocation; ///< ARL coefficient buffer is an alias of the global buffer and must not be free()'d
static Int* m_pcGlbArlCoeffY; ///< ARL coefficient buffer (Y)
static Int* m_pcGlbArlCoeffCb; ///< ARL coefficient buffer (Cb)
static Int* m_pcGlbArlCoeffCr; ///< ARL coefficient buffer (Cr)
#endif
Pel* m_pcIPCMSampleY; ///< PCM sample buffer (Y)
Pel* m_pcIPCMSampleCb; ///< PCM sample buffer (Cb)
Pel* m_pcIPCMSampleCr; ///< PCM sample buffer (Cr)
Int* m_piSliceSUMap; ///< pointer of slice ID map
std::vector<NDBFBlockInfo> m_vNDFBlock;
..................
.............
}
上面是HM9.0的class TComDataCU的部分代碼, 那些地方有疑問大家可以都貼出來一起討論啊!