from sface.config import config
from sface.screens.sfascreen import SfaScreen
+class nodeData(QVariant):
+ def __init__(self, *args, **kws):
+ QVariant.__init__(self, *args, **kws)
+ self.type = None
+
+ def setType(self, typ):
+ self.type = typ
+
+ def getType(self):
+ return self.type
+
class DomModel(QAbstractItemModel):
def __init__(self, document, parent = 0):
QAbstractItemModel.__init__(self, parent)
self.rootItem = DomItem(document, 0);
def data(self, index, role = Qt.DisplayRole):
- # sometimes it return a QString, sometimes a QVariant. not good.
+ # for interesting nodes, returns a dict wrapped into a QVariant.
if not index.isValid():
return QVariant()
if role != Qt.DisplayRole:
attr = attributeMap.item(i)
elem = ' %s="%s"' % (attr.nodeName(), attr.nodeValue())
qslist.append(elem)
- ElemNameAndAtts = QString('<%s%s>'% (node.nodeName(), qslist.join(' ')))
- obj = QObject()
- obj.setProperty('nodeType', QString('element'))
- obj.setProperty('content', ElemNameAndAtts)
- return obj
+ ElemNameAndAtts = '%s%s'% (node.nodeName(), qslist.join(' '))
+ answer = nodeData(ElemNameAndAtts)
+ answer.setType('element')
+ return answer
elif node.nodeType() == QDomNode.AttributeNode:
return QVariant()
elif node.nodeType() == QDomNode.TextNode:
- obj = QObject()
- obj.setProperty('nodeType', QString('text'))
- obj.setProperty('content', node.nodeValue())
- return obj
+ answer = nodeData(node.nodeValue())
+ answer.setType('text')
+ return answer
elif node.nodeType() == QDomNode.CDATASectionNode:
return QString('unsupported node type')
elif node.nodeType() == QDomNode.EntityReferenceNode:
elif node.nodeType() == QDomNode.ProcessingInstructionNode:
return QVariant()
elif node.nodeType() == QDomNode.CommentNode:
- obj = QObject()
- obj.setProperty('nodeType', QString('comment'))
- obj.setProperty('content', node.nodeValue())
- return obj
+ answer = nodeData(node.nodeValue())
+ answer.setType('comment')
+ return answer
elif node.nodeType() == QDomNode.DocumentNode:
return QString('unsupported node type')
elif node.nodeType() == QDomNode.DocumentTypeNode:
return self.rowNumber
class XmlView(QTreeView):
- def __init__(self, parent):
+ def __init__(self, parent=None):
QTreeView.__init__(self, parent)
+ delegate = XmlDelegate(self)
+ delegate.insertNodeDelegate('element', ElemNodeDelegate())
+ delegate.insertNodeDelegate('text', TextNodeDelegate())
+ delegate.insertNodeDelegate('comment', CommentNodeDelegate())
+ self.setItemDelegate(delegate)
+
self.setAnimated(True)
self.setItemsExpandable(True)
self.setRootIsDecorated(True)
self.setAttribute(Qt.WA_MacShowFocusRect, 0)
self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
-class XmlWindow(QDialog):
- def __init__(self, parent=None, title='XML Window'):
- QDialog.__init__(self, parent)
- self.setWindowTitle(title)
-
- self.document = None
- self.model = None
- self.title = title
-
- self.view = XmlView(self)
- self.delegate = XmlDelegate(self)
- self.view.setItemDelegate(self.delegate)
- self.delegate.insertNodeDelegate('element', ElemNodeDelegate())
- self.delegate.insertNodeDelegate('text', TextNodeDelegate())
- self.delegate.insertNodeDelegate('comment', CommentNodeDelegate())
- layout = QVBoxLayout()
- layout.addWidget(self.view)
- self.setLayout(layout)
-
- self.updateView()
-
- def show(self):
- self.updateView()
- QDialog.show(self)
-
- def updateView(self):
- del self.document
- del self.model
- self.document = None
- self.model = None
-
- self.document = QDomDocument(self.title)
- self.model = DomModel(self.document, self)
-
- self.view.setModel(self.model)
- self.view.expand(self.model.index(0, 0)) #expand first level only
-
- #move the code below to rspec window
- rspec_file = config.getSliceRSpecFile()
- if not os.path.exists(rspec_file):
- return
-
- self.document.setContent(open(rspec_file,'r').read())
-
-
-
class XmlDelegate(QItemDelegate):
def __init__(self, parent=None):
del self.delegates[nodeType]
def paint(self, painter, option, index):
- nodeType = index.model().data(index).property('nodeType')
- delegate = self.delegates.get(str(nodeType.toString()))
- print "TYPE:", str(type(str(nodeType.toString())))
- print "DELEGS DICT:", self.delegates
- print "NODETYPE:", nodeType.toString()
+ nodeData = index.model().data(index)
+ nodeType = nodeData.getType()
+ delegate = self.delegates.get(nodeType)
if delegate is not None:
- print "WOW DELEG ISNT NONE"
delegate.paint(painter, option, index)
else:
- print "ELSE BRANCH"
- # not sure this will ever work. this delegate
- # doesn't know about my QObject strategy.
QItemDelegate.paint(self, painter, option, index)
+ def sizeHint(self, option, index):
+ fm = option.fontMetrics
+ nodeData = index.model().data(index)
+ nodeType = nodeData.getType()
+ text = nodeData.toString()
+ if nodeType == 'element' or nodeType == 'comment':
+ numlines = 1
+ elif nodeType == 'text':
+ nl = text.count('\n')
+ numlines = nl if nl > 0 else 1
+ else:
+ numlines = 1
+ document = QTextDocument()
+ document.setDefaultFont(option.font)
+ document.setHtml(text)
+ # the +5 is for margin. The +4 is voodoo;
+ # fm.height just give it too small.
+ return QSize(document.idealWidth() + 5, (fm.height() + 4) * numlines)
+
class ElemNodeDelegate(QAbstractItemDelegate):
def paint(self, painter, option, index):
- print "ELEM DELEG CALLED"
- paint(self, painter, option, index)
-
- def sizeHint(self, option, index):
- sizeHint(self, option, index)
+ palette = QApplication.palette()
+ document = QTextDocument()
+ document.setDefaultFont(option.font)
+ nonHighGlobPattern = '<<b><font color="#b42be2">%s</font></b>%s>'
+ nonHighAttPattern = ' <b>%s</b>="<font color="#1e90ff">%s</font>"'
+ highGlobPattern = '<<b>%s</b>%s>'
+ highAttPattern = ' <b>%s</b>="%s"'
+ def getHtmlText(plainText, globPattern, attPattern):
+ tmp = plainText.split(' ', 1)
+ elemName = tmp[0]
+ AttListHtml = ''
+ if len(tmp) > 1:
+ # many elems don't have atts...
+ attList = tmp[1].split()
+ for att in attList:
+ tmp = att.split('=')
+ attName = tmp[0]
+ attValue = tmp[1][1:-1]
+ AttListHtml += (attPattern % (attName, attValue))
+ html = (globPattern % (elemName, AttListHtml))
+ return html
+ def colorize(color, text):
+ return '<font color=' + color + '>' + text + '</font>'
+ nodeData = index.model().data(index)
+ nodeType = nodeData.getType()
+ # Uff... QString Vs string...
+ text = str(nodeData.toString())
+ if option.state & QStyle.State_Selected:
+ htmlText = colorize(palette.highlightedText().color().name(),
+ getHtmlText(text, highGlobPattern, highAttPattern))
+ document.setHtml(QString(htmlText))
+ else:
+ htmlText = getHtmlText(text, nonHighGlobPattern, nonHighAttPattern)
+ document.setHtml(QString(htmlText))
+ color = palette.highlight().color() \
+ if option.state & QStyle.State_Selected \
+ else palette.base().color()
+ painter.save()
+ # voodoo: if not highlighted, filling the rect
+ # with the base color makes no difference
+ painter.fillRect(option.rect, color)
+ painter.translate(option.rect.x(), option.rect.y())
+ document.drawContents(painter)
+ painter.restore()
class TextNodeDelegate(QAbstractItemDelegate):
def paint(self, painter, option, index):
- print "TEXT DELEG CALLED"
- paint(self, painter, option, index)
-
- def sizeHint(self, option, index):
- sizeHint(self, option, index)
+ palette = QApplication.palette()
+ document = QTextDocument()
+ document.setDefaultFont(option.font)
+ def verbatimize(text):
+ text.replace('\n', '<br>')
+ return '<pre>' + text + '</pre'
+ def colorize(color, text):
+ return '<font color=' + color + '>' + text + '</font>'
+ nodeData = index.model().data(index)
+ nodeType = nodeData.getType()
+ text = nodeData.toString()
+ if option.state & QStyle.State_Selected:
+ htmlText = colorize(palette.highlightedText().color().name(),
+ verbatimize(text))
+ document.setHtml(QString(htmlText))
+ else:
+ htmlText = verbatimize(text)
+ document.setHtml(QString(htmlText))
+ color = palette.highlight().color() \
+ if option.state & QStyle.State_Selected \
+ else palette.base().color()
+ painter.save()
+ # voodoo: if not highlighted, filling the rect
+ # with the base color makes no difference
+ painter.fillRect(option.rect, color)
+ painter.translate(option.rect.x(), option.rect.y())
+ document.drawContents(painter)
+ painter.restore()
class CommentNodeDelegate(QAbstractItemDelegate):
def paint(self, painter, option, index):
- print "TEXT DELEG CALLED"
paint(self, painter, option, index)
- def sizeHint(self, option, index):
- sizeHint(self, option, index)
-
-
def paint(self, painter, option, index):
text = index.model().data(index).property('content').toString()
palette = QApplication.palette()
document.drawContents(painter)
painter.restore()
-def sizeHint(self, option, index):
- fm = option.fontMetrics
- text = index.model().data(index)
- document = QTextDocument()
- document.setDefaultFont(option.font)
- document.setHtml(text)
- # the +5 is for margin. The +4 is voodoo;
- # fm.height just give it too small.
- return QSize(document.idealWidth() + 5, fm.height() + 4)
+
git://git.onelab.eu / sface.git / blobdiff