#Parameters

B1 = 14    # starting mag. field in Amps
B2 = -14# final mag. field in Amps
BSTEP = 1   # step size mag. field in Amps
E1 = 779.3; #list of energies in eV
E2 = 770;
MODE = 1  #polarization (CIRC+ -> 1 or CIRC- -> 2)
OFFSET1 = -1.90 #ID1 offset
OFFSET2 = -1.90 #ID2 offset
FIELD_PRECISION = 0.05 #  in Amps
ENERGY_CHANGE_SLEEP = 0.0 # put the Enerrgy settling time if needed
FIELD_CHANGE_SLEEP =  0.0 # put the Field   settling time if needed
ACC_TIME = 1.0 # accumulation time

# List of scans. Each scan is defined as: (start,stop, step)
# to make a whole loop two scans are needed: [(B1, B2, BSTEP),(B2,B1,-BSTEP)]
# to make half of a loop one scan is needed: [(B1, B2, BSTEP),]  <- don't forget to put comma before ]
RANGES = [(B1, B2, -BSTEP),(B2,B1,BSTEP)] 


#Pre-actions
# Here polarization and offsets are set

caput ("X11PHS-E:OPT","PGM+ID2")
#caput(OTF_OFF1,-50) #detune ID1
caput(OTF_OFF2,OFFSET2) 
wait_channel(OTF_DONE, 1, type = 'i')

if MODE is 1 or 2:
    caput(OTF_MODE2,MODE)
else:
    raise Exception("Invalid polarization type: " +  MODE)
    
wait_channel(OTF_DONE, 1, type = 'i')


# setting number of samples to be averaged, which depends on the accumulation time
avg = ACC_TIME * 10
caput("X11MA-ES1-10ADC:AVG",avg)

# Generating a list of mag. fields (setpoints) needed for the loop
setpoints = []
for r in RANGES:
    setpoints = setpoints + frange(r[0], r[1], r[2], True)

# plot properties
set_preference(Preference.ENABLED_PLOTS,  ['field', 'tey_norm'])
set_preference(Preference.PLOT_TYPES, {'tey_norm':1})

#scan = ManualScan(['field', 'Energy'], ['TEY', 'I0', 'polarization', 'temperature', 'RingCurrent', 'tey_norm'] , [min(setpoints), min(ENERGIES)], [max(setpoints), max(ENERGIES)], [len(setpoints)-1, len(ENERGIES)-1])
scan = ManualScan(['field', 'Energy'], ['I0', 'TEY', 'TFY', 'polarization', 'temperature', 'RingCurrent', 'tey_norm','tfy_norm'] , [min(setpoints), min(ENERGIES)], [max(setpoints), max(ENERGIES)], [len(setpoints)-1, len(ENERGIES)-1])
scan.start()
os.chdir("/sls/X11MA/data/X11MA/beamtime_december/")
# Main loop
for B in setpoints:
    print "Setting field = ", B
    caput("X11MA-XMCD:I-SETraw",B)
    time.sleep( FIELD_CHANGE_SLEEP )     # Settling time

    #readback1 = caget("X11MA-XMCD:Ireadout")    

    #while abs(readback1-B) > FIELD_PRECISION:
     #   readback1 = caget("X11MA-XMCD:Ireadout")
      #  time.sleep(0.5)

    
    #edge
    print "Setting energy = ", E1
    caput(OTF_ESET, E1)
    wait_channel(OTF_DONE, 1, type = 'i')
    readback2  = energy.read()
    #if abs(readback2 - E[0]) > 0.1 : # TODO: Check accuracy
    #raise Exception('Energy could not be set to the value ' + str(E[0]))
    #sleep( ENERGY_CHANGE_SLEEP ) # Settling time

    detector1 = keithley_1a.read() #Keithley1
    detector2 = keithley_2a.read() #Keithley2
    detector4 = caget(OTF_MODE2)   #polarization in ID2
    M1 = detector2/detector1     
    #converting polarization strings to integer numbers: C+ -> 1 and C- -> 2
    if detector4 == 'CIRC +': detector4 = 1
    elif detector4 == 'CIRC -': detector4 = 2
    else: detector4 = 0

    
    #preedge
    print "Setting energy = ", E2
    caput(OTF_ESET, E2)
    wait_channel(OTF_DONE, 1, type = 'i')
    readback2  = energy.read()
    #if abs(readback2 - E[1]) > 0.1 : # TODO: Check accuracy
    #raise Exception('Energy could not be set to the value ' + str(E[0]))
    #sleep( ENERGY_CHANGE_SLEEP ) # Settling time

    detector1 = keithley_1a.read() #Keithley1
    detector2 = keithley_2a.read() #Keithley2
    detector4 = caget(OTF_MODE2)   #polarization in ID2
    M2 = detector2/detector1     
    #converting polarization strings to integer numbers: C+ -> 1 and C- -> 2
    if detector4 == 'CIRC +': detector4 = 1
    elif detector4 == 'CIRC -': detector4 = 2
    else: detector4 = 0

    tey_norm = M2/M1;
      
    #scan.append ([B, E], [readback1, readback2], [detector1, detector2, detector4, detector6, detector7, tey_norm])
    scan.append ([B, E], [readback2], [detector1, detector2, detector4, tey_norm])
    filename=open("data1.txt","a")
    filename.write("%s %s %s %s %s \n" %(B,E,detector1,detector2,tey_norm))
    
scan.end()
filename.close()

caput("X11MA-ES1-10ADC:AVG",1) 
caput("X11MA-XMCD:I-SETraw",0)

close_vg13()