mirror of
https://github.com/Escartem/AnimeWwise.git
synced 2026-06-05 07:50:23 +08:00
137 lines
3.1 KiB
Python
137 lines
3.1 KiB
Python
# reader for the .map format i've made to improve reading speed and mapping size
|
|
from filereader import FileReader
|
|
import os
|
|
import json
|
|
|
|
|
|
class Mapper:
|
|
def __init__(self, mapping_file):
|
|
file = open(mapping_file, "rb")
|
|
reader = FileReader(file, "little") # encoded as little
|
|
|
|
# check file
|
|
if reader.ReadBytes(4) != b"ESFM":
|
|
file.close()
|
|
raise Exception("mapping was invalid")
|
|
|
|
reader.ReadBytes(2)
|
|
|
|
map_version = reader.ReadBytes(2)
|
|
if map_version == b"\x56\x31":
|
|
print(f"Warning: you are using an old version of the mapping that is no longer supported, please use a newer one or download an older version of this tool.")
|
|
raise Exception("outdated mapping")
|
|
elif map_version == b"\x56\x32":
|
|
self.reader = reader
|
|
self.process_map()
|
|
else:
|
|
file.close()
|
|
raise Exception("invalid mapping version")
|
|
|
|
def process_map(self):
|
|
reader = self.reader
|
|
|
|
# utils
|
|
val = lambda length: int.from_bytes(reader.ReadBytes(length), "little")
|
|
raw = lambda length: reader.ReadBytes(length).rstrip(b"\x00").decode("utf-8")
|
|
|
|
# get map meta
|
|
reader.ReadBytes(2)
|
|
|
|
games = {
|
|
b"ys": "Genshin"
|
|
# more later
|
|
}
|
|
|
|
infos = {
|
|
"game": games[reader.ReadBytes(2)],
|
|
"version": list(raw(2)),
|
|
"null": reader.ReadBytes(4)
|
|
# more later
|
|
}
|
|
|
|
print(f"> Loading mapping for {infos['game']} v{infos['version'][0]}.{infos['version'][1]}, this may take a few seconds...")
|
|
|
|
# read prefixes
|
|
prefixes = {}
|
|
n_prefixes = reader.ReadUInt8()
|
|
|
|
for i in range(n_prefixes):
|
|
prefix = raw(4)
|
|
marker = reader.ReadBytes(1)
|
|
prefixes[marker] = prefix
|
|
|
|
# read languages
|
|
langs_offsets = {}
|
|
n_langs = reader.ReadUInt8()
|
|
|
|
for i in range(n_langs):
|
|
offset = reader.GetBufferPos()
|
|
langs_offsets[offset] = raw(11)
|
|
|
|
# read folders
|
|
folder_offsets = {}
|
|
n_folders = reader.ReadUInt8()
|
|
|
|
for i in range(n_folders):
|
|
offset = reader.GetBufferPos()
|
|
length = reader.ReadUInt8()
|
|
prefix = reader.ReadBytes(1)
|
|
folder = raw(length)
|
|
folder = f"{prefixes[prefix]}{folder}"
|
|
folder_offsets[offset] = folder
|
|
|
|
# read files
|
|
files_offsets = {}
|
|
n_files = val(3)
|
|
|
|
for i in range(n_files):
|
|
offset = reader.GetBufferPos()
|
|
path_length = reader.ReadUInt8()
|
|
path = []
|
|
for i in range(path_length):
|
|
path.append(folder_offsets[reader.ReadUInt16()])
|
|
|
|
name_length = reader.ReadUInt8()
|
|
prefix = prefixes[reader.ReadBytes(1)]
|
|
name = raw(name_length)
|
|
|
|
name = f"{prefix}{name}"
|
|
path.append(name)
|
|
path = "\\".join(path)
|
|
|
|
files_offsets[offset] = path
|
|
|
|
# read keys
|
|
keys_data = {}
|
|
n_keys = val(3)
|
|
|
|
for i in range(n_keys):
|
|
key = raw(16)
|
|
|
|
lang_offset = reader.ReadUInt8()
|
|
file_offset = val(3)
|
|
|
|
keys_data[key] = [files_offsets[file_offset], langs_offsets[lang_offset]]
|
|
|
|
self.keys_data = keys_data
|
|
|
|
# done
|
|
print(f"> Finished loading mapping")
|
|
print(f": {n_langs} supported languages")
|
|
print(f": {n_files} mapped files")
|
|
print(f": {n_keys} available keys")
|
|
|
|
|
|
def get_key(self, key, lang=False):
|
|
keys_data = self.keys_data
|
|
if key not in keys_data.keys():
|
|
return None
|
|
|
|
key_data = keys_data[key]
|
|
data = [key_data[0]]
|
|
|
|
if lang:
|
|
data.append(key_data[1])
|
|
|
|
return data
|