SM: Prevent unnecessary deep copies

ArchipelagoMW · Nov 1, 2024 · bccfb99 · bccfb99
1 parent 931e335
commit bccfb99
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/worlds/sm/variaRandomizer/logic/smboolmanager.py b/worlds/sm/variaRandomizer/logic/smboolmanager.py
@@ -7,6 +7,7 @@
 from ..utils.doorsmanager import DoorsManager
 from ..utils.objectives import Objectives
 from ..utils.parameters import Knows, isKnows
+import copy
 import logging
 import sys
 
@@ -240,6 +241,17 @@ def energyReserveCountOk(self, count, difficulty=0):
             return SMBool(True, difficulty, items = [items])
         else:
             return smboolFalse
+
+    def __deepcopy__(self, memo=None):
+        cls = self.__class__
+        cp = cls.__new__(cls)
+        if memo is None:
+            memo = {}
+        memo[id(self)] = cp
+        for k, v in self.__dict__.items():
+            setattr(cp, k, copy.deepcopy(v, memo))
+        return cp
+
 
 class SMBoolManagerPlando(SMBoolManager):
     def __init__(self):

diff --git a/worlds/sm/variaRandomizer/utils/objectives.py b/worlds/sm/variaRandomizer/utils/objectives.py
@@ -804,3 +804,13 @@ def writeChar(c, frameDelay=2):
         # write trailer, see intro_text.asm
         rom.writeWord(0xAE5B)
         rom.writeWord(0x9698)
+
+    def __deepcopy__(self, memo=None):
+        cls = self.__class__
+        cp = cls.__new__(cls)
+        if memo is None:
+            memo = {}
+        memo[id(self.graph)] = self.graph
+        for k, v in self.__dict__.items():
+            setattr(cp, k, copy.deepcopy(v, memo))
+        return cp