"""
Shows message dialog to user
"""
__version__ = "alpha"
__author__ = "Nard Janssens, Vinicius Silva, Robert Gowans, Ivan Antalec, Leonardo Marques - See Github PhotonFileUtils"
import os
import pygame
from pygame.locals import *
from GUI import *
########################################################################################################################
## Class MessageDialog
########################################################################################################################
class MessageDialog():
#Button constants
OK=0
YESNO=1
OKCANCEL=2
buttonChoice=OK
winrect=None
titlerect=None
margins=GRect(8,4,4,4)
waiting=False
dragDiff = None
bordercolor=(0,0,0)
titlebackcolor=defTitlebarBackground
titletextcolor=(0,0,0)
formcolor=defFormBackground
fontname = defFontName
fontsize = defFontSize
listbox=None
btnRight=None
titleheight=24
footerHeight=40
buttonHeight=28
buttonWidth=64
controls=[]
def reposControls(self): #called initially and after winrect is moved
""" Recalculates all positions after moving dialog box. """
self.winrect.height=self.titleheight+self.margins.y+self.label.rect.height+self.footerHeight+self.margins.height
self.titlerect = GRect(self.winrect.x, self.winrect.y, self.winrect.width, self.titleheight)
self.footerTop = self.winrect.y + self.winrect.height - self.margins.height - self.footerHeight
self.label.rect.x=self.winrect.x+self.margins.x
self.label.rect.y=self.winrect.y+self.titleheight+self.margins.y
self.btnRight.rect=GRect(self.winrect.x + self.winrect.width - self.margins.width - self.buttonWidth, self.footerTop + self.margins.x, self.buttonWidth, self.buttonHeight)
if not self.buttonChoice==self.OK: # two buttons
self.btnLeft.rect=GRect.copy(self.btnRight.rect)
self.btnLeft.rect.left=self.btnLeft.rect.left-self.margins.width - self.buttonWidth
def __init__(self, pyscreen,
pos, width=300,
center=True,
title="Message Dialog",message="Read this carefully... \n ...before entering Ok!",
dfontname=defFontName, dfontsize=defFontSize,
buttonChoice=OK,
handleOK=None,
handleCANCEL = None,
handleYES = None,
handleNO = None,
parentRedraw=None):
""" Saves all values to internal variables and calculates some extra internal vars. """
# Save variables
self.pyscreen = pyscreen
self.parentRedraw=parentRedraw
self.winrect=GRect(pos[0], pos[1], width, 160)
self.title=title
self.message=message
self.font = pygame.font.SysFont(dfontname, dfontsize)
self.buttonChoice=buttonChoice
self.handleOK=handleOK
self.handleCANCEL=handleCANCEL
self.handleNO = handleNO
self.handleYES = handleYES
# Calculate extra variables
dummy, textheight = self.font.size("MinimalText")
self.titleheight=textheight+self.margins.y+self.margins.height
self.footerTop = self.winrect.y + self.winrect.height - self.margins.height - self.footerHeight
# Add GUI.Label and GUI.Button(s)
rectLabel=GRect(0,0,self.winrect.width-self.margins.x-self.margins.width,0)
self.label=Label(pyscreen,text=message,fontname=dfontname,fontsize=dfontsize,rect=rectLabel,autoheight=True,center=center,backcolor=self.formcolor,autowrap=True)
self.controls.append(self.label)
self.btnRight=Button(pyscreen, text="OK", func_on_click=self.handleRight, rect=GRect())
self.controls.append(self.btnRight)
if buttonChoice==self.YESNO:
self.btnLeft = Button(pyscreen, text="YES", func_on_click=self.handleLeft, rect=GRect())
self.btnRight = Button(pyscreen, text="NO", func_on_click=self.handleRight, rect=GRect())
self.controls.append(self.btnLeft)
self.controls.append(self.btnRight)
if buttonChoice == self.OKCANCEL:
self.btnLeft = Button(pyscreen, text="OK", func_on_click=self.handleLeft, rect=GRect())
self.btnRight=Button(pyscreen, text="CANCEL", func_on_click=self.handleRight, rect=GRect())
self.controls.append(self.btnLeft)
self.controls.append(self.btnRight)
# (Re)calculate remaining variables
self.reposControls()
def show(self):
""" Returns selected button and closes dialog. """
self.waiting=True
self.waitforuser()
return self.lastaction
def redraw(self):
""" Redraws dialogbox. """
# First call parent / window to redraw itself
self.parentRedraw()
# Draw form background
pygame.draw.rect(self.pyscreen, self.formcolor, self.winrect.tuple(), 0)
# Draw title bar including title text
pygame.draw.rect(self.pyscreen, self.titlebackcolor,self.titlerect.tuple(), 0)
self.font.set_bold(True)
textsurface = self.font.render(self.title, True, self.titletextcolor)
self.pyscreen.blit(textsurface, (self.winrect.x + self.margins.x, self.winrect.y + self.margins.y))
self.font.set_bold(False)
# Draw form border
pygame.draw.rect(self.pyscreen, self.bordercolor, self.winrect.tuple(), 1)
# Call upon label and button to redraw themselves.
self.label.redraw()
self.btnRight.redraw()
if not self.buttonChoice==self.OK: self.btnLeft.redraw()
def waitforuser(self):
""" Blocks all events to Main window and wait for user to click OK. """
while self.waiting:
self.redraw()
pygame.display.flip()
for event in pygame.event.get():
pos = pygame.mouse.get_pos()
gpos=GPoint().fromTuple(pos)
if event.type == pygame.MOUSEBUTTONUP:
if self.dragDiff==None:
for ctrl in self.controls:
ctrl.handleMouseUp(pos,event.button)
else: # handle window move
self.dragDiff=None
if event.type == pygame.MOUSEBUTTONDOWN:
if gpos.inGRect(self.titlerect):
self.dragDiff = gpos - self.winrect.p1
else:
for ctrl in self.controls:
ctrl.handleMouseDown(pos,event.button)
if event.type == pygame.MOUSEMOTION:
if not self.dragDiff==None:
self.winrect.p1=gpos-self.dragDiff
self.reposControls()
else:
for ctrl in self.controls:
ctrl.handleMouseMove(pos)
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_ESCAPE:
print("Escape key pressed down.")
self.waiting = False
def handleRight(self):
""" If OK we tell main loop we are ready waiting. """
if self.buttonChoice==self.OK: self.lastaction="OK"
if self.buttonChoice==self.OKCANCEL: self.lastaction="CANCEL"
if self.buttonChoice == self.YESNO: self.lastaction = "NO"
self.waiting=False
def handleLeft(self):
""" If OK we tell main loop we are ready waiting. """
if self.buttonChoice==self.OKCANCEL: self.lastaction="OK"
if self.buttonChoice == self.YESNO: self.lastaction = "YES"
self.waiting=False
