136 lines
7.4 KiB
Python
136 lines
7.4 KiB
Python
|
from functools import cached_property
|
||
|
|
from typing import Union
|
||
|
|
|
||
|
|
from .ability_logic import AbilityLogicMixin
|
||
|
|
from .base_logic import BaseLogic, BaseLogicMixin
|
||
|
|
from .combat_logic import CombatLogicMixin
|
||
|
|
from .has_logic import HasLogicMixin
|
||
|
|
from .received_logic import ReceivedLogicMixin
|
||
|
|
from .region_logic import RegionLogicMixin
|
||
|
|
from ..strings.ap_names.event_names import Event
|
||
|
|
from ..options import ExcludeGingerIsland, Walnutsanity
|
||
|
|
from ..stardew_rule import StardewRule, False_, True_
|
||
|
|
from ..strings.ap_names.ap_option_names import OptionName
|
||
|
|
from ..strings.craftable_names import Furniture
|
||
|
|
from ..strings.crop_names import Fruit
|
||
|
|
from ..strings.metal_names import Mineral, Fossil
|
||
|
|
from ..strings.region_names import Region
|
||
|
|
from ..strings.seed_names import Seed
|
||
|
|
|
||
|
|
|
||
|
|
class WalnutLogicMixin(BaseLogicMixin):
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
super().__init__(*args, **kwargs)
|
||
|
|
self.walnut = WalnutLogic(*args, **kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
class WalnutLogic(BaseLogic[Union[WalnutLogicMixin, ReceivedLogicMixin, HasLogicMixin, RegionLogicMixin, CombatLogicMixin,
|
||
|
|
AbilityLogicMixin]]):
|
||
|
|
|
||
|
|
def has_walnut(self, number: int) -> StardewRule:
|
||
|
|
if self.options.exclude_ginger_island == ExcludeGingerIsland.option_true:
|
||
|
|
return False_()
|
||
|
|
if number <= 0:
|
||
|
|
return True_()
|
||
|
|
|
||
|
|
if self.options.walnutsanity == Walnutsanity.preset_none:
|
||
|
|
return self.can_get_walnuts(number)
|
||
|
|
if self.options.walnutsanity == Walnutsanity.preset_all:
|
||
|
|
return self.has_received_walnuts(number)
|
||
|
|
puzzle_walnuts = 61
|
||
|
|
bush_walnuts = 25
|
||
|
|
dig_walnuts = 18
|
||
|
|
repeatable_walnuts = 33
|
||
|
|
total_walnuts = puzzle_walnuts + bush_walnuts + dig_walnuts + repeatable_walnuts
|
||
|
|
walnuts_to_receive = 0
|
||
|
|
walnuts_to_collect = number
|
||
|
|
if OptionName.walnutsanity_puzzles in self.options.walnutsanity:
|
||
|
|
puzzle_walnut_rate = puzzle_walnuts / total_walnuts
|
||
|
|
puzzle_walnuts_required = round(puzzle_walnut_rate * number)
|
||
|
|
walnuts_to_receive += puzzle_walnuts_required
|
||
|
|
walnuts_to_collect -= puzzle_walnuts_required
|
||
|
|
if OptionName.walnutsanity_bushes in self.options.walnutsanity:
|
||
|
|
bush_walnuts_rate = bush_walnuts / total_walnuts
|
||
|
|
bush_walnuts_required = round(bush_walnuts_rate * number)
|
||
|
|
walnuts_to_receive += bush_walnuts_required
|
||
|
|
walnuts_to_collect -= bush_walnuts_required
|
||
|
|
if OptionName.walnutsanity_dig_spots in self.options.walnutsanity:
|
||
|
|
dig_walnuts_rate = dig_walnuts / total_walnuts
|
||
|
|
dig_walnuts_required = round(dig_walnuts_rate * number)
|
||
|
|
walnuts_to_receive += dig_walnuts_required
|
||
|
|
walnuts_to_collect -= dig_walnuts_required
|
||
|
|
if OptionName.walnutsanity_repeatables in self.options.walnutsanity:
|
||
|
|
repeatable_walnuts_rate = repeatable_walnuts / total_walnuts
|
||
|
|
repeatable_walnuts_required = round(repeatable_walnuts_rate * number)
|
||
|
|
walnuts_to_receive += repeatable_walnuts_required
|
||
|
|
walnuts_to_collect -= repeatable_walnuts_required
|
||
|
|
return self.has_received_walnuts(walnuts_to_receive) & self.can_get_walnuts(walnuts_to_collect)
|
||
|
|
|
||
|
|
def has_received_walnuts(self, number: int) -> StardewRule:
|
||
|
|
return self.logic.received(Event.received_walnuts, number)
|
||
|
|
|
||
|
|
def can_get_walnuts(self, number: int) -> StardewRule:
|
||
|
|
# https://stardewcommunitywiki.com/Golden_Walnut#Walnut_Locations
|
||
|
|
reach_south = self.logic.region.can_reach(Region.island_south)
|
||
|
|
reach_north = self.logic.region.can_reach(Region.island_north)
|
||
|
|
reach_west = self.logic.region.can_reach(Region.island_west)
|
||
|
|
reach_hut = self.logic.region.can_reach(Region.leo_hut)
|
||
|
|
reach_southeast = self.logic.region.can_reach(Region.island_south_east)
|
||
|
|
reach_field_office = self.logic.region.can_reach(Region.field_office)
|
||
|
|
reach_pirate_cove = self.logic.region.can_reach(Region.pirate_cove)
|
||
|
|
reach_outside_areas = self.logic.and_(reach_south, reach_north, reach_west, reach_hut)
|
||
|
|
reach_volcano_regions = [self.logic.region.can_reach(Region.volcano),
|
||
|
|
self.logic.region.can_reach(Region.volcano_secret_beach),
|
||
|
|
self.logic.region.can_reach(Region.volcano_floor_5),
|
||
|
|
self.logic.region.can_reach(Region.volcano_floor_10)]
|
||
|
|
reach_volcano = self.logic.or_(*reach_volcano_regions)
|
||
|
|
reach_all_volcano = self.logic.and_(*reach_volcano_regions)
|
||
|
|
reach_walnut_regions = [reach_south, reach_north, reach_west, reach_volcano, reach_field_office]
|
||
|
|
reach_caves = self.logic.and_(self.logic.region.can_reach(Region.qi_walnut_room), self.logic.region.can_reach(Region.dig_site),
|
||
|
|
self.logic.region.can_reach(Region.gourmand_frog_cave),
|
||
|
|
self.logic.region.can_reach(Region.colored_crystals_cave),
|
||
|
|
self.logic.region.can_reach(Region.shipwreck), self.logic.combat.has_slingshot)
|
||
|
|
reach_entire_island = self.logic.and_(reach_outside_areas, reach_all_volcano,
|
||
|
|
reach_caves, reach_southeast, reach_field_office, reach_pirate_cove)
|
||
|
|
if number <= 5:
|
||
|
|
return self.logic.or_(reach_south, reach_north, reach_west, reach_volcano)
|
||
|
|
if number <= 10:
|
||
|
|
return self.logic.count(2, *reach_walnut_regions)
|
||
|
|
if number <= 15:
|
||
|
|
return self.logic.count(3, *reach_walnut_regions)
|
||
|
|
if number <= 20:
|
||
|
|
return self.logic.and_(*reach_walnut_regions)
|
||
|
|
if number <= 50:
|
||
|
|
return reach_entire_island
|
||
|
|
gems = (Mineral.amethyst, Mineral.aquamarine, Mineral.emerald, Mineral.ruby, Mineral.topaz)
|
||
|
|
return reach_entire_island & self.logic.has(Fruit.banana) & self.logic.has_all(*gems) & \
|
||
|
|
self.logic.ability.can_mine_perfectly() & self.logic.ability.can_fish_perfectly() & \
|
||
|
|
self.logic.has(Furniture.flute_block) & self.logic.has(Seed.melon) & self.logic.has(Seed.wheat) & \
|
||
|
|
self.logic.has(Seed.garlic) & self.can_complete_field_office()
|
||
|
|
|
||
|
|
@cached_property
|
||
|
|
def can_start_field_office(self) -> StardewRule:
|
||
|
|
field_office = self.logic.region.can_reach(Region.field_office)
|
||
|
|
professor_snail = self.logic.received("Open Professor Snail Cave")
|
||
|
|
return field_office & professor_snail
|
||
|
|
|
||
|
|
def can_complete_large_animal_collection(self) -> StardewRule:
|
||
|
|
fossils = self.logic.has_all(Fossil.fossilized_leg, Fossil.fossilized_ribs, Fossil.fossilized_skull, Fossil.fossilized_spine, Fossil.fossilized_tail)
|
||
|
|
return self.can_start_field_office & fossils
|
||
|
|
|
||
|
|
def can_complete_snake_collection(self) -> StardewRule:
|
||
|
|
fossils = self.logic.has_all(Fossil.snake_skull, Fossil.snake_vertebrae)
|
||
|
|
return self.can_start_field_office & fossils
|
||
|
|
|
||
|
|
def can_complete_frog_collection(self) -> StardewRule:
|
||
|
|
fossils = self.logic.has_all(Fossil.mummified_frog)
|
||
|
|
return self.can_start_field_office & fossils
|
||
|
|
|
||
|
|
def can_complete_bat_collection(self) -> StardewRule:
|
||
|
|
fossils = self.logic.has_all(Fossil.mummified_bat)
|
||
|
|
return self.can_start_field_office & fossils
|
||
|
|
|
||
|
|
def can_complete_field_office(self) -> StardewRule:
|
||
|
|
return self.can_complete_large_animal_collection() & self.can_complete_snake_collection() & \
|
||
|
|
self.can_complete_frog_collection() & self.can_complete_bat_collection()
|