class DCParametersWidget(qt.QWidget):
def __init__(self, parent=None, name="parameter_widget"):
qt.QWidget.__init__(self, parent, name)
self._data_collection = None
self.add_dc_cb = None
self._tree_view_item = None
self.queue_model = None
self._beamline_setup_hwobj = None
self.caution_pixmap = Icons.load("Caution2.png")
self.path_widget = DataPathWidget(self, 'dc_params_path_widget')
self.acq_gbox = qt.QVGroupBox("Acquisition", self)
self.acq_gbox.setInsideMargin(2)
self.acq_widget = AcquisitionWidget(self.acq_gbox, layout='horizontal')
self.acq_widget.setFixedHeight(170)
widget_ui = os.path.join(os.path.dirname(__file__),
'ui_files/snapshot_widget_layout.ui')
widget = qtui.QWidgetFactory.create(widget_ui)
widget.reparent(self, qt.QPoint(0, 0))
self.position_widget = widget
self._processing_gbox = qt.QVGroupBox('Processing', self,
'processing_gbox')
self.processing_widget = ProcessingWidget(self._processing_gbox)
v_layout = qt.QVBoxLayout(self, 11, 10, "main_layout")
rone_hlayout = qt.QHBoxLayout(v_layout, 10, "rone")
rone_vlayout = qt.QVBoxLayout(rone_hlayout)
rone_sv_layout = qt.QVBoxLayout(rone_hlayout)
rone_vlayout.addWidget(self.path_widget)
rone_vlayout.addWidget(self.acq_gbox)
rtwo_hlayout = qt.QHBoxLayout(rone_vlayout, 10, "rtwo")
rone_vlayout.addStretch(10)
rone_sv_layout.addWidget(self.position_widget)
rone_sv_layout.addStretch(10)
rone_hlayout.addStretch()
rtwo_hlayout.addWidget(self._processing_gbox)
rtwo_hlayout.addStretch(10)
v_layout.addStretch()
self.connect(self.acq_widget, qt.PYSIGNAL('mad_energy_selected'),
self.mad_energy_selected)
self.connect(
self.path_widget.data_path_widget_layout.child('prefix_ledit'),
qt.SIGNAL("textChanged(const QString &)"),
self._prefix_ledit_change)
self.connect(
self.path_widget.data_path_widget_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.path_widget, qt.PYSIGNAL("path_template_changed"),
self.handle_path_conflict)
qt.QObject.connect(qt.qApp, qt.PYSIGNAL('tab_changed'),
self.tab_changed)
def set_beamline_setup(self, bl_setup):
self.acq_widget.set_beamline_setup(bl_setup)
self._beamline_setup_hwobj = bl_setup
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 handle_path_conflict(self, widget, new_value):
dc_tree_widget = self._tree_view_item.listView().parent()
dc_tree_widget.check_for_path_collisions()
path_template = self._data_collection.acquisitions[0].path_template
path_conflict = self.queue_model_hwobj.\
check_for_path_collisions(path_template)
if new_value != '':
if path_conflict:
logging.getLogger("user_level_log").\
error('The current path settings will overwrite data' +\
' from another task. Correct the problem before collecting')
widget.setPaletteBackgroundColor(widget_colors.LIGHT_RED)
else:
widget.setPaletteBackgroundColor(widget_colors.WHITE)
def __add_data_collection(self):
return self.add_dc_cb(self._data_collection, self.collection_type)
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 = self._beamline_setup_hwobj.queue_model_hwobj.\
get_next_run_number(path_template)
self.path_widget.set_run_number(run_number)
self.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_gbox.setEnabled(state)
self.path_widget.setEnabled(state)
self._processing_gbox.setEnabled(state)
def populate_parameter_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.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.acq_widget.use_osc_start(True)
self.path_widget.update_data_model(data_collection.\
acquisitions[0].path_template)
self.processing_widget.update_data_model(data_collection.\
processing_parameters)
if data_collection.acquisitions[0].acquisition_parameters.\
centred_position.snapshot_image:
image = data_collection.acquisitions[0].\
acquisition_parameters.centred_position.snapshot_image
image = image.scale(427, 320)
self.position_widget.child("svideo").setPixmap(qt.QPixmap(image))
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("user_level_log").\
warning(msg)
class DCParametersWidget(qt.QWidget):
def __init__(self, parent=None, name="parameter_widget"):
qt.QWidget.__init__(self, parent, name)
self._data_collection = None
self.add_dc_cb = None
self._tree_view_item = None
self.queue_model = None
self._beamline_setup_hwobj = None
self.caution_pixmap = Icons.load("Caution2.png")
self.path_widget = DataPathWidget(self, "dc_params_path_widget")
self.acq_gbox = qt.QVGroupBox("Acquisition", self)
self.acq_gbox.setInsideMargin(2)
self.acq_widget = AcquisitionWidget(self.acq_gbox, layout="horizontal")
self.acq_widget.setFixedHeight(170)
widget_ui = os.path.join(
os.path.dirname(__file__), "ui_files/snapshot_widget_layout.ui"
)
widget = qtui.QWidgetFactory.create(widget_ui)
widget.reparent(self, qt.QPoint(0, 0))
self.position_widget = widget
self._processing_gbox = qt.QVGroupBox("Processing", self, "processing_gbox")
self.processing_widget = ProcessingWidget(self._processing_gbox)
v_layout = qt.QVBoxLayout(self, 11, 10, "main_layout")
rone_hlayout = qt.QHBoxLayout(v_layout, 10, "rone")
rone_vlayout = qt.QVBoxLayout(rone_hlayout)
rone_sv_layout = qt.QVBoxLayout(rone_hlayout)
rone_vlayout.addWidget(self.path_widget)
rone_vlayout.addWidget(self.acq_gbox)
rtwo_hlayout = qt.QHBoxLayout(rone_vlayout, 10, "rtwo")
rone_vlayout.addStretch(10)
rone_sv_layout.addWidget(self.position_widget)
rone_sv_layout.addStretch(10)
rone_hlayout.addStretch()
rtwo_hlayout.addWidget(self._processing_gbox)
rtwo_hlayout.addStretch(10)
v_layout.addStretch()
self.connect(
self.acq_widget,
qt.PYSIGNAL("mad_energy_selected"),
self.mad_energy_selected,
)
self.connect(
self.path_widget.data_path_widget_layout.child("prefix_ledit"),
qt.SIGNAL("textChanged(const QString &)"),
self._prefix_ledit_change,
)
self.connect(
self.path_widget.data_path_widget_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.path_widget,
qt.PYSIGNAL("path_template_changed"),
self.handle_path_conflict,
)
qt.QObject.connect(qt.qApp, qt.PYSIGNAL("tab_changed"), self.tab_changed)
def set_beamline_setup(self, bl_setup):
self.acq_widget.set_beamline_setup(bl_setup)
self._beamline_setup_hwobj = bl_setup
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 handle_path_conflict(self, widget, new_value):
dc_tree_widget = self._tree_view_item.listView().parent()
dc_tree_widget.check_for_path_collisions()
path_template = self._data_collection.acquisitions[0].path_template
path_conflict = self.queue_model_hwobj.check_for_path_collisions(path_template)
if new_value != "":
if path_conflict:
logging.getLogger("user_level_log").error(
"The current path settings will overwrite data"
+ " from another task. Correct the problem before collecting"
)
widget.setPaletteBackgroundColor(widget_colors.LIGHT_RED)
else:
widget.setPaletteBackgroundColor(widget_colors.WHITE)
def __add_data_collection(self):
return self.add_dc_cb(self._data_collection, self.collection_type)
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 = self._beamline_setup_hwobj.queue_model_hwobj.get_next_run_number(
path_template
)
self.path_widget.set_run_number(run_number)
self.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_gbox.setEnabled(state)
self.path_widget.setEnabled(state)
self._processing_gbox.setEnabled(state)
def populate_parameter_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.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.acq_widget.use_osc_start(True)
self.path_widget.update_data_model(
data_collection.acquisitions[0].path_template
)
self.processing_widget.update_data_model(data_collection.processing_parameters)
if data_collection.acquisitions[
0
].acquisition_parameters.centred_position.snapshot_image:
image = data_collection.acquisitions[
0
].acquisition_parameters.centred_position.snapshot_image
image = image.scale(427, 320)
self.position_widget.child("svideo").setPixmap(qt.QPixmap(image))
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("user_level_log").warning(msg)
