Update to 2.0.0 tree from current Fremantle build

[opencv] / samples / swig_python / ffilldemo.py

diff --git a/samples/swig_python/ffilldemo.py b/samples/swig_python/ffilldemo.py
new file mode 100755 (executable)
index 0000000..d030712
--- /dev/null
+++ b/samples/swig_python/ffilldemo.py
@@ -0,0 +1,160 @@
+#!/usr/bin/python
+import sys
+import random
+from opencv.cv import *
+from opencv.highgui import *
+
+color_img0=None;
+mask=None;
+color_img=None;
+gray_img0 = None;
+gray_img = None;
+ffill_case = 1;
+lo_diff = 20
+up_diff = 20;
+connectivity = 4;
+is_color = 1;
+is_mask = 0;
+new_mask_val = 255;
+
+def update_lo( pos ):
+    lo_diff = pos
+def update_up( pos ):
+    up_diff = pos
+
+def on_mouse( event, x, y, flags, param ):
+
+    if( not color_img ):
+        return;
+
+    if event==CV_EVENT_LBUTTONDOWN:
+            comp = CvConnectedComp()
+            my_mask = None
+            seed = cvPoint(x,y);
+            if ffill_case==0:
+                lo = up = 0
+                flags = connectivity + (new_mask_val << 8)
+            else:
+                lo = lo_diff;
+                up = up_diff;
+                flags = connectivity + (new_mask_val << 8) + CV_FLOODFILL_FIXED_RANGE
+            b = random.randint(0,255)
+            g = random.randint(0,255)
+            r = random.randint(0,255)
+
+            if( is_mask ):
+                my_mask = mask
+                cvThreshold( mask, mask, 1, 128, CV_THRESH_BINARY );
+               
+            if( is_color ):
+            
+                color = CV_RGB( r, g, b );
+                cvFloodFill( color_img, seed, color, CV_RGB( lo, lo, lo ),
+                             CV_RGB( up, up, up ), comp, flags, my_mask );
+                cvShowImage( "image", color_img );
+            
+            else:
+            
+                brightness = cvRealScalar((r*2 + g*7 + b + 5)/10);
+                cvFloodFill( gray_img, seed, brightness, cvRealScalar(lo),
+                             cvRealScalar(up), comp, flags, my_mask );
+                cvShowImage( "image", gray_img );
+            
+
+            print "%g pixels were repainted" % comp.area;
+
+            if( is_mask ):
+                cvShowImage( "mask", mask );
+        
+    
+
+
+if __name__ == "__main__":
+    
+    filename = "../c/fruits.jpg"
+    if len(sys.argv)>1:
+        filename=argv[1]
+    
+    color_img0 = cvLoadImage(filename,1)
+    if not color_img0:
+        print "Could not open %s" % filename
+        sys.exit(-1)
+
+    print "Hot keys:"
+    print "\tESC - quit the program"
+    print "\tc - switch color/grayscale mode"
+    print "\tm - switch mask mode"
+    print "\tr - restore the original image"
+    print "\ts - use null-range floodfill"
+    print "\tf - use gradient floodfill with fixed(absolute) range"
+    print "\tg - use gradient floodfill with floating(relative) range"
+    print "\t4 - use 4-connectivity mode"
+    print "\t8 - use 8-connectivity mode"
+        
+    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", 1 );
+    cvCreateTrackbar( "lo_diff", "image", lo_diff, 255, update_lo);
+    cvCreateTrackbar( "up_diff", "image", up_diff, 255, update_up);
+
+    cvSetMouseCallback( "image", on_mouse );
+
+    while True: 
+        if( is_color ):
+            cvShowImage( "image", color_img );
+        else:
+            cvShowImage( "image", gray_img );
+
+        c = cvWaitKey(0);
+        if c=='\x1b':
+            print("Exiting ...");
+            sys.exit(0)
+        elif c=='c':
+            if( is_color ):
+            
+                print("Grayscale mode is set");
+                cvCvtColor( color_img, gray_img, CV_BGR2GRAY );
+                is_color = 0;
+            
+            else:
+            
+                print("Color mode is set");
+                cvCopy( color_img0, color_img, None );
+                cvZero( mask );
+                is_color = 1;
+            
+        elif c=='m':
+            if( is_mask ):
+                cvDestroyWindow( "mask" );
+                is_mask = 0;
+            
+            else:
+                cvNamedWindow( "mask", 0 );
+                cvZero( mask );
+                cvShowImage( "mask", mask );
+                is_mask = 1;
+            
+        elif c=='r':
+            print("Original image is restored");
+            cvCopy( color_img0, color_img, None );
+            cvCopy( gray_img0, gray_img, None );
+            cvZero( mask );
+        elif c=='s':
+            print("Simple floodfill mode is set");
+            ffill_case = 0;
+        elif c=='f':
+            print("Fixed Range floodfill mode is set");
+            ffill_case = 1;
+        elif c=='g':
+            print("Gradient (floating range) floodfill mode is set");
+            ffill_case = 2;
+        elif c=='4':
+            print("4-connectivity mode is set");
+            connectivity = 4;
+        elif c=='8':
+            print("8-connectivity mode is set");
+            connectivity = 8;