def beamline_with_3_empty_parameters():
"""
Returns: a beamline with three empty parameters, all in a mode
"""
one = EmptyBeamlineParameter("one")
two = EmptyBeamlineParameter("two")
three = EmptyBeamlineParameter("three")
beamline_parameters = [one, two, three]
mode = BeamlineMode("all", [
beamline_parameter.name
for beamline_parameter in beamline_parameters
])
naught_and_two = BeamlineMode(
"components1and3",
[beamline_parameters[0].name, beamline_parameters[2].name])
two = BeamlineMode("just2", [beamline_parameters[2].name])
beamline = Beamline([], beamline_parameters, [],
[mode, naught_and_two, two])
beamline.active_mode = mode.name
return beamline_parameters, beamline
def test_GIVEN_beamline_with_engineering_correction_containing_a_mode_update_correction_WHEN_update_mode_THEN_correct_readback_updated_fired(self):
comp = Component("comp", PositionAndAngle(0, 0, 90))
param = AxisParameter("comp", comp, ChangeAxis.POSITION)
mock_axis = create_mock_axis("mock", 0, 1)
driver = IocDriver(comp, ChangeAxis.POSITION, mock_axis, engineering_correction=self.mode_selection_correction)
mock_axis.trigger_rbv_change()
bl = Beamline(components=[comp], beamline_parameters=[param], drivers=[driver], modes=[self.mode1, self.mode2])
bl.active_mode = self.mode1.name
assert_that(self.correction_update.correction, is_(self.mode1_offset))
bl.active_mode = self.mode2.name
assert_that(self.correction_update.correction, is_(self.mode2_offset))
def beamline_with_one_mode_init_param_in_mode_and_at_off_init(
init_sm_angle, off_init, param_name):
super_mirror = ReflectingComponent(
"super mirror", PositionAndAngle(z=10, y=0, angle=90))
smangle = AxisParameter(param_name, super_mirror, ChangeAxis.ANGLE)
beamline_mode = BeamlineMode("mode name", [smangle.name],
{smangle.name: init_sm_angle})
beamline = Beamline([super_mirror], [smangle], [], [beamline_mode])
beamline.active_mode = beamline_mode.name
smangle.sp = off_init
return Beamline([super_mirror], [smangle], [], [beamline_mode])
def test_GIVEN_beam_line_where_autosave_and_engineering_correction_on_displacement_WHEN_init_THEN_beamline_is_at_given_place(self, param_float_autosave):
expected_setpoint = 1.0
multiple = 2.0
param_float_autosave.read_parameter.return_value = expected_setpoint
offset = expected_setpoint / multiple
comp = Component("comp", PositionAndAngle(0.0, 0, 90))
param = AxisParameter("param", comp, ChangeAxis.POSITION, autosave=True)
axis = create_mock_axis("MOT:MTR0101", offset + expected_setpoint, 1)
driver = IocDriver(comp, ChangeAxis.POSITION, axis,
engineering_correction=UserFunctionCorrection(lambda sp: sp / multiple))
nr_mode = BeamlineMode("NR", [param.name], {})
bl = Beamline([comp], [param], [driver], [nr_mode])
bl.active_mode = nr_mode.name
result = comp.beam_path_set_point.axis[ChangeAxis.POSITION].get_displacement()
assert_that(result, is_(close_to(expected_setpoint, 1e-6)))
def create_beamline():
"""
Returns: example beamline
"""
perp_to_floor = 90.0
beam_start = PositionAndAngle(y=0, z=0, angle=-2.5)
s0 = Component("s0", setup=PositionAndAngle(0, 0, perp_to_floor))
s1 = Component("s1", setup=PositionAndAngle(0, 1, perp_to_floor))
frame_overlap_mirror = ReflectingComponent("FOM",
setup=PositionAndAngle(
0, 2, perp_to_floor))
frame_overlap_mirror.beam_path_set_point.is_in_beam = False
polarising_mirror = ReflectingComponent("Polarising mirror",
setup=PositionAndAngle(
0, 3, perp_to_floor))
polarising_mirror.beam_path_set_point.is_in_beam = False
s2 = Component("s2", setup=PositionAndAngle(0, 4, perp_to_floor))
ideal_sample_point = ReflectingComponent("Ideal Sample Point",
setup=PositionAndAngle(
0, 5, perp_to_floor))
s3 = Component("s3", setup=PositionAndAngle(0, 6, perp_to_floor))
analyser = ReflectingComponent("analyser",
setup=PositionAndAngle(0, 7, perp_to_floor))
analyser.beam_path_set_point.is_in_beam = False
s4 = Component("s4", setup=PositionAndAngle(0, 8, perp_to_floor))
detector = Component("detector",
setup=PositionAndAngle(0, 10, perp_to_floor))
theta = AxisParameter("theta", ideal_sample_point, ChangeAxis.ANGLE)
nr_mode = BeamlineMode("NR", ["theta"])
beamline = Beamline([
s0, s1, frame_overlap_mirror, polarising_mirror, s2,
ideal_sample_point, s3, analyser, s4, detector
], [theta], [], [nr_mode], beam_start)
beamline.active_mode = nr_mode.name
return beamline
def beamline_sm_theta_detector(sm_angle,
theta,
det_offset=0,
autosave_theta_not_offset=True,
beam_angle=0.0,
sm_angle_engineering_correction=False):
"""
Create beamline with supermirror, theta and a tilting detector.
Args:
sm_angle (float): The initialisation value for supermirror angle
theta (float): The initialisation value for theta
det_offset (float): The initialisation value for detector offset
autosave_theta_not_offset (bool): true to autosave theta and not the offset, false otherwise
beam_angle (float): angle of the beam, worked out as the angle the components run along + 90
Returns: beamline, axes
"""
beam_start = PositionAndAngle(0.0, 0.0, 0.0)
perp_to_floor_angle_in_mantid = 90 + beam_angle
# COMPONENTS
z_sm_to_sample = 1
z_sample_to_det = 2
sm_comp = ReflectingComponent(
"sm_comp", PositionAndAngle(0.0, 0, perp_to_floor_angle_in_mantid))
detector_comp = TiltingComponent(
"detector_comp",
PositionAndAngle(0.0, z_sm_to_sample + z_sample_to_det,
perp_to_floor_angle_in_mantid))
theta_comp = ThetaComponent(
"theta_comp",
PositionAndAngle(0.0, z_sm_to_sample,
perp_to_floor_angle_in_mantid))
theta_comp.add_angle_to(detector_comp)
comps = [sm_comp, theta_comp, detector_comp]
# BEAMLINE PARAMETERS
sm_angle_param = AxisParameter("sm_angle", sm_comp, ChangeAxis.ANGLE)
theta_param = AxisParameter("theta",
theta_comp,
ChangeAxis.ANGLE,
autosave=autosave_theta_not_offset)
detector_position_param = AxisParameter(
"det_pos",
detector_comp,
ChangeAxis.POSITION,
autosave=not autosave_theta_not_offset)
detector_angle_param = AxisParameter("det_angle", detector_comp,
ChangeAxis.ANGLE)
params = [
sm_angle_param, theta_param, detector_position_param,
detector_angle_param
]
# DRIVERS
# engineering correction
if sm_angle_engineering_correction:
grid_data_provider = GridDataFileReader("linear_theta")
grid_data_provider.variables = ["Theta"]
grid_data_provider.points = np.array([[
-90,
], [
0.0,
], [
90.0,
]])
grid_data_provider.corrections = np.array([-45, 0.0, 45])
grid_data_provider.read = lambda: None
correction = InterpolateGridDataCorrectionFromProvider(
grid_data_provider, theta_param)
size_of_correction = theta / 2.0
else:
correction = None
size_of_correction = 0
# setup motors
beam_angle_after_sample = theta * 2 + sm_angle * 2
supermirror_segment = (z_sm_to_sample, sm_angle)
theta_segment = (z_sample_to_det, theta)
reflection_offset = DataMother._calc_reflection_offset(
beam_angle, [supermirror_segment, theta_segment])
sm_axis = create_mock_axis("MOT:MTR0101",
sm_angle + size_of_correction, 1)
det_axis = create_mock_axis("MOT:MTR0104",
reflection_offset + det_offset, 1)
det_angle_axis = create_mock_axis(
"MOT:MTR0105", beam_start.angle + beam_angle_after_sample, 1)
axes = {
"sm_axis": sm_axis,
"det_axis": det_axis,
"det_angle_axis": det_angle_axis
}
drives = [
IocDriver(sm_comp,
ChangeAxis.ANGLE,
sm_axis,
engineering_correction=correction),
IocDriver(detector_comp, ChangeAxis.POSITION, det_axis),
IocDriver(detector_comp, ChangeAxis.ANGLE, det_angle_axis)
]
# MODES
nr_inits = {}
nr_mode = BeamlineMode("NR", [param.name for param in params],
nr_inits)
modes = [nr_mode]
beam_start = PositionAndAngle(0.0, 0.0, 0.0)
bl = Beamline(comps, params, drives, modes, beam_start)
bl.active_mode = nr_mode.name
return bl, axes