ASCEE/pyqtgraph
5
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix: support colormaps of shape (h, 1)

Browse Source
This commit is contained in:
KIU Shueng Chuan 2021-04-10 14:47:58 +08:00
parent 941a436df7
commit a6bbb1c48c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pyqtgraph/graphicsItems/ImageItem.py
View File

@ -529,7 +529,7 @@ class ImageItem(GraphicsObject):
# levels and/or lut --> lut-only # levels and/or lut --> lut-only
fmt = QtGui.QImage.Format.Format_Indexed8 fmt = QtGui.QImage.Format.Format_Indexed8
if lut.ndim == 1 or lut.shape[1] == 1: if lut.ndim == 1 or lut.shape[1] == 1:
ctbl = [QtGui.qRgb(x,x,x) for x in lut.tolist()] ctbl = [QtGui.qRgb(x,x,x) for x in lut.ravel().tolist()]
elif lut.shape[1] == 3: elif lut.shape[1] == 3:
ctbl = [QtGui.qRgb(*rgb) for rgb in lut.tolist()] ctbl = [QtGui.qRgb(*rgb) for rgb in lut.tolist()]
elif lut.shape[1] == 4: elif lut.shape[1] == 4: