Archipelago/worlds/ff1/Items.py

78 lines
2.5 KiB
Python

import json
from pathlib import Path
from typing import Dict, Set, NamedTuple, List
from BaseClasses import Item, ItemClassification
class ItemData(NamedTuple):
name: str
code: int
item_type: str
classification: ItemClassification
FF1_BRIDGE = 'Bridge'
FF1_STARTER_ITEMS = [
"Ship"
]
FF1_PROGRESSION_LIST = [
"Rod", "Cube", "Lute", "Key", "Chime", "Oxyale",
"Ship", "Canoe", "Floater", "Canal",
"Crown", "Crystal", "Herb", "Tnt", "Adamant", "Slab", "Ruby", "Bottle",
"Shard",
"EarthOrb", "FireOrb", "WaterOrb", "AirOrb"
]
FF1_USEFUL_LIST = [
"Tail", "Masamune", "Xcalber", "Katana", "Vorpal",
"DragonArmor", "Opal", "AegisShield", "Ribbon"
]
class FF1Items:
_item_table: List[ItemData] = []
_item_table_lookup: Dict[str, ItemData] = {}
def _populate_item_table_from_data(self):
base_path = Path(__file__).parent
file_path = (base_path / "data/items.json").resolve()
with open(file_path) as file:
items = json.load(file)
# Hardcode progression and categories for now
self._item_table = [ItemData(name, code, "FF1Item", ItemClassification.progression if name in
FF1_PROGRESSION_LIST else ItemClassification.useful if name in FF1_USEFUL_LIST else
ItemClassification.filler) for name, code in items.items()]
self._item_table_lookup = {item.name: item for item in self._item_table}
def _get_item_table(self) -> List[ItemData]:
if not self._item_table or not self._item_table_lookup:
self._populate_item_table_from_data()
return self._item_table
def _get_item_table_lookup(self) -> Dict[str, ItemData]:
if not self._item_table or not self._item_table_lookup:
self._populate_item_table_from_data()
return self._item_table_lookup
def get_item_names_per_category(self) -> Dict[str, Set[str]]:
categories: Dict[str, Set[str]] = {}
for item in self._get_item_table():
categories.setdefault(item.item_type, set()).add(item.name)
return categories
def generate_item(self, name: str, player: int) -> Item:
item = self._get_item_table_lookup().get(name)
return Item(name, item.classification,
item.code, player)
def get_item_name_to_code_dict(self) -> Dict[str, int]:
return {name: item.code for name, item in self._get_item_table_lookup().items()}
def get_item(self, name: str) -> ItemData:
return self._get_item_table_lookup()[name]