def __init__(self, *args, **kwargs):
super(Dialog, self).__init__(*args, **kwargs)
self.image = pygame.Surface(self.rect.size)
self.image.fill((255, 255, 255))
# add ourselves
self.root.add(self, 0, 0)
self.rect.center = self.root.rect.center
self.title = kwargs.pop('title', '')
self._contents = None
self._label = Label(self.title, root=self.root)
self._label.setParent(self)
self._label.setPosition((self.rect.left + 5,
self.rect.top + 5))
def closeCallback(*args):
self.kill()
self._closeButton = SpriteButton('x16', closeCallback, root=self.root)
self._closeButton.setParent(self)
buttonPosition = (self.rect.right - self._closeButton.rect.width - 5,
self.rect.top + 5)
self._closeButton.setPosition(buttonPosition)
self.moveToFront()
def __init__(self, text, width, height, background=(255, 255, 255),
*args, **kwargs):
kwargs['border'] = (1, (0, 0, 0))
Container.__init__(self, width, height, *args, **kwargs)
self.cursorPosition = 0
self.text = ""
self.textDisplayedBeginIndex = 0
self.cursor = Image('textcursor', root=self.root)
self.add(self.cursor)
self.label = Label('', root=self.root, fontSize=30)
self.add(self.label)
class Dialog(Widget):
def __init__(self, *args, **kwargs):
super(Dialog, self).__init__(*args, **kwargs)
self.image = pygame.Surface(self.rect.size)
self.image.fill((255, 255, 255))
# add ourselves
self.root.add(self, 0, 0)
self.rect.center = self.root.rect.center
self.title = kwargs.pop('title', '')
self._contents = None
self._label = Label(self.title, root=self.root)
self._label.setParent(self)
self._label.setPosition((self.rect.left + 5,
self.rect.top + 5))
def closeCallback(*args):
self.kill()
self._closeButton = SpriteButton('x16', closeCallback, root=self.root)
self._closeButton.setParent(self)
buttonPosition = (self.rect.right - self._closeButton.rect.width - 5,
self.rect.top + 5)
self._closeButton.setPosition(buttonPosition)
self.moveToFront()
def setPosition(self, position):
pass
def handleEvent(self, event):
widgets = [self._closeButton, self._contents]
# handle mouse events by deciding which widgets are in the clicked area
for widget in widgets:
if event.type in [MOUSEBUTTONDOWN, MOUSEMOTION, MOUSEBUTTONUP]:
if widget.rect.collidepoint(pygame.mouse.get_pos()):
widget.handleEvent(event)
else:
widget.handleEvent(event)
def moveToFront(self):
super(Dialog, self).moveToFront()
self._label.moveToFront()
self._closeButton.moveToFront()
if self._contents is not None:
self._contents.moveToFront()
def moveToBack(self):
if self._contents is not None:
self._contents.moveToBack()
self._closeButton.moveToBack()
self._label.moveToBack()
super(Dialog, self).moveToBack()
def kill(self):
self._contents.kill()
self._label.kill()
self._closeButton.kill()
super(Dialog, self).kill()
def show(self):
self._contents.show()
self._label.show()
self._closeButton.show()
super(Dialog, self).show()
self.moveToFront()
def setContents(self, widget):
self._contents = widget
self._contents.setParent(self)
self._contents.setPosition((self.rect.left + 5,
self.rect.top + \
max(self._label.rect.height,
self._closeButton.rect.height) + 5))
self._contents.updateLayer()
class Text(Container):
def __init__(self, text, width, height, background=(255, 255, 255),
*args, **kwargs):
kwargs['border'] = (1, (0, 0, 0))
Container.__init__(self, width, height, *args, **kwargs)
self.cursorPosition = 0
self.text = ""
self.textDisplayedBeginIndex = 0
self.cursor = Image('textcursor', root=self.root)
self.add(self.cursor)
self.label = Label('', root=self.root, fontSize=30)
self.add(self.label)
@property
def textDisplayedEndIndex(self):
for index in xrange(len(self.text),
self.textDisplayedBeginIndex + 1,
-1):
text = self.text[self.textDisplayedBeginIndex:index]
if self.label.font.size(text)[0] < self.rect.width - 4:
return index
return self.textDisplayedBeginIndex + 1
def setPosition(self, position):
Container.setPosition(self, position)
x, y = position
self.cursor.setPosition(self._cursorPosition())
self.label.setPosition((x + 3, y))
self.label.rect.bottom = self.rect.bottom - 2
def update(self, *args):
Container.update(self, *args)
self.cursor.setPosition(self._cursorPosition())
self.label.setText(self._viewableText())
self.label.changeLayer(self.layer + 1)
self.cursor.changeLayer(self.layer + 2)
def _viewableText(self):
return self.text[self.textDisplayedBeginIndex:self.textDisplayedEndIndex]
def _relativeCursorPosition(self):
if self.cursorPosition < self.textDisplayedBeginIndex:
self.textDisplayedBeginIndex = self.cursorPosition
while self.textDisplayedEndIndex < self.cursorPosition:
self.textDisplayedBeginIndex += 1
return self.cursorPosition - self.textDisplayedBeginIndex
def _cursorPosition(self):
beforeCursorViewableText = self._viewableText()[:self._relativeCursorPosition()]
return (self.label.font.size(beforeCursorViewableText)[0] + \
self.rect.left,
self.rect.top + 2)
def moveCursorRight(self):
if self.cursorPosition >= len(self.text):
return
self.cursorPosition += 1
if self.cursor.rect.right > self.rect.right:
self.cursor.rect.right = self.rect.right
def moveCursorLeft(self):
if self.cursorPosition == 0:
return
self.cursorPosition -= 1
if self.cursor.rect.left < self.rect.left:
self.cursor.rect.left = self.rect.left
def deleteCharacter(self):
# Can't delete anything if there's no text.
if len(self.text) == 0:
return
self.text = self.text[:self.cursorPosition] + \
self.text[self.cursorPosition + 1:]
self.cursor.setPosition(self._cursorPosition())
def addCharacter(self, character):
self.text += character
self.moveCursorRight()
def handleEvent(self, event):
if event.type == MOUSEBUTTONDOWN and event.button == 1:
self.setActive()
# TODO support mouse insert-cursor selection
elif event.type == KEYDOWN:
if event.key == K_LEFT:
self.moveCursorLeft()
elif event.key == K_RIGHT:
self.moveCursorRight()
elif event.key == K_DELETE:
self.deleteCharacter()
elif event.key == K_BACKSPACE:
self.moveCursorLeft()
self.deleteCharacter()
elif event.unicode in string.printable and event.unicode not in '\n\r\x0b\x0c':
self.addCharacter(event.unicode)
def __init__(self, text, callback, background=(100, 100, 100),
padding=(10, 10), *args, **kwargs):
kwargs['background'] = background
kwargs['padding'] = padding
Label.__init__(self, text, *args, **kwargs)
BaseButton.__init__(self, callback)
def __init__(self, tabbedPages, tabName, *args, **kwargs):
Label.__init__(self, tabName, *args, **kwargs)
self.tabName = tabName
self._tabbedPages = tabbedPages