108 lines
4.0 KiB
Python
108 lines
4.0 KiB
Python
|
from ...locations.items import BOMB
|
||
|
|
from .base import LocationBase
|
||
|
|
from ...roomEditor import RoomEditor, Object, ObjectWarp
|
||
|
|
from ...entranceInfo import ENTRANCE_INFO
|
||
|
|
from ...assembler import ASM
|
||
|
|
from .entrance_info import INFO
|
||
|
|
|
||
|
|
|
||
|
|
class Entrance(LocationBase):
|
||
|
|
def __init__(self, room, x, y, entrance_name):
|
||
|
|
super().__init__(room, x, y)
|
||
|
|
self.entrance_name = entrance_name
|
||
|
|
self.entrance_info = ENTRANCE_INFO[entrance_name]
|
||
|
|
self.source_warp = None
|
||
|
|
self.target_warp_idx = None
|
||
|
|
|
||
|
|
self.inside_logic = None
|
||
|
|
|
||
|
|
def prepare(self, rom):
|
||
|
|
info = self.entrance_info
|
||
|
|
re = RoomEditor(rom, info.alt_room if info.alt_room is not None else info.room)
|
||
|
|
self.source_warp = re.getWarps()[info.index if info.index not in (None, "all") else 0]
|
||
|
|
re = RoomEditor(rom, self.source_warp.room)
|
||
|
|
for idx, warp in enumerate(re.getWarps()):
|
||
|
|
if warp.room == info.room or warp.room == info.alt_room:
|
||
|
|
self.target_warp_idx = idx
|
||
|
|
|
||
|
|
def update_room(self, rom, re: RoomEditor):
|
||
|
|
re.objects.append(self.source_warp)
|
||
|
|
|
||
|
|
target = RoomEditor(rom, self.source_warp.room)
|
||
|
|
warp = target.getWarps()[self.target_warp_idx]
|
||
|
|
warp.room = self.room.x | (self.room.y << 4)
|
||
|
|
warp.target_x = self.x * 16 + 8
|
||
|
|
warp.target_y = self.y * 16 + 18
|
||
|
|
target.store(rom)
|
||
|
|
|
||
|
|
def prepare_logic(self, configuration_options, world_setup, requirements_settings):
|
||
|
|
if self.entrance_name in INFO and INFO[self.entrance_name].logic is not None:
|
||
|
|
self.inside_logic = INFO[self.entrance_name].logic(configuration_options, world_setup, requirements_settings)
|
||
|
|
|
||
|
|
def connect_logic(self, logic_location):
|
||
|
|
if self.entrance_name not in INFO:
|
||
|
|
raise RuntimeError(f"WARNING: Logic connection to entrance unmapped! {self.entrance_name}")
|
||
|
|
if self.inside_logic:
|
||
|
|
req = None
|
||
|
|
if self.room.tiles[self.x + self.y * 10] == 0xBA:
|
||
|
|
req = BOMB
|
||
|
|
logic_location.connect(self.inside_logic, req)
|
||
|
|
if INFO[self.entrance_name].exits:
|
||
|
|
return [(name, logic(logic_location)) for name, logic in INFO[self.entrance_name].exits]
|
||
|
|
return None
|
||
|
|
|
||
|
|
def get_item_pool(self):
|
||
|
|
if self.entrance_name not in INFO:
|
||
|
|
return {}
|
||
|
|
return INFO[self.entrance_name].items or {}
|
||
|
|
|
||
|
|
|
||
|
|
class DummyEntrance(LocationBase):
|
||
|
|
def __init__(self, room, x, y):
|
||
|
|
super().__init__(room, x, y)
|
||
|
|
|
||
|
|
def update_room(self, rom, re: RoomEditor):
|
||
|
|
re.objects.append(ObjectWarp(0x01, 0x10, 0x2A3, 0x50, 0x7C))
|
||
|
|
|
||
|
|
def connect_logic(self, logic_location):
|
||
|
|
return
|
||
|
|
|
||
|
|
def get_item_pool(self):
|
||
|
|
return {}
|
||
|
|
|
||
|
|
|
||
|
|
class EggEntrance(LocationBase):
|
||
|
|
def __init__(self, room, x, y):
|
||
|
|
super().__init__(room, x, y)
|
||
|
|
|
||
|
|
def update_room(self, rom, re: RoomEditor):
|
||
|
|
# Setup the warps
|
||
|
|
re.objects.insert(0, Object(5, 3, 0xE1)) # Hide an entrance tile under the tile where the egg will open.
|
||
|
|
re.objects.append(ObjectWarp(0x01, 0x08, 0x270, 0x50, 0x7C))
|
||
|
|
re.entities.append((0, 0, 0xDE)) # egg song event
|
||
|
|
|
||
|
|
egg_inside = RoomEditor(rom, 0x270)
|
||
|
|
egg_inside.getWarps()[0].room = self.room.x
|
||
|
|
egg_inside.store(rom)
|
||
|
|
|
||
|
|
# Fix the alt room layout
|
||
|
|
alt = RoomEditor(rom, "Alt06")
|
||
|
|
tiles = re.getTileArray()
|
||
|
|
tiles[25] = 0xC1
|
||
|
|
tiles[35] = 0xCB
|
||
|
|
alt.buildObjectList(tiles, reduce_size=True)
|
||
|
|
alt.store(rom)
|
||
|
|
|
||
|
|
# Patch which room shows as Alt06
|
||
|
|
rom.patch(0x00, 0x31F1, ASM("cp $06"), ASM(f"cp ${self.room.x:02x}"))
|
||
|
|
rom.patch(0x00, 0x31F5, ASM("ld a, [$D806]"), ASM(f"ld a, [${0xD800 + self.room.x:04x}]"))
|
||
|
|
rom.patch(0x20, 0x2DE6, ASM("cp $06"), ASM(f"cp ${self.room.x:02x}"))
|
||
|
|
rom.patch(0x20, 0x2DEA, ASM("ld a, [$D806]"), ASM(f"ld a, [${0xD800 + self.room.x:04x}]"))
|
||
|
|
rom.patch(0x19, 0x0D1A, ASM("ld hl, $D806"), ASM(f"ld hl, ${0xD800 + self.room.x:04x}"))
|
||
|
|
|
||
|
|
def connect_logic(self, logic_location):
|
||
|
|
return
|
||
|
|
|
||
|
|
def get_item_pool(self):
|
||
|
|
return {}
|