Archipelago/worlds/sa2b/GateBosses.py

import typing
from BaseClasses import MultiWorld
from worlds.AutoWorld import World

speed_characters_1 = "Sonic vs Shadow 1"
speed_characters_2 = "Sonic vs Shadow 2"
mech_characters_1 = "Tails vs Eggman 1"
mech_characters_2 = "Tails vs Eggman 2"
hunt_characters_1 = "Knuckles vs Rouge 1"
big_foot = "F-6t BIG FOOT"
hot_shot = "B-3x HOT SHOT"
flying_dog = "R-1/A FLYING DOG"
egg_golem_sonic = "Egg Golem (Sonic)"
egg_golem_eggman = "Egg Golem (Eggman)"
king_boom_boo = "King Boom Boo"

gate_bosses_no_requirements_table = {
    speed_characters_1: 0,
    speed_characters_2: 1,
    mech_characters_1: 2,
    mech_characters_2: 3,
    hunt_characters_1: 4,
    big_foot: 5,
    hot_shot: 6,
    flying_dog: 7,
    egg_golem_sonic: 8,
    egg_golem_eggman: 9,
}

gate_bosses_with_requirements_table = {
    king_boom_boo: 10,
}

extra_boss_rush_bosses_table = {
    speed_characters_1: 11,
    speed_characters_2: 12,
    mech_characters_1: 13,
    mech_characters_2: 14,
    hunt_characters_1: 15,
}

all_gate_bosses_table = {
    **gate_bosses_no_requirements_table,
    **gate_bosses_with_requirements_table,
}


def get_boss_name(boss: int):
    for key, value in gate_bosses_no_requirements_table.items():
        if value == boss:
            return key
    for key, value in gate_bosses_with_requirements_table.items():
        if value == boss:
            return key
    for key, value in extra_boss_rush_bosses_table.items():
        if value == boss:
            return key


def boss_has_requirement(boss: int):
    return boss >= len(gate_bosses_no_requirements_table)


def get_gate_bosses(multiworld: MultiWorld, world: World):
    selected_bosses: typing.List[int] = []
    boss_gates: typing.List[int] = []
    available_bosses: typing.List[str] = list(gate_bosses_no_requirements_table.keys())
    multiworld.random.shuffle(available_bosses)
    halfway = False

    for x in range(world.options.number_of_level_gates):
        if (not halfway) and ((x + 1) / world.options.number_of_level_gates) > 0.5:
            available_bosses.extend(gate_bosses_with_requirements_table)
            multiworld.random.shuffle(available_bosses)
            halfway = True
        selected_bosses.append(all_gate_bosses_table[available_bosses[0]])
        boss_gates.append(x + 1)
        available_bosses.remove(available_bosses[0])

    bosses: typing.Dict[int, int] = dict(zip(boss_gates, selected_bosses))

    return bosses


def get_boss_rush_bosses(multiworld: MultiWorld, world: World):

    if world.options.boss_rush_shuffle == 0:
        boss_list_o = list(range(0, 16))
        boss_list_s = [5, 2, 0, 10, 8, 4, 3, 1, 6, 13, 7, 11, 9, 15, 14, 12]

        return dict(zip(boss_list_o, boss_list_s))
    elif world.options.boss_rush_shuffle == 1:
        boss_list_o = list(range(0, 16))
        boss_list_s = boss_list_o.copy()
        multiworld.random.shuffle(boss_list_s)

        return dict(zip(boss_list_o, boss_list_s))
    elif world.options.boss_rush_shuffle == 2:
        boss_list_o = list(range(0, 16))
        boss_list_s = [multiworld.random.choice(boss_list_o) for i in range(0, 16)]
        if 10 not in boss_list_s:
            boss_list_s[multiworld.random.randint(0, 15)] = 10

        return dict(zip(boss_list_o, boss_list_s))
    elif world.options.boss_rush_shuffle == 3:
        boss_list_o = list(range(0, 16))
        boss_list_s = [multiworld.random.choice(boss_list_o)] * len(boss_list_o)
        if 10 not in boss_list_s:
            boss_list_s[multiworld.random.randint(0, 15)] = 10

        return dict(zip(boss_list_o, boss_list_s))
    else:
        return dict()