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

REFERENCE_POSITION = {'X':1.2 ,'Y':0.0, 'Z':110.6, 'Theta':-9.1, 'Tilt':0.8, 'Phi':-90.0}
SAMPLE_POSITION = {'X':0.65 ,'Y':0.0, 'Z':115.0, 'Theta':-9.1, 'Tilt':0.8, 'Phi':-90.0}

def move_to_position(pdict):
    ManipulatorX.move(pdict['X'])
    ManipulatorY.move(pdict['Y'])
    ManipulatorZ.move(pdict['Z'])
    ManipulatorTheta.move(pdict['Theta'])
    ManipulatorTilt.move(pdict['Tilt'])
    ManipulatorPhi.move(pdict['Phi'])

#Au600
move_to_position(REFERENCE_POSITION)
rs = RangeSelection(347.0,354.0)
rs.setVars([1., 1., 1, 800.]) #Time, Size, Iteration, photon energy

run("XPSSpectrum", {"save_scienta_image":True, 
                    "ranges":[rs, ], 
                    "ENDSCAN":False, 
                    "pass_energy":100})

#set_exec_pars(open=False)                    

flush_data()

#In600
move_to_position(SAMPLE_POSITION)
#rs = RangeSelection(140.0,153.0)
rs = RangeSelection(347.0,354.0)

rs.setVars([1., 1., 1, 800.]) #Time, Size, Iteration, photon energy

run("XPSSpectrum", {"save_scienta_image":True, 
                    "ranges":[rs, ], 
                    "ENDSCAN":False, 
                    "pass_energy":100})

flush_data()