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:
elem = ' %s="%s"' % (attr.nodeName(), attr.nodeValue())
qslist.append(elem)
ElemNameAndAtts = '%s%s'% (node.nodeName(), qslist.join(' '))
- obj = QObject()
- obj.setProperty('nodeType', QString('element'))
- obj.setProperty('content', ElemNameAndAtts)
- return obj
+ return QVariant(
+ {QString('nodeType'):QVariant(QString('element')),
+ QString('content'):ElemNameAndAtts})
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
+ return QVariant(
+ {QString('nodeType'):QVariant(QString('text')),
+ QString('content'):node.nodeValue()})
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
+ return QVariant(
+ {QString('nodeType'):QVariant(QString('comment')),
+ QString('content'):node.nodeValue()})
elif node.nodeType() == QDomNode.DocumentNode:
return QString('unsupported node type')
elif node.nodeType() == QDomNode.DocumentTypeNode:
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())))
+ QVarMapAccess = lambda qv, key: qv.toMap()[QString(key)].toString()
+ dataAsQVarMap = index.model().data(index)
+ nodeType = str(QVarMapAccess(dataAsQVarMap, 'nodeType'))
+ delegate = self.delegates.get(nodeType)
+ print "WTF IS THIS TYPE:", type(index.model().data(index))
+ print "NODE TYPE:", nodeType
#print "DELEGS DICT:", self.delegates
#print "NODETYPE:", nodeType.toString()
if delegate is not None:
# doesn't know about my QObject strategy.
QItemDelegate.paint(self, painter, option, index)
- def sizeHint(self, option, index):
- fm = option.fontMetrics
- text = index.model().data(index).property('content').toString()
- 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)
+# 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):
- text = index.model().data(index)
palette = QApplication.palette()
document = QTextDocument()
document.setDefaultFont(option.font)
highGlobPattern = '<<b>%s</b>%s>'
highAttPattern = ' <b>%s</b>="%s"'
def getHtmlText(plainText, globPattern, attPattern):
- print "PLAIN TEXT:", plainText
tmp = plainText.split(' ', 1)
- print "TMP:", tmp
elemName = tmp[0]
AttListHtml = ''
if len(tmp) > 1:
return html
def colorize(color, text):
return '<font color=' + color + '>' + text + '</font>'
- text = str(index.model().data(index).property('content').toString())
- print "TEXT:", text
+ QVarMapAccess = lambda qv, key: qv.toMap()[QString(key)].toString()
+ dataAsQVarMap = index.model().data(index)
+ text = str(QVarMapAccess(dataAsQVarMap, 'content'))
if option.state & QStyle.State_Selected:
htmlText = colorize(palette.highlightedText().color().name(),
getHtmlText(text, highGlobPattern, highAttPattern))
if option.state & QStyle.State_Selected \
else palette.base().color()
painter.save()
- print "COLOR:", color.name()
# voodoo: if not highlighted, filling the rect
# with the base color makes no difference
painter.fillRect(option.rect, color)
document.drawContents(painter)
painter.restore()
- def sizeHint(self, option, index):
- sizeHint(self, option, index)
-
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>'
+ QVarMapAccess = lambda qv, key: qv.toMap()[QString(key)].toString()
+ 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):
- #print "TEXT DELEG CALLED"
paint(self, painter, option, index)
def paint(self, painter, option, index):