开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 多媒体 > 流媒体/Mpeg4/MP4 > 查看源码
mb.cpp
资源名称：NETVIDEO.rar　[点击查看]
上传用户：sun1608
上传日期：2007-02-02
资源大小：6116k
文件大小：13k
源码类别：

流媒体/Mpeg4/MP4

开发平台：

Visual C++

mb.cpp：源码内容
  1. /*************************************************************************
  3. This software module was originally developed by 
  5. Ming-Chieh Lee (mingcl@microsoft.com), Microsoft Corporation
  6. Wei-ge Chen (wchen@microsoft.com), Microsoft Corporation
  7. Bruce Lin (blin@microsoft.com), Microsoft Corporation
  8. Chuang Gu (chuanggu@microsoft.com), Microsoft Corporation
  9. (date: March, 1996)
  10. and edited by
  11.         Wei Wu (weiwu@stallion.risc.rockwell.com) Rockwell Science Center
  13. in the course of development of the MPEG-4 Video (ISO/IEC 14496-2). 
  14. This software module is an implementation of a part of one or more MPEG-4 Video tools 
  15. as specified by the MPEG-4 Video. 
  16. ISO/IEC gives users of the MPEG-4 Video free license to this software module or modifications 
  17. thereof for use in hardware or software products claiming conformance to the MPEG-4 Video. 
  18. Those intending to use this software module in hardware or software products are advised that its use may infringe existing patents. 
  19. The original developer of this software module and his/her cany, 
  20. the subsequent editors and their companies, 
  21. and ISO/IEC have no liability for use of this software module or modifications thereof in an implementation. 
  22. Copyright is not released for non MPEG-4 Video conforming products. 
  23. Microsoft retains full right to use the code for his/her own purpose, 
  24. assign or donate the code to a third party and to inhibit third parties from using the code for non <MPEG standard> conforming products. 
  25. This copyright notice must be included in all copies or derivative works. 
  27. Copyright (c) 1996, 1997.
  29. Module Name:
  31. MB.hpp
  33. Abstract:
  35. MacroBlock base class 
  37. Revision History:
  38.         May. 9   1998:  add chroma subsampling and field based MC padding by Hyundai Electronics
  39.                                   Cheol-Soo Park (cspark@super5.hyundai.co.kr)
  41. *************************************************************************/
  43. #include <stdio.h>
  45. #include "typeapi.h"
  46. #include "codehead.h"
  47. #include "mode.hpp"
  48. #include "dct.hpp"
  49. #include "vopses.hpp"
  51. #ifdef __MFC_
  52. #ifdef _DEBUG
  53. #undef THIS_FILE
  54. static char BASED_CODE THIS_FILE[] = __FILE__;
  55. #endif
  57. #define new DEBUG_NEW    
  58. #endif // __MFC_
  60. Void CVideoObject::decideTransparencyStatus (CMBMode* pmbmd, const PixelC* ppxlcMBBY)
  61. {
  62. const PixelC* ppxlcBlkBY1 = ppxlcMBBY;
  63. const PixelC* ppxlcBlkBY2 = ppxlcMBBY + BLOCK_SIZE;
  64. const PixelC* ppxlcBlkBY3 = ppxlcMBBY + MB_SIZE * BLOCK_SIZE;
  65. const PixelC* ppxlcBlkBY4 = ppxlcMBBY + MB_SIZE * BLOCK_SIZE + BLOCK_SIZE;
  66. // const PixelC* ppxlcBlkBUV = m_ppxlcCurrMBBUV;
  67. UInt uiNonTransPixels1 = 0, uiNonTransPixels2 = 0;
  68. UInt uiNonTransPixels3 = 0, uiNonTransPixels4 = 0;
  69. // UInt uiNonTransPixelsUV = 0;
  70. CoordI ix, iy;
  71. for (iy = 0; iy < BLOCK_SIZE; iy++) {
  72. for (ix = 0; ix < BLOCK_SIZE; ix++) {
  73. uiNonTransPixels1 += ppxlcBlkBY1 [ix];
  74. uiNonTransPixels2 += ppxlcBlkBY2 [ix];
  75. uiNonTransPixels3 += ppxlcBlkBY3 [ix];
  76. uiNonTransPixels4 += ppxlcBlkBY4 [ix];
  77. // uiNonTransPixelsUV += ppxlcBlkBUV [ix];
  78. }
  79. ppxlcBlkBY1 += MB_SIZE;
  80. ppxlcBlkBY2 += MB_SIZE;
  81. ppxlcBlkBY3 += MB_SIZE;
  82. ppxlcBlkBY4 += MB_SIZE;
  83. // ppxlcBlkBUV += BLOCK_SIZE;
  84. }
  85. uiNonTransPixels1 /= opaqueValue;
  86. uiNonTransPixels2 /= opaqueValue;
  87. uiNonTransPixels3 /= opaqueValue;
  88. uiNonTransPixels4 /= opaqueValue;
  89. // uiNonTransPixelsUV /= opaqueValue;
  91. pmbmd->m_rgNumNonTranspPixels [0] = 
  92. uiNonTransPixels1 + uiNonTransPixels2 + uiNonTransPixels3 + uiNonTransPixels4;
  93. pmbmd->m_rgNumNonTranspPixels [1] = uiNonTransPixels1;
  94. pmbmd->m_rgNumNonTranspPixels [2] = uiNonTransPixels2;
  95. pmbmd->m_rgNumNonTranspPixels [3] = uiNonTransPixels3;
  96. pmbmd->m_rgNumNonTranspPixels [4] = uiNonTransPixels4;
  97. // pmbmd->m_rgNumNonTranspPixels [5] = uiNonTransPixelsUV;
  98. if (pmbmd->m_rgNumNonTranspPixels [0] == 0) {
  99. pmbmd->m_rgTranspStatus [0] = ALL;
  100. pmbmd->m_dctMd = INTER;
  101. }
  102. else if (pmbmd->m_rgNumNonTranspPixels [0] == MB_SQUARE_SIZE)
  103. pmbmd -> m_rgTranspStatus [0] = NONE;
  104. else
  105. pmbmd -> m_rgTranspStatus [0] = PARTIAL;
  107. // for (ix = (UInt) Y_BLOCK1; ix <= (UInt) U_BLOCK; ix++) {
  108. for (ix = (UInt) Y_BLOCK1; ix < U_BLOCK; ix++) {
  109. if (pmbmd -> m_rgNumNonTranspPixels [ix] == 0)
  110. pmbmd -> m_rgTranspStatus [ix] = ALL;
  111. else if (pmbmd -> m_rgNumNonTranspPixels [ix] == BLOCK_SQUARE_SIZE)
  112. pmbmd -> m_rgTranspStatus [ix] = NONE;
  113. else
  114. pmbmd -> m_rgTranspStatus [ix] = PARTIAL;
  115. }
  116. //pmbmd->m_rgTranspStatus [V_BLOCK] = pmbmd->m_rgTranspStatus [U_BLOCK];
  117. pmbmd->m_rgTranspStatus [V_BLOCK] = pmbmd->m_rgTranspStatus [U_BLOCK] 
  118. = pmbmd -> m_rgTranspStatus [0];
  119. }
  121. Void CVideoObject::decideMBTransparencyStatus (CMBMode* pmbmd)
  122. {
  123. UInt uiNonTransPixels0 = 0;
  124. UInt ix;
  125. for (ix = 0; ix < MB_SQUARE_SIZE; ix++)
  126. uiNonTransPixels0 += m_ppxlcCurrMBBY [ix];
  127. uiNonTransPixels0 /= opaqueValue;
  129. pmbmd -> m_rgNumNonTranspPixels [0] = uiNonTransPixels0;
  130. if (uiNonTransPixels0 == 0) {
  131. pmbmd -> m_rgTranspStatus [0] = ALL;
  132. pmbmd -> m_dctMd = INTER;
  133. }
  134. else if (uiNonTransPixels0 == MB_SQUARE_SIZE)
  135. pmbmd -> m_rgTranspStatus [0] = NONE;
  136. else
  137. pmbmd -> m_rgTranspStatus [0] = PARTIAL;
  138. }
  140. // Added for Field Based MC Padding by Hyundai(1998-5-9)
  141. Void CVideoObject::decideFieldTransparencyStatus (
  142.         CMBMode* pmbmd,
  143.         const PixelC* ppxlcMBBY,
  144.         const PixelC* ppxlcMBBUV
  145. )
  146. {
  147.         Int *piNonTransPixels = new Int [5];
  148.         Int iYFieldSkip = 2*m_iFrameWidthY;
  149.         Int iCFieldSkip = 2*m_iFrameWidthUV;
  150.         CoordI ix, iy;
  151.  
  152.         // for field based mc-padding
  153.         memset (piNonTransPixels, 0, sizeof(Int)*5);
  155.         const PixelC* ppxlcTopFieldBY = ppxlcMBBY;
  156.         const PixelC* ppxlcBotFieldBY = ppxlcMBBY + m_iFrameWidthY;
  157.         for (iy = 0; iy < BLOCK_SIZE; iy++) {
  158.                 for (ix = 0; ix < MB_SIZE; ix++) {
  159.                         piNonTransPixels[1] += ppxlcTopFieldBY [ix];
  160.                         piNonTransPixels[2] += ppxlcBotFieldBY [ix];
  161.                 }
  162.                 ppxlcTopFieldBY += iYFieldSkip;
  163.                 ppxlcBotFieldBY += iYFieldSkip;
  164.         }
  165.         piNonTransPixels[1] /= opaqueValue;
  166.         piNonTransPixels[2] /= opaqueValue;
  167.         for (ix=1; ix<=2; ix++) {
  168.                 if (piNonTransPixels [ix] == 0)
  169.                         pmbmd-> m_rgFieldTranspStatus [ix] = ALL;
  170.                 else if (piNonTransPixels [ix] == MB_SQUARE_SIZE/2)
  171.                         pmbmd -> m_rgFieldTranspStatus [ix] = NONE;
  172.                 else    pmbmd -> m_rgFieldTranspStatus [ix] = PARTIAL;
  173.         }
  174.         ppxlcTopFieldBY = ppxlcMBBUV;
  175.         ppxlcBotFieldBY = ppxlcMBBUV + m_iFrameWidthUV;
  176.         for (iy = 0; iy < BLOCK_SIZE/2; iy++) {
  177.                 for (ix = 0; ix < BLOCK_SIZE; ix++) {
  178.                         piNonTransPixels[3] += ppxlcTopFieldBY [ix];
  179.                         piNonTransPixels[4] += ppxlcBotFieldBY [ix];
  180.                 }
  181.                 ppxlcTopFieldBY += iCFieldSkip;
  182.                 ppxlcBotFieldBY += iCFieldSkip;
  183.         }
  184.         piNonTransPixels[3] /= opaqueValue;
  185.         piNonTransPixels[4] /= opaqueValue;
  186.         for (ix=3; ix<=4; ix++) {
  187.                 if (piNonTransPixels [ix] == 0)
  188.                         pmbmd-> m_rgFieldTranspStatus [ix] = ALL;
  189.                 else if (piNonTransPixels [ix] == BLOCK_SQUARE_SIZE/2)
  190.                         pmbmd -> m_rgFieldTranspStatus [ix] = NONE;
  191.                 else    pmbmd -> m_rgFieldTranspStatus [ix] = PARTIAL;
  192.         }
  193.         
  194.         delete piNonTransPixels;
  195. }
  197. Void CVideoObject::fieldBasedDownSampleBY (
  198.         const PixelC* ppxlcMBBY,
  199.         PixelC* ppxlcMBBUV
  200. )
  201. {
  202.         UInt  ix, iy, x, iFieldSkip = 4*m_iFrameWidthY;
  203.         const PixelC* ppxlcTopFieldBY = ppxlcMBBY;
  204.         const PixelC* ppxlcBotFieldBY = ppxlcMBBY + m_iFrameWidthY;
  205.         const PixelC* ppxlcTopFieldBYNext = ppxlcTopFieldBY + 2*m_iFrameWidthY;
  206.         const PixelC* ppxlcBotFieldBYNext = ppxlcBotFieldBY + 2*m_iFrameWidthY;
  207.  
  208.         for (iy = 0; iy < BLOCK_SIZE/2; iy++) {
  209.                 for (ix = 0, x = 0; ix < BLOCK_SIZE; ix++, x = 2*ix)
  210.                         ppxlcMBBUV [ix] = (ppxlcTopFieldBY [x] | ppxlcTopFieldBY [x+1] |
  211.                                                 ppxlcTopFieldBYNext [x] | ppxlcTopFieldBYNext [x+1]);
  212.                 ppxlcMBBUV += m_iFrameWidthUV;
  213.                 for (ix = 0, x = 0; ix < BLOCK_SIZE; ix++, x = 2*ix)
  214.                         ppxlcMBBUV [ix] = (ppxlcBotFieldBY [x] | ppxlcBotFieldBY [x+1] |
  215.                                                 ppxlcBotFieldBYNext [x] | ppxlcBotFieldBYNext [x+1]);
  216.                 ppxlcMBBUV += m_iFrameWidthUV;
  217.                 ppxlcTopFieldBY += iFieldSkip;
  218.                 ppxlcBotFieldBY += iFieldSkip;
  219.                 ppxlcTopFieldBYNext += iFieldSkip;
  220.                 ppxlcBotFieldBYNext += iFieldSkip;
  221.         }
  222. }
  224. // End of Hyundai(1998-5-9)
  226. Void CVideoObject::downSampleBY (const PixelC* ppxlcMBBY, PixelC* ppxlcMBBUV)
  227. {
  228. const PixelC* ppxlcMBBYNextRow = ppxlcMBBY + MB_SIZE;
  229. UInt ix, iy;
  230. // Added for Chroma Subsampling by Hyundai(1998-5-9)
  231.         if (m_vopmd.bInterlace) {
  232.                 UInt  x, iFieldSkip = 4*MB_SIZE;
  233.                 const PixelC* ppxlcTopFieldBY = ppxlcMBBY;
  234.                 const PixelC* ppxlcBotFieldBY = ppxlcMBBY + MB_SIZE;
  235.                 const PixelC* ppxlcTopFieldBYNext = ppxlcTopFieldBY + 2*MB_SIZE;
  236.                 const PixelC* ppxlcBotFieldBYNext = ppxlcBotFieldBY + 2*MB_SIZE;
  237.                 for (iy = 0; iy < BLOCK_SIZE/2; iy++) {
  238.                         for (ix = 0, x = 0; ix < BLOCK_SIZE; ix++, x = 2*ix)
  239.                                 ppxlcMBBUV [ix] = (ppxlcTopFieldBY [x] | ppxlcTopFieldBY [x+1] |
  240.                                                         ppxlcTopFieldBYNext [x] | ppxlcTopFieldBYNext [x+1]);
  241.                         ppxlcMBBUV += BLOCK_SIZE;
  242.                         for (ix = 0, x = 0; ix < BLOCK_SIZE; ix++, x = 2*ix)
  243.                                 ppxlcMBBUV [ix] = (ppxlcBotFieldBY [x] | ppxlcBotFieldBY [x+1] |
  244.                                                         ppxlcBotFieldBYNext [x] | ppxlcBotFieldBYNext [x+1]);
  245.                         ppxlcMBBUV += BLOCK_SIZE;
  246.                         ppxlcTopFieldBY += iFieldSkip;
  247.                         ppxlcBotFieldBY += iFieldSkip;
  248.                         ppxlcTopFieldBYNext += iFieldSkip;
  249.                         ppxlcBotFieldBYNext += iFieldSkip;
  250.                 }
  251.                 return;
  252.         }
  253. // End of Hyundai(1998-5-9)
  254. for (iy = 0; iy < BLOCK_SIZE; iy++) {
  255. for (ix = 0; ix < BLOCK_SIZE; ix++) {
  256. ppxlcMBBUV [ix] = (
  257. (ppxlcMBBY [2 * ix] | ppxlcMBBY [2 * ix + 1]) |
  258. (ppxlcMBBYNextRow [2 * ix] | ppxlcMBBYNextRow [2 * ix + 1])
  259. );
  260. }
  261. ppxlcMBBUV += BLOCK_SIZE;
  262. ppxlcMBBY += 2 * MB_SIZE;
  263. ppxlcMBBYNextRow += 2 * MB_SIZE;
  264. }
  265. }
  267. Void CVideoObject::addErrorAndPredToCurrQ (
  268. PixelC* ppxlcQMBY, 
  269. PixelC* ppxlcQMBU, 
  270. PixelC* ppxlcQMBV
  271. )
  272. {
  273. CoordI ix, iy, ic = 0;
  274. for (iy = 0; iy < MB_SIZE; iy++) {
  275. for (ix = 0; ix < MB_SIZE; ix++, ic++) {
  276. ppxlcQMBY [ix] = m_rgiClipTab [m_ppxlcPredMBY [ic] + m_ppxliErrorMBY [ic]];
  278. }
  279. ppxlcQMBY += m_iFrameWidthY;
  280. }
  282. ic = 0;
  283. for (iy = 0; iy < BLOCK_SIZE; iy++) {
  284. for (ix = 0; ix < BLOCK_SIZE; ix++, ic++) {
  285. ppxlcQMBU [ix] = m_rgiClipTab [m_ppxlcPredMBU [ic] + m_ppxliErrorMBU [ic]];
  286. ppxlcQMBV [ix] = m_rgiClipTab [m_ppxlcPredMBV [ic] + m_ppxliErrorMBV [ic]];
  287. }
  288. ppxlcQMBU += m_iFrameWidthUV;
  289. ppxlcQMBV += m_iFrameWidthUV;
  290. }
  291. }
  293. Void CVideoObject::addAlphaErrorAndPredToCurrQ (PixelC* ppxlcQMBA)
  294. {
  295. CoordI ix, iy, ic = 0;
  296. for (iy = 0; iy < MB_SIZE; iy++) {
  297. for (ix = 0; ix < MB_SIZE; ix++, ic++)
  298. ppxlcQMBA [ix] = m_rgiClipTab [m_ppxlcPredMBA [ic] + m_ppxliErrorMBA [ic]];
  299. ppxlcQMBA += m_iFrameWidthY;
  300. }
  301. }
  303. Void CVideoObject::assignPredToCurrQ (
  304. PixelC* ppxlcQMBY, 
  305. PixelC* ppxlcQMBU, 
  306. PixelC* ppxlcQMBV
  307. )
  308. {
  309. CoordI ix;
  310. const PixelC* ppxlcPredMBY = m_ppxlcPredMBY;
  311. const PixelC* ppxlcPredMBU = m_ppxlcPredMBU;
  312. const PixelC* ppxlcPredMBV = m_ppxlcPredMBV;
  313. for (ix = 0; ix < BLOCK_SIZE; ix++) {
  314. memcpy (ppxlcQMBY, ppxlcPredMBY, MB_SIZE*sizeof(PixelC));
  315. memcpy (ppxlcQMBU, ppxlcPredMBU, BLOCK_SIZE*sizeof(PixelC));
  316. memcpy (ppxlcQMBV, ppxlcPredMBV, BLOCK_SIZE*sizeof(PixelC));
  318. ppxlcPredMBY += MB_SIZE;
  319. ppxlcPredMBU += BLOCK_SIZE;
  320. ppxlcPredMBV += BLOCK_SIZE;
  321. ppxlcQMBY += m_iFrameWidthY;
  322. ppxlcQMBU += m_iFrameWidthUV;
  323. ppxlcQMBV += m_iFrameWidthUV;
  325. memcpy (ppxlcQMBY, ppxlcPredMBY, MB_SIZE*sizeof(PixelC));
  326. ppxlcPredMBY += MB_SIZE;
  327. ppxlcQMBY += m_iFrameWidthY;
  328. }
  329. }
  331. Void CVideoObject::assignAlphaPredToCurrQ (PixelC* ppxlcQMBA)
  332. {
  333. CoordI ix;
  334. const PixelC* ppxlcPredMBA = m_ppxlcPredMBA;
  335. for (ix = 0; ix < MB_SIZE; ix++) {
  336. memcpy (ppxlcQMBA, ppxlcPredMBA, MB_SIZE*sizeof(PixelC));
  337. ppxlcPredMBA += MB_SIZE;
  338. ppxlcQMBA += m_iFrameWidthY;
  339. }
  340. }
  342. Void CVideoObject::findColocatedMB (Int iMBX, Int iMBY,
  343. const CMBMode*& pmbmdRef,
  344. const CMotionVector*& pmvRef)
  345. {
  346. Bool bColocatedMBExist = FALSE;
  348. if(m_bCodedFutureRef!=FALSE)
  349. if(iMBX<m_iNumMBXRef && iMBX>=0 && iMBY<m_iNumMBYRef && iMBY>=0)
  350. {
  351. pmbmdRef = &m_rgmbmdRef [iMBX + iMBY * m_iNumMBXRef];
  352. if(pmbmdRef -> m_rgTranspStatus [0] != ALL)
  353. {
  354. Int iOffset = min (max (0, iMBX), m_iNumMBXRef - 1) + 
  355.   min (max (0, iMBY), m_iNumMBYRef - 1) * m_iNumMBXRef;
  356. pmvRef = m_rgmvRef + iOffset * PVOP_MV_PER_REF_PER_MB;
  357. bColocatedMBExist = TRUE;
  358. }
  359. }
  361. if(bColocatedMBExist==FALSE)
  362. {
  363. pmbmdRef = NULL;
  364. pmvRef = NULL;
  365. }
  366. }