示例#1
0
文件: canvas.py 项目: adewes/pyview
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)
示例#2
0
文件: canvas.py 项目: adewes/pyview
    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()