Grinch-AP/worlds/yoshisisland/Regions.py

from typing import List, Dict, TYPE_CHECKING
from BaseClasses import Region, Location
from .Locations import LocationData
from .Options import MinigameChecks
from .level_logic import YoshiLogic
from .setup_bosses import BossReqs
if TYPE_CHECKING:
    from . import YoshisIslandWorld


class YoshisIslandLocation(Location):
    game: str = "Yoshi's Island"
    level_id: int

    def __init__(self, player: int, name: str = " ", address: int = None, parent=None, level_id: int = None):
        super().__init__(player, name, address, parent)
        self.level_id = level_id


def init_areas(world: "YoshisIslandWorld", locations: List[LocationData]) -> None:
    multiworld = world.multiworld
    player = world.player
    logic = YoshiLogic(world)

    locations_per_region = get_locations_per_region(locations)

    regions = [
        create_region(world, player, locations_per_region, "Menu"),
        create_region(world, player, locations_per_region, "Overworld"),
        create_region(world, player, locations_per_region, "World 1"),
        create_region(world, player, locations_per_region, "World 2"),
        create_region(world, player, locations_per_region, "World 3"),
        create_region(world, player, locations_per_region, "World 4"),
        create_region(world, player, locations_per_region, "World 5"),
        create_region(world, player, locations_per_region, "World 6"),

        create_region(world, player, locations_per_region, "1-1"),
        create_region(world, player, locations_per_region, "1-2"),
        create_region(world, player, locations_per_region, "1-3"),
        create_region(world, player, locations_per_region, "1-4"),
        create_region(world, player, locations_per_region, "Burt The Bashful's Boss Room"),
        create_region(world, player, locations_per_region, "1-5"),
        create_region(world, player, locations_per_region, "1-6"),
        create_region(world, player, locations_per_region, "1-7"),
        create_region(world, player, locations_per_region, "1-8"),
        create_region(world, player, locations_per_region, "Salvo The Slime's Boss Room"),

        create_region(world, player, locations_per_region, "2-1"),
        create_region(world, player, locations_per_region, "2-2"),
        create_region(world, player, locations_per_region, "2-3"),
        create_region(world, player, locations_per_region, "2-4"),
        create_region(world, player, locations_per_region, "Bigger Boo's Boss Room"),
        create_region(world, player, locations_per_region, "2-5"),
        create_region(world, player, locations_per_region, "2-6"),
        create_region(world, player, locations_per_region, "2-7"),
        create_region(world, player, locations_per_region, "2-8"),
        create_region(world, player, locations_per_region, "Roger The Ghost's Boss Room"),

        create_region(world, player, locations_per_region, "3-1"),
        create_region(world, player, locations_per_region, "3-2"),
        create_region(world, player, locations_per_region, "3-3"),
        create_region(world, player, locations_per_region, "3-4"),
        create_region(world, player, locations_per_region, "Prince Froggy's Boss Room"),
        create_region(world, player, locations_per_region, "3-5"),
        create_region(world, player, locations_per_region, "3-6"),
        create_region(world, player, locations_per_region, "3-7"),
        create_region(world, player, locations_per_region, "3-8"),
        create_region(world, player, locations_per_region, "Naval Piranha's Boss Room"),

        create_region(world, player, locations_per_region, "4-1"),
        create_region(world, player, locations_per_region, "4-2"),
        create_region(world, player, locations_per_region, "4-3"),
        create_region(world, player, locations_per_region, "4-4"),
        create_region(world, player, locations_per_region, "Marching Milde's Boss Room"),
        create_region(world, player, locations_per_region, "4-5"),
        create_region(world, player, locations_per_region, "4-6"),
        create_region(world, player, locations_per_region, "4-7"),
        create_region(world, player, locations_per_region, "4-8"),
        create_region(world, player, locations_per_region, "Hookbill The Koopa's Boss Room"),

        create_region(world, player, locations_per_region, "5-1"),
        create_region(world, player, locations_per_region, "5-2"),
        create_region(world, player, locations_per_region, "5-3"),
        create_region(world, player, locations_per_region, "5-4"),
        create_region(world, player, locations_per_region, "Sluggy The Unshaven's Boss Room"),
        create_region(world, player, locations_per_region, "5-5"),
        create_region(world, player, locations_per_region, "5-6"),
        create_region(world, player, locations_per_region, "5-7"),
        create_region(world, player, locations_per_region, "5-8"),
        create_region(world, player, locations_per_region, "Raphael The Raven's Boss Room"),

        create_region(world, player, locations_per_region, "6-1"),
        create_region(world, player, locations_per_region, "6-2"),
        create_region(world, player, locations_per_region, "6-3"),
        create_region(world, player, locations_per_region, "6-4"),
        create_region(world, player, locations_per_region, "Tap-Tap The Red Nose's Boss Room"),
        create_region(world, player, locations_per_region, "6-5"),
        create_region(world, player, locations_per_region, "6-6"),
        create_region(world, player, locations_per_region, "6-7"),
        create_region(world, player, locations_per_region, "6-8"),
        create_region(world, player, locations_per_region, "Bowser's Room"),
    ]

    if world.options.extras_enabled:
        regions.insert(68, create_region(world, player, locations_per_region, "6-Extra"))
        regions.insert(58, create_region(world, player, locations_per_region, "5-Extra"))
        regions.insert(48, create_region(world, player, locations_per_region, "4-Extra"))
        regions.insert(38, create_region(world, player, locations_per_region, "3-Extra"))
        regions.insert(28, create_region(world, player, locations_per_region, "2-Extra"))
        regions.insert(18, create_region(world, player, locations_per_region, "1-Extra"))

    if world.options.minigame_checks in {MinigameChecks.option_bonus_games, MinigameChecks.option_both}:
        regions.insert(74, create_region(world, player, locations_per_region, "6-Bonus"))
        regions.insert(63, create_region(world, player, locations_per_region, "5-Bonus"))
        regions.insert(52, create_region(world, player, locations_per_region, "4-Bonus"))
        regions.insert(41, create_region(world, player, locations_per_region, "3-Bonus"))
        regions.insert(29, create_region(world, player, locations_per_region, "2-Bonus"))
        regions.insert(19, create_region(world, player, locations_per_region, "1-Bonus"))

    multiworld.regions += regions

    connect_starting_region(world)

    bosses = BossReqs(world)

    multiworld.get_region("Overworld", player).add_exits(
    ["World 1", "World 2", "World 3", "World 4", "World 5", "World 6"],
         {
              "World 1": lambda state: state.has("World 1 Gate", player),
              "World 2": lambda state: state.has("World 2 Gate", player),
              "World 3": lambda state: state.has("World 3 Gate", player),
              "World 4": lambda state: state.has("World 4 Gate", player),
              "World 5": lambda state: state.has("World 5 Gate", player),
              "World 6": lambda state: state.has("World 6 Gate", player)
         }
    )

    for cur_world in range(1, 7):
        for cur_level in range(8):
            if cur_world != 6 or cur_level != 7:
                multiworld.get_region(f"World {cur_world}", player).add_exits(
                    [world.level_location_list[(cur_world - 1) * 8 + cur_level]]
                )

    multiworld.get_region("1-4", player).add_exits([world.boss_order[0]],{world.boss_order[0]: lambda state: logic._14Clear(state)})
    multiworld.get_region("1-8", player).add_exits([world.boss_order[1]],{world.boss_order[1]: lambda state: logic._18Clear(state)})
    multiworld.get_region("2-4", player).add_exits([world.boss_order[2]],{world.boss_order[2]: lambda state: logic._24Clear(state)})
    multiworld.get_region("2-8", player).add_exits([world.boss_order[3]],{world.boss_order[3]: lambda state: logic._28Clear(state)})
    multiworld.get_region("3-4", player).add_exits([world.boss_order[4]],{world.boss_order[4]: lambda state: logic._34Clear(state)})
    multiworld.get_region("3-8", player).add_exits([world.boss_order[5]],{world.boss_order[5]: lambda state: logic._38Clear(state)})
    multiworld.get_region("4-4", player).add_exits([world.boss_order[6]],{world.boss_order[6]: lambda state: logic._44Clear(state)})
    multiworld.get_region("4-8", player).add_exits([world.boss_order[7]],{world.boss_order[7]: lambda state: logic._48Clear(state)})
    multiworld.get_region("5-4", player).add_exits([world.boss_order[8]],{world.boss_order[8]: lambda state: logic._54Clear(state)})
    multiworld.get_region("5-8", player).add_exits([world.boss_order[9]],{world.boss_order[9]: lambda state: logic._58Clear(state)})
    multiworld.get_region("World 6", player).add_exits(["6-8"],{"6-8": lambda state: bosses.castle_access(state)})
    multiworld.get_region("6-4", player).add_exits([world.boss_order[10]],{world.boss_order[10]: lambda state: logic._64Clear(state)})
    multiworld.get_region("6-8", player).add_exits(["Bowser's Room"],{"Bowser's Room": lambda state: bosses.castle_clear(state)})

    if world.options.extras_enabled:
        multiworld.get_region("World 1", player).add_exits(
            ["1-Extra"],
            {"1-Extra": lambda state: state.has_any({"Extra Panels", "Extra 1"}, player)}
        )
        multiworld.get_region("World 2", player).add_exits(
            ["2-Extra"],
            {"2-Extra": lambda state: state.has_any({"Extra Panels", "Extra 2"}, player)}
        )
        multiworld.get_region(
            "World 3", player).add_exits(["3-Extra"],
            {"3-Extra": lambda state: state.has_any({"Extra Panels", "Extra 3"}, player)}
        )
        multiworld.get_region("World 4", player).add_exits(
            ["4-Extra"],
            {"4-Extra": lambda state: state.has_any({"Extra Panels", "Extra 4"}, player)}
        )
        multiworld.get_region("World 5", player).add_exits(
            ["5-Extra"],
            {"5-Extra": lambda state: state.has_any({"Extra Panels", "Extra 5"}, player)}
        )
        multiworld.get_region("World 6", player).add_exits(
            ["6-Extra"],
            {"6-Extra": lambda state: state.has_any({"Extra Panels", "Extra 6"}, player)}
        )

    if world.options.minigame_checks in {MinigameChecks.option_bonus_games, MinigameChecks.option_both}:
        multiworld.get_region("World 1", player).add_exits(
            ["1-Bonus"],
            {"1-Bonus": lambda state: state.has_any({"Bonus Panels", "Bonus 1"}, player)}
        )
        multiworld.get_region("World 2", player).add_exits(
            ["2-Bonus"],
            {"2-Bonus": lambda state: state.has_any({"Bonus Panels", "Bonus 2"}, player)}
        )
        multiworld.get_region("World 3", player).add_exits(
            ["3-Bonus"],
            {"3-Bonus": lambda state: state.has_any({"Bonus Panels", "Bonus 3"}, player)}
        )
        multiworld.get_region("World 4", player).add_exits(
            ["4-Bonus"],
            {"4-Bonus": lambda state: state.has_any({"Bonus Panels", "Bonus 4"}, player)}
        )
        multiworld.get_region("World 5", player).add_exits(
            ["5-Bonus"],
            {"5-Bonus": lambda state: state.has_any({"Bonus Panels", "Bonus 5"}, player)}
        )
        multiworld.get_region("World 6", player).add_exits(
            ["6-Bonus"],
            {"6-Bonus": lambda state: state.has_any({"Bonus Panels", "Bonus 6"}, player)}
        )


def create_location(player: int, location_data: LocationData, region: Region) -> Location:
    location = YoshisIslandLocation(player, location_data.name, location_data.code, region)
    location.access_rule = location_data.rule
    location.level_id = location_data.LevelID

    return location


def create_region(world: "YoshisIslandWorld", player: int, locations_per_region: Dict[str, List[LocationData]], name: str) -> Region:
    region = Region(name, player, world.multiworld)

    if name in locations_per_region:
        for location_data in locations_per_region[name]:
            location = create_location(player, location_data, region)
            region.locations.append(location)

    return region

def connect_starting_region(world: "YoshisIslandWorld") -> None:
    multiworld = world.multiworld
    player = world.player
    menu = multiworld.get_region("Menu", player)
    world_main = multiworld.get_region("Overworld", player)

    starting_region = multiworld.get_region(f"World {world.options.starting_world + 1}", player)

    menu.connect(world_main, "Start Game")
    world_main.connect(starting_region, "Overworld")


def get_locations_per_region(locations: List[LocationData]) -> Dict[str, List[LocationData]]:
    per_region: Dict[str, List[LocationData]] = {}

    for location in locations:
        per_region.setdefault(location.region, []).append(location)

    return per_region