# msearch.py ---
#
# Filename: msearch.py
# Description:
# Author:
# Maintainer:
# Created: Tue Jun 4 15:10:16 2013 (+0530)
# Version:
# Last-Updated: Thu Jun 6 18:26:02 2013 (+0530)
# By: subha
# Update #: 181
# URL:
# Keywords:
# Compatibility:
#
#
# Commentary:
#
#
#
#
# Change log:
#
#
#
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License as
# published by the Free Software Foundation; either version 3, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; see the file COPYING. If not, write to
# the Free Software Foundation, Inc., 51 Franklin Street, Fifth
# Floor, Boston, MA 02110-1301, USA.
#
#
# Code:
"""
Widget giving access to wildcardFind function in MOOSE
"""
import sys
from PyQt4 import QtGui, QtCore
from PyQt4.Qt import Qt
import moose
class SearchWidget(QtGui.QWidget):
"""Widget to search MOOSE model tree using wildcards.
SIGNALS:
executed(list of elements matching search criterion).
"""
executed = QtCore.pyqtSignal(list, name='executed')
def __init__(self, *args):
QtGui.QWidget.__init__(self, *args)
layout = QtGui.QVBoxLayout()
self.modeCombo = QtGui.QComboBox()
self.modeCombo.addItem('Basic', QtCore.QVariant(QtCore.QString('basic')))
self.modeCombo.addItem('Advanced', QtCore.QVariant(QtCore.QString('advanced')))
self.modeCombo.setCurrentIndex(0)
self.setLayout(layout)
layout.addWidget(self.modeCombo)
layout.addWidget(self.getBasicPanel())
layout.addWidget(self.getAdvancedPanel())
self.connect(self.modeCombo, QtCore.SIGNAL('currentIndexChanged(const QString &)'), self.switchModeSlot)
self.switchModeSlot('basic')
self.searchButton = QtGui.QPushButton('Search')
self.searchButton.clicked.connect(self.searchSlot)
layout.addWidget(self.searchButton)
self.__results = []
def switchModeSlot(self, text):
basic = (str(text).lower() == 'basic')
self.getBasicPanel().setVisible(basic)
self.getAdvancedPanel().setVisible(not basic)
def getBasicPanel(self):
"""Create a widget for basic level usage."""
if not hasattr(self, '_basicPanel'):
self._basicPanel = QtGui.QWidget()
self.searchRootEdit = QtGui.QLineEdit()
self.searchRootEdit.returnPressed.connect(self.searchSlot)
self.searchRootLabel = QtGui.QLabel('Search under')
self.typeEdit = QtGui.QLineEdit()
self.typeEdit.returnPressed.connect(self.searchSlot)
self.typeLabel = QtGui.QLabel('for elements of type')
self.fieldEdit = QtGui.QLineEdit()
self.fieldEdit.returnPressed.connect(self.searchSlot)
self.fieldLabel = QtGui.QLabel('or elements with field')
self.comparisonCombo = QtGui.QComboBox()
self.comparisonCombo.addItem('=', QtCore.QVariant(QtCore.QString('=')))
self.comparisonCombo.addItem('<', QtCore.QVariant(QtCore.QString('<')))
self.comparisonCombo.addItem('>', QtCore.QVariant(QtCore.QString('>')))
self.comparisonCombo.addItem('<=', QtCore.QVariant(QtCore.QString('<=')))
self.comparisonCombo.addItem('>=', QtCore.QVariant(QtCore.QString('>=')))
self.comparisonCombo.addItem('!=', QtCore.QVariant(QtCore.QString('!=')))
self.valueEdit = QtGui.QLineEdit()
self.valueEdit.returnPressed.connect(self.searchSlot)
self.recurseButton = QtGui.QRadioButton('Search inside children recursively')
self.recurseButton.setChecked(True)
layout = QtGui.QGridLayout()
# layout.setVerticalSpacing(2)
self._basicPanel.setLayout(layout)
layout.addWidget(self.searchRootLabel, 0, 0, 1, 1)
layout.addWidget(self.searchRootEdit, 0, 1, 1, 4)
layout.addWidget(self.typeLabel, 1, 0, 1, 1)
layout.addWidget(self.typeEdit, 1, 1, 1, 4)
layout.addWidget(self.fieldLabel, 2, 0, 1, 1)
layout.addWidget(self.fieldEdit, 2, 1, 1, 1)
layout.addWidget(self.comparisonCombo, 2, 2, 1, 1)
layout.addWidget(self.valueEdit, 2, 3, 1, 2)
layout.addWidget(self.recurseButton, 3, 0, 1, 1)
# layout.addItem(QtGui.QSpacerItem(1,1), 3, 0, 1, 7)
return self._basicPanel
def getAdvancedPanel(self):
"""Create a panel for advanced users who can enter the search string
directly."""
if not hasattr(self, '_advancedPanel'):
self._advancedPanel = QtGui.QWidget(self)
self._searchEdit = QtGui.QLineEdit()
self._searchEdit.returnPressed.connect(self.searchSlot)
self._searchLabel = QtGui.QLabel('Search string:')
layout = QtGui.QHBoxLayout()
self._advancedPanel.setLayout(layout)
layout.addWidget(self._searchLabel)
layout.addWidget(self._searchEdit)
return self._advancedPanel
def searchSlot(self):
"""Do the search and store the paths of the id's found. The results
can be obtained later with getResults()"""
search_str = ''
if str(self.modeCombo.currentText()).lower() == 'basic':
root = str(self.searchRootEdit.text()).strip()
if len(root) == 0:
root = '/'
search_str += root
if not search_str.endswith('/'):
search_str += '/'
if self.recurseButton.isChecked():
search_str += '##'
else:
search_str += '#'
type_ = str(self.typeEdit.text()).strip()
if len(type_) > 0:
search_str += '[ISA=%s]' % (type_)
field = str(self.fieldEdit.text()).strip()
if len(field) > 0:
search_str += '[FIELD(%s)'
operator = str(self.comparisonCombo.currentText())
value = str(self.valueEdit.text())
search_str += '%s%s]' % (operator, value)
else:
search_str = str(self._searchEdit.text()).strip()
ids = moose.wildcardFind(search_str)
self.__results = [el.path for el in ids]
self.executed.emit(list(self.__results))
def getResults(self):
return list(self.__results)
def setSearchRoot(self, root):
self.searchRootEdit.setText(moose.element(root).path)
def main():
"""Test main"""
model = moose.Neutral('/model')
moose.loadModel('../Demos/Genesis_files/Kholodenko.g', '/model/Kholodenko')
# tab = moose.element('/model/Kholodenko/graphs/conc1/MAPK_PP.Co')
# print tab
# for t in tab.children:
# print t
app = QtGui.QApplication(sys.argv)
mainwin = QtGui.QMainWindow()
mainwin.setWindowTitle('Model tree test')
wildcardWidget = SearchWidget()
mainwin.setCentralWidget(wildcardWidget)
mainwin.show()
sys.exit(app.exec_())
if __name__ == '__main__':
main()
#
# msearch.py ends here