class MyMplCanvas(FigureCanvas):
def onPress(self,event):
if event.button == 1:
self._pressed = True
self._pressEvent = event
self._moveEvent = event
def extraCode(self):
return self._extraCode
def setExtraCode(self,code):
self._extraCode = code
def execExtraCode(self):
if self._extraCode == None:
return
lv = self.__dict__
exec(self._extraCode,lv,lv)
def leaveEvent(self,e):
FigureCanvas.leaveEvent(self,e)
self._pressed = False
self.update()
def mouseMoveEvent(self,e):
try:
FigureCanvas.mouseMoveEvent(self,e)
except:
pass
def paintEvent(self,e):
FigureCanvas.paintEvent(self,e)
if self._pressed:
painter = QPainter(self)
painter.setPen(QPen(Qt.DotLine))
(x1,y1) = self.figure.transFigure.inverted().transform([self._pressEvent.x,self._pressEvent.y])
(x2,y2) = self.figure.transFigure.inverted().transform([self._moveEvent.x,self._moveEvent.y])
painter.drawRect(x1*self.width(),(1-y1)*self.height(),(x2-x1)*self.width(),-(y2-y1)*self.height())
def onMove(self,event):
self._moveLabel.show()
if self._pressed:
self._moveEvent = event
self.update()
if event.xdata == None:
self._moveLabel.hide()
return
self._moveLabel.setText(QString(r"x = {0:g}, y = {1:g}".format(event.xdata, event.ydata)))
self._moveLabel.adjustSize()
offset = 10
if self.width()-event.x < self._moveLabel.width():
offset = -10 - self._moveLabel.width()
self._moveLabel.move(event.x+offset,self.height()-event.y)
def zoomTo(self,rect):
self.axes.set_xlim(rect.left(),rect.right())
self.axes.set_ylim(rect.bottom(),rect.top())
self.axes.set_autoscale_on(False)
self.draw()
def title(self):
return "Figure"
def contextMenuEvent(self, event):
menu = QMenu(self)
autoscaleAction = menu.addAction("Autoscale / Zoom Out")
saveAsAction = menu.addAction("Save figure as...")
fastDisplayAction = menu.addAction("Open as PDF")
propertiesAction = menu.addAction("Properties")
fontSizeMenu = menu.addMenu("Font size")
fontSizes = dict()
for i in range(6,16,2):
fontSizes[i] = fontSizeMenu.addAction("{0:d} px".format(i))
action = menu.exec_(self.mapToGlobal(event.pos()))
if action == saveAsAction:
filename = QFileDialog.getSaveFileName()
if not filename == '':
self._fig.set_size_inches( self._width, self._height )
self._fig.savefig(str(filename))
elif action == fastDisplayAction:
dialog = QInputDialog()
dialog.setWindowTitle("Save and open as *.pdf")
dialog.setLabelText("Filename")
dialog.setTextValue("")
dialog.exec_()
if dialog.result() == QDialog.Accepted:
filename = dialog.textValue()
else:
return
if filename == "":
filename = "no name"
baseName = ""+filename
filename += ".pdf"
cnt = 1
while os.path.exists(filename):
filename = baseName+ "_{0:d}.pdf".format(cnt)
cnt+=1
(w,h) = self._fig.get_size_inches()
try:
services = QDesktopServices()
if not filename == '':
self._fig.set_size_inches( self._width, self._height )
self._fig.savefig(str(filename))
url = QUrl("file:///{0!s}".format(filename))
services.openUrl(url)
finally:
self._fig.set_size_inches( w,h )
self.draw()
elif action == propertiesAction:
try:
reload(sys.modules["pyview.lib.canvas"])
from pyview.lib.canvas import CanvasDialog
except ImportError:
pass
if not self._dialog == None:
self._dialog.hide()
self._dialog = CanvasDialog(self)
self._dialog.setModal(False)
self._dialog.show()
elif action == autoscaleAction:
self.autoScale()
for fontSize in fontSizes.keys():
if action == fontSizes[fontSize]:
print "Setting font size to {0:d}".format(fontSize)
rcParams['font.size'] = str(fontSize)
self.draw()
def autoScale(self):
self.axes.set_autoscale_on(True)
self.axes.relim()
self.axes.autoscale_view()
self.draw()
def onPaint(self,painter):
print "painting..."
def onRelease(self,event):
if event.button == 1:
self._pressed = False
oldRect = QRectF()
oldRect.setLeft(self.axes.get_xlim()[0])
oldRect.setRight(self.axes.get_xlim()[1])
oldRect.setBottom(self.axes.get_ylim()[0])
oldRect.setTop(self.axes.get_ylim()[1])
rect = QRectF()
(x1,y1) = self.axes.transData.inverted().transform([self._pressEvent.x,self._pressEvent.y])
(x2,y2) = self.axes.transData.inverted().transform([event.x,event.y])
rect.setLeft(min(x1,x2))
rect.setRight(max(x1,x2))
rect.setBottom(min(y1,y2))
rect.setTop(max(y1,y2))
if fabs(rect.width()) >= 0.01*fabs(oldRect.width()) and fabs(rect.height()) >=fabs(0.01*oldRect.height()):
self.zoomTo(rect)
"""Ultimately, this is a QWidget (as well as a FigureCanvasAgg, etc.)."""
def __init__(self, parent=None, width=5, height=5, dpi=60):
fig = Figure(figsize=(width, height), dpi=dpi)
self._fig = fig
self._dialog = None
self._width = width
self._height = height
self._dpi = dpi
self._pressed = False
FigureCanvas.__init__(self, fig)
# self.setFixedWidth(self._dpi*self._width)
# self.setFixedHeight(self._dpi*self._height)
self._isDrawing = False
self._extraCode = """#axes.set_title("test")
#axes.set_xlabel("frequency [GHz]")"""
self.axes = fig.add_subplot(111)
self.axes.set_autoscale_on(True)
self.axes.hold(True)
self.setParent(parent)
self._moveLabel = QLabel("",self)
self._moveLabel.setText("tes")
self._moveLabel.hide()
self._moveLabel.setStyleSheet("font-size:14px; margin:5px; padding:4px; background:#FFFFFF; border:2px solid #000;")
self.mpl_connect('button_press_event', self.onPress)
self.mpl_connect('button_release_event', self.onRelease)
self.mpl_connect('motion_notify_event', self.onMove)
FigureCanvas.setSizePolicy(self,
QSizePolicy.Expanding,
QSizePolicy.Expanding)
FigureCanvas.updateGeometry(self)
def contextMenuEvent(self, event):
menu = QMenu(self)
autoscaleAction = menu.addAction("Autoscale / Zoom Out")
saveAsAction = menu.addAction("Save figure as...")
fastDisplayAction = menu.addAction("Open as PDF")
propertiesAction = menu.addAction("Properties")
fontSizeMenu = menu.addMenu("Font size")
fontSizes = dict()
for i in range(6,16,2):
fontSizes[i] = fontSizeMenu.addAction("{0:d} px".format(i))
action = menu.exec_(self.mapToGlobal(event.pos()))
if action == saveAsAction:
filename = QFileDialog.getSaveFileName()
if not filename == '':
self._fig.set_size_inches( self._width, self._height )
self._fig.savefig(str(filename))
elif action == fastDisplayAction:
dialog = QInputDialog()
dialog.setWindowTitle("Save and open as *.pdf")
dialog.setLabelText("Filename")
dialog.setTextValue("")
dialog.exec_()
if dialog.result() == QDialog.Accepted:
filename = dialog.textValue()
else:
return
if filename == "":
filename = "no name"
baseName = ""+filename
filename += ".pdf"
cnt = 1
while os.path.exists(filename):
filename = baseName+ "_{0:d}.pdf".format(cnt)
cnt+=1
(w,h) = self._fig.get_size_inches()
try:
services = QDesktopServices()
if not filename == '':
self._fig.set_size_inches( self._width, self._height )
self._fig.savefig(str(filename))
url = QUrl("file:///{0!s}".format(filename))
services.openUrl(url)
finally:
self._fig.set_size_inches( w,h )
self.draw()
elif action == propertiesAction:
try:
reload(sys.modules["pyview.lib.canvas"])
from pyview.lib.canvas import CanvasDialog
except ImportError:
pass
if not self._dialog == None:
self._dialog.hide()
self._dialog = CanvasDialog(self)
self._dialog.setModal(False)
self._dialog.show()
elif action == autoscaleAction:
self.autoScale()
for fontSize in fontSizes.keys():
if action == fontSizes[fontSize]:
print "Setting font size to {0:d}".format(fontSize)
rcParams['font.size'] = str(fontSize)
self.draw()