Changeset 583 in 3DVCSoftware

Timestamp:

17 Aug 2013, 15:12:24 (11 years ago)

Author:

tech

Message:

Clean ups related to SCU_HS_FAST_DEPTH_INTRA_E0238

Location:

branches/HTM-DEV-2.0-dev3-SCU-Hisilicon/source/Lib

Files:

• TLibCommon/TComPrediction.cpp (modified) (1 diff)
• TLibCommon/TComRdCost.cpp (modified) (1 diff)
• TLibCommon/TComRdCost.h (modified) (1 diff)
• TLibEncoder/TEncSearch.cpp (modified) (3 diffs)

branches/HTM-DEV-2.0-dev3-SCU-Hisilicon/source/Lib/TLibCommon/TComPrediction.cpp

@@ -2142 +2142 @@
 
 #if SCU_HS_DMM4_REMOVE_DIV_E0242
-Int cuMaxLog2Size = g_aucConvertToBit[g_uiMaxCUWidth]+2;
-iDC = iDC >> (cuMaxLog2Size - pcCU->getDepth(0))*2;
+  Int cuMaxLog2Size = g_aucConvertToBit[g_uiMaxCUWidth]+2;
+  iDC = iDC >> (cuMaxLog2Size - pcCU->getDepth(0))*2;
 #else
   iDC /= (uiWidth*uiHeight);

branches/HTM-DEV-2.0-dev3-SCU-Hisilicon/source/Lib/TLibCommon/TComRdCost.cpp

@@ -486 +486 @@
 #if SCU_HS_FAST_DEPTH_INTRA_E0238
 
-UInt TComRdCost::calcVAR (Pel* pi0, Int uiWidth, Int uiHeight, Int cuDepth)
+UInt TComRdCost::calcVAR (Pel* pi0, Int width, Int height, Int stride, Int cuDepth)
 { 
-  Int x, y;
   Int temp = 0;
 
-  for (x = 0; x <  uiHeight; x++)
-  {
-     for (y = 0; y < uiWidth; y++)
-      {
-         temp += pi0[x*uiHeight+y]; 
-      }
-  }
+  for (Int y = 0; y < height; y++)
+  {
+    for (Int x = 0; x < width; x++)
+    {
+      temp += pi0[ y * stride + x ]; 
+    }
+  }
+
   Int cuMaxLog2Size = g_aucConvertToBit[g_uiMaxCUWidth]+2;
-  if (uiWidth == 4)
-  {
-     cuDepth = cuMaxLog2Size-2;
-  }
-
-  temp = temp >> (cuMaxLog2Size-cuDepth)*2;
-
-  UInt uiSum = 0;
-    for (x = 0; x <  uiHeight; x++)
-  {
-      for (y = 0; y < uiWidth; y++)
-      {
-        uiSum += (pi0[x*uiHeight+y]-temp)*(pi0[x*uiHeight+y]-temp);
-      }
-  }
-     return (uiSum >> (cuMaxLog2Size-cuDepth)*2);
+  
+  if ( width == 4 ) // GT: What is the rational behind this? 
+  {
+    cuDepth = cuMaxLog2Size - 2;
+  }
+
+  temp = temp >> (cuMaxLog2Size-cuDepth) * 2;
+
+  UInt sum = 0;
+  for (Int y = 0; y < height; y++)
+  {
+    for (Int x = 0; x <  width; x++)
+    {
+      sum += (pi0[ y * stride + x ] - temp ) * (pi0[ y * stride + x ] - temp );
+    }
+  }
+  return (sum >> (cuMaxLog2Size-cuDepth)*2);
 
 }

branches/HTM-DEV-2.0-dev3-SCU-Hisilicon/source/Lib/TLibCommon/TComRdCost.h

@@ -229 +229 @@
   UInt    calcHAD(Int bitDepth, Pel* pi0, Int iStride0, Pel* pi1, Int iStride1, Int iWidth, Int iHeight );
 #if SCU_HS_FAST_DEPTH_INTRA_E0238
-  UInt    calcVAR(Pel* pi0, Int uiWidth, Int uiHeight, Int cuDepth);
+  UInt    calcVAR(Pel* pi0, Int width, Int height, Int stride, Int cuDepth);
 #endif  
   // for motion cost

branches/HTM-DEV-2.0-dev3-SCU-Hisilicon/source/Lib/TLibEncoder/TEncSearch.cpp

@@ -2835 +2835 @@
 
 #if SCU_HS_FAST_DEPTH_INTRA_E0238
-Int threshold = max(((pcCU->getQP(0))>>3)-1,3);
-Int varThreshold = (Int)(threshold*threshold - 8);
-UInt uiVarCU=m_pcRdCost->calcVAR(piOrg, uiWidth,uiHeight,pcCU->getDepth(0));
+      Int  threshold    = max(((pcCU->getQP(0))>>3)-1,3);
+      Int  varThreshold = (Int)( threshold * threshold - 8 );
+      UInt varCU      = m_pcRdCost->calcVAR(piOrg, uiStride, uiWidth,uiHeight,pcCU->getDepth(0));
 #endif
 
@@ -2844 +2844 @@
       if( m_pcEncCfg->getUseDMM()
 #if SCU_HS_FAST_DEPTH_INTRA_E0238
-         && (uiRdModeList[0] != 0 || uiVarCU >= varThreshold)
+         && (uiRdModeList[0] != 0 || varCU >= varThreshold)
 #endif
         )
@@ -2922 +2922 @@
       if( m_pcEncCfg->getUseRBC()
 #if SCU_HS_FAST_DEPTH_INTRA_E0238
-          && (uiRdModeList[0] != 0 || uiVarCU >= varThreshold)
+          && (uiRdModeList[0] != 0 || varCU >= varThreshold)
 #endif
         )