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
|
from n64img.image import Image, CI4
from splat.util import log, options
from splat.segtypes.n64.segment import N64Segment
class N64SegAf_ci4(N64Segment):
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.width = args[0]
self.height = args[1]
self.palette_start = args[2]
self.palette_size = args[3]
def scan(self, rom_bytes: bytes):
self.n64img: Image = CI4(b"", 0, 0)
self.n64img.width = self.width
self.n64img.height = self.height
self.n64img.data = rom_bytes[self.rom_start : self.rom_end]
self.n64img.set_palette(rom_bytes[self.palette_start : self.palette_start + self.palette_size])
def split(self, rom_bytes: bytes):
path = options.opts.asset_path / self.dir / f"{self.name}.ci4.png"
path.parent.mkdir(parents=True, exist_ok=True)
self.n64img.write(path)
|
