Use clip_array instead of np.clip in GLVolumeItem

This commit is contained in:
Ogi Moore 2021-04-18 10:16:09 -07:00
parent f8cefa6284
commit f60fa3b534

View File

@ -11,6 +11,7 @@ import numpy as np
import pyqtgraph as pg import pyqtgraph as pg
from pyqtgraph.Qt import QtCore, QtGui from pyqtgraph.Qt import QtCore, QtGui
import pyqtgraph.opengl as gl import pyqtgraph.opengl as gl
from pyqtgraph.functions import clip_array
app = pg.mkQApp("GLVolumeItem Example") app = pg.mkQApp("GLVolumeItem Example")
w = gl.GLViewWidget() w = gl.GLViewWidget()
@ -39,8 +40,8 @@ def psi(i, j, k, offset=(50,50,100)):
data = np.fromfunction(psi, (100,100,200)) data = np.fromfunction(psi, (100,100,200))
positive = np.log(np.clip(data, 0, data.max())**2) positive = np.log(clip_array(data, 0, data.max())**2)
negative = np.log(np.clip(-data, 0, -data.min())**2) negative = np.log(clip_array(-data, 0, -data.min())**2)
d2 = np.empty(data.shape + (4,), dtype=np.ubyte) d2 = np.empty(data.shape + (4,), dtype=np.ubyte)
d2[..., 0] = positive * (255./positive.max()) d2[..., 0] = positive * (255./positive.max())