3af23725ca
* Added hovering demo to ScatterPlot example * Use Qt's serialization for SymbolAtlas.symbolMap keys Yields significant performance improvements when updating the scatter plot's options. See e.g. the plot hover example. * Further optimized scatter plot picking * Fix ScatterPlot example tool tip * Clean up while I'm here * Compatibility * Some simple optimizations for ScatterPlotItem Speedups for ScatterPlotSpeedTest.py: ~50% without pxMode ~ 0% pxMode with useCache ~30% pxMode without useCache * ~3x speedup in scatter plot speed test with pxMode * More optimization low-hanging fruit for the scatter plot * Removed hover example to lazily pass tests * Avoid segfault * Re-add hover example to ScatterPlot.py * Switch to id-based keying for scatter plot symbol atlas - Use cases exist where serialization-based keying is a significant bottleneck, e.g. updating without atlas invalidation when a large variety pens or brushes are present. - To avoid a performance hit, the onus is on the user to carefully reuse pen and brush objects. * Optimized caching in scatter plot hovering example * Fixed and optimized scatter plot hovering example * Minor scatter plot optimization * Cleanup * Store hovered points in a set for the hovering example * Keep a limited number symbol atlas entries around for future reuse * Added a docstring note to remind the user to reuse QPen and QBrush objects for better performance * Tidied up hovering example * Typo * Avoid unnecessary atlas rebuilds * Refactored SymbolAtlas * Efficient appending to SymbolAtlas * SymbolAtlas rewrite * Cleanup and profiling * Add randomized brushes to speed test * Add loc indexer to ScatterPlotItem * Profile ScatterPlotItem.paint to identify bottlenecks * Reuse targetRect to improve paint performance * Readability improvements (opinionated) * Only need to set x and y of targetRect - w and h can stay set to 0 (not entirely sure why) - this is a bit faster than setting all of x, y, w, h * Minor renaming * Strip off API changes and leave to another PR * Renaming * Compatibility * Use drawPixmap(x, y, pm, sx, sy, sw, sh) signature to avoid needing to update QRectFs * Use different drawing approaches for each Qt binding for performance reasons * Fix a bug introduced two commits ago Incidentally, I think there is a similar bug in the main branch currently. * Minor performance and readability improvements * Strip out source and target QRectF stuff * Bring source and target QRectF stuff back in a less coupled way * Leave deprecating getSpotOpts for another PR * Compatibility fix * Added docstrings and use SymbolAtlas__len__ where possible * Fix export issue * Add missing import * Add deprecation warnings * Avoid using deprecated methods * Fix and cleanup max spot size measurements * Make creation of style opts entries explicit * Add hovering API to ScatterPlotItem * Compatibility * Marshal pen and brush lists in setPen and setBrush * Fixed platform dependent bug
86 lines
2.3 KiB
Python
86 lines
2.3 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
For testing rapid updates of ScatterPlotItem under various conditions.
|
|
|
|
(Scatter plots are still rather slow to draw; expect about 20fps)
|
|
"""
|
|
|
|
|
|
|
|
## Add path to library (just for examples; you do not need this)
|
|
import initExample
|
|
|
|
|
|
from pyqtgraph.Qt import QtGui, QtCore, QT_LIB
|
|
import numpy as np
|
|
import pyqtgraph as pg
|
|
from pyqtgraph.ptime import time
|
|
#QtGui.QApplication.setGraphicsSystem('raster')
|
|
app = QtGui.QApplication([])
|
|
#mw = QtGui.QMainWindow()
|
|
#mw.resize(800,800)
|
|
if QT_LIB == 'PySide':
|
|
from ScatterPlotSpeedTestTemplate_pyside import Ui_Form
|
|
elif QT_LIB == 'PySide2':
|
|
from ScatterPlotSpeedTestTemplate_pyside2 import Ui_Form
|
|
elif QT_LIB == 'PyQt5':
|
|
from ScatterPlotSpeedTestTemplate_pyqt5 import Ui_Form
|
|
else:
|
|
from ScatterPlotSpeedTestTemplate_pyqt import Ui_Form
|
|
|
|
win = QtGui.QWidget()
|
|
win.setWindowTitle('pyqtgraph example: ScatterPlotSpeedTest')
|
|
ui = Ui_Form()
|
|
ui.setupUi(win)
|
|
win.show()
|
|
|
|
p = ui.plot
|
|
p.setRange(xRange=[-500, 500], yRange=[-500, 500])
|
|
|
|
count = 500
|
|
data = np.random.normal(size=(50,count), scale=100)
|
|
sizeArray = (np.random.random(count) * 20.).astype(int)
|
|
brushArray = [pg.mkBrush(x) for x in np.random.randint(0, 256, (count, 3))]
|
|
ptr = 0
|
|
lastTime = time()
|
|
fps = None
|
|
|
|
|
|
def update():
|
|
global curve, data, ptr, p, lastTime, fps
|
|
p.clear()
|
|
if ui.randCheck.isChecked():
|
|
size = sizeArray
|
|
brush = brushArray
|
|
else:
|
|
size = ui.sizeSpin.value()
|
|
brush = 'b'
|
|
curve = pg.ScatterPlotItem(x=data[ptr % 50], y=data[(ptr+1) % 50],
|
|
pen='w', brush=brush, size=size,
|
|
pxMode=ui.pixelModeCheck.isChecked())
|
|
p.addItem(curve)
|
|
ptr += 1
|
|
now = time()
|
|
dt = now - lastTime
|
|
lastTime = now
|
|
if fps is None:
|
|
fps = 1.0/dt
|
|
else:
|
|
s = np.clip(dt*3., 0, 1)
|
|
fps = fps * (1-s) + (1.0/dt) * s
|
|
p.setTitle('%0.2f fps' % fps)
|
|
p.repaint()
|
|
#app.processEvents() ## force complete redraw for every plot
|
|
timer = QtCore.QTimer()
|
|
timer.timeout.connect(update)
|
|
timer.start(0)
|
|
|
|
|
|
|
|
## Start Qt event loop unless running in interactive mode.
|
|
if __name__ == '__main__':
|
|
import sys
|
|
if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
|
|
QtGui.QApplication.instance().exec_()
|