def __init__(self, parent=None, name=None, fl=0):
CreateTaskBase.__init__(self, parent, name, fl, "Energy scan")
if not name:
self.setObjectName("create_energy_scan_widget")
# Hardware objects ----------------------------------------------------
# Internal variables --------------------------------------------------
self.init_models()
# Graphic elements ----------------------------------------------------
self._periodic_table_widget = PeriodicTableWidget(self)
self._data_path_widget = DataPathWidget(self,
data_model=self._path_template,
layout="vertical")
_parameters_gbox = qt_import.QGroupBox("Parameters", self)
self._adjust_transmission_cbox = qt_import.QCheckBox(
"Adjust transmission", _parameters_gbox)
self._adjust_transmission_cbox.setChecked(False)
self._adjust_transmission_cbox.setEnabled(True)
self._max_transmission_label = qt_import.QLabel(
"Maximum transmission:")
self._max_transmission_ledit = qt_import.QLineEdit(
"20", _parameters_gbox)
self._max_transmission_ledit.setFixedWidth(80)
self._max_transmission_ledit.setEnabled(False)
self._comments_widget = CommentsWidget(self)
# Layout --------------------------------------------------------------
_parameters_gbox_hlayout = qt_import.QGridLayout(_parameters_gbox)
_parameters_gbox_hlayout.addWidget(self._adjust_transmission_cbox, 0,
0)
_parameters_gbox_hlayout.addWidget(self._max_transmission_label, 1, 0)
_parameters_gbox_hlayout.addWidget(self._max_transmission_ledit, 1, 1)
_parameters_gbox_hlayout.setColumnStretch(2, 1)
_parameters_gbox_hlayout.setSpacing(2)
_main_vlayout = qt_import.QVBoxLayout(self)
_main_vlayout.addWidget(self._periodic_table_widget)
_main_vlayout.addWidget(self._data_path_widget)
_main_vlayout.addWidget(_parameters_gbox)
_main_vlayout.addWidget(self._comments_widget)
_main_vlayout.setContentsMargins(2, 2, 2, 2)
_main_vlayout.setSpacing(6)
_main_vlayout.addStretch(10)
# SizePolicies --------------------------------------------------------
self._comments_widget.setFixedHeight(100)
# Qt signal/slot connections ------------------------------------------
# self._periodic_table_widget.elementEdgeSelectedSignal.connect(\
# self.acq_parameters_changed)
self._data_path_widget.pathTemplateChangedSignal.connect(
self.path_template_changed)
self._adjust_transmission_cbox.stateChanged.connect(
self.adjust_transmission_state_changed)
self._max_transmission_ledit.textEdited.connect(
self.max_transmission_value_changed)
self._data_path_widget.data_path_layout.compression_cbox.setVisible(
False)
try:
self._periodic_table_widget.set_elements(
HWR.beamline.energy_scan.get_elements())
max_transmission_value = (
HWR.beamline.energy_scan.get_max_transmission_value())
self._adjust_transmission_cbox.setEnabled(True)
self._adjust_transmission_cbox.setChecked(True)
HWR.beamline.energy_scan.adjust_transmission(True)
if max_transmission_value:
self._max_transmission_ledit.setText("%.2f" %
max_transmission_value)
except BaseException:
pass
def __init__(self, parent=None, name=None, fl=0):
CreateTaskBase.__init__(
self, parent, name, qt_import.Qt.WindowFlags(fl), "Advanced"
)
if not name:
self.setObjectName("create_advanced_widget")
# Hardware objects ----------------------------------------------------
# Internal variables --------------------------------------------------
self._advanced_methods = None
self._grid_map = {}
self.spacing = [0, 0]
self.dc_selected = False
self.init_models()
# Graphic elements ----------------------------------------------------
self._advanced_methods_widget = qt_import.load_ui_file(
"advanced_methods_layout.ui"
)
self._acq_widget = AcquisitionWidget(
self,
"acquisition_widget",
layout="vertical",
acq_params=self._acquisition_parameters,
path_template=self._path_template,
)
self._acq_widget.grid_mode = True
self._data_path_widget = DataPathWidget(
self,
"create_dc_path_widget",
data_model=self._path_template,
layout="vertical",
)
self._comments_widget = CommentsWidget(self)
# Layout --------------------------------------------------------------
_main_vlayout = qt_import.QVBoxLayout(self)
_main_vlayout.addWidget(self._advanced_methods_widget)
_main_vlayout.addWidget(self._acq_widget)
_main_vlayout.addWidget(self._data_path_widget)
_main_vlayout.addWidget(self._comments_widget)
_main_vlayout.addStretch(0)
_main_vlayout.setSpacing(6)
_main_vlayout.setContentsMargins(2, 2, 2, 2)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self._acq_widget.acqParametersChangedSignal.connect(self.acq_parameters_changed)
self._acq_widget.madEnergySelectedSignal.connect(self.mad_energy_selected)
self._acq_widget.acq_widget_layout.osc_range_ledit.textEdited.connect(
self.grid_osc_range_ledit_changed
)
self._acq_widget.acq_widget_layout.osc_total_range_ledit.textEdited.connect(
self.grid_osc_total_range_ledit_changed
)
self._data_path_widget.pathTemplateChangedSignal.connect(
self.path_template_changed
)
self._advanced_methods_widget.grid_treewidget.itemSelectionChanged.connect(
self.grid_treewidget_item_selection_changed
)
self._advanced_methods_widget.draw_grid_button.clicked.connect(
self.draw_grid_button_clicked
)
self._advanced_methods_widget.remove_grid_button.clicked.connect(
self.remove_grid_button_clicked
)
self._advanced_methods_widget.hor_spacing_ledit.textEdited.connect(
self.hor_spacing_changed
)
self._advanced_methods_widget.ver_spacing_ledit.textEdited.connect(
self.ver_spacing_changed
)
self._advanced_methods_widget.move_right_button.clicked.connect(
lambda: self.move_grid("right")
)
self._advanced_methods_widget.move_left_button.clicked.connect(
lambda: self.move_grid("left")
)
self._advanced_methods_widget.move_up_button.clicked.connect(
lambda: self.move_grid("up")
)
self._advanced_methods_widget.move_down_button.clicked.connect(
lambda: self.move_grid("down")
)
self._advanced_methods_widget.overlay_cbox.toggled.connect(self.overlay_toggled)
self._advanced_methods_widget.overlay_slider.valueChanged.connect(
self.overlay_alpha_changed
)
self._advanced_methods_widget.overlay_color_button.clicked.connect(
self.overlay_change_color
)
self._advanced_methods_widget.move_to_grid_button.clicked.connect(
self.move_to_grid
)
self._advanced_methods_widget.method_combo.activated.connect(
self.method_combo_activated
)
# Other ---------------------------------------------------------------
self._acq_widget.use_osc_start(False)
self._acq_widget.use_kappa(False)
self._acq_widget.acq_widget_layout.num_images_label.setEnabled(False)
self._acq_widget.acq_widget_layout.num_images_ledit.setEnabled(False)
for col in range(self._advanced_methods_widget.grid_treewidget.columnCount()):
self._advanced_methods_widget.grid_treewidget.resizeColumnToContents(col)
self._acq_widget.acq_widget_layout.osc_total_range_label.setText(
"Total osc. range per line"
)
self._advanced_methods = list(HWR.beamline.advanced_methods)
if self._advanced_methods:
for method in self._advanced_methods:
self._advanced_methods_widget.method_combo.addItem(method)
else:
self.setEnabled(False)
self._acq_widget.acq_widget_layout.osc_start_label.setText(
"Oscillation middle:"
)
hor_size, ver_size = HWR.beamline.beam.get_beam_size()
self.spacing[0] = hor_size
self.spacing[1] = ver_size
self._advanced_methods_widget.hor_spacing_ledit.setText(
"%.1f" % (hor_size * 1000)
)
self._advanced_methods_widget.ver_spacing_ledit.setText(
"%.1f" % (ver_size * 1000)
)
HWR.beamline.sample_view.connect("shapeCreated", self.shape_created)
HWR.beamline.sample_view.connect("shapeChanged", self.shape_changed)
HWR.beamline.sample_view.connect("shapeDeleted", self.shape_deleted)
self._comments_widget.setHidden(True)
def __init__(self, parent=None, name="task_toolbox"):
qt_import.QWidget.__init__(self, parent)
self.setObjectName = name
# Internal variables --------------------------------------------------
self.tree_brick = None
self.previous_page_index = 0
self.is_running = None
self.path_conflict = False
self.acq_conflict = False
self.enable_collect = False
self.create_task_widgets = {}
# Graphic elements ----------------------------------------------------
self.method_label = qt_import.QLabel("Collection method", self)
self.method_label.setAlignment(qt_import.Qt.AlignCenter)
self.tool_box = qt_import.QToolBox(self)
self.tool_box.setObjectName("tool_box")
# self.tool_box.setFixedWidth(600)
self.discrete_page = CreateDiscreteWidget(self.tool_box, "Discrete")
self.char_page = CreateCharWidget(self.tool_box, "Characterise")
self.helical_page = CreateHelicalWidget(self.tool_box, "helical_page")
self.energy_scan_page = CreateEnergyScanWidget(self.tool_box,
"energy_scan")
self.xrf_spectrum_page = CreateXRFSpectrumWidget(
self.tool_box, "xrf_spectrum")
if HWR.beamline.gphl_workflow is not None:
self.gphl_workflow_page = CreateGphlWorkflowWidget(
self.tool_box, "gphl_workflow")
else:
self.gphl_workflow_page = None
self.advanced_page = CreateAdvancedWidget(self.tool_box,
"advanced_scan")
self.xray_imaging_page = CreateXrayImagingWidget(
self.tool_box, "xray_imaging")
self.ssx_page = CreateSsxWidget(self.tool_box, "ssx")
self.tool_box.addItem(self.discrete_page, "Standard Collection")
self.tool_box.addItem(self.char_page, "Characterisation")
self.tool_box.addItem(self.helical_page, "Helical Collection")
self.tool_box.addItem(self.energy_scan_page, "Energy Scan")
self.tool_box.addItem(self.xrf_spectrum_page, "XRF Spectrum")
if self.gphl_workflow_page is not None:
self.tool_box.addItem(self.gphl_workflow_page, "GPhL Workflows")
self.tool_box.addItem(self.advanced_page, "Advanced")
self.tool_box.addItem(self.xray_imaging_page, "Xray Imaging")
self.tool_box.addItem(self.ssx_page, "SSX")
self.button_box = qt_import.QWidget(self)
self.create_task_button = qt_import.QPushButton(
" Add to queue", self.button_box)
self.create_task_button.setIcon(icons.load_icon("add_row.png"))
msg = "Add the collection method to the selected sample"
self.create_task_button.setToolTip(msg)
self.create_task_button.setEnabled(False)
self.collect_now_button = qt_import.QPushButton(
"Collect Now", self.button_box)
self.collect_now_button.setIcon(icons.load_icon("VCRPlay2.png"))
self.collect_now_button.setToolTip(
"Add the collection method to the queue and execute it")
self.comments_widget = CommentsWidget(self)
self.comments_widget.setFixedHeight(100)
# Layout --------------------------------------------------------------
_button_box_hlayout = qt_import.QHBoxLayout(self.button_box)
_button_box_hlayout.addWidget(self.collect_now_button)
_button_box_hlayout.addStretch(0)
_button_box_hlayout.addWidget(self.create_task_button)
_button_box_hlayout.setSpacing(0)
_button_box_hlayout.setContentsMargins(0, 0, 0, 0)
_main_vlayout = qt_import.QVBoxLayout(self)
_main_vlayout.addWidget(self.method_label)
_main_vlayout.addWidget(self.tool_box)
_main_vlayout.addWidget(self.comments_widget)
_main_vlayout.addWidget(self.button_box)
_main_vlayout.setSpacing(0)
_main_vlayout.setContentsMargins(0, 0, 0, 0)
# SizePolicies --------------------------------------------------------
# self.setSizePolicy(QSizePolicy.Expanding,
# QSizePolicy.Expanding)
# Qt signal/slot connections ------------------------------------------
self.create_task_button.clicked.connect(self.create_task_button_click)
self.collect_now_button.clicked.connect(self.collect_now_button_click)
self.tool_box.currentChanged.connect(self.current_page_changed)
for i in range(0, self.tool_box.count()):
self.tool_box.widget(i).acqParametersConflictSignal.connect(
self.acq_parameters_conflict_changed)
self.tool_box.widget(i).pathTempleConflictSignal.connect(
self.path_template_conflict_changed)
# Other ---------------------------------------------------------------
in_plate_mode = HWR.beamline.diffractometer.in_plate_mode()
if (HWR.beamline.energy_scan is None or in_plate_mode
or not HWR.beamline.tunable_wavelength):
self.hide_task(self.energy_scan_page)
logging.getLogger("HWR").info("Energy scan task not available")
if HWR.beamline.xrf_spectrum is None or in_plate_mode:
self.hide_task(self.xrf_spectrum_page)
logging.getLogger("HWR").info("XRF spectrum task not available")
if not HWR.beamline.imaging or in_plate_mode:
self.hide_task(self.xray_imaging_page)
logging.getLogger("HWR").info("Xray Imaging task not available")
if HWR.beamline.gphl_connection and HWR.beamline.gphl_workflow:
self.gphl_workflow_page.initialise_workflows()
else:
logging.getLogger("HWR").info("GPhL workflow task not available")
def __init__(self, parent=None, name=None, fl=0):
CreateTaskBase.__init__(
self, parent, name, qt_import.Qt.WindowFlags(fl), "XRF spectrum"
)
if name is not None:
self.setObjectName(name)
# Hardware objects ----------------------------------------------------
# Internal variables --------------------------------------------------
self.count_time = None
self.xrf_spectrum_model = None
self.init_models()
# Graphic elements ----------------------------------------------------
self._data_path_widget = DataPathWidget(
self, data_model=self._path_template, layout="vertical"
)
_parameters_gbox = qt_import.QGroupBox("Parameters", self)
_count_time_label = qt_import.QLabel("Count time (sec.):", _parameters_gbox)
self.count_time_ledit = qt_import.QLineEdit("1", _parameters_gbox)
# self.count_time_ledit.setMaximumWidth(75)
self.adjust_transmission_cbox = qt_import.QCheckBox(
"Adjust transmission", _parameters_gbox
)
self.adjust_transmission_cbox.setChecked(True)
self._comments_widget = CommentsWidget(self)
# Layout --------------------------------------------------------------
_parameters_gbox_hlayout = qt_import.QHBoxLayout(_parameters_gbox)
_parameters_gbox_hlayout.addWidget(_count_time_label)
_parameters_gbox_hlayout.addWidget(self.count_time_ledit)
_parameters_gbox_hlayout.addWidget(self.adjust_transmission_cbox)
_parameters_gbox_hlayout.addStretch(0)
_parameters_gbox_hlayout.setSpacing(2)
_parameters_gbox_hlayout.setContentsMargins(0, 0, 0, 0)
_main_vlayout = qt_import.QVBoxLayout(self)
_main_vlayout.addWidget(self._data_path_widget)
_main_vlayout.addWidget(_parameters_gbox)
_main_vlayout.addWidget(self._comments_widget)
_main_vlayout.setSpacing(6)
_main_vlayout.setContentsMargins(2, 2, 2, 2)
_main_vlayout.addStretch(0)
# SizePolicies --------------------------------------------------------
self._comments_widget.setFixedHeight(100)
# Qt signal/slot connections ------------------------------------------
self._data_path_widget.pathTemplateChangedSignal.connect(
self.path_template_changed
)
self.adjust_transmission_cbox.stateChanged.connect(
self.adjust_transmission_state_changed
)
# Other ---------------------------------------------------------------
self._data_path_widget.data_path_layout.compression_cbox.setVisible(False)
def __init__(self, parent=None, name=None, fl=0):
CreateTaskBase.__init__(self, parent, name, fl, "Helical")
if not name:
self.setObjectName("create_helical_widget")
self.init_models()
# Hardware objects ----------------------------------------------------
# Internal variables --------------------------------------------------
self._lines_map = {}
# Graphic elements ----------------------------------------------------
self._lines_widget = qt_import.load_ui_file(
"helical_line_widget_layout.ui")
self._acq_widget = AcquisitionWidget(
self,
"acquisition_widget",
layout="vertical",
acq_params=self._acquisition_parameters,
path_template=self._path_template,
)
self._data_path_widget = DataPathWidget(
self,
"create_dc_path_widget",
data_model=self._path_template,
layout="vertical",
)
self._processing_widget = ProcessingWidget(
self, data_model=self._processing_parameters)
self._comments_widget = CommentsWidget(self)
# Layout --------------------------------------------------------------
_main_vlayout = qt_import.QVBoxLayout(self)
_main_vlayout.addWidget(self._lines_widget)
_main_vlayout.addWidget(self._acq_widget)
_main_vlayout.addWidget(self._data_path_widget)
_main_vlayout.addWidget(self._processing_widget)
_main_vlayout.addWidget(self._comments_widget)
_main_vlayout.addStretch(0)
_main_vlayout.setSpacing(6)
_main_vlayout.setContentsMargins(2, 2, 2, 2)
# SizePolicies --------------------------------------------------------
self._comments_widget.setFixedHeight(100)
# Qt signal/slot connections ------------------------------------------
self._lines_widget.lines_treewidget.itemSelectionChanged.connect(
self.lines_treewidget_selection_changed)
self._lines_widget.create_line_button.clicked.connect(
self.create_line_button_clicked)
self._lines_widget.create_auto_line_button.clicked.connect(
self.create_auto_line_button_clicked)
self._lines_widget.remove_line_button.clicked.connect(
self.remove_line_button_clicked)
self._lines_widget.overlay_cbox.stateChanged.connect(
self.overlay_toggled)
self._lines_widget.overlay_slider.valueChanged.connect(
self.overlay_alpha_changed)
self._lines_widget.swap_points_button.clicked.connect(
self.swap_points_clicked)
self._acq_widget.acqParametersChangedSignal.connect(
self.acq_parameters_changed)
self._acq_widget.madEnergySelectedSignal.connect(
self.mad_energy_selected)
self._data_path_widget.pathTemplateChangedSignal.connect(
self.path_template_changed)
self._processing_widget.enableProcessingSignal.connect(
self._run_processing_toggled)
# Other ---------------------------------------------------------------
for col in range(self._lines_widget.lines_treewidget.columnCount()):
self._lines_widget.lines_treewidget.resizeColumnToContents(col)
# self._processing_widget.processing_widget.\
# run_online_processing_cbox.setChecked(False)
self._processing_widget.processing_widget.run_online_processing_cbox.setChecked(
HWR.beamline.run_online_processing)
self.enable_widgets(False)
shapes = HWR.beamline.sample_view.get_shapes()
for shape in shapes:
if isinstance(shape, GraphicsItemLine):
self.shape_created(shape, "Line")
HWR.beamline.sample_view.connect("shapeCreated", self.shape_created)
HWR.beamline.sample_view.connect("shapeChanged", self.shape_changed)
HWR.beamline.sample_view.connect("shapeDeleted", self.shape_deleted)
self._comments_widget.setHidden(True)