diff --git a/tt_annotate.py b/tt_annotate.py new file mode 100755 index 0000000..d8db13e --- /dev/null +++ b/tt_annotate.py @@ -0,0 +1,32 @@ +#!/usr/bin/env python3 + +from PIL import Image, ImageDraw, ImageFont + +img = Image.open('pics/tinytapeout.png').convert('RGBA') +overlay = Image.new('RGBA', img.size, (0, 0, 0, 0)) +draw = ImageDraw.Draw(overlay) +font = ImageFont.truetype("/usr/share/fonts/dejavu/DejaVuSans.ttf", 30) + +# coordinate mapping +xmap = ((88, 272), (2676, 1646)) # two pairs of (chip x coord, picture x coord) +ymap = ((95, 1879), (3240, 213)) # two pairs of (chip y coord, picture y coord) +xshift, yshift = 40, -43 # text position relative to macro origin + +for line in open('openlane/user_project_wrapper/macro.cfg'): + macro, x, y, orient = line.split() + if macro == 'scan_controller': + continue + elif macro.startswith('scanchain_'): + continue + name, num = macro.rsplit('_', 1) + num, x, y = int(num), int(x), int(y) + x = int((x-xmap[0][0])/(xmap[1][0]-xmap[0][0])*(xmap[1][1]-xmap[0][1])+xmap[0][1]) + y = int((y-ymap[0][0])/(ymap[1][0]-ymap[0][0])*(ymap[1][1]-ymap[0][1])+ymap[0][1]) + x += xshift + y += yshift + msg = str(num) + w, h = draw.textsize(msg, font=font) + draw.text((x-w/2, y-h/2), msg, fill=(0, 0, 0, 255), font=font) + +combined = Image.alpha_composite(img, overlay).convert('RGB') +combined.save('pics/tinytapeout_numbered.png')