4 from PyQt4.QtCore import *
5 from PyQt4.QtGui import *
6 from PyQt4.QtXml import *
8 from sface.config import config
9 from sface.screens.sfascreen import SfaScreen
11 class DomModel(QAbstractItemModel):
12 def __init__(self, document, parent = 0):
13 QAbstractItemModel.__init__(self, parent)
14 self.domDocument = document
15 # one of the children of the rootItem is the 'xml' thing.
17 childList = document.childNodes()
18 for i in range(childList.count()):
19 currElem = childList.item(i)
20 if (currElem.nodeType() == QDomNode.ProcessingInstructionNode):
21 document.removeChild(currElem)
23 self.rootItem = DomItem(document, 0);
25 def data(self, index, role = Qt.DisplayRole):
26 # for interesting nodes, returns a dict wrapped into a QVariant.
27 if not index.isValid():
29 if role != Qt.DisplayRole:
31 node = index.internalPointer().node()
32 attributeMap = node.attributes()
36 if node.nodeType() == QDomNode.ElementNode:
37 qslist = QStringList()
38 for i in range(attributeMap.count()):
39 attr = attributeMap.item(i)
40 elem = ' %s="%s"' % (attr.nodeName(), attr.nodeValue())
42 ElemNameAndAtts = '%s%s'% (node.nodeName(), qslist.join(' '))
44 {QString('nodeType'):QVariant(QString('element')),
45 QString('content'):ElemNameAndAtts})
46 elif node.nodeType() == QDomNode.AttributeNode:
48 elif node.nodeType() == QDomNode.TextNode:
50 {QString('nodeType'):QVariant(QString('text')),
51 QString('content'):node.nodeValue()})
52 elif node.nodeType() == QDomNode.CDATASectionNode:
53 return QString('unsupported node type')
54 elif node.nodeType() == QDomNode.EntityReferenceNode:
55 return QString('unsupported node type')
56 elif node.nodeType() == QDomNode.EntityNode:
57 return QString('unsupported node type')
58 elif node.nodeType() == QDomNode.ProcessingInstructionNode:
60 elif node.nodeType() == QDomNode.CommentNode:
62 {QString('nodeType'):QVariant(QString('comment')),
63 QString('content'):node.nodeValue()})
64 elif node.nodeType() == QDomNode.DocumentNode:
65 return QString('unsupported node type')
66 elif node.nodeType() == QDomNode.DocumentTypeNode:
67 return QString('unsupported node type')
68 elif node.nodeType() == QDomNode.DocumentFragmentNode:
69 return QString('unsupported node type')
70 elif node.nodeType() == QDomNode.NotationNode:
71 return QString('unsupported node type')
72 elif node.nodeType() == QDomNode.BaseNode:
73 return QString('unsupported node type')
74 elif node.nodeType() == QDomNode.CharacterDataNode:
75 return QString('unsupported node type')
81 def flags(self, index):
82 if not index.isValid():
83 return Qt.ItemIsEnabled
84 return Qt.ItemIsEnabled | Qt.ItemIsSelectable
86 def headerData(self, section, orientation, role):
89 def index(self, row, column, parent=None):
90 if not parent or not parent.isValid():
91 parentItem = self.rootItem
93 parentItem = parent.internalPointer()
95 childItem = parentItem.child(row)
96 # childItem would be None to say "false"?
98 return self.createIndex(row, column, childItem)
102 def parent(self, child):
103 if not child.isValid():
105 childItem = child.internalPointer()
106 parentItem = childItem.parent()
108 if not parentItem or parentItem == self.rootItem:
110 return self.createIndex(parentItem.row(), 0, parentItem)
112 def rowCount(self, parent=None):
113 if not parent or not parent.isValid():
114 parentItem = self.rootItem
116 parentItem = parent.internalPointer()
118 return parentItem.node().childNodes().count()
120 def columnCount(self, parent):
121 # just one column we'll print tag name (and attributes) or the
127 # wrapper around PyQt4.QtXml.QDomNode it keeps an hash of
128 # childrens for performance reasons
130 def __init__(self, node, row, parent = 0):
131 # node is of type PyQt4.QtXml.QDomNode
133 self.parentItem = parent
138 if i in self.childItems:
139 return self.childItems[i]
140 if i >= 0 and i < self.domNode.childNodes().count():
141 childNode = self.domNode.childNodes().item(i)
142 childItem = DomItem(childNode, i, self)
143 self.childItems[i] = childItem
148 return self.parentItem
154 return self.rowNumber
156 class XmlView(QTreeView):
157 def __init__(self, parent):
158 QTreeView.__init__(self, parent)
160 self.setAnimated(True)
161 self.setItemsExpandable(True)
162 self.setRootIsDecorated(True)
163 self.setHeaderHidden(True)
164 self.setAttribute(Qt.WA_MacShowFocusRect, 0)
165 self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
167 class XmlWindow(QDialog):
168 def __init__(self, parent=None, title='XML Window'):
169 QDialog.__init__(self, parent)
170 self.setWindowTitle(title)
176 self.view = XmlView(self)
177 self.delegate = XmlDelegate(self)
178 self.view.setItemDelegate(self.delegate)
179 self.delegate.insertNodeDelegate('element', ElemNodeDelegate())
180 self.delegate.insertNodeDelegate('text', TextNodeDelegate())
181 self.delegate.insertNodeDelegate('comment', CommentNodeDelegate())
182 layout = QVBoxLayout()
183 layout.addWidget(self.view)
184 self.setLayout(layout)
192 def updateView(self):
198 self.document = QDomDocument(self.title)
199 self.model = DomModel(self.document, self)
201 self.view.setModel(self.model)
202 self.view.expand(self.model.index(0, 0)) #expand first level only
204 #move the code below to rspec window
205 rspec_file = config.getSliceRSpecFile()
206 if not os.path.exists(rspec_file):
209 self.document.setContent(open(rspec_file,'r').read())
213 class XmlDelegate(QItemDelegate):
215 def __init__(self, parent=None):
216 QAbstractItemDelegate.__init__(self, parent)
219 def insertNodeDelegate(self, nodeType, delegate):
220 delegate.setParent(self)
221 self.delegates[nodeType] = delegate
223 def removeNodeDelegate(self, nodeType, delegate):
224 if nodeType in self.delegates:
225 del self.delegates[nodeType]
227 def paint(self, painter, option, index):
228 QVarMapAccess = lambda qv, key: qv.toMap()[QString(key)].toString()
229 dataAsQVarMap = index.model().data(index)
230 nodeType = str(QVarMapAccess(dataAsQVarMap, 'nodeType'))
231 delegate = self.delegates.get(nodeType)
232 print "WTF IS THIS TYPE:", type(index.model().data(index))
233 print "NODE TYPE:", nodeType
234 #print "DELEGS DICT:", self.delegates
235 #print "NODETYPE:", nodeType.toString()
236 if delegate is not None:
237 #print "WOW DELEG ISNT NONE"
238 delegate.paint(painter, option, index)
241 # not sure this will ever work. this delegate
242 # doesn't know about my QObject strategy.
243 QItemDelegate.paint(self, painter, option, index)
245 # def sizeHint(self, option, index):
246 # fm = option.fontMetrics
247 # print "TYPE:", str(type(index.model().data(index).convert(QObject)))
248 # text = "the fish doesn't talk"
249 # #text = str(index.model().data(index).property('content').toString())
250 # nodeType = str(index.model().data(index).property('nodeType').toString())
251 # if nodeType == 'element' or nodeType == 'comment':
253 # elif nodeType == 'text':
254 # numlines = text.count('\n')
255 # sys.__stdout__.write("TEXT: \n" + text)
258 # document = QTextDocument()
259 # document.setDefaultFont(option.font)
260 # document.setHtml(text)
261 # # the +5 is for margin. The +4 is voodoo;
262 # # fm.height just give it too small.
263 # return QSize(document.idealWidth() + 5, (fm.height() + 4) * numlines)
265 class ElemNodeDelegate(QAbstractItemDelegate):
266 def paint(self, painter, option, index):
267 palette = QApplication.palette()
268 document = QTextDocument()
269 document.setDefaultFont(option.font)
270 nonHighGlobPattern = '<<b><font color="#b42be2">%s</font></b>%s>'
271 nonHighAttPattern = ' <b>%s</b>="<font color="#1e90ff">%s</font>"'
272 highGlobPattern = '<<b>%s</b>%s>'
273 highAttPattern = ' <b>%s</b>="%s"'
274 def getHtmlText(plainText, globPattern, attPattern):
275 tmp = plainText.split(' ', 1)
279 # many elems don't have atts...
280 attList = tmp[1].split()
284 attValue = tmp[1][1:-1]
285 AttListHtml += (nonHighAttPattern % (attName, attValue))
286 html = (globPattern % (elemName, AttListHtml))
288 def colorize(color, text):
289 return '<font color=' + color + '>' + text + '</font>'
290 QVarMapAccess = lambda qv, key: qv.toMap()[QString(key)].toString()
291 dataAsQVarMap = index.model().data(index)
292 text = str(QVarMapAccess(dataAsQVarMap, 'content'))
293 if option.state & QStyle.State_Selected:
294 htmlText = colorize(palette.highlightedText().color().name(),
295 getHtmlText(text, highGlobPattern, highAttPattern))
296 document.setHtml(QString(htmlText))
298 htmlText = getHtmlText(text, nonHighGlobPattern, nonHighAttPattern)
299 document.setHtml(QString(htmlText))
300 color = palette.highlight().color() \
301 if option.state & QStyle.State_Selected \
302 else palette.base().color()
304 # voodoo: if not highlighted, filling the rect
305 # with the base color makes no difference
306 painter.fillRect(option.rect, color)
307 painter.translate(option.rect.x(), option.rect.y())
308 document.drawContents(painter)
311 class TextNodeDelegate(QAbstractItemDelegate):
312 def paint(self, painter, option, index):
313 palette = QApplication.palette()
314 document = QTextDocument()
315 document.setDefaultFont(option.font)
316 def verbatimize(text):
317 text.replace('\n', '<br>')
318 return '<pre>' + text + '</pre'
319 def colorize(color, text):
320 return '<font color=' + color + '>' + text + '</font>'
321 QVarMapAccess = lambda qv, key: qv.toMap()[QString(key)].toString()
322 dataAsQVarMap = index.model().data(index)
323 text = str(QVarMapAccess(dataAsQVarMap, 'content'))
324 if option.state & QStyle.State_Selected:
325 htmlText = colorize(palette.highlightedText().color().name(),
327 document.setHtml(QString(htmlText))
329 htmlText = verbatimize(text)
330 document.setHtml(QString(htmlText))
331 color = palette.highlight().color() \
332 if option.state & QStyle.State_Selected \
333 else palette.base().color()
335 # voodoo: if not highlighted, filling the rect
336 # with the base color makes no difference
337 painter.fillRect(option.rect, color)
338 painter.translate(option.rect.x(), option.rect.y())
339 document.drawContents(painter)
342 class CommentNodeDelegate(QAbstractItemDelegate):
343 def paint(self, painter, option, index):
344 paint(self, painter, option, index)
346 def paint(self, painter, option, index):
347 text = index.model().data(index).property('content').toString()
348 palette = QApplication.palette()
349 document = QTextDocument()
350 document.setDefaultFont(option.font)
351 if option.state & QStyle.State_Selected:
352 rx = QRegExp(QString('<font .*>'))
354 # If selected, I remove the <font color="..."> by hand,
355 # and give the highlight color
356 document.setHtml(QString("<font color=%1>%2</font>") \
357 .arg(palette.highlightedText().color().name())\
358 .arg(text.replace(rx, QString('')).
359 replace(QString('</font>'),QString(''))))
361 document.setHtml(text)
362 color = palette.highlight().color() \
363 if option.state & QStyle.State_Selected \
364 else palette.base().color()
366 # voodoo: if not highlighted, filling the rect
367 # with the base color makes no difference
368 painter.fillRect(option.rect, color)
369 painter.translate(option.rect.x(), option.rect.y())
370 document.drawContents(painter)