#!/usr/bin/env python
# LeCrunch
# Copyright (C) 2010 Anthony LaTorre
#
# 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
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see .
"""
Save/load settings from a LeCroy DSO oscilloscope. Settings are saved as a
python dictionary in two ways: pickled to a stand-alone file, or attached as
attributes to a hdf5 file. Settings can be loaded from either source.
Examples:
Save/load settings to/from a pickled dictionary.
>>> config.py save mysettings.pkl
>>> config.py load mysettings.pkl
sending TIME_DIV
sending COMM_FORMAT
...
Save/load settings from a dataset.
>>> fetch.py test.hdf5 -n 100
Saving event: 1000
Completed 1000 events in 15.302 seconds.
Averaged 0.01530 seconds per acquisition.
Wrote to file 'test.hdf5'.
>>> settings.py load test.hdf5
sending TIME_DIV
sending COMM_FORMAT
...
"""
commands = ['TIME_DIV', 'COMM_FORMAT', 'COMM_HEADER', 'COMM_ORDER'] + \
['TRIG_DELAY', 'TRIG_SELECT', 'TRIG_MODE', 'TRIG_PATTERN'] + \
['C%i:COUPLING' % i for i in range(1,5)] + \
['C%i:VOLT_DIV' % i for i in range(1,5)] + \
['C%i:OFFSET' % i for i in range(1,5)] + \
['C%i:TRIG_COUPLING' % i for i in range(1,5)] + \
['C%i:TRIG_LEVEL' % i for i in range(1,5)] + \
['C%i:TRIG_SLOPE' % i for i in range(1,5)] + \
['C%i:TRACE' % i for i in range(1,5)]
def get_settings(scope):
settings = {}
for command in commands:
scope.send(command + '?')
settings[command] = scope.recv().strip()
scope.check_last_command()
return settings
def set_settings(scope, settings):
for command, setting in settings.items():
print 'sending %s' % command
scope.send(setting)
scope.check_last_command()
if __name__ == '__main__':
import sys
import optparse
import pickle
import h5py
usage = '%prog filename'
parser = optparse.OptionParser(usage)
options, args = parser.parse_args()
if len(args) < 2:
sys.exit(parser.format_help())
from scope import Scope
scope = Scope('scope01.hep.upenn.edu')
scope.connect()
scope.clear()
if args[0] == 'save':
settings = get_settings(scope)
for command, setting in settings.items():
print '%s, %s' % (command, setting)
f = open(args[1], 'wb')
pickle.dump(settings, f)
f.close()
elif args[0] == 'load':
try:
f = h5py.File(args[1], 'r')
settings = dict(f.attrs)
f.close()
except h5py.h5e.LowLevelIOError:
f = open(args[1], 'r')
settings = pickle.load(f)
f.close()
set_settings(scope, settings)
else:
raise Exception('unrecognized command %s' % args[0])