diff options
Diffstat (limited to 'apworld/player_logic.py')
| -rw-r--r-- | apworld/player_logic.py | 471 |
1 files changed, 416 insertions, 55 deletions
| diff --git a/apworld/player_logic.py b/apworld/player_logic.py index 6feef99..3ee8f38 100644 --- a/apworld/player_logic.py +++ b/apworld/player_logic.py | |||
| @@ -1,6 +1,11 @@ | |||
| 1 | from enum import IntEnum, auto | ||
| 2 | |||
| 1 | from .generated import data_pb2 as data_pb2 | 3 | from .generated import data_pb2 as data_pb2 |
| 4 | from .items import SYMBOL_ITEMS | ||
| 2 | from typing import TYPE_CHECKING, NamedTuple | 5 | from typing import TYPE_CHECKING, NamedTuple |
| 3 | 6 | ||
| 7 | from .options import ShuffleLetters, CyanDoorBehavior | ||
| 8 | |||
| 4 | if TYPE_CHECKING: | 9 | if TYPE_CHECKING: |
| 5 | from . import Lingo2World | 10 | from . import Lingo2World |
| 6 | 11 | ||
| @@ -12,64 +17,175 @@ def calculate_letter_histogram(solution: str) -> dict[str, int]: | |||
| 12 | real_l = l.upper() | 17 | real_l = l.upper() |
| 13 | histogram[real_l] = min(histogram.get(real_l, 0) + 1, 2) | 18 | histogram[real_l] = min(histogram.get(real_l, 0) + 1, 2) |
| 14 | 19 | ||
| 15 | for free_letter in "HINT": | ||
| 16 | if histogram.get(free_letter, 0) == 1: | ||
| 17 | del histogram[free_letter] | ||
| 18 | |||
| 19 | return histogram | 20 | return histogram |
| 20 | 21 | ||
| 21 | 22 | ||
| 22 | class AccessRequirements: | 23 | class AccessRequirements: |
| 23 | items: set[str] | 24 | items: set[str] |
| 25 | progressives: dict[str, int] | ||
| 24 | rooms: set[str] | 26 | rooms: set[str] |
| 25 | symbols: set[str] | ||
| 26 | letters: dict[str, int] | 27 | letters: dict[str, int] |
| 28 | cyans: bool | ||
| 27 | 29 | ||
| 28 | # This is an AND of ORs. | 30 | # This is an AND of ORs. |
| 29 | or_logic: list[list["AccessRequirements"]] | 31 | or_logic: list[list["AccessRequirements"]] |
| 30 | 32 | ||
| 33 | # When complete_at is set, at least that many of the requirements in possibilities must be accessible. This should | ||
| 34 | # only be used for doors with complete_at > 1, as or_logic is more efficient for complete_at == 1. | ||
| 35 | complete_at: int | None | ||
| 36 | possibilities: list["AccessRequirements"] | ||
| 37 | |||
| 31 | def __init__(self): | 38 | def __init__(self): |
| 32 | self.items = set() | 39 | self.items = set() |
| 40 | self.progressives = dict() | ||
| 33 | self.rooms = set() | 41 | self.rooms = set() |
| 34 | self.symbols = set() | ||
| 35 | self.letters = dict() | 42 | self.letters = dict() |
| 43 | self.cyans = False | ||
| 36 | self.or_logic = list() | 44 | self.or_logic = list() |
| 45 | self.complete_at = None | ||
| 46 | self.possibilities = list() | ||
| 37 | 47 | ||
| 38 | def add_solution(self, solution: str): | 48 | def copy(self) -> "AccessRequirements": |
| 39 | histogram = calculate_letter_histogram(solution) | 49 | reqs = AccessRequirements() |
| 40 | 50 | reqs.items = self.items.copy() | |
| 41 | for l, a in histogram.items(): | 51 | reqs.progressives = self.progressives.copy() |
| 42 | self.letters[l] = max(self.letters.get(l, 0), histogram.get(l)) | 52 | reqs.rooms = self.rooms.copy() |
| 53 | reqs.letters = self.letters.copy() | ||
| 54 | reqs.cyans = self.cyans | ||
| 55 | reqs.or_logic = [[other_req.copy() for other_req in disjunction] for disjunction in self.or_logic] | ||
| 56 | reqs.complete_at = self.complete_at | ||
| 57 | reqs.possibilities = self.possibilities.copy() | ||
| 58 | return reqs | ||
| 43 | 59 | ||
| 44 | def merge(self, other: "AccessRequirements"): | 60 | def merge(self, other: "AccessRequirements"): |
| 45 | for item in other.items: | 61 | for item in other.items: |
| 46 | self.items.add(item) | 62 | self.items.add(item) |
| 47 | 63 | ||
| 64 | for item, amount in other.progressives.items(): | ||
| 65 | self.progressives[item] = max(amount, self.progressives.get(item, 0)) | ||
| 66 | |||
| 48 | for room in other.rooms: | 67 | for room in other.rooms: |
| 49 | self.rooms.add(room) | 68 | self.rooms.add(room) |
| 50 | 69 | ||
| 51 | for symbol in other.symbols: | ||
| 52 | self.symbols.add(symbol) | ||
| 53 | |||
| 54 | for letter, level in other.letters.items(): | 70 | for letter, level in other.letters.items(): |
| 55 | self.letters[letter] = max(self.letters.get(letter, 0), level) | 71 | self.letters[letter] = max(self.letters.get(letter, 0), level) |
| 56 | 72 | ||
| 73 | self.cyans = self.cyans or other.cyans | ||
| 74 | |||
| 57 | for disjunction in other.or_logic: | 75 | for disjunction in other.or_logic: |
| 58 | self.or_logic.append(disjunction) | 76 | self.or_logic.append([sub_req.copy() for sub_req in disjunction]) |
| 77 | |||
| 78 | if other.complete_at is not None: | ||
| 79 | # Merging multiple requirements that use complete_at sucks, and is part of why we want to minimize use of | ||
| 80 | # it. If both requirements use complete_at, we will cheat by using the or_logic field, which supports | ||
| 81 | # conjunctions of requirements. | ||
| 82 | if self.complete_at is not None: | ||
| 83 | print("Merging requirements with complete_at > 1. This is messy and should be avoided!") | ||
| 84 | |||
| 85 | left_req = AccessRequirements() | ||
| 86 | left_req.complete_at = self.complete_at | ||
| 87 | left_req.possibilities = [sub_req.copy() for sub_req in self.possibilities] | ||
| 88 | self.or_logic.append([left_req]) | ||
| 89 | |||
| 90 | self.complete_at = None | ||
| 91 | self.possibilities = list() | ||
| 92 | |||
| 93 | right_req = AccessRequirements() | ||
| 94 | right_req.complete_at = other.complete_at | ||
| 95 | right_req.possibilities = [sub_req.copy() for sub_req in other.possibilities] | ||
| 96 | self.or_logic.append([right_req]) | ||
| 97 | else: | ||
| 98 | self.complete_at = other.complete_at | ||
| 99 | self.possibilities = [sub_req.copy() for sub_req in other.possibilities] | ||
| 100 | |||
| 101 | def is_empty(self) -> bool: | ||
| 102 | return (len(self.items) == 0 and len(self.progressives) == 0 and len(self.rooms) == 0 and len(self.letters) == 0 | ||
| 103 | and not self.cyans and len(self.or_logic) == 0 and self.complete_at is None) | ||
| 104 | |||
| 105 | def __eq__(self, other: "AccessRequirements"): | ||
| 106 | return (self.items == other.items and self.progressives == other.progressives and self.rooms == other.rooms and | ||
| 107 | self.letters == other.letters and self.cyans == other.cyans and self.or_logic == other.or_logic and | ||
| 108 | self.complete_at == other.complete_at and self.possibilities == other.possibilities) | ||
| 109 | |||
| 110 | def simplify(self): | ||
| 111 | resimplify = False | ||
| 112 | |||
| 113 | if len(self.or_logic) > 0: | ||
| 114 | old_or_logic = self.or_logic | ||
| 115 | |||
| 116 | def remove_redundant(sub_reqs: "AccessRequirements"): | ||
| 117 | new_reqs = sub_reqs.copy() | ||
| 118 | new_reqs.letters = {l: v for l, v in new_reqs.letters.items() if self.letters.get(l, 0) < v} | ||
| 119 | if new_reqs != sub_reqs: | ||
| 120 | return new_reqs | ||
| 121 | else: | ||
| 122 | return sub_reqs | ||
| 123 | |||
| 124 | self.or_logic = [] | ||
| 125 | for disjunction in old_or_logic: | ||
| 126 | new_disjunction = [] | ||
| 127 | for ssr in disjunction: | ||
| 128 | new_ssr = remove_redundant(ssr) | ||
| 129 | if not new_ssr.is_empty(): | ||
| 130 | new_disjunction.append(new_ssr) | ||
| 131 | else: | ||
| 132 | new_disjunction.clear() | ||
| 133 | break | ||
| 134 | if len(new_disjunction) == 1: | ||
| 135 | self.merge(new_disjunction[0]) | ||
| 136 | resimplify = True | ||
| 137 | elif len(new_disjunction) > 1: | ||
| 138 | if all(cjr == new_disjunction[0] for cjr in new_disjunction): | ||
| 139 | self.merge(new_disjunction[0]) | ||
| 140 | resimplify = True | ||
| 141 | else: | ||
| 142 | self.or_logic.append(new_disjunction) | ||
| 143 | |||
| 144 | if resimplify: | ||
| 145 | self.simplify() | ||
| 146 | |||
| 147 | def get_referenced_rooms(self): | ||
| 148 | result = set(self.rooms) | ||
| 149 | |||
| 150 | for disjunction in self.or_logic: | ||
| 151 | for sub_req in disjunction: | ||
| 152 | result = result.union(sub_req.get_referenced_rooms()) | ||
| 153 | |||
| 154 | for sub_req in self.possibilities: | ||
| 155 | result = result.union(sub_req.get_referenced_rooms()) | ||
| 156 | |||
| 157 | return result | ||
| 158 | |||
| 159 | def remove_room(self, room: str): | ||
| 160 | if room in self.rooms: | ||
| 161 | self.rooms.remove(room) | ||
| 162 | |||
| 163 | for disjunction in self.or_logic: | ||
| 164 | for sub_req in disjunction: | ||
| 165 | sub_req.remove_room(room) | ||
| 166 | |||
| 167 | for sub_req in self.possibilities: | ||
| 168 | sub_req.remove_room(room) | ||
| 59 | 169 | ||
| 60 | def __repr__(self): | 170 | def __repr__(self): |
| 61 | parts = [] | 171 | parts = [] |
| 62 | if len(self.items) > 0: | 172 | if len(self.items) > 0: |
| 63 | parts.append(f"items={self.items}") | 173 | parts.append(f"items={self.items}") |
| 174 | if len(self.progressives) > 0: | ||
| 175 | parts.append(f"progressives={self.progressives}") | ||
| 64 | if len(self.rooms) > 0: | 176 | if len(self.rooms) > 0: |
| 65 | parts.append(f"rooms={self.rooms}") | 177 | parts.append(f"rooms={self.rooms}") |
| 66 | if len(self.symbols) > 0: | ||
| 67 | parts.append(f"symbols={self.symbols}") | ||
| 68 | if len(self.letters) > 0: | 178 | if len(self.letters) > 0: |
| 69 | parts.append(f"letters={self.letters}") | 179 | parts.append(f"letters={self.letters}") |
| 180 | if self.cyans: | ||
| 181 | parts.append(f"cyans=True") | ||
| 70 | if len(self.or_logic) > 0: | 182 | if len(self.or_logic) > 0: |
| 71 | parts.append(f"or_logic={self.or_logic}") | 183 | parts.append(f"or_logic={self.or_logic}") |
| 72 | return f"AccessRequirements({", ".join(parts)})" | 184 | if self.complete_at is not None: |
| 185 | parts.append(f"complete_at={self.complete_at}") | ||
| 186 | if len(self.possibilities) > 0: | ||
| 187 | parts.append(f"possibilities={self.possibilities}") | ||
| 188 | return "AccessRequirements(" + ", ".join(parts) + ")" | ||
| 73 | 189 | ||
| 74 | 190 | ||
| 75 | class PlayerLocation(NamedTuple): | 191 | class PlayerLocation(NamedTuple): |
| @@ -77,13 +193,21 @@ class PlayerLocation(NamedTuple): | |||
| 77 | reqs: AccessRequirements | 193 | reqs: AccessRequirements |
| 78 | 194 | ||
| 79 | 195 | ||
| 196 | class LetterBehavior(IntEnum): | ||
| 197 | VANILLA = auto() | ||
| 198 | ITEM = auto() | ||
| 199 | UNLOCKED = auto() | ||
| 200 | |||
| 201 | |||
| 80 | class Lingo2PlayerLogic: | 202 | class Lingo2PlayerLogic: |
| 81 | world: "Lingo2World" | 203 | world: "Lingo2World" |
| 82 | 204 | ||
| 205 | shuffled_maps: set[int] | ||
| 206 | |||
| 83 | locations_by_room: dict[int, list[PlayerLocation]] | 207 | locations_by_room: dict[int, list[PlayerLocation]] |
| 84 | event_loc_item_by_room: dict[int, dict[str, str]] | 208 | event_loc_item_by_room: dict[int, dict[str, str]] |
| 85 | 209 | ||
| 86 | item_by_door: dict[int, str] | 210 | item_by_door: dict[int, tuple[str, int]] |
| 87 | 211 | ||
| 88 | panel_reqs: dict[int, AccessRequirements] | 212 | panel_reqs: dict[int, AccessRequirements] |
| 89 | proxy_reqs: dict[int, dict[str, AccessRequirements]] | 213 | proxy_reqs: dict[int, dict[str, AccessRequirements]] |
| @@ -91,6 +215,9 @@ class Lingo2PlayerLogic: | |||
| 91 | 215 | ||
| 92 | real_items: list[str] | 216 | real_items: list[str] |
| 93 | 217 | ||
| 218 | double_letter_amount: dict[str, int] | ||
| 219 | goal_room_id: int | ||
| 220 | |||
| 94 | def __init__(self, world: "Lingo2World"): | 221 | def __init__(self, world: "Lingo2World"): |
| 95 | self.world = world | 222 | self.world = world |
| 96 | self.locations_by_room = {} | 223 | self.locations_by_room = {} |
| @@ -100,47 +227,194 @@ class Lingo2PlayerLogic: | |||
| 100 | self.proxy_reqs = dict() | 227 | self.proxy_reqs = dict() |
| 101 | self.door_reqs = dict() | 228 | self.door_reqs = dict() |
| 102 | self.real_items = list() | 229 | self.real_items = list() |
| 230 | self.double_letter_amount = dict() | ||
| 231 | |||
| 232 | def should_shuffle_map(game_map) -> bool: | ||
| 233 | if game_map.type == data_pb2.MapType.NORMAL_MAP: | ||
| 234 | return True | ||
| 235 | elif game_map.type == data_pb2.MapType.ICARUS: | ||
| 236 | return bool(world.options.enable_icarus) | ||
| 237 | elif game_map.type == data_pb2.MapType.GIFT_MAP: | ||
| 238 | if game_map.name == "the_advanced": | ||
| 239 | return "The Advanced" in world.options.enable_gift_maps.value | ||
| 240 | elif game_map.name == "the_charismatic": | ||
| 241 | return "The Charismatic" in world.options.enable_gift_maps.value | ||
| 242 | elif game_map.name == "the_crystalline": | ||
| 243 | return "The Crystalline" in world.options.enable_gift_maps.value | ||
| 244 | elif game_map.name == "the_fuzzy": | ||
| 245 | return "The Fuzzy" in world.options.enable_gift_maps.value | ||
| 246 | elif game_map.name == "the_stellar": | ||
| 247 | return "The Stellar" in world.options.enable_gift_maps.value | ||
| 248 | |||
| 249 | return False | ||
| 250 | |||
| 251 | self.shuffled_maps = set(game_map.id for game_map in world.static_logic.objects.maps | ||
| 252 | if should_shuffle_map(game_map)) | ||
| 253 | |||
| 254 | maximum_masteries = 13 + len(world.options.enable_gift_maps.value) | ||
| 255 | if world.options.enable_icarus: | ||
| 256 | maximum_masteries += 1 | ||
| 257 | |||
| 258 | if world.options.masteries_requirement > maximum_masteries: | ||
| 259 | world.options.masteries_requirement.value = maximum_masteries | ||
| 260 | |||
| 261 | if "The Fuzzy" in world.options.enable_gift_maps.value: | ||
| 262 | self.real_items.append("Numbers") | ||
| 263 | |||
| 264 | if self.world.options.shuffle_doors: | ||
| 265 | for progressive in world.static_logic.objects.progressives: | ||
| 266 | for i in range(0, len(progressive.doors)): | ||
| 267 | door = world.static_logic.objects.doors[progressive.doors[i]] | ||
| 268 | if door.map_id not in self.shuffled_maps: | ||
| 269 | continue | ||
| 270 | |||
| 271 | self.item_by_door[progressive.doors[i]] = (progressive.name, i + 1) | ||
| 272 | self.real_items.append(progressive.name) | ||
| 273 | |||
| 274 | for door_group in world.static_logic.objects.door_groups: | ||
| 275 | if door_group.type == data_pb2.DoorGroupType.CONNECTOR: | ||
| 276 | if not self.world.options.shuffle_doors or self.world.options.shuffle_worldports: | ||
| 277 | continue | ||
| 278 | elif door_group.type == data_pb2.DoorGroupType.COLOR_CONNECTOR: | ||
| 279 | if not self.world.options.shuffle_control_center_colors or self.world.options.shuffle_worldports: | ||
| 280 | continue | ||
| 281 | elif door_group.type == data_pb2.DoorGroupType.SHUFFLE_GROUP: | ||
| 282 | if not self.world.options.shuffle_doors: | ||
| 283 | continue | ||
| 284 | else: | ||
| 285 | continue | ||
| 286 | |||
| 287 | shuffleable_doors = [door_id for door_id in door_group.doors | ||
| 288 | if world.static_logic.objects.doors[door_id].map_id in self.shuffled_maps] | ||
| 289 | |||
| 290 | if len(shuffleable_doors) > 0: | ||
| 291 | for door in shuffleable_doors: | ||
| 292 | self.item_by_door[door] = (door_group.name, 1) | ||
| 293 | |||
| 294 | self.real_items.append(door_group.name) | ||
| 103 | 295 | ||
| 104 | # We iterate through the doors in two parts because it is essential that we determine which doors are shuffled | 296 | # We iterate through the doors in two parts because it is essential that we determine which doors are shuffled |
| 105 | # before we calculate any access requirements. | 297 | # before we calculate any access requirements. |
| 106 | for door in world.static_logic.objects.doors: | 298 | for door in world.static_logic.objects.doors: |
| 107 | if door.type in [data_pb2.DoorType.STANDARD, data_pb2.DoorType.ITEM_ONLY] and self.world.options.shuffle_doors: | 299 | if door.map_id not in self.shuffled_maps: |
| 108 | door_item_name = self.world.static_logic.get_door_item_name(door.id) | 300 | continue |
| 109 | self.item_by_door[door.id] = door_item_name | 301 | |
| 110 | self.real_items.append(door_item_name) | 302 | if door.type in [data_pb2.DoorType.EVENT, data_pb2.DoorType.LOCATION_ONLY, data_pb2.DoorType.GRAVESTONE]: |
| 303 | continue | ||
| 304 | |||
| 305 | if door.id in self.item_by_door: | ||
| 306 | continue | ||
| 307 | |||
| 308 | if (door.type in [data_pb2.DoorType.STANDARD, data_pb2.DoorType.ITEM_ONLY] and | ||
| 309 | not self.world.options.shuffle_doors): | ||
| 310 | continue | ||
| 311 | |||
| 312 | if (door.type == data_pb2.DoorType.CONTROL_CENTER_COLOR and | ||
| 313 | not self.world.options.shuffle_control_center_colors): | ||
| 314 | continue | ||
| 315 | |||
| 316 | if door.type == data_pb2.DoorType.GALLERY_PAINTING and not self.world.options.shuffle_gallery_paintings: | ||
| 317 | continue | ||
| 318 | |||
| 319 | door_item_name = self.world.static_logic.get_door_item_name(door) | ||
| 320 | self.item_by_door[door.id] = (door_item_name, 1) | ||
| 321 | self.real_items.append(door_item_name) | ||
| 322 | |||
| 323 | # We handle cyan_door_behavior = Item after door shuffle, because cyan doors that are impacted by door shuffle | ||
| 324 | # should be exempt from cyan_door_behavior. | ||
| 325 | if world.options.cyan_door_behavior == CyanDoorBehavior.option_item: | ||
| 326 | for door_group in world.static_logic.objects.door_groups: | ||
| 327 | if door_group.type != data_pb2.DoorGroupType.CYAN_DOORS: | ||
| 328 | continue | ||
| 329 | |||
| 330 | shuffleable_doors = [door_id for door_id in door_group.doors | ||
| 331 | if world.static_logic.objects.doors[door_id].map_id in self.shuffled_maps | ||
| 332 | and door_id not in self.item_by_door] | ||
| 333 | |||
| 334 | if len(shuffleable_doors) > 0: | ||
| 335 | for door in shuffleable_doors: | ||
| 336 | self.item_by_door[door] = (door_group.name, 1) | ||
| 337 | |||
| 338 | self.real_items.append(door_group.name) | ||
| 111 | 339 | ||
| 112 | for door in world.static_logic.objects.doors: | 340 | for door in world.static_logic.objects.doors: |
| 341 | if door.map_id not in self.shuffled_maps: | ||
| 342 | continue | ||
| 343 | |||
| 113 | if door.type in [data_pb2.DoorType.STANDARD, data_pb2.DoorType.LOCATION_ONLY, data_pb2.DoorType.GRAVESTONE]: | 344 | if door.type in [data_pb2.DoorType.STANDARD, data_pb2.DoorType.LOCATION_ONLY, data_pb2.DoorType.GRAVESTONE]: |
| 114 | self.locations_by_room.setdefault(door.room_id, []).append(PlayerLocation(door.ap_id, | 345 | self.locations_by_room.setdefault(door.room_id, []).append(PlayerLocation(door.ap_id, |
| 115 | self.get_door_reqs(door.id))) | 346 | self.get_door_reqs(door.id))) |
| 116 | 347 | ||
| 117 | for letter in world.static_logic.objects.letters: | 348 | for letter in world.static_logic.objects.letters: |
| 349 | if world.static_logic.get_room_object_map_id(letter) not in self.shuffled_maps: | ||
| 350 | continue | ||
| 351 | |||
| 118 | self.locations_by_room.setdefault(letter.room_id, []).append(PlayerLocation(letter.ap_id, | 352 | self.locations_by_room.setdefault(letter.room_id, []).append(PlayerLocation(letter.ap_id, |
| 119 | AccessRequirements())) | 353 | AccessRequirements())) |
| 120 | 354 | behavior = self.get_letter_behavior(letter.key, letter.level2) | |
| 121 | letter_name = f"{letter.key.upper()}{'2' if letter.level2 else '1'}" | 355 | if behavior == LetterBehavior.VANILLA: |
| 122 | event_name = f"{letter_name} (Collected)" | 356 | if not world.for_tracker: |
| 123 | self.event_loc_item_by_room.setdefault(letter.room_id, {})[event_name] = letter.key.upper() | 357 | letter_name = f"{letter.key.upper()}{'2' if letter.level2 else '1'}" |
| 124 | 358 | event_name = f"{letter_name} (Collected)" | |
| 125 | if letter.level2: | 359 | self.event_loc_item_by_room.setdefault(letter.room_id, {})[event_name] = letter.key.upper() |
| 126 | event_name = f"{letter_name} (Double Collected)" | 360 | |
| 127 | self.event_loc_item_by_room.setdefault(letter.room_id, {})[event_name] = letter.key.upper() | 361 | if letter.level2: |
| 362 | event_name = f"{letter_name} (Double Collected)" | ||
| 363 | self.event_loc_item_by_room.setdefault(letter.room_id, {})[event_name] = letter.key.upper() | ||
| 364 | elif behavior == LetterBehavior.ITEM: | ||
| 365 | self.real_items.append(letter.key.upper()) | ||
| 366 | |||
| 367 | if behavior != LetterBehavior.UNLOCKED: | ||
| 368 | self.double_letter_amount[letter.key.upper()] = self.double_letter_amount.get(letter.key.upper(), 0) + 1 | ||
| 128 | 369 | ||
| 129 | for mastery in world.static_logic.objects.masteries: | 370 | for mastery in world.static_logic.objects.masteries: |
| 371 | if world.static_logic.get_room_object_map_id(mastery) not in self.shuffled_maps: | ||
| 372 | continue | ||
| 373 | |||
| 130 | self.locations_by_room.setdefault(mastery.room_id, []).append(PlayerLocation(mastery.ap_id, | 374 | self.locations_by_room.setdefault(mastery.room_id, []).append(PlayerLocation(mastery.ap_id, |
| 131 | AccessRequirements())) | 375 | AccessRequirements())) |
| 132 | 376 | ||
| 377 | if world.options.masteries_requirement > 0: | ||
| 378 | event_name = f"{world.static_logic.get_room_object_map_name(mastery)} - Mastery (Collected)" | ||
| 379 | self.event_loc_item_by_room.setdefault(mastery.room_id, {})[event_name] = "Mastery" | ||
| 380 | |||
| 133 | for ending in world.static_logic.objects.endings: | 381 | for ending in world.static_logic.objects.endings: |
| 134 | self.locations_by_room.setdefault(ending.room_id, []).append(PlayerLocation(ending.ap_id, | 382 | if world.static_logic.get_room_object_map_id(ending) not in self.shuffled_maps: |
| 135 | AccessRequirements())) | 383 | continue |
| 136 | 384 | ||
| 137 | event_name = f"{ending.name.capitalize()} Ending (Achieved)" | 385 | # Don't create a location for your selected ending. Also don't create a location for White Ending if it's |
| 138 | item_name = event_name | 386 | # necessarily in the postgame, i.e. it requires all 12 other endings. |
| 387 | if world.options.victory_condition.current_key.removesuffix("_ending").upper() != ending.name\ | ||
| 388 | and (ending.name != "WHITE" or world.options.endings_requirement < 12): | ||
| 389 | self.locations_by_room.setdefault(ending.room_id, []).append(PlayerLocation(ending.ap_id, | ||
| 390 | AccessRequirements())) | ||
| 139 | 391 | ||
| 140 | if world.options.victory_condition.current_key.removesuffix("_ending").upper() == ending.name: | 392 | if world.options.victory_condition.current_key.removesuffix("_ending").upper() == ending.name: |
| 141 | item_name = "Victory" | 393 | event_name = f"{ending.name.capitalize()} Ending (Goal)" |
| 394 | self.event_loc_item_by_room.setdefault(ending.room_id, {})[event_name] = "Victory" | ||
| 395 | self.goal_room_id = ending.room_id | ||
| 396 | |||
| 397 | if ending.name != "WHITE": | ||
| 398 | event_name = f"{ending.name.capitalize()} Ending (Achieved)" | ||
| 399 | self.event_loc_item_by_room.setdefault(ending.room_id, {})[event_name] = "Ending" | ||
| 400 | |||
| 401 | if self.world.options.keyholder_sanity: | ||
| 402 | for keyholder in world.static_logic.objects.keyholders: | ||
| 403 | if keyholder.HasField("key"): | ||
| 404 | if world.static_logic.get_room_object_map_id(keyholder) not in self.shuffled_maps: | ||
| 405 | continue | ||
| 406 | |||
| 407 | reqs = AccessRequirements() | ||
| 142 | 408 | ||
| 143 | self.event_loc_item_by_room.setdefault(ending.room_id, {})[event_name] = item_name | 409 | if self.get_letter_behavior(keyholder.key, False) != LetterBehavior.UNLOCKED: |
| 410 | reqs.letters[keyholder.key.upper()] = 1 | ||
| 411 | |||
| 412 | self.locations_by_room.setdefault(keyholder.room_id, []).append(PlayerLocation(keyholder.ap_id, | ||
| 413 | reqs)) | ||
| 414 | |||
| 415 | if self.world.options.shuffle_symbols: | ||
| 416 | for symbol_name in SYMBOL_ITEMS.values(): | ||
| 417 | self.real_items.append(symbol_name) | ||
| 144 | 418 | ||
| 145 | def get_panel_reqs(self, panel_id: int, answer: str | None) -> AccessRequirements: | 419 | def get_panel_reqs(self, panel_id: int, answer: str | None) -> AccessRequirements: |
| 146 | if answer is None: | 420 | if answer is None: |
| @@ -161,28 +435,38 @@ class Lingo2PlayerLogic: | |||
| 161 | reqs.rooms.add(self.world.static_logic.get_room_region_name(panel.room_id)) | 435 | reqs.rooms.add(self.world.static_logic.get_room_region_name(panel.room_id)) |
| 162 | 436 | ||
| 163 | if answer is not None: | 437 | if answer is not None: |
| 164 | reqs.add_solution(answer) | 438 | self.add_solution_reqs(reqs, answer) |
| 165 | elif len(panel.proxies) > 0: | 439 | elif len(panel.proxies) > 0: |
| 166 | possibilities = [] | 440 | possibilities = [] |
| 441 | already_filled = False | ||
| 167 | 442 | ||
| 168 | for proxy in panel.proxies: | 443 | for proxy in panel.proxies: |
| 169 | proxy_reqs = AccessRequirements() | 444 | proxy_reqs = AccessRequirements() |
| 170 | proxy_reqs.add_solution(proxy.answer) | 445 | self.add_solution_reqs(proxy_reqs, proxy.answer) |
| 171 | 446 | ||
| 172 | possibilities.append(proxy_reqs) | 447 | if not proxy_reqs.is_empty(): |
| 448 | possibilities.append(proxy_reqs) | ||
| 449 | else: | ||
| 450 | already_filled = True | ||
| 451 | break | ||
| 173 | 452 | ||
| 174 | if not any(proxy.answer == panel.answer for proxy in panel.proxies): | 453 | if not already_filled and not any(proxy.answer == panel.answer for proxy in panel.proxies): |
| 175 | proxy_reqs = AccessRequirements() | 454 | proxy_reqs = AccessRequirements() |
| 176 | proxy_reqs.add_solution(panel.answer) | 455 | self.add_solution_reqs(proxy_reqs, panel.answer) |
| 177 | 456 | ||
| 178 | possibilities.append(proxy_reqs) | 457 | if not proxy_reqs.is_empty(): |
| 458 | possibilities.append(proxy_reqs) | ||
| 459 | else: | ||
| 460 | already_filled = True | ||
| 179 | 461 | ||
| 180 | reqs.or_logic.append(possibilities) | 462 | if not already_filled: |
| 463 | reqs.or_logic.append(possibilities) | ||
| 181 | else: | 464 | else: |
| 182 | reqs.add_solution(panel.answer) | 465 | self.add_solution_reqs(reqs, panel.answer) |
| 183 | 466 | ||
| 184 | for symbol in panel.symbols: | 467 | if self.world.options.shuffle_symbols: |
| 185 | reqs.symbols.add(symbol) | 468 | for symbol in panel.symbols: |
| 469 | reqs.items.add(SYMBOL_ITEMS.get(symbol)) | ||
| 186 | 470 | ||
| 187 | if panel.HasField("required_door"): | 471 | if panel.HasField("required_door"): |
| 188 | door_reqs = self.get_door_open_reqs(panel.required_door) | 472 | door_reqs = self.get_door_open_reqs(panel.required_door) |
| @@ -205,22 +489,45 @@ class Lingo2PlayerLogic: | |||
| 205 | door = self.world.static_logic.objects.doors[door_id] | 489 | door = self.world.static_logic.objects.doors[door_id] |
| 206 | reqs = AccessRequirements() | 490 | reqs = AccessRequirements() |
| 207 | 491 | ||
| 208 | # TODO: control_center_color, switches | ||
| 209 | if not door.HasField("complete_at") or door.complete_at == 0: | 492 | if not door.HasField("complete_at") or door.complete_at == 0: |
| 210 | for proxy in door.panels: | 493 | for proxy in door.panels: |
| 211 | panel_reqs = self.get_panel_reqs(proxy.panel, proxy.answer if proxy.HasField("answer") else None) | 494 | panel_reqs = self.get_panel_reqs(proxy.panel, proxy.answer if proxy.HasField("answer") else None) |
| 212 | reqs.merge(panel_reqs) | 495 | reqs.merge(panel_reqs) |
| 213 | elif door.complete_at == 1: | 496 | elif door.complete_at == 1: |
| 214 | reqs.or_logic.append([self.get_panel_reqs(proxy.panel, | 497 | disjunction = [] |
| 215 | proxy.answer if proxy.HasField("answer") else None) | 498 | for proxy in door.panels: |
| 216 | for proxy in door.panels]) | 499 | proxy_reqs = self.get_panel_reqs(proxy.panel, proxy.answer if proxy.HasField("answer") else None) |
| 500 | if proxy_reqs.is_empty(): | ||
| 501 | disjunction.clear() | ||
| 502 | break | ||
| 503 | else: | ||
| 504 | disjunction.append(proxy_reqs) | ||
| 505 | if len(disjunction) > 0: | ||
| 506 | reqs.or_logic.append(disjunction) | ||
| 217 | else: | 507 | else: |
| 218 | # TODO: Handle complete_at > 1 | 508 | reqs.complete_at = door.complete_at |
| 219 | pass | 509 | for proxy in door.panels: |
| 510 | panel_reqs = self.get_panel_reqs(proxy.panel, proxy.answer if proxy.HasField("answer") else None) | ||
| 511 | reqs.possibilities.append(panel_reqs) | ||
| 512 | |||
| 513 | if door.HasField("control_center_color"): | ||
| 514 | reqs.rooms.add("Control Center - Main Area") | ||
| 515 | self.add_solution_reqs(reqs, door.control_center_color) | ||
| 516 | |||
| 517 | if door.double_letters: | ||
| 518 | if self.world.options.cyan_door_behavior == CyanDoorBehavior.option_collect_h2: | ||
| 519 | reqs.rooms.add("The Repetitive - Main Room") | ||
| 520 | elif self.world.options.cyan_door_behavior == CyanDoorBehavior.option_any_double_letter: | ||
| 521 | if self.world.options.shuffle_letters != ShuffleLetters.option_unlocked: | ||
| 522 | reqs.cyans = True | ||
| 523 | elif self.world.options.cyan_door_behavior == CyanDoorBehavior.option_item: | ||
| 524 | # There shouldn't be any locations that are cyan doors. | ||
| 525 | pass | ||
| 220 | 526 | ||
| 221 | for keyholder_uses in door.keyholders: | 527 | for keyholder_uses in door.keyholders: |
| 222 | key_name = keyholder_uses.key.upper() | 528 | key_name = keyholder_uses.key.upper() |
| 223 | if key_name not in reqs.letters: | 529 | if (self.get_letter_behavior(keyholder_uses.key, False) != LetterBehavior.UNLOCKED |
| 530 | and key_name not in reqs.letters): | ||
| 224 | reqs.letters[key_name] = 1 | 531 | reqs.letters[key_name] = 1 |
| 225 | 532 | ||
| 226 | keyholder = self.world.static_logic.objects.keyholders[keyholder_uses.keyholder] | 533 | keyholder = self.world.static_logic.objects.keyholders[keyholder_uses.keyholder] |
| @@ -229,10 +536,19 @@ class Lingo2PlayerLogic: | |||
| 229 | for room in door.rooms: | 536 | for room in door.rooms: |
| 230 | reqs.rooms.add(self.world.static_logic.get_room_region_name(room)) | 537 | reqs.rooms.add(self.world.static_logic.get_room_region_name(room)) |
| 231 | 538 | ||
| 539 | if door.white_ending: | ||
| 540 | if self.world.options.endings_requirement > 0: | ||
| 541 | reqs.progressives["Ending"] = self.world.options.endings_requirement.value | ||
| 542 | |||
| 543 | if self.world.options.masteries_requirement > 0: | ||
| 544 | reqs.progressives["Mastery"] = self.world.options.masteries_requirement.value | ||
| 545 | |||
| 232 | for sub_door_id in door.doors: | 546 | for sub_door_id in door.doors: |
| 233 | sub_reqs = self.get_door_open_reqs(sub_door_id) | 547 | sub_reqs = self.get_door_open_reqs(sub_door_id) |
| 234 | reqs.merge(sub_reqs) | 548 | reqs.merge(sub_reqs) |
| 235 | 549 | ||
| 550 | reqs.simplify() | ||
| 551 | |||
| 236 | return reqs | 552 | return reqs |
| 237 | 553 | ||
| 238 | # This gets the requirements to open a door within the world. When a door is shuffled, this means having the item | 554 | # This gets the requirements to open a door within the world. When a door is shuffled, this means having the item |
| @@ -240,8 +556,53 @@ class Lingo2PlayerLogic: | |||
| 240 | def get_door_open_reqs(self, door_id: int) -> AccessRequirements: | 556 | def get_door_open_reqs(self, door_id: int) -> AccessRequirements: |
| 241 | if door_id in self.item_by_door: | 557 | if door_id in self.item_by_door: |
| 242 | reqs = AccessRequirements() | 558 | reqs = AccessRequirements() |
| 243 | reqs.items.add(self.item_by_door.get(door_id)) | 559 | |
| 560 | item_name, amount = self.item_by_door.get(door_id) | ||
| 561 | if amount == 1: | ||
| 562 | reqs.items.add(item_name) | ||
| 563 | else: | ||
| 564 | reqs.progressives[item_name] = amount | ||
| 244 | 565 | ||
| 245 | return reqs | 566 | return reqs |
| 246 | else: | 567 | else: |
| 247 | return self.get_door_reqs(door_id) | 568 | return self.get_door_reqs(door_id) |
| 569 | |||
| 570 | def get_letter_behavior(self, letter: str, level2: bool) -> LetterBehavior: | ||
| 571 | if self.world.options.shuffle_letters == ShuffleLetters.option_unlocked: | ||
| 572 | return LetterBehavior.UNLOCKED | ||
| 573 | |||
| 574 | if self.world.options.shuffle_letters in [ShuffleLetters.option_vanilla_cyan, ShuffleLetters.option_item_cyan]: | ||
| 575 | if level2: | ||
| 576 | if self.world.options.shuffle_letters == ShuffleLetters.option_vanilla_cyan: | ||
| 577 | return LetterBehavior.VANILLA | ||
| 578 | else: | ||
| 579 | return LetterBehavior.ITEM | ||
| 580 | else: | ||
| 581 | return LetterBehavior.UNLOCKED | ||
| 582 | |||
| 583 | if not level2 and letter in ["h", "i", "n", "t"]: | ||
| 584 | return LetterBehavior.UNLOCKED | ||
| 585 | |||
| 586 | if self.world.options.shuffle_letters == ShuffleLetters.option_progressive: | ||
| 587 | return LetterBehavior.ITEM | ||
| 588 | |||
| 589 | return LetterBehavior.VANILLA | ||
| 590 | |||
| 591 | def add_solution_reqs(self, reqs: AccessRequirements, solution: str): | ||
| 592 | histogram = calculate_letter_histogram(solution) | ||
| 593 | |||
| 594 | for l, a in histogram.items(): | ||
| 595 | needed = min(a, 2) | ||
| 596 | level2 = (needed == 2) | ||
| 597 | |||
| 598 | if level2 and self.get_letter_behavior(l.lower(), True) == LetterBehavior.UNLOCKED: | ||
| 599 | needed = 1 | ||
| 600 | |||
| 601 | if self.get_letter_behavior(l.lower(), False) == LetterBehavior.UNLOCKED: | ||
| 602 | needed = needed - 1 | ||
| 603 | |||
| 604 | if needed > 0: | ||
| 605 | reqs.letters[l] = max(reqs.letters.get(l, 0), needed) | ||
| 606 | |||
| 607 | if any(l.isnumeric() for l in solution): | ||
| 608 | reqs.items.add("Numbers") | ||
