X-Git-Url: http://sigrok.org/gitweb/?a=blobdiff_plain;f=mainwindow.py;h=058fe5fb54e7edb589999d11ef058e499ad3874b;hb=68348e5abaa05fbbde7f3e6b0b28c2d12a7a2601;hp=06a958ed1d35a9d05b3671588d3e9f50eafda269;hpb=48723bbb0c3b8832d2b95414d5e64e81d6869b4b;p=sigrok-meter.git diff --git a/mainwindow.py b/mainwindow.py index 06a958e..058fe5f 100644 --- a/mainwindow.py +++ b/mainwindow.py @@ -19,14 +19,19 @@ ## Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA ## +import acquisition import datamodel +import icons +import multiplotwidget import os.path import qtcompat -import samplingthread import textwrap +import time +import util QtCore = qtcompat.QtCore QtGui = qtcompat.QtGui +pyqtgraph = qtcompat.pyqtgraph class EmptyMessageListView(QtGui.QListView): '''List view that shows a message if the model im empty.''' @@ -48,97 +53,370 @@ class EmptyMessageListView(QtGui.QListView): class MainWindow(QtGui.QMainWindow): '''The main window of the application.''' + # Number of seconds that the plots display. + BACKLOG = 30 + + # Update interval of the plots in milliseconds. + UPDATEINTERVAL = 100 + def __init__(self, context, drivers): super(self.__class__, self).__init__() + # Used to coordinate the stopping of the acquisition and + # the closing of the window. + self._closing = False + self.context = context + self.drivers = drivers self.delegate = datamodel.MultimeterDelegate(self, self.font()) self.model = datamodel.MeasurementDataModel(self) - self.model.rowsInserted.connect(self.modelRowsInserted) - self.setup_ui() + # Maps from 'unit' to the corresponding plot. + self._plots = {} + # Maps from '(plot, device)' to the corresponding curve. + self._curves = {} + + self._setup_ui() + + self._plot_update_timer = QtCore.QTimer() + self._plot_update_timer.setInterval(MainWindow.UPDATEINTERVAL) + self._plot_update_timer.timeout.connect(self._updatePlots) - self.thread = samplingthread.SamplingThread(self.context, drivers) - self.thread.measured.connect(self.model.update) - self.thread.error.connect(self.error) - self.thread.start() + QtCore.QTimer.singleShot(0, self._start_acquisition) - def setup_ui(self): + def _start_acquisition(self): + self.acquisition = acquisition.Acquisition(self.context) + self.acquisition.measured.connect(self.model.update) + self.acquisition.stopped.connect(self._stopped) + + try: + for (ds, cs) in self.drivers: + self.acquisition.add_device(ds, cs) + except Exception as e: + QtGui.QMessageBox.critical(self, 'Error', str(e)) + self.close() + return + + self.start_stop_acquisition() + + def _setup_ui(self): self.setWindowTitle('sigrok-meter') - # resizing the listView below will increase this again - self.resize(10, 10) + # Resizing the listView below will increase this again. + self.resize(350, 10) - p = os.path.abspath(os.path.dirname(__file__)) - p = os.path.join(p, 'sigrok-logo-notext.png') - self.setWindowIcon(QtGui.QIcon(p)) + self.setWindowIcon(QtGui.QIcon(':/logo.png')) - actionQuit = QtGui.QAction(self) - actionQuit.setText('&Quit') - actionQuit.setIcon(QtGui.QIcon.fromTheme('application-exit')) - actionQuit.setShortcut('Ctrl+Q') - actionQuit.triggered.connect(self.close) + self._setup_graphPage() + self._setup_addDevicePage() + self._setup_logPage() + self._setup_preferencesPage() - actionAbout = QtGui.QAction(self) - actionAbout.setText('&About') - actionAbout.setIcon(QtGui.QIcon.fromTheme('help-about')) - actionAbout.triggered.connect(self.show_about) + self._pages = [ + self.graphPage, + self.addDevicePage, + self.logPage, + self.preferencesPage + ] + + self.stackedWidget = QtGui.QStackedWidget(self) + for page in self._pages: + self.stackedWidget.addWidget(page) + + self._setup_sidebar() - menubar = self.menuBar() - menuFile = menubar.addMenu('&File') - menuFile.addAction(actionQuit) - menuHelp = menubar.addMenu('&Help') - menuHelp.addAction(actionAbout) - - self.listView = EmptyMessageListView('waiting for data...') - self.listView.setFrameShape(QtGui.QFrame.NoFrame) - self.listView.viewport().setBackgroundRole(QtGui.QPalette.Window) - self.listView.viewport().setAutoFillBackground(True) - self.listView.setMinimumWidth(260) - self.listView.setSelectionMode(QtGui.QAbstractItemView.NoSelection) - self.listView.setEditTriggers(QtGui.QAbstractItemView.NoEditTriggers) - self.listView.setVerticalScrollMode(QtGui.QAbstractItemView.ScrollPerPixel) - self.listView.setItemDelegate(self.delegate) - self.listView.setModel(self.model) - self.listView.setUniformItemSizes(True) - self.listView.setMinimumSize(self.delegate.sizeHint()) - - self.setCentralWidget(self.listView) + self.setCentralWidget(QtGui.QWidget()) self.centralWidget().setContentsMargins(0, 0, 0, 0) + layout = QtGui.QHBoxLayout(self.centralWidget()) + layout.addWidget(self.sideBar) + layout.addWidget(self.stackedWidget) + layout.setSpacing(0) + layout.setContentsMargins(0, 0, 0, 0) + + self.resize(900, 550) + + def _setup_sidebar(self): + self.sideBar = QtGui.QToolBar(self) + self.sideBar.setOrientation(QtCore.Qt.Vertical) + + actionGraph = self.sideBar.addAction('Instantaneous Values and Graphs') + actionGraph.setCheckable(True) + actionGraph.setIcon(icons.graph) + actionGraph.triggered.connect(self.showGraphPage) + + #actionAdd = self.sideBar.addAction('Add Device') + #actionAdd.setCheckable(True) + #actionAdd.setIcon(icons.add) + #actionAdd.triggered.connect(self.showAddDevicePage) + + #actionLog = self.sideBar.addAction('Logs') + #actionLog.setCheckable(True) + #actionLog.setIcon(icons.log) + #actionLog.triggered.connect(self.showLogPage) + + #actionPreferences = self.sideBar.addAction('Preferences') + #actionPreferences.setCheckable(True) + #actionPreferences.setIcon(icons.preferences) + #actionPreferences.triggered.connect(self.showPreferencesPage) + + # make the buttons at the top exclusive + self.actionGroup = QtGui.QActionGroup(self) + self.actionGroup.addAction(actionGraph) + #self.actionGroup.addAction(actionAdd) + #self.actionGroup.addAction(actionLog) + #self.actionGroup.addAction(actionPreferences) + + # show graph at startup + actionGraph.setChecked(True) + + # fill space between buttons on the top and on the bottom + fill = QtGui.QWidget(self) + fill.setSizePolicy(QtGui.QSizePolicy.Preferred, QtGui.QSizePolicy.Expanding) + self.sideBar.addWidget(fill) + + self.actionStartStop = self.sideBar.addAction('Start Acquisition') + self.actionStartStop.setIcon(icons.start) + self.actionStartStop.triggered.connect(self.start_stop_acquisition) + + actionAbout = self.sideBar.addAction('About') + actionAbout.setIcon(icons.about) + actionAbout.triggered.connect(self.show_about) + + actionQuit = self.sideBar.addAction('Quit') + actionQuit.setIcon(icons.exit) + actionQuit.triggered.connect(self.close) + + s = self.style().pixelMetric(QtGui.QStyle.PM_LargeIconSize) + self.sideBar.setIconSize(QtCore.QSize(s, s)) + + self.sideBar.setStyleSheet(''' + QToolBar { + background-color: white; + margin: 0px; + border: 0px; + border-right: 1px solid black; + } + + QToolButton { + padding: 10px; + border: 0px; + border-right: 1px solid black; + } + + QToolButton:checked, + QToolButton[checkable="false"]:hover { + background-color: #c0d0e8; + } + ''') + + def _setup_graphPage(self): + listView = EmptyMessageListView('waiting for data...') + listView.setFrameShape(QtGui.QFrame.NoFrame) + listView.viewport().setBackgroundRole(QtGui.QPalette.Window) + listView.viewport().setAutoFillBackground(True) + listView.setMinimumWidth(260) + listView.setSelectionMode(QtGui.QAbstractItemView.NoSelection) + listView.setEditTriggers(QtGui.QAbstractItemView.NoEditTriggers) + listView.setVerticalScrollMode(QtGui.QAbstractItemView.ScrollPerPixel) + listView.setItemDelegate(self.delegate) + listView.setModel(self.model) + listView.setUniformItemSizes(True) + listView.setMinimumSize(self.delegate.sizeHint()) + + self.plotwidget = multiplotwidget.MultiPlotWidget(self) + self.plotwidget.plotHidden.connect(self._on_plotHidden) + + self.graphPage = QtGui.QSplitter(QtCore.Qt.Horizontal, self) + self.graphPage.addWidget(listView) + self.graphPage.addWidget(self.plotwidget) + self.graphPage.setStretchFactor(0, 0) + self.graphPage.setStretchFactor(1, 1) + + def _setup_addDevicePage(self): + self.addDevicePage = QtGui.QWidget(self) + layout = QtGui.QVBoxLayout(self.addDevicePage) + label = QtGui.QLabel('add device page') + layout.addWidget(label) + + def _setup_logPage(self): + self.logPage = QtGui.QWidget(self) + layout = QtGui.QVBoxLayout(self.logPage) + label = QtGui.QLabel('log page') + layout.addWidget(label) + + def _setup_preferencesPage(self): + self.preferencesPage = QtGui.QWidget(self) + layout = QtGui.QVBoxLayout(self.preferencesPage) + label = QtGui.QLabel('preferences page') + layout.addWidget(label) + + def showPage(self, page): + self.stackedWidget.setCurrentIndex(self._pages.index(page)) + + @QtCore.Slot(bool) + def showGraphPage(self): + self.showPage(self.graphPage) + + @QtCore.Slot(bool) + def showAddDevicePage(self): + self.showPage(self.addDevicePage) + + @QtCore.Slot(bool) + def showLogPage(self): + self.showPage(self.logPage) + + @QtCore.Slot(bool) + def showPreferencesPage(self): + self.showPage(self.preferencesPage) + + def _getPlot(self, unit): + '''Looks up or creates a new plot for 'unit'.''' + + if unit in self._plots: + return self._plots[unit] + + # create a new plot for the unit + plot = self.plotwidget.addPlot() + plot.yaxis.setLabel(util.quantity_from_unit(unit), units=util.format_unit(unit)) + plot.view.setXRange(-MainWindow.BACKLOG, 0, update=False) + plot.view.setYRange(-1, 1) + plot.view.enableAutoRange(axis=pyqtgraph.ViewBox.YAxis) + # lock to the range calculated by the view using additional padding, + # looks nicer this way + r = plot.view.viewRange() + plot.view.setLimits(xMin=r[0][0], xMax=r[0][1]) + + self._plots[unit] = plot + return plot + + def _getCurve(self, plot, deviceID): + '''Looks up or creates a new curve for '(plot, deviceID)'.''' + + key = (plot, deviceID) + if key in self._curves: + return self._curves[key] + + # create a new curve + curve = pyqtgraph.PlotDataItem( + antialias=True, + symbolPen=pyqtgraph.mkPen(QtGui.QColor(QtCore.Qt.black)), + symbolBrush=pyqtgraph.mkBrush(QtGui.QColor(QtCore.Qt.black)), + symbolSize=1 + ) + plot.view.addItem(curve) + + self._curves[key] = curve + return curve + + def _updatePlots(self): + '''Updates all plots.''' + + # loop over all devices and channels + for row in range(self.model.rowCount()): + idx = self.model.index(row, 0) + deviceID = self.model.data(idx, + datamodel.MeasurementDataModel.idRole) + deviceID = tuple(deviceID) # PySide returns a list. + traces = self.model.data(idx, + datamodel.MeasurementDataModel.tracesRole) + + for unit, trace in traces.items(): + now = time.time() + + # remove old samples + l = now - MainWindow.BACKLOG + while trace.samples and trace.samples[0][0] < l: + trace.samples.pop(0) + + plot = self._getPlot(unit) + if not plot.visible: + if trace.new: + self.plotwidget.showPlot(plot) + + if plot.visible: + xdata = [s[0] - now for s in trace.samples] + ydata = [s[1] for s in trace.samples] + + color = self.model.data(idx, + datamodel.MeasurementDataModel.colorRole) + + curve = self._getCurve(plot, deviceID) + curve.setPen(pyqtgraph.mkPen(color=color)) + curve.setData(xdata, ydata) + + @QtCore.Slot(multiplotwidget.Plot) + def _on_plotHidden(self, plot): + plotunit = [u for u, p in self._plots.items() if p == plot][0] + + # Mark all traces of all devices/channels with the same unit as the + # plot as "old" ('trace.new = False'). As soon as a new sample arrives + # on one trace, the plot will be shown again + for row in range(self.model.rowCount()): + idx = self.model.index(row, 0) + traces = self.model.data(idx, datamodel.MeasurementDataModel.tracesRole) + + for traceunit, trace in traces.items(): + if traceunit == plotunit: + trace.new = False + + @QtCore.Slot() + def _stopped(self): + if self._closing: + # The acquisition was stopped by the 'closeEvent()', close the + # window again now that the acquisition has stopped. + self.close() + def closeEvent(self, event): - self.thread.stop() - event.accept() + if self.acquisition.is_running(): + # Stop the acquisition before closing the window. + self._closing = True + self.start_stop_acquisition() + event.ignore() + else: + event.accept() + + @QtCore.Slot() + def start_stop_acquisition(self): + if self.acquisition.is_running(): + self.acquisition.stop() + self._plot_update_timer.stop() + self.actionStartStop.setText('Start Acquisition') + self.actionStartStop.setIcon(icons.start) + else: + # before starting (again), remove all old samples and old curves + self.model.clear_samples() + + for key in self._curves: + plot, _ = key + curve = self._curves[key] + plot.view.removeItem(curve) + self._curves = {} + + self.acquisition.start() + self._plot_update_timer.start() + self.actionStartStop.setText('Stop Acquisition') + self.actionStartStop.setIcon(icons.stop) @QtCore.Slot() def show_about(self): text = textwrap.dedent('''\
- sigrok-meter
- 0.1.0
- Using libsigrok {} (lib version {}).
+ sigrok-meter 0.1.0

+ Using libsigrok {} (lib version {}).

http://www.sigrok.org

+ License: GNU GPL, version 3 or later
+
This program comes with ABSOLUTELY NO WARRANTY;
for details visit - http://www.gnu.org/licenses/gpl.html + http://www.gnu.org/licenses/gpl.html
+
+ Some icons by + the GNOME project
'''.format(self.context.package_version, self.context.lib_version)) QtGui.QMessageBox.about(self, 'About sigrok-meter', text) - - @QtCore.Slot(str) - def error(self, msg): - '''Error handler for the sampling thread.''' - QtGui.QMessageBox.critical(self, 'Error', msg) - self.close() - - @QtCore.Slot(object, int, int) - def modelRowsInserted(self, parent, start, end): - '''Resizes the list view to the size of the content.''' - - rows = self.model.rowCount() - dh = self.delegate.sizeHint().height() - self.listView.setMinimumHeight(dh * rows)