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 import signal
  28
  29 default_drivers = [('demo:analog_channels=4', 'samplerate=4')]
  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=None,
  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     signal.signal(signal.SIGINT, signal.SIG_DFL)
  91
  92     args = parse_cli()
  93
  94     import qtcompat
  95     qtcompat.load_modules(args.pyside)
  96     QtCore = qtcompat.QtCore
  97     QtGui = qtcompat.QtGui
  98     import mainwindow
  99
 100     app = QtGui.QApplication([])
 101
 102     # Initialize modules that need a QApplication to exist.
 103     import settings
 104     settings.init()
 105     import icons
 106     icons.load_icons()
 107
 108     context = sr.Context_create()
 109
 110     if args.loglevel != None:
 111         try:
 112             loglevel = sr.LogLevel.get(args.loglevel)
 113             settings.logging.level.setValue(loglevel)
 114         except:
 115             sys.exit('Error: invalid log level.')
 116
 117     s = mainwindow.MainWindow(context, args.drivers)
 118     s.show()
 119
 120     sys.exit(app.exec_())