work on demo.py shell

samples/python2/browse.py

@@ -1,20 +1,29 @@
 '''
-browse.py shows how to implement a simple hi resolution image navigation
+browse.py
+=========
+
+Sample shows how to implement a simple hi resolution image navigation
+
+Usage
+-----
+browse.py [image filename]
+
 '''
 
 import numpy as np
 import cv2
 import sys
 
-print 'This sample shows how to implement a simple hi resolution image navigation.'
-print 'USAGE: browse.py [image filename]'
-print 
+if __name__ == '__main__':
+    print 'This sample shows how to implement a simple hi resolution image navigation.'
+    print 'USAGE: browse.py [image filename]'
+    print 
 
-if len(sys.argv) > 1:
+    if len(sys.argv) > 1:
         fn = sys.argv[1]
         print 'loading %s ...' % fn
         img = cv2.imread(fn)
-else:
+    else:
         sz = 4096
         print 'generating %dx%d procedural image ...' % (sz, sz)
         img = np.zeros((sz, sz), np.uint8)
@@ -22,17 +31,17 @@ else:
         track = np.int32(track*10 + (sz/2, sz/2))
         cv2.polylines(img, [track], 0, 255, 1, cv2.CV_AA)
 
-small = img
-for i in xrange(3):
+    small = img
+    for i in xrange(3):
         small = cv2.pyrDown(small)
 
-def onmouse(event, x, y, flags, param):
+    def onmouse(event, x, y, flags, param):
         h, w = img.shape[:2]
         h1, w1 = small.shape[:2]
         x, y = 1.0*x*h/h1, 1.0*y*h/h1
         zoom = cv2.getRectSubPix(img, (800, 600), (x+0.5, y+0.5))
         cv2.imshow('zoom', zoom)
 
-cv2.imshow('preview', small)
-cv2.setMouseCallback('preview', onmouse)
-cv2.waitKey()
+    cv2.imshow('preview', small)
+    cv2.setMouseCallback('preview', onmouse)
+    cv2.waitKey()

samples/python2/color_histogram.py

@@ -5,27 +5,28 @@ import sys
 
 import video
 
-hsv_map = np.zeros((180, 256, 3), np.uint8)
-h, s = np.indices(hsv_map.shape[:2])
-hsv_map[:,:,0] = h
-hsv_map[:,:,1] = s
-hsv_map[:,:,2] = 255
-hsv_map = cv2.cvtColor(hsv_map, cv2.COLOR_HSV2BGR)
-cv2.imshow('hsv_map', hsv_map)
-
-cv2.namedWindow('hist', 0)
-hist_scale = 10
-def set_scale(val):
+if __name__ == '__main__':
+
+    hsv_map = np.zeros((180, 256, 3), np.uint8)
+    h, s = np.indices(hsv_map.shape[:2])
+    hsv_map[:,:,0] = h
+    hsv_map[:,:,1] = s
+    hsv_map[:,:,2] = 255
+    hsv_map = cv2.cvtColor(hsv_map, cv2.COLOR_HSV2BGR)
+    cv2.imshow('hsv_map', hsv_map)
+
+    cv2.namedWindow('hist', 0)
+    hist_scale = 10
+    def set_scale(val):
         global hist_scale
         hist_scale = val
-cv2.createTrackbar('scale', 'hist', hist_scale, 32, set_scale)
+    cv2.createTrackbar('scale', 'hist', hist_scale, 32, set_scale)
 
-try: fn = sys.argv[1]
-except: fn = 'synth:bg=../cpp/baboon.jpg:class=chess:noise=0.05'
-cam = video.create_capture(fn)
+    try: fn = sys.argv[1]
+    except: fn = 'synth:bg=../cpp/baboon.jpg:class=chess:noise=0.05'
+    cam = video.create_capture(fn)
 
-t = clock()
-while True:
+    while True:
         flag, frame = cam.read()
         cv2.imshow('camera', frame)
         
@@ -41,11 +42,6 @@ while True:
         vis = hsv_map*h[:,:,np.newaxis] / 255.0
         cv2.imshow('hist', vis)
         
-
-    t1 = clock()
-    #print (t1-t)*1000
-    t = t1
-
         ch = cv2.waitKey(1)
         if ch == 27:
             break

samples/python2/demo.py

@@ -0,0 +1,132 @@
+import Tkinter as tk
+from ScrolledText import ScrolledText
+from glob import glob
+from common import splitfn
+
+#from IPython.Shell import IPShellEmbed
+#ipshell = IPShellEmbed()
+
+exclude_list = ['demo', 'common']
+
+class LinkManager:
+    def __init__(self, text, url_callback = None):
+        self.text = text
+        self.text.tag_config("link", foreground="blue", underline=1)
+        self.text.tag_bind("link", "<Enter>", self._enter)
+        self.text.tag_bind("link", "<Leave>", self._leave)
+        self.text.tag_bind("link", "<Button-1>", self._click)
+
+        self.url_callback = url_callback
+        self.reset()
+
+    def reset(self):
+        self.links = {}
+    def add(self, action):
+        # add an action to the manager.  returns tags to use in
+        # associated text widget
+        tag = "link-%d" % len(self.links)
+        self.links[tag] = action
+        return "link", tag
+
+    def _enter(self, event):
+        self.text.config(cursor="hand2")
+    def _leave(self, event):
+        self.text.config(cursor="")
+    def _click(self, event):
+        for tag in self.text.tag_names(tk.CURRENT):
+            if tag.startswith("link-"):
+                proc = self.links[tag]
+                if callable(proc):
+                    proc()
+                else:
+                    if self.url_callback:
+                        self.url_callback(proc)
+
+class App:
+    def __init__(self):
+        root = tk.Tk()
+        root.title('OpenCV Demo')
+
+        self.win = win = tk.PanedWindow(root, orient=tk.HORIZONTAL, sashrelief=tk.RAISED, sashwidth=4)
+        self.win.pack(fill=tk.BOTH, expand=1)
+
+        left = tk.Frame(win)
+        right = tk.Frame(win)                 
+        win.add(left)
+        win.add(right)
+                            
+        scrollbar = tk.Scrollbar(left, orient=tk.VERTICAL)
+        self.demos_lb = demos_lb = tk.Listbox(left, yscrollcommand=scrollbar.set)
+        scrollbar.config(command=demos_lb.yview)
+        scrollbar.pack(side=tk.RIGHT, fill=tk.Y)
+        demos_lb.pack(side=tk.LEFT, fill=tk.BOTH, expand=1)
+        
+        self.samples = {}
+        for fn in glob('*.py'):
+            name = splitfn(fn)[1]
+            if fn[0] != '_' and name not in exclude_list:
+                demos_lb.insert(tk.END, name)
+                self.samples[name] = fn
+        demos_lb.bind('<<ListboxSelect>>', self.on_demo_select)
+
+        self.text = text = ScrolledText(right, font=('arial', 12, 'normal'), width = 30, wrap='word')
+        text.pack(fill='both', expand=1)
+        self.linker = linker = LinkManager(text, self.on_link)
+
+        self.text.tag_config("header1", font=('arial', 14, 'bold'))
+        self.text.tag_config("header2", font=('arial', 12, 'bold'))
+
+        text.config(state='disabled')
+
+    def on_link(self, url):
+        print url
+
+    def on_demo_select(self, evt):
+        name = self.demos_lb.get( self.demos_lb.curselection()[0] )
+        fn = self.samples[name]
+        loc = {}
+        execfile(fn, loc)
+        descr = loc.get('__doc__', 'no-description')
+        
+        self.linker.reset()
+        self.text.config(state='normal')
+        self.text.delete(1.0, tk.END)
+        self.format_text(descr)
+        self.text.config(state='disabled')
+
+    def format_text(self, s):
+        text = self.text
+        lines = s.splitlines()
+        for i, s in enumerate(lines):
+            s = s.rstrip()
+            if i == 0 and not s:
+                continue
+            sn = len(s)
+            if s and s == '='*sn:
+                text.tag_add('header1', 'end-2l', 'end-1l')
+            elif s and s == '-'*sn:
+                text.tag_add('header2', 'end-2l', 'end-1l')
+            else:
+                text.insert('end', s+'\n')
+
+    def format_line(self, s):
+        text = self.text
+        pos, n = 0, len(s)
+        while pos < n:
+            next = s.find('http://', pos)
+            if next < 0:
+                next = n
+                test.insert(tk.END, s[pos:next])
+            pos = next
+        
+
+        #text.insert(tk.END, "click here!", linker.add('http://asdfsdaf')) 
+        
+
+
+    def run(self):
+        tk.mainloop()
+
+
+if __name__ == '__main__':
+    App().run()

samples/python2/edge.py

@@ -2,18 +2,19 @@ import cv2
 import video
 import sys
 
-try: fn = sys.argv[1]
-except: fn = video.presets['chess']
+if __name__ == '__main__':
+    try: fn = sys.argv[1]
+    except: fn = video.presets['chess']
 
-def nothing(*arg):
+    def nothing(*arg):
         pass
 
-cv2.namedWindow('edge')
-cv2.createTrackbar('thrs1', 'edge', 2000, 5000, nothing)
-cv2.createTrackbar('thrs2', 'edge', 4000, 5000, nothing)
+    cv2.namedWindow('edge')
+    cv2.createTrackbar('thrs1', 'edge', 2000, 5000, nothing)
+    cv2.createTrackbar('thrs2', 'edge', 4000, 5000, nothing)
 
-cap = video.create_capture(fn)
-while True:
+    cap = video.create_capture(fn)
+    while True:
         flag, img = cap.read()
         gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
         thrs1 = cv2.getTrackbarPos('thrs1', 'edge')

samples/python2/turing.py

@@ -15,30 +15,32 @@ USAGE: turing.py [-o <output.avi>]
 
 Press ESC to stop.
 '''
-print help_message
 
-w, h = 512, 512
+if __name__ == '__main__':
+    print help_message
 
-args, args_list = getopt.getopt(sys.argv[1:], 'o:', [])
-args = dict(args)
-out = None
-if '-o' in args:
+    w, h = 512, 512
+
+    args, args_list = getopt.getopt(sys.argv[1:], 'o:', [])
+    args = dict(args)
+    out = None
+    if '-o' in args:
         fn = args['-o']
         out = cv2.VideoWriter(args['-o'], cv.CV_FOURCC(*'DIB '), 30.0, (w, h), False)
         print 'writing %s ...' % fn
 
-a = np.zeros((h, w), np.float32)
-cv2.randu(a, np.array([0]), np.array([1]))
+    a = np.zeros((h, w), np.float32)
+    cv2.randu(a, np.array([0]), np.array([1]))
 
-def process_scale(a_lods, lod):
+    def process_scale(a_lods, lod):
         d = a_lods[lod] - cv2.pyrUp(a_lods[lod+1])
         for i in xrange(lod):
             d = cv2.pyrUp(d)
         v = cv2.GaussianBlur(d*d, (3, 3), 0)
         return np.sign(d), v
         
-scale_num = 6
-for frame_i in count():
+    scale_num = 6
+    for frame_i in count():
         a_lods = [a]
         for i in xrange(scale_num):
             a_lods.append(cv2.pyrDown(a_lods[-1]))