Mercurial > pyntnclick

annotate pyntnclick/engine.py @ 608:a25cd1c6335a pyntnclick

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Pass screens through engine by name
author Neil Muller <neil@dip.sun.ac.za>
date Sat, 11 Feb 2012 20:57:28 +0200
parents f2b1ddcc4986
children cfc16ded7b34
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
596
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
1 """Game engine and top-level game loop."""
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
2
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
3 import pygame
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
4 import pygame.event
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
5 import pygame.display
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
6 import pygame.time
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
7 from pygame.locals import QUIT, USEREVENT
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
8
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
9
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
10 class Engine(object):
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
11 def __init__(self, game_description):
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
12 self._screen = None
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
13 self._game_description = game_description
608
a25cd1c6335a Pass screens through engine by name
Neil Muller <neil@dip.sun.ac.za>
parents: 607
diff changeset
14 self.screens = {}
596
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
15
608
a25cd1c6335a Pass screens through engine by name
Neil Muller <neil@dip.sun.ac.za>
parents: 607
diff changeset
16 def set_screen(self, screen_name):
596
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
17 if self._screen is not None:
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
18 self._screen.on_exit()
608
a25cd1c6335a Pass screens through engine by name
Neil Muller <neil@dip.sun.ac.za>
parents: 607
diff changeset
19 self._screen = self.screens.get(screen_name, None)
596
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
20 if self._screen is not None:
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
21 self._screen.on_enter()
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
22
608
a25cd1c6335a Pass screens through engine by name
Neil Muller <neil@dip.sun.ac.za>
parents: 607
diff changeset
23 def add_screen(self, name, screen):
a25cd1c6335a Pass screens through engine by name
Neil Muller <neil@dip.sun.ac.za>
parents: 607
diff changeset
24 self.screens[name] = screen
a25cd1c6335a Pass screens through engine by name
Neil Muller <neil@dip.sun.ac.za>
parents: 607
diff changeset
25
596
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
26 def run(self):
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
27 """Game loop."""
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
28
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
29 get_events = pygame.event.get
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
30 flip = pygame.display.flip
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
31 clock = pygame.time.Clock()
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
32 while True:
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
33 events = get_events()
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
34 for ev in events:
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
35 if ev.type == QUIT:
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
36 return
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
37 elif ScreenChangeEvent.matches(ev):
608
a25cd1c6335a Pass screens through engine by name
Neil Muller <neil@dip.sun.ac.za>
parents: 607
diff changeset
38 self.set_screen(ev.screen_name)
596
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
39 else:
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
40 self._screen.dispatch(ev)
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
41 surface = pygame.display.get_surface()
600
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
42 self._screen.draw(surface)
596
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
43 flip()
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
44 self._fps = 1000.0 / clock.tick(
600
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
45 self._game_description.constants.frame_rate)
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
46
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
47
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
48 class Screen(object):
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
49 """A top level object for the screen being displayed"""
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
50
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
51 def __init__(self, game_description):
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
52 # Avoid import loop
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
53 from pyntnclick.widgets.base import Container
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
54
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
55 self.game_description = game_description
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
56 self.resource = game_description.resource
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
57
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
58 self.surface_size = game_description.constants.screen
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
59 self.surface = None
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
60 self.container = Container(pygame.Rect((0, 0), self.surface_size))
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
61 self.setup()
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
62
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
63 def on_enter(self):
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
64 """Called when this becomes the current screen."""
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
65 # Create the surface here as flipping between editor and
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
66 # other things kills pygame.display
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
67 self.surface = pygame.Surface(self.surface_size)
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
68
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
69 def on_exit(self):
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
70 """Called when this stops being the current screen."""
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
71 self.surface = None
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
72
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
73 def setup(self):
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
74 """Override for initialization"""
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
75 pass
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
76
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
77 def dispatch(self, ev):
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
78 self.container.event(ev)
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
79
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
80 def draw_background(self):
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
81 self.surface.fill(pygame.Color('gray'))
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
82
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
83 def draw(self, surface):
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
84 if self.surface:
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
85 self.draw_background()
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
86 self.container.draw(self.surface)
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
87 surface.blit(self.surface, self.surface.get_rect())
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
88
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
89 def display_dialog(self, dialog):
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
90 self.container.paused = True
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
91 self.container.add(dialog)
fabce47e542f Stop using albow (at least for the menu). Breaks the world. Please fix it, kthx
Stefano Rivera <stefano@rivera.za.net>
parents: 596
diff changeset
92 dialog.grab_focus()
596
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
93
608
a25cd1c6335a Pass screens through engine by name
Neil Muller <neil@dip.sun.ac.za>
parents: 607
diff changeset
94 def change_screen(self, new_screen_name):
a25cd1c6335a Pass screens through engine by name
Neil Muller <neil@dip.sun.ac.za>
parents: 607
diff changeset
95 ScreenChangeEvent.post(new_screen_name)
607
f2b1ddcc4986 Make menu and endscreen work.
Jeremy Thurgood <firxen@gmail.com>
parents: 600
diff changeset
96
596
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
97
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
98 class UserEvent(object):
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
99 """A user event type allowing subclassing,
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
100 to provide an infinate number of user-defined events
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
101 """
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
102
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
103 TYPE = "UNKNOWN"
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
104
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
105 @classmethod
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
106 def post(cls, **kws):
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
107 ev = pygame.event.Event(USEREVENT, utype=cls.TYPE, **kws)
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
108 pygame.event.post(ev)
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
109
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
110 @classmethod
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
111 def matches(cls, ev):
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
112 return ev.type == USEREVENT and ev.utype == cls.TYPE
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
113
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
114
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
115 class ScreenChangeEvent(UserEvent):
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
116
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
117 TYPE = "SCREEN_CHANGE"
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
118
5496dcd16a84 Add pyntnclick.engine, based on mamba
Stefano Rivera <stefano@rivera.za.net>
parents:
diff changeset
119 @classmethod
608
a25cd1c6335a Pass screens through engine by name
Neil Muller <neil@dip.sun.ac.za>
parents: 607
diff changeset
120 def post(cls, screen_name):
a25cd1c6335a Pass screens through engine by name
Neil Muller <neil@dip.sun.ac.za>
parents: 607
diff changeset
121 super(ScreenChangeEvent, cls).post(screen_name=screen_name)