44 lines
2.2 KiB
Python
44 lines
2.2 KiB
Python
import typing
|
|
from ..generic.Rules import add_rule
|
|
from .Regions import connect_regions, v6areas
|
|
|
|
|
|
def _has_trinket_range(state, player, start, end) -> bool:
|
|
for i in range(start, end):
|
|
if not state.has("Trinket " + str(i + 1).zfill(2), player):
|
|
return False
|
|
return True
|
|
|
|
|
|
def set_rules(world, player, area_connections: typing.Dict[int, int], area_cost_map: typing.Dict[int, int]):
|
|
areashuffle = list(range(len(v6areas)))
|
|
if world.AreaRandomizer[player].value:
|
|
world.random.shuffle(areashuffle)
|
|
area_connections.update({(index + 1): (value + 1) for index, value in enumerate(areashuffle)})
|
|
area_connections.update({0: 0})
|
|
if world.AreaCostRandomizer[player].value:
|
|
world.random.shuffle(areashuffle)
|
|
area_cost_map.update({(index + 1): (value + 1) for index, value in enumerate(areashuffle)})
|
|
area_cost_map.update({0: 0})
|
|
|
|
for i in range(1, 5):
|
|
connect_regions(world, player, "Menu", v6areas[area_connections[i] - 1],
|
|
lambda state, i=i: _has_trinket_range(state, player,
|
|
world.DoorCost[player].value * (area_cost_map[i] - 1),
|
|
world.DoorCost[player].value * area_cost_map[i]))
|
|
|
|
# Special Rule for V
|
|
add_rule(world.get_location("V", player), lambda state: state.can_reach("Laboratory", 'Region', player) and
|
|
state.can_reach("The Tower", 'Region', player) and
|
|
state.can_reach("Space Station 2", 'Region', player) and
|
|
state.can_reach("Warp Zone", 'Region', player))
|
|
|
|
# Special Rule for NPC Trinket
|
|
add_rule(world.get_location("NPC Trinket", player),
|
|
lambda state: state.can_reach("Laboratory", 'Region', player) or
|
|
(state.can_reach("The Tower", 'Region', player) and
|
|
state.can_reach("Space Station 2", 'Region', player) and
|
|
state.can_reach("Warp Zone", 'Region', player)))
|
|
|
|
world.completion_condition[player] = lambda state: state.can_reach("V", 'Location', player)
|