Archipelago/worlds/jakanddaxter/Rules.py

import typing
from BaseClasses import MultiWorld
from .Options import JakAndDaxterOptions
from .Regions import JakAndDaxterLevel, JakAndDaxterSubLevel, JakAndDaxterRegion, level_table, subLevel_table
from .Items import item_table


def set_rules(multiworld: MultiWorld, options: JakAndDaxterOptions, player: int):
    region_menu = multiworld.get_region("Menu", player)
    region_gr = multiworld.get_region(level_table[JakAndDaxterLevel.GEYSER_ROCK], player)
    region_menu.connect(region_gr)

    connect_regions(multiworld, player,
                    JakAndDaxterLevel.GEYSER_ROCK,
                    JakAndDaxterLevel.SANDOVER_VILLAGE,
                    lambda state: state.has(item_table[0], player, 4))

    connect_regions(multiworld, player,
                    JakAndDaxterLevel.SANDOVER_VILLAGE,
                    JakAndDaxterLevel.FORBIDDEN_JUNGLE)

    connect_subregions(multiworld, player,
                       JakAndDaxterLevel.FORBIDDEN_JUNGLE,
                       JakAndDaxterSubLevel.FORBIDDEN_JUNGLE_PLANT_ROOM,
                       lambda state: state.has(item_table[2216], player))

    connect_regions(multiworld, player,
                    JakAndDaxterLevel.SANDOVER_VILLAGE,
                    JakAndDaxterLevel.SENTINEL_BEACH)

    connect_subregions(multiworld, player,
                       JakAndDaxterLevel.SENTINEL_BEACH,
                       JakAndDaxterSubLevel.SENTINEL_BEACH_CANNON_TOWER,
                       lambda state: state.has(item_table[2216], player))

    connect_regions(multiworld, player,
                    JakAndDaxterLevel.SANDOVER_VILLAGE,
                    JakAndDaxterLevel.MISTY_ISLAND,
                    lambda state: state.has(item_table[2213], player))

    connect_regions(multiworld, player,
                    JakAndDaxterLevel.SANDOVER_VILLAGE,
                    JakAndDaxterLevel.FIRE_CANYON,
                    lambda state: state.has(item_table[0], player, 20))

    connect_regions(multiworld, player,
                    JakAndDaxterLevel.FIRE_CANYON,
                    JakAndDaxterLevel.ROCK_VILLAGE)

    connect_regions(multiworld, player,
                    JakAndDaxterLevel.ROCK_VILLAGE,
                    JakAndDaxterLevel.PRECURSOR_BASIN)

    connect_regions(multiworld, player,
                    JakAndDaxterLevel.ROCK_VILLAGE,
                    JakAndDaxterLevel.LOST_PRECURSOR_CITY)

    connect_regions(multiworld, player,
                    JakAndDaxterLevel.ROCK_VILLAGE,
                    JakAndDaxterLevel.BOGGY_SWAMP,
                    lambda state: state.has(item_table[2217], player))

    connect_subregions(multiworld, player,
                       JakAndDaxterLevel.BOGGY_SWAMP,
                       JakAndDaxterSubLevel.BOGGY_SWAMP_FLUT_FLUT,
                       lambda state: state.has(item_table[2215], player))

    connect_regions(multiworld, player,
                    JakAndDaxterLevel.ROCK_VILLAGE,
                    JakAndDaxterLevel.MOUNTAIN_PASS,
                    lambda state: state.has(item_table[2217], player) and state.has(item_table[0], player, 45))

    connect_subregions(multiworld, player,
                       JakAndDaxterLevel.MOUNTAIN_PASS,
                       JakAndDaxterSubLevel.MOUNTAIN_PASS_SHORTCUT,
                       lambda state: state.has(item_table[2218], player))

    connect_regions(multiworld, player,
                    JakAndDaxterLevel.MOUNTAIN_PASS,
                    JakAndDaxterLevel.VOLCANIC_CRATER)

    connect_regions(multiworld, player,
                    JakAndDaxterLevel.VOLCANIC_CRATER,
                    JakAndDaxterLevel.SPIDER_CAVE)

    connect_regions(multiworld, player,
                    JakAndDaxterLevel.VOLCANIC_CRATER,
                    JakAndDaxterLevel.SNOWY_MOUNTAIN)

    connect_subregions(multiworld, player,
                       JakAndDaxterLevel.SNOWY_MOUNTAIN,
                       JakAndDaxterSubLevel.SNOWY_MOUNTAIN_FLUT_FLUT,
                       lambda state: state.has(item_table[2215], player))

    connect_subregions(multiworld, player,
                       JakAndDaxterLevel.SNOWY_MOUNTAIN,
                       JakAndDaxterSubLevel.SNOWY_MOUNTAIN_LURKER_FORT,
                       lambda state: state.has(item_table[2219], player))

    connect_regions(multiworld, player,
                    JakAndDaxterLevel.VOLCANIC_CRATER,
                    JakAndDaxterLevel.LAVA_TUBE,
                    lambda state: state.has(item_table[0], player, 72))

    connect_regions(multiworld, player,
                    JakAndDaxterLevel.LAVA_TUBE,
                    JakAndDaxterLevel.GOL_AND_MAIAS_CITADEL)

    connect_subregions(multiworld, player,
                       JakAndDaxterLevel.GOL_AND_MAIAS_CITADEL,
                       JakAndDaxterSubLevel.GOL_AND_MAIAS_CITADEL_ROTATING_TOWER,
                       # lambda state: state.has(item_table[96], player) and state.has(item_table[97], player) and state.has(item_table[98], player))
                       lambda state: state.has(item_table[0], player, 75))


def connect_regions(multiworld: MultiWorld, player: int, source: JakAndDaxterLevel, target: JakAndDaxterLevel, rule=None):
    source_region = multiworld.get_region(level_table[source], player)
    target_region = multiworld.get_region(level_table[target], player)
    source_region.connect(target_region, rule=rule)


def connect_subregions(multiworld: MultiWorld, player: int, source: JakAndDaxterLevel, target: JakAndDaxterSubLevel, rule=None):
    source_region = multiworld.get_region(level_table[source], player)
    target_region = multiworld.get_region(subLevel_table[target], player)
    source_region.connect(target_region, rule=rule)