153 lines
3.9 KiB
Python
153 lines
3.9 KiB
Python
from collections import *
|
|
from math import *
|
|
import axi
|
|
import fileinput
|
|
|
|
BOUNDS = axi.A3_BOUNDS
|
|
X, Y, W, H = BOUNDS
|
|
P = 0.25
|
|
R = 0.125
|
|
COLS = 16
|
|
ROWS = 16
|
|
N = ROWS * COLS
|
|
|
|
def rectangle(x, y, w, h):
|
|
return [
|
|
(x, y),
|
|
(x + w, y),
|
|
(x + w, y + h),
|
|
(x, y + h),
|
|
(x, y),
|
|
]
|
|
|
|
def padded_rectangle(x, y, w, h, p):
|
|
x += p
|
|
y += p
|
|
w -= p * 2
|
|
h -= p * 2
|
|
return rectangle(x, y, w, h)
|
|
|
|
def arc(cx, cy, r, a0, a1, n):
|
|
path = []
|
|
for i in range(n+1):
|
|
t = i / n
|
|
a = a0 + (a1 - a0) * t
|
|
x = cx + r * cos(a)
|
|
y = cy + r * sin(a)
|
|
path.append((x, y))
|
|
return path
|
|
|
|
def rounded_rectangle(x, y, w, h, r):
|
|
n = 18
|
|
x0, x1, x2, x3 = x, x + r, x + w - r, x + w
|
|
y0, y1, y2, y3 = y, y + r, y + h - r, y + h
|
|
path = []
|
|
path.extend([(x1, y0), (x2, y0)])
|
|
path.extend(arc(x2, y1, r, radians(270), radians(360), n))
|
|
path.extend([(x3, y1), (x3, y2)])
|
|
path.extend(arc(x2, y2, r, radians(0), radians(90), n))
|
|
path.extend([(x2, y3), (x1, y3)])
|
|
path.extend(arc(x1, y2, r, radians(90), radians(180), n))
|
|
path.extend([(x0, y2), (x0, y1)])
|
|
path.extend(arc(x1, y1, r, radians(180), radians(270), n))
|
|
return path
|
|
|
|
def padded_rounded_rectangle(x, y, w, h, r, p):
|
|
x += p
|
|
y += p
|
|
w -= p * 2
|
|
h -= p * 2
|
|
return rounded_rectangle(x, y, w, h, r)
|
|
|
|
def wall(x, y):
|
|
return [arc(x+0.5, y+0.5, 0.333, 0, 2*pi, 72)]
|
|
x0 = x + P
|
|
y0 = y + P
|
|
x1 = x + 1 - P
|
|
y1 = y + 1 - P
|
|
paths = [rectangle(x0, y0, x1 - x0, y1 - y0)]
|
|
paths.append([(x0, y0), (x1, y1)])
|
|
paths.append([(x0, y1), (x1, y0)])
|
|
return paths
|
|
|
|
def xy(i):
|
|
x = i % 6
|
|
y = i // 6
|
|
return (x, y)
|
|
|
|
def desc_paths(desc):
|
|
paths = []
|
|
lookup = defaultdict(list)
|
|
for i, c in enumerate(desc):
|
|
lookup[c].append(i)
|
|
for c in sorted(lookup):
|
|
ps = lookup[c]
|
|
if c == 'o':
|
|
continue
|
|
elif c == 'x':
|
|
for i in ps:
|
|
x, y = xy(i)
|
|
paths.extend(wall(x, y))
|
|
else:
|
|
stride = ps[1] - ps[0]
|
|
i0 = ps[0]
|
|
i1 = ps[-1]
|
|
x0, y0 = xy(i0)
|
|
x1, y1 = xy(i1)
|
|
dx = x1 - x0
|
|
dy = y1 - y0
|
|
# paths.append(padded_rounded_rectangle(x0, y0, dx + 1, dy + 1, R, P))
|
|
if c == 'A':
|
|
paths.append(padded_rectangle(x0, y0, dx + 1, dy + 1, 0.25))
|
|
else:
|
|
paths.append([(x0 + 0.5, y0 + 0.5), (x0 + dx + 0.5, y0 + dy + 0.5)])
|
|
# if c == 'A':
|
|
# if stride > 1:
|
|
if len(ps) == 3:
|
|
# if stride == 1:
|
|
# if len(ps) == 2:
|
|
# if False:
|
|
paths.append(padded_rectangle(x0, y0, dx + 1, dy + 1, 0.35))
|
|
# s = 0.1
|
|
# p = P + s
|
|
# while p < 0.5:
|
|
# paths.append(padded_rounded_rectangle(x0, y0, dx + 1, dy + 1, R, p))
|
|
# p += s
|
|
return paths
|
|
|
|
def main():
|
|
drawing = axi.Drawing()
|
|
font = axi.Font(axi.FUTURAL, 12)
|
|
n = 0
|
|
for line in fileinput.input():
|
|
fields = line.strip().split()
|
|
desc = fields[1]
|
|
moves = int(fields[0])
|
|
# if 'x' in desc:
|
|
# continue
|
|
paths = desc_paths(desc)
|
|
d = axi.Drawing(paths)
|
|
i = n % COLS
|
|
j = n // COLS
|
|
d = d.translate(i * 8, j * 10)
|
|
drawing.add(d)
|
|
|
|
d = font.wrap(str(moves), 10)
|
|
d = font.wrap(bin(moves)[2:].replace('1', '\\').replace('0', '/'), 10)
|
|
# d = d.scale(0.1, 0.1)
|
|
d = d.scale_to_fit_height(1)
|
|
d = d.move(i * 8 + 3, j * 10 + 6.5, 0.5, 0)
|
|
drawing.add(d)
|
|
|
|
n += 1
|
|
if n == N:
|
|
break
|
|
# d = axi.Drawing(paths)
|
|
d = drawing
|
|
d = d.rotate_and_scale_to_fit(W, H, step=90)
|
|
d.dump('rush.axi')
|
|
d.render(bounds=None, show_bounds=False, scale=300).write_to_png('rush.png')
|
|
|
|
if __name__ == '__main__':
|
|
main()
|