mirror of
https://github.com/pmret/papermario.git
synced 2024-11-08 12:02:30 +01:00
288bb8511f
* kooper cleanup * second pass --------- Co-authored-by: HailSanta <Hail2Santa@gmail.com>
88 lines
2.5 KiB
Python
88 lines
2.5 KiB
Python
from segtypes.n64.segment import N64Segment
|
|
from util import options
|
|
import png
|
|
|
|
def parse_ci4(data, width, height):
|
|
raster = bytearray()
|
|
|
|
for i in range(width * height // 2):
|
|
raster.append(data[i] >> 4)
|
|
raster.append(data[i] & 0xF)
|
|
|
|
return raster
|
|
|
|
def next_8(n):
|
|
if n % 8 != 0:
|
|
return next_8(n + 1)
|
|
else:
|
|
return n
|
|
|
|
def get_palette_idx(charset_name, char_id):
|
|
pal_id = 0
|
|
if charset_name == "standard":
|
|
if char_id == 0x98:
|
|
pal_id = 0x10
|
|
elif char_id == 0x99:
|
|
pal_id = 0x11
|
|
elif char_id == 0x9A:
|
|
pal_id = 0x15
|
|
elif char_id == 0x9B:
|
|
pal_id = 0x15
|
|
elif char_id == 0x9C:
|
|
pal_id = 0x15
|
|
elif char_id == 0x9D:
|
|
pal_id = 0x13
|
|
elif char_id == 0x9E:
|
|
pal_id = 0x13
|
|
elif char_id == 0x9F:
|
|
pal_id = 0x13
|
|
elif char_id == 0xA0:
|
|
pal_id = 0x13
|
|
elif char_id == 0xA1:
|
|
pal_id = 0x12
|
|
return pal_id
|
|
|
|
class N64SegPm_charset(N64Segment):
|
|
def scan(self, rom_bytes):
|
|
data = rom_bytes[self.rom_start:self.rom_end]
|
|
|
|
# start, type, name, WIDTH, HEIGHT, NUM_RASTERS
|
|
self.width = self.yaml[3]
|
|
self.height = self.yaml[4]
|
|
|
|
# pm_charset_palettes sibling
|
|
self.sibling = next(filter(lambda s: s.type == "pm_charset_palettes" and s.name == self.name, self.parent.subsegments))
|
|
|
|
self.rasters = []
|
|
|
|
for i in range(0, self.size, next_8(self.width * self.height // 2)):
|
|
raster = parse_ci4(data[i:], self.width, self.height)
|
|
self.rasters.append(raster)
|
|
|
|
def split(self, rom_bytes):
|
|
fs_dir = options.opts.asset_path / self.dir / self.name
|
|
fs_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
for i, raster in enumerate(self.rasters):
|
|
pal_idx = get_palette_idx(self.name, i)
|
|
palette = self.sibling.palettes[pal_idx]
|
|
|
|
w = png.Writer(self.width, self.height, palette=palette)
|
|
with open(fs_dir / f"{i:02X}.png", "wb") as f:
|
|
w.write_array(f, raster)
|
|
|
|
def get_linker_entries(self):
|
|
from segtypes.linker_entry import LinkerEntry
|
|
|
|
# start, type, name, WIDTH, HEIGHT
|
|
self.width = self.yaml[3]
|
|
self.height = self.yaml[4]
|
|
|
|
fs_dir = options.opts.asset_path / self.dir / self.name
|
|
|
|
return [LinkerEntry(
|
|
self,
|
|
[fs_dir / f"{i:02X}.png" for i in range(self.yaml[5])],
|
|
fs_dir.with_suffix(".dat"), ".data"),
|
|
]
|