forked from svenstaro/pong4p
-
Notifications
You must be signed in to change notification settings - Fork 1
/
menu_classes.py
379 lines (308 loc) · 10.1 KB
/
menu_classes.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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
# -*- coding: utf-8 -*-
import pygame
pygame.font.init()
MENUFONT=pygame.font.SysFont("Verdana",12,True)
class GameMenu(object):
"""This class defines all methods and attributes used to create a game menu."""
def __init__(self,x=0,y=0,style=None):
self.buttons=[]
self.fields=[]
self.x=x
self.y=y
self.style=style
def check_buttons(self,mouse,checkSubs=True):
do = True
for f in self.fields:
if f.active:
do = False
if do:
lastbutton='nobuttonpressed'
for button in self.get_buttons():
if button.check_over(mouse[0]-self.x,mouse[1]-self.y):
lastbutton=button.label
for button2 in self.get_buttons():
if button2 is not button:
button2.set_submenu_stat(False)
button.set_submenu_stat(True)
elif checkSubs and button.get_submenu_stat():
zwischen=button.submenu.check_buttons(mouse)
if zwischen!='nobuttonpressed':
lastbutton=zwischen
break
return lastbutton
else:
return -1
def click_fields(self,mouse):
x=self.x
y=self.y
act = False
for f in self.fields:
if f.active:
act = True
for f in self.fields:
if f.active and f.type=="pulldown":
f.click_subs(mouse,(x,y))
if not act:
if mouse[0]>f.x+x and mouse[0]<f.x+x+f.width and mouse[1]>f.y+y and mouse[1]<f.y+f.height+y:
f.setActive(not f.active)
else:
f.setActive(False)
else:
f.setActive(False)
for b in self.buttons:
if b.get_submenu_stat():
b.submenu.click_fields(mouse)
def toggle_fields(self):
for f in self.fields:
if f is TextField:
f.toggleCursor()
for b in self.buttons:
b.submenu.toggle_fields()
def key_down(self,uc,extra=None):
for f in self.fields:
if f.active:
f.write(uc,extra)
for b in self.buttons:
if b.get_submenu_stat():
b.submenu.key_down(uc,extra)
def set_pos(self, pos=(0,0)):
self.x=pos[0]
self.y=pos[1]
def add_button(self,button):
self.buttons.append(button)
def add_field(self,field):
self.fields.append(field)
def get_buttons(self):
return self.buttons
def get_fields(self):
return self.fields
def get_size(self,rand=10):
width = 0
height = 0
for f in self.fields:
if f.width + f.x +2*rand > width: width = f.width + f.x +2*rand
if f.height + f.y +2*rand > height: height = f.height + f.y +2*rand
for b in self.buttons:
if b.width + b.x +2*rand > width: width = b.width + b.x +2*rand
if b.height + b.y +2*rand > height: height = b.height + b.y +2*rand
return (width,height)
def draw(self, surface, mouse=(0,0)):
x=self.x
y=self.y
if not(self.style==None):
border = self.style[2]
w, h = self.get_size(self.style[3])
bbox = pygame.Surface((w,h)).convert()
box = pygame.Surface((w-2*border,h-2*border)).convert()
bbox.fill(self.style[1])
box.fill(self.style[0])
bbox.blit(box,(border,border))
surface.blit(bbox,(x-self.style[3],y-self.style[3]))
for button in self.buttons:
hover = True
for f in self.fields:
if f.active:
hover = False
box=button.draw((mouse[0]-self.x,mouse[1]-self.y),hover)
surface.blit(box,(button.x+x,button.y+y))
#mit hover: if (button.check_over(mouse[0]-self.x,mouse[1]-self.y) or button.get_submenu_stat()) and button.submenu!=0:
if button.get_submenu_stat():
button.draw_submenu(surface,mouse)
for field in self.fields:
m=mouse
for f in self.fields:
if f.active:
m=(-100,-100)
field.draw(surface,m,(x,y))
for f in self.fields:
if f.active and f.type=="pulldown":
f.draw_choice(surface,mouse,(x,y))
return surface
# end class 'GameMenu'
class MenuButton(object):
"""This class is used to create a single instance of a menu button."""
def __init__(self, style, label, x=0, y=0, width=170, height=40, submenu=0):
self.label = label
self.style = style
self.x = x
self.y = y
self.width = width
self.height = height
self.submenu=submenu
self.nosub=False
if self.submenu==0:
self.submenu=GameMenu()
self.nosub=True
self.submenu_status=False
def check_over(self,mouseX,mouseY):
erg=False
if self.x <= mouseX and self.x+self.width>=mouseX and self.y <=mouseY and self.y+self.height>=mouseY:
erg=True
return erg
def set_submenu_stat(self, status=False):
self.submenu_status=status
if status==False:
for button in self.submenu.get_buttons():
button.set_submenu_stat(False)
return self.submenu_status
def get_submenu_stat(self):
return self.submenu_status
def draw(self, mouse, hover):
w=self.width
s = self.style
border = self.style[3]
if (self.submenu_status and not self.nosub): state = 1 #active
elif self.check_over(mouse[0],mouse[1]) and hover: state = 2 #hover
else: state = 0 #normal
box = pygame.Surface((w-2*border,self.height-2*border)).convert()
bigbox = pygame.Surface((w,self.height)).convert()
box.fill(s[0][state])
bigbox.fill(s[2][state])
fontcolor=s[1][state]
box.blit(s[4].render(self.label, True, fontcolor),(10,(self.height-2*border)/2-s[4].size("lol")[1]/2+border))
bigbox.blit(box,(border,border))
return bigbox
def draw_submenu(self, surface,mouse=(0,0)):
surface=self.submenu.draw(surface,mouse)
return surface
# end class 'MenuButton'
class TextField(object):
def __init__(self,style,label,text,x=0,y=0,width=200,height=20):
self.label=label
self.style=style
self.text=text
self.x=x
self.y=y
self.width=width
self.height=height
self.font=style[4]
self.active=False
self.cursor="_"
self.type="text"
def setActive(self,state=True):
self.active=state
def check_over(self,surf,mouse,move=(0,0)):
return self.x+move[0] < mouse[0] and self.x+self.width+move[0]>mouse[0] and self.y+move[1] <mouse[1] and self.y+self.height+move[1]>=mouse[1]
def toggleCursor(self):
if self.cursor=="_": self.cursor=""
else: self.cursor="_"
def write(self,text,extra=None):
if extra==None: self.text+=text
elif extra == "back":
self.text=self.text[:-1]
def draw(self,surf,mouse,move=(0,0)):
s = self.style
border = s[3]
bbox=pygame.Surface((self.width,self.height)).convert()
box=pygame.Surface((self.width-2*border,self.height-2*border)).convert()
if self.active: state = 1
elif self.check_over(mouse,move): state = 2
else: state = 0
c = ""
if self.active: c = self.cursor
box.fill(s[0][state])
bbox.fill(s[2][state])
box.blit(self.style[4].render(self.label+": "+self.text+c,True,s[1][state]),(10,(self.height)/2-s[4].size("lol")[1]/2))
bbox.blit(box,(border,border))
surf.blit(bbox,(self.x+move[0],self.y+move[1]))
return surf
def draw_choice(self,surf,mouse,move=(0,0)):
pass
class PullDownMenu(object):
def __init__(self,style,label,text,items=[],x=0,y=0,width=200,height=20):
self.label=label
self.style=style
self.text=text
self.x=x
self.y=y
self.width=width
self.height=height
self.font=style[4]
self.items=items
self.active=True
self.type="pulldown"
def check_over(self,mouse,move):
return self.x+move[0] < mouse[0] and self.x+self.width+move[0]>mouse[0] and self.y+move[1] <mouse[1] and self.y+self.height+move[1]>=mouse[1]
def setActive(self,state=True):
self.active=state
def get_Size_All(self):
size_all=[0,0]
for item in self.items:
size=self.font.size(item)
size_all[1]+=size[1]+2*self.style[3]
size_all[0] = self.width - self.font.size(self.label+": ")[0] -10
return size_all
def draw(self,surf,mouse,move=(0,0)):
s = self.style
border = s[3]
bbox=pygame.Surface((self.width,self.height)).convert()
box=pygame.Surface((self.width-2*border,self.height-2*border)).convert()
if self.active: state=1
elif self.check_over(mouse,move): state = 2
else: state = 0
box.fill(s[0][state])
bbox.fill(s[2][state])
box.blit(self.font.render(self.label+": "+self.text,True,s[1][state]),(10,(self.height)/2-s[4].size("lol")[1]/2))
bbox.blit(box,(border,border))
surf.blit(bbox,(self.x+move[0],self.y+move[1]))
def draw_choice(self,surf,mouse,move=(0,0)):
s = self.style
border = s[3]
if self.active:
size_all=self.get_Size_All()
rand=pygame.Surface((size_all[0]+2*border,size_all[1]+2*border)).convert()
rand.fill(s[2][0])
box2=pygame.Surface((size_all[0],size_all[1])).convert()
box2.fill(s[0][0])
i=0
height=self.font.size("Empty")[1]
x_off=self.font.size(self.label+": ")[0]+10
for item in self.items:
size=self.font.size(item)
if mouse[0]>self.x+move[0]+x_off-10 and mouse[0]<self.x+move[0]+x_off-10+size_all[0] and mouse[1]>self.y+move[1]+self.height+i*(height+2*border) and mouse[1]<self.y+move[1]+self.height+i*(height+2*border)+size[1]+border:
cl=s[1][2]
bg=s[0][2]
backgr = pygame.Surface((size_all[0],size[1]+2*border)).convert()
backgr.fill(bg)
box2.blit(backgr,(0,i*(height+2*border)))
else:
cl=s[1][0]
bg=s[0][0]
box2.blit(self.font.render(item,True,cl),(10,i*(height+2*border)+border))
i+=1
surf.blit(rand,(self.x+move[0]+x_off-2*border,self.y+move[1]+self.height))
surf.blit(box2,(self.x+move[0]+x_off-border,self.y+move[1]+self.height+border))
return surf
def click_subs(self,mouse,move=(0,0)):
border = self.style[3]
if self.active:
size_all=self.get_Size_All()
i=0
height=self.font.size("Empty")[1]
x_off=self.font.size(self.label+": ")[0]
for item in self.items:
size=self.font.size(item)
if mouse[0]>self.x+move[0]+x_off-10 and mouse[0]<self.x+move[0]+x_off-10+size_all[0] and mouse[1]>self.y+move[1]+self.height+i*(height+2*border) and mouse[1]<self.y+move[1]+self.height+i*(height+2*border)+size[1]+border:
self.active=False
self.text = item
i+=1
class JpgAnimation(object):
def __init__(self,url,anz,format="jpg",digits = None):
self.url = url
self.anz = anz
self.format = format
self.images = [None]*anz
self.digits = digits
def load(self, num):
num = str(num)
if digits != None and len(num)<self.digits:
num = "0"*(self.digits - len(num)) +num
self.images[num] = pygame.image.load(self.url+num+"."+format).convert_alpha()
def load_all(self,scale = True):
i=0
for img in self.images:
self.load(i)
i+=1
def scale(self,RES=(800,620)):
for i in xrange(len(self.images)):
self.images[i] = pygame.transform.smoothscale(self.images[i],RES)