X-Git-Url: http://git.maemo.org/git/?p=opencv;a=blobdiff_plain;f=samples%2Fswig_python%2Fcontours.py;fp=samples%2Fswig_python%2Fcontours.py;h=5f41e1616ba5016cfd4152c931b065704a2740a1;hp=0000000000000000000000000000000000000000;hb=e4c14cdbdf2fe805e79cd96ded236f57e7b89060;hpb=454138ff8a20f6edb9b65a910101403d8b520643

diff --git a/samples/swig_python/contours.py b/samples/swig_python/contours.py
new file mode 100755
index 0000000..5f41e16
--- /dev/null
+++ b/samples/swig_python/contours.py
@@ -0,0 +1,137 @@
+#! /usr/bin/env python
+
+print "OpenCV Python version of contours"
+
+# import the necessary things for OpenCV
+from opencv import cv
+from opencv import highgui
+
+# some default constants
+_SIZE = 500
+_DEFAULT_LEVEL = 3
+
+# definition of some colors
+_red = cv.cvScalar (0, 0, 255, 0);
+_green = cv.cvScalar (0, 255, 0, 0);
+_white = cv.cvRealScalar (255)
+_black = cv.cvRealScalar (0)
+
+# the callback on the trackbar, to set the level of contours we want
+# to display
+def on_trackbar (position):
+
+    # create the image for putting in it the founded contours
+    contours_image = cv.cvCreateImage (cv.cvSize (_SIZE, _SIZE), 8, 3)
+
+    # compute the real level of display, given the current position
+    levels = position - 3
+
+    # initialisation
+    _contours = contours
+    
+    if levels <= 0:
+        # zero or negative value
+        # => get to the nearest face to make it look more funny
+        _contours = contours.h_next.h_next.h_next
+        
+    # first, clear the image where we will draw contours
+    cv.cvSetZero (contours_image)
+    
+    # draw contours in red and green
+    cv.cvDrawContours (contours_image, _contours,
+                       _red, _green,
+                       levels, 3, cv.CV_AA,
+                       cv.cvPoint (0, 0))
+
+    # finally, show the image
+    highgui.cvShowImage ("contours", contours_image)
+
+if __name__ == '__main__':
+
+    # create the image where we want to display results
+    image = cv.cvCreateImage (cv.cvSize (_SIZE, _SIZE), 8, 1)
+
+    # start with an empty image
+    cv.cvSetZero (image)
+
+    # draw the original picture
+    for i in range (6):
+        dx = (i % 2) * 250 - 30
+        dy = (i / 2) * 150
+        
+        cv.cvEllipse (image,
+                      cv.cvPoint (dx + 150, dy + 100),
+                      cv.cvSize (100, 70),
+                      0, 0, 360, _white, -1, 8, 0)
+        cv.cvEllipse (image,
+                      cv.cvPoint (dx + 115, dy + 70),
+                      cv.cvSize (30, 20),
+                      0, 0, 360, _black, -1, 8, 0)
+        cv.cvEllipse (image,
+                      cv.cvPoint (dx + 185, dy + 70),
+                      cv.cvSize (30, 20),
+                      0, 0, 360, _black, -1, 8, 0)
+        cv.cvEllipse (image,
+                      cv.cvPoint (dx + 115, dy + 70),
+                      cv.cvSize (15, 15),
+                      0, 0, 360, _white, -1, 8, 0)
+        cv.cvEllipse (image,
+                      cv.cvPoint (dx + 185, dy + 70),
+                      cv.cvSize (15, 15),
+                      0, 0, 360, _white, -1, 8, 0)
+        cv.cvEllipse (image,
+                      cv.cvPoint (dx + 115, dy + 70),
+                      cv.cvSize (5, 5),
+                      0, 0, 360, _black, -1, 8, 0)
+        cv.cvEllipse (image,
+                      cv.cvPoint (dx + 185, dy + 70),
+                      cv.cvSize (5, 5),
+                      0, 0, 360, _black, -1, 8, 0)
+        cv.cvEllipse (image,
+                      cv.cvPoint (dx + 150, dy + 100),
+                      cv.cvSize (10, 5),
+                      0, 0, 360, _black, -1, 8, 0)
+        cv.cvEllipse (image,
+                      cv.cvPoint (dx + 150, dy + 150),
+                      cv.cvSize (40, 10),
+                      0, 0, 360, _black, -1, 8, 0)
+        cv.cvEllipse (image,
+                      cv.cvPoint (dx + 27, dy + 100),
+                      cv.cvSize (20, 35),
+                      0, 0, 360, _white, -1, 8, 0)
+        cv.cvEllipse (image,
+                      cv.cvPoint (dx + 273, dy + 100),
+                      cv.cvSize (20, 35),
+                      0, 0, 360, _white, -1, 8, 0)
+
+    # create window and display the original picture in it
+    highgui.cvNamedWindow ("image", 1)
+    highgui.cvShowImage ("image", image)
+
+    # create the storage area
+    storage = cv.cvCreateMemStorage (0)
+    
+    # find the contours
+    nb_contours, contours = cv.cvFindContours (image,
+                                               storage,
+                                               cv.sizeof_CvContour,
+                                               cv.CV_RETR_TREE,
+                                               cv.CV_CHAIN_APPROX_SIMPLE,
+                                               cv.cvPoint (0,0))
+
+    # comment this out if you do not want approximation
+    contours = cv.cvApproxPoly (contours, cv.sizeof_CvContour,
+                                storage,
+                                cv.CV_POLY_APPROX_DP, 3, 1)
+    
+    # create the window for the contours
+    highgui.cvNamedWindow ("contours", 1)
+
+    # create the trackbar, to enable the change of the displayed level
+    highgui.cvCreateTrackbar ("levels+3", "contours", 3, 7, on_trackbar)
+
+    # call one time the callback, so we will have the 1st display done
+    on_trackbar (_DEFAULT_LEVEL)
+
+    # wait a key pressed to end
+    highgui.cvWaitKey (0)