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_TRANSFORM_HPP__
47 #define __OPENCV_CUDEV_PTR2D_TRANSFORM_HPP__
48
49 #include "../common.hpp"
50 #include "../grid/copy.hpp"
51 #include "traits.hpp"
52 #include "gpumat.hpp"
53
54 namespace cv { namespace cudev {
55
56 //! @addtogroup cudev
57 //! @{
58
59 // UnaryTransformPtr
60
61 template <class SrcPtr, class Op> struct UnaryTransformPtr
62 {
63 typedef typename Op::result_type value_type;
64 typedef typename PtrTraits<SrcPtr>::index_type index_type;
65
66 SrcPtr src;
67 Op op;
68
operator ()cv::cudev::UnaryTransformPtr69 __device__ __forceinline__ typename Op::result_type operator ()(typename PtrTraits<SrcPtr>::index_type y, typename PtrTraits<SrcPtr>::index_type x) const
70 {
71 return op(src(y, x));
72 }
73 };
74
75 template <class SrcPtr, class Op> struct UnaryTransformPtrSz : UnaryTransformPtr<SrcPtr, Op>
76 {
77 int rows, cols;
78
79 template <typename T>
assignTocv::cudev::UnaryTransformPtrSz80 __host__ void assignTo(GpuMat_<T>& dst, Stream& stream = Stream::Null()) const
81 {
82 gridCopy(*this, dst, stream);
83 }
84 };
85
86 template <class SrcPtr, class Op>
87 __host__ UnaryTransformPtrSz<typename PtrTraits<SrcPtr>::ptr_type, Op>
transformPtr(const SrcPtr & src,const Op & op)88 transformPtr(const SrcPtr& src, const Op& op)
89 {
90 UnaryTransformPtrSz<typename PtrTraits<SrcPtr>::ptr_type, Op> ptr;
91 ptr.src = shrinkPtr(src);
92 ptr.op = op;
93 ptr.rows = getRows(src);
94 ptr.cols = getCols(src);
95 return ptr;
96 }
97
98 template <class SrcPtr, class Op> struct PtrTraits< UnaryTransformPtrSz<SrcPtr, Op> > : PtrTraitsBase<UnaryTransformPtrSz<SrcPtr, Op>, UnaryTransformPtr<SrcPtr, Op> >
99 {
100 };
101
102 // BinaryTransformPtr
103
104 template <class Src1Ptr, class Src2Ptr, class Op> struct BinaryTransformPtr
105 {
106 typedef typename Op::result_type value_type;
107 typedef typename PtrTraits<Src1Ptr>::index_type index_type;
108
109 Src1Ptr src1;
110 Src2Ptr src2;
111 Op op;
112
operator ()cv::cudev::BinaryTransformPtr113 __device__ __forceinline__ typename Op::result_type operator ()(typename PtrTraits<Src1Ptr>::index_type y, typename PtrTraits<Src1Ptr>::index_type x) const
114 {
115 return op(src1(y, x), src2(y, x));
116 }
117 };
118
119 template <class Src1Ptr, class Src2Ptr, class Op> struct BinaryTransformPtrSz : BinaryTransformPtr<Src1Ptr, Src2Ptr, Op>
120 {
121 int rows, cols;
122
123 template <typename T>
assignTocv::cudev::BinaryTransformPtrSz124 __host__ void assignTo(GpuMat_<T>& dst, Stream& stream = Stream::Null()) const
125 {
126 gridCopy(*this, dst, stream);
127 }
128 };
129
130 template <class Src1Ptr, class Src2Ptr, class Op>
131 __host__ BinaryTransformPtrSz<typename PtrTraits<Src1Ptr>::ptr_type, typename PtrTraits<Src2Ptr>::ptr_type, Op>
transformPtr(const Src1Ptr & src1,const Src2Ptr & src2,const Op & op)132 transformPtr(const Src1Ptr& src1, const Src2Ptr& src2, const Op& op)
133 {
134 const int rows = getRows(src1);
135 const int cols = getCols(src1);
136
137 CV_Assert( getRows(src2) == rows && getCols(src2) == cols );
138
139 BinaryTransformPtrSz<typename PtrTraits<Src1Ptr>::ptr_type, typename PtrTraits<Src2Ptr>::ptr_type, Op> ptr;
140 ptr.src1 = shrinkPtr(src1);
141 ptr.src2 = shrinkPtr(src2);
142 ptr.op = op;
143 ptr.rows = rows;
144 ptr.cols = cols;
145 return ptr;
146 }
147
148 template <class Src1Ptr, class Src2Ptr, class Op> struct PtrTraits< BinaryTransformPtrSz<Src1Ptr, Src2Ptr, Op> > : PtrTraitsBase<BinaryTransformPtrSz<Src1Ptr, Src2Ptr, Op>, BinaryTransformPtr<Src1Ptr, Src2Ptr, Op> >
149 {
150 };
151
152 //! @}
153
154 }}
155
156 #endif
157