Source code for direct.tkpanels.TaskManagerPanel
"""Undocumented Module"""
__all__ = ['TaskManagerPanel', 'TaskManagerWidget']
from direct.tkwidgets.AppShell import *
from direct.showbase.DirectObject import DirectObject
import Pmw, sys
if sys.version_info >= (3, 0):
from tkinter import *
from tkinter.messagebox import askokcancel
else:
from Tkinter import *
from tkMessageBox import askokcancel
[docs]class TaskManagerPanel(AppShell):
# Override class variables here
appname = 'TaskManager Panel'
frameWidth = 300
frameHeight = 400
usecommandarea = 0
usestatusarea = 0
[docs] def __init__(self, taskMgr, parent = None, **kw):
INITOPT = Pmw.INITOPT
optiondefs = (
('title', self.appname, None),
)
self.defineoptions(kw, optiondefs)
self.taskMgr = taskMgr
# Call superclass initialization function
AppShell.__init__(self, parent = parent)
self.initialiseoptions(TaskManagerPanel)
[docs] def createInterface(self):
# FILE MENU
# Get a handle on the file menu so commands can be inserted
# before quit item
self.taskMgrWidget = TaskManagerWidget(
self.interior(), self.taskMgr)
[docs]class TaskManagerWidget(DirectObject):
"""
TaskManagerWidget class: this class contains methods for creating
a panel to control taskManager tasks.
"""
[docs] def __init__(self, parent, taskMgr):
"""
TaskManagerWidget class pops up a control panel to view/delete
tasks managed by the taskManager.
"""
# Make sure TK mainloop is running
from direct.showbase import TkGlobal
# Record parent (used by ok cancel dialog boxes)
self.parent = parent
# Record taskManager
self.taskMgr = taskMgr
# Init current task
self.currentTask = None
self.__taskDict = {}
# Create widgets
# Create a listbox
self.taskListBox = Pmw.ScrolledListBox(
parent,
labelpos = NW, label_text = 'Tasks:',
label_font=('MSSansSerif', 10, 'bold'),
listbox_takefocus = 1,
items = [],
selectioncommand = self.setCurrentTask)
self.taskListBox.pack(expand = 1, fill = BOTH)
self._popupMenu = Menu(self.taskListBox.component('listbox'),
tearoff = 0)
self._popupMenu.add_command(
label = 'Remove Task',
command = self.removeCurrentTask)
self._popupMenu.add_command(
label = 'Remove Matching Tasks',
command = self.removeMatchingTasks)
# Controls Frame
controlsFrame = Frame(parent)
self.removeButton = Button(controlsFrame, text = 'Remove Task',
command = self.removeCurrentTask)
#self.removeButton.pack(expand = 1, fill = X, side = LEFT)
self.removeButton.grid(row = 0, column = 0, sticky = EW)
self.removeMatchingButton = Button(controlsFrame,
text = 'Remove Matching Tasks',
command = self.removeMatchingTasks)
#self.removeMatchingButton.pack(expand = 1, fill = X, side = LEFT)
self.removeMatchingButton.grid(row = 0, column = 1, sticky = EW)
self.taskMgrVerbose = IntVar()
self.taskMgrVerbose.set(0)
self.update = Button(
controlsFrame,
text = 'Update',
command = self.updateTaskListBox)
#self.update.pack(expand = 1, fill = X, side = LEFT)
self.update.grid(row = 1, column = 0, sticky = EW)
self.dynamicUpdate = Checkbutton(
controlsFrame,
text = 'Dynamic Update',
variable = self.taskMgrVerbose,
command = self.toggleTaskMgrVerbose)
#self.dynamicUpdate.pack(expand = 1, fill = X, side = LEFT)
self.dynamicUpdate.grid(row = 1, column = 1, sticky = EW)
# Pack frames
controlsFrame.pack(fill = X)
controlsFrame.grid_columnconfigure(0, weight = 1)
controlsFrame.grid_columnconfigure(1, weight = 1)
# Add hook to spawnTaskEvents
self.accept('TaskManager-spawnTask', self.spawnTaskHook)
self.accept('TaskManager-removeTask', self.removeTaskHook)
# Get listbox
listbox = self.taskListBox.component('listbox')
# Bind updates to arrow buttons
listbox.bind('<KeyRelease-Up>', self.setCurrentTask)
listbox.bind('<KeyRelease-Down>', self.setCurrentTask)
listbox.bind('<ButtonPress-3>', self.popupMenu)
# And grab focus (to allow keyboard navigation)
listbox.focus_set()
# Update listbox values
self.updateTaskListBox()
[docs] def popupMenu(self, event):
"""
listbox = self.taskListBox.component('listbox')
index = listbox.nearest(event.y)
listbox.selection_clear(0)
listbox.activate(index)
self.taskListBox.select_set(index)
self.setCurrentTask()
"""
self._popupMenu.post(event.widget.winfo_pointerx(),
event.widget.winfo_pointery())
return "break"
[docs] def setCurrentTask(self, event = None):
if len(self.taskListBox.curselection()) > 0: # [gjeon] to avoid crash when nothing is selected
index = int(self.taskListBox.curselection()[0])
self.currentTask = self.__taskDict[index]
else:
self.currentTask = None
[docs] def updateTaskListBox(self):
# Get a list of task names
taskNames = []
self.__taskDict = {}
tasks = self.taskMgr.getTasks()
tasks.sort(key = lambda t: t.getName())
count = 0
for task in tasks:
taskNames.append(task.getName())
self.__taskDict[count] = task
count += 1
if taskNames:
self.taskListBox.setlist(taskNames)
# And set current index (so keypresses will start with index 0)
self.taskListBox.component('listbox').activate(0)
# Select first item
#self.taskListBox.select_set(0) # [gjeon] commented out to avoid focus problem with other lists
self.setCurrentTask()
[docs] def removeCurrentTask(self):
if self.currentTask:
name = self.currentTask.name
ok = 1
if ((name == 'dataLoop') or
(name == 'resetPrevTransform') or
(name == 'tkLoop') or
(name == 'eventManager') or
(name == 'igLoop')):
ok = askokcancel('TaskManagerControls',
'Remove: %s?' % name,
parent = self.parent,
default = 'cancel')
if ok:
self.taskMgr.remove(self.currentTask)
self.updateTaskListBox()
[docs] def removeMatchingTasks(self):
name = self.taskListBox.getcurselection()[0]
ok = 1
if ((name == 'dataLoop') or
(name == 'resetPrevTransform') or
(name == 'tkLoop') or
(name == 'eventManager') or
(name == 'igLoop')):
ok = askokcancel('TaskManagerControls',
'Remove tasks named: %s?' % name,
parent = self.parent,
default = 'cancel')
if ok:
self.taskMgr.remove(name)
self.updateTaskListBox()
[docs] def onDestroy(self):
self.ignore('TaskManager-spawnTask')
self.ignore('TaskManager-removeTask')