Diff: scripts/ad4110_data_capture.py

Revision:

0:6ca37a8f8ba9

diff -r 000000000000 -r 6ca37a8f8ba9 scripts/ad4110_data_capture.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/scripts/ad4110_data_capture.py	Wed Jul 27 17:04:15 2022 +0530
@@ -0,0 +1,167 @@
+# @file    ad4110_data_capture.py
+# @brief   Data capturing for the AD4110 device
+# Copyright (c) 2022 Analog Devices, Inc.
+# All rights reserved.
+# 
+# This software is proprietary to Analog Devices, Inc. and its licensors.
+# By using this software you agree to the terms of the associated
+# Analog Devices Software License Agreement.
+#
+
+import numpy
+from serial import Serial
+from time import sleep
+from pynput import keyboard
+import os
+import csv
+from adi.ad4110 import *
+import math
+
+# Global variables
+line = 0
+writer = 0
+run_continuous = False
+iterations = 0
+data_capture_abort = False
+chn_count = 0
+data_list = []
+
+def key_press_event(key):
+    global data_capture_abort
+    data_capture_abort = True
+
+def init_data_capture():
+    global device
+    global data_list
+    global chn_count
+    global listener
+    global samples_block
+
+    ######## User configuration ##########
+    # Configure the backend for PC to IIOD interface
+    uri = "serial:COM17,230400"  # For UART, baud rate must be same as set in the FW. COM port is physical Or VCOM.
+    device_name = "ad4110"      # Name of the device must be same as set in the FW.
+    ######################################
+
+    # Create an IIO device context
+    device = ad4110(uri, device_name)
+    device._ctx.set_timeout(100000)
+
+    ######## User configuration ##########
+    # Channels to be captured e.g. [0]: 1chn, [0,1]: 2chns, [0,1,2,3]: 4chns
+    device.rx_enabled_channels = [0,1,2,3]
+
+    # The block of samples to be captured. Total samples are received in multiple iterations or blocks
+    samples_block = 400    # The samples needs to be captured in smaller blocks due to limitations
+                           # of buffer size (RAM) in the firmware and IIO client timeout factor.
+                           # Request to capture samples more than buffer size, will be ignored by firmware.
+                           # Large time taken to read the samples from device, may cause timeout on IIO client.
+    ######################################
+
+    # Get the channels count from user
+    chn_count = len(device.rx_enabled_channels)
+
+    # Store the rx buffer size and rx data type based on input channels
+    device.rx_buffer_size = samples_block    # Size of the IIO buffer (buffer is submitted during call to rx() method)
+    device._rx_data_type = np.int32     # size of sample
+
+    listener = keyboard.Listener(on_press=key_press_event)
+    listener.start()
+
+def read_user_inputs():
+    global iterations
+    global run_continuous
+    global device
+    global samples_count
+
+    samples_count = int(input("Enter the number of samples to be captured \n\
+                                0: Unlimited \n\
+                                <50-1000000>: "))
+
+    if (samples_count == 0):
+        run_continuous = True
+    else:
+        run_continuous = False
+        if (samples_count <= samples_block):
+            device.rx_buffer_size = samples_count
+            iterations = 1
+        else:
+            iterations = math.ceil(samples_count / samples_block)
+
+def init_data_logger():
+    global writer
+    global chn_count
+    file_name = "adc_data_capture.csv"
+    current_dir = os.getcwd()
+    output_file = os.path.join(current_dir, file_name)
+    result_file = open(output_file, 'w', newline="")
+    writer = csv.writer(result_file)
+    row = []
+    # Write the channels list header
+    for chn in range(0,chn_count):
+        item = "Ch {}".format(chn)
+        row.insert(chn, item)
+    writer.writerow(row)
+
+
+def read_buffered_data():
+    global line
+    global writer
+    global device
+
+    # Receive data from device
+    data = device.rx()
+    if (line == 0):
+        print("Data capture started >>")
+    print("."*line, end="\r")
+
+    if (chn_count == 1):
+        # Convert 1-D array to 2-D array
+        data_arr = np.reshape(data, (-1,1))
+    else:
+        # Transpose data from N-D data array
+        data_arr = np.transpose(data)
+
+    writer.writerows(data_arr)
+
+    line = line+1
+    if (line == 100):
+        line = 1
+        print("\n", end="\r")
+
+def do_data_capture():
+    global iterations
+    global run_continuous
+    global data_capture_abort
+    done = False
+    if (run_continuous == True):
+        print("Press any key to stop data capture..")
+        sleep(2)
+        data_capture_abort = False
+        while not data_capture_abort:
+            read_buffered_data()
+    else:
+        for val in range(0,iterations):
+            read_buffered_data()
+
+    print("\r\nData capture finished\r\n")
+
+def exit():
+    global listener
+    global writer
+    global device
+
+    # Delete the objects
+    del listener
+    del writer
+    del device
+
+def main():
+    init_data_capture()
+    init_data_logger()
+    read_user_inputs()
+    do_data_capture()
+    exit()
+
+if __name__ == "__main__":
+    main()
\ No newline at end of file