HEVC学习（十三） —— SAO函数解析之一

这篇博客写得不错，觉得对我对大家刚开始学习时会有帮助，于是转载之。

原文地址：http://blog.csdn.net/hevc_cjl/article/details/8288479

关于SAO的原理和流程的解析，已经在我转载的一篇博客HEVC中SAO--自适应样点补偿 详细分析解读有了比较清楚的介绍了，本文就不再重复这个过程，而把主要精力放在具体函数实现的解析上。在我自己的一篇博客HEVC学习（八） —— 以SAO为例浅析跟踪代码方法里其实也作了相关的铺垫了，当中重点放在跟踪代码的方法上，本文在此基础上对重要的函数进行解析，它们的调用位置这里就不提了，有关这部分的内容请参考前述的两篇博客。

本文首先介绍

m_cEncSAO.create( getSourceWidth(), getSourceHeight(), g_uiMaxCUWidth, g_uiMaxCUHeight, _uiMaxCUDepth );

该函数为SAO的相关参数分配内存和初始化。

[cpp]  view plain copy

1. <span style="font-size:12px;">/** create SampleAdaptiveOffset memory. 
2.  * \param  
3.  */  
4. Void TComSampleAdaptiveOffset::create( UInt uiSourceWidth, UInt uiSourceHeight, UInt uiMaxCUWidth, UInt uiMaxCUHeight, UInt uiMaxCUDepth)  
5. {  
6.   m_iPicWidth  = uiSourceWidth;  
7.   m_iPicHeight = uiSourceHeight;  
8.   
9.   m_uiMaxCUWidth  = uiMaxCUWidth;  
10.   m_uiMaxCUHeight = uiMaxCUHeight;  
11.   
12.   m_iNumCuInWidth  = m_iPicWidth / m_uiMaxCUWidth;  
13.   m_iNumCuInWidth += ( m_iPicWidth % m_uiMaxCUWidth ) ? 1 : 0;  
14.   
15.   m_iNumCuInHeight  = m_iPicHeight / m_uiMaxCUHeight;  
16.   m_iNumCuInHeight += ( m_iPicHeight % m_uiMaxCUHeight ) ? 1 : 0;  
17.   //! 以下四句根据CU的实际尺寸计算可划分的最大深度  
18.   Int iMaxSplitLevelHeight = (Int)(logf((Float)m_iNumCuInHeight)/logf(2.0));  
19.   Int iMaxSplitLevelWidth  = (Int)(logf((Float)m_iNumCuInWidth )/logf(2.0));  
20.   
21.   m_uiMaxSplitLevel = (iMaxSplitLevelHeight < iMaxSplitLevelWidth)?(iMaxSplitLevelHeight):(iMaxSplitLevelWidth);  
22.   m_uiMaxSplitLevel = (m_uiMaxSplitLevel< m_uiMaxDepth)?(m_uiMaxSplitLevel):(m_uiMaxDepth);  
23.   /* various structures are overloaded to store per component data. 
24.    * m_iNumTotalParts must allow for sufficient storage in any allocated arrays */  
25.   /* 
26.   const Int TComSampleAdaptiveOffset::m_aiNumCulPartsLevel[5] = 
27.   { 
28.   1,   //level 0 
29.   5,   //level 1 
30.   21,  //level 2 
31.   85,  //level 3 
32.   341, //level 4 
33.   }; 
34.   */  
35.   m_iNumTotalParts  = max(3,m_aiNumCulPartsLevel[m_uiMaxSplitLevel]);  
36.   
37.   UInt uiPixelRangeY = 1 << g_bitDepthY; //!< Y分量的像素范围（256 for 8it-depth）  
38.   UInt uiBoRangeShiftY = g_bitDepthY - SAO_BO_BITS; //!< Band Offset (BO)中每个band的宽度（3 for 8bit-depth，即1<<3 = 8）（Y分量）  
39.   
40.   m_lumaTableBo = new Pel [uiPixelRangeY]; //!< Y分量在BO模式下的索引表，根据像素值可以索引到对应的band序号  
41.   for (Int k2=0; k2<uiPixelRangeY; k2++)  
42.   {  
43.     m_lumaTableBo[k2] = 1 + (k2>>uiBoRangeShiftY); //!< 总共分了32个band(1~32)  
44.   }  
45.   
46.   UInt uiPixelRangeC = 1 << g_bitDepthC; //!< CbCr分量的像素范围（256 for 8it-depth）  
47.   UInt uiBoRangeShiftC = g_bitDepthC - SAO_BO_BITS; //!< Band Offset (BO)中每个band的宽度（3 for 8bit-depth，即1<<3 = 8）（CbCr分量）  
48.   
49.   m_chromaTableBo = new Pel [uiPixelRangeC]; //!< CbCr分量在BO模式下的索引表，根据像素值可以索引到对应的band序号  
50.   for (Int k2=0; k2<uiPixelRangeC; k2++)  
51.   {  
52.     m_chromaTableBo[k2] = 1 + (k2>>uiBoRangeShiftC); //!< 总共分了32个band(1~32)  
53.   }  
54.   
55.   m_iUpBuff1 = new Int[m_iPicWidth+2];  
56.   m_iUpBuff2 = new Int[m_iPicWidth+2];  
57.   m_iUpBufft = new Int[m_iPicWidth+2];  
58.   
59.   m_iUpBuff1++;  
60.   m_iUpBuff2++;  
61.   m_iUpBufft++;  
62.   Pel i;  
63.   
64.   UInt uiMaxY  = (1 << g_bitDepthY) - 1; //!< Y分量像素最大值（255 for 8bit-depth）  
65.   UInt uiMinY  = 0;  
66.   
67.   Int iCRangeExt = uiMaxY>>1; //!< 对范围进行扩展  
68.   
69.   m_pClipTableBase = new Pel[uiMaxY+2*iCRangeExt];  
70.   m_iOffsetBo      = new Int[uiMaxY+2*iCRangeExt];  
71.   
72.   for(i=0;i<(uiMinY+iCRangeExt);i++)  
73.   {  
74.     m_pClipTableBase[i] = uiMinY;  
75.   }  
76.   
77.   for(i=uiMinY+iCRangeExt;i<(uiMaxY+  iCRangeExt);i++)  
78.   {  
79.     m_pClipTableBase[i] = i-iCRangeExt;  
80.   }  
81.   
82.   for(i=uiMaxY+iCRangeExt;i<(uiMaxY+2*iCRangeExt);i++)  
83.   {  
84.     m_pClipTableBase[i] = uiMaxY;  
85.   }  
86.   //! 0 0 0 ... 0 1 2 3 4 ... 255 255 ... 255//!  
87.   m_pClipTable = &(m_pClipTableBase[iCRangeExt]); //!< 查找表  
88.   
89.   UInt uiMaxC  = (1 << g_bitDepthC) - 1; //!< CbCr分量像素最大值（255 for 8bit-depth）  
90.   UInt uiMinC  = 0;  
91.   
92.   Int iCRangeExtC = uiMaxC>>1; //!< 对范围进行扩展  
93.   
94.   m_pChromaClipTableBase = new Pel[uiMaxC+2*iCRangeExtC];  
95.   m_iChromaOffsetBo      = new Int[uiMaxC+2*iCRangeExtC];  
96.   
97.   for(i=0;i<(uiMinC+iCRangeExtC);i++)  
98.   {  
99.     m_pChromaClipTableBase[i] = uiMinC;  
100.   }  
101.   
102.   for(i=uiMinC+iCRangeExtC;i<(uiMaxC+  iCRangeExtC);i++)  
103.   {  
104.     m_pChromaClipTableBase[i] = i-iCRangeExtC;  
105.   }  
106.   
107.   for(i=uiMaxC+iCRangeExtC;i<(uiMaxC+2*iCRangeExtC);i++)  
108.   {  
109.     m_pChromaClipTableBase[i] = uiMaxC;  
110.   }  
111.   //! 0 0 0 ... 0 ! 1 2 3 4 ... 255 255 ... 255 //!  
112.   m_pChromaClipTable = &(m_pChromaClipTableBase[iCRangeExtC]); //!< 查找表，实际上与Y分量是相同的  
113.   
114.   m_iLcuPartIdx = new Int [m_iNumCuInHeight*m_iNumCuInWidth];  
115.   m_pTmpL1 = new Pel [m_uiMaxCUHeight+1];  
116.   m_pTmpL2 = new Pel [m_uiMaxCUHeight+1];  
117.   m_pTmpU1 = new Pel [m_iPicWidth];  
118.   m_pTmpU2 = new Pel [m_iPicWidth];  
119. }</span>  

直观起见，不妨把m_lumaTableBo和m_pClipTable的内容打印出来，

/* m_lumaTableBo */

[cpp]  view plain copy

1. <span style="font-size:12px;">1     1   1   1   1   1   1   1     
2. 2   2   2   2   2   2   2   2     
3. 3   3   3   3   3   3   3   3     
4. 4   4   4   4   4   4   4   4     
5. 5   5   5   5   5   5   5   5     
6. 6   6   6   6   6   6   6   6     
7. 7   7   7   7   7   7   7   7     
8. 8   8   8   8   8   8   8   8     
9. 9   9   9   9   9   9   9   9     
10. 10  10  10  10  10  10  10  10    
11. 11  11  11  11  11  11  11  11    
12. 12  12  12  12  12  12  12  12    
13. 13  13  13  13  13  13  13  13    
14. 14  14  14  14  14  14  14  14    
15. 15  15  15  15  15  15  15  15    
16. 16  16  16  16  16  16  16  16    
17. 17  17  17  17  17  17  17  17    
18. 18  18  18  18  18  18  18  18    
19. 19  19  19  19  19  19  19  19    
20. 20  20  20  20  20  20  20  20    
21. 21  21  21  21  21  21  21  21    
22. 22  22  22  22  22  22  22  22    
23. 23  23  23  23  23  23  23  23    
24. 24  24  24  24  24  24  24  24    
25. 25  25  25  25  25  25  25  25    
26. 26  26  26  26  26  26  26  26    
27. 27  27  27  27  27  27  27  27    
28. 28  28  28  28  28  28  28  28    
29. 29  29  29  29  29  29  29  29    
30. 30  30  30  30  30  30  30  30    
31. 31  31  31  31  31  31  31  31    
32. 32  32  32  32  32  32  32  32    
33. </span>  

/* m_pClipTable */

[cpp]  view plain copy

1. <span style="font-size:12px;">0     1       2       3       4       5       6       7       8       9       10      11      12      13      14      15      16      17      18      19      20      21      22      23      24      25      26      27      28      29      30      31      32      33      34      35      36      37      38      39      40      41      42      43      44      45      46      47      48      49      50      51      52      53      54      55      56      57      58      59      60      61      62      63      64      65      66      67      68      69      70      71      72      73      74      75      76      77      78      79      80      81      82      83      84      85      86      87      88      89      90      91      92      93      94      95      96      97      98      99      100     101     102     103     104     105     106     107     108     109     110     111     112     113     114     115     116     117     118     119     120     121     122     123     124     125     126     127     128     129     130     131     132     133     134     135     136     137     138     139     140     141     142     143     144     145     146     147     148     149     150     151     152     153     154     155     156     157     158     159     160     161     162     163     164     165     166     167     168     169     170     171     172     173     174     175     176     177     178     179     180     181     182     183     184     185     186     187     188     189     190     191     192     193     194     195     196     197     198     199     200     201     202     203     204     205     206     207     208     209     210     211     212     213     214     215     216     217     218     219     220     221     222     223     224     225     226     227     228     229     230     231     232     233     234     235     236     237     238     239     240     241     242     243     244     245     246     247     248     249     250     251     252     253     254     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255     255 </span>   

更直观的做法就是绘制出像素值vs索引值的曲线图来，不过根据这些数据，相信大家很容易就能想象出这两条曲线是何种形状的了，这里就不必要画出图来了。