figure.py
4.1 kB · python · 132 lines
1import os23# STYLE45FONT = "Helvetica,Arial,sans-serif"6# PALETTE7INK = "#000000"8MUTED = "#8e8e93"9PALE = "#bababf"10PAPER = "#ffffff"11CONCEPT = {12 "gasket": "#00cad8",13 "carpet": "#008cff",14 "sponge": "#6768fa",15 "prime": "#ff325a",16 "bound": "#ff8f2c",17 "window": "#ff3d40",18 "control": "#8e8e93",19}20PALETTE = {21 "black": "#000000",22 "white": "#ffffff",23 "red": "#ff3d40",24 "red-light": "#ff9d95",25 "red-dark": "#a80016",26 "orange": "#ff8f2c",27 "orange-light": "#ffc093",28 "orange-dark": "#a25400",29 "yellow": "#ffd100",30 "yellow-light": "#ffe591",31 "yellow-dark": "#9e8100",32 "green": "#32cc58",33 "green-light": "#5eee79",34 "green-dark": "#007f2c",35 "mint": "#00d1bb",36 "mint-light": "#48efd8",37 "mint-dark": "#008173",38 "teal": "#00cad8",39 "teal-light": "#48e9f7",40 "teal-dark": "#007c85",41 "cyan": "#1ec9f3",42 "cyan-light": "#86e2ff",43 "cyan-dark": "#007c98",44 "blue": "#008cff",45 "blue-light": "#84bdff",46 "blue-dark": "#00559f",47 "indigo": "#6768fa",48 "indigo-light": "#9ea9ff",49 "indigo-dark": "#3c2abc",50 "purple": "#d332e9",51 "purple-light": "#f08aff",52 "purple-dark": "#870097",53 "pink": "#ff325a",54 "pink-light": "#ff9a9f",55 "pink-dark": "#a50030",56 "brown": "#b18462",57 "brown-light": "#dfaf8c",58 "brown-dark": "#754c2b",59 "gray": "#8e8e93",60 "gray-light": "#bababf",61 "gray-dark": "#56565a",62}63# PALETTE END64STROKE = 1.265THIN = 0.666MARGIN = 366768def esc(s):69 return s.replace("&", "&").replace("<", "<").replace(">", ">")7071def header(width, height):72 return ['<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 %d %d" width="%d" height="%d">' % (width, height, width, height), '<rect width="%d" height="%d" fill="%s"/>' % (width, height, PAPER)]7374def text(x, y, s, size=11, fill=INK, anchor="start", weight=None):75 w = ' font-weight="%s"' % weight if weight else ""76 return '<text x="%.1f" y="%.1f" font-family="%s" font-size="%d" fill="%s" text-anchor="%s"%s>%s</text>' % (x, y, FONT, size, fill, anchor, w, esc(s))7778def line(x1, y1, x2, y2, stroke=INK, width=STROKE):79 return '<line x1="%.1f" y1="%.1f" x2="%.1f" y2="%.1f" stroke="%s" stroke-width="%.1f"/>' % (x1, y1, x2, y2, stroke, width)8081def rect(x, y, w, h, fill, opacity=1.0):82 return '<rect x="%.1f" y="%.1f" width="%.1f" height="%.1f" fill="%s" opacity="%.2f"/>' % (x, y, w, h, fill, opacity)8384def circle(x, y, r, fill):85 return '<circle cx="%.1f" cy="%.1f" r="%.1f" fill="%s"/>' % (x, y, r, fill)8687def save(parts, path):88 with open(path, "w") as handle:89 handle.write("\n".join(parts + ["</svg>"]) + "\n")9091# STYLE END9293# DESIGN9495DARK = CONCEPT["sponge"]96LIGHT = PALETTE["indigo-light"]9798def kind(v):99 if sum(d == 1 for d in v) > 1:100 return "gone"101 return "dark" if sum(d == 0 for d in v) <= 1 else "pale"102103def main():104 here = os.path.dirname(os.path.abspath(__file__))105 out = os.path.join(here, "..", "figures", "design.svg")106 os.makedirs(os.path.dirname(out), exist_ok=True)107 cell = 34108 pad = 14109 gap = 30110 grid = 3 * cell111 width = 3 * grid + 2 * gap + 2 * pad112 height = grid + 2 * pad + 26113 rows = header(width, height)114 for c in range(3):115 ox = pad + c * (grid + gap)116 for a in range(3):117 for b in range(3):118 x = ox + a * cell119 y = pad + (2 - b) * cell120 style = kind((a, b, c))121 if style == "gone":122 rows.append(line(x + 8, y + 8, x + cell - 8, y + cell - 8, PALE, 2))123 rows.append(line(x + 8, y + cell - 8, x + cell - 8, y + 8, PALE, 2))124 else:125 rows.append(rect(x + 3, y + 3, cell - 6, cell - 6, DARK if style == "dark" else LIGHT))126 rows.append('<rect x="%d" y="%d" width="%d" height="%d" fill="none" stroke="%s" stroke-width="%.1f"/>' % (ox, pad, grid, grid, MUTED, STROKE))127 rows.append(text(ox + grid / 2, pad + grid + 20, "c = %d" % c, size=15, anchor="middle"))128 save(rows, out)129 print("drew figures/design.svg")130131if __name__ == "__main__":132 main()