From: Jens Steinhauser Date: Wed, 19 Nov 2014 02:15:21 +0000 (+0100) Subject: Split the program up into multiple files. X-Git-Url: http://sigrok.org/gitweb/?p=sigrok-meter.git;a=commitdiff_plain;h=48723bbb0c3b8832d2b95414d5e64e81d6869b4b Split the program up into multiple files. --- diff --git a/datamodel.py b/datamodel.py new file mode 100644 index 0000000..c36e295 --- /dev/null +++ b/datamodel.py @@ -0,0 +1,184 @@ +## +## This file is part of the sigrok-meter project. +## +## Copyright (C) 2014 Jens Steinhauser +## +## This program is free software; you can redistribute it and/or modify +## it under the terms of the GNU General Public License as published by +## the Free Software Foundation; either version 2 of the License, or +## (at your option) any later version. +## +## This program is distributed in the hope that it will be useful, +## but WITHOUT ANY WARRANTY; without even the implied warranty of +## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +## GNU General Public License for more details. +## +## You should have received a copy of the GNU General Public License +## along with this program; if not, write to the Free Software +## Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA +## + +import qtcompat +import sigrok.core as sr + +QtCore = qtcompat.QtCore +QtGui = qtcompat.QtGui + +class MeasurementDataModel(QtGui.QStandardItemModel): + '''Model to hold the measured values.''' + + '''Role used to identify and find the item.''' + _idRole = QtCore.Qt.UserRole + 1 + + '''Role used to store the device vendor and model.''' + descRole = QtCore.Qt.UserRole + 2 + + def __init__(self, parent): + super(self.__class__, self).__init__(parent) + + # Use the description text to sort the items for now, because the + # _idRole holds tuples, and using them to sort doesn't work. + self.setSortRole(MeasurementDataModel.descRole) + + # Used in 'format_value()' to check against. + self.inf = float('inf') + + def format_unit(self, u): + units = { + sr.Unit.VOLT: 'V', + sr.Unit.AMPERE: 'A', + sr.Unit.OHM: u'\u03A9', + sr.Unit.FARAD: 'F', + sr.Unit.KELVIN: 'K', + sr.Unit.CELSIUS: u'\u00B0C', + sr.Unit.FAHRENHEIT: u'\u00B0F', + sr.Unit.HERTZ: 'Hz', + sr.Unit.PERCENTAGE: '%', + # sr.Unit.BOOLEAN + sr.Unit.SECOND: 's', + sr.Unit.SIEMENS: 'S', + sr.Unit.DECIBEL_MW: 'dBu', + sr.Unit.DECIBEL_VOLT: 'dBV', + # sr.Unit.UNITLESS + sr.Unit.DECIBEL_SPL: 'dB', + # sr.Unit.CONCENTRATION + sr.Unit.REVOLUTIONS_PER_MINUTE: 'rpm', + sr.Unit.VOLT_AMPERE: 'VA', + sr.Unit.WATT: 'W', + sr.Unit.WATT_HOUR: 'Wh', + sr.Unit.METER_SECOND: 'm/s', + sr.Unit.HECTOPASCAL: 'hPa', + sr.Unit.HUMIDITY_293K: '%rF', + sr.Unit.DEGREE: u'\u00B0', + sr.Unit.HENRY: 'H' + } + + return units.get(u, '') + + def format_mqflags(self, mqflags): + if sr.QuantityFlag.AC in mqflags: + return 'AC' + elif sr.QuantityFlag.DC in mqflags: + return 'DC' + else: + return '' + + def format_value(self, mag): + if mag == self.inf: + return u'\u221E' + return '{:f}'.format(mag) + + def getItem(self, device, channel): + '''Returns the item for the device + channel combination from the model, + or creates a new item if no existing one matches.''' + + # unique identifier for the device + channel + # TODO: isn't there something better? + uid = ( + device.vendor, + device.model, + device.serial_number(), + device.connection_id(), + channel.index + ) + + # find the correct item in the model + for row in range(self.rowCount()): + item = self.item(row) + rid = item.data(MeasurementDataModel._idRole) + rid = tuple(rid) # PySide returns a list + if uid == rid: + return item + + # nothing found, create a new item + desc = '{} {}, channel "{}"'.format( + device.vendor, device.model, channel.name) + + item = QtGui.QStandardItem() + item.setData(uid, MeasurementDataModel._idRole) + item.setData(desc, MeasurementDataModel.descRole) + self.appendRow(item) + self.sort(0) + return item + + @QtCore.Slot(object, object, object) + def update(self, device, channel, data): + '''Updates the data for the device (+channel) with the most recent + measurement from the given payload.''' + + item = self.getItem(device, channel) + + value, unit, mqflags = data + value_str = self.format_value(value) + unit_str = self.format_unit(unit) + mqflags_str = self.format_mqflags(mqflags) + + disp = ' '.join([value_str, unit_str, mqflags_str]) + item.setData(disp, QtCore.Qt.DisplayRole) + +class MultimeterDelegate(QtGui.QStyledItemDelegate): + '''Delegate to show the data items from a MeasurementDataModel.''' + + def __init__(self, parent, font): + '''Initializes the delegate. + + :param font: Font used for the description text, the value is drawn + with a slightly bigger and bold variant of the font. + ''' + + super(self.__class__, self).__init__(parent) + + self._nfont = font + self._bfont = QtGui.QFont(self._nfont) + + self._bfont.setBold(True) + if self._bfont.pixelSize() != -1: + self._bfont.setPixelSize(self._bfont.pixelSize() * 1.8) + else: + self._bfont.setPointSizeF(self._bfont.pointSizeF() * 1.8) + + fi = QtGui.QFontInfo(self._nfont) + self._nfontheight = fi.pixelSize() + + fm = QtGui.QFontMetrics(self._bfont) + r = fm.boundingRect('-XX.XXXXXX X XX') + self._size = QtCore.QSize(r.width() * 1.2, r.height() * 3.5) + + def sizeHint(self, option=None, index=None): + return self._size + + def paint(self, painter, options, index): + value = index.data(QtCore.Qt.DisplayRole) + desc = index.data(MeasurementDataModel.descRole) + + # description in the top left corner + painter.setFont(self._nfont) + p = options.rect.topLeft() + p += QtCore.QPoint(self._nfontheight, 2 * self._nfontheight) + painter.drawText(p, desc) + + # value in the center + painter.setFont(self._bfont) + r = options.rect.adjusted(self._nfontheight, 2.5 * self._nfontheight, + 0, 0) + painter.drawText(r, QtCore.Qt.AlignCenter, value) diff --git a/mainwindow.py b/mainwindow.py new file mode 100644 index 0000000..06a958e --- /dev/null +++ b/mainwindow.py @@ -0,0 +1,144 @@ +## +## This file is part of the sigrok-meter project. +## +## Copyright (C) 2013 Uwe Hermann +## Copyright (C) 2014 Jens Steinhauser +## +## This program is free software; you can redistribute it and/or modify +## it under the terms of the GNU General Public License as published by +## the Free Software Foundation; either version 2 of the License, or +## (at your option) any later version. +## +## This program is distributed in the hope that it will be useful, +## but WITHOUT ANY WARRANTY; without even the implied warranty of +## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +## GNU General Public License for more details. +## +## You should have received a copy of the GNU General Public License +## along with this program; if not, write to the Free Software +## Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA +## + +import datamodel +import os.path +import qtcompat +import samplingthread +import textwrap + +QtCore = qtcompat.QtCore +QtGui = qtcompat.QtGui + +class EmptyMessageListView(QtGui.QListView): + '''List view that shows a message if the model im empty.''' + + def __init__(self, message, parent=None): + super(self.__class__, self).__init__(parent) + + self._message = message + + def paintEvent(self, event): + m = self.model() + if m and m.rowCount(): + super(self.__class__, self).paintEvent(event) + return + + painter = QtGui.QPainter(self.viewport()) + painter.drawText(self.rect(), QtCore.Qt.AlignCenter, self._message) + +class MainWindow(QtGui.QMainWindow): + '''The main window of the application.''' + + def __init__(self, context, drivers): + super(self.__class__, self).__init__() + + self.context = context + + self.delegate = datamodel.MultimeterDelegate(self, self.font()) + self.model = datamodel.MeasurementDataModel(self) + self.model.rowsInserted.connect(self.modelRowsInserted) + + self.setup_ui() + + self.thread = samplingthread.SamplingThread(self.context, drivers) + self.thread.measured.connect(self.model.update) + self.thread.error.connect(self.error) + self.thread.start() + + def setup_ui(self): + self.setWindowTitle('sigrok-meter') + # resizing the listView below will increase this again + self.resize(10, 10) + + p = os.path.abspath(os.path.dirname(__file__)) + p = os.path.join(p, 'sigrok-logo-notext.png') + self.setWindowIcon(QtGui.QIcon(p)) + + actionQuit = QtGui.QAction(self) + actionQuit.setText('&Quit') + actionQuit.setIcon(QtGui.QIcon.fromTheme('application-exit')) + actionQuit.setShortcut('Ctrl+Q') + actionQuit.triggered.connect(self.close) + + actionAbout = QtGui.QAction(self) + actionAbout.setText('&About') + actionAbout.setIcon(QtGui.QIcon.fromTheme('help-about')) + actionAbout.triggered.connect(self.show_about) + + 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.centralWidget().setContentsMargins(0, 0, 0, 0) + + def closeEvent(self, event): + self.thread.stop() + event.accept() + + @QtCore.Slot() + def show_about(self): + text = textwrap.dedent('''\ +
+ sigrok-meter
+ 0.1.0
+ Using libsigrok {} (lib version {}).
+ + http://www.sigrok.org
+
+ This program comes with ABSOLUTELY NO WARRANTY;
+ for details visit + + http://www.gnu.org/licenses/gpl.html +
+ '''.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) diff --git a/qtcompat.py b/qtcompat.py new file mode 100644 index 0000000..1457615 --- /dev/null +++ b/qtcompat.py @@ -0,0 +1,52 @@ +## +## This file is part of the sigrok-meter project. +## +## Copyright (C) 2014 Jens Steinhauser +## +## This program is free software; you can redistribute it and/or modify +## it under the terms of the GNU General Public License as published by +## the Free Software Foundation; either version 2 of the License, or +## (at your option) any later version. +## +## This program is distributed in the hope that it will be useful, +## but WITHOUT ANY WARRANTY; without even the implied warranty of +## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +## GNU General Public License for more details. +## +## You should have received a copy of the GNU General Public License +## along with this program; if not, write to the Free Software +## Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA +## + +def load_modules(force_pyside): + if force_pyside: + import PySide.QtCore as _QtCore + import PySide.QtGui as _QtGui + else: + try: + # Use version 2 API in all cases, because that's what PySide uses. + import sip + sip.setapi('QVariant', 2) + sip.setapi('QDate', 2) + sip.setapi('QDateTime', 2) + sip.setapi('QString', 2) + sip.setapi('QTextStream', 2) + sip.setapi('QTime', 2) + sip.setapi('QUrl', 2) + sip.setapi('QVariant', 2) + + import PyQt4.QtCore as _QtCore + import PyQt4.QtGui as _QtGui + + # Add PySide compatible names. + _QtCore.Signal = _QtCore.pyqtSignal + _QtCore.Slot = _QtCore.pyqtSlot + except: + sys.stderr.write('import of PyQt4 failed, using PySide\n') + import PySide.QtCore as _QtCore + import PySide.QtGui as _QtGui + + global QtCore + global QtGui + QtCore = _QtCore + QtGui = _QtGui diff --git a/samplingthread.py b/samplingthread.py new file mode 100644 index 0000000..708f99d --- /dev/null +++ b/samplingthread.py @@ -0,0 +1,128 @@ +## +## This file is part of the sigrok-meter project. +## +## Copyright (C) 2013 Uwe Hermann +## Copyright (C) 2014 Jens Steinhauser +## +## This program is free software; you can redistribute it and/or modify +## it under the terms of the GNU General Public License as published by +## the Free Software Foundation; either version 2 of the License, or +## (at your option) any later version. +## +## This program is distributed in the hope that it will be useful, +## but WITHOUT ANY WARRANTY; without even the implied warranty of +## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +## GNU General Public License for more details. +## +## You should have received a copy of the GNU General Public License +## along with this program; if not, write to the Free Software +## Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA +## + +import qtcompat +import sigrok.core as sr + +QtCore = qtcompat.QtCore +QtGui = qtcompat.QtGui + +class SamplingThread(QtCore.QObject): + '''A class that handles the reception of sigrok packets in the background.''' + + class Worker(QtCore.QObject): + '''Helper class that does the actual work in another thread.''' + + '''Signal emitted when new data arrived.''' + measured = QtCore.Signal(object, object, object) + + '''Signal emmited in case of an error.''' + error = QtCore.Signal(str) + + def __init__(self, context, drivers): + super(self.__class__, self).__init__() + + self.context = context + self.drivers = drivers + + self.sampling = False + + @QtCore.Slot() + def start_sampling(self): + devices = [] + for name, options in self.drivers: + try: + dr = self.context.drivers[name] + devices.append(dr.scan(**options)[0]) + except: + self.error.emit( + 'Unable to get device for driver "{}".'.format(name)) + return + + self.session = self.context.create_session() + for dev in devices: + self.session.add_device(dev) + dev.open() + self.session.add_datafeed_callback(self.callback) + self.session.start() + self.sampling = True + self.session.run() + + # If sampling is 'True' here, it means that 'stop_sampling()' was + # not called, therefore 'session.run()' ended too early, indicating + # an error. + if self.sampling: + self.error.emit('An error occured during the acquisition.') + + def stop_sampling(self): + if self.sampling: + self.sampling = False + self.session.stop() + + def callback(self, device, packet): + if not sr: + # In rare cases it can happen that the callback fires while + # the interpreter is shutting down. Then the sigrok module + # is already set to 'None'. + return + + if packet.type != sr.PacketType.ANALOG: + return + + if not len(packet.payload.channels): + return + + # TODO: find a device with multiple channels in one packet + channel = packet.payload.channels[0] + + # the most recent value + value = packet.payload.data[0][-1] + + self.measured.emit(device, channel, + (value, packet.payload.unit, packet.payload.mq_flags)) + + # signal used to start the worker across threads + _start_signal = QtCore.Signal() + + def __init__(self, context, drivers): + super(self.__class__, self).__init__() + + self.worker = self.Worker(context, drivers) + self.thread = QtCore.QThread() + self.worker.moveToThread(self.thread) + + self._start_signal.connect(self.worker.start_sampling) + + # expose the signals of the worker + self.measured = self.worker.measured + self.error = self.worker.error + + self.thread.start() + + def start(self): + '''Starts sampling''' + self._start_signal.emit() + + def stop(self): + '''Stops sampling and the background thread.''' + self.worker.stop_sampling() + self.thread.quit() + self.thread.wait() diff --git a/sigrok-meter b/sigrok-meter index 6f83443..4532844 100755 --- a/sigrok-meter +++ b/sigrok-meter @@ -22,8 +22,6 @@ ## import argparse -import datetime -import os.path import re import sigrok.core as sr import sys @@ -86,418 +84,19 @@ def parse_cli(): return result if __name__ == '__main__': - # The command line parsing and import of the Qt modules is done here, - # so that the modules are imported before the classes below derive - # from classes therein. The rest of the main function is at the bottom. - args = parse_cli() - if args['pyside']: - from PySide import QtCore, QtGui - else: - try: - # Use version 2 API in all cases, because that's what PySide uses. - import sip - sip.setapi('QVariant', 2) - sip.setapi('QDate', 2) - sip.setapi('QDateTime', 2) - sip.setapi('QString', 2) - sip.setapi('QTextStream', 2) - sip.setapi('QTime', 2) - sip.setapi('QUrl', 2) - sip.setapi('QVariant', 2) - - from PyQt4 import QtCore, QtGui - - # Add PySide compatible names. - QtCore.Signal = QtCore.pyqtSignal - QtCore.Slot = QtCore.pyqtSlot - except: - sys.stderr.write('import of PyQt4 failed, using PySide\n') - from PySide import QtCore, QtGui - -class SamplingThread(QtCore.QObject): - '''A class that handles the reception of sigrok packets in the background.''' - - class Worker(QtCore.QObject): - '''Helper class that does the actual work in another thread.''' - - '''Signal emitted when new data arrived.''' - measured = QtCore.Signal(object, object, object) - - '''Signal emmited in case of an error.''' - error = QtCore.Signal(str) - - def __init__(self, context, drivers): - super(self.__class__, self).__init__() - - self.context = context - self.drivers = drivers - - self.sampling = False - - @QtCore.Slot() - def start_sampling(self): - devices = [] - for name, options in self.drivers: - try: - dr = self.context.drivers[name] - devices.append(dr.scan(**options)[0]) - except: - self.error.emit( - 'Unable to get device for driver "{}".'.format(name)) - return - - self.session = self.context.create_session() - for dev in devices: - self.session.add_device(dev) - dev.open() - self.session.add_datafeed_callback(self.callback) - self.session.start() - self.sampling = True - self.session.run() - - # If sampling is 'True' here, it means that 'stop_sampling()' was - # not called, therefore 'session.run()' ended too early, indicating - # an error. - if self.sampling: - self.error.emit('An error occured during the acquisition.') - - def stop_sampling(self): - if self.sampling: - self.sampling = False - self.session.stop() - - def callback(self, device, packet): - if not sr: - # In rare cases it can happen that the callback fires while - # the interpreter is shutting down. Then the sigrok module - # is already set to 'None'. - return - - if packet.type != sr.PacketType.ANALOG: - return - - if not len(packet.payload.channels): - return - - # TODO: find a device with multiple channels in one packet - channel = packet.payload.channels[0] - - # the most recent value - value = packet.payload.data[0][-1] - - self.measured.emit(device, channel, - (value, packet.payload.unit, packet.payload.mq_flags)) - - # signal used to start the worker across threads - _start_signal = QtCore.Signal() - - def __init__(self, context, drivers): - super(self.__class__, self).__init__() - - self.worker = self.Worker(context, drivers) - self.thread = QtCore.QThread() - self.worker.moveToThread(self.thread) - - self._start_signal.connect(self.worker.start_sampling) - - # expose the signals of the worker - self.measured = self.worker.measured - self.error = self.worker.error - - self.thread.start() - - def start(self): - '''Starts sampling''' - self._start_signal.emit() - - def stop(self): - '''Stops sampling and the background thread.''' - self.worker.stop_sampling() - self.thread.quit() - self.thread.wait() - -class MeasurementDataModel(QtGui.QStandardItemModel): - '''Model to hold the measured values.''' - - '''Role used to identify and find the item.''' - _idRole = QtCore.Qt.UserRole + 1 - - '''Role used to store the device vendor and model.''' - descRole = QtCore.Qt.UserRole + 2 - - def __init__(self, parent): - super(self.__class__, self).__init__(parent) - - # Use the description text to sort the items for now, because the - # _idRole holds tuples, and using them to sort doesn't work. - self.setSortRole(MeasurementDataModel.descRole) - - # Used in 'format_value()' to check against. - self.inf = float('inf') - - def format_unit(self, u): - units = { - sr.Unit.VOLT: 'V', - sr.Unit.AMPERE: 'A', - sr.Unit.OHM: u'\u03A9', - sr.Unit.FARAD: 'F', - sr.Unit.KELVIN: 'K', - sr.Unit.CELSIUS: u'\u00B0C', - sr.Unit.FAHRENHEIT: u'\u00B0F', - sr.Unit.HERTZ: 'Hz', - sr.Unit.PERCENTAGE: '%', - # sr.Unit.BOOLEAN - sr.Unit.SECOND: 's', - sr.Unit.SIEMENS: 'S', - sr.Unit.DECIBEL_MW: 'dBu', - sr.Unit.DECIBEL_VOLT: 'dBV', - # sr.Unit.UNITLESS - sr.Unit.DECIBEL_SPL: 'dB', - # sr.Unit.CONCENTRATION - sr.Unit.REVOLUTIONS_PER_MINUTE: 'rpm', - sr.Unit.VOLT_AMPERE: 'VA', - sr.Unit.WATT: 'W', - sr.Unit.WATT_HOUR: 'Wh', - sr.Unit.METER_SECOND: 'm/s', - sr.Unit.HECTOPASCAL: 'hPa', - sr.Unit.HUMIDITY_293K: '%rF', - sr.Unit.DEGREE: u'\u00B0', - sr.Unit.HENRY: 'H' - } - - return units.get(u, '') - - def format_mqflags(self, mqflags): - if sr.QuantityFlag.AC in mqflags: - return 'AC' - elif sr.QuantityFlag.DC in mqflags: - return 'DC' - else: - return '' - - def format_value(self, mag): - if mag == self.inf: - return u'\u221E' - return '{:f}'.format(mag) - - def getItem(self, device, channel): - '''Returns the item for the device + channel combination from the model, - or creates a new item if no existing one matches.''' - - # unique identifier for the device + channel - # TODO: isn't there something better? - uid = ( - device.vendor, - device.model, - device.serial_number(), - device.connection_id(), - channel.index - ) - - # find the correct item in the model - for row in range(self.rowCount()): - item = self.item(row) - rid = item.data(MeasurementDataModel._idRole) - rid = tuple(rid) # PySide returns a list - if uid == rid: - return item + import qtcompat + qtcompat.load_modules(args['pyside']) + QtCore = qtcompat.QtCore + QtGui = qtcompat.QtGui + import mainwindow - # nothing found, create a new item - desc = '{} {}, channel "{}"'.format( - device.vendor, device.model, channel.name) - - item = QtGui.QStandardItem() - item.setData(uid, MeasurementDataModel._idRole) - item.setData(desc, MeasurementDataModel.descRole) - self.appendRow(item) - self.sort(0) - return item - - @QtCore.Slot(object, object, object) - def update(self, device, channel, data): - '''Updates the data for the device (+channel) with the most recent - measurement from the given payload.''' - - item = self.getItem(device, channel) - - value, unit, mqflags = data - value_str = self.format_value(value) - unit_str = self.format_unit(unit) - mqflags_str = self.format_mqflags(mqflags) - - disp = ' '.join([value_str, unit_str, mqflags_str]) - item.setData(disp, QtCore.Qt.DisplayRole) - -class MultimeterDelegate(QtGui.QStyledItemDelegate): - '''Delegate to show the data items from a MeasurementDataModel.''' - - def __init__(self, parent, font): - '''Initializes the delegate. - - :param font: Font used for the description text, the value is drawn - with a slightly bigger and bold variant of the font. - ''' - - super(self.__class__, self).__init__(parent) - - self._nfont = font - self._bfont = QtGui.QFont(self._nfont) - - self._bfont.setBold(True) - if self._bfont.pixelSize() != -1: - self._bfont.setPixelSize(self._bfont.pixelSize() * 1.8) - else: - self._bfont.setPointSizeF(self._bfont.pointSizeF() * 1.8) - - fi = QtGui.QFontInfo(self._nfont) - self._nfontheight = fi.pixelSize() - - fm = QtGui.QFontMetrics(self._bfont) - r = fm.boundingRect('-XX.XXXXXX X XX') - self._size = QtCore.QSize(r.width() * 1.2, r.height() * 3.5) - - def sizeHint(self, option=None, index=None): - return self._size - - def paint(self, painter, options, index): - value = index.data(QtCore.Qt.DisplayRole) - desc = index.data(MeasurementDataModel.descRole) - - # description in the top left corner - painter.setFont(self._nfont) - p = options.rect.topLeft() - p += QtCore.QPoint(self._nfontheight, 2 * self._nfontheight) - painter.drawText(p, desc) - - # value in the center - painter.setFont(self._bfont) - r = options.rect.adjusted(self._nfontheight, 2.5 * self._nfontheight, - 0, 0) - painter.drawText(r, QtCore.Qt.AlignCenter, value) - -class EmptyMessageListView(QtGui.QListView): - '''List view that shows a message if the model im empty.''' - - def __init__(self, message, parent=None): - super(self.__class__, self).__init__(parent) - - self._message = message - - def paintEvent(self, event): - m = self.model() - if m and m.rowCount(): - super(self.__class__, self).paintEvent(event) - return - - painter = QtGui.QPainter(self.viewport()) - painter.drawText(self.rect(), QtCore.Qt.AlignCenter, self._message) - -class SigrokMeter(QtGui.QMainWindow): - '''The main window of the application.''' - - def __init__(self, context, drivers): - super(SigrokMeter, self).__init__() - - self.context = context - - self.delegate = MultimeterDelegate(self, self.font()) - self.model = MeasurementDataModel(self) - self.model.rowsInserted.connect(self.modelRowsInserted) - - self.setup_ui() - - self.thread = SamplingThread(self.context, drivers) - self.thread.measured.connect(self.model.update) - self.thread.error.connect(self.error) - self.thread.start() - - def setup_ui(self): - self.setWindowTitle('sigrok-meter') - # resizing the listView below will increase this again - self.resize(10, 10) - - p = os.path.abspath(os.path.dirname(__file__)) - p = os.path.join(p, 'sigrok-logo-notext.png') - self.setWindowIcon(QtGui.QIcon(p)) - - actionQuit = QtGui.QAction(self) - actionQuit.setText('&Quit') - actionQuit.setIcon(QtGui.QIcon.fromTheme('application-exit')) - actionQuit.setShortcut('Ctrl+Q') - actionQuit.triggered.connect(self.close) - - actionAbout = QtGui.QAction(self) - actionAbout.setText('&About') - actionAbout.setIcon(QtGui.QIcon.fromTheme('help-about')) - actionAbout.triggered.connect(self.show_about) - - 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.centralWidget().setContentsMargins(0, 0, 0, 0) - - def closeEvent(self, event): - self.thread.stop() - event.accept() - - @QtCore.Slot() - def show_about(self): - text = textwrap.dedent('''\ -
- sigrok-meter
- 0.1.0
- Using libsigrok {} (lib version {}).
- - http://www.sigrok.org
-
- This program comes with ABSOLUTELY NO WARRANTY;
- for details visit - - http://www.gnu.org/licenses/gpl.html -
- '''.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) - -if __name__ == '__main__': context = sr.Context_create() context.log_level = args['loglevel'] app = QtGui.QApplication([]) - s = SigrokMeter(context, args['drivers']) + s = mainwindow.MainWindow(context, args['drivers']) s.show() sys.exit(app.exec_())