class AcquisitionWidget(QtGui.QWidget):
"""
Descript. :
"""
def __init__(self,
parent=None,
name=None,
fl=0,
acq_params=None,
path_template=None,
layout='horizontal'):
"""
Descript. :
"""
QtGui.QWidget.__init__(self, parent, QtCore.Qt.WindowFlags(fl))
if name is not None:
self.setObjectName(name)
# Hardware objects ----------------------------------------------------
self._beamline_setup_hwobj = None
# Internal variables --------------------------------------------------
self.previous_energy = 0
# Properties ----------------------------------------------------------
# Signals -------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Graphic elements ----------------------------------------------------
if acq_params is None:
self._acquisition_parameters = queue_model_objects.\
AcquisitionParameters()
else:
self._acquisition_parameters = acq_params
if path_template is None:
self._path_template = queue_model_objects.PathTemplate()
else:
self._path_template = path_template
self._acquisition_mib = DataModelInputBinder(
self._acquisition_parameters)
if layout == "horizontal":
self.acq_widget = uic.loadUi(
os.path.join(
os.path.dirname(__file__),
"ui_files/Qt4_acquisition_widget_horizontal_layout.ui"))
self.acq_widget.inverse_beam_cbx.hide()
self.acq_widget.subwedge_size_label.hide()
self.acq_widget.subwedge_size_ledit.hide()
else:
self.acq_widget = uic.loadUi(
os.path.join(
os.path.dirname(__file__),
"ui_files/Qt4_acquisition_widget_vertical_layout.ui"))
# Layout --------------------------------------------------------------
self.main_layout = QtGui.QVBoxLayout(self)
self.main_layout.addWidget(self.acq_widget)
self.main_layout.setSpacing(0)
self.main_layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(self.main_layout)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self.acq_widget.energies_combo.activated.connect(self.energy_selected)
self.acq_widget.mad_cbox.toggled.connect(self.use_mad)
self.acq_widget.inverse_beam_cbx.toggled.connect(
self.set_use_inverse_beam)
self.acq_widget.first_image_ledit.textChanged.connect(
self.first_image_ledit_change)
self.acq_widget.num_images_ledit.textChanged.connect(
self.num_images_ledit_change)
overlap_ledit = self.acq_widget.findChild(QtGui.QLineEdit,
"overlap_ledit")
if overlap_ledit is not None:
overlap_ledit.textChanged.connect(self.overlap_changed)
self.acq_widget.subwedge_size_ledit.textChanged.connect(
self.subwedge_size_ledit_change)
self.acq_widget.osc_start_cbox.toggled.connect(
self.osc_start_cbox_click)
# Other ---------------------------------------------------------------
self.acq_widget.subwedge_size_ledit.setDisabled(True)
self.acq_widget.energies_combo.setDisabled(True)
self.acq_widget.energies_combo.addItems(
['ip: -', 'pk: -', 'rm1: -', 'rm2: -'])
self.acq_widget.osc_start_ledit.setEnabled(False)
self.osc_start_validator = QtGui.QDoubleValidator(
-10000, 10000, 4, self)
self.osc_range_validator = QtGui.QDoubleValidator(
-10000, 10000, 4, self)
self.kappa_validator = QtGui.QDoubleValidator(0, 360, 4, self)
self.kappa_phi_validator = QtGui.QDoubleValidator(0, 360, 4, self)
self.energy_validator = QtGui.QDoubleValidator(0, 25, 4, self)
self.resolution_validator = QtGui.QDoubleValidator(0, 15, 3, self)
self.transmission_validator = QtGui.QDoubleValidator(0, 100, 3, self)
self.exp_time_validator = QtGui.QDoubleValidator(
0.0001, 10000, 4, self)
self.first_img_validator = QtGui.QIntValidator(0, 99999, self)
self.num_img_validator = QtGui.QIntValidator(1, 99999, self)
def osc_start_cbox_click(self, state):
"""
Descript. :
"""
self.update_osc_start(
self._beamline_setup_hwobj._get_omega_axis_position())
self.acq_widget.osc_start_ledit.setEnabled(state)
def update_osc_start(self, new_value):
"""
Descript. :
"""
if not self.acq_widget.osc_start_cbox.isChecked():
osc_start_value = 0
try:
osc_start_value = round(float(new_value), 2)
except TypeError:
pass
self.acq_widget.osc_start_ledit.setText("%.2f" % osc_start_value)
self._acquisition_parameters.osc_start = osc_start_value
def update_kappa(self, new_value):
self.acq_widget.kappa_ledit.setText("%.2f" % float(new_value))
#self._acquisition_parameters.kappa = float(new_value)
def update_kappa_phi(self, new_value):
self.acq_widget.kappa_phi_ledit.setText("%.2f" % float(new_value))
#self._acquisition_parameters.kappa_phi = float(new_value)
def use_osc_start(self, state):
"""
Descript. :
"""
self.acq_widget.osc_start_cbox.setChecked(state)
self.acq_widget.osc_start_cbox.setDisabled(state)
def use_kappa(self, state):
if self._beamline_setup.diffractometer_hwobj.get_head_type() == \
self._beamline_setup.diffractometer_hwobj.PLATE:
state = False
self.acq_widget_layout.kappa_ledit.setEnabled(state)
def use_kappa_phi(self, state):
if self._beamline_setup.diffractometer_hwobj.get_head_type() == \
self._beamline_setup.diffractometer_hwobj.PLATE:
state = False
self.acq_widget_layout.kappa_phi_ledit.setEnabled(state)
def set_beamline_setup(self, beamline_setup):
"""
Descript. :
"""
self._beamline_setup_hwobj = beamline_setup
limits_dict = self._beamline_setup_hwobj.get_acquisition_limit_values()
if 'osc_range' in limits_dict:
limits = tuple(map(float, limits_dict['osc_range'].split(',')))
(lower, upper) = limits
self.osc_start_validator.setRange(lower, upper, 4)
self.osc_range_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update(
'osc_start', self.acq_widget.osc_start_ledit, float,
self.osc_start_validator)
self._acquisition_mib.bind_value_update(
'osc_range', self.acq_widget.osc_range_ledit, float,
self.osc_range_validator)
if self.acq_widget.findChild(QtGui.QLineEdit, "kappa_ledit"):
if 'kappa' in limits_dict:
limits = tuple(map(float, limits_dict['kappa'].split(',')))
(lower, upper) = limits
self.kappa_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update(
'kappa', self.acq_widget.kappa_ledit, float,
self.kappa_validator)
if self.acq_widget.findChild(QtGui.QLineEdit, "kappa_phi_ledit"):
if 'kappa_phi' in limits_dict:
limits = tuple(map(float, limits_dict['kappa_phi'].split(',')))
(lower, upper) = limits
self.kappa_phi_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update(
'kappa_phi', self.acq_widget.kappa_phi_ledit, float,
self.kappa_phi_validator)
if 'exposure_time' in limits_dict:
limits = tuple(map(float, limits_dict['exposure_time'].split(',')))
(lower, upper) = limits
self.exp_time_valdidator.setRange(lower, upper, 5)
self._acquisition_mib.bind_value_update('exp_time',
self.acq_widget.exp_time_ledit,
float, self.exp_time_validator)
if 'number_of_images' in limits_dict:
limits = tuple(map(int,
limits_dict['number_of_images'].split(',')))
(lower, upper) = limits
self.num_img_valdidator.setRange(lower, upper)
self.first_img_valdidator.setRange(lower, upper)
self._acquisition_mib.bind_value_update(
'first_image', self.acq_widget.first_image_ledit, int,
self.first_img_validator)
self._acquisition_mib.bind_value_update(
'num_images', self.acq_widget.num_images_ledit, int,
self.num_img_validator)
num_passes = self.acq_widget.findChild(QtGui.QLineEdit,
"num_passes_ledit")
if num_passes:
self._acquisition_mib.\
bind_value_update('num_passes', num_passes, int,
QtGui.QIntValidator(1, 1000, self))
overlap_ledit = self.acq_widget.findChild(QtGui.QLineEdit,
"overlap_ledit")
if overlap_ledit:
self._acquisition_mib.\
bind_value_update('overlap', overlap_ledit, float,
QtGui.QDoubleValidator(-1000, 1000, 2, self))
self._acquisition_mib.\
bind_value_update('energy',
self.acq_widget.energy_ledit,
float,
self.energy_validator)
self._acquisition_mib.\
bind_value_update('transmission',
self.acq_widget.transmission_ledit,
float,
self.transmission_validator)
self._acquisition_mib.\
bind_value_update('resolution',
self.acq_widget.resolution_ledit,
float,
self.resolution_validator)
self._acquisition_mib.\
bind_value_update('inverse_beam',
self.acq_widget.inverse_beam_cbx,
bool,
None)
self._acquisition_mib.\
bind_value_update('shutterless',
self.acq_widget.shutterless_cbx,
bool,
None)
"""self.acq_widget_layout.osc_start_ledit.setToolTip(\
"Oscillation start limits %0.2f : %0.2f" %(\
self.osc_start_validator.bottom(),
self.osc_start_validator.top()))
self.acq_widget_layout.osc_range_ledit.setToolTip(\
"Oscillation range limits %0.2f : %0.2f" %(\
self.osc_ra_validator.bottom(),
self.osc_start_validator.top()))"""
self.set_tunable_energy(beamline_setup.tunable_wavelength())
has_shutter_less = self._beamline_setup_hwobj.detector_has_shutterless(
)
self.acq_widget.shutterless_cbx.setEnabled(has_shutter_less)
self.acq_widget.shutterless_cbx.setChecked(has_shutter_less)
if self._beamline_setup_hwobj.disable_num_passes():
num_passes = self.acq_widget.findChild(QtGui.QLineEdit,
"num_passes_ledit")
if num_passes:
num_passes.setDisabled(True)
def first_image_ledit_change(self, new_value):
"""
Descript. :
"""
if str(new_value).isdigit():
self._path_template.start_num = int(new_value)
widget = self.acq_widget.first_image_ledit
self.emit(QtCore.SIGNAL('pathTemplateChanged'), widget, new_value)
def num_images_ledit_change(self, new_value):
"""
Descript. :
"""
if str(new_value).isdigit():
self._path_template.num_files = int(new_value)
widget = self.acq_widget.num_images_ledit
self.emit(QtCore.SIGNAL('pathTemplateChanged'), widget, new_value)
def overlap_changed(self, new_value):
"""
Descript. :
"""
if self._beamline_setup_hwobj:
has_shutter_less = self._beamline_setup_hwobj.detector_has_shutterless(
)
else:
has_shutter_less = True
if has_shutter_less:
try:
new_value = float(new_value)
except ValueError:
pass
if new_value != 0:
self.acq_widget.shutterless_cbx.setEnabled(False)
self.acq_widget.shutterless_cbx.setChecked(False)
self._acquisition_parameters.shutterless = False
else:
self.acq_widget.shutterless_cbx.setEnabled(True)
self.acq_widget.shutterless_cbx.setChecked(True)
self._acquisition_parameters.shutterless = True
def use_mad(self, state):
"""
Descript. :
"""
self.acq_widget.energies_combo.setEnabled(state)
if state:
(name, energy) = self.get_mad_energy()
if energy != 0:
self.set_energy(energy, 0)
self.emit(QtCore.SIGNAL('mad_energy_selected'), name, energy,
state)
else:
self.set_energy(self.previous_energy, 0)
energy = self._beamline_setup_hwobj.energy_hwobj.getCurrentEnergy()
self.emit(QtCore.SIGNAL('mad_energy_selected'), '',
self.previous_energy, state)
def set_use_inverse_beam(self, state):
"""
Descript. :
"""
if state:
self.acq_widget.subwedge_size_ledit.setEnabled(True)
else:
self.acq_widget.subwedge_size_ledit.setDisabled(True)
def use_inverse_beam(self):
"""
Descript. :
"""
return self.acq_widget.inverse_beam_cbx.checkState == QtCore.Qt.Checked
def get_num_subwedges(self):
"""
Descript. :
"""
return int(self.acq_widget.subwedge_size_ledit.text())
def subwedge_size_ledit_change(self, new_value):
"""
Descript. :
"""
if int(new_value) > self._acquisition_parameters.num_images:
Qt4_widget_colors.set_widget_color(\
self.acq_widget.subwedge_size_ledit,
Qt4_widget_colors.LIGHT_RED)
else:
Qt4_widget_colors.set_widget_color(
self.acq_widget.subwedge_size_ledit, Qt4_widget_colors.WHITE)
def get_mad_energy(self):
"""
Descript. :
"""
energy_str = str(self.acq_widget.energies_combo.currentText())
(name, value) = energy_str.split(':')
name = name.strip()
value = value.strip()
value = 0 if (value == '-') else value
return (name, value)
def set_energies(self, energy_scan_result):
"""
Descript. :
"""
self.acq_widget.energies_combo.clear()
inflection = ('ip: %.4f' % energy_scan_result.inflection) if \
energy_scan_result.inflection else 'ip: -'
peak = ('pk: %.4f' % energy_scan_result.peak) if \
energy_scan_result.peak else 'pk: -'
first_remote = ('rm1: %.4f' % energy_scan_result.first_remote) if \
energy_scan_result.first_remote else 'rm1: -'
second_remote = ('rm2: %.4f' % energy_scan_result.second_remote) if \
energy_scan_result.second_remote else 'rm2: -'
self.acq_widget.energies_combo.\
addItems([inflection, peak, first_remote, second_remote])
def energy_selected(self, index):
"""
Descript. :
"""
if self.acq_widget.mad_cbox.isChecked():
(name, energy) = self.get_mad_energy()
if energy != 0:
self.set_energy(energy, 0)
self.emit(QtCore.SIGNAL('mad_energy_selected'), name, energy, True)
def set_energy(self, energy, wav):
"""
Descript. :
"""
self.acq_widget.energy_ledit.setText("%.4f" % float(energy))
def update_transmission(self, transmission):
"""
Descript. :
"""
self.acq_widget.transmission_ledit.setText("%.2f" %
float(transmission))
def update_resolution(self, resolution):
"""
Descript. :
"""
self.acq_widget.resolution_ledit.setText("%.3f" % float(resolution))
def update_energy_limits(self, limits):
if limits:
self.energy_validator.setBottom(limits[0])
self.energy_validator.setTop(limits[1])
self.acq_widget_layout.energy_ledit.setToolTip(\
"Energy limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_transmission_limits(self, limits):
if limits:
self.transmission_validator.setBottom(limits[0])
self.transmission_validator.setTop(limits[1])
self.acq_widget_layout.transmission_ledit.setToolTip(\
"Transmission limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_resolution_limits(self, limits):
if limits:
self.resolution_validator.setBottom(limits[0])
self.resolution_validator.setTop(limits[1])
self.acq_widget_layout.resolution_ledit.setToolTip(
"Resolution limits %0.3f : %0.3f" % (limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_detector_exp_time_limits(self, limits):
if limits:
self.exp_time_validator.setRange(limits[0], limits[1], 4)
self.acq_widget_layout.exp_time_ledit.setToolTip(
"Exposure time limits %0.3f : %0.3f" % (limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_osc_range_limits(self, exp_time):
osc_range = 40.0 * exp_time
self.osc_range_validator.setRange(0, osc_range, 4)
self.acq_widget_layout.osc_range_ledit.setToolTip(
"Oscillation range limits %0.2f : %0.2f" % (0, osc_range))
self._acquisition_mib.validate_all()
def update_data_model(self, acquisition_parameters, path_template):
"""
Descript. :
"""
self._acquisition_parameters = acquisition_parameters
self._path_template = path_template
self._acquisition_mib.set_model(acquisition_parameters)
#Update mad widgets
mad = True if self._path_template.mad_prefix != '' else False
if mad:
mad_prefix = self._path_template.mad_prefix
index = MAD_ENERGY_COMBO_NAMES[mad_prefix]
self.acq_widget.energies_combo.setCurrentIndex(index)
self.acq_widget.mad_cbox.setChecked(True)
self.acq_widget.energies_combo.setEnabled(True)
else:
self.acq_widget.mad_cbox.setChecked(False)
self.acq_widget.energies_combo.setEnabled(False)
self.acq_widget.energies_combo.setCurrentIndex(0)
def set_tunable_energy(self, state):
"""
Descript. :
"""
self.acq_widget.energy_ledit.setEnabled(state)
self.acq_widget.mad_cbox.setEnabled(state)
self.acq_widget.energies_combo.setEnabled(state)
def disable_inverse_beam(self, state):
"""
Descript. :
"""
if state:
self.acq_widget.inverse_beam_cbx.hide()
self.acq_widget.subwedge_size_label.hide()
self.acq_widget.subwedge_size_ledit.hide()
else:
self.acq_widget.inverse_beam_cbx.show()
self.acq_widget.subwedge_size_label.show()
self.acq_widget.subwedge_size_ledit.show()
def hide_aperture(self, state):
"""
Descript. :
"""
pass
class AcquisitionWidgetSimple(QtGui.QWidget):
"""
Descript. :
"""
acqParametersChangedSignal = QtCore.pyqtSignal()
madEnergySelectedSignal = QtCore.pyqtSignal(str, float, bool)
def __init__(self, parent = None, name = None, fl = 0, acq_params = None,
path_template = None, layout = None):
"""
Descript. :
"""
QtGui.QWidget.__init__(self, parent, QtCore.Qt.WindowFlags(fl))
if name is not None:
self.setObjectName(name)
# Hardware objects ----------------------------------------------------
self._beamline_setup_hwobj = None
# Internal variables --------------------------------------------------
# Properties ----------------------------------------------------------
# Signals -------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Graphic elements ----------------------------------------------------
if acq_params is None:
self._acquisition_parameters = queue_model_objects.AcquisitionParameters()
else:
self._acquisition_parameters = acq_params
if path_template is None:
self._path_template = queue_model_objects.PathTemplate()
else:
self._path_template = path_template
self._acquisition_mib = DataModelInputBinder(self._acquisition_parameters)
self.acq_widget_layout = uic.loadUi(os.path.join(os.path.dirname(\
__file__), "ui_files/Qt4_acquisition_widget_vertical_simple_layout.ui"))
# Layout --------------------------------------------------------------
main_layout = QtGui.QVBoxLayout(self)
main_layout.addWidget(self.acq_widget_layout)
main_layout.setSpacing(0)
main_layout.setContentsMargins(0, 0, 0, 0)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self.acq_widget_layout.osc_start_cbox.stateChanged.connect(\
self.use_osc_start)
self.acq_widget_layout.num_images_cbox.activated.connect(\
self.update_num_images)
self.acq_widget_layout.detector_roi_mode_combo.activated.connect(\
self.detector_roi_mode_changed)
# Other ---------------------------------------------------------------
self.osc_start_validator = QtGui.QDoubleValidator(-10000, 10000, 4, self)
self.osc_range_validator = QtGui.QDoubleValidator(-10000, 10000, 4, self)
self.kappa_validator = QtGui.QDoubleValidator(0, 360, 4, self)
self.kappa_phi_validator = QtGui.QDoubleValidator(0, 360, 4, self)
self.energy_validator = QtGui.QDoubleValidator(0, 25, 5, self)
self.resolution_validator = QtGui.QDoubleValidator(0, 15, 3, self)
self.transmission_validator = QtGui.QDoubleValidator(0, 100, 3, self)
self.exp_time_validator = QtGui.QDoubleValidator(0, 10000, 5, self)
self.acq_widget_layout.num_images_cbox.setCurrentIndex(1)
self.acq_widget_layout.detector_roi_mode_label.setEnabled(False)
self.acq_widget_layout.detector_roi_mode_combo.setEnabled(False)
def set_osc_start_limits(self, limits):
if not None in limits:
self.osc_start_validator.setRange(limits[0], limits[1], 4)
def update_osc_start(self, new_value):
"""
Descript. :
"""
if not self.acq_widget_layout.osc_start_cbox.hasFocus():
osc_start_value = 0
try:
osc_start_value = round(float(new_value), 2)
except TypeError:
pass
self.acq_widget_layout.osc_start_ledit.setText(\
"%.2f" % osc_start_value)
#self._acquisition_parameters.osc_start = osc_start_value
def update_kappa(self, new_value):
"""
Descript. :
"""
if not self.acq_widget_layout.kappa_ledit.hasFocus():
self.acq_widget_layout.kappa_ledit.setText(\
"%.2f" % float(new_value))
def update_kappa_phi(self, new_value):
"""
Descript. :
"""
if not self.acq_widget_layout.kappa_phi_ledit.hasFocus():
self.acq_widget_layout.kappa_phi_ledit.setText(\
"%.2f" % float(new_value))
def use_kappa(self, state):
"""
Descript. :
"""
if self._beamline_setup_hwobj is not None:
if self._beamline_setup_hwobj.diffractometer_hwobj.in_plate_mode():
state = False
self.acq_widget_layout.kappa_label.setEnabled(state)
self.acq_widget_layout.kappa_ledit.setEnabled(state)
self.acq_widget_layout.kappa_phi_label.setEnabled(state)
self.acq_widget_layout.kappa_phi_ledit.setEnabled(state)
def update_num_images(self, index = None, num_images = None):
"""
Descript. :
"""
if index is not None:
if index is 0:
self._acquisition_parameters.num_images = 1
self._path_template.num_files = 1
elif index is 1:
self._acquisition_parameters.num_images = 2
self._path_template.num_files = 2
elif index is 2:
self._acquisition_parameters.num_images = 4
self._path_template.num_files = 4
if num_images:
if self.acq_widget_layout.num_images_cbox.count() > 3:
self.acq_widget_layout.num_images_cbox.removeItem(4)
if num_images is 1:
self.acq_widget_layout.num_images_cbox.setCurrentIndex(0)
elif num_images is 2:
self.acq_widget_layout.num_images_cbox.setCurrentIndex(1)
elif num_images is 4:
self.acq_widget_layout.num_images_cbox.setCurrentIndex(2)
else:
self.acq_widget_layout.num_images_cbox.addItem(str(num_images))
self.acq_widget_layout.num_images_cbox.setCurrenIndex(3)
self._path_template.num_files = num_images
def use_mad(self, state):
"""
Descript. :
"""
pass
def get_mad_energy(self):
"""
Descript. :
"""
pass
def set_energies(self, energy_scan_result):
"""
Descript. :
"""
pass
def energy_selected(self, index):
"""
Descript. :
"""
pass
def set_beamline_setup(self, beamline_setup):
"""
Descript. :
"""
self._beamline_setup_hwobj = beamline_setup
limits_dict = self._beamline_setup_hwobj.get_acquisition_limit_values()
if 'osc_range' in limits_dict:
limits = tuple(map(float, limits_dict['osc_range'].split(',')))
(lower, upper) = limits
elf.osc_start_validator.setRange(lower, upper, 4)
self.osc_range_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update('osc_start',
self.acq_widget_layout.osc_start_ledit,
float,
self.osc_start_validator)
self._acquisition_mib.bind_value_update('osc_range',
self.acq_widget_layout.osc_range_ledit,
float,
self.osc_range_validator)
if 'kappa' in limits_dict:
limits = tuple(map(float, limits_dict['kappa'].split(',')))
(lower, upper) = limits
self.kappa_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update('kappa',
self.acq_widget_layout.kappa_ledit,
float,
self.kappa_validator)
if 'kappa_phi' in limits_dict:
limits = tuple(map(float, limits_dict['kappa_phi'].split(',')))
(lower, upper) = limits
self.kappa_phi_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update('kappa_phi',
self.acq_widget_layout.kappa_phi_ledit,
float,
self.kappa_phi_validator)
if 'exposure_time' in limits_dict:
limits = tuple(map(float, limits_dict['exposure_time'].split(',')))
(lower, upper) = limits
self.exp_time_validator.setRange(lower, upper, 5)
self._acquisition_mib.bind_value_update('exp_time',
self.acq_widget_layout.exp_time_ledit,
float,
self.exp_time_validator)
self._acquisition_mib.\
bind_value_update('energy',
self.acq_widget_layout.energy_ledit,
float,
self.energy_validator)
self.acq_widget_layout.energy_ledit.setToolTip(\
"Energy limits %0.3f : %0.3f" % \
(self.energy_validator.bottom(), self.energy_validator.top()))
self._acquisition_mib.\
bind_value_update('transmission',
self.acq_widget_layout.transmission_ledit,
float,
self.transmission_validator)
self._acquisition_mib.\
bind_value_update('resolution',
self.acq_widget_layout.resolution_ledit,
float,
self.resolution_validator)
self.set_tunable_energy(beamline_setup.tunable_wavelength())
def set_energy(self, energy, wav):
"""
Descript. :
"""
if not self.acq_widget_layout.energy_ledit.hasFocus():
self.acq_widget_layout.energy_ledit.setText(\
"%.4f" % float(energy))
def update_transmission(self, transmission):
"""
Descript. :
"""
if self.acq_widget_layout.transmission_ledit.hasFocus():
self.acq_widget_layout.transmission_ledit.setText(\
"%.2f" % float(transmission))
def update_resolution(self, resolution):
"""
Descript. :
"""
if not self.acq_widget_layout.resolution_ledit.hasFocus():
self.acq_widget_layout.resolution_ledit.setText(\
"%.3f" % float(resolution))
def update_energy_limits(self, limits):
"""
Descript. :
"""
if limits:
self.energy_validator.setBottom(limits[0])
self.energy_validator.setTop(limits[1])
self.acq_widget_layout.energy_ledit.setToolTip(
"Energy limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_transmission_limits(self, limits):
"""
Descript. :
"""
if limits:
self.transmission_validator.setBottom(limits[0])
self.transmission_validator.setTop(limits[1])
self.acq_widget_layout.transmission_ledit.setToolTip(
"Transmission limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_resolution_limits(self, limits):
"""
Descript. :
"""
if limits:
self.resolution_validator.setBottom(limits[0])
self.resolution_validator.setTop(limits[1])
self.acq_widget_layout.resolution_ledit.setToolTip(
"Resolution limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_detector_exp_time_limits(self, limits):
"""
Descript. :
"""
if limits:
self.exp_time_validator.setBottom(limits[0])
self.exp_time_validator.setTop(limits[1])
self.acq_widget_layout.exp_time_ledit.setToolTip(
"Exposure time limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def init_detector_roi_modes(self):
"""
Descript. :
"""
if self._beamline_setup_hwobj is not None:
roi_modes = self._beamline_setup_hwobj._get_roi_modes()
if (len(roi_modes) > 0 and
self.acq_widget_layout.detector_roi_mode_combo.count() == 0):
for roi_mode in roi_modes:
self.acq_widget_layout.detector_roi_mode_combo.\
addItem(roi_mode)
self.acq_widget_layout.detector_roi_mode_label.setEnabled(True)
self.acq_widget_layout.detector_roi_mode_combo.setEnabled(True)
#self._acquisition_mib.bind_value_update('detector_roi_mode',
# self.acq_widget_layout.detector_roi_mode_combo,
# str,
# None)
def update_detector_roi_mode(self, roi_mode_index):
"""
Descript. :
"""
if self.acq_widget_layout.detector_roi_mode_combo.count() > 0:
self.acq_widget_layout.detector_roi_mode_combo.\
setCurrentIndex(roi_mode_index)
def detector_roi_mode_changed(self, roi_mode_index):
"""
Descript. :
"""
if self._beamline_setup_hwobj is not None:
self._beamline_setup_hwobj.detector_hwobj.set_roi_mode(roi_mode_index)
def update_data_model(self, acquisition_parameters, path_template):
"""
Descript. :
"""
self._acquisition_parameters = acquisition_parameters
self._acquisition_mib.set_model(acquisition_parameters)
self._path_template = path_template
self.update_num_images(None, acquisition_parameters.num_images)
def set_tunable_energy(self, state):
"""
Descript. :
"""
self.acq_widget_layout.energy_ledit.setEnabled(state)
def use_osc_start(self, state):
"""
Descript. :
"""
self.acq_widget_layout.osc_start_ledit.setEnabled(state)
def check_parameter_conflict(self):
"""
Descript. :
"""
return len(self._acquisition_mib.validate_all()) > 0
class AcquisitionWidget(QWidget):
acqParametersChangedSignal = pyqtSignal(list)
madEnergySelectedSignal = pyqtSignal(str, float, bool)
def __init__(self, parent=None, name=None, fl=0, acq_params=None,
path_template=None, layout='horizontal'):
QWidget.__init__(self, parent, Qt.WindowFlags(fl))
if name is not None:
self.setObjectName(name)
# Hardware objects ----------------------------------------------------
self._beamline_setup_hwobj = None
self._diffractometer_hwobj = None
# Internal variables --------------------------------------------------
self.previous_energy = 0
# If the acq. widget is used with grids then total osc range is not
# equal to num_images * osc_range, but num_images_per_line * osc_range
# For grids the osc total range is updated when a grid is selected
self.grid_mode = False
# Properties ----------------------------------------------------------
# Signals -------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Graphic elements ----------------------------------------------------
if acq_params is None:
self._acquisition_parameters = queue_model_objects.\
AcquisitionParameters()
else:
self._acquisition_parameters = acq_params
if path_template is None:
self._path_template = queue_model_objects.PathTemplate()
else:
self._path_template = path_template
self._acquisition_mib = DataModelInputBinder(self._acquisition_parameters)
if layout == "horizontal":
self.acq_widget_layout = loadUi(os.path.join(\
os.path.dirname(__file__),
"ui_files/Qt4_acquisition_widget_horizontal_layout.ui"))
self.use_osc_start(False)
else:
self.acq_widget_layout = loadUi(os.path.join(\
os.path.dirname(__file__),
"ui_files/Qt4_acquisition_widget_vertical_layout.ui"))
# Layout --------------------------------------------------------------
__main_vlayout = QVBoxLayout(self)
__main_vlayout.addWidget(self.acq_widget_layout)
__main_vlayout.setSpacing(0)
__main_vlayout.setContentsMargins(0, 0, 0, 0)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self.acq_widget_layout.osc_start_cbox.stateChanged.connect(\
self.fix_osc_start)
self.acq_widget_layout.exp_time_ledit.textChanged.connect(\
self.exposure_time_ledit_changed)
self.acq_widget_layout.first_image_ledit.textChanged.connect(\
self.first_image_ledit_change)
self.acq_widget_layout.num_images_ledit.textChanged.connect(\
self.num_images_ledit_change)
self.acq_widget_layout.detector_roi_mode_combo.activated.connect(\
self.detector_roi_mode_changed)
self.acq_widget_layout.energies_combo.activated.connect(\
self.energy_selected)
self.acq_widget_layout.mad_cbox.toggled.connect(\
self.use_mad)
self.acq_widget_layout.osc_start_ledit.textEdited.connect(\
self.osc_start_ledit_changed)
self.acq_widget_layout.osc_range_ledit.textEdited.connect(\
self.osc_range_per_frame_ledit_changed)
self.acq_widget_layout.osc_total_range_ledit.textEdited.connect(\
self.osc_total_range_ledit_changed)
self.acq_widget_layout.energy_ledit.textEdited.connect(\
self.energy_ledit_changed)
self.acq_widget_layout.transmission_ledit.textEdited.connect(\
self.transmission_ledit_changed)
self.acq_widget_layout.resolution_ledit.textEdited.connect(\
self.resolution_ledit_changed)
self.acq_widget_layout.kappa_ledit.textEdited.connect(\
self.kappa_ledit_changed)
self.acq_widget_layout.kappa_phi_ledit.textEdited.connect(\
self.kappa_phi_ledit_changed)
if self.acq_widget_layout.findChild(QLineEdit, "overlap_ledit"):
self.acq_widget_layout.overlap_ledit.textChanged.connect(\
self.overlap_changed)
if self.acq_widget_layout.findChild(QCheckBox, "max_osc_range_cbx"):
self.acq_widget_layout.max_osc_range_cbx.toggled.connect(\
self.max_osc_range_toggled)
# Other ---------------------------------------------------------------
self.value_changed_list = []
self.acq_widget_layout.energies_combo.setDisabled(True)
self.acq_widget_layout.energies_combo.addItems(\
['ip: -', 'pk: -', 'rm1: -', 'rm2: -'])
self.osc_start_validator = QDoubleValidator(\
-10000, 10000, 4, self.acq_widget_layout.osc_start_ledit)
self.osc_range_per_frame_validator = QDoubleValidator(\
0, 10000, 4, self.acq_widget_layout.osc_range_ledit)
self.osc_total_range_validator = QDoubleValidator(\
0, 10000, 4, self.acq_widget_layout.osc_total_range_ledit)
self.kappa_validator = QDoubleValidator(\
0, 360, 4, self.acq_widget_layout.kappa_ledit)
self.kappa_phi_validator = QDoubleValidator(\
0, 360, 4, self.acq_widget_layout.kappa_phi_ledit)
self.energy_validator = QDoubleValidator(\
4, 25, 4, self.acq_widget_layout.energy_ledit)
self.resolution_validator = QDoubleValidator(\
0, 15, 3, self.acq_widget_layout.resolution_ledit)
self.transmission_validator = QDoubleValidator(\
0, 100, 3, self.acq_widget_layout.transmission_ledit)
self.exp_time_validator = QDoubleValidator(\
0.0001, 10000, 7, self.acq_widget_layout.exp_time_ledit)
self.first_img_validator = QIntValidator(\
0, 99999, self.acq_widget_layout.first_image_ledit)
self.num_img_validator = QIntValidator(\
1, 9999999, self.acq_widget_layout.num_images_ledit)
self.acq_widget_layout.detector_roi_mode_label.setEnabled(False)
self.acq_widget_layout.detector_roi_mode_combo.setEnabled(False)
def use_max_osc_range(self, state):
self.acq_widget_layout.max_osc_range_cbx.setEnabled(state)
def fix_osc_start(self, state):
"""
Fix osc start, so the lineEdit do not change when osc is changed
"""
self.acq_widget_layout.osc_start_ledit.setEnabled(state)
def update_osc_start(self, new_value):
"""
Updates osc line edit
"""
if "osc_start" not in self.value_changed_list and \
not self.acq_widget_layout.osc_start_ledit.hasFocus() and \
not self.acq_widget_layout.osc_start_cbox.isChecked():
osc_start_value = 0
try:
osc_start_value = round(float(new_value), 2)
except TypeError:
pass
self.acq_widget_layout.osc_start_ledit.setText(str(osc_start_value))
self._acquisition_parameters.osc_start = osc_start_value
#self.update_osc_total_range_limits()
#self.update_num_images_limits()
def osc_start_ledit_changed(self, osc_start):
"""Fixes osc start edit"""
if "osc_start" not in self.value_changed_list:
self.value_changed_list.append("osc_start")
self.update_osc_total_range_limits()
self.update_num_images_limits()
self.emit_acq_parameters_changed()
def update_osc_start_limits(self):
"""In the plate mode sets osc start limits"""
self.update_osc_total_range_limits()
self.update_num_images_limits()
def osc_range_per_frame_ledit_changed(self, new_value):
self.update_osc_total_range()
self.update_num_images_limits()
self.update_exp_time_limits()
self.emit_acq_parameters_changed()
def update_osc_range_per_frame_limits(self):
try:
max_osc_speed = self._diffractometer_hwobj.get_osc_max_speed()
top_limit = max_osc_speed * float(self.acq_widget_layout.exp_time_ledit.text())
self.osc_range_per_frame_validator.setTop(top_limit)
tool_tip = "Oscillation range per frame limits 0 : " +\
"%0.4f\n4 digits precision." % top_limit
self.acq_widget_layout.osc_range_ledit.setToolTip(tool_tip)
self._acquisition_mib.validate_all()
except:
pass
def update_osc_total_range(self):
self.acq_widget_layout.osc_total_range_ledit.blockSignals(True)
if not self.grid_mode:
try:
self.acq_widget_layout.osc_total_range_ledit.setText(str(\
float(self.acq_widget_layout.osc_range_ledit.text()) * \
float(self.acq_widget_layout.num_images_ledit.text())))
except:
pass
self.acq_widget_layout.osc_total_range_ledit.blockSignals(False)
def update_total_exp_time(self):
try:
self.acq_widget_layout.total_exp_time_ledit.setText("%.2f" % \
(float(self.acq_widget_layout.exp_time_ledit.text()) * \
float(self.acq_widget_layout.num_images_ledit.text())))
except:
pass
def osc_total_range_ledit_changed(self, new_value):
if not self.grid_mode:
try:
num_images = int(float(new_value) / \
float(self.acq_widget_layout.osc_range_ledit.text()))
self.acq_widget_layout.num_images_ledit.blockSignals(True)
self.acq_widget_layout.num_images_ledit.setText("%d" % num_images)
self._acquisition_parameters.num_images = num_images
self.acq_widget_layout.num_images_ledit.blockSignals(False)
except:
pass
self.emit_acq_parameters_changed()
def update_osc_total_range_limits(self, num_images=None):
"""Updates osc totol range. Limits are changed if a plate is used.
- For simple oscillation osc_range is defined by osc_start and
osc_start top limit.
- For mesh osc_range is defined by number of images per line
and osc in the middle of mesh
"""
if self._diffractometer_hwobj.in_plate_mode():
if hasattr(self.parent(), "set_osc_total_range"):
self.parent().set_osc_total_range(num_images)
self._acquisition_mib.validate_all()
def update_exp_time_limits(self):
try:
exp_time_limits = self._beamline_setup_hwobj.detector_hwobj.get_exposure_time_limits()
max_osc_speed = self._diffractometer_hwobj.get_osc_max_speed()
top_limit = float(self.acq_widget_layout.osc_range_ledit.text()) / max_osc_speed
limits = (max(exp_time_limits[0], top_limit), exp_time_limits[1])
self.update_detector_exp_time_limits(limits)
except:
pass
def update_kappa(self, new_value):
"""
Updates kappa value
"""
if not self.acq_widget_layout.kappa_ledit.hasFocus() and \
new_value is not None:
self.acq_widget_layout.kappa_ledit.setText(str(new_value))
self.emit_acq_parameters_changed()
def update_kappa_phi(self, new_value):
"""
Descript. :
"""
if not self.acq_widget_layout.kappa_phi_ledit.hasFocus() and \
new_value:
self.acq_widget_layout.kappa_phi_ledit.setText(str(new_value))
self.emit_acq_parameters_changed()
def use_osc_start(self, state):
"""
Descript. :
"""
self.acq_widget_layout.osc_start_cbox.setVisible(state)
self.acq_widget_layout.osc_start_label.setVisible(not state)
self.acq_widget_layout.osc_start_ledit.setEnabled(not state)
def use_kappa(self, state):
"""
Descript. :
"""
if self._diffractometer_hwobj is not None:
if self._diffractometer_hwobj.in_plate_mode():
state = False
self.acq_widget_layout.kappa_label.setEnabled(state)
self.acq_widget_layout.kappa_ledit.setEnabled(state)
self.acq_widget_layout.kappa_phi_label.setEnabled(state)
self.acq_widget_layout.kappa_phi_ledit.setEnabled(state)
def set_beamline_setup(self, beamline_setup):
"""
Descript. :
"""
self._beamline_setup_hwobj = beamline_setup
limits_dict = self._beamline_setup_hwobj.get_acquisition_limit_values()
self._diffractometer_hwobj = self._beamline_setup_hwobj.diffractometer_hwobj
if 'osc_range' in limits_dict:
limits = tuple(map(float, limits_dict['osc_range'].split(',')))
(lower, upper) = limits
self.osc_start_validator.setRange(lower, upper, 4)
self.osc_range_per_frame_validator.setRange(lower, upper, 4)
self.osc_total_range_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update('osc_start',
self.acq_widget_layout.osc_start_ledit,
float,
self.osc_start_validator)
self._acquisition_mib.bind_value_update('osc_range',
self.acq_widget_layout.osc_range_ledit,
float,
self.osc_range_per_frame_validator)
self._acquisition_mib.bind_value_update('osc_total_range',
self.acq_widget_layout.osc_total_range_ledit,
float,
self.osc_total_range_validator)
if 'kappa' in limits_dict:
limits = tuple(map(float, limits_dict['kappa'].split(',')))
(lower, upper) = limits
self.kappa_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update('kappa',
self.acq_widget_layout.kappa_ledit,
float,
self.kappa_validator)
if 'kappa_phi' in limits_dict:
limits = tuple(map(float, limits_dict['kappa_phi'].split(',')))
(lower, upper) = limits
self.kappa_phi_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update('kappa_phi',
self.acq_widget_layout.kappa_phi_ledit,
float,
self.kappa_phi_validator)
if 'exposure_time' in limits_dict:
limits = tuple(map(float, limits_dict['exposure_time'].split(',')))
(lower, upper) = limits
self.exp_time_validator.setRange(lower, upper, 6)
self._acquisition_mib.bind_value_update('exp_time',
self.acq_widget_layout.exp_time_ledit,
float,
self.exp_time_validator)
if 'number_of_images' in limits_dict:
limits = tuple(map(float, limits_dict['number_of_images'].split(',')))
(lower, upper) = limits
self.num_img_validator.setRange(lower, upper)
self.first_img_validator.setRange(lower, upper)
self._acquisition_mib.bind_value_update('first_image',
self.acq_widget_layout.first_image_ledit,
int,
self.first_img_validator)
self._acquisition_mib.bind_value_update('num_images',
self.acq_widget_layout.num_images_ledit,
int,
self.num_img_validator)
num_passes = self.acq_widget_layout.findChild(QLineEdit, "num_passes_ledit")
if num_passes:
self._acquisition_mib.\
bind_value_update('num_passes', num_passes, int,
QIntValidator(1, 1000, self))
overlap_ledit = self.acq_widget_layout.findChild(QLineEdit, "overlap_ledit")
if overlap_ledit:
self._acquisition_mib.\
bind_value_update('overlap', overlap_ledit, float,
QDoubleValidator(-1000, 1000, 2, self))
self._acquisition_mib.\
bind_value_update('energy',
self.acq_widget_layout.energy_ledit,
float,
self.energy_validator)
self.update_energy_limits((self.energy_validator.bottom(),
self.energy_validator.top()))
self._acquisition_mib.\
bind_value_update('transmission',
self.acq_widget_layout.transmission_ledit,
float,
self.transmission_validator)
self.update_transmission_limits((self.transmission_validator.bottom(),
self.transmission_validator.top()))
self._acquisition_mib.\
bind_value_update('resolution',
self.acq_widget_layout.resolution_ledit,
float,
self.resolution_validator)
#self.update_resolution_limits((self.resolution_validator.bottom(),
# self.resolution_validator.top()))
self._acquisition_mib.\
bind_value_update('shutterless',
self.acq_widget_layout.shutterless_cbx,
bool,
None)
self.set_tunable_energy(beamline_setup.tunable_wavelength())
has_shutter_less = self._beamline_setup_hwobj.detector_has_shutterless()
self.acq_widget_layout.shutterless_cbx.setEnabled(has_shutter_less)
self.acq_widget_layout.shutterless_cbx.setChecked(has_shutter_less)
if self._beamline_setup_hwobj.disable_num_passes():
num_passes = self.acq_widget_layout.findChild(QLineEdit, "num_passes_ledit")
if num_passes:
num_passes.setDisabled(True)
self.init_detector_roi_modes()
def first_image_ledit_change(self, new_value):
"""
Descript. :
"""
#if str(new_value).isdigit():
# #self._path_template.start_num = int(new_value)
# #widget = self.acq_widget_layout.first_image_ledit
self.emit_acq_parameters_changed()
def exposure_time_ledit_changed(self, new_values):
"""If the exposure time changes we have to check the osc speed
and if necessary update osc range per frame
"""
self.update_osc_range_per_frame_limits()
self.update_osc_total_range_limits()
self.update_total_exp_time()
self.emit_acq_parameters_changed()
def num_images_ledit_change(self, new_value):
"""
Descript. :
"""
if str(new_value).isdigit():
#self._path_template.num_files = int(new_value)
self.update_osc_range_per_frame_limits()
self.update_osc_total_range_limits()
self.update_osc_total_range()
self.update_total_exp_time()
self.emit_acq_parameters_changed()
def overlap_changed(self, new_value):
"""
Descript. :
"""
if self._beamline_setup_hwobj:
has_shutter_less = self._beamline_setup_hwobj.detector_has_shutterless()
else:
has_shutter_less = True
if has_shutter_less:
try:
new_value = float(new_value)
except ValueError:
pass
if new_value != 0:
self.acq_widget_layout.shutterless_cbx.setEnabled(False)
self.acq_widget_layout.shutterless_cbx.setChecked(False)
self._acquisition_parameters.shutterless = False
else:
self.acq_widget_layout.shutterless_cbx.setEnabled(True)
self.acq_widget_layout.shutterless_cbx.setChecked(True)
self._acquisition_parameters.shutterless = True
def use_mad(self, state):
"""
Descript. :
"""
self.acq_widget_layout.energies_combo.setEnabled(state)
if state:
(name, energy) = self.get_mad_energy()
if energy != 0:
self.update_energy(energy)
self.madEnergySelectedSignal.emit(name, energy, state)
else:
self.update_energy(self.previous_energy)
energy = self._beamline_setup_hwobj.energy_hwobj.getCurrentEnergy()
self.madEnergySelectedSignal.emit('', self.previous_energy, state)
def max_osc_range_toggled(self, state):
self.update_osc_total_range_limits()
def get_mad_energy(self):
"""
Descript. :
"""
energy_str = str(self.acq_widget_layout.energies_combo.currentText())
(name, value) = energy_str.split(':')
name = name.strip()
value = value.strip()
value = 0 if (value == '-') else float(value)
return (name, value)
def set_energies(self, energy_scan_result):
"""
Descript. :
"""
self.acq_widget_layout.energies_combo.clear()
inflection = ('ip: %.4f' % energy_scan_result.inflection) if \
energy_scan_result.inflection else 'ip: -'
peak = ('pk: %.4f' % energy_scan_result.peak) if \
energy_scan_result.peak else 'pk: -'
first_remote = ('rm1: %.4f' % energy_scan_result.first_remote) if \
energy_scan_result.first_remote else 'rm1: -'
second_remote = ('rm2: %.4f' % energy_scan_result.second_remote) if \
energy_scan_result.second_remote else 'rm2: -'
self.acq_widget_layout.energies_combo.\
addItems([inflection, peak, first_remote, second_remote])
def energy_selected(self, index):
"""
Descript. :
"""
if self.acq_widget_layout.mad_cbox.isChecked():
(name, energy) = self.get_mad_energy()
if energy != 0:
self.update_energy(energy)
self.madEnergySelectedSignal.emit(name, energy, True)
def energy_ledit_changed(self, new_value):
if "energy" not in self.value_changed_list:
self.value_changed_list.append("energy")
self.emit_acq_parameters_changed()
def update_energy(self, energy, wav=None):
"""
Descript. :
"""
if "energy" not in self.value_changed_list and \
not self.acq_widget_layout.energy_ledit.hasFocus():
self.acq_widget_layout.energy_ledit.setText(str(energy))
self.emit_acq_parameters_changed()
def transmission_ledit_changed(self, transmission):
if "transmission" not in self.value_changed_list:
self.value_changed_list.append("transmission")
self.emit_acq_parameters_changed()
def update_transmission(self, transmission):
if "transmission" not in self.value_changed_list:
self.acq_widget_layout.transmission_ledit.setText(str(transmission))
self.emit_acq_parameters_changed()
def resolution_ledit_changed(self, resolution):
if "resolution" not in self.value_changed_list:
self.value_changed_list.append("resolution")
self.emit_acq_parameters_changed()
def update_resolution(self, resolution):
if "resolution" not in self.value_changed_list and \
not self.acq_widget_layout.resolution_ledit.hasFocus():
self.acq_widget_layout.resolution_ledit.setText(str(resolution))
self.emit_acq_parameters_changed()
def update_energy_limits(self, limits):
if limits:
self.energy_validator.setBottom(limits[0])
self.energy_validator.setTop(limits[1])
self.acq_widget_layout.energy_ledit.setToolTip(\
"Energy limits %0.4f : %0.4f keV\n" % \
(limits[0], limits[1]) + \
"4 digits precision.")
self._acquisition_mib.validate_all()
def update_transmission_limits(self, limits):
if limits:
self.transmission_validator.setBottom(limits[0])
self.transmission_validator.setTop(limits[1])
self.acq_widget_layout.transmission_ledit.setToolTip(\
"Transmission limits %0.2f : %0.2f %%\n" % \
(limits[0], limits[1]) + \
"2 digits precision.")
self._acquisition_mib.validate_all()
def update_resolution_limits(self, limits):
if limits:
self.resolution_validator.setBottom(limits[0])
self.resolution_validator.setTop(limits[1])
self.acq_widget_layout.resolution_ledit.setToolTip(
"Resolution limits %0.4f : %0.4f %s\n" % \
(limits[0], limits[1], chr(197)) +
"4 digits precision.")
self._acquisition_mib.validate_all()
def update_detector_exp_time_limits(self, limits):
if limits:
self.exp_time_validator.setRange(limits[0], limits[1], 6)
self.acq_widget_layout.exp_time_ledit.setToolTip(
"Exposure time limits %0.6f s : %0.1f s\n" % \
(limits[0], limits[1]) + \
"6 digits precision.")
self._acquisition_mib.validate_all()
def update_num_images_limits(self, num_images_limits=None):
"""Updates number of images limit. Method used if plate mode.
"""
self._acquisition_mib.validate_all()
return
if self._diffractometer_hwobj.in_plate_mode():
if num_images_limits is None:
try:
osc_start = float(self.acq_widget_layout.osc_start_ledit.text())
osc_range = float(self.acq_widget_layout.osc_range_ledit.text())
except ValueError:
return
if osc_range == 0:
return
num_images_limits = int((self.osc_start_validator.top() - \
osc_start) / osc_range)
self.num_img_validator.setTop(num_images_limits)
self.acq_widget_layout.num_images_ledit.setToolTip(\
"Number of images limits : %d" % num_images_limits)
self._acquisition_mib.validate_all()
def init_detector_roi_modes(self):
"""
Descript. :
"""
if self._beamline_setup_hwobj is not None:
roi_modes = self._beamline_setup_hwobj.detector_hwobj.get_roi_modes()
if (len(roi_modes) > 0 and
self.acq_widget_layout.detector_roi_mode_combo.count() == 0):
for roi_mode in roi_modes:
self.acq_widget_layout.detector_roi_mode_combo.\
addItem(roi_mode)
self.acq_widget_layout.detector_roi_mode_label.setEnabled(True)
self.acq_widget_layout.detector_roi_mode_combo.setEnabled(True)
#self._acquisition_mib.bind_value_update('detector_roi_mode',
# self.acq_widget_layout.detector_roi_mode_combo,
## str,
# None)
def update_detector_roi_mode(self, roi_mode_index):
"""
Descript. :
"""
if roi_mode_index is not None and \
self.acq_widget_layout.detector_roi_mode_combo.count() > 0:
self.acq_widget_layout.detector_roi_mode_combo.\
setCurrentIndex(roi_mode_index)
def detector_roi_mode_changed(self, roi_mode_index):
"""
Descript. :
"""
if self._beamline_setup_hwobj is not None:
self._beamline_setup_hwobj.detector_hwobj.set_roi_mode(roi_mode_index)
def kappa_ledit_changed(self, new_value):
if "kappa" not in self.value_changed_list:
self.value_changed_list.append("kappa")
self.emit_acq_parameters_changed()
def kappa_phi_ledit_changed(self, new_value):
if "kappa_phi" not in self.value_changed_list:
self.value_changed_list.append("kappa_phi")
self.emit_acq_parameters_changed()
def update_data_model(self, acquisition_parameters, path_template):
"""
Descript. :
"""
self._acquisition_parameters = acquisition_parameters
self._path_template = path_template
self._acquisition_mib.set_model(acquisition_parameters)
#Update mad widgets
mad = True if self._path_template.mad_prefix != '' else False
if mad:
mad_prefix = str(self._path_template.mad_prefix)
index = MAD_ENERGY_COMBO_NAMES[str(mad_prefix)]
self.acq_widget_layout.energies_combo.setCurrentIndex(index)
self.acq_widget_layout.mad_cbox.setChecked(True)
self.acq_widget_layout.energies_combo.setEnabled(True)
else:
self.acq_widget_layout.mad_cbox.setChecked(False)
self.acq_widget_layout.energies_combo.setEnabled(False)
self.acq_widget_layout.energies_combo.setCurrentIndex(0)
#self.update_osc_start_limits()
#self.update_osc_total_range()
self.emit_acq_parameters_changed()
def set_tunable_energy(self, state):
"""
Descript. :
"""
self.acq_widget_layout.energy_ledit.setEnabled(state)
self.acq_widget_layout.mad_cbox.setEnabled(state)
self.acq_widget_layout.energies_combo.setEnabled(state)
def check_parameter_conflict(self):
return self._acquisition_mib.validate_all()
def emit_acq_parameters_changed(self):
self.acqParametersChangedSignal.emit(self._acquisition_mib.validate_all())
class AdvancedParametersWidget(QWidget):
def __init__(self, parent = None, name = "advanced_parameters_widget"):
QWidget.__init__(self, parent)
# Hardware objects ----------------------------------------------------
self._queue_model_hwobj = None
self._beamline_setup_hwobj = None
# Internal values -----------------------------------------------------
self._data_collection = None
self._tree_view_item = None
# Properties ----------------------------------------------------------
# Signals -------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Graphic elements ----------------------------------------------------
_dc_parameters_widget = QWidget(self)
self._data_path_widget = DataPathWidget(_dc_parameters_widget)
self._acq_widget = AcquisitionWidget(_dc_parameters_widget,
layout = 'horizontal')
# Layout --------------------------------------------------------------
_dc_parameters_widget_layout = 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 = 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 set_beamline_setup(self, bl_setup):
self._beamline_setup_hwobj = bl_setup
self._acq_widget.set_beamline_setup(bl_setup)
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._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)
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, Qt4_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 AdvancedParametersWidget(QWidget):
def __init__(self, parent = None, name = "advanced_parameters_widget"):
QWidget.__init__(self, parent)
# Hardware objects ----------------------------------------------------
self._queue_model_hwobj = None
self._beamline_setup_hwobj = None
# Internal values -----------------------------------------------------
self._data_collection = None
self._tree_view_item = None
# Properties ----------------------------------------------------------
# Signals -------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Graphic elements ----------------------------------------------------
_dc_parameters_widget = QWidget(self)
self._data_path_widget = DataPathWidget(_dc_parameters_widget)
self._acq_widget = AcquisitionWidget(_dc_parameters_widget,
layout = 'horizontal')
# Layout --------------------------------------------------------------
_dc_parameters_widget_layout = 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 = 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.num_images_ledit.setEnabled(False)
self._acq_widget.acq_widget_layout.shutterless_cbx.hide()
def set_beamline_setup(self, bl_setup):
self._beamline_setup_hwobj = bl_setup
self._acq_widget.set_beamline_setup(bl_setup)
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._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)
def populate_widget(self, tree_view_item):
self._tree_view_item = tree_view_item
if isinstance(tree_view_item, Qt4_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)
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(QtGui.QWidget):
def __init__(self, parent = None, name = "parameter_widget"):
QtGui.QWidget.__init__(self, parent)
if name is not None:
self.setObjectName(name)
# Properties ----------------------------------------------------------
# Signals ------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Hardware objects ----------------------------------------------------
self._beamline_setup_hwobj = None
# Internal variables --------------------------------------------------
self._data_collection = None
self.add_dc_cb = None
self._tree_view_item = None
# Graphic elements ----------------------------------------------------
_dc_parameters_widget = QtGui.QWidget(self)
self._data_path_widget = DataPathWidget(_dc_parameters_widget)
self._acq_widget = AcquisitionWidget(_dc_parameters_widget,
layout = 'horizontal')
#self._acq_widget.setFixedHeight(170)
self._processing_widget = ProcessingWidget(_dc_parameters_widget)
_snapshot_widget = QtGui.QWidget(self)
self.position_widget = uic.loadUi(os.path.join(os.path.dirname(__file__),
'ui_files/Qt4_snapshot_widget_layout.ui'))
# Layout --------------------------------------------------------------
_dc_parameters_widget_layout = QtGui.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(0)
_snapshots_vlayout = QtGui.QVBoxLayout(_snapshot_widget)
_snapshots_vlayout.addWidget(self.position_widget)
_snapshots_vlayout.setContentsMargins(0, 0, 0, 0)
_snapshots_vlayout.setSpacing(2)
_snapshots_vlayout.addStretch(10)
_main_hlayout = QtGui.QHBoxLayout(self)
_main_hlayout.addWidget(_dc_parameters_widget)
_main_hlayout.addWidget(_snapshot_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 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 acq_parameters_changed(self):
if self._tree_view_item is None:
#TODO fix this
return
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._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)
if data_collection.acquisitions[0].acquisition_parameters.\
centred_position.snapshot_image:
image = data_collection.acquisitions[0].\
acquisition_parameters.centred_position.snapshot_image
ration = image.height() / float(image.width())
image = image.scaled(400, 400 * ration, QtCore.Qt.KeepAspectRatio)
self.position_widget.svideo.setPixmap(QtGui.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 AcquisitionWidget(QtGui.QWidget):
"""
Descript. :
"""
acqParametersChangedSignal = QtCore.pyqtSignal()
madEnergySelectedSignal = QtCore.pyqtSignal(str, float, bool)
def __init__(self, parent = None, name = None, fl = 0, acq_params = None,
path_template = None, layout = 'horizontal'):
"""
Descript. :
"""
QtGui.QWidget.__init__(self, parent, QtCore.Qt.WindowFlags(fl))
if name is not None:
self.setObjectName(name)
# Hardware objects ----------------------------------------------------
self._beamline_setup_hwobj = None
# Internal variables --------------------------------------------------
self.previous_energy = 0
self.enable_parameter_update = True
# Properties ----------------------------------------------------------
# Signals -------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Graphic elements ----------------------------------------------------
if acq_params is None:
self._acquisition_parameters = queue_model_objects.\
AcquisitionParameters()
else:
self._acquisition_parameters = acq_params
if path_template is None:
self._path_template = queue_model_objects.PathTemplate()
else:
self._path_template = path_template
self._acquisition_mib = DataModelInputBinder(self._acquisition_parameters)
if layout == "horizontal":
self.acq_widget_layout = uic.loadUi(os.path.join(\
os.path.dirname(__file__),
"ui_files/Qt4_acquisition_widget_horizontal_layout.ui"))
else:
self.acq_widget_layout = uic.loadUi(os.path.join(\
os.path.dirname(__file__),
"ui_files/Qt4_acquisition_widget_vertical_layout.ui"))
# Layout --------------------------------------------------------------
__main_vlayout = QtGui.QVBoxLayout(self)
__main_vlayout.addWidget(self.acq_widget_layout)
__main_vlayout.setSpacing(0)
__main_vlayout.setContentsMargins(0, 0, 0, 0)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self.acq_widget_layout.energies_combo.activated.connect(\
self.energy_selected)
self.acq_widget_layout.mad_cbox.toggled.connect(\
self.use_mad)
self.acq_widget_layout.inverse_beam_cbx.toggled.connect(\
self.set_use_inverse_beam)
self.acq_widget_layout.first_image_ledit.textChanged.connect(\
self.first_image_ledit_change)
self.acq_widget_layout.num_images_ledit.textChanged.connect(\
self.num_images_ledit_change)
self.acq_widget_layout.detector_roi_mode_combo.activated.connect(\
self.detector_roi_mode_changed)
overlap_ledit = self.acq_widget_layout.findChild(\
QtGui.QLineEdit, "overlap_ledit")
if overlap_ledit is not None:
overlap_ledit.textChanged.connect(self.overlap_changed)
self.acq_widget_layout.subwedge_size_ledit.textChanged.connect(\
self.subwedge_size_ledit_change)
# Other ---------------------------------------------------------------
self.acq_widget_layout.subwedge_size_ledit.setDisabled(True)
self.acq_widget_layout.energies_combo.setDisabled(True)
self.acq_widget_layout.energies_combo.addItems(['ip: -', 'pk: -', 'rm1: -', 'rm2: -'])
self.osc_start_validator = QtGui.QDoubleValidator(-10000, 10000, 4, self)
self.osc_range_validator = QtGui.QDoubleValidator(-10000, 10000, 4, self)
self.kappa_validator = QtGui.QDoubleValidator(0, 360, 4, self)
self.kappa_phi_validator = QtGui.QDoubleValidator(0, 360, 4, self)
self.energy_validator = QtGui.QDoubleValidator(0, 25, 4, self)
self.resolution_validator = QtGui.QDoubleValidator(0, 15, 3, self)
self.transmission_validator = QtGui.QDoubleValidator(0, 100, 3, self)
self.exp_time_validator = QtGui.QDoubleValidator(0.0001, 10000, 4, self)
self.first_img_validator = QtGui.QIntValidator(0, 99999, self)
self.num_img_validator = QtGui.QIntValidator(1, 99999, self)
self.disable_inverse_beam(True)
self.acq_widget_layout.detector_roi_mode_label.setEnabled(False)
self.acq_widget_layout.detector_roi_mode_combo.setEnabled(False)
def set_enable_parameter_update(self, state):
self.enable_parameter_update = state
def update_osc_start(self, new_value):
"""
Descript. :
"""
if self.enable_parameter_update:
osc_start_value = 0
try:
osc_start_value = round(float(new_value), 2)
except TypeError:
pass
self.update_num_image_limits()
self.acq_widget_layout.osc_start_ledit.setText("%.2f" % osc_start_value)
#self._acquisition_parameters.osc_start = osc_start_value
def update_kappa(self, new_value):
if self.enable_parameter_update:
self.acq_widget_layout.kappa_ledit.setText("%.2f" % float(new_value))
#self._acquisition_parameters.kappa = float(new_value)
def update_kappa_phi(self, new_value):
if self.enable_parameter_update:
self.acq_widget_layout.kappa_phi_ledit.setText("%.2f" % float(new_value))
#self._acquisition_parameters.kappa_phi = float(new_value)
def use_osc_start(self, state):
self.acq_widget_layout.osc_start_label.setEnabled(state)
self.acq_widget_layout.osc_start_ledit.setEnabled(state)
def use_kappa(self, state):
self.acq_widget_layout.kappa_label.setEnabled(state)
self.acq_widget_layout.kappa_ledit.setEnabled(state)
def use_kappa_phi(self, state):
self.acq_widget_layout.kappa_phi_label.setEnabled(state)
self.acq_widget_layout.kappa_phi_ledit.setEnabled(state)
def set_beamline_setup(self, beamline_setup):
"""
Descript. :
"""
self._beamline_setup_hwobj = beamline_setup
limits_dict = self._beamline_setup_hwobj.get_acquisition_limit_values()
if 'osc_range' in limits_dict:
limits = tuple(map(float, limits_dict['osc_range'].split(',')))
(lower, upper) = limits
self.osc_start_validator.setRange(lower, upper, 4)
self.osc_range_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update('osc_start',
self.acq_widget_layout.osc_start_ledit,
float,
self.osc_start_validator)
self._acquisition_mib.bind_value_update('osc_range',
self.acq_widget_layout.osc_range_ledit,
float,
self.osc_range_validator)
if self.acq_widget_layout.findChild(QtGui.QLineEdit, "kappa_ledit"):
if 'kappa' in limits_dict:
limits = tuple(map(float, limits_dict['kappa'].split(',')))
(lower, upper) = limits
self.kappa_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update('kappa',
self.acq_widget_layout.kappa_ledit, float, self.kappa_validator)
if self.acq_widget_layout.findChild(QtGui.QLineEdit, "kappa_phi_ledit"):
if 'kappa_phi' in limits_dict:
limits = tuple(map(float, limits_dict['kappa_phi'].split(',')))
(lower, upper) = limits
self.kappa_phi_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update('kappa_phi',
self.acq_widget_layout.kappa_phi_ledit, float, self.kappa_phi_validator)
if 'exposure_time' in limits_dict:
limits = tuple(map(float, limits_dict['exposure_time'].split(',')))
(lower, upper) = limits
self.exp_time_valdidator.setRange(lower, upper, 5)
self._acquisition_mib.bind_value_update('exp_time',
self.acq_widget_layout.exp_time_ledit,
float,
self.exp_time_validator)
if 'number_of_images' in limits_dict:
limits = tuple(map(int, limits_dict['number_of_images'].split(',')))
(lower, upper) = limits
self.num_img_valdidator.setRange(lower, upper)
self.first_img_valdidator.setRange(lower, upper)
self._acquisition_mib.bind_value_update('first_image',
self.acq_widget_layout.first_image_ledit,
int,
self.first_img_validator)
self._acquisition_mib.bind_value_update('num_images',
self.acq_widget_layout.num_images_ledit,
int,
self.num_img_validator)
num_passes = self.acq_widget_layout.findChild(QtGui.QLineEdit, "num_passes_ledit")
if num_passes:
self._acquisition_mib.\
bind_value_update('num_passes', num_passes, int,
QtGui.QIntValidator(1, 1000, self))
overlap_ledit = self.acq_widget_layout.findChild(QtGui.QLineEdit, "overlap_ledit")
if overlap_ledit:
self._acquisition_mib.\
bind_value_update('overlap', overlap_ledit, float,
QtGui.QDoubleValidator(-1000, 1000, 2, self))
self._acquisition_mib.\
bind_value_update('energy',
self.acq_widget_layout.energy_ledit,
float,
self.energy_validator)
self._acquisition_mib.\
bind_value_update('transmission',
self.acq_widget_layout.transmission_ledit,
float,
self.transmission_validator)
self._acquisition_mib.\
bind_value_update('resolution',
self.acq_widget_layout.resolution_ledit,
float,
self.resolution_validator)
self._acquisition_mib.\
bind_value_update('inverse_beam',
self.acq_widget_layout.inverse_beam_cbx,
bool,
None)
self._acquisition_mib.\
bind_value_update('shutterless',
self.acq_widget_layout.shutterless_cbx,
bool,
None)
"""self.acq_widget_layout.osc_start_ledit.setToolTip(\
"Oscillation start limits %0.2f : %0.2f" %(\
self.osc_start_validator.bottom(),
self.osc_start_validator.top()))
self.acq_widget_layout.osc_range_ledit.setToolTip(\
"Oscillation range limits %0.2f : %0.2f" %(\
self.osc_ra_validator.bottom(),
self.osc_start_validator.top()))"""
self.set_tunable_energy(beamline_setup.tunable_wavelength())
has_shutter_less = self._beamline_setup_hwobj.detector_has_shutterless()
self.acq_widget_layout.shutterless_cbx.setEnabled(has_shutter_less)
self.acq_widget_layout.shutterless_cbx.setChecked(has_shutter_less)
if self._beamline_setup_hwobj.disable_num_passes():
num_passes = self.acq_widget_layout.findChild(QtGui.QLineEdit, "num_passes_ledit")
if num_passes:
num_passes.setDisabled(True)
def first_image_ledit_change(self, new_value):
"""
Descript. :
"""
if str(new_value).isdigit():
self._path_template.start_num = int(new_value)
widget = self.acq_widget_layout.first_image_ledit
self.acqParametersChangedSignal.emit()
def num_images_ledit_change(self, new_value):
"""
Descript. :
"""
if str(new_value).isdigit():
self._path_template.num_files = int(new_value)
widget = self.acq_widget_layout.num_images_ledit
self.acqParametersChangedSignal.emit()
def overlap_changed(self, new_value):
"""
Descript. :
"""
if self._beamline_setup_hwobj:
has_shutter_less = self._beamline_setup_hwobj.detector_has_shutterless()
else:
has_shutter_less = True
if has_shutter_less:
try:
new_value = float(new_value)
except ValueError:
pass
if new_value != 0:
self.acq_widget_layout.shutterless_cbx.setEnabled(False)
self.acq_widget_layout.shutterless_cbx.setChecked(False)
self._acquisition_parameters.shutterless = False
else:
self.acq_widget_layout.shutterless_cbx.setEnabled(True)
self.acq_widget_layout.shutterless_cbx.setChecked(True)
self._acquisition_parameters.shutterless = True
def use_mad(self, state):
"""
Descript. :
"""
self.acq_widget_layout.energies_combo.setEnabled(state)
if state:
(name, energy) = self.get_mad_energy()
if energy != 0:
self.set_energy(energy, 0)
self.madEnergySelectedSignal.emit(name, energy, state)
else:
self.set_energy(self.previous_energy, 0)
energy = self._beamline_setup_hwobj.energy_hwobj.getCurrentEnergy()
self.madEnergySelectedSignal.emit('', self.previous_energy, state)
def set_use_inverse_beam(self, state):
"""
Descript. :
"""
if state:
self.acq_widget_layout.subwedge_size_ledit.setEnabled(True)
else:
self.acq_widget_layout.subwedge_size_ledit.setDisabled(True)
def use_inverse_beam(self):
"""
Descript. :
"""
return self.acq_widget_layout.inverse_beam_cbx.isChecked()
def get_num_subwedges(self):
"""
Descript. :
"""
return int(self.acq_widget_layout.subwedge_size_ledit.text())
def subwedge_size_ledit_change(self, new_value):
"""
Descript. :
"""
accept_input = False
if str(new_value).isdigit() and \
int(new_value) < self._acquisition_parameters.num_images:
accept_input = True
if accept_input:
Qt4_widget_colors.set_widget_color(
self.acq_widget_layout.subwedge_size_ledit,
Qt4_widget_colors.WHITE,
QtGui.QPalette.Base)
else:
Qt4_widget_colors.set_widget_color(\
self.acq_widget_layout.subwedge_size_ledit,
Qt4_widget_colors.LIGHT_RED,
QtGui.QPalette.Base)
def get_mad_energy(self):
"""
Descript. :
"""
energy_str = str(self.acq_widget_layout.energies_combo.currentText())
(name, value) = energy_str.split(':')
name = name.strip()
value = value.strip()
value = 0 if (value == '-') else float(value)
return (name, value)
def set_energies(self, energy_scan_result):
"""
Descript. :
"""
self.acq_widget_layout.energies_combo.clear()
inflection = ('ip: %.4f' % energy_scan_result.inflection) if \
energy_scan_result.inflection else 'ip: -'
peak = ('pk: %.4f' % energy_scan_result.peak) if \
energy_scan_result.peak else 'pk: -'
first_remote = ('rm1: %.4f' % energy_scan_result.first_remote) if \
energy_scan_result.first_remote else 'rm1: -'
second_remote = ('rm2: %.4f' % energy_scan_result.second_remote) if \
energy_scan_result.second_remote else 'rm2: -'
self.acq_widget_layout.energies_combo.\
addItems([inflection, peak, first_remote, second_remote])
def energy_selected(self, index):
"""
Descript. :
"""
if self.acq_widget_layout.mad_cbox.isChecked():
(name, energy) = self.get_mad_energy()
if energy != 0:
self.set_energy(energy, 0)
self.madEnergySelectedSignal.emit(name, energy, True)
def set_energy(self, energy, wav):
"""
Descript. :
"""
self.acq_widget_layout.energy_ledit.setText("%.4f" % float(energy))
def update_transmission(self, transmission):
"""
Descript. :
"""
if self.enable_parameter_update:
self.acq_widget_layout.transmission_ledit.setText("%.2f" % float(transmission))
def update_resolution(self, resolution):
"""
Descript. :
"""
if self.enable_parameter_update:
self.acq_widget_layout.resolution_ledit.setText("%.3f" % float(resolution))
def update_energy_limits(self, limits):
if limits:
self.energy_validator.setBottom(limits[0])
self.energy_validator.setTop(limits[1])
self.acq_widget_layout.energy_ledit.setToolTip(\
"Energy limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_transmission_limits(self, limits):
if limits:
self.transmission_validator.setBottom(limits[0])
self.transmission_validator.setTop(limits[1])
self.acq_widget_layout.transmission_ledit.setToolTip(\
"Transmission limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_resolution_limits(self, limits):
if limits:
self.resolution_validator.setBottom(limits[0])
self.resolution_validator.setTop(limits[1])
self.acq_widget_layout.resolution_ledit.setToolTip(
"Resolution limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_detector_exp_time_limits(self, limits):
if limits:
self.exp_time_validator.setRange(limits[0], limits[1], 4)
self.acq_widget_layout.exp_time_ledit.setToolTip(
"Exposure time limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_osc_range_limits(self, exp_time):
osc_range = 40.0 * exp_time
self.osc_range_validator.setRange(0, osc_range, 4)
self.acq_widget_layout.osc_range_ledit.setToolTip(
"Oscillation range limits %0.2f : %0.2f" %(0, osc_range))
self._acquisition_mib.validate_all()
def update_num_image_limits(self):
try:
osc_start = float(self.acq_widget_layout.osc_start_ledit.text())
osc_range = float(self.acq_widget_layout.osc_range_ledit.text())
except ValueError:
return
if osc_range == 0:
return
if self._beamline_setup_hwobj.diffractometer_hwobj.in_plate_mode():
num_image_limit = int((self.osc_start_validator.top() - osc_start) / osc_range)
else:
num_image_limit = 99999
self.num_img_validator.setTop(num_image_limit)
self.acq_widget_layout.num_images_ledit.setToolTip(\
"Number of frames limit : %d" % num_image_limit)
self._acquisition_mib.validate_all()
def init_detector_roi_modes(self):
if self._beamline_setup_hwobj is not None:
roi_modes = self._beamline_setup_hwobj._get_roi_modes()
if (len(roi_modes) > 0 and
self.acq_widget_layout.detector_roi_mode_combo.count() == 0):
for roi_mode in roi_modes:
self.acq_widget_layout.detector_roi_mode_combo.\
addItem(roi_mode)
self.acq_widget_layout.detector_roi_mode_label.setEnabled(True)
self.acq_widget_layout.detector_roi_mode_combo.setEnabled(True)
#self._acquisition_mib.bind_value_update('detector_roi_mode',
# self.acq_widget_layout.detector_roi_mode_combo,
## str,
# None)
def update_detector_roi_mode(self, roi_mode_index):
if self.acq_widget_layout.detector_roi_mode_combo.count() > 0:
self.acq_widget_layout.detector_roi_mode_combo.\
setCurrentIndex(roi_mode_index)
def detector_roi_mode_changed(self, roi_mode_index):
if self._beamline_setup_hwobj is not None:
self._beamline_setup_hwobj.detector_hwobj.set_roi_mode(roi_mode_index)
def update_data_model(self, acquisition_parameters, path_template):
"""
Descript. :
"""
self._acquisition_parameters = acquisition_parameters
self._path_template = path_template
self._acquisition_mib.set_model(acquisition_parameters)
#Update mad widgets
mad = True if self._path_template.mad_prefix != '' else False
if mad:
mad_prefix = self._path_template.mad_prefix
index = MAD_ENERGY_COMBO_NAMES[mad_prefix]
self.acq_widget_layout.energies_combo.setCurrentIndex(index)
self.acq_widget_layout.mad_cbox.setChecked(True)
self.acq_widget_layout.energies_combo.setEnabled(True)
else:
self.acq_widget_layout.mad_cbox.setChecked(False)
self.acq_widget_layout.energies_combo.setEnabled(False)
self.acq_widget_layout.energies_combo.setCurrentIndex(0)
def set_tunable_energy(self, state):
"""
Descript. :
"""
self.acq_widget_layout.energy_ledit.setEnabled(state)
self.acq_widget_layout.mad_cbox.setEnabled(state)
self.acq_widget_layout.energies_combo.setEnabled(state)
def disable_inverse_beam(self, state):
"""
Descript. :
"""
self.acq_widget_layout.inverse_beam_cbx.setDisabled(state)
self.acq_widget_layout.subwedge_size_label.setDisabled(state)
self.acq_widget_layout.subwedge_size_ledit.setDisabled(state)
def hide_aperture(self, state):
"""
Descript. :
"""
pass
class AcquisitionWidgetSimple(QtGui.QWidget):
"""
Descript. :
"""
def __init__(self, parent = None, name = None, fl = 0, acq_params = None,
path_template = None, layout = None):
"""
Descript. :
"""
QtGui.QWidget.__init__(self, parent, QtCore.Qt.WindowFlags(fl))
if name is not None:
self.setObjectName(name)
# Hardware objects ----------------------------------------------------
self._beamline_setup_hwobj = None
# Internal variables --------------------------------------------------
self.enable_parameter_update = True
# Properties ----------------------------------------------------------
# Signals -------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Graphic elements ----------------------------------------------------
if acq_params is None:
self._acquisition_parameters = queue_model_objects.AcquisitionParameters()
else:
self._acquisition_parameters = acq_params
if path_template is None:
self._path_template = queue_model_objects.PathTemplate()
else:
self._path_template = path_template
self._acquisition_mib = DataModelInputBinder(self._acquisition_parameters)
self.acq_widget_layout = uic.loadUi(os.path.join(os.path.dirname(\
__file__), "ui_files/Qt4_acquisition_widget_vertical_simple_layout.ui"))
# Layout --------------------------------------------------------------
main_layout = QtGui.QVBoxLayout(self)
main_layout.addWidget(self.acq_widget_layout)
main_layout.setSpacing(0)
main_layout.setContentsMargins(0, 0, 0, 0)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self.acq_widget_layout.num_images_cbox.activated.connect(\
self.update_num_images)
self.acq_widget_layout.detector_roi_mode_combo.activated.connect(\
self.detector_roi_mode_changed)
# Other ---------------------------------------------------------------
self.energy_validator = QtGui.QDoubleValidator(0, 25, 5, self)
self.resolution_validator = QtGui.QDoubleValidator(0, 15, 3, self)
self.transmission_validator = QtGui.QDoubleValidator(0, 100, 3, self)
self.exp_time_validator = QtGui.QDoubleValidator(0, 10000, 5, self)
#self.acq_widget_layout.osc_start_ledit.setEnabled(False)
#self.acq_widget_layout.kappa_ledit.setEnabled(False)
#self.acq_widget_layout.kappa_phi_ledit.setEnabled(False)
self.acq_widget_layout.num_images_cbox.setCurrentIndex(1)
self.acq_widget_layout.detector_roi_mode_label.setEnabled(False)
self.acq_widget_layout.detector_roi_mode_combo.setEnabled(False)
def set_enable_parameter_update(self, state):
self.enable_parameter_update = state
def update_osc_start(self, new_value):
"""
Descript. :
"""
if self.enable_parameter_update:
osc_start_value = 0
try:
osc_start_value = round(float(new_value), 2)
except TypeError:
pass
self.acq_widget_layout.osc_start_ledit.setText(\
"%.2f" % osc_start_value)
#self._acquisition_parameters.osc_start = osc_start_value
def update_kappa(self, new_value):
"""
Descript. :
"""
if self.enable_parameter_update:
self.acq_widget_layout.kappa_ledit.setText(\
"%.2f" % float(new_value))
def update_kappa_phi(self, new_value):
"""
Descript. :
"""
if self.enable_parameter_update:
self.acq_widget_layout.kappa_phi_ledit.setText(\
"%.2f" % float(new_value))
def use_kappa(self, state):
"""
Descript. :
"""
self.acq_widget_layout.kappa_ledit.setDisabled(state)
def use_kappa_phi(self, state):
"""
Descript. :
"""
self.acq_widget_layout.kappa_phi_ledit.setDisabled(state)
def update_num_images(self, index = None, num_images = None):
"""
Descript. :
"""
if index is not None:
if index is 0:
self._acquisition_parameters.num_images = 1
self._path_template.num_files = 1
elif index is 1:
self._acquisition_parameters.num_images = 2
self._path_template.num_files = 2
elif index is 2:
self._acquisition_parameters.num_images = 4
self._path_template.num_files = 4
if num_images:
if self.acq_widget_layout.num_images_cbox.count() > 3:
self.acq_widget_layout.num_images_cbox.removeItem(4)
if num_images is 1:
self.acq_widget_layout.num_images_cbox.setCurrentIndex(0)
elif num_images is 2:
self.acq_widget_layout.num_images_cbox.setCurrentIndex(1)
elif num_images is 4:
self.acq_widget_layout.num_images_cbox.setCurrentIndex(2)
else:
self.acq_widget_layout.num_images_cbox.addItem(str(num_images))
self.acq_widget_layout.num_images_cbox.setCurrenIndex(3)
self._path_template.num_files = num_images
def use_mad(self, state):
"""
Descript. :
"""
pass
def get_mad_energy(self):
"""
Descript. :
"""
pass
def set_energies(self, energy_scan_result):
"""
Descript. :
"""
pass
def energy_selected(self, index):
"""
Descript. :
"""
pass
def set_beamline_setup(self, beamline_setup):
"""
Descript. :
"""
self._beamline_setup_hwobj = beamline_setup
limits_dict = self._beamline_setup_hwobj.get_acquisition_limit_values()
if 'osc_range' in limits_dict:
limits = tuple(map(float, limits_dict['osc_range'].split(',')))
(lower, upper) = limits
osc_start_validator = QtGui.QDoubleValidator(lower, upper, 4, self)
osc_range_validator = QtGui.QDoubleValidator(lower, upper, 4, self)
else:
osc_start_validator = QtGui.QDoubleValidator(-10000, 10000, 4, self)
osc_range_validator = QtGui.QDoubleValidator(-10000, 10000, 4, self)
self._acquisition_mib.bind_value_update('osc_start',
self.acq_widget_layout.osc_start_ledit,
float,
osc_start_validator)
self._acquisition_mib.bind_value_update('osc_range',
self.acq_widget_layout.osc_range_ledit,
float,
osc_range_validator)
kappa_validator = QtGui.QDoubleValidator(-30, 180, 2, self)
self._acquisition_mib.bind_value_update('kappa',
self.acq_widget_layout.kappa_ledit,
float,
kappa_validator)
kappa_phi_validator = QtGui.QDoubleValidator(-30, 180, 2, self)
self._acquisition_mib.bind_value_update('kappa_phi',
self.acq_widget_layout.kappa_phi_ledit,
float,
kappa_phi_validator)
self._acquisition_mib.bind_value_update('exp_time',
self.acq_widget_layout.exp_time_ledit,
float,
self.exp_time_validator)
self._acquisition_mib.\
bind_value_update('energy',
self.acq_widget_layout.energy_ledit,
float,
self.energy_validator)
self.acq_widget_layout.energy_ledit.setToolTip(\
"Energy limits %0.3f : %0.3f" % \
(self.energy_validator.bottom(), self.energy_validator.top()))
self._acquisition_mib.\
bind_value_update('transmission',
self.acq_widget_layout.transmission_ledit,
float,
self.transmission_validator)
self._acquisition_mib.\
bind_value_update('resolution',
self.acq_widget_layout.resolution_ledit,
float,
self.resolution_validator)
self.set_tunable_energy(beamline_setup.tunable_wavelength())
def set_energy(self, energy, wav):
"""
Descript. :
"""
#self._acquisition_parameters.energy = energy
if self.enable_parameter_update:
self.acq_widget_layout.energy_ledit.setText(\
"%.4f" % float(energy))
def update_transmission(self, transmission):
"""
Descript. :
"""
if self.enable_parameter_update:
self.acq_widget_layout.transmission_ledit.setText(\
"%.2f" % float(transmission))
#self._acquisition_parameters.transmission = float(transmission)
def update_resolution(self, resolution):
"""
Descript. :
"""
if self.enable_parameter_update:
self.acq_widget_layout.resolution_ledit.setText("%.3f" % float(resolution))
#self._acquisition_parameters.resolution = float(resolution)
def update_energy_limits(self, limits):
"""
Descript. :
"""
if limits:
self.energy_validator.setBottom(limits[0])
self.energy_validator.setTop(limits[1])
self.acq_widget_layout.energy_ledit.setToolTip(
"Energy limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_transmission_limits(self, limits):
"""
Descript. :
"""
if limits:
self.transmission_validator.setBottom(limits[0])
self.transmission_validator.setTop(limits[1])
self.acq_widget_layout.transmission_ledit.setToolTip(
"Transmission limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_resolution_limits(self, limits):
"""
Descript. :
"""
if limits:
self.resolution_validator.setBottom(limits[0])
self.resolution_validator.setTop(limits[1])
self.acq_widget_layout.resolution_ledit.setToolTip(
"Resolution limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_detector_exp_time_limits(self, limits):
"""
Descript. :
"""
if limits:
self.exp_time_validator.setBottom(limits[0])
self.exp_time_validator.setTop(limits[1])
self.acq_widget_layout.exp_time_ledit.setToolTip(
"Exposure time limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def init_detector_roi_modes(self):
"""
Descript. :
"""
if self._beamline_setup_hwobj is not None:
roi_modes = self._beamline_setup_hwobj._get_roi_modes()
if (len(roi_modes) > 0 and
self.acq_widget_layout.detector_roi_mode_combo.count() == 0):
for roi_mode in roi_modes:
self.acq_widget_layout.detector_roi_mode_combo.\
addItem(roi_mode)
self.acq_widget_layout.detector_roi_mode_label.setEnabled(True)
self.acq_widget_layout.detector_roi_mode_combo.setEnabled(True)
#self._acquisition_mib.bind_value_update('detector_roi_mode',
# self.acq_widget_layout.detector_roi_mode_combo,
# str,
# None)
def update_detector_roi_mode(self, roi_mode_index):
"""
Descript. :
"""
if self.acq_widget_layout.detector_roi_mode_combo.count() > 0:
self.acq_widget_layout.detector_roi_mode_combo.\
setCurrentIndex(roi_mode_index)
def detector_roi_mode_changed(self, roi_mode_index):
"""
Descript. :
"""
if self._beamline_setup_hwobj is not None:
self._beamline_setup_hwobj.detector_hwobj.set_roi_mode(roi_mode_index)
def update_data_model(self, acquisition_parameters, path_template):
"""
Descript. :
"""
self._acquisition_parameters = acquisition_parameters
self._acquisition_mib.set_model(acquisition_parameters)
self._path_template = path_template
self.update_num_images(None, acquisition_parameters.num_images)
def set_tunable_energy(self, state):
"""
Descript. :
"""
self.acq_widget_layout.energy_ledit.setEnabled(state)
def use_osc_start(self, state):
"""
Descript. :
"""
self.acq_widget_layout.osc_start_ledit.setEnabled(state)
def check_parameter_conflict(self):
"""
Descript. :
"""
return len(self._acquisition_mib.validate_all()) > 0
class AdvancedParametersWidget(QtGui.QWidget):
def __init__(self, parent = None, name = "advanced_parameters_widget"):
QtGui.QWidget.__init__(self, parent)
# Hardware objects ----------------------------------------------------
self._queue_model_hwobj = None
self._beamline_setup_hwobj = None
# Internal values -----------------------------------------------------
self._data_collection = None
self._tree_view_item = None
# Properties ----------------------------------------------------------
# Signals -------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Graphic elements ----------------------------------------------------
_dc_parameters_widget = QtGui.QWidget(self)
self._data_path_widget = DataPathWidget(_dc_parameters_widget)
self._acq_widget = AcquisitionWidget(_dc_parameters_widget,
layout = 'horizontal')
#self._acq_widget.setFixedHeight(170)
_snapshot_widget = QtGui.QWidget(self)
self.position_widget = uic.loadUi(os.path.join(os.path.dirname(__file__),
'ui_files/Qt4_snapshot_widget_layout.ui'))
# Layout --------------------------------------------------------------
_dc_parameters_widget_layout = QtGui.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)
_snapshots_vlayout = QtGui.QVBoxLayout(_snapshot_widget)
_snapshots_vlayout.addWidget(self.position_widget)
_snapshots_vlayout.setContentsMargins(0, 0, 0, 0)
_snapshots_vlayout.setSpacing(2)
_snapshots_vlayout.addStretch(10)
_main_hlayout = QtGui.QHBoxLayout(self)
_main_hlayout.addWidget(_dc_parameters_widget)
_main_hlayout.addWidget(_snapshot_widget)
_main_hlayout.setSpacing(2)
_main_hlayout.setContentsMargins(2, 2, 2, 2)
_main_hlayout.addStretch(0)
# 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.handle_path_conflict)
# 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.num_images_ledit.setEnabled(False)
self._acq_widget.acq_widget_layout.inverse_beam_cbx.hide()
self._acq_widget.acq_widget_layout.shutterless_cbx.hide()
self._acq_widget.acq_widget_layout.subwedge_size_label.hide()
self._acq_widget.acq_widget_layout.subwedge_size_ledit.hide()
def set_beamline_setup(self, bl_setup):
self._beamline_setup_hwobj = bl_setup
self._acq_widget.set_beamline_setup(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):
if self._tree_view_item:
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 path_conflict:
logging.getLogger("user_level_log").\
error('The current path settings will overwrite data' +\
' from another task. Correct the problem before collecting')
Qt4_widget_colors.set_widget_color(widget, Qt4_widget_colors.LIGHT_RED)
else:
Qt4_widget_colors.set_widget_color(widget, Qt4_widget_colors.LIGHT_WHITE)
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._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)
def populate_widget(self, tree_view_item):
self._tree_view_item = tree_view_item
advanced_model = tree_view_item.get_model()
self._data_collection = advanced_model.reference_image_collection
executed = self._data_collection.is_executed()
self._acq_widget.setEnabled(not executed)
self._data_path_widget.setEnabled(not executed)
image = advanced_model.grid_object.get_snapshot()
try:
image = image.scaled(427, 320, QtCore.Qt.KeepAspectRatio)
self.position_widget.svideo.setPixmap(QtGui.QPixmap(image))
except:
pass
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)
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 AcquisitionWidget(QWidget):
acqParametersChangedSignal = pyqtSignal(list)
madEnergySelectedSignal = pyqtSignal(str, float, bool)
def __init__(self,
parent=None,
name=None,
fl=0,
acq_params=None,
path_template=None,
layout='horizontal'):
QWidget.__init__(self, parent, Qt.WindowFlags(fl))
if name is not None:
self.setObjectName(name)
# Hardware objects ----------------------------------------------------
self._beamline_setup_hwobj = None
self._diffractometer_hwobj = None
# Internal variables --------------------------------------------------
self.previous_energy = 0
# If the acq. widget is used with grids then total osc range is not
# equal to num_images * osc_range, but num_images_per_line * osc_range
# For grids the osc total range is updated when a grid is selected
self.grid_mode = False
# Properties ----------------------------------------------------------
# Signals -------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Graphic elements ----------------------------------------------------
if acq_params is None:
self._acquisition_parameters = queue_model_objects.\
AcquisitionParameters()
else:
self._acquisition_parameters = acq_params
if path_template is None:
self._path_template = queue_model_objects.PathTemplate()
else:
self._path_template = path_template
self._acquisition_mib = DataModelInputBinder(
self._acquisition_parameters)
if layout == "horizontal":
self.acq_widget_layout = loadUi(os.path.join(\
os.path.dirname(__file__),
"ui_files/Qt4_acquisition_widget_horizontal_layout.ui"))
self.use_osc_start(False)
else:
self.acq_widget_layout = loadUi(os.path.join(\
os.path.dirname(__file__),
"ui_files/Qt4_acquisition_widget_vertical_layout.ui"))
# Layout --------------------------------------------------------------
__main_vlayout = QVBoxLayout(self)
__main_vlayout.addWidget(self.acq_widget_layout)
__main_vlayout.setSpacing(0)
__main_vlayout.setContentsMargins(0, 0, 0, 0)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self.acq_widget_layout.osc_start_cbox.stateChanged.connect(\
self.fix_osc_start)
self.acq_widget_layout.exp_time_ledit.textChanged.connect(\
self.exposure_time_ledit_changed)
self.acq_widget_layout.first_image_ledit.textChanged.connect(\
self.first_image_ledit_change)
self.acq_widget_layout.num_images_ledit.textChanged.connect(\
self.num_images_ledit_change)
self.acq_widget_layout.detector_roi_mode_combo.activated.connect(\
self.detector_roi_mode_changed)
self.acq_widget_layout.energies_combo.activated.connect(\
self.energy_selected)
self.acq_widget_layout.mad_cbox.toggled.connect(\
self.use_mad)
self.acq_widget_layout.osc_start_ledit.textEdited.connect(\
self.osc_start_ledit_changed)
self.acq_widget_layout.osc_range_ledit.textEdited.connect(\
self.osc_range_per_frame_ledit_changed)
self.acq_widget_layout.osc_total_range_ledit.textEdited.connect(\
self.osc_total_range_ledit_changed)
self.acq_widget_layout.energy_ledit.textEdited.connect(\
self.energy_ledit_changed)
self.acq_widget_layout.transmission_ledit.textEdited.connect(\
self.transmission_ledit_changed)
self.acq_widget_layout.resolution_ledit.textEdited.connect(\
self.resolution_ledit_changed)
self.acq_widget_layout.kappa_ledit.textEdited.connect(\
self.kappa_ledit_changed)
self.acq_widget_layout.kappa_phi_ledit.textEdited.connect(\
self.kappa_phi_ledit_changed)
if self.acq_widget_layout.findChild(QLineEdit, "overlap_ledit"):
self.acq_widget_layout.overlap_ledit.textChanged.connect(\
self.overlap_changed)
if self.acq_widget_layout.findChild(QCheckBox, "max_osc_range_cbx"):
self.acq_widget_layout.max_osc_range_cbx.toggled.connect(\
self.max_osc_range_toggled)
# Other ---------------------------------------------------------------
self.value_changed_list = []
self.acq_widget_layout.energies_combo.setDisabled(True)
self.acq_widget_layout.energies_combo.addItems(\
['ip: -', 'pk: -', 'rm1: -', 'rm2: -'])
self.osc_start_validator = QDoubleValidator(\
-10000, 10000, 4, self.acq_widget_layout.osc_start_ledit)
self.osc_range_per_frame_validator = QDoubleValidator(\
0, 10000, 4, self.acq_widget_layout.osc_range_ledit)
self.osc_total_range_validator = QDoubleValidator(\
0, 10000, 4, self.acq_widget_layout.osc_total_range_ledit)
self.kappa_validator = QDoubleValidator(\
0, 360, 4, self.acq_widget_layout.kappa_ledit)
self.kappa_phi_validator = QDoubleValidator(\
0, 360, 4, self.acq_widget_layout.kappa_phi_ledit)
self.energy_validator = QDoubleValidator(\
4, 25, 4, self.acq_widget_layout.energy_ledit)
self.resolution_validator = QDoubleValidator(\
0, 15, 3, self.acq_widget_layout.resolution_ledit)
self.transmission_validator = QDoubleValidator(\
0, 100, 3, self.acq_widget_layout.transmission_ledit)
self.exp_time_validator = QDoubleValidator(\
0.0001, 10000, 7, self.acq_widget_layout.exp_time_ledit)
self.first_img_validator = QIntValidator(\
0, 99999, self.acq_widget_layout.first_image_ledit)
self.num_img_validator = QIntValidator(\
1, 99999, self.acq_widget_layout.num_images_ledit)
self.acq_widget_layout.detector_roi_mode_label.setEnabled(False)
self.acq_widget_layout.detector_roi_mode_combo.setEnabled(False)
def use_max_osc_range(self, state):
self.acq_widget_layout.max_osc_range_cbx.setEnabled(state)
def fix_osc_start(self, state):
"""
Fix osc start, so the lineEdit do not change when osc is changed
"""
self.acq_widget_layout.osc_start_ledit.setEnabled(state)
def update_osc_start(self, new_value):
"""
Updates osc line edit
"""
if "osc_start" not in self.value_changed_list and \
not self.acq_widget_layout.osc_start_ledit.hasFocus() and \
not self.acq_widget_layout.osc_start_cbox.isChecked():
osc_start_value = 0
try:
osc_start_value = round(float(new_value), 2)
except TypeError:
pass
self.acq_widget_layout.osc_start_ledit.setText(
str(osc_start_value))
self._acquisition_parameters.osc_start = osc_start_value
#self.update_osc_total_range_limits()
#self.update_num_images_limits()
def osc_start_ledit_changed(self, osc_start):
"""Fixes osc start edit"""
if "osc_start" not in self.value_changed_list:
self.value_changed_list.append("osc_start")
self.update_osc_total_range_limits()
self.update_num_images_limits()
self.emit_acq_parameters_changed()
def update_osc_start_limits(self):
"""In the plate mode sets osc start limits"""
self.update_osc_total_range_limits()
self.update_num_images_limits()
def osc_range_per_frame_ledit_changed(self, new_value):
self.update_osc_total_range()
self.update_num_images_limits()
self.update_exp_time_limits()
self.emit_acq_parameters_changed()
def update_osc_range_per_frame_limits(self):
try:
max_osc_speed = self._diffractometer_hwobj.get_osc_max_speed()
top_limit = max_osc_speed * float(
self.acq_widget_layout.exp_time_ledit.text())
self.osc_range_per_frame_validator.setTop(top_limit)
tool_tip = "Oscillation range per frame limits 0 : " +\
"%0.4f\n4 digits precision." % top_limit
self.acq_widget_layout.osc_range_ledit.setToolTip(tool_tip)
self._acquisition_mib.validate_all()
except:
pass
def update_osc_total_range(self):
self.acq_widget_layout.osc_total_range_ledit.blockSignals(True)
if not self.grid_mode:
try:
self.acq_widget_layout.osc_total_range_ledit.setText(str(\
float(self.acq_widget_layout.osc_range_ledit.text()) * \
float(self.acq_widget_layout.num_images_ledit.text())))
except:
pass
self.acq_widget_layout.osc_total_range_ledit.blockSignals(False)
def update_total_exp_time(self):
try:
self.acq_widget_layout.total_exp_time_ledit.setText("%.2f" % \
(float(self.acq_widget_layout.exp_time_ledit.text()) * \
float(self.acq_widget_layout.num_images_ledit.text())))
except:
pass
def osc_total_range_ledit_changed(self, new_value):
if not self.grid_mode:
try:
num_images = int(float(new_value) / \
float(self.acq_widget_layout.osc_range_ledit.text()))
self.acq_widget_layout.num_images_ledit.blockSignals(True)
self.acq_widget_layout.num_images_ledit.setText("%d" %
num_images)
self._acquisition_parameters.num_images = num_images
self.acq_widget_layout.num_images_ledit.blockSignals(False)
except:
pass
self.emit_acq_parameters_changed()
def update_osc_total_range_limits(self, num_images=None):
"""Updates osc totol range. Limits are changed if a plate is used.
- For simple oscillation osc_range is defined by osc_start and
osc_start top limit.
- For mesh osc_range is defined by number of images per line
and osc in the middle of mesh
"""
if self._diffractometer_hwobj.in_plate_mode():
if hasattr(self.parent(), "set_osc_total_range"):
self.parent().set_osc_total_range(num_images)
self._acquisition_mib.validate_all()
def update_exp_time_limits(self):
try:
exp_time_limits = self._beamline_setup_hwobj.detector_hwobj.get_exposure_time_limits(
)
max_osc_speed = self._diffractometer_hwobj.get_osc_max_speed()
top_limit = float(
self.acq_widget_layout.osc_range_ledit.text()) / max_osc_speed
limits = (max(exp_time_limits[0], top_limit), exp_time_limits[1])
self.update_detector_exp_time_limits(limits)
except:
pass
def update_kappa(self, new_value):
"""
Updates kappa value
"""
if not self.acq_widget_layout.kappa_ledit.hasFocus() and \
new_value:
self.acq_widget_layout.kappa_ledit.setText(str(new_value))
self.emit_acq_parameters_changed()
def update_kappa_phi(self, new_value):
"""
Descript. :
"""
if not self.acq_widget_layout.kappa_phi_ledit.hasFocus() and \
new_value:
self.acq_widget_layout.kappa_phi_ledit.setText(str(new_value))
self.emit_acq_parameters_changed()
def use_osc_start(self, state):
"""
Descript. :
"""
self.acq_widget_layout.osc_start_cbox.setVisible(state)
self.acq_widget_layout.osc_start_label.setVisible(not state)
self.acq_widget_layout.osc_start_ledit.setEnabled(not state)
def use_kappa(self, state):
"""
Descript. :
"""
if self._diffractometer_hwobj is not None:
if self._diffractometer_hwobj.in_plate_mode():
state = False
self.acq_widget_layout.kappa_label.setEnabled(state)
self.acq_widget_layout.kappa_ledit.setEnabled(state)
self.acq_widget_layout.kappa_phi_label.setEnabled(state)
self.acq_widget_layout.kappa_phi_ledit.setEnabled(state)
def set_beamline_setup(self, beamline_setup):
"""
Descript. :
"""
self._beamline_setup_hwobj = beamline_setup
limits_dict = self._beamline_setup_hwobj.get_acquisition_limit_values()
self._diffractometer_hwobj = self._beamline_setup_hwobj.diffractometer_hwobj
if 'osc_range' in limits_dict:
limits = tuple(map(float, limits_dict['osc_range'].split(',')))
(lower, upper) = limits
self.osc_start_validator.setRange(lower, upper, 4)
self.osc_range_per_frame_validator.setRange(lower, upper, 4)
self.osc_total_range_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update(
'osc_start', self.acq_widget_layout.osc_start_ledit, float,
self.osc_start_validator)
self._acquisition_mib.bind_value_update(
'osc_range', self.acq_widget_layout.osc_range_ledit, float,
self.osc_range_per_frame_validator)
self._acquisition_mib.bind_value_update(
'osc_total_range', self.acq_widget_layout.osc_total_range_ledit,
float, self.osc_total_range_validator)
if 'kappa' in limits_dict:
limits = tuple(map(float, limits_dict['kappa'].split(',')))
(lower, upper) = limits
self.kappa_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update(
'kappa', self.acq_widget_layout.kappa_ledit, float,
self.kappa_validator)
if 'kappa_phi' in limits_dict:
limits = tuple(map(float, limits_dict['kappa_phi'].split(',')))
(lower, upper) = limits
self.kappa_phi_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update(
'kappa_phi', self.acq_widget_layout.kappa_phi_ledit, float,
self.kappa_phi_validator)
if 'exposure_time' in limits_dict:
limits = tuple(map(float, limits_dict['exposure_time'].split(',')))
(lower, upper) = limits
self.exp_time_validator.setRange(lower, upper, 6)
self._acquisition_mib.bind_value_update(
'exp_time', self.acq_widget_layout.exp_time_ledit, float,
self.exp_time_validator)
if 'number_of_images' in limits_dict:
limits = tuple(
map(float, limits_dict['number_of_images'].split(',')))
(lower, upper) = limits
self.num_img_validator.setRange(lower, upper)
self.first_img_validator.setRange(lower, upper)
self._acquisition_mib.bind_value_update(
'first_image', self.acq_widget_layout.first_image_ledit, int,
self.first_img_validator)
self._acquisition_mib.bind_value_update(
'num_images', self.acq_widget_layout.num_images_ledit, int,
self.num_img_validator)
num_passes = self.acq_widget_layout.findChild(QLineEdit,
"num_passes_ledit")
if num_passes:
self._acquisition_mib.\
bind_value_update('num_passes', num_passes, int,
QIntValidator(1, 1000, self))
overlap_ledit = self.acq_widget_layout.findChild(
QLineEdit, "overlap_ledit")
if overlap_ledit:
self._acquisition_mib.\
bind_value_update('overlap', overlap_ledit, float,
QDoubleValidator(-1000, 1000, 2, self))
self._acquisition_mib.\
bind_value_update('energy',
self.acq_widget_layout.energy_ledit,
float,
self.energy_validator)
self.update_energy_limits(
(self.energy_validator.bottom(), self.energy_validator.top()))
self._acquisition_mib.\
bind_value_update('transmission',
self.acq_widget_layout.transmission_ledit,
float,
self.transmission_validator)
self.update_transmission_limits((self.transmission_validator.bottom(),
self.transmission_validator.top()))
self._acquisition_mib.\
bind_value_update('resolution',
self.acq_widget_layout.resolution_ledit,
float,
self.resolution_validator)
#self.update_resolution_limits((self.resolution_validator.bottom(),
# self.resolution_validator.top()))
self._acquisition_mib.\
bind_value_update('shutterless',
self.acq_widget_layout.shutterless_cbx,
bool,
None)
self.set_tunable_energy(beamline_setup.tunable_wavelength())
has_shutter_less = self._beamline_setup_hwobj.detector_has_shutterless(
)
self.acq_widget_layout.shutterless_cbx.setEnabled(has_shutter_less)
self.acq_widget_layout.shutterless_cbx.setChecked(has_shutter_less)
if self._beamline_setup_hwobj.disable_num_passes():
num_passes = self.acq_widget_layout.findChild(
QLineEdit, "num_passes_ledit")
if num_passes:
num_passes.setDisabled(True)
self.init_detector_roi_modes()
def first_image_ledit_change(self, new_value):
"""
Descript. :
"""
#if str(new_value).isdigit():
# #self._path_template.start_num = int(new_value)
# #widget = self.acq_widget_layout.first_image_ledit
self.emit_acq_parameters_changed()
def exposure_time_ledit_changed(self, new_values):
"""If the exposure time changes we have to check the osc speed
and if necessary update osc range per frame
"""
self.update_osc_range_per_frame_limits()
self.update_osc_total_range_limits()
self.update_total_exp_time()
self.emit_acq_parameters_changed()
def num_images_ledit_change(self, new_value):
"""
Descript. :
"""
if str(new_value).isdigit():
#self._path_template.num_files = int(new_value)
self.update_osc_range_per_frame_limits()
self.update_osc_total_range_limits()
self.update_osc_total_range()
self.update_total_exp_time()
self.emit_acq_parameters_changed()
def overlap_changed(self, new_value):
"""
Descript. :
"""
if self._beamline_setup_hwobj:
has_shutter_less = self._beamline_setup_hwobj.detector_has_shutterless(
)
else:
has_shutter_less = True
if has_shutter_less:
try:
new_value = float(new_value)
except ValueError:
pass
if new_value != 0:
self.acq_widget_layout.shutterless_cbx.setEnabled(False)
self.acq_widget_layout.shutterless_cbx.setChecked(False)
self._acquisition_parameters.shutterless = False
else:
self.acq_widget_layout.shutterless_cbx.setEnabled(True)
self.acq_widget_layout.shutterless_cbx.setChecked(True)
self._acquisition_parameters.shutterless = True
def use_mad(self, state):
"""
Descript. :
"""
self.acq_widget_layout.energies_combo.setEnabled(state)
if state:
(name, energy) = self.get_mad_energy()
if energy != 0:
self.update_energy(energy)
self.madEnergySelectedSignal.emit(name, energy, state)
else:
self.update_energy(self.previous_energy)
energy = self._beamline_setup_hwobj.energy_hwobj.getCurrentEnergy()
self.madEnergySelectedSignal.emit('', self.previous_energy, state)
def max_osc_range_toggled(self, state):
self.update_osc_total_range_limits()
def get_mad_energy(self):
"""
Descript. :
"""
energy_str = str(self.acq_widget_layout.energies_combo.currentText())
(name, value) = energy_str.split(':')
name = name.strip()
value = value.strip()
value = 0 if (value == '-') else float(value)
return (name, value)
def set_energies(self, energy_scan_result):
"""
Descript. :
"""
self.acq_widget_layout.energies_combo.clear()
inflection = ('ip: %.4f' % energy_scan_result.inflection) if \
energy_scan_result.inflection else 'ip: -'
peak = ('pk: %.4f' % energy_scan_result.peak) if \
energy_scan_result.peak else 'pk: -'
first_remote = ('rm1: %.4f' % energy_scan_result.first_remote) if \
energy_scan_result.first_remote else 'rm1: -'
second_remote = ('rm2: %.4f' % energy_scan_result.second_remote) if \
energy_scan_result.second_remote else 'rm2: -'
self.acq_widget_layout.energies_combo.\
addItems([inflection, peak, first_remote, second_remote])
def energy_selected(self, index):
"""
Descript. :
"""
if self.acq_widget_layout.mad_cbox.isChecked():
(name, energy) = self.get_mad_energy()
if energy != 0:
self.update_energy(energy)
self.madEnergySelectedSignal.emit(name, energy, True)
def energy_ledit_changed(self, new_value):
if "energy" not in self.value_changed_list:
self.value_changed_list.append("energy")
self.emit_acq_parameters_changed()
def update_energy(self, energy, wav=None):
"""
Descript. :
"""
if "energy" not in self.value_changed_list and \
not self.acq_widget_layout.energy_ledit.hasFocus():
self.acq_widget_layout.energy_ledit.setText(str(energy))
self.emit_acq_parameters_changed()
def transmission_ledit_changed(self, transmission):
if "transmission" not in self.value_changed_list:
self.value_changed_list.append("transmission")
self.emit_acq_parameters_changed()
def update_transmission(self, transmission):
if "transmission" not in self.value_changed_list:
self.acq_widget_layout.transmission_ledit.setText(
str(transmission))
self.emit_acq_parameters_changed()
def resolution_ledit_changed(self, resolution):
if "resolution" not in self.value_changed_list:
self.value_changed_list.append("resolution")
self.emit_acq_parameters_changed()
def update_resolution(self, resolution):
if "resolution" not in self.value_changed_list and \
not self.acq_widget_layout.resolution_ledit.hasFocus():
self.acq_widget_layout.resolution_ledit.setText(str(resolution))
self.emit_acq_parameters_changed()
def update_energy_limits(self, limits):
if limits:
self.energy_validator.setBottom(limits[0])
self.energy_validator.setTop(limits[1])
self.acq_widget_layout.energy_ledit.setToolTip(\
"Energy limits %0.4f : %0.4f keV\n" % \
(limits[0], limits[1]) + \
"4 digits precision.")
self._acquisition_mib.validate_all()
def update_transmission_limits(self, limits):
if limits:
self.transmission_validator.setBottom(limits[0])
self.transmission_validator.setTop(limits[1])
self.acq_widget_layout.transmission_ledit.setToolTip(\
"Transmission limits %0.2f : %0.2f %%\n" % \
(limits[0], limits[1]) + \
"2 digits precision.")
self._acquisition_mib.validate_all()
def update_resolution_limits(self, limits):
if limits:
self.resolution_validator.setBottom(limits[0])
self.resolution_validator.setTop(limits[1])
self.acq_widget_layout.resolution_ledit.setToolTip(
"Resolution limits %0.4f : %0.4f %s\n" % \
(limits[0], limits[1], chr(197)) +
"4 digits precision.")
self._acquisition_mib.validate_all()
def update_detector_exp_time_limits(self, limits):
if limits:
self.exp_time_validator.setRange(limits[0], limits[1], 6)
self.acq_widget_layout.exp_time_ledit.setToolTip(
"Exposure time limits %0.6f s : %0.1f s\n" % \
(limits[0], limits[1]) + \
"6 digits precision.")
self._acquisition_mib.validate_all()
def update_num_images_limits(self, num_images_limits=None):
"""Updates number of images limit. Method used if plate mode.
"""
self._acquisition_mib.validate_all()
return
if self._diffractometer_hwobj.in_plate_mode():
if num_images_limits is None:
try:
osc_start = float(
self.acq_widget_layout.osc_start_ledit.text())
osc_range = float(
self.acq_widget_layout.osc_range_ledit.text())
except ValueError:
return
if osc_range == 0:
return
num_images_limits = int((self.osc_start_validator.top() - \
osc_start) / osc_range)
self.num_img_validator.setTop(num_images_limits)
self.acq_widget_layout.num_images_ledit.setToolTip(\
"Number of images limits : %d" % num_images_limits)
self._acquisition_mib.validate_all()
def init_detector_roi_modes(self):
"""
Descript. :
"""
if self._beamline_setup_hwobj is not None:
roi_modes = self._beamline_setup_hwobj.detector_hwobj.get_roi_modes(
)
if (len(roi_modes) > 0
and self.acq_widget_layout.detector_roi_mode_combo.count()
== 0):
for roi_mode in roi_modes:
self.acq_widget_layout.detector_roi_mode_combo.\
addItem(roi_mode)
self.acq_widget_layout.detector_roi_mode_label.setEnabled(True)
self.acq_widget_layout.detector_roi_mode_combo.setEnabled(True)
#self._acquisition_mib.bind_value_update('detector_roi_mode',
# self.acq_widget_layout.detector_roi_mode_combo,
## str,
# None)
def update_detector_roi_mode(self, roi_mode_index):
"""
Descript. :
"""
if roi_mode_index is not None and \
self.acq_widget_layout.detector_roi_mode_combo.count() > 0:
self.acq_widget_layout.detector_roi_mode_combo.\
setCurrentIndex(roi_mode_index)
def detector_roi_mode_changed(self, roi_mode_index):
"""
Descript. :
"""
if self._beamline_setup_hwobj is not None:
self._beamline_setup_hwobj.detector_hwobj.set_roi_mode(
roi_mode_index)
def kappa_ledit_changed(self, new_value):
if "kappa" not in self.value_changed_list:
self.value_changed_list.append("kappa")
self.emit_acq_parameters_changed()
def kappa_phi_ledit_changed(self, new_value):
if "kappa_phi" not in self.value_changed_list:
self.value_changed_list.append("kappa_phi")
self.emit_acq_parameters_changed()
def update_data_model(self, acquisition_parameters, path_template):
"""
Descript. :
"""
self._acquisition_parameters = acquisition_parameters
self._path_template = path_template
self._acquisition_mib.set_model(acquisition_parameters)
#Update mad widgets
mad = True if self._path_template.mad_prefix != '' else False
if mad:
mad_prefix = str(self._path_template.mad_prefix)
index = MAD_ENERGY_COMBO_NAMES[str(mad_prefix)]
self.acq_widget_layout.energies_combo.setCurrentIndex(index)
self.acq_widget_layout.mad_cbox.setChecked(True)
self.acq_widget_layout.energies_combo.setEnabled(True)
else:
self.acq_widget_layout.mad_cbox.setChecked(False)
self.acq_widget_layout.energies_combo.setEnabled(False)
self.acq_widget_layout.energies_combo.setCurrentIndex(0)
#self.update_osc_start_limits()
#self.update_osc_total_range()
self.emit_acq_parameters_changed()
def set_tunable_energy(self, state):
"""
Descript. :
"""
self.acq_widget_layout.energy_ledit.setEnabled(state)
self.acq_widget_layout.mad_cbox.setEnabled(state)
self.acq_widget_layout.energies_combo.setEnabled(state)
def check_parameter_conflict(self):
return self._acquisition_mib.validate_all()
def emit_acq_parameters_changed(self):
self.acqParametersChangedSignal.emit(
self._acquisition_mib.validate_all())
class AcquisitionStillWidget(QWidget):
acqParametersChangedSignal = pyqtSignal(list)
def __init__(self, parent=None, name=None, fl=0, acq_params=None,
path_template=None, layout='vertical'):
QWidget.__init__(self, parent, Qt.WindowFlags(fl))
if name is not None:
self.setObjectName(name)
# Hardware objects ----------------------------------------------------
self._beamline_setup_hwobj = None
self._diffractometer_hwobj = None
# Internal variables --------------------------------------------------
# Properties ----------------------------------------------------------
# Signals -------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Graphic elements ----------------------------------------------------
if acq_params is None:
self._acquisition_parameters = queue_model_objects.\
AcquisitionParameters()
else:
self._acquisition_parameters = acq_params
if path_template is None:
self._path_template = queue_model_objects.PathTemplate()
else:
self._path_template = path_template
self._acquisition_mib = DataModelInputBinder(self._acquisition_parameters)
self.acq_widget_layout = loadUi(os.path.join(\
os.path.dirname(__file__),
"ui_files/Qt4_acquisition_widget_vertical_still_layout.ui"))
# Layout --------------------------------------------------------------
__main_vlayout = QVBoxLayout(self)
__main_vlayout.addWidget(self.acq_widget_layout)
__main_vlayout.setSpacing(0)
__main_vlayout.setContentsMargins(0, 0, 0, 0)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self.acq_widget_layout.num_triggers_ledit.textChanged.connect(\
self.num_triggers_ledit_changed)
self.acq_widget_layout.num_images_per_trigger_ledit.textChanged.connect(\
self.num_images_per_trigger_ledit_changed)
self.acq_widget_layout.exp_time_ledit.textChanged.connect(\
self.exposure_time_ledit_changed)
self.acq_widget_layout.detector_roi_mode_combo.activated.connect(\
self.detector_roi_mode_changed)
self.acq_widget_layout.energy_ledit.textEdited.connect(\
self.energy_ledit_changed)
self.acq_widget_layout.transmission_ledit.textEdited.connect(\
self.transmission_ledit_changed)
self.acq_widget_layout.resolution_ledit.textEdited.connect(\
self.resolution_ledit_changed)
# Other ---------------------------------------------------------------
self.value_changed_list = []
self.energy_validator = QDoubleValidator(\
4, 25, 4, self.acq_widget_layout.energy_ledit)
self.resolution_validator = QDoubleValidator(\
0, 15, 3, self.acq_widget_layout.resolution_ledit)
self.transmission_validator = QDoubleValidator(\
0, 100, 3, self.acq_widget_layout.transmission_ledit)
self.exp_time_validator = QDoubleValidator(\
0.0001, 10000, 7, self.acq_widget_layout.exp_time_ledit)
self.num_triggers_validator = QIntValidator(\
1, 9999999, self.acq_widget_layout.num_triggers_ledit)
self.num_images_per_trigger_validator = QIntValidator(\
1, 9999999, self.acq_widget_layout.num_images_per_trigger_ledit)
self.num_img_validator = QIntValidator(\
1, 9999999, self.acq_widget_layout.num_images_ledit)
self.acq_widget_layout.detector_roi_mode_label.setEnabled(False)
self.acq_widget_layout.detector_roi_mode_combo.setEnabled(False)
def use_osc_start(self, status):
pass
def use_max_osc_range(self, status):
pass
def use_kappa(self, status):
pass
def set_beamline_setup(self, beamline_setup):
self._beamline_setup_hwobj = beamline_setup
limits_dict = self._beamline_setup_hwobj.get_acquisition_limit_values()
self._diffractometer_hwobj = self._beamline_setup_hwobj.diffractometer_hwobj
if 'exposure_time' in limits_dict:
limits = tuple(map(float, limits_dict['exposure_time'].split(',')))
(lower, upper) = limits
self.exp_time_validator.setRange(lower, upper, 6)
self._acquisition_mib.bind_value_update('exp_time',
self.acq_widget_layout.exp_time_ledit,
float,
self.exp_time_validator)
self._acquisition_mib.bind_value_update('num_triggers',
self.acq_widget_layout.num_triggers_ledit,
int,
self.num_triggers_validator)
self._acquisition_mib.bind_value_update('num_images_per_trigger',
self.acq_widget_layout.num_images_per_trigger_ledit,
int,
self.num_images_per_trigger_validator)
if 'number_of_images' in limits_dict:
limits = tuple(map(float, limits_dict['number_of_images'].split(',')))
(lower, upper) = limits
self.num_img_validator.setRange(lower, upper)
self._acquisition_mib.bind_value_update('num_images',
self.acq_widget_layout.num_images_ledit,
int,
self.num_img_validator)
self._acquisition_mib.\
bind_value_update('energy',
self.acq_widget_layout.energy_ledit,
float,
self.energy_validator)
self.update_energy_limits((self.energy_validator.bottom(),
self.energy_validator.top()))
self._acquisition_mib.\
bind_value_update('transmission',
self.acq_widget_layout.transmission_ledit,
float,
self.transmission_validator)
self.update_transmission_limits((self.transmission_validator.bottom(),
self.transmission_validator.top()))
self._acquisition_mib.\
bind_value_update('resolution',
self.acq_widget_layout.resolution_ledit,
float,
self.resolution_validator)
has_shutter_less = self._beamline_setup_hwobj.detector_has_shutterless()
self.init_detector_roi_modes()
def exposure_time_ledit_changed(self, new_values):
self.emit_acq_parameters_changed()
def energy_ledit_changed(self, new_value):
if "energy" not in self.value_changed_list:
self.value_changed_list.append("energy")
self.emit_acq_parameters_changed()
def update_energy(self, energy, wav=None):
if "energy" not in self.value_changed_list and \
not self.acq_widget_layout.energy_ledit.hasFocus():
self.acq_widget_layout.energy_ledit.setText(str(energy))
self.emit_acq_parameters_changed()
def transmission_ledit_changed(self, transmission):
if "transmission" not in self.value_changed_list:
self.value_changed_list.append("transmission")
self.emit_acq_parameters_changed()
def update_transmission(self, transmission):
if "transmission" not in self.value_changed_list:
self.acq_widget_layout.transmission_ledit.setText(str(transmission))
self.emit_acq_parameters_changed()
def resolution_ledit_changed(self, resolution):
if "resolution" not in self.value_changed_list:
self.value_changed_list.append("resolution")
self.emit_acq_parameters_changed()
def update_resolution(self, resolution):
if "resolution" not in self.value_changed_list and \
not self.acq_widget_layout.resolution_ledit.hasFocus():
self.acq_widget_layout.resolution_ledit.setText(str(resolution))
self.emit_acq_parameters_changed()
def update_energy_limits(self, limits):
if limits:
self.energy_validator.setBottom(limits[0])
self.energy_validator.setTop(limits[1])
self.acq_widget_layout.energy_ledit.setToolTip(\
"Energy limits %0.4f : %0.4f keV\n" % \
(limits[0], limits[1]) + \
"4 digits precision.")
self._acquisition_mib.validate_all()
def update_transmission_limits(self, limits):
if limits:
self.transmission_validator.setBottom(limits[0])
self.transmission_validator.setTop(limits[1])
self.acq_widget_layout.transmission_ledit.setToolTip(\
"Transmission limits %0.2f : %0.2f %%\n" % \
(limits[0], limits[1]) + \
"2 digits precision.")
self._acquisition_mib.validate_all()
def update_resolution_limits(self, limits):
if limits:
self.resolution_validator.setBottom(limits[0])
self.resolution_validator.setTop(limits[1])
self.acq_widget_layout.resolution_ledit.setToolTip(
"Resolution limits %0.4f : %0.4f %s\n" % \
(limits[0], limits[1], chr(197)) +
"4 digits precision.")
self._acquisition_mib.validate_all()
def update_detector_exp_time_limits(self, limits):
if limits:
self.exp_time_validator.setRange(limits[0], limits[1], 6)
self.acq_widget_layout.exp_time_ledit.setToolTip(
"Exposure time limits %0.6f s : %0.1f s\n" % \
(limits[0], limits[1]) + \
"6 digits precision.")
self._acquisition_mib.validate_all()
def init_detector_roi_modes(self):
if self._beamline_setup_hwobj is not None:
roi_modes = self._beamline_setup_hwobj.detector_hwobj.get_roi_modes()
if (len(roi_modes) > 0 and
self.acq_widget_layout.detector_roi_mode_combo.count() == 0):
for roi_mode in roi_modes:
self.acq_widget_layout.detector_roi_mode_combo.\
addItem(roi_mode)
self.acq_widget_layout.detector_roi_mode_label.setEnabled(len(roi_modes) > 1)
self.acq_widget_layout.detector_roi_mode_combo.setEnabled(len(roi_modes) > 1)
def update_detector_roi_mode(self, roi_mode_index):
if roi_mode_index is not None and \
self.acq_widget_layout.detector_roi_mode_combo.count() > 0:
self.acq_widget_layout.detector_roi_mode_combo.\
setCurrentIndex(roi_mode_index)
def detector_roi_mode_changed(self, roi_mode_index):
if self._beamline_setup_hwobj is not None:
self._beamline_setup_hwobj.detector_hwobj.set_roi_mode(roi_mode_index)
def update_osc_range_per_frame_limits(self):
pass
def update_exp_time_limits(self):
pass
def update_osc_start(self, value):
pass
def update_kappa(self, value):
pass
def update_kappa_phi(self, value):
pass
def update_data_model(self, acquisition_parameters, path_template):
self._acquisition_parameters = acquisition_parameters
self._path_template = path_template
self._acquisition_mib.set_model(acquisition_parameters)
self.emit_acq_parameters_changed()
def check_parameter_conflict(self):
return self._acquisition_mib.validate_all()
def emit_acq_parameters_changed(self):
self.acqParametersChangedSignal.emit(self._acquisition_mib.validate_all())
def set_energies(self, energies):
pass
def num_triggers_ledit_changed(self, value):
if "num_triggers" not in self.value_changed_list:
self.value_changed_list.append("num_triggers")
self.update_num_images()
self.emit_acq_parameters_changed()
def num_images_per_trigger_ledit_changed(self, values):
if "num_images_per_trigger" not in self.value_changed_list:
self.value_changed_list.append("num_images_per_trigger")
self.update_num_images()
self.emit_acq_parameters_changed()
def update_num_images(self):
self.acq_widget_layout.num_images_ledit.setText(\
str(self._acquisition_parameters.num_triggers * \
self._acquisition_parameters.num_images_per_trigger))
class DCParametersWidget(QWidget):
def __init__(self, parent=None, name="parameter_widget"):
QWidget.__init__(self, parent)
if name is not None:
self.setObjectName(name)
# Properties ----------------------------------------------------------
# Signals ------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Hardware objects ----------------------------------------------------
self._beamline_setup_hwobj = None
# Internal variables --------------------------------------------------
self._data_collection = None
self.add_dc_cb = None
self._tree_view_item = None
# Graphic elements ----------------------------------------------------
_dc_parameters_widget = 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 = 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 = 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 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 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 = self.queue_model_hwobj.\
check_for_path_collisions(path_template)
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 = str(name)
else:
path_template.mad_prefix = ''
run_number = self._beamline_setup_hwobj.queue_model_hwobj.\
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(QtGui.QWidget):
def __init__(self, parent=None, name="advanced_parameters_widget"):
QtGui.QWidget.__init__(self, parent)
# Hardware objects ----------------------------------------------------
self._queue_model_hwobj = None
self._beamline_setup_hwobj = None
# Internal values -----------------------------------------------------
self._data_collection = None
self._tree_view_item = None
# Properties ----------------------------------------------------------
# Signals -------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Graphic elements ----------------------------------------------------
_dc_parameters_widget = QtGui.QWidget(self)
self._data_path_widget = DataPathWidget(_dc_parameters_widget)
self._acq_widget = AcquisitionWidget(_dc_parameters_widget,
layout='horizontal')
#self._acq_widget.setFixedHeight(170)
_snapshot_widget = QtGui.QWidget(self)
self.position_widget = uic.loadUi(
os.path.join(os.path.dirname(__file__),
'ui_files/Qt4_snapshot_widget_layout.ui'))
# Layout --------------------------------------------------------------
_dc_parameters_widget_layout = QtGui.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)
_snapshots_vlayout = QtGui.QVBoxLayout(_snapshot_widget)
_snapshots_vlayout.addWidget(self.position_widget)
_snapshots_vlayout.setContentsMargins(0, 0, 0, 0)
_snapshots_vlayout.setSpacing(2)
_snapshots_vlayout.addStretch(10)
_main_hlayout = QtGui.QHBoxLayout(self)
_main_hlayout.addWidget(_dc_parameters_widget)
_main_hlayout.addWidget(_snapshot_widget)
_main_hlayout.setSpacing(2)
_main_hlayout.setContentsMargins(2, 2, 2, 2)
_main_hlayout.addStretch(0)
# 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.handle_path_conflict)
# 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.num_images_ledit.setEnabled(False)
self._acq_widget.acq_widget_layout.inverse_beam_cbx.hide()
self._acq_widget.acq_widget_layout.shutterless_cbx.hide()
self._acq_widget.acq_widget_layout.subwedge_size_label.hide()
self._acq_widget.acq_widget_layout.subwedge_size_ledit.hide()
def set_beamline_setup(self, bl_setup):
self._beamline_setup_hwobj = bl_setup
self._acq_widget.set_beamline_setup(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):
if self._tree_view_item:
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 path_conflict:
logging.getLogger("user_level_log").\
error('The current path settings will overwrite data' +\
' from another task. Correct the problem before collecting')
Qt4_widget_colors.set_widget_color(widget,
Qt4_widget_colors.LIGHT_RED)
else:
Qt4_widget_colors.set_widget_color(
widget, Qt4_widget_colors.LIGHT_WHITE)
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._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)
def populate_widget(self, tree_view_item):
self._tree_view_item = tree_view_item
advanced_model = tree_view_item.get_model()
self._data_collection = advanced_model.reference_image_collection
executed = self._data_collection.is_executed()
self._acq_widget.setEnabled(not executed)
self._data_path_widget.setEnabled(not executed)
image = advanced_model.grid_object.get_snapshot()
try:
image = image.scaled(427, 320, QtCore.Qt.KeepAspectRatio)
self.position_widget.svideo.setPixmap(QtGui.QPixmap(image))
except:
pass
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)
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 AcquisitionWidgetSimple(QWidget):
"""
Descript. :
"""
acqParametersChangedSignal = pyqtSignal(list)
madEnergySelectedSignal = pyqtSignal(str, float, bool)
def __init__(self, parent = None, name = None, fl = 0, acq_params = None,
path_template = None, layout = None):
"""
Descript. :
"""
QWidget.__init__(self, parent, Qt.WindowFlags(fl))
if name is not None:
self.setObjectName(name)
# Hardware objects ----------------------------------------------------
self._beamline_setup_hwobj = None
self._diffractometer_hwobj = None
# Internal variables --------------------------------------------------
self.value_changed_list = []
# Properties ----------------------------------------------------------
# Signals -------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Graphic elements ----------------------------------------------------
if acq_params is None:
self._acquisition_parameters = queue_model_objects.AcquisitionParameters()
else:
self._acquisition_parameters = acq_params
if path_template is None:
self._path_template = queue_model_objects.PathTemplate()
else:
self._path_template = path_template
self._acquisition_mib = DataModelInputBinder(self._acquisition_parameters)
self.acq_widget_layout = loadUi(os.path.join(os.path.dirname(\
__file__), "ui_files/Qt4_acquisition_widget_vertical_simple_layout.ui"))
# Layout --------------------------------------------------------------
main_layout = QVBoxLayout(self)
main_layout.addWidget(self.acq_widget_layout)
main_layout.setSpacing(0)
main_layout.setContentsMargins(0, 0, 0, 0)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self.acq_widget_layout.osc_start_cbox.stateChanged.connect(\
self.use_osc_start)
self.acq_widget_layout.num_images_cbox.activated.connect(\
self.update_num_images)
self.acq_widget_layout.detector_roi_mode_combo.activated.connect(\
self.detector_roi_mode_changed)
# Other ---------------------------------------------------------------
self.osc_start_validator = QDoubleValidator(\
-10000, 10000, 4, self.acq_widget_layout.osc_start_ledit)
self.osc_range_validator = QDoubleValidator(\
-10000, 10000, 4, self.acq_widget_layout.osc_range_ledit)
self.kappa_validator = QDoubleValidator(\
0, 360, 4, self.acq_widget_layout.kappa_ledit)
self.kappa_phi_validator = QDoubleValidator(\
0, 360, 4, self.acq_widget_layout.kappa_phi_ledit)
self.energy_validator = QDoubleValidator(\
0, 25, 5, self.acq_widget_layout.energy_ledit)
self.resolution_validator = QDoubleValidator(\
0, 15, 3, self.acq_widget_layout.resolution_ledit)
self.transmission_validator = QDoubleValidator(\
0, 100, 3, self.acq_widget_layout.transmission_ledit)
self.exp_time_validator = QDoubleValidator(\
0, 10000, 6, self.acq_widget_layout.exp_time_ledit)
self.acq_widget_layout.num_images_cbox.setCurrentIndex(1)
self.acq_widget_layout.detector_roi_mode_label.setEnabled(False)
self.acq_widget_layout.detector_roi_mode_combo.setEnabled(False)
def set_osc_start_limits(self, limits):
if not None in limits:
self.osc_start_validator.setRange(limits[0], limits[1], 4)
def update_osc_start(self, new_value):
"""
Descript. :
"""
if not self.acq_widget_layout.osc_start_cbox.hasFocus():
self.acq_widget_layout.osc_start_ledit.setText(str(new_value))
def update_kappa(self, new_value):
"""
Descript. :
"""
if not self.acq_widget_layout.kappa_ledit.hasFocus():
self.acq_widget_layout.kappa_ledit.setText(str(new_value))
def update_kappa_phi(self, new_value):
"""
Descript. :
"""
if not self.acq_widget_layout.kappa_phi_ledit.hasFocus():
self.acq_widget_layout.kappa_phi_ledit.setText(str(new_value))
def use_kappa(self, state):
"""
Descript. :
"""
if self._beamline_setup_hwobj is not None:
if self._beamline_setup_hwobj.diffractometer_hwobj.in_plate_mode():
state = False
self.acq_widget_layout.kappa_label.setEnabled(state)
self.acq_widget_layout.kappa_ledit.setEnabled(state)
self.acq_widget_layout.kappa_phi_label.setEnabled(state)
self.acq_widget_layout.kappa_phi_ledit.setEnabled(state)
def use_max_osc_range(self, state):
pass
def update_num_images(self, index = None, num_images = None):
"""
Descript. :
"""
if index is not None:
if index is 0:
self._acquisition_parameters.num_images = 1
self._path_template.num_files = 1
elif index is 1:
self._acquisition_parameters.num_images = 2
self._path_template.num_files = 2
elif index is 2:
self._acquisition_parameters.num_images = 4
self._path_template.num_files = 4
if num_images:
if self.acq_widget_layout.num_images_cbox.count() > 3:
self.acq_widget_layout.num_images_cbox.removeItem(4)
if num_images is 1:
self.acq_widget_layout.num_images_cbox.setCurrentIndex(0)
elif num_images is 2:
self.acq_widget_layout.num_images_cbox.setCurrentIndex(1)
elif num_images is 4:
self.acq_widget_layout.num_images_cbox.setCurrentIndex(2)
else:
self.acq_widget_layout.num_images_cbox.addItem(str(num_images))
self.acq_widget_layout.num_images_cbox.setCurrenIndex(3)
self._path_template.num_files = num_images
def use_mad(self, state):
"""
Descript. :
"""
pass
def get_mad_energy(self):
"""
Descript. :
"""
pass
def set_energies(self, energy_scan_result):
"""
Descript. :
"""
pass
def energy_selected(self, index):
"""
Descript. :
"""
pass
def set_beamline_setup(self, beamline_setup):
"""
Descript. :
"""
self._beamline_setup_hwobj = beamline_setup
self._diffractometer_hwobj = self._beamline_setup_hwobj.diffractometer_hwobj
limits_dict = self._beamline_setup_hwobj.get_acquisition_limit_values()
if 'osc_range' in limits_dict:
limits = tuple(map(float, limits_dict['osc_range'].split(',')))
(lower, upper) = limits
self.osc_start_validator.setRange(lower, upper, 4)
self.osc_range_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update('osc_start',
self.acq_widget_layout.osc_start_ledit,
float,
self.osc_start_validator)
self._acquisition_mib.bind_value_update('osc_range',
self.acq_widget_layout.osc_range_ledit,
float,
self.osc_range_validator)
if 'kappa' in limits_dict:
limits = tuple(map(float, limits_dict['kappa'].split(',')))
(lower, upper) = limits
self.kappa_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update('kappa',
self.acq_widget_layout.kappa_ledit,
float,
self.kappa_validator)
if 'kappa_phi' in limits_dict:
limits = tuple(map(float, limits_dict['kappa_phi'].split(',')))
(lower, upper) = limits
self.kappa_phi_validator.setRange(lower, upper, 4)
self._acquisition_mib.bind_value_update('kappa_phi',
self.acq_widget_layout.kappa_phi_ledit,
float,
self.kappa_phi_validator)
if 'exposure_time' in limits_dict:
limits = tuple(map(float, limits_dict['exposure_time'].split(',')))
(lower, upper) = limits
self.exp_time_validator.setRange(lower, upper, 6)
self._acquisition_mib.bind_value_update('exp_time',
self.acq_widget_layout.exp_time_ledit,
float,
self.exp_time_validator)
self._acquisition_mib.\
bind_value_update('energy',
self.acq_widget_layout.energy_ledit,
float,
self.energy_validator)
self.acq_widget_layout.energy_ledit.setToolTip(\
"Energy limits %0.3f : %0.3f" % \
(self.energy_validator.bottom(), self.energy_validator.top()))
self._acquisition_mib.\
bind_value_update('transmission',
self.acq_widget_layout.transmission_ledit,
float,
self.transmission_validator)
self._acquisition_mib.\
bind_value_update('resolution',
self.acq_widget_layout.resolution_ledit,
float,
self.resolution_validator)
self.set_tunable_energy(beamline_setup.tunable_wavelength())
if self._beamline_setup_hwobj.diffractometer_hwobj.in_plate_mode():
self.acq_widget_layout.num_images_cbox.clear()
self.acq_widget_layout.num_images_cbox.addItem("1")
self.acq_widget_layout.num_images_cbox.setCurrentIndex(0)
self.init_detector_roi_modes()
def set_energy(self, energy, wav):
"""
Descript. :
"""
if not self.acq_widget_layout.energy_ledit.hasFocus():
self.acq_widget_layout.energy_ledit.setText(str(energy))
def update_transmission(self, transmission):
"""
Descript. :
"""
if self.acq_widget_layout.transmission_ledit.hasFocus():
self.acq_widget_layout.transmission_ledit.setText(str(transmission))
def update_resolution(self, resolution):
"""
Descript. :
"""
if not self.acq_widget_layout.resolution_ledit.hasFocus():
self.acq_widget_layout.resolution_ledit.setText(str(resolution))
def update_energy_limits(self, limits):
"""
Descript. :
"""
if limits:
self.energy_validator.setBottom(limits[0])
self.energy_validator.setTop(limits[1])
self.acq_widget_layout.energy_ledit.setToolTip(
"Energy limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_transmission_limits(self, limits):
"""
Descript. :
"""
if limits:
self.transmission_validator.setBottom(limits[0])
self.transmission_validator.setTop(limits[1])
self.acq_widget_layout.transmission_ledit.setToolTip(
"Transmission limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_resolution_limits(self, limits):
"""
Descript. :
"""
if limits:
self.resolution_validator.setBottom(limits[0])
self.resolution_validator.setTop(limits[1])
self.acq_widget_layout.resolution_ledit.setToolTip(
"Resolution limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_detector_exp_time_limits(self, limits):
"""
Descript. :
"""
if limits:
self.exp_time_validator.setBottom(limits[0])
self.exp_time_validator.setTop(limits[1])
self.acq_widget_layout.exp_time_ledit.setToolTip(
"Exposure time limits %0.3f : %0.3f" %(limits[0], limits[1]))
self._acquisition_mib.validate_all()
def update_energy(self, energy, wav):
"""
Descript. :
"""
if "energy" not in self.value_changed_list and \
not self.acq_widget_layout.energy_ledit.hasFocus():
self.acq_widget_layout.energy_ledit.setText("%.4f" % float(energy))
def init_detector_roi_modes(self):
"""
Descript. :
"""
if self._beamline_setup_hwobj is not None:
roi_modes = self._beamline_setup_hwobj._get_roi_modes()
if (len(roi_modes) > 0 and
self.acq_widget_layout.detector_roi_mode_combo.count() == 0):
for roi_mode in roi_modes:
self.acq_widget_layout.detector_roi_mode_combo.\
addItem(roi_mode)
self.acq_widget_layout.detector_roi_mode_label.setEnabled(True)
self.acq_widget_layout.detector_roi_mode_combo.setEnabled(True)
#self._acquisition_mib.bind_value_update('detector_roi_mode',
# self.acq_widget_layout.detector_roi_mode_combo,
# str,
# None)
def update_osc_range_limits(self, limits=None):
pass
def update_exp_time_limits(self):
exp_time_limits = self._beamline_setup_hwobj.detector_hwobj.get_exposure_time_limits()
max_osc_speed = self._diffractometer_hwobj.get_osc_max_speed()
top_limit = float(self.acq_widget_layout.osc_range_ledit.text()) / max_osc_speed
limits = (max(exp_time_limits[0], top_limit), exp_time_limits[1])
self.update_detector_exp_time_limits(limits)
def update_detector_roi_mode(self, roi_mode_index):
"""
Descript. :
"""
if self.acq_widget_layout.detector_roi_mode_combo.count() > 0:
self.acq_widget_layout.detector_roi_mode_combo.\
setCurrentIndex(roi_mode_index)
def update_osc_range_limits(self, limits=None):
pass
def detector_roi_mode_changed(self, roi_mode_index):
"""
Descript. :
"""
if self._beamline_setup_hwobj is not None:
self._beamline_setup_hwobj.detector_hwobj.set_roi_mode(roi_mode_index)
def update_data_model(self, acquisition_parameters, path_template):
"""
Descript. :
"""
self._acquisition_parameters = acquisition_parameters
self._acquisition_mib.set_model(acquisition_parameters)
self._path_template = path_template
self.update_num_images(None, acquisition_parameters.num_images)
def set_tunable_energy(self, state):
"""
Descript. :
"""
self.acq_widget_layout.energy_ledit.setEnabled(state)
def use_osc_start(self, state):
"""
Descript. :
"""
self.acq_widget_layout.osc_start_ledit.setEnabled(state)
def check_parameter_conflict(self):
"""
Descript. :
"""
return self._acquisition_mib.validate_all()
class DCParametersWidget(QtGui.QWidget):
def __init__(self, parent=None, name="parameter_widget"):
QtGui.QWidget.__init__(self, parent)
if name is not None:
self.setObjectName(name)
# Properties ----------------------------------------------------------
# Signals ------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Hardware objects ----------------------------------------------------
self._beamline_setup_hwobj = None
# Internal variables --------------------------------------------------
self._data_collection = None
self.add_dc_cb = None
self._tree_view_item = None
# Graphic elements ----------------------------------------------------
_dc_parameters_widget = QtGui.QWidget(self)
self.caution_pixmap = Qt4_Icons.load("Caution2.png")
self.path_widget = DataPathWidget(_dc_parameters_widget)
self.acq_widget = AcquisitionWidget(_dc_parameters_widget,
layout='horizontal')
#self.acq_widget.setFixedHeight(170)
self.processing_widget = ProcessingWidget(_dc_parameters_widget)
self.position_widget = uic.loadUi(
os.path.join(os.path.dirname(__file__),
'ui_files/Qt4_snapshot_widget_layout.ui'))
#self.position_widget.setMinimumSize(310, 210)
# Layout --------------------------------------------------------------
_dc_parameters_widget_layout = QtGui.QVBoxLayout(self)
_dc_parameters_widget_layout.addWidget(self.path_widget)
_dc_parameters_widget_layout.addWidget(self.acq_widget)
_dc_parameters_widget_layout.addWidget(self.processing_widget)
_dc_parameters_widget_layout.setSpacing(2)
_dc_parameters_widget_layout.addStretch(0)
_dc_parameters_widget_layout.setContentsMargins(0, 0, 0, 0)
_dc_parameters_widget.setLayout(_dc_parameters_widget_layout)
_main_hlayout = QtGui.QHBoxLayout(self)
_main_hlayout.addWidget(_dc_parameters_widget)
_main_hlayout.addWidget(self.position_widget)
_main_hlayout.setSpacing(2)
_main_hlayout.setContentsMargins(0, 0, 0, 0)
_main_hlayout.addStretch(0)
self.setLayout(_main_hlayout)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self.path_widget.data_path_layout.prefix_ledit.textChanged.connect(
self._prefix_ledit_change)
self.path_widget.data_path_layout.run_number_ledit.textChanged.connect(
self._run_number_ledit_change)
self.connect(self.acq_widget, QtCore.SIGNAL('mad_energy_selected'),
self.mad_energy_selected)
self.connect(self.acq_widget, QtCore.SIGNAL("path_template_changed"),
self.handle_path_conflict)
#QtCore.QObject.connect(QtGui.QApplication, QtCore.SIGNAL('tab_changed'),
# self.tab_changed)
# Other ---------------------------------------------------------------
Qt4_widget_colors.set_widget_color(self.path_widget,
Qt4_widget_colors.GROUP_BOX_GRAY)
Qt4_widget_colors.set_widget_color(self.acq_widget,
Qt4_widget_colors.GROUP_BOX_GRAY)
Qt4_widget_colors.set_widget_color(self.processing_widget,
Qt4_widget_colors.GROUP_BOX_GRAY)
Qt4_widget_colors.set_widget_color(self.position_widget,
Qt4_widget_colors.GROUP_BOX_GRAY)
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):
if self._tree_view_item is None:
#TODO fix this
return
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_widget.setEnabled(state)
self.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.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
ration = image.height() / float(image.width())
image = image.scaled(400, 400 * ration, QtCore.Qt.KeepAspectRatio)
self.position_widget.svideo.setPixmap(QtGui.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(QtGui.QWidget):
def __init__(self, parent = None, name = "parameter_widget"):
QtGui.QWidget.__init__(self, parent)
if name is not None:
self.setObjectName(name)
# Properties ----------------------------------------------------------
# Signals ------------------------------------------------------------
# Slots ---------------------------------------------------------------
# Hardware objects ----------------------------------------------------
self._beamline_setup_hwobj = None
# Internal variables --------------------------------------------------
self._data_collection = None
self.add_dc_cb = None
self._tree_view_item = None
# Graphic elements ----------------------------------------------------
_dc_parameters_widget = QtGui.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 = QtGui.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 = QtGui.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 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 acq_parameters_changed(self):
if self._tree_view_item is None:
#TODO fix this
return
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)
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._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)
