filldemo.c
上传用户:banwdc
上传日期:2016-06-25
资源大小:2871k
文件大小:5k
- #include "cv.h"
- #include "highgui.h"
- #include <stdio.h>
- #include <stdlib.h>
- IplImage* color_img0;
- IplImage* mask;
- IplImage* color_img;
- IplImage* gray_img0 = NULL;
- IplImage* gray_img = NULL;
- int ffill_case = 1;
- int lo_diff = 20, up_diff = 20;
- int connectivity = 4;
- int is_color = 1;
- int is_mask = 0;
- int new_mask_val = 255;
- void on_mouse( int event, int x, int y, int flags, void* param )
- {
- if( !color_img )
- return;
- switch( event )
- {
- case CV_EVENT_LBUTTONDOWN:
- {
- CvPoint seed = cvPoint(x,y);
- int lo = ffill_case == 0 ? 0 : lo_diff;
- int up = ffill_case == 0 ? 0 : up_diff;
- int flags = connectivity + (new_mask_val << 8) +
- (ffill_case == 1 ? CV_FLOODFILL_FIXED_RANGE : 0);
- int b = rand() & 255, g = rand() & 255, r = rand() & 255;
- CvConnectedComp comp;
- if( is_mask )
- cvThreshold( mask, mask, 1, 128, CV_THRESH_BINARY );
-
- if( is_color )
- {
- CvScalar color = CV_RGB( r, g, b );
- cvFloodFill( color_img, seed, color, CV_RGB( lo, lo, lo ),
- CV_RGB( up, up, up ), &comp, flags, is_mask ? mask : NULL );
- cvShowImage( "image", color_img );
- }
- else
- {
- CvScalar brightness = cvRealScalar((r*2 + g*7 + b + 5)/10);
- cvFloodFill( gray_img, seed, brightness, cvRealScalar(lo),
- cvRealScalar(up), &comp, flags, is_mask ? mask : NULL );
- cvShowImage( "image", gray_img );
- }
- printf("%g pixels were repaintedn", comp.area );
- if( is_mask )
- cvShowImage( "mask", mask );
- }
- break;
- }
- }
- int main( int argc, char** argv )
- {
- char* filename = argc >= 2 ? argv[1] : (char*)"fruits.jpg";
- if( (color_img0 = cvLoadImage(filename,1)) == 0 )
- return 0;
- printf( "Hot keys: n"
- "tESC - quit the programn"
- "tc - switch color/grayscale moden"
- "tm - switch mask moden"
- "tr - restore the original imagen"
- "ts - use null-range floodfilln"
- "tf - use gradient floodfill with fixed(absolute) rangen"
- "tg - use gradient floodfill with floating(relative) rangen"
- "t4 - use 4-connectivity moden"
- "t8 - use 8-connectivity moden" );
-
- color_img = cvCloneImage( color_img0 );
- gray_img0 = cvCreateImage( cvSize(color_img->width, color_img->height), 8, 1 );
- cvCvtColor( color_img, gray_img0, CV_BGR2GRAY );
- gray_img = cvCloneImage( gray_img0 );
- mask = cvCreateImage( cvSize(color_img->width + 2, color_img->height + 2), 8, 1 );
- cvNamedWindow( "image", 0 );
- cvCreateTrackbar( "lo_diff", "image", &lo_diff, 255, NULL );
- cvCreateTrackbar( "up_diff", "image", &up_diff, 255, NULL );
- cvSetMouseCallback( "image", on_mouse, 0 );
- for(;;)
- {
- int c;
-
- if( is_color )
- cvShowImage( "image", color_img );
- else
- cvShowImage( "image", gray_img );
- c = cvWaitKey(0);
- switch( (char) c )
- {
- case 'x1b':
- printf("Exiting ...n");
- goto exit_main;
- case 'c':
- if( is_color )
- {
- printf("Grayscale mode is setn");
- cvCvtColor( color_img, gray_img, CV_BGR2GRAY );
- is_color = 0;
- }
- else
- {
- printf("Color mode is setn");
- cvCopy( color_img0, color_img, NULL );
- cvZero( mask );
- is_color = 1;
- }
- break;
- case 'm':
- if( is_mask )
- {
- cvDestroyWindow( "mask" );
- is_mask = 0;
- }
- else
- {
- cvNamedWindow( "mask", 0 );
- cvZero( mask );
- cvShowImage( "mask", mask );
- is_mask = 1;
- }
- break;
- case 'r':
- printf("Original image is restoredn");
- cvCopy( color_img0, color_img, NULL );
- cvCopy( gray_img0, gray_img, NULL );
- cvZero( mask );
- break;
- case 's':
- printf("Simple floodfill mode is setn");
- ffill_case = 0;
- break;
- case 'f':
- printf("Fixed Range floodfill mode is setn");
- ffill_case = 1;
- break;
- case 'g':
- printf("Gradient (floating range) floodfill mode is setn");
- ffill_case = 2;
- break;
- case '4':
- printf("4-connectivity mode is setn");
- connectivity = 4;
- break;
- case '8':
- printf("8-connectivity mode is setn");
- connectivity = 8;
- break;
- }
- }
- exit_main:
- cvDestroyWindow( "test" );
- cvReleaseImage( &gray_img );
- cvReleaseImage( &gray_img0 );
- cvReleaseImage( &color_img );
- cvReleaseImage( &color_img0 );
- cvReleaseImage( &mask );
- return 1;
- }