Add cyan door behavior option

1 files changed, 26 insertions, 10 deletions

@@ -3,7 +3,7 @@ from enum import IntEnum, auto
 from .generated import data_pb2 as data_pb2
 from typing import TYPE_CHECKING, NamedTuple
 
-from .options import VictoryCondition, ShuffleLetters
+from .options import VictoryCondition, ShuffleLetters, CyanDoorBehavior
 
 if TYPE_CHECKING:
     from . import Lingo2World
@@ -124,11 +124,13 @@ class Lingo2PlayerLogic:
                     self.real_items.append(progressive.name)
 
         for door_group in world.static_logic.objects.door_groups:
-            if door_group.type == data_pb2.DoorGroupType.CONNECTOR and not self.world.options.shuffle_doors:
-                continue
-
-            if (door_group.type == data_pb2.DoorGroupType.COLOR_CONNECTOR and
-                    not self.world.options.shuffle_control_center_colors):
+            if door_group.type == data_pb2.DoorGroupType.CONNECTOR:
+                if not self.world.options.shuffle_doors:
+                    continue
+            elif door_group.type == data_pb2.DoorGroupType.COLOR_CONNECTOR:
+                if not self.world.options.shuffle_control_center_colors:
+                    continue
+            else:
                 continue
 
             for door in door_group.doors:
@@ -157,6 +159,19 @@ class Lingo2PlayerLogic:
             self.item_by_door[door.id] = (door_item_name, 1)
             self.real_items.append(door_item_name)
 
+        # We handle cyan_door_behavior = Item after door shuffle, because cyan doors that are impacted by door shuffle
+        # should be exempt from cyan_door_behavior.
+        if world.options.cyan_door_behavior == CyanDoorBehavior.option_item:
+            for door_group in world.static_logic.objects.door_groups:
+                if door_group.type != data_pb2.DoorGroupType.CYAN_DOORS:
+                    continue
+
+                for door in door_group.doors:
+                    if not door in self.item_by_door:
+                        self.item_by_door[door] = (door_group.name, 1)
+
+                self.real_items.append(door_group.name)
+
         for door in world.static_logic.objects.doors:
             if door.type in [data_pb2.DoorType.STANDARD, data_pb2.DoorType.LOCATION_ONLY, data_pb2.DoorType.GRAVESTONE]:
                 self.locations_by_room.setdefault(door.room_id, []).append(PlayerLocation(door.ap_id,
@@ -295,12 +310,13 @@ class Lingo2PlayerLogic:
             self.add_solution_reqs(reqs, door.control_center_color)
 
         if door.double_letters:
-            if self.world.options.shuffle_letters in [ShuffleLetters.option_vanilla,
-                                                      ShuffleLetters.option_vanilla_cyan]:
+            if self.world.options.cyan_door_behavior == CyanDoorBehavior.option_collect_h2:
                 reqs.rooms.add("The Repetitive - Main Room")
-            elif self.world.options.shuffle_letters in [ShuffleLetters.option_progressive,
-                                                        ShuffleLetters.option_item_cyan]:
+            elif self.world.options.cyan_door_behavior == CyanDoorBehavior.option_any_double_letter:
                 reqs.cyans = True
+            elif self.world.options.cyan_door_behavior == CyanDoorBehavior.option_item:
+                # There shouldn't be any locations that are cyan doors.
+                pass
 
         for keyholder_uses in door.keyholders:
             key_name = keyholder_uses.key.upper()