def paintEvent(self, event):
super(InteractiveGRWidget, self).paintEvent(event)
self._painter.begin(self)
if self.getMouseSelectionEnabled() and self._tselect:
p0, p1 = self._tselect
coords = DeviceCoordConverter(self.dwidth, self.dheight)
coords.setNDC(p0.x, p0.y)
p0dc = coords.getDC()
coords.setNDC(p1.x, p1.y)
p1dc = coords.getDC()
if self._getPlotsForPoint(p0):
rect = QtCore.QRect(QtCore.QPoint(p0dc.x, p0dc.y),
QtCore.QPoint(p1dc.x, p1dc.y)).normalized()
self._painter.setOpacity(.75)
self._painter.drawRect(rect)
self._painter.setOpacity(1.)
self._painter.end()
def __init__(self, *args, **kwargs):
QtGui.QMainWindow.__init__(self, *args, **kwargs)
self._grw = InteractiveGRWidget()
self.setCentralWidget(self._grw)
viewport = [0.1, 0.88, 0.1, 0.88]
self.curve = DependentPlotCurve([0], [0], linecolor=4)
self.peakbars = PeakBars([], [], linecolor=2)
self.curve.dependent = [self.peakbars]
axes = ClampedPlotAxes(viewport, 50, 5, 1, 2)
axes.setWindow(50, 25000, 0, 100)
axes.addCurves(self.curve)
plot = Plot(viewport).addAxes(axes)
plot.title = "QtGR Power Spectrum"
plot.subTitle = "Capturing Microphone"
axes.scale = gr.OPTION_X_LOG
self._grw.addPlot(plot)
timer = QtCore.QTimer(self)
timer.timeout.connect(self.updateData)
timer.start(40)
def main(*args):
app = QtGui.QApplication(*args)
grw = InteractiveGRWidget()
viewport = [0.1, 0.88, 0.1, 0.88]
xd = [3, 3, 10, 18, 18, 10, 10, 5, 1, 15, 20, 5, 15, 10, 7, 13, 16]
yd = [3, 18, 18, 3, 18, 10, 1, 5, 10, 5, 10, 15, 15, 15, 20, 20, 8]
zd = [25, 25, 25, 25, 25, -5, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 25]
axes = ContourAxes(viewport)
axes.addCurves(PlotContour(xd, yd, zd, nx=40, ny=40))
axes.setGrid(False)
plot = Plot(viewport).addAxes(axes)
plot.title = "Plot contour lines using QtGR"
grw.addPlot(plot)
grw.resize(QtCore.QSize(500, 500))
grw.show()
win = axes.getWindow()
grw.update()
sys.exit(app.exec_())
def __init__(self, *args, **kwargs):
QtGui.QMainWindow.__init__(self, *args, **kwargs)
self._grw = InteractiveGRWidget()
self.setCentralWidget(self._grw)
viewport = [0.1, 0.88, 0.1, 0.88]
x = arange(0, 2 * pi, 0.01)
y = sin(x)
self.curve = PlotCurve(x, y)
axes = ScaledPlotAxes(viewport)
axes.addCurves(self.curve)
plot = Plot(viewport).addAxes(axes)
plot.title = "QtGR Timer example"
plot.subTitle = "Plotting Live-Data"
plot.xlabel = "t"
plot.ylabel = "sin(t)"
plot.autoscale = PlotAxes.SCALE_X | PlotAxes.SCALE_Y
self._grw.addPlot(plot)
timer = QtCore.QTimer(self)
timer.timeout.connect(self.updateData)
timer.start(40)
def main(*args):
app = QtWidgets.QApplication(*args)
grw = InteractiveGRWidget()
viewport = [0.1, 0.88, 0.1, 0.88]
z = readfile(
os.path.join(os.path.dirname(os.path.realpath(__file__)), "fecr.dat"))
zd = np.asarray(z)
xd = yd = np.arange(1, 201)
axes = ContourAxes(viewport)
axes.addCurves(PlotSurface(xd, yd, zd, option=gr.OPTION_CELL_ARRAY))
axes.addCurves(PlotContour(xd, yd, zd))
axes.setGrid(False)
plot = Plot(viewport).addAxes(axes)
plot.title = "Plot surface and contour lines using QtGR"
grw.addPlot(plot)
grw.resize(QtCore.QSize(500, 500))
grw.show()
win = axes.getWindow()
grw.update()
sys.exit(app.exec_())
def main(*args):
app = QtGui.QApplication(sys.argv)
mainWin = MainWindow()
mainWin.resize(QtCore.QSize(500, 500))
mainWin.show()
sys.exit(app.exec_())
class InteractiveGRWidget(GRWidget):
logXinDomain = QtCore.Signal(bool)
logYinDomain = QtCore.Signal(bool)
modePick = QtCore.Signal(bool)
GESTURE_RECOGNIZERS = [PanGestureRecognizer, SelectGestureRecognizer]
def __init__(self, *args, **kwargs):
super(InteractiveGRWidget, self).__init__(*args, **kwargs)
self._recognizers = []
# register gesture recognizers
for recognizer in self.GESTURE_RECOGNIZERS:
instance = recognizer()
self.grabGesture(recognizer.registerRecognizer(instance))
# keep a reference on existing QGestureRecognizers in order
# to circumvent that they will be accidentially freed when
# using PyQt4 versions which will not handle the membership
# correctly, e.g. PyQt4 4.9.1 shipped with Ubuntu 12.04.
self._recognizers.append(instance)
self._eventFilter = installEventFilter(self)
self._eventFilterEnabled = True
self.cbm.addHandler(MouseEvent.MOUSE_MOVE, self.mouseMove)
self.cbm.addHandler(MouseEvent.MOUSE_PRESS, self.mousePress)
self.cbm.addHandler(MouseEvent.MOUSE_RELEASE, self.mouseRelease)
self.cbm.addHandler(WheelEvent.WHEEL_MOVE, self.wheelMove)
self.cbm.addHandler(PickEvent.PICK_MOVE, self.pickMove)
self.cbm.addHandler(MouseGestureEvent.MOUSE_PAN, self._mousePan)
self.cbm.addHandler(MouseGestureEvent.MOUSE_SELECT, self._mouseSelect)
self.setMouseTracking(True)
self._tselect = None # select point tuple
self._logXinDomain = None
self._logYinDomain = None
self._pickMode = False
self._pickEvent = None
self._selectEnabled, self._panEnabled = True, True
self._roiEnabled = True
self._zoomEnabled = True
self._lstPlot = []
def draw(self, clear=None, update=None):
# obsolete kwargs clear, update (unused) just kept for compatibility
GRWidget.draw(self, clear, update)
gr.setwsviewport(0, self.mwidth, 0, self.mheight)
gr.setwswindow(0, self.sizex, 0, self.sizey)
for plot in self._lstPlot:
plot.sizex, plot.sizey = self.sizex, self.sizey
plot.drawGR()
# logDomainCheck
logXinDomain = plot.logXinDomain()
logYinDomain = plot.logYinDomain()
if logXinDomain != self._logXinDomain:
self._logXinDomain = logXinDomain
self.logXinDomain.emit(self._logXinDomain)
if logYinDomain != self._logYinDomain:
self._logYinDomain = logYinDomain
self.logYinDomain.emit(self._logYinDomain)
if self._pickEvent:
event = self._pickEvent
gr.setviewport(*event.viewportscaled)
wcPoint = event.getWC(event.viewport)
window = gr.inqwindow()
gr.setwindow(*event.getWindow())
gr.setmarkertype(gr.MARKERTYPE_PLUS)
gr.polymarker([wcPoint.x], [wcPoint.y])
gr.setwindow(*window)
def addPlot(self, *args, **kwargs):
for plot in args:
if plot and plot not in self._lstPlot:
self._lstPlot.append(plot)
self.update()
return self._lstPlot
@property
def cbm(self):
"""Get CallbackManager used by EventFilter"""
return self._eventFilter.manager
def setEventFilterEnabled(self, flag):
"""Dis-/Enable QtGR events"""
if self._eventFilterEnabled:
if not flag:
self.removeEventFilter(self._eventFilter)
elif flag:
self.installEventFilter(self._eventFilter)
self._eventFilterEnabled = flag
def plot(self, *args, **kwargs):
plot = Plot()
axes = PlotAxes(plot.viewport)
axes.plot(*args, **kwargs)
plot.addAxes(axes)
return self.addPlot(plot)
def adjustSelectRect(self, p0, p1):
# can be used to restrict select rectangle, e.g. to given aspect ratio
return p0, p1
def paintEvent(self, event):
super(InteractiveGRWidget, self).paintEvent(event)
self._painter.begin(self)
if self.getMouseSelectionEnabled() and self._tselect:
p0, p1 = self._tselect
coords = DeviceCoordConverter(self.dwidth, self.dheight)
coords.setNDC(p0.x, p0.y)
p0dc = coords.getDC()
coords.setNDC(p1.x, p1.y)
p1dc = coords.getDC()
if self._getPlotsForPoint(p0):
rect = QtCore.QRect(QtCore.QPoint(p0dc.x, p0dc.y),
QtCore.QPoint(p1dc.x, p1dc.y)).normalized()
self._painter.setOpacity(.75)
self._painter.drawRect(rect)
self._painter.setOpacity(1.)
self._painter.end()
def setAutoScale(self, mask):
for plot in self._lstPlot:
plot.autoscale = mask
def getPickMode(self):
return self._pickMode
def setPickMode(self, bool):
self._pickMode = bool
self.modePick.emit(self._pickMode)
def getMouseSelectionEnabled(self):
return self._selectEnabled
def setMouseSelectionEnabled(self, flag):
self._selectEnabled = flag
def getMousePanEnabled(self):
return self._panEnabled
def setMousePanEnabled(self, flag):
self._panEnabled = flag
def getMouseZoomEnabled(self):
return self._zoomEnabled
def setMouseZoomEnabled(self, flag):
self._zoomEnabled = flag
def _getPlotsForPoint(self, p0):
res = []
for plot in self._lstPlot:
xmin, xmax, ymin, ymax = plot.viewportscaled
if p0.x >= xmin and p0.x <= xmax and p0.y >= ymin and p0.y <= ymax:
res.append(plot)
return res
def _pick(self, p0, type):
for plot in self._getPlotsForPoint(p0):
(coord, axes, _curve) = plot.pick(p0, self.dwidth, self.dheight)
if coord:
dcPoint = coord.getDC()
QApplication.sendEvent(self, PickEvent(type,
self.dwidth,
self.dheight,
dcPoint.x,
dcPoint.y,
plot.viewport,
coord.getWindow(),
sizex=axes.sizex,
sizey=axes.sizey))
def _select(self, p0, p1):
self._pickEvent = None
change = False
for plot in self._getPlotsForPoint(p0):
plot.select(p0, p1, self.dwidth, self.dheight)
change = True
if change:
self.update()
def mouseSelect(self, event):
p0 = event.getNDC()
p0, p1 = self.adjustSelectRect(p0, p0 + event.getOffset())
self._tselect = (p0, p1)
if event.isFinished():
self._tselect = None
self._select(p0, p1)
self.update()
def _mouseSelect(self, event):
if self.getMouseSelectionEnabled():
self.mouseSelect(event)
def _pan(self, p0, dp):
self._pickEvent = None
change = False
for plot in self._getPlotsForPoint(p0):
plot.pan(dp, self.dwidth, self.dheight)
change = True
if change:
self.update()
def mousePan(self, event):
self._pan(event.getNDC(), event.getOffset())
def _mousePan(self, event):
if self.getMousePanEnabled():
self.mousePan(event)
# disable roi recognition during pannning
self._roiEnabled = event.isFinished()
def _zoom(self, dpercent, p0):
self._pickEvent = None
change = False
for plot in self._getPlotsForPoint(p0):
plot.zoom(dpercent, p0, self.dwidth, self.dheight)
change = True
if change:
self.update()
def _roi(self, p0, type, buttons, modifiers):
if self._roiEnabled:
for plot in self._lstPlot:
roi = plot.getROI(p0)
if roi:
if roi.regionType == RegionOfInterest.LEGEND:
eventObj = LegendEvent
else:
eventObj = ROIEvent
coords = DeviceCoordConverter(self.dwidth, self.dheight)
coords.setNDC(p0.x, p0.y)
p0dc = coords.getDC()
QApplication.sendEvent(self,
eventObj(type,
self.dwidth,
self.dheight,
p0dc.x, p0dc.y,
buttons, modifiers,
roi))
def mousePress(self, event):
if event.getButtons() & MouseEvent.LEFT_BUTTON:
if self.getPickMode():
self.setPickMode(False)
self._pick(event.getNDC(), PickEvent.PICK_PRESS)
else:
if event.getModifiers() & MouseEvent.CONTROL_MODIFIER:
self.setPickMode(True)
def mouseRelease(self, event):
self._roi(event.getNDC(), ROIEvent.ROI_CLICKED, event.getButtons(),
event.getModifiers())
def mouseMove(self, event):
if self.getPickMode():
self._pick(event.getNDC(), PickEvent.PICK_MOVE)
self._roi(event.getNDC(), ROIEvent.ROI_OVER, event.getButtons(),
event.getModifiers())
def wheelMove(self, event):
if self.getMouseZoomEnabled():
# delta percent
dpercent = event.getSteps() * .1
self._zoom(dpercent, event.getNDC())
def pickMove(self, event):
self._pickEvent = event
self.update()
self._zoom(dpercent, event.getNDC())
def pickMove(self, event):
self._pickEvent = event
self.update()
if __name__ == "__main__":
import sys
from gr import pygr
logging.basicConfig(level=logging.CRITICAL)
for name in [__name__, pygr.base.__name__, pygr.__name__]:
logging.getLogger(name).setLevel(logging.DEBUG)
app = QApplication(sys.argv)
grw = InteractiveGRWidget()
grw.resize(QtCore.QSize(500, 500))
viewport = [0.1, 0.45, 0.1, 0.88]
vp2 = [.6, .95, .1, .88]
x = [-3.3 + t * .1 for t in range(66)]
y = [t ** 5 - 13 * t ** 3 + 36 * t for t in x]
n = 100
pi2_n = 2.*math.pi / n
x2 = [i * pi2_n for i in range(0, n + 1)]
y2 = map(lambda xi: math.sin(xi), x2)
plot = Plot(viewport).addAxes(PlotAxes(viewport).plot(x, y),
PlotAxes(viewport).plot(x2, y2))
plot.title, plot.subTitle = "foo", "bar"
plot.xlabel, plot.ylabel = "x", "f(x)"
class InteractiveGRWidget(GRWidget):
logXinDomain = QtCore.Signal(bool)
logYinDomain = QtCore.Signal(bool)
modePick = QtCore.Signal(bool)
def __init__(self, *args, **kwargs):
super(InteractiveGRWidget, self).__init__(*args, **kwargs)
guiConn = GUIConnector(self)
guiConn.connect(qtgr.events.MouseEvent.MOUSE_MOVE, self.mouseMove)
guiConn.connect(qtgr.events.MouseEvent.MOUSE_PRESS, self.mousePress)
guiConn.connect(qtgr.events.MouseEvent.MOUSE_RELEASE,
self.mouseRelease)
guiConn.connect(qtgr.events.WheelEvent.WHEEL_MOVE, self.wheelMove)
guiConn.connect(qtgr.events.PickEvent.PICK_MOVE, self.pickMove)
self.setMouseTracking(True)
self._mouseLeft = False
self._mouseRight = False
self._startPoint = None
self._curPoint = None
self._logXinDomain = None
self._logYinDomain = None
self._pickMode = False
self._pickEvent = None
self._selectEnabled, self._panEnabled = True, True
self._zoomEnabled = True
self._lstPlot = []
def draw(self, clear=False, update=True):
if clear:
gr.clearws()
gr.setwsviewport(0, self.mwidth, 0, self.mheight)
gr.setwswindow(0, self.sizex, 0, self.sizey)
for plot in self._lstPlot:
plot.sizex, plot.sizey = self.sizex, self.sizey
plot.drawGR()
# logDomainCheck
logXinDomain = plot.logXinDomain()
logYinDomain = plot.logYinDomain()
if logXinDomain != self._logXinDomain:
self._logXinDomain = logXinDomain
self.logXinDomain.emit(self._logXinDomain)
if logYinDomain != self._logYinDomain:
self._logYinDomain = logYinDomain
self.logYinDomain.emit(self._logYinDomain)
if self._pickEvent:
event = self._pickEvent
gr.setviewport(*event.viewport)
wcPoint = event.getWC(event.viewport)
window = gr.inqwindow()
gr.setwindow(*event.getWindow())
gr.setmarkertype(gr.MARKERTYPE_PLUS)
gr.polymarker([wcPoint.x], [wcPoint.y])
gr.setwindow(*window)
def addPlot(self, *args, **kwargs):
for plot in args:
if plot and plot not in self._lstPlot:
self._lstPlot.append(plot)
self._draw(clear=True, update=True)
return self._lstPlot
def plot(self, *args, **kwargs):
plot = Plot()
axes = PlotAxes(plot.viewport)
axes.plot(*args, **kwargs)
plot.addAxes(axes)
return self.addPlot(plot)
def adjustSelectRect(self, p0, p1):
# can be used to restrict select rectangle, e.g. to given aspect ratio
return p0, p1
def paintEvent(self, event):
super(InteractiveGRWidget, self).paintEvent(event)
self._painter.begin(self)
if self._mouseLeft and self.getMouseSelectionEnabled():
p0 = self._startPoint.getNDC()
p1 = self._curPoint.getNDC()
p0, p1 = self.adjustSelectRect(p0, p1)
coords = DeviceCoordConverter(self.dwidth, self.dheight)
coords.setNDC(p0.x, p0.y)
p0dc = coords.getDC()
coords.setNDC(p1.x, p1.y)
p1dc = coords.getDC()
if self._getPlotsForPoint(p0):
rect = QtCore.QRect(QtCore.QPoint(p0dc.x, p0dc.y),
QtCore.QPoint(p1dc.x,
p1dc.y)).normalized()
self._painter.setOpacity(.75)
self._painter.drawRect(rect)
self._painter.setOpacity(1.)
self._painter.end()
def setAutoScale(self, mask):
for plot in self._lstPlot:
plot.autoscale = mask
def getPickMode(self):
return self._pickMode
def setPickMode(self, bool):
self._pickMode = bool
self.modePick.emit(self._pickMode)
def getMouseSelectionEnabled(self):
return self._selectEnabled
def setMouseSelectionEnabled(self, flag):
self._selectEnabled = flag
def getMousePanEnabled(self):
return self._panEnabled
def setMousePanEnabled(self, flag):
self._panEnabled = flag
def getMouseZoomEnabled(self):
return self._zoomEnabled
def setMouseZoomEnabled(self, flag):
self._zoomEnabled = flag
def _getPlotsForPoint(self, p0):
res = []
for plot in self._lstPlot:
xmin, xmax, ymin, ymax = plot.viewportscaled
if p0.x >= xmin and p0.x <= xmax and p0.y >= ymin and p0.y <= ymax:
res.append(plot)
return res
def _pick(self, p0, type):
for plot in self._getPlotsForPoint(p0):
(coord, _axes, _curve) = plot.pick(p0, self.dwidth, self.dheight)
if coord:
dcPoint = coord.getDC()
QtGui.QApplication.sendEvent(
self,
PickEvent(type, self.dwidth, self.dheight, dcPoint.x,
dcPoint.y, plot.viewport, coord.getWindow()))
def _select(self, p0, p1):
self._pickEvent = None
change = False
for plot in self._getPlotsForPoint(p0):
plot.select(p0, p1, self.dwidth, self.dheight)
change = True
if change:
self._draw(True)
self.update()
def _pan(self, p0, dp):
self._pickEvent = None
change = False
for plot in self._getPlotsForPoint(p0):
plot.pan(dp, self.dwidth, self.dheight)
change = True
if change:
self._draw(True)
self.update()
def _zoom(self, dpercent, p0):
self._pickEvent = None
change = False
for plot in self._getPlotsForPoint(p0):
plot.zoom(dpercent, p0, self.dwidth, self.dheight)
change = True
if change:
self._draw(True)
self.update()
def _roi(self, p0, type, buttons, modifiers):
for plot in self._lstPlot:
roi = plot.getROI(p0)
if roi:
if roi.regionType == RegionOfInterest.LEGEND:
eventObj = LegendEvent
else:
eventObj = ROIEvent
coords = DeviceCoordConverter(self.dwidth, self.dheight)
coords.setNDC(p0.x, p0.y)
p0dc = coords.getDC()
QtGui.QApplication.sendEvent(
self,
eventObj(type, self.dwidth, self.dheight, p0dc.x, p0dc.y,
buttons, modifiers, roi))
def mousePress(self, event):
if event.getButtons() & MouseEvent.LEFT_BUTTON:
if self.getPickMode():
self.setPickMode(False)
self._pick(event.getNDC(), PickEvent.PICK_PRESS)
else:
if event.getModifiers() & MouseEvent.CONTROL_MODIFIER:
self.setPickMode(True)
else:
self._mouseLeft = True
elif event.getButtons() & MouseEvent.RIGHT_BUTTON:
self._mouseRight = True
self._curPoint = event
self._startPoint = event
def mouseRelease(self, event):
if event.getButtons() & MouseEvent.LEFT_BUTTON and self._mouseLeft:
self._mouseLeft = False
self._curPoint = event
p0 = self._startPoint.getNDC()
p1 = self._curPoint.getNDC()
if p0 != p1:
if (self.getMouseSelectionEnabled()
and self._getPlotsForPoint(p0)):
p0, p1 = self.adjustSelectRect(p0, p1)
self._select(p0, p1)
else:
self._roi(p0, ROIEvent.ROI_CLICKED, event.getButtons(),
event.getModifiers())
elif event.getButtons() & MouseEvent.RIGHT_BUTTON and self._mouseRight:
self._mouseRight = False
self._roi(event.getNDC(), ROIEvent.ROI_CLICKED, event.getButtons(),
event.getModifiers())
self._curPoint = event
def mouseMove(self, event):
if self.getPickMode():
self._pick(event.getNDC(), PickEvent.PICK_MOVE)
if event.getButtons() & MouseEvent.LEFT_BUTTON:
self._curPoint = event
super(InteractiveGRWidget, self).update()
elif event.getButtons() & MouseEvent.RIGHT_BUTTON:
p0 = self._curPoint.getNDC() # point before now
p1 = event.getNDC()
dp = p1 - p0
self._curPoint = event
if self.getMousePanEnabled():
self._pan(self._startPoint.getNDC(), dp)
self._mouseRight = False
self._roi(event.getNDC(), ROIEvent.ROI_OVER, event.getButtons(),
event.getModifiers())
def wheelMove(self, event):
if self.getMouseZoomEnabled():
# delta percent
dpercent = event.getSteps() * .1
self._zoom(dpercent, event.getNDC())
def pickMove(self, event):
self._pickEvent = event
self.update()
