174 lines
11 KiB
Python
174 lines
11 KiB
Python
from dataclasses import dataclass
|
|
from typing import List, Tuple, Dict, Set, Callable
|
|
|
|
from ..mods.mod_data import ModNames
|
|
from ..mods.mod_monster_locations import modded_monsters_locations
|
|
from ..strings.monster_names import Monster, MonsterCategory
|
|
from ..strings.performance_names import Performance
|
|
from ..strings.region_names import Region
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class StardewMonster:
|
|
name: str
|
|
category: str
|
|
locations: Tuple[str]
|
|
difficulty: str
|
|
|
|
def __repr__(self):
|
|
return f"{self.name} [{self.category}] (Locations: {self.locations} |" \
|
|
f" Difficulty: {self.difficulty}) |"
|
|
|
|
|
|
slime_hutch = (Region.slime_hutch,)
|
|
mines_floor_20 = (Region.mines_floor_20,)
|
|
mines_floor_60 = (Region.mines_floor_60,)
|
|
mines_floor_100 = (Region.mines_floor_100,)
|
|
dangerous_mines_20 = (Region.dangerous_mines_20,)
|
|
dangerous_mines_60 = (Region.dangerous_mines_60,)
|
|
dangerous_mines_100 = (Region.dangerous_mines_100,)
|
|
quarry_mine = (Region.quarry_mine,)
|
|
mutant_bug_lair = (Region.mutant_bug_lair,)
|
|
skull_cavern = (Region.skull_cavern_25,)
|
|
skull_cavern_high = (Region.skull_cavern_75,)
|
|
skull_cavern_dangerous = (Region.dangerous_skull_cavern,)
|
|
tiger_slime_grove = (Region.island_west,)
|
|
volcano = (Region.volcano_floor_5,)
|
|
volcano_high = (Region.volcano_floor_10,)
|
|
|
|
all_monsters: List[StardewMonster] = []
|
|
monster_modifications_by_mod: Dict[str, Dict[str, Callable[[str, StardewMonster], StardewMonster]]] = {}
|
|
|
|
|
|
def create_monster(name: str, category: str, locations: Tuple[str, ...], difficulty: str) -> StardewMonster:
|
|
monster = StardewMonster(name, category, locations, difficulty)
|
|
all_monsters.append(monster)
|
|
return monster
|
|
|
|
|
|
def update_monster_locations(mod_name: str, monster: StardewMonster):
|
|
new_locations = modded_monsters_locations[mod_name][monster.name]
|
|
total_locations = tuple(sorted(set(monster.locations + new_locations)))
|
|
return StardewMonster(monster.name, monster.category, total_locations, monster.difficulty)
|
|
|
|
|
|
def register_monster_modification(mod_name: str, monster: StardewMonster, modification_function):
|
|
if mod_name not in monster_modifications_by_mod:
|
|
monster_modifications_by_mod[mod_name] = {}
|
|
monster_modifications_by_mod[mod_name][monster.name] = modification_function
|
|
|
|
|
|
green_slime = create_monster(Monster.green_slime, MonsterCategory.slime, mines_floor_20, Performance.basic)
|
|
blue_slime = create_monster(Monster.blue_slime, MonsterCategory.slime, mines_floor_60, Performance.decent)
|
|
red_slime = create_monster(Monster.red_slime, MonsterCategory.slime, mines_floor_100, Performance.good)
|
|
purple_slime = create_monster(Monster.purple_slime, MonsterCategory.slime, skull_cavern, Performance.great)
|
|
yellow_slime = create_monster(Monster.yellow_slime, MonsterCategory.slime, skull_cavern_high, Performance.galaxy)
|
|
black_slime = create_monster(Monster.black_slime, MonsterCategory.slime, slime_hutch, Performance.decent)
|
|
copper_slime = create_monster(Monster.copper_slime, MonsterCategory.slime, quarry_mine, Performance.decent)
|
|
iron_slime = create_monster(Monster.iron_slime, MonsterCategory.slime, quarry_mine, Performance.good)
|
|
tiger_slime = create_monster(Monster.tiger_slime, MonsterCategory.slime, tiger_slime_grove, Performance.galaxy)
|
|
|
|
shadow_shaman = create_monster(Monster.shadow_shaman, MonsterCategory.void_spirits, mines_floor_100, Performance.good)
|
|
shadow_shaman_dangerous = create_monster(Monster.shadow_shaman_dangerous, MonsterCategory.void_spirits, dangerous_mines_100, Performance.galaxy)
|
|
shadow_brute = create_monster(Monster.shadow_brute, MonsterCategory.void_spirits, mines_floor_100, Performance.good)
|
|
shadow_brute_dangerous = create_monster(Monster.shadow_brute_dangerous, MonsterCategory.void_spirits, dangerous_mines_100, Performance.galaxy)
|
|
shadow_sniper = create_monster(Monster.shadow_sniper, MonsterCategory.void_spirits, dangerous_mines_100, Performance.galaxy)
|
|
|
|
bat = create_monster(Monster.bat, MonsterCategory.bats, mines_floor_20, Performance.basic)
|
|
bat_dangerous = create_monster(Monster.bat_dangerous, MonsterCategory.bats, dangerous_mines_20, Performance.galaxy)
|
|
frost_bat = create_monster(Monster.frost_bat, MonsterCategory.bats, mines_floor_60, Performance.decent)
|
|
frost_bat_dangerous = create_monster(Monster.frost_bat_dangerous, MonsterCategory.bats, dangerous_mines_60, Performance.galaxy)
|
|
lava_bat = create_monster(Monster.lava_bat, MonsterCategory.bats, mines_floor_100, Performance.good)
|
|
iridium_bat = create_monster(Monster.iridium_bat, MonsterCategory.bats, skull_cavern_high, Performance.great)
|
|
|
|
skeleton = create_monster(Monster.skeleton, MonsterCategory.skeletons, mines_floor_100, Performance.good)
|
|
skeleton_dangerous = create_monster(Monster.skeleton_dangerous, MonsterCategory.skeletons, dangerous_mines_100, Performance.galaxy)
|
|
skeleton_mage = create_monster(Monster.skeleton_mage, MonsterCategory.skeletons, dangerous_mines_100, Performance.galaxy)
|
|
|
|
bug = create_monster(Monster.bug, MonsterCategory.cave_insects, mines_floor_20, Performance.basic)
|
|
bug_dangerous = create_monster(Monster.bug_dangerous, MonsterCategory.cave_insects, dangerous_mines_20, Performance.galaxy)
|
|
cave_fly = create_monster(Monster.cave_fly, MonsterCategory.cave_insects, mines_floor_20, Performance.basic)
|
|
cave_fly_dangerous = create_monster(Monster.cave_fly_dangerous, MonsterCategory.cave_insects, dangerous_mines_60, Performance.galaxy)
|
|
grub = create_monster(Monster.grub, MonsterCategory.cave_insects, mines_floor_20, Performance.basic)
|
|
grub_dangerous = create_monster(Monster.grub_dangerous, MonsterCategory.cave_insects, dangerous_mines_60, Performance.galaxy)
|
|
mutant_fly = create_monster(Monster.mutant_fly, MonsterCategory.cave_insects, mutant_bug_lair, Performance.good)
|
|
mutant_grub = create_monster(Monster.mutant_grub, MonsterCategory.cave_insects, mutant_bug_lair, Performance.good)
|
|
armored_bug = create_monster(Monster.armored_bug, MonsterCategory.cave_insects, skull_cavern, Performance.basic) # Requires 'Bug Killer' enchantment
|
|
armored_bug_dangerous = create_monster(Monster.armored_bug_dangerous, MonsterCategory.cave_insects, skull_cavern,
|
|
Performance.good) # Requires 'Bug Killer' enchantment
|
|
|
|
duggy = create_monster(Monster.duggy, MonsterCategory.duggies, mines_floor_20, Performance.basic)
|
|
duggy_dangerous = create_monster(Monster.duggy_dangerous, MonsterCategory.duggies, dangerous_mines_20, Performance.great)
|
|
magma_duggy = create_monster(Monster.magma_duggy, MonsterCategory.duggies, volcano, Performance.galaxy)
|
|
|
|
dust_sprite = create_monster(Monster.dust_sprite, MonsterCategory.dust_sprites, mines_floor_60, Performance.basic)
|
|
dust_sprite_dangerous = create_monster(Monster.dust_sprite_dangerous, MonsterCategory.dust_sprites, dangerous_mines_60, Performance.great)
|
|
|
|
rock_crab = create_monster(Monster.rock_crab, MonsterCategory.rock_crabs, mines_floor_20, Performance.basic)
|
|
rock_crab_dangerous = create_monster(Monster.rock_crab_dangerous, MonsterCategory.rock_crabs, dangerous_mines_20, Performance.great)
|
|
lava_crab = create_monster(Monster.lava_crab, MonsterCategory.rock_crabs, mines_floor_100, Performance.good)
|
|
lava_crab_dangerous = create_monster(Monster.lava_crab_dangerous, MonsterCategory.rock_crabs, dangerous_mines_100, Performance.galaxy)
|
|
iridium_crab = create_monster(Monster.iridium_crab, MonsterCategory.rock_crabs, skull_cavern, Performance.great)
|
|
|
|
mummy = create_monster(Monster.mummy, MonsterCategory.mummies, skull_cavern, Performance.great) # Requires bombs or "Crusader" enchantment
|
|
mummy_dangerous = create_monster(Monster.mummy_dangerous, MonsterCategory.mummies, skull_cavern_dangerous,
|
|
Performance.maximum) # Requires bombs or "Crusader" enchantment
|
|
|
|
pepper_rex = create_monster(Monster.pepper_rex, MonsterCategory.pepper_rex, skull_cavern, Performance.great)
|
|
|
|
serpent = create_monster(Monster.serpent, MonsterCategory.serpents, skull_cavern, Performance.galaxy)
|
|
royal_serpent = create_monster(Monster.royal_serpent, MonsterCategory.serpents, skull_cavern_dangerous, Performance.maximum)
|
|
|
|
magma_sprite = create_monster(Monster.magma_sprite, MonsterCategory.magma_sprites, volcano, Performance.galaxy)
|
|
magma_sparker = create_monster(Monster.magma_sparker, MonsterCategory.magma_sprites, volcano_high, Performance.galaxy)
|
|
|
|
register_monster_modification(ModNames.sve, shadow_brute_dangerous, update_monster_locations)
|
|
register_monster_modification(ModNames.sve, shadow_sniper, update_monster_locations)
|
|
register_monster_modification(ModNames.sve, shadow_shaman_dangerous, update_monster_locations)
|
|
register_monster_modification(ModNames.sve, mummy_dangerous, update_monster_locations)
|
|
register_monster_modification(ModNames.sve, royal_serpent, update_monster_locations)
|
|
register_monster_modification(ModNames.sve, skeleton_dangerous, update_monster_locations)
|
|
register_monster_modification(ModNames.sve, skeleton_mage, update_monster_locations)
|
|
register_monster_modification(ModNames.sve, dust_sprite_dangerous, update_monster_locations)
|
|
|
|
register_monster_modification(ModNames.deepwoods, shadow_brute, update_monster_locations)
|
|
register_monster_modification(ModNames.deepwoods, cave_fly, update_monster_locations)
|
|
register_monster_modification(ModNames.deepwoods, green_slime, update_monster_locations)
|
|
|
|
register_monster_modification(ModNames.boarding_house, pepper_rex, update_monster_locations)
|
|
register_monster_modification(ModNames.boarding_house, shadow_brute, update_monster_locations)
|
|
register_monster_modification(ModNames.boarding_house, iridium_bat, update_monster_locations)
|
|
register_monster_modification(ModNames.boarding_house, frost_bat, update_monster_locations)
|
|
register_monster_modification(ModNames.boarding_house, cave_fly, update_monster_locations)
|
|
register_monster_modification(ModNames.boarding_house, bat, update_monster_locations)
|
|
register_monster_modification(ModNames.boarding_house, grub, update_monster_locations)
|
|
register_monster_modification(ModNames.boarding_house, bug, update_monster_locations)
|
|
|
|
|
|
def all_monsters_by_name_given_mods(mods: Set[str]) -> Dict[str, StardewMonster]:
|
|
monsters_by_name = {}
|
|
for monster in all_monsters:
|
|
current_monster = monster
|
|
for mod in monster_modifications_by_mod:
|
|
if mod not in mods or monster.name not in monster_modifications_by_mod[mod]:
|
|
continue
|
|
modification_function = monster_modifications_by_mod[mod][monster.name]
|
|
current_monster = modification_function(mod, current_monster)
|
|
monsters_by_name[monster.name] = current_monster
|
|
return monsters_by_name
|
|
|
|
|
|
def all_monsters_by_category_given_mods(mods: Set[str]) -> Dict[str, Tuple[StardewMonster, ...]]:
|
|
monsters_by_category = {}
|
|
for monster in all_monsters:
|
|
current_monster = monster
|
|
for mod in monster_modifications_by_mod:
|
|
if mod not in mods or monster.name not in monster_modifications_by_mod[mod]:
|
|
continue
|
|
modification_function = monster_modifications_by_mod[mod][monster.name]
|
|
current_monster = modification_function(mod, current_monster)
|
|
if current_monster.category not in monsters_by_category:
|
|
monsters_by_category[monster.category] = ()
|
|
monsters_by_category[current_monster.category] = monsters_by_category[current_monster.category] + (current_monster,)
|
|
return monsters_by_category
|