import  ch.psi.pshell.epics.Positioner as Positioner
import  ch.psi.pshell.epics.ChannelDouble as ChannelDouble

dry_run = True
do_elog = True
is_panel = get_exec_pars().source != CommandSource.ui    #must be check before run
if is_panel:
    start = args[0]
    stop  = args[1]
    step  = args[2]
    nb    = int(args[3])
    lat   = args[4]
    plt   = args[5]
else:
    start   = -50.0
    stop    = 150.0
    step    = 1.0
    nb      = 2
    lat     = 0.010
    plt     = plot(None, title="Output")[0]
    
#Plot setup
plt.clear()
plt.removeMarker(None)
plt.setStyle(plt.Style.ErrorY)
plt.addSeries(LinePlotErrorSeries("Charge", Color.red))
plt.getAxis(plt.AxisId.X).setLabel("Gun Beam Phase (deg)")
plt.getAxis(plt.AxisId.Y).setLabel("SINEG01-DICT215:B1_CHARGE (pC)")
plt.setLegendVisible(True)

if dry_run:
    bph = Positioner("Beam phase", "SINEG01-RSYS:SET-BEAM-PHASE-SIM", "SINEG01-RSYS:SET-BEAM-PHASE-SIM")
    rph  = ChannelDouble("RF phase", "SINEG01-RSYS:SET-VSUM-PHASE-SIM")
    q = ChannelDouble("Charge", "SINEG01-DICT215:B1_CHARGE-SIM")
    q.initialize()
    q.monitored=True
else:
    bph = Positioner("Beam phase", "SINEG01-RSYS:SET-BEAM-PHASE", "SINEG01-RSYS:GET-BEAM-PHASE")
    rph  = ChannelDouble("RF phase", "SINEG01-RSYS:SET-VSUM-PHASE")
    #st = Stream("ICTstream", dispatcher)
    #q  = st.addScalar("Charge", "SINEG01-DICT215:B1_CHARGE", 1, 0)
    #st.initialize()
    #st.start()
    #st.waitValueChange(10000)
    q = ChannelDouble("Charge", "SINEG01-DICT215:B1_CHARGE")
    q.initialize()
    q.monitored=True
bph.config.minValue   = -360.0
bph.config.maxValue   = 360.0
bph.config.precision  = 3
bph.config.rotation   = False
bph.config.deadband = 0.01
bph.config.save()
bph.initialize()
rph.initialize()
rph.monitored=True
bph0 = bph.read()
rph0 = rph.read()

#Creating averagers
rph_averager = create_averager(rph, nb, 0.1)
q_averager   = create_averager(q, nb, 0.1)
q_averager.monitored=True

#Record callback: uptate of output plot
def after_sample(record, scan):
    plt.getSeries(0).appendData(record.positions[0], record.readables[1].mean, record.readables[1].stdev)

#The scan loop
try:
    r = lscan(bph, (rph_averager, q_averager), start, stop, step, latency=lat, after_read = after_sample)
finally:
    rph.write(rph0)
    bph.close()
    rph.close()
    q.close()
    #st.close()
beamphase  = r.getPositions(0)
rfphase    = [val.mean for val in r.getReadable(0)]
rfphaserms = [val.stdev for val in r.getReadable(0)]
charge     = [val.mean for val in r.getReadable(1)]
chargerms  = [val.stdev for val in r.getReadable(1)]

#Fitting and plotting
i, a, b = 0, 0, 0
while charge[i] < (max(charge) * 0.20) : a = i; i = i + 1
while charge[i] < (max(charge) * 0.80) : b = i; i = i + 1
x = beamphase[a:b+1]
y = charge[a:b+1]
p = (a0, a1) = fit_polynomial(y, x, 1)
f = PolynomialFunction(p)
x1 = -a0 / a1 if a1 != 0 else 0.0
x2 = beamphase[charge.index(max(charge))]
x_fit = frange(x1, x2, (x2-x1), True)
y_fit = [f.value(val) for val in x_fit]
plt.addSeries(LinePlotErrorSeries("Fit", Color.blue))
plt.getSeries(1).setData(to_array(x_fit, 'd'), y_fit, [0.0] * len(x_fit))
plt.getSeries(1).setPointsVisible(False)
plt.addMarker(x1, plt.AxisId.X, "%3.2f" % x1, plt.getSeries(1).color)

#Elog entry
if do_elog:
    if get_option("Generated data file:\n" + get_exec_pars().path +"\n\n" + "Save to ELOG?", "YesNo") == "Yes":
        Laser = str(caget("SLGTV-LMTO-M055:MOT-KNOWN-POS"))
        log_msg = "Data file: " + get_exec_pars().path + "\n\n"
        log_msg = log_msg + "Laser:            " + Laser + "\n"
        if Laser == "Alcor":
            log_msg = log_msg + "Energy plate:     %0.2f" % caget("SLGTH01-LMRM-M074:MOT.RBV") + " deg \n"
        else:
            log_msg = log_msg + "Energy plate:     %0.2f" % caget("SLGJG-LMRM-M031:MOT.RBV") + " deg \n"
        if caget("SLGTV-LMTO-M053:MOT-ACT-POS") == "IRIS":
            log_msg = log_msg + "Collimator:       IRIS %0.2f" % caget("SLGTV-LAPP:SIZE-GET") + " mm \n"
        else:
            log_msg = log_msg + "Collimator:       " + str(caget("SLGTV-LMTO-M053:MOT-ACT-POS")) + "\n"
        log_msg = log_msg + "Charge:           %0.2f" % caget("SINEG01-DICT215:B1_CHARGE-OP") + " pC at %0.2f" % bph0 + " deg beam phase\n"
        attachments = []
        if plt is not None:
            sleep(0.1) #Give some time to plot to be finished - it is not sync  with acquisition
            file_name = os.path.abspath(Setup.getContextPath() + "/SchottkyScanPlot.png")
            plt.saveSnapshot(file_name , "png")
            attachments = [file_name]
        elog("Schottky scan", log_msg, attachments)

#Setting the return value
bph_ref_guess = x1
set_return([bph_ref_guess])