sotn-decomp/tools/splat_ext/utils.py

import ctypes
import os
import sys

sys.path.append(f"{os.getcwd()}/tools/splat_ext")
from splat.util import log

subchar81_dict = {
    0x44: 0x2E,  # '.'
    0x43: 0x2C,  # ','
    0x48: 0x3F,  # '?'
    0x49: 0x21,  # '!'
    0x66: 0x27,  # '''
    0x68: 0x22,  # '"'
    0x69: 0x28,  # '('
    0x6A: 0x29,  # ')'
    0x6D: 0x5B,  # '['
    0x6E: 0x5D,  # ']'
    0x7B: 0x2B,  # '+'
    0x7C: 0x2D,  # '-'
    0x93: 0x25,  # '%'
}


def log_fatal(msg):
    log.write(msg, status="error")
    raise Exception(msg)


def log_warn(msg):
    log.write(msg, status="warn")


def from_s32(num):
    return num.to_bytes(4, byteorder="little", signed=True)


def to_s32(data: bytearray):
    return int.from_bytes(data[:4], byteorder="little", signed=True)


def from_u32(num):
    return bytes([(num >> i) & 0xFF for i in range(0, 32, 8)])


def to_u32(data: bytearray):
    return (data[0] << 0) | (data[1] << 8) | (data[2] << 16) | (data[3] << 24)


def from_16(num):
    return bytes([num & 0xFF, (num >> 8) & 0xFF])


def to_s16(data: bytearray):
    return ctypes.c_int16(data[0] | (data[1] << 8)).value


def to_u16(data: bytearray):
    return data[0] | (data[1] << 8)


def from_s8(num):
    if num < 0:
        num += 256
    return bytes([num])


def to_s8(data: bytearray):
    raw = data[0]
    if raw >= 128:
        return raw - 256
    return raw


def from_u8(num):
    return bytes([num])


def to_u8(data: bytearray):
    return data[0]


def from_bool(val):
    return bytes([int(val)])


def to_bool(data):
    return bool(data[0])


def from_ptr_str(ptr_str):
    return from_u32(int(ptr_str[2:], 16))


def to_ptr_str(data):
    return f"0x{to_u32(data):08X}"


def sotn_menu_name_to_str(data: bytearray) -> str:
    end_of_str = data.find(0xFF)
    assert end_of_str >= 0
    if end_of_str == 0:
        return ""

    utf8 = bytearray(data[:end_of_str])
    for i in range(len(utf8)):
        utf8[i] += 0x20
    return utf8.decode("utf-8")


def sotn_menu_desc_to_str(data: bytearray) -> str:
    utf8 = []
    while data[0] not in {0xFF, 0x00}:
        if data[0] == 0x81:
            ch = subchar81_dict.get(data[1])
            if ch is None:
                raise Exception(f"subchar {data[0]:02X} {data[1]:02X} not recognised")
            data = data[2:]
        elif data[0] == 0x82:
            if 0x4F <= data[1] <= 0x58:
                ch = data[1] - 0x4F + 0x30  # from '0' to '9'
            else:
                raise Exception(f"subchar {data[0]:02X} {data[1]:02X} not recognised")
            data = data[2:]
        else:
            ch = data[0]
            data = data[1:]
        utf8.append(ch)
    return bytes(utf8).decode("utf-8")