Better driver string handling and better error messages.

[sigrok-meter.git] / sigrok-meter

diff --git a/sigrok-meter b/sigrok-meter
index d53a288a2eb7d560943df6aee9e8461ea3f53681..72b8d49ee832d48a167aafec866ed3020789bbb9 100755 (executable)
--- a/sigrok-meter
+++ b/sigrok-meter
@@ -1,8 +1,10 @@
-#!/usr/bin/python3
+#!/usr/bin/env python
+
 ##
 ## This file is part of the sigrok-meter project.
 ##
 ## Copyright (C) 2013 Uwe Hermann <uwe@hermann-uwe.de>
+## Copyright (C) 2014 Jens Steinhauser <jens.steinhauser@gmail.com>
 ##
 ## 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
@@ -19,77 +21,71 @@
 ## Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
 ##
 
-from multiprocessing import Process, Queue
-from gi.repository import Gtk, GObject
-from sigrok.core import Context, Driver, Device, Session, Packet
-from sigrok.core import lowlevel as ll
-
-def init_and_run(queue):
-    def datafeed_in(device, packet):
-        if packet.type == Packet.ANALOG:
-            data = packet.payload.data
-            unit, unit_str = packet.payload.unit, ""
-            if unit == ll.SR_UNIT_VOLT:
-                unit_str = " V"
-            elif unit == ll.SR_UNIT_OHM:
-                unit_str = " Ohm"
-            elif unit == ll.SR_UNIT_AMPERE:
-                unit_str = " A"
-            mqflags, mqflags_str = packet.payload.unit, ""
-            if mqflags & ll.SR_MQFLAG_AC:
-                mqflags_str = " AC"
-            elif mqflags & ll.SR_MQFLAG_DC:
-                mqflags_str = " DC"
-            for i in range(packet.payload.num_samples):
-                queue.put("%f%s%s" % (data[i], unit_str, mqflags_str))
-
-    # ll.sr_log_loglevel_set(5)
-    context = Context()
-    driver = context.drivers['voltcraft-vc820']
-    device = driver.scan()[0]
-    device.limit_samples = 1000
-    session = Session(context)
-    session.add_device(device)
-    session.add_callback(datafeed_in)
-    session.start()
-    session.run()
-    session.stop()
-
-class SigrokMeter:
-    def __init__(self):
-        self.builder = Gtk.Builder()
-        self.builder.add_from_file("sigrok-meter.glade")
-        self.builder.connect_signals(self)
-        self.value_label = self.builder.get_object("value_label")
-        self.win = self.builder.get_object("mainwindow")
-        self.win.show_all()
-        self.queue = Queue()
-        GObject.timeout_add(100, self.update_label_if_needed)
-
-    def update_label_if_needed(self):
+import argparse
+import sigrok.core as sr
+import sys
+import textwrap
+
+default_drivers = ['demo:analog_channels=4']
+default_loglevel = sr.LogLevel.WARN
+
+def parse_cli():
+    parser = argparse.ArgumentParser(
+        description='Simple sigrok GUI for multimeters and dataloggers.',
+        epilog=textwrap.dedent('''\
+            The DRIVER string is the same as for sigrok-cli(1).
+
+            examples:
+
+              %(prog)s --driver tecpel-dmm-8061-ser:conn=/dev/ttyUSB0
+
+              %(prog)s --driver uni-t-ut61e:conn=1a86.e008
+        '''),
+        formatter_class=argparse.RawDescriptionHelpFormatter)
+
+    parser.add_argument('-d', '--driver',
+        action='append',
+        help='The driver to use')
+    parser.add_argument('-l', '--loglevel',
+        type=int,
+        help='Set loglevel (5 is most verbose)')
+    parser.add_argument('--pyside',
+        action='store_true',
+        default=False,
+        help='Force use of PySide (default is to use PyQt4)')
+    args = parser.parse_args()
+
+    result = {
+        'drivers': default_drivers,
+        'loglevel': default_loglevel,
+        'pyside': args.pyside
+    }
+
+    if args.driver:
+        result['drivers'] = args.driver
+
+    if args.loglevel != None:
         try:
-            t = self.queue.get_nowait()
-            self.value_label.set_text(t)
+            result['loglevel'] = sr.LogLevel.get(args.loglevel)
         except:
-            pass
-        GObject.timeout_add(100, self.update_label_if_needed)
-
-    def on_quit(self, *args):
-        Gtk.main_quit(*args)
+            sys.exit('error: invalid log level')
 
-    def on_about(self, action):
-        about = self.builder.get_object("aboutdialog")
-        sr_pkg = ll.sr_package_version_string_get()
-        sr_lib = ll.sr_lib_version_string_get()
-        s = "Using libsigrok %s (lib version %s)." % (sr_pkg, sr_lib)
-        about.set_comments(s)
-        about.run()
-        about.hide()
+    return result
 
 if __name__ == '__main__':
-        s = SigrokMeter()
-        process = Process(target=init_and_run, args=(s.queue,))
-        process.start()
-        Gtk.main()
-        process.terminate()
+    args = parse_cli()
+
+    import qtcompat
+    qtcompat.load_modules(args['pyside'])
+    QtCore = qtcompat.QtCore
+    QtGui = qtcompat.QtGui
+    import mainwindow
+
+    context = sr.Context_create()
+    context.log_level = args['loglevel']
+
+    app = QtGui.QApplication([])
+    s = mainwindow.MainWindow(context, args['drivers'])
+    s.show()
 
+    sys.exit(app.exec_())