X-Git-Url: http://git.tomasm.cz/imago.git/blobdiff_plain/33b793fe326a2706173b38d2fc4b454982fbf900..fedf00614bb6be8cea0941e431a01dc8a417e5c5:/src/intrsc.py?ds=sidebyside

diff --git a/src/intrsc.py b/src/intrsc.py
index 3c736d9..b0e798c 100644
--- a/src/intrsc.py
+++ b/src/intrsc.py
@@ -26,8 +26,8 @@ def dst_sort(lines):
     lines.sort(key=itemgetter(1))
     return lines
 
-def board(image, lines, show_all, do_something, logger):
-    """Compute intersections, find stone colors and return board situation."""
+def b_intersects(image, lines, show_all, do_something, logger):
+    """Compute intersections."""
     # TODO refactor show_all, do_something
     # TODO refactor this into smaller functions
     logger("finding the stones")
@@ -47,9 +47,15 @@ def board(image, lines, show_all, do_something, logger):
                 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")
+    return intersections
+
+def board(image, intersections, show_all, do_something, logger):
+    """Find stone colors and return board situation."""
+
+#    image_c = filters.color_enhance(image)
+#    if show_all:
+#        do_something(image_c, "white balance")
+    image_c = image
     
     board_raw = []
     
@@ -88,9 +94,17 @@ def board(image, lines, show_all, do_something, logger):
     #color_data = [(norm_s0(s[0]), norm_s1(s[1])) for s in board_raw]
     color_data = [(s[0], s[1]) for s in board_raw]
 
-    clusters = k_means.cluster(3, 2,zip(color_data, range(len(color_data))),
-                               [[0., 0.5], [0.5, 0.5], [1., 0.5]])
-
+    init_x = sum(c[0] for c in color_data) / float(len(color_data))
+
+    clusters, score = k_means.cluster(3, 2,zip(color_data, range(len(color_data))),
+                               [[0., 0.5], [init_x, 0.5], [1., 0.5]])
+#    clusters1, score1 = k_means.cluster(1, 2,zip(color_data, range(len(color_data))),
+#                               [[0.5, 0.5]])
+#    clusters2, score2 = k_means.cluster(2, 2,zip(color_data, range(len(color_data))),
+#                               [[0., 0.5], [0.75, 0.5]])
+#    import sys
+#    print >> sys.stderr, score1, score2, score
+#
     if show_all:
         fig = pyplot.figure(figsize=(8, 6))
         pyplot.scatter([d[0][0] for d in clusters[0]], [d[0][1] for d in clusters[0]],
@@ -125,7 +139,6 @@ def board(image, lines, show_all, do_something, logger):
     except StopIteration:
         pass
     
-
     return output.Board(19, board_r)
 
 def mean_luma(cluster):