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

流媒体/Mpeg4/MP4

开发平台:

Visual C++

  1. /*************************************************************************
  2. This software module was originally developed by 
  3. Ming-Chieh Lee (mingcl@microsoft.com), Microsoft Corporation
  4. (date: June, 1997)
  5. and edited by
  6.         Wei Wu (weiwu@stallion.risc.rockwell.com) Rockwell Science Center
  7. in the course of development of the MPEG-4 Video (ISO/IEC 14496-2). 
  8. This software module is an implementation of a part of one or more MPEG-4 Video tools 
  9. as specified by the MPEG-4 Video. 
  10. ISO/IEC gives users of the MPEG-4 Video free license to this software module or modifications 
  11. thereof for use in hardware or software products claiming conformance to the MPEG-4 Video. 
  12. Those intending to use this software module in hardware or software products are advised that its use may infringe existing patents. 
  13. The original developer of this software module and his/her company, 
  14. the subsequent editors and their companies, 
  15. and ISO/IEC have no liability for use of this software module or modifications thereof in an implementation. 
  16. Copyright is not released for non MPEG-4 Video conforming products. 
  17. Microsoft retains full right to use the code for his/her own purpose, 
  18. assign or donate the code to a third party and to inhibit third parties from using the code for non MPEG-4 Video conforming products. 
  19. This copyright notice must be included in all copies or derivative works. 
  20. Copyright (c) 1996, 1997.
  21. Module Name:
  22. padEnc.cpp
  23. Abstract:
  24. MB Padding for Intra-texture coding.
  25. Revision History:
  26. *************************************************************************/
  27. // NOTE: 
  28. // LPE padding is working on the current MB whose size is always a MB
  29. // MC padding is  working on the (reference) quantized VOP
  30. #include <stdio.h>
  31. #include <math.h>
  32. #include <stdlib.h>
  33. #include <iostream.h>
  34. #include "typeapi.h"
  35. #include "codehead.h"
  36. #include "global.hpp"
  37. #include "entropy/bitstrm.hpp"
  38. #include "entropy/entropy.hpp"
  39. #include "entropy/huffman.hpp"
  40. #include "mode.hpp"
  41. #include "vopses.hpp"
  42. #include "vopseenc.hpp"
  43. Void CVideoObjectEncoder::LPEPadding (const CMBMode* pmbmd)
  44. {
  45. const PixelC* ppxlcBlkB;
  46. PixelC* ppxlcBlk;
  47. if (pmbmd -> m_rgTranspStatus [1] == PARTIAL) {
  48. ppxlcBlkB = m_ppxlcCurrMBBY;
  49. ppxlcBlk = m_ppxlcCurrMBY;
  50. LPEPaddingBlk (ppxlcBlk, ppxlcBlkB, MB_SIZE);
  51. }
  52. if (pmbmd -> m_rgTranspStatus [2] == PARTIAL) {
  53. ppxlcBlkB = m_ppxlcCurrMBBY + BLOCK_SIZE;
  54. ppxlcBlk = m_ppxlcCurrMBY + BLOCK_SIZE;
  55. LPEPaddingBlk (ppxlcBlk, ppxlcBlkB, MB_SIZE);
  56. }
  57. if (pmbmd -> m_rgTranspStatus [3] == PARTIAL) {
  58. ppxlcBlkB = m_ppxlcCurrMBBY + MB_SIZE * BLOCK_SIZE;
  59. ppxlcBlk = m_ppxlcCurrMBY + MB_SIZE * BLOCK_SIZE;
  60. LPEPaddingBlk (ppxlcBlk, ppxlcBlkB, MB_SIZE);
  61. }
  62. if (pmbmd -> m_rgTranspStatus [4] == PARTIAL) {
  63. ppxlcBlkB = m_ppxlcCurrMBBY + MB_SIZE * BLOCK_SIZE + BLOCK_SIZE;
  64. ppxlcBlk = m_ppxlcCurrMBY + MB_SIZE * BLOCK_SIZE + BLOCK_SIZE;
  65. LPEPaddingBlk (ppxlcBlk, ppxlcBlkB, MB_SIZE);
  66. }
  67. if (pmbmd -> m_rgTranspStatus [5] == PARTIAL) {
  68. LPEPaddingBlk (m_ppxlcCurrMBU, m_ppxlcCurrMBBUV, BLOCK_SIZE);
  69. LPEPaddingBlk (m_ppxlcCurrMBV, m_ppxlcCurrMBBUV, BLOCK_SIZE);
  70. }
  71. if (m_volmd.fAUsage == EIGHT_BIT) {
  72. if (pmbmd -> m_rgTranspStatus [1] == PARTIAL) {
  73. ppxlcBlkB = m_ppxlcCurrMBBY;
  74. ppxlcBlk = m_ppxlcCurrMBA;
  75. LPEPaddingBlk (ppxlcBlk, ppxlcBlkB, MB_SIZE);
  76. }
  77. if (pmbmd -> m_rgTranspStatus [2] == PARTIAL) {
  78. ppxlcBlkB = m_ppxlcCurrMBBY + BLOCK_SIZE;
  79. ppxlcBlk = m_ppxlcCurrMBA + BLOCK_SIZE;
  80. LPEPaddingBlk (ppxlcBlk, ppxlcBlkB, MB_SIZE);
  81. }
  82. if (pmbmd -> m_rgTranspStatus [3] == PARTIAL) {
  83. ppxlcBlkB = m_ppxlcCurrMBBY + MB_SIZE * BLOCK_SIZE;
  84. ppxlcBlk = m_ppxlcCurrMBA + MB_SIZE * BLOCK_SIZE;
  85. LPEPaddingBlk (ppxlcBlk, ppxlcBlkB, MB_SIZE);
  86. }
  87. if (pmbmd -> m_rgTranspStatus [4] == PARTIAL) {
  88. ppxlcBlkB = m_ppxlcCurrMBBY + MB_SIZE * BLOCK_SIZE + BLOCK_SIZE;
  89. ppxlcBlk = m_ppxlcCurrMBA + MB_SIZE * BLOCK_SIZE + BLOCK_SIZE;
  90. LPEPaddingBlk (ppxlcBlk, ppxlcBlkB, MB_SIZE);
  91. }
  92. }
  93. }
  94. #define OUTSIDE_BLOCK_VALUE 1
  95. CU8Image uciBuff (CRct (0, 0, BLOCK_SIZE + 2, BLOCK_SIZE + 2), OUTSIDE_BLOCK_VALUE);
  96. PixelC* ppxlcBuff = (PixelC*) uciBuff.pixels () + BLOCK_SIZE + 3;
  97. #define BLOCK_SIZE_PLUS2 10
  98. Void CVideoObjectEncoder::LPEPaddingBlk (
  99. PixelC* ppxlcBlk, const PixelC* ppxlcBlkB,
  100. UInt uiSize)
  101. {
  102. Int iUnit = sizeof(PixelC); // NBIT: for memcpy
  103. UInt uiNumNonTranspPixels = 0;
  104. UInt ix, iy;
  105. UInt uiSum = 0;
  106. PixelC* ppxlcBuffTmp = ppxlcBuff;
  107. PixelC* ppxlcBlkTmp = ppxlcBlk;
  108. const PixelC* ppxlcBlkBTmp = ppxlcBlkB;
  109. for (iy = 0; iy < BLOCK_SIZE; iy++) {
  110. memcpy (ppxlcBuffTmp, ppxlcBlkBTmp, BLOCK_SIZE*iUnit);
  111. for (ix = 0; ix < BLOCK_SIZE; ix++) {
  112. if (ppxlcBlkBTmp [ix] != transpValue) {
  113. uiSum += ppxlcBlkTmp [ix];
  114. uiNumNonTranspPixels++;
  115. }
  116. }
  117. ppxlcBlkBTmp += uiSize;
  118. ppxlcBlkTmp += uiSize;
  119. ppxlcBuffTmp += BLOCK_SIZE_PLUS2;
  120. }
  121. assert (uiNumNonTranspPixels != 0);
  122. uiSum /= uiNumNonTranspPixels;
  123. UInt uiSumNeighbor, uiNumNonTranpNeighbor;
  124. ppxlcBuffTmp = ppxlcBuff;
  125. for (iy = 0; iy < BLOCK_SIZE; iy++) {
  126. for (ix = 0; ix < BLOCK_SIZE; ix++) {
  127. if (ppxlcBlkB [ix] == transpValue) {
  128. uiSumNeighbor = uiNumNonTranpNeighbor = 0;
  129. if (ppxlcBuffTmp [ix - 1] != OUTSIDE_BLOCK_VALUE) {
  130. uiNumNonTranpNeighbor++;
  131. uiSumNeighbor += (ppxlcBuffTmp [ix - 1] == transpValue) ? uiSum : ppxlcBlk [ix - 1];
  132. }
  133. if (ppxlcBuffTmp [ix + 1] != OUTSIDE_BLOCK_VALUE) {
  134. uiNumNonTranpNeighbor++;
  135. uiSumNeighbor += (ppxlcBuffTmp [ix + 1] == transpValue) ? uiSum : ppxlcBlk [ix + 1];
  136. }
  137. if (ppxlcBuffTmp [ix - BLOCK_SIZE_PLUS2] != OUTSIDE_BLOCK_VALUE) {
  138. uiNumNonTranpNeighbor++;
  139. uiSumNeighbor += (ppxlcBuffTmp [ix - BLOCK_SIZE_PLUS2] == transpValue) ? uiSum : ppxlcBlk [ix - uiSize];
  140. }
  141. if (ppxlcBuffTmp [ix + BLOCK_SIZE_PLUS2] != OUTSIDE_BLOCK_VALUE) {
  142. uiNumNonTranpNeighbor++;
  143. uiSumNeighbor += (ppxlcBuffTmp [ix + BLOCK_SIZE_PLUS2] == transpValue) ? uiSum : ppxlcBlk [ix + uiSize];
  144. }
  145. ppxlcBlk [ix] = (PixelC) (uiSumNeighbor / uiNumNonTranpNeighbor);
  146. }
  147. }
  148. ppxlcBlkB += uiSize;
  149. ppxlcBlk += uiSize;
  150. ppxlcBuffTmp += BLOCK_SIZE_PLUS2;
  151. }
  152. }