#Script imported from: Fe_hyst_plus.xml

#Parameters
START_FIELD = -0.2
END_FIELD = 0.2
ENERGIES = (707.90, 703.90)
RAMP_RATE = 2.0
ENERGY_CHANGE_SLEEP = 0.5 

#Pre-actions
#cawait('ACOAU-ACCU:OP-MODE', 'Light Available', type = 's')
#caput('X07MA-ID:MODE', 'CIRC +')
sleep(1.0)
#cawait('X07MA-ID:DONE', 'DONE', ype = 's')
#caput('X07MA-ID:ENERGY-OFFS', '-1.0')
#cawait('X07MA-ID:DONE', 'DONE', type = 's')
caputq('X07MA-PC:CSCALER.INPB', '1')
caputq('X07MA-PC-PS2:SET:DMD:RAMPRATE:TPM', RAMP_RATE)
#sleep(15.0)

if len(ENERGIES) ==2:
    dif_series = plot([],"Dif", title="Dif")[0].getSeries(0)
else:
    dif_series = None
    
set_preference(Preference.ENABLED_PLOTS,  ['field', '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'])
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.start()

#Stop condition
ramp_done = Channel('X07MA-PC-MAG:X:RAMP:DONE', type = 'i')

#Creating channels: dimension 1
#RegionPositioner field
#field = Channel('X07MA-PC:GO', type = 'd')
fieldReadback = Channel('X07MA-PC-PS2:STS:PMCURRENT', type = 'd')
#Creating channels: dimension 2
#ArrayPositioner Energy
Energy = sim_energy #Channel('X07MA-PHS-E:GO.A', type = 'd')
EnergyReadback = sim_energy_readback # Channel('X07MA-PGM:CERBK', type = 'd')
#ScalarDetector TEY
TEY = Channel('X07MA-ES1-AI:SIGNAL0', type = 'd')
#ScalarDetector I0
I0 = Channel('X07MA-ES1-AI:SIGNAL1', type = 'd')
#ScalarDetector trans
trans = Channel('X07MA-ES1-AI:SIGNAL2', type = 'd')
#ScalarDetector polarization
#polarization = Channel('X07MA-ID:MODE', type = 'd')
#ScalarDetector polAngle
#polAngle = Channel('X07MA-ID:ALPHA', type = 'd')
#ScalarDetector temperature
temperature = Channel('X07MA-PC-TC:STS:T1', type = 'd')
#ScalarDetector RingCurrent
RingCurrent = Channel('ARIDI-PCT:CURRENT', type = 'd')
#ScalarDetector fieldAnalogX
fieldAnalogX = Channel('X07MA-ES1-AI:SIGNAL4', type = 'd')


print "Waiting for start field"
"""
#WORKAROUND TO BUG IN DRIVER
if abs(fieldReadback.get() - START_FIELD) < 0.005:
    caput('X07MA-PC-PS2:M:GO.A', START_FIELD + 0.01)
caput('X07MA-PC-PS2:M:GO.A', START_FIELD)
#caputq('X07MA-PC-PS2:M:GO.A', START_FIELD)
"""

caput("X07MA-PC-MAG:X:DMD", START_FIELD)
caput("X07MA-PC-MAG:STARTRAMP.PROC", 1)
ramp_done.wait_for_value(1.0)
print "Set end field"
caput("X07MA-PC-MAG:X:DMD", END_FIELD)
caputq("X07MA-PC-MAG:STARTRAMP.PROC", 1)


index = 0
while(True):
    #ArrayPositioner Energy
    for setpoint2 in ENERGIES:
        Energy.put(setpoint2, timeout=None) # TODO: Set appropriate timeout
        sleep( ENERGY_CHANGE_SLEEP ) # Settling time

        readback2 = EnergyReadback.get()
        if abs(readback2 - setpoint2) > 0.1 : # TODO: Check accuracy
            raise Exception('Actor Energy could not be set to the value ' + str(setpoint2))

        #Detector field readback
        field_readback = fieldReadback.get()

        #Detector TEY
        detector1 = TEY.get()
        #Detector I0
        detector2 = I0.get()
        #Detector trans
        detector3 = trans.get()
        #Detector polarization
        detector4 = sim_energy_readback.read()  # polarization.get()
        #Detector polAngle
        detector5 = sim_energy_readback.read()  #polAngle.get()
        #Detector temperature
        detector6 = temperature.get()
        #Detector RingCurrent
        detector7 = RingCurrent.get()
        #Detector fieldAnalogX
        detector8 = fieldAnalogX.get()


        #Manipulation tey_norm
        #Variable Mappings
        
        #TODO: Move, if needed, this import to the file header: import math
        tey_norm = detector1/detector2
        		
        #Manipulation trans_norm
        #Variable Mappings
        
        #TODO: Move, if needed, this import to the file header: import math
        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)
        print [field_readback, setpoint2]  		
        print [field_readback, readback2]
        print [detector1, detector2, detector3, detector4, detector5, detector6, detector7, detector8, tey_norm, trans_norm]
        scan.append ([field_readback, setpoint2], [field_readback, readback2], [detector1, detector2, detector3, detector4, detector5, detector6, detector7, detector8, tey_norm, trans_norm])


    print "Field = " + str(fieldReadback.get())
    if ramp_done.get() == 1:    #If not ramping
        print "Not ramping, breaking execution"
        break
    index = index+1

#Closing channels
Energy.close()
EnergyReadback.close()
TEY.close()
I0.close()
trans.close()
#polarization.close()
#polAngle.close()
temperature.close()
RingCurrent.close()
fieldAnalogX.close()
#field.close()
fieldReadback.close()
ramp_done.close()

scan.end()