pokeplatinum/tools/scripts/make_pl_growtbl.py
2024-02-14 12:05:11 +00:00

52 lines
1.7 KiB
Python

#!/usr/bin/env python3
import argparse
import csv
import pathlib
import subprocess
argparser = argparse.ArgumentParser(
prog='make_pl_growtbl',
description='Packs the archive containing pokemon exp tables'
)
argparser.add_argument('-k', '--knarc',
required=True,
help='Path to knarc executable')
argparser.add_argument('-s', '--source-dir',
required=True,
help='Path to the source directory (res/pokemon)')
argparser.add_argument('-p', '--private-dir',
required=True,
help='Path to the private directory (where binaries will be made)')
argparser.add_argument('-o', '--output-dir',
required=True,
help='Path to the output directory (where the NARC will be made)')
argparser.add_argument('table',
help='Experience tables in csv format')
args = argparser.parse_args()
source_dir = pathlib.Path(args.source_dir)
private_dir = pathlib.Path(args.private_dir)
output_dir = pathlib.Path(args.output_dir)
private_dir.mkdir(parents=True, exist_ok=True)
with open(args.table) as data_file:
file_reader = csv.reader(data_file)
table_data = list(file_reader)
transpose = list(zip(*table_data))
count = 0
for table in transpose:
if table[0] == 'level':
continue
out = bytes()
for n in table[1:]:
out = out + int(n).to_bytes(4, byteorder='little')
target_fname = private_dir / f'pl_growtbl_{count}.bin'
with open(target_fname, 'wb+') as target_file:
target_file.write(out)
count += 1
subprocess.run([args.knarc, '-d', private_dir, '-p', output_dir / 'pl_growtbl.narc'])