git.maemo.org Git - opencv/blob - samples/c/ffilldemo.c

   1 #ifdef _CH_
   2 #pragma package <opencv>
   3 #endif
   4
   5 #ifndef _EiC
   6 #include "cv.h"
   7 #include "highgui.h"
   8 #include <stdio.h>
   9 #include <stdlib.h>
  10 #endif
  11
  12 IplImage* color_img0;
  13 IplImage* mask;
  14 IplImage* color_img;
  15 IplImage* gray_img0 = NULL;
  16 IplImage* gray_img = NULL;
  17 int ffill_case = 1;
  18 int lo_diff = 20, up_diff = 20;
  19 int connectivity = 4;
  20 int is_color = 1;
  21 int is_mask = 0;
  22 int new_mask_val = 255;
  23
  24 void on_mouse( int event, int x, int y, int flags, void* param )
  25 {
  26     if( !color_img )
  27         return;
  28
  29     switch( event )
  30     {
  31     case CV_EVENT_LBUTTONDOWN:
  32         {
  33             CvPoint seed = cvPoint(x,y);
  34             int lo = ffill_case == 0 ? 0 : lo_diff;
  35             int up = ffill_case == 0 ? 0 : up_diff;
  36             int flags = connectivity + (new_mask_val << 8) +
  37                         (ffill_case == 1 ? CV_FLOODFILL_FIXED_RANGE : 0);
  38             int b = rand() & 255, g = rand() & 255, r = rand() & 255;
  39             CvConnectedComp comp;
  40
  41             if( is_mask )
  42                 cvThreshold( mask, mask, 1, 128, CV_THRESH_BINARY );
  43
  44             if( is_color )
  45             {
  46                 CvScalar color = CV_RGB( r, g, b );
  47                 cvFloodFill( color_img, seed, color, CV_RGB( lo, lo, lo ),
  48                              CV_RGB( up, up, up ), &comp, flags, is_mask ? mask : NULL );
  49                 cvShowImage( "image", color_img );
  50             }
  51             else
  52             {
  53                 CvScalar brightness = cvRealScalar((r*2 + g*7 + b + 5)/10);
  54                 cvFloodFill( gray_img, seed, brightness, cvRealScalar(lo),
  55                              cvRealScalar(up), &comp, flags, is_mask ? mask : NULL );
  56                 cvShowImage( "image", gray_img );
  57             }
  58
  59             printf("%g pixels were repainted\n", comp.area );
  60
  61             if( is_mask )
  62                 cvShowImage( "mask", mask );
  63         }
  64         break;
  65     }
  66 }
  67
  68 int main( int argc, char** argv )
  69 {
  70     char* filename = argc >= 2 ? argv[1] : (char*)"fruits.jpg";
  71
  72     if( (color_img0 = cvLoadImage(filename,1)) == 0 )
  73         return 0;
  74
  75     printf( "Hot keys: \n"
  76             "\tESC - quit the program\n"
  77             "\tc - switch color/grayscale mode\n"
  78             "\tm - switch mask mode\n"
  79             "\tr - restore the original image\n"
  80             "\ts - use null-range floodfill\n"
  81             "\tf - use gradient floodfill with fixed(absolute) range\n"
  82             "\tg - use gradient floodfill with floating(relative) range\n"
  83             "\t4 - use 4-connectivity mode\n"
  84             "\t8 - use 8-connectivity mode\n" );
  85
  86     color_img = cvCloneImage( color_img0 );
  87     gray_img0 = cvCreateImage( cvSize(color_img->width, color_img->height), 8, 1 );
  88     cvCvtColor( color_img, gray_img0, CV_BGR2GRAY );
  89     gray_img = cvCloneImage( gray_img0 );
  90     mask = cvCreateImage( cvSize(color_img->width + 2, color_img->height + 2), 8, 1 );
  91
  92     cvNamedWindow( "image", 0 );
  93     cvCreateTrackbar( "lo_diff", "image", &lo_diff, 255, NULL );
  94     cvCreateTrackbar( "up_diff", "image", &up_diff, 255, NULL );
  95
  96     cvSetMouseCallback( "image", on_mouse, 0 );
  97
  98     for(;;)
  99     {
 100         int c;
 101
 102         if( is_color )
 103             cvShowImage( "image", color_img );
 104         else
 105             cvShowImage( "image", gray_img );
 106
 107         c = cvWaitKey(0);
 108         switch( (char) c )
 109         {
 110         case '\x1b':
 111             printf("Exiting ...\n");
 112             goto exit_main;
 113         case 'c':
 114             if( is_color )
 115             {
 116                 printf("Grayscale mode is set\n");
 117                 cvCvtColor( color_img, gray_img, CV_BGR2GRAY );
 118                 is_color = 0;
 119             }
 120             else
 121             {
 122                 printf("Color mode is set\n");
 123                 cvCopy( color_img0, color_img, NULL );
 124                 cvZero( mask );
 125                 is_color = 1;
 126             }
 127             break;
 128         case 'm':
 129             if( is_mask )
 130             {
 131                 cvDestroyWindow( "mask" );
 132                 is_mask = 0;
 133             }
 134             else
 135             {
 136                 cvNamedWindow( "mask", 0 );
 137                 cvZero( mask );
 138                 cvShowImage( "mask", mask );
 139                 is_mask = 1;
 140             }
 141             break;
 142         case 'r':
 143             printf("Original image is restored\n");
 144             cvCopy( color_img0, color_img, NULL );
 145             cvCopy( gray_img0, gray_img, NULL );
 146             cvZero( mask );
 147             break;
 148         case 's':
 149             printf("Simple floodfill mode is set\n");
 150             ffill_case = 0;
 151             break;
 152         case 'f':
 153             printf("Fixed Range floodfill mode is set\n");
 154             ffill_case = 1;
 155             break;
 156         case 'g':
 157             printf("Gradient (floating range) floodfill mode is set\n");
 158             ffill_case = 2;
 159             break;
 160         case '4':
 161             printf("4-connectivity mode is set\n");
 162             connectivity = 4;
 163             break;
 164         case '8':
 165             printf("8-connectivity mode is set\n");
 166             connectivity = 8;
 167             break;
 168         }
 169     }
 170
 171 exit_main:
 172
 173     cvDestroyWindow( "test" );
 174     cvReleaseImage( &gray_img );
 175     cvReleaseImage( &gray_img0 );
 176     cvReleaseImage( &color_img );
 177     cvReleaseImage( &color_img0 );
 178     cvReleaseImage( &mask );
 179
 180     return 1;
 181 }
 182
 183 #ifdef _EiC
 184 main(1,"ffilldemo.c");
 185 #endif