+""" 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
+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()
+
+ 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)
+
+ self._scene.render(self._screen, self._gamestate)
+ pygame.display.flip()
+ clock.tick(FPS)