from setup_cbp import *
import cbpassist as ca
import time


outdir = 'mondrik_temp_test/nt252_pulse_charge_map'

if not os.path.exists(outdir):
    os.mkdir(outdir)

wavelengths = np.arange(335, 1205, 5)
spec_exp_time = 2
n_repeats = 10

for i in range(n_repeats):
    for w in wavelengths:
        print('CYCLE: {}, WAVE: {}'.format(i, w))
        d['keithley.npulses'] = 90
        d['spectro.exptime'] = spec_exp_time
        d['laser.wavelength'] = str(w)
        sched.shoot(d, outdir)