from stardice import keithley
import numpy as np
from glob import glob
import time

template = 'zwo_stability_'
existing_files = glob(template +'*.npy')
existing_files = [int(f.replace('.npy', '').replace(template, '')) for f in existing_files]
if not existing_files:
    print('Warning: first file in the seri')
    odometer = 0
else:
    odometer = np.max(existing_files) + 1

k = keithley.Keithley6514('/dev/ttyUSB1')


k.setup(krange=2e-5, samples=100)

nrepeat =400
#nrepeat = 2
length = 10
result = []

for r in range(nrepeat):
        on = k.read(repeats=length)
        result.append([np.array(on[1]), np.array(on[0])])

#result = np.rec.fromrecords(result, names=['led', 'dark', 'on'])
result = np.hstack(result)
np.save(template+'%04d.npy' % odometer, result)