class AdvancedParametersWidget(QtImport.QWidget):
def __init__(self, parent=None, name="advanced_parameters_widget"):
QtImport.QWidget.__init__(self, parent)
# Internal values -----------------------------------------------------
self._data_collection = None
self._tree_view_item = None
self._acquisition_mib = None
# Properties ----------------------------------------------------------
# Signals -------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Graphic elements ----------------------------------------------------
_dc_parameters_widget = QtImport.QWidget(self)
self._data_path_widget = DataPathWidget(_dc_parameters_widget)
self._acq_widget = AcquisitionWidget(_dc_parameters_widget, layout="horizontal")
# Layout --------------------------------------------------------------
_dc_parameters_widget_layout = QtImport.QVBoxLayout(_dc_parameters_widget)
_dc_parameters_widget_layout.addWidget(self._data_path_widget)
_dc_parameters_widget_layout.addWidget(self._acq_widget)
_dc_parameters_widget_layout.setSpacing(2)
_dc_parameters_widget_layout.addStretch(10)
_dc_parameters_widget_layout.setContentsMargins(0, 0, 0, 0)
_main_hlayout = QtImport.QHBoxLayout(self)
_main_hlayout.addWidget(_dc_parameters_widget)
_main_hlayout.setSpacing(2)
_main_hlayout.setContentsMargins(2, 2, 2, 2)
_main_hlayout.addStretch(0)
# Qt signal/slot connections ------------------------------------------
# self._acq_widget.acqParametersChangedSignal.\
# connect(self.acq_parameters_changed)
# self._data_path_widget.pathTemplateChangedSignal.\
# connect(self.acq_parameters_changed)
self._acq_widget.madEnergySelectedSignal.connect(self.mad_energy_selected)
# Ohter ---------------------------------------------------------------
self._acq_widget.use_osc_start(False)
self._acq_widget.acq_widget_layout.mad_cbox.hide()
self._acq_widget.acq_widget_layout.energies_combo.hide()
self._acq_widget.acq_widget_layout.shutterless_cbx.hide()
def init_api(self):
self._acq_widget.init_api()
def mad_energy_selected(self, name, energy, state):
path_template = self._data_collection.acquisitions[0].path_template
if state:
path_template.mad_prefix = name
else:
path_template.mad_prefix = ""
run_number = api.queue_model.get_next_run_number(
path_template
)
self._data_path_widget.set_run_number(run_number)
self._data_path_widget.set_prefix(path_template.base_prefix)
model = self._tree_view_item.get_model()
model.set_name(path_template.get_prefix())
self._tree_view_item.setText(0, model.get_name())
def tab_changed(self):
if self._tree_view_item:
self.populate_widget(self._tree_view_item, None)
def populate_widget(self, tree_view_item, data_collection):
self._tree_view_item = tree_view_item
self._data_collection = data_collection
# if isinstance(tree_view_item, queue_item.XrayCenteringQueueItem):
# self._data_collection = tree_view_item.get_model().reference_image_collection
# else:
# self._data_collection = tree_view_item.get_model()
executed = self._data_collection.is_executed()
self._acq_widget.setEnabled(not executed)
self._data_path_widget.setEnabled(not executed)
self._acquisition_mib = DataModelInputBinder(
self._data_collection.acquisitions[0].acquisition_parameters
)
# The acq_widget sends a signal to the path_widget, and it relies
# on that both models upto date, we need to refactor this part
# so that both models are set before taking ceratin actions.
# This workaround, works for the time beeing.
self._data_path_widget._data_model = self._data_collection.acquisitions[
0
].path_template
self._data_path_widget.update_data_model(
self._data_collection.acquisitions[0].path_template
)
self._acq_widget.update_data_model(
self._data_collection.acquisitions[0].acquisition_parameters,
self._data_collection.acquisitions[0].path_template,
)
# self._acq_widget.use_osc_start(False)
self._acq_widget.acq_widget_layout.num_images_ledit.setDisabled(
data_collection.is_mesh()
)
invalid = self._acquisition_mib.validate_all()
if invalid:
msg = (
"This data collection has one or more incorrect parameters,"
+ " correct the fields marked in red to solve the problem."
)
logging.getLogger("GUI").warning(msg)
class DCParametersWidget(QtImport.QWidget):
def __init__(self, parent=None, name="parameter_widget"):
QtImport.QWidget.__init__(self, parent)
if name is not None:
self.setObjectName(name)
# Properties ----------------------------------------------------------
# Signals ------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Internal variables --------------------------------------------------
self._data_collection = None
self._tree_view_item = None
self._acquisition_mib = None
# Graphic elements ----------------------------------------------------
_dc_parameters_widget = QtImport.QWidget(self)
self._data_path_widget = DataPathWidget(_dc_parameters_widget)
self._acq_widget = AcquisitionWidget(_dc_parameters_widget,
layout="horizontal")
self._processing_widget = ProcessingWidget(_dc_parameters_widget)
# Layout --------------------------------------------------------------
_dc_parameters_widget_layout = QtImport.QVBoxLayout(
_dc_parameters_widget)
_dc_parameters_widget_layout.addWidget(self._data_path_widget)
_dc_parameters_widget_layout.addWidget(self._acq_widget)
_dc_parameters_widget_layout.addWidget(self._processing_widget)
_dc_parameters_widget_layout.setContentsMargins(0, 0, 0, 0)
_dc_parameters_widget_layout.setSpacing(2)
_dc_parameters_widget_layout.addStretch(10)
_main_hlayout = QtImport.QHBoxLayout(self)
_main_hlayout.addWidget(_dc_parameters_widget)
_main_hlayout.setContentsMargins(0, 0, 0, 0)
_main_hlayout.setSpacing(2)
_main_hlayout.addStretch(0)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self._data_path_widget.data_path_layout.prefix_ledit.textChanged.connect(
self._prefix_ledit_change)
self._data_path_widget.data_path_layout.run_number_ledit.textChanged.connect(
self._run_number_ledit_change)
self._acq_widget.madEnergySelectedSignal.connect(
self.mad_energy_selected)
self._acq_widget.acqParametersChangedSignal.connect(
self.acq_parameters_changed)
# Other ---------------------------------------------------------------
def _prefix_ledit_change(self, new_value):
prefix = self._data_collection.acquisitions[
0].path_template.get_prefix()
self._data_collection.set_name(prefix)
self._tree_view_item.setText(0, self._data_collection.get_name())
def _run_number_ledit_change(self, new_value):
if str(new_value).isdigit():
self._data_collection.set_number(int(new_value))
self._tree_view_item.setText(0, self._data_collection.get_name())
def acq_parameters_changed(self):
if self._tree_view_item is None:
return
# TODO get tree view in another way
dc_tree_widget = self._tree_view_item.listView().parent().parent()
dc_tree_widget.check_for_path_collisions()
path_template = self._data_collection.acquisitions[0].path_template
HWR.beamline.queue_model.check_for_path_collisions(path_template)
def mad_energy_selected(self, name, energy, state):
path_template = self._data_collection.acquisitions[0].path_template
if state:
path_template.mad_prefix = str(name)
else:
path_template.mad_prefix = ""
run_number = HWR.beamline.queue_model.get_next_run_number(
path_template)
self._data_path_widget.set_run_number(run_number)
self._data_path_widget.set_prefix(path_template.base_prefix)
model = self._tree_view_item.get_model()
model.set_name(path_template.get_prefix())
self._tree_view_item.setText(0, model.get_name())
def set_enabled(self, state):
self._acq_widget.setEnabled(state)
self._data_path_widget.setEnabled(state)
self._processing_widget.setEnabled(state)
def populate_widget(self, item):
data_collection = item.get_model()
self._tree_view_item = item
self._data_collection = data_collection
self._acquisition_mib = DataModelInputBinder(
self._data_collection.acquisitions[0].acquisition_parameters)
# The acq_widget sends a signal to the path_widget, and it relies
# on that both models upto date, we need to refactor this part
# so that both models are set before taking ceratin actions.
# This workaround, works for the time beeing.
self._data_path_widget._data_model = data_collection.acquisitions[
0].path_template
self._acq_widget.set_energies(
data_collection.crystal.energy_scan_result)
self._acq_widget.update_data_model(
data_collection.acquisitions[0].acquisition_parameters,
data_collection.acquisitions[0].path_template,
)
self._data_path_widget.update_data_model(
data_collection.acquisitions[0].path_template)
self._processing_widget.update_data_model(
data_collection.processing_parameters)
invalid = self._acquisition_mib.validate_all()
if invalid:
msg = (
"This data collection has one or more incorrect parameters," +
" correct the fields marked in red to solve the problem.")
logging.getLogger("GUI").warning(msg)
class DCParametersWidget(QtImport.QWidget):
def __init__(self, parent=None, name="parameter_widget"):
QtImport.QWidget.__init__(self, parent)
if name is not None:
self.setObjectName(name)
# Properties ----------------------------------------------------------
# Signals ------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Internal variables --------------------------------------------------
self._data_collection = None
self._tree_view_item = None
self._acquisition_mib = None
# Graphic elements ----------------------------------------------------
_dc_parameters_widget = QtImport.QWidget(self)
self._data_path_widget = DataPathWidget(_dc_parameters_widget)
self._acq_widget = AcquisitionWidget(_dc_parameters_widget, layout="horizontal")
self._processing_widget = ProcessingWidget(_dc_parameters_widget)
# Layout --------------------------------------------------------------
_dc_parameters_widget_layout = QtImport.QVBoxLayout(_dc_parameters_widget)
_dc_parameters_widget_layout.addWidget(self._data_path_widget)
_dc_parameters_widget_layout.addWidget(self._acq_widget)
_dc_parameters_widget_layout.addWidget(self._processing_widget)
_dc_parameters_widget_layout.setContentsMargins(0, 0, 0, 0)
_dc_parameters_widget_layout.setSpacing(2)
_dc_parameters_widget_layout.addStretch(10)
_main_hlayout = QtImport.QHBoxLayout(self)
_main_hlayout.addWidget(_dc_parameters_widget)
_main_hlayout.setContentsMargins(0, 0, 0, 0)
_main_hlayout.setSpacing(2)
_main_hlayout.addStretch(0)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self._data_path_widget.data_path_layout.prefix_ledit.textChanged.connect(
self._prefix_ledit_change
)
self._data_path_widget.data_path_layout.run_number_ledit.textChanged.connect(
self._run_number_ledit_change
)
self._acq_widget.madEnergySelectedSignal.connect(self.mad_energy_selected)
self._acq_widget.acqParametersChangedSignal.connect(self.acq_parameters_changed)
# Other ---------------------------------------------------------------
def _prefix_ledit_change(self, new_value):
prefix = self._data_collection.acquisitions[0].path_template.get_prefix()
self._data_collection.set_name(prefix)
self._tree_view_item.setText(0, self._data_collection.get_name())
def _run_number_ledit_change(self, new_value):
if str(new_value).isdigit():
self._data_collection.set_number(int(new_value))
self._tree_view_item.setText(0, self._data_collection.get_name())
def acq_parameters_changed(self):
if self._tree_view_item is None:
# TODO fix this
return
# TODO get tree view in another way
dc_tree_widget = self._tree_view_item.listView().parent().parent()
dc_tree_widget.check_for_path_collisions()
path_template = self._data_collection.acquisitions[0].path_template
path_conflict = api.queue_model.check_for_path_collisions(path_template)
def mad_energy_selected(self, name, energy, state):
path_template = self._data_collection.acquisitions[0].path_template
if state:
path_template.mad_prefix = str(name)
else:
path_template.mad_prefix = ""
run_number = api.queue_model.get_next_run_number(
path_template
)
self._data_path_widget.set_run_number(run_number)
self._data_path_widget.set_prefix(path_template.base_prefix)
model = self._tree_view_item.get_model()
model.set_name(path_template.get_prefix())
self._tree_view_item.setText(0, model.get_name())
def tab_changed(self):
if self._tree_view_item:
self.populate_parameter_widget(self._tree_view_item)
def set_enabled(self, state):
self._acq_widget.setEnabled(state)
self._data_path_widget.setEnabled(state)
self._processing_widget.setEnabled(state)
def populate_widget(self, item):
data_collection = item.get_model()
self._tree_view_item = item
self._data_collection = data_collection
self._acquisition_mib = DataModelInputBinder(
self._data_collection.acquisitions[0].acquisition_parameters
)
# The acq_widget sends a signal to the path_widget, and it relies
# on that both models upto date, we need to refactor this part
# so that both models are set before taking ceratin actions.
# This workaround, works for the time beeing.
self._data_path_widget._data_model = data_collection.acquisitions[
0
].path_template
self._acq_widget.set_energies(data_collection.crystal.energy_scan_result)
self._acq_widget.update_data_model(
data_collection.acquisitions[0].acquisition_parameters,
data_collection.acquisitions[0].path_template,
)
self._data_path_widget.update_data_model(
data_collection.acquisitions[0].path_template
)
self._processing_widget.update_data_model(data_collection.processing_parameters)
invalid = self._acquisition_mib.validate_all()
if invalid:
msg = (
"This data collection has one or more incorrect parameters,"
+ " correct the fields marked in red to solve the problem."
)
logging.getLogger("GUI").warning(msg)
class AdvancedParametersWidget(QtImport.QWidget):
def __init__(self, parent=None, name="advanced_parameters_widget"):
QtImport.QWidget.__init__(self, parent)
# Internal values -----------------------------------------------------
self._data_collection = None
self._tree_view_item = None
self._acquisition_mib = None
# Properties ----------------------------------------------------------
# Signals -------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Graphic elements ----------------------------------------------------
_dc_parameters_widget = QtImport.QWidget(self)
self._data_path_widget = DataPathWidget(_dc_parameters_widget)
self._acq_widget = AcquisitionWidget(_dc_parameters_widget,
layout="horizontal")
# Layout --------------------------------------------------------------
_dc_parameters_widget_layout = QtImport.QVBoxLayout(
_dc_parameters_widget)
_dc_parameters_widget_layout.addWidget(self._data_path_widget)
_dc_parameters_widget_layout.addWidget(self._acq_widget)
_dc_parameters_widget_layout.setSpacing(2)
_dc_parameters_widget_layout.addStretch(10)
_dc_parameters_widget_layout.setContentsMargins(0, 0, 0, 0)
_main_hlayout = QtImport.QHBoxLayout(self)
_main_hlayout.addWidget(_dc_parameters_widget)
_main_hlayout.setSpacing(2)
_main_hlayout.setContentsMargins(2, 2, 2, 2)
_main_hlayout.addStretch(0)
# Qt signal/slot connections ------------------------------------------
# self._acq_widget.acqParametersChangedSignal.\
# connect(self.acq_parameters_changed)
# self._data_path_widget.pathTemplateChangedSignal.\
# connect(self.acq_parameters_changed)
self._acq_widget.madEnergySelectedSignal.connect(
self.mad_energy_selected)
# Ohter ---------------------------------------------------------------
self._acq_widget.use_osc_start(False)
self._acq_widget.acq_widget_layout.mad_cbox.hide()
self._acq_widget.acq_widget_layout.energies_combo.hide()
self._acq_widget.acq_widget_layout.shutterless_cbx.hide()
def mad_energy_selected(self, name, energy, state):
path_template = self._data_collection.acquisitions[0].path_template
if state:
path_template.mad_prefix = name
else:
path_template.mad_prefix = ""
run_number = HWR.beamline.queue_model.get_next_run_number(
path_template)
self._data_path_widget.set_run_number(run_number)
self._data_path_widget.set_prefix(path_template.base_prefix)
model = self._tree_view_item.get_model()
model.set_name(path_template.get_prefix())
self._tree_view_item.setText(0, model.get_name())
def populate_widget(self, tree_view_item, data_collection):
self._tree_view_item = tree_view_item
self._data_collection = data_collection
# if isinstance(tree_view_item, queue_item.XrayCenteringQueueItem):
# self._data_collection = tree_view_item.get_model().reference_image_collection
# else:
# self._data_collection = tree_view_item.get_model()
executed = self._data_collection.is_executed()
self._acq_widget.setEnabled(not executed)
self._data_path_widget.setEnabled(not executed)
self._acquisition_mib = DataModelInputBinder(
self._data_collection.acquisitions[0].acquisition_parameters)
# The acq_widget sends a signal to the path_widget, and it relies
# on that both models upto date, we need to refactor this part
# so that both models are set before taking ceratin actions.
# This workaround, works for the time beeing.
self._data_path_widget.update_data_model(
self._data_collection.acquisitions[0].path_template)
self._acq_widget.update_data_model(
self._data_collection.acquisitions[0].acquisition_parameters,
self._data_collection.acquisitions[0].path_template,
)
# self._acq_widget.use_osc_start(False)
self._acq_widget.acq_widget_layout.num_images_ledit.setDisabled(
data_collection.is_mesh())
invalid = self._acquisition_mib.validate_all()
if invalid:
msg = (
"This data collection has one or more incorrect parameters," +
" correct the fields marked in red to solve the problem.")
logging.getLogger("GUI").warning(msg)