Update to 2.0.0 tree from current Fremantle build

[opencv] / samples / swig_python / edge.py

diff --git a/samples/swig_python/edge.py b/samples/swig_python/edge.py
new file mode 100755 (executable)
index 0000000..f0bb62d
--- /dev/null
+++ b/samples/swig_python/edge.py
@@ -0,0 +1,64 @@
+#! /usr/bin/env python
+
+print "OpenCV Python version of edge"
+
+import sys
+
+# import the necessary things for OpenCV
+from opencv import cv
+from opencv import highgui
+
+# some definitions
+win_name = "Edge"
+trackbar_name = "Threshold"
+
+# the callback on the trackbar
+def on_trackbar (position):
+
+    cv.cvSmooth (gray, edge, cv.CV_BLUR, 3, 3, 0)
+    cv.cvNot (gray, edge)
+
+    # run the edge dector on gray scale
+    cv.cvCanny (gray, edge, position, position * 3, 3)
+
+    # reset
+    cv.cvSetZero (col_edge)
+
+    # copy edge points
+    cv.cvCopy (image, col_edge, edge)
+    
+    # show the image
+    highgui.cvShowImage (win_name, col_edge)
+
+if __name__ == '__main__':
+    filename = "../c/fruits.jpg"
+
+    if len(sys.argv)>1:
+        filename = sys.argv[1]
+
+    # load the image gived on the command line
+    image = highgui.cvLoadImage (filename)
+
+    if not image:
+        print "Error loading image '%s'" % filename
+        sys.exit(-1)
+
+    # create the output image
+    col_edge = cv.cvCreateImage (cv.cvSize (image.width, image.height), 8, 3)
+
+    # convert to grayscale
+    gray = cv.cvCreateImage (cv.cvSize (image.width, image.height), 8, 1)
+    edge = cv.cvCreateImage (cv.cvSize (image.width, image.height), 8, 1)
+    cv.cvCvtColor (image, gray, cv.CV_BGR2GRAY)
+
+    # create the window
+    highgui.cvNamedWindow (win_name, highgui.CV_WINDOW_AUTOSIZE)
+
+    # create the trackbar
+    highgui.cvCreateTrackbar (trackbar_name, win_name, 1, 100, on_trackbar)
+
+    # show the image
+    on_trackbar (0)
+
+    # wait a key pressed to end
+    highgui.cvWaitKey (0)