from pysmu import Session, MODE
from Adalm1000_Logger.DeviceController import DeviceController
from Adalm1000_Logger.GUI import MainWindow, startGUI
import os


# --- Setup ---
sess = Session()
sess.add_all()
controllers = {dev.m_serial: DeviceController(dev) for dev in sess.m_devices}
serials = list(controllers.keys())

for i in enumerate(serials):
    c = controllers[serials[i]]
    c.set_mode(0, MODE.HI_Z)  # CH A Hi-Z
    c.set_mode(1, MODE.HI_Z)  # CH B Hi-Z
    MainWindow.add_list_item(MainWindow, serials[i], i, c.start, c.stop)

startGUI()

# # Beispiel: Gerät 1 starten, Gerät 2 später starten/stoppen

# c1 = controllers[serials[0]]
# c1.set_mode(0, MODE.HI_Z)  # CH A Hi-Z
# c1.set_mode(1, MODE.HI_Z)  # CH B Hi-Z
# c1.start()

# time.sleep(2.0)

# if len(serials) > 1:
#     c2 = controllers[serials[1]]
#     c2.set_mode(0, MODE.HI_Z)
#     c2.set_mode(1, MODE.HI_Z)
#     c2.start()
#     time.sleep(3.0)
#     c2.stop()  # nur Gerät 2 stoppen

# # sauber beenden (Strg+C-Handling etc. weglassen der Kürze halber)
# for c in controllers.values():
#     c.stop()
# for c in controllers.values():
#     c.shutdown()