samples/cpp/morphology2.cpp

   1 #include "opencv2/imgproc/imgproc.hpp"
   2 #include "opencv2/highgui/highgui.hpp"
   3 #include <stdlib.h>
   4 #include <stdio.h>
   5
   6 using namespace cv;
   7
   8 static void help()
   9 {
  10
  11 printf("\nShow off image morphology: erosion, dialation, open and close\n"
  12     "Call:\n   morphology2 [image]\n"
  13     "This program also shows use of rect, elipse and cross kernels\n\n");
  14 printf( "Hot keys: \n"
  15     "\tESC - quit the program\n"
  16     "\tr - use rectangle structuring element\n"
  17     "\te - use elliptic structuring element\n"
  18     "\tc - use cross-shaped structuring element\n"
  19     "\tSPACE - loop through all the options\n" );
  20 }
  21
  22 Mat src, dst;
  23
  24 int element_shape = MORPH_RECT;
  25
  26 //the address of variable which receives trackbar position update
  27 int max_iters = 10;
  28 int open_close_pos = 0;
  29 int erode_dilate_pos = 0;
  30
  31 // callback function for open/close trackbar
  32 static void OpenClose(int, void*)
  33 {
  34     int n = open_close_pos - max_iters;
  35     int an = n > 0 ? n : -n;
  36     Mat element = getStructuringElement(element_shape, Size(an*2+1, an*2+1), Point(an, an) );
  37     if( n < 0 )
  38         morphologyEx(src, dst, MORPH_OPEN, element);
  39     else
  40         morphologyEx(src, dst, MORPH_CLOSE, element);
  41     imshow("Open/Close",dst);
  42 }
  43
  44 // callback function for erode/dilate trackbar
  45 static void ErodeDilate(int, void*)
  46 {
  47     int n = erode_dilate_pos - max_iters;
  48     int an = n > 0 ? n : -n;
  49     Mat element = getStructuringElement(element_shape, Size(an*2+1, an*2+1), Point(an, an) );
  50     if( n < 0 )
  51         erode(src, dst, element);
  52     else
  53         dilate(src, dst, element);
  54     imshow("Erode/Dilate",dst);
  55 }
  56
  57
  58 int main( int argc, char** argv )
  59 {
  60     char* filename = argc == 2 ? argv[1] : (char*)"baboon.jpg";
  61     if( (src = imread(filename,1)).data == 0 )
  62         return -1;
  63
  64     help();
  65
  66     //create windows for output images
  67     namedWindow("Open/Close",1);
  68     namedWindow("Erode/Dilate",1);
  69
  70     open_close_pos = erode_dilate_pos = max_iters;
  71     createTrackbar("iterations", "Open/Close",&open_close_pos,max_iters*2+1,OpenClose);
  72     createTrackbar("iterations", "Erode/Dilate",&erode_dilate_pos,max_iters*2+1,ErodeDilate);
  73
  74     for(;;)
  75     {
  76         int c;
  77
  78         OpenClose(open_close_pos, 0);
  79         ErodeDilate(erode_dilate_pos, 0);
  80         c = waitKey(0);
  81
  82         if( (char)c == 27 )
  83             break;
  84         if( (char)c == 'e' )
  85             element_shape = MORPH_ELLIPSE;
  86         else if( (char)c == 'r' )
  87             element_shape = MORPH_RECT;
  88         else if( (char)c == 'c' )
  89             element_shape = MORPH_CROSS;
  90         else if( (char)c == ' ' )
  91             element_shape = (element_shape + 1) % 3;
  92     }
  93
  94     return 0;
  95 }