def __init__(self, parent=None, name=None, fl=0):
"""
Descript. :
"""
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 = QGroupBox('Parameters', self)
self._adjust_transmission_cbox = QCheckBox(\
"Adjust transmission", _parameters_gbox)
self._adjust_transmission_cbox.setChecked(False)
self._adjust_transmission_cbox.setEnabled(True)
self._max_transmission_label = QLabel("Maximum transmission:")
self._max_transmission_ledit = QLineEdit("20", _parameters_gbox)
self._max_transmission_ledit.setFixedWidth(80)
self._max_transmission_ledit.setEnabled(False)
# Layout --------------------------------------------------------------
_parameters_gbox_hlayout = 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 = QVBoxLayout(self)
_main_vlayout.addWidget(self._periodic_table_widget)
_main_vlayout.addWidget(self._data_path_widget)
_main_vlayout.addWidget(_parameters_gbox)
_main_vlayout.setContentsMargins(2, 2, 2, 2)
_main_vlayout.setSpacing(6)
_main_vlayout.addStretch(10)
# SizePolicies --------------------------------------------------------
# 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)
def __init__(self, parent = None,name = None, fl = 0):
"""
Descript. :
"""
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')
# Layout --------------------------------------------------------------
self.main_layout = QtGui.QVBoxLayout(self)
self.main_layout.addWidget(self._periodic_table_widget)
self.main_layout.addWidget(self._data_path_widget)
self.main_layout.setContentsMargins(0, 0, 0, 0)
self.main_layout.setSpacing(6)
self.main_layout.addStretch(10)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self._periodic_table_widget.elementEdgeSelectedSignal.\
connect(self.acq_parameters_changed)
self._data_path_widget.pathTemplateChangedSignal.\
connect(self.acq_parameters_changed)
def __init__(self, parent=None,name=None, fl=0):
"""
Descript. :
"""
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 = QGroupBox('Parameters', self)
self._adjust_transmission_cbox = QCheckBox(\
"Adjust transmission", _parameters_gbox)
self._adjust_transmission_cbox.setChecked(False)
self._adjust_transmission_cbox.setEnabled(True)
self._max_transmission_label = QLabel("Maximum transmission:")
self._max_transmission_ledit = QLineEdit("20", _parameters_gbox)
self._max_transmission_ledit.setFixedWidth(80)
self._max_transmission_ledit.setEnabled(False)
# Layout --------------------------------------------------------------
_parameters_gbox_hlayout = 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 = QVBoxLayout(self)
_main_vlayout.addWidget(self._periodic_table_widget)
_main_vlayout.addWidget(self._data_path_widget)
_main_vlayout.addWidget(_parameters_gbox)
_main_vlayout.setContentsMargins(2, 2, 2, 2)
_main_vlayout.setSpacing(6)
_main_vlayout.addStretch(10)
# SizePolicies --------------------------------------------------------
# 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)
def __init__(self, parent=None, name=None, fl=0):
"""
Descript. :
"""
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_gbox = QtGui.QGroupBox('Data location', self)
self._data_path_widget = DataPathWidget(self._data_path_gbox,
data_model=self._path_template,
layout='vertical')
# Layout --------------------------------------------------------------
self._data_path_gbox_vlayout = QtGui.QVBoxLayout(self)
self._data_path_gbox_vlayout.addWidget(self._data_path_widget)
self._data_path_gbox_vlayout.setSpacing(0)
self._data_path_gbox_vlayout.setContentsMargins(0, 0, 0, 0)
self._data_path_gbox.setLayout(self._data_path_gbox_vlayout)
self.main_layout = QtGui.QVBoxLayout(self)
self.main_layout.addWidget(self._periodic_table_widget)
self.main_layout.addWidget(self._data_path_gbox)
self.main_layout.addStretch(0)
self.main_layout.setSpacing(0)
self.main_layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(self.main_layout)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self._data_path_widget.data_path_layout.run_number_ledit.textChanged.\
connect(self._run_number_ledit_change)
self.connect(self._data_path_widget,
QtCore.SIGNAL("pathTemplateChanged"),
self.handle_path_conflict)
def __init__(self, parent=None, name=None, fl=0, parent_class=None):
"""
Descript. :
"""
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')
# Layout --------------------------------------------------------------
self.main_layout = QtGui.QVBoxLayout(self)
self.main_layout.addWidget(self._periodic_table_widget)
self.main_layout.addWidget(self._data_path_widget)
self.main_layout.setContentsMargins(0, 0, 0, 0)
self.main_layout.setSpacing(6)
self.main_layout.addStretch(10)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self._periodic_table_widget.elementEdgeSelectedSignal.connect(\
self.element_edge_selected)
self._data_path_widget.data_path_layout.run_number_ledit.textChanged.\
connect(self._run_number_ledit_change)
self._data_path_widget.pathTemplateChangedSignal.connect(
self.handle_path_conflict)
def __init__(self, parent = None,name = None, fl = 0):
"""
Descript. :
"""
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_gbox = QtGui.QGroupBox('Data location', self)
self._data_path_widget = DataPathWidget(self._data_path_gbox,
data_model = self._path_template,
layout = 'vertical')
# Layout --------------------------------------------------------------
self._data_path_gbox_vlayout = QtGui.QVBoxLayout(self)
self._data_path_gbox_vlayout.addWidget(self._data_path_widget)
self._data_path_gbox_vlayout.setSpacing(0)
self._data_path_gbox_vlayout.setContentsMargins(0, 0, 0, 0)
self._data_path_gbox.setLayout(self._data_path_gbox_vlayout)
self.main_layout = QtGui.QVBoxLayout(self)
self.main_layout.addWidget(self._periodic_table_widget)
self.main_layout.addWidget(self._data_path_gbox)
self.main_layout.addStretch(0)
self.main_layout.setSpacing(0)
self.main_layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(self.main_layout)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self._data_path_widget.data_path_layout.run_number_ledit.textChanged.\
connect(self._run_number_ledit_change)
self.connect(self._data_path_widget,
QtCore.SIGNAL("pathTemplateChanged"),
self.handle_path_conflict)
class CreateEnergyScanWidget(CreateTaskBase):
"""
Descript. :
"""
def __init__(self, parent = None,name = None, fl = 0):
"""
Descript. :
"""
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')
# Layout --------------------------------------------------------------
self.main_layout = QtGui.QVBoxLayout(self)
self.main_layout.addWidget(self._periodic_table_widget)
self.main_layout.addWidget(self._data_path_widget)
self.main_layout.setContentsMargins(0, 0, 0, 0)
self.main_layout.setSpacing(6)
self.main_layout.addStretch(10)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self._periodic_table_widget.elementEdgeSelectedSignal.\
connect(self.acq_parameters_changed)
self._data_path_widget.pathTemplateChangedSignal.\
connect(self.acq_parameters_changed)
def init_models(self):
"""
Descript. :
"""
CreateTaskBase.init_models(self)
self.enery_scan = queue_model_objects.EnergyScan()
self._path_template.start_num = 1
self._path_template.num_files = 1
self._path_template.suffix = 'raw'
def set_energy_scan_hwobj(self, energy_scan_hwobj):
"""
Descript. :
"""
self._periodic_table_widget.set_elements(\
energy_scan_hwobj.getElements())
def single_item_selection(self, tree_item):
"""
Descript. :
"""
CreateTaskBase.single_item_selection(self, tree_item)
escan_model = tree_item.get_model()
if isinstance(tree_item, Qt4_queue_item.EnergyScanQueueItem):
if tree_item.get_model().is_executed():
self.setDisabled(True)
else:
self.setDisabled(False)
if escan_model.get_path_template():
self._path_template = escan_model.get_path_template()
self._data_path_widget.update_data_model(self._path_template)
elif not(isinstance(tree_item, Qt4_queue_item.SampleQueueItem) or \
isinstance(tree_item, Qt4_queue_item.DataCollectionGroupQueueItem)):
self.setDisabled(True)
def approve_creation(self):
"""
Descript. :
"""
base_result = CreateTaskBase.approve_creation(self)
selected_element, selected_edge = self._periodic_table_widget.get_selected_element_edge()
if not selected_element:
logging.getLogger("GUI").\
info("No element selected, please select an element.")
return base_result and selected_element
# Called by the owning widget (task_toolbox_widget) to create
# a collection. When a data collection group is selected.
def _create_task(self, sample, shape):
"""
Descript. :
"""
data_collections = []
selected_element, selected_edge = self._periodic_table_widget.get_selected_element_edge()
if selected_element:
if not shape:
cpos = queue_model_objects.CentredPosition()
cpos.snapshot_image = self._graphics_manager_hwobj.get_scene_snapshot()
else:
# Shapes selected and sample is mounted, get the
# centred positions for the shapes
if isinstance(shape, GraphicsItemPoint):
snapshot = self._graphics_manager_hwobj.\
get_scene_snapshot(shape)
cpos = copy.deepcopy(shape.get_centred_position())
cpos.snapshot_image = snapshot
path_template = self._create_path_template(sample, self._path_template)
energy_scan = queue_model_objects.EnergyScan(sample,
path_template,
cpos)
energy_scan.set_name(path_template.get_prefix())
energy_scan.set_number(path_template.run_number)
energy_scan.element_symbol = selected_element
energy_scan.edge = selected_edge
data_collections.append(energy_scan)
self._path_template.run_number += 1
else:
logging.getLogger("GUI").\
info("No element selected, please select an element.")
return data_collections
def element_edge_selected(self, element, edge):
if len(self._current_selected_items) == 1:
item = self._current_selected_items[0]
if isinstance(item, Qt4_queue_item.EnergyScanQueueItem):
item.get_model().element_symbol = str(element)
item.get_model().edge = str(edge)
class CreateEnergyScanWidget(CreateTaskBase):
"""
Descript. :
"""
def __init__(self, parent=None, name=None, fl=0):
"""
Descript. :
"""
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 = QGroupBox('Parameters', self)
self._adjust_transmission_cbox = QCheckBox(\
"Adjust transmission", _parameters_gbox)
self._adjust_transmission_cbox.setChecked(False)
self._adjust_transmission_cbox.setEnabled(True)
self._max_transmission_label = QLabel("Maximum transmission:")
self._max_transmission_ledit = QLineEdit("20", _parameters_gbox)
self._max_transmission_ledit.setFixedWidth(80)
self._max_transmission_ledit.setEnabled(False)
# Layout --------------------------------------------------------------
_parameters_gbox_hlayout = 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 = QVBoxLayout(self)
_main_vlayout.addWidget(self._periodic_table_widget)
_main_vlayout.addWidget(self._data_path_widget)
_main_vlayout.addWidget(_parameters_gbox)
_main_vlayout.setContentsMargins(2, 2, 2, 2)
_main_vlayout.setSpacing(6)
_main_vlayout.addStretch(10)
# SizePolicies --------------------------------------------------------
# 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)
def set_expert_mode(self, state):
self._adjust_transmission_cbox.setEnabled(state)
self._max_transmission_label.setEnabled(state)
self._max_transmission_ledit.setEnabled(state)
def enable_compression(self, state):
CreateTaskBase.enable_compression(self, False)
def set_beamline_setup(self, bl_setup_hwobj):
CreateTaskBase.set_beamline_setup(self, bl_setup_hwobj)
self._periodic_table_widget.set_elements(\
self._beamline_setup_hwobj.energyscan_hwobj.getElements())
self.enable_compression(False)
try:
max_transmission_value = self._beamline_setup_hwobj.\
energyscan_hwobj.get_max_transmission_value()
self._adjust_transmission_cbox.setEnabled(True)
self._adjust_transmission_cbox.setChecked(True)
self._beamline_setup_hwobj.energyscan_hwobj.adjust_transmission(
True)
if max_transmission_value:
self._max_transmission_ledit.setText("%.2f" %
max_transmission_value)
except:
pass
def init_models(self):
"""
Descript. :
"""
CreateTaskBase.init_models(self)
self.enery_scan = queue_model_objects.EnergyScan()
self._path_template.start_num = 1
self._path_template.num_files = 1
self._path_template.suffix = 'raw'
self._path_template.compression = False
def single_item_selection(self, tree_item):
"""
Descript. :
"""
CreateTaskBase.single_item_selection(self, tree_item)
escan_model = tree_item.get_model()
if isinstance(tree_item, Qt4_queue_item.EnergyScanQueueItem):
if tree_item.get_model().is_executed():
self.setDisabled(True)
else:
self.setDisabled(False)
if escan_model.get_path_template():
self._path_template = escan_model.get_path_template()
self._data_path_widget.update_data_model(self._path_template)
elif not(isinstance(tree_item, Qt4_queue_item.SampleQueueItem) or \
isinstance(tree_item, Qt4_queue_item.DataCollectionGroupQueueItem)):
self.setDisabled(True)
def approve_creation(self):
"""
Descript. :
"""
base_result = CreateTaskBase.approve_creation(self)
selected_element, selected_edge = self._periodic_table_widget.get_selected_element_edge(
)
if not selected_element:
logging.getLogger("GUI").\
warning("No element selected, please select an element.")
return base_result and selected_element
# Called by the owning widget (task_toolbox_widget) to create
# a collection. When a data collection group is selected.
def _create_task(self, sample, shape):
"""
Descript. :
"""
data_collections = []
selected_element, selected_edge = self._periodic_table_widget.get_selected_element_edge(
)
if selected_element:
if not shape:
cpos = queue_model_objects.CentredPosition()
cpos.snapshot_image = self._graphics_manager_hwobj.get_scene_snapshot(
)
else:
# Shapes selected and sample is mounted, get the
# centred positions for the shapes
if isinstance(shape, GraphicsItemPoint):
snapshot = self._graphics_manager_hwobj.\
get_scene_snapshot(shape)
cpos = copy.deepcopy(shape.get_centred_position())
cpos.snapshot_image = snapshot
path_template = self._create_path_template(sample,
self._path_template)
energy_scan = queue_model_objects.EnergyScan(
sample, path_template, cpos)
energy_scan.set_name(path_template.get_prefix())
energy_scan.set_number(path_template.run_number)
energy_scan.element_symbol = selected_element
energy_scan.edge = selected_edge
data_collections.append(energy_scan)
self._path_template.run_number += 1
else:
logging.getLogger("GUI").\
info("No element selected, please select an element.")
return data_collections
def element_edge_selected(self, element, edge):
if len(self._current_selected_items) == 1:
item = self._current_selected_items[0]
if isinstance(item, Qt4_queue_item.EnergyScanQueueItem):
item.get_model().element_symbol = str(element)
item.get_model().edge = str(edge)
def adjust_transmission_state_changed(self, state):
self._max_transmission_ledit.setEnabled(state)
self._beamline_setup_hwobj.energyscan_hwobj.adjust_transmission(state)
def max_transmission_value_changed(self, value):
try:
max_transmission = float(value)
self._beamline_setup_hwobj.energyscan_hwobj.set_max_transmission(
max_transmission)
except:
pass
class CreateEnergyScanWidget(CreateTaskBase):
"""
Descript. :
"""
def __init__(self, parent=None, name=None, fl=0):
"""
Descript. :
"""
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')
# Layout --------------------------------------------------------------
self.main_layout = QtGui.QVBoxLayout(self)
self.main_layout.addWidget(self._periodic_table_widget)
self.main_layout.addWidget(self._data_path_widget)
self.main_layout.setContentsMargins(0, 0, 0, 0)
self.main_layout.setSpacing(6)
self.main_layout.addStretch(10)
# SizePolicies --------------------------------------------------------
# Qt signal/slot connections ------------------------------------------
self._periodic_table_widget.elementEdgeSelectedSignal.\
connect(self.acq_parameters_changed)
self._data_path_widget.pathTemplateChangedSignal.\
connect(self.acq_parameters_changed)
def init_models(self):
"""
Descript. :
"""
CreateTaskBase.init_models(self)
self.enery_scan = queue_model_objects.EnergyScan()
self._path_template.start_num = 1
self._path_template.num_files = 1
self._path_template.suffix = 'raw'
def set_energy_scan_hwobj(self, energy_scan_hwobj):
"""
Descript. :
"""
self._periodic_table_widget.set_elements(\
energy_scan_hwobj.getElements())
def single_item_selection(self, tree_item):
"""
Descript. :
"""
CreateTaskBase.single_item_selection(self, tree_item)
escan_model = tree_item.get_model()
if isinstance(tree_item, Qt4_queue_item.EnergyScanQueueItem):
if tree_item.get_model().is_executed():
self.setDisabled(True)
else:
self.setDisabled(False)
if escan_model.get_path_template():
self._path_template = escan_model.get_path_template()
self._data_path_widget.update_data_model(self._path_template)
elif not(isinstance(tree_item, Qt4_queue_item.SampleQueueItem) or \
isinstance(tree_item, Qt4_queue_item.DataCollectionGroupQueueItem)):
self.setDisabled(True)
def approve_creation(self):
"""
Descript. :
"""
base_result = CreateTaskBase.approve_creation(self)
selected_element, selected_edge = self._periodic_table_widget.get_selected_element_edge(
)
if not selected_element:
logging.getLogger("GUI").\
info("No element selected, please select an element.")
return base_result and selected_element
# Called by the owning widget (task_toolbox_widget) to create
# a collection. When a data collection group is selected.
def _create_task(self, sample, shape):
"""
Descript. :
"""
data_collections = []
selected_element, selected_edge = self._periodic_table_widget.get_selected_element_edge(
)
if selected_element:
if not shape:
cpos = queue_model_objects.CentredPosition()
cpos.snapshot_image = self._graphics_manager_hwobj.get_scene_snapshot(
)
else:
# Shapes selected and sample is mounted, get the
# centred positions for the shapes
if isinstance(shape, GraphicsItemPoint):
snapshot = self._graphics_manager_hwobj.\
get_scene_snapshot(shape)
cpos = copy.deepcopy(shape.get_centred_position())
cpos.snapshot_image = snapshot
path_template = self._create_path_template(sample,
self._path_template)
energy_scan = queue_model_objects.EnergyScan(
sample, path_template, cpos)
energy_scan.set_name(path_template.get_prefix())
energy_scan.set_number(path_template.run_number)
energy_scan.element_symbol = selected_element
energy_scan.edge = selected_edge
data_collections.append(energy_scan)
self._path_template.run_number += 1
else:
logging.getLogger("GUI").\
info("No element selected, please select an element.")
return data_collections
def element_edge_selected(self, element, edge):
if len(self._current_selected_items) == 1:
item = self._current_selected_items[0]
if isinstance(item, Qt4_queue_item.EnergyScanQueueItem):
item.get_model().element_symbol = str(element)
item.get_model().edge = str(edge)
class CreateEnergyScanWidget(CreateTaskBase):
"""
Descript. :
"""
def __init__(self, parent=None,name=None, fl=0):
"""
Descript. :
"""
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 = QGroupBox('Parameters', self)
self._adjust_transmission_cbox = QCheckBox(\
"Adjust transmission", _parameters_gbox)
self._adjust_transmission_cbox.setChecked(False)
self._adjust_transmission_cbox.setEnabled(True)
self._max_transmission_label = QLabel("Maximum transmission:")
self._max_transmission_ledit = QLineEdit("20", _parameters_gbox)
self._max_transmission_ledit.setFixedWidth(80)
self._max_transmission_ledit.setEnabled(False)
# Layout --------------------------------------------------------------
_parameters_gbox_hlayout = 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 = QVBoxLayout(self)
_main_vlayout.addWidget(self._periodic_table_widget)
_main_vlayout.addWidget(self._data_path_widget)
_main_vlayout.addWidget(_parameters_gbox)
_main_vlayout.setContentsMargins(2, 2, 2, 2)
_main_vlayout.setSpacing(6)
_main_vlayout.addStretch(10)
# SizePolicies --------------------------------------------------------
# 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)
def set_expert_mode(self, state):
self._adjust_transmission_cbox.setEnabled(state)
self._max_transmission_label.setEnabled(state)
self._max_transmission_ledit.setEnabled(state)
def enable_compression(self, state):
CreateTaskBase.enable_compression(self, False)
def set_beamline_setup(self, bl_setup_hwobj):
CreateTaskBase.set_beamline_setup(self, bl_setup_hwobj)
self._periodic_table_widget.set_elements(\
self._beamline_setup_hwobj.energyscan_hwobj.getElements())
self.enable_compression(False)
try:
max_transmission_value = self._beamline_setup_hwobj.\
energyscan_hwobj.get_max_transmission_value()
self._adjust_transmission_cbox.setEnabled(True)
self._adjust_transmission_cbox.setChecked(True)
self._beamline_setup_hwobj.energyscan_hwobj.adjust_transmission(True)
if max_transmission_value:
self._max_transmission_ledit.setText("%.2f" % max_transmission_value)
except:
pass
def init_models(self):
"""
Descript. :
"""
CreateTaskBase.init_models(self)
self.enery_scan = queue_model_objects.EnergyScan()
self._path_template.start_num = 1
self._path_template.num_files = 1
self._path_template.suffix = 'raw'
self._path_template.compression = False
def single_item_selection(self, tree_item):
"""
Descript. :
"""
CreateTaskBase.single_item_selection(self, tree_item)
escan_model = tree_item.get_model()
if isinstance(tree_item, Qt4_queue_item.EnergyScanQueueItem):
if tree_item.get_model().is_executed():
self.setDisabled(True)
else:
self.setDisabled(False)
if escan_model.get_path_template():
self._path_template = escan_model.get_path_template()
self._data_path_widget.update_data_model(self._path_template)
elif not(isinstance(tree_item, Qt4_queue_item.SampleQueueItem) or \
isinstance(tree_item, Qt4_queue_item.DataCollectionGroupQueueItem)):
self.setDisabled(True)
def approve_creation(self):
"""
Descript. :
"""
base_result = CreateTaskBase.approve_creation(self)
selected_element, selected_edge = self._periodic_table_widget.get_selected_element_edge()
if not selected_element:
logging.getLogger("GUI").\
warning("No element selected, please select an element.")
return base_result and selected_element
# Called by the owning widget (task_toolbox_widget) to create
# a collection. When a data collection group is selected.
def _create_task(self, sample, shape):
"""
Descript. :
"""
data_collections = []
selected_element, selected_edge = self._periodic_table_widget.get_selected_element_edge()
if selected_element:
if not shape:
cpos = queue_model_objects.CentredPosition()
cpos.snapshot_image = self._graphics_manager_hwobj.get_scene_snapshot()
else:
# Shapes selected and sample is mounted, get the
# centred positions for the shapes
if isinstance(shape, GraphicsItemPoint):
snapshot = self._graphics_manager_hwobj.\
get_scene_snapshot(shape)
cpos = copy.deepcopy(shape.get_centred_position())
cpos.snapshot_image = snapshot
path_template = self._create_path_template(sample, self._path_template)
energy_scan = queue_model_objects.EnergyScan(sample,
path_template,
cpos)
energy_scan.set_name(path_template.get_prefix())
energy_scan.set_number(path_template.run_number)
energy_scan.element_symbol = selected_element
energy_scan.edge = selected_edge
data_collections.append(energy_scan)
self._path_template.run_number += 1
else:
logging.getLogger("GUI").\
info("No element selected, please select an element.")
return data_collections
def element_edge_selected(self, element, edge):
if len(self._current_selected_items) == 1:
item = self._current_selected_items[0]
if isinstance(item, Qt4_queue_item.EnergyScanQueueItem):
item.get_model().element_symbol = str(element)
item.get_model().edge = str(edge)
def adjust_transmission_state_changed(self, state):
self._max_transmission_ledit.setEnabled(state)
self._beamline_setup_hwobj.energyscan_hwobj.adjust_transmission(state)
def max_transmission_value_changed(self, value):
try:
max_transmission = float(value)
self._beamline_setup_hwobj.energyscan_hwobj.set_max_transmission(max_transmission)
except:
pass
