more comments

[imago.git] / imago_pack / intrsc.py

diff --git a/imago_pack/intrsc.py b/imago_pack/intrsc.py
index 45308b1..d577205 100644 (file)
--- a/imago_pack/intrsc.py
+++ b/imago_pack/intrsc.py
@@ -1,10 +1,11 @@
-"""Imago intersections module"""
+"""Imago intersections module."""

 
 from math import cos, tan, pi
 from operator import itemgetter
 
 import ImageDraw
 
 
 from math import cos, tan, pi
 from operator import itemgetter
 
 import ImageDraw
 

+import filters

 import k_means
 import output
 
 import k_means
 import output
 


@@ -36,10 +37,14 @@ def board(image, lines, show_all, do_something):
                 draw.point((x , y), fill=(120, 255, 120))
         do_something(image_g, "intersections")
 
                 draw.point((x , y), fill=(120, 255, 120))
         do_something(image_g, "intersections")
 

+    image_c = filters.color_enhance(image)
+    if show_all:
+        do_something(image_c, "white balance")
+    

     board_raw = []
     
     for line in intersections:
     board_raw = []
     
     for line in intersections:

-        board_raw.append([stone_color_raw(image, intersection) for intersection in
+        board_raw.append([stone_color_raw(image_c, intersection) for intersection in

                       line])
     board_raw = sum(board_raw, [])
 
                       line])
     board_raw = sum(board_raw, [])
 


@@ -95,6 +100,7 @@ def board(image, lines, show_all, do_something):
     return output.Board(19, board_r)
 
 def mean_luma(cluster):
     return output.Board(19, board_r)
 
 def mean_luma(cluster):

+    """Return mean luma of the *cluster* of points."""

     return sum(c[0][0] for c in cluster) / float(len(cluster))
 
 def intersections_from_angl_dist(lines, size, get_all=True):
     return sum(c[0][0] for c in cluster) / float(len(cluster))
 
 def intersections_from_angl_dist(lines, size, get_all=True):