Slightly better menu item positions (stop gap while I do something better).
[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 = 200
32         x_offset = 300
33
34         y_offset += 50
35         resume = TextWidget(
36             (x_offset, y_offset), 'Resume Game', colour=PALETTE.WHITE,
37             unselectable_colour=PALETTE.GREY)
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 += 50
43         new = TextWidget(
44             (x_offset, y_offset), 'New Game', colour=PALETTE.WHITE)
45         new.add_callback('click', self.scene_callback(NewGameScene))
46         selector.add(new)
47
48         y_offset += 50
49         load = TextWidget((x_offset, y_offset), 'Load', colour=PALETTE.WHITE)
50         load.add_callback('click', self.scene_callback(LoadGameScene))
51         selector.add(load)
52
53         y_offset += 50
54         save = TextWidget(
55             (x_offset, y_offset), 'Save', colour=PALETTE.WHITE,
56             unselectable_colour=PALETTE.GREY)
57         save.add_callback('click', self.scene_callback(SaveGameScene))
58         save.set_selectable_callback(lambda: state is not None)
59         selector.add(save)
60
61         y_offset += 50
62         howto = TextWidget(
63             (x_offset, y_offset), 'How To Play', colour=PALETTE.WHITE)
64         howto.add_callback('click', self.scene_callback(HowtoScene))
65         selector.add(howto)
66
67         y_offset += 50
68         credits = TextWidget(
69             (x_offset, y_offset), 'Credits', colour=PALETTE.WHITE)
70         credits.add_callback('click', self.scene_callback(CreditsScene))
71         selector.add(credits)
72
73         y_offset += 50
74         quit = TextWidget((x_offset, y_offset), 'Quit', colour=PALETTE.WHITE)
75         quit.add_callback('click', lambda event: QuitGameEvent.post())
76         selector.add(quit)
77
78         self.konami = []
79
80     def scene_callback(self, scene_class):
81         return lambda event: SceneChangeEvent.post(scene_class)
82
83     def handle_event(self, ev):
84         if ev.type == pgl.KEYDOWN and self.konami is not None:
85             if ev.key in KEYS.KONAMI:
86                 self.konami.append(ev.key)
87                 if self.konami == KONAMI_CODE:
88                     options['cheat_enabled'] = True
89                     print "Cheats enabled."
90                     print "  Try hitting some number keys."
91                     self.konami = None
92                 elif self.konami != KONAMI_CODE[:len(self.konami)]:
93                     self.konami = None
94         return super(MenuScene, self).handle_event(ev)
95
96     def handle_scene_event(self, ev):
97         if ev.type == pgl.KEYDOWN:
98             if ev.key in KEYS.QUIT:
99                 QuitGameEvent.post()
100                 return