mirror of
https://github.com/pmret/papermario.git
synced 2024-11-10 04:52:34 +01:00
6c606383e8
* nok_04 ok * thread done * push block doc * nok done * almost done pra * pra done * cleanup pra_31 mtx names * git subrepo pull --force tools/splat subrepo: subdir: "tools/splat" merged: "3bbc02af68" upstream: origin: "https://github.com/ethteck/splat.git" branch: "master" commit: "3bbc02af68" git-subrepo: version: "0.4.5" origin: "???" commit: "???" --------- Co-authored-by: HailSanta <Hail2Santa@gmail.com>
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
from typing import Dict
|
|
|
|
from segtypes.common.group import CommonSegGroup
|
|
from segtypes.n64.ci import N64SegCi
|
|
from segtypes.n64.palette import N64SegPalette as Palette
|
|
|
|
|
|
# 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)
|