import  ch.psi.pshell.epics.ChannelDouble as ChannelDouble
import ch.psi.pshell.device.DummyRegister as DummyRegister
A1 = ChannelDouble("Phase (deg.)",            "SARUN03-UPHS060:PHI_SET")
S1 = ChannelDouble("energy per pulse (uJ)",   "SARFE10-PBPG050:PHOTON-ENERGY-PER-PULSE-US")
#S2 = ChannelDouble("HAMP",                    "SARFE10-PBPG050:HAMP-INTENSITY-CAL")

A1.initialize()
S1.initialize()
#S2.initialize()
A1_init = A1.read()
A1i    = 0.
A1f    = 360.
nstep  = 20
lat    = 1.
nav    = 10
plt    = plot(None, title="Output")[0]
plt.clear()
plt.setStyle(plt.Style.ErrorY)
plt.addSeries(LinePlotErrorSeries("Sensor1", Color.red))
def after_sample(record, scan):
    plt.getSeries(0).appendData(record.positions[0], record.readables[0].mean, record.readables[0].stdev)
try:
    S1_averager = create_averager(S1, nav, lat)
    #S2_averager = create_averager(S2, nav, lat)
   # S2_averager.monitored=True
    r = lscan(A1, S1_averager, A1i, A1f, nstep, latency=2.0, after_read = after_sample)
    #r = lscan(A1, (S1_averager , S2_averager), A1i, A1f, nstep, latency=5.0, after_read = after_sample)
    Act1     = r.getPositions(0)
    S1mean   = [val.mean for val in r.getReadable(0)]
    S1rms    = [val.stdev for val in r.getReadable(0)]
    #S2mean   = [val.mean for val in r.getReadable(1)]
    #S2rmsn   = [val.stdev for val in r.getReadable(1)]
finally:
    A1.write(A1_init)
    A1.close()
    S1.close()
    #S2.close()