X-Git-Url: http://git.tomasm.cz/imago.git/blobdiff_plain/81185816bcf9c1f257ed23670fdeed539557da94..be7942c3c75a64979e60b50cf76aa1000ec57413:/src/intrsc.py

diff --git a/src/intrsc.py b/src/intrsc.py
index 2488d14..7455d0c 100644
--- a/src/intrsc.py
+++ b/src/intrsc.py
@@ -75,13 +75,17 @@ def board(image, lines, show_all, do_something):
         image_p = Image.fromstring('RGB', size, buff, 'raw')
         do_something(image_p, "color distribution")
 
-    max_s0 = max(s[0] for s in board_raw)
-    min_s0 = min(s[0] for s in board_raw)
-    norm_s0 = lambda x: (x - min_s0) / (max_s0 - min_s0)
-    max_s1 = max(s[1] for s in board_raw)
-    min_s1 = min(s[1] for s in board_raw)
-    norm_s1 = lambda x: (x - min_s1) / (max_s1 - min_s1)
-    color_data = [(norm_s0(s[0]), norm_s1(s[1])) for s in board_raw]
+    #max_s0 = max(s[0] for s in board_raw)
+    #min_s0 = min(s[0] for s in board_raw)
+    #norm_s0 = lambda x: (x - min_s0) / (max_s0 - min_s0)
+    #max_s1 = max(s[1] for s in board_raw)
+    #min_s1 = min(s[1] for s in board_raw)
+    #norm_s1 = lambda x: (x - min_s1) / (max_s1 - min_s1)
+    #max_s1 = max(s[1] for s in board_raw)
+    #min_s1 = min(s[1] for s in board_raw)
+    #norm_s1 = lambda x: (x - min_s1) / (max_s1 - min_s1)
+    #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]])
@@ -190,4 +194,5 @@ def stone_color_raw(image, (x, y)):
              sum(p[1] for p in points) / norm,
              sum(p[2] for p in points) / norm)
     hue, luma, saturation = colorsys.rgb_to_hls(*color)
-    return luma, saturation, color
+    color = colorsys.hls_to_rgb(hue, 0.5, 1.)
+    return luma, saturation, color, hue