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

init so that pointers are present

Browse Source
This commit is contained in:
KIU Shueng Chuan 2021-06-15 06:37:22 +08:00
parent a1845cddbc
commit abeae0b7fa
1 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
pyqtgraph/graphicsItems/ScatterPlotItem.py
View File

@ -173,19 +173,21 @@ def _mkBrush(*args, **kwargs):
class PixmapFragments: class PixmapFragments:
def __init__(self): def __init__(self):
self._array = np.empty((0, 10), dtype=np.float64) self.alloc(0)
self.ptrs = None
def alloc(self, size):
self.arr = np.empty((size, 10), dtype=np.float64)
if QT_LIB.startswith('PyQt'):
self.ptrs = list(map(sip.wrapinstance,
itertools.count(self.arr.ctypes.data, self.arr.strides[0]),
itertools.repeat(QtGui.QPainter.PixmapFragment, self.arr.shape[0])))
else:
self.ptrs = wrapInstance(self.arr.ctypes.data, QtGui.QPainter.PixmapFragment)
def array(self, size): def array(self, size):
if size > self._array.shape[0]: if size > self.arr.shape[0]:
self._array = np.empty((size + 16, 10), dtype=np.float64) self.alloc(size + 16)
if QT_LIB.startswith('PyQt'): return self.arr[:size]
self.ptrs = list(map(sip.wrapinstance,
itertools.count(self._array.ctypes.data, self._array.strides[0]),
itertools.repeat(QtGui.QPainter.PixmapFragment, self._array.shape[0])))
else:
self.ptrs = wrapInstance(self._array.ctypes.data, QtGui.QPainter.PixmapFragment)
return self._array[:size]
def draw(self, painter, size, pixmap): def draw(self, painter, size, pixmap):
if QT_LIB.startswith('PyQt'): if QT_LIB.startswith('PyQt'):