Adds Link's Awakening: DX. Fully imports and forks LADXR, with permission - https://github.com/daid/LADXR
		
			
				
	
	
		
			254 lines
		
	
	
		
			9.5 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			254 lines
		
	
	
		
			9.5 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
from typing import Dict, Set
 | 
						|
from ..roomEditor import RoomEditor
 | 
						|
 | 
						|
 | 
						|
animated_tiles = {0x0E, 0x1B, 0x1E, 0x1F, 0x44, 0x91, 0xCF, 0xD0, 0xD1, 0xD2, 0xD9, 0xDC, 0xE9, 0xEB, 0xEC, 0xED, 0xEE, 0xEF}
 | 
						|
entrance_tiles = {0xE1, 0xE2, 0xE3, 0xBA, 0xC6}
 | 
						|
 | 
						|
solid_tiles = set()
 | 
						|
open_tiles = set()
 | 
						|
walkable_tiles = set()
 | 
						|
vertical_edge_tiles = set()
 | 
						|
horizontal_edge_tiles = set()
 | 
						|
 | 
						|
 | 
						|
class TileInfo:
 | 
						|
    def __init__(self, key):
 | 
						|
        self.key = key
 | 
						|
        self.up = set()
 | 
						|
        self.right = set()
 | 
						|
        self.down = set()
 | 
						|
        self.left = set()
 | 
						|
        self.up_freq = {}
 | 
						|
        self.right_freq = {}
 | 
						|
        self.down_freq = {}
 | 
						|
        self.left_freq = {}
 | 
						|
        self.frequency = 0
 | 
						|
 | 
						|
    def copy(self):
 | 
						|
        result = TileInfo(self.key)
 | 
						|
        result.up = self.up.copy()
 | 
						|
        result.right = self.right.copy()
 | 
						|
        result.down = self.down.copy()
 | 
						|
        result.left = self.left.copy()
 | 
						|
        result.up_freq = self.up_freq.copy()
 | 
						|
        result.right_freq = self.right_freq.copy()
 | 
						|
        result.down_freq = self.down_freq.copy()
 | 
						|
        result.left_freq = self.left_freq.copy()
 | 
						|
        result.frequency = self.frequency
 | 
						|
        return result
 | 
						|
 | 
						|
    def remove(self, tile_id):
 | 
						|
        if tile_id in self.up:
 | 
						|
            self.up.remove(tile_id)
 | 
						|
            del self.up_freq[tile_id]
 | 
						|
        if tile_id in self.down:
 | 
						|
            self.down.remove(tile_id)
 | 
						|
            del self.down_freq[tile_id]
 | 
						|
        if tile_id in self.left:
 | 
						|
            self.left.remove(tile_id)
 | 
						|
            del self.left_freq[tile_id]
 | 
						|
        if tile_id in self.right:
 | 
						|
            self.right.remove(tile_id)
 | 
						|
            del self.right_freq[tile_id]
 | 
						|
 | 
						|
    def update(self, other: "TileInfo", tile_filter: Set[int]):
 | 
						|
        self.frequency += other.frequency
 | 
						|
        self.up.update(other.up.intersection(tile_filter))
 | 
						|
        self.down.update(other.down.intersection(tile_filter))
 | 
						|
        self.left.update(other.left.intersection(tile_filter))
 | 
						|
        self.right.update(other.right.intersection(tile_filter))
 | 
						|
        for k, v in other.up_freq.items():
 | 
						|
            if k not in tile_filter:
 | 
						|
                continue
 | 
						|
            self.up_freq[k] = self.up_freq.get(k, 0) + v
 | 
						|
        for k, v in other.down_freq.items():
 | 
						|
            if k not in tile_filter:
 | 
						|
                continue
 | 
						|
            self.down_freq[k] = self.down_freq.get(k, 0) + v
 | 
						|
        for k, v in other.left_freq.items():
 | 
						|
            if k not in tile_filter:
 | 
						|
                continue
 | 
						|
            self.left_freq[k] = self.left_freq.get(k, 0) + v
 | 
						|
        for k, v in other.down_freq.items():
 | 
						|
            if k not in tile_filter:
 | 
						|
                continue
 | 
						|
            self.right_freq[k] = self.right_freq.get(k, 0) + v
 | 
						|
 | 
						|
    def __repr__(self):
 | 
						|
        return f"<{self.key}>\n U{[f'{n:02x}' for n in self.up]}\n R{[f'{n:02x}' for n in self.right]}\n D{[f'{n:02x}' for n in self.down]}\n L{[f'{n:02x}' for n in self.left]}>"
 | 
						|
 | 
						|
 | 
						|
class TileSet:
 | 
						|
    def __init__(self, *, main_id=None, animation_id=None):
 | 
						|
        self.main_id = main_id
 | 
						|
        self.animation_id = animation_id
 | 
						|
        self.palette_id = None
 | 
						|
        self.attr_bank = None
 | 
						|
        self.attr_addr = None
 | 
						|
        self.tiles: Dict[int, "TileInfo"] = {}
 | 
						|
        self.all: Set[int] = set()
 | 
						|
 | 
						|
    def copy(self) -> "TileSet":
 | 
						|
        result = TileSet(main_id=self.main_id, animation_id=self.animation_id)
 | 
						|
        for k, v in self.tiles.items():
 | 
						|
            result.tiles[k] = v.copy()
 | 
						|
        result.all = self.all.copy()
 | 
						|
        return result
 | 
						|
 | 
						|
    def remove(self, tile_id):
 | 
						|
        self.all.remove(tile_id)
 | 
						|
        del self.tiles[tile_id]
 | 
						|
        for k, v in self.tiles.items():
 | 
						|
            v.remove(tile_id)
 | 
						|
 | 
						|
    # Look at the "other" tileset and merge information about tiles known in this tileset
 | 
						|
    def learn_from(self, other: "TileSet"):
 | 
						|
        for key, other_info in other.tiles.items():
 | 
						|
            if key not in self.all:
 | 
						|
                continue
 | 
						|
            self.tiles[key].update(other_info, self.all)
 | 
						|
 | 
						|
    def combine(self, other: "TileSet"):
 | 
						|
        if other.main_id and not self.main_id:
 | 
						|
            self.main_id = other.main_id
 | 
						|
        if other.animation_id and not self.animation_id:
 | 
						|
            self.animation_id = other.animation_id
 | 
						|
        for key, other_info in other.tiles.items():
 | 
						|
            if key not in self.all:
 | 
						|
                self.tiles[key] = other_info.copy()
 | 
						|
            else:
 | 
						|
                self.tiles[key].update(other_info, self.all)
 | 
						|
        self.all.update(other.all)
 | 
						|
 | 
						|
 | 
						|
def loadTileInfo(rom) -> Dict[str, TileSet]:
 | 
						|
    for n in range(0x100):
 | 
						|
        physics_flag = rom.banks[8][0x0AD4 + n]
 | 
						|
        if n == 0xEF:
 | 
						|
            physics_flag = 0x01  # One of the sky tiles is marked as a pit instead of solid, which messes with the generation of sky
 | 
						|
        if physics_flag in {0x00, 0x05, 0x06, 0x07}:
 | 
						|
            open_tiles.add(n)
 | 
						|
            walkable_tiles.add(n)
 | 
						|
            vertical_edge_tiles.add(n)
 | 
						|
            horizontal_edge_tiles.add(n)
 | 
						|
        elif physics_flag in {0x01, 0x04, 0x60}:
 | 
						|
            solid_tiles.add(n)
 | 
						|
            vertical_edge_tiles.add(n)
 | 
						|
            horizontal_edge_tiles.add(n)
 | 
						|
        elif physics_flag in {0x08}:  # Bridge
 | 
						|
            open_tiles.add(n)
 | 
						|
            walkable_tiles.add(n)
 | 
						|
        elif physics_flag in {0x02}:  # Stairs
 | 
						|
            open_tiles.add(n)
 | 
						|
            walkable_tiles.add(n)
 | 
						|
            horizontal_edge_tiles.add(n)
 | 
						|
        elif physics_flag in {0x03}:  # Entrances
 | 
						|
            open_tiles.add(n)
 | 
						|
        elif physics_flag in {0x30}:  # bushes/rocks
 | 
						|
            open_tiles.add(n)
 | 
						|
        elif physics_flag in {0x50}:  # pits
 | 
						|
            open_tiles.add(n)
 | 
						|
    world_tiles = {}
 | 
						|
    for ry in range(0, 16):
 | 
						|
        for rx in range(0, 16):
 | 
						|
            tileset_id = rom.banks[0x3F][0x3F00 + rx + (ry << 4)]
 | 
						|
            re = RoomEditor(rom, rx | (ry << 4))
 | 
						|
            tiles = re.getTileArray()
 | 
						|
            for y in range(8):
 | 
						|
                for x in range(10):
 | 
						|
                    tile_id = tiles[x+y*10]
 | 
						|
                    world_tiles[(rx*10+x, ry*8+y)] = (tile_id, tileset_id, re.animation_id | 0x100)
 | 
						|
 | 
						|
    # Fix up wrong tiles
 | 
						|
    world_tiles[(150, 24)] = (0x2A, world_tiles[(150, 24)][1], world_tiles[(150, 24)][2])  # Left of the raft house, a tree has the wrong tile.
 | 
						|
 | 
						|
    rom_tilesets: Dict[int, TileSet] = {}
 | 
						|
    for (x, y), (key, tileset_id, animation_id) in world_tiles.items():
 | 
						|
        if key in animated_tiles:
 | 
						|
            if animation_id not in rom_tilesets:
 | 
						|
                rom_tilesets[animation_id] = TileSet(animation_id=animation_id&0xFF)
 | 
						|
            tileset = rom_tilesets[animation_id]
 | 
						|
        else:
 | 
						|
            if tileset_id not in rom_tilesets:
 | 
						|
                rom_tilesets[tileset_id] = TileSet(main_id=tileset_id)
 | 
						|
            tileset = rom_tilesets[tileset_id]
 | 
						|
        tileset.all.add(key)
 | 
						|
        if key not in tileset.tiles:
 | 
						|
            tileset.tiles[key] = TileInfo(key)
 | 
						|
        ti = tileset.tiles[key]
 | 
						|
        ti.frequency += 1
 | 
						|
        if (x, y - 1) in world_tiles:
 | 
						|
            tile_id = world_tiles[(x, y - 1)][0]
 | 
						|
            ti.up.add(tile_id)
 | 
						|
            ti.up_freq[tile_id] = ti.up_freq.get(tile_id, 0) + 1
 | 
						|
        if (x + 1, y) in world_tiles:
 | 
						|
            tile_id = world_tiles[(x + 1, y)][0]
 | 
						|
            ti.right.add(tile_id)
 | 
						|
            ti.right_freq[tile_id] = ti.right_freq.get(tile_id, 0) + 1
 | 
						|
        if (x, y + 1) in world_tiles:
 | 
						|
            tile_id = world_tiles[(x, y + 1)][0]
 | 
						|
            ti.down.add(tile_id)
 | 
						|
            ti.down_freq[tile_id] = ti.down_freq.get(tile_id, 0) + 1
 | 
						|
        if (x - 1, y) in world_tiles:
 | 
						|
            tile_id = world_tiles[(x - 1, y)][0]
 | 
						|
            ti.left.add(tile_id)
 | 
						|
            ti.left_freq[tile_id] = ti.left_freq.get(tile_id, 0) + 1
 | 
						|
 | 
						|
    tilesets = {
 | 
						|
        "basic": rom_tilesets[0x0F].copy()
 | 
						|
    }
 | 
						|
    for key, tileset in rom_tilesets.items():
 | 
						|
        tilesets["basic"].learn_from(tileset)
 | 
						|
    tilesets["mountains"] = rom_tilesets[0x3E].copy()
 | 
						|
    tilesets["mountains"].combine(rom_tilesets[0x10B])
 | 
						|
    tilesets["mountains"].remove(0xB6)  # Remove the raft house roof
 | 
						|
    tilesets["mountains"].remove(0xB7)  # Remove the raft house roof
 | 
						|
    tilesets["mountains"].remove(0x66)  # Remove the raft house roof
 | 
						|
    tilesets["mountains"].learn_from(rom_tilesets[0x1C])
 | 
						|
    tilesets["mountains"].learn_from(rom_tilesets[0x3C])
 | 
						|
    tilesets["mountains"].learn_from(rom_tilesets[0x30])
 | 
						|
    tilesets["mountains"].palette_id = 0x15
 | 
						|
    tilesets["mountains"].attr_bank = 0x27
 | 
						|
    tilesets["mountains"].attr_addr = 0x5A20
 | 
						|
 | 
						|
    tilesets["egg"] = rom_tilesets[0x3C].copy()
 | 
						|
    tilesets["egg"].combine(tilesets["mountains"])
 | 
						|
    tilesets["egg"].palette_id = 0x13
 | 
						|
    tilesets["egg"].attr_bank = 0x27
 | 
						|
    tilesets["egg"].attr_addr = 0x5620
 | 
						|
 | 
						|
    tilesets["forest"] = rom_tilesets[0x20].copy()
 | 
						|
    tilesets["forest"].palette_id = 0x00
 | 
						|
    tilesets["forest"].attr_bank = 0x25
 | 
						|
    tilesets["forest"].attr_addr = 0x4000
 | 
						|
 | 
						|
    tilesets["town"] = rom_tilesets[0x26].copy()
 | 
						|
    tilesets["town"].combine(rom_tilesets[0x103])
 | 
						|
    tilesets["town"].palette_id = 0x03
 | 
						|
    tilesets["town"].attr_bank = 0x25
 | 
						|
    tilesets["town"].attr_addr = 0x4C00
 | 
						|
 | 
						|
    tilesets["swamp"] = rom_tilesets[0x36].copy()
 | 
						|
    tilesets["swamp"].combine(rom_tilesets[0x103])
 | 
						|
    tilesets["swamp"].palette_id = 0x0E
 | 
						|
    tilesets["swamp"].attr_bank = 0x22
 | 
						|
    tilesets["swamp"].attr_addr = 0x7400
 | 
						|
 | 
						|
    tilesets["beach"] = rom_tilesets[0x22].copy()
 | 
						|
    tilesets["beach"].combine(rom_tilesets[0x102])
 | 
						|
    tilesets["beach"].palette_id = 0x01
 | 
						|
    tilesets["beach"].attr_bank = 0x22
 | 
						|
    tilesets["beach"].attr_addr = 0x5000
 | 
						|
 | 
						|
    tilesets["water"] = rom_tilesets[0x3E].copy()
 | 
						|
    tilesets["water"].combine(rom_tilesets[0x103])
 | 
						|
    tilesets["water"].learn_from(tilesets["basic"])
 | 
						|
    tilesets["water"].remove(0x7A)
 | 
						|
    tilesets["water"].remove(0xC8)
 | 
						|
    tilesets["water"].palette_id = 0x09
 | 
						|
    tilesets["water"].attr_bank = 0x22
 | 
						|
    tilesets["water"].attr_addr = 0x6400
 | 
						|
 | 
						|
    return tilesets
 |