#Parameters
FILE  = '3-Er_test'
START_LAMBDA = 1536.00  # in nm
END_LAMBDA   = 1540.00  # in nm
SWEEP_RATE   = 0.01   # in nm/sec
#ENERGIES = (1300.90, 1302.90)   # X-ray energy in eV
ENERGIES = [ 1400.6 ]   # X-ray energy in eV

ENERGY_CHANGE_SLEEP = 0.5 
MODE = 'CIRC +'
OFFSET = -5.0

set_slit(30.0)
set_har(3.0)

if NO_BEAM_CHECK == False:
    wait_beam()
if MODE is not None:
    pol_mode.write(MODE)
if OFFSET is not None:     
    pol_offset.write(OFFSET) #caput('X07MA-ID:ENERGY-OFFS', OFFSET)
caputq('X07MA-PC:CSCALER.INPB', '1')

# open shutter
#caput('X07MA-FE-PH1:CLOSE4BL',1) 
#cawait('X07MA-FE-PH1:MS2', 1)


if len(ENERGIES) ==2:
  #  dif_series = plot([],"Dif", get_context()="Dif")[0].getSeries(0)
    dif_series = plot([],"Dif")[0].getSeries(0)
else:
    dif_series = None
    
set_preference(Preference.ENABLED_PLOTS,  ['wavelength', 'tey_norm', 'trans_norm'])
set_preference(Preference.PLOT_TYPES, {'tey_norm':1, 'trans_norm':1})

#scan = ManualScan(['field', 'Energy'], ['TEY', 'I0', 'trans', 'polarization', 'polAngle', 'temperature', 'RingCurrent', 'fieldAnalogX', 'tey_norm','trans_norm'], [0.0, ENERGIES[0]], [0.0, ENERGIES[-1]], [0, len(ENERGIES)-1])
scan = ManualScan(['Wavelength', 'Energy'], ['TEY', 'I0', 'trans', 'polarization', 'polAngle', 'temperature', 'RingCurrent', 'fieldAnalogX', 'fieldX', 'tey_norm','trans_norm'], [0.0, ENERGIES[0]], [0.0, ENERGIES[-1]], [0, len(ENERGIES)-1])
scan.start()

write_logs()

print "Setting laser wavelength..."

caput("X07MA-ES1-DLCP:SPEED",SWEEP_RATE)
caput("X07MA-ES1-DLCP:WAVELENGTH",START_LAMBDA)
caput("X07MA-ES1-DLCP:BEGIN",START_LAMBDA)
caput("X07MA-ES1-DLCP:END",END_LAMBDA)

open_valve()
time.sleep(5.0)
while True:
    time.sleep(0.5)
    lambda_readback = caget("X07MA-ES1-DLCP:WAVELENGTH_RBV")
    if abs(lambda_readback - START_LAMBDA) < 0.02:  
        print "Start wavelength reached."
        break
    
#open_valve(2.0)

print "Set energy..."
setpoint2 = ENERGIES[0]
print setpoint2
energy.write(setpoint2)
sleep( ENERGY_CHANGE_SLEEP ) # Settling time     

wait_device(energy_done, 1 )

wait_device(pol_done, "DONE") 


#if WAIT_STABLE_TEMPERATURE == True:
#    wait_temp()

 
print "Start laser scan..."

caput("X07MA-ES1-DLCP:CONTROL",3)

index = 0
while(True):
    #Dimension 2START_FIELD = -0.2
    #ArrayPositioner Energy    
    for setpoint2 in ENERGIES:        
        if len (ENERGIES)>1:    
            #Energy.put(setpoint2, timeout=None) # TODO: Set appropriate timeout
            energy.write(setpoint2)
            sleep( ENERGY_CHANGE_SLEEP ) # Settling time
        else: 
            sleep( 0.25 ) # TODO: Check
        
        #TODO: change to Controlled variable?
        readback2 = energy_readback.read() #EnergyReadback.get()
        if abs(readback2 - setpoint2) > 0.1 : # TODO: Check accuracy
#            raise Exception('Energy could not be set to the value ' + str(setpoint2))
            #print 'Energy could not be set to the value, try again. ' + str(setpoint2)
            energy.write(setpoint2)
            sleep( ENERGY_CHANGE_SLEEP ) # Settling time

        #Detector field readback
        lambda_readback = caget("X07MA-ES1-DLCP:WAVELENGTH_RBV")
        #field_readback = field.readback.read() #fieldReadback.get()

        detector1 = signal_tey.read() #TEY.get()
        detector2 = signal_i0.read() #I0.get()
        detector3 = signal_trans.read() #trans.get()
        detector4 = float(pol_mode.getPositions().index(pol_mode.readback.read())) #polarization.get()
        detector5 = pol_angle.read() #polAngle.get()
        detector6 = temperature.readback.read() #temperature.get()
        detector7 = current.read()
        detector8 = signal_field_analog_x.read() # fieldAnalogX.get()
        detector9 = field_x.read()

        _tey_norm = detector1/detector2
        _trans_norm = detector3/detector2

        if dif_series is not None:
            if setpoint2 == ENERGIES[0]:
                first = _tey_norm
            else:
                dif = _tey_norm-first
                dif_series.appendData(field_readback,dif)
        scan.append ([lambda_readback, setpoint2], [lambda_readback, readback2], [detector1, detector2, detector3, detector4, detector5, detector6, detector7, detector8, detector9, _tey_norm, _trans_norm])

        after_sample() #To call check_id_error()
    
    if abs(lambda_readback - END_LAMBDA) < 0.02:  
        print "Target wavelength has been reached, breaking execution."
        break
    index = index+1

scan.end()
log_scan_filename()
close_valve()