First gpu code draft. Interface might be significantly changed in future.
Simple implementation of StereoBM_GPU. It is excluded from compilation now.
This commit is contained in:
73
modules/gpu/include/opencv2/gpu/devmem2d.hpp
Normal file
73
modules/gpu/include/opencv2/gpu/devmem2d.hpp
Normal file
@@ -0,0 +1,73 @@
|
||||
/*M///////////////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
|
||||
//
|
||||
// By downloading, copying, installing or using the software you agree to this license.
|
||||
// If you do not agree to this license, do not download, install,
|
||||
// copy or use the software.
|
||||
//
|
||||
//
|
||||
// License Agreement
|
||||
// For Open Source Computer Vision Library
|
||||
//
|
||||
// Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
|
||||
// Copyright (C) 2009, Willow Garage Inc., all rights reserved.
|
||||
// Third party copyrights are property of their respective owners.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without modification,
|
||||
// are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistribution's of source code must retain the above copyright notice,
|
||||
// this list of conditions and the following disclaimer.
|
||||
//
|
||||
// * Redistribution's in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other GpuMaterials provided with the distribution.
|
||||
//
|
||||
// * The name of the copyright holders may not be used to endorse or promote products
|
||||
// derived from this software without specific prior written permission.
|
||||
//
|
||||
// This software is provided by the copyright holders and contributors "as is" and
|
||||
// any express or implied warranties, including, but not limited to, the implied
|
||||
// warranties of merchantability and fitness for a particular purpose are disclaimed.
|
||||
// In no event shall the Intel Corporation or contributors be liable for any direct,
|
||||
// indirect, incidental, special, exemplary, or consequential damages
|
||||
// (including, but not limited to, procurement of substitute goods or services;
|
||||
// loss of use, data, or profits; or business interruption) however caused
|
||||
// and on any theory of liability, whether in contract, strict liability,
|
||||
// or tort (including negligence or otherwise) arising in any way out of
|
||||
// the use of this software, even if advised of the possibility of such damage.
|
||||
//
|
||||
//M*/
|
||||
|
||||
#ifndef __OPENCV_GPU_DEVMEM2D_HPP__
|
||||
#define __OPENCV_GPU_DEVMEM2D_HPP__
|
||||
|
||||
namespace cv
|
||||
{
|
||||
namespace gpu
|
||||
{
|
||||
// Simple lightweight structure that encapsulates image ptr on device, its pitch and its sizes.
|
||||
// It is intended to pass to nvcc-compiled code.
|
||||
|
||||
template<typename T = unsigned char>
|
||||
struct DevMem2D_
|
||||
{
|
||||
enum { elem_size = sizeof(T) };
|
||||
|
||||
int cols;
|
||||
int rows;
|
||||
T* ptr;
|
||||
size_t step;
|
||||
|
||||
DevMem2D_(int rows_, int cols_, T *ptr_, size_t step_)
|
||||
: cols(cols_), rows(rows_), ptr(ptr_), step(step_) {}
|
||||
|
||||
size_t elemSize() const { return elem_size; }
|
||||
};
|
||||
|
||||
typedef DevMem2D_<> DevMem2D;
|
||||
}
|
||||
}
|
||||
|
||||
#endif /* __OPENCV_GPU_DEVMEM2D_HPP__ */
|
276
modules/gpu/include/opencv2/gpu/gpu.hpp
Normal file
276
modules/gpu/include/opencv2/gpu/gpu.hpp
Normal file
@@ -0,0 +1,276 @@
|
||||
/*M///////////////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
|
||||
//
|
||||
// By downloading, copying, installing or using the software you agree to this license.
|
||||
// If you do not agree to this license, do not download, install,
|
||||
// copy or use the software.
|
||||
//
|
||||
//
|
||||
// License Agreement
|
||||
// For Open Source Computer Vision Library
|
||||
//
|
||||
// Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
|
||||
// Copyright (C) 2009, Willow Garage Inc., all rights reserved.
|
||||
// Third party copyrights are property of their respective owners.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without modification,
|
||||
// are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistribution's of source code must retain the above copyright notice,
|
||||
// this list of conditions and the following disclaimer.
|
||||
//
|
||||
// * Redistribution's in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other GpuMaterials provided with the distribution.
|
||||
//
|
||||
// * The name of the copyright holders may not be used to endorse or promote products
|
||||
// derived from this software without specific prior written permission.
|
||||
//
|
||||
// This software is provided by the copyright holders and contributors "as is" and
|
||||
// any express or implied warranties, including, but not limited to, the implied
|
||||
// warranties of merchantability and fitness for a particular purpose are disclaimed.
|
||||
// In no event shall the Intel Corporation or contributors be liable for any direct,
|
||||
// indirect, incidental, special, exemplary, or consequential damages
|
||||
// (including, but not limited to, procurement of substitute goods or services;
|
||||
// loss of use, data, or profits; or business interruption) however caused
|
||||
// and on any theory of liability, whether in contract, strict liability,
|
||||
// or tort (including negligence or otherwise) arising in any way out of
|
||||
// the use of this software, even if advised of the possibility of such damage.
|
||||
//
|
||||
//M*/
|
||||
|
||||
#ifndef __OPENCV_GPU_HPP__
|
||||
#define __OPENCV_GPU_HPP__
|
||||
|
||||
#include "opencv2/core/core.hpp"
|
||||
#include "opencv2/gpu/devmem2d.hpp"
|
||||
|
||||
namespace cv
|
||||
{
|
||||
namespace gpu
|
||||
{
|
||||
//////////////////////////////// Initialization ////////////////////////
|
||||
|
||||
CV_EXPORTS int getCudaEnabledDeviceCount();
|
||||
CV_EXPORTS string getDeviceName(int device);
|
||||
CV_EXPORTS void setDevice(int device);
|
||||
|
||||
enum { CV_GPU_CC_10, CV_GPU_CC_11, CV_GPU_CC_12, CV_GPU_CC_13, CV_GPU_CC_20 };
|
||||
|
||||
CV_EXPORTS int getComputeCapability(int device);
|
||||
CV_EXPORTS int getNumberOfSMs(int device);
|
||||
|
||||
//////////////////////////////// GpuMat ////////////////////////////////
|
||||
|
||||
class CV_EXPORTS GpuMat
|
||||
{
|
||||
public:
|
||||
//! default constructor
|
||||
GpuMat();
|
||||
//! constructs GpuMatrix of the specified size and type
|
||||
// (_type is CV_8UC1, CV_64FC3, CV_32SC(12) etc.)
|
||||
GpuMat(int _rows, int _cols, int _type);
|
||||
GpuMat(Size _size, int _type);
|
||||
//! constucts GpuMatrix and fills it with the specified value _s.
|
||||
GpuMat(int _rows, int _cols, int _type, const Scalar& _s);
|
||||
GpuMat(Size _size, int _type, const Scalar& _s);
|
||||
//! copy constructor
|
||||
GpuMat(const GpuMat& m);
|
||||
|
||||
//! constructor for GpuMatrix headers pointing to user-allocated data
|
||||
GpuMat(int _rows, int _cols, int _type, void* _data, size_t _step = Mat::AUTO_STEP);
|
||||
GpuMat(Size _size, int _type, void* _data, size_t _step = Mat::AUTO_STEP);
|
||||
|
||||
//! creates a matrix header for a part of the bigger matrix
|
||||
GpuMat(const GpuMat& m, const Range& rowRange, const Range& colRange);
|
||||
GpuMat(const GpuMat& m, const Rect& roi);
|
||||
|
||||
//! builds GpuMat from Mat. Perfom blocking upload to device.
|
||||
GpuMat (const Mat& m);
|
||||
|
||||
//! destructor - calls release()
|
||||
~GpuMat();
|
||||
|
||||
//! assignment operators
|
||||
GpuMat& operator = (const GpuMat& m);
|
||||
//! assignment operator. Perfom blocking upload to device.
|
||||
GpuMat& operator = (const Mat& m);
|
||||
|
||||
//! returns lightweight DevMem2D_ structure for passing to nvcc-compiled code.
|
||||
// Contains just image size, data ptr and step.
|
||||
template <class T> operator DevMem2D_<T>() const;
|
||||
|
||||
//! pefroms blocking upload data to GpuMat. .
|
||||
void upload(const cv::Mat& m);
|
||||
|
||||
//! Downloads data from device to host memory. Blocking calls.
|
||||
operator Mat() const;
|
||||
void download(cv::Mat& m) const;
|
||||
|
||||
//! returns a new GpuMatrix header for the specified row
|
||||
GpuMat row(int y) const;
|
||||
//! returns a new GpuMatrix header for the specified column
|
||||
GpuMat col(int x) const;
|
||||
//! ... for the specified row span
|
||||
GpuMat rowRange(int startrow, int endrow) const;
|
||||
GpuMat rowRange(const Range& r) const;
|
||||
//! ... for the specified column span
|
||||
GpuMat colRange(int startcol, int endcol) const;
|
||||
GpuMat colRange(const Range& r) const;
|
||||
|
||||
//! returns deep copy of the GpuMatrix, i.e. the data is copied
|
||||
GpuMat clone() const;
|
||||
//! copies the GpuMatrix content to "m".
|
||||
// It calls m.create(this->size(), this->type()).
|
||||
void copyTo( GpuMat& m ) const;
|
||||
//! copies those GpuMatrix elements to "m" that are marked with non-zero mask elements.
|
||||
void copyTo( GpuMat& m, const GpuMat& mask ) const;
|
||||
//! converts GpuMatrix to another datatype with optional scalng. See cvConvertScale.
|
||||
void convertTo( GpuMat& m, int rtype, double alpha=1, double beta=0 ) const;
|
||||
|
||||
void assignTo( GpuMat& m, int type=-1 ) const;
|
||||
|
||||
//! sets every GpuMatrix element to s
|
||||
GpuMat& operator = (const Scalar& s);
|
||||
//! sets some of the GpuMatrix elements to s, according to the mask
|
||||
GpuMat& setTo(const Scalar& s, const GpuMat& mask=GpuMat());
|
||||
//! creates alternative GpuMatrix header for the same data, with different
|
||||
// number of channels and/or different number of rows. see cvReshape.
|
||||
GpuMat reshape(int _cn, int _rows=0) const;
|
||||
|
||||
//! allocates new GpuMatrix data unless the GpuMatrix already has specified size and type.
|
||||
// previous data is unreferenced if needed.
|
||||
void create(int _rows, int _cols, int _type);
|
||||
void create(Size _size, int _type);
|
||||
//! decreases reference counter;
|
||||
// deallocate the data when reference counter reaches 0.
|
||||
void release();
|
||||
|
||||
//! swaps with other smart pointer
|
||||
void swap(GpuMat& mat);
|
||||
|
||||
//! locates GpuMatrix header within a parent GpuMatrix. See below
|
||||
void locateROI( Size& wholeSize, Point& ofs ) const;
|
||||
//! moves/resizes the current GpuMatrix ROI inside the parent GpuMatrix.
|
||||
GpuMat& adjustROI( int dtop, int dbottom, int dleft, int dright );
|
||||
//! extracts a rectangular sub-GpuMatrix
|
||||
// (this is a generalized form of row, rowRange etc.)
|
||||
GpuMat operator()( Range rowRange, Range colRange ) const;
|
||||
GpuMat operator()( const Rect& roi ) const;
|
||||
|
||||
//! returns true iff the GpuMatrix data is continuous
|
||||
// (i.e. when there are no gaps between successive rows).
|
||||
// similar to CV_IS_GpuMat_CONT(cvGpuMat->type)
|
||||
bool isContinuous() const;
|
||||
//! returns element size in bytes,
|
||||
// similar to CV_ELEM_SIZE(cvMat->type)
|
||||
size_t elemSize() const;
|
||||
//! returns the size of element channel in bytes.
|
||||
size_t elemSize1() const;
|
||||
//! returns element type, similar to CV_MAT_TYPE(cvMat->type)
|
||||
int type() const;
|
||||
//! returns element type, similar to CV_MAT_DEPTH(cvMat->type)
|
||||
int depth() const;
|
||||
//! returns element type, similar to CV_MAT_CN(cvMat->type)
|
||||
int channels() const;
|
||||
//! returns step/elemSize1()
|
||||
size_t step1() const;
|
||||
//! returns GpuMatrix size:
|
||||
// width == number of columns, height == number of rows
|
||||
Size size() const;
|
||||
//! returns true if GpuMatrix data is NULL
|
||||
bool empty() const;
|
||||
|
||||
//! returns pointer to y-th row
|
||||
uchar* ptr(int y=0);
|
||||
const uchar* ptr(int y=0) const;
|
||||
|
||||
//! template version of the above method
|
||||
template<typename _Tp> _Tp* ptr(int y=0);
|
||||
template<typename _Tp> const _Tp* ptr(int y=0) const;
|
||||
|
||||
/*! includes several bit-fields:
|
||||
- the magic signature
|
||||
- continuity flag
|
||||
- depth
|
||||
- number of channels
|
||||
*/
|
||||
int flags;
|
||||
//! the number of rows and columns
|
||||
int rows, cols;
|
||||
//! a distance between successive rows in bytes; includes the gap if any
|
||||
size_t step;
|
||||
//! pointer to the data
|
||||
uchar* data;
|
||||
|
||||
//! pointer to the reference counter;
|
||||
// when GpuMatrix points to user-allocated data, the pointer is NULL
|
||||
int* refcount;
|
||||
|
||||
//! helper fields used in locateROI and adjustROI
|
||||
uchar* datastart;
|
||||
uchar* dataend;
|
||||
};
|
||||
|
||||
//////////////////////////////// CudaStream ////////////////////////////////
|
||||
|
||||
class CudaStream
|
||||
{
|
||||
public:
|
||||
CudaStream();
|
||||
~CudaStream();
|
||||
|
||||
bool queryIfComplete();
|
||||
void waitForCompletion();
|
||||
|
||||
//calls cudaMemcpyAsync
|
||||
void enqueueDownload(const GpuMat& src, Mat& dst);
|
||||
void enqueueUpload(const Mat& src, GpuMat& dst);
|
||||
void enqueueCopy(const GpuMat& src, GpuMat& dst);
|
||||
|
||||
// calls cudaMemset2D asynchronous for single channel. Invoke kernel for some multichannel.
|
||||
void enqueueMemSet(const GpuMat& src, Scalar val);
|
||||
|
||||
// invoke kernel asynchronous because of mask
|
||||
void enqueueMemSet(const GpuMat& src, Scalar val, const GpuMat& mask);
|
||||
|
||||
// converts matrix type, ex from float to uchar depending on type
|
||||
void enqueueConvert(const GpuMat& src, GpuMat& dst, int type);
|
||||
|
||||
//CUstream_st& getStream();
|
||||
private:
|
||||
void *impl;
|
||||
|
||||
CudaStream(const CudaStream&);
|
||||
CudaStream& operator=(const CudaStream&);
|
||||
};
|
||||
|
||||
//////////////////////////////// StereoBM_GPU ////////////////////////////////
|
||||
|
||||
class CV_EXPORTS StereoBM_GPU
|
||||
{
|
||||
public:
|
||||
enum { BASIC_PRESET=0, PREFILTER_XSOBEL = 1 };
|
||||
|
||||
//! the default constructor
|
||||
StereoBM_GPU();
|
||||
//! the full constructor taking the camera-specific preset, number of disparities and the SAD window size
|
||||
//! ndisparities should be multiple of 8. SSD WindowsSize is fixed to 19 now
|
||||
StereoBM_GPU(int preset, int ndisparities=0);
|
||||
//! the stereo correspondence operator. Finds the disparity for the specified rectified stereo pair
|
||||
//! Output disparity has CV_8U type.
|
||||
void operator() ( const GpuMat& left, const GpuMat& right, GpuMat& disparity) const;
|
||||
private:
|
||||
mutable GpuMat minSSD;
|
||||
int preset;
|
||||
int ndisp;
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
#include "opencv2/gpu/gpumat.hpp"
|
||||
|
||||
#endif /* __OPENCV_GPU_HPP__ */
|
350
modules/gpu/include/opencv2/gpu/gpumat.hpp
Normal file
350
modules/gpu/include/opencv2/gpu/gpumat.hpp
Normal file
@@ -0,0 +1,350 @@
|
||||
/*M///////////////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
|
||||
//
|
||||
// By downloading, copying, installing or using the software you agree to this license.
|
||||
// If you do not agree to this license, do not download, install,
|
||||
// copy or use the software.
|
||||
//
|
||||
//
|
||||
// License Agreement
|
||||
// For Open Source Computer Vision Library
|
||||
//
|
||||
// Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
|
||||
// Copyright (C) 2009, Willow Garage Inc., all rights reserved.
|
||||
// Third party copyrights are property of their respective owners.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without modification,
|
||||
// are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistribution's of source code must retain the above copyright notice,
|
||||
// this list of conditions and the following disclaimer.
|
||||
//
|
||||
// * Redistribution's in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other GpuMaterials provided with the distribution.
|
||||
//
|
||||
// * The name of the copyright holders may not be used to endorse or promote products
|
||||
// derived from this software without specific prior written permission.
|
||||
//
|
||||
// This software is provided by the copyright holders and contributors "as is" and
|
||||
// any express or implied warranties, including, but not limited to, the implied
|
||||
// warranties of merchantability and fitness for a particular purpose are disclaimed.
|
||||
// In no event shall the Intel Corporation or contributors be liable for any direct,
|
||||
// indirect, incidental, special, exemplary, or consequential damages
|
||||
// (including, but not limited to, procurement of substitute goods or services;
|
||||
// loss of use, data, or profits; or business interruption) however caused
|
||||
// and on any theory of liability, whether in contract, strict liability,
|
||||
// or tort (including negligence or otherwise) arising in any way out of
|
||||
// the use of this software, even if advised of the possibility of such damage.
|
||||
//
|
||||
//M*/
|
||||
|
||||
#ifndef __OPENCV_GPU_MATRIX_OPERATIONS_HPP__
|
||||
#define __OPENCV_GPU_MATRIX_OPERATIONS_HPP__
|
||||
|
||||
|
||||
namespace cv
|
||||
{
|
||||
|
||||
namespace gpu
|
||||
{
|
||||
|
||||
//////////////////////////////// GpuMat ////////////////////////////////
|
||||
|
||||
inline GpuMat::GpuMat()
|
||||
: flags(0), rows(0), cols(0), step(0), data(0), refcount(0), datastart(0), dataend(0) {}
|
||||
|
||||
inline GpuMat::GpuMat(int _rows, int _cols, int _type)
|
||||
: flags(0), rows(0), cols(0), step(0), data(0), refcount(0), datastart(0), dataend(0)
|
||||
{
|
||||
if( _rows > 0 && _cols > 0 )
|
||||
create( _rows, _cols, _type );
|
||||
}
|
||||
|
||||
inline GpuMat::GpuMat(Size _size, int _type)
|
||||
: flags(0), rows(0), cols(0), step(0), data(0), refcount(0), datastart(0), dataend(0)
|
||||
{
|
||||
if( _size.height > 0 && _size.width > 0 )
|
||||
create( _size.height, _size.width, _type );
|
||||
}
|
||||
|
||||
inline GpuMat::GpuMat(int _rows, int _cols, int _type, const Scalar& _s)
|
||||
: flags(0), rows(0), cols(0), step(0), data(0), refcount(0), datastart(0), dataend(0)
|
||||
{
|
||||
if(_rows > 0 && _cols > 0)
|
||||
{
|
||||
create(_rows, _cols, _type);
|
||||
*this = _s;
|
||||
}
|
||||
}
|
||||
|
||||
inline GpuMat::GpuMat(Size _size, int _type, const Scalar& _s)
|
||||
: flags(0), rows(0), cols(0), step(0), data(0), refcount(0), datastart(0), dataend(0)
|
||||
{
|
||||
if( _size.height > 0 && _size.width > 0 )
|
||||
{
|
||||
create( _size.height, _size.width, _type );
|
||||
*this = _s;
|
||||
}
|
||||
}
|
||||
|
||||
inline GpuMat::GpuMat(const GpuMat& m)
|
||||
: flags(m.flags), rows(m.rows), cols(m.cols), step(m.step), data(m.data), refcount(m.refcount), datastart(m.datastart), dataend(m.dataend)
|
||||
{
|
||||
if( refcount )
|
||||
CV_XADD(refcount, 1);
|
||||
}
|
||||
|
||||
inline GpuMat::GpuMat(int _rows, int _cols, int _type, void* _data, size_t _step)
|
||||
: flags(Mat::MAGIC_VAL + (_type & TYPE_MASK)), rows(_rows), cols(_cols), step(_step), data((uchar*)_data), refcount(0),
|
||||
datastart((uchar*)_data), dataend((uchar*)_data)
|
||||
{
|
||||
size_t minstep = cols*elemSize();
|
||||
if( step == Mat::AUTO_STEP )
|
||||
{
|
||||
step = minstep;
|
||||
flags |= Mat::CONTINUOUS_FLAG;
|
||||
}
|
||||
else
|
||||
{
|
||||
if( rows == 1 ) step = minstep;
|
||||
CV_DbgAssert( step >= minstep );
|
||||
flags |= step == minstep ? Mat::CONTINUOUS_FLAG : 0;
|
||||
}
|
||||
dataend += step*(rows-1) + minstep;
|
||||
}
|
||||
|
||||
inline GpuMat::GpuMat(Size _size, int _type, void* _data, size_t _step)
|
||||
: flags(Mat::MAGIC_VAL + (_type & TYPE_MASK)), rows(_size.height), cols(_size.width),
|
||||
step(_step), data((uchar*)_data), refcount(0),
|
||||
datastart((uchar*)_data), dataend((uchar*)_data)
|
||||
{
|
||||
size_t minstep = cols*elemSize();
|
||||
if( step == Mat::AUTO_STEP )
|
||||
{
|
||||
step = minstep;
|
||||
flags |= Mat::CONTINUOUS_FLAG;
|
||||
}
|
||||
else
|
||||
{
|
||||
if( rows == 1 ) step = minstep;
|
||||
CV_DbgAssert( step >= minstep );
|
||||
flags |= step == minstep ? Mat::CONTINUOUS_FLAG : 0;
|
||||
}
|
||||
dataend += step*(rows-1) + minstep;
|
||||
}
|
||||
|
||||
|
||||
inline GpuMat::GpuMat(const GpuMat& m, const Range& rowRange, const Range& colRange)
|
||||
{
|
||||
flags = m.flags;
|
||||
step = m.step; refcount = m.refcount;
|
||||
data = m.data; datastart = m.datastart; dataend = m.dataend;
|
||||
|
||||
if( rowRange == Range::all() )
|
||||
rows = m.rows;
|
||||
else
|
||||
{
|
||||
CV_Assert( 0 <= rowRange.start && rowRange.start <= rowRange.end && rowRange.end <= m.rows );
|
||||
rows = rowRange.size();
|
||||
data += step*rowRange.start;
|
||||
}
|
||||
|
||||
if( colRange == Range::all() )
|
||||
cols = m.cols;
|
||||
else
|
||||
{
|
||||
CV_Assert( 0 <= colRange.start && colRange.start <= colRange.end && colRange.end <= m.cols );
|
||||
cols = colRange.size();
|
||||
data += colRange.start*elemSize();
|
||||
flags &= cols < m.cols ? ~Mat::CONTINUOUS_FLAG : -1;
|
||||
}
|
||||
|
||||
if( rows == 1 )
|
||||
flags |= Mat::CONTINUOUS_FLAG;
|
||||
|
||||
if( refcount )
|
||||
CV_XADD(refcount, 1);
|
||||
if( rows <= 0 || cols <= 0 )
|
||||
rows = cols = 0;
|
||||
}
|
||||
|
||||
inline GpuMat::GpuMat(const GpuMat& m, const Rect& roi)
|
||||
: flags(m.flags), rows(roi.height), cols(roi.width),
|
||||
step(m.step), data(m.data + roi.y*step), refcount(m.refcount),
|
||||
datastart(m.datastart), dataend(m.dataend)
|
||||
{
|
||||
flags &= roi.width < m.cols ? ~Mat::CONTINUOUS_FLAG : -1;
|
||||
data += roi.x*elemSize();
|
||||
CV_Assert( 0 <= roi.x && 0 <= roi.width && roi.x + roi.width <= m.cols &&
|
||||
0 <= roi.y && 0 <= roi.height && roi.y + roi.height <= m.rows );
|
||||
if( refcount )
|
||||
CV_XADD(refcount, 1);
|
||||
if( rows <= 0 || cols <= 0 )
|
||||
rows = cols = 0;
|
||||
}
|
||||
|
||||
inline GpuMat::GpuMat(const Mat& m)
|
||||
: flags(0), rows(0), cols(0), step(0), data(0), refcount(0), datastart(0), dataend(0) { upload(m); }
|
||||
|
||||
inline GpuMat::~GpuMat() { release(); }
|
||||
|
||||
inline GpuMat& GpuMat::operator = (const GpuMat& m)
|
||||
{
|
||||
if( this != &m )
|
||||
{
|
||||
if( m.refcount )
|
||||
CV_XADD(m.refcount, 1);
|
||||
release();
|
||||
flags = m.flags;
|
||||
rows = m.rows; cols = m.cols;
|
||||
step = m.step; data = m.data;
|
||||
datastart = m.datastart; dataend = m.dataend;
|
||||
refcount = m.refcount;
|
||||
}
|
||||
return *this;
|
||||
}
|
||||
|
||||
inline GpuMat& GpuMat::operator = (const Mat& m) { upload(m); return *this; }
|
||||
|
||||
template <class T> inline GpuMat::operator DevMem2D_<T>() const { return DevMem2D_<T>(rows, cols, (T*)data, step); }
|
||||
|
||||
//CPP: void GpuMat::upload(const Mat& m);
|
||||
|
||||
inline GpuMat::operator Mat() const
|
||||
{
|
||||
Mat m;
|
||||
download(m);
|
||||
return m;
|
||||
}
|
||||
|
||||
//CPP void GpuMat::download(cv::Mat& m) const;
|
||||
|
||||
inline GpuMat GpuMat::row(int y) const { return GpuMat(*this, Range(y, y+1), Range::all()); }
|
||||
inline GpuMat GpuMat::col(int x) const { return GpuMat(*this, Range::all(), Range(x, x+1)); }
|
||||
inline GpuMat GpuMat::rowRange(int startrow, int endrow) const { return GpuMat(*this, Range(startrow, endrow), Range::all()); }
|
||||
inline GpuMat GpuMat::rowRange(const Range& r) const { return GpuMat(*this, r, Range::all()); }
|
||||
inline GpuMat GpuMat::colRange(int startcol, int endcol) const { return GpuMat(*this, Range::all(), Range(startcol, endcol)); }
|
||||
inline GpuMat GpuMat::colRange(const Range& r) const { return GpuMat(*this, Range::all(), r); }
|
||||
|
||||
inline GpuMat GpuMat::clone() const
|
||||
{
|
||||
GpuMat m;
|
||||
copyTo(m);
|
||||
return m;
|
||||
}
|
||||
|
||||
//CPP void GpuMat::copyTo( GpuMat& m ) const;
|
||||
//CPP void GpuMat::copyTo( GpuMat& m, const GpuMat& mask ) const;
|
||||
//CPP void GpuMat::convertTo( GpuMat& m, int rtype, double alpha=1, double beta=0 ) const;
|
||||
|
||||
inline void GpuMat::assignTo( GpuMat& m, int type ) const
|
||||
{
|
||||
if( type < 0 )
|
||||
m = *this;
|
||||
else
|
||||
convertTo(m, type);
|
||||
}
|
||||
|
||||
//CPP GpuMat& GpuMat::operator = (const Scalar& s);
|
||||
//CPP GpuMat& GpuMat::setTo(const Scalar& s, const GpuMat& mask=GpuMat());
|
||||
|
||||
//CPP GpuMat GpuMat::reshape(int _cn, int _rows=0) const;
|
||||
|
||||
//CPP void GpuMat::create(int _rows, int _cols, int _type);
|
||||
inline void GpuMat::create(Size _size, int _type) { create(_size.height, _size.width, _type); }
|
||||
|
||||
//CPP void GpuMat::release();
|
||||
|
||||
inline void GpuMat::swap(GpuMat& b)
|
||||
{
|
||||
std::swap( flags, b.flags );
|
||||
std::swap( rows, b.rows ); std::swap( cols, b.cols );
|
||||
std::swap( step, b.step ); std::swap( data, b.data );
|
||||
std::swap( datastart, b.datastart );
|
||||
std::swap( dataend, b.dataend );
|
||||
std::swap( refcount, b.refcount );
|
||||
}
|
||||
|
||||
inline void GpuMat::locateROI( Size& wholeSize, Point& ofs ) const
|
||||
{
|
||||
size_t esz = elemSize(), minstep;
|
||||
ptrdiff_t delta1 = data - datastart, delta2 = dataend - datastart;
|
||||
CV_DbgAssert( step > 0 );
|
||||
if( delta1 == 0 )
|
||||
ofs.x = ofs.y = 0;
|
||||
else
|
||||
{
|
||||
ofs.y = (int)(delta1/step);
|
||||
ofs.x = (int)((delta1 - step*ofs.y)/esz);
|
||||
CV_DbgAssert( data == datastart + ofs.y*step + ofs.x*esz );
|
||||
}
|
||||
minstep = (ofs.x + cols)*esz;
|
||||
wholeSize.height = (int)((delta2 - minstep)/step + 1);
|
||||
wholeSize.height = std::max(wholeSize.height, ofs.y + rows);
|
||||
wholeSize.width = (int)((delta2 - step*(wholeSize.height-1))/esz);
|
||||
wholeSize.width = std::max(wholeSize.width, ofs.x + cols);
|
||||
}
|
||||
|
||||
inline GpuMat& GpuMat::adjustROI( int dtop, int dbottom, int dleft, int dright )
|
||||
{
|
||||
Size wholeSize; Point ofs;
|
||||
size_t esz = elemSize();
|
||||
locateROI( wholeSize, ofs );
|
||||
int row1 = std::max(ofs.y - dtop, 0), row2 = std::min(ofs.y + rows + dbottom, wholeSize.height);
|
||||
int col1 = std::max(ofs.x - dleft, 0), col2 = std::min(ofs.x + cols + dright, wholeSize.width);
|
||||
data += (row1 - ofs.y)*step + (col1 - ofs.x)*esz;
|
||||
rows = row2 - row1; cols = col2 - col1;
|
||||
if( esz*cols == step || rows == 1 )
|
||||
flags |= Mat::CONTINUOUS_FLAG;
|
||||
else
|
||||
flags &= ~Mat::CONTINUOUS_FLAG;
|
||||
return *this;
|
||||
}
|
||||
|
||||
inline GpuMat GpuMat::operator()( Range rowRange, Range colRange ) const { return GpuMat(*this, rowRange, colRange); }
|
||||
inline GpuMat GpuMat::operator()( const Rect& roi ) const { return GpuMat(*this, roi); }
|
||||
|
||||
inline bool GpuMat::isContinuous() const { return (flags & Mat::CONTINUOUS_FLAG) != 0; }
|
||||
inline size_t GpuMat::elemSize() const { return CV_ELEM_SIZE(flags); }
|
||||
inline size_t GpuMat::elemSize1() const { return CV_ELEM_SIZE1(flags); }
|
||||
inline int GpuMat::type() const { return CV_MAT_TYPE(flags); }
|
||||
inline int GpuMat::depth() const { return CV_MAT_DEPTH(flags); }
|
||||
inline int GpuMat::channels() const { return CV_MAT_CN(flags); }
|
||||
inline size_t GpuMat::step1() const { return step/elemSize1(); }
|
||||
inline Size GpuMat::size() const { return Size(cols, rows); }
|
||||
inline bool GpuMat::empty() const { return data == 0; }
|
||||
|
||||
inline uchar* GpuMat::ptr(int y)
|
||||
{
|
||||
CV_DbgAssert( (unsigned)y < (unsigned)rows );
|
||||
return data + step*y;
|
||||
}
|
||||
|
||||
inline const uchar* GpuMat::ptr(int y) const
|
||||
{
|
||||
CV_DbgAssert( (unsigned)y < (unsigned)rows );
|
||||
return data + step*y;
|
||||
}
|
||||
|
||||
template<typename _Tp> inline _Tp* GpuMat::ptr(int y)
|
||||
{
|
||||
CV_DbgAssert( (unsigned)y < (unsigned)rows );
|
||||
return (_Tp*)(data + step*y);
|
||||
}
|
||||
|
||||
template<typename _Tp> inline const _Tp* GpuMat::ptr(int y) const
|
||||
{
|
||||
CV_DbgAssert( (unsigned)y < (unsigned)rows );
|
||||
return (const _Tp*)(data + step*y);
|
||||
}
|
||||
|
||||
static inline void swap( GpuMat& a, GpuMat& b ) { a.swap(b); }
|
||||
|
||||
|
||||
} /* end of namespace gpu */
|
||||
|
||||
} /* end of namespace cv */
|
||||
|
||||
#endif /* __OPENCV_GPU_MATRIX_OPERATIONS_HPP__ */
|
265
modules/gpu/include/opencv2/gpu/matpl.hpp
Normal file
265
modules/gpu/include/opencv2/gpu/matpl.hpp
Normal file
@@ -0,0 +1,265 @@
|
||||
/*M///////////////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
|
||||
//
|
||||
// By downloading, copying, installing or using the software you agree to this license.
|
||||
// If you do not agree to this license, do not download, install,
|
||||
// copy or use the software.
|
||||
//
|
||||
//
|
||||
// License Agreement
|
||||
// For Open Source Computer Vision Library
|
||||
//
|
||||
// Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
|
||||
// Copyright (C) 2009, Willow Garage Inc., all rights reserved.
|
||||
// Third party copyrights are property of their respective owners.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without modification,
|
||||
// are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistribution's of source code must retain the above copyright notice,
|
||||
// this list of conditions and the following disclaimer.
|
||||
//
|
||||
// * Redistribution's in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other GpuMaterials provided with the distribution.
|
||||
//
|
||||
// * The name of the copyright holders may not be used to endorse or promote products
|
||||
// derived from this software without specific prior written permission.
|
||||
//
|
||||
// This software is provided by the copyright holders and contributors "as is" and
|
||||
// any express or implied warranties, including, but not limited to, the implied
|
||||
// warranties of merchantability and fitness for a particular purpose are disclaimed.
|
||||
// In no event shall the Intel Corporation or contributors be liable for any direct,
|
||||
// indirect, incidental, special, exemplary, or consequential damages
|
||||
// (including, but not limited to, procurement of substitute goods or services;
|
||||
// loss of use, data, or profits; or business interruption) however caused
|
||||
// and on any theory of liability, whether in contract, strict liability,
|
||||
// or tort (including negligence or otherwise) arising in any way out of
|
||||
// the use of this software, even if advised of the possibility of such damage.
|
||||
//
|
||||
//M*/
|
||||
|
||||
#ifndef __OPENCV_GPU_MATPL_HPP__
|
||||
#define __OPENCV_GPU_MATPL_HPP__
|
||||
|
||||
#include "opencv2/core/core.hpp"
|
||||
|
||||
namespace cv
|
||||
{
|
||||
namespace gpu
|
||||
{
|
||||
|
||||
//////////////////////////////// MatPL ////////////////////////////////
|
||||
|
||||
//class CV_EXPORTS MatPL : private Mat
|
||||
//{
|
||||
//public:
|
||||
// MatPL() {}
|
||||
// MatPL(int _rows, int _cols, int _type) : Mat(_rows, _cols, _type) {}
|
||||
// MatPL(Size _size, int _type) : Mat(_size, _type) {}
|
||||
//
|
||||
// Mat(int _rows, int _cols, int _type, const Scalar& _s) : Mat
|
||||
// MatPL(Size _size, int _type, const Scalar& _s);
|
||||
// //! copy constructor
|
||||
// MatPL(const Mat& m);
|
||||
// //! constructor for matrix headers pointing to user-allocated data
|
||||
// MatPL(int _rows, int _cols, int _type, void* _data, size_t _step=AUTO_STEP);
|
||||
// MatPL(Size _size, int _type, void* _data, size_t _step=AUTO_STEP);
|
||||
// //! creates a matrix header for a part of the bigger matrix
|
||||
// MatPL(const Mat& m, const Range& rowRange, const Range& colRange);
|
||||
// MatPL(const Mat& m, const Rect& roi);
|
||||
// //! converts old-style CvMat to the new matrix; the data is not copied by default
|
||||
// Mat(const CvMat* m, bool copyData=false);
|
||||
// MatPL converts old-style IplImage to the new matrix; the data is not copied by default
|
||||
// MatPL(const IplImage* img, bool copyData=false);
|
||||
// //! builds matrix from std::vector with or without copying the data
|
||||
// template<typename _Tp> explicit Mat(const vector<_Tp>& vec, bool copyData=false);
|
||||
// //! builds matrix from cv::Vec; the data is copied by default
|
||||
// template<typename _Tp, int n> explicit Mat(const Vec<_Tp, n>& vec,
|
||||
// bool copyData=true);
|
||||
// //! builds matrix from cv::Matx; the data is copied by default
|
||||
// template<typename _Tp, int m, int n> explicit Mat(const Matx<_Tp, m, n>& mtx,
|
||||
// bool copyData=true);
|
||||
// //! builds matrix from a 2D point
|
||||
// template<typename _Tp> explicit Mat(const Point_<_Tp>& pt);
|
||||
// //! builds matrix from a 3D point
|
||||
// template<typename _Tp> explicit Mat(const Point3_<_Tp>& pt);
|
||||
// //! builds matrix from comma initializer
|
||||
// template<typename _Tp> explicit Mat(const MatCommaInitializer_<_Tp>& commaInitializer);
|
||||
// //! helper constructor to compile matrix expressions
|
||||
// Mat(const MatExpr_Base& expr);
|
||||
// //! destructor - calls release()
|
||||
// ~Mat();
|
||||
// //! assignment operators
|
||||
// Mat& operator = (const Mat& m);
|
||||
// Mat& operator = (const MatExpr_Base& expr);
|
||||
|
||||
// operator MatExpr_<Mat, Mat>() const;
|
||||
|
||||
// //! returns a new matrix header for the specified row
|
||||
// Mat row(int y) const;
|
||||
// //! returns a new matrix header for the specified column
|
||||
// Mat col(int x) const;
|
||||
// //! ... for the specified row span
|
||||
// Mat rowRange(int startrow, int endrow) const;
|
||||
// Mat rowRange(const Range& r) const;
|
||||
// //! ... for the specified column span
|
||||
// Mat colRange(int startcol, int endcol) const;
|
||||
// Mat colRange(const Range& r) const;
|
||||
// //! ... for the specified diagonal
|
||||
// // (d=0 - the main diagonal,
|
||||
// // >0 - a diagonal from the lower half,
|
||||
// // <0 - a diagonal from the upper half)
|
||||
// Mat diag(int d=0) const;
|
||||
// //! constructs a square diagonal matrix which main diagonal is vector "d"
|
||||
// static Mat diag(const Mat& d);
|
||||
|
||||
// //! returns deep copy of the matrix, i.e. the data is copied
|
||||
// Mat clone() const;
|
||||
// //! copies the matrix content to "m".
|
||||
// // It calls m.create(this->size(), this->type()).
|
||||
// void copyTo( Mat& m ) const;
|
||||
// //! copies those matrix elements to "m" that are marked with non-zero mask elements.
|
||||
// void copyTo( Mat& m, const Mat& mask ) const;
|
||||
// //! converts matrix to another datatype with optional scalng. See cvConvertScale.
|
||||
// void convertTo( Mat& m, int rtype, double alpha=1, double beta=0 ) const;
|
||||
|
||||
// void assignTo( Mat& m, int type=-1 ) const;
|
||||
|
||||
// //! sets every matrix element to s
|
||||
// Mat& operator = (const Scalar& s);
|
||||
// //! sets some of the matrix elements to s, according to the mask
|
||||
// Mat& setTo(const Scalar& s, const Mat& mask=Mat());
|
||||
// //! creates alternative matrix header for the same data, with different
|
||||
// // number of channels and/or different number of rows. see cvReshape.
|
||||
// Mat reshape(int _cn, int _rows=0) const;
|
||||
|
||||
// //! matrix transposition by means of matrix expressions
|
||||
// MatExpr_<MatExpr_Op2_<Mat, double, Mat, MatOp_T_<Mat> >, Mat>
|
||||
// t() const;
|
||||
// //! matrix inversion by means of matrix expressions
|
||||
// MatExpr_<MatExpr_Op2_<Mat, int, Mat, MatOp_Inv_<Mat> >, Mat>
|
||||
// inv(int method=DECOMP_LU) const;
|
||||
// MatExpr_<MatExpr_Op4_<Mat, Mat, double, char, Mat, MatOp_MulDiv_<Mat> >, Mat>
|
||||
// //! per-element matrix multiplication by means of matrix expressions
|
||||
// mul(const Mat& m, double scale=1) const;
|
||||
// MatExpr_<MatExpr_Op4_<Mat, Mat, double, char, Mat, MatOp_MulDiv_<Mat> >, Mat>
|
||||
// mul(const MatExpr_<MatExpr_Op2_<Mat, double, Mat, MatOp_Scale_<Mat> >, Mat>& m, double scale=1) const;
|
||||
// MatExpr_<MatExpr_Op4_<Mat, Mat, double, char, Mat, MatOp_MulDiv_<Mat> >, Mat>
|
||||
// mul(const MatExpr_<MatExpr_Op2_<Mat, double, Mat, MatOp_DivRS_<Mat> >, Mat>& m, double scale=1) const;
|
||||
|
||||
// //! computes cross-product of 2 3D vectors
|
||||
// Mat cross(const Mat& m) const;
|
||||
// //! computes dot-product
|
||||
// double dot(const Mat& m) const;
|
||||
|
||||
// //! Matlab-style matrix initialization
|
||||
// static MatExpr_Initializer zeros(int rows, int cols, int type);
|
||||
// static MatExpr_Initializer zeros(Size size, int type);
|
||||
// static MatExpr_Initializer ones(int rows, int cols, int type);
|
||||
// static MatExpr_Initializer ones(Size size, int type);
|
||||
// static MatExpr_Initializer eye(int rows, int cols, int type);
|
||||
// static MatExpr_Initializer eye(Size size, int type);
|
||||
|
||||
// //! allocates new matrix data unless the matrix already has specified size and type.
|
||||
// // previous data is unreferenced if needed.
|
||||
// void create(int _rows, int _cols, int _type);
|
||||
// void create(Size _size, int _type);
|
||||
// //! increases the reference counter; use with care to avoid memleaks
|
||||
// void addref();
|
||||
// //! decreases reference counter;
|
||||
// // deallocate the data when reference counter reaches 0.
|
||||
// void release();
|
||||
|
||||
// //! locates matrix header within a parent matrix. See below
|
||||
// void locateROI( Size& wholeSize, Point& ofs ) const;
|
||||
// //! moves/resizes the current matrix ROI inside the parent matrix.
|
||||
// Mat& adjustROI( int dtop, int dbottom, int dleft, int dright );
|
||||
// //! extracts a rectangular sub-matrix
|
||||
// // (this is a generalized form of row, rowRange etc.)
|
||||
// Mat operator()( Range rowRange, Range colRange ) const;
|
||||
// Mat operator()( const Rect& roi ) const;
|
||||
|
||||
// //! converts header to CvMat; no data is copied
|
||||
// operator CvMat() const;
|
||||
// //! converts header to IplImage; no data is copied
|
||||
// operator IplImage() const;
|
||||
|
||||
// //! returns true iff the matrix data is continuous
|
||||
// // (i.e. when there are no gaps between successive rows).
|
||||
// // similar to CV_IS_MAT_CONT(cvmat->type)
|
||||
// bool isContinuous() const;
|
||||
// //! returns element size in bytes,
|
||||
// // similar to CV_ELEM_SIZE(cvmat->type)
|
||||
// size_t elemSize() const;
|
||||
// //! returns the size of element channel in bytes.
|
||||
// size_t elemSize1() const;
|
||||
// //! returns element type, similar to CV_MAT_TYPE(cvmat->type)
|
||||
// int type() const;
|
||||
// //! returns element type, similar to CV_MAT_DEPTH(cvmat->type)
|
||||
// int depth() const;
|
||||
// //! returns element type, similar to CV_MAT_CN(cvmat->type)
|
||||
// int channels() const;
|
||||
// //! returns step/elemSize1()
|
||||
// size_t step1() const;
|
||||
// //! returns matrix size:
|
||||
// // width == number of columns, height == number of rows
|
||||
// Size size() const;
|
||||
// //! returns true if matrix data is NULL
|
||||
// bool empty() const;
|
||||
|
||||
// //! returns pointer to y-th row
|
||||
// uchar* ptr(int y=0);
|
||||
// const uchar* ptr(int y=0) const;
|
||||
|
||||
// //! template version of the above method
|
||||
// template<typename _Tp> _Tp* ptr(int y=0);
|
||||
// template<typename _Tp> const _Tp* ptr(int y=0) const;
|
||||
|
||||
// //! template methods for read-write or read-only element access.
|
||||
// // note that _Tp must match the actual matrix type -
|
||||
// // the functions do not do any on-fly type conversion
|
||||
// template<typename _Tp> _Tp& at(int y, int x);
|
||||
// template<typename _Tp> _Tp& at(Point pt);
|
||||
// template<typename _Tp> const _Tp& at(int y, int x) const;
|
||||
// template<typename _Tp> const _Tp& at(Point pt) const;
|
||||
// template<typename _Tp> _Tp& at(int i);
|
||||
// template<typename _Tp> const _Tp& at(int i) const;
|
||||
|
||||
// //! template methods for iteration over matrix elements.
|
||||
// // the iterators take care of skipping gaps in the end of rows (if any)
|
||||
// template<typename _Tp> MatIterator_<_Tp> begin();
|
||||
// template<typename _Tp> MatIterator_<_Tp> end();
|
||||
// template<typename _Tp> MatConstIterator_<_Tp> begin() const;
|
||||
// template<typename _Tp> MatConstIterator_<_Tp> end() const;
|
||||
|
||||
// enum { MAGIC_VAL=0x42FF0000, AUTO_STEP=0, CONTINUOUS_FLAG=CV_MAT_CONT_FLAG };
|
||||
|
||||
// /*! includes several bit-fields:
|
||||
// - the magic signature
|
||||
// - continuity flag
|
||||
// - depth
|
||||
// - number of channels
|
||||
// */
|
||||
// int flags;
|
||||
// //! the number of rows and columns
|
||||
// int rows, cols;
|
||||
// //! a distance between successive rows in bytes; includes the gap if any
|
||||
// size_t step;
|
||||
// //! pointer to the data
|
||||
// uchar* data;
|
||||
|
||||
// //! pointer to the reference counter;
|
||||
// // when matrix points to user-allocated data, the pointer is NULL
|
||||
// int* refcount;
|
||||
|
||||
// //! helper fields used in locateROI and adjustROI
|
||||
// uchar* datastart;
|
||||
// uchar* dataend;
|
||||
//};
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
#endif /* __OPENCV_GPU_MATPL_HPP__ */
|
Reference in New Issue
Block a user