231 lines
7.4 KiB
Python
231 lines
7.4 KiB
Python
"""PickList module: contains the PickList class"""
|
|
from ShowBaseGlobal import *
|
|
from GuiGlobals import *
|
|
import PandaObject
|
|
import Frame
|
|
import Button
|
|
|
|
class PickList(PandaObject.PandaObject):
|
|
"""PickList class: display a menu of choices and report users
|
|
choice (via mouse or keyboard) as an event with the choice as
|
|
an extra argument
|
|
"""
|
|
|
|
# special methods
|
|
def __init__(self, name, choiceList,
|
|
scale = 0.1,
|
|
width = None,
|
|
drawOrder = getDefaultDrawOrder(),
|
|
font = getDefaultFont()):
|
|
|
|
self.name = name
|
|
self.frame = Frame.Frame(name)
|
|
|
|
# initialization
|
|
self.choice = -1
|
|
self.choiceList = []
|
|
self.eventName = self.name
|
|
self.frame.setOffset(0.015)
|
|
|
|
# display the menu
|
|
self.__displayChoices(choiceList, scale, width, drawOrder, font)
|
|
self.isClean = 0
|
|
return None
|
|
|
|
def cleanup(self):
|
|
"""cleanup(self)
|
|
"""
|
|
if self.isClean == 0:
|
|
self.isClean = 1
|
|
self.disable()
|
|
self.frame.unmanage()
|
|
self.frame = None
|
|
self.choiceList = []
|
|
return None
|
|
|
|
# accessing
|
|
def getName(self):
|
|
return self.name
|
|
|
|
def getEventName(self):
|
|
return self.eventName
|
|
|
|
def setEventName(self, eventName):
|
|
self.eventName = eventName
|
|
|
|
# actions
|
|
def __displayChoices(self, choiceList,
|
|
scale, width, drawOrder, font):
|
|
"""__displayChoices(self, string[])
|
|
Display the list of choices
|
|
"""
|
|
|
|
if width == None:
|
|
# First, compute the maximum width of the buttons. We do this
|
|
# ahead of time so the Gui code doesn't have to do it and take
|
|
# forever about it.
|
|
width = 0
|
|
text = TextNode()
|
|
text.setFont(font)
|
|
for choice in choiceList:
|
|
w = text.calcWidth(choice) + 0.2
|
|
width = max(width, w)
|
|
|
|
# Now create all the buttons.
|
|
for choice in choiceList:
|
|
# create a button for each choice
|
|
button = Button.Button(choice, scale = scale, width = width,
|
|
drawOrder = drawOrder, font = font)
|
|
choiceIndex = choiceList.index(choice)
|
|
# set the rollover-up event
|
|
eventName = self.name + "-up-" + str(choiceIndex)
|
|
button.button.setUpRolloverEvent(eventName)
|
|
# set the rollover-down event
|
|
eventName = self.name + "-down-" + str(choiceIndex)
|
|
button.button.setDownRolloverEvent(eventName)
|
|
# set exit event
|
|
eventName = self.name + "-exit-" + str(choiceIndex)
|
|
button.button.setUpEvent(eventName)
|
|
# keep a list of the choice buttons
|
|
self.frame.addItem(button)
|
|
self.choiceList.append(button)
|
|
|
|
# set up the frame
|
|
self.frame.makeVertical()
|
|
|
|
return None
|
|
|
|
def manage(self):
|
|
self.enable()
|
|
self.frame.manage()
|
|
return None
|
|
|
|
def unmanage(self):
|
|
self.disable()
|
|
self.frame.unmanage()
|
|
return None
|
|
|
|
def enable(self):
|
|
# turn the buttons on
|
|
self.activate()
|
|
|
|
# listen for keyboard events
|
|
self.accept("up-up", self.__decrementChoice)
|
|
self.accept("down-up", self.__incrementChoice)
|
|
self.accept("enter-up", self.__makeChoice, [1])
|
|
|
|
for choice in self.choiceList:
|
|
choiceIndex = self.choiceList.index(choice)
|
|
# set the rollover-up event
|
|
eventName = self.name + "-up-" + str(choiceIndex)
|
|
self.accept(eventName, self.__updateButtonChoice, [choiceIndex])
|
|
# set the rollover-down event
|
|
eventName = self.name + "-down-" + str(choiceIndex)
|
|
self.accept(eventName, self.__makeChoice)
|
|
# set exit event
|
|
eventName = self.name + "-exit-" + str(choiceIndex)
|
|
self.accept(eventName, self.__exitChoice)
|
|
|
|
|
|
|
|
def disable(self, button=None):
|
|
# turn the buttons off
|
|
self.deactivate(button)
|
|
|
|
# ignore all hooks
|
|
self.ignore("up-up")
|
|
self.ignore("down-up")
|
|
self.ignore("enter-up")
|
|
|
|
# ignore all the buttons
|
|
for item in self.frame.getItems():
|
|
self.ignore(item.getGuiItem().getUpEvent())
|
|
self.ignore(item.getGuiItem().getUpRolloverEvent())
|
|
self.ignore(item.getGuiItem().getDownRolloverEvent())
|
|
|
|
def activate(self):
|
|
# make sure items are active
|
|
for choice in self.choiceList:
|
|
choice.getGuiItem().up()
|
|
|
|
def deactivate(self, button=None):
|
|
# make sure all items are inactive, if a button
|
|
# is passed in do NOT deactivate it.
|
|
for choice in self.choiceList:
|
|
if (choice.getGuiItem().isActive()):
|
|
if not (button == None):
|
|
if not (self.choiceList.index(choice) == button):
|
|
choice.getGuiItem().inactive()
|
|
else:
|
|
choice.getGuiItem().inactive()
|
|
|
|
def recompute(self):
|
|
self.frame.recompute()
|
|
|
|
def setPos(self, x, y):
|
|
self.frame.setPos(x, y)
|
|
|
|
# event handlers
|
|
def __incrementChoice(self):
|
|
# handle the up arrow
|
|
if (self.choice >= 0):
|
|
# exit lest choice, if it exists
|
|
self.choiceList[self.choice].getGuiItem().exit()
|
|
self.choice = self.choice + 1
|
|
if (self.choice > len(self.choiceList) - 1):
|
|
self.choice = 0
|
|
# enter the new choice
|
|
self.choiceList[self.choice].getGuiItem().enter()
|
|
|
|
|
|
def __decrementChoice(self):
|
|
# handle the down arrow
|
|
if (self.choice >= 0):
|
|
self.choiceList[self.choice].getGuiItem().exit()
|
|
self.choice = self.choice - 1
|
|
if (self.choice < 0):
|
|
self.choice = len(self.choiceList) - 1
|
|
# enter this choice
|
|
self.choiceList[self.choice].getGuiItem().enter()
|
|
|
|
def __updateButtonChoice(self, newChoice, *args):
|
|
# handle the mouse rollovers
|
|
self.choice = newChoice
|
|
# make sure all the other buttons have exited
|
|
for choice in self.choiceList:
|
|
if not (self.choiceList.index(choice) == self.choice):
|
|
choice.getGuiItem().exit()
|
|
# throw event
|
|
messenger.send(self.name + "-rollover")
|
|
|
|
def __exitChoice(self, *args):
|
|
# reset choice when mouse exits a button
|
|
self.choice = -1
|
|
|
|
def __makeChoice(self, button=0, *args):
|
|
# handle the enter key
|
|
if (self.choice == -1):
|
|
# nothing selected yet
|
|
return None
|
|
else:
|
|
# if the keyboard was used, update the button manually
|
|
if (button):
|
|
self.choiceList[self.choice].getGuiItem().down()
|
|
def buttonUp(state):
|
|
state.choice.getGuiItem().up()
|
|
return Task.done
|
|
task = Task.Task(buttonUp)
|
|
task.choice = self.choiceList[self.choice]
|
|
taskMgr.spawnTaskNamed(Task.doLater(.035, task,
|
|
"buttonUp-later"),
|
|
"doLater-buttonUp-later")
|
|
else:
|
|
# let everyone know a choice was made
|
|
messenger.send(self.eventName, [self.choice])
|
|
self.disable(self.choice)
|
|
|
|
|
|
|
|
|
|
|