5 from PyQt4.QtCore import *
6 from PyQt4.QtGui import *
7 from PyQt4.QtXml import *
9 from sface.config import config
10 from sface.screens.sfascreen import SfaScreen
12 class DomModel(QAbstractItemModel):
13 def __init__(self, document, parent = 0):
14 QAbstractItemModel.__init__(self, parent)
15 self.domDocument = document
16 # one of the children of the rootItem is the 'xml' thing.
18 childList = document.childNodes()
19 for i in range(childList.count()):
20 currElem = childList.item(i)
21 if (currElem.nodeType() == QDomNode.ProcessingInstructionNode):
22 document.removeChild(currElem)
24 self.rootItem = DomItem(document, 0);
26 def data(self, index, role = Qt.DisplayRole):
27 # sometimes it return a QString, sometimes a QVariant. not good.
28 if not index.isValid():
30 if role != Qt.DisplayRole:
32 node = index.internalPointer().node()
33 attributeMap = node.attributes()
37 if node.nodeType() == QDomNode.ElementNode:
38 qslist = QStringList()
39 for i in range(attributeMap.count()):
40 attr = attributeMap.item(i)
41 elem = ' %s="%s"' % (attr.nodeName(), attr.nodeValue())
43 ElemNameAndAtts = '%s%s'% (node.nodeName(), qslist.join(' '))
45 obj.setProperty('nodeType', QString('element'))
46 obj.setProperty('content', ElemNameAndAtts)
48 elif node.nodeType() == QDomNode.AttributeNode:
50 elif node.nodeType() == QDomNode.TextNode:
52 obj.setProperty('nodeType', QString('text'))
53 obj.setProperty('content', node.nodeValue())
55 elif node.nodeType() == QDomNode.CDATASectionNode:
56 return QString('unsupported node type')
57 elif node.nodeType() == QDomNode.EntityReferenceNode:
58 return QString('unsupported node type')
59 elif node.nodeType() == QDomNode.EntityNode:
60 return QString('unsupported node type')
61 elif node.nodeType() == QDomNode.ProcessingInstructionNode:
63 elif node.nodeType() == QDomNode.CommentNode:
65 obj.setProperty('nodeType', QString('comment'))
66 obj.setProperty('content', node.nodeValue())
68 elif node.nodeType() == QDomNode.DocumentNode:
69 return QString('unsupported node type')
70 elif node.nodeType() == QDomNode.DocumentTypeNode:
71 return QString('unsupported node type')
72 elif node.nodeType() == QDomNode.DocumentFragmentNode:
73 return QString('unsupported node type')
74 elif node.nodeType() == QDomNode.NotationNode:
75 return QString('unsupported node type')
76 elif node.nodeType() == QDomNode.BaseNode:
77 return QString('unsupported node type')
78 elif node.nodeType() == QDomNode.CharacterDataNode:
79 return QString('unsupported node type')
85 def flags(self, index):
86 if not index.isValid():
87 return Qt.ItemIsEnabled
88 return Qt.ItemIsEnabled | Qt.ItemIsSelectable
90 def headerData(self, section, orientation, role):
93 def index(self, row, column, parent=None):
94 if not parent or not parent.isValid():
95 parentItem = self.rootItem
97 parentItem = parent.internalPointer()
99 childItem = parentItem.child(row)
100 # childItem would be None to say "false"?
102 return self.createIndex(row, column, childItem)
106 def parent(self, child):
107 if not child.isValid():
109 childItem = child.internalPointer()
110 parentItem = childItem.parent()
112 if not parentItem or parentItem == self.rootItem:
114 return self.createIndex(parentItem.row(), 0, parentItem)
116 def rowCount(self, parent=None):
117 if not parent or not parent.isValid():
118 parentItem = self.rootItem
120 parentItem = parent.internalPointer()
122 return parentItem.node().childNodes().count()
124 def columnCount(self, parent):
125 # just one column we'll print tag name (and attributes) or the
131 # wrapper around PyQt4.QtXml.QDomNode it keeps an hash of
132 # childrens for performance reasons
134 def __init__(self, node, row, parent = 0):
135 # node is of type PyQt4.QtXml.QDomNode
137 self.parentItem = parent
142 if i in self.childItems:
143 return self.childItems[i]
144 if i >= 0 and i < self.domNode.childNodes().count():
145 childNode = self.domNode.childNodes().item(i)
146 childItem = DomItem(childNode, i, self)
147 self.childItems[i] = childItem
152 return self.parentItem
158 return self.rowNumber
160 class XmlView(QTreeView):
161 def __init__(self, parent):
162 QTreeView.__init__(self, parent)
164 self.setAnimated(True)
165 self.setItemsExpandable(True)
166 self.setRootIsDecorated(True)
167 self.setHeaderHidden(True)
168 self.setAttribute(Qt.WA_MacShowFocusRect, 0)
169 self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
171 class XmlWindow(QDialog):
172 def __init__(self, parent=None, title='XML Window'):
173 QDialog.__init__(self, parent)
174 self.setWindowTitle(title)
180 self.view = self.initView()
181 self.delegate = XmlDelegate(self)
182 self.view.setItemDelegate(self.delegate)
183 self.delegate.insertNodeDelegate('element', ElemNodeDelegate())
184 self.delegate.insertNodeDelegate('text', TextNodeDelegate())
185 self.delegate.insertNodeDelegate('comment', CommentNodeDelegate())
186 layout = QVBoxLayout()
187 layout.addWidget(self.view)
188 self.setLayout(layout)
199 def updateView(self):
205 self.document = QDomDocument(self.title)
206 self.model = DomModel(self.document, self)
208 self.view.setModel(self.model)
209 self.view.expand(self.model.index(0, 0)) #expand first level only
211 #move the code below to rspec window
212 rspec_file = config.getSliceRSpecFile()
213 if not os.path.exists(rspec_file):
216 self.document.setContent(open(rspec_file,'r').read())
220 class XmlDelegate(QItemDelegate):
222 def __init__(self, parent=None):
223 QAbstractItemDelegate.__init__(self, parent)
226 def insertNodeDelegate(self, nodeType, delegate):
227 delegate.setParent(self)
228 self.delegates[nodeType] = delegate
230 def removeNodeDelegate(self, nodeType, delegate):
231 if nodeType in self.delegates:
232 del self.delegates[nodeType]
234 def paint(self, painter, option, index):
235 if isinstance(index.model().data(index),QVariant):
237 nodeType = index.model().data(index).property('nodeType')
238 delegate = self.delegates.get(str(nodeType.toString()))
239 #print "TYPE:", str(type(str(nodeType.toString())))
240 #print "DELEGS DICT:", self.delegates
241 #print "NODETYPE:", nodeType.toString()
242 if delegate is not None:
243 #print "WOW DELEG ISNT NONE"
244 delegate.paint(painter, option, index)
247 # not sure this will ever work. this delegate
248 # doesn't know about my QObject strategy.
249 QItemDelegate.paint(self, painter, option, index)
251 def sizeHint(self, option, index):
252 fm = option.fontMetrics
253 if isinstance(index.model().data(index),QVariant):
255 text = index.model().data(index).property('content').toString()
256 document = QTextDocument()
257 document.setDefaultFont(option.font)
258 document.setHtml(text)
259 # the +5 is for margin. The +4 is voodoo;
260 # fm.height just give it too small.
261 return QSize(document.idealWidth() + 5, fm.height() + 4)
263 class ElemNodeDelegate(QAbstractItemDelegate):
264 def paint(self, painter, option, index):
265 text = index.model().data(index)
266 palette = QApplication.palette()
267 document = QTextDocument()
268 document.setDefaultFont(option.font)
269 nonHighGlobPattern = '<<b><font color="#b42be2">%s</font></b>%s>'
270 nonHighAttPattern = ' <b>%s</b>="<font color="#1e90ff">%s</font>"'
271 highGlobPattern = '<<b>%s</b>%s>'
272 highAttPattern = ' <b>%s</b>="%s"'
273 def getHtmlText(plainText, globPattern, attPattern):
274 # print "PLAIN TEXT:", plainText
275 tmp = plainText.split(' ', 1)
280 # many elems don't have atts...
281 # use shlex.split so we can handle quoted strings with spaces
282 # in them, like <link enpoints="foo bar">. Note that there are
283 # documented problems with shlex.split and unicode, so we
284 # convert any potential unicode to a string first.
285 attList = shlex.split(str(tmp[1]))
287 tmp = att.split('=',1)
292 # this shouldn't happen, but if it does, pretend the
293 # attribute value is blank.
296 AttListHtml += (nonHighAttPattern % (attName, attValue))
297 html = (globPattern % (elemName, AttListHtml))
299 def colorize(color, text):
300 return '<font color=' + color + '>' + text + '</font>'
301 text = str(index.model().data(index).property('content').toString())
302 # print "TEXT:", text
303 if option.state & QStyle.State_Selected:
304 htmlText = colorize(palette.highlightedText().color().name(),
305 getHtmlText(text, highGlobPattern, highAttPattern))
306 document.setHtml(QString(htmlText))
308 htmlText = getHtmlText(text, nonHighGlobPattern, nonHighAttPattern)
309 document.setHtml(QString(htmlText))
310 color = palette.highlight().color() \
311 if option.state & QStyle.State_Selected \
312 else palette.base().color()
314 # print "COLOR:", color.name()
315 # voodoo: if not highlighted, filling the rect
316 # with the base color makes no difference
317 painter.fillRect(option.rect, color)
318 painter.translate(option.rect.x(), option.rect.y())
319 document.drawContents(painter)
322 def sizeHint(self, option, index):
323 sizeHint(self, option, index)
325 class TextNodeDelegate(QAbstractItemDelegate):
326 def paint(self, painter, option, index):
327 #print "TEXT DELEG CALLED"
328 paint(self, painter, option, index)
330 def sizeHint(self, option, index):
331 sizeHint(self, option, index)
333 class CommentNodeDelegate(QAbstractItemDelegate):
334 def paint(self, painter, option, index):
335 #print "TEXT DELEG CALLED"
336 paint(self, painter, option, index)
338 def paint(self, painter, option, index):
339 text = index.model().data(index).property('content').toString()
340 palette = QApplication.palette()
341 document = QTextDocument()
342 document.setDefaultFont(option.font)
343 if option.state & QStyle.State_Selected:
344 rx = QRegExp(QString('<font .*>'))
346 # If selected, I remove the <font color="..."> by hand,
347 # and give the highlight color
348 document.setHtml(QString("<font color=%1>%2</font>") \
349 .arg(palette.highlightedText().color().name())\
350 .arg(text.replace(rx, QString('')).
351 replace(QString('</font>'),QString(''))))
353 document.setHtml(text)
354 color = palette.highlight().color() \
355 if option.state & QStyle.State_Selected \
356 else palette.base().color()
358 # voodoo: if not highlighted, filling the rect
359 # with the base color makes no difference
360 painter.fillRect(option.rect, color)
361 painter.translate(option.rect.x(), option.rect.y())
362 document.drawContents(painter)