Creation

script/PbpgPosScan.py

@@ -0,0 +1,34 @@
+from mathutils import fit_polynomial
+from mathutils import PolynomialFunction
+import math
+from plotutils import plot_function
+print "Starting"
+
+
+#Creating averaging devices
+av_adc_xh1 = create_averager(adc_xh1, count = 10, interval = -1, name = "av_adc_xh1")
+av_adc_xh2 = create_averager(adc_yh2, count = 10, interval = -1, name = "av_adc_xh2")
+av_adc_xh2.monitored = True
+
+#The actuals scan
+r=lscan(pbpg_mx, [av_adc_xh1, av_adc_xh2], 0.0, 0.5, 10, latency = 0.0)
+
+#Fitting 
+values = to_array(r.getReadable(0), 'd')
+positions = r.getPositions(0)
+pars_polynomial = (a0, a1, a2) = fit_polynomial(values, positions, 2)
+
+#Writing metadata to data file
+path = get_exec_pars().scanPath
+set_attribute(path, "a0", a0)      
+set_attribute(path, "a1", a1)      
+set_attribute(path, "a2", a2)      
+
+#Plotting fit and writing fitting parameters
+outp = "a0="+ ("%0.4f" % a0) + "a1="+  ("%0.4f" % a1) + "a2="+  ("%0.4f" % a2)
+print outp
+p = get_plots()[0]
+p.addText((min(positions) + max(positions))/2, max(values), outp, Color.BLACK)
+plot_function(p, PolynomialFunction(pars_polynomial), "Fit",positions, show_points = False, show_lines = True, color = Color.BLUE)
+
+

script/local.groovy

@@ -0,0 +1,3 @@
+///////////////////////////////////////////////////////////////////////////////////////////////////
+// Deployment specific global definitions - executed after startup.groovy
+///////////////////////////////////////////////////////////////////////////////////////////////////

script/local.js

@@ -0,0 +1,4 @@
+///////////////////////////////////////////////////////////////////////////////////////////////////
+// Deployment specific global definitions - executed after startup.js
+///////////////////////////////////////////////////////////////////////////////////////////////////
+

script/local.py

@@ -0,0 +1,99 @@
+###################################################################################################
+#  Deployment specific global definitions - executed after startup.py
+###################################################################################################
+
+#Device pool customization
+pbpg_mx.setTrustedWrite(False)
+pbpg_my.setTrustedWrite(False)
+
+
+#Utilities
+def fit(ydata, xdata = None):
+    """
+    Gaussian fit
+    """
+    if xdata is None:
+        xdata = frange(0, len(ydata), 1)
+    #ydata = to_list(ydata)
+    #xdata = to_list(xdata)
+    max_y= max(ydata)
+    index_max = ydata.index(max_y)
+    max_x= xdata[index_max]
+    print "Max index:" + str(index_max),
+    print " x:" + str(max_x),
+    print " y:" + str(max_y)
+    gaussians = fit_gaussians(ydata, xdata, [index_max,])
+    (norm, mean, sigma) = gaussians[0]
+    p = plot([ydata],["data"],[xdata], title="Fit" )[0]
+    fitted_gaussian_function = Gaussian(norm, mean, sigma)
+    scale_x = [float(min(xdata)), float(max(xdata)) ]
+    points = max((len(xdata)+1), 100)
+    resolution = (scale_x[1]-scale_x[0]) / points
+    fit_y = []
+    fit_x = frange(scale_x[0],scale_x[1],resolution, True)
+    for x in fit_x:
+        fit_y.append(fitted_gaussian_function.value(x))
+    p.addSeries(LinePlotSeries("fit"))
+    p.getSeries(1).setData(fit_x, fit_y)
+
+    if abs(mean - xdata[index_max]) < ((scale_x[0] + scale_x[1])/2):
+        print "Mean -> " +  str(mean)
+        p.addMarker(mean, None, "Mean="+str(round(norm,2)), Color.MAGENTA.darker())
+        return (norm, mean, sigma)
+    else:
+        p.addMarker(max_x, None, "Max="+str(round(max_x,2)), Color.GRAY)
+        print "Invalid gaussian fit: " +  str(mean)
+        return (None, None, None)
+
+
+def elog(title, message, attachments = [], author = None, category = "Info", domain = "", logbook = "SwissFEL commissioning data", encoding=1):
+    """
+    Add entry to ELOG.
+    """
+    if author is None:
+        author = "pshell" #get_context().user.name
+    typ = "pshell"
+    entry = ""
+
+    cmd =           'G_CS_ELOG_add -l "' + logbook       + '" '
+    cmd =     cmd + '-a "Author='        + author        + '" '
+    cmd =     cmd + '-a "Type='          + typ           + '" '
+    cmd =     cmd + '-a "Entry='         + entry         + '" '
+    cmd =     cmd + '-a "Title='         + title         + '" '
+    cmd =     cmd + '-a "Category='      + category      + '" '
+    cmd =     cmd + '-a "Domain='        + domain        + '" '
+    for attachment in attachments:
+        cmd = cmd + '-f "'               + attachment    + '" '
+    cmd =     cmd + '-n '                + str(encoding)
+    cmd =     cmd + ' "'                 + message       + '"'
+    #print cmd
+    #os.system (cmd)
+    #print os.popen(cmd).read()
+    import subprocess
+    proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True)
+    (out, err) = proc.communicate()
+    if (err is not None) and err!="":
+        raise Exception(err)
+    print out
+
+    
+def bsget(channel):
+    """
+    """
+    st = Stream(None, dispatcher)
+    try:    
+        st.addScalar(channel, channel, 10, 0)
+        st.initialize()
+        st.start();
+        st.waitValueNot(None, 5000)
+        return st.getValue(channel)
+    finally:
+        st.close()
+
+#Machine utilities    
+def is_timing_ok():
+    return caget("SIN-TIMAST-TMA:SOS-COUNT-CHECK") == 0
+
+def get_repetition_rate():
+    return caget("SIN-TIMAST-TMA:Evt-15-Freq-I")
+