def wait_pv(pv, wait_val, max_timeout_sec=-1):
# wait on a pv to be a value until max_timeout (default forever)
# delay for pv to change
time.sleep(.01)
startTime = time.time()
while (True):
pv_val = pv.get()
if type(pv_val) == float:
if abs(pv_val - wait_val) < EPSILON:
return True
if (pv_val != wait_val):
if max_timeout_sec > -1:
curTime = time.time()
diffTime = curTime - startTime
if diffTime >= max_timeout_sec:
log.error(' *** ERROR: DROPPED IMAGES ***')
log.error(
' *** wait_pv(%s, %d, %5.2f reached max timeout. Return False'
% (pv.pvname, wait_val, max_timeout_sec))
return False
time.sleep(.01)
else:
return True
def set(global_PVs, params):
if (params.detector_prefix == '2bmbSP1:'):
log.info(' ')
log.info(' *** setup FLIR camera')
global_PVs['Cam1Acquire'].put(DetectorIdle)
pv.wait_pv(global_PVs['Cam1Acquire'], DetectorIdle, 2)
global_PVs['Cam1TriggerMode'].put('Off', wait=True)
global_PVs['Cam1TriggerSource'].put('Line2', wait=True)
global_PVs['Cam1TriggerOverlap'].put('ReadOut', wait=True)
global_PVs['Cam1ExposureMode'].put('Timed', wait=True)
global_PVs['Cam1TriggerSelector'].put('FrameStart', wait=True)
global_PVs['Cam1TriggerActivation'].put('RisingEdge', wait=True)
global_PVs['Cam1ImageMode'].put('Multiple')
global_PVs['Cam1ArrayCallbacks'].put('Enable')
global_PVs['Cam1FrameRateOnOff'].put(0)
global_PVs['Cam1AcquireTimeAuto'].put('Off')
global_PVs['Cam1AcquireTime'].put(float(params.exposure_time))
wait_time_sec = int(params.exposure_time) + 5
global_PVs['Cam1TriggerMode'].put('On', wait=True)
log.info(' *** setup FLIR camera: Done!')
else:
log.error('Detector %s is not supported' % params.detector_prefix)
return
def init(global_PVs, params):
if (params.detector_prefix == '2bmbSP1:'):
log.info(' ')
log.info(' *** init FLIR camera')
log.info(' *** *** set detector to idle')
global_PVs['Cam1Acquire'].put(DetectorIdle)
pv.wait_pv(global_PVs['Cam1Acquire'], DetectorIdle, 2)
log.info(' *** *** set detector to idle: Done')
time.sleep(2)
log.info(' *** *** set trigger mode to Off')
global_PVs['Cam1TriggerMode'].put('Off', wait=True) #
log.info(' *** *** set trigger mode to Off: done')
time.sleep(7)
log.info(' *** *** set image mode to single')
global_PVs['Cam1ImageMode'].put(
'Single', wait=True
) # here is where it crashes with (ValueError: invalid literal for int() with base 0: 'Single') Added 7 s delay before
log.info(' *** *** set image mode to single: done')
log.info(' *** *** set cam display to 1')
global_PVs['Cam1Display'].put(1)
log.info(' *** *** set cam display to 1: done')
log.info(' *** *** set cam acquire')
global_PVs['Cam1Acquire'].put(DetectorAcquire)
pv.wait_pv(global_PVs['Cam1Acquire'], DetectorAcquire, 2)
log.info(' *** *** set cam acquire: done')
log.info(' *** init FLIR camera: Done!')
else:
log.error('Detector %s is not supported' % params.detector_prefix)
return
def take_image(global_PVs, params):
log.info(' *** *** taking a single image')
nRow = global_PVs['Cam1SizeY_RBV'].get()
nCol = global_PVs['Cam1SizeX_RBV'].get()
image_size = nRow * nCol
global_PVs['Cam1NumImages'].put(1, wait=True)
global_PVs['Cam1TriggerMode'].put('Off', wait=True)
wait_time_sec = int(params.exposure_time) + 5
global_PVs['Cam1Acquire'].put(DetectorAcquire, wait=True, timeout=1000.0)
time.sleep(0.1)
if pv.wait_pv(global_PVs['Cam1Acquire'], DetectorIdle,
wait_time_sec) == False: # adjust wait time
global_PVs['Cam1Acquire'].put(DetectorIdle)
# Get the image loaded in memory
img_vect = global_PVs['Image'].get(count=image_size)
img = np.reshape(img_vect, [nRow, nCol])
pixelFormat = global_PVs['Cam1PixelFormat_RBV'].get(as_string=True)
if (pixelFormat == "Mono16"):
pixel_f = 16
elif (pixelFormat == "Mono8"):
pixel_f = 8
else:
log.error(' *** *** bit %s format not supported' % pixelFormat)
exit()
img_uint = np.mod(img, 2**pixel_f)
return img_uint
def adjust(what, params):
global_PVs = pv.init_general_PVs(params)
try:
detector_sn = global_PVs['Cam1SerialNumber'].get()
if ((detector_sn == None) or (detector_sn == 'Unknown')):
log.error('*** The detector with EPICS IOC prefix %s is down' %
params.detector_prefix)
log.error(' *** Failed!')
else:
log.info('*** The detector with EPICS IOC prefix %s and serial number %s is on' \
% (params.detector_prefix, detector_sn))
if (what == 'resolution'):
detector.init(global_PVs, params)
detector.set(global_PVs, params)
dark_field, white_field = detector.take_dark_and_white(
global_PVs, params)
find_resolution(params,
dark_field,
white_field,
angle_shift=-0.7)
config.update_sphere(params)
else:
if (params.image_pixel_size == None):
# resolution must be measured at least once
log.error(
' *** Detector resolution is not determined. Please run adjust resolution first!'
)
time.sleep(
2
) # to avoid a calling callback function/epics.ca.ChannelAccessException
exit()
else:
dark_field, white_field = detector.take_dark_and_white(
global_PVs, params)
if (what == 'focus'):
adjust_focus(params)
if (what == 'center'):
adjust_center(params, dark_field, white_field)
if (what == 'roll'):
adjust_roll(params,
dark_field,
white_field,
angle_shift=-0.7)
if (what == 'pitch'):
adjust_pitch(params,
dark_field,
white_field,
angle_shift=-0.7)
if (what == 'roll') or (what == 'pitch'):
# align center again for higher accuracy
adjust_center(params, dark_field, white_field)
config.update_sphere(params)
except KeyError:
log.error(' *** Some PV assignment failed!')
pass
def init_general_PVs(params):
global_PVs = {}
global_PVs['ShutterOpen'] = PV(params.shutter_open_pv_name + '.VAL')
global_PVs['ShutterClose'] = PV(params.shutter_close_pv_name + '.VAL')
global_PVs['ShutterStatus'] = PV(params.shutter_status_pv_name + '.VAL')
global_PVs['SampleX'] = PV(params.sample_x_pv_name + '.VAL')
global_PVs['SampleXSet'] = PV(params.sample_x_pv_name + '.SET')
global_PVs['SampleY'] = PV(params.sample_y_pv_name + '.VAL')
global_PVs['SampleYSet'] = PV(params.sample_y_pv_name + '.SET')
global_PVs['Rotation'] = PV(params.rotation_pv_name + '.VAL')
global_PVs['RotationRBV'] = PV(params.rotation_pv_name + '.RBV')
global_PVs['RotationCnen'] = PV(params.rotation_pv_name + '.CNEN')
global_PVs['RotationAccl'] = PV(params.rotation_pv_name + '.ACCL')
global_PVs['RotationStop'] = PV(params.rotation_pv_name + '.STOP')
global_PVs['RotationSet'] = PV(params.rotation_pv_name + '.SET')
global_PVs['RotationVelo'] = PV(params.rotation_pv_name + '.VELO')
global_PVs['SampleXCent'] = PV(params.sample_x_center_pv_name + '.VAL')
global_PVs['SampleZCent'] = PV(params.sample_z_center_pv_name + '.VAL')
global_PVs['SamplePitch'] = PV(params.sample_pitch_pv_name + '.VAL')
global_PVs['SampleRoll'] = PV(params.sample_roll_pv_name + '.VAL')
global_PVs['Focus'] = PV(params.focus_pv_name + '.VAL')
global_PVs['ImagePixelSize'] = PV(params.image_pixel_size_pv_name + '.VAL')
# detector pv's
camera_prefix = params.detector_prefix + 'cam1:'
global_PVs['CamManufacturer'] = PV(camera_prefix + 'Manufacturer_RBV')
global_PVs['CamModel'] = PV(camera_prefix + 'Model_RBV')
global_PVs['Cam1SerialNumber'] = PV(camera_prefix + 'SerialNumber_RBV')
global_PVs['Cam1ImageMode'] = PV(camera_prefix + 'ImageMode')
global_PVs['Cam1ArrayCallbacks'] = PV(camera_prefix + 'ArrayCallbacks')
global_PVs['Cam1AcquirePeriod'] = PV(camera_prefix + 'AcquirePeriod')
global_PVs['Cam1SoftwareTrigger'] = PV(camera_prefix + 'SoftwareTrigger')
global_PVs['Cam1AcquireTime'] = PV(camera_prefix + 'AcquireTime')
global_PVs['Cam1FrameType'] = PV(camera_prefix + 'FrameType')
global_PVs['Cam1AttributeFile'] = PV(camera_prefix + 'NDAttributesFile')
global_PVs['Cam1SizeX'] = PV(camera_prefix + 'SizeX')
global_PVs['Cam1SizeY'] = PV(camera_prefix + 'SizeY')
global_PVs['Cam1NumImages'] = PV(camera_prefix + 'NumImages')
global_PVs['Cam1TriggerMode'] = PV(camera_prefix + 'TriggerMode')
global_PVs['Cam1Acquire'] = PV(camera_prefix + 'Acquire')
global_PVs['Cam1SizeX_RBV'] = PV(camera_prefix + 'SizeX_RBV')
global_PVs['Cam1SizeY_RBV'] = PV(camera_prefix + 'SizeY_RBV')
global_PVs['Cam1MaxSizeX_RBV'] = PV(camera_prefix + 'MaxSizeX_RBV')
global_PVs['Cam1MaxSizeY_RBV'] = PV(camera_prefix + 'MaxSizeY_RBV')
global_PVs['Cam1PixelFormat_RBV'] = PV(camera_prefix + 'PixelFormat_RBV')
image_prefix = params.detector_prefix + 'image1:'
global_PVs['Image'] = PV(image_prefix + 'ArrayData')
global_PVs['Cam1Display'] = PV(image_prefix + 'EnableCallbacks')
manufacturer = global_PVs['CamManufacturer'].get(as_string=True)
model = global_PVs['CamModel'].get(as_string=True)
if model == 'Oryx ORX-10G-51S5M':
log.info('Detector %s model %s:' % (manufacturer, model))
global_PVs['Cam1AcquireTimeAuto'] = PV(params.detector_prefix +
'AcquireTimeAuto')
global_PVs['Cam1FrameRateOnOff'] = PV(params.detector_prefix +
'FrameRateEnable')
global_PVs['Cam1TriggerSource'] = PV(params.detector_prefix +
'TriggerSource')
global_PVs['Cam1TriggerOverlap'] = PV(params.detector_prefix +
'TriggerOverlap')
global_PVs['Cam1ExposureMode'] = PV(params.detector_prefix +
'ExposureMode')
global_PVs['Cam1TriggerSelector'] = PV(params.detector_prefix +
'TriggerSelector')
global_PVs['Cam1TriggerActivation'] = PV(params.detector_prefix +
'TriggerActivation')
else:
log.error('Detector %s model %s is not supported' %
(manufacturer, model))
return None
return global_PVs
def adjust_center(params, dark_field, white_field):
global_PVs = pv.init_general_PVs(params)
log.warning(' *** Adjusting center ***')
for ang in [params.adjust_center_angle_1, params.adjust_center_angle_2]:
log.warning(' *** take 3 spheres angular %f deg ***' % float(ang))
#sphere 0
log.info(' *** sphere 0')
log.info(' *** *** moving rotary stage to %f deg position ***' %
float(0))
global_PVs["Rotation"].put(float(0), wait=True, timeout=600.0)
log.error(' *** *** acquire sphere at %f deg position ***' %
float(0))
sphere_0 = util.normalize(detector.take_image(global_PVs, params),
white_field, dark_field)
#sphere 1
log.info(' *** sphere 1')
log.info(' *** *** moving rotary stage to %f deg position ***' %
float(ang))
global_PVs["Rotation"].put(float(ang), wait=True, timeout=600.0)
log.error(' *** *** acquire sphere at %f deg position ***' %
float(ang))
sphere_1 = util.normalize(detector.take_image(global_PVs, params),
white_field, dark_field)
#sphere 2
log.info(' *** sphere 2')
log.info(' *** *** moving rotary stage to %f deg position ***' %
float(2 * ang))
global_PVs["Rotation"].put(float(2 * ang), wait=True, timeout=600.0)
log.error(' *** *** acquire sphere at %f deg position ***' %
float(2 * ang))
sphere_2 = util.normalize(detector.take_image(global_PVs, params),
white_field, dark_field)
# find shifts
shift0 = register_translation(sphere_1, sphere_0, 100)[0][1]
shift1 = register_translation(sphere_2, sphere_1, 100)[0][1]
a = ang * np.pi / 180
# x=-(1/4) (d1+d2-2 d1 Cos[a]) Csc[a/2]^2,
x = -(1 / 4) * (shift0 + shift1 - 2 * shift0 * np.cos(a)) * 1 / np.sin(
a / 2)**2
# r = 1/2 Csc[a/2]^2 Csc[a] Sqrt[(d1^2+d2^2-2 d1 d2 Cos[a]) Sin[a/2]^2]
r = 1 / 2 * 1 / np.sin(a / 2)**2 * 1 / np.sin(a) * np.sqrt(
np.abs((shift0**2 + shift1**2 - 2 * shift0 * shift1 * np.cos(a)) *
np.sin(a / 2)**2))
# g = ArcCos[((-d1-d2+2 d1 Cos[a]) Sin[a])/(2 Sqrt[(d1^2+d2^2-2 d1 d2 Cos[a]) Sin[a/2]^2])]
g = np.arccos(
((-shift0 - shift1 + 2 * shift0 * np.cos(a)) * np.sin(a)) /
(2 * np.sqrt(
np.abs(
(shift0**2 + shift1**2 - 2 * shift0 * shift1 * np.cos(a)) *
np.sin(a / 2)**2))))
y = r * np.sin(g) * np.sign(shift0)
# find center of mass
cmass_0 = util.center_of_mass(sphere_0)
log.info(' ')
log.info(
' *** position of the initial sphere wrt to the rotation center (%f,%f) ***'
% (x, y))
log.info(' *** center of mass for the initial sphere (%f,%f) ***' %
(cmass_0[1], cmass_0[0]))
log.info(
' *** moving sphere to the position of the rotation center ***')
if (params.ask):
if util.yes_or_no(' *** Yes or No'):
move_center(params, cmass_0, x, y)
check_center(params, white_field, dark_field)
else:
log.warning(' No motion ')
exit()
else:
move_center(params, cmass_0, x, y)
check_center(params, white_field, dark_field)