78be4f66f7
Conflicts: CMakeLists.txt modules/calib3d/src/calibration.cpp modules/ocl/src/cl_programcache.cpp modules/ocl/src/filtering.cpp modules/ocl/src/imgproc.cpp samples/ocl/adaptive_bilateral_filter.cpp samples/ocl/bgfg_segm.cpp samples/ocl/clahe.cpp samples/ocl/facedetect.cpp samples/ocl/pyrlk_optical_flow.cpp samples/ocl/squares.cpp samples/ocl/surf_matcher.cpp samples/ocl/tvl1_optical_flow.cpp
306 lines
14 KiB
C++
306 lines
14 KiB
C++
/*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.
|
|
// Third party copyrights are property of their respective owners.
|
|
//
|
|
// @Authors
|
|
// Wang Weiyan, wangweiyanster@gmail.com
|
|
// Peng Xiao, pengxiao@multicorewareinc.com
|
|
//
|
|
// 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 materials 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 "opencl_kernels.hpp"
|
|
|
|
using namespace cv;
|
|
using namespace cv::ocl;
|
|
|
|
#ifndef CV_DESCALE
|
|
#define CV_DESCALE(x, n) (((x) + (1 << ((n)-1))) >> (n))
|
|
#endif
|
|
|
|
#ifndef FLT_EPSILON
|
|
#define FLT_EPSILON 1.192092896e-07F
|
|
#endif
|
|
|
|
namespace
|
|
{
|
|
|
|
void RGB2Gray_caller(const oclMat &src, oclMat &dst, int bidx)
|
|
{
|
|
int channels = src.oclchannels();
|
|
int src_offset = src.offset / src.elemSize1(), src_step = src.step1();
|
|
int dst_offset = dst.offset / dst.elemSize1(), dst_step = dst.step1();
|
|
|
|
String build_options = format("-D DEPTH_%d", src.depth());
|
|
|
|
std::vector<std::pair<size_t , const void *> > args;
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src.cols));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src.rows));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src_step));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&dst_step));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&channels));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&bidx));
|
|
args.push_back( std::make_pair( sizeof(cl_mem) , (void *)&src.data));
|
|
args.push_back( std::make_pair( sizeof(cl_mem) , (void *)&dst.data));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src_offset ));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&dst_offset ));
|
|
|
|
size_t gt[3] = {src.cols, src.rows, 1}, lt[3] = {16, 16, 1};
|
|
openCLExecuteKernel(src.clCxt, &cvt_color, "RGB2Gray", gt, lt, args, -1, -1, build_options.c_str());
|
|
}
|
|
|
|
void Gray2RGB_caller(const oclMat &src, oclMat &dst)
|
|
{
|
|
String build_options = format("-D DEPTH_%d", src.depth());
|
|
int src_offset = src.offset / src.elemSize1(), src_step = src.step1();
|
|
int dst_offset = dst.offset / dst.elemSize1(), dst_step = dst.step1();
|
|
|
|
std::vector<std::pair<size_t , const void *> > args;
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src.cols));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src.rows));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src_step));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&dst_step));
|
|
args.push_back( std::make_pair( sizeof(cl_mem) , (void *)&src.data));
|
|
args.push_back( std::make_pair( sizeof(cl_mem) , (void *)&dst.data));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src_offset ));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&dst_offset ));
|
|
|
|
size_t gt[3] = {src.cols, src.rows, 1}, lt[3] = {16, 16, 1};
|
|
openCLExecuteKernel(src.clCxt, &cvt_color, "Gray2RGB", gt, lt, args, -1, -1, build_options.c_str());
|
|
}
|
|
|
|
void RGB2YUV_caller(const oclMat &src, oclMat &dst, int bidx)
|
|
{
|
|
int channels = src.oclchannels();
|
|
String build_options = format("-D DEPTH_%d", src.depth());
|
|
int src_offset = src.offset / src.elemSize1(), src_step = src.step1();
|
|
int dst_offset = dst.offset / dst.elemSize1(), dst_step = dst.step1();
|
|
|
|
std::vector<std::pair<size_t , const void *> > args;
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src.cols));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src.rows));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src_step));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&dst_step));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&channels));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&bidx));
|
|
args.push_back( std::make_pair( sizeof(cl_mem) , (void *)&src.data));
|
|
args.push_back( std::make_pair( sizeof(cl_mem) , (void *)&dst.data));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src_offset ));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&dst_offset ));
|
|
|
|
size_t gt[3] = {src.cols, src.rows, 1}, lt[3] = {16, 16, 1};
|
|
openCLExecuteKernel(src.clCxt, &cvt_color, "RGB2YUV", gt, lt, args, -1, -1, build_options.c_str());
|
|
}
|
|
|
|
void YUV2RGB_caller(const oclMat &src, oclMat &dst, int bidx)
|
|
{
|
|
int channels = src.oclchannels();
|
|
int src_offset = src.offset / src.elemSize1(), src_step = src.step1();
|
|
int dst_offset = dst.offset / dst.elemSize1(), dst_step = dst.step1();
|
|
|
|
String buildOptions = format("-D DEPTH_%d", src.depth());
|
|
|
|
std::vector<std::pair<size_t , const void *> > args;
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src.cols));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src.rows));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src_step));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&dst_step));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&channels));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&bidx));
|
|
args.push_back( std::make_pair( sizeof(cl_mem) , (void *)&src.data));
|
|
args.push_back( std::make_pair( sizeof(cl_mem) , (void *)&dst.data));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src_offset ));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&dst_offset ));
|
|
|
|
size_t gt[3] = {src.cols, src.rows, 1}, lt[3] = {16, 16, 1};
|
|
openCLExecuteKernel(src.clCxt, &cvt_color, "YUV2RGB", gt, lt, args, -1, -1, buildOptions.c_str());
|
|
}
|
|
|
|
void YUV2RGB_NV12_caller(const oclMat &src, oclMat &dst, int bidx)
|
|
{
|
|
String build_options = format("-D DEPTH_%d", src.depth());
|
|
int src_offset = src.offset / src.elemSize1(), src_step = src.step1();
|
|
int dst_offset = dst.offset / dst.elemSize1(), dst_step = dst.step1();
|
|
|
|
std::vector<std::pair<size_t , const void *> > args;
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src.cols));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src.rows));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src_step));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&dst_step));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&bidx));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&dst.cols));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&dst.rows));
|
|
args.push_back( std::make_pair( sizeof(cl_mem) , (void *)&src.data));
|
|
args.push_back( std::make_pair( sizeof(cl_mem) , (void *)&dst.data));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src_offset ));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&dst_offset ));
|
|
|
|
size_t gt[3] = {dst.cols / 2, dst.rows / 2, 1}, lt[3] = {16, 16, 1};
|
|
openCLExecuteKernel(src.clCxt, &cvt_color, "YUV2RGBA_NV12", gt, lt, args, -1, -1, build_options.c_str());
|
|
}
|
|
|
|
void RGB2YCrCb_caller(const oclMat &src, oclMat &dst, int bidx)
|
|
{
|
|
int channels = src.oclchannels();
|
|
String build_options = format("-D DEPTH_%d", src.depth());
|
|
int src_offset = src.offset / src.elemSize1(), src_step = src.step1();
|
|
int dst_offset = dst.offset / dst.elemSize1(), dst_step = dst.step1();
|
|
|
|
std::vector<std::pair<size_t , const void *> > args;
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src.cols));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src.rows));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src_step));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&dst_step));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&channels));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&bidx));
|
|
args.push_back( std::make_pair( sizeof(cl_mem) , (void *)&src.data));
|
|
args.push_back( std::make_pair( sizeof(cl_mem) , (void *)&dst.data));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&src_offset ));
|
|
args.push_back( std::make_pair( sizeof(cl_int) , (void *)&dst_offset ));
|
|
|
|
size_t gt[3] = {src.cols, src.rows, 1}, lt[3] = {16, 16, 1};
|
|
openCLExecuteKernel(src.clCxt, &cvt_color, "RGB2YCrCb", gt, lt, args, -1, -1, build_options.c_str());
|
|
}
|
|
|
|
void cvtColor_caller(const oclMat &src, oclMat &dst, int code, int dcn)
|
|
{
|
|
Size sz = src.size();
|
|
int scn = src.oclchannels(), depth = src.depth(), bidx;
|
|
|
|
CV_Assert(depth == CV_8U || depth == CV_16U || depth == CV_32F);
|
|
|
|
switch (code)
|
|
{
|
|
/*
|
|
case COLOR_BGR2BGRA: case COLOR_RGB2BGRA: case COLOR_BGRA2BGR:
|
|
case COLOR_RGBA2BGR: case COLOR_RGB2BGR: case COLOR_BGRA2RGBA:
|
|
case COLOR_BGR2BGR565: case COLOR_BGR2BGR555: case COLOR_RGB2BGR565: case COLOR_RGB2BGR555:
|
|
case COLOR_BGRA2BGR565: case COLOR_BGRA2BGR555: case COLOR_RGBA2BGR565: case COLOR_RGBA2BGR555:
|
|
case COLOR_BGR5652BGR: case COLOR_BGR5552BGR: case COLOR_BGR5652RGB: case COLOR_BGR5552RGB:
|
|
case COLOR_BGR5652BGRA: case COLOR_BGR5552BGRA: case COLOR_BGR5652RGBA: case COLOR_BGR5552RGBA:
|
|
*/
|
|
case COLOR_BGR2GRAY:
|
|
case COLOR_BGRA2GRAY:
|
|
case COLOR_RGB2GRAY:
|
|
case COLOR_RGBA2GRAY:
|
|
{
|
|
CV_Assert(scn == 3 || scn == 4);
|
|
bidx = code == COLOR_BGR2GRAY || code == COLOR_BGRA2GRAY ? 0 : 2;
|
|
dst.create(sz, CV_MAKETYPE(depth, 1));
|
|
RGB2Gray_caller(src, dst, bidx);
|
|
break;
|
|
}
|
|
case COLOR_GRAY2BGR:
|
|
case COLOR_GRAY2BGRA:
|
|
{
|
|
CV_Assert(scn == 1);
|
|
dcn = code == COLOR_GRAY2BGRA ? 4 : 3;
|
|
dst.create(sz, CV_MAKETYPE(depth, dcn));
|
|
Gray2RGB_caller(src, dst);
|
|
break;
|
|
}
|
|
case COLOR_BGR2YUV:
|
|
case COLOR_RGB2YUV:
|
|
{
|
|
CV_Assert(scn == 3 || scn == 4);
|
|
bidx = code == COLOR_RGB2YUV ? 0 : 2;
|
|
dst.create(sz, CV_MAKETYPE(depth, 3));
|
|
RGB2YUV_caller(src, dst, bidx);
|
|
break;
|
|
}
|
|
case COLOR_YUV2BGR:
|
|
case COLOR_YUV2RGB:
|
|
{
|
|
CV_Assert(scn == 3 || scn == 4);
|
|
bidx = code == COLOR_YUV2RGB ? 0 : 2;
|
|
dst.create(sz, CV_MAKETYPE(depth, 3));
|
|
YUV2RGB_caller(src, dst, bidx);
|
|
break;
|
|
}
|
|
case COLOR_YUV2RGB_NV12:
|
|
case COLOR_YUV2BGR_NV12:
|
|
case COLOR_YUV2RGBA_NV12:
|
|
case COLOR_YUV2BGRA_NV12:
|
|
{
|
|
CV_Assert(scn == 1);
|
|
CV_Assert( sz.width % 2 == 0 && sz.height % 3 == 0 && depth == CV_8U );
|
|
dcn = code == COLOR_YUV2BGRA_NV12 || code == COLOR_YUV2RGBA_NV12 ? 4 : 3;
|
|
bidx = code == COLOR_YUV2BGRA_NV12 || code == COLOR_YUV2BGR_NV12 ? 0 : 2;
|
|
|
|
Size dstSz(sz.width, sz.height * 2 / 3);
|
|
dst.create(dstSz, CV_MAKETYPE(depth, dcn));
|
|
YUV2RGB_NV12_caller(src, dst, bidx);
|
|
break;
|
|
}
|
|
case COLOR_BGR2YCrCb:
|
|
case COLOR_RGB2YCrCb:
|
|
{
|
|
CV_Assert(scn == 3 || scn == 4);
|
|
bidx = code == COLOR_BGR2YCrCb ? 0 : 2;
|
|
dst.create(sz, CV_MAKETYPE(depth, 3));
|
|
RGB2YCrCb_caller(src, dst, bidx);
|
|
break;
|
|
}
|
|
case COLOR_YCrCb2BGR:
|
|
case COLOR_YCrCb2RGB:
|
|
{
|
|
break;
|
|
}
|
|
/*
|
|
case COLOR_BGR5652GRAY: case COLOR_BGR5552GRAY:
|
|
case COLOR_GRAY2BGR565: case COLOR_GRAY2BGR555:
|
|
case COLOR_BGR2YCrCb: case COLOR_RGB2YCrCb:
|
|
case COLOR_BGR2XYZ: case COLOR_RGB2XYZ:
|
|
case COLOR_XYZ2BGR: case COLOR_XYZ2RGB:
|
|
case COLOR_BGR2HSV: case COLOR_RGB2HSV: case COLOR_BGR2HSV_FULL: case COLOR_RGB2HSV_FULL:
|
|
case COLOR_BGR2HLS: case COLOR_RGB2HLS: case COLOR_BGR2HLS_FULL: case COLOR_RGB2HLS_FULL:
|
|
case COLOR_HSV2BGR: case COLOR_HSV2RGB: case COLOR_HSV2BGR_FULL: case COLOR_HSV2RGB_FULL:
|
|
case COLOR_HLS2BGR: case COLOR_HLS2RGB: case COLOR_HLS2BGR_FULL: case COLOR_HLS2RGB_FULL:
|
|
*/
|
|
default:
|
|
CV_Error(Error::StsBadFlag, "Unknown/unsupported color conversion code" );
|
|
}
|
|
}
|
|
}
|
|
|
|
void cv::ocl::cvtColor(const oclMat &src, oclMat &dst, int code, int dcn)
|
|
{
|
|
cvtColor_caller(src, dst, code, dcn);
|
|
}
|