/
menu.py
65 lines (60 loc) · 1.86 KB
/
menu.py
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
65
import core
import data
class Menu:
opts = []
active = 0
height = 0
skipSpace = 20
spanSpace = 5
def __init__(self):
pass
#enddef
def draw(self):
y = (core.height - (self.height + (len(self.opts)-1)*self.spanSpace))/2
for i, e in enumerate(self.opts):
if e[0] == None:
y += self.skipSpace
continue
sf = e[0]
if (i == self.active):
sf = e[1]
r = sf.get_rect()
sf.draw(core.screen, (core.width-r.width)/2, y)
y += r.height + self.spanSpace
#endfor
#enddef
def behave(self):
for opt in self.opts:
if opt[0] != None:
opt[0].behave()
opt[1].behave()
dir = 0
if core.controls.down:
dir = +1
core.controls.down = False
if core.controls.up:
dir = -1
core.controls.up = False
self.active += dir
while self.active>0 and self.active < len(self.opts) and self.opts[self.active][0]==None:
self.active += dir
if self.active<0:
self.active = len(self.opts) - 1
if self.active >= len(self.opts):
self.active = 0
if core.controls.fire or core.controls.enter:
core.controls.enter = False
core.controls.fire = False
if self.opts[self.active][2]:
self.opts[self.active][2]()
#endif
# enddef
def set(self, no, entry, selectedEntry, callback = None):
if len(self.opts) <= no:
addLen = 1+no-len(self.opts)
self.opts += [(None, )]*(addLen)
self.height += (addLen - 1) * self.skipSpace
self.opts[no] = (entry, selectedEntry, callback )
self.height += self.opts[no][0].get_rect().height
#enddef
#endclass