mirror of
https://github.com/pmret/papermario.git
synced 2024-11-08 12:02:30 +01:00
142 lines
4.1 KiB
Python
142 lines
4.1 KiB
Python
from dataclasses import dataclass
|
|
from pathlib import Path
|
|
from typing import List
|
|
from splat.segtypes.n64.segment import N64Segment
|
|
from splat.util import options
|
|
import yaml as yaml_loader
|
|
|
|
|
|
@dataclass
|
|
class Effect:
|
|
name: str
|
|
args: str
|
|
gfx: str
|
|
empty: bool
|
|
returns_void: bool
|
|
|
|
def get_macro_def(self) -> str:
|
|
ret_type = "void" if self.returns_void else "EffectInstance*"
|
|
|
|
return f"#define EFFECT_DEF_{self.name.upper()}(func_name) {ret_type} func_name({self.args})"
|
|
|
|
def get_macro_call(self, func_name: str) -> str:
|
|
return f"EFFECT_DEF_{self.name.upper()}({func_name})"
|
|
|
|
|
|
def effects_from_yaml(yaml_path: Path) -> List[Effect]:
|
|
with open(yaml_path) as f:
|
|
effects_yaml = yaml_loader.load(f.read(), Loader=yaml_loader.SafeLoader)
|
|
|
|
effects: List[Effect] = []
|
|
for effect_yaml in effects_yaml:
|
|
name = str(effect_yaml.get("name", f"{len(effects):02X}"))
|
|
effects.append(
|
|
Effect(
|
|
name=name,
|
|
args=effect_yaml.get("args", ""),
|
|
gfx=effect_yaml.get("gfx", name),
|
|
empty="name" not in effect_yaml,
|
|
returns_void=effect_yaml.get("void", False),
|
|
)
|
|
)
|
|
|
|
return effects
|
|
|
|
|
|
class N64SegPm_effect_loads(N64Segment):
|
|
effects: List[Effect] = []
|
|
|
|
@staticmethod
|
|
def get_effect_asm(index, name):
|
|
return f""".include "macro.inc"
|
|
|
|
# assembler directives
|
|
.set noat # allow manual use of $at
|
|
.set noreorder # don't insert nops after branches
|
|
.set gp=64 # allow use of 64-bit general purpose registers
|
|
|
|
.section .text, "ax"
|
|
|
|
glabel fx_{name}
|
|
/* 00 27BDFFD0 */ addiu $sp, $sp, -0x30
|
|
/* 04 AFA40010 */ sw $a0, 0x10($sp)
|
|
/* 08 AFA50014 */ sw $a1, 0x14($sp)
|
|
/* 0C AFA60018 */ sw $a2, 0x18($sp)
|
|
/* 10 AFA7001C */ sw $a3, 0x1c($sp)
|
|
/* 14 E7AC0020 */ swc1 $f12, 0x20($sp)
|
|
/* 18 E7AE0024 */ swc1 $f14, 0x24($sp)
|
|
/* 1C E7B00028 */ swc1 $f16, 0x28($sp)
|
|
/* 20 AFBF002C */ sw $ra, 0x2c($sp)
|
|
/* 24 0C016959 */ jal load_effect
|
|
/* 28 ???????? */ addiu $a0, $zero, 0x{index:X}
|
|
/* 2C 8FA40010 */ lw $a0, 0x10($sp)
|
|
/* 30 8FA50014 */ lw $a1, 0x14($sp)
|
|
/* 34 8FA60018 */ lw $a2, 0x18($sp)
|
|
/* 38 8FA7001C */ lw $a3, 0x1c($sp)
|
|
/* 3C C7AC0020 */ lwc1 $f12, 0x20($sp)
|
|
/* 40 C7AE0024 */ lwc1 $f14, 0x24($sp)
|
|
/* 44 C7B00028 */ lwc1 $f16, 0x28($sp)
|
|
/* 48 8FBF002C */ lw $ra, 0x2c($sp)
|
|
/* 4C 27BD0030 */ addiu $sp, $sp, 0x30
|
|
/* 50 3C01???? */ lui $at, %hi(gEffectTable + 0x{index * 24:X})
|
|
/* 54 8C21???? */ lw $at, %lo(gEffectTable + 0x{index * 24:X})($at)
|
|
/* 58 00200008 */ jr $at
|
|
/* 5C 00000000 */ nop
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
rom_start,
|
|
rom_end,
|
|
type,
|
|
name,
|
|
vram_start,
|
|
args,
|
|
yaml,
|
|
):
|
|
super().__init__(
|
|
rom_start,
|
|
rom_end,
|
|
type,
|
|
name,
|
|
vram_start,
|
|
args=args,
|
|
yaml=yaml,
|
|
)
|
|
|
|
self.effects = effects_from_yaml(options.opts.src_path / "effects.yaml")
|
|
|
|
def effect_s_path(self, effect_name: str):
|
|
return options.opts.build_path / "asm" / "effects" / f"{effect_name}.s"
|
|
|
|
def split(self, rom_bytes):
|
|
for i, effect in enumerate(self.effects):
|
|
# .s file for effect
|
|
effect_asm = N64SegPm_effect_loads.get_effect_asm(i, effect.name)
|
|
|
|
self.effect_s_path("").parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
with open(self.effect_s_path(effect.name), "w") as f:
|
|
f.write(effect_asm)
|
|
|
|
def get_linker_entries(self):
|
|
from splat.segtypes.linker_entry import LinkerEntry
|
|
|
|
ret = []
|
|
|
|
for effect in self.effects:
|
|
ret.append(
|
|
LinkerEntry(
|
|
self,
|
|
[self.effect_s_path(effect.name)],
|
|
self.effect_s_path(effect.name),
|
|
".text",
|
|
".text",
|
|
)
|
|
)
|
|
|
|
return ret
|
|
|
|
def should_split(self) -> bool:
|
|
return options.opts.is_mode_active(self.type)
|