intrsc.py

   1 from math import cos, tan
   2 from operator import itemgetter
   3
   4 import ImageDraw
   5
   6 def board(image, lines, show_all, do_something):
   7     intersections = intersections_from_angl_dist(lines, image.size)
   8     image_g = image.copy()
   9     draw = ImageDraw.Draw(image_g)
  10     for line in intersections:
  11         for (x, y) in line:
  12             draw.point((x , y), fill=(120, 255, 120))
  13
  14     if show_all:
  15         do_something(image_g, "intersections")
  16
  17     board_r = []
  18
  19     for line in intersections:
  20         board_r.append([stone_color(image, intersection) for intersection in
  21                       line])
  22     return board_r
  23
  24 def intersections_from_angl_dist(lines, size, get_all=False):
  25     intersections = []
  26     for (angl1, dist1) in sorted(lines[1], key=itemgetter(1)):
  27         line = []
  28         for (angl2, dist2) in sorted(lines[0], key=itemgetter(1)):
  29             if abs(angl1 - angl2) > 0.4:
  30                 x =  (- ((dist2 / cos(angl2)) - (dist1 / cos(angl1)))
  31                         / (tan(angl1) - tan(angl2)))
  32                 y = (tan(angl1) * x) - (dist1 / cos(angl1))
  33                 if get_all or (-size[0] / 2 < x < size[0] / 2 and
  34                     -size[1] / 2 < y < size[1] / 2):
  35                     line.append((int(x + size[0] / 2), int(y + size[1] / 2)))
  36         intersections.append(line)
  37     return intersections
  38
  39 def stone_color(image, (x, y)):
  40     suma = 0.
  41     for i in range(-2, 3):
  42         for j in range(-2, 3):
  43             try:
  44                 suma += sum(image.getpixel((x + i, y + j)))
  45             except IndexError:
  46                 pass
  47     suma /= 3 * 25
  48     if suma < 55:
  49         return 'B'
  50     elif suma < 200:
  51         return '.'
  52     else:
  53         return 'W'