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

流媒体/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. Simon Winder (swinder@microsoft.com), Microsoft Corporation
  8. (date: March, 1996)
  9. and edited by
  10. Wei Wu (weiwu@stallion.risc.rockwell.com), Rockwell Science Center
  11. in the course of development of the MPEG-4 Video (ISO/IEC 14496-2). 
  12. This software module is an implementation of a part of one or more MPEG-4 Video tools 
  13. as specified by the MPEG-4 Video. 
  14. ISO/IEC gives users of the MPEG-4 Video free license to this software module or modifications 
  15. thereof for use in hardware or software products claiming conformance to the MPEG-4 Video. 
  16. Those intending to use this software module in hardware or software products are advised that its use may infringe existing patents. 
  17. The original developer of this software module and his/her company, 
  18. the subsequent editors and their companies, 
  19. and ISO/IEC have no liability for use of this software module or modifications thereof in an implementation. 
  20. Copyright is not released for non MPEG-4 Video conforming products. 
  21. Microsoft retains full right to use the code for his/her own purpose, 
  22. assign or donate the code to a third party and to inhibit third parties from using the code for non <MPEG standard> conforming products. 
  23. This copyright notice must be included in all copies or derivative works. 
  24. Copyright (c) 1996, 1997.
  25. Module Name:
  26. block.cpp
  27. Abstract:
  28. Block base class
  29. Revision History:
  30.         May. 9   1998:  add boundary by Hyundai Electronics 
  31.                                   Cheol-Soo Park (cspark@super5.hyundai.co.kr) 
  32. *************************************************************************/
  33. #include <stdlib.h>
  34. #include <math.h>
  35. #include "typeapi.h"
  36. #include "codehead.h"
  37. #include "mode.hpp"
  38. #include "global.hpp"
  39. #include "entropy/bitstrm.hpp"
  40. #include "entropy/entropy.hpp"
  41. #include "entropy/huffman.hpp"
  42. #include "dct.hpp"
  43. #include "vopses.hpp"
  44. #ifdef __MFC_
  45. #ifdef _DEBUG
  46. #undef THIS_FILE
  47. static char BASED_CODE THIS_FILE[] = __FILE__;
  48. #endif
  49. #define new DEBUG_NEW    
  50. #endif // __MFC_
  51. Int divroundnearest(Int i, Int iDenom)
  52. {
  53. assert(iDenom>0);
  54. if(i>=0)
  55. return (i+(iDenom>>1))/iDenom;
  56. else
  57. return (i-(iDenom>>1))/iDenom;
  58. }
  59. Void CVideoObject::inverseQuantizeIntraDc (Int* rgiCoefQ, Int iDcScaler)
  60. {
  61. #ifdef NO_APPENDIXF
  62. m_rgiDCTcoef [0] = (Float) rgiCoefQ [0] * 8;
  63. #else
  64. m_rgiDCTcoef [0] = rgiCoefQ [0] * iDcScaler;
  65. /*
  66. if (iQP <= 4)
  67. m_rgiDCTcoef [0] = rgiCoefQ [0] * 8;
  68. else if (iQP >= 5 && iQP <= 8)
  69. m_rgiDCTcoef [0] = rgiCoefQ [0] * 2 * iQP;
  70. else if (iQP >= 9 && iQP <= 24)
  71. m_rgiDCTcoef [0] = rgiCoefQ [0] * (iQP + 8);
  72. else
  73. m_rgiDCTcoef [0] = rgiCoefQ [0] * (iQP * 2 - 16);
  74. */
  75. #endif
  76. }
  77. Void CVideoObject::inverseQuantizeDCTcoefH263 (Int* rgiCoefQ, Int iStart, Int iQP)
  78. {
  79. Int i;
  80. for (i = iStart; i < BLOCK_SQUARE_SIZE; i++) {
  81. if (rgiCoefQ[i]) {
  82. if (iQP % 2 == 1) //?
  83. m_rgiDCTcoef [i] = iQP * (2 * abs (rgiCoefQ[i]) + 1);
  84. else
  85. m_rgiDCTcoef [i] = iQP * (2 * abs (rgiCoefQ[i]) + 1) - 1;
  86. m_rgiDCTcoef [i] = sign (rgiCoefQ[i]) * m_rgiDCTcoef [i] ;
  87. }
  88. else
  89. m_rgiDCTcoef [i] = 0;
  90. }
  91. }
  92. Void CVideoObject::inverseQuantizeIntraDCTcoefMPEG (Int* rgiCoefQ, Int iStart, Int iQP,
  93. Bool bUseAlphaMatrix)
  94. {
  95. assert (iQP != 0);
  96. Int iSum = m_rgiDCTcoef [0];
  97. Bool bCoefQAllZero = (m_rgiDCTcoef [0] == 0)? TRUE : FALSE;
  98. Int *piQuantizerMatrix;
  99. if(bUseAlphaMatrix)
  100. piQuantizerMatrix = m_volmd.rgiIntraQuantizerMatrixAlpha;
  101. else
  102. piQuantizerMatrix = m_volmd.rgiIntraQuantizerMatrix;
  103. Int iMaxVal = 1<<(m_volmd.nBits+3); // NBIT
  104. UInt i;
  105. for (i = iStart; i < BLOCK_SQUARE_SIZE; i++) {
  106. if (rgiCoefQ [i] == 0)
  107. m_rgiDCTcoef [i] = 0;
  108. else {
  109. m_rgiDCTcoef [i] = iQP * rgiCoefQ [i] * piQuantizerMatrix [i] / 8;
  110. /* NBIT: change
  111. m_rgiDCTcoef [i] = checkrange (m_rgiDCTcoef [i], -2048, 2047);
  112. */
  113. m_rgiDCTcoef [i] = checkrange (m_rgiDCTcoef [i], -iMaxVal, iMaxVal-1);
  114. bCoefQAllZero = FALSE;
  115. }
  116. iSum ^= m_rgiDCTcoef [i];
  117. }
  118. if (!bCoefQAllZero) {
  119. if ((iSum & 0x00000001) == 0)
  120. m_rgiDCTcoef [i - 1] ^= 0x00000001;
  121. }
  122. }
  123. Void CVideoObject::inverseQuantizeInterDCTcoefMPEG (Int* rgiCoefQ, Int iStart, Int iQP,
  124. Bool bUseAlphaMatrix)
  125. {
  126. assert (iQP != 0);
  127. Int iSum = 0;
  128. Bool bCoefQAllZero = TRUE;
  129. Int *piQuantizerMatrix;
  130. if(bUseAlphaMatrix)
  131. piQuantizerMatrix = m_volmd.rgiInterQuantizerMatrixAlpha;
  132. else
  133. piQuantizerMatrix = m_volmd.rgiInterQuantizerMatrix;
  134. Int iMaxVal = 1<<(m_volmd.nBits+3); // NBIT
  135. UInt i;
  136. for (i = iStart; i < BLOCK_SQUARE_SIZE; i++) {
  137. if (rgiCoefQ [i] == 0)
  138. m_rgiDCTcoef [i] = 0;
  139. else {
  140. m_rgiDCTcoef [i] = (iQP * (rgiCoefQ [i] * 2 + sign(rgiCoefQ [i])) * piQuantizerMatrix [i]) / 16;
  141. /* NBIT: change
  142. m_rgiDCTcoef [i] = checkrange (m_rgiDCTcoef [i], -2048, 2047);
  143. */
  144. m_rgiDCTcoef [i] = checkrange (m_rgiDCTcoef [i], -iMaxVal, iMaxVal-1);
  145. bCoefQAllZero = FALSE;
  146. }
  147. iSum ^= m_rgiDCTcoef [i];
  148. }
  149. if (!bCoefQAllZero) {
  150. if ((iSum & 0x00000001) == 0)
  151. m_rgiDCTcoef [i - 1] ^= 0x00000001;
  152. }
  153. }
  154. const BlockMemory CVideoObject::findPredictorBlock (
  155. BlockNum iBlk, 
  156. IntraPredDirection predDir,
  157. const MacroBlockMemory* pmbmLeft, 
  158. const MacroBlockMemory* pmbmTop,
  159. const MacroBlockMemory* pmbmLeftTop,
  160. const MacroBlockMemory* pmbmCurr,
  161. const CMBMode* pmbmdLeft, 
  162. const CMBMode* pmbmdTop,
  163. const CMBMode* pmbmdLeftTop,
  164. const CMBMode* pmbmdCurr,
  165. Int& iQPpred
  166. )
  167. {
  168. const BlockMemory blkmRet = NULL;
  169. /*BBM// Added for Boundary by Hyundai(1998-5-9)
  170. if (m_vopmd.bInterlace && pmbmdCurr->m_bMerged [0])
  171.                 swapTransparentModes ((CMBMode*)pmbmdCurr, BBM);
  172. // End of Hyundai(1998-5-9)*/
  173. if (predDir == HORIZONTAL) {
  174. switch (iBlk) {
  175. case Y_BLOCK1:
  176. if (pmbmLeft != NULL && 
  177. (pmbmdLeft->m_dctMd == INTRA || pmbmdLeft->m_dctMd == INTRAQ) &&
  178. pmbmdLeft->m_rgTranspStatus [Y_BLOCK2] != ALL) {
  179. blkmRet = pmbmLeft->rgblkm [Y_BLOCK2 - 1];
  180. iQPpred = pmbmdLeft->m_stepSize;
  181. }
  182. break;
  183. case Y_BLOCK2:
  184. if (pmbmdCurr->m_rgTranspStatus [Y_BLOCK1] != ALL) {
  185. blkmRet = pmbmCurr->rgblkm [Y_BLOCK1 - 1];
  186. iQPpred = pmbmdCurr->m_stepSize;
  187. }
  188. break;
  189. case Y_BLOCK3:
  190. if (pmbmLeft != NULL && 
  191. (pmbmdLeft->m_dctMd == INTRA || pmbmdLeft->m_dctMd == INTRAQ) &&
  192. pmbmdLeft->m_rgTranspStatus [Y_BLOCK4] != ALL) {
  193. blkmRet = pmbmLeft->rgblkm [Y_BLOCK4 - 1];
  194. iQPpred = pmbmdLeft->m_stepSize;
  195. }
  196. break;
  197. case Y_BLOCK4:
  198. if (pmbmdCurr->m_rgTranspStatus [Y_BLOCK3] != ALL) {
  199. blkmRet = pmbmCurr->rgblkm [Y_BLOCK3 - 1];
  200. iQPpred = pmbmdCurr->m_stepSize;
  201. }
  202. break;
  203. case A_BLOCK1:
  204. if (pmbmLeft != NULL && 
  205. (pmbmdLeft->m_dctMd == INTRA || pmbmdLeft->m_dctMd == INTRAQ) &&
  206. pmbmdLeft->m_rgTranspStatus [Y_BLOCK2] != ALL) {
  207. blkmRet = pmbmLeft->rgblkm [A_BLOCK2 - 1];
  208. iQPpred = pmbmdLeft->m_stepSizeAlpha;
  209. }
  210. break;
  211. case A_BLOCK2:
  212. if (pmbmdCurr->m_rgTranspStatus [Y_BLOCK1] != ALL) {
  213. blkmRet = pmbmCurr->rgblkm [A_BLOCK1 - 1];
  214. iQPpred = pmbmdCurr->m_stepSizeAlpha;
  215. }
  216. break;
  217. case A_BLOCK3:
  218. if (pmbmLeft != NULL && 
  219. (pmbmdLeft->m_dctMd == INTRA || pmbmdLeft->m_dctMd == INTRAQ) &&
  220. pmbmdLeft->m_rgTranspStatus [Y_BLOCK4] != ALL) {
  221. blkmRet = pmbmLeft->rgblkm [A_BLOCK4 - 1];
  222. iQPpred = pmbmdLeft->m_stepSizeAlpha;
  223. }
  224. break;
  225. case A_BLOCK4:
  226. if (pmbmdCurr->m_rgTranspStatus [Y_BLOCK3] != ALL) {
  227. blkmRet = pmbmCurr->rgblkm [A_BLOCK3 - 1];
  228. iQPpred = pmbmdCurr->m_stepSizeAlpha;
  229. }
  230. break;
  231. default: //U, V block
  232. if (pmbmLeft != NULL && 
  233. (pmbmdLeft->m_dctMd == INTRA || pmbmdLeft->m_dctMd == INTRAQ) &&
  234. pmbmdLeft->m_rgTranspStatus [ALL_Y_BLOCKS] != ALL) {
  235. blkmRet = pmbmLeft->rgblkm [iBlk - 1];
  236. iQPpred = pmbmdLeft->m_stepSize;
  237. }
  238. }
  239. }
  240. else if  (predDir == VERTICAL) {
  241. switch (iBlk) {
  242. case Y_BLOCK1:
  243. if (pmbmTop != NULL && 
  244. (pmbmdTop->m_dctMd == INTRA || pmbmdTop->m_dctMd == INTRAQ) &&
  245. pmbmdTop->m_rgTranspStatus [Y_BLOCK3] != ALL) {
  246. blkmRet = pmbmTop->rgblkm [Y_BLOCK3 - 1];
  247. iQPpred = pmbmdTop->m_stepSize;
  248. }
  249. break;
  250. case Y_BLOCK2:
  251. if (pmbmTop != NULL && 
  252. (pmbmdTop->m_dctMd == INTRA || pmbmdTop->m_dctMd == INTRAQ) &&
  253. pmbmdTop->m_rgTranspStatus [Y_BLOCK4] != ALL) {
  254. blkmRet = pmbmTop->rgblkm [Y_BLOCK4 - 1];
  255. iQPpred = pmbmdTop->m_stepSize;
  256. }
  257. break;
  258. case Y_BLOCK3:
  259. if (pmbmdCurr->m_rgTranspStatus [Y_BLOCK1] != ALL) {
  260. blkmRet = pmbmCurr->rgblkm [Y_BLOCK1 - 1];
  261. iQPpred = pmbmdCurr->m_stepSize;
  262. }
  263. break;
  264. case Y_BLOCK4:
  265. if (pmbmdCurr->m_rgTranspStatus [Y_BLOCK2] != ALL) {
  266. blkmRet = pmbmCurr->rgblkm [Y_BLOCK2 - 1];
  267. iQPpred = pmbmdCurr->m_stepSize;
  268. }
  269. break;
  270. case A_BLOCK1:
  271. if (pmbmTop != NULL && 
  272. (pmbmdTop->m_dctMd == INTRA || pmbmdTop->m_dctMd == INTRAQ) &&
  273. pmbmdTop->m_rgTranspStatus [Y_BLOCK3] != ALL) {
  274. blkmRet = pmbmTop->rgblkm [A_BLOCK3 - 1];
  275. iQPpred = pmbmdTop->m_stepSizeAlpha;
  276. }
  277. break;
  278. case A_BLOCK2:
  279. if (pmbmTop != NULL && 
  280. (pmbmdTop->m_dctMd == INTRA || pmbmdTop->m_dctMd == INTRAQ) &&
  281. pmbmdTop->m_rgTranspStatus [Y_BLOCK4] != ALL) {
  282. blkmRet = pmbmTop->rgblkm [A_BLOCK4 - 1];
  283. iQPpred = pmbmdTop->m_stepSizeAlpha;
  284. }
  285. break;
  286. case A_BLOCK3:
  287. if (pmbmdCurr->m_rgTranspStatus [Y_BLOCK1] != ALL) {
  288. blkmRet = pmbmCurr->rgblkm [A_BLOCK1 - 1];
  289. iQPpred = pmbmdCurr->m_stepSizeAlpha;
  290. }
  291. break;
  292. case A_BLOCK4:
  293. if (pmbmdCurr->m_rgTranspStatus [Y_BLOCK2] != ALL) {
  294. blkmRet = pmbmCurr->rgblkm [A_BLOCK2 - 1];
  295. iQPpred = pmbmdCurr->m_stepSizeAlpha;
  296. }
  297. break;
  298. default: //U, V block
  299. if (pmbmTop != NULL && 
  300. (pmbmdTop->m_dctMd == INTRA || pmbmdTop->m_dctMd == INTRAQ) &&
  301. pmbmdTop->m_rgTranspStatus [ALL_Y_BLOCKS] != ALL) {
  302. blkmRet = pmbmTop->rgblkm [iBlk - 1];
  303. iQPpred = pmbmdTop->m_stepSize;
  304. }
  305. }
  306. }
  307. else if (predDir == DIAGONAL) {
  308. switch (iBlk) {
  309. case Y_BLOCK1:
  310. if (pmbmLeftTop != NULL && 
  311. (pmbmdLeftTop->m_dctMd == INTRA || pmbmdLeftTop->m_dctMd == INTRAQ) &&
  312. pmbmdLeftTop->m_rgTranspStatus [Y_BLOCK4] != ALL) {
  313. blkmRet = pmbmLeftTop->rgblkm [Y_BLOCK4 - 1];
  314. iQPpred = pmbmdLeftTop->m_stepSize;
  315. }
  316. break;
  317. case Y_BLOCK2:
  318. if (pmbmTop != NULL && 
  319. (pmbmdTop->m_dctMd == INTRA || pmbmdTop->m_dctMd == INTRAQ) &&
  320. pmbmdTop->m_rgTranspStatus [Y_BLOCK3] != ALL) {
  321. blkmRet = pmbmTop->rgblkm [Y_BLOCK3 - 1];
  322. iQPpred = pmbmdTop->m_stepSize;
  323. }
  324. break;
  325. case Y_BLOCK3:
  326. if (pmbmLeft != NULL && 
  327. (pmbmdLeft->m_dctMd == INTRA || pmbmdLeft->m_dctMd == INTRAQ) &&
  328. pmbmdLeft->m_rgTranspStatus [Y_BLOCK2] != ALL) {
  329. blkmRet = pmbmLeft->rgblkm [Y_BLOCK2 - 1];
  330. iQPpred = pmbmdLeft->m_stepSize;
  331. }
  332. break;
  333. case Y_BLOCK4:
  334. if (pmbmdCurr->m_rgTranspStatus [Y_BLOCK1] != ALL) {
  335. blkmRet = pmbmCurr->rgblkm [Y_BLOCK1 - 1];
  336. iQPpred = pmbmdCurr->m_stepSize;
  337. }
  338. break;
  339. case A_BLOCK1:
  340. if (pmbmLeftTop != NULL && 
  341. (pmbmdLeftTop->m_dctMd == INTRA || pmbmdLeftTop->m_dctMd == INTRAQ) &&
  342. pmbmdLeftTop->m_rgTranspStatus [Y_BLOCK4] != ALL) {
  343. blkmRet = pmbmLeftTop->rgblkm [A_BLOCK4 - 1];
  344. iQPpred = pmbmdLeftTop->m_stepSizeAlpha;
  345. }
  346. break;
  347. case A_BLOCK2:
  348. if (pmbmTop != NULL && 
  349. (pmbmdTop->m_dctMd == INTRA || pmbmdTop->m_dctMd == INTRAQ) &&
  350. pmbmdTop->m_rgTranspStatus [Y_BLOCK3] != ALL) {
  351. blkmRet = pmbmTop->rgblkm [A_BLOCK3 - 1];
  352. iQPpred = pmbmdTop->m_stepSizeAlpha;
  353. }
  354. break;
  355. case A_BLOCK3:
  356. if (pmbmLeft != NULL && 
  357. (pmbmdLeft->m_dctMd == INTRA || pmbmdLeft->m_dctMd == INTRAQ) &&
  358. pmbmdLeft->m_rgTranspStatus [Y_BLOCK2] != ALL) {
  359. blkmRet = pmbmLeft->rgblkm [A_BLOCK2 - 1];
  360. iQPpred = pmbmdLeft->m_stepSizeAlpha;
  361. }
  362. break;
  363. case A_BLOCK4:
  364. if (pmbmdCurr->m_rgTranspStatus [Y_BLOCK1] != ALL) {
  365. blkmRet = pmbmCurr->rgblkm [A_BLOCK1 - 1];
  366. iQPpred = pmbmdCurr->m_stepSizeAlpha;
  367. }
  368. break;
  369. default: //U, V block
  370. if (pmbmLeftTop != NULL && 
  371. (pmbmdLeftTop->m_dctMd == INTRA || pmbmdLeftTop->m_dctMd == INTRAQ) &&
  372. pmbmdLeftTop->m_rgTranspStatus [ALL_Y_BLOCKS] != ALL) {
  373. blkmRet = pmbmLeftTop->rgblkm [iBlk - 1];
  374. iQPpred = pmbmdLeftTop->m_stepSize;
  375. }
  376. }
  377. }
  378. else 
  379. assert (FALSE);
  380. /*BBM// Added for Boundary by Hyundai(1998-5-9)
  381. if (m_vopmd.bInterlace && pmbmdCurr->m_bMerged [0])
  382.                 swapTransparentModes ((CMBMode*)pmbmdCurr, BBS);
  383. // End of Hyundai(1998-5-9)*/
  384. return (blkmRet);
  385. }