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_EXTRAPOLATION_HPP__ 47 #define __OPENCV_CUDEV_PTR2D_EXTRAPOLATION_HPP__ 48 49 #include "../common.hpp" 50 #include "../util/vec_traits.hpp" 51 #include "traits.hpp" 52 53 namespace cv { namespace cudev { 54 55 //! @addtogroup cudev 56 //! @{ 57 58 // BrdConstant 59 60 template <class SrcPtr> struct BrdConstant 61 { 62 typedef typename PtrTraits<SrcPtr>::value_type value_type; 63 typedef int index_type; 64 65 SrcPtr src; 66 int rows, cols; 67 typename PtrTraits<SrcPtr>::value_type val; 68 69 __device__ __forceinline__ typename PtrTraits<SrcPtr>::value_type operator ()(int y, int x) const 70 { 71 return (x >= 0 && x < cols && y >= 0 && y < rows) ? src(y, x) : val; 72 } 73 }; 74 75 template <class SrcPtr> 76 __host__ BrdConstant<typename PtrTraits<SrcPtr>::ptr_type> brdConstant(const SrcPtr& src, typename PtrTraits<SrcPtr>::value_type val) 77 { 78 BrdConstant<typename PtrTraits<SrcPtr>::ptr_type> b; 79 b.src = shrinkPtr(src); 80 b.rows = getRows(src); 81 b.cols = getCols(src); 82 b.val = val; 83 return b; 84 } 85 86 template <class SrcPtr> 87 __host__ BrdConstant<typename PtrTraits<SrcPtr>::ptr_type> brdConstant(const SrcPtr& src) 88 { 89 return brdConstant(src, VecTraits<typename PtrTraits<SrcPtr>::value_type>::all(0)); 90 } 91 92 // BrdBase 93 94 template <class BrdImpl, class SrcPtr> struct BrdBase 95 { 96 typedef typename PtrTraits<SrcPtr>::value_type value_type; 97 typedef int index_type; 98 99 SrcPtr src; 100 int rows, cols; 101 102 __device__ __forceinline__ int idx_row(int y) const 103 { 104 return BrdImpl::idx_low(BrdImpl::idx_high(y, rows), rows); 105 } 106 107 __device__ __forceinline__ int idx_col(int x) const 108 { 109 return BrdImpl::idx_low(BrdImpl::idx_high(x, cols), cols); 110 } 111 112 __device__ __forceinline__ typename PtrTraits<SrcPtr>::value_type operator ()(int y, int x) const 113 { 114 return src(idx_row(y), idx_col(x)); 115 } 116 }; 117 118 // BrdReplicate 119 120 struct BrdReplicate 121 { 122 __device__ __forceinline__ static int idx_low(int i, int len) 123 { 124 return ::max(i, 0); 125 } 126 127 __device__ __forceinline__ static int idx_high(int i, int len) 128 { 129 return ::min(i, len - 1); 130 } 131 }; 132 133 template <class SrcPtr> 134 __host__ BrdBase<BrdReplicate, typename PtrTraits<SrcPtr>::ptr_type> brdReplicate(const SrcPtr& src) 135 { 136 BrdBase<BrdReplicate, typename PtrTraits<SrcPtr>::ptr_type> b; 137 b.src = shrinkPtr(src); 138 b.rows = getRows(src); 139 b.cols = getCols(src); 140 return b; 141 } 142 143 // BrdReflect101 144 145 struct BrdReflect101 146 { 147 __device__ __forceinline__ static int idx_low(int i, int len) 148 { 149 return ::abs(i) % len; 150 } 151 152 __device__ __forceinline__ static int idx_high(int i, int len) 153 { 154 const int last_ind = len - 1; 155 return ::abs(last_ind - ::abs(last_ind - i)) % len; 156 } 157 }; 158 159 template <class SrcPtr> 160 __host__ BrdBase<BrdReflect101, typename PtrTraits<SrcPtr>::ptr_type> brdReflect101(const SrcPtr& src) 161 { 162 BrdBase<BrdReflect101, typename PtrTraits<SrcPtr>::ptr_type> b; 163 b.src = shrinkPtr(src); 164 b.rows = getRows(src); 165 b.cols = getCols(src); 166 return b; 167 } 168 169 // BrdReflect 170 171 struct BrdReflect 172 { 173 __device__ __forceinline__ static int idx_low(int i, int len) 174 { 175 return (::abs(i) - (i < 0)) % len; 176 } 177 178 __device__ __forceinline__ static int idx_high(int i, int len) 179 { 180 const int last_ind = len - 1; 181 return (last_ind - ::abs(last_ind - i) + (i > last_ind)); 182 } 183 }; 184 185 template <class SrcPtr> 186 __host__ BrdBase<BrdReflect, typename PtrTraits<SrcPtr>::ptr_type> brdReflect(const SrcPtr& src) 187 { 188 BrdBase<BrdReflect, typename PtrTraits<SrcPtr>::ptr_type> b; 189 b.src = shrinkPtr(src); 190 b.rows = getRows(src); 191 b.cols = getCols(src); 192 return b; 193 } 194 195 // BrdWrap 196 197 struct BrdWrap 198 { 199 __device__ __forceinline__ static int idx_low(int i, int len) 200 { 201 return (i >= 0) * i + (i < 0) * (i - ((i - len + 1) / len) * len); 202 } 203 204 __device__ __forceinline__ static int idx_high(int i, int len) 205 { 206 return (i < len) * i + (i >= len) * (i % len); 207 } 208 }; 209 210 template <class SrcPtr> 211 __host__ BrdBase<BrdWrap, typename PtrTraits<SrcPtr>::ptr_type> brdWrap(const SrcPtr& src) 212 { 213 BrdBase<BrdWrap, typename PtrTraits<SrcPtr>::ptr_type> b; 214 b.src = shrinkPtr(src); 215 b.rows = getRows(src); 216 b.cols = getCols(src); 217 return b; 218 } 219 220 //! @} 221 222 }} 223 224 #endif 225
