Added GL picking, matrix retrieval methods
This commit is contained in:
parent
934e50ad55
commit
a20e732f65
@ -80,14 +80,17 @@ class GLViewWidget(QtOpenGL.QGLWidget):
|
||||
#self.update()
|
||||
|
||||
def setProjection(self, region=None):
|
||||
m = self.projectionMatrix(region)
|
||||
glMatrixMode(GL_PROJECTION)
|
||||
glLoadIdentity()
|
||||
a = np.array(m.copyDataTo()).reshape((4,4))
|
||||
glMultMatrixf(a.transpose())
|
||||
|
||||
def projectionMatrix(self, region=None):
|
||||
# Xw = (Xnd + 1) * width/2 + X
|
||||
if region is None:
|
||||
region = (0, 0, self.width(), self.height())
|
||||
## Create the projection matrix
|
||||
glMatrixMode(GL_PROJECTION)
|
||||
glLoadIdentity()
|
||||
#w = self.width()
|
||||
#h = self.height()
|
||||
|
||||
x0, y0, w, h = self.getViewport()
|
||||
dist = self.opts['distance']
|
||||
fov = self.opts['fov']
|
||||
@ -105,20 +108,45 @@ class GLViewWidget(QtOpenGL.QGLWidget):
|
||||
bottom = t * ((region[1]-y0) * (2.0/h) - 1)
|
||||
top = t * ((region[1]+region[3]-y0) * (2.0/h) - 1)
|
||||
|
||||
glFrustum( left, right, bottom, top, nearClip, farClip)
|
||||
#glFrustum(-r, r, -t, t, nearClip, farClip)
|
||||
tr = QtGui.QMatrix4x4()
|
||||
tr.frustum(left, right, bottom, top, nearClip, farClip)
|
||||
return tr
|
||||
|
||||
def setModelview(self):
|
||||
glMatrixMode(GL_MODELVIEW)
|
||||
glLoadIdentity()
|
||||
glTranslatef( 0.0, 0.0, -self.opts['distance'])
|
||||
glRotatef(self.opts['elevation']-90, 1, 0, 0)
|
||||
glRotatef(self.opts['azimuth']+90, 0, 0, -1)
|
||||
m = self.viewMatrix()
|
||||
a = np.array(m.copyDataTo()).reshape((4,4))
|
||||
glMultMatrixf(a.transpose())
|
||||
|
||||
def viewMatrix(self):
|
||||
tr = QtGui.QMatrix4x4()
|
||||
tr.translate( 0.0, 0.0, -self.opts['distance'])
|
||||
tr.rotate(self.opts['elevation']-90, 1, 0, 0)
|
||||
tr.rotate(self.opts['azimuth']+90, 0, 0, -1)
|
||||
center = self.opts['center']
|
||||
glTranslatef(-center.x(), -center.y(), -center.z())
|
||||
tr.translate(-center.x(), -center.y(), -center.z())
|
||||
return tr
|
||||
|
||||
def itemsAt(self, region=None):
|
||||
#buf = np.zeros(100000, dtype=np.uint)
|
||||
buf = glSelectBuffer(100000)
|
||||
try:
|
||||
glRenderMode(GL_SELECT)
|
||||
glInitNames()
|
||||
glPushName(0)
|
||||
self._itemNames = {}
|
||||
self.paintGL(region=region, useItemNames=True)
|
||||
|
||||
def paintGL(self, region=None, viewport=None):
|
||||
finally:
|
||||
hits = glRenderMode(GL_RENDER)
|
||||
|
||||
items = [(h.near, h.names[0]) for h in hits]
|
||||
items.sort(key=lambda i: i[0])
|
||||
|
||||
return [self._itemNames[i[1]] for i in items]
|
||||
|
||||
def paintGL(self, region=None, viewport=None, useItemNames=False):
|
||||
"""
|
||||
viewport specifies the arguments to glViewport. If None, then we use self.opts['viewport']
|
||||
region specifies the sub-region of self.opts['viewport'] that should be rendered.
|
||||
@ -131,9 +159,9 @@ class GLViewWidget(QtOpenGL.QGLWidget):
|
||||
self.setProjection(region=region)
|
||||
self.setModelview()
|
||||
glClear( GL_DEPTH_BUFFER_BIT | GL_COLOR_BUFFER_BIT )
|
||||
self.drawItemTree()
|
||||
self.drawItemTree(useItemNames=useItemNames)
|
||||
|
||||
def drawItemTree(self, item=None):
|
||||
def drawItemTree(self, item=None, useItemNames=False):
|
||||
if item is None:
|
||||
items = [x for x in self.items if x.parentItem() is None]
|
||||
else:
|
||||
@ -146,6 +174,9 @@ class GLViewWidget(QtOpenGL.QGLWidget):
|
||||
if i is item:
|
||||
try:
|
||||
glPushAttrib(GL_ALL_ATTRIB_BITS)
|
||||
if useItemNames:
|
||||
glLoadName(id(i))
|
||||
self._itemNames[id(i)] = i
|
||||
i.paint()
|
||||
except:
|
||||
import pyqtgraph.debug
|
||||
@ -168,7 +199,7 @@ class GLViewWidget(QtOpenGL.QGLWidget):
|
||||
tr = i.transform()
|
||||
a = np.array(tr.copyDataTo()).reshape((4,4))
|
||||
glMultMatrixf(a.transpose())
|
||||
self.drawItemTree(i)
|
||||
self.drawItemTree(i, useItemNames=useItemNames)
|
||||
finally:
|
||||
glMatrixMode(GL_MODELVIEW)
|
||||
glPopMatrix()
|
||||
|
Loading…
Reference in New Issue
Block a user