def __init__(self, parent=None, state=None, settings=None):
super(StitcherWidget, self).__init__(parent, state, settings)
class DataFrame(QtGui.QFrame, ui.ui_stitcher.Ui_Frame):
def __init__(self, parent=None):
QtGui.QFrame.__init__(self, parent)
self.setupUi(self)
self._content = DataFrame(self)
self.initialize_content()
self._layout.addWidget(self._content)
# General GUI settings
if settings is None:
settings = GeneralSettings()
self._settings = settings
# Connect do UI data update
self._settings.data_updated.connect(self._data_updated)
self._low_q_data = None
self._medium_q_data = None
self._high_q_data = None
# Flag to know when a field has been modified by hand
self._low_q_modified = False
self._medium_q_modified = False
self._high_q_modified = False
self._referenceID = 0
self._graph = "StitchedData"
self._output_dir = self._settings.data_output_dir
self._stitcher = None
self._plotted = False
def __init__(self,
parent=None,
state=None,
settings=None,
name="BIOSANS",
data_proxy=None):
super(SANSInstrumentWidget, self).__init__(parent,
state,
settings,
data_proxy=data_proxy)
class SummaryFrame(QtGui.QFrame, ui.sans.ui_hfir_instrument.Ui_Frame):
def __init__(self, parent=None):
QtGui.QFrame.__init__(self, parent)
self.setupUi(self)
self._summary = SummaryFrame(self)
self.initialize_content()
self._layout.addWidget(self._summary)
self._masked_detectors = []
if state is not None:
self.set_state(state)
else:
instr = ReductionOptions()
instr.instrument_name = name
self.set_state(instr)
# General GUI settings
if settings is None:
settings = GeneralSettings()
self._settings = settings
# Connect do UI data update
self._settings.data_updated.connect(self._data_updated)
def __init__(self,
parent=None,
state=None,
settings=None,
catalog_cls=None):
super(SANSCatalogWidget, self).__init__(parent, state, settings)
self._catalog_cls = catalog_cls
class DataFrame(QtGui.QFrame, ui.ui_data_catalog.Ui_Frame):
def __init__(self, parent=None):
QtGui.QFrame.__init__(self, parent)
self.setupUi(self)
self._content = DataFrame(self)
self.initialize_content()
self._layout.addWidget(self._content)
# General GUI settings
if settings is None:
settings = GeneralSettings()
self._settings = settings
# Connect do UI data update
self._settings.data_updated.connect(self._data_updated)
def __init__(self, parent=None, state=None, settings=None, data_type=None, ui_class=None, data_proxy=None):
QtGui.QWidget.__init__(self, parent)
self._layout = QtGui.QHBoxLayout()
self.setLayout(self._layout)
if ui_class is not None:
self._content = ui_class(self)
self._layout.addWidget(self._content)
# Data filter for file dialog
self._data_type="Data files (*.xml)"
if data_type is not None:
self._data_type = data_type
# General GUI settings
if settings is None:
settings = GeneralSettings()
self._settings = settings
if ui_class is not None:
self.setLayout(self._layout)
self.initialize_content()
self._instrument_view = None
self._data_set_viewed = ''
self._data_proxy = data_proxy
self._in_mantidplot = IS_IN_MANTIDPLOT and self._data_proxy is not None
self._is_running = True
def __init__(self,
parent=None,
state=None,
settings=None,
catalog_cls=None):
super(SANSCatalogWidget, self).__init__(parent, state, settings)
self._catalog_cls = catalog_cls
class DataFrame(QFrame):
def __init__(self, parent=None):
QFrame.__init__(self, parent)
self.ui = load_ui(__file__,
'../../../ui/data_catalog.ui',
baseinstance=self)
self._content = DataFrame(self)
self.initialize_content()
self._layout.addWidget(self._content)
# General GUI settings
if settings is None:
settings = GeneralSettings()
self._settings = settings
# Connect do UI data update
self._settings.data_updated.connect(self._data_updated)
def __init__(self, instrument=None, instrument_list=None):
QtGui.QMainWindow.__init__(self)
if REDUCTION_WARNING:
message = "The reduction application has problems starting:\n\n"
message += WARNING_MESSAGE
QtGui.QMessageBox.warning(self, "WARNING", message)
# Application settings
settings = QtCore.QSettings()
# Name handle for the instrument
if instrument is None:
instrument = unicode(settings.value("instrument_name", QtCore.QVariant("")).toString())
self._instrument = instrument
# List of allowed instrument
self._instrument_list = instrument_list
# Reduction interface
self._interface = None
# Recent files
self._recent_files = settings.value("recent_files", QtCore.QVariant([])).toStringList()
# Folder to open files in
self._last_directory = unicode(settings.value("last_directory", QtCore.QVariant(".")).toString())
self._last_export_directory = unicode(settings.value("last_export_directory", QtCore.QVariant(".")).toString())
# Current file name
self._filename = None
# Internal flag for clearing all settings and restarting the application
self._clear_and_restart = False
# General settings shared by all widgets
self.general_settings = GeneralSettings(settings)
self.setupUi(self)
# Event connections
if not HAS_MANTID:
self.reduce_button.hide()
self.connect(self.export_button, QtCore.SIGNAL("clicked()"), self._export)
self.connect(self.reduce_button, QtCore.SIGNAL("clicked()"), self.reduce_clicked)
self.connect(self.save_button, QtCore.SIGNAL("clicked()"), self._save)
self.connect(self.interface_chk, QtCore.SIGNAL("clicked(bool)"), self._interface_choice)
self.interface_chk.setChecked(self.general_settings.advanced)
# Of the widgets that are part of the application, one is the ApplicationWindow.
# The ApplicationWindow will send a shutting_down() signal when quitting,
# after which we should close this window.
# Note: there is no way to identify which Widget is the ApplicationWindow.
for w in QtCore.QCoreApplication.instance().topLevelWidgets():
self.connect(w, QtCore.SIGNAL("shutting_down()"), self.close)
self.general_settings.progress.connect(self._progress_updated)
def __init__(self, parent=None, state=None, settings=None):
super(RemoteJobsWidget, self).__init__(parent, state, settings)
class DataFrame(QtGui.QFrame, ui.ui_cluster_status.Ui_Frame):
def __init__(self, parent=None):
QtGui.QFrame.__init__(self, parent)
self.setupUi(self)
self._content = DataFrame(self)
self.initialize_content()
self._layout.addWidget(self._content)
# General GUI settings
if settings is None:
settings = GeneralSettings()
self._settings = settings
def __init__(self,
parent=None,
state=None,
settings=None,
name="EQSANS",
data_type=None,
data_proxy=None):
super(SANSInstrumentWidget, self).__init__(parent,
state,
settings,
data_type=data_type,
data_proxy=data_proxy)
class SummaryFrame(QFrame):
def __init__(self, parent=None):
QFrame.__init__(self, parent)
self.ui = load_ui(__file__,
'../../../ui/sans/eqsans_instrument.ui',
baseinstance=self)
self._summary = SummaryFrame(self)
self.initialize_content()
self._layout.addWidget(self._summary)
self._masked_detectors = []
if state is not None:
self.set_state(state)
else:
instr = ReductionOptions()
instr.instrument_name = name
self.set_state(instr)
# General GUI settings
if settings is None:
settings = GeneralSettings()
self._settings = settings
# Connect do UI data update
self._settings.data_updated.connect(self._data_updated)
def __init__(self, instrument=None, instrument_list=None):
QtGui.QMainWindow.__init__(self)
if REDUCTION_WARNING:
message = "The reduction application has problems starting:\n\n"
message += WARNING_MESSAGE
QtGui.QMessageBox.warning(self, "WARNING", message)
# Application settings
settings = QtCore.QSettings()
# Name handle for the instrument
if instrument is None:
instrument = unicode(settings.value("instrument_name", ''))
if instrument_list is not None and instrument not in instrument_list:
instrument = None
self._instrument = instrument
self._facility = None
# List of allowed instrument
self._instrument_list = instrument_list
# Reduction interface
self._interface = None
# Recent files
self._recent_files = settings.value("recent_files", [])
if self._recent_files is None: # An empty list saved to QSettings comes back as 'None'
self._recent_files = []
# Folder to open files in
self._last_directory = unicode(settings.value("last_directory", '.'))
self._last_export_directory = unicode(settings.value("last_export_directory", '.'))
# Current file name
self._filename = None
# Cluster credentials and options
self._cluster_details_set = False
self._number_of_nodes = 1
self._cores_per_node = 16
self._compute_resources = ['Fermi']
if IS_IN_MANTIDPLOT \
and hasattr(ConfigService.Instance().getFacility(), "computeResources"):
self._compute_resources = ConfigService.Instance().getFacility().computeResources()
# Internal flag for clearing all settings and restarting the application
self._clear_and_restart = False
# General settings shared by all widgets
self.general_settings = GeneralSettings(settings)
self.setupUi(self)
# Event connections
if not IS_IN_MANTIDPLOT:
self.reduce_button.hide()
self.cluster_button.hide()
self.connect(self.export_button, QtCore.SIGNAL("clicked()"), self._export)
self.connect(self.reduce_button, QtCore.SIGNAL("clicked()"), self.reduce_clicked)
self.connect(self.save_button, QtCore.SIGNAL("clicked()"), self._save)
self.connect(self.interface_chk, QtCore.SIGNAL("clicked(bool)"), self._interface_choice)
self.interface_chk.setChecked(self.general_settings.advanced)
# Of the widgets that are part of the application, one is the ApplicationWindow.
# The ApplicationWindow will send a shutting_down() signal when quitting,
# after which we should close this window.
# Note: there is no way to identify which Widget is the ApplicationWindow.
for w in QtCore.QCoreApplication.instance().topLevelWidgets():
self.connect(w, QtCore.SIGNAL("shutting_down()"), self.close)
self.general_settings.progress.connect(self._progress_updated)
class ReductionGUI(QtGui.QMainWindow, ui.ui_reduction_main.Ui_SANSReduction):
def __init__(self, instrument=None, instrument_list=None):
QtGui.QMainWindow.__init__(self)
# Application settings
settings = QtCore.QSettings()
# Name handle for the instrument
if instrument is None:
instrument = unicode(settings.value("instrument_name", QtCore.QVariant('')).toString())
self._instrument = instrument
# List of allowed instrument
self._instrument_list = instrument_list
# Reduction interface
self._interface = None
# Recent files
self._recent_files = settings.value("recent_files", QtCore.QVariant([])).toStringList()
# Folder to open files in
self._last_directory = unicode(settings.value("last_directory", QtCore.QVariant('.')).toString())
self._last_export_directory = unicode(settings.value("last_export_directory", QtCore.QVariant('.')).toString())
# Current file name
self._filename = None
# Internal flag for clearing all settings and restarting the application
self._clear_and_restart = False
# General settings shared by all widgets
self.general_settings = GeneralSettings(settings)
self.setupUi(self)
# Event connections
if not HAS_MANTID:
self.reduce_button.hide()
self.connect(self.export_button, QtCore.SIGNAL("clicked()"), self._export)
self.connect(self.reduce_button, QtCore.SIGNAL("clicked()"), self.reduce_clicked)
# Of the widgets that are part of the application, one is the ApplicationWindow.
# The ApplicationWindow will send a shutting_down() signal when quitting,
# after which we should close this window.
# Note: there is no way to identify which Widget is the ApplicationWindow.
for w in QtCore.QCoreApplication.instance().topLevelWidgets():
self.connect(w, QtCore.SIGNAL("shutting_down()"), self.close)
self.general_settings.progress.connect(self._progress_updated)
def _set_window_title(self):
"""
Sets the window title using the instrument name and the
current settings file
"""
title = "%s Reduction" % self._instrument
if self._filename is not None:
title += ": %s" % self._filename
self.setWindowTitle(title)
def _progress_updated(self, value):
self.progress_bar.setValue(value)
def setup_layout(self):
"""
Sets up the instrument-specific part of the UI layout
"""
if self._instrument == '':
self._change_instrument()
if self._instrument == '':
self.close()
return
self._update_file_menu()
# Clean up the widgets that have already been created
self.tabWidget.clear()
if self._interface is not None:
self._interface.destroy()
self._interface = instrument_factory(self._instrument, settings=self.general_settings)
if self._interface is not None:
tab_list = self._interface.get_tabs()
for tab in tab_list:
self.tabWidget.addTab(tab[1], tab[0])
self._set_window_title()
self.progress_bar.hide()
else:
self.close()
def _update_file_menu(self):
"""
Set up the File menu and update the menu with recent files
"""
self.file_menu.clear()
newAction = QtGui.QAction("&New...", self)
newAction.setShortcut("Ctrl+N")
newAction.setStatusTip("Start a new reduction")
self.connect(newAction, QtCore.SIGNAL("triggered()"), self._new)
openAction = QtGui.QAction("&Open...", self)
openAction.setShortcut("Ctrl+O")
openAction.setStatusTip("Open an XML file containing reduction parameters")
self.connect(openAction, QtCore.SIGNAL("triggered()"), self._file_open)
saveAsAction = QtGui.QAction("Save as...", self)
saveAsAction.setStatusTip("Save the reduction parameters to XML")
self.connect(saveAsAction, QtCore.SIGNAL("triggered()"), self._save_as)
saveAction = QtGui.QAction("&Save...", self)
saveAction.setShortcut("Ctrl+S")
saveAction.setStatusTip("Save the reduction parameters to XML")
self.connect(saveAction, QtCore.SIGNAL("triggered()"), self._save)
exportAction = QtGui.QAction("&Export...", self)
exportAction.setShortcut("Ctrl+E")
exportAction.setStatusTip("Export to python script for Mantid")
self.connect(exportAction, QtCore.SIGNAL("triggered()"), self._export)
quitAction = QtGui.QAction("&Quit", self)
quitAction.setShortcut("Ctrl+Q")
self.connect(quitAction, QtCore.SIGNAL("triggered()"), self.close)
self.file_menu.addAction(newAction)
self.file_menu.addAction(openAction)
self.file_menu.addAction(saveAction)
self.file_menu.addAction(saveAsAction)
self.file_menu.addAction(exportAction)
self.file_menu.addSeparator()
if self.general_settings.debug:
clearAction = QtGui.QAction("&Clear settings and quit", self)
clearAction.setStatusTip("Restore initial application settings and close the application")
self.connect(clearAction, QtCore.SIGNAL("triggered()"), self._clear_and_close)
self.file_menu.addAction(clearAction)
self.file_menu.addAction(quitAction)
# TOOLS menu
instrAction = QtGui.QAction("Change &instrument...", self)
instrAction.setShortcut("Ctrl+I")
instrAction.setStatusTip("Select a new instrument")
self.connect(instrAction, QtCore.SIGNAL("triggered()"), self._change_instrument)
debug_menu_item_str = "Turn debug mode ON"
if self.general_settings.debug:
debug_menu_item_str = "Turn debug mode OFF"
debugAction = QtGui.QAction(debug_menu_item_str, self)
debugAction.setStatusTip(debug_menu_item_str)
self.connect(debugAction, QtCore.SIGNAL("triggered()"), self._debug_mode)
self.tools_menu.clear()
self.tools_menu.addAction(instrAction)
self.tools_menu.addAction(debugAction)
recent_files = []
for fname in self._recent_files:
if fname != self._filename and QtCore.QFile.exists(fname) and not fname in recent_files:
recent_files.append(fname)
if len(recent_files)>0:
self.file_menu.addSeparator()
for i, fname in enumerate(recent_files):
action = QtGui.QAction("&%d %s" % (i+1, QtCore.QFileInfo(fname).fileName()), self)
action.setData(QtCore.QVariant(fname))
self.connect(action, QtCore.SIGNAL("triggered()"), self.open_file)
self.file_menu.addAction(action)
def _debug_mode(self, mode=None):
"""
Set debug mode
@param mode: debug mode (True or False). If None, the debug mode will simply be flipped
"""
if mode is None:
mode = not self.general_settings.debug
self.general_settings.debug = mode
self._new()
self.setup_layout()
def _change_instrument(self):
"""
Invoke an instrument selection dialog
"""
class InstrDialog(QtGui.QDialog, ui.ui_instrument_dialog.Ui_Dialog):
def __init__(self, instrument_list=None):
QtGui.QDialog.__init__(self)
self.instrument_list = instrument_list
self.setupUi(self)
self.instr_combo.clear()
self.facility_combo.clear()
instruments = INSTRUMENT_DICT.keys()
instruments.sort()
for facility in instruments:
self.facility_combo.addItem(QtGui.QApplication.translate("Dialog", facility, None, QtGui.QApplication.UnicodeUTF8))
self._facility_changed(instruments[0])
self.connect(self.facility_combo, QtCore.SIGNAL("activated(QString)"), self._facility_changed)
def _facility_changed(self, facility):
self.instr_combo.clear()
for item in INSTRUMENT_DICT[unicode(facility)]:
if self.instrument_list is None or item in self.instrument_list:
self.instr_combo.addItem(QtGui.QApplication.translate("Dialog", item, None, QtGui.QApplication.UnicodeUTF8))
if self.general_settings.debug:
dialog = InstrDialog()
else:
dialog = InstrDialog(self._instrument_list)
dialog.exec_()
if dialog.result()==1:
self._instrument = dialog.instr_combo.currentText()
self.setup_layout()
self._new()
def _clear_and_close(self):
"""
Clear all QSettings parameters
"""
self._clear_and_restart = True
self.close()
# If we make it here, the user canceled the close, which
# means that we need to reset the clear&close flag so
# that the state is properly saved on the next close.
self._clear_and_restart = False
def closeEvent(self, event):
"""
Executed when the application closes
"""
if False:
reply = QtGui.QMessageBox.question(self, 'Message',
"Are you sure you want to quit this application?", QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
if reply == QtGui.QMessageBox.Yes:
event.accept()
else:
event.ignore()
# Save application settings
if self._clear_and_restart:
self._clear_and_restart = False
QtCore.QSettings().clear()
else:
settings = QtCore.QSettings()
if self._instrument is not None:
instrument = QtCore.QVariant(QtCore.QString(self._instrument))
else:
instrument = QtCore.QVariant()
settings.setValue("instrument_name", instrument)
if self._filename is not None:
filename = QtCore.QVariant(QtCore.QString(self._filename))
else:
filename = QtCore.QVariant()
settings.setValue("last_file", filename)
if self._recent_files is not []:
recent_files = QtCore.QVariant(self._recent_files)
else:
recent_files = QtCore.QVariant()
settings.setValue("recent_files", recent_files)
last_dir = QtCore.QVariant(QtCore.QString(self._last_directory))
settings.setValue("last_directory", last_dir)
last_export_dir = QtCore.QVariant(QtCore.QString(self._last_export_directory))
settings.setValue("last_export_directory", last_export_dir)
# General settings
self.general_settings.to_settings(settings)
def reduce_clicked(self):
"""
Create an object capable of using the information in the
interface and turn it into a reduction process.
"""
self.reduce_button.setEnabled(False)
self.export_button.setEnabled(False)
if IS_IN_MANTIDPLOT:
qti.app.mantidUI.setIsRunning(True)
self._interface.reduce()
if IS_IN_MANTIDPLOT:
qti.app.mantidUI.setIsRunning(False)
self.reduce_button.setEnabled(True)
self.export_button.setEnabled(True)
def open_file(self, file_path=None):
"""
Open an XML file and populate the UI
@param file_path: path to the file to be loaded
"""
if file_path is None:
action = self.sender()
if isinstance(action, QtGui.QAction):
file_path = unicode(action.data().toString())
# Check whether the file describes the current instrument
found_instrument = self._interface.scripter.verify_instrument(file_path)
if not found_instrument == self._instrument:
self._instrument = found_instrument
self.setup_layout()
self.reduce_button.setEnabled(False)
self.export_button.setEnabled(False)
self._interface.load_file(file_path)
self.reduce_button.setEnabled(True)
self.export_button.setEnabled(True)
self._filename = file_path
self._update_file_menu()
self._set_window_title()
if file_path in self._recent_files:
self._recent_files.removeAll(file_path)
self._recent_files.prepend(file_path)
while self._recent_files.count() > 10:
self._recent_files.takeLast()
def _new(self, *argv):
"""
Start new reduction
"""
self._interface.reset()
self._filename = None
self._update_file_menu()
self._set_window_title()
def _file_open(self, *argv):
"""
File chooser for loading UI parameters
"""
fname_qstr = QtGui.QFileDialog.getOpenFileName(self, "Reduction settings - Choose a settings file",
self._last_directory,
"Settings files (*.xml)")
fname = str(QtCore.QFileInfo(fname_qstr).filePath())
if fname:
# Store the location of the loaded file
self._last_directory = str(QtCore.QFileInfo(fname_qstr).path())
self.open_file(fname)
def _save(self):
"""
Present a file dialog to the user and saves the content of the
UI in XML format
"""
if self._filename is None:
self._save_as()
else:
try:
self._interface.save_file(self._filename)
self._update_file_menu()
self.statusBar().showMessage("Saved as %s" % self._filename)
self._set_window_title()
except:
#TODO: put this in a log window, and in a file
print sys.exc_value
self.statusBar().showMessage("Failed to save %s" % self._filename)
def _save_as(self):
"""
Present a file dialog to the user and saves the content of
the UI in XML format.
"""
if self._filename is not None:
fname = self._filename
else:
fname = '.'
fname_qstr = QtGui.QFileDialog.getSaveFileName(self, "Reduction settings - Save settings",
self._last_directory,
"Settings files (*.xml)")
fname = str(QtCore.QFileInfo(fname_qstr).filePath())
if len(fname)>0:
if not fname.endswith('.xml'):
fname += ".xml"
if fname in self._recent_files:
self._recent_files.removeAll(fname)
self._recent_files.prepend(fname)
while self._recent_files.count() > 10:
self._recent_files.takeLast()
self._last_directory = str(QtCore.QFileInfo(fname_qstr).path())
self._filename = fname
self._save()
def _export(self):
"""
Exports the current content of the UI to a python script that can
be run within MantidPlot
"""
fname = '.'
if self._filename is not None:
(root, ext) = os.path.splitext(self._filename)
fname = root
fname = unicode(QtGui.QFileDialog.getSaveFileName(self, "Mantid Python script - Save script",
self._last_export_directory,
"Python script (*.py)"))
if len(fname)>0:
if not fname.endswith('.py'):
fname += ".py"
(folder, file_name) = os.path.split(fname)
self._last_export_directory = folder
script = self._interface.export(fname)
if script is not None:
self.statusBar().showMessage("Saved as %s" % fname)
else:
self.statusBar().showMessage("Could not save file")
def __init__(self, instrument=None, instrument_list=None):
QMainWindow.__init__(self)
self.ui = load_ui(__file__, 'ui/reduction_main.ui', baseinstance=self)
if STARTUP_WARNING:
message = "The reduction application has problems starting:\n\n"
message += STARTUP_WARNING
QMessageBox.warning(self, "WARNING", message)
# Application settings
settings = QSettings()
# Name handle for the instrument
if instrument is None:
instrument = unicode(settings.value("instrument_name", ''))
if instrument_list is not None and instrument not in instrument_list:
instrument = None
self._instrument = instrument
self._facility = None
# List of allowed instrument
self._instrument_list = instrument_list
# Reduction interface
self._interface = None
# Recent files
self._recent_files = settings.value("recent_files", [])
if self._recent_files is None: # An empty list saved to QSettings comes back as 'None'
self._recent_files = []
# Folder to open files in
self._last_directory = unicode(settings.value("last_directory", '.'))
self._last_export_directory = unicode(
settings.value("last_export_directory", '.'))
# Current file name
self._filename = None
# Cluster credentials and options
self._cluster_details_set = False
self._number_of_nodes = 1
self._cores_per_node = 16
self._compute_resources = ['Fermi']
if CAN_REDUCE and hasattr(ConfigService.Instance().getFacility(),
"computeResources"):
self._compute_resources = ConfigService.Instance().getFacility(
).computeResources()
# Internal flag for clearing all settings and restarting the application
self._clear_and_restart = False
# General settings shared by all widgets
self.general_settings = GeneralSettings(settings)
# Event connections
if not CAN_REDUCE:
self.reduce_button.hide()
self.export_button.clicked.connect(self._export)
self.reduce_button.clicked.connect(self.reduce_clicked)
self.save_button.clicked.connect(self._save)
self.interface_chk.clicked.connect(self._interface_choice)
self.interface_chk.setChecked(self.general_settings.advanced)
self.general_settings.progress.connect(self._progress_updated)
def __init__(self, instrument=None, instrument_list=None):
QtGui.QMainWindow.__init__(self)
if REDUCTION_WARNING:
message = "The reduction application has problems starting:\n\n"
message += WARNING_MESSAGE
QtGui.QMessageBox.warning(self, "WARNING", message)
# Application settings
settings = QtCore.QSettings()
# Name handle for the instrument
if instrument is None:
instrument = unicode(settings.value("instrument_name", ''))
if instrument_list is not None and instrument not in instrument_list:
instrument = None
self._instrument = instrument
self._facility = None
# List of allowed instrument
self._instrument_list = instrument_list
# Reduction interface
self._interface = None
# Recent files
self._recent_files = settings.value("recent_files", [])
if self._recent_files is None: # An empty list saved to QSettings comes back as 'None'
self._recent_files = []
# Folder to open files in
self._last_directory = unicode(settings.value("last_directory", '.'))
self._last_export_directory = unicode(settings.value("last_export_directory", '.'))
# Current file name
self._filename = None
# Cluster credentials and options
self._cluster_details_set = False
self._number_of_nodes = 1
self._cores_per_node = 16
self._compute_resources = ['Fermi']
if IS_IN_MANTIDPLOT \
and hasattr(ConfigService.Instance().getFacility(), "computeResources"):
self._compute_resources = ConfigService.Instance().getFacility().computeResources()
# Internal flag for clearing all settings and restarting the application
self._clear_and_restart = False
# General settings shared by all widgets
self.general_settings = GeneralSettings(settings)
self.setupUi(self)
# Event connections
if not IS_IN_MANTIDPLOT:
self.reduce_button.hide()
self.cluster_button.hide()
self.connect(self.export_button, QtCore.SIGNAL("clicked()"), self._export)
self.connect(self.reduce_button, QtCore.SIGNAL("clicked()"), self.reduce_clicked)
self.connect(self.save_button, QtCore.SIGNAL("clicked()"), self._save)
self.connect(self.interface_chk, QtCore.SIGNAL("clicked(bool)"), self._interface_choice)
self.interface_chk.setChecked(self.general_settings.advanced)
# Of the widgets that are part of the application, one is the ApplicationWindow.
# The ApplicationWindow will send a shutting_down() signal when quitting,
# after which we should close this window.
# Note: there is no way to identify which Widget is the ApplicationWindow.
for w in QtCore.QCoreApplication.instance().topLevelWidgets():
self.connect(w, QtCore.SIGNAL("shutting_down()"), self.close)
self.general_settings.progress.connect(self._progress_updated)
class ReductionGUI(QtGui.QMainWindow, ui.ui_reduction_main.Ui_SANSReduction):
def __init__(self, instrument=None, instrument_list=None):
QtGui.QMainWindow.__init__(self)
if REDUCTION_WARNING:
message = "The reduction application has problems starting:\n\n"
message += WARNING_MESSAGE
QtGui.QMessageBox.warning(self, "WARNING", message)
# Application settings
settings = QtCore.QSettings()
# Name handle for the instrument
if instrument is None:
instrument = unicode(settings.value("instrument_name", ''))
if instrument_list is not None and instrument not in instrument_list:
instrument = None
self._instrument = instrument
self._facility = None
# List of allowed instrument
self._instrument_list = instrument_list
# Reduction interface
self._interface = None
# Recent files
self._recent_files = settings.value("recent_files", [])
if self._recent_files is None: # An empty list saved to QSettings comes back as 'None'
self._recent_files = []
# Folder to open files in
self._last_directory = unicode(settings.value("last_directory", '.'))
self._last_export_directory = unicode(settings.value("last_export_directory", '.'))
# Current file name
self._filename = None
# Cluster credentials and options
self._cluster_details_set = False
self._number_of_nodes = 1
self._cores_per_node = 16
self._compute_resources = ['Fermi']
if IS_IN_MANTIDPLOT \
and hasattr(ConfigService.Instance().getFacility(), "computeResources"):
self._compute_resources = ConfigService.Instance().getFacility().computeResources()
# Internal flag for clearing all settings and restarting the application
self._clear_and_restart = False
# General settings shared by all widgets
self.general_settings = GeneralSettings(settings)
self.setupUi(self)
# Event connections
if not IS_IN_MANTIDPLOT:
self.reduce_button.hide()
self.cluster_button.hide()
self.connect(self.export_button, QtCore.SIGNAL("clicked()"), self._export)
self.connect(self.reduce_button, QtCore.SIGNAL("clicked()"), self.reduce_clicked)
self.connect(self.save_button, QtCore.SIGNAL("clicked()"), self._save)
self.connect(self.interface_chk, QtCore.SIGNAL("clicked(bool)"), self._interface_choice)
self.interface_chk.setChecked(self.general_settings.advanced)
# Of the widgets that are part of the application, one is the ApplicationWindow.
# The ApplicationWindow will send a shutting_down() signal when quitting,
# after which we should close this window.
# Note: there is no way to identify which Widget is the ApplicationWindow.
for w in QtCore.QCoreApplication.instance().topLevelWidgets():
self.connect(w, QtCore.SIGNAL("shutting_down()"), self.close)
self.general_settings.progress.connect(self._progress_updated)
def _set_window_title(self):
"""
Sets the window title using the instrument name and the
current settings file
"""
title = "%s Reduction" % self._instrument
if self._filename is not None:
title += ": %s" % self._filename
self.setWindowTitle(title)
def _progress_updated(self, value):
self.progress_bar.setValue(value)
def setup_layout(self, load_last=False):
"""
Sets up the instrument-specific part of the UI layout
"""
# Clean up the widgets that have already been created
self.tabWidget.clear()
self.progress_bar.hide()
if self._instrument == '' or self._instrument is None:
return self._change_instrument()
self._update_file_menu()
if self._interface is not None:
self._interface.destroy()
self.general_settings.instrument_name = self._instrument
# Find corresponding facility
if self._facility is None:
for facility in INSTRUMENT_DICT.keys():
if self._instrument in INSTRUMENT_DICT[facility].keys():
self._facility = facility
break
if self._facility is None:
self._facility = ConfigService.Instance().getFacility().name()
self.general_settings.facility_name = self._facility
self._interface = instrument_factory(self._instrument, settings=self.general_settings)
if self._interface is not None:
tab_list = self._interface.get_tabs()
for tab in tab_list:
self.tabWidget.addTab(tab[1], tab[0])
self._set_window_title()
# Show the "advanced interface" check box if needed
if self._interface.has_advanced_version():
self.interface_chk.show()
else:
self.interface_chk.hide()
# Show the parallel reduction button if enabled
if self._interface.is_cluster_enabled() and IS_IN_MANTIDPLOT \
and CLUSTER_ENABLED:
config = ConfigService.Instance()
if config.hasProperty("cluster.submission") \
and config.getString("cluster.submission").lower()=='on':
self.cluster_button.show()
self.connect(self.cluster_button, QtCore.SIGNAL("clicked()"), self.cluster_clicked)
else:
self.cluster_button.hide()
if load_last:
self._interface.load_last_reduction()
else:
print "Could not generate an interface for instrument %s" % self._instrument
self.close()
return True
def _update_file_menu(self):
"""
Set up the File menu and update the menu with recent files
"""
self.file_menu.clear()
newAction = QtGui.QAction("&New Reduction...", self)
newAction.setShortcut("Ctrl+N")
newAction.setStatusTip("Start a new reduction")
self.connect(newAction, QtCore.SIGNAL("triggered()"), self._new)
openAction = QtGui.QAction("&Open...", self)
openAction.setShortcut("Ctrl+O")
openAction.setStatusTip("Open an XML file containing reduction parameters")
self.connect(openAction, QtCore.SIGNAL("triggered()"), self._file_open)
saveAsAction = QtGui.QAction("Save as...", self)
saveAsAction.setStatusTip("Save the reduction parameters to XML")
self.connect(saveAsAction, QtCore.SIGNAL("triggered()"), self._save_as)
saveAction = QtGui.QAction("&Save...", self)
saveAction.setShortcut("Ctrl+S")
saveAction.setStatusTip("Save the reduction parameters to XML")
self.connect(saveAction, QtCore.SIGNAL("triggered()"), self._save)
exportAction = QtGui.QAction("&Export...", self)
exportAction.setShortcut("Ctrl+E")
exportAction.setStatusTip("Export to python script for Mantid")
self.connect(exportAction, QtCore.SIGNAL("triggered()"), self._export)
quitAction = QtGui.QAction("&Quit", self)
quitAction.setShortcut("Ctrl+Q")
self.connect(quitAction, QtCore.SIGNAL("triggered()"), self.close)
self.file_menu.addAction(newAction)
self.file_menu.addAction(openAction)
self.file_menu.addAction(saveAction)
self.file_menu.addAction(saveAsAction)
self.file_menu.addAction(exportAction)
self.file_menu.addSeparator()
if self.general_settings.debug:
clearAction = QtGui.QAction("&Clear settings and quit", self)
clearAction.setStatusTip("Restore initial application settings and close the application")
self.connect(clearAction, QtCore.SIGNAL("triggered()"), self._clear_and_close)
self.file_menu.addAction(clearAction)
self.file_menu.addAction(quitAction)
# TOOLS menu
instrAction = QtGui.QAction("Change &instrument...", self)
instrAction.setShortcut("Ctrl+I")
instrAction.setStatusTip("Select a new instrument")
self.connect(instrAction, QtCore.SIGNAL("triggered()"), self._change_instrument)
debug_menu_item_str = "Turn debug mode ON"
if self.general_settings.debug:
debug_menu_item_str = "Turn debug mode OFF"
debugAction = QtGui.QAction(debug_menu_item_str, self)
debugAction.setStatusTip(debug_menu_item_str)
self.connect(debugAction, QtCore.SIGNAL("triggered()"), self._debug_mode)
self.tools_menu.clear()
self.tools_menu.addAction(instrAction)
self.tools_menu.addAction(debugAction)
# Cluster submission details
if IS_IN_MANTIDPLOT and CLUSTER_ENABLED:
jobAction = QtGui.QAction("Remote submission details", self)
jobAction.setShortcut("Ctrl+R")
jobAction.setStatusTip("Set the cluster information for remote job submission")
self.connect(jobAction, QtCore.SIGNAL("triggered()"), self._cluster_details_dialog)
self.tools_menu.addAction(jobAction)
recent_files = []
for fname in self._recent_files:
if fname != self._filename and QtCore.QFile.exists(fname) and not fname in recent_files:
recent_files.append(fname)
if len(recent_files)>0:
self.file_menu.addSeparator()
for i, fname in enumerate(recent_files):
action = QtGui.QAction("&%d %s" % (i+1, QtCore.QFileInfo(fname).fileName()), self)
action.setData(fname)
self.connect(action, QtCore.SIGNAL("triggered()"), self.open_file)
self.file_menu.addAction(action)
def _debug_mode(self, mode=None):
"""
Set debug mode
@param mode: debug mode (True or False). If None, the debug mode will simply be flipped
"""
if mode is None:
mode = not self.general_settings.debug
self.general_settings.debug = mode
self._new()
self.setup_layout()
def _interface_choice(self, advanced_ui=None):
if advanced_ui is None:
advanced_ui = self.general_settings.advanced
self.general_settings.advanced = advanced_ui
self._new()
self.setup_layout()
def _change_instrument(self):
"""
Invoke an instrument selection dialog
"""
class InstrDialog(QtGui.QDialog, ui.ui_instrument_dialog.Ui_Dialog):
def __init__(self, instrument_list=None):
QtGui.QDialog.__init__(self)
self.instrument_list = instrument_list
self.setupUi(self)
self.instr_combo.clear()
self.facility_combo.clear()
instruments = INSTRUMENT_DICT.keys()
instruments.sort()
instruments.reverse()
for facility in instruments:
self.facility_combo.addItem(QtGui.QApplication.translate("Dialog", facility, None, QtGui.QApplication.UnicodeUTF8))
self._facility_changed(instruments[0])
self.connect(self.facility_combo, QtCore.SIGNAL("activated(QString)"), self._facility_changed)
def _facility_changed(self, facility):
self.instr_combo.clear()
instr_list = INSTRUMENT_DICT[unicode(facility)].keys()
instr_list.sort()
for item in instr_list:
if self.instrument_list is None or item in self.instrument_list:
self.instr_combo.addItem(QtGui.QApplication.translate("Dialog", item, None, QtGui.QApplication.UnicodeUTF8))
if self.general_settings.debug:
dialog = InstrDialog()
else:
dialog = InstrDialog(self._instrument_list)
dialog.exec_()
if dialog.result()==1:
self._instrument = dialog.instr_combo.currentText()
self._facility = dialog.facility_combo.currentText()
self.setup_layout()
self._new()
return True
else:
self.close()
return False
def _cluster_details_dialog(self):
"""
Show dialog to get cluster submission details
"""
class ClusterDialog(QtGui.QDialog, ui.ui_cluster_details_dialog.Ui_Dialog):
def __init__(self, compute_resources=None):
QtGui.QDialog.__init__(self)
self.setupUi(self)
self.resource_combo.clear()
for res in compute_resources:
self.resource_combo.addItem(QtGui.QApplication.translate("Dialog", res, None, QtGui.QApplication.UnicodeUTF8))
# Fill out the defaults
dialog = ClusterDialog(self._compute_resources)
if self.general_settings.cluster_user is not None:
dialog.username_edit.setText(str(self.general_settings.cluster_user))
dialog.pass_edit.setText(str(self.general_settings.cluster_pass))
dialog.nodes_box.setValue(int(self._number_of_nodes))
dialog.cores_box.setValue(int(self._cores_per_node))
for i in range(dialog.resource_combo.count()):
if dialog.resource_combo.itemText(i)==self.general_settings.compute_resource:
dialog.resource_combo.setCurrentIndex(i)
break
dialog.exec_()
if dialog.result()==1:
self.general_settings.cluster_user = str(dialog.username_edit.text())
self.general_settings.cluster_pass = str(dialog.pass_edit.text())
self._cluster_details_set = True
self._number_of_nodes = int(dialog.nodes_box.value())
self._cores_per_node = int(dialog.cores_box.value())
self.general_settings.compute_resource = dialog.resource_combo.currentText()
def _clear_and_close(self):
"""
Clear all QSettings parameters
"""
self._clear_and_restart = True
self.close()
# If we make it here, the user canceled the close, which
# means that we need to reset the clear&close flag so
# that the state is properly saved on the next close.
self._clear_and_restart = False
def closeEvent(self, event):
"""
Executed when the application closes
"""
if False:
reply = QtGui.QMessageBox.question(self, 'Message',\
"Are you sure you want to quit this application?", QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
if reply == QtGui.QMessageBox.Yes:
event.accept()
else:
event.ignore()
# Save application settings
if self._clear_and_restart:
self._clear_and_restart = False
QtCore.QSettings().clear()
else:
settings = QtCore.QSettings()
settings.setValue("instrument_name", self._instrument)
settings.setValue("last_file", self._filename)
settings.setValue("recent_files", self._recent_files)
settings.setValue("last_directory", str(self._last_directory))
settings.setValue("last_export_directory", str(self._last_export_directory))
# General settings
self.general_settings.to_settings(settings)
def reduce_clicked(self):
"""
Create an object capable of using the information in the
interface and turn it into a reduction process.
"""
self.reduce_button.setEnabled(False)
self.export_button.setEnabled(False)
self.save_button.setEnabled(False)
self.interface_chk.setEnabled(False)
self.file_menu.setEnabled(False)
self.tools_menu.setEnabled(False)
if self._interface is not None:
self._interface.reduce()
self.reduce_button.setEnabled(True)
self.export_button.setEnabled(True)
self.save_button.setEnabled(True)
self.interface_chk.setEnabled(True)
self.file_menu.setEnabled(True)
self.tools_menu.setEnabled(True)
def cluster_clicked(self):
"""
Submit for parallel reduction
"""
if not self._cluster_details_set:
self._cluster_details_dialog()
if self._interface is not None \
and self.general_settings.cluster_user is not None \
and self.general_settings.cluster_pass is not None:
# Chose a name for the job
if self._filename is not None:
job_name = os.path.basename(self._filename).strip()
toks = os.path.splitext(job_name)
job_name = toks[0]
else:
job_name = ''
self._interface.cluster_submit(self.general_settings.cluster_user,
self.general_settings.cluster_pass,
resource=self.general_settings.compute_resource,
nodes=self._number_of_nodes,
cores_per_node=self._cores_per_node,
job_name=job_name)
def open_file(self, file_path=None):
"""
Open an XML file and populate the UI
@param file_path: path to the file to be loaded
"""
if file_path is None:
action = self.sender()
if isinstance(action, QtGui.QAction):
file_path = unicode(action.data())
# Check whether the file describes the current instrument
try:
found_instrument = self._interface.scripter.verify_instrument(file_path)
except:
msg = "The file you attempted to load doesn't have a recognized format.\n\n"
msg += "Please make sure it has been produced by this application."
QtGui.QMessageBox.warning(self, "Error loading reduction parameter file", msg)
print sys.exc_value
return
if not found_instrument == self._instrument:
self._instrument = found_instrument
self.setup_layout()
self.reduce_button.setEnabled(False)
self.export_button.setEnabled(False)
self.save_button.setEnabled(False)
self.interface_chk.setEnabled(False)
self._interface.load_file(file_path)
self.reduce_button.setEnabled(True)
self.export_button.setEnabled(True)
self.save_button.setEnabled(True)
self.interface_chk.setEnabled(True)
self._filename = file_path
self._update_file_menu()
self._set_window_title()
if file_path in self._recent_files:
self._recent_files.remove(file_path)
self._recent_files.insert(0,file_path)
while len(self._recent_files) > 10:
self._recent_files.pop()
def _new(self, *argv):
"""
Start new reduction
"""
self._interface.reset()
self._filename = None
self._update_file_menu()
self._set_window_title()
def _file_open(self, *argv):
"""
File chooser for loading UI parameters
"""
fname_qstr = QtGui.QFileDialog.getOpenFileName(self, "Reduction settings - Choose a settings file",
self._last_directory,
"Settings files (*.xml)")
fname = str(QtCore.QFileInfo(fname_qstr).filePath())
if fname:
# Store the location of the loaded file
self._last_directory = str(QtCore.QFileInfo(fname_qstr).path())
self.open_file(fname)
def _save(self):
"""
Present a file dialog to the user and saves the content of the
UI in XML format
"""
if self._filename is None:
self._save_as()
else:
try:
self._interface.save_file(self._filename)
self._update_file_menu()
self.statusBar().showMessage("Saved as %s" % self._filename)
self._set_window_title()
except:
#TODO: put this in a log window, and in a file
print sys.exc_value
self.statusBar().showMessage("Failed to save %s" % self._filename)
def _save_as(self):
"""
Present a file dialog to the user and saves the content of
the UI in XML format.
"""
if self._filename is not None:
fname = self._filename
else:
fname = self._instrument + '_'
fname_qstr = QtGui.QFileDialog.getSaveFileName(self, "Reduction settings - Save settings",
self._last_directory + '/' + fname,
"Settings files (*.xml)")
fname = str(QtCore.QFileInfo(fname_qstr).filePath())
if len(fname)>0:
if not fname.endswith('.xml'):
fname += ".xml"
if fname in self._recent_files:
self._recent_files.remove(fname)
self._recent_files.insert(0,fname)
while len(self._recent_files) > 10:
self._recent_files.pop()
self._last_directory = str(QtCore.QFileInfo(fname_qstr).path())
self._filename = fname
self._save()
def _export(self):
"""
Exports the current content of the UI to a python script that can
be run within MantidPlot
"""
if self._interface is None:
return
fname = '.'
if self._filename is not None:
(root, ext) = os.path.splitext(self._filename)
fname = root
fname = unicode(QtGui.QFileDialog.getSaveFileName(self, "Mantid Python script - Save script",
self._last_export_directory,
"Python script (*.py)"))
if len(fname)>0:
if not fname.endswith('.py'):
fname += ".py"
(folder, file_name) = os.path.split(fname)
self._last_export_directory = folder
script = self._interface.export(fname)
if script is not None:
self.statusBar().showMessage("Saved as %s" % fname)
else:
self.statusBar().showMessage("Could not save file")
class ReductionGUI(QtGui.QMainWindow, ui.ui_reduction_main.Ui_SANSReduction):
def __init__(self, instrument=None, instrument_list=None):
QtGui.QMainWindow.__init__(self)
if REDUCTION_WARNING:
message = "The reduction application has problems starting:\n\n"
message += WARNING_MESSAGE
QtGui.QMessageBox.warning(self, "WARNING", message)
# Application settings
settings = QtCore.QSettings()
# Name handle for the instrument
if instrument is None:
instrument = unicode(settings.value("instrument_name", ''))
if instrument_list is not None and instrument not in instrument_list:
instrument = None
self._instrument = instrument
self._facility = None
# List of allowed instrument
self._instrument_list = instrument_list
# Reduction interface
self._interface = None
# Recent files
self._recent_files = settings.value("recent_files", [])
if self._recent_files is None: # An empty list saved to QSettings comes back as 'None'
self._recent_files = []
# Folder to open files in
self._last_directory = unicode(settings.value("last_directory", '.'))
self._last_export_directory = unicode(
settings.value("last_export_directory", '.'))
# Current file name
self._filename = None
# Cluster credentials and options
self._cluster_details_set = False
self._number_of_nodes = 1
self._cores_per_node = 16
self._compute_resources = ['Fermi']
if IS_IN_MANTIDPLOT \
and hasattr(ConfigService.Instance().getFacility(), "computeResources"):
self._compute_resources = ConfigService.Instance().getFacility(
).computeResources()
# Internal flag for clearing all settings and restarting the application
self._clear_and_restart = False
# General settings shared by all widgets
self.general_settings = GeneralSettings(settings)
self.setupUi(self)
# Event connections
if not IS_IN_MANTIDPLOT:
self.reduce_button.hide()
self.cluster_button.hide()
self.connect(self.export_button, QtCore.SIGNAL("clicked()"),
self._export)
self.connect(self.reduce_button, QtCore.SIGNAL("clicked()"),
self.reduce_clicked)
self.connect(self.save_button, QtCore.SIGNAL("clicked()"), self._save)
self.connect(self.interface_chk, QtCore.SIGNAL("clicked(bool)"),
self._interface_choice)
self.interface_chk.setChecked(self.general_settings.advanced)
# Of the widgets that are part of the application, one is the ApplicationWindow.
# The ApplicationWindow will send a shutting_down() signal when quitting,
# after which we should close this window.
# Note: there is no way to identify which Widget is the ApplicationWindow.
for w in QtCore.QCoreApplication.instance().topLevelWidgets():
self.connect(w, QtCore.SIGNAL("shutting_down()"), self.close)
self.general_settings.progress.connect(self._progress_updated)
def _set_window_title(self):
"""
Sets the window title using the instrument name and the
current settings file
"""
title = "%s Reduction" % self._instrument
if self._filename is not None:
title += ": %s" % self._filename
self.setWindowTitle(title)
def _progress_updated(self, value):
self.progress_bar.setValue(value)
def setup_layout(self, load_last=False):
"""
Sets up the instrument-specific part of the UI layout
"""
# Clean up the widgets that have already been created
self.tabWidget.clear()
self.progress_bar.hide()
if self._instrument == '' or self._instrument is None:
return self._change_instrument()
self._update_file_menu()
if self._interface is not None:
self._interface.destroy()
self.general_settings.instrument_name = self._instrument
# Find corresponding facility
if self._facility is None:
for facility in INSTRUMENT_DICT.keys():
if self._instrument in INSTRUMENT_DICT[facility].keys():
self._facility = facility
break
if self._facility is None:
self._facility = ConfigService.Instance().getFacility().name()
self.general_settings.facility_name = self._facility
self._interface = instrument_factory(self._instrument,
settings=self.general_settings)
if self._interface is not None:
tab_list = self._interface.get_tabs()
for tab in tab_list:
self.tabWidget.addTab(tab[1], tab[0])
self._set_window_title()
# Show the "advanced interface" check box if needed
if self._interface.has_advanced_version():
self.interface_chk.show()
else:
self.interface_chk.hide()
# Show the parallel reduction button if enabled
if self._interface.is_cluster_enabled() and IS_IN_MANTIDPLOT \
and CLUSTER_ENABLED:
config = ConfigService.Instance()
if config.hasProperty("cluster.submission") \
and config.getString("cluster.submission").lower()=='on':
self.cluster_button.show()
self.connect(self.cluster_button,
QtCore.SIGNAL("clicked()"),
self.cluster_clicked)
else:
self.cluster_button.hide()
if load_last:
self._interface.load_last_reduction()
else:
print "Could not generate an interface for instrument %s" % self._instrument
self.close()
return True
def _update_file_menu(self):
"""
Set up the File menu and update the menu with recent files
"""
self.file_menu.clear()
newAction = QtGui.QAction("&New Reduction...", self)
newAction.setShortcut("Ctrl+N")
newAction.setStatusTip("Start a new reduction")
self.connect(newAction, QtCore.SIGNAL("triggered()"), self._new)
openAction = QtGui.QAction("&Open...", self)
openAction.setShortcut("Ctrl+O")
openAction.setStatusTip(
"Open an XML file containing reduction parameters")
self.connect(openAction, QtCore.SIGNAL("triggered()"), self._file_open)
saveAsAction = QtGui.QAction("Save as...", self)
saveAsAction.setStatusTip("Save the reduction parameters to XML")
self.connect(saveAsAction, QtCore.SIGNAL("triggered()"), self._save_as)
saveAction = QtGui.QAction("&Save...", self)
saveAction.setShortcut("Ctrl+S")
saveAction.setStatusTip("Save the reduction parameters to XML")
self.connect(saveAction, QtCore.SIGNAL("triggered()"), self._save)
exportAction = QtGui.QAction("&Export...", self)
exportAction.setShortcut("Ctrl+E")
exportAction.setStatusTip("Export to python script for Mantid")
self.connect(exportAction, QtCore.SIGNAL("triggered()"), self._export)
quitAction = QtGui.QAction("&Quit", self)
quitAction.setShortcut("Ctrl+Q")
self.connect(quitAction, QtCore.SIGNAL("triggered()"), self.close)
self.file_menu.addAction(newAction)
self.file_menu.addAction(openAction)
self.file_menu.addAction(saveAction)
self.file_menu.addAction(saveAsAction)
self.file_menu.addAction(exportAction)
self.file_menu.addSeparator()
if self.general_settings.debug:
clearAction = QtGui.QAction("&Clear settings and quit", self)
clearAction.setStatusTip(
"Restore initial application settings and close the application"
)
self.connect(clearAction, QtCore.SIGNAL("triggered()"),
self._clear_and_close)
self.file_menu.addAction(clearAction)
self.file_menu.addAction(quitAction)
# TOOLS menu
instrAction = QtGui.QAction("Change &instrument...", self)
instrAction.setShortcut("Ctrl+I")
instrAction.setStatusTip("Select a new instrument")
self.connect(instrAction, QtCore.SIGNAL("triggered()"),
self._change_instrument)
debug_menu_item_str = "Turn debug mode ON"
if self.general_settings.debug:
debug_menu_item_str = "Turn debug mode OFF"
debugAction = QtGui.QAction(debug_menu_item_str, self)
debugAction.setStatusTip(debug_menu_item_str)
self.connect(debugAction, QtCore.SIGNAL("triggered()"),
self._debug_mode)
self.tools_menu.clear()
self.tools_menu.addAction(instrAction)
self.tools_menu.addAction(debugAction)
# Cluster submission details
if IS_IN_MANTIDPLOT and CLUSTER_ENABLED:
jobAction = QtGui.QAction("Remote submission details", self)
jobAction.setShortcut("Ctrl+R")
jobAction.setStatusTip(
"Set the cluster information for remote job submission")
self.connect(jobAction, QtCore.SIGNAL("triggered()"),
self._cluster_details_dialog)
self.tools_menu.addAction(jobAction)
recent_files = []
for fname in self._recent_files:
if fname != self._filename and QtCore.QFile.exists(
fname) and not fname in recent_files:
recent_files.append(fname)
if len(recent_files) > 0:
self.file_menu.addSeparator()
for i, fname in enumerate(recent_files):
action = QtGui.QAction(
"&%d %s" % (i + 1, QtCore.QFileInfo(fname).fileName()),
self)
action.setData(fname)
self.connect(action, QtCore.SIGNAL("triggered()"),
self.open_file)
self.file_menu.addAction(action)
def _debug_mode(self, mode=None):
"""
Set debug mode
@param mode: debug mode (True or False). If None, the debug mode will simply be flipped
"""
if mode is None:
mode = not self.general_settings.debug
self.general_settings.debug = mode
self._new()
self.setup_layout()
def _interface_choice(self, advanced_ui=None):
if advanced_ui is None:
advanced_ui = self.general_settings.advanced
self.general_settings.advanced = advanced_ui
self._new()
self.setup_layout()
def _change_instrument(self):
"""
Invoke an instrument selection dialog
"""
class InstrDialog(QtGui.QDialog, ui.ui_instrument_dialog.Ui_Dialog):
def __init__(self, instrument_list=None):
QtGui.QDialog.__init__(self)
self.instrument_list = instrument_list
self.setupUi(self)
self.instr_combo.clear()
self.facility_combo.clear()
instruments = INSTRUMENT_DICT.keys()
instruments.sort()
instruments.reverse()
for facility in instruments:
self.facility_combo.addItem(
QtGui.QApplication.translate(
"Dialog", facility, None,
QtGui.QApplication.UnicodeUTF8))
self._facility_changed(instruments[0])
self.connect(self.facility_combo,
QtCore.SIGNAL("activated(QString)"),
self._facility_changed)
def _facility_changed(self, facility):
self.instr_combo.clear()
instr_list = INSTRUMENT_DICT[unicode(facility)].keys()
instr_list.sort()
for item in instr_list:
if self.instrument_list is None or item in self.instrument_list:
self.instr_combo.addItem(
QtGui.QApplication.translate(
"Dialog", item, None,
QtGui.QApplication.UnicodeUTF8))
if self.general_settings.debug:
dialog = InstrDialog()
else:
dialog = InstrDialog(self._instrument_list)
dialog.exec_()
if dialog.result() == 1:
self._instrument = dialog.instr_combo.currentText()
self._facility = dialog.facility_combo.currentText()
self.setup_layout()
self._new()
return True
else:
self.close()
return False
def _cluster_details_dialog(self):
"""
Show dialog to get cluster submission details
"""
class ClusterDialog(QtGui.QDialog,
ui.ui_cluster_details_dialog.Ui_Dialog):
def __init__(self, compute_resources=None):
QtGui.QDialog.__init__(self)
self.setupUi(self)
self.resource_combo.clear()
for res in compute_resources:
self.resource_combo.addItem(
QtGui.QApplication.translate(
"Dialog", res, None,
QtGui.QApplication.UnicodeUTF8))
# Fill out the defaults
dialog = ClusterDialog(self._compute_resources)
if self.general_settings.cluster_user is not None:
dialog.username_edit.setText(
str(self.general_settings.cluster_user))
dialog.pass_edit.setText(str(self.general_settings.cluster_pass))
dialog.nodes_box.setValue(int(self._number_of_nodes))
dialog.cores_box.setValue(int(self._cores_per_node))
for i in range(dialog.resource_combo.count()):
if dialog.resource_combo.itemText(
i) == self.general_settings.compute_resource:
dialog.resource_combo.setCurrentIndex(i)
break
dialog.exec_()
if dialog.result() == 1:
self.general_settings.cluster_user = str(
dialog.username_edit.text())
self.general_settings.cluster_pass = str(dialog.pass_edit.text())
self._cluster_details_set = True
self._number_of_nodes = int(dialog.nodes_box.value())
self._cores_per_node = int(dialog.cores_box.value())
self.general_settings.compute_resource = dialog.resource_combo.currentText(
)
def _clear_and_close(self):
"""
Clear all QSettings parameters
"""
self._clear_and_restart = True
self.close()
# If we make it here, the user canceled the close, which
# means that we need to reset the clear&close flag so
# that the state is properly saved on the next close.
self._clear_and_restart = False
def closeEvent(self, event):
"""
Executed when the application closes
"""
if False:
reply = QtGui.QMessageBox.question(
self, 'Message',
"Are you sure you want to quit this application?",
QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
if reply == QtGui.QMessageBox.Yes:
event.accept()
else:
event.ignore()
# Save application settings
if self._clear_and_restart:
self._clear_and_restart = False
QtCore.QSettings().clear()
else:
settings = QtCore.QSettings()
settings.setValue("instrument_name", self._instrument)
settings.setValue("last_file", self._filename)
settings.setValue("recent_files", self._recent_files)
settings.setValue("last_directory", str(self._last_directory))
settings.setValue("last_export_directory",
str(self._last_export_directory))
# General settings
self.general_settings.to_settings(settings)
def reduce_clicked(self):
"""
Create an object capable of using the information in the
interface and turn it into a reduction process.
"""
self.reduce_button.setEnabled(False)
self.export_button.setEnabled(False)
self.save_button.setEnabled(False)
self.interface_chk.setEnabled(False)
self.file_menu.setEnabled(False)
self.tools_menu.setEnabled(False)
if self._interface is not None:
self._interface.reduce()
self.reduce_button.setEnabled(True)
self.export_button.setEnabled(True)
self.save_button.setEnabled(True)
self.interface_chk.setEnabled(True)
self.file_menu.setEnabled(True)
self.tools_menu.setEnabled(True)
def cluster_clicked(self):
"""
Submit for parallel reduction
"""
if not self._cluster_details_set:
self._cluster_details_dialog()
if self._interface is not None \
and self.general_settings.cluster_user is not None \
and self.general_settings.cluster_pass is not None:
# Chose a name for the job
if self._filename is not None:
job_name = os.path.basename(self._filename).strip()
toks = os.path.splitext(job_name)
job_name = toks[0]
else:
job_name = ''
self._interface.cluster_submit(
self.general_settings.cluster_user,
self.general_settings.cluster_pass,
resource=self.general_settings.compute_resource,
nodes=self._number_of_nodes,
cores_per_node=self._cores_per_node,
job_name=job_name)
def open_file(self, file_path=None):
"""
Open an XML file and populate the UI
@param file_path: path to the file to be loaded
"""
if file_path is None:
action = self.sender()
if isinstance(action, QtGui.QAction):
file_path = unicode(action.data())
# Check whether the file describes the current instrument
try:
found_instrument = self._interface.scripter.verify_instrument(
file_path)
except:
msg = "The file you attempted to load doesn't have a recognized format.\n\n"
msg += "Please make sure it has been produced by this application."
QtGui.QMessageBox.warning(
self, "Error loading reduction parameter file", msg)
print sys.exc_value
return
if not found_instrument == self._instrument:
self._instrument = found_instrument
self.setup_layout()
self.reduce_button.setEnabled(False)
self.export_button.setEnabled(False)
self.save_button.setEnabled(False)
self.interface_chk.setEnabled(False)
self._interface.load_file(file_path)
self.reduce_button.setEnabled(True)
self.export_button.setEnabled(True)
self.save_button.setEnabled(True)
self.interface_chk.setEnabled(True)
self._filename = file_path
self._update_file_menu()
self._set_window_title()
if file_path in self._recent_files:
self._recent_files.remove(file_path)
self._recent_files.insert(0, file_path)
while len(self._recent_files) > 10:
self._recent_files.pop()
def _new(self, *argv):
"""
Start new reduction
"""
self._interface.reset()
self._filename = None
self._update_file_menu()
self._set_window_title()
def _file_open(self, *argv):
"""
File chooser for loading UI parameters
"""
fname_qstr = QtGui.QFileDialog.getOpenFileName(
self, "Reduction settings - Choose a settings file",
self._last_directory, "Settings files (*.xml)")
fname = str(QtCore.QFileInfo(fname_qstr).filePath())
if fname:
# Store the location of the loaded file
self._last_directory = str(QtCore.QFileInfo(fname_qstr).path())
self.open_file(fname)
def _save(self):
"""
Present a file dialog to the user and saves the content of the
UI in XML format
"""
if self._filename is None:
self._save_as()
else:
try:
self._interface.save_file(self._filename)
self._update_file_menu()
self.statusBar().showMessage("Saved as %s" % self._filename)
self._set_window_title()
except:
#TODO: put this in a log window, and in a file
print sys.exc_value
self.statusBar().showMessage("Failed to save %s" %
self._filename)
def _save_as(self):
"""
Present a file dialog to the user and saves the content of
the UI in XML format.
"""
if self._filename is not None:
fname = self._filename
else:
fname = self._instrument + '_'
fname_qstr = QtGui.QFileDialog.getSaveFileName(
self, "Reduction settings - Save settings",
self._last_directory + '/' + fname, "Settings files (*.xml)")
fname = str(QtCore.QFileInfo(fname_qstr).filePath())
if len(fname) > 0:
if not fname.endswith('.xml'):
fname += ".xml"
if fname in self._recent_files:
self._recent_files.remove(fname)
self._recent_files.insert(0, fname)
while len(self._recent_files) > 10:
self._recent_files.pop()
self._last_directory = str(QtCore.QFileInfo(fname_qstr).path())
self._filename = fname
self._save()
def _export(self):
"""
Exports the current content of the UI to a python script that can
be run within MantidPlot
"""
if self._interface is None:
return
fname = '.'
if self._filename is not None:
(root, ext) = os.path.splitext(self._filename)
fname = root
fname = unicode(
QtGui.QFileDialog.getSaveFileName(
self, "Mantid Python script - Save script",
self._last_export_directory, "Python script (*.py)"))
if len(fname) > 0:
if not fname.endswith('.py'):
fname += ".py"
(folder, file_name) = os.path.split(fname)
self._last_export_directory = folder
script = self._interface.export(fname)
if script is not None:
self.statusBar().showMessage("Saved as %s" % fname)
else:
self.statusBar().showMessage("Could not save file")
