from setup_cbp import *
import astropy.io.fits as pyfits
import pyds9
ds9 = pyds9.DS9()

d['sbig.EXPTIME'] = 1
d['sbig.filter'] = 'EMPTY'
d['mount.alpha'] = sched.mount.get_alpha()
d['mount.delta'] = sched.mount.get_delta()
d['cbpmount.alt'] = sched.cbpmount.get_alt()
d['cbpmount.az'] = sched.cbpmount.get_az()
d['keithley'] = False
d['spectro'] = False
d['laser.wavelength'] = 450
d['cbpwheel.pinhole'] = 'grid'
d['cbpwheel.mask'] = 2
d['ndfilter.pos'] = 'IDLE'

#var = ['mount.alpha',]# 'mount.delta', 'cbpmount.alt', 'cbpmount.az']
var = ['mount.delta',]
steps = [64,]#32,1,1]
def show(f):
    if True:
        fid = pyfits.open(f)
        ds9.set_pyfits(fid)
        fid.close()

def move_shoot(v, h):
    d[v] = d[v] + h
    f=sched.shoot(d)
    show(f)
'''    
f = sched.shoot(d)
for v, h in zip(var, steps):
    d[v] = d[v] + h
    f = sched.shoot(d)
    show(f)
    d[v] = d[v] - h
    f = sched.shoot(d)
    show(f)
    d[v] = d[v] - h
    f = sched.shoot(d)
    show(f)
    d[v] = d[v] + h
    f = sched.shoot(d)
    show(f)

d['cbpmount.alt'] = sched.cbpmount.get_alt() + 1
f = sched.shoot(d)
fid = pyfits.open(f)
v.set_pyfits(fid)
fid.close()

d['cbpmount.az'] = sched.cbpmount.get_az() + 1
f = sched.shoot(d)
fid = pyfits.open(f)
v.set_pyfits(fid)
fid.close()

d['cbpmount.alt'] = sched.cbpmount.get_alt() - 1
f = sched.shoot(d)
fid = pyfits.open(f)
v.set_pyfits(fid)
fid.close()

d['cbpmount.az'] = sched.cbpmount.get_az() - 1
f = sched.shoot(d)
fid = pyfits.open(f)
v.set_pyfits(fid)
fid.close()
'''

'''
d['mount.alpha'] = 0
f = sched.shoot(d)
fid = pyfits.open(f)
v.set_pyfits(fid)
fid.close()

d['mount.delta'] = 16
f = sched.shoot(d)
fid = pyfits.open(f)
v.set_pyfits(fid)
fid.close()
d['mount.delta'] = 0
f = sched.shoot(d)
fid = pyfits.open(f)
v.set_pyfits(fid)
fid.close()
'''