Mercurial > pygpibtoolkit / file comparison
comparison: hpgl_qt.py
hpgl_qt.py
- changeset 21
- 66b7cd43bac1
- child 23
- cb97962a1ae9
equal
deleted
inserted
replaced
| 20:de60b63b2275 | 21:66b7cd43bac1 |
|---|---|
| 1 # -*- coding: utf-8 -*- | |
| 2 import numpy | |
| 3 import pylab | |
| 4 import math | |
| 5 import new # monkey patching... | |
| 6 | |
| 7 from PyQt4 import QtGui, QtCore | |
| 8 from PyQt4.QtCore import Qt, SIGNAL | |
| 9 | |
| 10 from hpgl_parser import HPGLParser | |
| 11 | |
| 12 class QHPGLPlotterWidget(HPGLParser, QtGui.QWidget): | |
| 13 pen_styles = {2: [0.4, 0.6], | |
| 14 3: [0.6, 0.4], | |
| 15 4: [0.7, 0.1, 0.1, 0.1], | |
| 16 5: [0.6, 0.1, 0.2, 0.1], | |
| 17 6: [0.45, 0.1, 0.1, 0.2, 0.1, 0.05], | |
| 18 } | |
| 19 pen_colors = ["black", "green", "red", "blue", "yellow", "cyan", "magenta", | |
| 20 "darkred", "darkblue"] | |
| 21 | |
| 22 def __init__(self, parent=None): | |
| 23 QtGui.QWidget.__init__(self, parent) | |
| 24 | |
| 25 self.qpen = QtGui.QPen(QtCore.Qt.blue) | |
| 26 self.qbrush = QtGui.QBrush(QtCore.Qt.blue) | |
| 27 self.qfont = QtGui.QFont('Courier') | |
| 28 self.qantialiased = False | |
| 29 self.qtransformed = False | |
| 30 self.qscene = QtGui.QGraphicsScene() | |
| 31 self.qview = QtGui.QGraphicsView(self.qscene, self) | |
| 32 self.qview.scale(0.5,-0.5) | |
| 33 l = QtGui.QVBoxLayout(self) | |
| 34 l.addWidget(self.qview) | |
| 35 self.setBackgroundRole(QtGui.QPalette.Base) | |
| 36 self.setSizePolicy(QtGui.QSizePolicy.Expanding, | |
| 37 QtGui.QSizePolicy.Expanding) | |
| 38 HPGLParser.__init__(self) | |
| 39 | |
| 40 def _get_PW(self): | |
| 41 return self._pen_width | |
| 42 def _set_PW(self, value): | |
| 43 value = float(value) | |
| 44 #print "set pen width", value, value*300.0/75.0 | |
| 45 #self.qpen.setWidthF(value*300.0/75.0) | |
| 46 self._pen_width = value | |
| 47 pen_width = property(_get_PW, _set_PW) | |
| 48 | |
| 49 def LT(self): | |
| 50 HPGLParser.LT(self) | |
| 51 if self.line_type == 0: | |
| 52 self.qpen.setStyle(Qt.SolidLine) | |
| 53 elif self.line_type == 1: | |
| 54 self.qpen.setStyle(Qt.DotLine) | |
| 55 else: | |
| 56 pattern = numpy.array(self.pen_styles[self.line_type]) | |
| 57 x0, x1, y0, y1 = self.scale | |
| 58 dist = math.sqrt((x1-x0)**2 + (y1-y0)**2) | |
| 59 pattern = self.pattern_len*dist/100.0 * pattern | |
| 60 if self.qpen.widthF()>0: | |
| 61 pattern = pattern/self.qpen.widthF() | |
| 62 self.qpen.setDashPattern(pattern.tolist()) | |
| 63 | |
| 64 def SP(self): | |
| 65 HPGLParser.SP(self) | |
| 66 self.qpen.setColor(QtGui.QColor(self.pen_colors[self.pen])) | |
| 67 | |
| 68 def parse(self, data): | |
| 69 HPGLParser.parse(self, data) | |
| 70 self.update() | |
| 71 self.qview.fitInView(self.qscene.sceneRect(), Qt.KeepAspectRatio) | |
| 72 | |
| 73 def minimumSizeHint(self): | |
| 74 return QtCore.QSize(100, 100) | |
| 75 | |
| 76 def sizeHint(self): | |
| 77 return QtCore.QSize(400, 400) | |
| 78 | |
| 79 def resizeEvent(self, event): | |
| 80 self.qview.fitInView(self.qscene.sceneRect(), Qt.KeepAspectRatio) | |
| 81 QtGui.QWidget.resizeEvent(self, event) | |
| 82 | |
| 83 def plot_lines(self, points): | |
| 84 if len(points) == 1: | |
| 85 return | |
| 86 pa = QtGui.QPainterPath() | |
| 87 pa.addPolygon(QtGui.QPolygonF([QtCore.QPointF(*p) for p in points])) | |
| 88 self.qscene.addPath(pa, self.qpen) | |
| 89 | |
| 90 def plot_symbols(self, points): | |
| 91 pass | |
| 92 | |
| 93 def plot_string(self, s): | |
| 94 cw, ch = self.get_char_size() | |
| 95 # this is very ugly and so, but I don't understand haw is | |
| 96 # computed string positionning in HPGL... | |
| 97 t0 = self.qscene.addSimpleText(" ", self.qfont) | |
| 98 br = t0.boundingRect() | |
| 99 mw = br.width() | |
| 100 mh = br.height() | |
| 101 self.qscene.removeItem(t0) | |
| 102 # don't ask me why theses constants are here, they seem OK, that's all | |
| 103 t = self.qscene.addSimpleText(s, self.qfont) | |
| 104 t.scale(1.5 * cw/mw, -2.0 * ch/mh) | |
| 105 x, y = self.pos | |
| 106 t.moveBy(x,y) | |
| 107 br = t.boundingRect() | |
| 108 t.moveBy(-0.5*cw, math.sqrt(2)*ch) | |
| 109 | |
| 110 | |
| 111 if __name__=='__main__': | |
| 112 import sys | |
| 113 a = QtGui.QApplication(sys.argv) | |
| 114 mw = QtGui.QMainWindow() | |
| 115 w = QtGui.QWidget(mw) | |
| 116 mw.setCentralWidget(w) | |
| 117 | |
| 118 l = QtGui.QVBoxLayout(w) | |
| 119 p = QHPGLPlotterWidget(w) | |
| 120 l.addWidget(p, 1) | |
| 121 b = QtGui.QPushButton("Replot", w) | |
| 122 l.addWidget(b) | |
| 123 def plot(): | |
| 124 p.parse(open(sys.argv[1]).read()) | |
| 125 b.connect(b, SIGNAL('pressed()'), plot) | |
| 126 mw.show() | |
| 127 plot() | |
| 128 a.exec_() | |
| 129 |
