mirror of https://github.com/pmret/papermario.git
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from typing import Dict, Union
|
|
from segtypes.n64.ci import N64SegCi
|
|
from segtypes.n64.palette import N64SegPalette as Palette
|
|
from segtypes.common.group import CommonSegGroup
|
|
|
|
# Resolve Raster#palette and Palette#raster links
|
|
def initialize(all_segments):
|
|
def process(segments):
|
|
raster_map: Dict[str, N64SegCi] = {}
|
|
palette_map: Dict[str, Palette] = {}
|
|
|
|
for segment in segments:
|
|
if isinstance(segment, Palette):
|
|
palette_map[segment.name] = segment
|
|
|
|
if isinstance(segment, N64SegCi):
|
|
raster_map[segment.name] = segment
|
|
|
|
if isinstance(segment, CommonSegGroup):
|
|
process(segment.subsegments)
|
|
|
|
for raster_name in raster_map:
|
|
raster = raster_map[raster_name]
|
|
# print(f"{raster_name} -> {raster.palette_name}")
|
|
raster.palette = palette_map.get(raster.palette_name)
|
|
|
|
for palette_name in palette_map:
|
|
palette = palette_map[palette_name]
|
|
# print(f"{palette_name} -> {palette.raster_name}")
|
|
palette.raster = raster_map.get(palette.raster_name)
|
|
|
|
process(all_segments)
|