macrovision/scripts/process-model.py

import sys
import re
import json
import glob
import os
import subprocess
import pathlib

# an affront to god
def combine(base_path, highlight_path, vivid_path, output_path):
    base = open(base_path, "r", encoding="utf-8").read()
    highlight = open(highlight_path, "r", encoding="utf-8").read()
    vivid = open(vivid_path, "r", encoding="utf-8").read()
    
    base_data = re.search("<g.*", base, flags=re.DOTALL)[0][:-7]
    highlight_data = re.search("<g.*", highlight, flags=re.DOTALL)[0][:-7]
    vivid_data = vivid.replace("</metadata>", "</metadata>" + base_data + "\n" + highlight_data)

    with open(output_path, "w", encoding="utf-8") as f:
        f.write(vivid_data)

    subprocess.run([INKSCAPE, "--without-gui", "--export-plain-svg=" + output_path.resolve().__str__(), "--export-area-drawing", output_path], shell=True)

configdir = pathlib.Path(__file__).parent
configpath = configdir.joinpath("config.json")
config = json.load(open(configpath, encoding="utf-8"))

workdir = pathlib.Path(config["work-directory"])

sourcedir = workdir.joinpath(sys.argv[1])
macrodir = pathlib.Path(config["macrovision-directory"])

print(sourcedir, macrodir) 
side_strings = []

POTRACE = config["potrace"]
INKSCAPE = config["inkscape"]

output = {}

with open(sourcedir.joinpath("data.json"), "r", encoding="utf-8") as file:
    all_data = json.load(file)

group_name = all_data["name"]
category = all_data["kind"]

outputdir = macrodir.joinpath("media").joinpath(category).joinpath(group_name)
os.makedirs(outputdir, exist_ok=True)

base_lut = configdir.joinpath("luts").joinpath("base-lut.png").__str__()
highlight_lut = configdir.joinpath("luts").joinpath("highlight-lut.png").__str__()
vivid_lut = configdir.joinpath("luts").joinpath("vivid-lut.png").__str__()

for data in all_data["forms"]:
    name = data["name"]
    for view in data["views"]:
        view_name = view["name"]
        input = sourcedir.joinpath(name + "-" + view_name + ".png")
        result = outputdir.joinpath(name + "-" + view_name + ".svg")
        print(result)
        if os.path.exists(result) and os.path.getmtime(input) < os.path.getmtime(result):
            print("Skipping ", input)
            # continue

        input_base = sourcedir.joinpath(name + "-" + view_name + "-base.bmp")
        input_highlight = sourcedir.joinpath(name + "-" + view_name + "-highlight.bmp")
        input_vivid = sourcedir.joinpath(name + "-" + view_name + "-vivid.bmp")

        subprocess.run(["convert", input, base_lut, "-channel", "RGB", "-clut", "-background", "#FFFFFF", "-flatten", input_base], shell=True)
        subprocess.run(["convert", input, highlight_lut, "-channel", "RGB", "-clut", "-background", "#FFFFFF", "-flatten", input_highlight], shell=True)
        subprocess.run(["convert", input, vivid_lut, "-channel", "RGB", "-clut", "-background", "#FFFFFF", "-flatten", input_vivid], shell=True)
        
        output_base = sourcedir.joinpath(name + "-" + view_name + "-base.svg")
        output_highlight = sourcedir.joinpath(name + "-" + view_name + "-highlight.svg")
        output_vivid = sourcedir.joinpath(name + "-" + view_name + "-vivid.svg")

        subprocess.run([POTRACE, input_base, "-b", "svg", "-o", output_base], shell=True)
        subprocess.run([POTRACE, input_highlight, "-b", "svg", "-C", "#1a1a1a", "-o", output_highlight], shell=True)
        subprocess.run([POTRACE, input_vivid, "-b", "svg", "-C", "#333333", "-o", output_vivid], shell=True)
        combine(output_base, output_highlight, output_vivid, result)
        # os.unlink(input_base)
        # os.unlink(input_highlight)