#
# Import UniTAP module.
#

import UniTAP
import time

#
# To initialize UniTAP library wrapper user should create UniTAP.TsiLib() object.
#
lUniTAP = UniTAP.TsiLib()

#
# For opening device, please, put serial number of the device as 8 symbol str or put index of device.
#
# dev = lUniTAP.open("NNNNNNNN")
dev = lUniTAP.open(0)

# After opening device as in UCD Console device role should be selected.

role = dev.select_role(UniTAP.dev.UCD323.HDMISinkEARC)
role.hdrx.earc.configure(mode=UniTAP.EarcMode.eArc, latency=0)

print("Starting audio generator...")
print("eARC TX Status:", role.hdrx.earc.get_status())
role.hdrx.earc.ag.setup(audio_mode=UniTAP.AudioMode(), audio_pattern=UniTAP.AudioPattern.SignalSine)
role.hdrx.earc.ag.apply()

print("Audio generator is running...")
print("eARC TX Status:", role.hdrx.earc.get_status())
print("AG status:", role.hdrx.earc.ag.status)
time.sleep(5)

print("Starting againg...")
print("eARC TX Status:", role.hdrx.earc.get_status())
print("AG status:", role.hdrx.earc.ag.status)
role.hdrx.earc.ag.apply()

time.sleep(5)

print("Stopping audio generator...")
print("eARC TX Status:", role.hdrx.earc.get_status())
print("AG status:", role.hdrx.earc.ag.status)
role.hdrx.earc.ag.stop_generate()

print("eARC RX EDID:", role.hdrx.earc.get_edid())

#
# Since the 3.5 version, TsiLib and TSIDevice objects have the option to be closed earlier.
# TSIDevice can be closed with the TsiLib method close(). TsiLib can be closed with cleanup().
# Clean up will close all opened devices and block ability to open any devices
# with same TsiLib object.
#
lUniTAP.close(dev)

lUniTAP.cleanup()