mirror of
https://github.com/ArchipelagoMW/Archipelago.git
synced 2026-03-27 21:53:21 -07:00
62 lines
2.6 KiB
Python
62 lines
2.6 KiB
Python
from typing import TYPE_CHECKING, Callable, List, Dict
|
|
from BaseClasses import CollectionState
|
|
from .Items import get_item_by_civ_name
|
|
from .Data import get_boosts_data
|
|
from .Enum import CivVICheckType
|
|
from .ProgressiveDistricts import convert_items_to_have_progression
|
|
|
|
from worlds.generic.Rules import forbid_item, set_rule
|
|
|
|
|
|
if TYPE_CHECKING:
|
|
from . import CivVIWorld
|
|
|
|
|
|
def generate_has_required_items_lambda(prereqs: List[str], required_count: int, player: int) -> Callable[[CollectionState, int], bool]:
|
|
def has_required_items_lambda(state: CollectionState):
|
|
return has_required_items(state, prereqs, required_count, player)
|
|
return has_required_items_lambda
|
|
|
|
|
|
def create_boost_rules(world: 'CivVIWorld'):
|
|
boost_data_list = get_boosts_data()
|
|
boost_locations = [location for location in world.location_table.values() if location.location_type == CivVICheckType.BOOST]
|
|
for location in boost_locations:
|
|
boost_data = next((boost for boost in boost_data_list if boost.Type == location.name), None)
|
|
world_location = world.multiworld.get_location(location.name, world.player)
|
|
forbid_item(world_location, "Progressive Era", world.player)
|
|
if not boost_data or boost_data.PrereqRequiredCount == 0:
|
|
continue
|
|
|
|
set_rule(world_location,
|
|
generate_has_required_items_lambda(boost_data.Prereq, boost_data.PrereqRequiredCount, world.player)
|
|
)
|
|
|
|
|
|
def has_required_items(state: CollectionState, prereqs: List[str], required_count: int, player: int) -> bool:
|
|
world: 'CivVIWorld' = state.multiworld.worlds[player]
|
|
has_progressive_items = world.options.progression_style != "none"
|
|
if has_progressive_items:
|
|
count = 0
|
|
items = [get_item_by_civ_name(item, world.item_table).name for item in convert_items_to_have_progression(prereqs)]
|
|
progressive_items: Dict[str, int] = {}
|
|
for item in items:
|
|
if "Progressive" in item:
|
|
if not progressive_items.get(item):
|
|
progressive_items[item] = 0
|
|
progressive_items[item] += 1
|
|
else:
|
|
if state.has(item, player):
|
|
count += 1
|
|
|
|
for item, required_progressive_item_count in progressive_items.items():
|
|
if state.has(item, player, required_progressive_item_count):
|
|
count += required_progressive_item_count
|
|
return count >= required_count
|
|
else:
|
|
total = state.count_from_list_unique(
|
|
[
|
|
get_item_by_civ_name(prereq, state.multiworld.worlds[player].item_table).name for prereq in prereqs
|
|
], player)
|
|
return total >= required_count
|