123 lines
		
	
	
		
			6.8 KiB
		
	
	
	
		
			Python
		
	
	
	
			
		
		
	
	
			123 lines
		
	
	
		
			6.8 KiB
		
	
	
	
		
			Python
		
	
	
	
from typing import Dict, Set, Tuple, NamedTuple, Optional
 | 
						|
from BaseClasses import ItemClassification
 | 
						|
 | 
						|
class ItemData(NamedTuple):
 | 
						|
    category: str
 | 
						|
    code: Optional[int]
 | 
						|
    classification: ItemClassification
 | 
						|
    amount: Optional[int] = 1
 | 
						|
 | 
						|
item_table: Dict[str, ItemData] = {
 | 
						|
    "! Switch": ItemData("Items", 0x302050, ItemClassification.progression),
 | 
						|
    "Dashed Platform": ItemData("Items", 0x302051, ItemClassification.progression),
 | 
						|
    "Dashed Stairs": ItemData("Items", 0x302052, ItemClassification.progression),
 | 
						|
    "Beanstalk": ItemData("Items", 0x302053, ItemClassification.progression),
 | 
						|
    "Helicopter Morph": ItemData("Morphs", 0x302054, ItemClassification.progression),
 | 
						|
    "Spring Ball": ItemData("Items", 0x302055, ItemClassification.progression),
 | 
						|
    "Large Spring Ball": ItemData("Items", 0x302056, ItemClassification.progression),
 | 
						|
    "Arrow Wheel": ItemData("Items", 0x302057, ItemClassification.progression),
 | 
						|
    "Vanishing Arrow Wheel": ItemData("Items", 0x302058, ItemClassification.progression),
 | 
						|
    "Mole Tank Morph": ItemData("Morphs", 0x302059, ItemClassification.progression),
 | 
						|
    "Watermelon": ItemData("Items", 0x30205A, ItemClassification.progression),
 | 
						|
    "Ice Melon": ItemData("Items", 0x30205B, ItemClassification.progression),
 | 
						|
    "Fire Melon": ItemData("Items", 0x30205C, ItemClassification.progression),
 | 
						|
    "Super Star": ItemData("Items", 0x30205D, ItemClassification.progression),
 | 
						|
    "Car Morph": ItemData("Morphs", 0x30205E, ItemClassification.progression),
 | 
						|
    "Flashing Eggs": ItemData("Items", 0x30205F, ItemClassification.progression),
 | 
						|
    "Giant Eggs": ItemData("Items", 0x302060, ItemClassification.progression),
 | 
						|
    "Egg Launcher": ItemData("Items", 0x302061, ItemClassification.progression),
 | 
						|
    "Egg Plant": ItemData("Items", 0x302062, ItemClassification.progression),
 | 
						|
    "Submarine Morph": ItemData("Morphs", 0x302063, ItemClassification.progression),
 | 
						|
    "Chomp Rock": ItemData("Items", 0x302064, ItemClassification.progression),
 | 
						|
    "Poochy": ItemData("Items", 0x302065, ItemClassification.progression),
 | 
						|
    "Platform Ghost": ItemData("Items", 0x302066, ItemClassification.progression),
 | 
						|
    "Skis": ItemData("Items", 0x302067, ItemClassification.progression),
 | 
						|
    "Train Morph": ItemData("Morphs", 0x302068, ItemClassification.progression),
 | 
						|
    "Key": ItemData("Items", 0x302069, ItemClassification.progression),
 | 
						|
    "Middle Ring": ItemData("Items", 0x30206A, ItemClassification.progression),
 | 
						|
    "Bucket": ItemData("Items", 0x30206B, ItemClassification.progression),
 | 
						|
    "Tulip": ItemData("Items", 0x30206C, ItemClassification.progression),
 | 
						|
    "Egg Capacity Upgrade": ItemData("Items", 0x30206D, ItemClassification.progression, 5),
 | 
						|
    "Secret Lens": ItemData("Items", 0x302081, ItemClassification.progression),
 | 
						|
 | 
						|
    "World 1 Gate": ItemData("Gates", 0x30206E, ItemClassification.progression),
 | 
						|
    "World 2 Gate": ItemData("Gates", 0x30206F, ItemClassification.progression),
 | 
						|
    "World 3 Gate": ItemData("Gates", 0x302070, ItemClassification.progression),
 | 
						|
    "World 4 Gate": ItemData("Gates", 0x302071, ItemClassification.progression),
 | 
						|
    "World 5 Gate": ItemData("Gates", 0x302072, ItemClassification.progression),
 | 
						|
    "World 6 Gate": ItemData("Gates", 0x302073, ItemClassification.progression),
 | 
						|
 | 
						|
    "Extra 1": ItemData("Panels", 0x302074, ItemClassification.progression),
 | 
						|
    "Extra 2": ItemData("Panels", 0x302075, ItemClassification.progression),
 | 
						|
    "Extra 3": ItemData("Panels", 0x302076, ItemClassification.progression),
 | 
						|
    "Extra 4": ItemData("Panels", 0x302077, ItemClassification.progression),
 | 
						|
    "Extra 5": ItemData("Panels", 0x302078, ItemClassification.progression),
 | 
						|
    "Extra 6": ItemData("Panels", 0x302079, ItemClassification.progression),
 | 
						|
    "Extra Panels": ItemData("Panels", 0x30207A, ItemClassification.progression),
 | 
						|
 | 
						|
    "Bonus 1": ItemData("Panels", 0x30207B, ItemClassification.progression),
 | 
						|
    "Bonus 2": ItemData("Panels", 0x30207C, ItemClassification.progression),
 | 
						|
    "Bonus 3": ItemData("Panels", 0x30207D, ItemClassification.progression),
 | 
						|
    "Bonus 4": ItemData("Panels", 0x30207E, ItemClassification.progression),
 | 
						|
    "Bonus 5": ItemData("Panels", 0x30207F, ItemClassification.progression),
 | 
						|
    "Bonus 6": ItemData("Panels", 0x302080, ItemClassification.progression),
 | 
						|
    "Bonus Panels": ItemData("Panels", 0x302082, ItemClassification.progression),
 | 
						|
 | 
						|
    "Anytime Egg": ItemData("Consumable", 0x302083, ItemClassification.useful, 0),
 | 
						|
    "Anywhere Pow": ItemData("Consumable", 0x302084, ItemClassification.filler, 0),
 | 
						|
    "Winged Cloud Maker": ItemData("Consumable", 0x302085, ItemClassification.filler, 0),
 | 
						|
    "Pocket Melon": ItemData("Consumable", 0x302086, ItemClassification.filler, 0),
 | 
						|
    "Pocket Fire Melon": ItemData("Consumable", 0x302087, ItemClassification.filler, 0),
 | 
						|
    "Pocket Ice Melon": ItemData("Consumable", 0x302088, ItemClassification.filler, 0),
 | 
						|
    "Magnifying Glass": ItemData("Consumable", 0x302089, ItemClassification.filler, 0),
 | 
						|
    "+10 Stars": ItemData("Consumable", 0x30208A, ItemClassification.useful, 0),
 | 
						|
    "+20 Stars": ItemData("Consumable", 0x30208B, ItemClassification.useful, 0),
 | 
						|
    "1-Up": ItemData("Lives", 0x30208C, ItemClassification.filler, 0),
 | 
						|
    "2-Up": ItemData("Lives", 0x30208D, ItemClassification.filler, 0),
 | 
						|
    "3-Up": ItemData("Lives", 0x30208E, ItemClassification.filler, 0),
 | 
						|
    "10-Up": ItemData("Lives", 0x30208F, ItemClassification.useful, 5),
 | 
						|
    "Bonus Consumables": ItemData("Events", None, ItemClassification.progression, 0),
 | 
						|
    "Bandit Consumables": ItemData("Events", None, ItemClassification.progression, 0),
 | 
						|
    "Bandit Watermelons": ItemData("Events", None, ItemClassification.progression, 0),
 | 
						|
 | 
						|
    "Fuzzy Trap": ItemData("Traps", 0x302090, ItemClassification.trap, 0),
 | 
						|
    "Reversal Trap": ItemData("Traps", 0x302091, ItemClassification.trap, 0),
 | 
						|
    "Darkness Trap": ItemData("Traps", 0x302092, ItemClassification.trap, 0),
 | 
						|
    "Freeze Trap": ItemData("Traps", 0x302093, ItemClassification.trap, 0),
 | 
						|
 | 
						|
    "Boss Clear": ItemData("Events", None, ItemClassification.progression, 0),
 | 
						|
    "Piece of Luigi": ItemData("Items", 0x302095, ItemClassification.progression, 0),
 | 
						|
    "Saved Baby Luigi": ItemData("Events", None, ItemClassification.progression, 0)
 | 
						|
}
 | 
						|
 | 
						|
filler_items: Tuple[str, ...] = (
 | 
						|
    "Anytime Egg",
 | 
						|
    "Anywhere Pow",
 | 
						|
    "Winged Cloud Maker",
 | 
						|
    "Pocket Melon",
 | 
						|
    "Pocket Fire Melon",
 | 
						|
    "Pocket Ice Melon",
 | 
						|
    "Magnifying Glass",
 | 
						|
    "+10 Stars",
 | 
						|
    "+20 Stars",
 | 
						|
    "1-Up",
 | 
						|
    "2-Up",
 | 
						|
    "3-Up"
 | 
						|
)
 | 
						|
 | 
						|
trap_items: Tuple[str, ...] = (
 | 
						|
    "Fuzzy Trap",
 | 
						|
    "Reversal Trap",
 | 
						|
    "Darkness Trap",
 | 
						|
    "Freeze Trap"
 | 
						|
)
 | 
						|
 | 
						|
def get_item_names_per_category() -> Dict[str, Set[str]]:
 | 
						|
    categories: Dict[str, Set[str]] = {}
 | 
						|
 | 
						|
    for name, data in item_table.items():
 | 
						|
        if data.category != "Events":
 | 
						|
            categories.setdefault(data.category, set()).add(name)
 | 
						|
 | 
						|
    return categories
 |