Python implementation of Arbitrary Waveform Generator and oscilloscope

Moku's Arbitrary Waveform Generator (AWG) can be deployed within Python to drive output signals. At the same time, the Python AWG can be used as an oscilloscope to view the output signal. In order to do so, you would need to loop back output 1 to input 1.

This is implemented in the Python below :

#
# Moku example: Plotting Oscilloscope with Artibrary Waveform Generator
#
# This example demonstrates how you can configure the ArbitraryWaveformGenerator instrument,
# and view triggered time-voltage data frames in real-time.
#
# (c) 2023 Liquid Instruments Pty. Ltd.
#
from moku.instruments import ArbitraryWaveformGenerator, MultiInstrument, Oscilloscope

import numpy as np
import matplotlib
import matplotlib.pyplot as plt

# Launch MiM and connect to your device through IP
m = MultiInstrument('192.168.###.###', force_connect=True, platform_id=4)
awg = m.set_instrument(1, ArbitraryWaveformGenerator)
awg.set_defaults()
osc = m.set_instrument(2, Oscilloscope)
osc.set_defaults()
connections = [dict(source="Slot1OutA", destination="Slot2InB"),
               dict(source="Slot1OutA", destination="Output1"),
               dict(source="Slot2OutA", destination="Slot2InA")]
m.set_connections(connections=connections)

current_sine = np.zeros(1000)

g = 0
for x in current_sine:
    current_sine[g] = np.sin((float(g)*2*np.pi/len(current_sine)))
    g +=1

try:
    # Generate an output sinewave on Channel 2, 500mVpp, 1MHz, 0V offset
    awg.generate_waveform(channel=1, frequency=1e6, amplitude = 0.5, sample_rate='Auto', lut_data=list(current_sine), interpolation = True) 


    # Trigger on input Channel 2, rising edge, 0V with 0.1V hysteresis)
    osc.set_trigger(source='ChannelA', type="Edge", level=0, hysteresis=0.1)

    # View +-5usec, i.e. trigger in the centre
    osc.set_timebase(-5e-6, 5e-6)

    osc.generate_waveform(1, 'Sine', amplitude=0.5, frequency=1e6, symmetry=0.0)


    # Get initial data frame to set up plotting parameters. This can be done once
    # if we know that the axes aren't going to change (otherwise we'd do
    # this in the loop)
    plt_dt = osc.get_data()

    # Set up the plotting parameters
    plt.ion()
    plt.show()
    plt.grid(True)
    plt.ylim([-1,1])
    plt.xlim([plt_dt['time'][0], plt_dt['time'][-1]])

    line1, = plt.plot([])
    line2, = plt.plot([])

    ax=plt.gca()

    # This loops continuously updates the plot with new data
    while True:
        # Get new data
        data = osc.get_data()

        # Update the plot
        line1.set_ydata(data['ch1'])
        line1.set_xdata(data['time'])
        line2.set_ydata(data['ch2'])
        line2.set_xdata(data['time'])
        plt.pause(0.001)

except Exception as e:
        print(f'Exception occured: {e}')

finally:
    # Close the connection to the Moku device
    # This ensures network resources and released correctly
    m.relinquish_ownership()