#Script imported from: Fe_hyst_plus.xml

#Parameters
"""
START_FIELD = -0.1
END_FIELD = 0.1
ENERGIES = (707.90, 703.90)
RAMP_RATE = 0.2
ENERGY_CHANGE_SLEEP = 0.5 
MODE = 'CIRC +'
OFFSET = -1.0
"""

controller.dataManager.openOutput()
print "Data folder: " + controller.dataManager.output
#TODO: Print filename


FIELD_PRECISION = 0.01

#Pre-actions
cawait('ACOAU-ACCU:OP-MODE', 'Light Available', type = 's')
caput('X07MA-ID:MODE', MODE)
caput('X07MA-ID:ENERGY-OFFS', OFFSET)
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", context="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 = Channel('X07MA-PHS-E:GO.A', type = 'd')
EnergyReadback = 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 "SKIPPED Waiting for start field"


#if abs(fieldReadback.get() - START_FIELD) > FIELD_PRECISION:
#    caput('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)

cawait('X07MA-ID:DONE', 'DONE', type = 's')

caput('X07MA-OP-VG13:WT_SET', 'Try open')
time.sleep(2.0)

print "Set end field"
#caput("X07MA-PC-MAG:X:DMD", END_FIELD)
#caputq("X07MA-PC-MAG:STARTRAMP.PROC", 1)
caputq('X07MA-PC-PS2:M:GO.A', END_FIELD)




index = 0
while(True):
    #Dimension 2START_FIELD = -0.2
    #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 = polarization.get()
        #Detector polAngle
        detector5 = 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()