papermario/tools/splat_ext/pm_effect_shims.py
Ethan Roseman 262428c68a
Splat update (#1136)
* Remove splat from repo

* refactor scraps

* requirements update
2024-01-03 02:16:18 +09:00

73 lines
1.9 KiB
Python

from typing import List
from yaml.loader import Loader
from splat.segtypes.n64.segment import N64Segment
from splat.util import options
import yaml as yaml_loader
class N64SegPm_effect_shims(N64Segment):
shims: List[str] = []
@staticmethod
def get_shim_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 {name}
/* 0 3C01E020 */ lui $at, %hi(effectFuncs + 0x{index * 4:X})
/* 4 8C210694 */ lw $at, %lo(effectFuncs + 0x{index * 4:X})($at)
/* 8 00200008 */ jr $at
/* C 00000000 */ nop
"""
def shim_path(self, shim: str):
return options.opts.build_path / "asm" / "effect_shims" / f"{shim}.s"
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,
)
with open(options.opts.src_path / "effect_shims.yaml") as f:
self.shims = yaml_loader.load(f.read(), Loader=yaml_loader.SafeLoader)
def split(self, rom_bytes):
for i, shim in enumerate(self.shims):
shim_asm = N64SegPm_effect_shims.get_shim_asm(i, shim)
self.shim_path("").parent.mkdir(parents=True, exist_ok=True)
with open(self.shim_path(shim), "w") as f:
f.write(shim_asm)
def get_linker_entries(self):
from splat.segtypes.linker_entry import LinkerEntry
ret = []
for shim in self.shims:
ret.append(LinkerEntry(self, [self.shim_path(shim)], self.shim_path(shim), ".text", ".text"))
return ret