mirror of
https://github.com/opencv/opencv.git
synced 2024-12-12 07:09:12 +08:00
165 lines
5.3 KiB
C++
165 lines
5.3 KiB
C++
// This file is part of OpenCV project.
|
|
// It is subject to the license terms in the LICENSE file found in the top-level directory
|
|
// of this distribution and at http://opencv.org/license.html
|
|
|
|
#include "opencv2/core.hpp"
|
|
#include "opencv2/core/ocl.hpp"
|
|
#include "opencv2/highgui.hpp"
|
|
#include "opencv2/imgcodecs.hpp"
|
|
#include "opencv2/imgproc.hpp"
|
|
|
|
#include <iostream>
|
|
|
|
using namespace std;
|
|
using namespace cv;
|
|
|
|
static const char* opencl_kernel_src =
|
|
"__kernel void magnutude_filter_8u(\n"
|
|
" __global const uchar* src, int src_step, int src_offset,\n"
|
|
" __global uchar* dst, int dst_step, int dst_offset, int dst_rows, int dst_cols,\n"
|
|
" float scale)\n"
|
|
"{\n"
|
|
" int x = get_global_id(0);\n"
|
|
" int y = get_global_id(1);\n"
|
|
" if (x < dst_cols && y < dst_rows)\n"
|
|
" {\n"
|
|
" int dst_idx = y * dst_step + x + dst_offset;\n"
|
|
" if (x > 0 && x < dst_cols - 1 && y > 0 && y < dst_rows - 2)\n"
|
|
" {\n"
|
|
" int src_idx = y * src_step + x + src_offset;\n"
|
|
" int dx = (int)src[src_idx]*2 - src[src_idx - 1] - src[src_idx + 1];\n"
|
|
" int dy = (int)src[src_idx]*2 - src[src_idx - 1*src_step] - src[src_idx + 1*src_step];\n"
|
|
" dst[dst_idx] = convert_uchar_sat(sqrt((float)(dx*dx + dy*dy)) * scale);\n"
|
|
" }\n"
|
|
" else\n"
|
|
" {\n"
|
|
" dst[dst_idx] = 0;\n"
|
|
" }\n"
|
|
" }\n"
|
|
"}\n";
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
const char* keys =
|
|
"{ i input | | specify input image }"
|
|
"{ h help | | print help message }";
|
|
|
|
cv::CommandLineParser args(argc, argv, keys);
|
|
if (args.has("help"))
|
|
{
|
|
cout << "Usage : " << argv[0] << " [options]" << endl;
|
|
cout << "Available options:" << endl;
|
|
args.printMessage();
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|
|
cv::ocl::Context ctx = cv::ocl::Context::getDefault();
|
|
if (!ctx.ptr())
|
|
{
|
|
cerr << "OpenCL is not available" << endl;
|
|
return 1;
|
|
}
|
|
cv::ocl::Device device = cv::ocl::Device::getDefault();
|
|
if (!device.compilerAvailable())
|
|
{
|
|
cerr << "OpenCL compiler is not available" << endl;
|
|
return 1;
|
|
}
|
|
|
|
|
|
UMat src;
|
|
{
|
|
string image_file = args.get<string>("i");
|
|
if (!image_file.empty())
|
|
{
|
|
Mat image = imread(samples::findFile(image_file));
|
|
if (image.empty())
|
|
{
|
|
cout << "error read image: " << image_file << endl;
|
|
return 1;
|
|
}
|
|
cvtColor(image, src, COLOR_BGR2GRAY);
|
|
}
|
|
else
|
|
{
|
|
Mat frame(cv::Size(640, 480), CV_8U, Scalar::all(128));
|
|
Point p(frame.cols / 2, frame.rows / 2);
|
|
line(frame, Point(0, frame.rows / 2), Point(frame.cols, frame.rows / 2), 1);
|
|
circle(frame, p, 200, Scalar(32, 32, 32), 8, LINE_AA);
|
|
string str = "OpenCL";
|
|
int baseLine = 0;
|
|
Size box = getTextSize(str, FONT_HERSHEY_COMPLEX, 2, 5, &baseLine);
|
|
putText(frame, str, Point((frame.cols - box.width) / 2, (frame.rows - box.height) / 2 + baseLine),
|
|
FONT_HERSHEY_COMPLEX, 2, Scalar(255, 255, 255), 5, LINE_AA);
|
|
frame.copyTo(src);
|
|
}
|
|
}
|
|
|
|
|
|
cv::String module_name; // empty to disable OpenCL cache
|
|
|
|
{
|
|
cout << "OpenCL program source: " << endl;
|
|
cout << "======================================================================================================" << endl;
|
|
cout << opencl_kernel_src << endl;
|
|
cout << "======================================================================================================" << endl;
|
|
//! [Define OpenCL program source]
|
|
cv::ocl::ProgramSource source(module_name, "simple", opencl_kernel_src, "");
|
|
//! [Define OpenCL program source]
|
|
|
|
//! [Compile/build OpenCL for current OpenCL device]
|
|
cv::String errmsg;
|
|
cv::ocl::Program program(source, "", errmsg);
|
|
if (program.ptr() == NULL)
|
|
{
|
|
cerr << "Can't compile OpenCL program:" << endl << errmsg << endl;
|
|
return 1;
|
|
}
|
|
//! [Compile/build OpenCL for current OpenCL device]
|
|
|
|
if (!errmsg.empty())
|
|
{
|
|
cout << "OpenCL program build log:" << endl << errmsg << endl;
|
|
}
|
|
|
|
//! [Get OpenCL kernel by name]
|
|
cv::ocl::Kernel k("magnutude_filter_8u", program);
|
|
if (k.empty())
|
|
{
|
|
cerr << "Can't get OpenCL kernel" << endl;
|
|
return 1;
|
|
}
|
|
//! [Get OpenCL kernel by name]
|
|
|
|
UMat result(src.size(), CV_8UC1);
|
|
|
|
//! [Define kernel parameters and run]
|
|
size_t globalSize[2] = {(size_t)src.cols, (size_t)src.rows};
|
|
size_t localSize[2] = {8, 8};
|
|
bool executionResult = k
|
|
.args(
|
|
cv::ocl::KernelArg::ReadOnlyNoSize(src), // size is not used (similar to 'dst' size)
|
|
cv::ocl::KernelArg::WriteOnly(result),
|
|
(float)2.0
|
|
)
|
|
.run(2, globalSize, localSize, true);
|
|
if (!executionResult)
|
|
{
|
|
cerr << "OpenCL kernel launch failed" << endl;
|
|
return 1;
|
|
}
|
|
//! [Define kernel parameters and run]
|
|
|
|
imshow("Source", src);
|
|
imshow("Result", result);
|
|
|
|
for (;;)
|
|
{
|
|
int key = waitKey();
|
|
if (key == 27/*ESC*/ || key == 'q' || key == 'Q')
|
|
break;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|