X-Git-Url: http://git.tomasm.cz/imago.git/blobdiff_plain/e081eb1ecc06066d8015da7ad40947916607e964..5575c909e2d345773367d1da42149b758f7e4865:/imago.py?ds=sidebyside

diff --git a/imago.py b/imago.py
index 1cda8ba..9e1697c 100755
--- a/imago.py
+++ b/imago.py
@@ -47,6 +47,9 @@ def main():
     except IOError, msg:
         print >> sys.stderr, msg
         return 1
+    if image.mode == 'P':
+        image = image.convert('RGB')
+    
     if image.size[0] > args.w:
         image = image.resize((args.w, int((float(args.w)/image.size[0]) *
                               image.size[1])), Image.ANTIALIAS)
@@ -127,6 +130,13 @@ def main():
             do_something(im_c, "hough x lines")
         lines.append(hough1.all_lines(im_c))
 
+    image_g = image.copy()
+    draw = ImageDraw.Draw(image_g)
+    for line in [l for s in lines for l in s]:
+        draw.line(line_from_angl_dist(line, image.size), fill=(120, 255, 120))
+    if show_all:
+        do_something(image_g, "the grid")
+
     intersections = intersections_from_angl_dist(lines, image.size)
     image_g = image.copy()
     draw = ImageDraw.Draw(image_g)
@@ -139,7 +149,7 @@ def main():
                        line])
 
     if show_all:
-        do_something(image_g, "the grid")
+        do_something(image_g, "intersections")
 
     return 0