forked from Mirror/GodMode9
67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
from argparse import ArgumentParser
|
||
|
|
from os import path
|
||
|
|
|
||
|
|
import json
|
||
|
|
import struct
|
||
|
|
|
||
|
|
parser = ArgumentParser(description="Creates an TRF translation for GodMode9 from a translation JSON")
|
||
|
|
parser.add_argument("input", type=str, help="JSON to convert from")
|
||
|
|
parser.add_argument("output", type=str, help="to output to")
|
||
|
|
parser.add_argument("version", type=int, help="translation version, from language.inl")
|
||
|
|
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
with open(args.input, "r") as f:
|
||
|
|
# read JSON
|
||
|
|
strings = json.load(f)
|
||
|
|
if "GM9_LANGUAGE" not in strings:
|
||
|
|
print("Fatal: Input is not a valid JSON file")
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
# Encode strings to UTF-8 bytestrings
|
||
|
|
strings = {item: strings[item].encode("utf-8") + b"\0" for item in strings}
|
||
|
|
|
||
|
|
# Remove language name from strings
|
||
|
|
lang_name = strings["GM9_LANGUAGE"]
|
||
|
|
del strings["GM9_LANGUAGE"]
|
||
|
|
|
||
|
|
# sort map
|
||
|
|
# fontMap = sorted(fontMap, key=lambda x: x["mapping"])
|
||
|
|
|
||
|
|
# write file
|
||
|
|
with open(args.output, "wb") as out:
|
||
|
|
out.write(b"RIFF")
|
||
|
|
out.write(struct.pack("<L", 0)) # Filled at end
|
||
|
|
|
||
|
|
# metadata
|
||
|
|
out.write(b"META")
|
||
|
|
out.write(struct.pack("<LLL32s", 40, args.version, len(strings), lang_name))
|
||
|
|
|
||
|
|
# character data
|
||
|
|
out.write(b"SDAT")
|
||
|
|
sectionSize = sum(len(strings[item]) for item in strings)
|
||
|
|
padding = 4 - sectionSize % 4 if sectionSize % 4 else 0
|
||
|
|
out.write(struct.pack("<L", sectionSize + padding))
|
||
|
|
out.write(b"".join(strings.values()))
|
||
|
|
out.write(b"\0" * padding)
|
||
|
|
|
||
|
|
# character map
|
||
|
|
out.write(b"SMAP")
|
||
|
|
sectionSize = len(strings) * 2
|
||
|
|
padding = 4 - sectionSize % 4 if sectionSize % 4 else 0
|
||
|
|
out.write(struct.pack("<L", sectionSize + padding))
|
||
|
|
offset = 0
|
||
|
|
for string in strings.values():
|
||
|
|
out.write(struct.pack("<H", offset))
|
||
|
|
offset += len(string)
|
||
|
|
out.write(b"\0" * padding)
|
||
|
|
|
||
|
|
# write final size
|
||
|
|
outSize = out.tell()
|
||
|
|
out.seek(4)
|
||
|
|
out.write(struct.pack("<L", outSize - 8))
|
||
|
|
|
||
|
|
print("Info: %s created with %d strings." % (args.output, len(strings)))
|