2012-07-17 01:08:14 +08:00
|
|
|
/*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) 2010-2012, Institute Of Software Chinese Academy Of Science, all rights reserved.
|
|
|
|
// Copyright (C) 2010-2012, Advanced Micro Devices, Inc., all rights reserved.
|
2013-02-08 11:41:46 +08:00
|
|
|
// Copyright (C) 2010-2012, Multicoreware, Inc., all rights reserved.
|
2012-07-17 01:08:14 +08:00
|
|
|
// Third party copyrights are property of their respective owners.
|
|
|
|
//
|
|
|
|
// @Authors
|
|
|
|
// Niko Li, newlife20080214@gmail.com
|
|
|
|
// Jia Haipeng, jiahaipeng95@gmail.com
|
|
|
|
// Shengen Yan, yanshengen@gmail.com
|
|
|
|
// Jiang Liyuan, jlyuan001.good@163.com
|
|
|
|
// Rock Li, Rock.Li@amd.com
|
|
|
|
// Zailong Wu, bullet@yeah.net
|
2013-05-17 15:34:22 +08:00
|
|
|
// Peng Xiao, pengxiao@outlook.com
|
2012-07-17 01:08:14 +08:00
|
|
|
//
|
|
|
|
// 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 oclMaterials 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*/
|
|
|
|
|
|
|
|
#include "precomp.hpp"
|
|
|
|
#include <iomanip>
|
|
|
|
|
|
|
|
|
|
|
|
using namespace cv;
|
|
|
|
using namespace cv::ocl;
|
|
|
|
using namespace std;
|
|
|
|
|
|
|
|
namespace cv
|
|
|
|
{
|
|
|
|
namespace ocl
|
|
|
|
{
|
2013-09-24 17:34:55 +08:00
|
|
|
//////////////////////////////// OpenCL kernel strings /////////////////////
|
|
|
|
|
2012-07-17 01:08:14 +08:00
|
|
|
extern const char *transpose_kernel;
|
|
|
|
extern const char *arithm_nonzero;
|
|
|
|
extern const char *arithm_sum;
|
|
|
|
extern const char *arithm_sum_3;
|
|
|
|
extern const char *arithm_minMax;
|
|
|
|
extern const char *arithm_minMax_mask;
|
|
|
|
extern const char *arithm_minMaxLoc;
|
|
|
|
extern const char *arithm_minMaxLoc_mask;
|
|
|
|
extern const char *arithm_LUT;
|
|
|
|
extern const char *arithm_add;
|
2013-09-24 17:34:55 +08:00
|
|
|
extern const char *arithm_add_mask;
|
2012-07-17 01:08:14 +08:00
|
|
|
extern const char *arithm_add_scalar;
|
|
|
|
extern const char *arithm_add_scalar_mask;
|
2013-04-15 10:36:03 +08:00
|
|
|
extern const char *arithm_bitwise_binary;
|
|
|
|
extern const char *arithm_bitwise_binary_mask;
|
|
|
|
extern const char *arithm_bitwise_binary_scalar;
|
|
|
|
extern const char *arithm_bitwise_binary_scalar_mask;
|
2012-07-17 01:08:14 +08:00
|
|
|
extern const char *arithm_bitwise_not;
|
|
|
|
extern const char *arithm_compare_eq;
|
|
|
|
extern const char *arithm_compare_ne;
|
|
|
|
extern const char *arithm_transpose;
|
|
|
|
extern const char *arithm_flip;
|
|
|
|
extern const char *arithm_flip_rc;
|
|
|
|
extern const char *arithm_magnitude;
|
|
|
|
extern const char *arithm_cartToPolar;
|
|
|
|
extern const char *arithm_polarToCart;
|
|
|
|
extern const char *arithm_exp;
|
|
|
|
extern const char *arithm_log;
|
|
|
|
extern const char *arithm_addWeighted;
|
|
|
|
extern const char *arithm_phase;
|
|
|
|
extern const char *arithm_pow;
|
2013-09-09 16:08:23 +08:00
|
|
|
extern const char *arithm_setidentity;
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
/////////////////////// add subtract multiply divide /////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
/////////////////////// add subtract multiply divide /////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
enum { ADD = 0, SUB, MUL, DIV, ABS_DIFF };
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
static void arithmetic_run_generic(const oclMat &src1, const oclMat &src2, const Scalar & scalar, const oclMat & mask,
|
|
|
|
oclMat &dst, int op_type, bool use_scalar = false)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-09-24 17:34:55 +08:00
|
|
|
Context *clCxt = src1.clCxt;
|
|
|
|
bool hasDouble = clCxt->supportsFeature(Context::CL_DOUBLE);
|
|
|
|
if (!hasDouble && (src1.depth() == CV_64F || src2.depth() == CV_64F || dst.depth() == CV_64F))
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-09-24 17:34:55 +08:00
|
|
|
CV_Error(CV_GpuNotSupported, "Selected device doesn't support double\r\n");
|
2012-07-17 01:08:14 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
CV_Assert(src2.empty() || (!src2.empty() && src1.type() == src2.type() && src1.size() == src2.size()));
|
|
|
|
CV_Assert(mask.empty() || (!mask.empty() && mask.type() == CV_8UC1 && mask.size() == src1.size()));
|
|
|
|
CV_Assert(op_type >= ADD && op_type <= ABS_DIFF);
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
dst.create(src1.size(), src1.type());
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
int oclChannels = src1.oclchannels(), depth = src1.depth();
|
|
|
|
int src1step1 = src1.step / src1.elemSize(), src1offset1 = src1.offset / src1.elemSize();
|
|
|
|
int src2step1 = src2.step / src2.elemSize(), src2offset1 = src2.offset / src2.elemSize();
|
|
|
|
int maskstep1 = mask.step, maskoffset1 = mask.offset / mask.elemSize();
|
|
|
|
int dststep1 = dst.step / dst.elemSize(), dstoffset1 = dst.offset / dst.elemSize();
|
|
|
|
oclMat m;
|
|
|
|
|
|
|
|
size_t localThreads[3] = { 16, 16, 1 };
|
|
|
|
size_t globalThreads[3] = { dst.cols, dst.rows, 1 };
|
|
|
|
|
|
|
|
std::string kernelName = op_type == ABS_DIFF ? "arithm_absdiff" : "arithm_binary_op";
|
|
|
|
|
|
|
|
const char * const typeMap[] = { "uchar", "char", "ushort", "short", "int", "float", "double" };
|
|
|
|
const char * const WTypeMap[] = { "short", "short", "int", "int", "int", "float", "double" };
|
|
|
|
const char operationsMap[] = { '+', '-', '*', '/', '-' };
|
|
|
|
const char * const channelMap[] = { "", "", "2", "4", "4" };
|
|
|
|
bool haveScalar = use_scalar || src2.empty();
|
|
|
|
|
|
|
|
int WDepth = depth;
|
|
|
|
if (haveScalar)
|
|
|
|
WDepth = hasDouble && WDepth == CV_64F ? CV_64F : CV_32F;
|
|
|
|
if (op_type == DIV)
|
|
|
|
WDepth = hasDouble ? CV_64F : CV_32F;
|
|
|
|
else if (op_type == MUL)
|
|
|
|
WDepth = hasDouble && (depth == CV_32S || depth == CV_64F) ? CV_64F : CV_32F;
|
|
|
|
|
|
|
|
std::string buildOptions = format("-D T=%s%s -D WT=%s%s -D convertToT=convert_%s%s%s -D Operation=%c"
|
|
|
|
" -D convertToWT=convert_%s%s",
|
|
|
|
typeMap[depth], channelMap[oclChannels],
|
|
|
|
WTypeMap[WDepth], channelMap[oclChannels],
|
|
|
|
typeMap[depth], channelMap[oclChannels], (depth >= CV_32F ? "" : (depth == CV_32S ? "_rte" : "_sat_rte")),
|
|
|
|
operationsMap[op_type], WTypeMap[WDepth], channelMap[oclChannels]);
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src1.data ));
|
2013-09-24 17:34:55 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1step1 ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1offset1 ));
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
if (!src2.empty())
|
2013-04-06 13:37:36 +08:00
|
|
|
{
|
2013-09-24 17:34:55 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src2.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2step1 ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2offset1 ));
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
kernelName += "_mat";
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
if (haveScalar)
|
2013-01-25 18:31:34 +08:00
|
|
|
{
|
2013-09-24 17:34:55 +08:00
|
|
|
const int WDepthMap[] = { CV_16S, CV_16S, CV_32S, CV_32S, CV_32S, CV_32F, CV_64F };
|
|
|
|
m.create(1, 1, CV_MAKE_TYPE(WDepthMap[WDepth], oclChannels));
|
|
|
|
m.setTo(scalar);
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&m.data ));
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
kernelName += "_scalar";
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
if (!mask.empty())
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-09-24 17:34:55 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&mask.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&maskstep1 ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&maskoffset1 ));
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
kernelName += "_mask";
|
2012-08-03 14:08:36 +08:00
|
|
|
}
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
if (op_type == DIV)
|
|
|
|
kernelName += "_div";
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dststep1 ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dstoffset1 ));
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.rows ));
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
openCLExecuteKernel(clCxt, mask.empty() ?
|
|
|
|
(!src2.empty() ? &arithm_add : &arithm_add_scalar) :
|
|
|
|
(!src2.empty() ? &arithm_add_mask : &arithm_add_scalar_mask),
|
|
|
|
kernelName, globalThreads, localThreads,
|
|
|
|
args, -1, -1, buildOptions.c_str());
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
2013-09-24 17:34:55 +08:00
|
|
|
|
|
|
|
void cv::ocl::add(const oclMat &src1, const oclMat &src2, oclMat &dst, const oclMat &mask)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-09-24 17:34:55 +08:00
|
|
|
arithmetic_run_generic(src1, src2, Scalar(), mask, dst, ADD);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void cv::ocl::add(const oclMat &src1, const Scalar &src2, oclMat &dst, const oclMat &mask)
|
|
|
|
{
|
2013-09-24 17:34:55 +08:00
|
|
|
arithmetic_run_generic(src1, oclMat(), src2, mask, dst, ADD);
|
|
|
|
}
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
void cv::ocl::subtract(const oclMat &src1, const oclMat &src2, oclMat &dst, const oclMat &mask)
|
|
|
|
{
|
|
|
|
arithmetic_run_generic(src1, src2, Scalar(), mask, dst, SUB);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void cv::ocl::subtract(const oclMat &src1, const Scalar &src2, oclMat &dst, const oclMat &mask)
|
|
|
|
{
|
2013-09-24 17:34:55 +08:00
|
|
|
arithmetic_run_generic(src1, oclMat(), src2, mask, dst, SUB);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
2013-09-24 17:34:55 +08:00
|
|
|
|
|
|
|
void cv::ocl::multiply(const oclMat &src1, const oclMat &src2, oclMat &dst, double scalar)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-09-24 17:34:55 +08:00
|
|
|
const bool use_scalar = !(std::abs(scalar - 1.0) < std::numeric_limits<double>::epsilon());
|
|
|
|
arithmetic_run_generic(src1, src2, Scalar::all(scalar), oclMat(), dst, MUL, use_scalar);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
2013-09-24 17:34:55 +08:00
|
|
|
|
2013-05-17 15:34:22 +08:00
|
|
|
void cv::ocl::multiply(double scalar, const oclMat &src, oclMat &dst)
|
|
|
|
{
|
2013-09-24 17:34:55 +08:00
|
|
|
arithmetic_run_generic(src, oclMat(), Scalar::all(scalar), oclMat(), dst, MUL);
|
2013-05-17 15:34:22 +08:00
|
|
|
}
|
2013-09-24 17:34:55 +08:00
|
|
|
|
|
|
|
void cv::ocl::divide(const oclMat &src1, const oclMat &src2, oclMat &dst, double scalar)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-09-24 17:34:55 +08:00
|
|
|
const bool use_scalar = !(std::abs(scalar - 1.0) < std::numeric_limits<double>::epsilon());
|
|
|
|
arithmetic_run_generic(src1, src2, Scalar::all(scalar), oclMat(), dst, DIV, use_scalar);
|
|
|
|
}
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:34:55 +08:00
|
|
|
void cv::ocl::divide(double scalar, const oclMat &src, oclMat &dst)
|
|
|
|
{
|
|
|
|
arithmetic_run_generic(src, oclMat(), Scalar::all(scalar), oclMat(), dst, DIV);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
2013-09-24 17:34:55 +08:00
|
|
|
|
2012-07-17 01:08:14 +08:00
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////// Absdiff ///////////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2013-09-24 17:34:55 +08:00
|
|
|
|
2012-07-17 01:08:14 +08:00
|
|
|
void cv::ocl::absdiff(const oclMat &src1, const oclMat &src2, oclMat &dst)
|
|
|
|
{
|
2013-09-24 17:34:55 +08:00
|
|
|
arithmetic_run_generic(src1, src2, Scalar(), oclMat(), dst, ABS_DIFF);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
2013-09-24 17:34:55 +08:00
|
|
|
|
2012-07-17 01:08:14 +08:00
|
|
|
void cv::ocl::absdiff(const oclMat &src1, const Scalar &src2, oclMat &dst)
|
|
|
|
{
|
2013-09-24 17:34:55 +08:00
|
|
|
arithmetic_run_generic(src1, oclMat(), src2, oclMat(), dst, ABS_DIFF);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
2013-09-24 17:34:55 +08:00
|
|
|
|
2012-07-17 01:08:14 +08:00
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////////// compare ///////////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2013-01-25 18:31:34 +08:00
|
|
|
static void compare_run(const oclMat &src1, const oclMat &src2, oclMat &dst, string kernelName, const char **kernelString)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
dst.create(src1.size(), CV_8UC1);
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Assert(src1.oclchannels() == 1);
|
2012-07-17 01:08:14 +08:00
|
|
|
CV_Assert(src1.type() == src2.type());
|
|
|
|
Context *clCxt = src1.clCxt;
|
|
|
|
int depth = src1.depth();
|
|
|
|
int vector_lengths[7] = {4, 0, 4, 4, 4, 4, 4};
|
|
|
|
size_t vector_length = vector_lengths[depth];
|
|
|
|
int offset_cols = (dst.offset / dst.elemSize1()) & (vector_length - 1);
|
|
|
|
int cols = divUp(dst.cols + offset_cols, vector_length);
|
|
|
|
size_t localThreads[3] = { 64, 4, 1 };
|
2013-09-16 19:11:56 +08:00
|
|
|
size_t globalThreads[3] = { cols, dst.rows, 1 };
|
|
|
|
|
2012-07-17 01:08:14 +08:00
|
|
|
int dst_step1 = dst.cols * dst.elemSize();
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src1.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src2.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.rows ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst_step1 ));
|
|
|
|
openCLExecuteKernel(clCxt, kernelString, kernelName, globalThreads, localThreads, args, -1, depth);
|
|
|
|
}
|
|
|
|
|
|
|
|
void cv::ocl::compare(const oclMat &src1, const oclMat &src2, oclMat &dst , int cmpOp)
|
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src1.clCxt->supportsFeature(Context::CL_DOUBLE) && src1.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
cout << "Selected device do not support double" << endl;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
string kernelName;
|
|
|
|
const char **kernelString = NULL;
|
|
|
|
switch( cmpOp )
|
|
|
|
{
|
|
|
|
case CMP_EQ:
|
|
|
|
kernelName = "arithm_compare_eq";
|
|
|
|
kernelString = &arithm_compare_eq;
|
|
|
|
break;
|
|
|
|
case CMP_GT:
|
|
|
|
kernelName = "arithm_compare_gt";
|
|
|
|
kernelString = &arithm_compare_eq;
|
|
|
|
break;
|
|
|
|
case CMP_GE:
|
|
|
|
kernelName = "arithm_compare_ge";
|
|
|
|
kernelString = &arithm_compare_eq;
|
|
|
|
break;
|
|
|
|
case CMP_NE:
|
|
|
|
kernelName = "arithm_compare_ne";
|
|
|
|
kernelString = &arithm_compare_ne;
|
|
|
|
break;
|
|
|
|
case CMP_LT:
|
|
|
|
kernelName = "arithm_compare_lt";
|
|
|
|
kernelString = &arithm_compare_ne;
|
|
|
|
break;
|
|
|
|
case CMP_LE:
|
|
|
|
kernelName = "arithm_compare_le";
|
|
|
|
kernelString = &arithm_compare_ne;
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
CV_Error(CV_StsBadArg, "Unknown comparison method");
|
|
|
|
}
|
|
|
|
compare_run(src1, src2, dst, kernelName, kernelString);
|
|
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
////////////////////////////////// sum //////////////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
|
|
|
|
//type = 0 sum,type = 1 absSum,type = 2 sqrSum
|
2013-01-25 18:31:34 +08:00
|
|
|
static void arithmetic_sum_buffer_run(const oclMat &src, cl_mem &dst, int vlen , int groupnum, int type = 0)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
int all_cols = src.step / (vlen * src.elemSize1());
|
|
|
|
int pre_cols = (src.offset % src.step) / (vlen * src.elemSize1());
|
|
|
|
int sec_cols = all_cols - (src.offset % src.step + src.cols * src.elemSize() - 1) / (vlen * src.elemSize1()) - 1;
|
|
|
|
int invalid_cols = pre_cols + sec_cols;
|
|
|
|
int cols = all_cols - invalid_cols , elemnum = cols * src.rows;;
|
|
|
|
int offset = src.offset / (vlen * src.elemSize1());
|
|
|
|
int repeat_s = src.offset / src.elemSize1() - offset * vlen;
|
2012-10-11 16:22:47 +08:00
|
|
|
int repeat_e = (offset + cols) * vlen - src.offset / src.elemSize1() - src.cols * src.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
char build_options[512];
|
|
|
|
CV_Assert(type == 0 || type == 1 || type == 2);
|
|
|
|
sprintf(build_options, "-D DEPTH_%d -D REPEAT_S%d -D REPEAT_E%d -D FUNC_TYPE_%d", src.depth(), repeat_s, repeat_e, type);
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&invalid_cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&offset));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&elemnum));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&groupnum));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&src.data));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&dst ));
|
|
|
|
size_t gt[3] = {groupnum * 256, 1, 1}, lt[3] = {256, 1, 1};
|
2012-10-11 16:22:47 +08:00
|
|
|
if(src.oclchannels() != 3)
|
2012-07-17 01:08:14 +08:00
|
|
|
openCLExecuteKernel(src.clCxt, &arithm_sum, "arithm_op_sum", gt, lt, args, -1, -1, build_options);
|
|
|
|
else
|
|
|
|
openCLExecuteKernel(src.clCxt, &arithm_sum_3, "arithm_op_sum_3", gt, lt, args, -1, -1, build_options);
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename T>
|
2012-10-11 16:22:47 +08:00
|
|
|
Scalar arithmetic_sum(const oclMat &src, int type = 0)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
size_t groupnum = src.clCxt->computeUnits();
|
2012-07-17 01:08:14 +08:00
|
|
|
CV_Assert(groupnum != 0);
|
2012-10-22 15:14:22 +08:00
|
|
|
int vlen = src.oclchannels() == 3 ? 12 : 8, dbsize = groupnum * vlen;
|
2012-07-17 01:08:14 +08:00
|
|
|
Context *clCxt = src.clCxt;
|
|
|
|
T *p = new T[dbsize];
|
2012-10-11 16:22:47 +08:00
|
|
|
cl_mem dstBuffer = openCLCreateBuffer(clCxt, CL_MEM_WRITE_ONLY, dbsize * sizeof(T));
|
2012-07-17 01:08:14 +08:00
|
|
|
Scalar s;
|
|
|
|
s.val[0] = 0.0;
|
|
|
|
s.val[1] = 0.0;
|
|
|
|
s.val[2] = 0.0;
|
|
|
|
s.val[3] = 0.0;
|
2012-10-11 16:22:47 +08:00
|
|
|
arithmetic_sum_buffer_run(src, dstBuffer, vlen, groupnum, type);
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
memset(p, 0, dbsize * sizeof(T));
|
2012-10-11 16:22:47 +08:00
|
|
|
openCLReadBuffer(clCxt, dstBuffer, (void *)p, dbsize * sizeof(T));
|
2012-07-17 01:08:14 +08:00
|
|
|
for(int i = 0; i < dbsize;)
|
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
for(int j = 0; j < src.oclchannels(); j++, i++)
|
2012-07-17 01:08:14 +08:00
|
|
|
s.val[j] += p[i];
|
|
|
|
}
|
|
|
|
delete[] p;
|
|
|
|
openCLFree(dstBuffer);
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
2012-10-11 16:22:47 +08:00
|
|
|
typedef Scalar (*sumFunc)(const oclMat &src, int type);
|
2012-07-17 01:08:14 +08:00
|
|
|
Scalar cv::ocl::sum(const oclMat &src)
|
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src.clCxt->supportsFeature(Context::CL_DOUBLE) && src.depth() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Error(CV_GpuNotSupported, "select device don't support double");
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
static sumFunc functab[2] =
|
|
|
|
{
|
|
|
|
arithmetic_sum<float>,
|
|
|
|
arithmetic_sum<double>
|
|
|
|
};
|
|
|
|
|
|
|
|
sumFunc func;
|
2013-03-18 05:59:24 +08:00
|
|
|
func = functab[(int)src.clCxt->supportsFeature(Context::CL_DOUBLE)];
|
2012-10-11 16:22:47 +08:00
|
|
|
return func(src, 0);
|
|
|
|
}
|
|
|
|
|
2012-10-26 17:43:59 +08:00
|
|
|
Scalar cv::ocl::absSum(const oclMat &src)
|
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src.clCxt->supportsFeature(Context::CL_DOUBLE) && src.depth() == CV_64F)
|
2012-10-26 17:43:59 +08:00
|
|
|
{
|
|
|
|
CV_Error(CV_GpuNotSupported, "select device don't support double");
|
|
|
|
}
|
|
|
|
static sumFunc functab[2] =
|
|
|
|
{
|
|
|
|
arithmetic_sum<float>,
|
|
|
|
arithmetic_sum<double>
|
|
|
|
};
|
|
|
|
|
|
|
|
sumFunc func;
|
2013-03-18 05:59:24 +08:00
|
|
|
func = functab[(int)src.clCxt->supportsFeature(Context::CL_DOUBLE)];
|
2012-10-26 17:43:59 +08:00
|
|
|
return func(src, 1);
|
|
|
|
}
|
2012-10-11 16:22:47 +08:00
|
|
|
|
|
|
|
Scalar cv::ocl::sqrSum(const oclMat &src)
|
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src.clCxt->supportsFeature(Context::CL_DOUBLE) && src.depth() == CV_64F)
|
2012-10-11 16:22:47 +08:00
|
|
|
{
|
|
|
|
CV_Error(CV_GpuNotSupported, "select device don't support double");
|
|
|
|
}
|
|
|
|
static sumFunc functab[2] =
|
|
|
|
{
|
|
|
|
arithmetic_sum<float>,
|
|
|
|
arithmetic_sum<double>
|
|
|
|
};
|
|
|
|
|
|
|
|
sumFunc func;
|
2013-03-18 05:59:24 +08:00
|
|
|
func = functab[(int)src.clCxt->supportsFeature(Context::CL_DOUBLE)];
|
2012-10-11 16:22:47 +08:00
|
|
|
return func(src, 2);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
//////////////////////////////// meanStdDev //////////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
void cv::ocl::meanStdDev(const oclMat &src, Scalar &mean, Scalar &stddev)
|
|
|
|
{
|
|
|
|
CV_Assert(src.depth() <= CV_32S);
|
|
|
|
cv::Size sz(1, 1);
|
2012-10-11 16:22:47 +08:00
|
|
|
int channels = src.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
Mat m1(sz, CV_MAKETYPE(CV_32S, channels), cv::Scalar::all(0)),
|
|
|
|
m2(sz, CV_MAKETYPE(CV_32S, channels), cv::Scalar::all(0));
|
|
|
|
oclMat dst1(m1), dst2(m2);
|
|
|
|
//arithmetic_sum_run(src, dst1,"arithm_op_sum");
|
|
|
|
//arithmetic_sum_run(src, dst2,"arithm_op_squares_sum");
|
|
|
|
m1 = (Mat)dst1;
|
|
|
|
m2 = (Mat)dst2;
|
|
|
|
int i = 0, *p = (int *)m1.data, *q = (int *)m2.data;
|
|
|
|
for(; i < channels; i++)
|
|
|
|
{
|
|
|
|
mean.val[i] = (double)p[i] / (src.cols * src.rows);
|
|
|
|
stddev.val[i] = std::sqrt(std::max((double) q[i] / (src.cols * src.rows) - mean.val[i] * mean.val[i] , 0.));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
//////////////////////////////////// minMax /////////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2013-01-25 18:31:34 +08:00
|
|
|
static void arithmetic_minMax_run(const oclMat &src, const oclMat &mask, cl_mem &dst, int vlen , int groupnum, string kernelName)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
int all_cols = src.step / (vlen * src.elemSize1());
|
|
|
|
int pre_cols = (src.offset % src.step) / (vlen * src.elemSize1());
|
|
|
|
int sec_cols = all_cols - (src.offset % src.step + src.cols * src.elemSize() - 1) / (vlen * src.elemSize1()) - 1;
|
|
|
|
int invalid_cols = pre_cols + sec_cols;
|
|
|
|
int cols = all_cols - invalid_cols , elemnum = cols * src.rows;;
|
|
|
|
int offset = src.offset / (vlen * src.elemSize1());
|
|
|
|
int repeat_s = src.offset / src.elemSize1() - offset * vlen;
|
2012-10-11 16:22:47 +08:00
|
|
|
int repeat_e = (offset + cols) * vlen - src.offset / src.elemSize1() - src.cols * src.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
char build_options[50];
|
|
|
|
sprintf(build_options, "-D DEPTH_%d -D REPEAT_S%d -D REPEAT_E%d", src.depth(), repeat_s, repeat_e);
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&invalid_cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&offset));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&elemnum));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&groupnum));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&src.data));
|
|
|
|
if(!mask.empty())
|
|
|
|
{
|
|
|
|
int mall_cols = mask.step / (vlen * mask.elemSize1());
|
|
|
|
int mpre_cols = (mask.offset % mask.step) / (vlen * mask.elemSize1());
|
|
|
|
int msec_cols = mall_cols - (mask.offset % mask.step + mask.cols * mask.elemSize() - 1) / (vlen * mask.elemSize1()) - 1;
|
|
|
|
int minvalid_cols = mpre_cols + msec_cols;
|
|
|
|
int moffset = mask.offset / (vlen * mask.elemSize1());
|
|
|
|
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&minvalid_cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&moffset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&mask.data ));
|
|
|
|
}
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&dst ));
|
|
|
|
size_t gt[3] = {groupnum * 256, 1, 1}, lt[3] = {256, 1, 1};
|
|
|
|
openCLExecuteKernel(src.clCxt, &arithm_minMax, kernelName, gt, lt, args, -1, -1, build_options);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2013-01-25 18:31:34 +08:00
|
|
|
static void arithmetic_minMax_mask_run(const oclMat &src, const oclMat &mask, cl_mem &dst, int vlen, int groupnum, string kernelName)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
size_t gt[3] = {groupnum * 256, 1, 1}, lt[3] = {256, 1, 1};
|
|
|
|
char build_options[50];
|
2012-10-11 16:22:47 +08:00
|
|
|
if(src.oclchannels() == 1)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
int cols = (src.cols - 1) / vlen + 1;
|
|
|
|
int invalid_cols = src.step / (vlen * src.elemSize1()) - cols;
|
|
|
|
int offset = src.offset / src.elemSize1();
|
|
|
|
int repeat_me = vlen - (mask.cols % vlen == 0 ? vlen : mask.cols % vlen);
|
|
|
|
int minvalid_cols = mask.step / (vlen * mask.elemSize1()) - cols;
|
|
|
|
int moffset = mask.offset / mask.elemSize1();
|
|
|
|
int elemnum = cols * src.rows;
|
|
|
|
sprintf(build_options, "-D DEPTH_%d -D REPEAT_E%d", src.depth(), repeat_me);
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&invalid_cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&offset));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&elemnum));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&groupnum));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&src.data));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&minvalid_cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&moffset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&mask.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&dst ));
|
|
|
|
// printf("elemnum:%d,cols:%d,invalid_cols:%d,offset:%d,minvalid_cols:%d,moffset:%d,repeat_e:%d\r\n",
|
|
|
|
// elemnum,cols,invalid_cols,offset,minvalid_cols,moffset,repeat_me);
|
|
|
|
openCLExecuteKernel(src.clCxt, &arithm_minMax_mask, kernelName, gt, lt, args, -1, -1, build_options);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-05-22 13:46:42 +08:00
|
|
|
template <typename T> void arithmetic_minMax(const oclMat &src, double *minVal, double *maxVal,
|
|
|
|
const oclMat &mask, oclMat &buf)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
size_t groupnum = src.clCxt->computeUnits();
|
2012-07-17 01:08:14 +08:00
|
|
|
CV_Assert(groupnum != 0);
|
|
|
|
groupnum = groupnum * 2;
|
|
|
|
int vlen = 8;
|
2012-10-22 15:14:22 +08:00
|
|
|
int dbsize = groupnum * 2 * vlen * sizeof(T) ;
|
2013-05-22 13:46:42 +08:00
|
|
|
|
|
|
|
ensureSizeIsEnough(1, dbsize, CV_8UC1, buf);
|
|
|
|
|
|
|
|
cl_mem buf_data = reinterpret_cast<cl_mem>(buf.data);
|
|
|
|
|
2012-07-17 01:08:14 +08:00
|
|
|
if (mask.empty())
|
|
|
|
{
|
2013-05-22 13:46:42 +08:00
|
|
|
arithmetic_minMax_run(src, mask, buf_data, vlen, groupnum, "arithm_op_minMax");
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2013-05-22 13:46:42 +08:00
|
|
|
arithmetic_minMax_mask_run(src, mask, buf_data, vlen, groupnum, "arithm_op_minMax_mask");
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
2013-05-22 13:46:42 +08:00
|
|
|
|
|
|
|
Mat matbuf = Mat(buf);
|
|
|
|
T *p = matbuf.ptr<T>();
|
|
|
|
if(minVal != NULL)
|
|
|
|
{
|
|
|
|
*minVal = std::numeric_limits<double>::max();
|
2012-10-26 17:43:59 +08:00
|
|
|
for(int i = 0; i < vlen * (int)groupnum; i++)
|
|
|
|
{
|
|
|
|
*minVal = *minVal < p[i] ? *minVal : p[i];
|
|
|
|
}
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
2013-05-22 13:46:42 +08:00
|
|
|
if(maxVal != NULL)
|
|
|
|
{
|
|
|
|
*maxVal = -std::numeric_limits<double>::max();
|
2012-10-26 17:43:59 +08:00
|
|
|
for(int i = vlen * (int)groupnum; i < 2 * vlen * (int)groupnum; i++)
|
|
|
|
{
|
|
|
|
*maxVal = *maxVal > p[i] ? *maxVal : p[i];
|
|
|
|
}
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-05-22 13:46:42 +08:00
|
|
|
typedef void (*minMaxFunc)(const oclMat &src, double *minVal, double *maxVal, const oclMat &mask, oclMat &buf);
|
2012-07-17 01:08:14 +08:00
|
|
|
void cv::ocl::minMax(const oclMat &src, double *minVal, double *maxVal, const oclMat &mask)
|
2013-05-22 13:46:42 +08:00
|
|
|
{
|
|
|
|
oclMat buf;
|
|
|
|
minMax_buf(src, minVal, maxVal, mask, buf);
|
|
|
|
}
|
|
|
|
void cv::ocl::minMax_buf(const oclMat &src, double *minVal, double *maxVal, const oclMat &mask, oclMat &buf)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Assert(src.oclchannels() == 1);
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src.clCxt->supportsFeature(Context::CL_DOUBLE) && src.depth() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Error(CV_GpuNotSupported, "select device don't support double");
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
static minMaxFunc functab[8] =
|
|
|
|
{
|
|
|
|
arithmetic_minMax<uchar>,
|
|
|
|
arithmetic_minMax<char>,
|
|
|
|
arithmetic_minMax<ushort>,
|
|
|
|
arithmetic_minMax<short>,
|
|
|
|
arithmetic_minMax<int>,
|
|
|
|
arithmetic_minMax<float>,
|
|
|
|
arithmetic_minMax<double>,
|
|
|
|
0
|
|
|
|
};
|
|
|
|
minMaxFunc func;
|
|
|
|
func = functab[src.depth()];
|
2013-05-22 13:46:42 +08:00
|
|
|
func(src, minVal, maxVal, mask, buf);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
/////////////////////////////////// norm /////////////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
double cv::ocl::norm(const oclMat &src1, int normType)
|
|
|
|
{
|
|
|
|
return norm(src1, oclMat(src1.size(), src1.type(), Scalar::all(0)), normType);
|
|
|
|
}
|
|
|
|
|
|
|
|
double cv::ocl::norm(const oclMat &src1, const oclMat &src2, int normType)
|
|
|
|
{
|
|
|
|
bool isRelative = (normType & NORM_RELATIVE) != 0;
|
|
|
|
normType &= 7;
|
|
|
|
CV_Assert(src1.depth() <= CV_32S && src1.type() == src2.type() && ( normType == NORM_INF || normType == NORM_L1 || normType == NORM_L2));
|
2012-10-11 16:22:47 +08:00
|
|
|
int channels = src1.oclchannels(), i = 0, *p;
|
2012-07-17 01:08:14 +08:00
|
|
|
double r = 0;
|
|
|
|
oclMat gm1(src1.size(), src1.type());
|
|
|
|
int min_int = (normType == NORM_INF ? CL_INT_MIN : 0);
|
|
|
|
Mat m(1, 1, CV_MAKETYPE(CV_32S, channels), cv::Scalar::all(min_int));
|
|
|
|
oclMat gm2(m), emptyMat;
|
|
|
|
switch(normType)
|
|
|
|
{
|
|
|
|
case NORM_INF:
|
|
|
|
// arithmetic_run(src1, src2, gm1, "arithm_op_absdiff");
|
|
|
|
//arithmetic_minMax_run(gm1,emptyMat, gm2,"arithm_op_max");
|
|
|
|
m = (gm2);
|
|
|
|
p = (int *)m.data;
|
|
|
|
r = -std::numeric_limits<double>::max();
|
|
|
|
for(i = 0; i < channels; i++)
|
|
|
|
{
|
|
|
|
r = std::max(r, (double)p[i]);
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case NORM_L1:
|
|
|
|
//arithmetic_run(src1, src2, gm1, "arithm_op_absdiff");
|
|
|
|
//arithmetic_sum_run(gm1, gm2,"arithm_op_sum");
|
|
|
|
m = (gm2);
|
|
|
|
p = (int *)m.data;
|
|
|
|
for(i = 0; i < channels; i++)
|
|
|
|
{
|
|
|
|
r = r + (double)p[i];
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case NORM_L2:
|
|
|
|
//arithmetic_run(src1, src2, gm1, "arithm_op_absdiff");
|
|
|
|
//arithmetic_sum_run(gm1, gm2,"arithm_op_squares_sum");
|
|
|
|
m = (gm2);
|
|
|
|
p = (int *)m.data;
|
|
|
|
for(i = 0; i < channels; i++)
|
|
|
|
{
|
|
|
|
r = r + (double)p[i];
|
|
|
|
}
|
|
|
|
r = std::sqrt(r);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if(isRelative)
|
|
|
|
r = r / norm(src2, normType);
|
|
|
|
return r;
|
|
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
////////////////////////////////// flip //////////////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2013-01-25 18:31:34 +08:00
|
|
|
static void arithmetic_flip_rows_run(const oclMat &src, oclMat &dst, string kernelName)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src.clCxt->supportsFeature(Context::CL_DOUBLE) && src.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Error(CV_GpuNotSupported, "Selected device don't support double\r\n");
|
2012-07-17 01:08:14 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
CV_Assert(src.cols == dst.cols && src.rows == dst.rows);
|
|
|
|
|
|
|
|
CV_Assert(src.type() == dst.type());
|
|
|
|
|
|
|
|
Context *clCxt = src.clCxt;
|
2012-10-11 16:22:47 +08:00
|
|
|
int channels = dst.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
int depth = dst.depth();
|
|
|
|
|
|
|
|
int vector_lengths[4][7] = {{4, 4, 4, 4, 1, 1, 1},
|
|
|
|
{4, 4, 4, 4, 1, 1, 1},
|
|
|
|
{4, 4, 4, 4, 1, 1, 1},
|
|
|
|
{4, 4, 4, 4, 1, 1, 1}
|
|
|
|
};
|
|
|
|
|
2012-10-11 16:22:47 +08:00
|
|
|
size_t vector_length = vector_lengths[channels - 1][depth];
|
2012-07-17 01:08:14 +08:00
|
|
|
int offset_cols = ((dst.offset % dst.step) / dst.elemSize1()) & (vector_length - 1);
|
|
|
|
|
|
|
|
int cols = divUp(dst.cols * channels + offset_cols, vector_length);
|
|
|
|
int rows = divUp(dst.rows, 2);
|
|
|
|
|
|
|
|
size_t localThreads[3] = { 64, 4, 1 };
|
2013-09-16 19:11:56 +08:00
|
|
|
size_t globalThreads[3] = { cols, rows, 1 };
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
int dst_step1 = dst.cols * dst.elemSize();
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.rows ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&rows ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst_step1 ));
|
|
|
|
|
|
|
|
openCLExecuteKernel(clCxt, &arithm_flip, kernelName, globalThreads, localThreads, args, -1, depth);
|
|
|
|
}
|
2013-01-25 18:31:34 +08:00
|
|
|
static void arithmetic_flip_cols_run(const oclMat &src, oclMat &dst, string kernelName, bool isVertical)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src.clCxt->supportsFeature(Context::CL_DOUBLE) && src.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Error(CV_GpuNotSupported, "Selected device don't support double\r\n");
|
2012-07-17 01:08:14 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
CV_Assert(src.cols == dst.cols && src.rows == dst.rows);
|
|
|
|
CV_Assert(src.type() == dst.type());
|
|
|
|
|
|
|
|
Context *clCxt = src.clCxt;
|
2012-10-11 16:22:47 +08:00
|
|
|
int channels = dst.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
int depth = dst.depth();
|
|
|
|
|
|
|
|
int vector_lengths[4][7] = {{1, 1, 1, 1, 1, 1, 1},
|
|
|
|
{1, 1, 1, 1, 1, 1, 1},
|
|
|
|
{1, 1, 1, 1, 1, 1, 1},
|
|
|
|
{1, 1, 1, 1, 1, 1, 1}
|
|
|
|
};
|
|
|
|
|
2012-10-11 16:22:47 +08:00
|
|
|
size_t vector_length = vector_lengths[channels - 1][depth];
|
2012-07-17 01:08:14 +08:00
|
|
|
int offset_cols = ((dst.offset % dst.step) / dst.elemSize()) & (vector_length - 1);
|
|
|
|
int cols = divUp(dst.cols + offset_cols, vector_length);
|
|
|
|
cols = isVertical ? cols : divUp(cols, 2);
|
|
|
|
int rows = isVertical ? divUp(dst.rows, 2) : dst.rows;
|
|
|
|
|
|
|
|
size_t localThreads[3] = { 64, 4, 1 };
|
2013-09-16 19:11:56 +08:00
|
|
|
size_t globalThreads[3] = { cols, rows, 1 };
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
int dst_step1 = dst.cols * dst.elemSize();
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.rows ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.cols ));
|
|
|
|
|
|
|
|
if(isVertical)
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&rows ));
|
|
|
|
else
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&cols ));
|
|
|
|
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst_step1 ));
|
|
|
|
|
|
|
|
const char **kernelString = isVertical ? &arithm_flip_rc : &arithm_flip;
|
|
|
|
|
2012-10-11 16:22:47 +08:00
|
|
|
openCLExecuteKernel(clCxt, kernelString, kernelName, globalThreads, localThreads, args, src.oclchannels(), depth);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
void cv::ocl::flip(const oclMat &src, oclMat &dst, int flipCode)
|
|
|
|
{
|
|
|
|
dst.create(src.size(), src.type());
|
|
|
|
if(flipCode == 0)
|
|
|
|
{
|
|
|
|
arithmetic_flip_rows_run(src, dst, "arithm_flip_rows");
|
|
|
|
}
|
|
|
|
else if(flipCode > 0)
|
|
|
|
arithmetic_flip_cols_run(src, dst, "arithm_flip_cols", false);
|
|
|
|
else
|
|
|
|
arithmetic_flip_cols_run(src, dst, "arithm_flip_rc", true);
|
|
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
////////////////////////////////// LUT //////////////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2013-09-24 17:49:38 +08:00
|
|
|
|
|
|
|
static void arithmetic_lut_run(const oclMat &src, const oclMat &lut, oclMat &dst, string kernelName)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-09-24 17:49:38 +08:00
|
|
|
Context *clCxt = src.clCxt;
|
|
|
|
int sdepth = src.depth();
|
|
|
|
int src_step1 = src.step1(), dst_step1 = dst.step1();
|
|
|
|
int src_offset1 = src.offset / src.elemSize1(), dst_offset1 = dst.offset / dst.elemSize1();
|
|
|
|
int lut_offset1 = lut.offset / lut.elemSize1() + (sdepth == CV_8U ? 0 : 128) * lut.channels();
|
|
|
|
int cols1 = src.cols * src.oclchannels();
|
|
|
|
|
|
|
|
size_t localSize[] = { 16, 16, 1 };
|
|
|
|
size_t globalSize[] = { lut.channels() == 1 ? cols1 : src.cols, src.rows, 1 };
|
|
|
|
|
|
|
|
const char * const typeMap[] = { "uchar", "char", "ushort", "short", "int", "float", "double" };
|
|
|
|
std::string buildOptions = format("-D srcT=%s -D dstT=%s", typeMap[sdepth], typeMap[dst.depth()]);
|
|
|
|
|
2012-07-17 01:08:14 +08:00
|
|
|
vector<pair<size_t , const void *> > args;
|
2013-09-24 17:49:38 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&lut.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&cols1));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src.rows ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src_offset1 ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&lut_offset1 ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst_offset1 ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src_step1 ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst_step1 ));
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:49:38 +08:00
|
|
|
openCLExecuteKernel(clCxt, &arithm_LUT, kernelName, globalSize, localSize,
|
|
|
|
args, lut.oclchannels(), -1, buildOptions.c_str());
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void cv::ocl::LUT(const oclMat &src, const oclMat &lut, oclMat &dst)
|
|
|
|
{
|
2013-09-24 17:49:38 +08:00
|
|
|
int cn = src.channels(), depth = src.depth();
|
|
|
|
CV_Assert(depth == CV_8U || depth == CV_8S);
|
|
|
|
CV_Assert(lut.channels() == 1 || lut.channels() == src.channels());
|
|
|
|
CV_Assert(lut.rows == 1 && lut.cols == 256);
|
2012-07-17 01:08:14 +08:00
|
|
|
dst.create(src.size(), CV_MAKETYPE(lut.depth(), cn));
|
|
|
|
string kernelName = "LUT";
|
|
|
|
arithmetic_lut_run(src, lut, dst, kernelName);
|
|
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
//////////////////////////////// exp log /////////////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2013-01-25 18:31:34 +08:00
|
|
|
static void arithmetic_exp_log_run(const oclMat &src, oclMat &dst, string kernelName, const char **kernelString)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
dst.create(src.size(), src.type());
|
|
|
|
CV_Assert(src.cols == dst.cols &&
|
|
|
|
src.rows == dst.rows );
|
|
|
|
|
|
|
|
CV_Assert(src.type() == dst.type());
|
|
|
|
CV_Assert( src.type() == CV_32F || src.type() == CV_64F);
|
|
|
|
|
|
|
|
Context *clCxt = src.clCxt;
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!clCxt->supportsFeature(Context::CL_DOUBLE) && src.type() == CV_64F)
|
2012-08-03 14:08:36 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Error(CV_GpuNotSupported, "Selected device don't support double\r\n");
|
2012-08-03 14:08:36 +08:00
|
|
|
return;
|
|
|
|
}
|
2012-10-11 16:22:47 +08:00
|
|
|
//int channels = dst.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
int depth = dst.depth();
|
|
|
|
|
|
|
|
size_t localThreads[3] = { 64, 4, 1 };
|
2013-09-16 19:11:56 +08:00
|
|
|
size_t globalThreads[3] = { dst.cols, dst.rows, 1 };
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src.rows ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src.cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst.data ));
|
|
|
|
|
|
|
|
openCLExecuteKernel(clCxt, kernelString, kernelName, globalThreads, localThreads, args, -1, depth);
|
|
|
|
}
|
|
|
|
void cv::ocl::exp(const oclMat &src, oclMat &dst)
|
|
|
|
{
|
|
|
|
arithmetic_exp_log_run(src, dst, "arithm_exp", &arithm_exp);
|
|
|
|
}
|
|
|
|
|
|
|
|
void cv::ocl::log(const oclMat &src, oclMat &dst)
|
|
|
|
{
|
|
|
|
arithmetic_exp_log_run(src, dst, "arithm_log", &arithm_log);
|
|
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
////////////////////////////// magnitude phase ///////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2013-01-25 18:31:34 +08:00
|
|
|
static void arithmetic_magnitude_phase_run(const oclMat &src1, const oclMat &src2, oclMat &dst, string kernelName)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src1.clCxt->supportsFeature(Context::CL_DOUBLE) && src1.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Error(CV_GpuNotSupported, "Selected device don't support double\r\n");
|
2012-07-17 01:08:14 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
Context *clCxt = src1.clCxt;
|
2012-10-11 16:22:47 +08:00
|
|
|
int channels = dst.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
int depth = dst.depth();
|
|
|
|
|
|
|
|
size_t vector_length = 1;
|
|
|
|
int offset_cols = ((dst.offset % dst.step) / dst.elemSize1()) & (vector_length - 1);
|
|
|
|
int cols = divUp(dst.cols * channels + offset_cols, vector_length);
|
|
|
|
|
|
|
|
size_t localThreads[3] = { 64, 4, 1 };
|
2013-09-16 19:11:56 +08:00
|
|
|
size_t globalThreads[3] = { cols, dst.rows, 1 };
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src1.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src2.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.rows ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&cols ));
|
|
|
|
|
|
|
|
openCLExecuteKernel(clCxt, &arithm_magnitude, kernelName, globalThreads, localThreads, args, -1, depth);
|
|
|
|
}
|
|
|
|
|
|
|
|
void cv::ocl::magnitude(const oclMat &src1, const oclMat &src2, oclMat &dst)
|
|
|
|
{
|
|
|
|
CV_Assert(src1.type() == src2.type() && src1.size() == src2.size() &&
|
|
|
|
(src1.depth() == CV_32F || src1.depth() == CV_64F));
|
|
|
|
|
|
|
|
dst.create(src1.size(), src1.type());
|
|
|
|
arithmetic_magnitude_phase_run(src1, src2, dst, "arithm_magnitude");
|
|
|
|
}
|
|
|
|
|
2013-01-25 18:31:34 +08:00
|
|
|
static void arithmetic_phase_run(const oclMat &src1, const oclMat &src2, oclMat &dst, string kernelName, const char **kernelString)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src1.clCxt->supportsFeature(Context::CL_DOUBLE) && src1.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Error(CV_GpuNotSupported, "Selected device don't support double\r\n");
|
2012-07-17 01:08:14 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
CV_Assert(src1.cols == src2.cols && src2.cols == dst.cols && src1.rows == src2.rows && src2.rows == dst.rows);
|
|
|
|
CV_Assert(src1.type() == src2.type() && src1.type() == dst.type());
|
|
|
|
|
|
|
|
Context *clCxt = src1.clCxt;
|
2012-10-11 16:22:47 +08:00
|
|
|
int channels = dst.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
int depth = dst.depth();
|
|
|
|
|
|
|
|
size_t vector_length = 1;
|
|
|
|
int offset_cols = ((dst.offset % dst.step) / dst.elemSize1()) & (vector_length - 1);
|
|
|
|
int cols = divUp(dst.cols * channels + offset_cols, vector_length);
|
|
|
|
|
|
|
|
size_t localThreads[3] = { 64, 4, 1 };
|
2013-09-16 19:11:56 +08:00
|
|
|
size_t globalThreads[3] = { cols, dst.rows, 1 };
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
int dst_step1 = dst.cols * dst.elemSize();
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src1.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src2.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.rows ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst_step1 ));
|
|
|
|
|
|
|
|
openCLExecuteKernel(clCxt, kernelString, kernelName, globalThreads, localThreads, args, -1, depth);
|
|
|
|
}
|
|
|
|
void cv::ocl::phase(const oclMat &x, const oclMat &y, oclMat &Angle , bool angleInDegrees)
|
|
|
|
{
|
|
|
|
CV_Assert(x.type() == y.type() && x.size() == y.size() && (x.depth() == CV_32F || x.depth() == CV_64F));
|
|
|
|
Angle.create(x.size(), x.type());
|
|
|
|
string kernelName = angleInDegrees ? "arithm_phase_indegrees" : "arithm_phase_inradians";
|
|
|
|
if(angleInDegrees)
|
|
|
|
arithmetic_phase_run(x, y, Angle, kernelName, &arithm_phase);
|
|
|
|
else
|
|
|
|
arithmetic_phase_run(x, y, Angle, kernelName, &arithm_phase);
|
|
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
////////////////////////////////// cartToPolar ///////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2013-01-25 18:31:34 +08:00
|
|
|
static void arithmetic_cartToPolar_run(const oclMat &src1, const oclMat &src2, oclMat &dst_mag, oclMat &dst_cart,
|
2012-07-17 01:08:14 +08:00
|
|
|
string kernelName, bool angleInDegrees)
|
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src1.clCxt->supportsFeature(Context::CL_DOUBLE) && src1.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Error(CV_GpuNotSupported, "Selected device don't support double\r\n");
|
2012-07-17 01:08:14 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
Context *clCxt = src1.clCxt;
|
2012-10-11 16:22:47 +08:00
|
|
|
int channels = src1.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
int depth = src1.depth();
|
|
|
|
|
|
|
|
int cols = src1.cols * channels;
|
|
|
|
|
|
|
|
size_t localThreads[3] = { 64, 4, 1 };
|
2013-09-16 19:11:56 +08:00
|
|
|
size_t globalThreads[3] = { cols, src1.rows, 1 };
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
int tmp = angleInDegrees ? 1 : 0;
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src1.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src2.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst_mag.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst_mag.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst_mag.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst_cart.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst_cart.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst_cart.offset ));
|
2013-09-16 19:11:56 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.rows ));
|
2012-07-17 01:08:14 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&tmp ));
|
|
|
|
|
|
|
|
openCLExecuteKernel(clCxt, &arithm_cartToPolar, kernelName, globalThreads, localThreads, args, -1, depth);
|
|
|
|
}
|
|
|
|
void cv::ocl::cartToPolar(const oclMat &x, const oclMat &y, oclMat &mag, oclMat &angle, bool angleInDegrees)
|
|
|
|
{
|
|
|
|
CV_Assert(x.type() == y.type() && x.size() == y.size() && (x.depth() == CV_32F || x.depth() == CV_64F));
|
|
|
|
|
|
|
|
mag.create(x.size(), x.type());
|
|
|
|
angle.create(x.size(), x.type());
|
|
|
|
|
|
|
|
arithmetic_cartToPolar_run(x, y, mag, angle, "arithm_cartToPolar", angleInDegrees);
|
|
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
////////////////////////////////// polarToCart ///////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2013-01-25 18:31:34 +08:00
|
|
|
static void arithmetic_ptc_run(const oclMat &src1, const oclMat &src2, oclMat &dst1, oclMat &dst2, bool angleInDegrees,
|
2012-07-17 01:08:14 +08:00
|
|
|
string kernelName)
|
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src1.clCxt->supportsFeature(Context::CL_DOUBLE) && src1.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Error(CV_GpuNotSupported, "Selected device don't support double\r\n");
|
2012-07-17 01:08:14 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
Context *clCxt = src2.clCxt;
|
2012-10-11 16:22:47 +08:00
|
|
|
int channels = src2.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
int depth = src2.depth();
|
|
|
|
|
|
|
|
int cols = src2.cols * channels;
|
|
|
|
int rows = src2.rows;
|
|
|
|
|
|
|
|
size_t localThreads[3] = { 64, 4, 1 };
|
2013-09-16 19:11:56 +08:00
|
|
|
size_t globalThreads[3] = { cols, rows, 1 };
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
int tmp = angleInDegrees ? 1 : 0;
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
if(src1.data)
|
|
|
|
{
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src1.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.offset ));
|
|
|
|
}
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src2.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst1.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst1.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst1.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst2.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst2.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst2.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&rows ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&tmp ));
|
|
|
|
|
|
|
|
openCLExecuteKernel(clCxt, &arithm_polarToCart, kernelName, globalThreads, localThreads, args, -1, depth);
|
|
|
|
}
|
|
|
|
|
|
|
|
void cv::ocl::polarToCart(const oclMat &magnitude, const oclMat &angle, oclMat &x, oclMat &y, bool angleInDegrees)
|
|
|
|
{
|
|
|
|
CV_Assert(angle.depth() == CV_32F || angle.depth() == CV_64F);
|
|
|
|
|
|
|
|
x.create(angle.size(), angle.type());
|
|
|
|
y.create(angle.size(), angle.type());
|
|
|
|
|
|
|
|
if( magnitude.data )
|
|
|
|
{
|
|
|
|
CV_Assert( magnitude.size() == angle.size() && magnitude.type() == angle.type() );
|
|
|
|
arithmetic_ptc_run(magnitude, angle, x, y, angleInDegrees, "arithm_polarToCart_mag");
|
|
|
|
}
|
|
|
|
else
|
|
|
|
arithmetic_ptc_run(magnitude, angle, x, y, angleInDegrees, "arithm_polarToCart");
|
|
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
/////////////////////////////////// minMaxLoc ////////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2013-01-25 18:31:34 +08:00
|
|
|
static void arithmetic_minMaxLoc_run(const oclMat &src, cl_mem &dst, int vlen , int groupnum)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
int all_cols = src.step / (vlen * src.elemSize1());
|
|
|
|
int pre_cols = (src.offset % src.step) / (vlen * src.elemSize1());
|
|
|
|
int sec_cols = all_cols - (src.offset % src.step + src.cols * src.elemSize1() - 1) / (vlen * src.elemSize1()) - 1;
|
|
|
|
int invalid_cols = pre_cols + sec_cols;
|
|
|
|
int cols = all_cols - invalid_cols , elemnum = cols * src.rows;;
|
|
|
|
int offset = src.offset / (vlen * src.elemSize1());
|
|
|
|
int repeat_s = src.offset / src.elemSize1() - offset * vlen;
|
|
|
|
int repeat_e = (offset + cols) * vlen - src.offset / src.elemSize1() - src.cols;
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&invalid_cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&offset));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&elemnum));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&groupnum));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&src.data));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&dst ));
|
|
|
|
char build_options[50];
|
|
|
|
sprintf(build_options, "-D DEPTH_%d -D REPEAT_S%d -D REPEAT_E%d", src.depth(), repeat_s, repeat_e);
|
|
|
|
size_t gt[3] = {groupnum * 256, 1, 1}, lt[3] = {256, 1, 1};
|
|
|
|
openCLExecuteKernel(src.clCxt, &arithm_minMaxLoc, "arithm_op_minMaxLoc", gt, lt, args, -1, -1, build_options);
|
|
|
|
}
|
|
|
|
|
2013-01-25 18:31:34 +08:00
|
|
|
static void arithmetic_minMaxLoc_mask_run(const oclMat &src, const oclMat &mask, cl_mem &dst, int vlen, int groupnum)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
size_t gt[3] = {groupnum * 256, 1, 1}, lt[3] = {256, 1, 1};
|
|
|
|
char build_options[50];
|
2012-10-11 16:22:47 +08:00
|
|
|
if(src.oclchannels() == 1)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
int cols = (src.cols - 1) / vlen + 1;
|
|
|
|
int invalid_cols = src.step / (vlen * src.elemSize1()) - cols;
|
|
|
|
int offset = src.offset / src.elemSize1();
|
|
|
|
int repeat_me = vlen - (mask.cols % vlen == 0 ? vlen : mask.cols % vlen);
|
|
|
|
int minvalid_cols = mask.step / (vlen * mask.elemSize1()) - cols;
|
|
|
|
int moffset = mask.offset / mask.elemSize1();
|
|
|
|
int elemnum = cols * src.rows;
|
|
|
|
sprintf(build_options, "-D DEPTH_%d -D REPEAT_E%d", src.depth(), repeat_me);
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&invalid_cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&offset));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&elemnum));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&groupnum));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&src.data));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&minvalid_cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&moffset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&mask.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&dst ));
|
|
|
|
// printf("elemnum:%d,cols:%d,invalid_cols:%d,offset:%d,minvalid_cols:%d,moffset:%d,repeat_e:%d\r\n",
|
|
|
|
// elemnum,cols,invalid_cols,offset,minvalid_cols,moffset,repeat_me);
|
|
|
|
openCLExecuteKernel(src.clCxt, &arithm_minMaxLoc_mask, "arithm_op_minMaxLoc_mask", gt, lt, args, -1, -1, build_options);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
template<typename T>
|
|
|
|
void arithmetic_minMaxLoc(const oclMat &src, double *minVal, double *maxVal,
|
2012-10-11 16:22:47 +08:00
|
|
|
Point *minLoc, Point *maxLoc, const oclMat &mask)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Assert(src.oclchannels() == 1);
|
2013-03-18 05:59:24 +08:00
|
|
|
size_t groupnum = src.clCxt->computeUnits();
|
2012-07-17 01:08:14 +08:00
|
|
|
CV_Assert(groupnum != 0);
|
|
|
|
int minloc = -1 , maxloc = -1;
|
2012-10-22 15:14:22 +08:00
|
|
|
int vlen = 4, dbsize = groupnum * vlen * 4 * sizeof(T) ;
|
2012-07-17 01:08:14 +08:00
|
|
|
Context *clCxt = src.clCxt;
|
2012-10-11 16:22:47 +08:00
|
|
|
cl_mem dstBuffer = openCLCreateBuffer(clCxt, CL_MEM_WRITE_ONLY, dbsize);
|
2012-07-17 01:08:14 +08:00
|
|
|
*minVal = std::numeric_limits<double>::max() , *maxVal = -std::numeric_limits<double>::max();
|
|
|
|
if (mask.empty())
|
|
|
|
{
|
|
|
|
arithmetic_minMaxLoc_run(src, dstBuffer, vlen, groupnum);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
arithmetic_minMaxLoc_mask_run(src, mask, dstBuffer, vlen, groupnum);
|
|
|
|
}
|
|
|
|
T *p = new T[groupnum * vlen * 4];
|
|
|
|
memset(p, 0, dbsize);
|
2012-10-11 16:22:47 +08:00
|
|
|
openCLReadBuffer(clCxt, dstBuffer, (void *)p, dbsize);
|
2012-10-22 15:14:22 +08:00
|
|
|
for(int i = 0; i < vlen * (int)groupnum; i++)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
*minVal = (*minVal < p[i] || p[i + 2 * vlen * groupnum] == -1) ? *minVal : p[i];
|
2012-10-22 15:14:22 +08:00
|
|
|
minloc = (*minVal < p[i] || p[i + 2 * vlen * groupnum] == -1) ? minloc : cvRound(p[i + 2 * vlen * groupnum]);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
2012-10-22 15:14:22 +08:00
|
|
|
for(int i = vlen * (int)groupnum; i < 2 * vlen * (int)groupnum; i++)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
*maxVal = (*maxVal > p[i] || p[i + 2 * vlen * groupnum] == -1) ? *maxVal : p[i];
|
2012-10-22 15:14:22 +08:00
|
|
|
maxloc = (*maxVal > p[i] || p[i + 2 * vlen * groupnum] == -1) ? maxloc : cvRound(p[i + 2 * vlen * groupnum]);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
int pre_rows = src.offset / src.step;
|
|
|
|
int pre_cols = (src.offset % src.step) / src.elemSize1();
|
|
|
|
int wholecols = src.step / src.elemSize1();
|
|
|
|
if( minLoc )
|
|
|
|
{
|
|
|
|
if( minloc >= 0 )
|
|
|
|
{
|
|
|
|
minLoc->y = minloc / wholecols - pre_rows;
|
|
|
|
minLoc->x = minloc % wholecols - pre_cols;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
minLoc->x = minLoc->y = -1;
|
|
|
|
}
|
|
|
|
if( maxLoc )
|
|
|
|
{
|
|
|
|
if( maxloc >= 0 )
|
|
|
|
{
|
|
|
|
maxLoc->y = maxloc / wholecols - pre_rows;
|
|
|
|
maxLoc->x = maxloc % wholecols - pre_cols;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
maxLoc->x = maxLoc->y = -1;
|
|
|
|
}
|
|
|
|
delete[] p;
|
|
|
|
openCLSafeCall(clReleaseMemObject(dstBuffer));
|
|
|
|
}
|
|
|
|
|
|
|
|
typedef void (*minMaxLocFunc)(const oclMat &src, double *minVal, double *maxVal,
|
2012-10-11 16:22:47 +08:00
|
|
|
Point *minLoc, Point *maxLoc, const oclMat &mask);
|
2012-07-17 01:08:14 +08:00
|
|
|
void cv::ocl::minMaxLoc(const oclMat &src, double *minVal, double *maxVal,
|
|
|
|
Point *minLoc, Point *maxLoc, const oclMat &mask)
|
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src.clCxt->supportsFeature(Context::CL_DOUBLE) && src.depth() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Error(CV_GpuNotSupported, "select device don't support double");
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
static minMaxLocFunc functab[2] =
|
|
|
|
{
|
|
|
|
arithmetic_minMaxLoc<float>,
|
|
|
|
arithmetic_minMaxLoc<double>
|
|
|
|
};
|
|
|
|
|
|
|
|
minMaxLocFunc func;
|
2013-03-18 05:59:24 +08:00
|
|
|
func = functab[(int)src.clCxt->supportsFeature(Context::CL_DOUBLE)];
|
2012-10-11 16:22:47 +08:00
|
|
|
func(src, minVal, maxVal, minLoc, maxLoc, mask);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////////////////// countNonZero ///////////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2013-01-25 18:31:34 +08:00
|
|
|
static void arithmetic_countNonZero_run(const oclMat &src, cl_mem &dst, int vlen , int groupnum, string kernelName)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
int all_cols = src.step / (vlen * src.elemSize1());
|
|
|
|
int pre_cols = (src.offset % src.step) / (vlen * src.elemSize1());
|
|
|
|
int sec_cols = all_cols - (src.offset % src.step + src.cols * src.elemSize() - 1) / (vlen * src.elemSize1()) - 1;
|
|
|
|
int invalid_cols = pre_cols + sec_cols;
|
|
|
|
int cols = all_cols - invalid_cols , elemnum = cols * src.rows;;
|
|
|
|
int offset = src.offset / (vlen * src.elemSize1());
|
|
|
|
int repeat_s = src.offset / src.elemSize1() - offset * vlen;
|
2012-10-11 16:22:47 +08:00
|
|
|
int repeat_e = (offset + cols) * vlen - src.offset / src.elemSize1() - src.cols * src.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
char build_options[50];
|
|
|
|
sprintf(build_options, "-D DEPTH_%d -D REPEAT_S%d -D REPEAT_E%d", src.depth(), repeat_s, repeat_e);
|
|
|
|
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&invalid_cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&offset));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&elemnum));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&groupnum));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&src.data));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&dst ));
|
|
|
|
size_t gt[3] = {groupnum * 256, 1, 1}, lt[3] = {256, 1, 1};
|
|
|
|
openCLExecuteKernel(src.clCxt, &arithm_nonzero, kernelName, gt, lt, args, -1, -1, build_options);
|
|
|
|
}
|
|
|
|
|
|
|
|
int cv::ocl::countNonZero(const oclMat &src)
|
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
size_t groupnum = src.clCxt->computeUnits();
|
|
|
|
if(!src.clCxt->supportsFeature(Context::CL_DOUBLE) && src.depth() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Error(CV_GpuNotSupported, "select device don't support double");
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
CV_Assert(groupnum != 0);
|
|
|
|
groupnum = groupnum * 2;
|
2012-10-22 15:14:22 +08:00
|
|
|
int vlen = 8 , dbsize = groupnum * vlen;
|
2012-07-17 01:08:14 +08:00
|
|
|
//cl_ulong start, end;
|
|
|
|
Context *clCxt = src.clCxt;
|
|
|
|
string kernelName = "arithm_op_nonzero";
|
|
|
|
int *p = new int[dbsize], nonzero = 0;
|
2012-10-11 16:22:47 +08:00
|
|
|
cl_mem dstBuffer = openCLCreateBuffer(clCxt, CL_MEM_WRITE_ONLY, dbsize * sizeof(int));
|
2012-07-17 01:08:14 +08:00
|
|
|
arithmetic_countNonZero_run(src, dstBuffer, vlen, groupnum, kernelName);
|
|
|
|
|
|
|
|
memset(p, 0, dbsize * sizeof(int));
|
2012-10-11 16:22:47 +08:00
|
|
|
openCLReadBuffer(clCxt, dstBuffer, (void *)p, dbsize * sizeof(int));
|
2012-07-17 01:08:14 +08:00
|
|
|
for(int i = 0; i < dbsize; i++)
|
|
|
|
{
|
|
|
|
nonzero += p[i];
|
|
|
|
}
|
|
|
|
delete[] p;
|
|
|
|
openCLSafeCall(clReleaseMemObject(dstBuffer));
|
|
|
|
return nonzero;
|
|
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
////////////////////////////////bitwise_op////////////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
2013-01-25 18:31:34 +08:00
|
|
|
static void bitwise_run(const oclMat &src1, oclMat &dst, string kernelName, const char **kernelString)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
dst.create(src1.size(), src1.type());
|
|
|
|
|
|
|
|
|
|
|
|
Context *clCxt = src1.clCxt;
|
2012-10-11 16:22:47 +08:00
|
|
|
int channels = dst.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
int depth = dst.depth();
|
|
|
|
|
|
|
|
int vector_lengths[4][7] = {{4, 4, 4, 4, 1, 1, 1},
|
|
|
|
{4, 4, 4, 4, 1, 1, 1},
|
|
|
|
{4, 4, 4, 4, 1, 1, 1},
|
|
|
|
{4, 4, 4, 4, 1, 1, 1}
|
|
|
|
};
|
|
|
|
|
2012-10-11 16:22:47 +08:00
|
|
|
size_t vector_length = vector_lengths[channels - 1][depth];
|
2012-07-17 01:08:14 +08:00
|
|
|
int offset_cols = (dst.offset / dst.elemSize1()) & (vector_length - 1);
|
|
|
|
int cols = divUp(dst.cols * channels + offset_cols, vector_length);
|
|
|
|
|
|
|
|
size_t localThreads[3] = { 64, 4, 1 };
|
2013-09-16 19:11:56 +08:00
|
|
|
size_t globalThreads[3] = { cols, dst.rows, 1 };
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
int dst_step1 = dst.cols * dst.elemSize();
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src1.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.rows ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst_step1 ));
|
|
|
|
|
|
|
|
openCLExecuteKernel(clCxt, kernelString, kernelName, globalThreads, localThreads, args, -1, depth);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
template<typename T>
|
2013-04-15 10:36:03 +08:00
|
|
|
void bitwise_run(const oclMat &src1, const oclMat &src2, oclMat &dst, string kernelName,
|
|
|
|
const char **kernelString, void *_scalar, const char* _opt = NULL)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
dst.create(src1.size(), src1.type());
|
|
|
|
CV_Assert(src1.cols == src2.cols && src2.cols == dst.cols &&
|
|
|
|
src1.rows == src2.rows && src2.rows == dst.rows);
|
|
|
|
|
|
|
|
CV_Assert(src1.type() == src2.type() && src1.type() == dst.type());
|
|
|
|
|
|
|
|
Context *clCxt = src1.clCxt;
|
2012-10-11 16:22:47 +08:00
|
|
|
int channels = dst.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
int depth = dst.depth();
|
|
|
|
|
|
|
|
int vector_lengths[4][7] = {{4, 4, 4, 4, 1, 1, 1},
|
|
|
|
{4, 4, 4, 4, 1, 1, 1},
|
|
|
|
{4, 4, 4, 4, 1, 1, 1},
|
|
|
|
{4, 4, 4, 4, 1, 1, 1}
|
|
|
|
};
|
|
|
|
|
2012-10-11 16:22:47 +08:00
|
|
|
size_t vector_length = vector_lengths[channels - 1][depth];
|
2012-07-17 01:08:14 +08:00
|
|
|
int offset_cols = (dst.offset / dst.elemSize1()) & (vector_length - 1);
|
|
|
|
int cols = divUp(dst.cols * channels + offset_cols, vector_length);
|
|
|
|
|
|
|
|
size_t localThreads[3] = { 64, 4, 1 };
|
2013-09-16 19:11:56 +08:00
|
|
|
size_t globalThreads[3] = { cols, dst.rows, 1 };
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
int dst_step1 = dst.cols * dst.elemSize();
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src1.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src2.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.rows ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst_step1 ));
|
|
|
|
|
2013-05-28 17:27:55 +08:00
|
|
|
T scalar;
|
2012-07-17 01:08:14 +08:00
|
|
|
if(_scalar != NULL)
|
|
|
|
{
|
|
|
|
double scalar1 = *((double *)_scalar);
|
2013-05-28 17:27:55 +08:00
|
|
|
scalar = (T)scalar1;
|
2012-07-17 01:08:14 +08:00
|
|
|
args.push_back( make_pair( sizeof(T), (void *)&scalar ));
|
|
|
|
}
|
|
|
|
|
2013-04-15 10:36:03 +08:00
|
|
|
openCLExecuteKernel(clCxt, kernelString, kernelName, globalThreads, localThreads, args, -1, depth, _opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
2013-04-15 10:36:03 +08:00
|
|
|
static void bitwise_run(const oclMat &src1, const oclMat &src2, oclMat &dst,
|
|
|
|
string kernelName, const char **kernelString, const char* _opt = NULL)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-04-15 10:36:03 +08:00
|
|
|
bitwise_run<char>(src1, src2, dst, kernelName, kernelString, (void *)NULL, _opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
2013-04-15 10:36:03 +08:00
|
|
|
static void bitwise_run(const oclMat &src1, const oclMat &src2, oclMat &dst,
|
|
|
|
const oclMat &mask, string kernelName, const char **kernelString, const char* _opt = NULL)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
dst.create(src1.size(), src1.type());
|
|
|
|
CV_Assert(src1.cols == src2.cols && src2.cols == dst.cols &&
|
|
|
|
src1.rows == src2.rows && src2.rows == dst.rows &&
|
|
|
|
src1.rows == mask.rows && src1.cols == mask.cols);
|
|
|
|
|
|
|
|
CV_Assert(src1.type() == src2.type() && src1.type() == dst.type());
|
|
|
|
CV_Assert(mask.type() == CV_8U);
|
|
|
|
|
|
|
|
Context *clCxt = src1.clCxt;
|
2012-10-11 16:22:47 +08:00
|
|
|
int channels = dst.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
int depth = dst.depth();
|
|
|
|
|
|
|
|
int vector_lengths[4][7] = {{4, 4, 2, 2, 1, 1, 1},
|
|
|
|
{2, 2, 1, 1, 1, 1, 1},
|
|
|
|
{4, 4, 2, 2 , 1, 1, 1},
|
|
|
|
{1, 1, 1, 1, 1, 1, 1}
|
|
|
|
};
|
|
|
|
|
2012-10-11 16:22:47 +08:00
|
|
|
size_t vector_length = vector_lengths[channels - 1][depth];
|
2012-07-17 01:08:14 +08:00
|
|
|
int offset_cols = ((dst.offset % dst.step) / dst.elemSize()) & (vector_length - 1);
|
|
|
|
int cols = divUp(dst.cols + offset_cols, vector_length);
|
|
|
|
|
|
|
|
size_t localThreads[3] = { 64, 4, 1 };
|
2013-09-16 19:11:56 +08:00
|
|
|
size_t globalThreads[3] = { cols, dst.rows, 1 };
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
int dst_step1 = dst.cols * dst.elemSize();
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src1.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src2.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&mask.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&mask.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&mask.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.rows ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst_step1 ));
|
|
|
|
|
2013-04-15 10:36:03 +08:00
|
|
|
openCLExecuteKernel(clCxt, kernelString, kernelName, globalThreads, localThreads, args, channels, depth, _opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2012-10-11 16:22:47 +08:00
|
|
|
template <typename WT , typename CL_WT>
|
2013-04-15 10:36:03 +08:00
|
|
|
void bitwise_scalar_run(const oclMat &src1, const Scalar &src2, oclMat &dst,
|
|
|
|
const oclMat &mask, string kernelName, const char **kernelString, int isMatSubScalar, const char* opt = NULL)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
dst.create(src1.size(), src1.type());
|
|
|
|
|
|
|
|
CV_Assert(src1.cols == dst.cols && src1.rows == dst.rows &&
|
|
|
|
src1.type() == dst.type());
|
|
|
|
|
|
|
|
|
|
|
|
if(mask.data)
|
2013-01-25 18:31:34 +08:00
|
|
|
{
|
2012-07-17 01:08:14 +08:00
|
|
|
CV_Assert(mask.type() == CV_8U && src1.rows == mask.rows && src1.cols == mask.cols);
|
2013-01-25 18:31:34 +08:00
|
|
|
}
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
Context *clCxt = src1.clCxt;
|
2012-10-11 16:22:47 +08:00
|
|
|
int channels = dst.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
int depth = dst.depth();
|
|
|
|
|
|
|
|
WT s[4] = { saturate_cast<WT>(src2.val[0]), saturate_cast<WT>(src2.val[1]),
|
|
|
|
saturate_cast<WT>(src2.val[2]), saturate_cast<WT>(src2.val[3])
|
|
|
|
};
|
|
|
|
|
|
|
|
int vector_lengths[4][7] = {{4, 4, 2, 2, 1, 1, 1},
|
|
|
|
{2, 2, 1, 1, 1, 1, 1},
|
|
|
|
{4, 4, 2, 2 , 1, 1, 1},
|
|
|
|
{1, 1, 1, 1, 1, 1, 1}
|
|
|
|
};
|
|
|
|
|
2012-10-11 16:22:47 +08:00
|
|
|
size_t vector_length = vector_lengths[channels - 1][depth];
|
2012-07-17 01:08:14 +08:00
|
|
|
int offset_cols = ((dst.offset % dst.step) / dst.elemSize()) & (vector_length - 1);
|
|
|
|
int cols = divUp(dst.cols + offset_cols, vector_length);
|
|
|
|
|
|
|
|
size_t localThreads[3] = { 64, 4, 1 };
|
2013-09-16 19:11:56 +08:00
|
|
|
size_t globalThreads[3] = { cols, dst.rows, 1 };
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
int dst_step1 = dst.cols * dst.elemSize();
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&src1.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&src1.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&src1.offset));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&dst.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&dst.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&dst.offset));
|
|
|
|
|
|
|
|
if(mask.data)
|
|
|
|
{
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem) , (void *)&mask.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&mask.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&mask.offset));
|
|
|
|
}
|
|
|
|
args.push_back( make_pair( sizeof(CL_WT) , (void *)&s ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&src1.rows ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&dst_step1 ));
|
|
|
|
if(isMatSubScalar != 0)
|
|
|
|
{
|
|
|
|
isMatSubScalar = isMatSubScalar > 0 ? 1 : 0;
|
|
|
|
args.push_back( make_pair( sizeof(cl_int) , (void *)&isMatSubScalar));
|
|
|
|
}
|
|
|
|
|
2013-04-15 10:36:03 +08:00
|
|
|
openCLExecuteKernel(clCxt, kernelString, kernelName, globalThreads, localThreads, args, channels, depth, opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2013-04-15 10:36:03 +08:00
|
|
|
typedef void (*BitwiseFuncS)(const oclMat &src1, const Scalar &src2, oclMat &dst,
|
|
|
|
const oclMat &mask, string kernelName, const char **kernelString, int isMatSubScalar, const char* opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
|
2013-04-15 10:36:03 +08:00
|
|
|
static void bitwise_scalar(const oclMat &src1, const Scalar &src2, oclMat &dst,
|
|
|
|
const oclMat &mask, string kernelName, const char **kernelString, int isMatSubScalar, const char* opt)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
static BitwiseFuncS tab[8] =
|
|
|
|
{
|
|
|
|
#if 0
|
|
|
|
bitwise_scalar_run<unsigned char>,
|
|
|
|
bitwise_scalar_run<char>,
|
|
|
|
bitwise_scalar_run<unsigned short>,
|
|
|
|
bitwise_scalar_run<short>,
|
|
|
|
bitwise_scalar_run<int>,
|
|
|
|
bitwise_scalar_run<float>,
|
|
|
|
bitwise_scalar_run<double>,
|
|
|
|
0
|
|
|
|
#else
|
|
|
|
|
2012-10-11 16:22:47 +08:00
|
|
|
bitwise_scalar_run<unsigned char, cl_uchar4>,
|
|
|
|
bitwise_scalar_run<char, cl_char4>,
|
|
|
|
bitwise_scalar_run<unsigned short, cl_ushort4>,
|
|
|
|
bitwise_scalar_run<short, cl_short4>,
|
|
|
|
bitwise_scalar_run<int, cl_int4>,
|
|
|
|
bitwise_scalar_run<float, cl_float4>,
|
|
|
|
bitwise_scalar_run<double, cl_double4>,
|
2012-07-17 01:08:14 +08:00
|
|
|
0
|
|
|
|
#endif
|
|
|
|
};
|
|
|
|
BitwiseFuncS func = tab[src1.depth()];
|
|
|
|
if(func == 0)
|
|
|
|
cv::ocl::error("Unsupported arithmetic operation", __FILE__, __LINE__);
|
2013-04-15 10:36:03 +08:00
|
|
|
func(src1, src2, dst, mask, kernelName, kernelString, isMatSubScalar, opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
2013-04-15 10:36:03 +08:00
|
|
|
static void bitwise_scalar(const oclMat &src1, const Scalar &src2, oclMat &dst,
|
|
|
|
const oclMat &mask, string kernelName, const char **kernelString, const char * opt = NULL)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-04-15 10:36:03 +08:00
|
|
|
bitwise_scalar(src1, src2, dst, mask, kernelName, kernelString, 0, opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void cv::ocl::bitwise_not(const oclMat &src, oclMat &dst)
|
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src.clCxt->supportsFeature(Context::CL_DOUBLE) && src.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
cout << "Selected device do not support double" << endl;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
dst.create(src.size(), src.type());
|
|
|
|
string kernelName = "arithm_bitwise_not";
|
|
|
|
bitwise_run(src, dst, kernelName, &arithm_bitwise_not);
|
|
|
|
}
|
|
|
|
|
|
|
|
void cv::ocl::bitwise_or(const oclMat &src1, const oclMat &src2, oclMat &dst, const oclMat &mask)
|
|
|
|
{
|
|
|
|
// dst.create(src1.size(),src1.type());
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src1.clCxt->supportsFeature(Context::CL_DOUBLE) && src1.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
cout << "Selected device do not support double" << endl;
|
|
|
|
return;
|
|
|
|
}
|
2013-04-15 10:36:03 +08:00
|
|
|
|
|
|
|
string kernelName = mask.empty() ? "arithm_bitwise_binary" : "arithm_bitwise_binary_with_mask";
|
|
|
|
static const char opt [] = "-D OP_BINARY=|";
|
2012-07-17 01:08:14 +08:00
|
|
|
if (mask.empty())
|
2013-04-15 10:36:03 +08:00
|
|
|
bitwise_run(src1, src2, dst, kernelName, &arithm_bitwise_binary, opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
else
|
2013-04-15 10:36:03 +08:00
|
|
|
bitwise_run(src1, src2, dst, mask, kernelName, &arithm_bitwise_binary_mask, opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void cv::ocl::bitwise_or(const oclMat &src1, const Scalar &src2, oclMat &dst, const oclMat &mask)
|
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src1.clCxt->supportsFeature(Context::CL_DOUBLE) && src1.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
cout << "Selected device do not support double" << endl;
|
|
|
|
return;
|
|
|
|
}
|
2013-04-15 10:36:03 +08:00
|
|
|
static const char opt [] = "-D OP_BINARY=|";
|
|
|
|
string kernelName = mask.data ? "arithm_s_bitwise_binary_with_mask" : "arithm_s_bitwise_binary";
|
2012-07-17 01:08:14 +08:00
|
|
|
if (mask.data)
|
2013-04-15 10:36:03 +08:00
|
|
|
bitwise_scalar( src1, src2, dst, mask, kernelName, &arithm_bitwise_binary_scalar_mask, opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
else
|
2013-04-15 10:36:03 +08:00
|
|
|
bitwise_scalar( src1, src2, dst, mask, kernelName, &arithm_bitwise_binary_scalar, opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void cv::ocl::bitwise_and(const oclMat &src1, const oclMat &src2, oclMat &dst, const oclMat &mask)
|
|
|
|
{
|
|
|
|
// dst.create(src1.size(),src1.type());
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src1.clCxt->supportsFeature(Context::CL_DOUBLE) && src1.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
cout << "Selected device do not support double" << endl;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
oclMat emptyMat;
|
|
|
|
|
2013-04-15 10:36:03 +08:00
|
|
|
string kernelName = mask.empty() ? "arithm_bitwise_binary" : "arithm_bitwise_binary_with_mask";
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-04-15 10:36:03 +08:00
|
|
|
static const char opt [] = "-D OP_BINARY=&";
|
2012-07-17 01:08:14 +08:00
|
|
|
if (mask.empty())
|
2013-04-15 10:36:03 +08:00
|
|
|
bitwise_run(src1, src2, dst, kernelName, &arithm_bitwise_binary, opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
else
|
2013-04-15 10:36:03 +08:00
|
|
|
bitwise_run(src1, src2, dst, mask, kernelName, &arithm_bitwise_binary_mask, opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void cv::ocl::bitwise_and(const oclMat &src1, const Scalar &src2, oclMat &dst, const oclMat &mask)
|
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src1.clCxt->supportsFeature(Context::CL_DOUBLE) && src1.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
cout << "Selected device do not support double" << endl;
|
|
|
|
return;
|
|
|
|
}
|
2013-04-15 10:36:03 +08:00
|
|
|
static const char opt [] = "-D OP_BINARY=&";
|
|
|
|
string kernelName = mask.data ? "arithm_s_bitwise_binary_with_mask" : "arithm_s_bitwise_binary";
|
2012-07-17 01:08:14 +08:00
|
|
|
if (mask.data)
|
2013-04-15 10:36:03 +08:00
|
|
|
bitwise_scalar(src1, src2, dst, mask, kernelName, &arithm_bitwise_binary_scalar_mask, opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
else
|
2013-04-15 10:36:03 +08:00
|
|
|
bitwise_scalar(src1, src2, dst, mask, kernelName, &arithm_bitwise_binary_scalar, opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void cv::ocl::bitwise_xor(const oclMat &src1, const oclMat &src2, oclMat &dst, const oclMat &mask)
|
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src1.clCxt->supportsFeature(Context::CL_DOUBLE) && src1.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
cout << "Selected device do not support double" << endl;
|
|
|
|
return;
|
|
|
|
}
|
2013-04-15 10:36:03 +08:00
|
|
|
string kernelName = mask.empty() ? "arithm_bitwise_binary" : "arithm_bitwise_binary_with_mask";
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-04-15 10:36:03 +08:00
|
|
|
static const char opt [] = "-D OP_BINARY=^";
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
if (mask.empty())
|
2013-04-15 10:36:03 +08:00
|
|
|
bitwise_run(src1, src2, dst, kernelName, &arithm_bitwise_binary, opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
else
|
2013-04-15 10:36:03 +08:00
|
|
|
bitwise_run(src1, src2, dst, mask, kernelName, &arithm_bitwise_binary_mask, opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void cv::ocl::bitwise_xor(const oclMat &src1, const Scalar &src2, oclMat &dst, const oclMat &mask)
|
|
|
|
{
|
|
|
|
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src1.clCxt->supportsFeature(Context::CL_DOUBLE) && src1.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
cout << "Selected device do not support double" << endl;
|
|
|
|
return;
|
|
|
|
}
|
2013-04-15 10:36:03 +08:00
|
|
|
string kernelName = mask.data ? "arithm_s_bitwise_binary_with_mask" : "arithm_s_bitwise_binary";
|
|
|
|
static const char opt [] = "-D OP_BINARY=^";
|
2012-07-17 01:08:14 +08:00
|
|
|
if (mask.data)
|
2013-04-15 10:36:03 +08:00
|
|
|
bitwise_scalar( src1, src2, dst, mask, kernelName, &arithm_bitwise_binary_scalar_mask, opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
else
|
2013-04-15 10:36:03 +08:00
|
|
|
bitwise_scalar( src1, src2, dst, mask, kernelName, &arithm_bitwise_binary_scalar, opt);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
2013-02-22 04:30:27 +08:00
|
|
|
oclMat cv::ocl::operator ~ (const oclMat &src)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-02-16 18:41:43 +08:00
|
|
|
return oclMatExpr(src, oclMat(), MAT_NOT);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
2013-02-22 04:30:27 +08:00
|
|
|
oclMat cv::ocl::operator | (const oclMat &src1, const oclMat &src2)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-02-16 18:41:43 +08:00
|
|
|
return oclMatExpr(src1, src2, MAT_OR);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
2013-02-22 04:30:27 +08:00
|
|
|
oclMat cv::ocl::operator & (const oclMat &src1, const oclMat &src2)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-02-16 18:41:43 +08:00
|
|
|
return oclMatExpr(src1, src2, MAT_AND);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
2013-02-22 04:30:27 +08:00
|
|
|
oclMat cv::ocl::operator ^ (const oclMat &src1, const oclMat &src2)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-02-16 18:41:43 +08:00
|
|
|
return oclMatExpr(src1, src2, MAT_XOR);
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
2013-02-16 11:05:23 +08:00
|
|
|
cv::ocl::oclMatExpr cv::ocl::operator + (const oclMat &src1, const oclMat &src2)
|
2013-02-08 11:41:46 +08:00
|
|
|
{
|
2013-02-16 18:41:43 +08:00
|
|
|
return oclMatExpr(src1, src2, cv::ocl::MAT_ADD);
|
2013-02-08 11:41:46 +08:00
|
|
|
}
|
|
|
|
|
2013-02-16 11:05:23 +08:00
|
|
|
cv::ocl::oclMatExpr cv::ocl::operator - (const oclMat &src1, const oclMat &src2)
|
2013-02-08 11:41:46 +08:00
|
|
|
{
|
2013-02-16 18:41:43 +08:00
|
|
|
return oclMatExpr(src1, src2, cv::ocl::MAT_SUB);
|
2013-02-08 11:41:46 +08:00
|
|
|
}
|
|
|
|
|
2013-02-16 11:05:23 +08:00
|
|
|
cv::ocl::oclMatExpr cv::ocl::operator * (const oclMat &src1, const oclMat &src2)
|
2013-02-08 11:41:46 +08:00
|
|
|
{
|
2013-02-16 18:41:43 +08:00
|
|
|
return oclMatExpr(src1, src2, cv::ocl::MAT_MUL);
|
2013-02-08 11:41:46 +08:00
|
|
|
}
|
|
|
|
|
2013-02-16 11:05:23 +08:00
|
|
|
cv::ocl::oclMatExpr cv::ocl::operator / (const oclMat &src1, const oclMat &src2)
|
2013-02-08 11:41:46 +08:00
|
|
|
{
|
2013-02-16 18:41:43 +08:00
|
|
|
return oclMatExpr(src1, src2, cv::ocl::MAT_DIV);
|
2013-02-08 11:41:46 +08:00
|
|
|
}
|
|
|
|
|
2013-02-16 11:05:23 +08:00
|
|
|
void oclMatExpr::assign(oclMat& m) const
|
|
|
|
{
|
|
|
|
switch (op)
|
|
|
|
{
|
|
|
|
case MAT_ADD:
|
|
|
|
add(a, b, m);
|
|
|
|
break;
|
|
|
|
case MAT_SUB:
|
|
|
|
subtract(a, b, m);
|
|
|
|
break;
|
|
|
|
case MAT_MUL:
|
|
|
|
multiply(a, b, m);
|
|
|
|
break;
|
|
|
|
case MAT_DIV:
|
|
|
|
divide(a, b, m);
|
|
|
|
break;
|
2013-02-16 18:41:43 +08:00
|
|
|
case MAT_NOT:
|
|
|
|
bitwise_not(a, m);
|
|
|
|
break;
|
|
|
|
case MAT_AND:
|
|
|
|
bitwise_and(a, b, m);
|
|
|
|
break;
|
|
|
|
case MAT_OR:
|
|
|
|
bitwise_or(a, b, m);
|
|
|
|
break;
|
|
|
|
case MAT_XOR:
|
|
|
|
bitwise_xor(a, b, m);
|
|
|
|
break;
|
2013-02-16 11:05:23 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
oclMatExpr::operator oclMat() const
|
|
|
|
{
|
|
|
|
oclMat m;
|
|
|
|
assign(m);
|
|
|
|
return m;
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
/////////////////////////////// transpose ////////////////////////////////////
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
#define TILE_DIM (32)
|
|
|
|
#define BLOCK_ROWS (256/TILE_DIM)
|
2013-01-25 18:31:34 +08:00
|
|
|
static void transpose_run(const oclMat &src, oclMat &dst, string kernelName)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src.clCxt->supportsFeature(Context::CL_DOUBLE) && src.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Error(CV_GpuNotSupported, "Selected device don't support double\r\n");
|
2012-07-17 01:08:14 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
CV_Assert(src.cols == dst.rows && src.rows == dst.cols);
|
|
|
|
|
|
|
|
Context *clCxt = src.clCxt;
|
2012-10-11 16:22:47 +08:00
|
|
|
int channels = src.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
int depth = src.depth();
|
|
|
|
|
|
|
|
int vector_lengths[4][7] = {{1, 0, 0, 0, 1, 1, 0},
|
|
|
|
{0, 0, 1, 1, 0, 0, 0},
|
|
|
|
{0, 0, 0, 0 , 0, 0, 0},
|
|
|
|
{1, 1, 0, 0, 0, 0, 0}
|
|
|
|
};
|
|
|
|
|
2012-10-11 16:22:47 +08:00
|
|
|
size_t vector_length = vector_lengths[channels - 1][depth];
|
2012-07-17 01:08:14 +08:00
|
|
|
int offset_cols = ((dst.offset % dst.step) / dst.elemSize()) & (vector_length - 1);
|
|
|
|
int cols = divUp(src.cols + offset_cols, vector_length);
|
|
|
|
|
|
|
|
size_t localThreads[3] = { TILE_DIM, BLOCK_ROWS, 1 };
|
2013-09-16 19:11:56 +08:00
|
|
|
size_t globalThreads[3] = { cols, src.rows, 1 };
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src.rows ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&cols ));
|
|
|
|
|
|
|
|
openCLExecuteKernel(clCxt, &arithm_transpose, kernelName, globalThreads, localThreads, args, channels, depth);
|
|
|
|
}
|
|
|
|
|
|
|
|
void cv::ocl::transpose(const oclMat &src, oclMat &dst)
|
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
CV_Assert(src.type() == CV_8UC1 || src.type() == CV_8UC3 || src.type() == CV_8UC4 || src.type() == CV_8SC3 || src.type() == CV_8SC4 ||
|
2012-07-17 01:08:14 +08:00
|
|
|
src.type() == CV_16UC2 || src.type() == CV_16SC2 || src.type() == CV_32SC1 || src.type() == CV_32FC1);
|
|
|
|
|
|
|
|
oclMat emptyMat;
|
|
|
|
|
|
|
|
if( src.data == dst.data && dst.cols == dst.rows )
|
|
|
|
transpose_run( src, emptyMat, "transposeI_");
|
|
|
|
else
|
|
|
|
{
|
|
|
|
dst.create(src.cols, src.rows, src.type());
|
|
|
|
transpose_run( src, dst, "transpose");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void cv::ocl::addWeighted(const oclMat &src1, double alpha, const oclMat &src2, double beta, double gama, oclMat &dst)
|
|
|
|
{
|
2013-09-24 17:51:37 +08:00
|
|
|
Context *clCxt = src1.clCxt;
|
|
|
|
bool hasDouble = clCxt->supportsFeature(Context::CL_DOUBLE);
|
|
|
|
if (!hasDouble && src1.depth() == CV_64F)
|
|
|
|
{
|
|
|
|
CV_Error(CV_GpuNotSupported, "Selected device doesn't support double\r\n");
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
CV_Assert(src1.size() == src2.size() && src1.type() == src2.type());
|
2012-07-17 01:08:14 +08:00
|
|
|
dst.create(src1.size(), src1.type());
|
|
|
|
|
2012-10-11 16:22:47 +08:00
|
|
|
int channels = dst.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
int depth = dst.depth();
|
|
|
|
|
2013-09-24 17:51:37 +08:00
|
|
|
int cols1 = src1.cols * channels;
|
|
|
|
int src1step1 = src1.step1(), src1offset1 = src1.offset / src1.elemSize1();
|
|
|
|
int src2step1 = src2.step1(), src2offset1 = src2.offset / src1.elemSize1();
|
|
|
|
int dststep1 = dst.step1(), dstoffset1 = dst.offset / dst.elemSize1();
|
2012-07-17 01:08:14 +08:00
|
|
|
|
2013-09-24 17:51:37 +08:00
|
|
|
const char * const typeMap[] = { "uchar", "char", "ushort", "short", "int", "float", "double" };
|
|
|
|
std::string buildOptions = format("-D T=%s -D WT=%s -D convertToT=convert_%s%s",
|
|
|
|
typeMap[depth], hasDouble ? "double" : "float", typeMap[depth],
|
|
|
|
depth >= CV_32F ? "" : "_sat_rte");
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
size_t localThreads[3] = { 256, 1, 1 };
|
2013-09-24 17:51:37 +08:00
|
|
|
size_t globalThreads[3] = { cols1, dst.rows, 1};
|
|
|
|
|
|
|
|
float alpha_f = static_cast<float>(alpha),
|
|
|
|
beta_f = static_cast<float>(beta),
|
|
|
|
gama_f = static_cast<float>(gama);
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
vector<pair<size_t , const void *> > args;
|
2012-08-03 14:08:36 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src1.data ));
|
2013-09-24 17:51:37 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1step1 ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1offset1));
|
2012-08-03 14:08:36 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src2.data ));
|
2013-09-24 17:51:37 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2step1 ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src2offset1));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dststep1 ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dstoffset1));
|
2012-08-03 14:08:36 +08:00
|
|
|
|
2013-09-24 17:51:37 +08:00
|
|
|
if (!hasDouble)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
2012-09-21 16:51:45 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_float), (void *)&alpha_f ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_float), (void *)&beta_f ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_float), (void *)&gama_f ));
|
2012-10-11 16:22:47 +08:00
|
|
|
}
|
2013-09-24 17:51:37 +08:00
|
|
|
else
|
|
|
|
{
|
|
|
|
args.push_back( make_pair( sizeof(cl_double), (void *)&alpha ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_double), (void *)&beta ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_double), (void *)&gama ));
|
|
|
|
}
|
2012-08-03 14:08:36 +08:00
|
|
|
|
2013-09-24 17:51:37 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&cols1 ));
|
2012-08-03 14:08:36 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.rows ));
|
|
|
|
|
2013-09-24 17:51:37 +08:00
|
|
|
openCLExecuteKernel(clCxt, &arithm_addWeighted, "addWeighted", globalThreads, localThreads,
|
|
|
|
args, -1, -1, buildOptions.c_str());
|
2012-07-17 01:08:14 +08:00
|
|
|
}
|
|
|
|
|
2013-01-25 18:31:34 +08:00
|
|
|
static void arithmetic_pow_run(const oclMat &src1, double p, oclMat &dst, string kernelName, const char **kernelString)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
CV_Assert(src1.cols == dst.cols && src1.rows == dst.rows);
|
|
|
|
CV_Assert(src1.type() == dst.type());
|
|
|
|
|
|
|
|
Context *clCxt = src1.clCxt;
|
2012-10-11 16:22:47 +08:00
|
|
|
int channels = dst.oclchannels();
|
2012-07-17 01:08:14 +08:00
|
|
|
int depth = dst.depth();
|
|
|
|
|
|
|
|
size_t vector_length = 1;
|
|
|
|
int offset_cols = ((dst.offset % dst.step) / dst.elemSize1()) & (vector_length - 1);
|
|
|
|
int cols = divUp(dst.cols * channels + offset_cols, vector_length);
|
|
|
|
int rows = dst.rows;
|
|
|
|
|
|
|
|
size_t localThreads[3] = { 64, 4, 1 };
|
2013-09-16 19:11:56 +08:00
|
|
|
size_t globalThreads[3] = { cols, rows, 1 };
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
int dst_step1 = dst.cols * dst.elemSize();
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src1.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src1.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&dst.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.step ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.offset ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst.rows ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&cols ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&dst_step1 ));
|
2013-05-28 17:27:55 +08:00
|
|
|
float pf = p;
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!src1.clCxt->supportsFeature(Context::CL_DOUBLE))
|
2012-09-21 16:51:45 +08:00
|
|
|
{
|
2012-10-11 16:22:47 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_float), (void *)&pf ));
|
2012-09-21 16:51:45 +08:00
|
|
|
}
|
|
|
|
else
|
2012-10-11 16:22:47 +08:00
|
|
|
args.push_back( make_pair( sizeof(cl_double), (void *)&p ));
|
2012-07-17 01:08:14 +08:00
|
|
|
|
|
|
|
openCLExecuteKernel(clCxt, kernelString, kernelName, globalThreads, localThreads, args, -1, depth);
|
|
|
|
}
|
|
|
|
void cv::ocl::pow(const oclMat &x, double p, oclMat &y)
|
|
|
|
{
|
2013-03-18 05:59:24 +08:00
|
|
|
if(!x.clCxt->supportsFeature(Context::CL_DOUBLE) && x.type() == CV_64F)
|
2012-07-17 01:08:14 +08:00
|
|
|
{
|
|
|
|
cout << "Selected device do not support double" << endl;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2013-08-22 21:38:55 +08:00
|
|
|
CV_Assert(x.depth() == CV_32F || x.depth() == CV_64F);
|
2012-07-17 01:08:14 +08:00
|
|
|
y.create(x.size(), x.type());
|
|
|
|
string kernelName = "arithm_pow";
|
|
|
|
|
|
|
|
arithmetic_pow_run(x, p, y, kernelName, &arithm_pow);
|
|
|
|
}
|
2013-09-09 16:08:23 +08:00
|
|
|
void cv::ocl::setIdentity(oclMat& src, double scalar)
|
|
|
|
{
|
|
|
|
CV_Assert(src.empty() == false && src.rows == src.cols);
|
|
|
|
CV_Assert(src.type() == CV_32SC1 || src.type() == CV_32FC1);
|
|
|
|
int src_step = src.step/src.elemSize();
|
|
|
|
Context *clCxt = Context::getContext();
|
|
|
|
size_t local_threads[] = {16, 16, 1};
|
|
|
|
size_t global_threads[] = {src.cols, src.rows, 1};
|
|
|
|
|
|
|
|
string kernelName = "setIdentityKernel";
|
|
|
|
if(src.type() == CV_32FC1)
|
|
|
|
kernelName += "_F1";
|
|
|
|
else if(src.type() == CV_32SC1)
|
|
|
|
kernelName += "_I1";
|
|
|
|
else
|
|
|
|
{
|
|
|
|
kernelName += "_D1";
|
|
|
|
if(!(clCxt->supportsFeature(Context::CL_DOUBLE)))
|
|
|
|
{
|
|
|
|
oclMat temp;
|
|
|
|
src.convertTo(temp, CV_32FC1);
|
|
|
|
temp.copyTo(src);
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
vector<pair<size_t , const void *> > args;
|
|
|
|
args.push_back( make_pair( sizeof(cl_mem), (void *)&src.data ));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src.rows));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src.cols));
|
|
|
|
args.push_back( make_pair( sizeof(cl_int), (void *)&src_step ));
|
|
|
|
|
|
|
|
int scalar_i = 0;
|
|
|
|
float scalar_f = 0.0f;
|
|
|
|
if(clCxt->supportsFeature(Context::CL_DOUBLE))
|
|
|
|
{
|
|
|
|
if(src.type() == CV_32SC1)
|
|
|
|
{
|
|
|
|
scalar_i = (int)scalar;
|
|
|
|
args.push_back(make_pair(sizeof(cl_int), (void*)&scalar_i));
|
|
|
|
}else
|
|
|
|
args.push_back(make_pair(sizeof(cl_double), (void*)&scalar));
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
if(src.type() == CV_32SC1)
|
|
|
|
{
|
|
|
|
scalar_i = (int)scalar;
|
|
|
|
args.push_back(make_pair(sizeof(cl_int), (void*)&scalar_i));
|
|
|
|
}else
|
|
|
|
{
|
|
|
|
scalar_f = (float)scalar;
|
|
|
|
args.push_back(make_pair(sizeof(cl_float), (void*)&scalar_f));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
openCLExecuteKernel(clCxt, &arithm_setidentity, kernelName, global_threads, local_threads, args, -1, -1);
|
|
|
|
}
|