mirror of
https://github.com/pmret/papermario.git
synced 2024-11-18 08:52:40 +01:00
ae66312d8c
* Add Python linter to github actions * wip * Add back splat_ext * Format files * C++ -> C * format 2 files * split workflow into separate file, line length 120, fix excludes * -l 120 in ci * update black locally and apply formatting changes * pyproject.toject --------- Co-authored-by: Ethan Roseman <ethteck@gmail.com>
49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
import argparse
|
|
from pathlib import Path
|
|
import sys
|
|
|
|
sys.path.append(str(Path(__file__).parent.parent))
|
|
sys.path.append(str(Path(__file__).parent.parent / "splat"))
|
|
|
|
from splat_ext.pm_effect_loads import effects_from_yaml
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Builds effect table, function declarations, macros, and enum")
|
|
parser.add_argument("in_yaml")
|
|
parser.add_argument("out_dir", type=Path)
|
|
args = parser.parse_args()
|
|
|
|
effect_enum_text = "enum EffectID {\n"
|
|
effect_table_text = "EffectTableEntry gEffectTable[] = {\n"
|
|
fx_decls_text = '#include "effects/effect_macros.h"\n\n'
|
|
main_decls_text = '#include "effects/effect_macros.h"\n\n'
|
|
macro_defs = ""
|
|
|
|
effects = effects_from_yaml(args.in_yaml)
|
|
|
|
for i, effect in enumerate(effects):
|
|
enum_name = effect.name.upper()
|
|
if not enum_name.startswith("EFFECT_"):
|
|
enum_name = "EFFECT_" + enum_name
|
|
|
|
effect_enum_text += f" {enum_name} = 0x{i:02X},\n"
|
|
if not effect.empty:
|
|
effect_table_text += f" FX_ENTRY({effect.name}, effect_gfx_{effect.gfx}),\n"
|
|
fx_decls_text += effect.get_macro_call("fx_" + effect.name) + ";\n"
|
|
main_decls_text += effect.get_macro_call(effect.name + "_main") + ";\n"
|
|
macro_defs += effect.get_macro_def() + "\n"
|
|
else:
|
|
effect_table_text += " {},\n"
|
|
|
|
args.out_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
with open(args.out_dir / "effect_macros.h", "w") as f:
|
|
f.write(macro_defs)
|
|
|
|
with open(args.out_dir / "effect_table.c", "w") as f:
|
|
f.write(main_decls_text + "\n" + effect_table_text + "};\n")
|
|
|
|
with open(args.out_dir / "effect_defs.h", "w") as f:
|
|
f.write(effect_enum_text + "};\n\n" + fx_decls_text)
|