blob: c9074433efd563fcc60463489307c75bd0dbe842 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
""" mainmenu.mainmenu - Let's user choose """
from enum import Enum
import pygame
from screens import mainmenu_screen
from game.anykey import anykey
class MenuItems(Enum):
""" Items in the menu. Should match mainmenu_screen """
NEWGAME = 0
HISCORES = 1
INSTRUCT = 2
QUIT = 3
class Mainmenu(pygame.sprite.Group):
""" sprite group with imputs to make selection """
def __init__(self):
super().__init__()
self.add(mainmenu_screen(0))
self.explode = False
self.active = True
self.fadeout = 1_000
self.selection = 0
def update(self, dt = 0):
""" Does it all. Reads keyboard and updates screen """
if not self.active:
return
if self.explode:
for sprite in self.sprites():
sprite.update(dt = dt, explode = self.explode)
if self.fadeout <= 0:
self.active = False
else:
if anykey():
self.fadeout = 0
self.active = False
self.fadeout -= dt
return
for event in pygame.event.get():
if event.type == pygame.QUIT:
self.selection = MenuItems.QUIT
self.explode = True
break
if event.type == pygame.KEYDOWN:
match event.key:
case pygame.K_KP_ENTER | pygame.K_RETURN | pygame.K_RIGHT:
self.explode = True
break
case pygame.K_ESCAPE | pygame.K_q | pygame.K_LEFT:
self.selection = MenuItems.QUIT
self.explode = True
break
case pygame.K_UP:
self.selection -= 1
self.selection %= len(MenuItems)
case pygame.K_DOWN:
self.selection += 1
self.selection %= len(MenuItems)
self.empty()
self.add(mainmenu_screen(self.selection))
|
