112 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			112 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| from typing import Dict, List, NamedTuple, Optional, TYPE_CHECKING
 | |
| 
 | |
| from BaseClasses import Item, ItemClassification
 | |
| from .options import ShuffleDoors
 | |
| 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
 | |
| 
 | |
| if TYPE_CHECKING:
 | |
|     from . import LingoWorld
 | |
| 
 | |
| 
 | |
| class ItemData(NamedTuple):
 | |
|     """
 | |
|     ItemData for an item in Lingo
 | |
|     """
 | |
|     code: int
 | |
|     classification: ItemClassification
 | |
|     mode: Optional[str]
 | |
|     has_doors: bool
 | |
|     painting_ids: List[str]
 | |
| 
 | |
|     def should_include(self, world: "LingoWorld") -> bool:
 | |
|         if self.mode == "colors":
 | |
|             return world.options.shuffle_colors > 0
 | |
|         elif self.mode == "doors":
 | |
|             return world.options.shuffle_doors != ShuffleDoors.option_none
 | |
|         elif self.mode == "complex door":
 | |
|             return world.options.shuffle_doors == ShuffleDoors.option_complex
 | |
|         elif self.mode == "door group":
 | |
|             return world.options.shuffle_doors == ShuffleDoors.option_simple
 | |
|         elif self.mode == "special":
 | |
|             return False
 | |
|         else:
 | |
|             return True
 | |
| 
 | |
| 
 | |
| class LingoItem(Item):
 | |
|     """
 | |
|     Item from the game Lingo
 | |
|     """
 | |
|     game: str = "Lingo"
 | |
| 
 | |
| 
 | |
| ALL_ITEM_TABLE: Dict[str, ItemData] = {}
 | |
| ITEMS_BY_GROUP: Dict[str, List[str]] = {}
 | |
| 
 | |
| 
 | |
| 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,
 | |
|         "Slowness Trap":             ItemClassification.trap,
 | |
|         "Iceland Trap":              ItemClassification.trap,
 | |
|         "Atbash Trap":               ItemClassification.trap,
 | |
|         "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()
 | 
