sigrok-meter

   1 #!/usr/bin/env python
   2 ##
   3 ## This file is part of the sigrok-meter project.
   4 ##
   5 ## Copyright (C) 2013 Uwe Hermann <uwe@hermann-uwe.de>
   6 ## Copyright (C) 2014 Jens Steinhauser <jens.steinhauser@gmail.com>
   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 import sigrok.core as sr
  25 import sys
  26 import textwrap
  27
  28 default_drivers = [('demo:analog_channels=4', 'samplerate=4')]
  29 default_loglevel = 2
  30
  31 def parse_cli():
  32     parser = argparse.ArgumentParser(
  33         description='Simple sigrok GUI for multimeters and dataloggers.',
  34         epilog=textwrap.dedent('''\
  35             The DRIVER string is the same as for sigrok-cli(1). Multiple
  36             DRIVER and CONFIG items can be supplied. The nth CONFIG is applied
  37             to the nth DRIVER. If there are more drivers than configs, the
  38             remaining drivers use the default configuration.
  39
  40             Examples:
  41
  42               %(prog)s --driver tecpel-dmm-8061-ser:conn=/dev/ttyUSB0
  43
  44               %(prog)s --driver uni-t-ut61e:conn=1a86.e008
  45
  46               %(prog)s --driver demo:analog_channels=1 \\
  47                        --config samplerate=10
  48
  49               %(prog)s --driver voltcraft-k204:conn=/dev/ttyUSB0 \\
  50                        --driver uni-t-ut61d:conn=1a86.e008 \\
  51                        --driver uni-t-ut61e-ser:conn=/dev/ttyUSB1
  52         '''),
  53         formatter_class=argparse.RawDescriptionHelpFormatter)
  54
  55     parser.add_argument('-d', '--driver',
  56         action='append',
  57         default=[],
  58         help='The driver to use')
  59     parser.add_argument('-c', '--config',
  60         action='append',
  61         default=[],
  62         help='Specify device configuration options')
  63     parser.add_argument('-l', '--loglevel',
  64         type=int,
  65         default=default_loglevel,
  66         help='Set loglevel (5 is most verbose)')
  67     parser.add_argument('--pyside',
  68         action='store_true',
  69         default=False,
  70         help='Force use of PySide (default is to use PyQt4)')
  71     args = parser.parse_args()
  72
  73     if len(args.config) > len(args.driver):
  74         sys.exit('Error: More configurations than drivers given.')
  75
  76     # Merge drivers and configurations into a list of tuples.
  77     setattr(args, 'drivers', [])
  78     if not args.driver:
  79         args.drivers = default_drivers
  80         sys.stderr.write('No driver given, using demo driver.\n')
  81     if args.driver:
  82         args.config.extend([''] * (len(args.driver) - len(args.config)))
  83         args.drivers = zip(args.driver, args.config)
  84     del args.driver
  85     del args.config
  86
  87     return args
  88
  89 if __name__ == '__main__':
  90     args = parse_cli()
  91
  92     import qtcompat
  93     qtcompat.load_modules(args.pyside)
  94     QtCore = qtcompat.QtCore
  95     QtGui = qtcompat.QtGui
  96     import mainwindow
  97
  98     context = sr.Context_create()
  99     try:
 100         loglevel = sr.LogLevel.get(args.loglevel)
 101         context.log_level = loglevel
 102     except:
 103         sys.exit('Error: invalid log level.')
 104
 105     app = QtGui.QApplication([])
 106     s = mainwindow.MainWindow(context, args.drivers)
 107     s.show()
 108
 109     sys.exit(app.exec_())