X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sface%2Fxmlwidget.py;h=b24de60248a918289c4fb3d6e77ff03135e34196;hb=9c27ebe87eabf14d9bcc3d7b54e88ce6b33b3be6;hp=a52522a512aaa36d414660f03f964feb30d35a09;hpb=349fb978a1b7a88e741beef6aea4ac3d98a56efd;p=sface.git
diff --git a/sface/xmlwidget.py b/sface/xmlwidget.py
index a52522a..b24de60 100644
--- a/sface/xmlwidget.py
+++ b/sface/xmlwidget.py
@@ -8,6 +8,19 @@ from PyQt4.QtXml import *
from sface.config import config
from sface.screens.sfascreen import SfaScreen
+def QVarMapAccess(qv, key):
+ # helper function. qv is a dict wrapped into a QVariant
+ print 10*'='
+ print "DICT:", qv.toMap()
+ if len(qv.toMap().keys()) == 0:
+ print "EMPTY!"
+ import traceback
+ traceback.print_stack()
+ return None
+ else:
+ return qv.toMap()[QString(key)].toString()
+
+
class DomModel(QAbstractItemModel):
def __init__(self, document, parent = 0):
QAbstractItemModel.__init__(self, parent)
@@ -19,11 +32,12 @@ class DomModel(QAbstractItemModel):
currElem = childList.item(i)
if (currElem.nodeType() == QDomNode.ProcessingInstructionNode):
document.removeChild(currElem)
+ print "REMOVED!"
break
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:
@@ -37,39 +51,61 @@ class DomModel(QAbstractItemModel):
qslist = QStringList()
for i in range(attributeMap.count()):
attr = attributeMap.item(i)
- elem = ' %s="%s"' % (attr.nodeName(), attr.nodeValue())
+ elem = ' %s="%s"' % (attr.nodeName(), attr.nodeValue())
qslist.append(elem)
- return QString('<%s%s>'% (node.nodeName(), qslist.join(' ')))
+ ElemNameAndAtts = '%s%s'% (node.nodeName(), qslist.join(' '))
+ print "1"
+ return QVariant(
+ {'nodeType':QVariant(QString('element')),
+ 'content':ElemNameAndAtts})
elif node.nodeType() == QDomNode.AttributeNode:
- return QString('Whozat?!')
+ print "2"
+ return QVariant()
elif node.nodeType() == QDomNode.TextNode:
- return node.nodeValue()
+ print "3"
+ return QVariant(
+ {'nodeType':QVariant(QString('text')),
+ 'content':node.nodeValue()})
elif node.nodeType() == QDomNode.CDATASectionNode:
+ print "4"
return QString('unsupported node type')
elif node.nodeType() == QDomNode.EntityReferenceNode:
+ print "5"
return QString('unsupported node type')
elif node.nodeType() == QDomNode.EntityNode:
+ print "6"
return QString('unsupported node type')
elif node.nodeType() == QDomNode.ProcessingInstructionNode:
+ print "7"
return QVariant()
- #return node.nodeName()
elif node.nodeType() == QDomNode.CommentNode:
- return QString('#').append(node.nodeValue())
+ print "8"
+ return QVariant(
+ {'nodeType':QVariant(QString('comment')),
+ 'content':node.nodeValue()})
elif node.nodeType() == QDomNode.DocumentNode:
+ print "9"
return QString('unsupported node type')
elif node.nodeType() == QDomNode.DocumentTypeNode:
+ print "10"
return QString('unsupported node type')
elif node.nodeType() == QDomNode.DocumentFragmentNode:
+ print "12"
return QString('unsupported node type')
elif node.nodeType() == QDomNode.NotationNode:
+ print "13"
return QString('unsupported node type')
elif node.nodeType() == QDomNode.BaseNode:
+ print "14"
return QString('unsupported node type')
elif node.nodeType() == QDomNode.CharacterDataNode:
+ print "15"
return QString('unsupported node type')
else:
+ print "16"
return QVariant()
else:
+ print "17"
return QVariant()
def flags(self, index):
@@ -148,9 +184,15 @@ class DomItem:
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)
@@ -158,70 +200,91 @@ class XmlView(QTreeView):
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)
- 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(QAbstractItemDelegate):
+class XmlDelegate(QItemDelegate):
def __init__(self, parent=None):
QAbstractItemDelegate.__init__(self, parent)
+ self.delegates = {}
+
+ def insertNodeDelegate(self, nodeType, delegate):
+ delegate.setParent(self)
+ self.delegates[nodeType] = delegate
+ def removeNodeDelegate(self, nodeType, delegate):
+ if nodeType in self.delegates:
+ del self.delegates[nodeType]
+
def paint(self, painter, option, index):
- text = index.model().data(index)
+ print "ASKING FOR DATA"
+ dataAsQVarMap = index.model().data(index)
+ print "GOT DATA"
+ nodeType = str(QVarMapAccess(dataAsQVarMap, 'nodeType'))
+ delegate = self.delegates.get(nodeType)
+ #print "DELEGS DICT:", self.delegates
+ #print "NODETYPE:", nodeType.toString()
+ 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 QMap strategy.
+ QItemDelegate.paint(self, painter, option, index)
+
+# def sizeHint(self, option, index):
+# fm = option.fontMetrics
+# print "TYPE:", str(type(index.model().data(index).convert(QObject)))
+# text = "the fish doesn't talk"
+# #text = str(index.model().data(index).property('content').toString())
+# nodeType = str(index.model().data(index).property('nodeType').toString())
+# if nodeType == 'element' or nodeType == 'comment':
+# numlines = 1
+# elif nodeType == 'text':
+# numlines = text.count('\n')
+# sys.__stdout__.write("TEXT: \n" + text)
+# 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):
palette = QApplication.palette()
document = QTextDocument()
document.setDefaultFont(option.font)
+ nonHighGlobPattern = '<%s%s>'
+ nonHighAttPattern = ' %s="%s"'
+ highGlobPattern = '<%s%s>'
+ highAttPattern = ' %s="%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 += (nonHighAttPattern % (attName, attValue))
+ html = (globPattern % (elemName, AttListHtml))
+ return html
+ def colorize(color, text):
+ return '' + text + ''
+ dataAsQVarMap = index.model().data(index)
+ text = str(QVarMapAccess(dataAsQVarMap, 'content'))
if option.state & QStyle.State_Selected:
- rx = QRegExp(QString(''))
- rx.setMinimal(True)
- # If selected, I remove the by hand,
- # and give the highlight color
- document.setHtml(QString("%2") \
- .arg(palette.highlightedText().color().name())\
- .arg(text.replace(rx, QString('')).
- replace(QString(''),QString(''))))
+ htmlText = colorize(palette.highlightedText().color().name(),
+ getHtmlText(text, highGlobPattern, highAttPattern))
+ document.setHtml(QString(htmlText))
else:
- document.setHtml(text)
+ htmlText = getHtmlText(text, nonHighGlobPattern, nonHighAttPattern)
+ document.setHtml(QString(htmlText))
color = palette.highlight().color() \
if option.state & QStyle.State_Selected \
else palette.base().color()
@@ -233,12 +296,65 @@ class XmlDelegate(QAbstractItemDelegate):
document.drawContents(painter)
painter.restore()
- def sizeHint(self, option, index):
- fm = option.fontMetrics
- text = index.model().data(index)
+class TextNodeDelegate(QAbstractItemDelegate):
+ def paint(self, painter, option, index):
+ palette = QApplication.palette()
document = QTextDocument()
document.setDefaultFont(option.font)
+ def verbatimize(text):
+ text.replace('\n', '
')
+ return '' + text + '
' + text + ''
+ dataAsQVarMap = index.model().data(index)
+ text = str(QVarMapAccess(dataAsQVarMap, 'content'))
+ 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):
+ paint(self, painter, option, index)
+
+def paint(self, painter, option, index):
+ text = index.model().data(index).property('content').toString()
+ palette = QApplication.palette()
+ document = QTextDocument()
+ document.setDefaultFont(option.font)
+ if option.state & QStyle.State_Selected:
+ rx = QRegExp(QString(''))
+ rx.setMinimal(True)
+ # If selected, I remove the by hand,
+ # and give the highlight color
+ document.setHtml(QString("%2") \
+ .arg(palette.highlightedText().color().name())\
+ .arg(text.replace(rx, QString('')).
+ replace(QString(''),QString(''))))
+ else:
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)
+ 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()
+
+