import  ch.psi.pshell.epics.ChannelDouble as ChannelDouble
A1 = ChannelDouble("x1",                 "S10MA01-DBPM060:X-REF-FB")
A2 = ChannelDouble("x2",                 "S10MA01-DBPM120:X-REF-FB")
S1 = ChannelDouble("LLM1",               "S30CB05-DLLM110:CH4_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()
A2.initialize()
S1.initialize()
S2.initialize()
S3.initialize()
S4.initialize()
S5.initialize()
A1_init = A1.read()
A2_init = A2.read()
A1i    = A1_init-1.6
A1f    = A1_init+3.0
A2i    = A2_init-1.6
A2f    = A2_init+3.0
nstep  = 47
lat    = 10
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 = ascan((A1,A2), (S1_averager, S2_averager, S3_averager, S4_averager, S5_averager), (A1i,A2i), (A1f,A2f), (nstep,nstep), latency=lat, 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()