mirror of
https://github.com/boostorg/compute.git
synced 2026-01-24 17:52:37 +00:00
Add interop support
This adds interoperability support between Boost.Compute and various other C/C++ libraries (Eigen, OpenCV, OpenGL, Qt and VTK). This eases development for users using external libraries with Boost.Compute.
This commit is contained in:
101
example/opencv_filter.cpp
Normal file
101
example/opencv_filter.cpp
Normal file
@@ -0,0 +1,101 @@
|
||||
//---------------------------------------------------------------------------//
|
||||
// Copyright (c) 2013-2014 Kyle Lutz <kyle.r.lutz@gmail.com>
|
||||
//
|
||||
// Distributed under the Boost Software License, Version 1.0
|
||||
// See accompanying file LICENSE_1_0.txt or copy at
|
||||
// http://www.boost.org/LICENSE_1_0.txt
|
||||
//
|
||||
// See http://kylelutz.github.com/compute for more information.
|
||||
//---------------------------------------------------------------------------//
|
||||
|
||||
#include <iostream>
|
||||
|
||||
#include <opencv2/core/core.hpp>
|
||||
#include <opencv2/highgui/highgui.hpp>
|
||||
#include <opencv2/imgproc/imgproc.hpp>
|
||||
|
||||
#include <boost/compute/source.hpp>
|
||||
#include <boost/compute/system.hpp>
|
||||
#include <boost/compute/interop/opencv/core.hpp>
|
||||
#include <boost/compute/interop/opencv/highgui.hpp>
|
||||
|
||||
namespace compute = boost::compute;
|
||||
|
||||
// this example shows how to read an image with OpenCV, transfer the
|
||||
// image to the GPU, and apply a simple flip filter written in OpenCL
|
||||
int main(int argc, char *argv[])
|
||||
{
|
||||
// check command line
|
||||
if(argc < 2){
|
||||
std::cerr << "usage: " << argv[0] << " FILENAME" << std::endl;
|
||||
return -1;
|
||||
}
|
||||
|
||||
// read image with opencv
|
||||
cv::Mat cv_image = cv::imread(argv[1], CV_LOAD_IMAGE_COLOR);
|
||||
if(!cv_image.data){
|
||||
std::cerr << "failed to load image" << std::endl;
|
||||
return -1;
|
||||
}
|
||||
|
||||
// get default device and setup context
|
||||
compute::device gpu = compute::system::default_device();
|
||||
compute::context context(gpu);
|
||||
compute::command_queue queue(context, gpu);
|
||||
|
||||
// convert image to BGRA (OpenCL requires 16-byte aligned data)
|
||||
cv::cvtColor(cv_image, cv_image, CV_BGR2BGRA);
|
||||
|
||||
// transfer image to gpu
|
||||
compute::image2d input_image =
|
||||
compute::opencv_create_image2d_with_mat(
|
||||
cv_image, compute::image2d::read_write, queue
|
||||
);
|
||||
|
||||
// create output image
|
||||
compute::image2d output_image(
|
||||
context,
|
||||
compute::image2d::write_only,
|
||||
input_image.get_format(),
|
||||
input_image.width(),
|
||||
input_image.height()
|
||||
);
|
||||
|
||||
// create flip program
|
||||
const char source[] = BOOST_COMPUTE_STRINGIZE_SOURCE(
|
||||
__kernel void flip_kernel(__read_only image2d_t input,
|
||||
__write_only image2d_t output)
|
||||
{
|
||||
const sampler_t sampler = CLK_ADDRESS_NONE | CLK_FILTER_NEAREST;
|
||||
int height = get_image_height(input);
|
||||
int2 input_coord = { get_global_id(0), get_global_id(1) };
|
||||
int2 output_coord = { input_coord.x, height - input_coord.y - 1 };
|
||||
float4 value = read_imagef(input, sampler, input_coord);
|
||||
write_imagef(output, output_coord, value);
|
||||
}
|
||||
);
|
||||
|
||||
compute::program flip_program =
|
||||
compute::program::create_with_source(source, context);
|
||||
flip_program.build();
|
||||
|
||||
// create flip kernel and set arguments
|
||||
compute::kernel flip_kernel(flip_program, "flip_kernel");
|
||||
flip_kernel.set_arg(0, input_image);
|
||||
flip_kernel.set_arg(1, output_image);
|
||||
|
||||
// run flip kernel
|
||||
size_t origin[2] = { 0, 0 };
|
||||
size_t region[2] = { input_image.width(), input_image.height() };
|
||||
queue.enqueue_nd_range_kernel(flip_kernel, 2, origin, region, 0);
|
||||
|
||||
// show host image
|
||||
cv::imshow("opencv image", cv_image);
|
||||
|
||||
// show gpu image
|
||||
compute::opencv_imshow("filtered image", output_image, queue);
|
||||
|
||||
// wait and return
|
||||
cv::waitKey(0);
|
||||
return 0;
|
||||
}
|
||||
Reference in New Issue
Block a user