about summary refs log tree commit diff stats
path: root/data/maps/the_nuanced/connections.txtpb
diff options
context:
space:
mode:
authorStar Rauchenberger <fefferburbia@gmail.com>2025-08-30 10:58:54 -0400
committerStar Rauchenberger <fefferburbia@gmail.com>2025-08-30 10:58:54 -0400
commit3b77044a6a53d38a6960eb2a5855283a00b24d75 (patch)
treeb32a19aa6b1cf3646e1796660ba11d0306c36947 /data/maps/the_nuanced/connections.txtpb
parent9b9f1248528090f5cdd37770a7df5b15d8804d86 (diff)
downloadlingo2-archipelago-3b77044a6a53d38a6960eb2a5855283a00b24d75.tar.gz
lingo2-archipelago-3b77044a6a53d38a6960eb2a5855283a00b24d75.tar.bz2
lingo2-archipelago-3b77044a6a53d38a6960eb2a5855283a00b24d75.zip
[Apworld] Require CC access + letters for CC color doors
Diffstat (limited to 'data/maps/the_nuanced/connections.txtpb')
0 files changed, 0 insertions, 0 deletions
137' href='#n137'>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
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_data_version(self) -> int:
        return self.objects.version