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_LUT_HPP__
     47 #define __OPENCV_CUDEV_PTR2D_LUT_HPP__
     48 
     49 #include "../common.hpp"
     50 #include "../util/vec_traits.hpp"
     51 #include "../grid/copy.hpp"
     52 #include "traits.hpp"
     53 #include "gpumat.hpp"
     54 
     55 namespace cv { namespace cudev {
     56 
     57 //! @addtogroup cudev
     58 //! @{
     59 
     60 template <class SrcPtr, class TablePtr> struct LutPtr
     61 {
     62     typedef typename PtrTraits<TablePtr>::value_type value_type;
     63     typedef typename PtrTraits<SrcPtr>::index_type   index_type;
     64 
     65     SrcPtr src;
     66     TablePtr tbl;
     67 
     68     __device__ __forceinline__ typename PtrTraits<TablePtr>::value_type operator ()(typename PtrTraits<SrcPtr>::index_type y, typename PtrTraits<SrcPtr>::index_type x) const
     69     {
     70         typedef typename PtrTraits<TablePtr>::index_type tbl_index_type;
     71         return tbl(VecTraits<tbl_index_type>::all(0), src(y, x));
     72     }
     73 };
     74 
     75 template <class SrcPtr, class TablePtr> struct LutPtrSz : LutPtr<SrcPtr, TablePtr>
     76 {
     77     int rows, cols;
     78 
     79     template <typename T>
     80     __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 TablePtr>
     87 __host__ LutPtrSz<typename PtrTraits<SrcPtr>::ptr_type, typename PtrTraits<TablePtr>::ptr_type> lutPtr(const SrcPtr& src, const TablePtr& tbl)
     88 {
     89     LutPtrSz<typename PtrTraits<SrcPtr>::ptr_type, typename PtrTraits<TablePtr>::ptr_type> ptr;
     90     ptr.src = shrinkPtr(src);
     91     ptr.tbl = shrinkPtr(tbl);
     92     ptr.rows = getRows(src);
     93     ptr.cols = getCols(src);
     94     return ptr;
     95 }
     96 
     97 template <class SrcPtr, class TablePtr> struct PtrTraits< LutPtrSz<SrcPtr, TablePtr> > : PtrTraitsBase<LutPtrSz<SrcPtr, TablePtr>, LutPtr<SrcPtr, TablePtr> >
     98 {
     99 };
    100 
    101 //! @}
    102 
    103 }}
    104 
    105 #endif
    106