1 files changed, 7 insertions, 2 deletions

@@ -15,6 +15,9 @@ class Lingo2StaticLogic:
 
     letter_weights: dict[str, int]
 
+    door_id_by_ap_id: dict[int, int]
+    port_id_by_ap_id: dict[int, int]
+
     def __init__(self):
         self.item_id_to_name = {}
         self.location_id_to_name = {}
@@ -31,8 +34,7 @@ class Lingo2StaticLogic:
                 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,
-                                 data_pb2.DoorType.LEGACY_LOCATION]:
+            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
 
@@ -84,6 +86,9 @@ class Lingo2StaticLogic:
             for letter in panel.answer.upper():
                 self.letter_weights[letter] = self.letter_weights.get(letter, 0) + 1
 
+        self.door_id_by_ap_id = {door.ap_id: door.id for door in self.objects.doors if door.HasField("ap_id")}
+        self.port_id_by_ap_id = {port.ap_id: port.id for port in self.objects.ports if port.HasField("ap_id")}
+
     def get_door_item_name(self, door: data_pb2.Door) -> str:
         return f"{self.get_map_object_map_name(door)} - {door.name}"