set_wave_and_aai.py

# Simple script to write random values to the waveform and aai records
from epics import PV
import random

PV_ROOT = "SIMPLE:"


def generate_data():
    data = [1, 5, 10, 3, 0]
    random.shuffle(data)
    return data

def generate_char_data():
    data = [chr(97), chr(98), chr(99), chr(100), chr(101)]
    random.shuffle(data)
    return "".join(data)

def generate_str_data():
    data = ["hello", "goodbye", "ciao", "hej", "shrug"]
    random.shuffle(data)
    return data


def update_record(name, data, as_str=False):
    print(name)
    p = PV(name)
    p.put(data, wait=True)
    print(p.get(as_string=as_str))


update_record(f"{PV_ROOT}WAVE", generate_data())
update_record(f"{PV_ROOT}CHARWAVE", generate_char_data(), True)
update_record(f"{PV_ROOT}STRWAVE", generate_str_data(), True)
update_record(f"{PV_ROOT}AAI", generate_data())