axi/examples/rush.py

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()