about summary refs log tree commit diff stats
path: root/client
diff options
context:
space:
mode:
authorStar Rauchenberger <fefferburbia@gmail.com>2025-10-23 12:13:09 -0400
committerStar Rauchenberger <fefferburbia@gmail.com>2025-10-23 12:13:09 -0400
commit8cfee29fe6500e96c3b8669c12ac944716e9aae4 (patch)
treebf4723a1790d30155bb70fc710d416f008790b27 /client
parent3cb2ced48dec3b9ea60e008b2a3598a322458adf (diff)
downloadlingo2-archipelago-8cfee29fe6500e96c3b8669c12ac944716e9aae4.tar.gz
lingo2-archipelago-8cfee29fe6500e96c3b8669c12ac944716e9aae4.tar.bz2
lingo2-archipelago-8cfee29fe6500e96c3b8669c12ac944716e9aae4.zip
Added "Enable Gift Maps" option
Only supports The Advanced so far. Also added the mastery to The Advanced. Location listeners are now created after any map edits are made since some locations may require custom nodes (like The Advanced's mastery).
Diffstat (limited to 'client')
0 files changed, 0 insertions, 0 deletions
#n115'>115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174
from .generated import data_pb2 as data_pb2
from .items import SYMBOL_ITEMS, ANTI_COLLECTABLE_TRAPS
import pkgutil


class Lingo2StaticLogic:
    item_id_to_name: dict[int, str]
    location_id_to_name: dict[int, str]

    item_name_to_id: dict[str, int]
    location_name_to_id: dict[str, int]

    item_name_groups: dict[str, list[str]]
    location_name_groups: dict[str, list[str]]

    letter_weights: dict[str, int]

    def __init__(self):
        self.item_id_to_name = {}
        self.location_id_to_name = {}
        self.item_name_groups = {}
        self.location_name_groups = {}
        self.letter_weights = {}

        file = pkgutil.get_data(__name__, "generated/data.binpb")
        self.objects = data_pb2.AllObjects()
        self.objects.ParseFromString(bytearray(file))

        for door in self.objects.doors:
            if door.type in [data_pb2.DoorType.STANDARD, data_pb2.DoorType.LOCATION_ONLY, data_pb2.DoorType.GRAVESTONE]:
                location_name = self.get_door_location_name(door)
                self.location_id_to_name[door.ap_id] = location_name

            if door.type not in [data_pb2.DoorType.EVENT, data_pb2.DoorType.LOCATION_ONLY, data_pb2.DoorType.GRAVESTONE]:
                item_name = self.get_door_item_name(door)
                self.item_id_to_name[door.ap_id] = item_name

        for letter in self.objects.letters:
            letter_name = f"{letter.key.upper()}{'2' if letter.level2 else '1'}"
            location_name = f"{self.get_room_object_map_name(letter)} - {letter_name}"
            self.location_id_to_name[letter.ap_id] = location_name
            self.location_name_groups.setdefault("Letters", []).append(location_name)

            if not letter.level2:
                self.item_id_to_name[letter.ap_id] = letter.key.upper()
                self.item_name_groups.setdefault("Letters", []).append(letter.key.upper())

        for mastery in self.objects.masteries:
            location_name = f"{self.get_room_object_map_name(mastery)} - Mastery"
            self.location_id_to_name[mastery.ap_id] = location_name
            self.location_name_groups.setdefault("Masteries", []).append(location_name)

        for ending in self.objects.endings:
            location_name = f"{self.get_room_object_map_name(ending)} - {ending.name.title()} Ending"
            self.location_id_to_name[ending.ap_id] = location_name
            self.location_name_groups.setdefault("Endings", []).append(location_name)

        for progressive in self.objects.progressives:
            self.item_id_to_name[progressive.ap_id] = progressive.name

        for door_group in self.objects.door_groups:
            self.item_id_to_name[door_group.ap_id] = door_group.name

        for keyholder in self.objects.keyholders:
            if keyholder.HasField("key"):
                location_name = f"{self.get_room_object_location_prefix(keyholder)} - {keyholder.key.upper()} Keyholder"
                self.location_id_to_name[keyholder.ap_id] = location_name
                self.location_name_groups.setdefault("Keyholders", []).append(location_name)

        self.item_id_to_name[self.objects.special_ids["A Job Well Done"]] = "A Job Well Done"

        for symbol_name in SYMBOL_ITEMS.values():
            self.item_id_to_name[self.objects.special_ids[symbol_name]] = symbol_name

        for trap_name in ANTI_COLLECTABLE_TRAPS:
            self.item_id_to_name[self.objects.special_ids[trap_name]] = trap_name

        self.item_name_to_id = {name: ap_id for ap_id, name in self.item_id_to_name.items()}
        self.location_name_to_id = {name: ap_id for ap_id, name in self.location_id_to_name.items()}

        for panel in self.objects.panels:
            for letter in panel.answer.upper():
                self.letter_weights[letter] = self.letter_weights.get(letter, 0) + 1

    def get_door_item_name(self, door: data_pb2.Door) -> str:
        return f"{self.get_map_object_map_name(door)} - {door.name}"

    def get_door_item_name_by_id(self, door_id: int) -> str:
        door = self.objects.doors[door_id]
        return self.get_door_item_name(door_id)

    def get_door_location_name(self, door: data_pb2.Door) -> str:
        map_part = self.get_room_object_location_prefix(door)

        if door.HasField("location_name"):
            return f"{map_part} - {door.location_name}"

        generated_location_name = self.get_generated_door_location_name(door)
        if generated_location_name is not None:
            return generated_location_name

        return f"{map_part} - {door.name}"

    def get_generated_door_location_name(self, door: data_pb2.Door) -> str | None:
        if door.type != data_pb2.DoorType.STANDARD:
            return None

        if len(door.keyholders) > 0 or len(door.endings) > 0 or door.HasField("complete_at"):
            return None

        if len(door.panels) > 4:
            return None

        map_areas = set()
        for panel_id in door.panels:
            panel = self.objects.panels[panel_id.panel]
            panel_room = self.objects.rooms[panel.room_id]
            # It's okay if panel_display_name is not present because then it's coalesced with other unnamed areas.
            map_areas.add(panel_room.panel_display_name)

        if len(map_areas) > 1:
            return None

        game_map = self.objects.maps[door.map_id]
        map_area = map_areas.pop()
        if map_area == "":
            map_part = game_map.display_name
        else:
            map_part = f"{game_map.display_name} ({map_area})"

        def get_panel_display_name(panel: data_pb2.ProxyIdentifier) -> str:
            panel_data = self.objects.panels[panel.panel]
            panel_name = panel_data.display_name if panel_data.HasField("display_name") else panel_data.name

            if panel.HasField("answer"):
                return f"{panel_name}/{panel.answer.upper()}"
            else:
                return panel_name

        panel_names = [get_panel_display_name(panel_id)
                       for panel_id in door.panels]
        panel_names.sort()

        return map_part + " - " + ", ".join(panel_names)

    def get_door_location_name_by_id(self, door_id: int) -> str:
        door = self.objects.doors[door_id]
        return self.get_door_location_name(door)

    def get_room_region_name(self, room_id: int) -> str:
        room = self.objects.rooms[room_id]
        return f"{self.get_map_object_map_name(room)} - {room.name}"

    def get_map_object_map_name(self, obj) -> str:
        return self.objects.maps[obj.map_id].display_name

    def get_room_object_map_name(self, obj) -> str:
        return self.get_map_object_map_name(self.objects.rooms[obj.room_id])

    def get_room_object_location_prefix(self, obj) -> str:
        room = self.objects.rooms[obj.room_id]
        game_map = self.objects.maps[room.map_id]

        if room.HasField("panel_display_name"):
            return f"{game_map.display_name} ({room.panel_display_name})"
        else:
            return game_map.display_name

    def get_room_object_map_id(self, obj) -> int:
        return self.objects.rooms[obj.room_id].map_id

    def get_data_version(self) -> list[int]:
        version = self.objects.version
        return [version.major, version.minor, version.patch]