samples/c/morphology.c

   1 #include "opencv2/highgui/highgui.hpp"
   2 #include "opencv2/imgproc/imgproc_c.h"
   3
   4 #include <stdio.h>
   5
   6 IplImage* src = 0;
   7 IplImage* dst = 0;
   8
   9 IplConvKernel* element = 0;
  10 int element_shape = CV_SHAPE_RECT;
  11
  12 //the address of variable which receives trackbar position update
  13 int max_iters = 10;
  14 int open_close_pos = 0;
  15 int erode_dilate_pos = 0;
  16
  17 // callback function for open/close trackbar
  18 void OpenClose(int pos)
  19 {
  20     int n = open_close_pos - max_iters;
  21     int an = n > 0 ? n : -n;
  22     element = cvCreateStructuringElementEx( an*2+1, an*2+1, an, an, element_shape, 0 );
  23     if( n < 0 )
  24     {
  25         cvErode(src,dst,element,1);
  26         cvDilate(dst,dst,element,1);
  27     }
  28     else
  29     {
  30         cvDilate(src,dst,element,1);
  31         cvErode(dst,dst,element,1);
  32     }
  33     cvReleaseStructuringElement(&element);
  34     cvShowImage("Open/Close",dst);
  35 }
  36
  37 // callback function for erode/dilate trackbar
  38 void ErodeDilate(int pos)
  39 {
  40     int n = erode_dilate_pos - max_iters;
  41     int an = n > 0 ? n : -n;
  42     element = cvCreateStructuringElementEx( an*2+1, an*2+1, an, an, element_shape, 0 );
  43     if( n < 0 )
  44     {
  45         cvErode(src,dst,element,1);
  46     }
  47     else
  48     {
  49         cvDilate(src,dst,element,1);
  50     }
  51     cvReleaseStructuringElement(&element);
  52     cvShowImage("Erode/Dilate",dst);
  53 }
  54
  55
  56 int main( int argc, char** argv )
  57 {
  58     char* filename = argc == 2 ? argv[1] : (char*)"baboon.jpg";
  59     if( (src = cvLoadImage(filename,1)) == 0 )
  60         return -1;
  61
  62     printf( "Hot keys: \n"
  63         "\tESC - quit the program\n"
  64         "\tr - use rectangle structuring element\n"
  65         "\te - use elliptic structuring element\n"
  66         "\tc - use cross-shaped structuring element\n"
  67         "\tSPACE - loop through all the options\n" );
  68
  69     dst = cvCloneImage(src);
  70
  71     //create windows for output images
  72     cvNamedWindow("Open/Close",1);
  73     cvNamedWindow("Erode/Dilate",1);
  74
  75     open_close_pos = erode_dilate_pos = max_iters;
  76     cvCreateTrackbar("iterations", "Open/Close",&open_close_pos,max_iters*2+1,OpenClose);
  77     cvCreateTrackbar("iterations", "Erode/Dilate",&erode_dilate_pos,max_iters*2+1,ErodeDilate);
  78
  79     for(;;)
  80     {
  81         int c;
  82
  83         OpenClose(open_close_pos);
  84         ErodeDilate(erode_dilate_pos);
  85         c = cvWaitKey(0);
  86
  87         if( (char)c == 27 )
  88             break;
  89         if( (char)c == 'e' )
  90             element_shape = CV_SHAPE_ELLIPSE;
  91         else if( (char)c == 'r' )
  92             element_shape = CV_SHAPE_RECT;
  93         else if( (char)c == 'c' )
  94             element_shape = CV_SHAPE_CROSS;
  95         else if( (char)c == ' ' )
  96             element_shape = (element_shape + 1) % 3;
  97     }
  98
  99     //release images
 100     cvReleaseImage(&src);
 101     cvReleaseImage(&dst);
 102
 103     //destroy windows
 104     cvDestroyWindow("Open/Close");
 105     cvDestroyWindow("Erode/Dilate");
 106
 107     return 0;
 108 }