#!/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')