Startup

2016-06-27 15:15:24 +02:00
parent d5b17c2686
commit 9e56d92e46
1 changed files with 82 additions and 0 deletions
@@ -0,0 +1,82 @@
+"""
+Parameters:
+args = [prefix ]
+"""
+
+import  ch.psi.pshell.epics.ControlledVariable as ControlledVariable
+
+
+prefix = args[0]
+start = caget(prefix + ":SET-SCAN-START")
+stop = caget("PSHELL:scanstop")
+step = caget("PSHELL:scanstep")
+latency = 0.05
+
+
+print "Prefix = ", prefix
+print "Start = ", start
+print "Stop = ", stop
+print "Step = ", step
+
+
+rf_phase_var =  ControlledVariable("Phase", prefix + ":SET-VSUM-PHASE", prefix + ":GET-VSUM-PHASE")
+(rf_phase_var.config.minValue, rf_phase_var.config.maxValue)= (-180.0, 180.0)
+rf_phase_var.config.resolution = 0.001
+rf_phase_var.initialize()
+rf_ampl_readback =  Channel(prefix + ":GET-VSUM-AMPLT", type = 'd', alias='Amplitude Readback')
+
+#TODO: update the plot dinamically
+def after(rec):
+    pass
+
+try:
+    sim.step = step    #TODO: Remove if not simulated
+    r = lscan(rf_phase_var, [rf_ampl_readback, sim], start, stop, step , latency=latency, after_read = after)
+
+    #Update plot arrays
+    
+
+    f = r.getReadable(1) # r.getReadable(2)
+    f_ampl = r.getReadable(0) # r.getReadable(1)
+    xdata = r.getPositions(0) # r.getReadable(0)
+
+    #plot(r.getReadable(1), xdata = r.getPositions(0), title = "data")
+    phase_fit_max = None
+    try:
+        (amplitude, angular_frequency, phase, phase_fit_max) = hfit(f , xdata = xdata)
+    except:
+        raise Exception("Fit failure")        
+
+    start,end = min(xdata), max(xdata)
+    if not (start <=phase_fit_max <=end):             
+        raise Exception("Fit maximum outside scan range")
+
+    rf_phase_var.write(phase_fit_max)
+    time.sleep(latency)    
+    a0 = rf_ampl_readback.read()
+    power = 1.0 #TODO        
+finally:
+    #rf_phase_setpoint.close()   
+    #rf_phase_readback.close()    
+    rf_phase_var.close()
+    rf_ampl_readback.close()
+
+
+
+print ("------------------------------------")
+print ("Valid fit")
+
+
+
+energy_gain = amplitude
+phase_offset = - phase_fit_max
+amplitude_scale = energy_gain / a0
+power_scale = power / math.pow(a0,2)
+
+#caput(prefix + ":SET-VSUM-PHASE-OFFSET", phase_offset)
+#caput(prefix + ":SET-VSUM-AMPLT-SCALE", amplitude_scale)
+#caput(prefix + ":SET-VOLT-POWER-SCALE", power_scale)
+
+
+#set_return(r.print())
+set_return("Energy Gain: " + str(energy_gain) + "\nPhase Offset: " + str(phase_offset) +  "\nAmplitude Scale: " + str(amplitude_scale) + "\nPower Scale: "+ str(power_scale))