Super Metroid: Avoid unnecessary deep copies to improve generation speed.

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):

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