summaryrefslogtreecommitdiff
path: root/Master/texmf-dist/scripts/puyotikz/puyotikz.py
diff options
context:
space:
mode:
Diffstat (limited to 'Master/texmf-dist/scripts/puyotikz/puyotikz.py')
-rwxr-xr-xMaster/texmf-dist/scripts/puyotikz/puyotikz.py294
1 files changed, 294 insertions, 0 deletions
diff --git a/Master/texmf-dist/scripts/puyotikz/puyotikz.py b/Master/texmf-dist/scripts/puyotikz/puyotikz.py
new file mode 100755
index 00000000000..66f536b4cae
--- /dev/null
+++ b/Master/texmf-dist/scripts/puyotikz/puyotikz.py
@@ -0,0 +1,294 @@
+# PuyoTikZ package for LaTeX.
+# Created by amosborne (GitHub) / terramyst (Twitter).
+# MIT License
+
+import numpy as np
+import itertools
+import string
+import re
+
+
+# Wraps a function which returns a string. Prints the result with a ";".
+def print_tikz(func):
+ def wrapper(*args, **kwargs):
+ print(func(*args, **kwargs) + ";")
+
+ return wrapper
+
+
+PUYO_RADIUS, PUYO_JOIN = 0.45, 0.25
+COLORS = {
+ "r": "red",
+ "y": "yellow",
+ "g": "green",
+ "b": "blue",
+ "p": "purple",
+ "n": "gray",
+}
+for key, value in COLORS.items():
+ COLORS[key] = value + "puyo"
+
+
+def puyoboard(cols, rows, hrows, boardpuyos, nextpuyos, label):
+ if not cols > 0:
+ raise UserWarning("{0} columns must be atleast 1.".format(cols))
+ if not rows > 0:
+ raise UserWarning("{0} rows must be atleast 1.".format(rows))
+ if hrows < 0:
+ raise UserWarning("{0} hidden rows must be atleast 0.".format(hrows))
+
+ draw_board(cols, rows, hrows, boardpuyos)
+ draw_nextpuyos(cols, rows, nextpuyos)
+ if label:
+ draw_labels(cols, rows, hrows, nextpuyos)
+
+
+def puyomarker(config):
+ pattern = "^([a-z]+)(\d+)([rygbpn])([A-Z])$"
+
+ @print_tikz
+ def singlemark(subconfig):
+ match = re.search(pattern, subconfig)
+ if match is None:
+ raise UserWarning("Bad puyo marker format string.")
+ else:
+ colid, rowid, puyo, lid = match.groups()
+
+ rowidx = int(rowid) - 1
+ colidx = next(
+ idx for idx, x in enumerate(excel_cols(upper=False)) if x == colid
+ )
+ pos = colidx + 0.5, rowidx + 0.5
+
+ tikz = "\\draw[{0},thick]".format("dark" + COLORS[puyo])
+ tikz += " ({0},{1}) circle ({2});".format(*pos, PUYO_RADIUS)
+ tikz += " \\node[anchor=center, font=\sffamily]"
+ tikz += " at ({0},{1}) {{\\small {2}}}".format(*pos, lid)
+ return tikz
+
+ subconfigs = config.split("/")
+ if subconfigs == [""]:
+ return
+ else:
+ for subconfig in subconfigs:
+ singlemark(subconfig)
+
+
+def draw_board(cols, rows, hrows, boardpuyos):
+ # draw the board grid
+ draw_grid(params=["gray", "ultra thin"], size=(cols, rows))
+ draw_grid(
+ params=["gray", "ultra thin", "line cap=round"],
+ origin=(0, rows),
+ size=(cols, hrows),
+ )
+ draw_grid(
+ params=["black", "line cap=round"], stepsize=(cols, rows), size=(cols, rows)
+ )
+
+ # draw the puyos on the board
+ draw_puyos(puyos=boardpuyos.split("/"), size=(cols, rows), nhidden=hrows)
+
+
+def draw_nextpuyos(cols, rows, nextpuyos):
+ nextpuyos = nextpuyos.split("/")
+ if nextpuyos == [""]:
+ return
+ for idx, puyos in enumerate(nextpuyos):
+ if not len(puyos) == 2:
+ raise UserWarning("Next puyos must be of length 2.")
+ draw_puyos(
+ puyos=[puyos], origin=(cols + 0.75, rows - 1.5 - 3 * idx), size=(1, 2)
+ )
+
+
+def draw_labels(cols, rows, hrows, nextpuyos):
+ @print_tikz
+ def draw_label(pos, text, size="\\normalsize", style="\\rmfamily"):
+ return "\\node[anchor=mid, font={4}] at ({0},{1}) {{{3} {2}}}".format(
+ *pos, text, size, style
+ )
+
+ for idx in range(0, rows + hrows):
+ draw_label((-0.5, 0.5 + idx), idx + 1)
+
+ for idx, label in zip(range(0, cols), excel_cols(upper=False)):
+ draw_label((0.5 + idx, -0.5), label)
+
+ nextpuyoscount = len(nextpuyos.split("/"))
+ if nextpuyos.split("/") == [""]:
+ return
+ for idx, label in zip(range(0, nextpuyoscount), excel_cols(upper=True)):
+ draw_label((cols + 0.75, rows - 1.5 - 3 * idx), label, "\\small", "\\sffamily")
+
+
+@print_tikz
+def draw_grid(params=[], origin=(0, 0), stepsize=(1, 1), size=(6, 12)):
+ extent = tuple([x + y for x, y in zip(origin, size)])
+ tikz = "\\draw"
+ tikz += "[" + ",".join(params) + "]"
+ tikz += " ({0},{1})".format(*origin)
+ tikz += " grid [xstep={0},ystep={1}]".format(*stepsize)
+ tikz += " ({0},{1})".format(*extent)
+ return tikz
+
+
+def draw_puyos(puyos=[], origin=(0, 0), size=(6, 12), nhidden=0):
+ # construct a 2D matrix of strings (representing puyos)
+ board = np.full((size[0], size[1] + nhidden), "")
+ for ridx, col in enumerate(puyos):
+ for cidx, puyo in enumerate(list(col)):
+ if puyo not in COLORS.keys():
+ raise UserWarning(
+ "{0} is not a valid puyo identifier (rygbpn).".format(puyo)
+ )
+ try:
+ board[ridx, cidx] = puyo
+ except IndexError:
+ raise UserWarning("Puyo layout string has too many rows or columns.")
+
+ # helper function for converting true position
+ def true_pos(pos):
+ return tuple([x + y + 0.5 for x, y in zip(pos, origin)])
+
+ # draw all the puyo circles
+ for pos, puyo in np.ndenumerate(board):
+ if puyo:
+ draw_puyo(true_pos(pos), puyo)
+
+ # connect to adjacent puyos (except nuisance and hidden row)
+ for pos, puyo in np.ndenumerate(board):
+ for direc in {(1, 0), (-1, 0), (0, 1), (0, -1)}:
+ if not 0 <= pos[0] + direc[0] < board.shape[0]:
+ continue
+ elif not 0 <= pos[1] + direc[1] < board.shape[1] - nhidden:
+ continue
+ elif not 0 <= pos[1] < board.shape[1] - nhidden:
+ continue
+
+ other = board[pos[0] + direc[0], pos[1] + direc[1]]
+ if puyo and other == puyo and puyo != "n":
+ connect_puyo(true_pos(pos), direc, puyo)
+
+ # draw all the puyo faces
+ for pos, puyo in np.ndenumerate(board):
+ if puyo:
+ draw_face(true_pos(pos), puyo)
+
+ # outline the puyos with a darker color
+ for pos, puyo in np.ndenumerate(board):
+ for direc in {(1, 0), (-1, 0), (0, 1), (0, -1)}:
+ if not 0 <= pos[0] + direc[0] < board.shape[0]:
+ if puyo:
+ outline_puyo(true_pos(pos), direc, puyo, join=False)
+ continue
+ elif not 0 <= pos[1] + direc[1] < board.shape[1] - nhidden:
+ if puyo:
+ outline_puyo(true_pos(pos), direc, puyo, join=False)
+ continue
+ elif not 0 <= pos[1] < board.shape[1] - nhidden:
+ if puyo:
+ outline_puyo(true_pos(pos), direc, puyo, join=False)
+ continue
+
+ other = board[pos[0] + direc[0], pos[1] + direc[1]]
+ if puyo and other == puyo and puyo != "n":
+ outline_puyo(true_pos(pos), direc, puyo, join=True)
+ elif puyo:
+ outline_puyo(true_pos(pos), direc, puyo, join=False)
+
+
+@print_tikz
+def draw_face(pos=(0, 0), puyo="r"):
+ def tikz_shift(shift=(0, 0)):
+ total_shift = pos[0] + shift[0], pos[1] + shift[1]
+ return "[xshift={0}cm,yshift={1}cm]".format(*total_shift)
+
+ tikz = "\\draw[white,semithick]"
+ if puyo == "p":
+ tikz += " ({0}150:{1}) -- ({0}-30:{1});".format(tikz_shift(), PUYO_RADIUS)
+ elif puyo == "r":
+ tikz += " ({0}150:{1}) -- ({0}0,0) -- ({0}30:{1});".format(
+ tikz_shift(), PUYO_RADIUS
+ )
+ elif puyo == "g":
+ tikz += " ({0}-90:{1}) arc (-90:0:{1})".format(
+ tikz_shift((-PUYO_RADIUS, PUYO_RADIUS)), PUYO_RADIUS
+ )
+ tikz += " ({0}-90:{1}) arc (-90:-180:{1});".format(
+ tikz_shift((PUYO_RADIUS, PUYO_RADIUS)), PUYO_RADIUS
+ )
+ elif puyo == "b":
+ tikz += " ({0}180:{1}) -- ({0}90:{2}) -- ({0}0:{1});".format(
+ tikz_shift(), PUYO_RADIUS, PUYO_RADIUS * np.sin(np.deg2rad(30))
+ )
+ elif puyo == "y":
+ tikz += " ({0}150:{1}) -- ({0}30:{1});".format(tikz_shift(), PUYO_RADIUS)
+ elif puyo == "n":
+ tikz += " ({0}210:{1}) -- ({0}-30:{1});".format(tikz_shift(), PUYO_RADIUS)
+ else:
+ tikz = ""
+
+ return tikz
+
+
+@print_tikz
+def draw_puyo(pos=(0, 0), puyo="r"):
+ tikz = "\\draw[{0},fill={0}]".format(COLORS[puyo])
+ tikz += " ({0},{1}) circle ({2})".format(*pos, PUYO_RADIUS)
+ return tikz
+
+
+@print_tikz
+def connect_puyo(pos=(0, 0), direc=(1, 0), puyo="r"):
+ phi, join1, join2 = connect_puyo_params(pos, direc)
+ tikz = "\\draw[{0},fill={0}]".format(COLORS[puyo])
+ tikz += " ([xshift={0}cm,yshift={1}cm]".format(*pos)
+ tikz += "{0}:{2}) arc ({0}:{1}:{2})".format(phi - 45, phi + 45, PUYO_RADIUS)
+ tikz += " to[out={0},in={1}] ({2},{3})".format(phi - 45, phi - 180, *join1)
+ tikz += " -- ({0},{1})".format(*join2)
+ tikz += " to[out={0},in={1}] cycle".format(phi - 180, phi + 45)
+ return tikz
+
+
+@print_tikz
+def outline_puyo(pos=(0, 0), direc=(1, 0), puyo="r", join=False):
+ phi, join1, join2 = connect_puyo_params(pos, direc)
+ tikz = "\\draw[{0}]".format("dark" + COLORS[puyo])
+
+ if join:
+ tikz += " ([xshift={0}cm,yshift={1}cm]".format(*pos)
+ tikz += "{0}:{1})".format(phi + 45, PUYO_RADIUS)
+ tikz += " to[out={0},in={1}] ({2},{3})".format(phi - 45, phi - 180, *join1)
+ tikz += " ({0},{1})".format(*join2)
+ tikz += " to[out={0},in={1}]".format(phi - 180, phi + 45)
+ tikz += " ([xshift={0}cm,yshift={1}cm]".format(*pos)
+ tikz += "{0}:{1})".format(phi - 45, PUYO_RADIUS)
+ else:
+ tikz += " ([xshift={0}cm,yshift={1}cm]".format(*pos)
+ tikz += "{0}:{2}) arc ({0}:{1}:{2})".format(phi - 45, phi + 45, PUYO_RADIUS)
+
+ return tikz
+
+
+def connect_puyo_params(pos, direc):
+ """Determine the path angle and midpoint locations."""
+ phi = np.rad2deg(np.arctan2(direc[1], direc[0])) % 360
+ x, y1, y2 = 0.5, PUYO_JOIN, -PUYO_JOIN
+
+ c, s = np.cos(np.deg2rad(phi)), np.sin(np.deg2rad(-phi))
+ j = np.matrix([[c, s], [-s, c]])
+
+ def rotate_point(y):
+ m = np.dot(j, [x, y])
+ return float(m.T[0]) + pos[0], float(m.T[1]) + pos[1]
+
+ return phi, rotate_point(y1), rotate_point(y2)
+
+
+def excel_cols(upper=True):
+ n = 1
+ letters = string.ascii_uppercase if upper else string.ascii_lowercase
+ while True:
+ yield from ("".join(group) for group in itertools.product(letters, repeat=n))
+ n += 1