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
operator ()cv::cudev::AffineMapPtr68 __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
affineMap(Size dstSize,const GpuMat_<float> & warpMat)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>
warpAffinePtr(const SrcPtr & src,Size dstSize,const GpuMat_<float> & warpMat)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
operator ()cv::cudev::PerspectiveMapPtr114 __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
perspectiveMap(Size dstSize,const GpuMat_<float> & warpMat)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>
warpPerspectivePtr(const SrcPtr & src,Size dstSize,const GpuMat_<float> & warpMat)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