HEVC学习（十六） —— SAO函数解析之四

这篇博客写得不错，觉得对我对大家刚开始学习时会有帮助，于是转载之。原文地址：http://blog.csdn.net/hevc_cjl/article/details/8288601

 

 

1. Void TEncSampleAdaptiveOffset::saoComponentParamDist(Int allowMergeLeft, Int allowMergeUp, SAOParam *saoParam, Int addr, Int addrUp, Int addrLeft, Int yCbCr, Double lambda, SaoLcuParam *compSaoParam, Double *compDistortion)
2. {
3. Int typeIdx;
5. Int64 estDist;
6. Int classIdx;
7. Int shift = 2 * DISTORTION_PRECISION_ADJUSTMENT(((yCbCr==0)?g_bitDepthY:g_bitDepthC)-8); //!< 0 for 8bit-depth
8. Int64 bestDist;
10. SaoLcuParam* saoLcuParam = &(saoParam->saoLcuParam[yCbCr][addr]);
11. SaoLcuParam* saoLcuParamNeighbor = NULL;
13. resetSaoUnit(saoLcuParam);
14. resetSaoUnit(&compSaoParam[0]); //!< 左邻块的SAO参数
15. resetSaoUnit(&compSaoParam[1]); //!< 上邻块的SAO参数
18. Double dCostPartBest = MAX_DOUBLE;
20. Double bestRDCostTableBo = MAX_DOUBLE;
21. Int bestClassTableBo = 0;
22. Int currentDistortionTableBo[MAX_NUM_SAO_CLASS];
23. Double currentRdCostTableBo[MAX_NUM_SAO_CLASS];
26. SaoLcuParam saoLcuParamRdo;
27. Double estRate = 0;
29. resetSaoUnit(&saoLcuParamRdo);
31. m_pcRDGoOnSbacCoder->load(m_pppcRDSbacCoder[0][CI_TEMP_BEST]);
32. m_pcRDGoOnSbacCoder->resetBits();
33. m_pcEntropyCoder->encodeSaoOffset(&saoLcuParamRdo, yCbCr);
35. dCostPartBest = m_pcEntropyCoder->getNumberOfWrittenBits()*lambda ;
36. copySaoUnit(saoLcuParam, &saoLcuParamRdo );
37. bestDist = 0;
41. for (typeIdx=0; typeIdx<MAX_NUM_SAO_TYPE; typeIdx++) //!< 遍历所有的滤波类型
42. {
43. estDist = estSaoTypeDist(yCbCr, typeIdx, shift, lambda, currentDistortionTableBo, currentRdCostTableBo);//!< 得到当前滤波类型的失真值
45. if( typeIdx == SAO_BO )
46. {
47. // Estimate Best Position
48. Double currentRDCost = 0.0;
50. for(Int i=0; i< SAO_MAX_BO_CLASSES -SAO_BO_LEN +1; i++)
51. {
52. currentRDCost = 0.0;
53. for(UInt uj = i; uj < i+SAO_BO_LEN; uj++) //!< 依次以4个band为单位进行RDcost计算
54. {
55. currentRDCost += currentRdCostTableBo[uj];
56. }
58. if( currentRDCost < bestRDCostTableBo) //!< 更新最佳值
59. {
60. bestRDCostTableBo = currentRDCost;
61. bestClassTableBo = i;
62. }
63. }
65. // Re code all Offsets
66. // Code Center
67. estDist = 0;
68. for(classIdx = bestClassTableBo; classIdx < bestClassTableBo+SAO_BO_LEN; classIdx++)
69. {
70. estDist += currentDistortionTableBo[classIdx];
71. }
72. } //!< if( typeIdx == SAO_BO )
73. resetSaoUnit(&saoLcuParamRdo);
74. saoLcuParamRdo.length = m_iNumClass[typeIdx];
75. saoLcuParamRdo.typeIdx = typeIdx;
76. saoLcuParamRdo.mergeLeftFlag = 0;
77. saoLcuParamRdo.mergeUpFlag = 0;
78. saoLcuParamRdo.subTypeIdx = (typeIdx == SAO_BO) ? bestClassTableBo : 0;
79. for (classIdx = 0; classIdx < saoLcuParamRdo.length; classIdx++)
80. {
81. saoLcuParamRdo.offset[classIdx] = (Int)m_iOffset[yCbCr][typeIdx][classIdx+saoLcuParamRdo.subTypeIdx+1];
82. }
83. m_pcRDGoOnSbacCoder->load(m_pppcRDSbacCoder[0][CI_TEMP_BEST]);
84. m_pcRDGoOnSbacCoder->resetBits();
85. m_pcEntropyCoder->encodeSaoOffset(&saoLcuParamRdo, yCbCr);
87. estRate = m_pcEntropyCoder->getNumberOfWrittenBits();
88. m_dCost[yCbCr][typeIdx] = (Double)((Double)estDist + lambda * (Double) estRate);
90. if(m_dCost[yCbCr][typeIdx] < dCostPartBest) //!< 更新最佳值
91. {
92. dCostPartBest = m_dCost[yCbCr][typeIdx];
93. copySaoUnit(saoLcuParam, &saoLcuParamRdo );
94. bestDist = estDist;
95. }
96. } //!< for (typeIdx=0; typeIdx<MAX_NUM_SAO_TYPE; typeIdx++)
97. compDistortion[0] += ((Double)bestDist/lambda);
98. m_pcRDGoOnSbacCoder->load(m_pppcRDSbacCoder[0][CI_TEMP_BEST]);
99. m_pcEntropyCoder->encodeSaoOffset(saoLcuParam, yCbCr);
100. m_pcRDGoOnSbacCoder->store( m_pppcRDSbacCoder[0][CI_TEMP_BEST] );
103. // merge left or merge up
105. for (Int idxNeighbor=0;idxNeighbor<2;idxNeighbor++)
106. {
107. saoLcuParamNeighbor = NULL;
108. if (allowMergeLeft && addrLeft>=0 && idxNeighbor ==0) //!< 左邻块可用
109. {
110. saoLcuParamNeighbor = &(saoParam->saoLcuParam[yCbCr][addrLeft]); //!< 取左邻块的SAO参数
111. }
112. else if (allowMergeUp && addrUp>=0 && idxNeighbor ==1) //!< 上邻块可用
113. {
114. saoLcuParamNeighbor = &(saoParam->saoLcuParam[yCbCr][addrUp]); //!< 取上邻块的SAO参数
115. }
116. if (saoLcuParamNeighbor!=NULL)
117. {
118. estDist = 0;
119. typeIdx = saoLcuParamNeighbor->typeIdx;
120. if (typeIdx>=0) //!< typeIdx为有效的
121. {
122. Int mergeBandPosition = (typeIdx == SAO_BO)?saoLcuParamNeighbor->subTypeIdx:0;
123. Int merge_iOffset;
124. for(classIdx = 0; classIdx < m_iNumClass[typeIdx]; classIdx++)
125. {
126. merge_iOffset = saoLcuParamNeighbor->offset[classIdx];
127. estDist += estSaoDist(m_iCount [yCbCr][typeIdx][classIdx+mergeBandPosition+1], merge_iOffset, m_iOffsetOrg[yCbCr][typeIdx][classIdx+mergeBandPosition+1], shift);
128. }
129. }
130. else
131. {
132. estDist = 0;
133. }
135. copySaoUnit(&compSaoParam[idxNeighbor], saoLcuParamNeighbor );
136. compSaoParam[idxNeighbor].mergeUpFlag = idxNeighbor;
137. compSaoParam[idxNeighbor].mergeLeftFlag = !idxNeighbor;
139. compDistortion[idxNeighbor+1] += ((Double)estDist/lambda);
140. }
141. }
142. }

Void TEncSampleAdaptiveOffset::saoComponentParamDist(Int allowMergeLeft, Int allowMergeUp, SAOParam *saoParam, Int addr, Int addrUp, Int addrLeft, Int yCbCr, Double lambda, SaoLcuParam *compSaoParam, Double *compDistortion)
{
  Int typeIdx;

  Int64 estDist;
  Int classIdx;
  Int shift = 2 * DISTORTION_PRECISION_ADJUSTMENT(((yCbCr==0)?g_bitDepthY:g_bitDepthC)-8); //!< 0 for 8bit-depth
  Int64 bestDist;

  SaoLcuParam*  saoLcuParam = &(saoParam->saoLcuParam[yCbCr][addr]);
  SaoLcuParam*  saoLcuParamNeighbor = NULL; 

  resetSaoUnit(saoLcuParam);
  resetSaoUnit(&compSaoParam[0]); //!< 左邻块的SAO参数
  resetSaoUnit(&compSaoParam[1]); //!< 上邻块的SAO参数


  Double dCostPartBest = MAX_DOUBLE;

  Double  bestRDCostTableBo = MAX_DOUBLE;
  Int     bestClassTableBo    = 0;
  Int     currentDistortionTableBo[MAX_NUM_SAO_CLASS];
  Double  currentRdCostTableBo[MAX_NUM_SAO_CLASS];


  SaoLcuParam   saoLcuParamRdo;   
  Double   estRate = 0;

  resetSaoUnit(&saoLcuParamRdo);

  m_pcRDGoOnSbacCoder->load(m_pppcRDSbacCoder[0][CI_TEMP_BEST]);
  m_pcRDGoOnSbacCoder->resetBits();
 m_pcEntropyCoder->encodeSaoOffset(&saoLcuParamRdo, yCbCr);
  
  dCostPartBest = m_pcEntropyCoder->getNumberOfWrittenBits()*lambda ; 
  copySaoUnit(saoLcuParam, &saoLcuParamRdo );
  bestDist = 0;
  


  for (typeIdx=0; typeIdx<MAX_NUM_SAO_TYPE; typeIdx++) //!< 遍历所有的滤波类型
  {
    estDist = estSaoTypeDist(yCbCr, typeIdx, shift, lambda, currentDistortionTableBo, currentRdCostTableBo);//!< 得到当前滤波类型的失真值

    if( typeIdx == SAO_BO )
    {
      // Estimate Best Position
      Double currentRDCost = 0.0;

      for(Int i=0; i< SAO_MAX_BO_CLASSES -SAO_BO_LEN +1; i++)
      {
        currentRDCost = 0.0;
        for(UInt uj = i; uj < i+SAO_BO_LEN; uj++) //!< 依次以4个band为单位进行RDcost计算
        {
          currentRDCost += currentRdCostTableBo[uj];
        }

        if( currentRDCost < bestRDCostTableBo) //!< 更新最佳值
        {
          bestRDCostTableBo = currentRDCost;
          bestClassTableBo  = i;
        }
      }

      // Re code all Offsets
      // Code Center
      estDist = 0;
      for(classIdx = bestClassTableBo; classIdx < bestClassTableBo+SAO_BO_LEN; classIdx++) 
      {
        estDist += currentDistortionTableBo[classIdx];
      }
    } //!< if( typeIdx == SAO_BO )
    resetSaoUnit(&saoLcuParamRdo);
    saoLcuParamRdo.length = m_iNumClass[typeIdx];
    saoLcuParamRdo.typeIdx = typeIdx;
    saoLcuParamRdo.mergeLeftFlag = 0;
    saoLcuParamRdo.mergeUpFlag   = 0;
    saoLcuParamRdo.subTypeIdx = (typeIdx == SAO_BO) ? bestClassTableBo : 0;
    for (classIdx = 0; classIdx < saoLcuParamRdo.length; classIdx++)
    {
      saoLcuParamRdo.offset[classIdx] = (Int)m_iOffset[yCbCr][typeIdx][classIdx+saoLcuParamRdo.subTypeIdx+1];
    }
    m_pcRDGoOnSbacCoder->load(m_pppcRDSbacCoder[0][CI_TEMP_BEST]);
    m_pcRDGoOnSbacCoder->resetBits();
    m_pcEntropyCoder->encodeSaoOffset(&saoLcuParamRdo, yCbCr);

    estRate = m_pcEntropyCoder->getNumberOfWrittenBits();
    m_dCost[yCbCr][typeIdx] = (Double)((Double)estDist + lambda * (Double) estRate);

    if(m_dCost[yCbCr][typeIdx] < dCostPartBest) //!< 更新最佳值
    {
      dCostPartBest = m_dCost[yCbCr][typeIdx];
      copySaoUnit(saoLcuParam, &saoLcuParamRdo );
      bestDist = estDist;       
    }
  } //!< for (typeIdx=0; typeIdx<MAX_NUM_SAO_TYPE; typeIdx++)
  compDistortion[0] += ((Double)bestDist/lambda);
  m_pcRDGoOnSbacCoder->load(m_pppcRDSbacCoder[0][CI_TEMP_BEST]);
 m_pcEntropyCoder->encodeSaoOffset(saoLcuParam, yCbCr);
  m_pcRDGoOnSbacCoder->store( m_pppcRDSbacCoder[0][CI_TEMP_BEST] );


  // merge left or merge up

  for (Int idxNeighbor=0;idxNeighbor<2;idxNeighbor++) 
  {
    saoLcuParamNeighbor = NULL;
    if (allowMergeLeft && addrLeft>=0 && idxNeighbor ==0) //!< 左邻块可用
    {
      saoLcuParamNeighbor = &(saoParam->saoLcuParam[yCbCr][addrLeft]); //!< 取左邻块的SAO参数
    }
    else if (allowMergeUp && addrUp>=0 && idxNeighbor ==1) //!< 上邻块可用
    {
      saoLcuParamNeighbor = &(saoParam->saoLcuParam[yCbCr][addrUp]); //!< 取上邻块的SAO参数
    }
    if (saoLcuParamNeighbor!=NULL)
    {
      estDist = 0;
      typeIdx = saoLcuParamNeighbor->typeIdx;
      if (typeIdx>=0) //!< typeIdx为有效的 
      {
        Int mergeBandPosition = (typeIdx == SAO_BO)?saoLcuParamNeighbor->subTypeIdx:0;
        Int   merge_iOffset;
        for(classIdx = 0; classIdx < m_iNumClass[typeIdx]; classIdx++)
        {
          merge_iOffset = saoLcuParamNeighbor->offset[classIdx];
          estDist   += estSaoDist(m_iCount [yCbCr][typeIdx][classIdx+mergeBandPosition+1], merge_iOffset, m_iOffsetOrg[yCbCr][typeIdx][classIdx+mergeBandPosition+1],  shift);
        }
      }
      else
      {
        estDist = 0;
      }

      copySaoUnit(&compSaoParam[idxNeighbor], saoLcuParamNeighbor );
      compSaoParam[idxNeighbor].mergeUpFlag   = idxNeighbor;
      compSaoParam[idxNeighbor].mergeLeftFlag = !idxNeighbor;

      compDistortion[idxNeighbor+1] += ((Double)estDist/lambda);
    } 
  } 
}