import  ch.psi.pshell.epics.ChannelDouble as ChannelDouble
A1 = ChannelDouble("Phase",               "S20SY02-MKAC040:PHASE-SET")
S1 = ChannelDouble("BPM-Y",               "SARMA02-DBPM040:Y1")
#S2 = ChannelDouble("SATSY01-DBPM010:Y1",      "SATSY01-DBPM010:Y1")
#S3 = ChannelDouble("SATSY01-DBPM010:Q1",      "SATSY01-DBPM010:Q1")
#S4 = ChannelDouble("S20SY02-DBPM120:Y1",      "S20SY02-DBPM120:Y1")
A1.initialize()
S1.initialize()
#S2.initialize()
#S3.initialize()
#S4.initialize()
A1_init = A1.read()
A1i    = 265.0
A1f    = 275.0
step   = 0.5
lat    = 1.0
nav    = 4
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
    #S3_averager = create_averager(S3, nav, lat)
    #S3_averager.monitored=True
    #S4_averager = create_averager(S4, nav, lat)
    #S4_averager.monitored=True
    time.sleep(4.0)
    #r = lscan(A1, (S1_averager, S2_averager, S3_averager, S4_averager), A1i, A1f, step, latency=0.5, after_read = after_sample)
    r = lscan(A1, (S1_averager), A1i, A1f, step, latency=0.5, 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)]
    #S3mean   = [val.mean for val in r.getReadable(2)]
    #S3rmsn   = [val.stdev for val in r.getReadable(2)]
    #S4mean   = [val.mean for val in r.getReadable(3)]
    #S4rmsn   = [val.stdev for val in r.getReadable(3)]
finally:
    A1.write(A1_init)
    A1.close()
    S1.close()
    #S2.close()
    #S3.close()
    #S4.close()