Skip to content

Instantly share code, notes, and snippets.

Last active November 2, 2021 11:04
What would you like to do?
Load waveform and save it as csv file for RIGOL MSO5000 series oscilloscope
import visa
import argparse
import csv
def load_waveform(chidx, points_request):
inst.write(f':WAV:SOUR CHAN{chidx}')
inst.write(':WAV:MODE RAW')
inst.write(f':WAV:POIN {points_request}')
inst.write(':WAV:FORM BYTE')
preample = inst.query(':WAV:PRE?').split(',')
points = int(preample[2])
xinc = float(preample[4])
xorg = float(preample[5])
xref = float(preample[6])
yinc = float(preample[7])
yorg = float(preample[8])
yref = float(preample[9])
print(f'loading CH{chidx} {points}pts')
data_bin = inst.query_binary_values('WAV:DATA?', datatype='B', container=bytes)
t = [(float(i) - xref)*xinc + xorg for i in range(points)]
v = [(float(byte_data) - yref)*yinc + yorg for byte_data in data_bin]
return t,v
argparser = argparse.ArgumentParser()
argparser.add_argument('-a', '--address', required=True, help='VISA address like "TCPIP::{ipaddress}::INSTR"')
argparser.add_argument('-o', '--output', help='Output file name (default: "waveform.csv")', default='waveform.csv')
argparser.add_argument('-p', '--points', type=int, help='Points of the data to be load', default=1000)
argparser.add_argument('-c', '--chlist', help='Specify channels which waveforms will be load from like "-c 1,2,3,4"', default='1')
args = argparser.parse_args()
chlist = [int(x) for x in args.chlist.split(',')]
inst = visa.ResourceManager().open_resource(args.address)
load_data = {}
for chidx in chlist:
t, v = load_waveform(chidx, args.points)
if 'time' not in load_data.keys():
load_data['time'] = t
load_data[f'CH{chidx}'] = v
with open(args.output, 'w', newline='') as f:
writer = csv.writer(f, delimiter=',')
Copy link

idt12312 commented Jun 9, 2020


How to use

The following command means that the script loads 10000pts waveforms from the ch1,2 of the oscilloscope whose visa address is TCPIP:: and saves them to test.csv.
$ python -a TCPIP:: -c 1,2 -o test.csv -p 10000

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment