restored 8u->32f cvGetQuadrangleSubPix() mode (need better test for it)

modules/imgproc/src/samplers.cpp

@@ -267,6 +267,99 @@ static void getRectSubPix_8u32f
     }
 }
 
+
+static void
+getQuadrangleSubPix_8u32f_CnR( const uchar* src, int src_step, Size src_size,
+                               float* dst, int dst_step, Size win_size,
+                               const double *matrix, int cn )
+{
+    int x, y, k;
+    double A11 = matrix[0], A12 = matrix[1], A13 = matrix[2];
+    double A21 = matrix[3], A22 = matrix[4], A23 = matrix[5];
+
+    src_step /= sizeof(src[0]);
+    dst_step /= sizeof(dst[0]);
+
+    for( y = 0; y < win_size.height; y++, dst += dst_step )
+    {
+        double xs = A12*y + A13;
+        double ys = A22*y + A23;
+        double xe = A11*(win_size.width-1) + A12*y + A13;
+        double ye = A21*(win_size.width-1) + A22*y + A23;
+
+        if( (unsigned)(cvFloor(xs)-1) < (unsigned)(src_size.width - 3) &&
+            (unsigned)(cvFloor(ys)-1) < (unsigned)(src_size.height - 3) &&
+            (unsigned)(cvFloor(xe)-1) < (unsigned)(src_size.width - 3) &&
+            (unsigned)(cvFloor(ye)-1) < (unsigned)(src_size.height - 3))
+        {
+            for( x = 0; x < win_size.width; x++ )
+            {
+                int ixs = cvFloor( xs );
+                int iys = cvFloor( ys );
+                const uchar *ptr = src + src_step*iys;
+                float a = (float)(xs - ixs), b = (float)(ys - iys), a1 = 1.f - a, b1 = 1.f - b;
+                float w00 = a1*b1, w01 = a*b1, w10 = a1*b, w11 = a*b;
+                xs += A11;
+                ys += A21;
+
+                if( cn == 1 )
+                {
+                    ptr += ixs;
+                    dst[x] = ptr[0]*w00 + ptr[1]*w01 + ptr[src_step]*w10 + ptr[src_step+1]*w11;
+                }
+                else if( cn == 3 )
+                {
+                    ptr += ixs*3;
+                    float t0 = ptr[0]*w00 + ptr[3]*w01 + ptr[src_step]*w10 + ptr[src_step+3]*w11;
+                    float t1 = ptr[1]*w00 + ptr[4]*w01 + ptr[src_step+1]*w10 + ptr[src_step+4]*w11;
+                    float t2 = ptr[2]*w00 + ptr[5]*w01 + ptr[src_step+2]*w10 + ptr[src_step+5]*w11;
+
+                    dst[x*3] = t0;
+                    dst[x*3+1] = t1;
+                    dst[x*3+2] = t2;
+                }
+                else
+                {
+                    ptr += ixs*cn;
+                    for( k = 0; k < cn; k++ )
+                        dst[x*cn+k] = ptr[k]*w00 + ptr[k+cn]*w01 +
+                                    ptr[src_step+k]*w10 + ptr[src_step+k+cn]*w11;
+                }
+            }
+        }
+        else
+        {
+            for( x = 0; x < win_size.width; x++ )
+            {
+                int ixs = cvFloor( xs ), iys = cvFloor( ys );
+                float a = (float)(xs - ixs), b = (float)(ys - iys), a1 = 1.f - a, b1 = 1.f - b;
+                float w00 = a1*b1, w01 = a*b1, w10 = a1*b, w11 = a*b;
+                const uchar *ptr0, *ptr1;
+                xs += A11; ys += A21;
+                
+                if( (unsigned)iys < (unsigned)(src_size.height-1) )
+                    ptr0 = src + src_step*iys, ptr1 = ptr0 + src_step;
+                else
+                    ptr0 = ptr1 = src + (iys < 0 ? 0 : src_size.height-1)*src_step;
+                
+                if( (unsigned)ixs < (unsigned)(src_size.width-1) )
+                {
+                    ptr0 += ixs*cn; ptr1 += ixs*cn;
+                    for( k = 0; k < cn; k++ )
+                        dst[x*cn + k] = ptr0[k]*w00 + ptr0[k+cn]*w01 + ptr1[k]*w10 + ptr1[k+cn]*w11;
+                }
+                else
+                {
+                    ixs = ixs < 0 ? 0 : src_size.width - 1;
+                    ptr0 += ixs*cn; ptr1 += ixs*cn;
+                    for( k = 0; k < cn; k++ )
+                        dst[x*cn + k] = ptr0[k]*b1 + ptr1[k]*b;
+                }
+            }
+        }
+    }
+}
+
 }
 
 void cv::getRectSubPix( InputArray _image, Size patchSize, Point2f center,
@@ -275,12 +368,12 @@ void cv::getRectSubPix( InputArray _image, Size patchSize, Point2f center,
     Mat image = _image.getMat();
     int depth = image.depth(), cn = image.channels();
     int ddepth = patchType < 0 ? depth : CV_MAT_DEPTH(patchType);
-    
+
     CV_Assert( cn == 1 || cn == 3 );
-    
+
     _patch.create(patchSize, CV_MAKETYPE(ddepth, cn));
     Mat patch = _patch.getMat();
-    
+
     if( depth == CV_8U && ddepth == CV_8U )
         getRectSubPix_Cn_<uchar, uchar, int, scale_fixpt, cast_8u>
         (image.data, image.step, image.size(), patch.data, patch.step, patch.size(), center, cn);
@@ -311,6 +404,9 @@ cvGetQuadrangleSubPix( const void* srcarr, void* dstarr, const CvMat* mat )
 {
     cv::Mat src = cv::cvarrToMat(srcarr), m = cv::cvarrToMat(mat);
     const cv::Mat dst = cv::cvarrToMat(dstarr);
+
+    CV_Assert( src.channels() == dst.channels() );
+
     cv::Size win_size = dst.size();
     double matrix[6];
     cv::Mat M(2, 3, CV_64F, matrix);
@@ -320,9 +416,17 @@ cvGetQuadrangleSubPix( const void* srcarr, void* dstarr, const CvMat* mat )
     matrix[2] -= matrix[0]*dx + matrix[1]*dy;
     matrix[5] -= matrix[3]*dx + matrix[4]*dy;
 
-    cv::warpAffine(src, dst, M, dst.size(),
-                   cv::INTER_LINEAR + cv::WARP_INVERSE_MAP,
-                   cv::BORDER_REPLICATE);
+    if( src.depth() == CV_8U && dst.depth() == CV_32F )
+        cv::getQuadrangleSubPix_8u32f_CnR( src.data, src.step, src.size(),
+                                           (float*)dst.data, dst.step, dst.size(),
+                                           matrix, src.channels());
+    else
+    {
+        CV_Assert( src.depth() == dst.depth() );
+        cv::warpAffine(src, dst, M, dst.size(),
+                       cv::INTER_LINEAR + cv::WARP_INVERSE_MAP,
+                       cv::BORDER_REPLICATE);
+    }
 }