"""
Flying hologram scan (experimental)

Arguments:

THETA_RANGE  (tuple (min, max))
PHI_RANGE (tuple (min, max))
THETA_STEP (scalar)
PHI_STEP (scalar)

ZIGZAG (BOOLEAN)
LATENCY (float) in seconds
SENSORS (list of devices)
"""

THETA_RANGE = (-9.0, 81.0)
THETA_STEP = 1.0
#PHI_RANGE = (-40.0, +40.0)
PHI_RANGE = (-160.0, +160.0)
PHI_STEP = 40.0

LATENCY = 0.0
ZIGZAG = True
ENDSCAN = True
MOTORS = (ManipulatorTheta)
#SENSORS = (Counts, Scienta.spectrum, SampleCurrent, RefCurrent, MachineCurrent)
SENSORS = (Counts, Scienta.dataMatrix, SampleCurrent, RefCurrent, MachineCurrent)

#set_preference(Preference.ENABLED_PLOTS, [ManipulatorPhi, ManipulatorTheta, Scienta.dataMatrix, ImageIntegrator])
#set_preference(Preference.PLOT_TYPES,{'ImageIntegrator':1})
adjust_sensors()
set_adc_averaging()
set_preference(Preference.PLOT_TYPES, {'Scienta spectrum':1})

# time per scienta acquisition in seconds
time1 = time.time()
trig_scienta()
time2 = time.time()
scienta_time = (time2 - time1)
print "scienta_time: ", scienta_time

# time for one theta scan in seconds
THETA_NSTEPS = int((THETA_RANGE[1] - THETA_RANGE[0]) / THETA_STEP) + 1
theta_time = scienta_time * THETA_NSTEPS
print "theta_time: ", theta_time

PHI_NSTEPS = int((PHI_RANGE[1] - PHI_RANGE[0]) / PHI_STEP) + 1
phi_positions = [PHI_RANGE[0] + PHI_STEP * i for i in range(PHI_NSTEPS)]
print "phi_positions: ", phi_positions


def before_pass(index, scan):
    global phi_positions
    print "Starting pass: ", index
    phi = phi_positions[index-1]
    ManipulatorPhi.write(phi)
    print "phi = ", phi
    ManipulatorPhi.waitValueInRange(phi, 1.0, 100)    

    
try:
    cscan(MOTORS, SENSORS, THETA_RANGE[0], THETA_RANGE[1], THETA_NSTEPS - 1, time=theta_time, passes=len(phi_positions), zigzag=ZIGZAG, before_read=before_readout, after_read = after_readout, before_pass = before_pass, check_positions = False)
    
finally:
    if ENDSCAN:
        after_scan()