summaryrefslogtreecommitdiffstats
path: root/src/boost/libs/compute/example/opencv_flip.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/boost/libs/compute/example/opencv_flip.cpp')
-rw-r--r--src/boost/libs/compute/example/opencv_flip.cpp101
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;
+}