src/manual_lines.py

   1 """Computing the grid"""
   2
   3 from math import sqrt, acos, copysign
   4 from geometry import l2ad, line, intersection
   5
   6 def lines(corners):
   7     # TODO Error on triangle
   8     corners.sort() # TODO does this help?
   9     # TODO refactor this vvv
  10     cor_d = [(corners[0], (c[0] - corners[0][0], c[1] - corners[0][1]), c) for c in
  11              corners[1:]]
  12     cor_d = [(float(a[0] * b[0] + a[1] * b[1]) / (sqrt(a[0] ** 2 + a[1] ** 2) *
  13               sqrt(b[0] **2 + b[1] ** 2)), a[0] * b[1] - b[0] * a[1], c) for a, b, c in cor_d]
  14     cor_d = sorted([(copysign(acos(min(a, 1)), b), c) for a, b, c in cor_d])
  15     corners = [corners[0]] + [c for _, c in cor_d]
  16     return (_lines(corners, 0) +
  17             [(corners[0], corners[3]), (corners[1], corners[2])],
  18             _lines(corners[1:4] + [corners[0]], 0) +
  19             [(corners[0], corners[1]), (corners[2], corners[3])])
  20
  21 def _lines(corners, n):
  22     # TODO what is this?
  23     if n == 0:
  24         x = half_line(corners)
  25         return (_lines([corners[0], x[0], x[1], corners[3]], 1) + [x] +
  26                 _lines([x[0], corners[1], corners[2], x[1]], 1))
  27     else:
  28         x = half_line(corners)
  29         c = intersection(line(x[0], corners[2]), line(corners[1], corners[3]))
  30         d = intersection(line(corners[0], corners[3]), line(corners[1], corners[2]))
  31         if d:
  32             l = (intersection(line(corners[0], corners[1]), line(c, d)),
  33                  intersection(line(corners[2], corners[3]), line(c, d)))
  34         else:
  35             lx = line(c, (c[0] + corners[0][0] - corners[3][0],
  36                       c[1] + corners[0][1] - corners[3][1]))
  37             l = (intersection(line(corners[0], corners[1]), lx),
  38                  intersection(line(corners[2], corners[3]), lx))
  39         l2 = half_line([corners[0], l[0], l[1], corners[3]])
  40         if n == 1:
  41             return ([l, l2] + _lines([l[0], l2[0], l2[1], l[1]], 2)
  42                     + _lines([corners[0], l2[0], l2[1], corners[3]], 2)
  43                     + _lines([l[0], corners[1], corners[2], l[1]], 2))
  44         if n == 2:
  45             return [l, l2]
  46
  47
  48 def half_line(corners):
  49     # TODO what is this?
  50     c = center(corners)
  51     d = intersection(line(corners[0], corners[3]), line(corners[1], corners[2]))
  52     if d:
  53         l = line(c, d)
  54     else:
  55         l = line(c, (c[0] + corners[0][0] - corners[3][0],
  56                      c[1] + corners[0][1] - corners[3][1]))
  57     p1 = intersection(l, line(corners[0], corners[1]))
  58     p2 = intersection(l, line(corners[2], corners[3]))
  59     return (p1, p2)
  60
  61 def center(corners):
  62     """Given a list of four corner points, return the center of the square."""
  63     return intersection(line(corners[0], corners[2]),
  64                         line(corners[1], corners[3]))