import  ch.psi.pshell.epics.ChannelDouble as ChannelDouble
A1 = ChannelDouble("BPM-X",              "S10MA01-DBPM060:X-REF-FB ")
S1 = ChannelDouble("LLM1",               "S10CB09-DLLM110:CH7_LOSS-MPS_REL_L1")
S2 = ChannelDouble("LLM2",               "S10CB09-DLLM110:CH8_LOSS-MPS_REL_L1")
S3 = ChannelDouble("BLM",                "S20CB02-DBLM435:B1_LOSS_MPS_REL_L1")
S4 = ChannelDouble("DRPS",               "SGE-EDRPS01-DU0601:DOSE-REL-OP")
S5 = ChannelDouble("CDR",                "S10MA01-DCDR080:AL1-B1")
A1.initialize()
S1.initialize()
S2.initialize()
S3.initialize()
S4.initialize()
S5.initialize()
A1_init = A1.read()
A1i    = -1.0
A1f    = 1.0
step   = 21
lat    = 0.11
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
    S3_averager = create_averager(S3, nav, lat)
    S3_averager.monitored=True
    S4_averager = create_averager(S4, nav, lat)
    S4_averager.monitored=True
    S5_averager = create_averager(S5, nav, lat)
    S5_averager.monitored=True
    A1.write(A1i)
    time.sleep(1.0)
    r = lscan(A1, (S1_averager, S2_averager, S3_averager, S4_averager, S5_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)]
    S2rms    = [val.stdev for val in r.getReadable(1)]
    S3mean   = [val.mean for val in r.getReadable(2)]
    S3rms    = [val.stdev for val in r.getReadable(2)]
    S4mean   = [val.mean for val in r.getReadable(3)]
    S4rms    = [val.stdev for val in r.getReadable(3)]
    S5mean   = [val.mean for val in r.getReadable(4)]
    S5rms    = [val.stdev for val in r.getReadable(4)]
finally:
    A1.write(A1_init)
    A1.close()
    S1.close()
    S2.close()
    S3.close()
    S4.close()
    S5.close()