85 lines
3.5 KiB
Python
85 lines
3.5 KiB
Python
from typing import Dict, List, Set, cast
|
|
|
|
from BaseClasses import ItemClassification
|
|
|
|
from . import static_logic as static_witness_logic
|
|
from .item_definition_classes import DoorItemDefinition, ItemCategory, ItemData
|
|
from .static_locations import ID_START
|
|
|
|
ITEM_DATA: Dict[str, ItemData] = {}
|
|
ITEM_GROUPS: Dict[str, Set[str]] = {}
|
|
|
|
# Useful items that are treated specially at generation time and should not be automatically added to the player's
|
|
# item list during get_progression_items.
|
|
_special_usefuls: List[str] = ["Puzzle Skip"]
|
|
|
|
ALWAYS_GOOD_SYMBOL_ITEMS: Set[str] = {"Dots", "Black/White Squares", "Symmetry", "Shapers", "Stars"}
|
|
|
|
MODE_SPECIFIC_GOOD_ITEMS: Dict[str, Set[str]] = {
|
|
"none": set(),
|
|
"sigma_normal": set(),
|
|
"sigma_expert": {"Triangles"},
|
|
"umbra_variety": {"Triangles"}
|
|
}
|
|
|
|
MODE_SPECIFIC_GOOD_DISCARD_ITEMS: Dict[str, Set[str]] = {
|
|
"none": {"Triangles"},
|
|
"sigma_normal": {"Triangles"},
|
|
"sigma_expert": {"Arrows"},
|
|
"umbra_variety": set() # Variety Discards use both Arrows and Triangles, so neither of them are that useful alone
|
|
}
|
|
|
|
|
|
def populate_items() -> None:
|
|
for item_name, definition in static_witness_logic.ALL_ITEMS.items():
|
|
ap_item_code = definition.local_code + ID_START
|
|
classification: ItemClassification = ItemClassification.filler
|
|
local_only: bool = False
|
|
|
|
if definition.category is ItemCategory.SYMBOL:
|
|
classification = ItemClassification.progression
|
|
ITEM_GROUPS.setdefault("Symbols", set()).add(item_name)
|
|
elif definition.category is ItemCategory.DOOR:
|
|
classification = ItemClassification.progression
|
|
|
|
first_entity_hex = cast(DoorItemDefinition, definition).panel_id_hexes[0]
|
|
entity_type = static_witness_logic.ENTITIES_BY_HEX[first_entity_hex]["entityType"]
|
|
|
|
if entity_type == "Door":
|
|
ITEM_GROUPS.setdefault("Doors", set()).add(item_name)
|
|
elif entity_type == "Panel":
|
|
ITEM_GROUPS.setdefault("Panel Keys", set()).add(item_name)
|
|
elif entity_type in {"EP", "Obelisk Side", "Obelisk"}:
|
|
ITEM_GROUPS.setdefault("Obelisk Keys", set()).add(item_name)
|
|
else:
|
|
raise ValueError(f"Couldn't figure out what type of door item {definition} is.")
|
|
|
|
elif definition.category is ItemCategory.LASER:
|
|
classification = ItemClassification.progression_skip_balancing
|
|
ITEM_GROUPS.setdefault("Lasers", set()).add(item_name)
|
|
elif definition.category is ItemCategory.USEFUL:
|
|
classification = ItemClassification.useful
|
|
elif definition.category is ItemCategory.FILLER:
|
|
if item_name in ["Energy Fill (Small)"]:
|
|
local_only = True
|
|
classification = ItemClassification.filler
|
|
elif definition.category is ItemCategory.TRAP:
|
|
classification = ItemClassification.trap
|
|
elif definition.category is ItemCategory.JOKE:
|
|
classification = ItemClassification.filler
|
|
|
|
ITEM_DATA[item_name] = ItemData(ap_item_code, definition,
|
|
classification, local_only)
|
|
|
|
|
|
def get_item_to_door_mappings() -> Dict[int, List[int]]:
|
|
output: Dict[int, List[int]] = {}
|
|
for item_name, item_data in ITEM_DATA.items():
|
|
if not isinstance(item_data.definition, DoorItemDefinition) or item_data.ap_code is None:
|
|
continue
|
|
output[item_data.ap_code] = [int(hex_string, 16) for hex_string in item_data.definition.panel_id_hexes]
|
|
return output
|
|
|
|
|
|
populate_items()
|