HM之基于哈达玛变换的SATD计算

1. // 函数功能是计算8×8块的哈达玛变换，并计算其SATD值，m2[8][8]存放最终的变换结果  

转载https://blog.csdn.net/king20168/article/details/52759843

[cpp]  view plain  copy

1. Int xCalcHADs8x8_ISlice(Pel *piOrg, Int iStrideOrg)  
2. {  
3. Int k, i, j, jj;  
4. Int diff[64], m1[8][8], m2[8][8], m3[8][8], iSumHad = 0;  
5.   
6. //把8×8块像素存放在一维数组中  
7. for( k = 0; k < 64; k += 8 )  
8. {  
9. diff[k+0] = piOrg[0] ;  
10. diff[k+1] = piOrg[1] ;  
11. diff[k+2] = piOrg[2] ;  
12. diff[k+3] = piOrg[3] ;  
13. diff[k+4] = piOrg[4] ;  
14. diff[k+5] = piOrg[5] ;  
15. diff[k+6] = piOrg[6] ;  
16. diff[k+7] = piOrg[7] ;  
17.   
18. piOrg += iStrideOrg; //指针指向下一行  
19. }  
20.   
21. //horizontal 水平变换，计算V=X×H，  
22. for (j=0; j < 8; j++)  
23. {  
24. jj = j << 3; //一次算8个，乘8换行  
25. m2[j][0] = diff[jj ] + diff[jj+4];  
26. m2[j][1] = diff[jj+1] + diff[jj+5];  
27. m2[j][2] = diff[jj+2] + diff[jj+6];  
28. m2[j][3] = diff[jj+3] + diff[jj+7];  
29. m2[j][4] = diff[jj ] - diff[jj+4];  
30. m2[j][5] = diff[jj+1] - diff[jj+5];  
31. m2[j][6] = diff[jj+2] - diff[jj+6];  
32. m2[j][7] = diff[jj+3] - diff[jj+7];  
33.   
34. m1[j][0] = m2[j][0] + m2[j][2];  
35. m1[j][1] = m2[j][1] + m2[j][3];  
36. m1[j][2] = m2[j][0] - m2[j][2];  
37. m1[j][3] = m2[j][1] - m2[j][3];  
38. m1[j][4] = m2[j][4] + m2[j][6];  
39. m1[j][5] = m2[j][5] + m2[j][7];  
40. m1[j][6] = m2[j][4] - m2[j][6];  
41. m1[j][7] = m2[j][5] - m2[j][7];  
42.   
43. m2[j][0] = m1[j][0] + m1[j][1];  
44. m2[j][1] = m1[j][0] - m1[j][1];  
45. m2[j][2] = m1[j][2] + m1[j][3];  
46. m2[j][3] = m1[j][2] - m1[j][3];  
47. m2[j][4] = m1[j][4] + m1[j][5];  
48. m2[j][5] = m1[j][4] - m1[j][5];  
49. m2[j][6] = m1[j][6] + m1[j][7];  
50. m2[j][7] = m1[j][6] - m1[j][7];  
51. }  
52.   
53. //vertical 垂直变换，计算H×V  
54. for (i=0; i < 8; i++)  
55. {  
56. m3[0][i] = m2[0][i] + m2[4][i];  
57. m3[1][i] = m2[1][i] + m2[5][i];  
58. m3[2][i] = m2[2][i] + m2[6][i];  
59. m3[3][i] = m2[3][i] + m2[7][i];  
60. m3[4][i] = m2[0][i] - m2[4][i];  
61. m3[5][i] = m2[1][i] - m2[5][i];  
62. m3[6][i] = m2[2][i] - m2[6][i];  
63. m3[7][i] = m2[3][i] - m2[7][i];  
64.   
65. m1[0][i] = m3[0][i] + m3[2][i];  
66. m1[1][i] = m3[1][i] + m3[3][i];  
67. m1[2][i] = m3[0][i] - m3[2][i];  
68. m1[3][i] = m3[1][i] - m3[3][i];  
69. m1[4][i] = m3[4][i] + m3[6][i];  
70. m1[5][i] = m3[5][i] + m3[7][i];  
71. m1[6][i] = m3[4][i] - m3[6][i];  
72. m1[7][i] = m3[5][i] - m3[7][i];  
73.   
74. m2[0][i] = m1[0][i] + m1[1][i];  
75. m2[1][i] = m1[0][i] - m1[1][i];  
76. m2[2][i] = m1[2][i] + m1[3][i];  
77. m2[3][i] = m1[2][i] - m1[3][i];  
78. m2[4][i] = m1[4][i] + m1[5][i];  
79. m2[5][i] = m1[4][i] - m1[5][i];  
80. m2[6][i] = m1[6][i] + m1[7][i];  
81. m2[7][i] = m1[6][i] - m1[7][i];  
82. }  
83.   
84. for (i = 0; i < 8; i++)  
85. {  
86. for (j = 0; j < 8; j++)  
87. {  
88. iSumHad += abs(m2[i][j]);  
89. }  
90. }  
91. iSumHad -= abs(m2[0][0]);  
92. iSumHad =(iSumHad+2)>>2; //除以系数4，并向上取整  
93. return(iSumHad);  
94. }