ROBChain/inject.py

38 lines
987 B
Python
Raw Normal View History

2020-02-11 23:00:37 +01:00
import os, zlib, argparse
PACK_FILE_PATH = "robot_packed"
OUTPUT_FOLDER = "data/fighter/robot/script/"
OUTPUT_FILE = "packed"
OUTPUT_PATH = OUTPUT_FOLDER + OUTPUT_FILE
PACK_OFFSET = 0xB380
COMPRESSED_SIZE = 0x1B1D0
UNCOMPRESSED_SIZE = 0x5B2E0
if not os.path.exists(OUTPUT_FOLDER):
os.makedirs(OUTPUT_FOLDER)
parser = argparse.ArgumentParser(description="pack ROB's mscsb files")
parser.add_argument('in_file')
in_file = parser.parse_args().in_file
with open(in_file, 'rb') as f:
data = f.read()
assert len(data) <= UNCOMPRESSED_SIZE
data += b'\0' * (UNCOMPRESSED_SIZE - len(data))
compressed_data = zlib.compress(data)
assert len(compressed_data) <= COMPRESSED_SIZE
compressed_data += b'\0' * (COMPRESSED_SIZE - len(compressed_data))
with open(PACK_FILE_PATH, 'rb') as f:
pack_file_data = f.read()
with open(OUTPUT_PATH, 'wb') as f:
f.write(pack_file_data[:PACK_OFFSET])
f.write(compressed_data)
f.write(pack_file_data[PACK_OFFSET + COMPRESSED_SIZE:])