Acquire data from two DMMs at the same time.

[sigrok-meter.git] / sigrok-meter

diff --git a/sigrok-meter b/sigrok-meter
index 3b018bd5b7be4772570abd82276b4d5c2ae3d39d..caa6fdd1dea3821cee17808765aaa3cad51edc3f 100755 (executable)
--- a/sigrok-meter
+++ b/sigrok-meter
@@ -19,10 +19,85 @@
 ## Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
 ##
 
-from gi.repository import Gtk
+from multiprocessing import Process, Queue
+from gi.repository import Gtk, GObject
+from sigrok.core import Context, Driver, Device, Session, Packet, Log
+from sigrok.core import lowlevel as ll
 
-win = Gtk.Window()
-win.connect("delete-event", Gtk.main_quit)
-win.show_all()
-Gtk.main()
+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.mqflags, ""
+            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):
+                dev = "%s" % device.vendor
+                val = "%f%s%s" % (data[i], unit_str, mqflags_str)
+                queue.put((dev, val))
+
+    # log = Log()
+    # log.level = Log.SPEW
+    context = Context()
+    drivers_to_use = ['voltcraft-vc820', 'victor-dmm']
+    drivers = [context.drivers[d] for d in drivers_to_use]
+    devices = [d.scan()[0] for d in drivers]
+    for dev in devices:
+        dev.limit_samples = 1000
+    session = Session(context)
+    for dev in devices:
+        session.add_device(dev)
+    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.value_label2 = self.builder.get_object("value_label2")
+        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):
+        try:
+            t = self.queue.get_nowait()
+            l = self.value_label if t[0] != "Victor" else self.value_label2
+            l.set_text("%s: %s" % (t[0], t[1]))
+        except:
+            pass
+        GObject.timeout_add(100, self.update_label_if_needed)
+
+    def on_quit(self, *args):
+        Gtk.main_quit(*args)
+
+    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()
+
+if __name__ == '__main__':
+        s = SigrokMeter()
+        process = Process(target=init_and_run, args=(s.queue,))
+        process.start()
+        Gtk.main()
+        process.terminate()