papermario/tools/mapheaders.py

77 lines
2.7 KiB
Python
Raw Normal View History

2020-10-23 14:31:56 +02:00
#!/usr/bin/python3
import argparse
from collections import OrderedDict
import os
import re
import pickle
import sys
from pathlib import Path
script_dir = os.path.dirname(os.path.realpath(__file__))
root_dir = script_dir + "/../"
src_dir = root_dir + "src/world/"
asm_dir = root_dir + "ver/current/asm/nonmatchings/world/"
2020-10-23 14:31:56 +02:00
2021-04-03 19:21:49 +02:00
if len(sys.argv) <= 1 or not (sys.argv[1] == "y" or sys.argv[1] == "-y"):
print(f"Warning! This script is destructive, pass -y if you really want to run it")
exit()
2020-10-24 03:09:49 +02:00
for root, dirs, files in os.walk(src_dir):
for dir_name in dirs:
dir_path = os.path.join(root, dir_name)
map_name = Path(dir_path).parent.name
if map_name.startswith("area_"):
common_funcs = []
2020-10-23 14:31:56 +02:00
2020-10-24 03:09:49 +02:00
for c_file in os.scandir(dir_path):
if c_file.name.endswith(".c"):
with open(c_file) as f:
c_lines = f.readlines()
for line in c_lines:
if "world/common" in line:
common_funcs.append(line.split("/")[-1].split(".")[0])
2020-10-23 14:31:56 +02:00
2020-10-24 03:09:49 +02:00
header_name = dir_name + ".h"
guard_name = f"_{dir_name.upper()}_"
2020-10-23 14:31:56 +02:00
2020-10-24 03:09:49 +02:00
out_lines = []
# out_lines.append(f"#ifndef {guard_name}")
# out_lines.append(f"#define {guard_name}")
# out_lines.append("")
out_lines.append("#include \"common.h\"")
out_lines.append("#include \"map.h\"")
out_lines.append("")
2020-12-28 08:09:39 +01:00
out_lines.append(f"#define NAMESPACE {dir_name}")
2020-10-24 03:09:49 +02:00
out_lines.append("")
2020-10-23 14:31:56 +02:00
2020-10-24 03:09:49 +02:00
# if len(common_funcs) > 0:
# for func_name in common_funcs:
# out_lines.append(f"#include \"world/common/{func_name}.h\"")
# out_lines.append("#endif")
# out_lines.append("")
2020-10-23 14:31:56 +02:00
2020-10-24 03:09:49 +02:00
with open(os.path.join(dir_path, header_name), "w", newline="\n") as f:
f.write("\n".join(out_lines))
2020-10-23 14:31:56 +02:00
2020-10-24 03:09:49 +02:00
# for root, dirs, files in os.walk(src_dir):
# for f_name in files:
# if f_name.endswith(".c"):
# f_path = os.path.join(root, f_name)
# area_name = Path(f_path).parent.name
# with open(f_path) as f:
# f_text_orig = f.readlines()
2020-10-24 03:09:49 +02:00
# f_text = []
# f_text.append(f"#include \"{area_name}.h\"\n")
2020-10-23 14:31:56 +02:00
2020-10-24 03:09:49 +02:00
# for line in f_text_orig:
# if not line.startswith("#include") or line.startswith("#include \"world"):
# f_text.append(line)
# #new_line = re.sub(r"INCLUDE_ASM.*\);", sub_func, line)
2020-10-23 14:31:56 +02:00
2020-10-24 03:09:49 +02:00
# if f_text != f_text_orig:
# with open(f_path, "w", newline="\n") as f:
# f.writelines(f_text)