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