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)