mirror of
https://github.com/pmret/papermario.git
synced 2024-11-08 20:12:30 +01:00
88 lines
3.4 KiB
Python
Executable File
88 lines
3.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import re
|
|
from glob import glob
|
|
from stringcase import constcase, pascalcase # pip install stringcase
|
|
|
|
DIR = os.path.dirname(__file__)
|
|
SR_DATABASE = os.path.join(DIR, "star-rod/database")
|
|
FILES = (
|
|
glob(os.path.join(SR_DATABASE, "types/*.enum")) +
|
|
glob(os.path.join(SR_DATABASE, "types/**/*.enum")) +
|
|
glob(os.path.join(SR_DATABASE, "types/*.flags")) +
|
|
glob(os.path.join(SR_DATABASE, "types/**/*.flags"))
|
|
)
|
|
|
|
with open(os.path.join(DIR, "../include/enums.h"), "w") as h:
|
|
h.write("""#ifndef _ENUMS_H_
|
|
#define _ENUMS_H_
|
|
|
|
#include "ultra64.h"
|
|
#include "types.h"
|
|
|
|
""")
|
|
|
|
for filename in FILES:
|
|
with open(filename, "r") as file:
|
|
lines = file.readlines()
|
|
|
|
# Get enum attributes
|
|
namespace = pascalcase(re.match(r"[^ \t]*", lines[0]).group(0))
|
|
library_name = re.match(r"[^ \t]*", lines[1]).group(0)
|
|
reverse = True if re.match(r"[^ \t]*", lines[2]).group(0) == "true" else False
|
|
|
|
# Renamed namespaces
|
|
if namespace == "Item": namespace = "ItemId"
|
|
if namespace == "Sound": namespace = "SoundId"
|
|
if namespace == "Partner": namespace = "PartnerId"
|
|
if namespace == "Sprite": namespace = "SpriteId"
|
|
if namespace == "Outcome": namespace = "EncounterOutcome"
|
|
#if namespace == "Phase": namespace = "BattlePhase"
|
|
if namespace == "Actor": namespace = "ActorId"
|
|
#if namespace == "Status": namespace = "ActorStatus"
|
|
#if namespace == "StatusFlags": namespace = "ActorStatusFlags"
|
|
#if namespace == "Event": namespace = "BattleEvent"
|
|
if namespace == "Decoration": namespace = "DecorationId"
|
|
if namespace == "Npc": namespace = "NpcId"
|
|
if namespace == "Trigger": namespace = "TriggerFlags"
|
|
if namespace == "Anim": continue
|
|
if namespace == "Entity": continue # just ram addresses
|
|
|
|
# Get a list of tuples containing (name, value)
|
|
items = []
|
|
name_max_len = 0
|
|
comment = False
|
|
for line in lines[4:]:
|
|
line = line.rstrip("\n")
|
|
if not comment and not line.startswith("%") and "=" in line:
|
|
m = re.match(r"[ \t]*?([^ \t]*)[ \t]*=[ \t]*([^ \t]*)", line)
|
|
if m:
|
|
if reverse:
|
|
name = m.group(1)
|
|
value = int(m.group(2), 16)
|
|
else:
|
|
name = m.group(2)
|
|
value = int(m.group(1), 16)
|
|
if value < 0:
|
|
str_value = f"{value}"
|
|
else:
|
|
str_value = "0x" + f"{value:08x}".upper()
|
|
|
|
if name.upper() != name:
|
|
name = constcase(name)
|
|
items.append((name, str_value))
|
|
name_max_len = max(len(name), name_max_len)
|
|
elif "/%" in line:
|
|
comment = True
|
|
elif "%/" in line:
|
|
comment = False
|
|
|
|
# Write a section to the header file for this enum
|
|
h.write(f"typedef UNK_TYPE {namespace};\n")
|
|
for (name, value) in items:
|
|
h.write(f"#define {namespace}_{name.ljust(name_max_len)} {value}\n")
|
|
h.write("\n")
|
|
|
|
h.write("#endif\n")
|