1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
import struct
from splat.util import options, log
from splat.segtypes.n64.palette import N64SegPalette
class N64SegAf_palette(N64SegPalette):
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.data_only = isinstance(yaml, dict) and yaml.get("data_only", False)
def scan(self, rom_bytes: bytes):
data = rom_bytes[self.rom_start : self.rom_end]
lines = []
if not self.data_only:
lines.append(options.opts.generated_c_preamble)
lines.append("")
lines.append(f"u16 {self.name}[] = {{")
for short in struct.iter_unpack(">H", data):
lines.append(f" 0x{short[0]:04X},")
if not self.data_only:
lines.append("};")
lines.append("")
self.file_text = "\n".join(lines)
def split(self, rom_bytes: bytes):
path = options.opts.asset_path / self.dir / f"{self.name}.palette.inc.c"
if self.file_text and path:
path.parent.mkdir(parents=True, exist_ok=True)
with open(path, "w", newline="\n") as f:
f.write(self.file_text)
|
