#####################
## INPUT PARAMETERS##
POS_START = 3
POS_END = 34.5
POS_STEP = 5

ENERGIES = [287.0, 290.0]

#####################

open_valve()

class Dummy(Writable):
    def write(self, val):
        pass
dummy = Dummy()
#caput("X07MA-OP-VG13:WT_SET",1)
#time.sleep(0.1)

sensors = [tey_norm, signal_tey,signal_i0,  trans_norm, sample_hor, energy_readback]


"""
def before_pass(pass_num, scan):
    print "Setting energy = " , ENERGIES[pass_num-1]
    #set_energy(ENERGIES[pass_num-1])
#TODO: check whi must incluide range="auto"
#lscan(sample_vert,sensors, POS_START, POS_END, POS_STEP, latency =0.5,  passes = len(ENERGIES), before_pass=before_pass, range="auto")
lscan(dummy,sensors, POS_START, POS_END, POS_STEP, latency =0.5,  passes = len(ENERGIES), before_pass=before_pass, range="auto")
"""


plots = plot([None]*len(sensors), name = [d.name for d in sensors])
for p in plots: 
    p.clear()
    p.legendVisible = True

def after_readout(record, scan):
    for i in range(len(plots)):
        plots[i].getSeries(plots[i].numberOfSeries-1).appendData(record.positions[0], record.values[i])    
def before_pass(pass_num):
    en = ENERGIES[pass_num-1]
    #set_energy(en)  
    for p in plots: p.addSeries(LinePlotSeries(str(en))) 

ret= lscan(dummy, sensors, POS_START, POS_END, POS_STEP, latency = 0.2, passes = len(ENERGIES), before_pass = before_pass, after_read=after_readout, plot_disabled=True, range="auto")



close_valve()