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

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

流媒体/Mpeg4/MP4

开发平台：

Visual C++

shape.cpp：源码内容
  1. /*************************************************************************
  3. This software module was originally developed by 
  5. Wei-ge Chen (wchen@microsoft.com), Microsoft Corporation
  6. (April, 1997)
  7. and edited by
  8.         Wei Wu (weiwu@stallion.risc.rockwell.com) Rockwell Science Center
  10. and also edited by
  11. Yoshihiro Kikuchi (TOSHIBA CORPORATION)
  12. Takeshi Nagai (TOSHIBA CORPORATION)
  13. Toshiaki Watanabe (TOSHIBA CORPORATION)
  14. Noboru Yamaguchi (TOSHIBA CORPORATION)
  16. in the course of development of the MPEG-4 Video (ISO/IEC 14496-2). 
  17. This software module is an implementation of a part of one or more MPEG-4 Video tools 
  18. as specified by the MPEG-4 Video. 
  19. ISO/IEC gives users of the MPEG-4 Video free license to this software module or modifications 
  20. thereof for use in hardware or software products claiming conformance to the MPEG-4 Video. 
  21. Those intending to use this software module in hardware or software products are advised that its use may infringe existing patents. 
  22. The original developer of this software module and his/her company, 
  23. the subsequent editors and their companies, 
  24. and ISO/IEC have no liability for use of this software module or modifications thereof in an implementation. 
  25. Copyright is not released for non MPEG-4 Video conforming products. 
  26. Microsoft retains full right to use the code for his/her own purpose, 
  27. assign or donate the code to a third party and to inhibit third parties from using the code for non <MPEG standard> conforming products. 
  28. This copyright notice must be included in all copies or derivative works. 
  30. Copyright (c) 1996, 1997.
  32. Module Name:
  34. shpenc.hpp
  36. Abstract:
  38. binary shape encoder with context-based arithmatic coder
  40. Revision History:
  42. *************************************************************************/
  44. #include "typeapi.h"
  45. #include "entropy/entropy.hpp"
  46. #include "entropy/huffman.hpp"
  47. #include "entropy/bitstrm.hpp"
  48. #include "global.hpp"
  49. #include "mode.hpp"
  50. #include "codehead.h"
  51. #include "cae.h"
  52. #include "vopses.hpp"
  55. #ifdef __MFC_
  56. #ifdef _DEBUG
  57. #undef THIS_FILE
  58. static char BASED_CODE THIS_FILE[] = __FILE__;
  59. #endif
  61. #define new DEBUG_NEW    
  62. #endif // __MFC_
  64. own Int* CVideoObject::computeShapeSubBlkIndex (Int iSubBlkSize, Int iSrcSize)
  65. {
  66. Int* rgiShapeSubBlkIndex = new Int [MB_SIZE * MB_SIZE / iSubBlkSize / iSubBlkSize];
  67. Int nBorderLine = (iSrcSize - MB_SIZE) / 2;
  68. CoordI iX, iY, i = 0;
  69. for (iY = nBorderLine; iY < (MB_SIZE + nBorderLine); iY += iSubBlkSize) {
  70. for (iX = nBorderLine; iX < (MB_SIZE + nBorderLine); iX += iSubBlkSize)
  71. rgiShapeSubBlkIndex [i++] = iY * iSrcSize + iX;
  72. }
  73. return rgiShapeSubBlkIndex;
  74. }
  76. Void CVideoObject::downSampleShapeMCPred(const PixelC*ppxlcSrc,
  77. PixelC* ppxlcDst,Int iRate)
  78. {
  79. assert(iRate==1 || iRate==2 || iRate==4);
  80. static Int rgiScan[16] = {0,1,MC_BAB_SIZE,MC_BAB_SIZE+1,
  81. 2,3,MC_BAB_SIZE+2,MC_BAB_SIZE+3,
  82. 2*MC_BAB_SIZE,2*MC_BAB_SIZE+1,2*MC_BAB_SIZE+2,2*MC_BAB_SIZE+3,
  83. 3*MC_BAB_SIZE,3*MC_BAB_SIZE+1,3*MC_BAB_SIZE+2,3*MC_BAB_SIZE+3};
  84. static Int rgiThresh[5] = {0,0,MPEG4_OPAQUE,0,7*MPEG4_OPAQUE};
  85. Int iBdrThresh=0;
  86. if(iRate>2)
  87. iBdrThresh=MPEG4_OPAQUE;
  88. Int iThreshold = rgiThresh[iRate];
  90. #ifdef __TRACE_AND_STATS_
  91. //m_pbitstrmOut->trace ((PixelC*)ppxlcSrc, MC_BAB_SIZE, MC_BAB_SIZE, "MB_MC_BAB_ORIG");
  92. #endif //__TRACE_AND_STATS_
  94. const PixelC *ppxlcSrcScan = ppxlcSrc+(MC_BAB_SIZE+1)*MC_BAB_BORDER;
  95. Int iDstSize = 2*MC_BAB_BORDER+(MC_BAB_SIZE-2*MC_BAB_BORDER)/iRate;
  96. PixelC *ppxlcDstScan = ppxlcDst+(iDstSize+1)*MC_BAB_BORDER;
  97. Int iX,iY,i,iSum,iSum1,iSum2,iSum3,iSum4;
  98. Int iRateSqd=iRate*iRate;
  99. const PixelC *ppxlcSrcBdr1 = ppxlcSrc+MC_BAB_SIZE*MC_BAB_BORDER;
  100. const PixelC *ppxlcSrcBdr2 = ppxlcSrc+2*MC_BAB_SIZE-1;
  101. const PixelC *ppxlcSrcBdr3 = ppxlcSrc+MC_BAB_BORDER;
  102. const PixelC *ppxlcSrcBdr4 = ppxlcSrc+MC_BAB_SIZE*(MC_BAB_SIZE-1)+MC_BAB_BORDER;
  103. PixelC *ppxlcDstBdr1 = ppxlcDst+iDstSize*MC_BAB_BORDER;
  104. PixelC *ppxlcDstBdr2 = ppxlcDst+2*iDstSize-1;
  105. PixelC *ppxlcDstBdr3 = ppxlcDst+MC_BAB_BORDER;
  106. PixelC *ppxlcDstBdr4 = ppxlcDst+iDstSize*(iDstSize-1)+MC_BAB_BORDER;
  108. for(iY=MC_BAB_BORDER;iY<iDstSize-MC_BAB_BORDER;iY++)
  109. {
  110. for(iX=MC_BAB_BORDER;iX<iDstSize-MC_BAB_BORDER;iX++,ppxlcDstScan++)
  111. {
  112. iSum=0;
  113. for(i=0;i<iRateSqd;i++)
  114. iSum+=ppxlcSrcScan[rgiScan[i]];
  115. *ppxlcDstScan=(iSum>iThreshold)?MPEG4_OPAQUE:MPEG4_TRANSPARENT;
  116. ppxlcSrcScan+=iRate;
  117. }
  118. ppxlcDstScan+=2*MC_BAB_BORDER;
  119. ppxlcSrcScan+=MC_BAB_SIZE*iRate-MC_BAB_SIZE+2*MC_BAB_BORDER;
  121. // border
  122. iSum1=iSum2=iSum3=iSum4=0;
  123. for(i=0;i<iRate;i++)
  124. {
  125. iSum1+=ppxlcSrcBdr1[i*MC_BAB_SIZE];
  126. iSum2+=ppxlcSrcBdr2[i*MC_BAB_SIZE];
  127. iSum3+=ppxlcSrcBdr3[i];
  128. iSum4+=ppxlcSrcBdr4[i];
  129. }
  130. *ppxlcDstBdr1=(iSum1>iBdrThresh)?MPEG4_OPAQUE:MPEG4_TRANSPARENT;
  131. *ppxlcDstBdr2=(iSum2>iBdrThresh)?MPEG4_OPAQUE:MPEG4_TRANSPARENT;
  132. *ppxlcDstBdr3++=(iSum3>iBdrThresh)?MPEG4_OPAQUE:MPEG4_TRANSPARENT;
  133. *ppxlcDstBdr4++=(iSum4>iBdrThresh)?MPEG4_OPAQUE:MPEG4_TRANSPARENT;
  134. ppxlcDstBdr1+=iDstSize;
  135. ppxlcDstBdr2+=iDstSize;
  136. ppxlcSrcBdr1+=MC_BAB_SIZE*iRate;
  137. ppxlcSrcBdr2+=MC_BAB_SIZE*iRate;
  138. ppxlcSrcBdr3+=iRate;
  139. ppxlcSrcBdr4+=iRate;
  140. }
  142. // corners
  143. ppxlcDst[0]=ppxlcSrc[0];
  144. ppxlcDst[iDstSize-1]=ppxlcSrc[MC_BAB_SIZE-1];
  145. ppxlcDst[(iDstSize-1)*iDstSize]=ppxlcSrc[(MC_BAB_SIZE-1)*MC_BAB_SIZE];
  146. ppxlcDst[iDstSize*iDstSize-1]=ppxlcSrc[MC_BAB_SIZE*MC_BAB_SIZE-1];
  148. #ifdef __TRACE_AND_STATS_
  149. //m_pbitstrmOut->trace ((PixelC*)ppxlcDst, iDstSize, iDstSize, "MB_MC_BAB_DOWN");
  150. #endif // __TRACE_AND_STATS_
  151. }
  153. Void CVideoObject::upSampleShape(PixelC*ppxlcBYFrm,const PixelC* rgpxlcSrc, PixelC* rgpxlcDst)
  154. {
  155. if(m_iInverseCR==2)
  156. adaptiveUpSampleShape(rgpxlcSrc, rgpxlcDst,8,8);
  157. else
  158. {
  159. static PixelC rgpxlcTmp[12*12];
  161. assert(m_iInverseCR==4);
  162. adaptiveUpSampleShape(rgpxlcSrc, rgpxlcTmp,4,4);
  164. Int iSizeTmp=12;
  165. Int iSizeSrc=8;
  166. Int i,j;
  168. // top left corner 
  169. rgpxlcTmp[0]=rgpxlcSrc[0];
  170. rgpxlcTmp[1]=rgpxlcSrc[1];
  171. rgpxlcTmp[iSizeTmp]=rgpxlcSrc[iSizeSrc];
  172. rgpxlcTmp[iSizeTmp+1]=rgpxlcSrc[iSizeSrc+1];
  174. // top right corner
  175. rgpxlcTmp[iSizeTmp-2]=rgpxlcSrc[iSizeSrc-2];
  176. rgpxlcTmp[iSizeTmp-1]=rgpxlcSrc[iSizeSrc-1];
  177. rgpxlcTmp[(iSizeTmp<<1)-2]=rgpxlcSrc[(iSizeSrc<<1)-2];
  178. rgpxlcTmp[(iSizeTmp<<1)-1]=rgpxlcSrc[(iSizeSrc<<1)-1];
  180. // top border
  181. for(j=0;j<2;j++)
  182. for(i=BAB_BORDER;i<iSizeTmp-BAB_BORDER;i++)
  183. rgpxlcTmp[j*iSizeTmp+i]=rgpxlcSrc[j*iSizeSrc+i/2+1];
  185. // left border 
  186. for(i=0;i<2;i++)
  187. for(j=BAB_BORDER;j<iSizeTmp-BAB_BORDER;j++)
  188. rgpxlcTmp[j*iSizeTmp+i]=rgpxlcSrc[(j/2+1)*iSizeSrc+i];
  190. /*Int iTmp = m_iWidthCurrBAB;
  191. m_iInverseCR = 2;
  192. m_iWidthCurrBAB = 12;
  193. subsampleLeftTopBorderFromVOP (ppxlcBYFrm, rgpxlcTmp);
  194. m_iInverseCR = 4;
  195. m_iWidthCurrBAB = iTmp;*/
  197. adaptiveUpSampleShape(rgpxlcTmp, rgpxlcDst,8,8);
  198. }
  199. }
  201. Int CVideoObject::getContextUS( PixelC a,
  202.                                 PixelC b,
  203.                                 PixelC c,
  204.                                 PixelC d,
  205.                                 PixelC e,
  206.                                 PixelC f,
  207.                                 PixelC g,
  208.                                 PixelC h)
  209. {
  210.         Int     ret =(Int)(a+(b<<1)+(c<<2)+(d<<3)+(e<<4)+(f<<5)+(g<<6)+(h<<7));
  211.  
  212.         return  ret;
  213. }
  215. PixelC CVideoObject::getRefValue(const PixelC* ppxlcRow,
  216.                                 Int x_adr, Int y_adr,
  217.                                 Int h_size, Int v_size)
  218. {
  219.         assert ((x_adr>=-2) && (x_adr<=h_size+1) && (y_adr>=-2) && (y_adr<=v_size+1));
  220.  
  221.         Int     x_lim, y_lim;
  222.         PixelC  ret;
  223.  
  224.         if(x_adr>=0 && x_adr<h_size && y_adr>=0 && y_adr<v_size) {
  225.                 ret=(ppxlcRow[y_adr*(h_size+BAB_BORDER_BOTH)+x_adr]>0) ? 1 : 0;
  226.         } else {
  227.                 if(y_adr<0 || (x_adr<0 && y_adr<v_size)) {
  228. ret=(ppxlcRow[y_adr*(h_size+BAB_BORDER_BOTH)+x_adr]>0) ? 1 : 0;
  229.                 } else {
  230.                         x_lim=(x_adr<0)? 0: (x_adr>=h_size)? h_size-1 :x_adr;
  231.                         y_lim=(y_adr<0)? 0: (y_adr>=v_size)? v_size-1 :y_adr;
  232.                         ret=(ppxlcRow[y_lim*(h_size+BAB_BORDER_BOTH)+x_lim]>0) ? 1 : 0;
  233.                 }
  234.         }
  235.  
  236.         return ret;
  237. }
  239. Void CVideoObject::adaptiveUpSampleShape (const PixelC* rgpxlcSrc, 
  240. PixelC* rgpxlcDst,
  241. Int h_size, Int v_size)
  242. {
  243. Int x[12], y[12], other_total, context, rvalue, th, m;
  244. Int width_up = h_size << 1;
  245. Int width_up_border = width_up + BAB_BORDER_BOTH;
  246. Int j, i, py, px, py_st, py_en, px_st, px_en, py_p;
  247. PixelC val[12];
  248. const PixelC* ppxlcRow = rgpxlcSrc + BAB_BORDER+BAB_BORDER * (h_size + BAB_BORDER_BOTH);
  250. /* --  4  5 -- */
  251. /*  6  0  1  7 */
  252. /*  8  2  3  9 */
  253. /* -- 10 11 -- */
  254. for (j = -1; j < v_size; j++) {
  256. y[0] = y[1] = y[6] =y[7] = j;
  257. y[2] = y[3] = y[8] =y[9] = j + 1;
  258. y[4] = y[5] = j - 1;
  259. y[10]= y[11]= j + 2;
  260. if (j > -1) 
  261. py_st = 0;
  262. else  
  263. py_st = 1;
  264. if (j < v_size - 1) 
  265. py_en = 2;
  266. else        
  267. py_en = 1;
  269. for(i = -1; i < h_size; i++) {
  270. x[0] = x[2] = x[4] = x[10] = i;
  271. x[1] = x[3] = x[5] = x[11] = i+1;
  272. x[6] = x[8] = i - 1;
  273. x[7] = x[9] = i + 2;
  275. for(m = 0; m < 12; m++)
  276. val[m] = getRefValue (ppxlcRow, x[m], y[m], h_size, v_size);
  278. if (i > -1) 
  279. px_st = 0;
  280. else     
  281. px_st = 1;
  283. if (i < h_size-1) 
  284. px_en = 2;
  285. else   
  286. px_en=1;
  288. other_total = val[4] + val[5] + val[6] + val[7] + val[8] + val[9] + val[10] + val[11];
  289. for (py = py_st; py < py_en; py++) {
  290. py_p = ((j << 1) + 1 + py + BAB_BORDER) * width_up_border + (i << 1) + 1;
  291. for (px = px_st; px < px_en; px++) {
  292. if(px < 1 && py < 1) {
  293. context = getContextUS (val[5], val[4], val[6], val[8], val[10], val[11], val[9], val[7]);
  294. th = grgchInterpolationFilterTh[context];
  295. rvalue = (val[0] << 2) + ((val[1] + val[2] + val[3]) << 1) + other_total;
  296. } 
  297. else if (py < 1) {
  298. context = getContextUS(val[9], val[7], val[5], val[4], val[6], val[8], val[10], val[11]);
  299. th = grgchInterpolationFilterTh [context];
  300. rvalue = (val[1] << 2) + ((val[0] + val[2] + val[3]) << 1) + other_total;
  301. } 
  302. else if (px < 1) {
  303. context = getContextUS(val[6], val[8], val[10], val[11], val[9], val[7], val[5], val[4]);
  304. th = grgchInterpolationFilterTh [context];
  305. rvalue = (val[2] << 2) + ((val[1] + val[0] + val[3]) << 1) + other_total;
  306. } 
  307. else {
  308. context = getContextUS(val[10], val[11], val[9], val[7], val[5], val[4], val[6], val[8]);
  309. th = grgchInterpolationFilterTh [context];
  310. rvalue = (val[3] << 2) + ((val[1] + val[2] + val[0]) << 1) + other_total;
  311. }
  312. rgpxlcDst [py_p+(px+BAB_BORDER)] = (rvalue > th) ? opaqueValue: transpValue;
  313. }
  314. }
  315. }
  316. }
  317. }
  319. Void CVideoObject::copyLeftTopBorderFromVOP (PixelC* ppxlcSrc, PixelC* ppxlcDst)
  320. {
  321. PixelC* ppxlcSrcTop1  = ppxlcSrc - BAB_BORDER  * m_iFrameWidthY - BAB_BORDER;
  322. PixelC* ppxlcSrcTop2  = ppxlcSrcTop1 + m_iFrameWidthY;
  323. PixelC* ppxlcSrcLeft1 = ppxlcSrcTop1;
  324. PixelC* ppxlcSrcLeft2 = ppxlcSrcLeft1 + 1;
  325. PixelC* ppxlcDstTop1  = ppxlcDst;
  326. PixelC* ppxlcDstTop2  = ppxlcDst + TOTAL_BAB_SIZE;
  327. PixelC* ppxlcDstLeft  = ppxlcDst;
  328. CoordI iPixel;
  329. // Modified for error resilient mode by Toshiba(1997-11-14)
  330. for (iPixel = 0; iPixel < BAB_BORDER; iPixel++) {
  331. if (!m_bVPNoLeftTop) {
  332. ppxlcDstTop1 [iPixel]  = *ppxlcSrcTop1;
  333. //src pixel never out of bound due to padding of Ref1
  334. ppxlcDstTop2 [iPixel]  = *ppxlcSrcTop2;
  335. //BY should be intialized as zero outside of VOP
  336. }
  337. else {
  338. ppxlcDstTop1 [iPixel]  = (PixelC) 0;
  339. ppxlcDstTop2 [iPixel]  = (PixelC) 0;
  340. }
  341. ppxlcSrcTop1++; 
  342. ppxlcSrcTop2++; 
  343. }
  344. for (iPixel = BAB_BORDER; iPixel < TOTAL_BAB_SIZE-BAB_BORDER; iPixel++) {
  345. if (!m_bVPNoTop) {
  346. ppxlcDstTop1 [iPixel]  = *ppxlcSrcTop1;
  347. //src pixel never out of bound due to padding of Ref1
  348. ppxlcDstTop2 [iPixel]  = *ppxlcSrcTop2;
  349. //BY should be intialized as zero outside of VOP
  350. }
  351. else {
  352. ppxlcDstTop1 [iPixel]  = (PixelC) 0;
  353. ppxlcDstTop2 [iPixel]  = (PixelC) 0;
  354. }
  355. ppxlcSrcTop1++; 
  356. ppxlcSrcTop2++; 
  357. }
  358. for (iPixel = TOTAL_BAB_SIZE-BAB_BORDER; iPixel < TOTAL_BAB_SIZE; iPixel++) {
  359. if (!m_bVPNoRightTop) {
  360. ppxlcDstTop1 [iPixel]  = *ppxlcSrcTop1;
  361. //src pixel never out of bound due to padding of Ref1
  362. ppxlcDstTop2 [iPixel]  = *ppxlcSrcTop2;
  363. //BY should be intialized as zero outside of VOP
  364. }
  365. else {
  366. ppxlcDstTop1 [iPixel]  = (PixelC) 0;
  367. ppxlcDstTop2 [iPixel]  = (PixelC) 0;
  368. }
  369. ppxlcSrcTop1++; 
  370. ppxlcSrcTop2++; 
  371. }
  372. ppxlcSrcLeft1 += BAB_BORDER*m_iFrameWidthY; 
  373. ppxlcSrcLeft2 += BAB_BORDER*m_iFrameWidthY; 
  374. ppxlcDstLeft += BAB_BORDER*TOTAL_BAB_SIZE;
  375. for (iPixel = BAB_BORDER; iPixel < TOTAL_BAB_SIZE; iPixel++) {
  376. if (!m_bVPNoLeft) {
  377. *ppxlcDstLeft = *ppxlcSrcLeft1;
  378. *(ppxlcDstLeft + 1) = *ppxlcSrcLeft2;
  379. }
  380. else {
  381. *ppxlcDstLeft = (PixelC) 0;
  382. *(ppxlcDstLeft + 1) = (PixelC) 0;
  383. }
  384. ppxlcSrcLeft1 += m_iFrameWidthY; 
  385. ppxlcSrcLeft2 += m_iFrameWidthY; 
  386. ppxlcDstLeft += TOTAL_BAB_SIZE; //last two values of left border will be over written after the loop
  387. }
  388. //repeat pad the left border (this will be done again in right bottom border; improve in the future)
  389. if (!m_bVPNoLeft) {
  390. Int iLastValidSrcLeft1 = *(ppxlcSrcLeft1 - (BAB_BORDER+1) * m_iFrameWidthY);
  391. Int iLastValidSrcLeft2 = *(ppxlcSrcLeft2 - (BAB_BORDER+1) * m_iFrameWidthY);
  392. for (iPixel = 0; iPixel < BAB_BORDER; iPixel++) {
  393. ppxlcDstLeft -= TOTAL_BAB_SIZE;
  394. *ppxlcDstLeft       = iLastValidSrcLeft1;
  395. *(ppxlcDstLeft + 1) = iLastValidSrcLeft2;
  396. }
  397. }
  398. // End Toshiba(1997-11-14)
  399. }
  401. Void CVideoObject::subsampleLeftTopBorderFromVOP (PixelC* ppxlcSrc, PixelC* ppxlcDst)
  402. {
  403. PixelC* ppxlcSrcTop1 = ppxlcSrc - BAB_BORDER  * m_iFrameWidthY - BAB_BORDER;
  404. PixelC* ppxlcSrcTop2 = ppxlcSrcTop1 + m_iFrameWidthY;
  405. PixelC* ppxlcSrcLft1 = ppxlcSrcTop1;
  406. PixelC* ppxlcSrcLft2 = ppxlcSrcLft1 + 1;
  407. PixelC* ppxlcDstTop1 = ppxlcDst;
  408. PixelC* ppxlcDstTop2 = ppxlcDst + m_iWidthCurrBAB;
  409. PixelC* ppxlcDstLft1 = ppxlcDst;
  410. PixelC* ppxlcDstLft2 = ppxlcDst + 1;
  411. Int iThresh = (m_iInverseCR == 2) ? 0 : opaqueValue;
  412. CoordI iPixel, iPixelSub, iSampleInterval;
  413. // Modified for error resilient mode by Toshiba(1997-11-14)
  414. for (iPixelSub = BAB_BORDER, iSampleInterval = BAB_BORDER; 
  415. iPixelSub < m_iWidthCurrBAB - BAB_BORDER; 
  416. iPixelSub++, iSampleInterval += m_iInverseCR) { //get each subsample
  417. Int iTempSumTop1 = 0, iTempSumTop2 = 0; 
  418. Int iTempSumLft1 = 0, iTempSumLft2 = 0;
  419. for (iPixel = 0; iPixel < m_iInverseCR; iPixel++) {
  420. iTempSumTop1 += ppxlcSrcTop1 [iSampleInterval + iPixel];
  421. iTempSumTop2 += ppxlcSrcTop2 [iSampleInterval + iPixel];
  422. iTempSumLft1 += ppxlcSrcLft1 [(iSampleInterval + iPixel) * m_iFrameWidthY];
  423. iTempSumLft2 += ppxlcSrcLft2 [(iSampleInterval + iPixel) * m_iFrameWidthY];
  424. }
  425. if (!m_bVPNoTop) {
  426. ppxlcDstTop1 [iPixelSub] = (iTempSumTop1 > iThresh) ? opaqueValue : transpValue;
  427. ppxlcDstTop2 [iPixelSub] = (iTempSumTop2 > iThresh) ? opaqueValue : transpValue;
  428. }
  429. else {
  430. ppxlcDstTop1 [iPixelSub] = 0;
  431. ppxlcDstTop2 [iPixelSub] = 0;
  432. }
  433. if (!m_bVPNoLeft) {
  434. ppxlcDstLft1 [iPixelSub * m_iWidthCurrBAB] = (iTempSumLft1 > iThresh) ? opaqueValue : transpValue;
  435. ppxlcDstLft2 [iPixelSub * m_iWidthCurrBAB] = (iTempSumLft2 > iThresh) ? opaqueValue : transpValue;
  436. }
  437. else {
  438. ppxlcDstLft1 [iPixelSub * m_iWidthCurrBAB] = 0;
  439. ppxlcDstLft2 [iPixelSub * m_iWidthCurrBAB] = 0;
  440. }
  441. }
  442. //boundary conditions 0, 1, 18, 19 
  443. CoordI iBorder;
  444. for (iBorder = 0; iBorder < BAB_BORDER; iBorder++) {
  445. if (!m_bVPNoLeftTop) {
  446. ppxlcDstTop1 [iBorder] = ppxlcSrcTop1 [iBorder];
  447. ppxlcDstTop2 [iBorder] = ppxlcSrcTop2 [iBorder];
  448. }
  449. else {
  450. ppxlcDstTop1 [iBorder] = 0;
  451. ppxlcDstTop2 [iBorder] = 0;
  452. }
  453. if (!m_bVPNoRightTop) {
  454. ppxlcDstTop1 [m_iWidthCurrBAB - 1 - iBorder] = ppxlcSrcTop1 [TOTAL_BAB_SIZE - 1 - iBorder];
  455. ppxlcDstTop2 [m_iWidthCurrBAB - 1 - iBorder] = ppxlcSrcTop2 [TOTAL_BAB_SIZE - 1 - iBorder];
  456. }
  457. else {
  458. ppxlcDstTop1 [m_iWidthCurrBAB - 1 - iBorder] = 0;
  459. ppxlcDstTop2 [m_iWidthCurrBAB - 1 - iBorder] = 0;
  460. }
  461. }
  462. // End Toshiba(1997-11-14)
  463. //repeat pad left-bottom corner
  464. ppxlcDstLft1 [(m_iWidthCurrBAB - 1) * m_iWidthCurrBAB] = 
  465. ppxlcDstLft1 [(m_iWidthCurrBAB - 2) * m_iWidthCurrBAB] =
  466. ppxlcDstLft1 [(m_iWidthCurrBAB - 3) * m_iWidthCurrBAB];
  467. ppxlcDstLft2 [(m_iWidthCurrBAB - 1) * m_iWidthCurrBAB] =
  468. ppxlcDstLft2 [(m_iWidthCurrBAB - 2) * m_iWidthCurrBAB] =
  469. ppxlcDstLft2 [(m_iWidthCurrBAB - 3) * m_iWidthCurrBAB];
  470. }
  472. Void CVideoObject::makeRightBottomBorder (PixelC* ppxlcSrc, Int iWidth)
  473. {
  474. Int i;
  475. PixelC* ppxlcDst = ppxlcSrc + BAB_BORDER * iWidth + iWidth - BAB_BORDER; //make right border
  476. for (i = 0; i < iWidth - 2*BAB_BORDER; i++) {
  477. *ppxlcDst = *(ppxlcDst - 1);
  478. *(ppxlcDst + 1) = *ppxlcDst;
  479. ppxlcDst += iWidth;
  480. }
  481. ppxlcDst = ppxlcSrc + (iWidth - BAB_BORDER) * iWidth; //make bottom border
  482. for (i = 0; i < iWidth; i++) {
  483. *ppxlcDst = *(ppxlcDst - iWidth);
  484. *(ppxlcDst + iWidth) = *ppxlcDst;
  485. ppxlcDst++;
  486. }
  487. }
  489. Int CVideoObject::contextIntra (const PixelC* ppxlcSrc)
  490. {
  491. Int iContext = 0, i;
  492. static Int rgiNeighbourIndx [10];
  493. rgiNeighbourIndx [0] = -1;
  494. rgiNeighbourIndx [1] = -2;
  495. rgiNeighbourIndx [2] = -m_iWidthCurrBAB + 2;
  496. rgiNeighbourIndx [3] = -m_iWidthCurrBAB + 1;
  497. rgiNeighbourIndx [4] = -m_iWidthCurrBAB;
  498. rgiNeighbourIndx [5] = -m_iWidthCurrBAB - 1;
  499. rgiNeighbourIndx [6] = -m_iWidthCurrBAB - 2;
  500. rgiNeighbourIndx [7] = -2 * m_iWidthCurrBAB + 1;
  501. rgiNeighbourIndx [8] = -2 * m_iWidthCurrBAB;
  502. rgiNeighbourIndx [9] = -2 * m_iWidthCurrBAB - 1;
  503. for (i = 0; i < 10; i++)
  504. iContext += (ppxlcSrc [rgiNeighbourIndx [i]] == MPEG4_OPAQUE) << i;
  505. assert (iContext >= 0 && iContext < 1024);
  506. return iContext;
  507. }
  509. Int CVideoObject::contextIntraTranspose (const PixelC* ppxlcSrc)
  510. {
  511. Int iContext = 0, i;
  512. static Int rgiNeighbourIndx [10];
  513. rgiNeighbourIndx [0] = -m_iWidthCurrBAB;
  514. rgiNeighbourIndx [1] = -2 * m_iWidthCurrBAB;
  515. rgiNeighbourIndx [2] = -1 + 2 * m_iWidthCurrBAB;
  516. rgiNeighbourIndx [3] = -1 + m_iWidthCurrBAB;
  517. rgiNeighbourIndx [4] = -1;
  518. rgiNeighbourIndx [5] = -m_iWidthCurrBAB - 1;
  519. rgiNeighbourIndx [6] = -1 - 2 * m_iWidthCurrBAB;
  520. rgiNeighbourIndx [7] = -2 + m_iWidthCurrBAB;
  521. rgiNeighbourIndx [8] = -2;
  522. rgiNeighbourIndx [9] = -2 - m_iWidthCurrBAB;
  523. for (i = 0; i < 10; i++)
  524. iContext += (ppxlcSrc [rgiNeighbourIndx [i]] == MPEG4_OPAQUE) << i;
  525. assert (iContext >= 0 && iContext < 1024);
  526. return iContext;
  527. }
  529. Int CVideoObject::contextInter (const PixelC* ppxlcSrc, const PixelC* ppxlcPred)
  530. {
  531. Int i, iContext = 0;
  532. static Int rgiNeighbourIndx [9];
  533. Int iSizePredBAB = m_iWidthCurrBAB - 2*BAB_BORDER + 2*MC_BAB_BORDER;
  535. rgiNeighbourIndx [0] = -1;
  536. rgiNeighbourIndx [1] = -m_iWidthCurrBAB + 1;
  537. rgiNeighbourIndx [2] = -m_iWidthCurrBAB;
  538. rgiNeighbourIndx [3] = -m_iWidthCurrBAB - 1;
  539. rgiNeighbourIndx [4] = iSizePredBAB;
  540. rgiNeighbourIndx [5] = 1;
  541. rgiNeighbourIndx [6] = 0;
  542. rgiNeighbourIndx [7] = -1;
  543. rgiNeighbourIndx [8] = -iSizePredBAB;
  545. for (i = 0; i < 4; i++) {
  546. iContext += (ppxlcSrc [rgiNeighbourIndx [i]] == MPEG4_OPAQUE) << i;
  547. }
  549. for (i = 4; i < 9; i++) {
  550. iContext += (ppxlcPred [rgiNeighbourIndx [i]] == MPEG4_OPAQUE) << i;
  551. }
  552. assert (iContext >= 0 && iContext < 1024);
  553. return iContext;
  554. }
  556. Int CVideoObject::contextInterTranspose (const PixelC* ppxlcSrc, const PixelC* ppxlcPred)
  557. {
  558. Int i, iContext = 0;
  559. static Int rgiNeighbourIndx [9];
  560. Int iSizePredBAB = m_iWidthCurrBAB - 2*BAB_BORDER + 2*MC_BAB_BORDER;
  562. rgiNeighbourIndx [0] = -m_iWidthCurrBAB;
  563. rgiNeighbourIndx [1] = m_iWidthCurrBAB - 1;
  564. rgiNeighbourIndx [2] = -1;
  565. rgiNeighbourIndx [3] = -m_iWidthCurrBAB - 1;
  566. rgiNeighbourIndx [4] = 1;
  567. rgiNeighbourIndx [5] = iSizePredBAB;
  568. rgiNeighbourIndx [6] = 0;
  569. rgiNeighbourIndx [7] = -iSizePredBAB;
  570. rgiNeighbourIndx [8] = -1;
  572. for (i = 0; i < 4; i++)
  573. iContext += (ppxlcSrc [rgiNeighbourIndx [i]] == MPEG4_OPAQUE) << i;
  575. for (i = 4; i < 9; i++)
  576. iContext += (ppxlcPred [rgiNeighbourIndx [i]] == MPEG4_OPAQUE) << i;
  577. assert (iContext >= 0 && iContext < 1024);
  578. return iContext;
  579. }
  581. Void CVideoObject::copyReconShapeToMbAndRef (
  582. PixelC* ppxlcDstMB, 
  583. PixelC* ppxlcRefFrm, PixelC pxlcSrc
  584. )
  585. {
  586. pxlcmemset (ppxlcDstMB, pxlcSrc, MB_SQUARE_SIZE); // Modified by swinder
  587. for (Int i = 0; i < MB_SIZE; i++) {
  588. pxlcmemset (ppxlcRefFrm, pxlcSrc, MB_SIZE);
  589. ppxlcRefFrm += m_iFrameWidthY;
  590. }
  591. }
  593. Void CVideoObject::copyReconShapeUVToRef (
  594. PixelC* ppxlcRefFrm, const PixelC* ppxlcSrc)
  595. {
  596. for (Int i = 0; i < BLOCK_SIZE; i++) {
  597. memcpy (ppxlcRefFrm, ppxlcSrc, BLOCK_SIZE*sizeof(PixelC));
  598. ppxlcRefFrm += m_iFrameWidthUV;
  599. ppxlcSrc += BLOCK_SIZE;
  600. }
  601. }
  603. Void CVideoObject::copyReconShapeToMbAndRef (
  604. PixelC* ppxlcDstMB, 
  605. PixelC* ppxlcRefFrm, 
  606. const PixelC* ppxlcSrc, Int iSrcWidth, Int iBorder
  607. )
  608. {
  609. Int iUnit = sizeof(PixelC); // NBIT: for memcpy
  610. ppxlcSrc += iSrcWidth * iBorder + iBorder;
  611. for (Int i = 0; i < MB_SIZE; i++) {
  612. memcpy (ppxlcDstMB, ppxlcSrc, MB_SIZE*iUnit);
  613. memcpy (ppxlcRefFrm, ppxlcSrc, MB_SIZE*iUnit);
  614. ppxlcRefFrm += m_iFrameWidthY;
  615. ppxlcDstMB  += MB_SIZE;
  616. ppxlcSrc += iSrcWidth;
  617. }
  618. }
  620. CMotionVector CVideoObject::findShapeMVP (
  621. const CMotionVector* pmv, const CMotionVector* pmvBY, 
  622. const CMBMode* pmbmd,
  623. Int iMBX, Int iMBY
  624. ) const
  625. {
  626. CMotionVector mvRet;
  627. // try shape vector first
  629. Bool bLeftBndry, bRightBndry, bTopBndry;
  630. Int iMBnum = VPMBnum(iMBX, iMBY);
  631. bLeftBndry = bVPNoLeft(iMBnum, iMBX);
  632. bTopBndry = bVPNoTop(iMBnum);
  633. bRightBndry = bVPNoRightTop(iMBnum, iMBX);
  635. mvRet = *(pmvBY - 1);
  636. // Modified for error resilient mode by Toshiba(1997-11-14)
  637. // if ((iMBX > 0) && ((pmbmd - 1)->m_shpmd == INTER_CAE_MVDZ ||
  638. if (!bLeftBndry && ((pmbmd - 1)->m_shpmd == INTER_CAE_MVDZ ||
  639. (pmbmd - 1)->m_shpmd == INTER_CAE_MVDNZ ||
  640. (pmbmd - 1)->m_shpmd == MVDZ_NOUPDT ||
  641. (pmbmd - 1)->m_shpmd == MVDNZ_NOUPDT))
  642. return mvRet;
  643. if (iMBY > 0)   {
  644. mvRet = *(pmvBY - m_iNumMBX);
  645. // Modified for error resilient mode by Toshiba(1997-11-14)
  646. // if ((pmbmd - m_iNumMBX)->m_shpmd == INTER_CAE_MVDZ ||
  647. if (!bTopBndry && ((pmbmd - m_iNumMBX)->m_shpmd == INTER_CAE_MVDZ ||
  648. (pmbmd - m_iNumMBX)->m_shpmd == INTER_CAE_MVDNZ ||
  649. (pmbmd - m_iNumMBX)->m_shpmd == MVDZ_NOUPDT ||
  650. (pmbmd - m_iNumMBX)->m_shpmd == MVDNZ_NOUPDT
  651. // Modified for error resilient mode by Toshiba(1997-11-14)
  652. // )
  653. ))
  654. return mvRet;
  655. mvRet = *(pmvBY - m_iNumMBX + 1);
  656. // Modified for error resilient mode by Toshiba(1997-11-14)
  657. // if ((iMBX < m_iNumMBX - 1) && ((pmbmd - m_iNumMBX + 1)->m_shpmd == INTER_CAE_MVDZ ||
  658. if (!bRightBndry && ((pmbmd - m_iNumMBX + 1)->m_shpmd == INTER_CAE_MVDZ ||
  659. (pmbmd - m_iNumMBX + 1)->m_shpmd == INTER_CAE_MVDNZ ||
  660. (pmbmd - m_iNumMBX + 1)->m_shpmd == MVDZ_NOUPDT ||
  661. (pmbmd - m_iNumMBX + 1)->m_shpmd == MVDNZ_NOUPDT)
  662. )
  663. return mvRet;
  664. }
  666. // try texture vector then; truncate half pel
  667. if(m_volmd.bShapeOnly==FALSE && m_vopmd.vopPredType==PVOP)
  668. {
  669. // Modified for error resilient mode by Toshiba(1997-11-14)
  670. // if (iMBX != 0 && validBlock (pmbmd - 1, gIndexOfCandBlk [1] [0]))
  671. if (!bLeftBndry && validBlock (pmbmd, pmbmd - 1, gIndexOfCandBlk [1] [0]))
  672. return CMotionVector ((pmv - PVOP_MV_PER_REF_PER_MB + gIndexOfCandBlk [1] [0])->m_vctTrueHalfPel.x / 2,
  673.   (pmv - PVOP_MV_PER_REF_PER_MB + gIndexOfCandBlk [1] [0])->m_vctTrueHalfPel.y / 2);
  674. if (iMBY != 0)   {
  675. // Modified for error resilient mode by Toshiba(1997-11-14)
  676. // if (validBlock (pmbmd - m_iNumMBX, gIndexOfCandBlk [1] [1]))
  677. if (!bTopBndry && validBlock (pmbmd, pmbmd - m_iNumMBX, gIndexOfCandBlk [1] [1]))
  678. return CMotionVector ((pmv - m_iNumOfTotalMVPerRow + gIndexOfCandBlk [1] [1])->m_vctTrueHalfPel.x / 2,
  679.   (pmv - m_iNumOfTotalMVPerRow + gIndexOfCandBlk [1] [1])->m_vctTrueHalfPel.y / 2);
  680. // Modified for error resilient mode by Toshiba(1997-11-14)
  681. // if (iMBX < m_iNumMBX - 1 && validBlock (pmbmd - m_iNumMBX + 1, gIndexOfCandBlk [1] [2]))
  682. if (!bRightBndry && validBlock (pmbmd, pmbmd - m_iNumMBX + 1, gIndexOfCandBlk [1] [2]))
  683. return CMotionVector ((pmv - m_iNumOfTotalMVPerRow + PVOP_MV_PER_REF_PER_MB + gIndexOfCandBlk [1] [2])->m_vctTrueHalfPel.x / 2,
  684.   (pmv - m_iNumOfTotalMVPerRow + PVOP_MV_PER_REF_PER_MB + gIndexOfCandBlk [1] [2])->m_vctTrueHalfPel.y / 2);
  685. }
  686. }
  687. return CMotionVector ();
  688. }