1 files changed, 67 insertions, 0 deletions

@@ -0,0 +1,67 @@
+from BaseClasses import MultiWorld, CollectionState, ItemClassification
+from NetUtils import NetworkItem
+from . import Lingo2World, Lingo2Item
+from .regions import connect_ports_from_ut
+from .options import Lingo2Options
+
+PLAYER_NUM = 1
+
+
+class Tracker:
+    multiworld: MultiWorld
+
+    collected_items: dict[int, int]
+    checked_locations: set[int]
+    accessible_locations: set[int]
+
+    state: CollectionState
+
+    def __init__(self):
+        self.collected_items = {}
+        self.checked_locations = set()
+        self.accessible_locations = set()
+
+    def setup_slot(self, slot_data):
+        self.multiworld = MultiWorld(players=PLAYER_NUM)
+        world = Lingo2World(self.multiworld, PLAYER_NUM)
+        self.multiworld.worlds[1] = world
+        world.options = Lingo2Options(**{k: t(slot_data.get(k, t.default))
+                                         for k, t in Lingo2Options.type_hints.items()})
+
+        world.generate_early()
+        world.create_regions()
+
+        if world.options.shuffle_worldports:
+            port_pairings = {int(fp): int(tp) for fp, tp in slot_data["port_pairings"].items()}
+            connect_ports_from_ut(port_pairings, world)
+
+        self.state = CollectionState(self.multiworld)
+
+    def set_checked_locations(self, checked_locations: set[int]):
+        self.checked_locations = checked_locations.copy()
+
+    def set_collected_items(self, network_items: list[NetworkItem]):
+        self.collected_items = {}
+
+        for item in network_items:
+            self.collected_items[item.item] = self.collected_items.get(item.item, 0) + 1
+
+        self.refresh_state()
+
+    def refresh_state(self):
+        self.state = CollectionState(self.multiworld)
+
+        for item_id, item_amount in self.collected_items.items():
+            for i in range(item_amount):
+                self.state.collect(Lingo2Item(Lingo2World.static_logic.item_id_to_name.get(item_id),
+                                              ItemClassification.progression, item_id, PLAYER_NUM), prevent_sweep=True)
+
+        self.state.sweep_for_advancements()
+
+        self.accessible_locations = set()
+
+        for region in self.state.reachable_regions[PLAYER_NUM]:
+            for location in region.locations:
+                if location.address not in self.checked_locations and location.access_rule(self.state):
+                    if location.address is not None:
+                        self.accessible_locations.add(location.address)