def __init__(self, *args):
QtGui.QMainWindow.__init__(self, *args)
self.setWindowTitle('MOOSE')
self.pluginNames = None
self.plugin = None
self.fileMenu = None
self.editMenu = None
self.helpMenu = None
self.helpActions = None
self.viewActions = None
self.editActions = None
self.connectMenu = None
self.toolBars = []
self._loadedPlugins = {}
self._plugins = {}
self._loadedModels = []
self.setDockOptions(self.AnimatedDocks and self.AllowNestedDocks and self.AllowTabbedDocks)
self.mdiArea = MdiArea()
self.quitAction = QtGui.QAction('&Quit', self)
self.connect(self.quitAction, QtCore.SIGNAL('triggered()'), self.quit)
self.quitAction.setShortcut(QtGui.QApplication.translate("MainWindow", "Ctrl+Q", None, QtGui.QApplication.UnicodeUTF8))
self.getMyDockWidgets()
self.setCentralWidget(self.mdiArea)
self.setWindowIcon(QIcon(APPLICATION_ICON_PATH))
# pixmap = QPixmap("icons/moose_icon.png")
# pixmap = pixmap.scaled(self.mdiArea.size())
# self.mdiArea.setStyleSheet("QMdiArea { background-image: url(icons/moose_icon_large.png); }")
# palette = QPalette()
# palette.setBrush(QPalette.Background, QBrush(pixmap))
# self.setPalette(palette)
# self.mdiArea.setStyleSheet("border-image: url(icons/moose_icon_large.png)")
# self.mdiArea.setBackground(QBrush(pixmap))
self.mdiArea.setViewMode(QtGui.QMdiArea.TabbedView)
self.mdiArea.subWindowActivated.connect(self.switchSubwindowSlot)
self.setPlugin('default', '/')
self.plugin.getEditorView().getCentralWidget().parent().close()
self.popup = None
self.createPopup()
class MWindow(QtGui.QMainWindow):
"""The main window for MOOSE GUI.
This is the driver class that uses the mplugin API. mplugin based
classes will provide the toolbar, plugin specific menu items and a
set of panes to be displayed on the docks.
1. Setting a plugin
When a plugin is set as the current plugin, the view and the
menus are updated.
1.a) Updating menus:
the plugin can provide its own list of menus by implementing the
function getMenus().
the view of the plugin can also provide its own list of
menus by implementing the function getMenus().
the currentView provides a set of toolbars that are added to the
main window.
1.b) Updating views
central widget is set to the currentView (a ViewBase instance) of
the plugin.
the currentView provides a set of panes that are inserted in the
right dock area one by one.
"""
def __init__(self, *args):
QtGui.QMainWindow.__init__(self, *args)
self.setWindowTitle('MOOSE')
self.pluginNames = None
self.plugin = None
self.fileMenu = None
self.editMenu = None
self.helpMenu = None
self.helpActions = None
self.viewActions = None
self.editActions = None
self.connectMenu = None
self.toolBars = []
self._loadedPlugins = {}
self._plugins = {}
self._loadedModels = []
self.setDockOptions(self.AnimatedDocks and self.AllowNestedDocks and self.AllowTabbedDocks)
self.mdiArea = MdiArea()
self.quitAction = QtGui.QAction('&Quit', self)
self.connect(self.quitAction, QtCore.SIGNAL('triggered()'), self.quit)
self.quitAction.setShortcut(QtGui.QApplication.translate("MainWindow", "Ctrl+Q", None, QtGui.QApplication.UnicodeUTF8))
self.getMyDockWidgets()
self.setCentralWidget(self.mdiArea)
self.setWindowIcon(QIcon(APPLICATION_ICON_PATH))
# pixmap = QPixmap("icons/moose_icon.png")
# pixmap = pixmap.scaled(self.mdiArea.size())
# self.mdiArea.setStyleSheet("QMdiArea { background-image: url(icons/moose_icon_large.png); }")
# palette = QPalette()
# palette.setBrush(QPalette.Background, QBrush(pixmap))
# self.setPalette(palette)
# self.mdiArea.setStyleSheet("border-image: url(icons/moose_icon_large.png)")
# self.mdiArea.setBackground(QBrush(pixmap))
self.mdiArea.setViewMode(QtGui.QMdiArea.TabbedView)
self.mdiArea.subWindowActivated.connect(self.switchSubwindowSlot)
self.setPlugin('default', '/')
self.plugin.getEditorView().getCentralWidget().parent().close()
self.popup = None
self.createPopup()
def createPopup(self):
self.popup = dialog = QDialog(self)
dialog.setWindowFlags(Qt.Qt.Dialog | Qt.Qt.FramelessWindowHint)
# dialog.setModal(True)
layout = QGridLayout()
createKineticModelButton = QPushButton("Create Kinetic Model")
loadKineticModelButton = QPushButton("Load Model")
loadNeuronalModelButton = QPushButton("Load Neuronal Model")
# createKineticModelButton.setStyleSheet(
# """QPushButton { font-size : 18pt;
# font-weight: bold;
# color : #000000;
# }
# """ )
layout.setContentsMargins(QtCore.QMargins(20,20,20,20))
# layout.addWidget(QLabel(""), 0, 0)
layout.addWidget(createKineticModelButton)
# layout.addWidget(QLabel(" "), 0, 2)
# layout.addWidget(QLabel(" "), 1, 0)
layout.addWidget(loadKineticModelButton)
# layout.addWidget(QLabel(" "), 1, 2)
# layout.addWidget(QLabel(" "), 2, 0)
# layout.addWidget(loadNeuronalModelButton)
# layout.addWidget(QLabel(" "), 2, 2)
dialog.setLayout(layout)
createKineticModelButton.clicked.connect(self.newModelDialogSlot)
loadKineticModelButton.clicked.connect(self.loadModelDialogSlot)
loadNeuronalModelButton.clicked.connect(self.loadModelDialogSlot)
dialog.show()
return dialog
def quit(self):
QtGui.qApp.closeAllWindows()
def handleException(self, t, v, s):
"""This handler will show warning messages for error exceptions. Show
info at status bar for non-error exceptions. It will replace
sys.excepthook and has the same signature (except being bound
to this object).
t : exception type
v : exception value
s: traceback object.
"""
traceback.print_exception(t, v, s)
title = ''.join(traceback.format_exception_only(t, v))
trace = ''.join(traceback.format_exception(t, v, s))
if isinstance(v, mexception.MooseInfo):
self.statusBar().showMessage(title, 5000)
elif isinstance(v, mexception.MooseWarning):
QtGui.QMessageBox.warning(self, title, '\n'.join((title, trace)))
else:
QtGui.QMessageBox.critical(self, title, '\n'.join((title, trace)))
def getPluginNames(self):
"""Return pluginNames attribute or create it by retrieving
available plugin names from plugin/list.txt file.
"""
if self.pluginNames is None:
with open(os.path.join(config.MOOSE_GUI_DIR,
'plugins',
'list.txt')) as lfile:
self.pluginNames = [line.strip() for line in lfile]
self.pluginNames = [name for name in self.pluginNames if name]
return self.pluginNames
def loadPluginModule(self, name, re=False):
"""Load a plugin module by name.
First check if the plugin is already loaded. If so return the
existing one. Otherwise, search load the plugin as a python
module from {MOOSE_GUI_DIRECTORY}/plugins directory.
If re is True, the plugin is reloaded.
"""
if (not re) and name in sys.modules:
return sys.modules[name]
fp, pathname, description = imp.find_module(name, [config.MOOSE_PLUGIN_DIR])
try:
module = imp.load_module(name, fp, pathname, description)
finally:
if fp:
fp.close()
return module
def getMyDockWidgets(self):
"""Return a list of dockwidgets that belong to the top
level. This is needed to keep them separate from those
provided by the plugins.
Currently we only have shell for this."""
if not hasattr(self, 'dockWidgets') or self.dockWidgets is None:
self.dockWidgets = {}
dockWidget = QtGui.QDockWidget('Python')
dockWidget.setWidget(self.getShellWidget())
self.dockWidgets[dockWidget] = True
self.addDockWidget(Qt.Qt.BottomDockWidgetArea, dockWidget)
dockWidget.setVisible(False)
dockWidget = ObjectEditDockWidget('/')
self.dockWidgets[dockWidget] = True
self.objectEditDockWidget = dockWidget
self.addDockWidget(Qt.Qt.RightDockWidgetArea, dockWidget)
dockWidget.setVisible(False)
return self.dockWidgets.keys()
def getShellWidget(self):
"""Create an instance of shell widget. This can be either a
QSciQScintialla widget or a PyCute widget (extends QTextArea)
if the first is not available"""
if not hasattr(self, 'shellWidget') or self.shellWidget is None:
self.shellWidget = get_shell_class()( code.InteractiveInterpreter()
, message='MOOSE version %s' % (moose._moose.__version__)
)
self.shellWidget.interpreter.runsource('from moose import *')
self.shellWidget.setVisible(False)
return self.shellWidget
def loadPluginClass(self, name, re=False):
"""Load the plugin class from a plugin module.
A plugin module should have only one subclass of
MoosePluginBase. Otherwise the first such class found will be
loaded.
"""
try:
return self._loadedPlugins[name]
except KeyError:
pluginModule = self.loadPluginModule(name, re=re)
for classname, classobj in inspect.getmembers(pluginModule, inspect.isclass):
if issubclass(classobj, mplugin.MoosePluginBase):
self._loadedPlugins[name] = classobj
# classobj.getEditorView().getCentralWidget().editObject.connect(self.objectEditSlot)
return self._loadedPlugins[name]
raise Exception('No plugin with name: %s' % (name))
def setPlugin(self, name, root='/'):
"""Set the current plugin to use.
This -
1. sets the `plugin` attribute.
2. updates menus by clearing and reinstating menus including
anything provided by the plugin.
3. sets the current view to the plugins editor view.
"""
busyCursor()
for model in self._loadedModels:
self.disableModel(model[0])
for i in range(0, len(self._loadedModels)):
if self._loadedModels[i][0]== root:
c = moose.Clock('/clock')
compt = moose.wildcardFind(root+'/##[ISA=ChemCompt]')
if compt:
c.tickDt[11] = self._loadedModels[i][3]
c.tickDt[16] = self._loadedModels[i][4]
if moose.exists(compt[0].path+'/ksolve'):
ksolve = moose.Ksolve( compt[0].path+'/ksolve' )
ksolve.tick = 16
if moose.exists(compt[0].path+'/gsolve'):
gsolve = moose.Gsolve( compt[0].path+'/gsolve' )
gsolve.tick = 16
for x in moose.wildcardFind( root+'/data/graph#/#' ):
x.tick = 18
else:
c.tickDt[7] = self._loadedModels[i][3]
c.tickDt[8] = self._loadedModels[i][4]
neurons = moose.wildcardFind(root + "/model/cells/##[ISA=Neuron]")
for neuron in neurons:
#print(neuron)
solver = moose.element(neuron.path + "/hsolve")
# print("Disabling => ", solver)
solver.tick = 7
for x in moose.wildcardFind( root+'/data/graph#/#' ):
x.tick = 8
break
self.plugin = self.loadPluginClass(str(name))(str(root), self)
moose.reinit()
# if root != '/' and root not in self._loadedModels:
# self._loadedModels[root] = name
# for k,v in self._loadedModels.items():
# compt = moose.wildcardFind(k+'/##[ISA=ChemCompt]')
# if compt:
# if moose.exists(compt[0].path+'/ksolve'):
# ksolve = moose.Ksolve( compt[0].path+'/ksolve' )
# ksolve.tick = -1
# if moose.exists(compt[0].path+'/stoich'):
# stoich = moose.Stoich( compt[0].path+'/stoich' )
# stoich.tick = -1
# for x in moose.wildcardFind( k+'/data/graph#/#' ):
# x.tick = -1
# if root != '/' and root not in self._loadedModels:
# self._loadedModels[root] = name
# try:
# self.plugin = self._plugins[str(name)]
# print 'PLUGIN', self.plugin
# self.plugin.setModelRoot(root)
# except KeyError:
# self.plugin = self.loadPluginClass(str(name))(str(root), self)
# self._plugins[str(name)] = self.plugin
#self.plugin.getEditorView().getCentralWidget().editObject.connect(self.objectEditSlot, QtCore.Qt.UniqueConnection)
self.updateMenus()
for action in self.pluginsMenu.actions():
if str(action.text()) == str(name):
action.setChecked(True)
elif action.isChecked():
action.setChecked(False)
for subwin in self.mdiArea.subWindowList():
subwin.close()
if name != "default" :
self.setCurrentView('editor')
self.setCurrentView('run')
if name == 'kkit':
self.objectEditDockWidget.objectNameChanged.connect(self.plugin.getEditorView().getCentralWidget().updateItemSlot)
self.objectEditDockWidget.colorChanged.connect(self.plugin.getEditorView().getCentralWidget().updateColorSlot)
self.setCurrentView('editor')
freeCursor()
return self.plugin
def updateExistingMenu(self, menu):
"""Check if a menu with same title
already exists. If so, update the same and return
True. Otherwise return False.
"""
if not isinstance(menu, QtGui.QMenu):
return False
for action in self.menuBar().actions():
if menu.title() == action.text():
# if not action.menu().isEmpty():
# action.menu().addSeparator()
action.menu().addActions(menu.actions())
return True
return False
def updateMenus(self):
"""Clear the menubar and reinstate the basic menus. Go
through the menus provided by current plugin and add those to
menubar.
If a menu provided by a plugin has same name as one of the
core menus, the menu items provided by the plugin are appended
to the existing menu after a separator.
"""
self.menuBar().clear()
self.getPluginsMenu()
menus = [self.getFileMenu(),
self.getEditMenu(),
self.getViewMenu(),
#self.getRunMenu(),
#self.getConnectMenu(),
self.getHelpMenu()]
for menu in menus:
self.menuBar().addMenu(menu)
for menu in self.plugin.getMenus():
if not self.updateExistingMenu(menu):
if not self.menuBar().isEmpty():
action.menu().addSeparator()
self.menuBar().addMenu(menu)
menus[0].addSeparator()
menus[0].addAction(self.quitAction)
def updateToolbars(self):
for toolbar in self.toolBars:
self.removeToolBar(toolbar)
self.toolBars = []
self.toolBars.extend(self.getMyToolBars())
self.toolBars.extend(self.plugin.getToolBars())
self.toolBars.extend(self.plugin.getCurrentView().getToolBars())
for toolbar in self.toolBars:
self.addToolBar(toolbar)
toolbar.setVisible(True)
def switchSubwindowSlot(self, window):
"""Change view based on what subwindow `window` is activated."""
if not window:
#print 'Window is None'
return
view = str(window.windowTitle()).partition(':')[0]
#print 'activated', window.windowTitle(), 'view=', view
#print 'setting current view'
self.setCurrentView(view)
def setCurrentView(self, view):
"""Set current view to a particular one: options are 'editor',
'plot', 'run'. A plugin can provide more views if necessary.
"""
self.plugin.setCurrentView(view)
if view =='run':
#Harsha: This will clear out object editor's objectpath and make it invisible
self.objectEditSlot('/',False)
targetView = None
newSubWindow = True
widget = self.plugin.getCurrentView().getCentralWidget()
current = self.mdiArea.activeSubWindow()
subwin = None
for subwin in self.mdiArea.subWindowList():
if subwin.widget() == widget:
newSubWindow = False
break
if newSubWindow:
subwin = self.mdiArea.addSubWindow(widget)
subwin.setWindowTitle('%s: %s' % (view, widget.modelRoot))
subwin.setSizePolicy(QtGui.QSizePolicy.Minimum |
QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Minimum |
QtGui.QSizePolicy.Expanding)
subwin.resize(600, 400)
# Make dockwidgets from other views invisible and make those
# from current view visible or add them if not already part of
# main window.
dockWidgets = set([dockWidget for dockWidget in self.findChildren(QtGui.QDockWidget)])
for dockWidget in dockWidgets:
if dockWidget not in self.dockWidgets:
dockWidget.setVisible(False)
for dockWidget in self.plugin.getCurrentView().getToolPanes():
if dockWidget not in dockWidgets:
if view == "run":
if dockWidget.windowTitle() == "Scheduling":
self.addDockWidget(Qt.Qt.TopDockWidgetArea, dockWidget)
else:
self.addDockWidget(Qt.Qt.RightDockWidgetArea, dockWidget)
dockWidget.setVisible(True)
subwin.setVisible(True)
self.mdiArea.setActiveSubWindow(subwin)
self.updateMenus()
for menu in self.plugin.getCurrentView().getMenus():
if not self.updateExistingMenu(menu):
self.menuBar().addMenu(menu)
self.updateToolbars()
return subwin
def getMyToolBars(self):
self._toolBars = []
'''
#Harsha: removing the toolbars (plot,run,edit) from the Gui
if not hasattr(self, 'viewToolBar'):
self.viewToolBar = QtGui.QToolBar('View')
#Harsha:removing plotView from the ToolBar
for t in self.getViewActions():
if t.text() != "&Plot view":
self.viewToolBar.addAction(t)
#self.viewToolBar.addActions(self.getViewActions())
#return [self.viewToolBar]
'''
return self._toolBars
def getFileMenu(self):
if self.fileMenu is None:
self.fileMenu = QtGui.QMenu('&File')
else:
self.fileMenu.clear()
if not hasattr(self, 'newModelAction'):
self.newModelAction = QtGui.QAction('New', self)
self.newModelAction.setShortcut(QtGui.QApplication.translate("MainWindow", "Ctrl+N", None, QtGui.QApplication.UnicodeUTF8))
self.connect(self.newModelAction, QtCore.SIGNAL('triggered()'), self.newModelDialogSlot)
self.fileMenu.addAction(self.newModelAction)
if not hasattr(self, 'loadModelAction'):
self.loadModelAction = QtGui.QAction('L&oad model', self)
self.loadModelAction.setShortcut(QtGui.QApplication.translate("MainWindow", "Ctrl+O", None, QtGui.QApplication.UnicodeUTF8))
self.connect(self.loadModelAction, QtCore.SIGNAL('triggered()'), self.loadModelDialogSlot)
self.fileMenu.addAction(self.loadModelAction)
if not hasattr(self,'loadedModels'):
self.loadedModelAction = QtGui.QAction('Recently Loaded Models',self)
self.loadedModelAction.setCheckable(False)
#self.fileMenu.addAction(QtGui.QAction(self.loadedModelAction,checkable=True))
if bool(self._loadedModels):
self.fileMenu.addSeparator()
self.fileMenu.addAction(self.loadedModelAction)
self.loadedModelAction.setEnabled(False)
for (model, modeltype, action,simdt,plotdt) in reversed(self._loadedModels):
self.fileMenu.addAction(action)
self.fileMenu.addSeparator()
if not hasattr(self,'connectBioModelAction'):
self.connectBioModelAction = QtGui.QAction('&Connect BioModels', self)
self.connectBioModelAction.setShortcut(QtGui.QApplication.translate("MainWindow", "Ctrl+B", None, QtGui.QApplication.UnicodeUTF8))
self.connect(self.connectBioModelAction, QtCore.SIGNAL('triggered()'), self.connectBioModel)
self.fileMenu.addAction(self.connectBioModelAction)
return self.fileMenu
def getEditMenu(self):
if self.editMenu is None:
self.editMenu = QtGui.QMenu('&Edit')
else:
self.editMenu.clear()
#self.editMenu.addActions(self.getEditActions())
return self.editMenu
def getPluginsMenu(self):
"""Populate plugins menu if it does not exist already."""
if (not hasattr(self, 'pluginsMenu')) or (self.pluginsMenu is None):
self.pluginsMenu = QtGui.QMenu('&Plugins')
mapper = QtCore.QSignalMapper(self)
pluginsGroup = QtGui.QActionGroup(self)
pluginsGroup.setExclusive(True)
for pluginName in self.getPluginNames():
action = QtGui.QAction(pluginName, self)
action.setObjectName(pluginName)
action.setCheckable(True)
mapper.setMapping(action, QtCore.QString(pluginName))
self.connect(action, QtCore.SIGNAL('triggered()'), mapper, QtCore.SLOT('map()'))
self.pluginsMenu.addAction(action)
pluginsGroup.addAction(action)
self.connect(mapper, QtCore.SIGNAL('mapped(const QString &)'), self.setPlugin)
#self.pluginsMenu.addMenu(self.defaultPluginMenu)
#self.pluginsMenu.addMenu(self.kkitPluginMenu)
#self.pluginsMenu.addMenu(self.neurokitPluginMenu)
#openRootAction = self.defaultPluginMenu.addAction("/")
#openRootAction.triggered.connect(lambda : self.setPlugin("default", "/") )
# if (not hasattr(self, 'loadedModelAction')) or (self.loadedModelAction is None) :
# self.loadedModelAction = QtGui.QAction("kkit",self)
# self.loadedModelAction.addMenu('test')
# self.pluginsMenu.addAction(self.loadedModelAction)
# self.pluginsMenu.addMenu(self.insertkkitMenu)
# self.insertMapperkkit = QtCore.QSignalMapper(self)
#insertMapperkkit,actions = self.getInsertkkitActions(self.loadedModels)
# ignored_bases = ['ZPool', 'Msg', 'Panel', 'SolverBase', 'none']
# ignored_classes = ['ZPool','ZReac','ZMMenz','ZEnz','CplxEnzBase']
# classlist = [ch[0].name for ch in moose.element('/classes').children
# if (ch[0].baseClass not in ignored_bases)
# and (ch[0].name not in (ignored_bases + ignored_classes))
# and not ch[0].name.startswith('Zombie')
# and not ch[0].name.endswith('Base')
# ]
# insertMapper, actions = self.getInsertActions(classlist)
# for action in actions:
# self.insertMenu.addAction(action)
# self.connect(insertMapper, QtCore.SIGNAL('mapped(const QString&)'), self.tree.insertElementSlot)
# self.editAction = QtGui.QAction('Edit', self.treeMenu)
# self.editAction.triggered.connect(self.editCurrentObjectSlot)
# self.tree.elementInserted.connect(self.elementInsertedSlot)
# self.treeMenu.addAction(self.editAction)
return self.pluginsMenu
def getHelpMenu(self):
if self.helpMenu is None:
self.helpMenu = QtGui.QMenu('&Help')
else:
self.helpMenu.clear()
self.helpMenu.addActions(self.getHelpActions())
return self.helpMenu
'''
def getConnectMenu(self):
if self.connectMenu is None:
self.connectMenu = QtGui.QMenu('&Connect')
else:
self.connectMenu.clear()
self.connectMenu.addActions(self.getConnectActions())
return self.connectMenu
'''
def getViewMenu(self):
if (not hasattr(self, 'viewMenu')) or (self.viewMenu is None):
self.viewMenu = QtGui.QMenu('&View')
else:
self.viewMenu.clear()
self.viewMenu.addActions(self.getViewActions())
self.docksMenu = self.viewMenu.addMenu('&Dock widgets')
self.docksMenu.addActions(self.getDockWidgetsToggleActions())
self.subWindowMenu = self.viewMenu.addMenu('&Subwindows')
self.subWindowMenu.addActions(self.getSubWindowActions())
return self.viewMenu
# def getSubWindowVisibilityActions(self):
# if not hasattr(self, 'subwindowToToggles'):
# self.subWindowToToggle = QtGui.QSignalMapper()
# self.toggleToSubWindow = QtGui.QSignalMapper()
# for subwindow in self.mdiArea.subWindowList():
# if self.subWindowToToggle.mapping(subwindow) is None:
# action = QtGui.QAction('Show: %s' % (subwindow.windowTitle), self)
# self.toggleToSubWindow.setMapping(action, subwindow)
# self.connect(action, QtCore.SIGNAL('triggered()'),
# self.toggleToSubWindow,
# QtCore.SLOT('mapped(QWidget*)'))
# self.subWindowToToggle.setMapping(subwindow, action)
# self.connect(subwindow, QtCore.SIGNAL('closed()')
# self.subWindowVisibilityMenu = QtGui.Q
# for subwin in self.mdiArea.subWindowList():
# Removed from the menu
# def getRunMenu(self):
# if (not hasattr(self, 'runMenu')) or (self.runMenu is None):
# self.runMenu = QtGui.QMenu('&Run')
# else:
# self.runMenu.clear()
# self.runMenu.addActions(self.getRunActions())
# return self.runMenu
def getEditActions(self):
# self.editActions = []
# if (not hasattr(self, 'editActions')) or (self.editActions is None):
# self.setModelRootAction = QtGui.QAction('&Set model root', self)
# self.setModelRootAction.triggered.connect(self.showSetModelRootDialog)
# self.setDataRootAction = QtGui.QAction('Set &data root', self)
# self.setDataRootAction.triggered.connect(self.showSetDataRootDialog)
# self.editActions = [self.setModelRootAction, self.setDataRootAction]
# return self.editActions
return None
def showSetModelRootDialog(self):
root, ok = QtGui.QInputDialog.getText(self, 'Model Root', 'Enter the model root path:', text=moose.element(self.plugin.modelRoot).path)
if not ok:
return
root = str(root) #convert from QString to python str
self.plugin.setModelRoot(root)
for subwin in self.mdiArea.subWindowList():
if hasattr(subwin.widget(), 'modelRoot'):
subwin.setWindowTitle(root)
def showSetDataRootDialog(self):
root, ok = QtGui.QInputDialog.getText(self, 'Data Root', 'Enter the data root path:', text=moose.element(self.plugin.dataRoot).path)
if not ok:
return
root = str(root) #convert from QString to python str
self.plugin.setDataRoot(root)
for subwin in self.mdiArea.subWindowList():
if hasattr(subwin.widget(), 'dataRoot'):
subwin.setWindowTitle(root)
def getViewActions(self):
if (not hasattr(self, 'viewActions')) or (self.viewActions is None):
self.editorViewAction = QtGui.QAction('&Editor view', self)
self.editorViewAction.triggered.connect(self.openEditorView)
#self.plotViewAction = QtGui.QAction('&Plot view', self)
#self.plotViewAction.triggered.connect(self.openPlotView)
self.runViewAction = QtGui.QAction('&Run view', self)
self.runViewAction.triggered.connect(self.openRunView)
#self.viewActions = [self.editorViewAction, self.plotViewAction, self.runViewAction]
self.viewActions = [self.editorViewAction, self.runViewAction]
return self.viewActions
def getSubWindowActions(self):
if not hasattr(self, 'subWindowActions') or self.subWindowActions is None:
self.tabbedViewAction = QtGui.QAction('&Tabbed view', self)
self.tabbedViewAction.triggered.connect(lambda : self.mdiArea.setViewMode(QtGui.QMdiArea.TabbedView))
self.subWindowViewAction = QtGui.QAction('&SubWindow view', self)
self.subWindowViewAction.triggered.connect(lambda : self.mdiArea.setViewMode(QtGui.QMdiArea.SubWindowView))
self.tileSubWindowsAction = QtGui.QAction('Ti&le subwindows', self)
self.tileSubWindowsAction.triggered.connect(self.mdiArea.tileSubWindows)
self.cascadeSubWindowsAction = QtGui.QAction('&Cascade subwindows', self)
self.cascadeSubWindowsAction.triggered.connect(self.mdiArea.cascadeSubWindows)
self.subWindowActions = [self.tabbedViewAction, self.subWindowViewAction, self.tileSubWindowsAction, self.cascadeSubWindowsAction]
return self.subWindowActions
def getDockWidgetsToggleActions(self):
"""Get a list of actions for toggling visibility of dock
widgets
"""
return [widget.toggleViewAction() for widget in self.findChildren(QtGui.QDockWidget)]
# Removed form the menu item
# def getRunActions(self):
# if (not hasattr(self, 'runActions')) or \
# (self.runActions is None):
# self.startAction = QtGui.QAction('Start', self)
# self.startAction.triggered.connect(self.resetAndStartSimulation)
# self.pauseAction = QtGui.QAction('Pause', self)
# self.pauseAction.triggered.connect(self.pauseSimulation)
# self.continueAction = QtGui.QAction('Continue', self)
# self.continueAction.triggered.connect(self.continueSimulation)
# self.runActions = [self.startAction, self.pauseAction, self.continueAction]
# return self.runActions
def getHelpActions(self):
if (not hasattr(self, 'helpActions')) or (self.helpActions is None):
self.actionAbout = QtGui.QAction('About MOOSE', self)
self.connect(self.actionAbout, QtCore.SIGNAL('triggered()'), self.showAboutMoose)
self.actionBuiltInDocumentation = QtGui.QAction('Built-in documentation', self)
self.connect(self.actionBuiltInDocumentation, QtCore.SIGNAL('triggered()'), self.showBuiltInDocumentation)
self.actionBug = QtGui.QAction('Report a bug', self)
self.connect(self.actionBug, QtCore.SIGNAL('triggered()'), self.reportBug)
self.helpActions = [self.actionAbout, self.actionBuiltInDocumentation, self.actionBug]
return self.helpActions
# Removed from the main menu item replace with File menu
# def getConnectActions(self):
# if(not hasattr(self,'connectActions')) or(self.connectActions is None):
# self.actionBioModel = QtGui.QAction('BioModels',self)
# self.connect(self.actionBioModel, QtCore.SIGNAL('triggered()'), self.connectBioModel)
# self.connectActions = [self.actionBioModel]
# return self.connectActions
def connectBioModel(self):
connecttoBioModel = BioModelsClientWidget()
if connecttoBioModel.exec_():
pass
filepath = connecttoBioModel.filePath
if filepath:
head, fileName = os.path.split(filepath)
modelName = os.path.splitext(fileName)[0]
pwe = moose.getCwe()
ret = loadFile(str(filepath), '/model/%s' % (modelName), merge=False)
self.objectEditSlot('/',False)
pluginLookup = '%s/%s' % (ret['modeltype'], ret['subtype'])
try:
pluginName = subtype_plugin_map['%s/%s' % (ret['modeltype'], ret['subtype'])]
except KeyError:
pluginName = 'default'
print 'Loaded model', ret['model'].path,subtype
self._loadedModels.append([ret['model'].path,pluginName])
if len(self._loadedModels)>5:
self._loadedModels.pop(0)
if not moose.exists(ret['model'].path+'/info'):
moose.Annotator(ret['model'].path+'/info')
modelAnno = moose.Annotator(ret['model'].path+'/info')
if ret['subtype']:
modelAnno.modeltype = ret['subtype']
else:
modelAnno.modeltype = ret['modeltype']
modelAnno.dirpath = str(dialog.directory().absolutePath())
self.loadedModelsAction(ret['model'].path,pluginName)
self.setPlugin(pluginName, ret['model'].path)
def showAboutMoose(self):
with open(config.MOOSE_ABOUT_FILE, 'r') as aboutfile:
QtGui.QMessageBox.about(self, 'About MOOSE', ''.join(aboutfile.readlines()))
def showDocumentation(self, source):
if not hasattr(self, 'documentationViewer'):
self.documentationViewer = QtGui.QTextBrowser()
self.documentationViewer.setOpenLinks(True)
self.documentationViewer.setOpenExternalLinks(True)
print " path ",config.settings[config.KEY_DOCS_DIR], os.path.join(config.settings[config.KEY_DOCS_DIR], 'html'), os.path.join(config.settings[config.KEY_DOCS_DIR], 'images')
self.documentationViewer.setSearchPaths([config.settings[config.KEY_DOCS_DIR],
os.path.join(config.settings[config.KEY_DOCS_DIR], 'html'),
os.path.join(config.settings[config.KEY_DOCS_DIR], 'images')])
self.documentationViewer.setMinimumSize(800, 480)
self.documentationViewer.setSource(QtCore.QUrl(source))
result = self.documentationViewer.loadResource(QtGui.QTextDocument.HtmlResource, self.documentationViewer.source())
if not result.isValid():
QtGui.QMessageBox.warning(self, 'Could not access documentation', 'The link %s could not be accessed' % (source))
return
self.documentationViewer.setWindowTitle(source)
self.documentationViewer.reload()
self.documentationViewer.setVisible(True)
def reportBug(self):
QtGui.QDesktopServices.openUrl(QtCore.QUrl(config.MOOSE_REPORT_BUG_URL))
def showBuiltInDocumentation(self):
self.showDocumentation('moose_builtins.html')
# openEditorView, openPlotView and openRunView are identical
# except the view they ask from the plugin. Consider using a
# mapper.
def openEditorView(self):
"""Switch to the editor view of current plugin. If there is
already a subwindow for this, make that the active
one. Otherwise create a new one.
"""
self.setCurrentView('editor')
def openPlotView(self):
self.setCurrentView('plot')
def openRunView(self):
self.setCurrentView('run')
def resetAndStartSimulation(self):
"""TODO this should provide a clean scheduling through all kinds
of simulation or default scheduling should be implemented in MOOSE
itself. We need to define a policy for handling scheduling. It can
be pushed to the plugin-developers who should have knowledge of
the scheduling criteria for their domain."""
settings = config.MooseSetting()
try:
simdt_kinetics = float(settings[config.KEY_KINETICS_SIMDT])
except ValueError:
simdt_kinetics = 0.1
try:
simdt_electrical = float(settings[config.KEY_ELECTRICAL_SIMDT])
except ValueError:
simdt_electrical = 0.25e-4
try:
plotdt_kinetics = float(settings[config.KEY_KINETICS_PLOTDT])
except ValueError:
plotdt_kinetics = 0.1
try:
plotdt_electrical = float(settings[config.KEY_ELECTRICAL_PLOTDT])
except ValueError:
plotdt_electrical = 0.25e-3
try:
simtime = float(settings[config.KEY_SIMTIME])
except ValueError:
simtime = 1.0
moose.reinit()
view = self.plugin.getRunView()
moose.start(simtime)
if view.getCentralWidget().plotAll:
view.getCentralWidget().plotAllData()
self.setCurrentView('run')
def pauseSimulation(self):
moose.stop()
'''
def continueSimulation(self):
"""TODO implement this somewhere else"""
try:
simtime = float(config.MooseSetting()[config.KEY_SIMTIME])
except ValueError:
simtime = 1.0
moose.start(simtime)
'''
#Harsha: added visible=True so that loadModelDialogSlot and NewModelDialogSlot call this function
# to clear out object path
def objectEditSlot(self, mobj, visible=True):
"""Slot for switching the current object in object editor."""
self.objectEditDockWidget.setObject(mobj)
self.objectEditDockWidget.setVisible(visible)
def loadedModelsAction(self,modelPath,pluginName):
#Harsha: added under file Menu, Recently Loaded Models
#All the previously loaded chemical models, solver's and table's ticks are made -1
for model in self._loadedModels:
self.disableModel(model[0])
action = QAction(modelPath[1:],self)
action.triggered.connect(lambda : self.setPlugin(pluginName, modelPath))
compt = moose.wildcardFind(modelPath + '/##[ISA=ChemCompt]')
c = moose.Clock('/clock')
self.simulationdt = c.tickDt[7]
self.plotdt = c.tickDt[8]
if compt:
self.simulationdt = c.tickDt[11]
self.plotdt = c.tickDt[16]
self._loadedModels.append([modelPath,pluginName,action,self.simulationdt,self.plotdt])
if len(self._loadedModels)>5:
self._loadedModels.pop(0)
def disableModel(self, modelPath):
compt = moose.wildcardFind(modelPath + '/##[ISA=ChemCompt]')
if compt:
if moose.exists(compt[0].path+'/ksolve'):
ksolve = moose.Ksolve( compt[0].path+'/ksolve' )
ksolve.tick = -1
if moose.exists(compt[0].path+'/gsolve'):
gsolve = moose.Gsolve( compt[0].path+'/gsolve' )
gsolve.tick = -1
# if moose.exists(compt[0].path+'/stoich'):
# stoich = moose.Stoich( compt[0].path+'/stoich' )
# stoich.tick = -1
else :
neurons = moose.wildcardFind(modelPath + "/model/cells/##[ISA=Neuron]")
for neuron in neurons:
#print(neuron)
solver = moose.element(neuron.path + "/hsolve")
# print("Disabling => ", solver)
solver.tick = -1
for table in moose.wildcardFind( modelPath+'/data/graph#/#' ):
table.tick = -1
def loadModelDialogSlot(self):
"""Start a file dialog to choose a model file.
Once the dialog succeeds, we should hand-over the duty of
actual model loading to something else. Then refresh the
views. Things to check from the user:
1) The file type
2) Target element
3) Whether we should update the current window or start a new
window.
4) Plugin to use for displaying this model (can be automated
by looking into the model file for a regular expression)
"""
self.popup.close()
activeWindow = None # This to be used later to refresh the current widget with newly loaded model
dialog = LoaderDialog(self,
self.tr('Load model from file'))
if dialog.exec_():
fileNames = dialog.selectedFiles()
for fileName in fileNames:
modelName = dialog.getTargetPath()
if '/' in modelName:
raise mexception.ElementNameError('Model name cannot contain `/`')
ret = loadFile(str(fileName),'%s' %(modelName),merge=False)
#ret = loadFile(str(fileName), '/model/%s' % (modelName), merge=False)
#Harsha: This will clear out object editor's objectpath and make it invisible
self.objectEditSlot('/',False)
# Harsha: if subtype is None, in case of cspace then pluginLookup = /cspace/None
# which will not call kkit plugin so cleaning to /cspace
pluginLookup = '%s/%s' % (ret['modeltype'], ret['subtype'])
try:
pluginName = subtype_plugin_map['%s/%s' % (ret['modeltype'], ret['subtype'])]
except KeyError:
pluginName = 'default'
print 'Loaded model', ret['model'].path
if not moose.exists(ret['model'].path+'/info'):
moose.Annotator(ret['model'].path+'/info')
modelAnno = moose.Annotator(ret['model'].path+'/info')
if ret['subtype']:
modelAnno.modeltype = ret['subtype']
else:
modelAnno.modeltype = ret['modeltype']
modelAnno.dirpath = str(dialog.directory().absolutePath())
self.loadedModelsAction(ret['model'].path,pluginName)
self.setPlugin(pluginName, ret['model'].path)
if pluginName == 'kkit':
QtCore.QCoreApplication.sendEvent(self.plugin.getEditorView().getCentralWidget().view, QtGui.QKeyEvent(QtCore.QEvent.KeyPress, Qt.Qt.Key_A, Qt.Qt.NoModifier))
def newModelDialogSlot(self):
#Harsha: Create a new dialog widget for model building
self.popup.close()
newModelDialog = DialogWidget()
if newModelDialog.exec_():
modelPath = str(newModelDialog.modelPathEdit.text()).strip()
if len(modelPath) == 0:
raise mexception.ElementNameError('Model path cannot be empty')
if re.search('[ /]',modelPath) is not None:
raise mexception.ElementNameError('Model path should not containe / or whitespace')
#plugin = str(newModelDialog.submenu.currentText())
plugin = str(newModelDialog.getcurrentRadioButton())
#print "plugin ",plugin
#Harsha: All model will be forced to load/build under /model,
#2014 sep 10: All the model will be forced to load/build model under /modelName/model
'''
modelContainer = moose.Neutral('/model')
modelRoot = moose.Neutral('%s/%s' % (modelContainer.path, modelPath))
'''
modelContainer = moose.Neutral('%s' %(modelPath))
modelRoot = moose.Neutral('%s/%s' %(modelContainer.path,"model"))
if not moose.exists(modelContainer.path+'/info'):
moose.Annotator(modelContainer.path+'/info')
modelAnno = moose.element(modelContainer.path+'/info')
modelAnno.modeltype = "new_kkit"
modelAnno.dirpath = " "
self.loadedModelsAction(modelRoot.path,plugin)
self.setPlugin(plugin, modelRoot.path)
#Harsha: This will clear out object editor's objectpath and make it invisible
self.objectEditSlot('/', False)