class ObjectSelector(DirectObject.DirectObject):
def __init__(self):
self.menu = DirectOptionMenu(
text = "Attach camera to",
scale = 0.05,
items = [],
initialitem = 0,
highlightColor = (0.65,0.65,0.65,1),
command = self.select,
textMayChange = 1
)
self.visible = False
self.menu.hide()
self.camera_poses = {}
#-------------------------------------------------------------------------------
def select(self, item):
mat = Mat4(base.camera.getMat())
mat.invertInPlace()
key = base.camera.getParent().getName()
self.camera_poses[key] = mat
base.camera.reparentTo(framework.world.meshes[item])
if self.camera_poses.has_key(item):
mat = self.camera_poses[item]
base.mouseInterfaceNode.setMat(mat)
else:
base.camera.setPos(-10, -5, 5)
base.camera.lookAt(5, 0, 0)
mat = Mat4(base.camera.getMat())
mat.invertInPlace()
base.mouseInterfaceNode.setMat(mat)
self.toggle()
#-------------------------------------------------------------------------------
def toggle(self):
self.visible = not self.visible
if self.visible:
items = self.menu["items"]
del items[:]
i = 0
selected = 0
for k, v in framework.world.meshes.iteritems():
items.append(k)
if k == base.camera.getParent().getName():
selected = i
i = i + 1
self.menu["items"] = items
self.menu.set(selected, 0)
self.menu.show()
else:
self.menu.hide()
class ObjectManager(DirectObject.DirectObject):
''' Gui for object selection and displaying basic proerties.'''
V_SPACING = 1.3
H_SPACING = 1.3
FRAME_COLOR = (200, 200, 200, 0.5)
TEXT_COLOR = (0.8, 0.8, 0.8, 0.8)
NO_COLOR = (0, 0, 0, 0)
HL_COLOR = (0.65, 0.65, 0.65, 1)
SCALE = 0.05
def __init__(self):
self.top = 0.5 / self.SCALE
self.frame = DirectFrame(
relief = DGG.GROOVE,
frameColor = self.FRAME_COLOR,
frameSize = (0, 0.8 / self.SCALE, 0, self.top),
pos = (-0.5, 0, -0.5),
scale = self.SCALE
)
self.label("Current Object", 1, 1)
self.objectMenu = DirectOptionMenu(
parent = self.frame,
items = [],
initialitem = 0,
command = self.objectMenuSelect,
textMayChange = 1
)
self.layout(self.objectMenu, 8, 1)
self.label ("Position", 1, 2,)
self.positionLabel = self.label("", 8, 2)
self.label ("Orientation", 1, 3,)
self.orientationLabel = self.label("", 8, 3)
self.frame.hide()
self.accept('window-event', self.windowEvent)
self.objects = {}
self.camera_poses = {}
framework.scheduler.addTask("ObjectManager/Update", self.update)
#-------------------------------------------------------------------------------
def label(self, text, x, y):
result = OnscreenText(
parent = self.frame,
mayChange = True,
text = text,
scale = 1,
fg = self.TEXT_COLOR,
align = panda.TextNode.ALeft
)
result.setPos(x, self.top - y * self.V_SPACING)
return result
#-------------------------------------------------------------------------------
def layout(self, widget, x, y):
widget["text_fg"] = self.TEXT_COLOR
widget["text_align"] = panda.TextNode.ALeft
widget["highlightColor"] = self.HL_COLOR,
widget.setPos(x, 0, self.top - y * self.V_SPACING)
widget.resetFrameSize()
#-------------------------------------------------------------------------------
def objectMenuSelect(self, item):
mat = panda.Mat4(base.camera.getMat())
mat.invertInPlace()
key = base.camera.getParent().getName()
self.camera_poses[key] = mat
base.camera.reparentTo(self.objects[item])
if self.camera_poses.has_key(item):
mat = self.camera_poses[item]
base.mouseInterfaceNode.setMat(mat)
else:
base.camera.setPos(-10, -5, 5)
base.camera.lookAt(5, 0, 0)
mat = panda.Mat4(base.camera.getMat())
mat.invertInPlace()
base.mouseInterfaceNode.setMat(mat)
self.objectMenuUpdate()
#-------------------------------------------------------------------------------
def toggle(self):
self.frame.toggleVis()
if not self.frame.isHidden():
self.update()
self.frame.setPos(0, 0, 0)
self.windowEvent(base.win)
#-------------------------------------------------------------------------------
def update(self, task = None):
if not self.frame.isHidden():
self.objectPropertiesUpdate()
if not task:
self.objectMenuUpdate()
return True
#-------------------------------------------------------------------------------
def objectPropertiesUpdate(self):
current = base.camera.getParent()
pos = current.getPos()
self.positionLabel.setText("%.2f, %.2f, %.2f" % (pos[0], pos[1], pos[2]))
hpr = current.getHpr()
self.orientationLabel.setText("%.2f, %.2f, %.2f" % (hpr[0], hpr[1], hpr[2]))
#-------------------------------------------------------------------------------
def objectMenuUpdate(self):
items = self.objectMenu["items"]
del items[:]
self.objects = {}
i = 0
selected = 0
if framework.world and framework.world.scene:
for mesh in framework.world.scene.meshes:
self.objects[mesh.getName()] = mesh
items.append(mesh.getName())
if mesh == base.camera.getParent():
selected = i
i = i + 1
self.objectMenu["items"] = items
if len(items) > 0:
self.objectMenu.set(selected, 0)
self.objectMenu["text_align"] = panda.TextNode.ALeft
self.objectMenu.resetFrameSize()
#-------------------------------------------------------------------------------
def windowEvent(self, window):
self.frame.setPos(-0.5, 0, -0.5)