diff options
Diffstat (limited to 'src/boost/libs/compute/example/opencv_flip.cpp')
-rw-r--r-- | src/boost/libs/compute/example/opencv_flip.cpp | 101 |
1 files changed, 101 insertions, 0 deletions
diff --git a/src/boost/libs/compute/example/opencv_flip.cpp b/src/boost/libs/compute/example/opencv_flip.cpp new file mode 100644 index 00000000..8cc26ef2 --- /dev/null +++ b/src/boost/libs/compute/example/opencv_flip.cpp @@ -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://boostorg.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/system.hpp> +#include <boost/compute/interop/opencv/core.hpp> +#include <boost/compute/interop/opencv/highgui.hpp> +#include <boost/compute/utility/source.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, + input_image.width(), + input_image.height(), + input_image.format(), + compute::image2d::write_only + ); + + // 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; +} |