from setup_cbp import *
import cbpassist as ca
import time

d['cbp.pinhole'] = 'grid'
d['cbp.alt'] =  'IDLE'
d['cbp.az'] = 'IDLE'
d['sbig.filter'] = 'EMPTY'

outdir = 'mondrik_temp_test/linearity_scan_2'

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

wl = 660
scale_factors = 10.**np.arange(-2,1.05,0.05)


for scale_factor in scale_factors:
    exptime, filter_pos = ca.calc_exp_time(wl)
    d['sbig.EXPTIME'] = exptime*scale_factor+10.
    d['keithley.exptime'] = str(exptime*scale_factor)
    d['laser.wavelength'] = str(wl)
    d['ndfilter.pos'] = str(filter_pos)
    if exptime*scale_factor > 100:
        d['spectro.exptime'] = 1.
    else:
        d['spectro.exptime'] = 0.1

    #  Do dark
    d['keithley'] = False
    print('Taking Dark')
    sched.shoot(d, outdir)
    d['keithley'] = True
    print('Taking Light')
    sched.shoot(d, outdir)