Changeset 56 in 3DVCSoftware for trunk/source/Lib/TLibCommon/TComWedgelet.h

Timestamp:

11 May 2012, 21:20:17 (13 years ago)

Author:

hschwarz

Message:

updated trunk (move to HM6.1)

File:

• trunk/source/Lib/TLibCommon/TComWedgelet.h (modified) (5 diffs)

trunk/source/Lib/TLibCommon/TComWedgelet.h

@@ -43 +43 @@
 #include <vector>
 
-
 enum WedgeResolution
 {
@@ -49 +48 @@
   FULL_PEL,
   HALF_PEL
-};
-
-enum WedgeDist
-{
-  WedgeDist_SAD  = 0,
-  WedgeDist_SSE  = 4,
 };
 
@@ -101 +94 @@
 
   Bool  checkNotPlain();
-  Bool  checkNotIdentical( Bool* pbRefPattern );
-  Bool  checkNotInvIdentical( Bool* pbRefPattern );
-
+  Bool  checkIdentical( Bool* pbRefPattern );
+  Bool  checkInvIdentical( Bool* pbRefPattern );
+
+#if HHI_DMM_WEDGE_INTRA
   Bool  checkPredDirAbovePossible( UInt uiPredDirBlockSize, UInt uiPredDirBlockOffsett );
   Bool  checkPredDirLeftPossible ( UInt uiPredDirBlockSize, UInt uiPredDirBlockOffsett );
@@ -109 +103 @@
   Void  getPredDirStartEndAbove( UChar& ruhXs, UChar& ruhYs, UChar& ruhXe, UChar& ruhYe, UInt uiPredDirBlockSize, UInt uiPredDirBlockOffset, Int iDeltaEnd );
   Void  getPredDirStartEndLeft ( UChar& ruhXs, UChar& ruhYs, UChar& ruhXe, UChar& ruhYe, UInt uiPredDirBlockSize, UInt uiPredDirBlockOffset, Int iDeltaEnd );
+#endif
 };  // END CLASS DEFINITION TComWedgelet
 
@@ -142 +137 @@
 typedef std::vector<TComWedgeRef> WedgeRefList;
 
+#if HHI_DMM_PRED_TEX
+enum WedgeDist
+{
+  WedgeDist_SAD  = 0,
+  WedgeDist_SSE  = 4,
+};
+
+class WedgeDistParam;
+typedef UInt (*FpWedgeDistFunc) (WedgeDistParam*);
+
+/// distortion parameter class
+class WedgeDistParam
+{
+public:
+  Pel*  pOrg;
+  Pel*  pCur;
+  Int   iStrideOrg;
+  Int   iStrideCur;
+  Int   iRows;
+  Int   iCols;
+  Int   iStep;
+  FpWedgeDistFunc DistFunc;
+  Int   iSubShift;
+
+  WedgeDistParam()
+  {
+    pOrg = NULL;
+    pCur = NULL;
+    iStrideOrg = 0;
+    iStrideCur = 0;
+    iRows = 0;
+    iCols = 0;
+    iStep = 1;
+    DistFunc = NULL;
+    iSubShift = 0;
+  }
+};
+
+// ====================================================================================================================
+// Class definition TComWedgeDist
+// ====================================================================================================================
+class TComWedgeDist
+{
+private:
+  Int                     m_iBlkWidth;
+  Int                     m_iBlkHeight;
+  FpWedgeDistFunc         m_afpDistortFunc[8];
+
+public:
+  TComWedgeDist();
+  virtual ~TComWedgeDist();
+
+  Void init();
+  Void setDistParam( UInt uiBlkWidth, UInt uiBlkHeight, WedgeDist eWDist, WedgeDistParam& rcDistParam );
+  UInt getDistPart( Pel* piCur, Int iCurStride,  Pel* piOrg, Int iOrgStride, UInt uiBlkWidth, UInt uiBlkHeight, WedgeDist eWDist = WedgeDist_SAD );
+
+private:
+  static UInt xGetSAD4          ( WedgeDistParam* pcDtParam );
+  static UInt xGetSAD8          ( WedgeDistParam* pcDtParam );
+  static UInt xGetSAD16         ( WedgeDistParam* pcDtParam );
+  static UInt xGetSAD32         ( WedgeDistParam* pcDtParam );
+  //static UInt xGetSAD64         ( WedgeDistParam* pcDtParam );
+
+  static UInt xGetSSE4          ( WedgeDistParam* pcDtParam );
+  static UInt xGetSSE8          ( WedgeDistParam* pcDtParam );
+  static UInt xGetSSE16         ( WedgeDistParam* pcDtParam );
+  static UInt xGetSSE32         ( WedgeDistParam* pcDtParam );
+  //static UInt xGetSSE64         ( WedgeDistParam* pcDtParam );
+
+};// END CLASS DEFINITION TComWedgeDist
+#endif
+
 // ====================================================================================================================
 // Function definition roftoi (mathematically correct rounding of float to int)