280 lines
18 KiB
Python
280 lines
18 KiB
Python
import itertools
|
|
from typing import Dict, List
|
|
|
|
from BaseClasses import MultiWorld
|
|
from worlds.generic import Rules as MultiWorldRules
|
|
from . import options, locations
|
|
from .bundles import Bundle
|
|
from .data.museum_data import all_museum_items, all_mineral_items, all_artifact_items, \
|
|
dwarf_scrolls, skeleton_front, \
|
|
skeleton_middle, skeleton_back, all_museum_items_by_name
|
|
from .locations import LocationTags
|
|
from .logic import StardewLogic, And, month_end_per_skill_level, tool_prices, week_days
|
|
from .options import StardewOptions
|
|
|
|
|
|
def set_rules(multi_world: MultiWorld, player: int, world_options: StardewOptions, logic: StardewLogic,
|
|
current_bundles: Dict[str, Bundle]):
|
|
all_location_names = list(location.name for location in multi_world.get_locations(player))
|
|
|
|
for floor in range(5, 120 + 5, 5):
|
|
MultiWorldRules.set_rule(multi_world.get_entrance(f"Dig to The Mines - Floor {floor}", player),
|
|
logic.can_mine_to_floor(floor).simplify())
|
|
|
|
MultiWorldRules.set_rule(multi_world.get_entrance("Enter Tide Pools", player),
|
|
logic.received("Beach Bridge").simplify())
|
|
MultiWorldRules.set_rule(multi_world.get_entrance("Enter Quarry", player),
|
|
logic.received("Bridge Repair").simplify())
|
|
MultiWorldRules.set_rule(multi_world.get_entrance("Enter Secret Woods", player),
|
|
logic.has_tool("Axe", "Iron").simplify())
|
|
MultiWorldRules.set_rule(multi_world.get_entrance("Forest to Sewers", player),
|
|
logic.has_rusty_key().simplify())
|
|
MultiWorldRules.set_rule(multi_world.get_entrance("Town to Sewers", player),
|
|
logic.has_rusty_key().simplify())
|
|
MultiWorldRules.set_rule(multi_world.get_entrance("Take Bus to Desert", player),
|
|
logic.received("Bus Repair").simplify())
|
|
MultiWorldRules.set_rule(multi_world.get_entrance("Enter Skull Cavern", player),
|
|
logic.received("Skull Key").simplify())
|
|
MultiWorldRules.set_rule(multi_world.get_entrance("Mine to Skull Cavern Floor 100", player),
|
|
logic.can_mine_perfectly_in_the_skull_cavern().simplify())
|
|
|
|
MultiWorldRules.set_rule(multi_world.get_entrance("Use Desert Obelisk", player),
|
|
logic.received("Desert Obelisk").simplify())
|
|
MultiWorldRules.set_rule(multi_world.get_entrance("Use Island Obelisk", player),
|
|
logic.received("Island Obelisk").simplify())
|
|
MultiWorldRules.set_rule(multi_world.get_entrance("Talk to Traveling Merchant", player),
|
|
logic.has_traveling_merchant())
|
|
MultiWorldRules.set_rule(multi_world.get_entrance("Enter Greenhouse", player),
|
|
logic.received("Greenhouse"))
|
|
|
|
# Those checks do not exist if ToolProgression is vanilla
|
|
if world_options[options.ToolProgression] != options.ToolProgression.option_vanilla:
|
|
MultiWorldRules.add_rule(multi_world.get_location("Purchase Fiberglass Rod", player),
|
|
(logic.has_skill_level("Fishing", 2) & logic.can_spend_money(1800)).simplify())
|
|
MultiWorldRules.add_rule(multi_world.get_location("Purchase Iridium Rod", player),
|
|
(logic.has_skill_level("Fishing", 6) & logic.can_spend_money(7500)).simplify())
|
|
|
|
materials = [None, "Copper", "Iron", "Gold", "Iridium"]
|
|
tool = ["Hoe", "Pickaxe", "Axe", "Watering Can", "Trash Can"]
|
|
for (previous, material), tool in itertools.product(zip(materials[:4], materials[1:]), tool):
|
|
if previous is None:
|
|
MultiWorldRules.add_rule(multi_world.get_location(f"{material} {tool} Upgrade", player),
|
|
(logic.has(f"{material} Ore") &
|
|
logic.can_spend_money(tool_prices[material])).simplify())
|
|
else:
|
|
MultiWorldRules.add_rule(multi_world.get_location(f"{material} {tool} Upgrade", player),
|
|
(logic.has(f"{material} Ore") & logic.has_tool(tool, previous) &
|
|
logic.can_spend_money(tool_prices[material])).simplify())
|
|
|
|
# Skills
|
|
if world_options[options.SkillProgression] != options.SkillProgression.option_vanilla:
|
|
for i in range(1, 11):
|
|
MultiWorldRules.set_rule(multi_world.get_location(f"Level {i} Farming", player),
|
|
(logic.received("Month End", month_end_per_skill_level["Farming", i])).simplify())
|
|
MultiWorldRules.set_rule(multi_world.get_location(f"Level {i} Fishing", player),
|
|
(logic.can_get_fishing_xp() &
|
|
logic.received("Month End", month_end_per_skill_level["Fishing", i])).simplify())
|
|
MultiWorldRules.add_rule(multi_world.get_location(f"Level {i} Foraging", player),
|
|
logic.received("Month End", month_end_per_skill_level["Foraging", i]).simplify())
|
|
if i >= 6:
|
|
MultiWorldRules.add_rule(multi_world.get_location(f"Level {i} Foraging", player),
|
|
logic.has_tool("Axe", "Iron").simplify())
|
|
MultiWorldRules.set_rule(multi_world.get_location(f"Level {i} Mining", player),
|
|
logic.received("Month End", month_end_per_skill_level["Mining", i]).simplify())
|
|
MultiWorldRules.set_rule(multi_world.get_location(f"Level {i} Combat", player),
|
|
(logic.received("Month End", month_end_per_skill_level["Combat", i]) &
|
|
logic.has_any_weapon()).simplify())
|
|
|
|
# Bundles
|
|
for bundle in current_bundles.values():
|
|
location = multi_world.get_location(bundle.get_name_with_bundle(), player)
|
|
rules = logic.can_complete_bundle(bundle.requirements, bundle.number_required)
|
|
simplified_rules = rules.simplify()
|
|
MultiWorldRules.set_rule(location, simplified_rules)
|
|
MultiWorldRules.add_rule(multi_world.get_location("Complete Crafts Room", player),
|
|
And(logic.can_reach_location(bundle.name)
|
|
for bundle in locations.locations_by_tag[LocationTags.CRAFTS_ROOM_BUNDLE]).simplify())
|
|
MultiWorldRules.add_rule(multi_world.get_location("Complete Pantry", player),
|
|
And(logic.can_reach_location(bundle.name)
|
|
for bundle in locations.locations_by_tag[LocationTags.PANTRY_BUNDLE]).simplify())
|
|
MultiWorldRules.add_rule(multi_world.get_location("Complete Fish Tank", player),
|
|
And(logic.can_reach_location(bundle.name)
|
|
for bundle in locations.locations_by_tag[LocationTags.FISH_TANK_BUNDLE]).simplify())
|
|
MultiWorldRules.add_rule(multi_world.get_location("Complete Boiler Room", player),
|
|
And(logic.can_reach_location(bundle.name)
|
|
for bundle in locations.locations_by_tag[LocationTags.BOILER_ROOM_BUNDLE]).simplify())
|
|
MultiWorldRules.add_rule(multi_world.get_location("Complete Bulletin Board", player),
|
|
And(logic.can_reach_location(bundle.name)
|
|
for bundle
|
|
in locations.locations_by_tag[LocationTags.BULLETIN_BOARD_BUNDLE]).simplify())
|
|
MultiWorldRules.add_rule(multi_world.get_location("Complete Vault", player),
|
|
And(logic.can_reach_location(bundle.name)
|
|
for bundle in locations.locations_by_tag[LocationTags.VAULT_BUNDLE]).simplify())
|
|
|
|
# Buildings
|
|
if world_options[options.BuildingProgression] != options.BuildingProgression.option_vanilla:
|
|
for building in locations.locations_by_tag[LocationTags.BUILDING_BLUEPRINT]:
|
|
MultiWorldRules.set_rule(multi_world.get_location(building.name, player),
|
|
logic.building_rules[building.name.replace(" Blueprint", "")].simplify())
|
|
|
|
# Story Quests
|
|
for quest in locations.locations_by_tag[LocationTags.QUEST]:
|
|
MultiWorldRules.set_rule(multi_world.get_location(quest.name, player),
|
|
logic.quest_rules[quest.name].simplify())
|
|
|
|
# Help Wanted Quests
|
|
desired_number_help_wanted: int = world_options[options.HelpWantedLocations] // 7
|
|
for i in range(0, desired_number_help_wanted):
|
|
prefix = "Help Wanted:"
|
|
delivery = "Item Delivery"
|
|
rule = logic.received("Month End", i)
|
|
fishing_rule = rule & logic.can_fish()
|
|
slay_rule = rule & logic.has_any_weapon()
|
|
item_delivery_index = (i * 4) + 1
|
|
for j in range(item_delivery_index, item_delivery_index + 4):
|
|
location_name = f"{prefix} {delivery} {j}"
|
|
MultiWorldRules.set_rule(multi_world.get_location(location_name, player), rule.simplify())
|
|
|
|
MultiWorldRules.set_rule(multi_world.get_location(f"{prefix} Gathering {i+1}", player),
|
|
rule.simplify())
|
|
MultiWorldRules.set_rule(multi_world.get_location(f"{prefix} Fishing {i+1}", player),
|
|
fishing_rule.simplify())
|
|
MultiWorldRules.set_rule(multi_world.get_location(f"{prefix} Slay Monsters {i+1}", player),
|
|
slay_rule.simplify())
|
|
|
|
set_fishsanity_rules(all_location_names, logic, multi_world, player)
|
|
set_museumsanity_rules(all_location_names, logic, multi_world, player, world_options)
|
|
set_friendsanity_rules(all_location_names, logic, multi_world, player)
|
|
set_backpack_rules(logic, multi_world, player, world_options)
|
|
|
|
MultiWorldRules.add_rule(multi_world.get_location("Old Master Cannoli", player),
|
|
logic.has("Sweet Gem Berry").simplify())
|
|
MultiWorldRules.add_rule(multi_world.get_location("Galaxy Sword Shrine", player),
|
|
logic.has("Prismatic Shard").simplify())
|
|
|
|
set_traveling_merchant_rules(logic, multi_world, player)
|
|
set_arcade_machine_rules(logic, multi_world, player, world_options)
|
|
|
|
|
|
def set_fishsanity_rules(all_location_names: List[str], logic: StardewLogic, multi_world: MultiWorld, player: int):
|
|
fish_prefix = "Fishsanity: "
|
|
for fish_location in locations.locations_by_tag[LocationTags.FISHSANITY]:
|
|
if fish_location.name in all_location_names:
|
|
fish_name = fish_location.name[len(fish_prefix):]
|
|
MultiWorldRules.set_rule(multi_world.get_location(fish_location.name, player),
|
|
logic.has(fish_name).simplify())
|
|
|
|
|
|
def set_museumsanity_rules(all_location_names: List[str], logic: StardewLogic, multi_world: MultiWorld, player: int,
|
|
world_options: StardewOptions):
|
|
museum_prefix = "Museumsanity: "
|
|
if world_options[options.Museumsanity] == options.Museumsanity.option_milestones:
|
|
for museum_milestone in locations.locations_by_tag[LocationTags.MUSEUM_MILESTONES]:
|
|
set_museum_milestone_rule(logic, multi_world, museum_milestone, museum_prefix, player)
|
|
elif world_options[options.Museumsanity] != options.Museumsanity.option_none:
|
|
set_museum_individual_donations_rules(all_location_names, logic, multi_world, museum_prefix, player)
|
|
|
|
|
|
def set_museum_individual_donations_rules(all_location_names, logic, multi_world, museum_prefix, player):
|
|
all_donations = sorted(locations.locations_by_tag[LocationTags.MUSEUM_DONATIONS],
|
|
key=lambda x: all_museum_items_by_name[x.name[len(museum_prefix):]].difficulty, reverse=True)
|
|
counter = 0
|
|
number_donations = len(all_donations)
|
|
for museum_location in all_donations:
|
|
if museum_location.name in all_location_names:
|
|
donation_name = museum_location.name[len(museum_prefix):]
|
|
required_detectors = counter * 5 // number_donations
|
|
rule = logic.has(donation_name) & logic.received("Traveling Merchant Metal Detector", required_detectors)
|
|
MultiWorldRules.set_rule(multi_world.get_location(museum_location.name, player),
|
|
rule.simplify())
|
|
counter += 1
|
|
|
|
|
|
def set_museum_milestone_rule(logic: StardewLogic, multi_world: MultiWorld, museum_milestone, museum_prefix: str,
|
|
player: int):
|
|
milestone_name = museum_milestone.name[len(museum_prefix):]
|
|
donations_suffix = " Donations"
|
|
minerals_suffix = " Minerals"
|
|
artifacts_suffix = " Artifacts"
|
|
metal_detector = "Traveling Merchant Metal Detector"
|
|
rule = None
|
|
if milestone_name.endswith(donations_suffix):
|
|
rule = get_museum_item_count_rule(logic, donations_suffix, milestone_name, all_museum_items)
|
|
elif milestone_name.endswith(minerals_suffix):
|
|
rule = get_museum_item_count_rule(logic, minerals_suffix, milestone_name, all_mineral_items)
|
|
elif milestone_name.endswith(artifacts_suffix):
|
|
rule = get_museum_item_count_rule(logic, artifacts_suffix, milestone_name, all_artifact_items)
|
|
elif milestone_name == "Dwarf Scrolls":
|
|
rule = logic.has([item.name for item in dwarf_scrolls]) & logic.received(metal_detector, 4)
|
|
elif milestone_name == "Skeleton Front":
|
|
rule = logic.has([item.name for item in skeleton_front]) & logic.received(metal_detector, 4)
|
|
elif milestone_name == "Skeleton Middle":
|
|
rule = logic.has([item.name for item in skeleton_middle]) & logic.received(metal_detector, 4)
|
|
elif milestone_name == "Skeleton Back":
|
|
rule = logic.has([item.name for item in skeleton_back]) & logic.received(metal_detector, 4)
|
|
elif milestone_name == "Ancient Seed":
|
|
rule = logic.has("Ancient Seed") & logic.received(metal_detector, 4)
|
|
if rule is None:
|
|
return
|
|
MultiWorldRules.set_rule(multi_world.get_location(museum_milestone.name, player), rule.simplify())
|
|
|
|
|
|
def get_museum_item_count_rule(logic, suffix, milestone_name, accepted_items):
|
|
metal_detector = "Traveling Merchant Metal Detector"
|
|
num = int(milestone_name[:milestone_name.index(suffix)])
|
|
required_detectors = (num - 1) * 5 // len(accepted_items)
|
|
rule = logic.has([item.name for item in accepted_items], num) & logic.received(metal_detector, required_detectors)
|
|
return rule
|
|
|
|
|
|
def set_backpack_rules(logic: StardewLogic, multi_world: MultiWorld, player: int, world_options):
|
|
if world_options[options.BackpackProgression] != options.BackpackProgression.option_vanilla:
|
|
MultiWorldRules.set_rule(multi_world.get_location("Large Pack", player),
|
|
logic.can_spend_money(2000).simplify())
|
|
MultiWorldRules.set_rule(multi_world.get_location("Deluxe Pack", player),
|
|
(logic.can_spend_money(10000) & logic.received("Progressive Backpack")).simplify())
|
|
|
|
|
|
def set_traveling_merchant_rules(logic: StardewLogic, multi_world: MultiWorld, player: int):
|
|
for day in week_days:
|
|
item_for_day = f"Traveling Merchant: {day}"
|
|
for i in range(1, 4):
|
|
location_name = f"Traveling Merchant {day} Item {i}"
|
|
MultiWorldRules.set_rule(multi_world.get_location(location_name, player),
|
|
logic.received(item_for_day))
|
|
|
|
|
|
def set_arcade_machine_rules(logic: StardewLogic, multi_world: MultiWorld, player: int, world_options):
|
|
if world_options[options.ArcadeMachineLocations] == options.ArcadeMachineLocations.option_full_shuffling:
|
|
MultiWorldRules.add_rule(multi_world.get_entrance("Play Junimo Kart", player),
|
|
(logic.received("Skull Key") & logic.has("Junimo Kart Small Buff")).simplify())
|
|
MultiWorldRules.add_rule(multi_world.get_entrance("Reach Junimo Kart 2", player),
|
|
logic.has("Junimo Kart Medium Buff").simplify())
|
|
MultiWorldRules.add_rule(multi_world.get_entrance("Reach Junimo Kart 3", player),
|
|
logic.has("Junimo Kart Big Buff").simplify())
|
|
MultiWorldRules.add_rule(multi_world.get_location("Junimo Kart: Sunset Speedway (Victory)", player),
|
|
logic.has("Junimo Kart Max Buff").simplify())
|
|
MultiWorldRules.add_rule(multi_world.get_entrance("Play Journey of the Prairie King", player),
|
|
logic.has("JotPK Small Buff").simplify())
|
|
MultiWorldRules.add_rule(multi_world.get_entrance("Reach JotPK World 2", player),
|
|
logic.has("JotPK Medium Buff").simplify())
|
|
MultiWorldRules.add_rule(multi_world.get_entrance("Reach JotPK World 3", player),
|
|
logic.has("JotPK Big Buff").simplify())
|
|
MultiWorldRules.add_rule(multi_world.get_location("Journey of the Prairie King Victory", player),
|
|
logic.has("JotPK Max Buff").simplify())
|
|
|
|
|
|
def set_friendsanity_rules(all_location_names: List[str], logic: StardewLogic, multi_world: MultiWorld, player: int):
|
|
friend_prefix = "Friendsanity: "
|
|
friend_suffix = " <3"
|
|
for friend_location in locations.locations_by_tag[LocationTags.FRIENDSANITY]:
|
|
if not friend_location.name in all_location_names:
|
|
continue
|
|
friend_location_without_prefix = friend_location.name[len(friend_prefix):]
|
|
friend_location_trimmed = friend_location_without_prefix[:friend_location_without_prefix.index(friend_suffix)]
|
|
parts = friend_location_trimmed.split(" ")
|
|
friend_name = parts[0]
|
|
num_hearts = int(parts[1])
|
|
MultiWorldRules.set_rule(multi_world.get_location(friend_location.name, player),
|
|
logic.can_earn_relationship(friend_name, num_hearts).simplify())
|