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
 | 
