vcs.maemo.org Git - opencv/blob - samples/c/morphology.c

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