class CreateDiscreteWidget(CreateTaskBase):
def __init__(self, parent=None, name=None, fl=0):
CreateTaskBase.__init__(self, parent, name, fl, "Standard")
if not name:
self.setName("create_discrete_widget")
self.previous_energy = None
self.init_models()
#
# Layout
#
v_layout = qt.QVBoxLayout(self, 2, 5, "v_layout")
self._acq_gbox = qt.QVGroupBox("Acquisition", self, "acq_gbox")
self._acq_widget = AcquisitionWidget(
self._acq_gbox,
"acquisition_widget",
layout="vertical",
acq_params=self._acquisition_parameters,
path_template=self._path_template,
)
self._data_path_gbox = qt.QVGroupBox("Data location", self, "data_path_gbox")
self._data_path_widget = DataPathWidget(
self._data_path_gbox,
"create_dc_path_widget",
data_model=self._path_template,
layout="vertical",
)
self._processing_gbox = qt.QVGroupBox("Processing", self, "processing_gbox")
self._processing_widget = ProcessingWidget(
self._processing_gbox, data_model=self._processing_parameters
)
v_layout.addWidget(self._acq_gbox)
v_layout.addWidget(self._data_path_gbox)
v_layout.addWidget(self._processing_gbox)
v_layout.addStretch()
dp_layout = self._data_path_widget.data_path_widget_layout
self.connect(
self._acq_widget,
qt.PYSIGNAL("mad_energy_selected"),
self.mad_energy_selected,
)
self.connect(
dp_layout.child("prefix_ledit"),
qt.SIGNAL("textChanged(const QString &)"),
self._prefix_ledit_change,
)
self.connect(
dp_layout.child("run_number_ledit"),
qt.SIGNAL("textChanged(const QString &)"),
self._run_number_ledit_change,
)
self.connect(
self._acq_widget,
qt.PYSIGNAL("path_template_changed"),
self.handle_path_conflict,
)
self.connect(
self._data_path_widget,
qt.PYSIGNAL("path_template_changed"),
self.handle_path_conflict,
)
def init_models(self):
CreateTaskBase.init_models(self)
self._energy_scan_result = queue_model_objects.EnergyScanResult()
self._processing_parameters = queue_model_objects.ProcessingParameters()
def set_tunable_energy(self, state):
self._acq_widget.set_tunable_energy(state)
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 mad_energy_selected(self, name, energy, state):
item = self._current_selected_items[0]
model = item.get_model()
if state:
self._path_template.mad_prefix = name
else:
self._path_template.mad_prefix = ""
run_number = self._beamline_setup_hwobj.queue_model_hwobj.get_next_run_number(
self._path_template
)
data_path_widget = self.get_data_path_widget()
data_path_widget.set_run_number(run_number)
data_path_widget.set_prefix(self._path_template.base_prefix)
if self.isEnabled():
if isinstance(item, queue_item.TaskQueueItem) and not isinstance(
item, queue_item.DataCollectionGroupQueueItem
):
model.set_name(self._path_template.get_prefix())
item.setText(0, model.get_name())
def single_item_selection(self, tree_item):
CreateTaskBase.single_item_selection(self, tree_item)
if isinstance(tree_item, queue_item.SampleQueueItem):
sample_model = tree_item.get_model()
# self._processing_parameters = copy.deepcopy(self._processing_parameters)
self._processing_parameters = sample_model.processing_parameters
self._processing_widget.update_data_model(self._processing_parameters)
self._acq_widget.disable_inverse_beam(False)
elif isinstance(tree_item, queue_item.BasketQueueItem):
self.setDisabled(False)
elif isinstance(tree_item, queue_item.DataCollectionQueueItem):
dc = tree_item.get_model()
if dc.experiment_type != queue_model_enumerables.EXPERIMENT_TYPE.HELICAL:
if dc.is_executed():
self.setDisabled(True)
else:
self.setDisabled(False)
sample_data_model = self.get_sample_item(tree_item).get_model()
energy_scan_result = sample_data_model.crystals[0].energy_scan_result
self._acq_widget.set_energies(energy_scan_result)
self._acq_widget.disable_inverse_beam(True)
self._path_template = dc.get_path_template()
self._data_path_widget.update_data_model(self._path_template)
self._acquisition_parameters = dc.acquisitions[0].acquisition_parameters
self._acq_widget.update_data_model(
self._acquisition_parameters, self._path_template
)
self.get_acquisition_widget().use_osc_start(True)
if len(dc.acquisitions) == 1:
self.select_shape_with_cpos(
self._acquisition_parameters.centred_position
)
self._processing_parameters = dc.processing_parameters
self._processing_widget.update_data_model(self._processing_parameters)
else:
self.setDisabled(True)
else:
self.setDisabled(True)
def approve_creation(self):
result = CreateTaskBase.approve_creation(self)
selected_shapes = self._shape_history.selected_shapes
for shape in selected_shapes:
if isinstance(shape, shape_history.Line) or isinstance(
shape, shape_history.CanvasGrid
):
result = False
return result
# Called by the owning widget (task_toolbox_widget) to create
# a collection. When a data collection group is selected.
def _create_task(self, sample, shape):
tasks = []
cpos = queue_model_objects.CentredPosition()
cpos.snapshot_image = self._shape_history.get_snapshot([])
if shape is not None:
# Shapes selected and sample is mounted, get the
# centred positions for the shapes
if isinstance(shape, shape_history.Point):
snapshot = self._shape_history.get_snapshot([shape.qub_point])
cpos = copy.deepcopy(shape.get_centred_positions()[0])
cpos.snapshot_image = snapshot
if self._acq_widget.use_inverse_beam():
total_num_images = self._acquisition_parameters.num_images
subwedge_size = self._acq_widget.get_num_subwedges()
osc_range = self._acquisition_parameters.osc_range
osc_start = self._acquisition_parameters.osc_start
run_number = self._path_template.run_number
subwedges = queue_model_objects.create_inverse_beam_sw(
total_num_images, subwedge_size, osc_range, osc_start, run_number
)
self._acq_widget.set_use_inverse_beam(False)
for sw in subwedges:
tasks.extend(
self.create_dc(
sample, sw[3], sw[0], sw[1], sw[2], cpos=cpos, inverse_beam=True
)
)
self._path_template.run_number += 1
else:
tasks.extend(self.create_dc(sample, cpos=cpos))
self._path_template.run_number += 1
return tasks
def create_dc(
self,
sample,
run_number=None,
start_image=None,
num_images=None,
osc_start=None,
sc=None,
cpos=None,
inverse_beam=False,
):
tasks = []
# Acquisition for start position
acq = self._create_acq(sample)
if run_number:
acq.path_template.run_number = run_number
if start_image:
acq.acquisition_parameters.first_image = start_image
acq.path_template.start_num = start_image
if num_images:
acq.acquisition_parameters.num_images = num_images
acq.path_template.num_files = num_images
if osc_start:
acq.acquisition_parameters.osc_start = osc_start
if inverse_beam:
acq.acquisition_parameters.inverse_beam = False
acq.acquisition_parameters.centred_position = cpos
processing_parameters = copy.deepcopy(self._processing_parameters)
dc = queue_model_objects.DataCollection(
[acq], sample.crystals[0], processing_parameters
)
dc.set_name(acq.path_template.get_prefix())
dc.set_number(acq.path_template.run_number)
dc.experiment_type = queue_model_enumerables.EXPERIMENT_TYPE.NATIVE
tasks.append(dc)
return tasks
class CreateHelicalWidget(CreateTaskBase):
def __init__(self, parent = None,name = None, fl = 0):
CreateTaskBase.__init__(self, parent, name, fl, 'Helical')
if not name:
self.setName("create_helical_widget")
#
# Data attributes
#
self.init_models()
self._prev_pos = None
self._current_pos = None
self._list_item_map = {}
self.init_models()
#
# Layout
#
v_layout = qt.QVBoxLayout(self, 2, 5, "v_layout")
self._lines_gbox = qt.QGroupBox('Lines', self, "lines_gbox")
self._lines_gbox.setColumnLayout(0, qt.Qt.Vertical)
self._lines_gbox.layout().setSpacing(6)
self._lines_gbox.layout().setMargin(11)
lines_gbox_layout = qt.QHBoxLayout(self._lines_gbox.layout())
lines_gbox_layout.setAlignment(qt.Qt.AlignTop)
self._list_box = qt.QListBox(self._lines_gbox, "helical_page")
self._list_box.setSelectionMode(qt.QListBox.Extended)
self._list_box.setFixedWidth(200)
self._list_box.setFixedHeight(75)
list_box_tool_tip = "Select the line(s) to perfrom helical scan on"
qt.QToolTip.add(self._list_box, list_box_tool_tip)
lines_gbox_layout.addWidget(self._list_box)
button_layout = qt.QVBoxLayout(None, 0, 6, "button_layout")
button_layout.setSpacing(5)
add_button = qt.QPushButton("+", self._lines_gbox, "add_button")
add_button.setFixedWidth(20)
add_button.setFixedHeight(20)
remove_button = qt.QPushButton("-", self._lines_gbox, "add_button")
remove_button.setFixedWidth(20)
remove_button.setFixedHeight(20)
button_layout.addWidget(add_button)
button_layout.addWidget(remove_button)
lines_gbox_layout.addLayout(button_layout)
add_button_tool_tip = "Add a line between two saved positions, " \
"CTRL click to select more than one position"
qt.QToolTip.add(add_button, add_button_tool_tip)
remove_button_tool_tip = "Remove selected line(s)"
qt.QToolTip.add(remove_button, remove_button_tool_tip)
self._acq_gbox = qt.QVGroupBox('Acquisition', self, 'acq_gbox')
self._acq_widget = \
AcquisitionWidget(self._acq_gbox,
"acquisition_widget", layout='vertical',
acq_params=self._acquisition_parameters,
path_template=self._path_template)
self._acq_widget.disable_inverse_beam(True)
self._data_path_gbox = qt.QVGroupBox('Data location', self,
'data_path_gbox')
self._data_path_widget = \
DataPathWidget(self._data_path_gbox,
data_model = self._path_template,
layout = 'vertical')
self._processing_gbox = qt.QVGroupBox('Processing', self,
'processing_gbox')
self._processing_widget = \
ProcessingWidget(self._processing_gbox,
data_model = self._processing_parameters)
v_layout.addWidget(self._lines_gbox)
v_layout.addWidget(self._acq_gbox)
v_layout.addWidget(self._data_path_gbox)
v_layout.addWidget(self._processing_gbox)
qt.QObject.connect(add_button, qt.SIGNAL("clicked()"),
self.add_clicked)
qt.QObject.connect(remove_button, qt.SIGNAL("clicked()"),
self.remove_clicked)
qt.QObject.connect(self._list_box, qt.SIGNAL("selectionChanged()"),
self.list_box_selection_changed)
prefix_ledit = self._data_path_widget.\
data_path_widget_layout.child('prefix_ledit')
run_number_ledit = self._data_path_widget.\
data_path_widget_layout.child('run_number_ledit')
self.connect(prefix_ledit,
qt.SIGNAL("textChanged(const QString &)"),
self._prefix_ledit_change)
self.connect(run_number_ledit,
qt.SIGNAL("textChanged(const QString &)"),
self._run_number_ledit_change)
self.connect(self._data_path_widget,
qt.PYSIGNAL("path_template_changed"),
self.handle_path_conflict)
def init_models(self):
CreateTaskBase.init_models(self)
self._energy_scan_result = qmo.EnergyScanResult()
self._processing_parameters = qmo.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")
else:
self._acquisition_parameters = qmo.AcquisitionParameters()
self._path_template = qmo.PathTemplate()
def add_clicked(self):
selected_shapes = list(self._shape_history.selected_shapes.values())
if len(selected_shapes) == 2:
p1 = selected_shapes[0]
p2 = selected_shapes[1]
line = shape_history.\
Line(self._shape_history.get_drawing(),
p1.qub_point, p2.qub_point,
p1.centred_position, p2.centred_position)
line.show()
self._shape_history.add_shape(line)
points_index = line.get_points_index()
if points_index:
display_name = "Line (points: %d, %d)" % \
(points_index[0], points_index[1])
else:
display_name = "Line (points: #, #)"
list_box_item = qt.QListBoxText(self._list_box, display_name)
#list_box_item = qt.QListBoxText(self._list_box, 'Line')
self._list_item_map[list_box_item] = line
# De select previous items
for item in self.selected_items():
self._list_box.setSelected(item, False)
self._list_box.setSelected(list_box_item, True)
def remove_clicked(self):
selected_items = self.selected_items()
if selected_items:
for item in selected_items:
self._list_box.removeItem(self._list_box.index(item))
line = self._list_item_map[item]
self._shape_history.delete_shape(line)
del self._list_item_map[item]
# Calback from shape_history, called when a shape is deleted
def shape_deleted(self, shape):
if isinstance(shape, shape_history.Point):
items_to_remove = []
for (list_item, line) in self._list_item_map.items():
for qub_object in shape.get_qub_objects():
if qub_object in line.get_qub_objects():
items_to_remove.append((list_item, line))
for (list_item, line) in items_to_remove:
self._list_box.removeItem(self._list_box.index(list_item))
del self._list_item_map[list_item]
def centred_position_selection(self, positions):
if len(positions) == 1:
self._prev_pos = positions[0]
elif len(positions) == 2:
for pos in positions:
if pos != self._prev_pos:
self._current_pos = pos
else:
self._prev_pos = None
self._current_pos = None
self._list_box.clearSelection()
def list_box_selection_changed(self):
self.show_selected_lines()
def selected_items(self):
selected_items = []
for item_index in range(0, self._list_box.numRows()):
if self._list_box.isSelected(item_index):
selected_items.append(self._list_box.item(item_index))
return selected_items
def show_selected_lines(self):
selected_items = self.selected_items()
for list_item in list(self._list_item_map.keys()):
line = self._list_item_map[list_item]
if list_item in selected_items:
self._shape_history.select_shape(line)
else:
self._shape_history.de_select_shape(line)
def approve_creation(self):
base_result = CreateTaskBase.approve_creation(self)
selected_lines = False
if self.selected_items():
selected_lines = True
else:
logging.getLogger("user_level_log").\
warning("No lines selected, please select one or more lines.")
return base_result and selected_lines
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):
self._shape_history.de_select_all()
selected_line = None
for shape in self._shape_history.get_shapes():
if isinstance(shape, shape_history.Line):
if shape.get_centred_positions()[0] == start_cpos and\
shape.get_centred_positions()[1] == end_cpos:
self._shape_history.select_shape(shape)
selected_line = shape
#de-select previous selected list items and
#select the current shape (Line).
for (list_item, shape) in self._list_item_map.items():
if selected_line is shape:
self._list_box.setSelected(list_item, True)
else:
self._list_box.setSelected(list_item, False)
def single_item_selection(self, tree_item):
CreateTaskBase.single_item_selection(self, tree_item)
if isinstance(tree_item, 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, queue_item.BasketQueueItem):
self.setDisabled(False)
elif isinstance(tree_item, queue_item.DataCollectionQueueItem):
data_collection = tree_item.get_model()
if data_collection.experiment_type == EXPERIMENT_TYPE.HELICAL:
if tree_item.get_model().is_executed():
self.setDisabled(True)
else:
self.setDisabled(False)
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
self.select_shape_with_cpos(start_cpos, end_cpos)
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, queue_item.SampleQueueItem) or \
isinstance(tree_item, queue_item.DataCollectionGroupQueueItem) or \
isinstance(tree_item, 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 = []
if isinstance(shape, shape_history.Line ):
if shape.get_qub_objects() is not None:
snapshot = self._shape_history.get_snapshot(shape.get_qub_objects())
else:
snapshot = self._shape_history.get_snapshot([])
# Acquisition for start position
start_acq = self._create_acq(sample)
start_acq.acquisition_parameters.\
centred_position = copy.deepcopy(shape.start_cpos)
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 = shape.end_cpos
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 = qmo.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.experiment_type = EXPERIMENT_TYPE.HELICAL
data_collections.append(dc)
self._path_template.run_number += 1
return data_collections
class CreateDiscreteWidget(CreateTaskBase):
def __init__(self, parent=None, name=None, fl=0):
CreateTaskBase.__init__(self, parent, name, fl, 'Standard')
if not name:
self.setName("create_discrete_widget")
self.previous_energy = None
self.init_models()
#
# Layout
#
v_layout = qt.QVBoxLayout(self, 2, 5, "v_layout")
self._acq_gbox = qt.QVGroupBox('Acquisition', self, 'acq_gbox')
self._acq_widget = \
AcquisitionWidget(self._acq_gbox,
"acquisition_widget",
layout='vertical',
acq_params=self._acquisition_parameters,
path_template=self._path_template)
self._data_path_gbox = qt.QVGroupBox('Data location', self,
'data_path_gbox')
self._data_path_widget = \
DataPathWidget(self._data_path_gbox,
'create_dc_path_widget',
data_model=self._path_template,
layout='vertical')
self._processing_gbox = qt.QVGroupBox('Processing', self,
'processing_gbox')
self._processing_widget = \
ProcessingWidget(self._processing_gbox,
data_model=self._processing_parameters)
v_layout.addWidget(self._acq_gbox)
v_layout.addWidget(self._data_path_gbox)
v_layout.addWidget(self._processing_gbox)
v_layout.addStretch()
dp_layout = self._data_path_widget.data_path_widget_layout
self.connect(self._acq_widget, qt.PYSIGNAL('mad_energy_selected'),
self.mad_energy_selected)
self.connect(dp_layout.child('prefix_ledit'),
qt.SIGNAL("textChanged(const QString &)"),
self._prefix_ledit_change)
self.connect(dp_layout.child('run_number_ledit'),
qt.SIGNAL("textChanged(const QString &)"),
self._run_number_ledit_change)
self.connect(self._acq_widget, qt.PYSIGNAL("path_template_changed"),
self.handle_path_conflict)
self.connect(self._data_path_widget,
qt.PYSIGNAL("path_template_changed"),
self.handle_path_conflict)
def init_models(self):
CreateTaskBase.init_models(self)
self._energy_scan_result = qmo.EnergyScanResult()
self._processing_parameters = qmo.ProcessingParameters()
def set_tunable_energy(self, state):
self._acq_widget.set_tunable_energy(state)
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 mad_energy_selected(self, name, energy, state):
item = self._current_selected_items[0]
model = item.get_model()
if state:
self._path_template.mad_prefix = name
else:
self._path_template.mad_prefix = ''
run_number = self._beamline_setup_hwobj.queue_model_hwobj.\
get_next_run_number(self._path_template)
data_path_widget = self.get_data_path_widget()
data_path_widget.set_run_number(run_number)
data_path_widget.set_prefix(self._path_template.base_prefix)
if self.isEnabled():
if isinstance(item, queue_item.TaskQueueItem) and \
not isinstance(item, queue_item.DataCollectionGroupQueueItem):
model.set_name(self._path_template.get_prefix())
item.setText(0, model.get_name())
def single_item_selection(self, tree_item):
CreateTaskBase.single_item_selection(self, tree_item)
if isinstance(tree_item, queue_item.SampleQueueItem):
self._processing_parameters = copy.deepcopy(
self._processing_parameters)
self._processing_widget.update_data_model(
self._processing_parameters)
self._acq_widget.disable_inverse_beam(False)
elif isinstance(tree_item, queue_item.DataCollectionQueueItem):
dc = tree_item.get_model()
if dc.experiment_type != queue_model_enumerables.EXPERIMENT_TYPE.HELICAL:
if dc.is_executed():
self.setDisabled(True)
else:
self.setDisabled(False)
sample_data_model = self.get_sample_item(tree_item).get_model()
energy_scan_result = sample_data_model.crystals[
0].energy_scan_result
self._acq_widget.set_energies(energy_scan_result)
self._acq_widget.disable_inverse_beam(True)
self._path_template = dc.get_path_template()
self._data_path_widget.update_data_model(self._path_template)
self._acquisition_parameters = dc.acquisitions[
0].acquisition_parameters
self._acq_widget.update_data_model(
self._acquisition_parameters, self._path_template)
self.get_acquisition_widget().use_osc_start(True)
if len(dc.acquisitions) == 1:
self.select_shape_with_cpos(self._acquisition_parameters.\
centred_position)
self._processing_parameters = dc.processing_parameters
self._processing_widget.update_data_model(
self._processing_parameters)
else:
self.setDisabled(True)
else:
self.setDisabled(True)
def approve_creation(self):
result = CreateTaskBase.approve_creation(self)
selected_shapes = self._shape_history.selected_shapes
for shape in selected_shapes:
if isinstance(shape, shape_history.Line):
result = False
return result
# Called by the owning widget (task_toolbox_widget) to create
# a collection. When a data collection group is selected.
def _create_task(self, sample, shape):
tasks = []
if not shape:
cpos = qmo.CentredPosition()
cpos.snapshot_image = self._shape_history.get_snapshot([])
else:
# Shapes selected and sample is mounted, get the
# centred positions for the shapes
if isinstance(shape, shape_history.Point):
snapshot = self._shape_history.\
get_snapshot([shape.qub_point])
cpos = copy.deepcopy(shape.get_centred_positions()[0])
cpos.snapshot_image = snapshot
if self._acq_widget.use_inverse_beam():
total_num_images = self._acquisition_parameters.num_images
subwedge_size = self._acq_widget.get_num_subwedges()
osc_range = self._acquisition_parameters.osc_range
osc_start = self._acquisition_parameters.osc_start
run_number = self._path_template.run_number
subwedges = qmo.create_inverse_beam_sw(total_num_images,
subwedge_size, osc_range,
osc_start, run_number)
self._acq_widget.set_use_inverse_beam(False)
for sw in subwedges:
tasks.extend(
self.create_dc(sample,
sw[3],
sw[0],
sw[1],
sw[2],
cpos=cpos,
inverse_beam=True))
self._path_template.run_number += 1
else:
tasks.extend(self.create_dc(sample, cpos=cpos))
self._path_template.run_number += 1
return tasks
def create_dc(self,
sample,
run_number=None,
start_image=None,
num_images=None,
osc_start=None,
sc=None,
cpos=None,
inverse_beam=False):
tasks = []
# Acquisition for start position
acq = qmo.Acquisition()
acq.acquisition_parameters = \
copy.deepcopy(self._acquisition_parameters)
acq.acquisition_parameters.collect_agent = \
queue_model_enumerables.COLLECTION_ORIGIN.MXCUBE
acq.path_template = copy.deepcopy(self._path_template)
acq.acquisition_parameters.centred_position = cpos
if '<sample_name>' in acq.path_template.directory:
name = sample.get_name().replace(':', '-')
acq.path_template.directory = acq.path_template.directory.\
replace('<sample_name>', name)
acq.path_template.process_directory = acq.path_template.process_directory.\
replace('<sample_name>', name)
if '<acronym>-<name>' in acq.path_template.base_prefix:
acq.path_template.base_prefix = self.get_default_prefix(sample)
acq.path_template.run_numer = self._beamline_setup_hwobj.queue_model_hwobj.\
get_next_run_number(acq.path_template)
if run_number:
acq.path_template.run_number = run_number
if start_image:
acq.acquisition_parameters.first_image = start_image
acq.path_template.start_num = start_image
if num_images:
acq.acquisition_parameters.num_images = num_images
acq.path_template.num_files = num_images
if osc_start:
acq.acquisition_parameters.osc_start = osc_start
if inverse_beam:
acq.acquisition_parameters.inverse_beam = False
if self._beamline_setup_hwobj.in_plate_mode():
acq.acquisition_parameters.take_snapshots = False
else:
acq.acquisition_parameters.take_snapshots = True
processing_parameters = copy.deepcopy(self._processing_parameters)
dc = qmo.DataCollection([acq], sample.crystals[0],
processing_parameters)
dc.set_name(acq.path_template.get_prefix())
dc.set_number(acq.path_template.run_number)
dc.experiment_type = queue_model_enumerables.EXPERIMENT_TYPE.NATIVE
tasks.append(dc)
return tasks
class CreateHelicalWidget(CreateTaskBase):
def __init__(self, parent = None,name = None, fl = 0):
CreateTaskBase.__init__(self, parent, name, fl, 'Helical')
if not name:
self.setName("create_helical_widget")
#
# Data attributes
#
self.init_models()
self._prev_pos = None
self._current_pos = None
self._list_item_map = {}
self.init_models()
#
# Layout
#
v_layout = qt.QVBoxLayout(self, 2, 5, "v_layout")
self._lines_gbox = qt.QGroupBox('Lines', self, "lines_gbox")
self._lines_gbox.setColumnLayout(0, qt.Qt.Vertical)
self._lines_gbox.layout().setSpacing(6)
self._lines_gbox.layout().setMargin(11)
lines_gbox_layout = qt.QHBoxLayout(self._lines_gbox.layout())
lines_gbox_layout.setAlignment(qt.Qt.AlignTop)
self._list_box = qt.QListBox(self._lines_gbox, "helical_page")
self._list_box.setSelectionMode(qt.QListBox.Extended)
self._list_box.setFixedWidth(175)
lines_gbox_layout.addWidget(self._list_box)
button_layout = qt.QVBoxLayout(None, 0, 6, "button_layout")
button_layout.setSpacing(20)
add_button = qt.QPushButton("+", self._lines_gbox, "add_button")
add_button.setFixedWidth(25)
remove_button = qt.QPushButton("-", self._lines_gbox, "add_button")
remove_button.setFixedWidth(25)
button_layout.addWidget(add_button)
button_layout.addWidget(remove_button)
lines_gbox_layout.addLayout(button_layout)
self._acq_gbox = qt.QVGroupBox('Acquisition', self, 'acq_gbox')
self._acq_widget = \
AcquisitionWidget(self._acq_gbox,
"acquisition_widget", layout='vertical',
acq_params=self._acquisition_parameters,
path_template=self._path_template)
self._acq_widget.disable_inverse_beam(True)
self._data_path_gbox = qt.QVGroupBox('Data location', self,
'data_path_gbox')
self._data_path_widget = \
DataPathWidget(self._data_path_gbox,
data_model = self._path_template,
layout = DataPathWidgetVerticalLayout)
self._processing_gbox = qt.QVGroupBox('Processing', self,
'processing_gbox')
self._processing_widget = \
ProcessingWidget(self._processing_gbox,
data_model = self._processing_parameters)
v_layout.addWidget(self._lines_gbox)
v_layout.addWidget(self._acq_gbox)
v_layout.addWidget(self._data_path_gbox)
v_layout.addWidget(self._processing_gbox)
qt.QObject.connect(add_button, qt.SIGNAL("clicked()"),
self.add_clicked)
qt.QObject.connect(remove_button, qt.SIGNAL("clicked()"),
self.remove_clicked)
qt.QObject.connect(self._list_box, qt.SIGNAL("selectionChanged()"),
self.list_box_selection_changed)
prefix_ledit = self._data_path_widget.\
data_path_widget_layout.prefix_ledit
run_number_ledit = self._data_path_widget.\
data_path_widget_layout.run_number_ledit
self.connect(prefix_ledit,
qt.SIGNAL("textChanged(const QString &)"),
self._prefix_ledit_change)
self.connect(run_number_ledit,
qt.SIGNAL("textChanged(const QString &)"),
self._run_number_ledit_change)
self.connect(self._data_path_widget,
qt.PYSIGNAL("path_template_changed"),
self.handle_path_conflict)
def init_models(self):
CreateTaskBase.init_models(self)
self._energy_scan_result = qmo.EnergyScanResult()
self._processing_parameters = qmo.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()
try:
transmission = self._beamline_setup_hwobj.\
transmission_hwobj.getAttFactor()
transmission = round(float(transmission), 1)
except AttributeError:
transmission = 0
try:
resolution = self._beamline_setup_hwobj.\
resolution_hwobj.getPosition()
resolution = round(float(resolution), 4)
except AttributeError:
resolution = 0
try:
energy = self._beamline_setup_hwobj.energy_hwobj.\
getCurrentEnergy()
energy = round(float(energy), 2)
except AttributeError:
energy = 0
self._acquisition_parameters.resolution = resolution
self._acquisition_parameters.energy = energy
self._acquisition_parameters.transmission = transmission
else:
self._acquisition_parameters = qmo.AcquisitionParameters()
self._path_template = qmo.PathTemplate()
def add_clicked(self):
selected_shapes = self._shape_history.selected_shapes.values()
if len(selected_shapes) == 2:
p1 = selected_shapes[1]
p2 = selected_shapes[0]
line = shape_history.\
Line(self._shape_history.get_drawing(),
p1.qub_point, p2.qub_point,
p1.centred_position, p2.centred_position)
line.show()
self._shape_history.add_shape(line)
list_box_item = qt.QListBoxText(self._list_box, 'Line')
self._list_item_map[list_box_item] = line
self._list_box.setSelected(list_box_item, True)
def remove_clicked(self):
selected_items = self.selected_items()
if selected_items:
for item in selected_items:
self._list_box.removeItem(self._list_box.index(item))
line = self._list_item_map[item]
self._shape_history.delete_shape(line)
del self._list_item_map[item]
# Calback from shape_history, called when a shape is deleted
def shape_deleted(self, shape):
if isinstance(shape, shape_history.Point):
items_to_remove = []
for (list_item, line) in self._list_item_map.iteritems():
for qub_object in shape.get_qub_objects():
if qub_object in line.get_qub_objects():
items_to_remove.append((list_item, line))
for (list_item, line) in items_to_remove:
self._list_box.removeItem(self._list_box.index(list_item))
del self._list_item_map[list_item]
def centred_position_selection(self, positions):
if len(positions) == 1:
self._prev_pos = positions[0]
elif len(positions) == 2:
for pos in positions:
if pos != self._prev_pos:
self._current_pos = pos
else:
self._prev_pos = None
self._current_pos = None
def list_box_selection_changed(self):
self.show_selected_lines()
def selected_items(self):
selected_items = []
for item_index in range(0, self._list_box.numRows()):
if self._list_box.isSelected(item_index):
selected_items.append(self._list_box.item(item_index))
return selected_items
def show_selected_lines(self):
selected_items = self.selected_items()
for list_item in self._list_item_map.keys():
line = self._list_item_map[list_item]
if list_item in selected_items:
line.highlight()
else:
line.unhighlight()
def approve_creation(self):
base_result = CreateTaskBase.approve_creation(self)
selected_lines = False
if self.selected_items():
selected_lines = True
else:
logging.getLogger("user_level_log").\
info("No lines selected, please select one or more lines.")
return base_result and selected_lines
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):
self._shape_history._drawing_event.de_select_all()
for shape in self._shape_history.get_shapes():
if len(shape.get_centred_positions()) == 2:
if shape.get_centred_positions()[0] is start_cpos and\
shape.get_centred_positions()[1] is end_cpos:
self._shape_history._drawing_event.set_selected(shape)
def single_item_selection(self, tree_item):
CreateTaskBase.single_item_selection(self, tree_item)
if isinstance(tree_item, queue_item.SampleQueueItem) or \
isinstance(tree_item, queue_item.DataCollectionGroupQueueItem):
self._acquisition_parameters = copy.deepcopy(self._acquisition_parameters)
self._processing_parameters = copy.deepcopy(self._processing_parameters)
elif isinstance(tree_item, queue_item.DataCollectionQueueItem):
data_collection = tree_item.get_model()
if data_collection.experiment_type == EXPERIMENT_TYPE.HELICAL:
self.setDisabled(False)
self._path_template = data_collection.acquisitions[0].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
self.select_shape_with_cpos(start_cpos, end_cpos)
self._energy_scan_result = qmo.EnergyScanResult()
self._processing_parameters = data_collection.processing_parameters
self._energy_scan_result = data_collection.crystal.energy_scan_result
self._acq_widget.set_energies(self._energy_scan_result)
self._processing_widget.update_data_model(self._processing_parameters)
self._acq_widget.update_data_model(self._acquisition_parameters,
self._path_template)
else:
self.setDisabled(True)
else:
self.setDisabled(True)
if isinstance(tree_item, queue_item.SampleQueueItem) or \
isinstance(tree_item, queue_item.DataCollectionGroupQueueItem) or \
isinstance(tree_item, 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):
data_collections = []
selected_items = self.selected_items()
for item in selected_items:
shape = self._list_item_map[item]
snapshot = None
if isinstance(shape, shape_history.Line ):
if shape.get_qub_objects() is not None:
snapshot = self._shape_history.get_snapshot(shape.get_qub_objects())
else:
snapshot = self._shape_history.get_snapshot([])
# Acquisition for start position
start_acq = qmo.Acquisition()
start_acq.acquisition_parameters = \
copy.deepcopy(self._acquisition_parameters)
start_acq.acquisition_parameters.collect_agent = \
COLLECTION_ORIGIN.MXCUBE
start_acq.acquisition_parameters.\
centred_position = shape.start_cpos
start_acq.path_template = copy.deepcopy(self._path_template)
start_acq.acquisition_parameters.centred_position.\
snapshot_image = snapshot
start_acq.path_template.suffix = self._session_hwobj.suffix
if '<sample_name>' in start_acq.path_template.directory:
name = sample.get_name().replace(':', '-')
start_acq.path_template.directory = start_acq.path_template.directory.\
replace('<sample_name>', name)
start_acq.path_template.process_directory = start_acq.path_template.process_directory.\
replace('<sample_name>', name)
if '<acronym>-<name>' in start_acq.path_template.base_prefix:
start_acq.path_template.base_prefix = self.get_default_prefix(sample)
start_acq.path_template.run_numer = self._beamline_setup_hwobj.queue_model_hwobj.\
get_next_run_number(start_acq.path_template)
# Add another acquisition for the end position
end_acq = qmo.Acquisition()
end_acq.acquisition_parameters = \
copy.deepcopy(self._acquisition_parameters)
end_acq.acquisition_parameters.collect_agent = \
COLLECTION_ORIGIN.MXCUBE
end_acq.acquisition_parameters.\
centred_position = shape.end_cpos
end_acq.path_template = copy.deepcopy(self._path_template)
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 = qmo.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.experiment_type = EXPERIMENT_TYPE.HELICAL
data_collections.append(dc)
self._path_template.run_number += 1
return data_collections
class CreateDiscreteWidget(CreateTaskBase):
def __init__(self, parent=None, name=None, fl=0):
CreateTaskBase.__init__(self, parent, name, fl, 'Discrete')
if not name:
self.setName("create_discrete_widget")
#
# Data attributes
#
self.previous_energy = None
self.init_models()
#
# Layout
#
v_layout = qt.QVBoxLayout(self, 2, 5, "v_layout")
self._acq_gbox = qt.QVGroupBox('Acquisition', self, 'acq_gbox')
self._acq_widget = \
AcquisitionWidget(self._acq_gbox,
"acquisition_widget",
layout='vertical',
acq_params=self._acquisition_parameters,
path_template=self._path_template)
self._data_path_gbox = qt.QVGroupBox('Data location',
self, 'data_path_gbox')
self._data_path_widget = \
DataPathWidget(self._data_path_gbox,
'create_dc_path_widget',
data_model=self._path_template,
layout=DataPathWidgetVerticalLayout)
self._processing_gbox = qt.QVGroupBox('Processing', self,
'processing_gbox')
self._processing_widget = \
ProcessingWidget(self._processing_gbox,
data_model=self._processing_parameters)
v_layout.addWidget(self._acq_gbox)
v_layout.addWidget(self._data_path_gbox)
v_layout.addWidget(self._processing_gbox)
v_layout.addStretch()
self.connect(self._acq_widget, qt.PYSIGNAL('mad_energy_selected'),
self.mad_energy_selected)
self.connect(self._data_path_widget.data_path_widget_layout.prefix_ledit,
qt.SIGNAL("textChanged(const QString &)"),
self._prefix_ledit_change)
self.connect(self._data_path_widget.data_path_widget_layout.run_number_ledit,
qt.SIGNAL("textChanged(const QString &)"),
self._run_number_ledit_change)
self.connect(self._acq_widget,
qt.PYSIGNAL("path_template_changed"),
self.handle_path_conflict)
self.connect(self._data_path_widget,
qt.PYSIGNAL("path_template_changed"),
self.handle_path_conflict)
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()
try:
transmission = self._beamline_setup_hwobj.transmission_hwobj.getAttFactor()
transmission = round(float(transmission), 2)
except AttributeError:
transmission = 0
try:
resolution = self._beamline_setup_hwobj.resolution_hwobj.getPosition()
resolution = round(float(resolution), 2)
except AttributeError:
resolution = 0
try:
energy = self._beamline_setup_hwobj.\
energy_hwobj.getCurrentEnergy()
if energy:
energy = round(float(energy), 4)
else:
energy = round(float(-1), 4)
except AttributeError:
energy = 0
self._acquisition_parameters.resolution = resolution
self._acquisition_parameters.energy = energy
self._acquisition_parameters.transmission = transmission
else:
self._acquisition_parameters = queue_model_objects.\
AcquisitionParameters()
def set_tunable_energy(self, state):
self._acq_widget.set_tunable_energy(state)
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 mad_energy_selected(self, name, energy, state):
item = self._current_selected_items[0]
model = item.get_model()
if state:
self._path_template.mad_prefix = name
else:
self._path_template.mad_prefix = ''
run_number = self._beamline_setup_hwobj.queue_model_hwobj.\
get_next_run_number(self._path_template)
data_path_widget = self.get_data_path_widget()
data_path_widget.set_run_number(run_number)
data_path_widget.set_prefix(self._path_template.base_prefix)
if self.isEnabled():
if isinstance(item, queue_item.TaskQueueItem) and \
not isinstance(item, queue_item.DataCollectionGroupQueueItem):
model.set_name(self._path_template.get_prefix())
item.setText(0, model.get_name())
def single_item_selection(self, tree_item):
CreateTaskBase.single_item_selection(self, tree_item)
if isinstance(tree_item, queue_item.SampleQueueItem) or \
isinstance(tree_item, queue_item.DataCollectionGroupQueueItem):
self._acquisition_parameters = copy.deepcopy(self._acquisition_parameters)
self._processing_parameters = copy.deepcopy(self._processing_parameters)
self._acq_widget.disable_inverse_beam(False)
elif isinstance(tree_item, queue_item.DataCollectionQueueItem):
data_collection = tree_item.get_model()
if data_collection.experiment_type != queue_model_enumerables.\
EXPERIMENT_TYPE.HELICAL:
self.setDisabled(False)
self._acq_widget.disable_inverse_beam(True)
if data_collection.get_path_template():
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) == 1:
self.select_shape_with_cpos(self._acquisition_parameters.\
centred_position)
self._processing_parameters = data_collection.processing_parameters
else:
self.setDisabled(True)
else:
# Disable control
self.setDisabled(True)
if isinstance(tree_item, queue_item.SampleQueueItem) or \
isinstance(tree_item, queue_item.DataCollectionGroupQueueItem) or \
isinstance(tree_item, 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 approve_creation(self):
result = CreateTaskBase.approve_creation(self)
selected_shapes = self._shape_history.selected_shapes
for shape in selected_shapes:
if isinstance(shape, shape_history.Line):
result = False
return result
def get_subwedges(self, total_num_images, subwedge_size,
osc_range, osc_offset):
number_of_subwedges = total_num_images / subwedge_size
subwedges = []
for subwedge_num in range(0, number_of_subwedges):
osc_start = osc_offset + (osc_range * subwedge_size * subwedge_num)
subwedges.append((subwedge_num * subwedge_size + 1,
subwedge_size, osc_start))
return subwedges
# Called by the owning widget (task_toolbox_widget) to create
# a collection. When a data collection group is selected.
def _create_task(self, sample):
tasks = []
sample_is_mounted = self._beamline_setup_hwobj.sample_changer_hwobj.\
is_mounted_sample(sample)
if (not self._selected_positions) or (not sample_is_mounted):
# No centred positions selected, or selected sample not
# mounted create sample centring task.
sc = self.create_sample_centring(sample)
tasks.append(sc)
cpos = queue_model_objects.CentredPosition()
cpos.snapshot_image = self._shape_history.get_snapshot([])
cpos_list = [cpos]
else:
# Shapes selected and sample is mounted, get the
# centred positions for the shapes
selected_shapes = self._shape_history.selected_shapes
cpos_list = self.get_centred_positions(selected_shapes)
sc = None
if self._acq_widget.use_inverse_beam():
total_num_images = self._acquisition_parameters.num_images
subwedge_size = self._acq_widget.get_num_subwedges()
osc_range = self._acquisition_parameters.osc_range
osc_start = self._acquisition_parameters.osc_start
run_number = self._path_template.run_number
wedge_one = self.get_subwedges(total_num_images, subwedge_size,
osc_range, osc_start)
wedge_one = [pair + (run_number,) for pair in wedge_one]
wedge_two = self.get_subwedges(total_num_images,
subwedge_size, osc_range,
180 + osc_start)
wedge_two = [pair + (run_number + 1,) for pair in wedge_two]
subwedges = [sw_pair for pair in zip(wedge_one, wedge_two) for sw_pair in pair]
self._acq_widget.set_use_inverse_beam(False)
for cpos in cpos_list:
for sw in subwedges:
tasks.extend(self.create_dc(sample, sw[3], sw[0], sw[1],
sw[2], sc=sc, cpos=cpos, inverse_beam = True))
self._path_template.run_number += 1
else:
for cpos in cpos_list:
tasks.extend(self.create_dc(sample, sc=sc, cpos=cpos))
self._path_template.run_number += 1
return tasks
def create_sample_centring(self, sample):
sc = queue_model_objects.SampleCentring()
sc.set_name('sample-centring')
return sc
def get_centred_positions(self, shapes):
centred_positions = []
for shape in shapes:
snapshot = None
if isinstance(shape, shape_history.Point):
if shape.qub_point is not None:
snapshot = self._shape_history.\
get_snapshot([shape.qub_point])
else:
snapshot = self._shape_history.get_snapshot([])
centred_position = shape.get_centred_positions()[0]
centred_position.snapshot_image = snapshot
centred_positions.append(centred_position)
return centred_positions
def create_dc(self, sample, run_number=None, start_image=None,
num_images=None, osc_start=None, sc=None,
cpos=None, inverse_beam = False):
tasks = []
# Acquisition for start position
acq = queue_model_objects.Acquisition()
acq.acquisition_parameters = \
copy.deepcopy(self._acquisition_parameters)
acq.acquisition_parameters.collect_agent = \
queue_model_enumerables.COLLECTION_ORIGIN.MXCUBE
acq.path_template = copy.deepcopy(self._path_template)
acq.acquisition_parameters.centred_position = cpos
if '<sample_name>' in acq.path_template.directory:
name = sample.get_name().replace(':', '-')
acq.path_template.directory = acq.path_template.directory.\
replace('<sample_name>', name)
acq.path_template.process_directory = acq.path_template.process_directory.\
replace('<sample_name>', name)
if '<acronym>-<name>' in acq.path_template.base_prefix:
acq.path_template.base_prefix = self.get_default_prefix(sample)
acq.path_template.run_numer = self._beamline_setup_hwobj.queue_model_hwobj.\
get_next_run_number(acq.path_template)
if run_number:
acq.path_template.run_number = run_number
if start_image:
acq.acquisition_parameters.first_image = start_image
acq.path_template.start_num = start_image
if num_images:
acq.acquisition_parameters.num_images = num_images
acq.path_template.num_files = num_images
if osc_start:
acq.acquisition_parameters.osc_start = osc_start
if inverse_beam:
acq.acquisition_parameters.inverse_beam = False
processing_parameters = copy.deepcopy(self._processing_parameters)
dc = queue_model_objects.DataCollection([acq],
sample.crystals[0],
processing_parameters)
dc.set_name(acq.path_template.get_prefix())
dc.set_number(acq.path_template.run_number)
dc.experiment_type = queue_model_enumerables.EXPERIMENT_TYPE.NATIVE
if sc:
sc.add_task(dc)
tasks.append(dc)
self._data_path_widget.update_data_model(self._path_template)
self._acq_widget.update_data_model(self._acquisition_parameters,
self._path_template)
return tasks
class CreateHelicalWidget(CreateTaskBase):
def __init__(self, parent=None, name=None, fl=0):
CreateTaskBase.__init__(self, parent, name, fl, 'Helical')
if not name:
self.setName("create_helical_widget")
#
# Data attributes
#
self.init_models()
self._prev_pos = None
self._current_pos = None
self._list_item_map = {}
self.init_models()
#
# Layout
#
v_layout = qt.QVBoxLayout(self, 2, 5, "v_layout")
self._lines_gbox = qt.QGroupBox('Lines', self, "lines_gbox")
self._lines_gbox.setColumnLayout(0, qt.Qt.Vertical)
self._lines_gbox.layout().setSpacing(6)
self._lines_gbox.layout().setMargin(11)
lines_gbox_layout = qt.QHBoxLayout(self._lines_gbox.layout())
lines_gbox_layout.setAlignment(qt.Qt.AlignTop)
self._list_box = qt.QListBox(self._lines_gbox, "helical_page")
self._list_box.setSelectionMode(qt.QListBox.Extended)
self._list_box.setFixedWidth(200)
self._list_box.setFixedHeight(75)
list_box_tool_tip = "Select the line(s) to perfrom helical scan on"
qt.QToolTip.add(self._list_box, list_box_tool_tip)
lines_gbox_layout.addWidget(self._list_box)
button_layout = qt.QVBoxLayout(None, 0, 6, "button_layout")
button_layout.setSpacing(5)
add_button = qt.QPushButton("+", self._lines_gbox, "add_button")
add_button.setFixedWidth(20)
add_button.setFixedHeight(20)
remove_button = qt.QPushButton("-", self._lines_gbox, "add_button")
remove_button.setFixedWidth(20)
remove_button.setFixedHeight(20)
button_layout.addWidget(add_button)
button_layout.addWidget(remove_button)
lines_gbox_layout.addLayout(button_layout)
add_button_tool_tip = "Add a line between two saved positions, " \
"CTRL click to select more than one position"
qt.QToolTip.add(add_button, add_button_tool_tip)
remove_button_tool_tip = "Remove selected line(s)"
qt.QToolTip.add(remove_button, remove_button_tool_tip)
self._acq_gbox = qt.QVGroupBox('Acquisition', self, 'acq_gbox')
self._acq_widget = \
AcquisitionWidget(self._acq_gbox,
"acquisition_widget", layout='vertical',
acq_params=self._acquisition_parameters,
path_template=self._path_template)
self._acq_widget.disable_inverse_beam(True)
self._data_path_gbox = qt.QVGroupBox('Data location', self,
'data_path_gbox')
self._data_path_widget = \
DataPathWidget(self._data_path_gbox,
data_model = self._path_template,
layout = 'vertical')
self._processing_gbox = qt.QVGroupBox('Processing', self,
'processing_gbox')
self._processing_widget = \
ProcessingWidget(self._processing_gbox,
data_model = self._processing_parameters)
v_layout.addWidget(self._lines_gbox)
v_layout.addWidget(self._acq_gbox)
v_layout.addWidget(self._data_path_gbox)
v_layout.addWidget(self._processing_gbox)
qt.QObject.connect(add_button, qt.SIGNAL("clicked()"),
self.add_clicked)
qt.QObject.connect(remove_button, qt.SIGNAL("clicked()"),
self.remove_clicked)
qt.QObject.connect(self._list_box, qt.SIGNAL("selectionChanged()"),
self.list_box_selection_changed)
prefix_ledit = self._data_path_widget.\
data_path_widget_layout.child('prefix_ledit')
run_number_ledit = self._data_path_widget.\
data_path_widget_layout.child('run_number_ledit')
self.connect(prefix_ledit, qt.SIGNAL("textChanged(const QString &)"),
self._prefix_ledit_change)
self.connect(run_number_ledit,
qt.SIGNAL("textChanged(const QString &)"),
self._run_number_ledit_change)
self.connect(self._data_path_widget,
qt.PYSIGNAL("path_template_changed"),
self.handle_path_conflict)
def init_models(self):
CreateTaskBase.init_models(self)
self._energy_scan_result = qmo.EnergyScanResult()
self._processing_parameters = qmo.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")
else:
self._acquisition_parameters = qmo.AcquisitionParameters()
self._path_template = qmo.PathTemplate()
def add_clicked(self):
selected_shapes = self._shape_history.selected_shapes.values()
if len(selected_shapes) == 2:
p1 = selected_shapes[0]
p2 = selected_shapes[1]
line = shape_history.\
Line(self._shape_history.get_drawing(),
p1.qub_point, p2.qub_point,
p1.centred_position, p2.centred_position)
line.show()
self._shape_history.add_shape(line)
points_index = line.get_points_index()
if points_index:
display_name = "Line (points: %d, %d)" % \
(points_index[0], points_index[1])
else:
display_name = "Line (points: #, #)"
list_box_item = qt.QListBoxText(self._list_box, display_name)
#list_box_item = qt.QListBoxText(self._list_box, 'Line')
self._list_item_map[list_box_item] = line
# De select previous items
for item in self.selected_items():
self._list_box.setSelected(item, False)
self._list_box.setSelected(list_box_item, True)
def remove_clicked(self):
selected_items = self.selected_items()
if selected_items:
for item in selected_items:
self._list_box.removeItem(self._list_box.index(item))
line = self._list_item_map[item]
self._shape_history.delete_shape(line)
del self._list_item_map[item]
# Calback from shape_history, called when a shape is deleted
def shape_deleted(self, shape):
if isinstance(shape, shape_history.Point):
items_to_remove = []
for (list_item, line) in self._list_item_map.iteritems():
for qub_object in shape.get_qub_objects():
if qub_object in line.get_qub_objects():
items_to_remove.append((list_item, line))
for (list_item, line) in items_to_remove:
self._list_box.removeItem(self._list_box.index(list_item))
del self._list_item_map[list_item]
def centred_position_selection(self, positions):
if len(positions) == 1:
self._prev_pos = positions[0]
elif len(positions) == 2:
for pos in positions:
if pos != self._prev_pos:
self._current_pos = pos
else:
self._prev_pos = None
self._current_pos = None
self._list_box.clearSelection()
def list_box_selection_changed(self):
self.show_selected_lines()
def selected_items(self):
selected_items = []
for item_index in range(0, self._list_box.numRows()):
if self._list_box.isSelected(item_index):
selected_items.append(self._list_box.item(item_index))
return selected_items
def show_selected_lines(self):
selected_items = self.selected_items()
for list_item in self._list_item_map.keys():
line = self._list_item_map[list_item]
if list_item in selected_items:
self._shape_history.select_shape(line)
else:
self._shape_history.de_select_shape(line)
def approve_creation(self):
base_result = CreateTaskBase.approve_creation(self)
selected_lines = False
if self.selected_items():
selected_lines = True
else:
logging.getLogger("user_level_log").\
warning("No lines selected, please select one or more lines.")
return base_result and selected_lines
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):
self._shape_history.de_select_all()
selected_line = None
for shape in self._shape_history.get_shapes():
if isinstance(shape, shape_history.Line):
if shape.get_centred_positions()[0] == start_cpos and\
shape.get_centred_positions()[1] == end_cpos:
self._shape_history.select_shape(shape)
selected_line = shape
#de-select previous selected list items and
#select the current shape (Line).
for (list_item, shape) in self._list_item_map.iteritems():
if selected_line is shape:
self._list_box.setSelected(list_item, True)
else:
self._list_box.setSelected(list_item, False)
def single_item_selection(self, tree_item):
CreateTaskBase.single_item_selection(self, tree_item)
if isinstance(tree_item, 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, queue_item.BasketQueueItem):
self.setDisabled(False)
elif isinstance(tree_item, queue_item.DataCollectionQueueItem):
data_collection = tree_item.get_model()
if data_collection.experiment_type == EXPERIMENT_TYPE.HELICAL:
if tree_item.get_model().is_executed():
self.setDisabled(True)
else:
self.setDisabled(False)
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
self.select_shape_with_cpos(start_cpos, end_cpos)
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, queue_item.SampleQueueItem) or \
isinstance(tree_item, queue_item.DataCollectionGroupQueueItem) or \
isinstance(tree_item, 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 = []
if isinstance(shape, shape_history.Line):
if shape.get_qub_objects() is not None:
snapshot = self._shape_history.get_snapshot(
shape.get_qub_objects())
else:
snapshot = self._shape_history.get_snapshot([])
# Acquisition for start position
start_acq = self._create_acq(sample)
start_acq.acquisition_parameters.\
centred_position = copy.deepcopy(shape.start_cpos)
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 = shape.end_cpos
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 = qmo.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.experiment_type = EXPERIMENT_TYPE.HELICAL
data_collections.append(dc)
self._path_template.run_number += 1
return data_collections
class CreateDiscreteWidget(CreateTaskBase):
def __init__(self, parent=None, name=None, fl=0):
CreateTaskBase.__init__(self, parent, name, fl, 'Discrete')
if not name:
self.setName("create_discrete_widget")
self.previous_energy = None
self.init_models()
#
# Layout
#
v_layout = qt.QVBoxLayout(self, 2, 5, "v_layout")
self._acq_gbox = qt.QVGroupBox('Acquisition', self, 'acq_gbox')
self._acq_widget = \
AcquisitionWidget(self._acq_gbox,
"acquisition_widget",
layout='vertical',
acq_params=self._acquisition_parameters,
path_template=self._path_template)
self._data_path_gbox = qt.QVGroupBox('Data location',
self, 'data_path_gbox')
self._data_path_widget = \
DataPathWidget(self._data_path_gbox,
'create_dc_path_widget',
data_model=self._path_template,
layout=DataPathWidgetVerticalLayout)
self._processing_gbox = qt.QVGroupBox('Processing', self,
'processing_gbox')
self._processing_widget = \
ProcessingWidget(self._processing_gbox,
data_model=self._processing_parameters)
v_layout.addWidget(self._acq_gbox)
v_layout.addWidget(self._data_path_gbox)
v_layout.addWidget(self._processing_gbox)
v_layout.addStretch()
dp_layout = self._data_path_widget.data_path_widget_layout
self.connect(self._acq_widget, qt.PYSIGNAL('mad_energy_selected'),
self.mad_energy_selected)
self.connect(dp_layout.prefix_ledit,
qt.SIGNAL("textChanged(const QString &)"),
self._prefix_ledit_change)
self.connect(dp_layout.run_number_ledit,
qt.SIGNAL("textChanged(const QString &)"),
self._run_number_ledit_change)
self.connect(self._acq_widget,
qt.PYSIGNAL("path_template_changed"),
self.handle_path_conflict)
self.connect(self._data_path_widget,
qt.PYSIGNAL("path_template_changed"),
self.handle_path_conflict)
def init_models(self):
CreateTaskBase.init_models(self)
self._energy_scan_result = qmo.EnergyScanResult()
self._processing_parameters = qmo.ProcessingParameters()
def set_tunable_energy(self, state):
self._acq_widget.set_tunable_energy(state)
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 mad_energy_selected(self, name, energy, state):
item = self._current_selected_items[0]
model = item.get_model()
if state:
self._path_template.mad_prefix = name
else:
self._path_template.mad_prefix = ''
run_number = self._beamline_setup_hwobj.queue_model_hwobj.\
get_next_run_number(self._path_template)
data_path_widget = self.get_data_path_widget()
data_path_widget.set_run_number(run_number)
data_path_widget.set_prefix(self._path_template.base_prefix)
if self.isEnabled():
if isinstance(item, queue_item.TaskQueueItem) and \
not isinstance(item, queue_item.DataCollectionGroupQueueItem):
model.set_name(self._path_template.get_prefix())
item.setText(0, model.get_name())
def single_item_selection(self, tree_item):
CreateTaskBase.single_item_selection(self, tree_item)
if isinstance(tree_item, queue_item.SampleQueueItem) or \
isinstance(tree_item, queue_item.DataCollectionGroupQueueItem):
self._processing_parameters = copy.deepcopy(self._processing_parameters)
self._processing_widget.update_data_model(self._processing_parameters)
self._acq_widget.disable_inverse_beam(False)
elif isinstance(tree_item, queue_item.DataCollectionQueueItem):
dc = tree_item.get_model()
if dc.experiment_type != queue_model_enumerables.EXPERIMENT_TYPE.HELICAL:
self.setDisabled(False)
self._acq_widget.disable_inverse_beam(True)
self._path_template = dc.get_path_template()
self._data_path_widget.update_data_model(self._path_template)
self._acquisition_parameters = dc.acquisitions[0].acquisition_parameters
self._acq_widget.update_data_model(self._acquisition_parameters,
self._path_template)
if len(dc.acquisitions) == 1:
self.select_shape_with_cpos(self._acquisition_parameters.\
centred_position)
self._processing_parameters = dc.processing_parameters
self._processing_widget.update_data_model(self._processing_parameters)
else:
self.setDisabled(True)
else:
self.setDisabled(True)
def approve_creation(self):
result = CreateTaskBase.approve_creation(self)
selected_shapes = self._shape_history.selected_shapes
for shape in selected_shapes:
if isinstance(shape, shape_history.Line):
result = False
return result
# Called by the owning widget (task_toolbox_widget) to create
# a collection. When a data collection group is selected.
def _create_task(self, sample, shape):
tasks = []
if not shape:
cpos = qmo.CentredPosition()
cpos.snapshot_image = self._shape_history.get_snapshot([])
else:
# Shapes selected and sample is mounted, get the
# centred positions for the shapes
if isinstance(shape, shape_history.Point):
snapshot = self._shape_history.\
get_snapshot([shape.qub_point])
cpos = shape.get_centred_positions()[0]
cpos.snapshot_image = snapshot
if self._acq_widget.use_inverse_beam():
total_num_images = self._acquisition_parameters.num_images
subwedge_size = self._acq_widget.get_num_subwedges()
osc_range = self._acquisition_parameters.osc_range
osc_start = self._acquisition_parameters.osc_start
run_number = self._path_template.run_number
subwedges = qmo.create_inverse_beam_sw(total_num_images,
subwedge_size, osc_range, osc_start, run_number)
self._acq_widget.set_use_inverse_beam(False)
for sw in subwedges:
tasks.extend(self.create_dc(sample, sw[3], sw[0], sw[1],
sw[2], cpos=cpos,
inverse_beam = True))
self._path_template.run_number += 1
else:
tasks.extend(self.create_dc(sample, cpos=cpos))
self._path_template.run_number += 1
return tasks
def create_dc(self, sample, run_number=None, start_image=None,
num_images=None, osc_start=None, sc=None,
cpos=None, inverse_beam = False):
tasks = []
# Acquisition for start position
acq = qmo.Acquisition()
acq.acquisition_parameters = \
copy.deepcopy(self._acquisition_parameters)
acq.acquisition_parameters.collect_agent = \
queue_model_enumerables.COLLECTION_ORIGIN.MXCUBE
acq.path_template = copy.deepcopy(self._path_template)
acq.acquisition_parameters.centred_position = cpos
if '<sample_name>' in acq.path_template.directory:
name = sample.get_name().replace(':', '-')
acq.path_template.directory = acq.path_template.directory.\
replace('<sample_name>', name)
acq.path_template.process_directory = acq.path_template.process_directory.\
replace('<sample_name>', name)
if '<acronym>-<name>' in acq.path_template.base_prefix:
acq.path_template.base_prefix = self.get_default_prefix(sample)
acq.path_template.run_numer = self._beamline_setup_hwobj.queue_model_hwobj.\
get_next_run_number(acq.path_template)
if run_number:
acq.path_template.run_number = run_number
if start_image:
acq.acquisition_parameters.first_image = start_image
acq.path_template.start_num = start_image
if num_images:
acq.acquisition_parameters.num_images = num_images
acq.path_template.num_files = num_images
if osc_start:
acq.acquisition_parameters.osc_start = osc_start
if inverse_beam:
acq.acquisition_parameters.inverse_beam = False
processing_parameters = copy.deepcopy(self._processing_parameters)
dc = qmo.DataCollection([acq], sample.crystals[0],
processing_parameters)
dc.set_name(acq.path_template.get_prefix())
dc.set_number(acq.path_template.run_number)
dc.experiment_type = queue_model_enumerables.EXPERIMENT_TYPE.NATIVE
tasks.append(dc)
self._data_path_widget.update_data_model(self._path_template)
self._acq_widget.update_data_model(self._acquisition_parameters,
self._path_template)
return tasks
