PickRect code snippet

[Gladsomebeast]

Example of window.pickRect that I though could be handy for other folks out in vizland:

import viz
import vizact

viz.go()

model = viz.add("mini.osg")
viz.MainView.setPosition([0, 1.2, -5])

viz.mouse.setOverride(viz.ON)
vizact.onkeydown(' ', viz.mouse.setOverride, viz.TOGGLE)

print model.getNodeNames()
parts = []
for name in model.getNodeNames():
part = model.getChild(name)
part.nodeName = name
parts.append(part)


#Rectangle Pick
viz.startLayer(viz.LINE_LOOP)
viz.vertex([0, 0, 0])
viz.vertex([0, 0, 0])
viz.vertex([0, 0, 0])
viz.vertex([0, 0, 0])
box = viz.endLayer(parent=viz.SCREEN)
box.alpha(.8)

firstPoint = [0,0]
def startPickRect():
global firstPoint
firstPoint = viz.mouse.getPosition()
box.setVertex(0, firstPoint+[0])
vizact.onmousedown(viz.MOUSEBUTTON_LEFT, startPickRect)

def updateBoxVisual():
if viz.mouse.getState() & viz. MOUSEBUTTON_LEFT:
mouseScreenPos = viz.mouse.getPosition()
box.setVertex(1, [mouseScreenPos[0],firstPoint[1],0])
box.setVertex(2, mouseScreenPos+[0])
box.setVertex(3, [firstPoint[0],mouseScreenPos[1],0])
vizact.onupdate(viz.PRIORITY_DEFAULT, updateBoxVisual)

def endPickRect():
mouseScreenPos = viz.mouse.getPosition()
#update box visual
box.setVertex(1, [mouseScreenPos[0],firstPoint[1],0])
box.setVertex(2, mouseScreenPos+[0])
box.setVertex(3, [firstPoint[0],mouseScreenPos[1],0])

rect = firstPoint + mouseScreenPos
# Get all nodes within the rectangle
picked = viz.MainWindow.pickRect(rect)
for node in picked:
print node.nodeName
vizact.onmouseup(viz.MOUSEBUTTON_LEFT, endPickRect)