Added ippiPyrUp_Gauss5x5 to cv::pyrUp
authorAlexander Karsakov <alexander.karsakov@itseez.com>
Wed, 16 Apr 2014 07:59:32 +0000 (11:59 +0400)
committerAlexander Karsakov <alexander.karsakov@itseez.com>
Wed, 16 Apr 2014 07:59:32 +0000 (11:59 +0400)
modules/imgproc/src/pyramids.cpp

index e6ce7bc..3f51908 100644 (file)
@@ -509,31 +509,36 @@ void cv::pyrDown( InputArray _src, OutputArray _dst, const Size& _dsz, int borde
 #endif
 
 #if (defined(HAVE_IPP) && !defined(HAVE_IPP_ICV_ONLY) && IPP_VERSION_X100 >= 801)
-    typedef IppStatus (CV_STDCALL * ippiPyrDown)(const void* pSrc, int srcStep, void* pDst, int dstStep, IppiSize srcRoi, Ipp8u* buffer);
-    int type = src.type();
-    CV_SUPPRESS_DEPRECATED_START
-    ippiPyrDown pyrDownFunc = type == CV_8UC1 ? (ippiPyrDown) ippiPyrDown_Gauss5x5_8u_C1R :
-                              type == CV_8UC3 ? (ippiPyrDown) ippiPyrDown_Gauss5x5_8u_C3R :
-                              type == CV_32FC1 ? (ippiPyrDown) ippiPyrDown_Gauss5x5_32f_C1R :
-                              type == CV_32FC3 ? (ippiPyrDown) ippiPyrDown_Gauss5x5_32f_C3R : 0;
-    CV_SUPPRESS_DEPRECATED_END
-
-    if (pyrDownFunc)
+    bool isolated = (borderType & BORDER_ISOLATED) != 0;
+    int borderTypeNI = borderType & ~BORDER_ISOLATED;
+    if (borderTypeNI == BORDER_DEFAULT && (!src.isSubmatrix() || isolated))
     {
-        int bufferSize;
-        IppiSize srcRoi = { src.cols, src.rows };
-        IppDataType dataType = depth == CV_8U ? ipp8u : ipp32f;
+        typedef IppStatus (CV_STDCALL * ippiPyrDown)(const void* pSrc, int srcStep, void* pDst, int dstStep, IppiSize srcRoi, Ipp8u* buffer);
+        int type = src.type();
         CV_SUPPRESS_DEPRECATED_START
-        IppStatus ok = ippiPyrDownGetBufSize_Gauss5x5(srcRoi.width, dataType, src.channels(), &bufferSize);
+        ippiPyrDown pyrDownFunc = type == CV_8UC1 ? (ippiPyrDown) ippiPyrDown_Gauss5x5_8u_C1R :
+                                  type == CV_8UC3 ? (ippiPyrDown) ippiPyrDown_Gauss5x5_8u_C3R :
+                                  type == CV_32FC1 ? (ippiPyrDown) ippiPyrDown_Gauss5x5_32f_C1R :
+                                  type == CV_32FC3 ? (ippiPyrDown) ippiPyrDown_Gauss5x5_32f_C3R : 0;
         CV_SUPPRESS_DEPRECATED_END
-        if (ok >= 0)
-        {
-            Ipp8u* buffer = ippsMalloc_8u(bufferSize);
-            ok = pyrDownFunc(src.data, (int) src.step, dst.data, (int) dst.step, srcRoi, buffer);
-            ippsFree(buffer);
 
+        if (pyrDownFunc)
+        {
+            int bufferSize;
+            IppiSize srcRoi = { src.cols, src.rows };
+            IppDataType dataType = depth == CV_8U ? ipp8u : ipp32f;
+            CV_SUPPRESS_DEPRECATED_START
+            IppStatus ok = ippiPyrDownGetBufSize_Gauss5x5(srcRoi.width, dataType, src.channels(), &bufferSize);
+            CV_SUPPRESS_DEPRECATED_END
             if (ok >= 0)
-                return;
+            {
+                Ipp8u* buffer = ippsMalloc_8u(bufferSize);
+                ok = pyrDownFunc(src.data, (int) src.step, dst.data, (int) dst.step, srcRoi, buffer);
+                ippsFree(buffer);
+
+                if (ok >= 0)
+                    return;
+            }
         }
     }
 #endif
@@ -564,13 +569,48 @@ void cv::pyrUp( InputArray _src, OutputArray _dst, const Size& _dsz, int borderT
     Size dsz = _dsz.area() == 0 ? Size(src.cols*2, src.rows*2) : _dsz;
     _dst.create( dsz, src.type() );
     Mat dst = _dst.getMat();
+    int depth = src.depth();
 
 #ifdef HAVE_TEGRA_OPTIMIZATION
     if(borderType == BORDER_DEFAULT && tegra::pyrUp(src, dst))
         return;
 #endif
 
-    int depth = src.depth();
+#if (defined(HAVE_IPP) && !defined(HAVE_IPP_ICV_ONLY) && IPP_VERSION_X100 >= 801)
+    bool isolated = (borderType & BORDER_ISOLATED) != 0;
+    int borderTypeNI = borderType & ~BORDER_ISOLATED;
+    if (borderTypeNI == BORDER_DEFAULT && (!src.isSubmatrix() || isolated))
+    {
+        typedef IppStatus (CV_STDCALL * ippiPyrUp)(const void* pSrc, int srcStep, void* pDst, int dstStep, IppiSize srcRoi, Ipp8u* buffer);
+        int type = src.type();
+        CV_SUPPRESS_DEPRECATED_START
+        ippiPyrUp pyrUpFunc = type == CV_8UC1 ? (ippiPyrUp) ippiPyrUp_Gauss5x5_8u_C1R :
+                                  type == CV_8UC3 ? (ippiPyrUp) ippiPyrUp_Gauss5x5_8u_C3R :
+                                  type == CV_32FC1 ? (ippiPyrUp) ippiPyrUp_Gauss5x5_32f_C1R :
+                                  type == CV_32FC3 ? (ippiPyrUp) ippiPyrUp_Gauss5x5_32f_C3R : 0;
+        CV_SUPPRESS_DEPRECATED_END
+
+        if (pyrUpFunc)
+        {
+            int bufferSize;
+            IppiSize srcRoi = { src.cols, src.rows };
+            IppDataType dataType = depth == CV_8U ? ipp8u : ipp32f;
+            CV_SUPPRESS_DEPRECATED_START
+            IppStatus ok = ippiPyrUpGetBufSize_Gauss5x5(srcRoi.width, dataType, src.channels(), &bufferSize);
+            CV_SUPPRESS_DEPRECATED_END
+            if (ok >= 0)
+            {
+                Ipp8u* buffer = ippsMalloc_8u(bufferSize);
+                ok = pyrUpFunc(src.data, (int) src.step, dst.data, (int) dst.step, srcRoi, buffer);
+                ippsFree(buffer);
+
+                if (ok >= 0)
+                    return;
+            }
+        }
+    }
+#endif
+
     PyrFunc func = 0;
     if( depth == CV_8U )
         func = pyrUp_<FixPtCast<uchar, 6>, NoVec<int, uchar> >;