Display power usage per hour.

[tabakrolletjie.git] / tabakrolletjie / engine.py

diff --git a/tabakrolletjie/engine.py b/tabakrolletjie/engine.py
index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..cdea1608316b5b68c1e4af4eb5856bd21a9be2ad 100644 (file)
--- a/tabakrolletjie/engine.py
+++ b/tabakrolletjie/engine.py
@@ -0,0 +1,57 @@
+""" Engine for moving controll scenes and tranistions between them.
+"""
+
+import pygame.display
+import pygame.event
+import pygame.time
+
+import pygame.locals as pgl
+
+from .constants import FPS, DRAW_FPS, FONTS, FPS_FRAMES
+from .loader import loader
+from .events import QuitEvent, SceneChangeEvent
+
+
+class Engine(object):
+    def __init__(self, screen, gamestate):
+        self._screen = screen
+        self._gamestate = gamestate
+        self._scene = None
+
+    def set_scene(self, scene):
+        if self._scene is not None:
+            self._scene.exit(self._gamestate)
+        self._scene = scene
+        if scene is not None:
+            self._scene.enter(self._gamestate)
+
+    def run(self):
+        clock = pygame.time.Clock()
+        fpses = [FPS] * FPS_FRAMES
+        frame = 0
+        font = loader.load_font(FONTS['sans'], size=16)
+
+        while True:
+            events = pygame.event.get()
+            for ev in events:
+                if QuitEvent.matches(ev) or ev.type == pgl.QUIT:
+                    self.set_scene(None)
+                    return
+                elif SceneChangeEvent.matches(ev):
+                    self.set_scene(ev.scene)
+                else:
+                    self._scene.event(ev, self._gamestate)
+
+            # Advance time on the world
+            # Time is assumed to flow perfectly, so no dt parameter for now
+            self._scene.tick(self._gamestate)
+            self._scene.render(self._screen, self._gamestate)
+            if DRAW_FPS:
+                fps = sum(fpses) / FPS_FRAMES
+                text = font.render("FPS: %.2f" % fps, True, (255, 255, 255))
+                self._screen.blit(text, (10, 10), None)
+            pygame.display.flip()
+            fpses[frame] = 1000.0 / clock.tick(FPS)
+            frame += 1
+            if frame >= FPS_FRAMES:
+                frame = 0