Archipelago/worlds/v6/Rules.py

39 lines
2.0 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: _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("Space Station 2",'Region',player))
world.completion_condition[player] = lambda state: state.can_reach("V",'Location',player)