mirror of
https://github.com/TinyTapeout/tt-support-tools.git
synced 2026-11-03 04:36:09 +01:00
106 lines
3.4 KiB
Python
Executable File
106 lines
3.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
|
|
import gdstk
|
|
from git.repo import Repo
|
|
from PIL import Image, ImageDraw, ImageFont
|
|
|
|
from config import Config
|
|
|
|
|
|
class LogoGenerator:
|
|
def __init__(self, tt_dir, config: Config | None = None):
|
|
self.tt_dir = tt_dir
|
|
self.config = config
|
|
|
|
def gen_logo(self, variant: str, gds_file: str, shuttle=None, commit=None):
|
|
|
|
assert variant in ("top", "bottom")
|
|
if variant == "top":
|
|
# use included bitmap
|
|
img = Image.open(f"{self.tt_dir}/logo/tt_logo.png").convert("L")
|
|
|
|
elif variant == "bottom":
|
|
# generate bitmap from shuttle ID & commit hash
|
|
if shuttle is None:
|
|
assert self.config is not None
|
|
shuttle = self.config["id"]
|
|
if commit is None:
|
|
commit = Repo(".").commit().hexsha
|
|
|
|
img = Image.new("L", (200, 200), (0,))
|
|
draw = ImageDraw.Draw(img)
|
|
|
|
font_file = f"{self.tt_dir}/logo/font/UbuntuSansMono.ttf"
|
|
|
|
def font(size):
|
|
f = ImageFont.truetype(font_file, size)
|
|
f.set_variation_by_axes([700])
|
|
return f
|
|
|
|
shuttle_font_size = 88
|
|
shuttle_y_offset = -15
|
|
if len(shuttle) > 4:
|
|
shuttle = shuttle[:-1].upper() + shuttle[-1]
|
|
shuttle_font_size = 44
|
|
shuttle_y_offset = 10
|
|
draw.text(
|
|
(2, shuttle_y_offset), shuttle, fill=255, font=font(shuttle_font_size)
|
|
)
|
|
draw.text((2, 68), commit[:8], fill=255, font=font(44))
|
|
draw.text((2, 107), commit[8:18], fill=255, font=font(35))
|
|
draw.text((1, 137), commit[18:29], fill=255, font=font(32))
|
|
draw.text((1, 165), commit[29:], fill=255, font=font(32))
|
|
|
|
PRBOUNDARY_LAYER = 235
|
|
PRBOUNDARY_DATATYPE = 4
|
|
MET4_LAYER = 71
|
|
DRAWING_DATATYPE = 20
|
|
PIXEL_SIZE = 0.5 # um
|
|
|
|
lib = gdstk.Library()
|
|
cell = lib.new_cell(f"tt_logo_{variant}")
|
|
boundary = gdstk.rectangle(
|
|
(0, 0),
|
|
(img.width * PIXEL_SIZE, img.height * PIXEL_SIZE),
|
|
layer=PRBOUNDARY_LAYER,
|
|
datatype=PRBOUNDARY_DATATYPE,
|
|
)
|
|
cell.add(boundary)
|
|
|
|
for y in range(img.height):
|
|
for x in range(img.width):
|
|
color: int = img.getpixel((x, y)) # type: ignore
|
|
if color >= 128:
|
|
flipped_y = img.height - y - 1 # flip vertically
|
|
rect = gdstk.rectangle(
|
|
(x * PIXEL_SIZE, flipped_y * PIXEL_SIZE),
|
|
((x + 1) * PIXEL_SIZE, (flipped_y + 1) * PIXEL_SIZE),
|
|
layer=MET4_LAYER,
|
|
datatype=DRAWING_DATATYPE,
|
|
)
|
|
cell.add(rect)
|
|
|
|
lib.write_gds(gds_file)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
try:
|
|
if sys.argv[1] == "--top":
|
|
LogoGenerator(".").gen_logo("top", "logo/tt_logo_top.gds")
|
|
|
|
elif sys.argv[1] == "--bottom":
|
|
shuttle = sys.argv[2] # e.g. "tt10"
|
|
commit = sys.argv[3] # e.g. "0123456789abcdef0123456789abcdef01234567"
|
|
LogoGenerator(".").gen_logo(
|
|
"bottom", "logo/tt_logo_bottom.gds", shuttle, commit
|
|
)
|
|
|
|
except IndexError:
|
|
print(
|
|
f"Usage:\n {sys.argv[0]} --top\n {sys.argv[0]} --bottom <shuttle> <commit>",
|
|
file=sys.stderr,
|
|
)
|