From 22e24497a510c174e6de7718ad918a423d1973dd Mon Sep 17 00:00:00 2001 From: Andrej Rode Date: Thu, 9 Feb 2017 11:16:16 -0800 Subject: python: Initial commit of Python API Initial commit of the Python API using Boost.Python. Bind the MultiUSRP API for use in Python. Bindings intended to provide as complete coverage as possible. - Wrap most multi_usrp calls - Adding multi channel send/recv examples in examples/python - Adding setuptools support - Initial attempt at binding the UHD types and filters --- host/examples/python/pyuhd_tx_waveforms.py | 64 ++++++++++++++++++++++++++++++ 1 file changed, 64 insertions(+) create mode 100755 host/examples/python/pyuhd_tx_waveforms.py (limited to 'host/examples/python/pyuhd_tx_waveforms.py') diff --git a/host/examples/python/pyuhd_tx_waveforms.py b/host/examples/python/pyuhd_tx_waveforms.py new file mode 100755 index 000000000..d7d431f44 --- /dev/null +++ b/host/examples/python/pyuhd_tx_waveforms.py @@ -0,0 +1,64 @@ +#! /usr/bin/env python +# +# Copyright 2017 Ettus Research LLC +# +# 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 . +# + +import numpy as np +import uhd +import argparse + + +waveforms = { + "sine": lambda n, tone_offset, rate: np.exp(n * 2j * np.pi * tone_offset / rate), + "square": lambda n, tone_offset, rate: np.sign(waveforms["sine"](n, tone_offset, rate)), + "const": lambda n, tone_offset, rate: 1 + 1j, + "ramp": lambda n, tone_offset, rate: 2*(n*(tone_offset/rate) - np.floor(float(0.5 + n*(tone_offset/rate)))) +} + + +def parse_args(): + parser = argparse.ArgumentParser() + parser.add_argument("-a", "--args", default="", type=str) + parser.add_argument( + "-w", "--waveform", default="sine", choices=waveforms.keys(), type=str) + parser.add_argument("-f", "--freq", type=float, required=True) + parser.add_argument("-r", "--rate", default=1e6, type=float) + parser.add_argument("-d", "--duration", default=5.0, type=float) + parser.add_argument("-c", "--channels", default=0, nargs="+", type=int) + parser.add_argument("-g", "--gain", type=int, default=10) + parser.add_argument("--wave-freq", default=1e4, type=float) + parser.add_argument("--wave-ampl", default=0.3, type=float) + return parser.parse_args() + + +def main(): + args = parse_args() + usrp = uhd.multi_usrp(args.args) + if not isinstance(args.channels, list): + args.channels = [args.channels] + data = np.array( + map(lambda n: args.wave_ampl * waveforms[args.waveform](n, args.wave_freq, args.rate), + np.arange( + int(10 * np.floor(args.rate / args.wave_freq)), + dtype=np.complex64)), + dtype=np.complex64) # One period + + usrp.send_waveform(data, args.duration, args.freq, args.rate, + args.channels, args.gain) + + +if __name__ == "__main__": + main() -- cgit v1.2.3