1. 程式人生 > 其它 >HEVC程式碼閱讀- - xPredIntraPlanar函式

HEVC程式碼閱讀- - xPredIntraPlanar函式

技術標籤: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 );
    }
  }
}