Mercurial > pygpibtoolkit / changeset

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/hpgl_qt.py	Fri Jan 11 18:50:08 2008 +0100
@@ -0,0 +1,129 @@
+# -*- coding: utf-8 -*-
+import numpy
+import pylab
+import math
+import new # monkey patching...
+
+from PyQt4 import QtGui, QtCore
+from PyQt4.QtCore import Qt, SIGNAL
+
+from hpgl_parser import HPGLParser
+
+class QHPGLPlotterWidget(HPGLParser, QtGui.QWidget):
+    pen_styles = {2: [0.4, 0.6],
+                  3: [0.6, 0.4],
+                  4: [0.7, 0.1, 0.1, 0.1],
+                  5: [0.6, 0.1, 0.2, 0.1],
+                  6: [0.45, 0.1, 0.1, 0.2, 0.1, 0.05],
+                  }
+    pen_colors = ["black", "green", "red", "blue", "yellow", "cyan", "magenta",
+                  "darkred", "darkblue"]
+    
+    def __init__(self, parent=None):
+        QtGui.QWidget.__init__(self, parent)
+
+        self.qpen = QtGui.QPen(QtCore.Qt.blue)
+        self.qbrush = QtGui.QBrush(QtCore.Qt.blue)
+        self.qfont = QtGui.QFont('Courier') 
+        self.qantialiased = False
+        self.qtransformed = False
+        self.qscene = QtGui.QGraphicsScene()
+        self.qview = QtGui.QGraphicsView(self.qscene, self)
+        self.qview.scale(0.5,-0.5)
+        l = QtGui.QVBoxLayout(self)
+        l.addWidget(self.qview)
+        self.setBackgroundRole(QtGui.QPalette.Base)
+        self.setSizePolicy(QtGui.QSizePolicy.Expanding,
+                           QtGui.QSizePolicy.Expanding)
+        HPGLParser.__init__(self)
+
+    def _get_PW(self):
+        return self._pen_width
+    def _set_PW(self, value):
+        value = float(value)
+        #print "set pen width", value, value*300.0/75.0
+        #self.qpen.setWidthF(value*300.0/75.0)
+        self._pen_width = value
+    pen_width = property(_get_PW, _set_PW)
+        
+    def LT(self):
+        HPGLParser.LT(self)
+        if self.line_type == 0:
+            self.qpen.setStyle(Qt.SolidLine)
+        elif self.line_type == 1:
+            self.qpen.setStyle(Qt.DotLine)
+        else:
+            pattern = numpy.array(self.pen_styles[self.line_type])
+            x0, x1, y0, y1 = self.scale
+            dist = math.sqrt((x1-x0)**2 + (y1-y0)**2)
+            pattern = self.pattern_len*dist/100.0 * pattern
+            if self.qpen.widthF()>0:
+                pattern = pattern/self.qpen.widthF()
+            self.qpen.setDashPattern(pattern.tolist())
+            
+    def SP(self):
+        HPGLParser.SP(self)
+        self.qpen.setColor(QtGui.QColor(self.pen_colors[self.pen]))
+        
+    def parse(self, data):
+        HPGLParser.parse(self, data)
+        self.update()
+        self.qview.fitInView(self.qscene.sceneRect(), Qt.KeepAspectRatio)
+        
+    def minimumSizeHint(self):
+        return QtCore.QSize(100, 100)
+
+    def sizeHint(self):
+        return QtCore.QSize(400, 400)
+
+    def resizeEvent(self, event):
+        self.qview.fitInView(self.qscene.sceneRect(), Qt.KeepAspectRatio)
+        QtGui.QWidget.resizeEvent(self, event)
+
+    def plot_lines(self, points):
+        if len(points) == 1:
+            return
+        pa = QtGui.QPainterPath()
+        pa.addPolygon(QtGui.QPolygonF([QtCore.QPointF(*p) for p in points]))
+        self.qscene.addPath(pa, self.qpen)
+        
+    def plot_symbols(self, points):
+        pass
+    
+    def plot_string(self, s):
+        cw, ch = self.get_char_size()
+        # this is very ugly and so, but I don't understand haw is
+        # computed string positionning in HPGL...
+        t0 = self.qscene.addSimpleText(" ", self.qfont) 
+        br = t0.boundingRect()
+        mw = br.width()
+        mh = br.height()
+        self.qscene.removeItem(t0)
+        # don't ask me why theses constants are here, they seem OK, that's all
+        t = self.qscene.addSimpleText(s, self.qfont)
+        t.scale(1.5 * cw/mw, -2.0 * ch/mh)
+        x, y = self.pos
+        t.moveBy(x,y)
+        br = t.boundingRect()
+        t.moveBy(-0.5*cw, math.sqrt(2)*ch)
+
+        
+if __name__=='__main__':
+    import sys
+    a = QtGui.QApplication(sys.argv)
+    mw = QtGui.QMainWindow()
+    w = QtGui.QWidget(mw)
+    mw.setCentralWidget(w)
+
+    l = QtGui.QVBoxLayout(w)
+    p = QHPGLPlotterWidget(w)
+    l.addWidget(p, 1)
+    b = QtGui.QPushButton("Replot", w)
+    l.addWidget(b)
+    def plot():
+        p.parse(open(sys.argv[1]).read())
+    b.connect(b, SIGNAL('pressed()'), plot)
+    mw.show()
+    plot()
+    a.exec_()
+