Simplified engine by adding new class to extend

1 files changed, 5 insertions, 16 deletions

diff --git a/MAFH2/TermBox.py b/MAFH2/TermBox.py
index 7eb9471..c9d09f8 100644
--- a/MAFH2/TermBox.py
+++ b/MAFH2/TermBox.py
@@ -1,29 +1,18 @@
 import pygame
+from GameEngine import GameEngineElement
+class TermBox(GameEngineElement):
+    def __init__(self, x,y,width,height,lines):
+        GameEngineElement.__init__(self, has_draw=True, has_event=False)
 
-class TermBox:
-    def __init__(self, ge, x,y,width,height,lines):
-        self.ge = ge
         self.max_lines = lines
-
         self.x = x
         self.y = y
         self.width = width
         self.height = height
         self.font = pygame.font.Font(None, 32)
-
         self.__lines = []
-        self.__inEngine = False
-        self.add_to_engine()
 
-    def add_to_engine(self):
-        if not self.__inEngine:
-            self.__inEngine = True
-            self.ge.add_draw_callback( self.draw )
-
-    def remove_from_engine(self):
-        if self.__inEngine:
-            self.__inEngine = False
-            self.ge.remove_draw_callback( self.draw )
+        self.add_to_engine()
 
     def add_line(self, line):
         self.__lines.append( line )