Added basic 3D line plot class
This commit is contained in:
parent
51e88bd436
commit
5374afeca8
48
examples/GLLinePlotItem.py
Normal file
48
examples/GLLinePlotItem.py
Normal file
@ -0,0 +1,48 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
## Add path to library (just for examples; you do not need this)
|
||||||
|
import sys, os
|
||||||
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
|
||||||
|
|
||||||
|
from pyqtgraph.Qt import QtCore, QtGui
|
||||||
|
import pyqtgraph.opengl as gl
|
||||||
|
import pyqtgraph as pg
|
||||||
|
import numpy as np
|
||||||
|
|
||||||
|
app = QtGui.QApplication([])
|
||||||
|
w = gl.GLViewWidget()
|
||||||
|
w.opts['distance'] = 40
|
||||||
|
w.show()
|
||||||
|
|
||||||
|
gx = gl.GLGridItem()
|
||||||
|
gx.rotate(90, 0, 1, 0)
|
||||||
|
gx.translate(-10, 0, 0)
|
||||||
|
w.addItem(gx)
|
||||||
|
gy = gl.GLGridItem()
|
||||||
|
gy.rotate(90, 1, 0, 0)
|
||||||
|
gy.translate(0, -10, 0)
|
||||||
|
w.addItem(gy)
|
||||||
|
gz = gl.GLGridItem()
|
||||||
|
gz.translate(0, 0, -10)
|
||||||
|
w.addItem(gz)
|
||||||
|
|
||||||
|
def fn(x, y):
|
||||||
|
return np.cos((x**2 + y**2)**0.5)
|
||||||
|
|
||||||
|
n = 51
|
||||||
|
y = np.linspace(-10,10,n)
|
||||||
|
x = np.linspace(-10,10,100)
|
||||||
|
for i in range(n):
|
||||||
|
yi = np.array([y[i]]*100)
|
||||||
|
d = (x**2 + yi**2)**0.5
|
||||||
|
z = 10 * np.cos(d) / (d+1)
|
||||||
|
pts = np.vstack([x,yi,z]).transpose()
|
||||||
|
plt = gl.GLLinePlotItem(pos=pts, color=pg.glColor((i,n*1.3)))
|
||||||
|
w.addItem(plt)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
## Start Qt event loop unless running in interactive mode.
|
||||||
|
if __name__ == '__main__':
|
||||||
|
import sys
|
||||||
|
if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
|
||||||
|
QtGui.QApplication.instance().exec_()
|
81
pyqtgraph/opengl/items/GLLinePlotItem.py
Normal file
81
pyqtgraph/opengl/items/GLLinePlotItem.py
Normal file
@ -0,0 +1,81 @@
|
|||||||
|
from OpenGL.GL import *
|
||||||
|
from OpenGL.arrays import vbo
|
||||||
|
from .. GLGraphicsItem import GLGraphicsItem
|
||||||
|
from .. import shaders
|
||||||
|
from pyqtgraph import QtGui
|
||||||
|
import numpy as np
|
||||||
|
|
||||||
|
__all__ = ['GLLinePlotItem']
|
||||||
|
|
||||||
|
class GLLinePlotItem(GLGraphicsItem):
|
||||||
|
"""Draws line plots in 3D."""
|
||||||
|
|
||||||
|
def __init__(self, **kwds):
|
||||||
|
GLGraphicsItem.__init__(self)
|
||||||
|
glopts = kwds.pop('glOptions', 'additive')
|
||||||
|
self.setGLOptions(glopts)
|
||||||
|
self.pos = None
|
||||||
|
self.width = 1.
|
||||||
|
self.color = (1.0,1.0,1.0,1.0)
|
||||||
|
self.setData(**kwds)
|
||||||
|
|
||||||
|
def setData(self, **kwds):
|
||||||
|
"""
|
||||||
|
Update the data displayed by this item. All arguments are optional;
|
||||||
|
for example it is allowed to update spot positions while leaving
|
||||||
|
colors unchanged, etc.
|
||||||
|
|
||||||
|
==================== ==================================================
|
||||||
|
Arguments:
|
||||||
|
------------------------------------------------------------------------
|
||||||
|
pos (N,3) array of floats specifying point locations.
|
||||||
|
color tuple of floats (0.0-1.0) specifying
|
||||||
|
a color for the entire item.
|
||||||
|
width float specifying line width
|
||||||
|
==================== ==================================================
|
||||||
|
"""
|
||||||
|
args = ['pos', 'color', 'width', 'connected']
|
||||||
|
for k in kwds.keys():
|
||||||
|
if k not in args:
|
||||||
|
raise Exception('Invalid keyword argument: %s (allowed arguments are %s)' % (k, str(args)))
|
||||||
|
|
||||||
|
for arg in args:
|
||||||
|
if arg in kwds:
|
||||||
|
setattr(self, arg, kwds[arg])
|
||||||
|
#self.vbo.pop(arg, None)
|
||||||
|
self.update()
|
||||||
|
|
||||||
|
def initializeGL(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
#def setupGLState(self):
|
||||||
|
#"""Prepare OpenGL state for drawing. This function is called immediately before painting."""
|
||||||
|
##glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA) ## requires z-sorting to render properly.
|
||||||
|
#glBlendFunc(GL_SRC_ALPHA, GL_ONE)
|
||||||
|
#glEnable( GL_BLEND )
|
||||||
|
#glEnable( GL_ALPHA_TEST )
|
||||||
|
#glDisable( GL_DEPTH_TEST )
|
||||||
|
|
||||||
|
##glEnable( GL_POINT_SMOOTH )
|
||||||
|
|
||||||
|
##glHint(GL_POINT_SMOOTH_HINT, GL_NICEST)
|
||||||
|
##glPointParameterfv(GL_POINT_DISTANCE_ATTENUATION, (0, 0, -1e-3))
|
||||||
|
##glPointParameterfv(GL_POINT_SIZE_MAX, (65500,))
|
||||||
|
##glPointParameterfv(GL_POINT_SIZE_MIN, (0,))
|
||||||
|
|
||||||
|
def paint(self):
|
||||||
|
if self.pos is None:
|
||||||
|
return
|
||||||
|
self.setupGLState()
|
||||||
|
|
||||||
|
glEnableClientState(GL_VERTEX_ARRAY)
|
||||||
|
try:
|
||||||
|
glVertexPointerf(self.pos)
|
||||||
|
glColor4f(*self.color)
|
||||||
|
|
||||||
|
glPointSize(self.width)
|
||||||
|
glDrawArrays(GL_LINE_STRIP, 0, self.pos.size / self.pos.shape[-1])
|
||||||
|
finally:
|
||||||
|
glDisableClientState(GL_VERTEX_ARRAY)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user