refactoring to game base class

equipment.py

@@ -6,6 +6,7 @@ from hsvfilter import HsvFilter
 import pydirectinput
 from utils import dig_point
 from game_base_class import GameBase
+from window_capture import WindowCapture
 
 EMITTER_MAIN = "main"
 EMITTER_MUSH = "mushroom"
@@ -30,6 +31,9 @@ class Equipment(GameBase):
         self.masks = []
         self.emitters = []
 
+        # initialize the StunWindowCapture class
+        self.capture_window = WindowCapture(None, "equip", self.config)
+
         self.EMITTER_TO_USE = str(overlay.emitter_use.get())
         self.SPAWN_COUNT = int(overlay.spawn_use.get())
         self.ENERGY_TO_USE = int(overlay.energy_use.get())