103 lines
4.2 KiB
Python
103 lines
4.2 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
from BaseClasses import Entrance, Region
|
|
from entrance_rando import EntranceType, randomize_entrances
|
|
from .connections import RANDOMIZED_CONNECTIONS, TRANSITIONS
|
|
from .options import ShuffleTransitions, TransitionPlando
|
|
|
|
if TYPE_CHECKING:
|
|
from . import MessengerWorld
|
|
|
|
|
|
def disconnect_entrances(world: "MessengerWorld") -> None:
|
|
def disconnect_entrance() -> None:
|
|
child = entrance.connected_region.name
|
|
child_region = entrance.connected_region
|
|
child_region.entrances.remove(entrance)
|
|
entrance.connected_region = None
|
|
|
|
er_type = EntranceType.ONE_WAY if child == "Glacial Peak - Left" else \
|
|
EntranceType.TWO_WAY if child in RANDOMIZED_CONNECTIONS else EntranceType.ONE_WAY
|
|
if er_type == EntranceType.TWO_WAY:
|
|
mock_entrance = entrance.parent_region.create_er_target(entrance.name)
|
|
else:
|
|
mock_entrance = child_region.create_er_target(child)
|
|
|
|
entrance.randomization_type = er_type
|
|
mock_entrance.randomization_type = er_type
|
|
|
|
|
|
for parent, child in RANDOMIZED_CONNECTIONS.items():
|
|
if child == "Corrupted Future":
|
|
entrance = world.get_entrance("Artificer's Portal")
|
|
elif child == "Tower of Time - Left":
|
|
entrance = world.get_entrance("Artificer's Challenge")
|
|
else:
|
|
entrance = world.get_entrance(f"{parent} -> {child}")
|
|
disconnect_entrance()
|
|
|
|
def connect_plando(world: "MessengerWorld", plando_connections: TransitionPlando) -> None:
|
|
def remove_dangling_exit(region: Region) -> None:
|
|
# find the disconnected exit and remove references to it
|
|
for _exit in region.exits:
|
|
if not _exit.connected_region:
|
|
break
|
|
else:
|
|
raise ValueError(f"Unable to find randomized transition for {plando_connection}")
|
|
region.exits.remove(_exit)
|
|
|
|
def remove_dangling_entrance(region: Region) -> None:
|
|
# find the disconnected entrance and remove references to it
|
|
for _entrance in region.entrances:
|
|
if not _entrance.parent_region:
|
|
break
|
|
else:
|
|
raise ValueError(f"Invalid target region for {plando_connection}")
|
|
region.entrances.remove(_entrance)
|
|
|
|
for plando_connection in plando_connections:
|
|
# get the connecting regions
|
|
# need to handle these special because the names are unique but have the same parent region
|
|
if plando_connection.entrance in ("Artificer", "Tower HQ"):
|
|
reg1 = world.get_region("Tower HQ")
|
|
if plando_connection.entrance == "Artificer":
|
|
dangling_exit = world.get_entrance("Artificer's Portal")
|
|
else:
|
|
dangling_exit = world.get_entrance("Artificer's Challenge")
|
|
reg1.exits.remove(dangling_exit)
|
|
else:
|
|
reg1 = world.get_region(plando_connection.entrance)
|
|
remove_dangling_exit(reg1)
|
|
|
|
reg2 = world.get_region(plando_connection.exit)
|
|
remove_dangling_entrance(reg2)
|
|
# connect the regions
|
|
reg1.connect(reg2)
|
|
|
|
# pretend the user set the plando direction as "both" regardless of what they actually put on coupled
|
|
if ((world.options.shuffle_transitions == ShuffleTransitions.option_coupled
|
|
or plando_connection.direction == "both")
|
|
and plando_connection.exit in RANDOMIZED_CONNECTIONS):
|
|
remove_dangling_exit(reg2)
|
|
remove_dangling_entrance(reg1)
|
|
reg2.connect(reg1)
|
|
|
|
|
|
def shuffle_transitions(world: "MessengerWorld") -> None:
|
|
coupled = world.options.shuffle_transitions == ShuffleTransitions.option_coupled
|
|
|
|
plando = world.options.plando_connections
|
|
if plando:
|
|
connect_plando(world, plando)
|
|
|
|
result = randomize_entrances(world, coupled, {0: [0]})
|
|
|
|
world.transitions = sorted(result.placements, key=lambda entrance: TRANSITIONS.index(entrance.parent_region.name))
|
|
|
|
for transition in world.transitions:
|
|
if "->" not in transition.name:
|
|
continue
|
|
transition.parent_region.exits.remove(transition)
|
|
transition.name = f"{transition.parent_region.name} -> {transition.connected_region.name}"
|
|
transition.parent_region.exits.append(transition)
|