Centre menu items
[naja.git] / naja / scenes / menu.py
1 """
2 Main menu scene.
3 """
4
5 import pygame.locals as pgl
6
7 from naja.constants import KEYS, PALETTE, KONAMI_CODE
8 from naja.events import SceneChangeEvent, QuitGameEvent
9 from naja.options import options
10 from naja.scenes.scene import Scene
11 from naja.scenes.credits import CreditsScene
12 from naja.scenes.game import GameScene
13 from naja.scenes.howto import HowtoScene
14 from naja.scenes.load_save import LoadGameScene, SaveGameScene
15 from naja.scenes.new_game import NewGameScene
16 from naja.widgets.selector import SelectorWidget
17 from naja.widgets.text import TextWidget
18 from naja.widgets.image_box import ImageBox
19
20
21 class MenuScene(Scene):
22     def __init__(self, state):
23         super(MenuScene, self).__init__(state)
24
25         background = ImageBox(
26             (0, 0), "screens/splash.png")
27         self.add(background)
28
29         selector = SelectorWidget()
30         self.add(selector)
31         y_offset, y_diff = 270, 40
32         x_offset = 400
33
34         y_offset += y_diff
35         resume = TextWidget(
36             (x_offset, y_offset), 'Resume Game', colour=PALETTE.WHITE,
37             unselectable_colour=PALETTE.GREY, centre=True)
38         resume.add_callback('click', self.scene_callback(GameScene))
39         resume.set_selectable_callback(lambda: state is not None)
40         selector.add(resume)
41
42         y_offset += y_diff
43         new = TextWidget(
44             (x_offset, y_offset), 'New Game', colour=PALETTE.WHITE,
45             centre=True)
46         new.add_callback('click', self.scene_callback(NewGameScene))
47         selector.add(new)
48
49         y_offset += y_diff
50         load = TextWidget((x_offset, y_offset), 'Load', colour=PALETTE.WHITE,
51                            centre=True)
52         load.add_callback('click', self.scene_callback(LoadGameScene))
53         selector.add(load)
54
55         y_offset += y_diff
56         save = TextWidget(
57             (x_offset, y_offset), 'Save', colour=PALETTE.WHITE,
58             unselectable_colour=PALETTE.GREY, centre=True)
59         save.add_callback('click', self.scene_callback(SaveGameScene))
60         save.set_selectable_callback(lambda: state is not None)
61         selector.add(save)
62
63         y_offset += y_diff
64         howto = TextWidget(
65             (x_offset, y_offset), 'How To Play', colour=PALETTE.WHITE,
66             centre=True)
67         howto.add_callback('click', self.scene_callback(HowtoScene))
68         selector.add(howto)
69
70         y_offset += y_diff
71         credits = TextWidget(
72             (x_offset, y_offset), 'Credits', colour=PALETTE.WHITE,
73             centre=True)
74         credits.add_callback('click', self.scene_callback(CreditsScene))
75         selector.add(credits)
76
77         y_offset += y_diff
78         quit = TextWidget((x_offset, y_offset), 'Quit', colour=PALETTE.WHITE,
79                            centre=True)
80         quit.add_callback('click', lambda event: QuitGameEvent.post())
81         selector.add(quit)
82
83         self.konami = []
84
85     def scene_callback(self, scene_class):
86         return lambda event: SceneChangeEvent.post(scene_class)
87
88     def handle_event(self, ev):
89         if ev.type == pgl.KEYDOWN and self.konami is not None:
90             if ev.key in KEYS.KONAMI:
91                 self.konami.append(ev.key)
92                 if self.konami == KONAMI_CODE:
93                     options['cheat_enabled'] = True
94                     print "Cheats enabled."
95                     print "  Try hitting some number keys."
96                     self.konami = None
97                 elif self.konami != KONAMI_CODE[:len(self.konami)]:
98                     self.konami = None
99         return super(MenuScene, self).handle_event(ev)
100
101     def handle_scene_event(self, ev):
102         if ev.type == pgl.KEYDOWN:
103             if ev.key in KEYS.QUIT:
104                 QuitGameEvent.post()
105                 return