gpu morphology sample

diff --git a/samples/gpu/morfology.cpp b/samples/gpu/morfology.cpp
new file mode 100644 (file)
index 0000000..702125b
--- /dev/null
+++ b/samples/gpu/morfology.cpp
@@ -0,0 +1,107 @@
+\r
+#include <opencv2/imgproc/imgproc.hpp>\r
+#include <opencv2/highgui/highgui.hpp>\r
+#include <opencv2/gpu/gpu.hpp>\r
+#include <stdlib.h>\r
+#include <stdio.h>\r
+\r
+using namespace cv;\r
+using namespace cv::gpu;\r
+\r
+void help()\r
+{\r
+\r
+printf("\nShow off image morphology: erosion, dialation, open and close\n"\r
+       "Call:\n   morphology2 [image]\n"\r
+       "This program also shows use of rect, elipse and cross kernels\n\n");\r
+printf( "Hot keys: \n"\r
+    "\tESC - quit the program\n"\r
+    "\tr - use rectangle structuring element\n"\r
+    "\te - use elliptic structuring element\n"\r
+    "\tc - use cross-shaped structuring element\n"\r
+    "\tSPACE - loop through all the options\n" );\r
+}\r
+\r
+GpuMat src, dst;\r
+\r
+int element_shape = MORPH_RECT;\r
+\r
+//the address of variable which receives trackbar position update\r
+int max_iters = 10;\r
+int open_close_pos = 0;\r
+int erode_dilate_pos = 0;\r
+\r
+// callback function for open/close trackbar\r
+void OpenClose(int, void*)\r
+{\r
+    int n = open_close_pos - max_iters;\r
+    int an = n > 0 ? n : -n;\r
+    Mat element = getStructuringElement(element_shape, Size(an*2+1, an*2+1), Point(an, an) );\r
+    if( n < 0 )\r
+               cv::gpu::morphologyEx(src, dst, CV_MOP_OPEN, element);\r
+    else\r
+        cv::gpu::morphologyEx(src, dst, CV_MOP_CLOSE, element);\r
+    imshow("Open/Close",dst);\r
+}\r
+\r
+// callback function for erode/dilate trackbar\r
+void ErodeDilate(int, void*)\r
+{\r
+    int n = erode_dilate_pos - max_iters;\r
+    int an = n > 0 ? n : -n;\r
+    Mat element = getStructuringElement(element_shape, Size(an*2+1, an*2+1), Point(an, an) );\r
+    if( n < 0 )\r
+        cv::gpu::erode(src, dst, element);\r
+    else\r
+        cv::gpu::dilate(src, dst, element);\r
+    imshow("Erode/Dilate",dst);\r
+}\r
+\r
+\r
+int main( int argc, char** argv )\r
+{\r
+    char* filename = argc == 2 ? argv[1] : (char*)"baboon.jpg";\r
+    if( (src = imread(filename,1)).data == 0 )\r
+        return help(), -1;\r
+\r
+    help();\r
+\r
+       \r
+       if (src.channels() == 3)\r
+       {\r
+               // gpu support only 4th channel images\r
+               GpuMat src4ch;\r
+               cv::gpu::cvtColor(src, src4ch, CV_BGR2BGRA); \r
+               src = src4ch;\r
+       }\r
+\r
+    //create windows for output images\r
+    namedWindow("Open/Close",1);\r
+    namedWindow("Erode/Dilate",1);\r
+\r
+    open_close_pos = erode_dilate_pos = max_iters;\r
+    createTrackbar("iterations", "Open/Close",&open_close_pos,max_iters*2+1,OpenClose);\r
+    createTrackbar("iterations", "Erode/Dilate",&erode_dilate_pos,max_iters*2+1,ErodeDilate);\r
+\r
+    for(;;)\r
+    {\r
+        int c;\r
+\r
+        OpenClose(open_close_pos, 0);\r
+        ErodeDilate(erode_dilate_pos, 0);\r
+        c = cvWaitKey(0);\r
+\r
+        if( (char)c == 27 )\r
+            break;\r
+        if( (char)c == 'e' )\r
+            element_shape = MORPH_ELLIPSE;\r
+        else if( (char)c == 'r' )\r
+            element_shape = MORPH_RECT;\r
+        else if( (char)c == 'c' )\r
+            element_shape = MORPH_CROSS;\r
+        else if( (char)c == ' ' )\r
+            element_shape = (element_shape + 1) % 3;\r
+    }\r
+\r
+    return 0;\r
+}\r