根据pyqtgraph.opengl中的信号值调整曲面图项的颜色

时间:2013-09-16 21:30:54

标签: python opengl pyqtgraph

这是尝试进行瀑布式表示。 我需要以不同的颜色表示不同颜色的信号(数组)值(幅度/电平/密度)。 因为我是一个算法和信号处理工程师。而不是软件开发人员,我不熟悉彩色地图和这些东西。因此,如果有人可以使用代码将我的颜色与信号值相关联。

from pyqtgraph.Qt import QtCore, QtGui
import pyqtgraph.opengl as gl
import scipy.ndimage as ndi
import numpy as np

Nf = 90     # No. of frames
Ns = 100    # Signal length

app = QtGui.QApplication([])

w_SA = QtGui.QWidget(); w_SA.setFixedSize(400, 400)

# Create a GL View widget to display data
plt_SA1 = gl.GLViewWidget(w_SA); plt_SA1.move(10, 10); plt_SA1.resize(380, 380)
plt_SA1.setCameraPosition(elevation=90.0, azimuth=0.0, distance=70)
p1 = gl.GLSurfacePlotItem(shader='shaded', color=(0.5, 0.5, 1, 1), smooth=False)
p1.translate(-Nf/2, -Ns/2, 0)
plt_SA1.addItem(p1)

Arx = np.zeros([Nf, Ns])
def update():
    global Arx
    Arx = np.roll(Arx, 1, axis=0)
    Arx[0] = ndi.gaussian_filter(np.random.normal(size=(1,Ns)), (1,1))
    p1.setData(z=Arx)
timer = QtCore.QTimer()
timer.timeout.connect(update)
timer.start(30)

w_SA.show()

if __name__ == '__main__':
    import sys
    if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
        QtGui.QApplication.instance().exec_()

1 个答案:

答案 0 :(得分:2)

您能更具体地了解图像的着色方式吗?如果您不需要OpenGL,这里有一个使用pyqtgraph.ImageView的更简单的解决方案。您可以右键单击右侧的渐变条以更改用于为图像着色的查找表。根据所需的效果,还可以通过多种方式手动设置此表。

from pyqtgraph.Qt import QtCore, QtGui
import pyqtgraph as pg
import scipy.ndimage as ndi
import numpy as np

Nf = 90     # No. of frames
Ns = 100    # Signal length

app = QtGui.QApplication([])

Arx = np.zeros([Nf, Ns])
win = pg.image(Arx)
win.view.setAspectLocked()
def update():
    global Arx
    Arx = np.roll(Arx, 1, axis=0)
    Arx[0] = ndi.gaussian_filter(np.random.normal(size=(1,Ns)), (1, 1))
    win.setImage(Arx.T, autoRange=False)

timer = QtCore.QTimer()
timer.timeout.connect(update)
timer.start(30)

if __name__ == '__main__':
    import sys
    if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
        QtGui.QApplication.instance().exec_()