"""Lays the joint markers over the dimmed model render (Pillow). Run: python compose_rig_preview.py <previews dir>"""
import sys, os
from PIL import Image, ImageEnhance

prev = sys.argv[1]
for tag in ("above", "below"):
    a = os.path.join(prev, "_rig_%s_model.png" % tag)
    b = os.path.join(prev, "_rig_%s_markers.png" % tag)
    model = Image.open(a).convert("RGB")
    model.save(os.path.join(prev, "DockArm_rig_posed_%s.png" % tag))
    dim = ImageEnhance.Brightness(model).enhance(0.45).convert("RGBA")
    out = Image.alpha_composite(dim, Image.open(b).convert("RGBA")).convert("RGB")
    out.save(os.path.join(prev, "DockArm_rig_joints_%s.png" % tag))
    os.remove(a); os.remove(b)
    print("wrote DockArm_rig_joints_%s.png and DockArm_rig_posed_%s.png" % (tag, tag))