Added a few new examples
This commit is contained in:
Luke Campagnola
parent
193367a56b
commit
dc1af8946e
41
examples/BarGraphItem.py
Normal file
41
examples/BarGraphItem.py
Normal file
@ -0,0 +1,41 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
"""
|
||||||
|
Simple example using BarGraphItem
|
||||||
|
"""
|
||||||
|
import initExample ## Add path to library (just for examples; you do not need this)
|
||||||
|
|
||||||
|
import pyqtgraph as pg
|
||||||
|
from pyqtgraph.Qt import QtCore, QtGui
|
||||||
|
import numpy as np
|
||||||
|
|
||||||
|
win = pg.plot()
|
||||||
|
win.setWindowTitle('pyqtgraph example: BarGraphItem')
|
||||||
|
|
||||||
|
x = np.arange(10)
|
||||||
|
y1 = np.sin(x)
|
||||||
|
y2 = 1.1 * np.sin(x+1)
|
||||||
|
y3 = 1.2 * np.sin(x+2)
|
||||||
|
|
||||||
|
bg1 = pg.BarGraphItem(x=x, height=y1, width=0.3, brush='r')
|
||||||
|
bg2 = pg.BarGraphItem(x=x+0.33, height=y2, width=0.3, brush='g')
|
||||||
|
bg3 = pg.BarGraphItem(x=x+0.66, height=y3, width=0.3, brush='b')
|
||||||
|
|
||||||
|
win.addItem(bg1)
|
||||||
|
win.addItem(bg2)
|
||||||
|
win.addItem(bg3)
|
||||||
|
|
||||||
|
|
||||||
|
# Final example shows how to handle mouse clicks:
|
||||||
|
class BarGraph(pg.BarGraphItem):
|
||||||
|
def mouseClickEvent(self, event):
|
||||||
|
print "clicked"
|
||||||
|
|
||||||
|
|
||||||
|
bg = BarGraph(x=x, y=y1*0.3+2, height=0.4+y1*0.2, width=0.8)
|
||||||
|
win.addItem(bg)
|
||||||
|
|
||||||
|
## Start Qt event loop unless running in interactive mode or using pyside.
|
||||||
|
if __name__ == '__main__':
|
||||||
|
import sys
|
||||||
|
if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
|
||||||
|
QtGui.QApplication.instance().exec_()
|
||||||
136
examples/CustomGraphItem.py
Normal file
136
examples/CustomGraphItem.py
Normal file
@ -0,0 +1,136 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
"""
|
||||||
|
Simple example of subclassing GraphItem.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import initExample ## Add path to library (just for examples; you do not need this)
|
||||||
|
|
||||||
|
import pyqtgraph as pg
|
||||||
|
from pyqtgraph.Qt import QtCore, QtGui
|
||||||
|
import numpy as np
|
||||||
|
|
||||||
|
# Enable antialiasing for prettier plots
|
||||||
|
pg.setConfigOptions(antialias=True)
|
||||||
|
|
||||||
|
w = pg.GraphicsWindow()
|
||||||
|
w.setWindowTitle('pyqtgraph example: CustomGraphItem')
|
||||||
|
v = w.addViewBox()
|
||||||
|
v.setAspectLocked()
|
||||||
|
|
||||||
|
class Graph(pg.GraphItem):
|
||||||
|
def __init__(self):
|
||||||
|
self.dragPoint = None
|
||||||
|
self.dragOffset = None
|
||||||
|
self.textItems = []
|
||||||
|
pg.GraphItem.__init__(self)
|
||||||
|
self.scatter.sigClicked.connect(self.clicked)
|
||||||
|
|
||||||
|
def setData(self, **kwds):
|
||||||
|
self.text = kwds.pop('text', [])
|
||||||
|
self.data = kwds
|
||||||
|
if 'pos' in self.data:
|
||||||
|
npts = self.data['pos'].shape[0]
|
||||||
|
self.data['data'] = np.empty(npts, dtype=[('index', int)])
|
||||||
|
self.data['data']['index'] = np.arange(npts)
|
||||||
|
self.setTexts(self.text)
|
||||||
|
self.updateGraph()
|
||||||
|
|
||||||
|
def setTexts(self, text):
|
||||||
|
for i in self.textItems:
|
||||||
|
i.scene().removeItem(i)
|
||||||
|
self.textItems = []
|
||||||
|
for t in text:
|
||||||
|
item = pg.TextItem(t)
|
||||||
|
self.textItems.append(item)
|
||||||
|
item.setParentItem(self)
|
||||||
|
|
||||||
|
def updateGraph(self):
|
||||||
|
pg.GraphItem.setData(self, **self.data)
|
||||||
|
for i,item in enumerate(self.textItems):
|
||||||
|
item.setPos(*self.data['pos'][i])
|
||||||
|
|
||||||
|
|
||||||
|
def mouseDragEvent(self, ev):
|
||||||
|
if ev.button() != QtCore.Qt.LeftButton:
|
||||||
|
ev.ignore()
|
||||||
|
return
|
||||||
|
|
||||||
|
if ev.isStart():
|
||||||
|
# We are already one step into the drag.
|
||||||
|
# Find the point(s) at the mouse cursor when the button was first
|
||||||
|
# pressed:
|
||||||
|
pos = ev.buttonDownPos()
|
||||||
|
pts = self.scatter.pointsAt(pos)
|
||||||
|
if len(pts) == 0:
|
||||||
|
ev.ignore()
|
||||||
|
return
|
||||||
|
self.dragPoint = pts[0]
|
||||||
|
ind = pts[0].data()[0]
|
||||||
|
self.dragOffset = self.data['pos'][ind] - pos
|
||||||
|
elif ev.isFinish():
|
||||||
|
self.dragPoint = None
|
||||||
|
return
|
||||||
|
else:
|
||||||
|
if self.dragPoint is None:
|
||||||
|
ev.ignore()
|
||||||
|
return
|
||||||
|
|
||||||
|
ind = self.dragPoint.data()[0]
|
||||||
|
self.data['pos'][ind] = ev.pos() + self.dragOffset
|
||||||
|
self.updateGraph()
|
||||||
|
ev.accept()
|
||||||
|
|
||||||
|
def clicked(self, pts):
|
||||||
|
print("clicked: %s" % pts)
|
||||||
|
|
||||||
|
|
||||||
|
g = Graph()
|
||||||
|
v.addItem(g)
|
||||||
|
|
||||||
|
## Define positions of nodes
|
||||||
|
pos = np.array([
|
||||||
|
[0,0],
|
||||||
|
[10,0],
|
||||||
|
[0,10],
|
||||||
|
[10,10],
|
||||||
|
[5,5],
|
||||||
|
[15,5]
|
||||||
|
], dtype=float)
|
||||||
|
|
||||||
|
## Define the set of connections in the graph
|
||||||
|
adj = np.array([
|
||||||
|
[0,1],
|
||||||
|
[1,3],
|
||||||
|
[3,2],
|
||||||
|
[2,0],
|
||||||
|
[1,5],
|
||||||
|
[3,5],
|
||||||
|
])
|
||||||
|
|
||||||
|
## Define the symbol to use for each node (this is optional)
|
||||||
|
symbols = ['o','o','o','o','t','+']
|
||||||
|
|
||||||
|
## Define the line style for each connection (this is optional)
|
||||||
|
lines = np.array([
|
||||||
|
(255,0,0,255,1),
|
||||||
|
(255,0,255,255,2),
|
||||||
|
(255,0,255,255,3),
|
||||||
|
(255,255,0,255,2),
|
||||||
|
(255,0,0,255,1),
|
||||||
|
(255,255,255,255,4),
|
||||||
|
], dtype=[('red',np.ubyte),('green',np.ubyte),('blue',np.ubyte),('alpha',np.ubyte),('width',float)])
|
||||||
|
|
||||||
|
## Define text to show next to each symbol
|
||||||
|
texts = ["Point %d" % i for i in range(6)]
|
||||||
|
|
||||||
|
## Update the graph
|
||||||
|
g.setData(pos=pos, adj=adj, pen=lines, size=1, symbol=symbols, pxMode=False, text=texts)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
## Start Qt event loop unless running in interactive mode or using pyside.
|
||||||
|
if __name__ == '__main__':
|
||||||
|
import sys
|
||||||
|
if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
|
||||||
|
QtGui.QApplication.instance().exec_()
|
||||||
50
examples/FillBetweenItem.py
Normal file
50
examples/FillBetweenItem.py
Normal file
@ -0,0 +1,50 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
"""
|
||||||
|
Demonstrates use of FillBetweenItem to fill the space between two plot curves.
|
||||||
|
"""
|
||||||
|
import initExample ## Add path to library (just for examples; you do not need this)
|
||||||
|
|
||||||
|
import pyqtgraph as pg
|
||||||
|
from pyqtgraph.Qt import QtGui, QtCore
|
||||||
|
import numpy as np
|
||||||
|
|
||||||
|
win = pg.plot()
|
||||||
|
win.setWindowTitle('pyqtgraph example: FillBetweenItem')
|
||||||
|
win.setXRange(-10, 10)
|
||||||
|
win.setYRange(-10, 10)
|
||||||
|
|
||||||
|
N = 200
|
||||||
|
x = np.linspace(-10, 10, N)
|
||||||
|
gauss = np.exp(-x**2 / 20.)
|
||||||
|
mn = mx = np.zeros(len(x))
|
||||||
|
curves = [win.plot(x=x, y=np.zeros(len(x)), pen='k') for i in range(4)]
|
||||||
|
brushes = [0.5, (100, 100, 255), 0.5]
|
||||||
|
fills = [pg.FillBetweenItem(curves[i], curves[i+1], brushes[i]) for i in range(3)]
|
||||||
|
for f in fills:
|
||||||
|
win.addItem(f)
|
||||||
|
|
||||||
|
def update():
|
||||||
|
global mx, mn, curves, gauss, x
|
||||||
|
a = 5 / abs(np.random.normal(loc=1, scale=0.2))
|
||||||
|
y1 = -np.abs(a*gauss + np.random.normal(size=len(x)))
|
||||||
|
y2 = np.abs(a*gauss + np.random.normal(size=len(x)))
|
||||||
|
|
||||||
|
s = 0.01
|
||||||
|
mn = np.where(y1<mn, y1, mn) * (1-s) + y1 * s
|
||||||
|
mx = np.where(y2>mx, y2, mx) * (1-s) + y2 * s
|
||||||
|
curves[0].setData(x, mn)
|
||||||
|
curves[1].setData(x, y1)
|
||||||
|
curves[2].setData(x, y2)
|
||||||
|
curves[3].setData(x, mx)
|
||||||
|
|
||||||
|
|
||||||
|
timer = QtCore.QTimer()
|
||||||
|
timer.timeout.connect(update)
|
||||||
|
timer.start(30)
|
||||||
|
|
||||||
|
|
||||||
|
## Start Qt event loop unless running in interactive mode or using pyside.
|
||||||
|
if __name__ == '__main__':
|
||||||
|
import sys
|
||||||
|
if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
|
||||||
|
QtGui.QApplication.instance().exec_()
|
||||||
90
examples/ViewBoxFeatures.py
Normal file
90
examples/ViewBoxFeatures.py
Normal file
@ -0,0 +1,90 @@
|
|||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
"""
|
||||||
|
ViewBox is the general-purpose graphical container that allows the user to
|
||||||
|
zoom / pan to inspect any area of a 2D coordinate system.
|
||||||
|
|
||||||
|
This example demonstrates many of the features ViewBox provides.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import initExample ## Add path to library (just for examples; you do not need this)
|
||||||
|
|
||||||
|
import pyqtgraph as pg
|
||||||
|
from pyqtgraph.Qt import QtCore, QtGui
|
||||||
|
import numpy as np
|
||||||
|
|
||||||
|
x = np.arange(1000, dtype=float)
|
||||||
|
y = np.random.normal(size=1000)
|
||||||
|
y += 5 * np.sin(x/100)
|
||||||
|
|
||||||
|
win = pg.GraphicsWindow()
|
||||||
|
win.setWindowTitle('pyqtgraph example: ____')
|
||||||
|
win.resize(1000, 800)
|
||||||
|
win.ci.setBorder((50, 50, 100))
|
||||||
|
|
||||||
|
sub1 = win.addLayout()
|
||||||
|
sub1.addLabel("<b>Standard mouse interaction:</b><br>left-drag to pan, right-drag to zoom.")
|
||||||
|
sub1.nextRow()
|
||||||
|
v1 = sub1.addViewBox()
|
||||||
|
l1 = pg.PlotDataItem(y)
|
||||||
|
v1.addItem(l1)
|
||||||
|
|
||||||
|
|
||||||
|
sub2 = win.addLayout()
|
||||||
|
sub2.addLabel("<b>One-button mouse interaction:</b><br>left-drag zoom to box, wheel to zoom out.")
|
||||||
|
sub2.nextRow()
|
||||||
|
v2 = sub2.addViewBox()
|
||||||
|
v2.setMouseMode(v2.RectMode)
|
||||||
|
l2 = pg.PlotDataItem(y)
|
||||||
|
v2.addItem(l2)
|
||||||
|
|
||||||
|
win.nextRow()
|
||||||
|
|
||||||
|
sub3 = win.addLayout()
|
||||||
|
sub3.addLabel("<b>Locked aspect ratio when zooming.</b>")
|
||||||
|
sub3.nextRow()
|
||||||
|
v3 = sub3.addViewBox()
|
||||||
|
v3.setAspectLocked(1.0)
|
||||||
|
l3 = pg.PlotDataItem(y)
|
||||||
|
v3.addItem(l3)
|
||||||
|
|
||||||
|
sub4 = win.addLayout()
|
||||||
|
sub4.addLabel("<b>View limits:</b><br>prevent panning or zooming past limits.")
|
||||||
|
sub4.nextRow()
|
||||||
|
v4 = sub4.addViewBox()
|
||||||
|
v4.setLimits(xMin=-100, xMax=1100,
|
||||||
|
minXRange=20, maxXRange=500,
|
||||||
|
yMin=-10, yMax=10,
|
||||||
|
minYRange=1, maxYRange=10)
|
||||||
|
l4 = pg.PlotDataItem(y)
|
||||||
|
v4.addItem(l4)
|
||||||
|
|
||||||
|
win.nextRow()
|
||||||
|
|
||||||
|
sub5 = win.addLayout()
|
||||||
|
sub5.addLabel("<b>Linked axes:</b> Data in this plot is always X-aligned to<br>the plot above.")
|
||||||
|
sub5.nextRow()
|
||||||
|
v5 = sub5.addViewBox()
|
||||||
|
v5.setXLink(v3)
|
||||||
|
l5 = pg.PlotDataItem(y)
|
||||||
|
v5.addItem(l5)
|
||||||
|
|
||||||
|
sub6 = win.addLayout()
|
||||||
|
sub6.addLabel("<b>Disable mouse:</b> Per-axis control over mouse input.<br>"
|
||||||
|
"<b>Auto-scale-visible:</b> Automatically fit *visible* data within view<br>"
|
||||||
|
"(try panning left-right).")
|
||||||
|
sub6.nextRow()
|
||||||
|
v6 = sub6.addViewBox()
|
||||||
|
v6.setMouseEnabled(x=True, y=False)
|
||||||
|
v6.enableAutoRange(x=False, y=True)
|
||||||
|
v6.setXRange(300, 450)
|
||||||
|
v6.setAutoVisible(x=False, y=True)
|
||||||
|
l6 = pg.PlotDataItem(y)
|
||||||
|
v6.addItem(l6)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
## Start Qt event loop unless running in interactive mode or using pyside.
|
||||||
|
if __name__ == '__main__':
|
||||||
|
import sys
|
||||||
|
if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
|
||||||
|
QtGui.QApplication.instance().exec_()
|
||||||
Loading…
Reference in New Issue
Block a user