Home | History | Annotate | Download | only in ptr2d
      1 /*M///////////////////////////////////////////////////////////////////////////////////////
      2 //
      3 //  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
      4 //
      5 //  By downloading, copying, installing or using the software you agree to this license.
      6 //  If you do not agree to this license, do not download, install,
      7 //  copy or use the software.
      8 //
      9 //
     10 //                          License Agreement
     11 //                For Open Source Computer Vision Library
     12 //
     13 // Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
     14 // Copyright (C) 2009, Willow Garage Inc., all rights reserved.
     15 // Copyright (C) 2013, OpenCV Foundation, all rights reserved.
     16 // Third party copyrights are property of their respective owners.
     17 //
     18 // Redistribution and use in source and binary forms, with or without modification,
     19 // are permitted provided that the following conditions are met:
     20 //
     21 //   * Redistribution's of source code must retain the above copyright notice,
     22 //     this list of conditions and the following disclaimer.
     23 //
     24 //   * Redistribution's in binary form must reproduce the above copyright notice,
     25 //     this list of conditions and the following disclaimer in the documentation
     26 //     and/or other materials provided with the distribution.
     27 //
     28 //   * The name of the copyright holders may not be used to endorse or promote products
     29 //     derived from this software without specific prior written permission.
     30 //
     31 // This software is provided by the copyright holders and contributors "as is" and
     32 // any express or implied warranties, including, but not limited to, the implied
     33 // warranties of merchantability and fitness for a particular purpose are disclaimed.
     34 // In no event shall the Intel Corporation or contributors be liable for any direct,
     35 // indirect, incidental, special, exemplary, or consequential damages
     36 // (including, but not limited to, procurement of substitute goods or services;
     37 // loss of use, data, or profits; or business interruption) however caused
     38 // and on any theory of liability, whether in contract, strict liability,
     39 // or tort (including negligence or otherwise) arising in any way out of
     40 // the use of this software, even if advised of the possibility of such damage.
     41 //
     42 //M*/
     43 
     44 #pragma once
     45 
     46 #ifndef __OPENCV_CUDEV_PTR2D_WARPING_HPP__
     47 #define __OPENCV_CUDEV_PTR2D_WARPING_HPP__
     48 
     49 #include "../common.hpp"
     50 #include "traits.hpp"
     51 #include "remap.hpp"
     52 #include "gpumat.hpp"
     53 
     54 namespace cv { namespace cudev {
     55 
     56 //! @addtogroup cudev
     57 //! @{
     58 
     59 // affine
     60 
     61 struct AffineMapPtr
     62 {
     63     typedef float2 value_type;
     64     typedef float  index_type;
     65 
     66     const float* warpMat;
     67 
     68     __device__ __forceinline__ float2 operator ()(float y, float x) const
     69     {
     70         const float xcoo = warpMat[0] * x + warpMat[1] * y + warpMat[2];
     71         const float ycoo = warpMat[3] * x + warpMat[4] * y + warpMat[5];
     72 
     73         return make_float2(xcoo, ycoo);
     74     }
     75 };
     76 
     77 struct AffineMapPtrSz : AffineMapPtr
     78 {
     79     int rows, cols;
     80 };
     81 
     82 template <> struct PtrTraits<AffineMapPtrSz> : PtrTraitsBase<AffineMapPtrSz, AffineMapPtr>
     83 {
     84 };
     85 
     86 __host__ static AffineMapPtrSz affineMap(Size dstSize, const GpuMat_<float>& warpMat)
     87 {
     88     CV_Assert( warpMat.rows == 2 && warpMat.cols == 3 );
     89     CV_Assert( warpMat.isContinuous() );
     90 
     91     AffineMapPtrSz map;
     92     map.warpMat = warpMat[0];
     93     map.rows = dstSize.height;
     94     map.cols = dstSize.width;
     95     return map;
     96 }
     97 
     98 template <class SrcPtr>
     99 __host__ RemapPtr1Sz<typename PtrTraits<SrcPtr>::ptr_type, AffineMapPtr>
    100 warpAffinePtr(const SrcPtr& src, Size dstSize, const GpuMat_<float>& warpMat)
    101 {
    102     return remapPtr(src, affineMap(dstSize, warpMat));
    103 }
    104 
    105 // perspective
    106 
    107 struct PerspectiveMapPtr
    108 {
    109     typedef float2 value_type;
    110     typedef float  index_type;
    111 
    112     const float* warpMat;
    113 
    114     __device__ __forceinline__ float2 operator ()(float y, float x) const
    115     {
    116         const float coeff = 1.0f / (warpMat[6] * x + warpMat[7] * y + warpMat[8]);
    117 
    118         const float xcoo = coeff * (warpMat[0] * x + warpMat[1] * y + warpMat[2]);
    119         const float ycoo = coeff * (warpMat[3] * x + warpMat[4] * y + warpMat[5]);
    120 
    121         return make_float2(xcoo, ycoo);
    122     }
    123 };
    124 
    125 struct PerspectiveMapPtrSz : PerspectiveMapPtr
    126 {
    127     int rows, cols;
    128 };
    129 
    130 template <> struct PtrTraits<PerspectiveMapPtrSz> : PtrTraitsBase<PerspectiveMapPtrSz, PerspectiveMapPtr>
    131 {
    132 };
    133 
    134 __host__ static PerspectiveMapPtrSz perspectiveMap(Size dstSize, const GpuMat_<float>& warpMat)
    135 {
    136     CV_Assert( warpMat.rows == 3 && warpMat.cols == 3 );
    137     CV_Assert( warpMat.isContinuous() );
    138 
    139     PerspectiveMapPtrSz map;
    140     map.warpMat = warpMat[0];
    141     map.rows = dstSize.height;
    142     map.cols = dstSize.width;
    143     return map;
    144 }
    145 
    146 template <class SrcPtr>
    147 __host__ RemapPtr1Sz<typename PtrTraits<SrcPtr>::ptr_type, PerspectiveMapPtr>
    148 warpPerspectivePtr(const SrcPtr& src, Size dstSize, const GpuMat_<float>& warpMat)
    149 {
    150     return remapPtr(src, perspectiveMap(dstSize, warpMat));
    151 }
    152 
    153 //! @}
    154 
    155 }}
    156 
    157 #endif
    158