mirror of
https://github.com/panda3d/panda3d.git
synced 2025-10-04 02:42:49 -04:00
419 lines
14 KiB
Python
419 lines
14 KiB
Python
#################################################################
|
|
# seTree.py
|
|
# Originally from Tree.py
|
|
# Altered by Yi-Hong Lin, yihhongl@andrew.cmu.edu, 2004
|
|
#
|
|
# This class actually decides the behavior of the sceneGraphExplorer
|
|
# You might feel it realy looks like the original one, but we actually did a lots of change in it.
|
|
# such as, when selection happend in other place, such as picking directly inside the scene. or,
|
|
# when user removed something by hot key.
|
|
# The rename process has also been changed. It won't be rename in here anymore.
|
|
# Instead, here we will send out a message to sceneEditor to reaname the target.
|
|
#
|
|
#################################################################
|
|
|
|
import os, sys, string, Pmw, Tkinter
|
|
from direct.showbase.DirectObject import DirectObject
|
|
from Tkinter import IntVar, Menu, PhotoImage, Label, Frame, Entry
|
|
from pandac.PandaModules import *
|
|
|
|
# Initialize icon directory
|
|
ICONDIR = getModelPath().findFile(Filename('icons')).toOsSpecific()
|
|
if not os.path.isdir(ICONDIR):
|
|
raise RuntimeError, "can't find DIRECT icon directory (%s)" % `ICONDIR`
|
|
|
|
class TreeNode:
|
|
|
|
def __init__(self, canvas, parent, item, menuList = []):
|
|
self.canvas = canvas
|
|
self.parent = parent
|
|
self.item = item
|
|
self.state = 'collapsed'
|
|
self.selected = 0
|
|
self.children = {}
|
|
self.kidKeys = []
|
|
self.x = self.y = None
|
|
self.iconimages = {} # cache of PhotoImage instances for icons
|
|
self.menuList = menuList
|
|
self.menuVar = IntVar()
|
|
self.menuVar.set(0)
|
|
self._popupMenu = None
|
|
self.image_id = None
|
|
if self.menuList:
|
|
if self.menuList[-1] == 'Separator':
|
|
self.menuList = self.menuList[:-1]
|
|
self._popupMenu = Menu(self.canvas, tearoff = 0)
|
|
for i in range(len(self.menuList)):
|
|
item = self.menuList[i]
|
|
if item == 'Separator':
|
|
self._popupMenu.add_separator()
|
|
else:
|
|
self._popupMenu.add_radiobutton(
|
|
label = item,
|
|
variable = self.menuVar,
|
|
value = i,
|
|
indicatoron = 0,
|
|
command = self.popupMenuCommand)
|
|
|
|
def destroy(self):
|
|
for key in self.kidKeys:
|
|
c = self.children[key]
|
|
del self.children[key]
|
|
c.destroy()
|
|
self.parent = None
|
|
|
|
def geticonimage(self, name):
|
|
try:
|
|
return self.iconimages[name]
|
|
except KeyError:
|
|
pass
|
|
file, ext = os.path.splitext(name)
|
|
ext = ext or ".gif"
|
|
fullname = os.path.join(ICONDIR, file + ext)
|
|
image = PhotoImage(master=self.canvas, file=fullname)
|
|
self.iconimages[name] = image
|
|
return image
|
|
|
|
def select(self, event=None):
|
|
if self.selected:
|
|
return
|
|
self.deselectall()
|
|
self.selected = 1
|
|
if self.parent != None:
|
|
if self.parent.state == 'expanded':
|
|
self.canvas.delete(self.image_id)
|
|
self.drawicon()
|
|
self.drawtext()
|
|
self.item.OnSelect(event)
|
|
|
|
def deselect(self, event=None):
|
|
if not self.selected:
|
|
return
|
|
self.selected = 0
|
|
if self.parent != None:
|
|
if self.parent.state == 'expanded':
|
|
self.canvas.delete(self.image_id)
|
|
self.drawicon()
|
|
self.drawtext()
|
|
|
|
def deselectall(self):
|
|
if self.parent:
|
|
self.parent.deselectall()
|
|
else:
|
|
self.deselecttree()
|
|
|
|
def deselecttree(self):
|
|
if self.selected:
|
|
self.deselect()
|
|
for key in self.kidKeys:
|
|
child = self.children[key]
|
|
child.deselecttree()
|
|
|
|
def flip(self, event=None):
|
|
if self.state == 'expanded':
|
|
self.collapse()
|
|
else:
|
|
self.expand()
|
|
self.item.OnDoubleClick()
|
|
return "break"
|
|
|
|
def popupMenu(self, event=None):
|
|
if self._popupMenu:
|
|
self._popupMenu.post(event.widget.winfo_pointerx(),
|
|
event.widget.winfo_pointery())
|
|
return "break"
|
|
|
|
def popupMenuCommand(self):
|
|
command = self.menuList[self.menuVar.get()]
|
|
self.item.MenuCommand(command)
|
|
if self.parent and (command != 'Update Explorer'):
|
|
# Update parent to try to keep explorer up to date
|
|
self.parent.update()
|
|
|
|
def expand(self, event=None):
|
|
if not self.item.IsExpandable():
|
|
return
|
|
if self.state != 'expanded':
|
|
self.state = 'expanded'
|
|
self.update()
|
|
self.view()
|
|
|
|
def collapse(self, event=None):
|
|
if self.state != 'collapsed':
|
|
self.state = 'collapsed'
|
|
self.update()
|
|
|
|
def view(self):
|
|
top = self.y - 2
|
|
bottom = self.lastvisiblechild().y + 17
|
|
height = bottom - top
|
|
visible_top = self.canvas.canvasy(0)
|
|
visible_height = self.canvas.winfo_height()
|
|
visible_bottom = self.canvas.canvasy(visible_height)
|
|
if visible_top <= top and bottom <= visible_bottom:
|
|
return
|
|
x0, y0, x1, y1 = self.canvas._getints(self.canvas['scrollregion'])
|
|
if top >= visible_top and height <= visible_height:
|
|
fraction = top + height - visible_height
|
|
else:
|
|
fraction = top
|
|
fraction = float(fraction) / y1
|
|
self.canvas.yview_moveto(fraction)
|
|
|
|
def reveal(self):
|
|
# Make sure all parent nodes are marked as expanded
|
|
parent = self.parent
|
|
while parent:
|
|
if parent.state == 'collapsed':
|
|
parent.state = 'expanded'
|
|
parent = parent.parent
|
|
else:
|
|
break
|
|
# Redraw tree accordingly
|
|
self.update()
|
|
# Bring this item into view
|
|
self.view()
|
|
|
|
def lastvisiblechild(self):
|
|
if self.kidKeys and self.state == 'expanded':
|
|
return self.children[self.kidKeys[-1]].lastvisiblechild()
|
|
else:
|
|
return self
|
|
|
|
def update(self):
|
|
if self.parent:
|
|
self.parent.update()
|
|
else:
|
|
oldcursor = self.canvas['cursor']
|
|
self.canvas['cursor'] = "watch"
|
|
self.canvas.update()
|
|
self.canvas.delete(Tkinter.ALL) # XXX could be more subtle
|
|
self.draw(7, 2)
|
|
x0, y0, x1, y1 = self.canvas.bbox(Tkinter.ALL)
|
|
self.canvas.configure(scrollregion=(0, 0, x1, y1))
|
|
self.canvas['cursor'] = oldcursor
|
|
|
|
def draw(self, x, y):
|
|
# XXX This hard-codes too many geometry constants!
|
|
self.x, self.y = x, y
|
|
self.drawicon()
|
|
self.drawtext()
|
|
if self.state != 'expanded':
|
|
return y+17
|
|
# draw children
|
|
sublist = self.item._GetSubList()
|
|
if not sublist:
|
|
# IsExpandable() was mistaken; that's allowed
|
|
return y+17
|
|
self.kidKeys = []
|
|
for item in sublist:
|
|
key = item.GetKey()
|
|
if self.children.has_key(key):
|
|
child = self.children[key]
|
|
else:
|
|
child = TreeNode(self.canvas, self, item, self.menuList)
|
|
self.children[key] = child
|
|
self.kidKeys.append(key)
|
|
# Remove unused children
|
|
for key in self.children.keys():
|
|
if key not in self.kidKeys:
|
|
del(self.children[key])
|
|
cx = x+20
|
|
cy = y+17
|
|
cylast = 0
|
|
for key in self.kidKeys:
|
|
child = self.children[key]
|
|
cylast = cy
|
|
self.canvas.create_line(x+9, cy+7, cx, cy+7, fill="gray50")
|
|
cy = child.draw(cx, cy)
|
|
if child.item.IsExpandable():
|
|
if child.state == 'expanded':
|
|
iconname = "minusnode"
|
|
callback = child.collapse
|
|
else:
|
|
iconname = "plusnode"
|
|
callback = child.expand
|
|
image = self.geticonimage(iconname)
|
|
id = self.canvas.create_image(x+9, cylast+7, image=image)
|
|
# XXX This leaks bindings until canvas is deleted:
|
|
self.canvas.tag_bind(id, "<1>", callback)
|
|
self.canvas.tag_bind(id, "<Double-1>", lambda x: None)
|
|
id = self.canvas.create_line(x+9, y+10, x+9, cylast+7,
|
|
##stipple="gray50", # XXX Seems broken in Tk 8.0.x
|
|
fill="gray50")
|
|
self.canvas.tag_lower(id) # XXX .lower(id) before Python 1.5.2
|
|
return cy
|
|
|
|
def drawicon(self):
|
|
if self.selected:
|
|
imagename = (self.item.GetSelectedIconName() or
|
|
self.item.GetIconName() or
|
|
"openfolder")
|
|
else:
|
|
imagename = self.item.GetIconName() or "folder"
|
|
image = self.geticonimage(imagename)
|
|
id = self.canvas.create_image(self.x, self.y, anchor="nw", image=image)
|
|
self.image_id = id
|
|
self.canvas.tag_bind(id, "<1>", self.select)
|
|
self.canvas.tag_bind(id, "<Double-1>", self.flip)
|
|
self.canvas.tag_bind(id, "<3>", self.popupMenu)
|
|
|
|
def drawtext(self, text=None):
|
|
textx = self.x+20-1
|
|
texty = self.y-1
|
|
labeltext = self.item.GetLabelText()
|
|
if labeltext:
|
|
id = self.canvas.create_text(textx, texty, anchor="nw",
|
|
text=labeltext)
|
|
self.canvas.tag_bind(id, "<1>", self.select)
|
|
self.canvas.tag_bind(id, "<Double-1>", self.flip)
|
|
x0, y0, x1, y1 = self.canvas.bbox(id)
|
|
textx = max(x1, 200) + 10
|
|
if text==None:
|
|
text = self.item.GetText() or "<no text>"
|
|
try:
|
|
self.entry
|
|
except AttributeError:
|
|
pass
|
|
else:
|
|
self.edit_finish()
|
|
try:
|
|
label = self.label
|
|
except AttributeError:
|
|
# padding carefully selected (on Windows) to match Entry widget:
|
|
self.label = Label(self.canvas, text=text, bd=0, padx=2, pady=2)
|
|
if self.selected:
|
|
self.label.configure(fg="white", bg="darkblue")
|
|
else:
|
|
fg = self.item.GetTextFg()
|
|
self.label.configure(fg=fg, bg="white")
|
|
id = self.canvas.create_window(textx, texty,
|
|
anchor="nw", window=self.label)
|
|
self.label.bind("<1>", self.select_or_edit)
|
|
self.label.bind("<Double-1>", self.flip)
|
|
self.label.bind("<3>", self.popupMenu)
|
|
# Update text if necessary
|
|
if text != self.label['text']:
|
|
self.label['text'] = text
|
|
self.text_id = id
|
|
|
|
def select_or_edit(self, event=None):
|
|
if self.selected and self.item.IsEditable():
|
|
text = self.item.GetTextForEdit()
|
|
self.label['text'] = text
|
|
self.drawtext(text)
|
|
self.edit(event)
|
|
else:
|
|
self.select(event)
|
|
|
|
def edit(self, event=None):
|
|
self.entry = Entry(self.label, bd=0, highlightthickness=1, width=0)
|
|
self.entry.insert(0, self.label['text'])
|
|
self.entry.selection_range(0, Tkinter.END)
|
|
self.entry.pack(ipadx=5)
|
|
self.entry.focus_set()
|
|
self.entry.bind("<Return>", self.edit_finish)
|
|
self.entry.bind("<Escape>", self.edit_cancel)
|
|
|
|
def edit_finish(self, event=None):
|
|
try:
|
|
entry = self.entry
|
|
del self.entry
|
|
except AttributeError:
|
|
return
|
|
text = entry.get()
|
|
entry.destroy()
|
|
if text and text != self.item.GetText():
|
|
self.item.SetText(text)
|
|
text = self.item.GetText()
|
|
self.label['text'] = text
|
|
self.drawtext()
|
|
self.canvas.focus_set()
|
|
|
|
def edit_cancel(self, event=None):
|
|
self.drawtext()
|
|
self.canvas.focus_set()
|
|
|
|
def find(self, searchKey):
|
|
# Search for a node who's key matches the given key
|
|
# Is it this node
|
|
if searchKey == self.item.GetKey():
|
|
return self
|
|
# Nope, check the children
|
|
sublist = self.item._GetSubList()
|
|
for item in sublist:
|
|
key = item.GetKey()
|
|
# Use existing child or create new TreeNode if none exists
|
|
if self.children.has_key(key):
|
|
child = self.children[key]
|
|
else:
|
|
child = TreeNode(self.canvas, self, item, self.menuList)
|
|
# Update local list of children and keys
|
|
self.children[key] = child
|
|
self.kidKeys.append(key)
|
|
# See if node is child (or one of child's descendants)
|
|
retVal = child.find(searchKey)
|
|
if retVal:
|
|
return retVal
|
|
# Not here
|
|
return None
|
|
|
|
class TreeItem:
|
|
|
|
"""Abstract class representing tree items.
|
|
|
|
Methods should typically be overridden, otherwise a default action
|
|
is used.
|
|
|
|
"""
|
|
|
|
def __init__(self):
|
|
"""Constructor. Do whatever you need to do."""
|
|
|
|
def GetText(self):
|
|
"""Return text string to display."""
|
|
|
|
def GetTextFg(self):
|
|
return "black"
|
|
|
|
def GetLabelText(self):
|
|
"""Return label text string to display in front of text (if any)."""
|
|
|
|
def IsExpandable(self):
|
|
"""Return whether there are subitems."""
|
|
return 1
|
|
|
|
def _GetSubList(self):
|
|
"""Do not override! Called by TreeNode."""
|
|
if not self.IsExpandable():
|
|
return []
|
|
sublist = self.GetSubList()
|
|
return sublist
|
|
|
|
def IsEditable(self):
|
|
"""Return whether the item's text may be edited."""
|
|
|
|
def SetText(self, text):
|
|
"""Change the item's text (if it is editable)."""
|
|
|
|
def GetIconName(self):
|
|
"""Return name of icon to be displayed normally."""
|
|
|
|
def GetSelectedIconName(self):
|
|
"""Return name of icon to be displayed when selected."""
|
|
|
|
def GetSubList(self):
|
|
"""Return list of items forming sublist."""
|
|
|
|
def OnDoubleClick(self):
|
|
"""Called on a double-click on the item."""
|
|
|
|
def OnSelect(self):
|
|
"""Called when item selected."""
|
|
|
|
def GetTextForEdit(self):
|
|
"""Called before editting the item."""
|
|
|
|
|
|
|