HEVC程式碼閱讀- - xPredIntraPlanar函式
阿新 • • 發佈:2021-02-03
技術標籤:HEVC
xPredIntraPlanar函式
實現功能:Planar模式的預測
對應的公式:
Void TComPrediction::xPredIntraPlanar( Int* pSrc, Int srcStride, Pel* rpDst, Int dstStride, UInt width, UInt height ) { assert(width == height); // k,l表示行和列 // bottomLeft和topRight表示左下、右上畫素值 Int k, l, bottomLeft, topRight; Int horPred; Int leftColumn[MAX_CU_SIZE], topRow[MAX_CU_SIZE], bottomRow[MAX_CU_SIZE], rightColumn[MAX_CU_SIZE]; UInt blkSize = width; UInt offset2D = width; // shift1D 表示 width對應的位數 // g_aucConvertToBit(x) 函式:log2(x)-2 UInt shift1D = g_aucConvertToBit[ width ] + 2; UInt shift2D = shift1D + 1; // 獲取左側列和上方行的參考陣列 for(k=0;k<blkSize+1;k++) { // srcStride 表示 pSrc陣列的跨度 topRow[k] = pSrc[k-srcStride]; leftColumn[k] = pSrc[k*srcStride-1]; } // 準備插值所用的中間變數 bottomLeft = leftColumn[blkSize]; topRight = topRow[blkSize]; for (k=0;k<blkSize;k++) { bottomRow[k] = bottomLeft - topRow[k]; rightColumn[k] = topRight - leftColumn[k]; topRow[k] <<= shift1D; leftColumn[k] <<= shift1D; } // 生成預測訊號 for (k=0;k<blkSize;k++) { horPred = leftColumn[k] + offset2D; for (l=0;l<blkSize;l++) { horPred += rightColumn[k]; topRow[l] += bottomRow[l]; // rpDst[k*dstStride+l]表示 (k,l)處的預測訊號值 // 此陣列為一維陣列,使用dstStride(跨度)來間接的表示二維 rpDst[k*dstStride+l] = ( (horPred + topRow[l]) >> shift2D ); } } }