pygame_gui/DerGUI.py

   1 #!/usr/bin/python
   2
   3 import os, sys
   4 from os import *
   5 from os.path import *
   6
   7 from xml.dom.minidom import parse, parseString
   8
   9 from zipfile import *
  10
  11 import pygame
  12 from pygame import *
  13
  14 import StringIO
  15
  16 import copy
  17
  18 FONT_NAME = 'serif'
  19 FONT_SIZE = 30
  20 FONT_COLOR = (255, 255, 255)
  21
  22 class GUISystem:
  23
  24     def __init__(self, gui_def, theme):
  25
  26         assert is_zipfile(theme)
  27         assert exists(gui_def)
  28
  29         self.zipfile = ZipFile(theme)
  30
  31         self.definitions = self.parse_gui_def(gui_def)
  32
  33         pygame.init()
  34
  35         assert pygame.font.get_init()
  36
  37         self.the_font = pygame.font.SysFont(FONT_NAME, FONT_SIZE)
  38
  39         self.images = {}
  40
  41         assert self.getImages()
  42
  43         self.widgets = []
  44
  45     def parse_gui_def(self, gui_def):
  46         results = {}
  47
  48         dom = parse(gui_def)
  49
  50         if dom.childNodes[0].nodeName == 'widgets':
  51
  52             widgets = {}
  53
  54             for node in dom.childNodes[0].childNodes:
  55
  56                 image_list = []
  57
  58                 if node.nodeName == 'images':
  59
  60                     for sub_node in node.childNodes:
  61                         if sub_node.nodeName == 'image':
  62
  63                             image = {'filename': '', name: ''}
  64
  65                             for sub_sub_node in sub_node.childNodes:
  66
  67                                 if sub_sub_node.nodeName == 'filename':
  68                                   image['filename'] = sub_sub_node.childNodes[0].nodeValue
  69                                 if sub_sub_node.nodeName == 'name':
  70                                   image['name'] = sub_sub_node.childNodes[0].nodeValue
  71
  72                             if 'posix' in image:
  73                                 del image['posix']
  74                             image_list.append(image)
  75
  76                     results['image_list'] = image_list
  77
  78                 if node.nodeName == 'widget':
  79
  80                     widget = {}
  81
  82                     widget['parameters'] = {}
  83                     widget['parts'] = {}
  84                     widget['behaviors'] = {}
  85
  86                     for sub_node in node.childNodes:
  87                         if sub_node.nodeName == 'name' and len(sub_node.childNodes):
  88                             widget['name'] = sub_node.childNodes[0].nodeValue
  89
  90                         if sub_node.nodeName == 'parameter':
  91
  92                             parameter = {}
  93                             parameter['name'] = sub_node.attributes['name'].value
  94
  95                             widget['parameters'][parameter['name']] = sub_node.attributes['default']
  96
  97                         if sub_node.nodeName == 'part':
  98
  99                           part_name = sub_node.getElementsByTagName('name')[0].childNodes[0].nodeValue
 100                           widget['parts'][part_name] = {}
 101
 102                           for sub_sub_node in sub_node.childNodes:
 103
 104                             if sub_sub_node.nodeName == 'type':
 105                               widget['parts'][part_name]['type'] = sub_sub_node.childNodes[0].nodeValue.encode()
 106
 107                             if sub_sub_node.nodeName == 'source':
 108                               widget['parts'][part_name]['source'] = sub_sub_node.childNodes[0].nodeValue.encode()
 109
 110                             if sub_sub_node.nodeName == 'position':
 111                               widget['parts'][part_name]['position'] = sub_sub_node.childNodes[0].nodeValue.encode()
 112
 113                             if sub_sub_node.nodeName == 'size':
 114                               widget['parts'][part_name]['size'] = sub_sub_node.childNodes[0].nodeValue.encode()
 115
 116                         if sub_node.nodeName == 'behavior':
 117
 118                           event_type = sub_node.getElementsByTagName('on_event')[0].childNodes[0].nodeValue
 119                           the_do = sub_node.getElementsByTagName('do')[0].childNodes[0].nodeValue
 120
 121                           widget['behaviors'][event_type] = the_do
 122
 123                     widgets[widget['name']] = widget
 124
 125             results['widgets'] = widgets
 126
 127         return results
 128
 129     def getImages(self):
 130
 131         image_list = self.definitions['image_list']
 132         name_list = []
 133
 134         name_list = self.zipfile.namelist()
 135
 136         for image in image_list:
 137
 138             if image['filename'] not in name_list:
 139                 return 0
 140
 141             else:
 142                 the_string_io = StringIO.StringIO()
 143                 print >>the_string_io, self.zipfile.read(image['filename'])
 144                 the_string_io.seek(0)
 145
 146                 self.images[image['name']] = pygame.image.load(the_string_io).convert_alpha()
 147
 148         return 1
 149
 150     def event(self, the_event):
 151
 152         new_event = self.convertEvent(the_event)
 153
 154         if not new_event['type'].find('MOUSE') == -1:
 155
 156             for widget in self.widgets:
 157
 158                 widget_rect = widget.getRect()
 159                 if widget_rect.collidepoint(new_event['data']['pos']):
 160                     if new_event['type'] != 'MOUSEMOTION':
 161                         self.focus = widget
 162                         self.focus.event(new_event)
 163         else:
 164             self.focus.event(new_event)
 165
 166         return
 167
 168     def convertEvent(self, e):
 169
 170         data = {}
 171
 172         if e.type == QUIT:
 173             data['none'] = ''
 174
 175         if e.type == ACTIVEEVENT:
 176             data['gain'] = e.gain
 177             data['state'] = e.state
 178
 179         if e.type == KEYDOWN:
 180             data['unicode'] = e.unicode
 181             data['key'] = e.key
 182             data['mod'] = e.mod
 183
 184         if e.type == KEYUP:
 185             data['key'] = e.key
 186             data['mod'] = e.mod
 187
 188         if e.type == MOUSEMOTION:
 189             data['pos'] = e.pos
 190             data['rel'] = e.rel
 191             data['buttons'] = e.buttons
 192
 193         if e.type == MOUSEBUTTONUP:
 194             data['pos'] = e.pos
 195             data['button'] = e.button
 196
 197         if e.type == MOUSEBUTTONDOWN:
 198             data['pos'] = e.pos
 199             data['button'] = e.button
 200
 201         if e.type == JOYAXISMOTION:
 202             data['joy'] = e.joy
 203             data['axis'] = e.axis
 204             data['value'] = e.value
 205
 206         if e.type == JOYBALLMOTION:
 207             data['joy'] = e.joy
 208             data['ball'] = e.ball
 209             data['rel'] = e.rel
 210
 211         if e.type == JOYHATMOTION:
 212             data['joy'] = e.joy
 213             data['hat'] = e.hat
 214             data['value'] = e.value
 215
 216         if e.type == JOYBUTTONUP:
 217             data['joy'] = e.joy
 218             data['button'] = e.button
 219
 220         if e.type == JOYBUTTONDOWN:
 221             data['joy'] = e.joy
 222             data['button'] = e.button
 223
 224         if e.type == VIDEORESIZE:
 225             data['size'] = e.size
 226             data['w'] = e.w
 227             data['h'] = e.h
 228
 229         if e.type == VIDEOEXPOSE:
 230             data['none'] = ''
 231
 232         if e.type == USEREVENT:
 233             data['code'] = e.code
 234
 235         type = ''
 236
 237         if e.type == QUIT: type = "QUIT"
 238         if e.type == ACTIVEEVENT: type = "ACTIVEEVENT"
 239         if e.type == KEYDOWN: type = "KEYDOWN"
 240         if e.type == KEYUP: type = "KEYUP"
 241         if e.type == MOUSEMOTION    : type = "MOUSEMOTION"
 242         if e.type == MOUSEBUTTONUP: type = "MOUSEBUTTONUP"
 243         if e.type == MOUSEBUTTONDOWN: type = "MOUSEBUTTONDOWN"
 244         if e.type == JOYAXISMOTION: type = "JOYAXISMOTION"
 245         if e.type == JOYBALLMOTION: type = "JOYBALLMOTION"
 246         if e.type == JOYHATMOTION: type = "JOYHATMOTION"
 247         if e.type == JOYBUTTONUP: type = "JOYBUTTONUP"
 248         if e.type == JOYBUTTONDOWN: type = "JOYBUTTONDOWN"
 249         if e.type == VIDEORESIZE: type = "VIDEORESIZE"
 250         if e.type == VIDEOEXPOSE: type = "VIDEOEXPOSE"
 251         if e.type == USEREVENT: type = "USEREVENT"
 252
 253         new_event = {}
 254
 255         new_event['type'] = type
 256         new_event['data'] = data
 257
 258         return new_event
 259
 260     def draw(self, dest_surface):
 261
 262         for widget in self.widgets:
 263
 264             if widget.params['parameters']['visible']:
 265                 widget.draw(dest_surface)
 266
 267         return
 268
 269     def makeWidget(self, widget_name, params):
 270
 271         if self.definitions['widgets'].has_key(widget_name):
 272
 273             for param in self.definitions['widgets'][widget_name]['parameters']:
 274                 if not params.has_key(param):
 275                     if self.definitions['widgets'][widget_name][parameters][param] == 'kill':
 276                         print "GUISystem.makeWidget() Not all required parameters were passed."
 277                         assert 0
 278                     else:
 279                         params[param] = self.definitions['widgets'][widget_name][parameters][param]
 280
 281             new_params = copy.deepcopy(self.definitions['widgets'][widget_name])
 282             new_params['parameters'] = params
 283
 284             the_widget = Widget(self, widget_name, new_params)
 285
 286             self.widgets.append(the_widget)
 287
 288             if len(self.widgets) == 1:
 289                 self.focus = the_widget
 290
 291         return the_widget
 292
 293 class Widget:
 294
 295     def __init__(self, the_gui_system, widget_name, params):
 296
 297         self.name = widget_name
 298         self.params = params
 299         self.the_gui_system = the_gui_system
 300
 301         assert self.params['parameters'].has_key('x')
 302         assert self.params['parameters'].has_key('y')
 303
 304         if not self.params['parameters'].has_key('visible'):
 305             self.params['parameters']['visible'] = 1
 306
 307         self.callbacks = {}
 308
 309         return
 310
 311     def move(self, (new_x, new_y)):
 312
 313         self.params['parameters']['x'], self.params['parameters']['y'] = new_x, new_y
 314
 315         return
 316
 317     def hide(self):
 318
 319         self.params['parameters']['visible'] = 0
 320
 321         return
 322
 323     def show(self):
 324
 325         self.params['parameters']['visible'] = 1
 326
 327         return
 328
 329     def connect(self, event, function, params={}):
 330
 331         self.callbacks[event] = {'function': function, 'params': params}
 332
 333         return
 334
 335     def getRect(self):
 336
 337         coords = (self.params['parameters']['x'], self.params['parameters']['y'])
 338         dimensions = (self.params['parameters']['width'], self.params['parameters']['height'])
 339
 340         return pygame.Rect(coords, dimensions)
 341
 342     def event(self, the_event):
 343
 344         if self.params['behaviors'].has_key(the_event['type']):
 345
 346             the_value = self.params['behaviors'][the_event['type']]
 347
 348             exec(the_value)
 349
 350         return
 351
 352     def setPart(self, old_part, new_part):
 353
 354         self.params['parts'][old_part]['source'] = new_part
 355         return
 356
 357     def raiseEvent(self, the_event, parameters={}):
 358
 359         if not self.callbacks.has_key(the_event):
 360             return
 361
 362         the_function = self.callbacks[the_event]['function']
 363
 364         try:
 365             the_function(self.callbacks[the_event]['params'], parameters)
 366         except:
 367             print "Widget.raiseEvent(): Error in callback."
 368
 369         return
 370
 371     def draw(self, dest_surface):
 372
 373         code = ''
 374
 375         for image in self.the_gui_system.images:
 376
 377             code = ''.join([code, '\n', image, ' = self.the_gui_system.images["', image, '"]'])
 378
 379         try:
 380             exec(code)
 381         except:
 382             assert 0
 383
 384         for part in self.params['parts']:
 385
 386             if self.params['parts'][part]['type'] == 'text':
 387
 388                 # here be dragons
 389                 try:
 390                     exec(''.join([part, ' = self.the_gui_system.the_font.render(self.params["parameters"][self.params["parts"][part]["source"]], FONT_SIZE, FONT_COLOR)\nthe_source =', part]))
 391                 except:
 392                     assert 0
 393
 394             width = self.params['parameters']['width']
 395             height = self.params['parameters']['height']
 396
 397             position = eval(self.params['parts'][part]['position'])
 398             size = eval(self.params['parts'][part]['size'])
 399
 400             position = [position[0], position[1]]
 401
 402             position[0] += self.params['parameters']['x']
 403             position[1] += self.params['parameters']['y']
 404
 405             if self.params['parts'][part]['type'] == 'image':
 406
 407                 if size[0] > 0 and size[1] > 0:
 408                     the_source = pygame.transform.scale(self.the_gui_system.images[self.params['parts'][part]['source']], size)
 409                 dest_surface.blit(the_source, position)
 410
 411             if self.params['parts'][part]['type'] == 'text':
 412
 413                 dest_surface.blit(the_source, position, pygame.Rect((0, 0), size))
 414
 415             if self.params['parts'][part]['type'] == 'widget':
 416
 417                 pass # implement nested widgets
 418
 419         return
 420
 421 # end