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