filldemo.c
上传用户:banwdc
上传日期:2016-06-25
资源大小:2871k
文件大小:5k
源码类别:

OpenCV

开发平台:

Visual C++

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