pyqtgraph/examples/test_PlotWidget.py

74 lines
1.8 KiB
Python
Raw Normal View History

2010-03-22 05:48:52 +00:00
#!/usr/bin/python
# -*- coding: utf-8 -*-
## Add path to library (just for examples; you do not need this)
import sys, os
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..'))
2010-03-22 05:48:52 +00:00
from PyQt4 import QtGui, QtCore
import numpy as np
import pyqtgraph as pg
2010-03-22 05:48:52 +00:00
#QtGui.QApplication.setGraphicsSystem('raster')
2010-03-22 05:48:52 +00:00
app = QtGui.QApplication([])
mw = QtGui.QMainWindow()
mw.resize(800,800)
2010-03-22 05:48:52 +00:00
cw = QtGui.QWidget()
mw.setCentralWidget(cw)
l = QtGui.QVBoxLayout()
cw.setLayout(l)
pw = pg.PlotWidget(name='Plot1') ## giving the plots names allows us to link their axes together
2010-03-22 05:48:52 +00:00
l.addWidget(pw)
pw2 = pg.PlotWidget(name='Plot2')
2010-03-22 05:48:52 +00:00
l.addWidget(pw2)
pw3 = pg.PlotWidget()
2010-03-22 05:48:52 +00:00
l.addWidget(pw3)
mw.show()
2010-03-22 05:48:52 +00:00
## Create an empty plot curve to be filled later, set its pen
2010-03-22 05:48:52 +00:00
p1 = pw.plot()
p1.setPen((200,200,100))
## Add in some extra graphics
2010-03-22 05:48:52 +00:00
rect = QtGui.QGraphicsRectItem(QtCore.QRectF(0, 0, 1, 1))
rect.setPen(QtGui.QPen(QtGui.QColor(100, 200, 100)))
pw.addItem(rect)
def rand(n):
data = np.random.random(n)
2010-03-22 05:48:52 +00:00
data[int(n*0.1):int(n*0.13)] += .5
data[int(n*0.18)] += 2
data[int(n*0.1):int(n*0.13)] *= 5
data[int(n*0.18)] *= 20
data *= 1e-12
return data, np.arange(n, n+len(data)) / float(n)
2010-03-22 05:48:52 +00:00
def updateData():
yd, xd = rand(10000)
p1.updateData(yd, x=xd)
## Start a timer to rapidly update the plot in pw
2010-03-22 05:48:52 +00:00
t = QtCore.QTimer()
t.timeout.connect(updateData)
2010-03-22 05:48:52 +00:00
t.start(50)
## Multiple parameterized plots--we can autogenerate averages for these.
2010-03-22 05:48:52 +00:00
for i in range(0, 5):
for j in range(0, 3):
yd, xd = rand(10000)
pw2.plot(y=yd*(j+1), x=xd, params={'iter': i, 'val': j})
## Test large numbers
curve = pw3.plot(np.random.normal(size=100)*1e6)
curve.setPen('w') ## white pen
curve.setShadowPen(pg.mkPen((70,70,30), width=6, cosmetic=True))
## Start Qt event loop unless running in interactive mode.
if sys.flags.interactive != 1:
app.exec_()