python: added iio_readdev with Python bindings.

Implemented the iio_readdev module with Python bindings.

Signed-off-by: Cristi Iacob <[email protected]>

bindings/python/examples/iio_readdev.py

@@ -0,0 +1,161 @@
+#!/usr/bin/env python
+#
+# Copyright (C) 2020 Analog Devices, Inc.
+# Author: Cristian Iacob <[email protected]>
+#
+# 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 2
+# 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, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+
+import iio
+import sys
+import signal
+import argparse
+
+parser = argparse.ArgumentParser(description='iio_readdev')
+parser.add_argument('-n', '--network', type=str, metavar='',
+                    help='Use the network backend with the provided hostname.')
+parser.add_argument('-u', '--uri', type=str, metavar='',
+                    help='Use the context with the provided URI.')
+parser.add_argument('-b', '--buffer-size', type=int, metavar='',
+                    help='Size of the capture buffer. Default is 256.')
+parser.add_argument('-s', '--samples', type=int, metavar='',
+                    help='Number of samples to capture, 0 = infinite. Default is 0.')
+parser.add_argument('-T', '--timeout', type=int, metavar='',
+                    help='Buffer timeout in milliseconds. 0 = no timeout')
+parser.add_argument('-a', '--auto', action='store_true',
+                    help='Scan for available contexts and if only one is available use it.')
+parser.add_argument('device', type=str, nargs=1)
+parser.add_argument('channel', type=str, nargs='*')
+
+arg_ip = ""
+arg_uri = ""
+scan_for_context = False
+buffer_size = 256
+num_samples = 0
+timeout = 0
+device_name = None
+channels = None
+
+
+def read_arguments():
+    global arg_ip, arg_uri, scan_for_context, buffer_size, num_samples, timeout, device_name, channels
+
+    args = parser.parse_args()
+
+    if args.network is not None:
+        arg_ip = str(args.network)
+
+    if args.uri is not None:
+        arg_uri = str(args.uri)
+
+    if args.auto is True:
+        scan_for_context = True
+
+    if args.buffer_size is not None:
+        buffer_size = int(args.buffer_size)
+
+    if args.samples is not None:
+        num_samples = int(args.samples)
+
+    if args.timeout is not None:
+        timeout = int(args.timeout)
+
+    device_name = args.device[0]
+    channels = args.channel
+
+
+def create_context(scan_for_context, arg_uri, arg_ip):
+    ctx = None
+
+    try:
+        if scan_for_context:
+            contexts = iio.scan_contexts()
+            if len(contexts) == 0:
+                sys.stderr.write("No IIO context found.\n")
+                exit(1)
+            elif len(contexts) == 1:
+                uri, _ = contexts.popitem()
+                ctx = iio.Context(_context=uri)
+            else:
+                print("Multiple contexts found. Please select one using --uri!")
+
+                for uri, _ in contexts:
+                    print(uri)
+        elif arg_uri != "":
+            ctx = iio.Context(_context=arg_uri)
+        elif arg_ip != "":
+            ctx = iio.NetworkContext(arg_ip)
+        else:
+            ctx = iio.Context()
+    except FileNotFoundError:
+        sys.stderr.write('Unable to create IIO context\n')
+        exit(1)
+
+    return ctx
+
+
+def keyboard_interrupt_handler(signal, frame):
+    sys.exit(0)
+
+
+signal.signal(signal.SIGINT, keyboard_interrupt_handler)
+
+
+def read_data(buffer, num_samples):
+    if buffer is None:
+        sys.stderr.write('Unable to create buffer!\n')
+        exit(1)
+
+    while True:
+        buffer.refill()
+        samples = buffer.read()
+
+        if num_samples > 0:
+            sys.stdout.buffer.write(samples[:min(num_samples, len(samples))])
+            num_samples -= min(num_samples, len(samples))
+
+            if num_samples == 0:
+                break
+        else:
+            sys.stdout.buffer.write(bytes(samples))
+
+
+def main():
+    read_arguments()
+
+    ctx = create_context(scan_for_context, arg_uri, arg_ip)
+
+    if timeout >= 0:
+        ctx.set_timeout(timeout)
+
+    dev = ctx.find_device(device_name)
+
+    if dev is None:
+        sys.stderr.write('Device %s not found!\n' % device_name)
+        exit(1)
+
+    if len(channels) == 0:
+        for channel in dev.channels:
+            channel.enabled = True
+    else:
+        for channel_idx in channels:
+            dev.channels[int(channel_idx)].enabled = True
+
+    buffer = iio.Buffer(dev, buffer_size)
+
+    read_data(buffer, num_samples)
+
+
+if __name__ == '__main__':
+    main()