diff -r e8f3c9276f3f -r 0f8f2621418f qgpib_plotter.py --- a/qgpib_plotter.py Sun Jan 20 17:09:31 2008 +0100 +++ b/qgpib_plotter.py Fri Jan 25 20:25:39 2008 +0100 @@ -1,11 +1,13 @@ # import os, sys +import time from PyQt4 import QtGui, QtCore, uic from PyQt4.QtCore import SIGNAL, Qt -from gpib_plotter import GPIBplotter +#from gpib_plotter import GPIBplotter +from gpib_plotter_mockup import GPIBplotter from hpgl_qt import QHPGLPlotterWidget form_class, base_class = uic.loadUiType(os.path.join(os.path.dirname(__file__), "qhpgl_plotter.ui")) @@ -21,6 +23,7 @@ _pos = PreferenceItem(basetype=QtCore.QPoint) _size = PreferenceItem(basetype=QtCore.QSize) _appState = PreferenceItem(basetype=QtCore.QByteArray) + class QtHPGLPlotter(QtGui.QMainWindow, form_class): def __init__(self, parent=None): @@ -105,6 +108,21 @@ if not self.plotsView.currentIndex().isValid(): self.plotsView.setCurrentIndex(self.plotsView.model().index(0, 0)) + + def plotReceived(self, num): + print "plot received", num + self._receiving = False + self.setReceivingLed() + plot, timestamp = self.captureThread.getPlot(num) + name = "plot_%s"%(num) + lst = self.plotsView.model().stringList() + lst.append(name) + self._plots[name] = plot + self.plotsView.model().setStringList(lst) + + def plotStarted(self): + self._receiving = True + self.setReceivingLed() def saveTriggered(self, checked=False): print "save" @@ -112,15 +130,22 @@ print "saveAs" def initializeGPIB(self): + self._online = False try: - self.gpib_plotter = GPIBplotter(device=self._prefs.device, + self.gpib_plotter = QGPIBplotter(device=self._prefs.device, address=self._prefs.address, ) - except: + self.captureThread = GPIBReceiver(self.gpib_plotter) + self.connect(self.captureThread, SIGNAL('plotReceived(int)'), + self.plotReceived) + self.connect(self.captureThread, SIGNAL('plotStarted()'), + self.plotStarted) + self.captureThread.start() + except Exception, e: + #print e self.gpib_plotter = None - self._online = False self.setCaptureLed() - + def captureToggled(self, state): if state: if self.gpib_plotter is None: @@ -129,22 +154,102 @@ QtGui.QMessageBox.critical(self, self.tr("GPIB error"), self.tr("Unable to initialize GPIB connection.
Please check your GPIB dongle and settings.")) self._online = False - else: - # todo - self._online = True + self.setCaptureLed() + return + self._online = True + self.captureThread.startCapture() else: + if self.captureThread: + self.captureThread.stopCapture() self._online = False self.setCaptureLed() - - + def setCaptureLed(self): if self._online: - icn = QtGui.QIcon(':/icons/led_green.png') + icn = QtGui.QIcon(':/icons/led_green.svg') else: - icn = QtGui.QIcon(':/icons/led_green_off.png') + icn = QtGui.QIcon(':/icons/led_green_off.svg') self.captureButton.setIcon(icn) self.captureButton.setChecked(self._online) + + def setReceivingLed(self): + if self._receiving: + icn = QtGui.QIcon(':/icons/led_red.svg') + else: + icn = QtGui.QIcon(':/icons/led_red_off.svg') + self.receivingButton.setIcon(icn) + +class QGPIBplotter(GPIBplotter): + def __init__(self, device="/dev/ttyUSB0", baudrate=115200, timeout=0.1, + address=5): + GPIBplotter.__init__(self, device, baudrate, timeout, address) + self.emitter = None + + def plotStarted(self): + if self.emitter: + self.emitter.emit(SIGNAL('plotStarted()')) + #self.emitter.msleep(1) +class GPIBReceiver(QtCore.QThread): + def __init__(self, cnx): + QtCore.QThread.__init__(self) + self.gpibplotter = cnx + self.gpibplotter.emitter = self + + self._cancelmutex = QtCore.QMutex() + self._cancel = False + #self._nreceived = 0 + self._plotsmutex = QtCore.QMutex() + self._plots = [] + self._startstopmutex = QtCore.QMutex() + self._startstop = QtCore.QWaitCondition() + self._capturing = False + + def cancel(self): + self._cancelmutex.lock() + self._cancel = True + self._cancelmutex.unlock() + + def startCapture(self): + self._startstop.wakeOne() + + def stopCapture(self): + self._startstopmutex.lock() + self._capturing = False + self._startstopmutex.unlock() + + def run(self): + while 1: + self._cancelmutex.lock() + if self._cancel: + return + self._cancelmutex.unlock() + self._startstopmutex.lock() + if not self._capturing: + self._startstop.wait(self._startstopmutex) + self._capturing = True + self._startstopmutex.unlock() + + plot = self.gpibplotter.load_plot(wait_timeout=0.1) + timestamp = time.time() + if plot: + self._plotsmutex.lock() + self._plots.append((plot, timestamp)) + n = len(self._plots) + self._plotsmutex.unlock() + self.emit(SIGNAL('plotReceived(int)'), n-1) + self.msleep(10) + + def getPlot(self, num): + self._plotsmutex.lock() + try: + return self._plots[num] + finally: + self._plotsmutex.unlock() + + + + if __name__ == '__main__': a = QtGui.QApplication([]) w = QtHPGLPlotter()