Mercurial > skaapsteker
annotate skaapsteker/engine.py @ 126:9cae256b91d7
Fix bug in FPS calculation. Phew -- we're still fast.
author | Simon Cross <hodgestar@gmail.com> |
---|---|
date | Mon, 04 Apr 2011 23:38:03 +0200 |
parents | 592477d8b09b |
children | e1dd3b785269 |
rev | line source |
---|---|
11
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
1 """Top-level engine for switching scenes.""" |
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
2 |
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
3 import pygame.display |
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
4 import pygame.time |
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
5 import pygame.event |
14 | 6 from pygame.locals import QUIT, USEREVENT |
11
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
7 |
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
8 class Engine(object): |
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
9 |
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
10 def __init__(self): |
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
11 self._framerate = 60 |
14 | 12 self._current_scene = None |
123 | 13 self._fpss = [self._framerate] * 100 |
14 self._cur_frame = 0 | |
14 | 15 |
16 def change_scene(self, next_scene): | |
17 if self._current_scene is not None: | |
18 self._current_scene.leave() | |
19 self._current_scene = next_scene | |
20 self._current_scene.enter() | |
11
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
21 |
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
22 def run(self): |
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
23 """Run the game loop dispatching events as necessary.""" |
14 | 24 assert self._current_scene is not None |
11
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
25 clock = pygame.time.Clock() |
14 | 26 surface = pygame.display.get_surface() |
11
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
27 while True: |
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
28 events = pygame.event.get() |
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
29 for ev in events: |
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
30 if ev.type is QUIT: |
249814efa91b
Create game engine. Enable quitting.
Simon Cross <hodgestar@gmail.com>
parents:
diff
changeset
|
31 return |
14 | 32 if ChangeScene.matches(ev): |
33 self.change_scene(ev.next_scene) | |
34 break | |
35 self._current_scene.dispatch(ev) | |
123 | 36 self._current_scene.draw(surface, self) |
14 | 37 pygame.display.flip() |
126
9cae256b91d7
Fix bug in FPS calculation. Phew -- we're still fast.
Simon Cross <hodgestar@gmail.com>
parents:
123
diff
changeset
|
38 self._fpss[self._cur_frame] = 1000.0 / clock.tick(self._framerate) |
123 | 39 self._cur_frame = self._cur_frame + 1 if self._cur_frame < 99 else 0 |
40 | |
41 def get_fps(self): | |
42 return sum(self._fpss) / 100 | |
14 | 43 |
44 | |
45 class Scene(object): | |
46 | |
47 def __init__(self): | |
79
fe8652a060df
Simple menu screen and TextChoice widget
Stefano Rivera <stefano@rivera.za.net>
parents:
14
diff
changeset
|
48 self.widgets = [] |
14 | 49 |
50 def post(self, ev): | |
51 """Post an event to pygame's event loop.""" | |
52 pygame.event.post(ev) | |
53 | |
54 def enter(self): | |
55 """Enter the scene.""" | |
56 pass | |
57 | |
58 def leave(self): | |
59 """Exit the scene.""" | |
60 pass | |
61 | |
62 def dispatch(self, ev): | |
63 """Dispatch an event.""" | |
79
fe8652a060df
Simple menu screen and TextChoice widget
Stefano Rivera <stefano@rivera.za.net>
parents:
14
diff
changeset
|
64 for widget in self.widgets: |
fe8652a060df
Simple menu screen and TextChoice widget
Stefano Rivera <stefano@rivera.za.net>
parents:
14
diff
changeset
|
65 widget.dispatch(ev) |
14 | 66 |
67 def draw(self, surface): | |
68 """Update the scene surface.""" | |
79
fe8652a060df
Simple menu screen and TextChoice widget
Stefano Rivera <stefano@rivera.za.net>
parents:
14
diff
changeset
|
69 for widget in self.widgets: |
fe8652a060df
Simple menu screen and TextChoice widget
Stefano Rivera <stefano@rivera.za.net>
parents:
14
diff
changeset
|
70 widget.draw(surface) |
14 | 71 |
72 | |
73 class UserEvent(object): | |
74 | |
75 utype = "UNKNOWN" | |
76 | |
77 @classmethod | |
78 def post(cls, **kws): | |
79 ev = pygame.event.Event(USEREVENT, utype=cls.utype, **kws) | |
80 pygame.event.post(ev) | |
81 | |
82 @classmethod | |
83 def matches(cls, ev): | |
84 return ev.type is USEREVENT and ev.utype == cls.utype | |
85 | |
86 class ChangeScene(UserEvent): | |
87 | |
88 utype = "CHANGE_SCENE" | |
89 | |
90 @classmethod | |
91 def post(cls, next_scene): | |
92 super(ChangeScene, cls).post(next_scene=next_scene) |