Add quit event.

[koperkapel.git] / koperkapel / scenes / base.py

diff --git a/koperkapel/scenes/base.py b/koperkapel/scenes/base.py
index 9bad13cd9f03b9e1d172c9c977ae575002a51c88..58f3874f204b87a874751d593f4360a785791a70 100644 (file)
--- a/koperkapel/scenes/base.py
+++ b/koperkapel/scenes/base.py
@@ -30,6 +30,10 @@ class Engine:
         self._scene = scene
         self._scene.enter()
 
+    def quit_game(self):
+        from pgzero.game import exit
+        exit()
+
     @apply_events
     def update(self, dt):
         return self._scene.update(dt)
@@ -77,9 +81,37 @@ class ChangeSceneEvent(Event):
     ENGINE_METHOD = "change_scene"
 
 
+class QuitEvent(Event):
+    """ Quit the game. """
+
+    ENGINE_METHOD = "quit_game"
+
+
+class Actors:
+    """ A list of actors. """
+
+    def __init__(self):
+        self._actors = []
+
+    def add(self, actor):
+        self._actors.append(actor)
+        return actor
+
+    def remove(self, actor):
+        self._actors.remove(actor)
+        return actor
+
+    def draw(self, screen):
+        for actor in self._actors:
+            actor.draw()  # TODO: allow an option screen to be passed in
+
+
 class Scene:
     """ Base class for scenes. """
 
+    def __init__(self):
+        self.actors = Actors()
+
     def enter(self):
         pass
 
@@ -90,7 +122,8 @@ class Scene:
         pass
 
     def draw(self, screen):
-        pass
+        screen.clear()
+        self.actors.draw(screen)
 
     def on_mouse_down(self, pos, button):
         pass