better manual mode, grid-fitting preparation

[imago.git] / linef.py

diff --git a/linef.py b/linef.py
index 1dd35d4..34c4d07 100755 (executable)
--- a/linef.py
+++ b/linef.py
@@ -100,6 +100,12 @@ def find_lines(image, show_all, do_something, verbose):
     if show_all:
         do_something(im_c, "hough x lines")
 
     if show_all:
         do_something(im_c, "hough x lines")
 

+    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, "lines")

    
     return lines
 
    
     return lines