about summary refs log tree commit diff stats
path: root/util
Commit message (Collapse)AuthorAgeFilesLines
* Implemented color shuffleStar Rauchenberger2023-04-162-0/+51
a> 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93
from typing import Dict, List, NamedTuple, Optional, TYPE_CHECKING

from BaseClasses import Item, ItemClassification
from .static_logic import DOORS_BY_ROOM, PROGRESSION_BY_ROOM, PROGRESSIVE_ITEMS, get_door_group_item_id, \
    get_door_item_id, get_progressive_item_id, get_special_item_id


class ItemData(NamedTuple):
    """
    ItemData for an item in Lingo
    """
    code: int
    classification: ItemClassification
    mode: Optional[str]
    has_doors: bool
    painting_ids: List[str]


class LingoItem(Item):
    """
    Item from the game Lingo
    """
    game: str = "Lingo"


ALL_ITEM_TABLE: Dict[str, ItemData] = {}
ITEMS_BY_GROUP: Dict[str, List[str]] = {}

TRAP_ITEMS: List[str] = ["Slowness Trap", "Iceland Trap", "Atbash Trap"]


def load_item_data():
    global ALL_ITEM_TABLE, ITEMS_BY_GROUP

    for color in ["Black", "Red", "Blue", "Yellow", "Green", "Orange", "Gray", "Brown", "Purple"]:
        ALL_ITEM_TABLE[color] = ItemData(get_special_item_id(color), ItemClassification.progression,
                                         "colors", [], [])
        ITEMS_BY_GROUP.setdefault("Colors", []).append(color)

    door_groups: Dict[str, List[str]] = {}
    for room_name, doors in DOORS_BY_ROOM.items():
        for door_name, door in doors.items():
            if door.skip_item is True or door.event is True:
                continue

            if door.door_group is None:
                door_mode = "doors"
            else:
                door_mode = "complex door"
                door_groups.setdefault(door.door_group, [])

            if room_name in PROGRESSION_BY_ROOM and door_name in PROGRESSION_BY_ROOM[room_name]:
                door_mode = "special"

            ALL_ITEM_TABLE[door.item_name] = \
                ItemData(get_door_item_id(room_name, door_name),
                         ItemClassification.filler if door.junk_item else ItemClassification.progression, door_mode,
                         door.has_doors, door.painting_ids)
            ITEMS_BY_GROUP.setdefault("Doors", []).append(door.item_name)

            if door.item_group is not None:
                ITEMS_BY_GROUP.setdefault(door.item_group, []).append(door.item_name)

    for group, group_door_ids in door_groups.items():
        ALL_ITEM_TABLE[group] = ItemData(get_door_group_item_id(group),
                                         ItemClassification.progression, "door group", True, [])
        ITEMS_BY_GROUP.setdefault("Doors", []).append(group)

    special_items: Dict[str, ItemClassification] = {
        ":)":                        ItemClassification.filler,
        "The Feeling of Being Lost": ItemClassification.filler,
        "Wanderlust":                ItemClassification.filler,
        "Empty White Hallways":      ItemClassification.filler,
        **{trap_name: ItemClassification.trap for trap_name in TRAP_ITEMS},
        "Puzzle Skip":               ItemClassification.useful,
    }

    for item_name, classification in special_items.items():
        ALL_ITEM_TABLE[item_name] = ItemData(get_special_item_id(item_name), classification,
                                             "special", False, [])

        if classification == ItemClassification.filler:
            ITEMS_BY_GROUP.setdefault("Junk", []).append(item_name)
        elif classification == ItemClassification.trap:
            ITEMS_BY_GROUP.setdefault("Traps", []).append(item_name)

    for item_name in PROGRESSIVE_ITEMS:
        ALL_ITEM_TABLE[item_name] = ItemData(get_progressive_item_id(item_name),
                                             ItemClassification.progression, "special", False, [])


# Initialize the item data at module scope.
load_item_data()