import time
import interface.prologix_gpib as prologix
import devices.schlumberger_7150plus as schlumberger

import numpy as np
import matplotlib.pyplot as plt

iface = prologix.usb(com='ASRL31::INSTR', baudrate=19200, timeout=5000)
iface.loc()  # Lokaler Modus
schlumi = schlumberger.device(iface, 13)

values = np.zeros(100)              # leeres Array mit 100 Werten
for i in range(0, len(values)):
    schlumi.trigger()               # single shot
    values[i] = schlumi.read()      # Wert lesen
    time.sleep(0.1)                 # 100ms warten

plt.plot(values)                    # plotten
plt.ylabel('U [V]')
plt.xlabel('Messung Nr.')

import interface.prologix_gpib as prologix
import devices.rohde_schwarz_ngmo2 as ngmo

iface = prologix.usb(com='ASRL1::INSTR', baudrate=19200, timeout=2000)

battery = ngmo.device(iface, 7)  # ngmo at address 7
battery.set_voltage('A', 12.4)   # set to 12.4 V
battery.set_output('A', True)    # turn on