sigrok-meter

   1 #!/usr/bin/env python
   2
   3 ##
   4 ## This file is part of the sigrok-meter project.
   5 ##
   6 ## Copyright (C) 2013 Uwe Hermann <uwe@hermann-uwe.de>
   7 ##
   8 ## This program is free software; you can redistribute it and/or modify
   9 ## it under the terms of the GNU General Public License as published by
  10 ## the Free Software Foundation; either version 2 of the License, or
  11 ## (at your option) any later version.
  12 ##
  13 ## This program is distributed in the hope that it will be useful,
  14 ## but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 ## GNU General Public License for more details.
  17 ##
  18 ## You should have received a copy of the GNU General Public License
  19 ## along with this program; if not, write to the Free Software
  20 ## Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  21 ##
  22
  23 import argparse
  24 from multiprocessing import Process, Queue
  25 from gi.repository import Gtk, GObject
  26 import re
  27 import sigrok.core as sr
  28 import sys
  29 import textwrap
  30
  31 default_drivers = [('demo', {'analog_channels': 1})]
  32 default_loglevel = sr.LogLevel.WARN
  33
  34 def init_and_run(queue, drivers, loglevel):
  35     def datafeed_in(device, packet):
  36         if packet.type == sr.PacketType.ANALOG:
  37             data = packet.payload.data
  38             unit, unit_str = packet.payload.unit, ""
  39             if unit == sr.Unit.VOLT:
  40                 unit_str = " V"
  41             elif unit == sr.Unit.OHM:
  42                 unit_str = " Ohm"
  43             elif unit == sr.Unit.AMPERE:
  44                 unit_str = " A"
  45             mqflags, mqflags_str = packet.payload.mq_flags, ""
  46             if sr.QuantityFlag.AC in mqflags:
  47                 mqflags_str = " AC"
  48             elif sr.QuantityFlag.DC in mqflags:
  49                 mqflags_str = " DC"
  50             for i in range(packet.payload.num_samples):
  51                 dev = "%s %s" % (device.vendor, device.model)
  52                 val = "%f%s%s" % (data[0][i], unit_str, mqflags_str)
  53                 queue.put((dev, val))
  54
  55     context = sr.Context_create()
  56     context.log_level = loglevel
  57
  58     devices = []
  59     for name, options in drivers:
  60         try:
  61             dr = context.drivers[name]
  62             devices.append(dr.scan(**options)[0])
  63         except:
  64             print('error getting device for driver "{}", skipping'.format(name))
  65
  66     if not devices:
  67         print('no devices found')
  68         return
  69
  70     session = context.create_session()
  71     for dev in devices:
  72         session.add_device(dev)
  73         dev.open()
  74     session.add_datafeed_callback(datafeed_in)
  75     session.start()
  76     session.run()
  77     session.stop()
  78
  79 class SigrokMeter:
  80     def __init__(self):
  81         self.builder = Gtk.Builder()
  82         self.builder.add_from_file("sigrok-meter.glade")
  83         self.builder.connect_signals(self)
  84         self.value_label = self.builder.get_object("value_label")
  85         self.value_label2 = self.builder.get_object("value_label2")
  86         self.win = self.builder.get_object("mainwindow")
  87         self.win.show_all()
  88         self.queue = Queue()
  89         GObject.timeout_add(100, self.update_label_if_needed)
  90
  91     def update_label_if_needed(self):
  92         try:
  93             t = self.queue.get_nowait()
  94             l = self.value_label if t[0] != "Victor" else self.value_label2
  95             l.set_text("%s: %s" % (t[0], t[1]))
  96         except:
  97             pass
  98         GObject.timeout_add(100, self.update_label_if_needed)
  99
 100     def on_quit(self, *args):
 101         Gtk.main_quit(*args)
 102
 103     def on_about(self, action):
 104         about = self.builder.get_object("aboutdialog")
 105         context = sr.Context_create()
 106         sr_pkg = context.package_version
 107         sr_lib = context.lib_version
 108         s = "Using libsigrok %s (lib version %s)." % (sr_pkg, sr_lib)
 109         about.set_comments(s)
 110         about.run()
 111         about.hide()
 112
 113 def parse_cli():
 114     parser = argparse.ArgumentParser(
 115         description='Simple sigrok GUI for multimeters and dataloggers.',
 116         epilog=textwrap.dedent('''\
 117             The DRIVER string is the same as for sigrok-cli(1).
 118
 119             examples:
 120
 121               %(prog)s --driver tecpel-dmm-8061-ser:conn=/dev/ttyUSB0
 122
 123               %(prog)s --driver uni-t-ut61e:conn=1a86.e008
 124         '''),
 125         formatter_class=argparse.RawDescriptionHelpFormatter)
 126
 127     parser.add_argument('-d', '--driver',
 128         action='append',
 129         help='The driver to use')
 130     parser.add_argument('-l', '--loglevel',
 131         type=int,
 132         help='Set loglevel (5 is most verbose)')
 133     args = parser.parse_args()
 134
 135     result = {
 136         'drivers': default_drivers,
 137         'loglevel': default_loglevel
 138     }
 139
 140     if args.driver:
 141         result['drivers'] = []
 142         for d in args.driver:
 143             m = re.match('(?P<name>[^:]+)(?P<opts>(:[^:=]+=[^:=]+)*)', d)
 144             if not m:
 145                 sys.exit('error parsing option "{}"'.format(d))
 146
 147             opts = m.group('opts').split(':')[1:]
 148             opts = [tuple(kv.split('=')) for kv in opts]
 149             opts = dict(opts)
 150
 151             result['drivers'].append((m.group('name'), opts))
 152
 153     if args.loglevel != None:
 154         try:
 155             result['loglevel'] = sr.LogLevel.get(args.loglevel)
 156         except:
 157             sys.exit('error: invalid log level')
 158
 159     return result
 160
 161 if __name__ == '__main__':
 162     args = parse_cli()
 163     s = SigrokMeter()
 164     process = Process(target=init_and_run,
 165             args=(s.queue, args['drivers'], args['loglevel']))
 166     process.start()
 167     Gtk.main()
 168     process.terminate()
 169