dct.cpp
上传用户:sun1608
上传日期:2007-02-02
资源大小:6116k
文件大小:8k
源码类别:

流媒体/Mpeg4/MP4

开发平台:

Visual C++

  1. /*************************************************************************
  2. This software module was originally developed by 
  3. Ming-Chieh Lee (mingcl@microsoft.com), Microsoft Corporation
  4. Wei-ge Chen (wchen@microsoft.com), Microsoft Corporation
  5. Bruce Lin (blin@microsoft.com), Microsoft Corporation
  6. Chuang Gu (chuanggu@microsoft.com), Microsoft Corporation
  7. (date: March, 1996)
  8. and edited by
  9. Wei Wu (weiwu@stallion.risc.rockwell.com) Rockwell Science Center
  10. in the course of development of the MPEG-4 Video (ISO/IEC 14496-2). 
  11. This software module is an implementation of a part of one or more MPEG-4 Video tools 
  12. as specified by the MPEG-4 Video. 
  13. ISO/IEC gives users of the MPEG-4 Video free license to this software module or modifications 
  14. thereof for use in hardware or software products claiming conformance to the MPEG-4 Video. 
  15. Those intending to use this software module in hardware or software products are advised that its use may infringe existing patents. 
  16. The original developer of this software module and his/her company, 
  17. the subsequent editors and their companies, 
  18. and ISO/IEC have no liability for use of this software module or modifications thereof in an implementation. 
  19. Copyright is not released for non MPEG-4 Video conforming products. 
  20. Microsoft retains full right to use the code for his/her own purpose, 
  21. assign or donate the code to a third party and to inhibit third parties from using the code for non <MPEG standard> conforming products. 
  22. This copyright notice must be included in all copies or derivative works. 
  23. Copyright (c) 1996, 1997.
  24. Module Name:
  25. vopSes.cpp
  26. Abstract:
  27. Base class for the encoder for one VOP session.
  28. Revision History:
  29. *************************************************************************/
  30. #include <stdio.h>
  31. #include "typeapi.h"
  32. #include "dct.hpp"
  33. #ifdef __MFC_
  34. #ifdef _DEBUG
  35. #undef THIS_FILE
  36. static char BASED_CODE THIS_FILE[] = __FILE__;
  37. #endif
  38. #define new DEBUG_NEW    
  39. #endif // __MFC_
  40. CBlockDCT::~CBlockDCT ()
  41. {
  42. /* NBIT: change
  43. m_rgchClipTbl -= 512;
  44. */
  45. m_rgchClipTbl -= (1<<(m_nBits+1));
  46. delete [] m_rgchClipTbl;
  47. }
  48. /* NBIT
  49. CBlockDCT::CBlockDCT ()
  50. */
  51. CBlockDCT::CBlockDCT (UInt nBits) : m_nBits(nBits)
  52. {
  53. Int ClipTblSize = 1<<(m_nBits+2); // NBIT
  54. Int offset = ClipTblSize/2; // NBIT
  55. Int maxVal = (1<<m_nBits)-1; // NBIT
  56. /* NBIT: change
  57. m_rgchClipTbl = new PixelC [1024];
  58. m_rgchClipTbl += 512;
  59. Int i;
  60. for (i = -512; i < 512; i++) {
  61. if (i < 0)
  62. m_rgchClipTbl [i] = 0;
  63. else if (i >= 0 && i < 256)
  64. m_rgchClipTbl [i] = i;
  65. else 
  66. m_rgchClipTbl [i] = 255;
  67. }
  68. */
  69. m_rgchClipTbl = new PixelC [ClipTblSize];
  70. m_rgchClipTbl += offset;
  71. Int i;
  72. for (i = -offset; i < offset; i++) {
  73. if (i < 0)
  74. m_rgchClipTbl [i] = 0;
  75. else if (i >= 0 && i <= maxVal)
  76. m_rgchClipTbl [i] = i;
  77. else 
  78. m_rgchClipTbl [i] = maxVal;
  79. }
  80. m_c0 = 0.7071068;
  81. m_c1 = 0.4903926;
  82. m_c2 = 0.4619398;
  83. m_c3 = 0.4157348;
  84. m_c4 = 0.3535534;
  85. m_c5 = 0.2777851;
  86. m_c6 = 0.1913417;
  87. m_c7 = 0.0975452;
  88. }
  89. Void CBlockDCT::apply (const PixelC* rgchSrc, Int nColSrc, Int* rgiDst, Int nColDst)
  90. {
  91. // transform here
  92. const PixelC* pchRowSrc = rgchSrc;
  93. for (CoordI iRow = 0; iRow < BLOCK_SIZE; iRow++) {
  94. xformRow (pchRowSrc, iRow);
  95. pchRowSrc += nColSrc;
  96. }
  97. Int* piColDst = rgiDst;
  98. for (CoordI iCol = 0; iCol < BLOCK_SIZE; iCol++) {
  99. xformColumn (piColDst, iCol, nColDst);
  100. piColDst++;
  101. }
  102. }
  103. Void CBlockDCT::apply (const Int* rgiSrc, Int nColSrc, PixelC* rgchDst, Int nColDst)
  104. {
  105. // transform here
  106. const PixelI* piRowSrc = rgiSrc;
  107. for (CoordI iRow = 0; iRow < BLOCK_SIZE; iRow++) {
  108. xformRow (piRowSrc, iRow);
  109. piRowSrc += nColSrc;
  110. }
  111. PixelC* pchColDst = rgchDst;
  112. for (CoordI iCol = 0; iCol < BLOCK_SIZE; iCol++) {
  113. xformColumn (pchColDst, iCol, nColDst);
  114. pchColDst++;
  115. }
  116. }
  117. Void CBlockDCT::apply (const Int* rgiSrc, Int nColSrc, Int* rgiDst, Int nColDst)
  118. {
  119. // transform here
  120. const PixelI* piRowSrc = rgiSrc;
  121. for (CoordI iRow = 0; iRow < BLOCK_SIZE; iRow++) {
  122. xformRow (piRowSrc, iRow);
  123. piRowSrc += nColSrc;
  124. }
  125. Int* piColDst = rgiDst;
  126. for (CoordI iCol = 0; iCol < BLOCK_SIZE; iCol++) {
  127. xformColumn (piColDst, iCol, nColDst);
  128. piColDst++;
  129. }
  130. }
  131. Void CBlockDCT::xformRow (const PixelC* ppxlcRowSrc, CoordI i)
  132. {
  133. UInt j;
  134. for (j = 0; j < BLOCK_SIZE; j++) 
  135. m_rgfltBuf1 [j] = (Float) *ppxlcRowSrc++;
  136. oneDimensionalDCT ();
  137. for (j = 0; j < BLOCK_SIZE; j++) 
  138. m_rgfltAfterRowXform [i] [j] = m_rgfltAfter1dXform [j];
  139. }
  140. Void CBlockDCT::xformRow (const PixelI* ppxlfRowSrc, CoordI i)
  141. {
  142. UInt j;
  143. for (j = 0; j < BLOCK_SIZE; j++) 
  144. m_rgfltBuf1 [j] = (Float) *ppxlfRowSrc++;
  145. oneDimensionalDCT ();
  146. for (j = 0; j < BLOCK_SIZE; j++) 
  147. m_rgfltAfterRowXform [i] [j] = m_rgfltAfter1dXform [j];
  148. }
  149. Void CBlockDCT::xformColumn (PixelC* ppxlcColDst, CoordI i, Int nColDst)
  150. {
  151. UInt j;
  152. for (j = 0; j < BLOCK_SIZE; j++)
  153. m_rgfltBuf1[j] = m_rgfltAfterRowXform[j][i];
  154. oneDimensionalDCT ();
  155. for (j = 0; j < BLOCK_SIZE; j++) { 
  156. Int iValue = 
  157. (m_rgfltAfter1dXform[j] >= 0) ? (PixelI) (m_rgfltAfter1dXform[j] + .5) :
  158. (PixelI) (m_rgfltAfter1dXform[j] - .5);
  159. *ppxlcColDst = m_rgchClipTbl [iValue];
  160. ppxlcColDst += nColDst;
  161. }
  162. }
  163. Void CBlockDCT::xformColumn (PixelI* ppxliColDst, CoordI i, Int nColDst)
  164. {
  165. UInt j;
  166. for (j = 0; j < BLOCK_SIZE; j++)
  167. m_rgfltBuf1[j] = m_rgfltAfterRowXform[j][i];
  168. oneDimensionalDCT ();
  169. for (j = 0; j < BLOCK_SIZE; j++) { 
  170. PixelI vl = 
  171. (m_rgfltAfter1dXform[j] >= 0) ? (PixelI) (m_rgfltAfter1dXform[j] + .5) :
  172. (PixelI) (m_rgfltAfter1dXform[j] - .5);
  173. *ppxliColDst = vl;
  174. ppxliColDst += nColDst;
  175. }
  176. }
  177. /* NBIT: change
  178. CFwdBlockDCT::CFwdBlockDCT () : CBlockDCT ()
  179. */
  180. CFwdBlockDCT::CFwdBlockDCT (UInt nBits) : CBlockDCT (nBits)
  181. {
  182. }
  183. Void CFwdBlockDCT::oneDimensionalDCT ()
  184. {
  185. Int j, j1;
  186. for (j = 0; j < 4; j++) {
  187. j1 = 7 - j;
  188. m_rgfltBuf2[j] = m_rgfltBuf1[j] + m_rgfltBuf1[j1];
  189. m_rgfltBuf2[j1] = m_rgfltBuf1[j] - m_rgfltBuf1[j1];
  190. }
  191. m_rgfltBuf1[0] = m_rgfltBuf2[0] + m_rgfltBuf2[3];
  192. m_rgfltBuf1[1] = m_rgfltBuf2[1] + m_rgfltBuf2[2];
  193. m_rgfltBuf1[2] = m_rgfltBuf2[1] - m_rgfltBuf2[2];
  194. m_rgfltBuf1[3] = m_rgfltBuf2[0] - m_rgfltBuf2[3];
  195. m_rgfltBuf1[4] = m_rgfltBuf2[4];
  196. m_rgfltBuf1[5] = (m_rgfltBuf2[6] - m_rgfltBuf2[5]) * m_c0;
  197. m_rgfltBuf1[6] = (m_rgfltBuf2[6] + m_rgfltBuf2[5]) * m_c0;
  198. m_rgfltBuf1[7] = m_rgfltBuf2[7];
  199. m_rgfltAfter1dXform[0] = (m_rgfltBuf1[0] + m_rgfltBuf1[1]) * m_c4;
  200. m_rgfltAfter1dXform[4] = (m_rgfltBuf1[0] - m_rgfltBuf1[1]) * m_c4;
  201. m_rgfltAfter1dXform[2] = m_rgfltBuf1[2] * m_c6 + m_rgfltBuf1[3] * m_c2;
  202. m_rgfltAfter1dXform[6] = m_rgfltBuf1[3] * m_c6 - m_rgfltBuf1[2] * m_c2;
  203. m_rgfltBuf2[4] = m_rgfltBuf1[4] + m_rgfltBuf1[5];
  204. m_rgfltBuf2[7] = m_rgfltBuf1[7] + m_rgfltBuf1[6];
  205. m_rgfltBuf2[5] = m_rgfltBuf1[4] - m_rgfltBuf1[5];
  206. m_rgfltBuf2[6] = m_rgfltBuf1[7] - m_rgfltBuf1[6];
  207. m_rgfltAfter1dXform[1] = m_rgfltBuf2[4] * m_c7 + m_rgfltBuf2[7] * m_c1;
  208. m_rgfltAfter1dXform[5] = m_rgfltBuf2[5] * m_c3 + m_rgfltBuf2[6] * m_c5;
  209. m_rgfltAfter1dXform[7] = m_rgfltBuf2[7] * m_c7 - m_rgfltBuf2[4] * m_c1;
  210. m_rgfltAfter1dXform[3] = m_rgfltBuf2[6] * m_c3 - m_rgfltBuf2[5] * m_c5;
  211. }
  212. /* NBIT: change
  213. CInvBlockDCT::CInvBlockDCT () : CBlockDCT ()
  214. */
  215. CInvBlockDCT::CInvBlockDCT (UInt nBits) : CBlockDCT (nBits)
  216. {
  217. }
  218. Void CInvBlockDCT::oneDimensionalDCT ()
  219. {
  220. Float flt1 = m_rgfltBuf1[1] * m_c7 - m_rgfltBuf1[7] * m_c1;
  221. Float flt2 = m_rgfltBuf1[7] * m_c7 + m_rgfltBuf1[1] * m_c1;
  222. Float flt3 = m_rgfltBuf1[5] * m_c3 - m_rgfltBuf1[3] * m_c5;
  223. Float flt4 = m_rgfltBuf1[3] * m_c3 + m_rgfltBuf1[5] * m_c5;
  224. m_rgfltBuf2[0] = (m_rgfltBuf1[0] + m_rgfltBuf1[4]) * m_c4;
  225. m_rgfltBuf2[1] = (m_rgfltBuf1[0] - m_rgfltBuf1[4]) * m_c4;
  226. m_rgfltBuf2[2] = m_rgfltBuf1[2] * m_c6 - m_rgfltBuf1[6] * m_c2;
  227. m_rgfltBuf2[3] = m_rgfltBuf1[6] * m_c6 + m_rgfltBuf1[2] * m_c2;
  228. m_rgfltBuf1[4] = flt1 + flt3;
  229. m_rgfltBuf2[5] = flt1 - flt3;
  230. m_rgfltBuf2[6] = flt2 - flt4;
  231. m_rgfltBuf1[7] = flt2 + flt4;
  232.     
  233. m_rgfltBuf1[5] = (m_rgfltBuf2[6] - m_rgfltBuf2[5]) * m_c0;
  234. m_rgfltBuf1[6] = (m_rgfltBuf2[6] + m_rgfltBuf2[5]) * m_c0;
  235. m_rgfltBuf1[0] = m_rgfltBuf2[0] + m_rgfltBuf2[3];
  236. m_rgfltBuf1[1] = m_rgfltBuf2[1] + m_rgfltBuf2[2];
  237. m_rgfltBuf1[2] = m_rgfltBuf2[1] - m_rgfltBuf2[2];
  238. m_rgfltBuf1[3] = m_rgfltBuf2[0] - m_rgfltBuf2[3];
  239. Int j, j1;
  240. for (j = 0; j < 4; j++) {
  241. j1 = 7 - j;
  242. m_rgfltAfter1dXform[j] = m_rgfltBuf1[j] + m_rgfltBuf1[j1];
  243. m_rgfltAfter1dXform[j1] = m_rgfltBuf1[j] - m_rgfltBuf1[j1];
  244. }
  245. }