def test_that_model_initialises_with_correct_values(self):
beam_centre_model = BeamCentreModel()
self.assertEqual(beam_centre_model.max_iterations, 10)
self.assertEqual(beam_centre_model.r_min, 60)
self.assertEqual(beam_centre_model.r_max, 280)
self.assertEqual(beam_centre_model.left_right, True)
self.assertEqual(beam_centre_model.up_down, True)
self.assertEqual(beam_centre_model.tolerance, 0.000125)
self.assertEqual(beam_centre_model.lab_pos_1, '')
self.assertEqual(beam_centre_model.lab_pos_2, '')
self.assertEqual(beam_centre_model.hab_pos_2, '')
self.assertEqual(beam_centre_model.hab_pos_1, '')
self.assertEqual(beam_centre_model.scale_1, 1000)
self.assertEqual(beam_centre_model.scale_2, 1000)
def test_that_correct_values_are_set_for_LOQ(self):
beam_centre_model = BeamCentreModel()
beam_centre_model.reset_to_defaults_for_instrument(SANSInstrument.LOQ)
self.assertEqual(beam_centre_model.r_max, 200)
def test_that_correct_values_are_set_for_LARMOR(self):
beam_centre_model = BeamCentreModel()
beam_centre_model.reset_to_defaults_for_instrument(
SANSInstrument.LARMOR)
self.assertEqual(beam_centre_model.scale_1, 1.0)
def test_that_can_update_model_values(self):
beam_centre_model = BeamCentreModel()
beam_centre_model.scale_2 = 1.0
self.assertEqual(beam_centre_model.scale_2, 1.0)
class BeamCentrePresenter(object):
class ConcreteBeamCentreListener(BeamCentre.BeamCentreListener):
def __init__(self, presenter):
self._presenter = presenter
def on_run_clicked(self):
self._presenter.on_run_clicked()
class CentreFinderListener(WorkHandler.WorkListener):
def __init__(self, presenter):
super(BeamCentrePresenter.CentreFinderListener, self).__init__()
self._presenter = presenter
def on_processing_finished(self, result):
self._presenter.on_processing_finished_centre_finder(result)
def on_processing_error(self, error):
self._presenter.on_processing_error_centre_finder(error)
def __init__(self,
parent_presenter,
SANSCentreFinder,
work_handler=None,
beam_centre_model=None):
self._view = None
self._parent_presenter = parent_presenter
self._work_handler = WorkHandler(
) if not work_handler else work_handler
self._logger = Logger("SANS")
self._beam_centre_model = BeamCentreModel(
SANSCentreFinder) if not beam_centre_model else beam_centre_model
def set_view(self, view):
if view:
self._view = view
# Set up run listener
listener = BeamCentrePresenter.ConcreteBeamCentreListener(self)
self._view.add_listener(listener)
# Set the default gui
self._view.set_options(self._beam_centre_model)
# Connect view signals
self.connect_signals()
def connect_signals(self):
self._view.r_min_line_edit.textChanged.connect(
self._validate_radius_values)
self._view.r_max_line_edit.textChanged.connect(
self._validate_radius_values)
def on_update_instrument(self, instrument):
self._beam_centre_model.set_scaling(instrument)
self._view.on_update_instrument(instrument)
def on_update_rows(self):
self._beam_centre_model.reset_inst_defaults(
self._parent_presenter.instrument)
def on_processing_finished_centre_finder(self, result):
# Enable button
self._view.set_run_button_to_normal()
# Update Centre Positions in model and GUI
if self._beam_centre_model.update_lab:
self._beam_centre_model.lab_pos_1 = result['pos1']
self._beam_centre_model.lab_pos_2 = result['pos2']
self._view.lab_pos_1 = self._beam_centre_model.lab_pos_1 * self._beam_centre_model.scale_1
self._view.lab_pos_2 = self._beam_centre_model.lab_pos_2 * self._beam_centre_model.scale_2
if self._beam_centre_model.update_hab:
self._beam_centre_model.hab_pos_1 = result['pos1']
self._beam_centre_model.hab_pos_2 = result['pos2']
self._view.hab_pos_1 = self._beam_centre_model.hab_pos_1 * self._beam_centre_model.scale_1
self._view.hab_pos_2 = self._beam_centre_model.hab_pos_2 * self._beam_centre_model.scale_2
def on_processing_error_centre_finder(self, error):
self._logger.warning(
"There has been an error. See more: {}".format(error))
self._view.set_run_button_to_normal()
def on_processing_error(self, error):
self._view.set_run_button_to_normal()
def on_run_clicked(self):
self._work_handler.wait_for_done()
# Get the state information for the first row.
state = self._parent_presenter.get_state_for_row(0)
if not state:
self._logger.information(
"You can only calculate the beam centre if a user file has been loaded and there"
"valid sample scatter entry has been provided in the selected row."
)
return
# Disable the button
self._view.set_run_button_to_processing()
#Update model
self._update_beam_model_from_view()
# Run the task
listener = BeamCentrePresenter.CentreFinderListener(self)
state_copy = copy.copy(state)
self._work_handler.process(listener,
self._beam_centre_model.find_beam_centre, 0,
state_copy)
def _update_beam_model_from_view(self):
self._beam_centre_model.r_min = self._view.r_min
self._beam_centre_model.r_max = self._view.r_max
self._beam_centre_model.max_iterations = self._view.max_iterations
self._beam_centre_model.tolerance = self._view.tolerance
self._beam_centre_model.left_right = self._view.left_right
self._beam_centre_model.verbose = self._view.verbose
self._beam_centre_model.COM = self._view.COM
self._beam_centre_model.up_down = self._view.up_down
self._beam_centre_model.lab_pos_1 = self._view.lab_pos_1 / self._beam_centre_model.scale_1
self._beam_centre_model.lab_pos_2 = self._view.lab_pos_2 / self._beam_centre_model.scale_2
self._beam_centre_model.hab_pos_1 = self._view.hab_pos_1 / self._beam_centre_model.scale_1
self._beam_centre_model.hab_pos_2 = self._view.hab_pos_2 / self._beam_centre_model.scale_2
self._beam_centre_model.q_min = self._view.q_min
self._beam_centre_model.q_max = self._view.q_max
self._beam_centre_model.component = self._view.component
self._beam_centre_model.update_hab = self._view.update_hab
self._beam_centre_model.update_lab = self._view.update_lab
def update_centre_positions(self, state_model):
lab_pos_1 = getattr(state_model, 'lab_pos_1')
lab_pos_2 = getattr(state_model, 'lab_pos_2')
hab_pos_1 = getattr(state_model, 'hab_pos_1') if getattr(
state_model, 'hab_pos_1') else lab_pos_1
hab_pos_2 = getattr(state_model, 'hab_pos_2') if getattr(
state_model, 'hab_pos_2') else lab_pos_2
self._view.lab_pos_1 = lab_pos_1
self._view.lab_pos_2 = lab_pos_2
self._view.hab_pos_1 = hab_pos_1
self._view.hab_pos_2 = hab_pos_2
def update_hab_selected(self):
self._beam_centre_model.update_hab = True
self._beam_centre_model.update_lab = False
# HAB is selected, so ensure update HAB is enabled and checked
self._view.enable_update_hab(True)
# Disable and deselect update LAB
self._view.enable_update_lab(False)
def update_lab_selected(self):
self._beam_centre_model.update_hab = False
self._beam_centre_model.update_lab = True
# LAB is selected, so ensure update LAB is enabled and checked
self._view.enable_update_lab(True)
# Disable and deselect update HAB
self._view.enable_update_hab(False)
def update_all_selected(self):
self._beam_centre_model.update_hab = True
self._beam_centre_model.update_lab = True
self._view.enable_update_hab(True)
self._view.enable_update_lab(True)
def set_on_state_model(self, attribute_name, state_model):
attribute = getattr(self._view, attribute_name)
if attribute or isinstance(attribute, bool):
setattr(state_model, attribute_name, attribute)
def set_on_view(self, attribute_name, state_model):
attribute = getattr(state_model, attribute_name)
if attribute or isinstance(
attribute, bool
): # We need to be careful here. We don't want to set empty strings, or None, but we want to set boolean values. # noqa
setattr(self._view, attribute_name, attribute)
def _validate_radius_values(self):
min_value = getattr(self._view, "r_min_line_edit").text()
max_value = getattr(self._view, "r_max_line_edit").text()
try:
min_value = float(min_value)
max_value = float(max_value)
except ValueError:
# one of the values is empty
pass
else:
if min_value == max_value == 0:
self._view.run_button.setEnabled(False)
return
if min_value >= max_value:
if self._view.run_button.isEnabled():
# Only post to logger once per disabling
self._logger.notice(
"Minimum radius is larger than maximum radius. "
"Cannot find beam centre with current settings.")
self._view.run_button.setEnabled(False)
else:
self._view.run_button.setEnabled(True)
class BeamCentreModelTest(unittest.TestCase):
def setUp(self):
self.result = {'pos1': 300, 'pos2': -300}
self.centre_finder_instance = mock.MagicMock(return_value=self.result)
self.SANSCentreFinder = mock.MagicMock(
return_value=self.centre_finder_instance)
self.beam_centre_model = BeamCentreModel(self.SANSCentreFinder)
def test_that_model_initialises_with_correct_values(self):
self.assertEqual(self.beam_centre_model.max_iterations, 10)
self.assertEqual(self.beam_centre_model.r_min, 60)
self.assertEqual(self.beam_centre_model.r_max, 280)
self.assertEqual(self.beam_centre_model.left_right, True)
self.assertEqual(self.beam_centre_model.up_down, True)
self.assertEqual(self.beam_centre_model.tolerance, 0.0001251)
self.assertEqual(self.beam_centre_model.lab_pos_1, '')
self.assertEqual(self.beam_centre_model.lab_pos_2, '')
self.assertEqual(self.beam_centre_model.hab_pos_2, '')
self.assertEqual(self.beam_centre_model.hab_pos_1, '')
self.assertEqual(self.beam_centre_model.scale_1, 1000)
self.assertEqual(self.beam_centre_model.scale_2, 1000)
self.assertEqual(self.beam_centre_model.COM, False)
self.assertEqual(self.beam_centre_model.verbose, False)
self.assertEqual(self.beam_centre_model.q_min, 0.01)
self.assertEqual(self.beam_centre_model.q_max, 0.1)
self.assertEqual(self.beam_centre_model.component, DetectorType.LAB)
self.assertTrue(self.beam_centre_model.update_lab)
self.assertTrue(self.beam_centre_model.update_hab)
def test_all_other_hardcoded_inst_values_taken(self):
for inst in {
SANSInstrument.NO_INSTRUMENT, SANSInstrument.SANS2D,
SANSInstrument.ZOOM
}:
self.beam_centre_model.reset_inst_defaults(instrument=inst)
self.assertEqual(60, self.beam_centre_model.r_min)
self.assertEqual(280, self.beam_centre_model.r_max)
self.assertEqual(1000, self.beam_centre_model.scale_1)
def test_loq_values_updated(self):
self.beam_centre_model.reset_inst_defaults(SANSInstrument.LOQ)
self.assertEqual(96, self.beam_centre_model.r_min)
self.assertEqual(216, self.beam_centre_model.r_max)
def test_that_can_update_model_values(self):
self.beam_centre_model.scale_2 = 1.0
self.assertEqual(self.beam_centre_model.scale_2, 1.0)
def test_that_correct_values_are_set_for_LARMOR(self):
self.beam_centre_model.reset_inst_defaults(
instrument=SANSInstrument.LARMOR)
self.assertEqual(self.beam_centre_model.scale_1, 1.0)
def test_that_find_beam_centre_calls_centre_finder_once_when_COM_is_False(
self):
state = mock.MagicMock()
self.beam_centre_model.find_beam_centre(state)
self.SANSCentreFinder.return_value.assert_called_once_with(
state,
r_min=self.beam_centre_model.r_min,
r_max=self.beam_centre_model.r_max,
max_iter=self.beam_centre_model.max_iterations,
x_start=self.beam_centre_model.lab_pos_1,
y_start=self.beam_centre_model.lab_pos_2,
tolerance=self.beam_centre_model.tolerance,
find_direction=FindDirectionEnum.ALL,
reduction_method=True,
verbose=False,
component=DetectorType.LAB)
def test_that_find_beam_centre_calls_centre_finder_twice_when_COM_is_TRUE(
self):
state = mock.MagicMock()
self.beam_centre_model.COM = True
self.beam_centre_model.find_beam_centre(state)
self.assertEqual(self.SANSCentreFinder.return_value.call_count, 2)
self.SANSCentreFinder.return_value.assert_called_with(
state,
r_min=self.beam_centre_model.r_min,
r_max=self.beam_centre_model.r_max,
max_iter=self.beam_centre_model.max_iterations,
x_start=self.result['pos1'],
y_start=self.result['pos2'],
tolerance=self.beam_centre_model.tolerance,
find_direction=FindDirectionEnum.ALL,
reduction_method=True,
verbose=False,
component=DetectorType.LAB)
self.SANSCentreFinder.return_value.assert_any_call(
state,
r_min=self.beam_centre_model.r_min,
r_max=self.beam_centre_model.r_max,
max_iter=self.beam_centre_model.max_iterations,
x_start=self.beam_centre_model.lab_pos_1,
y_start=self.beam_centre_model.lab_pos_2,
tolerance=self.beam_centre_model.tolerance,
find_direction=FindDirectionEnum.ALL,
reduction_method=False,
component=DetectorType.LAB)
def setUp(self):
self.result = {'pos1': 300, 'pos2': -300}
self.centre_finder_instance = mock.MagicMock(return_value=self.result)
self.SANSCentreFinder = mock.MagicMock(
return_value=self.centre_finder_instance)
self.beam_centre_model = BeamCentreModel(self.SANSCentreFinder)
class BeamCentrePresenter(object):
class ConcreteBeamCentreListener(BeamCentre.BeamCentreListener):
def __init__(self, presenter):
self._presenter = presenter
def on_run_clicked(self):
self._presenter.on_run_clicked()
def __init__(self, parent_presenter, beam_centre_model=None):
self._view = None
self._parent_presenter = parent_presenter
self._logger = Logger("SANS")
self._beam_centre_model = BeamCentreModel(
) if not beam_centre_model else beam_centre_model
self._worker = BeamCentreAsync(parent_presenter=self)
def set_view(self, view):
if view:
self._view = view
# Set up run listener
listener = BeamCentrePresenter.ConcreteBeamCentreListener(self)
self._view.add_listener(listener)
# Set the default gui
self._view.set_options(self._beam_centre_model)
# Connect view signals
self.connect_signals()
def connect_signals(self):
self._view.r_min_line_edit.textChanged.connect(
self._validate_radius_values)
self._view.r_max_line_edit.textChanged.connect(
self._validate_radius_values)
def on_update_instrument(self, instrument):
self._view.on_update_instrument(instrument)
def on_update_rows(self):
self._beam_centre_model.reset_inst_defaults(
self._parent_presenter.instrument)
self.update_centre_positions()
def on_update_centre_values(self, new_vals: Dict):
self._beam_centre_model.update_centre_positions(new_vals)
def on_processing_finished_centre_finder(self):
# Enable button
self._view.set_run_button_to_normal()
# Update Centre Positions in model and GUI
self._view.rear_pos_1 = self._round(self._beam_centre_model.rear_pos_1)
self._view.rear_pos_2 = self._round(self._beam_centre_model.rear_pos_2)
self._view.front_pos_1 = self._round(
self._beam_centre_model.front_pos_1)
self._view.front_pos_2 = self._round(
self._beam_centre_model.front_pos_2)
def on_processing_error_centre_finder(self, error):
self._logger.warning(
"There has been an error. See more: {}".format(error))
self._view.set_run_button_to_normal()
def on_processing_finished(self):
# Signal from run tab presenter
self._view.set_run_button_to_normal()
def on_processing_error(self, error):
# Signal from run tab presenter
self._view.set_run_button_to_normal()
def on_run_clicked(self):
UsageService.registerFeatureUsage(
FeatureType.Feature, ["ISIS SANS", "Beam Centre Finder - Run"],
False)
# Get the state information for the first row.
state = self._parent_presenter.get_state_for_row(0)
if not state:
self._logger.information(
"You can only calculate the beam centre if a user file has been loaded and there"
"valid sample scatter entry has been provided in the selected row."
)
return
# Disable the button
self._view.set_run_button_to_processing()
self._update_beam_model_from_view()
# Run the task
state_copy = copy.copy(state)
self._worker.find_beam_centre(
state_copy, self._beam_centre_model.pack_beam_centre_settings())
def _update_beam_model_from_view(self):
self._beam_centre_model.r_min = self._view.r_min
self._beam_centre_model.r_max = self._view.r_max
self._beam_centre_model.max_iterations = self._view.max_iterations
self._beam_centre_model.tolerance = self._view.tolerance
self._beam_centre_model.left_right = self._view.left_right
self._beam_centre_model.verbose = self._view.verbose
self._beam_centre_model.COM = self._view.COM
self._beam_centre_model.up_down = self._view.up_down
self._beam_centre_model.rear_pos_1 = self._view.rear_pos_1
self._beam_centre_model.rear_pos_2 = self._view.rear_pos_2
self._beam_centre_model.front_pos_1 = self._view.front_pos_1
self._beam_centre_model.front_pos_2 = self._view.front_pos_2
self._beam_centre_model.q_min = self._view.q_min
self._beam_centre_model.q_max = self._view.q_max
self._beam_centre_model.component = self._view.component
self._beam_centre_model.update_front = self._view.update_front
self._beam_centre_model.update_rear = self._view.update_rear
def copy_centre_positions(self, state_model):
"""
Copies rear / front positions from an external model
"""
rear_pos_1 = getattr(state_model, 'rear_pos_1')
rear_pos_2 = getattr(state_model, 'rear_pos_2')
self._beam_centre_model.rear_pos_1 = rear_pos_1
self._beam_centre_model.rear_pos_2 = rear_pos_2
self._beam_centre_model.front_pos_1 = \
getattr(state_model, 'front_pos_1') if getattr(state_model, 'front_pos_1') else rear_pos_1
self._beam_centre_model.front_pos_2 = \
getattr(state_model, 'front_pos_2') if getattr(state_model, 'front_pos_2') else rear_pos_2
def update_centre_positions(self):
rear_pos_1 = self._beam_centre_model.rear_pos_1
rear_pos_2 = self._beam_centre_model.rear_pos_2
front_pos_1 = self._beam_centre_model.front_pos_1 if self._beam_centre_model.front_pos_1 == '' else rear_pos_1
front_pos_2 = self._beam_centre_model.front_pos_2 if self._beam_centre_model.front_pos_2 == '' else rear_pos_2
self._view.rear_pos_1 = self._round(rear_pos_1)
self._view.rear_pos_2 = self._round(rear_pos_2)
self._view.front_pos_1 = self._round(front_pos_1)
self._view.front_pos_2 = self._round(front_pos_2)
def update_front_selected(self):
self._beam_centre_model.update_front = True
self._beam_centre_model.update_rear = False
# front is selected, so ensure update front is enabled and checked
self._view.enable_update_front(True)
# Disable and deselect update rear
self._view.enable_update_rear(False)
def update_rear_selected(self):
self._beam_centre_model.update_front = False
self._beam_centre_model.update_rear = True
# rear is selected, so ensure update rear is enabled and checked
self._view.enable_update_rear(True)
# Disable and deselect update front
self._view.enable_update_front(False)
def update_all_selected(self):
self._beam_centre_model.update_front = True
self._beam_centre_model.update_rear = True
self._view.enable_update_front(True)
self._view.enable_update_rear(True)
def set_on_state_model(self, attribute_name, state_model):
attribute = getattr(self._view, attribute_name)
if attribute or isinstance(attribute, bool):
setattr(state_model, attribute_name, attribute)
def set_on_view(self, attribute_name, state_model):
attribute = getattr(state_model, attribute_name)
# We need to be careful here. We don't want to set empty strings, or None, but we want to set boolean values.
if attribute or isinstance(attribute, bool):
setattr(self._view, attribute_name, attribute)
def _round(self, val):
DECIMAL_PLACES_CENTRE_POS = 3
try:
val = float(val)
except ValueError:
return val
return round(val, DECIMAL_PLACES_CENTRE_POS)
def _validate_radius_values(self):
min_value = getattr(self._view, "r_min_line_edit").text()
max_value = getattr(self._view, "r_max_line_edit").text()
try:
min_value = float(min_value)
max_value = float(max_value)
except ValueError:
# one of the values is empty
pass
else:
if min_value == max_value == 0:
self._view.run_button.setEnabled(False)
return
if min_value >= max_value:
if self._view.run_button.isEnabled():
# Only post to logger once per disabling
self._logger.notice(
"Minimum radius is larger than maximum radius. "
"Cannot find beam centre with current settings.")
self._view.run_button.setEnabled(False)
else:
self._view.run_button.setEnabled(True)
class BeamCentreModelTest(unittest.TestCase):
def setUp(self):
self.result = {'pos1':300, 'pos2':-300}
self.centre_finder_instance = mock.MagicMock(return_value = self.result)
self.SANSCentreFinder = mock.MagicMock(return_value = self.centre_finder_instance)
self.beam_centre_model = BeamCentreModel(self.SANSCentreFinder)
def test_that_model_initialises_with_correct_values(self):
self.assertEqual(self.beam_centre_model.max_iterations, 10)
self.assertEqual(self.beam_centre_model.r_min, 60)
self.assertEqual(self.beam_centre_model.r_max, 280)
self.assertEqual(self.beam_centre_model.left_right, True)
self.assertEqual(self.beam_centre_model.up_down, True)
self.assertEqual(self.beam_centre_model.tolerance, 0.0001251)
self.assertEqual(self.beam_centre_model.lab_pos_1, '')
self.assertEqual(self.beam_centre_model.lab_pos_2, '')
self.assertEqual(self.beam_centre_model.hab_pos_2, '')
self.assertEqual(self.beam_centre_model.hab_pos_1, '')
self.assertEqual(self.beam_centre_model.scale_1, 1000)
self.assertEqual(self.beam_centre_model.scale_2, 1000)
self.assertEqual(self.beam_centre_model.COM, False)
self.assertEqual(self.beam_centre_model.verbose, False)
self.assertEqual(self.beam_centre_model.q_min, 0.01)
self.assertEqual(self.beam_centre_model.q_max, 0.1)
self.assertEqual(self.beam_centre_model.component, DetectorType.LAB)
self.assertTrue(self.beam_centre_model.update_lab)
self.assertTrue(self.beam_centre_model.update_hab)
def test_that_can_update_model_values(self):
self.beam_centre_model.scale_2 = 1.0
self.assertEqual(self.beam_centre_model.scale_2, 1.0)
def test_that_correct_values_are_set_for_LARMOR(self):
facility = SANSFacility.ISIS
file_information = SANSFileInformationMock(run_number=2260, instrument=SANSInstrument.LARMOR)
data_builder = get_data_builder(facility, file_information)
data_builder.set_sample_scatter("LARMOR00002260")
data_state = data_builder.build()
self.beam_centre_model.reset_to_defaults_for_instrument(data_state)
self.assertEqual(self.beam_centre_model.scale_1, 1.0)
def test_that_correct_values_are_set_for_LOQ(self):
facility = SANSFacility.ISIS
file_information = SANSFileInformationMock(run_number=74044, instrument=SANSInstrument.LOQ)
data_builder = get_data_builder(facility, file_information)
data_builder.set_sample_scatter("LOQ74044")
data_state = data_builder.build()
self.beam_centre_model.reset_to_defaults_for_instrument(data_state)
self.assertEqual(self.beam_centre_model.r_max, 200)
def test_that_find_beam_centre_calls_centre_finder_once_when_COM_is_False(self):
state = mock.MagicMock()
self.beam_centre_model.find_beam_centre(state)
self.SANSCentreFinder.return_value.assert_called_once_with(state, r_min=self.beam_centre_model.r_min,
r_max=self.beam_centre_model.r_max,
max_iter= self.beam_centre_model.max_iterations,
x_start=self.beam_centre_model.lab_pos_1,
y_start=self.beam_centre_model.lab_pos_2,
tolerance=self.beam_centre_model.tolerance,
find_direction=FindDirectionEnum.All,
reduction_method=True,
verbose=False, component=DetectorType.LAB)
def test_that_find_beam_centre_calls_centre_finder_twice_when_COM_is_TRUE(self):
state = mock.MagicMock()
self.beam_centre_model.COM = True
self.beam_centre_model.find_beam_centre(state)
self.assertEqual(self.SANSCentreFinder.return_value.call_count, 2)
self.SANSCentreFinder.return_value.assert_called_with(state, r_min=self.beam_centre_model.r_min,
r_max=self.beam_centre_model.r_max,
max_iter= self.beam_centre_model.max_iterations,
x_start=self.result['pos1'],
y_start=self.result['pos2'],
tolerance=self.beam_centre_model.tolerance,
find_direction=FindDirectionEnum.All,
reduction_method=True,
verbose=False, component=DetectorType.LAB)
self.SANSCentreFinder.return_value.assert_any_call(state, r_min=self.beam_centre_model.r_min,
r_max=self.beam_centre_model.r_max,
max_iter=self.beam_centre_model.max_iterations,
x_start=self.beam_centre_model.lab_pos_1,
y_start=self.beam_centre_model.lab_pos_2,
tolerance=self.beam_centre_model.tolerance,
find_direction=FindDirectionEnum.All,
reduction_method=False, component=DetectorType.LAB)
class BeamCentrePresenter(object):
class ConcreteBeamCentreListener(BeamCentre.BeamCentreListener):
def __init__(self, presenter):
self._presenter = presenter
def on_run_clicked(self):
self._presenter.on_run_clicked()
class CentreFinderListener(WorkHandler.WorkListener):
def __init__(self, presenter):
super(BeamCentrePresenter.CentreFinderListener, self).__init__()
self._presenter = presenter
def on_processing_finished(self, result):
self._presenter.on_processing_finished_centre_finder(result)
def on_processing_error(self, error):
self._presenter.on_processing_error_centre_finder(error)
def __init__(self, parent_presenter):
super(BeamCentrePresenter, self).__init__()
self._view = None
self._parent_presenter = parent_presenter
self._work_handler = WorkHandler()
self._logger = Logger("SANS")
self._beam_centre_model = BeamCentreModel()
def set_view(self, view):
if view:
self._view = view
# Set up run listener
listener = BeamCentrePresenter.ConcreteBeamCentreListener(self)
self._view.add_listener(listener)
# Set the default gui
self._view.set_options(self._beam_centre_model)
def on_update_rows(self):
if self._parent_presenter._file_information:
instrument = self._parent_presenter._file_information.get_instrument(
)
self._beam_centre_model.set_scaling(instrument)
self._view.set_options(self._beam_centre_model)
def on_processing_finished_centre_finder(self, result):
# Enable button
self._view.set_run_button_to_normal()
# Update Centre Positions in model and GUI
self._beam_centre_model.lab_pos_1 = result['pos1']
self._beam_centre_model.lab_pos_2 = result['pos2']
self._beam_centre_model.hab_pos_1 = result['pos1']
self._beam_centre_model.hab_pos_2 = result['pos2']
self._view.lab_pos_1 = self._beam_centre_model.lab_pos_1 * self._beam_centre_model.scale_1
self._view.lab_pos_2 = self._beam_centre_model.lab_pos_2 * self._beam_centre_model.scale_2
self._view.hab_pos_1 = self._beam_centre_model.hab_pos_1 * self._beam_centre_model.scale_1
self._view.hab_pos_2 = self._beam_centre_model.hab_pos_2 * self._beam_centre_model.scale_2
def on_processing_error_centre_finder(self, error):
self._logger.warning(
"There has been an error. See more: {}".format(error))
def on_processing_error(self, error):
pass
def on_run_clicked(self):
# Get the state information for the first row.
state = self._parent_presenter.get_state_for_row(0)
if not state:
self._logger.information(
"You can only calculate the beam centre if a user file has been loaded and there"
"valid sample scatter entry has been provided in the selected row."
)
return
# Disable the button
self._view.set_run_button_to_processing()
#Update model
self._update_beam_model_from_view()
# Run the task
listener = BeamCentrePresenter.CentreFinderListener(self)
state_copy = copy.copy(state)
self._work_handler.process(listener, find_beam_centre, state_copy,
self._beam_centre_model)
def _update_beam_model_from_view(self):
self._beam_centre_model.r_min = self._view.r_min
self._beam_centre_model.r_max = self._view.r_max
self._beam_centre_model.max_iterations = self._view.max_iterations
self._beam_centre_model.tolerance = self._view.tolerance
self._beam_centre_model.left_right = self._view.left_right
self._beam_centre_model.up_down = self._view.up_down
self._beam_centre_model.lab_pos_1 = self._view.lab_pos_1 / self._beam_centre_model.scale_1
self._beam_centre_model.lab_pos_2 = self._view.lab_pos_2 / self._beam_centre_model.scale_2
self._beam_centre_model.hab_pos_1 = self._view.hab_pos_1 / self._beam_centre_model.scale_1
self._beam_centre_model.hab_pos_2 = self._view.hab_pos_2 / self._beam_centre_model.scale_2
def set_on_state_model(self, attribute_name, state_model):
attribute = getattr(self._view, attribute_name)
if attribute or isinstance(attribute, bool):
setattr(state_model, attribute_name, attribute)
def set_on_view(self, attribute_name, state_model):
attribute = getattr(state_model, attribute_name)
if attribute or isinstance(
attribute, bool
): # We need to be careful here. We don't want to set empty strings, or None, but we want to set boolean values. # noqa
setattr(self._view, attribute_name, attribute)
class BeamCentreModelTest(unittest.TestCase):
def setUp(self):
self.result = {'pos1': 300, 'pos2': -300}
self.centre_finder_instance = mock.MagicMock(return_value=self.result)
self.SANSCentreFinder = mock.MagicMock(
return_value=self.centre_finder_instance)
self.beam_centre_model = BeamCentreModel()
def test_that_model_initialises_with_correct_values(self):
self.assertEqual(self.beam_centre_model.max_iterations, 10)
self.assertEqual(self.beam_centre_model.r_min, 60)
self.assertEqual(self.beam_centre_model.r_max, 280)
self.assertEqual(self.beam_centre_model.left_right, True)
self.assertEqual(self.beam_centre_model.up_down, True)
self.assertEqual(self.beam_centre_model.tolerance, 0.0001251)
self.assertEqual(self.beam_centre_model.rear_pos_1, '')
self.assertEqual(self.beam_centre_model.rear_pos_2, '')
self.assertEqual(self.beam_centre_model.front_pos_2, '')
self.assertEqual(self.beam_centre_model.front_pos_1, '')
self.assertEqual(self.beam_centre_model.COM, False)
self.assertEqual(self.beam_centre_model.verbose, False)
self.assertEqual(self.beam_centre_model.q_min, 0.01)
self.assertEqual(self.beam_centre_model.q_max, 0.1)
self.assertEqual(self.beam_centre_model.component, DetectorType.LAB)
self.assertTrue(self.beam_centre_model.update_rear)
self.assertTrue(self.beam_centre_model.update_front)
def test_all_other_hardcoded_inst_values_taken(self):
for inst in {
SANSInstrument.NO_INSTRUMENT, SANSInstrument.SANS2D,
SANSInstrument.ZOOM
}:
self.beam_centre_model.reset_inst_defaults(instrument=inst)
self.assertEqual(60, self.beam_centre_model.r_min)
self.assertEqual(280, self.beam_centre_model.r_max)
def test_update_centre_positions_front_mode(self):
expected_vals = {"pos1": 101.123, "pos2": 202.234}
rear_vals_before = (self.beam_centre_model.rear_pos_1,
self.beam_centre_model.rear_pos_2)
self.beam_centre_model.component = DetectorType.HAB # Where HAB == front
self.beam_centre_model.update_centre_positions(expected_vals)
# mm -> m scaling
self.assertEqual(expected_vals["pos1"] * 1000,
self.beam_centre_model.front_pos_1)
self.assertEqual(expected_vals["pos2"] * 1000,
self.beam_centre_model.front_pos_2)
self.assertEqual(rear_vals_before[0],
self.beam_centre_model.rear_pos_1)
self.assertEqual(rear_vals_before[1],
self.beam_centre_model.rear_pos_2)
def test_update_centre_positions_rear_mode(self):
expected_vals = {"pos1": 303.345, "pos2": 404.456}
front_vals_before = (self.beam_centre_model.front_pos_1,
self.beam_centre_model.front_pos_2)
self.beam_centre_model.component = DetectorType.LAB # Where LAB == rear
self.beam_centre_model.update_centre_positions(expected_vals)
# mm -> m scaling
self.assertEqual(expected_vals["pos1"] * 1000,
self.beam_centre_model.rear_pos_1)
self.assertEqual(expected_vals["pos2"] * 1000,
self.beam_centre_model.rear_pos_2)
self.assertEqual(front_vals_before[0],
self.beam_centre_model.front_pos_1)
self.assertEqual(front_vals_before[1],
self.beam_centre_model.front_pos_2)
def test_loq_values_updated(self):
self.beam_centre_model.reset_inst_defaults(SANSInstrument.LOQ)
self.assertEqual(96, self.beam_centre_model.r_min)
self.assertEqual(216, self.beam_centre_model.r_max)
def test_that_can_update_model_values(self):
self.beam_centre_model.r_max = 1.0
self.assertEqual(self.beam_centre_model.r_max, 1.0)
def test_beam_centre_scales_to_mills(self):
self.assertIsNot(SANSInstrument.LARMOR,
self.beam_centre_model.instrument)
value_in_mm = 1200
self.beam_centre_model.rear_pos_1 = value_in_mm
self.beam_centre_model.front_pos_2 = value_in_mm * 2
self.assertEqual(value_in_mm, self.beam_centre_model.rear_pos_1)
self.assertEqual((value_in_mm * 2), self.beam_centre_model.front_pos_2)
# Should internally be in m
self.assertEqual((value_in_mm / 1000),
self.beam_centre_model._rear_pos_1)
self.assertEqual((value_in_mm * 2 / 1000),
self.beam_centre_model._front_pos_2)
def test_beam_centre_does_not_scale(self):
self.beam_centre_model.instrument = SANSInstrument.LARMOR
value_in_m = 1.2
self.beam_centre_model.rear_pos_1 = value_in_m
self.beam_centre_model.front_pos_1 = value_in_m * 2
self.assertEqual(value_in_m, self.beam_centre_model.rear_pos_1)
self.assertEqual(value_in_m, self.beam_centre_model._rear_pos_1)
self.assertEqual((value_in_m * 2), self.beam_centre_model.front_pos_1)
self.assertEqual((value_in_m * 2), self.beam_centre_model._front_pos_1)
def test_instrument_is_set(self):
for inst in SANSInstrument:
self.beam_centre_model.reset_inst_defaults(inst)
self.assertEqual(inst, self.beam_centre_model.instrument)
def test_scaling_does_not_affect_non_rear_front_values(self):
value_in_mm = 100 # 0.1 m
for inst in [SANSInstrument.LARMOR, SANSInstrument.LOQ]:
self.beam_centre_model.instrument = inst
self.beam_centre_model.tolerance = value_in_mm
self.assertEqual((value_in_mm / 1000),
self.beam_centre_model._tolerance)
self.assertEqual(self.beam_centre_model.tolerance, value_in_mm)
def test_scaling_can_handle_non_float_types(self):
self.beam_centre_model.instrument = SANSInstrument.NO_INSTRUMENT
# When in doubt it should just forward the value as is
self.beam_centre_model.rear_pos_1 = 'a'
self.assertEqual(self.beam_centre_model.rear_pos_1, 'a')
def test_scaling_ignores_zero_vals(self):
self.beam_centre_model.lab_pos_1 = 0.0
self.beam_centre_model.lab_pos_2 = 0.0
self.beam_centre_model.hab_pos_1 = 0.0
self.beam_centre_model.hab_pos_2 = 0.0
self.assertEqual(0.0, self.beam_centre_model.lab_pos_1)
self.assertEqual(0.0, self.beam_centre_model.lab_pos_2)
self.assertEqual(0.0, self.beam_centre_model.hab_pos_1)
self.assertEqual(0.0, self.beam_centre_model.hab_pos_2)
class BeamCentreModelTest(unittest.TestCase):
def setUp(self):
self.result = {'pos1': 300, 'pos2': -300}
self.centre_finder_instance = mock.MagicMock(return_value=self.result)
self.SANSCentreFinder = mock.MagicMock(
return_value=self.centre_finder_instance)
self.beam_centre_model = BeamCentreModel(self.SANSCentreFinder)
def test_that_model_initialises_with_correct_values(self):
self.assertEqual(self.beam_centre_model.max_iterations, 10)
self.assertEqual(self.beam_centre_model.r_min, 60)
self.assertEqual(self.beam_centre_model.r_max, 280)
self.assertEqual(self.beam_centre_model.left_right, True)
self.assertEqual(self.beam_centre_model.up_down, True)
self.assertEqual(self.beam_centre_model.tolerance, 0.000125)
self.assertEqual(self.beam_centre_model.lab_pos_1, '')
self.assertEqual(self.beam_centre_model.lab_pos_2, '')
self.assertEqual(self.beam_centre_model.hab_pos_2, '')
self.assertEqual(self.beam_centre_model.hab_pos_1, '')
self.assertEqual(self.beam_centre_model.scale_1, 1000)
self.assertEqual(self.beam_centre_model.scale_2, 1000)
self.assertEqual(self.beam_centre_model.COM, False)
self.assertEqual(self.beam_centre_model.verbose, False)
self.assertEqual(self.beam_centre_model.q_min, 0.01)
self.assertEqual(self.beam_centre_model.q_max, 0.1)
def test_that_can_update_model_values(self):
self.beam_centre_model.scale_2 = 1.0
self.assertEqual(self.beam_centre_model.scale_2, 1.0)
def test_that_correct_values_are_set_for_LARMOR(self):
self.beam_centre_model.reset_to_defaults_for_instrument(
SANSInstrument.LARMOR)
self.assertEqual(self.beam_centre_model.scale_1, 1.0)
def test_that_correct_values_are_set_for_LOQ(self):
self.beam_centre_model.reset_to_defaults_for_instrument(
SANSInstrument.LOQ)
self.assertEqual(self.beam_centre_model.r_max, 200)
def test_that_find_beam_centre_calls_centre_finder_once_when_COM_is_False(
self):
state = mock.MagicMock()
self.beam_centre_model.find_beam_centre(state)
self.SANSCentreFinder.return_value.assert_called_once_with(
state,
r_min=self.beam_centre_model.r_min,
r_max=self.beam_centre_model.r_max,
max_iter=self.beam_centre_model.max_iterations,
x_start=self.beam_centre_model.lab_pos_1,
y_start=self.beam_centre_model.lab_pos_2,
tolerance=self.beam_centre_model.tolerance,
find_direction=FindDirectionEnum.All,
reduction_method=True,
verbose=False)
self.assertEqual(state.convert_to_q.q_min,
self.beam_centre_model.q_min)
self.assertEqual(state.convert_to_q.q_max,
self.beam_centre_model.q_max)
def test_that_find_beam_centre_calls_centre_finder_twice_when_COM_is_TRUE(
self):
state = mock.MagicMock()
self.beam_centre_model.COM = True
self.beam_centre_model.find_beam_centre(state)
self.assertEqual(self.SANSCentreFinder.return_value.call_count, 2)
self.SANSCentreFinder.return_value.assert_called_with(
state,
r_min=self.beam_centre_model.r_min,
r_max=self.beam_centre_model.r_max,
max_iter=self.beam_centre_model.max_iterations,
x_start=self.result['pos1'],
y_start=self.result['pos2'],
tolerance=self.beam_centre_model.tolerance,
find_direction=FindDirectionEnum.All,
reduction_method=True,
verbose=False)
self.SANSCentreFinder.return_value.assert_any_call(
state,
r_min=self.beam_centre_model.r_min,
r_max=self.beam_centre_model.r_max,
max_iter=self.beam_centre_model.max_iterations,
x_start=self.beam_centre_model.lab_pos_1,
y_start=self.beam_centre_model.lab_pos_2,
tolerance=self.beam_centre_model.tolerance,
find_direction=FindDirectionEnum.All,
reduction_method=False)
def setUp(self):
self.result = {'pos1':300, 'pos2':-300}
self.centre_finder_instance = mock.MagicMock(return_value = self.result)
self.SANSCentreFinder = mock.MagicMock(return_value = self.centre_finder_instance)
self.beam_centre_model = BeamCentreModel(self.SANSCentreFinder)
class BeamCentreModelTest(unittest.TestCase):
def setUp(self):
self.result = {'pos1':300, 'pos2':-300}
self.centre_finder_instance = mock.MagicMock(return_value = self.result)
self.SANSCentreFinder = mock.MagicMock(return_value = self.centre_finder_instance)
self.beam_centre_model = BeamCentreModel(self.SANSCentreFinder)
def test_that_model_initialises_with_correct_values(self):
self.assertEqual(self.beam_centre_model.max_iterations, 10)
self.assertEqual(self.beam_centre_model.r_min, 60)
self.assertEqual(self.beam_centre_model.r_max, 280)
self.assertEqual(self.beam_centre_model.left_right, True)
self.assertEqual(self.beam_centre_model.up_down, True)
self.assertEqual(self.beam_centre_model.tolerance, 0.0001251)
self.assertEqual(self.beam_centre_model.lab_pos_1, '')
self.assertEqual(self.beam_centre_model.lab_pos_2, '')
self.assertEqual(self.beam_centre_model.hab_pos_2, '')
self.assertEqual(self.beam_centre_model.hab_pos_1, '')
self.assertEqual(self.beam_centre_model.scale_1, 1000)
self.assertEqual(self.beam_centre_model.scale_2, 1000)
self.assertEqual(self.beam_centre_model.COM, False)
self.assertEqual(self.beam_centre_model.verbose, False)
self.assertEqual(self.beam_centre_model.q_min, 0.01)
self.assertEqual(self.beam_centre_model.q_max, 0.1)
self.assertEqual(self.beam_centre_model.component, DetectorType.LAB)
self.assertTrue(self.beam_centre_model.update_lab)
self.assertTrue(self.beam_centre_model.update_hab)
def test_that_can_update_model_values(self):
self.beam_centre_model.scale_2 = 1.0
self.assertEqual(self.beam_centre_model.scale_2, 1.0)
def test_that_correct_values_are_set_for_LARMOR(self):
file_information = SANSFileInformationMock(run_number=2260, instrument=SANSInstrument.LARMOR)
self.beam_centre_model.reset_to_defaults_for_instrument(file_information)
self.assertEqual(self.beam_centre_model.scale_1, 1.0)
def test_that_correct_values_are_set_for_LOQ(self):
file_information = SANSFileInformationMock(run_number=74044, instrument=SANSInstrument.LOQ)
self.beam_centre_model.reset_to_defaults_for_instrument(file_information)
self.assertEqual(self.beam_centre_model.r_max, 200)
def test_that_find_beam_centre_calls_centre_finder_once_when_COM_is_False(self):
state = mock.MagicMock()
self.beam_centre_model.find_beam_centre(state)
self.SANSCentreFinder.return_value.assert_called_once_with(state, r_min=self.beam_centre_model.r_min,
r_max=self.beam_centre_model.r_max,
max_iter= self.beam_centre_model.max_iterations,
x_start=self.beam_centre_model.lab_pos_1,
y_start=self.beam_centre_model.lab_pos_2,
tolerance=self.beam_centre_model.tolerance,
find_direction=FindDirectionEnum.All,
reduction_method=True,
verbose=False, component=DetectorType.LAB)
def test_that_find_beam_centre_calls_centre_finder_twice_when_COM_is_TRUE(self):
state = mock.MagicMock()
self.beam_centre_model.COM = True
self.beam_centre_model.find_beam_centre(state)
self.assertEqual(self.SANSCentreFinder.return_value.call_count, 2)
self.SANSCentreFinder.return_value.assert_called_with(state, r_min=self.beam_centre_model.r_min,
r_max=self.beam_centre_model.r_max,
max_iter= self.beam_centre_model.max_iterations,
x_start=self.result['pos1'],
y_start=self.result['pos2'],
tolerance=self.beam_centre_model.tolerance,
find_direction=FindDirectionEnum.All,
reduction_method=True,
verbose=False, component=DetectorType.LAB)
self.SANSCentreFinder.return_value.assert_any_call(state, r_min=self.beam_centre_model.r_min,
r_max=self.beam_centre_model.r_max,
max_iter=self.beam_centre_model.max_iterations,
x_start=self.beam_centre_model.lab_pos_1,
y_start=self.beam_centre_model.lab_pos_2,
tolerance=self.beam_centre_model.tolerance,
find_direction=FindDirectionEnum.All,
reduction_method=False, component=DetectorType.LAB)