from setup_cbp import *
import cbpassist as ca
import time


outdir = 'mondrik_temp_test/nt252_pollution_test_scan_5nm_highflux'

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

wavelengths = np.arange(350, 1150, 5)
spec_exp_time = 2

for w in wavelengths:
    if w >= 670:
        d['keithley.npulses'] = 30
    else:
        d['keithley.npulses'] = 250
    d['spectro.exptime'] = spec_exp_time
    d['laser.wavelength'] = str(w)
    sched.shoot(d, outdir)