#Script imported from: Fe_hyst_plus.xml

#Pre-actions
#cawait('ACOAU-ACCU:OP-MODE', 'Light Available', type = 's')
#caput('X07MA-ID:MODE', 'CIRC +')
sleep(1.0)
#cawait('X07MA-ID:DONE', 'DONE', type = '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', '2.0')
sleep(15.0)

#TODO: Set the diplay names of positioners and detectors
scan = ManualScan(['field', 'Energy'], ['TEY', 'I0', 'trans', 'polarization', 'polAngle', 'temperature', 'RingCurrent', 'fieldAnalogX', 'tey_norm', 'trans_norm'] , [0.0, 703.9], [80.0, 707.9], [80, 1])
scan.start()

#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')

#Dimension 1
#RegionPositioner field
for setpoint1 in frange(0.0, 80.0, 1.0, True):
    #Region 1 pre-actions
    if setpoint1 == 0.0:
        caputq('X07MA-PC-PS2:M:GO.A', '6.8')
    field.put(setpoint1, timeout=None) # TODO: Set appropriate timeout
    readback1 = fieldReadback.get()
    if abs(readback1 - setpoint1) > 0.5 : # TODO: Check accuracy
        raise Exception('Actor field could not be set to the value ' + str(setpoint1))
    #Dimension 2
    #ArrayPositioner Energy
    for setpoint2 in (707.90, 703.90):
        Energy.put(setpoint2, timeout=None) # TODO: Set appropriate timeout
        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))
        sleep( 0.5 ) # Settling time
        #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
        		
        scan.append ([setpoint1, setpoint2], [readback1, readback2], [detector1, detector2, detector3, detector4, detector5, detector6, detector7, detector8, tey_norm, trans_norm])

#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()

scan.end()