sm64pc/tools/texture_converter.py

#!/usr/bin/env python3

import sys
import os
import os.path
import glob
import re
from pathlib import Path

prefixes = ["ALIGNED8 static const u8", "static const u8", "ALIGNED8 static u8", "ALIGNED8 u8", "ALIGNED8 const u8", "u8"]

def parse_model(file, lines, overwrite):
    searchInclude = False
    searchLineIndex = 0
    searchLineOriginal = ""
    needsReplace = False
    includePath = ""
    for lineIndex, uline in enumerate(lines, 0):
        line = uline.strip()
        if not searchInclude:
            for prefix in prefixes:
                if line.rstrip().startswith(prefix) and "[] = {" in line:
                    searchInclude = True
                    searchLineIndex = lineIndex
                    searchLineOriginal = line
        else:
            if line.startswith("#include"):
                includePath = line.replace("#include ", "").replace(".inc.c", "")
            elif line.endswith("};") and includePath != "":
                lines[searchLineIndex] = searchLineOriginal.replace("[] = {", f"[] = {includePath};")
                lines[searchLineIndex + 1] = ""
                lines[searchLineIndex + 2] = "\n"
                searchInclude = False
                searchLineIndex = 0
                searchLineOriginal = ""
                needsReplace = True
                includePath = ""
            else:
                searchInclude = False
    nfile = os.path.abspath(file) if overwrite else os.path.abspath(file.replace("\\", "/").replace("Render96ex/", "Render96ex/overwriten/"))
    npath = os.path.dirname(nfile)

    if needsReplace:
        if not os.path.isdir(npath):
            os.makedirs(npath)
        f = open(nfile, "w")
        f.writelines(lines)
        f.close()

if not len(sys.argv) > 1:
    path = f"{str(Path(__file__).resolve().parents[1])}/**/*.c"

    print(f"Converting repo textures this may take a long time")

    for file in glob.iglob(path, recursive=True):
        try:
            if not "\\build" in file:
                parse_model(file, open(file, 'r').readlines(), False)
        except Exception as err:
            print(f"Error on {file}")
            print(err)
else:
    file = os.path.abspath(sys.argv[1])
    try:
        if os.path.isfile(file):
            print(f"Converting {file} textures")
            parse_model(file, open(file, 'r').readlines(), True)
        else:
            print(f"Invalid Path")
    except Exception as err:
        print(f"Error on {file}")
        print(err)