323 lines
		
	
	
		
			15 KiB
		
	
	
	
		
			Python
		
	
	
	
			
		
		
	
	
			323 lines
		
	
	
		
			15 KiB
		
	
	
	
		
			Python
		
	
	
	
import logging
 | 
						|
from typing import Dict, Any, Iterable, Optional, Union, Set, List
 | 
						|
 | 
						|
from BaseClasses import Region, Entrance, Location, Item, Tutorial, CollectionState, ItemClassification, MultiWorld, Group as ItemLinkGroup
 | 
						|
from Options import PerGameCommonOptions
 | 
						|
from worlds.AutoWorld import World, WebWorld
 | 
						|
from . import rules
 | 
						|
from .bundles import get_all_bundles, Bundle
 | 
						|
from .items import item_table, create_items, ItemData, Group, items_by_group, get_all_filler_items, remove_limited_amount_packs
 | 
						|
from .locations import location_table, create_locations, LocationData
 | 
						|
from .logic import StardewLogic, StardewRule, True_, MAX_MONTHS
 | 
						|
from .options import StardewValleyOptions, SeasonRandomization, Goal, BundleRandomization, BundlePrice, NumberOfLuckBuffs, NumberOfMovementBuffs, \
 | 
						|
    BackpackProgression, BuildingProgression, ExcludeGingerIsland, TrapItems
 | 
						|
from .presets import sv_options_presets
 | 
						|
from .regions import create_regions
 | 
						|
from .rules import set_rules
 | 
						|
from worlds.generic.Rules import set_rule
 | 
						|
from .strings.goal_names import Goal as GoalName
 | 
						|
 | 
						|
client_version = 0
 | 
						|
 | 
						|
 | 
						|
class StardewLocation(Location):
 | 
						|
    game: str = "Stardew Valley"
 | 
						|
 | 
						|
    def __init__(self, player: int, name: str, address: Optional[int], parent=None):
 | 
						|
        super().__init__(player, name, address, parent)
 | 
						|
        self.event = not address
 | 
						|
 | 
						|
 | 
						|
class StardewItem(Item):
 | 
						|
    game: str = "Stardew Valley"
 | 
						|
 | 
						|
 | 
						|
class StardewWebWorld(WebWorld):
 | 
						|
    theme = "dirt"
 | 
						|
    bug_report_page = "https://github.com/agilbert1412/StardewArchipelago/issues/new?labels=bug&title=%5BBug%5D%3A+Brief+Description+of+bug+here"
 | 
						|
    options_presets = sv_options_presets
 | 
						|
 | 
						|
    tutorials = [
 | 
						|
        Tutorial(
 | 
						|
            "Multiworld Setup Guide",
 | 
						|
            "A guide to playing Stardew Valley with Archipelago.",
 | 
						|
            "English",
 | 
						|
            "setup_en.md",
 | 
						|
            "setup/en",
 | 
						|
            ["KaitoKid", "Jouramie", "Witchybun (Mod Support)", "Exempt-Medic (Proofreading)"]
 | 
						|
        )]
 | 
						|
 | 
						|
 | 
						|
class StardewValleyWorld(World):
 | 
						|
    """
 | 
						|
    Stardew Valley is an open-ended country-life RPG. You can farm, fish, mine, fight, complete quests,
 | 
						|
    befriend villagers, and uncover dark secrets.
 | 
						|
    """
 | 
						|
    game = "Stardew Valley"
 | 
						|
    topology_present = False
 | 
						|
 | 
						|
    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()}
 | 
						|
 | 
						|
    data_version = 3
 | 
						|
    required_client_version = (0, 4, 0)
 | 
						|
 | 
						|
    options_dataclass = StardewValleyOptions
 | 
						|
    options: StardewValleyOptions
 | 
						|
    logic: StardewLogic
 | 
						|
 | 
						|
    web = StardewWebWorld()
 | 
						|
    modified_bundles: Dict[str, Bundle]
 | 
						|
    randomized_entrances: Dict[str, str]
 | 
						|
    all_progression_items: Set[str]
 | 
						|
 | 
						|
    def __init__(self, world: MultiWorld, player: int):
 | 
						|
        super().__init__(world, player)
 | 
						|
        self.all_progression_items = set()
 | 
						|
        self.filler_item_pool_names = []
 | 
						|
 | 
						|
    def generate_early(self):
 | 
						|
        self.force_change_options_if_incompatible()
 | 
						|
 | 
						|
        self.logic = StardewLogic(self.player, self.options)
 | 
						|
        self.modified_bundles = get_all_bundles(self.multiworld.random,
 | 
						|
                                                self.logic,
 | 
						|
                                                self.options.bundle_randomization,
 | 
						|
                                                self.options.bundle_price)
 | 
						|
 | 
						|
    def force_change_options_if_incompatible(self):
 | 
						|
        goal_is_walnut_hunter = self.options.goal == Goal.option_greatest_walnut_hunter
 | 
						|
        goal_is_perfection = self.options.goal == Goal.option_perfection
 | 
						|
        goal_is_island_related = goal_is_walnut_hunter or goal_is_perfection
 | 
						|
        exclude_ginger_island = self.options.exclude_ginger_island == ExcludeGingerIsland.option_true
 | 
						|
        if goal_is_island_related and exclude_ginger_island:
 | 
						|
            self.options.exclude_ginger_island.value = ExcludeGingerIsland.option_false
 | 
						|
            goal_name = self.options.goal.current_key
 | 
						|
            player_name = self.multiworld.player_name[self.player]
 | 
						|
            logging.warning(f"Goal '{goal_name}' requires Ginger Island. Exclude Ginger Island setting forced to 'False' for player {self.player} ({player_name})")
 | 
						|
 | 
						|
    def create_regions(self):
 | 
						|
        def create_region(name: str, exits: Iterable[str]) -> Region:
 | 
						|
            region = Region(name, self.player, self.multiworld)
 | 
						|
            region.exits = [Entrance(self.player, exit_name, region) for exit_name in exits]
 | 
						|
            return region
 | 
						|
 | 
						|
        world_regions, self.randomized_entrances = create_regions(create_region, self.multiworld.random, self.options)
 | 
						|
 | 
						|
        def add_location(name: str, code: Optional[int], region: str):
 | 
						|
            region = world_regions[region]
 | 
						|
            location = StardewLocation(self.player, name, code, region)
 | 
						|
            location.access_rule = lambda _: True
 | 
						|
            region.locations.append(location)
 | 
						|
 | 
						|
        create_locations(add_location, self.options, self.multiworld.random)
 | 
						|
        self.multiworld.regions.extend(world_regions.values())
 | 
						|
 | 
						|
    def create_items(self):
 | 
						|
        self.precollect_starting_season()
 | 
						|
        items_to_exclude = [excluded_items
 | 
						|
                            for excluded_items in self.multiworld.precollected_items[self.player]
 | 
						|
                            if not item_table[excluded_items.name].has_any_group(Group.RESOURCE_PACK,
 | 
						|
                                                                                 Group.FRIENDSHIP_PACK)]
 | 
						|
 | 
						|
        if self.options.season_randomization == SeasonRandomization.option_disabled:
 | 
						|
            items_to_exclude = [item for item in items_to_exclude
 | 
						|
                                if item_table[item.name] not in items_by_group[Group.SEASON]]
 | 
						|
 | 
						|
        locations_count = len([location
 | 
						|
                               for location in self.multiworld.get_locations(self.player)
 | 
						|
                               if not location.event])
 | 
						|
 | 
						|
        created_items = create_items(self.create_item, locations_count, items_to_exclude, self.options,
 | 
						|
                                     self.multiworld.random)
 | 
						|
 | 
						|
        self.multiworld.itempool += created_items
 | 
						|
 | 
						|
        self.setup_early_items()
 | 
						|
        self.setup_month_events()
 | 
						|
        self.setup_victory()
 | 
						|
 | 
						|
    def precollect_starting_season(self) -> Optional[StardewItem]:
 | 
						|
        if self.options.season_randomization == SeasonRandomization.option_progressive:
 | 
						|
            return
 | 
						|
 | 
						|
        season_pool = items_by_group[Group.SEASON]
 | 
						|
 | 
						|
        if self.options.season_randomization == SeasonRandomization.option_disabled:
 | 
						|
            for season in season_pool:
 | 
						|
                self.multiworld.push_precollected(self.create_item(season))
 | 
						|
            return
 | 
						|
 | 
						|
        if [item for item in self.multiworld.precollected_items[self.player]
 | 
						|
            if item.name in {season.name for season in items_by_group[Group.SEASON]}]:
 | 
						|
            return
 | 
						|
 | 
						|
        if self.options.season_randomization == SeasonRandomization.option_randomized_not_winter:
 | 
						|
            season_pool = [season for season in season_pool if season.name != "Winter"]
 | 
						|
 | 
						|
        starting_season = self.create_item(self.multiworld.random.choice(season_pool))
 | 
						|
        self.multiworld.push_precollected(starting_season)
 | 
						|
 | 
						|
    def setup_early_items(self):
 | 
						|
        if (self.options.building_progression ==
 | 
						|
                BuildingProgression.option_progressive_early_shipping_bin):
 | 
						|
            self.multiworld.early_items[self.player]["Shipping Bin"] = 1
 | 
						|
 | 
						|
        if self.options.backpack_progression == BackpackProgression.option_early_progressive:
 | 
						|
            self.multiworld.early_items[self.player]["Progressive Backpack"] = 1
 | 
						|
 | 
						|
    def setup_month_events(self):
 | 
						|
        for i in range(0, MAX_MONTHS):
 | 
						|
            month_end = LocationData(None, "Stardew Valley", f"Month End {i + 1}")
 | 
						|
            if i == 0:
 | 
						|
                self.create_event_location(month_end, True_(), "Month End")
 | 
						|
                continue
 | 
						|
 | 
						|
            self.create_event_location(month_end, self.logic.received("Month End", i).simplify(), "Month End")
 | 
						|
 | 
						|
    def setup_victory(self):
 | 
						|
        if self.options.goal == Goal.option_community_center:
 | 
						|
            self.create_event_location(location_table[GoalName.community_center],
 | 
						|
                                       self.logic.can_complete_community_center().simplify(),
 | 
						|
                                       "Victory")
 | 
						|
        elif self.options.goal == Goal.option_grandpa_evaluation:
 | 
						|
            self.create_event_location(location_table[GoalName.grandpa_evaluation],
 | 
						|
                                       self.logic.can_finish_grandpa_evaluation().simplify(),
 | 
						|
                                       "Victory")
 | 
						|
        elif self.options.goal == Goal.option_bottom_of_the_mines:
 | 
						|
            self.create_event_location(location_table[GoalName.bottom_of_the_mines],
 | 
						|
                                       self.logic.can_mine_to_floor(120).simplify(),
 | 
						|
                                       "Victory")
 | 
						|
        elif self.options.goal == Goal.option_cryptic_note:
 | 
						|
            self.create_event_location(location_table[GoalName.cryptic_note],
 | 
						|
                                       self.logic.can_complete_quest("Cryptic Note").simplify(),
 | 
						|
                                       "Victory")
 | 
						|
        elif self.options.goal == Goal.option_master_angler:
 | 
						|
            self.create_event_location(location_table[GoalName.master_angler],
 | 
						|
                                       self.logic.can_catch_every_fish().simplify(),
 | 
						|
                                       "Victory")
 | 
						|
        elif self.options.goal == Goal.option_complete_collection:
 | 
						|
            self.create_event_location(location_table[GoalName.complete_museum],
 | 
						|
                                       self.logic.can_complete_museum().simplify(),
 | 
						|
                                       "Victory")
 | 
						|
        elif self.options.goal == Goal.option_full_house:
 | 
						|
            self.create_event_location(location_table[GoalName.full_house],
 | 
						|
                                       (self.logic.has_children(2) & self.logic.can_reproduce()).simplify(),
 | 
						|
                                       "Victory")
 | 
						|
        elif self.options.goal == Goal.option_greatest_walnut_hunter:
 | 
						|
            self.create_event_location(location_table[GoalName.greatest_walnut_hunter],
 | 
						|
                                       self.logic.has_walnut(130).simplify(),
 | 
						|
                                       "Victory")
 | 
						|
        elif self.options.goal == Goal.option_perfection:
 | 
						|
            self.create_event_location(location_table[GoalName.perfection],
 | 
						|
                                       self.logic.has_everything(self.all_progression_items).simplify(),
 | 
						|
                                       "Victory")
 | 
						|
 | 
						|
        self.multiworld.completion_condition[self.player] = lambda state: state.has("Victory", self.player)
 | 
						|
 | 
						|
    def create_item(self, item: Union[str, ItemData]) -> StardewItem:
 | 
						|
        if isinstance(item, str):
 | 
						|
            item = item_table[item]
 | 
						|
 | 
						|
        if item.classification == ItemClassification.progression:
 | 
						|
            self.all_progression_items.add(item.name)
 | 
						|
        return StardewItem(item.name, item.classification, item.code, self.player)
 | 
						|
 | 
						|
    def create_event_location(self, location_data: LocationData, rule: StardewRule, item: Optional[str] = None):
 | 
						|
        if item is None:
 | 
						|
            item = location_data.name
 | 
						|
 | 
						|
        region = self.multiworld.get_region(location_data.region, self.player)
 | 
						|
        location = StardewLocation(self.player, location_data.name, None, region)
 | 
						|
        location.access_rule = rule
 | 
						|
        region.locations.append(location)
 | 
						|
        location.place_locked_item(self.create_item(item))
 | 
						|
 | 
						|
    def set_rules(self):
 | 
						|
        set_rules(self)
 | 
						|
        self.force_first_month_once_all_early_items_are_found()
 | 
						|
 | 
						|
    def force_first_month_once_all_early_items_are_found(self):
 | 
						|
        """
 | 
						|
        The Fill algorithm sweeps all event when calculating the early location. This causes an issue where
 | 
						|
        location only locked behind event are considered early, which they are not really...
 | 
						|
 | 
						|
        This patches the issue, by adding a dependency to the first month end on all early items, so all the locations
 | 
						|
        that depends on it will not be considered early. This requires at least one early item to be progression, or
 | 
						|
        it just won't work...
 | 
						|
        """
 | 
						|
 | 
						|
        early_items = []
 | 
						|
        for player, item_count in self.multiworld.early_items.items():
 | 
						|
            for item, count in item_count.items():
 | 
						|
                if self.multiworld.worlds[player].create_item(item).advancement:
 | 
						|
                    early_items.append((player, item, count))
 | 
						|
 | 
						|
        for item, count in self.multiworld.local_early_items[self.player].items():
 | 
						|
            if self.create_item(item).advancement:
 | 
						|
                early_items.append((self.player, item, count))
 | 
						|
 | 
						|
        def first_month_require_all_early_items(state: CollectionState) -> bool:
 | 
						|
            for player, item, count in early_items:
 | 
						|
                if not state.has(item, player, count):
 | 
						|
                    return False
 | 
						|
 | 
						|
            return True
 | 
						|
 | 
						|
        first_month_end = self.multiworld.get_location("Month End 1", self.player)
 | 
						|
        set_rule(first_month_end, first_month_require_all_early_items)
 | 
						|
 | 
						|
    def generate_basic(self):
 | 
						|
        pass
 | 
						|
 | 
						|
    def get_filler_item_name(self) -> str:
 | 
						|
        if not self.filler_item_pool_names:
 | 
						|
            self.generate_filler_item_pool_names()
 | 
						|
        return self.random.choice(self.filler_item_pool_names)
 | 
						|
 | 
						|
    def generate_filler_item_pool_names(self):
 | 
						|
        include_traps, exclude_island = self.get_filler_item_rules()
 | 
						|
        available_filler = get_all_filler_items(include_traps, exclude_island)
 | 
						|
        available_filler = remove_limited_amount_packs(available_filler)
 | 
						|
        self.filler_item_pool_names = [item.name for item in available_filler]
 | 
						|
 | 
						|
    def get_filler_item_rules(self):
 | 
						|
        if self.player in self.multiworld.groups:
 | 
						|
            link_group: ItemLinkGroup = self.multiworld.groups[self.player]
 | 
						|
            include_traps = True
 | 
						|
            exclude_island = False
 | 
						|
            for player in link_group["players"]:
 | 
						|
                player_options = self.multiworld.worlds[player].options
 | 
						|
                if self.multiworld.game[player] != self.game:
 | 
						|
 | 
						|
                    continue
 | 
						|
                if player_options.trap_items == TrapItems.option_no_traps:
 | 
						|
                    include_traps = False
 | 
						|
                if player_options.exclude_ginger_island == ExcludeGingerIsland.option_true:
 | 
						|
                    exclude_island = True
 | 
						|
            return include_traps, exclude_island
 | 
						|
        else:
 | 
						|
            return self.options.trap_items != TrapItems.option_no_traps, self.options.exclude_ginger_island == ExcludeGingerIsland.option_true
 | 
						|
 | 
						|
    def fill_slot_data(self) -> Dict[str, Any]:
 | 
						|
 | 
						|
        modified_bundles = {}
 | 
						|
        for bundle_key in self.modified_bundles:
 | 
						|
            key, value = self.modified_bundles[bundle_key].to_pair()
 | 
						|
            modified_bundles[key] = value
 | 
						|
 | 
						|
        excluded_options = [BundleRandomization, BundlePrice, NumberOfMovementBuffs, NumberOfLuckBuffs]
 | 
						|
        excluded_option_names = [option.internal_name for option in excluded_options]
 | 
						|
        generic_option_names = [option_name for option_name in PerGameCommonOptions.type_hints]
 | 
						|
        excluded_option_names.extend(generic_option_names)
 | 
						|
        included_option_names: List[str] = [option_name for option_name in self.options_dataclass.type_hints if option_name not in excluded_option_names]
 | 
						|
        slot_data = self.options.as_dict(*included_option_names)
 | 
						|
        slot_data.update({
 | 
						|
            "seed": self.multiworld.per_slot_randoms[self.player].randrange(1000000000),  # Seed should be max 9 digits
 | 
						|
            "randomized_entrances": self.randomized_entrances,
 | 
						|
            "modified_bundles": modified_bundles,
 | 
						|
            "client_version": "4.0.0",
 | 
						|
        })
 | 
						|
 | 
						|
        return slot_data
 |