150 lines
3.7 KiB
Python
Executable File
150 lines
3.7 KiB
Python
Executable File
# Tool to align the solenoid on the gun.
|
|
# S. Bettoni, A. Gobbo, D. Voulot
|
|
# 06/06/2016
|
|
|
|
|
|
# Procedure:
|
|
# switch off all the magnets between the gun solenoid and the screen or BPM used for the measurement
|
|
# change the current of the gun soleoid
|
|
# look at the centroid position (BPM or screen) downstream of the gun.
|
|
|
|
|
|
|
|
import org.apache.commons.math3.linear.Array2DRowRealMatrix as Matrix
|
|
import ch.psi.utils.Convert.toBidimensional as mono_to_bidi
|
|
import datetime
|
|
|
|
zero_solenoids=False
|
|
do_elog = False
|
|
|
|
if get_context().source == CommandSource.ui:
|
|
I1 = 95.0
|
|
I2 = 100.0
|
|
dI = 1.0
|
|
settling_time = 0.1
|
|
plot_image = False
|
|
number_images = 1
|
|
use_background = True
|
|
multiple_background = True
|
|
number_backgrounds = 5
|
|
centroid_excursion_plot = True
|
|
else:
|
|
centroid_excursion_plot = False
|
|
|
|
print I1
|
|
print I2
|
|
print dI
|
|
print settling_time
|
|
print plot_image
|
|
print number_images
|
|
print use_background
|
|
print multiple_background
|
|
print number_backgrounds
|
|
|
|
|
|
|
|
|
|
plot_name = datetime.datetime.fromtimestamp(time.time()).strftime('%H%M%S')
|
|
|
|
cam_x = Channel("SINEG01-DSCR190:profile.X_stats.com", alias = "cam_x")
|
|
cam_y = Channel("SINEG01-DSCR190:profile.Y_stats.com", alias = "cam_y")
|
|
|
|
profile_x = Channel("SINEG01-DSCR190:profile.X", alias = "profile_x")
|
|
profile_y = Channel("SINEG01-DSCR190:profile.Y", alias = "profile_y")
|
|
|
|
cam_raw_data = Channel("SINEG01-DSCR190:data")
|
|
|
|
|
|
class CameraImage(ReadableMatrix):
|
|
def read(self):
|
|
raw = cam_raw_data.read()
|
|
return Matrix(mono_to_bidi(raw, self.getHeight(), self.getWidth())).transpose().getData() #data is transposed
|
|
#return mono_to_bidi(raw, self.getWidth(), self.getHeight())
|
|
|
|
def getWidth(self):
|
|
return self._width
|
|
|
|
def getHeight(self):
|
|
return self._height
|
|
|
|
cam_img = CameraImage()
|
|
cam_img._width = len(profile_x.read())
|
|
cam_img._height = len(profile_y.read())
|
|
|
|
|
|
def ccr(mag):
|
|
n = 1
|
|
while n > 0:
|
|
sleep(0.5)
|
|
n = caget(mag + ":I-COMP")
|
|
|
|
def laser_on():
|
|
caput("SIN-TIMAST-TMA:Beam-Las-Delay-Sel", 0)
|
|
|
|
def laser_off():
|
|
caput("SIN-TIMAST-TMA:Beam-Las-Delay-Sel", 1)
|
|
|
|
|
|
def run_pipeline():
|
|
caput("camtool...:start")
|
|
|
|
while !caget("camtool...:done"):
|
|
sleep(0.1)
|
|
|
|
caget("camtool....:data")
|
|
#caget("camtool....:X_stats.com")
|
|
#caget("camtool....:X_stats.com")
|
|
|
|
|
|
|
|
# Switch off magnets
|
|
mag = [ "SINEG01-MCRX120","SINEG01-MCRY120",
|
|
"SINEG01-MQUA140",
|
|
"SINEG01-MQUA150",
|
|
"SINEG01-MCRX160","SINEG01-MCRY160",
|
|
"SINEG01-MCRX180","SINEG01-MCRY180",
|
|
"SINEG01-MCRX200","SINEG01-MCRY200",
|
|
"SINEG01-MCRX220","SINEG01-MCRY220",
|
|
"SINEG01-MQUA310",
|
|
"SINEG01-MQUA320" ]
|
|
if zero_solenoids:
|
|
for m in mag:
|
|
caput(m + ":I-SET", 0.0)
|
|
for m in mag:
|
|
ccr(m)
|
|
|
|
# add here gun phase setting see wiki page
|
|
|
|
def br():
|
|
pass
|
|
def ar():
|
|
pass
|
|
|
|
r = None
|
|
laser_on()
|
|
try:
|
|
sensors = [cam_x, cam_y, cam_img] if plot_image else [cam_x, cam_y]
|
|
r = lscan(gun_solenoid, sensors , I1, I2, dI, settling_time, before_read = br, after_read = ar)
|
|
finally:
|
|
laser_off()
|
|
|
|
# take the result of the scan and generate convex hull plot
|
|
if centroid_excursion_plot:
|
|
(hx,hy)=add_convex_hull_plot ("Centroid excursion", r.getReadable(0),r.getReadable(1), plot_name)
|
|
else:
|
|
(hx,hy)= convex_hull(x=r.getReadable(0), y=r.getReadable(1))
|
|
hx.append(hx[0]); hy.append(hy[0])
|
|
|
|
# save the entry in the logbook
|
|
if do_elog:
|
|
msg = str(r)
|
|
msg = msg + "\nFile: " + get_context().path
|
|
msg = msg + "\n\n" + r.print()
|
|
elog("Gun solenoid current scan", msg , get_plot_snapshots())
|
|
|
|
|
|
set_return([r, hx,hy])
|
|
|
|
|
|
|