source: nagslang/engine.py@ 20:347667c941de

"""Top-level event dispatching and area transition."""

import pygame

from nagslang import constants
from nagslang.screens.menu import MenuScreen
from nagslang.screens.area import AreaScreen
from nagslang.events import ScreenChange


class Engine(object):
    def __init__(self, surface):
        self._surface = surface
        self._clock = pygame.time.Clock()
        self._fps = constants.FPS
        self._world = None  # TODO: create the world
        self._current_screen = None
        self._screens = {
            'menu': MenuScreen,
            'level1': AreaScreen,
        }
        self.change_screen('menu')

    def change_screen(self, new_screen):
        if self._current_screen is not None:
            self._current_screen.teardown()
        self._current_screen = self._screens[new_screen]()
        self._current_screen.setup(self._world)

    def run(self):
        running = True
        while running:
            for ev in pygame.event.get():
                if ev.type == pygame.locals.QUIT:
                    running = False
                elif ScreenChange.matches(ev):
                    self.change_screen(ev.screen)
                else:
                    self._current_screen.handle_event(ev)
            self._current_screen.render(self._surface)
            pygame.display.flip()
            self._clock.tick(self._fps)