Example: Publishers: publish queued.

"""Example: Publishers: publish queued.

Demonstrates publishing measurements/commands to a dataset (Nominal Core publisher).

"""

import time

from nominal_instro.instruments import NominalPSU
from nominal_instro.lib import ConnectConfig
from nominal_instro.lib.publishers import NominalCorePublisher, QueuedPublisher

DATASET_RID = "<dataset_rid>" # Replace with your dataset RID. 
VISA_RESOURCE = "TCPIP0::127.0.1::5025::SOCKET"


# Create instrument instances
psu = NominalPSU.auto_create("myPSU", resource=ConnectConfig(visa_resource=VISA_RESOURCE), num_channels=2)
core_publisher = NominalCorePublisher(dataset_rid=DATASET_RID)
queued_publisher = QueuedPublisher(core_publisher, max_queue_size=100, wait_for_queue=True)

psu.add_publisher(queued_publisher)

try:
    # Set up initial state of test
    psu.open()

    psu.output_enable(False, channel=2)
    psu.set_current_limit(0.2, channel=2)
    psu.set_voltage(0, channel=2)

    psu.get_current(channel=2)
    psu.get_voltage(channel=2)

    # Start
    psu.output_enable(True, channel=2)

    for v in range(10):
        start = time.time()

        psu.set_voltage(v, channel=2)
        time.sleep(1)
        psu.get_current(channel=2)
        psu.get_voltage(channel=2)

        print(f"Time taken: {time.time() - start}")

    psu.output_enable(False, channel=2)

    print("Done, waiting for queue to be empty...")

finally:
    # Shut it down
    psu.close()