# script for checking the symmetry of the machine bump on Au poly
#
# set the beamline to FE = 6 x 0.5, Ephot = 800 (G2), EX-SL = 100
# acquire the bump control lock manually

import ch.psi.pshell.plot.RangeSelectionPlot.RangeSelection as RangeSelection

rs = RangeSelection(706., 714.)
rs.setVars([0.3, 0.025, 1, 800.]) #Time, Size, Iteration, photon energy


# DynamicBumpLock.write("X03DA")
# DynamicBumpLock.write("none")

BUMP_TIMEOUT = 30000

print "switching to linear"
DynamicBumpMode.write("off")
DynamicBumpYAngle.waitValueInRange(0.0, 0.002, BUMP_TIMEOUT)
print "running XPS spectrum"
run("XPSSpectrum", {"save_scienta_image":True, 
                    "ranges":[rs, ], 
                    "ENDSCAN":False, 
                    "pass_energy":50})

print "switching to positive helicity"
DynamicBumpMode.write("pos")
DynamicBumpYAngle.waitValueInRange(0.2, 0.002, BUMP_TIMEOUT)
print "running XPS spectrum"
run("XPSSpectrum", {"save_scienta_image":True, 
                    "ranges":[rs, ], 
                    "ENDSCAN":False, 
                    "pass_energy":50})

print "switching to negative helicity"
DynamicBumpMode.write("neg")
DynamicBumpYAngle.waitValueInRange(-0.2, 0.002, BUMP_TIMEOUT)
print "running XPS spectrum"
run("XPSSpectrum", {"save_scienta_image":True, 
                    "ranges":[rs, ], 
                    "ENDSCAN":False, 
                    "pass_energy":50})

print "switching to linear"
DynamicBumpMode.write("off")
DynamicBumpYAngle.waitValueInRange(0.0, 0.002, BUMP_TIMEOUT)
print "done"