class CreateHelicalWidget(CreateTaskBase):
def __init__(self, parent = None,name = None, fl = 0):
CreateTaskBase.__init__(self, parent, name, fl, 'Helical')
if not name:
self.setObjectName("create_helical_widget")
self.init_models()
# Hardware objects ----------------------------------------------------
# Internal variables --------------------------------------------------
self._lines_map = {}
# Graphic elements ----------------------------------------------------
self._lines_widget = loadUi(os.path.join(\
os.path.dirname(__file__), "ui_files/Qt4_helical_line_widget_layout.ui"))
self._acq_widget = AcquisitionWidget(self, "acquisition_widget",
layout='vertical', acq_params=self._acquisition_parameters,
path_template=self._path_template)
self._data_path_widget = DataPathWidget(self, 'create_dc_path_widget',
data_model=self._path_template, layout='vertical')
self._processing_widget = ProcessingWidget(self,
data_model=self._processing_parameters)
# Layout --------------------------------------------------------------
_main_vlayout = QVBoxLayout(self)
_main_vlayout.addWidget(self._lines_widget)
_main_vlayout.addWidget(self._acq_widget)
_main_vlayout.addWidget(self._data_path_widget)
_main_vlayout.addWidget(self._processing_widget)
_main_vlayout.addStretch(0)
_main_vlayout.setSpacing(6)
_main_vlayout.setContentsMargins(2, 2, 2, 2)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self._lines_widget.lines_treewidget.itemSelectionChanged.connect(\
self.lines_treewidget_selection_changed)
self._lines_widget.create_line_button.clicked.connect(\
self.create_line_button_clicked)
self._lines_widget.create_auto_line_button.clicked.connect(\
self.create_auto_line_button_clicked)
self._lines_widget.remove_line_button.clicked.connect(\
self.remove_line_button_clicked)
self._lines_widget.overlay_cbox.stateChanged.connect(\
self.overlay_toggled)
self._lines_widget.overlay_slider.valueChanged.connect(\
self.overlay_alpha_changed)
self._lines_widget.swap_points_button.clicked.connect(\
self.swap_points_clicked)
self._acq_widget.acqParametersChangedSignal.\
connect(self.acq_parameters_changed)
self._acq_widget.madEnergySelectedSignal.connect(\
self.mad_energy_selected)
self._data_path_widget.pathTemplateChangedSignal.\
connect(self.path_template_changed)
self._processing_widget.enableProcessingSignal.connect(\
self._run_processing_toggled)
# Other ---------------------------------------------------------------
for col in range(self._lines_widget.lines_treewidget.columnCount()):
self._lines_widget.lines_treewidget.resizeColumnToContents(col)
#self._processing_widget.processing_widget.\
# run_processing_parallel_cbox.setChecked(False)
self.enable_widgets(False)
def enable_widgets(self, state):
self._acq_widget.setEnabled(state)
self._data_path_widget.setEnabled(state)
self._processing_widget.setEnabled(state)
def init_models(self):
CreateTaskBase.init_models(self)
self._energy_scan_result = queue_model_objects.EnergyScanResult()
self._processing_parameters = queue_model_objects.ProcessingParameters()
if self._beamline_setup_hwobj is not None:
has_shutter_less = self._beamline_setup_hwobj.\
detector_has_shutterless()
self._acquisition_parameters.shutterless = has_shutter_less
self._acquisition_parameters = self._beamline_setup_hwobj.\
get_default_acquisition_parameters("default_helical_values")
self._processing_widget.processing_widget.run_processing_parallel_cbox.setChecked(\
self._beamline_setup_hwobj._get_run_processing_parallel())
def set_beamline_setup(self, bl_setup_hwobj):
CreateTaskBase.set_beamline_setup(self, bl_setup_hwobj)
# At startup, if scene loaded from file, then update listwidget
shapes = self._graphics_manager_hwobj.get_shapes()
for shape in shapes:
if isinstance(shape, GraphicsItemLine):
self.shape_created(shape, "Line")
def shape_created(self, shape, shape_type):
if shape_type == "Line":
self._lines_widget.lines_treewidget.clearSelection()
#info_str_list = QStringList()
info_str_list = []
info_str_list.append(shape.get_display_name())
info_str_list.append("%d" % shape.get_points_index()[0])
info_str_list.append("%d" % shape.get_points_index()[1])
lines_treewidget_item = QTreeWidgetItem(\
self._lines_widget.lines_treewidget,
info_str_list)
lines_treewidget_item.setSelected(True)
self._lines_map[shape] = lines_treewidget_item
self.lines_treewidget_selection_changed()
def shape_deleted(self, shape, shape_type):
if shape_type == "Line" and self._lines_map.get(shape):
shape_index = self._lines_widget.lines_treewidget.\
indexFromItem(self._lines_map[shape])
self._lines_widget.lines_treewidget.\
takeTopLevelItem(shape_index.row())
self._lines_map.pop(shape)
def shape_changed(self, shape, shape_type):
lines_treewidget_item = self._lines_map.get(shape)
if lines_treewidget_item:
lines_treewidget_item.setText(1, "%d" % shape.get_points_index()[0])
lines_treewidget_item.setText(2, "%d" % shape.get_points_index()[1])
def approve_creation(self):
base_result = CreateTaskBase.approve_creation(self)
if len(self._lines_widget.lines_treewidget.selectedItems()) == 0:
logging.getLogger("GUI").\
warning("No lines selected, please select one or more lines.")
return False
else:
return base_result
def update_processing_parameters(self, crystal):
self._processing_parameters.space_group = crystal.space_group
self._processing_parameters.cell_a = crystal.cell_a
self._processing_parameters.cell_alpha = crystal.cell_alpha
self._processing_parameters.cell_b = crystal.cell_b
self._processing_parameters.cell_beta = crystal.cell_beta
self._processing_parameters.cell_c = crystal.cell_c
self._processing_parameters.cell_gamma = crystal.cell_gamma
self._processing_widget.update_data_model(self._processing_parameters)
def select_shape_with_cpos(self, start_cpos, end_cpos, num_images):
selected_line = None
self._lines_widget.overlay_slider.setEnabled(False)
self._lines_widget.overlay_cbox.setEnabled(False)
self._graphics_manager_hwobj.de_select_all()
for shape in self._graphics_manager_hwobj.get_shapes():
if isinstance(shape, GraphicsItemLine):
(start_cpos_index, end_cpos_index) = shape.get_points_index()
if start_cpos_index == start_cpos.index and \
end_cpos_index == end_cpos.index:
self._graphics_manager_hwobj.select_shape(shape)
shape.set_num_images(num_images)
selected_line = shape
self._lines_widget.overlay_slider.setEnabled(True)
self._lines_widget.overlay_cbox.setEnabled(True)
#de-select previous selected list items and
#select the current shape (Line).
"""
for (list_item, shape) in self._lines_map.iteritems():
if selected_line == shape:
list_item.setSelected(True)
else:
list_item.setSelected(False)
"""
def single_item_selection(self, tree_item):
CreateTaskBase.single_item_selection(self, tree_item)
if isinstance(tree_item, Qt4_queue_item.SampleQueueItem):
sample_model = tree_item.get_model()
self._processing_parameters = sample_model.processing_parameters
#self._processing_parameters = copy.deepcopy(self._processing_parameters)
self._processing_widget.update_data_model(self._processing_parameters)
elif isinstance(tree_item, Qt4_queue_item.BasketQueueItem):
self.setDisabled(False)
elif isinstance(tree_item, Qt4_queue_item.DataCollectionQueueItem):
data_collection = tree_item.get_model()
if data_collection.is_helical():
self.setDisabled(tree_item.get_model().is_executed())
self._path_template = data_collection.get_path_template()
self._data_path_widget.update_data_model(self._path_template)
self._acquisition_parameters = data_collection.acquisitions[0].\
acquisition_parameters
if len(data_collection.acquisitions) == 2:
start_cpos = data_collection.acquisitions[0].acquisition_parameters.\
centred_position
end_cpos = data_collection.acquisitions[1].acquisition_parameters.\
centred_position
num_images = data_collection.acquisitions[0].acquisition_parameters.\
num_images
self.select_shape_with_cpos(start_cpos, end_cpos, num_images)
self._acq_widget.update_data_model(self._acquisition_parameters,
self._path_template)
#self.get_acquisition_widget().use_osc_start(True)
self._processing_parameters = data_collection.processing_parameters
self._processing_widget.update_data_model(self._processing_parameters)
else:
self.setDisabled(True)
else:
self.setDisabled(True)
if isinstance(tree_item, Qt4_queue_item.SampleQueueItem) or \
isinstance(tree_item, Qt4_queue_item.DataCollectionGroupQueueItem) or \
isinstance(tree_item, Qt4_queue_item.DataCollectionQueueItem):
self._processing_widget.update_data_model(self._processing_parameters)
self._acq_widget.update_data_model(self._acquisition_parameters,
self._path_template)
def _create_task(self, sample, shape):
data_collections = []
for shape in self.get_selected_shapes():
snapshot = self._graphics_manager_hwobj.get_scene_snapshot(shape)
# Acquisition for start position
start_acq = self._create_acq(sample)
start_graphical_point, end_graphical_point = \
shape.get_graphical_points()
start_acq.acquisition_parameters.\
centred_position = copy.deepcopy(start_graphical_point.get_centred_position())
start_acq.acquisition_parameters.centred_position.\
snapshot_image = snapshot
start_acq.path_template.suffix = self._session_hwobj.suffix
# Add another acquisition for the end position
end_acq = self._create_acq(sample)
end_acq.acquisition_parameters.\
centred_position = copy.deepcopy(end_graphical_point.get_centred_position())
end_acq.acquisition_parameters.centred_position.\
snapshot_image = snapshot
end_acq.path_template.suffix = self._session_hwobj.suffix
processing_parameters = copy.deepcopy(self._processing_parameters)
dc = queue_model_objects.DataCollection([start_acq, end_acq],
sample.crystals[0],
processing_parameters)
dc.set_name(start_acq.path_template.get_prefix())
dc.set_number(start_acq.path_template.run_number)
dc.set_experiment_type(EXPERIMENT_TYPE.HELICAL)
dc.set_requires_centring(False)
dc.run_processing_after = self._processing_widget.processing_widget.\
run_processing_after_cbox.isChecked()
if self._processing_widget.processing_widget.\
run_processing_parallel_cbox.isChecked():
dc.run_processing_parallel = "LineScan"
data_collections.append(dc)
self._path_template.run_number += 1
return data_collections
def lines_treewidget_selection_changed(self):
self.enable_widgets(\
len(self._lines_widget.lines_treewidget.selectedItems()) > 0)
self._lines_widget.remove_line_button.setEnabled(\
len(self._lines_widget.lines_treewidget.selectedItems()) > 0)
self._lines_widget.swap_points_button.setEnabled(\
len(self._lines_widget.lines_treewidget.selectedItems()) > 0)
for shape, list_item in self._lines_map.iteritems():
self._graphics_manager_hwobj.select_shape(shape, list_item.isSelected())
self._acq_widget.emit_acq_parameters_changed()
def create_line_button_clicked(self):
self._graphics_manager_hwobj.create_line()
def create_auto_line_button_clicked(self):
self._graphics_manager_hwobj.create_auto_line()
def remove_line_button_clicked(self):
line_to_delete = None
for line, treewidget_item in self._lines_map.iteritems():
if treewidget_item.isSelected():
line_to_delete = line
break
if line_to_delete:
self._graphics_manager_hwobj.delete_shape(line_to_delete)
self.lines_treewidget_selection_changed()
def get_selected_shapes(self):
selected_lines = []
for line, treewidget_item in self._lines_map.iteritems():
if treewidget_item.isSelected():
selected_lines.append(line)
return selected_lines
def overlay_toggled(self, state):
self._graphics_manager_hwobj.set_display_overlay(state)
def overlay_alpha_changed(self, alpha_value):
for line, treewidget_item in self._lines_map.iteritems():
if treewidget_item.isSelected():
line.set_fill_alpha(alpha_value)
def swap_points_clicked(self):
for line, treewidget_item in self._lines_map.iteritems():
if treewidget_item.isSelected():
self._graphics_manager_hwobj.swap_line_points(line)
class CreateHelicalWidget(CreateTaskBase):
def __init__(self, parent=None, name=None, fl=0):
CreateTaskBase.__init__(self, parent, name, fl, 'Helical')
if not name:
self.setObjectName("create_helical_widget")
self.init_models()
# Hardware objects ----------------------------------------------------
# Internal variables --------------------------------------------------
self._lines_map = {}
# Graphic elements ----------------------------------------------------
self._lines_widget = loadUi(os.path.join(\
os.path.dirname(__file__), "ui_files/Qt4_helical_line_widget_layout.ui"))
self._acq_widget = AcquisitionWidget(
self,
"acquisition_widget",
layout='vertical',
acq_params=self._acquisition_parameters,
path_template=self._path_template)
self._data_path_widget = DataPathWidget(self,
'create_dc_path_widget',
data_model=self._path_template,
layout='vertical')
self._processing_widget = ProcessingWidget(
self, data_model=self._processing_parameters)
# Layout --------------------------------------------------------------
_main_vlayout = QVBoxLayout(self)
_main_vlayout.addWidget(self._lines_widget)
_main_vlayout.addWidget(self._acq_widget)
_main_vlayout.addWidget(self._data_path_widget)
_main_vlayout.addWidget(self._processing_widget)
_main_vlayout.addStretch(0)
_main_vlayout.setSpacing(6)
_main_vlayout.setContentsMargins(2, 2, 2, 2)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self._lines_widget.lines_treewidget.itemSelectionChanged.connect(\
self.lines_treewidget_selection_changed)
self._lines_widget.create_line_button.clicked.connect(\
self.create_line_button_clicked)
self._lines_widget.create_auto_line_button.clicked.connect(\
self.create_auto_line_button_clicked)
self._lines_widget.remove_line_button.clicked.connect(\
self.remove_line_button_clicked)
self._lines_widget.overlay_cbox.stateChanged.connect(\
self.overlay_toggled)
self._lines_widget.overlay_slider.valueChanged.connect(\
self.overlay_alpha_changed)
self._lines_widget.swap_points_button.clicked.connect(\
self.swap_points_clicked)
self._acq_widget.acqParametersChangedSignal.\
connect(self.acq_parameters_changed)
self._acq_widget.madEnergySelectedSignal.connect(\
self.mad_energy_selected)
self._acq_widget.acq_widget_layout.set_max_osc_range_button.clicked.\
connect(self.set_max_osc_total_range_clicked)
self._data_path_widget.pathTemplateChangedSignal.\
connect(self.path_template_changed)
self._processing_widget.enableProcessingSignal.connect(\
self._run_processing_toggled)
# Other ---------------------------------------------------------------
for col in range(self._lines_widget.lines_treewidget.columnCount()):
self._lines_widget.lines_treewidget.resizeColumnToContents(col)
#self._processing_widget.processing_widget.\
# run_processing_parallel_cbox.setChecked(False)
self.enable_widgets(False)
def enable_widgets(self, state):
self._acq_widget.setEnabled(state)
self._data_path_widget.setEnabled(state)
self._processing_widget.setEnabled(state)
def init_models(self):
CreateTaskBase.init_models(self)
self._energy_scan_result = queue_model_objects.EnergyScanResult()
self._processing_parameters = queue_model_objects.ProcessingParameters(
)
if self._beamline_setup_hwobj is not None:
has_shutter_less = self._beamline_setup_hwobj.\
detector_has_shutterless()
self._acquisition_parameters.shutterless = has_shutter_less
self._acquisition_parameters = self._beamline_setup_hwobj.\
get_default_acquisition_parameters("default_helical_values")
def set_beamline_setup(self, bl_setup_hwobj):
CreateTaskBase.set_beamline_setup(self, bl_setup_hwobj)
# At startup, if scene loaded from file, then update listwidget
shapes = self._graphics_manager_hwobj.get_shapes()
for shape in shapes:
if isinstance(shape, GraphicsItemLine):
self.shape_created(shape, "Line")
def shape_created(self, shape, shape_type):
if shape_type == "Line":
self._lines_widget.lines_treewidget.clearSelection()
#info_str_list = QStringList()
info_str_list = []
info_str_list.append(shape.get_display_name())
info_str_list.append("%d" % shape.get_points_index()[0])
info_str_list.append("%d" % shape.get_points_index()[1])
lines_treewidget_item = QTreeWidgetItem(\
self._lines_widget.lines_treewidget,
info_str_list)
lines_treewidget_item.setSelected(True)
self._lines_map[shape] = lines_treewidget_item
self.lines_treewidget_selection_changed()
def shape_deleted(self, shape, shape_type):
if shape_type == "Line" and self._lines_map.get(shape):
shape_index = self._lines_widget.lines_treewidget.\
indexFromItem(self._lines_map[shape])
self._lines_widget.lines_treewidget.\
takeTopLevelItem(shape_index.row())
self._lines_map.pop(shape)
def shape_changed(self, shape, shape_type):
lines_treewidget_item = self._lines_map.get(shape)
if lines_treewidget_item:
lines_treewidget_item.setText(1,
"%d" % shape.get_points_index()[0])
lines_treewidget_item.setText(2,
"%d" % shape.get_points_index()[1])
def approve_creation(self):
base_result = CreateTaskBase.approve_creation(self)
if len(self._lines_widget.lines_treewidget.selectedItems()) == 0:
logging.getLogger("GUI").\
warning("No lines selected, please select one or more lines.")
return False
else:
return base_result
def update_processing_parameters(self, crystal):
self._processing_parameters.space_group = crystal.space_group
self._processing_parameters.cell_a = crystal.cell_a
self._processing_parameters.cell_alpha = crystal.cell_alpha
self._processing_parameters.cell_b = crystal.cell_b
self._processing_parameters.cell_beta = crystal.cell_beta
self._processing_parameters.cell_c = crystal.cell_c
self._processing_parameters.cell_gamma = crystal.cell_gamma
self._processing_widget.update_data_model(self._processing_parameters)
def select_shape_with_cpos(self, start_cpos, end_cpos, num_images):
selected_line = None
self._lines_widget.overlay_slider.setEnabled(False)
self._lines_widget.overlay_cbox.setEnabled(False)
self._graphics_manager_hwobj.de_select_all()
for shape in self._graphics_manager_hwobj.get_shapes():
if isinstance(shape, GraphicsItemLine):
(start_cpos_index, end_cpos_index) = shape.get_points_index()
if start_cpos_index == start_cpos.index and \
end_cpos_index == end_cpos.index:
self._graphics_manager_hwobj.select_shape(shape)
shape.set_num_images(num_images)
selected_line = shape
self._lines_widget.overlay_slider.setEnabled(True)
self._lines_widget.overlay_cbox.setEnabled(True)
#de-select previous selected list items and
#select the current shape (Line).
"""
for (list_item, shape) in self._lines_map.iteritems():
if selected_line == shape:
list_item.setSelected(True)
else:
list_item.setSelected(False)
"""
def single_item_selection(self, tree_item):
CreateTaskBase.single_item_selection(self, tree_item)
if isinstance(tree_item, Qt4_queue_item.SampleQueueItem):
sample_model = tree_item.get_model()
self._processing_parameters = sample_model.processing_parameters
#self._processing_parameters = copy.deepcopy(self._processing_parameters)
self._processing_widget.update_data_model(
self._processing_parameters)
elif isinstance(tree_item, Qt4_queue_item.BasketQueueItem):
self.setDisabled(False)
elif isinstance(tree_item, Qt4_queue_item.DataCollectionQueueItem):
data_collection = tree_item.get_model()
if data_collection.is_helical():
self.setDisabled(tree_item.get_model().is_executed())
self._path_template = data_collection.get_path_template()
self._data_path_widget.update_data_model(self._path_template)
self._acquisition_parameters = data_collection.acquisitions[0].\
acquisition_parameters
if len(data_collection.acquisitions) == 2:
start_cpos = data_collection.acquisitions[0].acquisition_parameters.\
centred_position
end_cpos = data_collection.acquisitions[1].acquisition_parameters.\
centred_position
num_images = data_collection.acquisitions[0].acquisition_parameters.\
num_images
self.select_shape_with_cpos(start_cpos, end_cpos,
num_images)
self._acq_widget.update_data_model(
self._acquisition_parameters, self._path_template)
#self.get_acquisition_widget().use_osc_start(True)
self._processing_parameters = data_collection.processing_parameters
self._processing_widget.update_data_model(
self._processing_parameters)
else:
self.setDisabled(True)
else:
self.setDisabled(True)
if isinstance(tree_item, Qt4_queue_item.SampleQueueItem) or \
isinstance(tree_item, Qt4_queue_item.DataCollectionGroupQueueItem) or \
isinstance(tree_item, Qt4_queue_item.DataCollectionQueueItem):
self._processing_widget.update_data_model(
self._processing_parameters)
self._acq_widget.update_data_model(self._acquisition_parameters,
self._path_template)
def _create_task(self, sample, shape):
data_collections = []
for shape in self.get_selected_shapes():
snapshot = self._graphics_manager_hwobj.get_scene_snapshot(shape)
# Acquisition for start position
start_acq = self._create_acq(sample)
start_graphical_point, end_graphical_point = \
shape.get_graphical_points()
start_acq.acquisition_parameters.\
centred_position = copy.deepcopy(start_graphical_point.get_centred_position())
start_acq.acquisition_parameters.centred_position.\
snapshot_image = snapshot
start_acq.path_template.suffix = self._session_hwobj.suffix
# Add another acquisition for the end position
end_acq = self._create_acq(sample)
end_acq.acquisition_parameters.\
centred_position = copy.deepcopy(end_graphical_point.get_centred_position())
end_acq.acquisition_parameters.centred_position.\
snapshot_image = snapshot
end_acq.path_template.suffix = self._session_hwobj.suffix
processing_parameters = copy.deepcopy(self._processing_parameters)
dc = queue_model_objects.DataCollection([start_acq, end_acq],
sample.crystals[0],
processing_parameters)
dc.set_name(start_acq.path_template.get_prefix())
dc.set_number(start_acq.path_template.run_number)
dc.set_experiment_type(EXPERIMENT_TYPE.HELICAL)
dc.set_requires_centring(False)
dc.run_processing_after = self._processing_widget.processing_widget.\
run_processing_after_cbox.isChecked()
dc.run_processing_parallel = self._processing_widget.processing_widget.\
run_processing_parallel_cbox.isChecked()
data_collections.append(dc)
self._path_template.run_number += 1
return data_collections
def set_max_osc_total_range_clicked(self):
num_images = int(
self._acq_widget.acq_widget_layout.num_images_ledit.text())
(lower,
upper), exp_time = self._acq_widget.update_osc_total_range_limits()
self._acq_widget.acq_widget_layout.osc_start_ledit.setText(\
"%.2f" % lower)
self._acq_widget.acq_widget_layout.osc_total_range_ledit.setText(\
"%.2f" % abs(upper - lower))
self._acq_widget.acq_widget_layout.osc_range.setText(\
"%.2f" % (abs(lower - upper) / 2 / num_images))
def lines_treewidget_selection_changed(self):
self.enable_widgets(\
len(self._lines_widget.lines_treewidget.selectedItems()) > 0)
self._lines_widget.remove_line_button.setEnabled(\
len(self._lines_widget.lines_treewidget.selectedItems()) > 0)
self._lines_widget.swap_points_button.setEnabled(\
len(self._lines_widget.lines_treewidget.selectedItems()) > 0)
for shape, list_item in self._lines_map.iteritems():
self._graphics_manager_hwobj.select_shape(shape,
list_item.isSelected())
self._acq_widget.emit_acq_parameters_changed()
def create_line_button_clicked(self):
self._graphics_manager_hwobj.create_line()
def create_auto_line_button_clicked(self):
self._graphics_manager_hwobj.create_auto_line()
def remove_line_button_clicked(self):
line_to_delete = None
for line, treewidget_item in self._lines_map.iteritems():
if treewidget_item.isSelected():
line_to_delete = line
break
if line_to_delete:
self._graphics_manager_hwobj.delete_shape(line_to_delete)
self.lines_treewidget_selection_changed()
def get_selected_shapes(self):
selected_lines = []
for line, treewidget_item in self._lines_map.iteritems():
if treewidget_item.isSelected():
selected_lines.append(line)
return selected_lines
def overlay_toggled(self, state):
self._graphics_manager_hwobj.set_display_overlay(state)
def overlay_alpha_changed(self, alpha_value):
for line, treewidget_item in self._lines_map.iteritems():
if treewidget_item.isSelected():
line.set_fill_alpha(alpha_value)
def swap_points_clicked(self):
for line, treewidget_item in self._lines_map.iteritems():
if treewidget_item.isSelected():
self._graphics_manager_hwobj.swap_line_points(line)
