pyqtgraph: проблема с настройкой smooth = False для объекта GLSurfacePlotItem - PullRequest
1 голос
/ 08 февраля 2020

В настоящее время я пытаюсь построить базовый c цветовой кодированный «срез» в 3D с использованием pyqtgraph. Когда я инициализирую GLSurfacePlotItem объект с smooth=True, все работает нормально. Однако я получаю огромное количество ошибок и предупреждений, когда просто устанавливаю smooth=False. К сожалению, документация не особо объясняет этот параметр.

Ниже приведен пример кода. Чего мне не хватает?

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import sys
import numpy as np
import matplotlib
import matplotlib.pyplot as plt

from pyqtgraph.Qt import QtCore, QtGui
import pyqtgraph.opengl as gl
import pyqtgraph as pg

x, y, z = 30, 20, 10
data = np.random.rand(x,y,z)

cmap = matplotlib.cm.get_cmap('jet')
norm = matplotlib.colors.PowerNorm(vmin=0.0, vmax=1.0, gamma=2.5)
m = matplotlib.cm.ScalarMappable(norm=norm, cmap=cmap)

colors = np.zeros((x,y,z,4), dtype=np.float16)
for i in range(x):
    colors[i,:,:,:] = m.to_rgba(data[i,:,:])

app = QtGui.QApplication(sys.argv)
w = gl.GLViewWidget()

zslice = gl.GLSurfacePlotItem(x=np.arange(0,x,1), y=np.arange(0,y,1), z=np.ones((x,y))*0.5, colors=colors[:,:,0,:], smooth=True, computeNormals=False)

w.addItem(zslice)
w.setGeometry(100, 10, 1280, 720)
w.show()

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

Сообщение об ошибке, которое я получаю при установке smooth=False:

[13:16:04]

    |==============================>>
    |  Traceback (most recent call last):
    |    File "./test5.py", line 34, in <module>
    |      QtGui.QApplication.instance().exec_()
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 179, in paintGL
    |      self.drawItemTree(useItemNames=useItemNames)
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 219, in drawItemTree
    |      self.drawItemTree(i, useItemNames=useItemNames)
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 200, in drawItemTree
    |      debug.printExc()
    |    --- exception caught here ---
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 197, in drawItemTree
    |      i.paint()
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/items/GLMeshItem.py", line 167, in paint
    |      self.parseMeshData()
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/items/GLMeshItem.py", line 151, in parseMeshData
    |      self.colors = md.vertexColors(indexed='faces')
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/MeshData.py", line 239, in vertexColors
    |      self._vertexColorsIndexedByFaces = self._vertexColors[self.faces()]
    |  IndexError: index 30 is out of bounds for axis 0 with size 30
    |==============================<<
Error while drawing item <pyqtgraph.opengl.items.GLSurfacePlotItem.GLSurfacePlotItem object at 0x7f9f0cf21410>.
[13:16:04]

    |==============================>>
    |  Traceback (most recent call last):
    |    File "./test5.py", line 34, in <module>
    |      QtGui.QApplication.instance().exec_()
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 179, in paintGL
    |      self.drawItemTree(useItemNames=useItemNames)
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 219, in drawItemTree
    |      self.drawItemTree(i, useItemNames=useItemNames)
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 200, in drawItemTree
    |      debug.printExc()
    |    --- exception caught here ---
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 197, in drawItemTree
    |      i.paint()
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/items/GLMeshItem.py", line 167, in paint
    |      self.parseMeshData()
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/items/GLMeshItem.py", line 151, in parseMeshData
    |      self.colors = md.vertexColors(indexed='faces')
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/MeshData.py", line 239, in vertexColors
    |      self._vertexColorsIndexedByFaces = self._vertexColors[self.faces()]
    |  IndexError: index 30 is out of bounds for axis 0 with size 30
    |==============================<<
Error while drawing item <pyqtgraph.opengl.items.GLSurfacePlotItem.GLSurfacePlotItem object at 0x7f9f0cf21410>.
[13:16:04]

    |==============================>>
    |  Traceback (most recent call last):
    |    File "./test5.py", line 34, in <module>
    |      QtGui.QApplication.instance().exec_()
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 179, in paintGL
    |      self.drawItemTree(useItemNames=useItemNames)
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 219, in drawItemTree
    |      self.drawItemTree(i, useItemNames=useItemNames)
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 200, in drawItemTree
    |      debug.printExc()
    |    --- exception caught here ---
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 197, in drawItemTree
    |      i.paint()
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/items/GLMeshItem.py", line 167, in paint
    |      self.parseMeshData()
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/items/GLMeshItem.py", line 151, in parseMeshData
    |      self.colors = md.vertexColors(indexed='faces')
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/MeshData.py", line 239, in vertexColors
    |      self._vertexColorsIndexedByFaces = self._vertexColors[self.faces()]
    |  IndexError: index 30 is out of bounds for axis 0 with size 30
    |==============================<<
Error while drawing item <pyqtgraph.opengl.items.GLSurfacePlotItem.GLSurfacePlotItem object at 0x7f9f0cf21410>.
[13:16:06]

    |==============================>>
    |  Traceback (most recent call last):
    |    File "./test5.py", line 34, in <module>
    |      QtGui.QApplication.instance().exec_()
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 179, in paintGL
    |      self.drawItemTree(useItemNames=useItemNames)
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 219, in drawItemTree
    |      self.drawItemTree(i, useItemNames=useItemNames)
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 200, in drawItemTree
    |      debug.printExc()
    |    --- exception caught here ---
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/GLViewWidget.py", line 197, in drawItemTree
    |      i.paint()
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/items/GLMeshItem.py", line 167, in paint
    |      self.parseMeshData()
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/items/GLMeshItem.py", line 151, in parseMeshData
    |      self.colors = md.vertexColors(indexed='faces')
    |    File "/home/steve/anaconda3/lib/python3.7/site-packages/pyqtgraph/opengl/MeshData.py", line 239, in vertexColors
    |      self._vertexColorsIndexedByFaces = self._vertexColors[self.faces()]
    |  IndexError: index 30 is out of bounds for axis 0 with size 30
    |==============================<<
Error while drawing item <pyqtgraph.opengl.items.GLSurfacePlotItem.GLSurfacePlotItem object at 0x7f9f0cf21410>.

1 Ответ

0 голосов
/ 17 февраля 2020

В сообщении об ошибке говорится: «IndexError: индекс 30 выходит за пределы оси 0 с размером 30», что говорит о том, что вы или используемая вами библиотека могли бы выполнять цикл от 1 до 30 вместо от 0 до 29. (Настройка smooth = False может отправить ваш код в часть библиотеки с ошибкой в ​​ней.) Поскольку ваша переменная x равна 30, я предполагаю, что это точка происхождения. В сообщении об ошибке говорится, что проблема в ./test5.py ", строка 34, поэтому я бы начал с нее.

...