from croaks import IFile
from setup_bench import *

sched.ar.close()
sched.ar.led(5, 5000)

xs = np.hstack([np.arange(0,145700,500)])
sched.dkd.set_nsamples(10)

with IFile(['pos', 'lev', 'std', 'zwo', 'sbig'], template='background_') as fid:
    for x in xs:
        sched.xy.move(x=int(x), y=0)
        r = sched.dkd.server.read(1)[0]
        fid.append(x, np.mean(r), np.std(r), sched.zwo.server.photodiode(), sched.sbig.photodiode(1))
        r = sched.dkd.server.read(1)[0]
        fid.append(x, np.mean(r), np.std(r), sched.zwo.server.photodiode(), sched.sbig.photodiode(1))

sched.ar.open()
with IFile(['pos', 'lev', 'std', 'zwo', 'sbig'], template='background_') as fid:
    for x in xs:
        sched.xy.move(x=int(x), y=0)
        r = sched.dkd.server.read(1)[0]
        fid.append(x, np.mean(r), np.std(r), sched.zwo.server.photodiode(), sched.sbig.photodiode(1))
        r = sched.dkd.server.read(1)[0]
        fid.append(x, np.mean(r), np.std(r), sched.zwo.server.photodiode(), sched.sbig.photodiode(1))