Grinch-AP/worlds/kh1/__init__.py

import logging
from typing import List

from BaseClasses import Tutorial
from worlds.AutoWorld import WebWorld, World
from .Items import KH1Item, KH1ItemData, event_item_table, get_items_by_category, item_table, item_name_groups
from .Locations import KH1Location, location_table, get_locations_by_category, location_name_groups
from .Options import KH1Options, kh1_option_groups
from .Regions import connect_entrances, create_regions
from .Rules import set_rules
from .Presets import kh1_option_presets
from worlds.LauncherComponents import Component, components, Type, launch as launch_component


def launch_client():
    from .Client import launch
    launch_component(launch, name="KH1 Client")


components.append(Component("KH1 Client", "KH1Client", func=launch_client, component_type=Type.CLIENT))


class KH1Web(WebWorld):
    theme = "ocean"
    tutorials = [Tutorial(
            "Multiworld Setup Guide",
            "A guide to setting up the Kingdom Hearts Randomizer software on your computer."
            "This guide covers single-player, multiworld, and related software.",
            "English",
            "kh1_en.md",
            "kh1/en",
            ["Gicu"]
    )]
    option_groups = kh1_option_groups
    options_presets = kh1_option_presets


class KH1World(World):
    """
    Kingdom Hearts is an action RPG following Sora on his journey 
    through many worlds to find Riku and Kairi.
    """
    game = "Kingdom Hearts"
    options_dataclass = KH1Options
    options: KH1Options
    topology_present = True
    web = KH1Web()

    item_name_to_id = {name: data.code for name, data in item_table.items()}
    location_name_to_id = {name: data.code for name, data in location_table.items()}
    item_name_groups = item_name_groups
    location_name_groups = location_name_groups
    fillers = {}
    fillers.update(get_items_by_category("Item"))
    fillers.update(get_items_by_category("Camping"))
    fillers.update(get_items_by_category("Stat Ups"))

    def create_items(self):
        self.place_predetermined_items()
        # Handle starting worlds
        starting_worlds = []
        if self.options.starting_worlds > 0:
            possible_starting_worlds = ["Wonderland", "Olympus Coliseum", "Deep Jungle", "Agrabah", "Monstro", "Halloween Town", "Neverland", "Hollow Bastion"]
            if self.options.atlantica:
                possible_starting_worlds.append("Atlantica")
            if self.options.end_of_the_world_unlock == "item":
                possible_starting_worlds.append("End of the World")
            starting_worlds = self.random.sample(possible_starting_worlds, min(self.options.starting_worlds.value, len(possible_starting_worlds)))
            for starting_world in starting_worlds:
                self.multiworld.push_precollected(self.create_item(starting_world))
        
        item_pool: List[KH1Item] = []
        possible_level_up_item_pool = []
        level_up_item_pool = []
        
        # Calculate Level Up Items
        # Fill pool with mandatory items
        for _ in range(self.options.item_slot_increase):
            level_up_item_pool.append("Item Slot Increase")
        for _ in range(self.options.accessory_slot_increase):
            level_up_item_pool.append("Accessory Slot Increase")

        # Create other pool
        for _ in range(self.options.strength_increase):
            possible_level_up_item_pool.append("Strength Increase")
        for _ in range(self.options.defense_increase):
            possible_level_up_item_pool.append("Defense Increase")
        for _ in range(self.options.hp_increase):
            possible_level_up_item_pool.append("Max HP Increase")
        for _ in range(self.options.mp_increase):
            possible_level_up_item_pool.append("Max MP Increase")
        for _ in range(self.options.ap_increase):
            possible_level_up_item_pool.append("Max AP Increase")

        # Fill remaining pool with items from other pool
        self.random.shuffle(possible_level_up_item_pool)
        level_up_item_pool = level_up_item_pool + possible_level_up_item_pool[:(100 - len(level_up_item_pool))]

        level_up_locations = list(get_locations_by_category("Levels").keys())
        self.random.shuffle(level_up_item_pool)
        current_level_for_placing_stats = self.options.force_stats_on_levels.value
        while len(level_up_item_pool) > 0 and current_level_for_placing_stats <= self.options.level_checks:
            self.get_location(level_up_locations[current_level_for_placing_stats - 1]).place_locked_item(self.create_item(level_up_item_pool.pop()))
            current_level_for_placing_stats += 1
        
        # Calculate prefilled locations and items
        prefilled_items = []
        if self.options.vanilla_emblem_pieces:
            prefilled_items = prefilled_items + ["Emblem Piece (Flame)", "Emblem Piece (Chest)", "Emblem Piece (Fountain)", "Emblem Piece (Statue)"]
        
        total_locations = len(self.multiworld.get_unfilled_locations(self.player))
        
        non_filler_item_categories = ["Key", "Magic", "Worlds", "Trinities", "Cups", "Summons", "Abilities", "Shared Abilities", "Keyblades", "Accessory", "Weapons", "Puppies"]
        if self.options.hundred_acre_wood:
            non_filler_item_categories.append("Torn Pages")
        for name, data in item_table.items():
            quantity = data.max_quantity
            if data.category not in non_filler_item_categories:
                continue
            if name in starting_worlds:
                continue
            if data.category == "Puppies":
                if self.options.puppies == "triplets" and "-" in name:
                    item_pool += [self.create_item(name) for _ in range(quantity)]
                if self.options.puppies == "individual" and "Puppy" in name:
                    item_pool += [self.create_item(name) for _ in range(0, quantity)]
                if self.options.puppies == "full" and name == "All Puppies":
                    item_pool += [self.create_item(name) for _ in range(0, quantity)]
            elif name == "Atlantica":
                if self.options.atlantica:
                    item_pool += [self.create_item(name) for _ in range(0, quantity)]
            elif name == "Mermaid Kick":
                if self.options.atlantica:
                    if self.options.extra_shared_abilities:
                        item_pool += [self.create_item(name) for _ in range(0, 2)]
                    else:
                        item_pool += [self.create_item(name) for _ in range(0, quantity)]
            elif name == "Crystal Trident":
                if self.options.atlantica:
                    item_pool += [self.create_item(name) for _ in range(0, quantity)]
            elif name == "High Jump":
                if self.options.extra_shared_abilities:
                    item_pool += [self.create_item(name) for _ in range(0, 3)]
                else:
                    item_pool += [self.create_item(name) for _ in range(0, quantity)]
            elif name == "Progressive Glide":
                if self.options.extra_shared_abilities:
                    item_pool += [self.create_item(name) for _ in range(0, 4)]
                else:
                    item_pool += [self.create_item(name) for _ in range(0, quantity)]
            elif name == "End of the World":
                if self.options.end_of_the_world_unlock.current_key == "item":
                    item_pool += [self.create_item(name) for _ in range(0, quantity)]
            elif name == "EXP Zero":
                if self.options.exp_zero_in_pool:
                    item_pool += [self.create_item(name) for _ in range(0, quantity)]
            elif name not in prefilled_items:
                item_pool += [self.create_item(name) for _ in range(0, quantity)]
        
        for i in range(self.determine_reports_in_pool()):
            item_pool += [self.create_item("Ansem's Report " + str(i+1))]
        
        while len(item_pool) < total_locations and len(level_up_item_pool) > 0:
            item_pool += [self.create_item(level_up_item_pool.pop())]
        
        # Fill any empty locations with filler items.
        while len(item_pool) < total_locations:
            item_pool.append(self.create_item(self.get_filler_item_name()))
        
        self.multiworld.itempool += item_pool

    def place_predetermined_items(self) -> None:
        goal_dict = {
            "sephiroth":       "Olympus Coliseum Defeat Sephiroth Ansem's Report 12",
            "unknown":         "Hollow Bastion Defeat Unknown Ansem's Report 13",
            "postcards":       "Traverse Town Mail Postcard 10 Event",
            "final_ansem":     "Final Ansem",
            "puppies":         "Traverse Town Piano Room Return 99 Puppies Reward 2",
            "final_rest":      "End of the World Final Rest Chest"
        }
        self.get_location(goal_dict[self.options.goal.current_key]).place_locked_item(self.create_item("Victory"))
        if self.options.vanilla_emblem_pieces:
            self.get_location("Hollow Bastion Entrance Hall Emblem Piece (Flame)").place_locked_item(self.create_item("Emblem Piece (Flame)"))
            self.get_location("Hollow Bastion Entrance Hall Emblem Piece (Statue)").place_locked_item(self.create_item("Emblem Piece (Statue)"))
            self.get_location("Hollow Bastion Entrance Hall Emblem Piece (Fountain)").place_locked_item(self.create_item("Emblem Piece (Fountain)"))
            self.get_location("Hollow Bastion Entrance Hall Emblem Piece (Chest)").place_locked_item(self.create_item("Emblem Piece (Chest)"))

    def get_filler_item_name(self) -> str:
        weights = [data.weight for data in self.fillers.values()]
        return self.random.choices([filler for filler in self.fillers.keys()], weights)[0]

    def fill_slot_data(self) -> dict:
        slot_data = {"xpmult": int(self.options.exp_multiplier)/16,
                    "required_reports_eotw": self.determine_reports_required_to_open_end_of_the_world(),
                    "required_reports_door": self.determine_reports_required_to_open_final_rest_door(),
                    "door": self.options.final_rest_door.current_key,
                    "seed": self.multiworld.seed_name,
                    "advanced_logic": bool(self.options.advanced_logic),
                    "hundred_acre_wood": bool(self.options.hundred_acre_wood),
                    "atlantica": bool(self.options.atlantica),
                    "goal": str(self.options.goal.current_key)}
        if self.options.randomize_keyblade_stats:
            min_str_bonus = min(self.options.keyblade_min_str.value, self.options.keyblade_max_str.value)
            max_str_bonus = max(self.options.keyblade_min_str.value, self.options.keyblade_max_str.value)
            self.options.keyblade_min_str.value = min_str_bonus
            self.options.keyblade_max_str.value = max_str_bonus
            min_mp_bonus = min(self.options.keyblade_min_mp.value, self.options.keyblade_max_mp.value)
            max_mp_bonus = max(self.options.keyblade_min_mp.value, self.options.keyblade_max_mp.value)
            self.options.keyblade_min_mp.value = min_mp_bonus
            self.options.keyblade_max_mp.value = max_mp_bonus
            slot_data["keyblade_stats"] = ""
            for i in range(22):
                if i < 4 and self.options.bad_starting_weapons:
                    slot_data["keyblade_stats"] = slot_data["keyblade_stats"] + "1,0,"
                else:
                    str_bonus = int(self.random.randint(min_str_bonus, max_str_bonus))
                    mp_bonus = int(self.random.randint(min_mp_bonus, max_mp_bonus))
                    slot_data["keyblade_stats"] = slot_data["keyblade_stats"] + str(str_bonus) + "," + str(mp_bonus) + ","
            slot_data["keyblade_stats"] = slot_data["keyblade_stats"][:-1]
        if self.options.donald_death_link:
            slot_data["donalddl"] = ""
        if self.options.goofy_death_link:
            slot_data["goofydl"] = ""
        if self.options.keyblades_unlock_chests:
            slot_data["chestslocked"] = ""
        else:
            slot_data["chestsunlocked"] = ""
        if self.options.interact_in_battle:
            slot_data["interactinbattle"] = ""
        return slot_data
    
    def create_item(self, name: str) -> KH1Item:
        data = item_table[name]
        return KH1Item(name, data.classification, data.code, self.player)

    def create_event(self, name: str) -> KH1Item:
        data = event_item_table[name]
        return KH1Item(name, data.classification, data.code, self.player)

    def set_rules(self):
        set_rules(self)

    def create_regions(self):
        create_regions(self.multiworld, self.player, self.options)

    def connect_entrances(self):
        connect_entrances(self.multiworld, self.player)
    
    def generate_early(self):
        value_names = ["Reports to Open End of the World", "Reports to Open Final Rest Door", "Reports in Pool"]
        initial_report_settings = [self.options.required_reports_eotw.value, self.options.required_reports_door.value, self.options.reports_in_pool.value]
        self.change_numbers_of_reports_to_consider()
        new_report_settings = [self.options.required_reports_eotw.value, self.options.required_reports_door.value, self.options.reports_in_pool.value]
        for i in range(3):
            if initial_report_settings[i] != new_report_settings[i]:
                logging.info(f"{self.player_name}'s value {initial_report_settings[i]} for \"{value_names[i]}\" was invalid\n"
                             f"Setting \"{value_names[i]}\" value to {new_report_settings[i]}")
    
    def change_numbers_of_reports_to_consider(self) -> None:
        if self.options.end_of_the_world_unlock == "reports" and self.options.final_rest_door == "reports":
            self.options.required_reports_eotw.value, self.options.required_reports_door.value, self.options.reports_in_pool.value = sorted(
                [self.options.required_reports_eotw.value, self.options.required_reports_door.value, self.options.reports_in_pool.value])

        elif self.options.end_of_the_world_unlock == "reports":
            self.options.required_reports_eotw.value, self.options.reports_in_pool.value = sorted(
                [self.options.required_reports_eotw.value, self.options.reports_in_pool.value])

        elif self.options.final_rest_door == "reports":
            self.options.required_reports_door.value, self.options.reports_in_pool.value = sorted(
                [self.options.required_reports_door.value, self.options.reports_in_pool.value])

    def determine_reports_in_pool(self) -> int:
        if self.options.end_of_the_world_unlock == "reports" or self.options.final_rest_door == "reports":
            return self.options.reports_in_pool.value
        return 0
    
    def determine_reports_required_to_open_end_of_the_world(self) -> int:
        if self.options.end_of_the_world_unlock == "reports":
            return self.options.required_reports_eotw.value
        return 14
    
    def determine_reports_required_to_open_final_rest_door(self) -> int:
        if self.options.final_rest_door == "reports":
            return self.options.required_reports_door.value
        return 14