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 self.rootItem = DomItem(document, 0);
18 def data(self, index, role = Qt.DisplayRole):
19 # sometimes it return a QString, sometimes a QVariant. not good.
20 if not index.isValid():
22 if role != Qt.DisplayRole:
24 node = index.internalPointer().node()
25 attributeMap = node.attributes()
29 if node.nodeType() == QDomNode.ElementNode:
30 qslist = QStringList()
31 for i in range(attributeMap.count()):
32 attr = attributeMap.item(i)
33 elem = ' %s="%s"' % (attr.nodeName(), attr.nodeValue())
35 ElemNameAndAtts = '%s%s'% (node.nodeName(), qslist.join(' '))
37 obj.setProperty('nodeType', QString('element'))
38 obj.setProperty('content', ElemNameAndAtts)
40 elif node.nodeType() == QDomNode.AttributeNode:
42 elif node.nodeType() == QDomNode.TextNode:
44 obj.setProperty('nodeType', QString('text'))
45 obj.setProperty('content', node.nodeValue())
47 elif node.nodeType() == QDomNode.CDATASectionNode:
48 return QString('unsupported node type')
49 elif node.nodeType() == QDomNode.EntityReferenceNode:
50 return QString('unsupported node type')
51 elif node.nodeType() == QDomNode.EntityNode:
52 return QString('unsupported node type')
53 elif node.nodeType() == QDomNode.ProcessingInstructionNode:
55 obj.setProperty('nodeType', QString('element'))
56 obj.setProperty('content', node.nodeName() + " " + node.nodeValue())
58 elif node.nodeType() == QDomNode.CommentNode:
60 obj.setProperty('nodeType', QString('comment'))
61 obj.setProperty('content', node.nodeValue())
63 elif node.nodeType() == QDomNode.DocumentNode:
64 return QString('unsupported node type')
65 elif node.nodeType() == QDomNode.DocumentTypeNode:
66 return QString('unsupported node type')
67 elif node.nodeType() == QDomNode.DocumentFragmentNode:
68 return QString('unsupported node type')
69 elif node.nodeType() == QDomNode.NotationNode:
70 return QString('unsupported node type')
71 elif node.nodeType() == QDomNode.BaseNode:
72 return QString('unsupported node type')
73 elif node.nodeType() == QDomNode.CharacterDataNode:
74 return QString('unsupported node type')
80 def flags(self, index):
81 if not index.isValid():
82 return Qt.ItemIsEnabled
83 return Qt.ItemIsEnabled | Qt.ItemIsSelectable
85 def headerData(self, section, orientation, role):
88 def index(self, row, column, parent=None):
89 if not parent or not parent.isValid():
90 parentItem = self.rootItem
92 parentItem = parent.internalPointer()
94 childItem = parentItem.child(row)
95 # childItem would be None to say "false"?
97 return self.createIndex(row, column, childItem)
101 def parent(self, child):
102 if not child.isValid():
104 childItem = child.internalPointer()
105 parentItem = childItem.parent()
107 if not parentItem or parentItem == self.rootItem:
109 return self.createIndex(parentItem.row(), 0, parentItem)
111 def rowCount(self, parent=None):
112 if not parent or not parent.isValid():
113 parentItem = self.rootItem
115 parentItem = parent.internalPointer()
117 return parentItem.node().childNodes().count()
119 def columnCount(self, parent):
120 # just one column we'll print tag name (and attributes) or the
126 # wrapper around PyQt4.QtXml.QDomNode it keeps an hash of
127 # childrens for performance reasons
129 def __init__(self, node, row, parent = 0):
130 # node is of type PyQt4.QtXml.QDomNode
132 self.parentItem = parent
137 if i in self.childItems:
138 return self.childItems[i]
139 if i >= 0 and i < self.domNode.childNodes().count():
140 childNode = self.domNode.childNodes().item(i)
141 childItem = DomItem(childNode, i, self)
142 self.childItems[i] = childItem
147 return self.parentItem
153 return self.rowNumber
155 class XmlView(QTreeView):
156 def __init__(self, parent):
157 QTreeView.__init__(self, parent)
159 self.setAnimated(True)
160 self.setItemsExpandable(True)
161 self.setRootIsDecorated(True)
162 self.setHeaderHidden(True)
163 self.setAttribute(Qt.WA_MacShowFocusRect, 0)
164 self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
166 class XmlWindow(QDialog):
167 def __init__(self, parent=None, title='XML Window'):
168 QDialog.__init__(self, parent)
169 self.setWindowTitle(title)
175 self.view = self.initView()
176 self.delegate = XmlDelegate(self)
177 self.view.setItemDelegate(self.delegate)
178 self.delegate.insertNodeDelegate('element', ElemNodeDelegate())
179 self.delegate.insertNodeDelegate('text', TextNodeDelegate())
180 self.delegate.insertNodeDelegate('comment', CommentNodeDelegate())
181 layout = QVBoxLayout()
182 layout.addWidget(self.view)
183 self.setLayout(layout)
194 def updateView(self):
200 self.document = QDomDocument(self.title)
201 self.model = DomModel(self.document, self)
203 self.view.setModel(self.model)
205 #move the code below to rspec window
206 rspec_file = config.getSliceRSpecFile()
207 if not os.path.exists(rspec_file):
210 self.document.setContent(open(rspec_file,'r').read())
212 if self.document.childNodes().count() == 0:
213 # empty document - do nothing
215 elif self.document.childNodes().item(0).nodeType() == QDomNode.ProcessingInstructionNode:
216 # the first item is the <xml> tag, so expand the second
217 self.view.expand(self.model.index(1,0))
219 # the document didn't start with an <xml> tag; let's try to expand
220 # the first item on the assumption that it is our root level xml
222 self.view.expand(self.model.index(0,0))
226 class XmlDelegate(QItemDelegate):
228 def __init__(self, parent=None):
229 QAbstractItemDelegate.__init__(self, parent)
232 def insertNodeDelegate(self, nodeType, delegate):
233 delegate.setParent(self)
234 self.delegates[nodeType] = delegate
236 def removeNodeDelegate(self, nodeType, delegate):
237 if nodeType in self.delegates:
238 del self.delegates[nodeType]
240 def paint(self, painter, option, index):
241 if isinstance(index.model().data(index),QVariant):
243 nodeType = index.model().data(index).property('nodeType')
244 delegate = self.delegates.get(str(nodeType.toString()))
245 #print "TYPE:", str(type(str(nodeType.toString())))
246 #print "DELEGS DICT:", self.delegates
247 #print "NODETYPE:", nodeType.toString()
248 if delegate is not None:
249 #print "WOW DELEG ISNT NONE"
250 delegate.paint(painter, option, index)
253 # not sure this will ever work. this delegate
254 # doesn't know about my QObject strategy.
255 QItemDelegate.paint(self, painter, option, index)
257 def sizeHint(self, option, index):
258 fm = option.fontMetrics
259 if isinstance(index.model().data(index),QVariant):
261 text = index.model().data(index).property('content').toString()
262 document = QTextDocument()
263 document.setDefaultFont(option.font)
264 document.setHtml(text)
265 # the +5 is for margin. The +4 is voodoo;
266 # fm.height just give it too small.
267 return QSize(document.idealWidth() + 5, fm.height() + 4)
269 class ElemNodeDelegate(QAbstractItemDelegate):
270 def paint(self, painter, option, index):
271 text = index.model().data(index)
272 palette = QApplication.palette()
273 document = QTextDocument()
274 document.setDefaultFont(option.font)
275 nonHighGlobPattern = '<<b><font color="#b42be2">%s</font></b>%s>'
276 nonHighAttPattern = ' <b>%s</b>="<font color="#1e90ff">%s</font>"'
277 highGlobPattern = '<<b>%s</b>%s>'
278 highAttPattern = ' <b>%s</b>="%s"'
279 def getHtmlText(plainText, globPattern, attPattern):
280 # print "PLAIN TEXT:", plainText
281 tmp = plainText.split(' ', 1)
286 # many elems don't have atts...
287 # use shlex.split so we can handle quoted strings with spaces
288 # in them, like <link enpoints="foo bar">. Note that there are
289 # documented problems with shlex.split and unicode, so we
290 # convert any potential unicode to a string first.
291 attList = shlex.split(str(tmp[1]))
293 tmp = att.split('=',1)
298 # this shouldn't happen, but if it does, pretend the
299 # attribute value is blank.
302 AttListHtml += (nonHighAttPattern % (attName, attValue))
303 html = (globPattern % (elemName, AttListHtml))
305 def colorize(color, text):
306 return '<font color=' + color + '>' + text + '</font>'
307 text = str(index.model().data(index).property('content').toString())
308 # print "TEXT:", text
309 if option.state & QStyle.State_Selected:
310 htmlText = colorize(palette.highlightedText().color().name(),
311 getHtmlText(text, highGlobPattern, highAttPattern))
312 document.setHtml(QString(htmlText))
314 htmlText = getHtmlText(text, nonHighGlobPattern, nonHighAttPattern)
315 document.setHtml(QString(htmlText))
316 color = palette.highlight().color() \
317 if option.state & QStyle.State_Selected \
318 else palette.base().color()
320 # print "COLOR:", color.name()
321 # voodoo: if not highlighted, filling the rect
322 # with the base color makes no difference
323 painter.fillRect(option.rect, color)
324 painter.translate(option.rect.x(), option.rect.y())
325 document.drawContents(painter)
328 def sizeHint(self, option, index):
329 sizeHint(self, option, index)
331 class TextNodeDelegate(QAbstractItemDelegate):
332 def paint(self, painter, option, index):
333 #print "TEXT DELEG CALLED"
334 paint(self, painter, option, index)
336 def sizeHint(self, option, index):
337 sizeHint(self, option, index)
339 class CommentNodeDelegate(QAbstractItemDelegate):
340 def paint(self, painter, option, index):
341 #print "TEXT DELEG CALLED"
342 paint(self, painter, option, index)
344 def paint(self, painter, option, index):
345 text = index.model().data(index).property('content').toString()
346 palette = QApplication.palette()
347 document = QTextDocument()
348 document.setDefaultFont(option.font)
349 if option.state & QStyle.State_Selected:
350 rx = QRegExp(QString('<font .*>'))
352 # If selected, I remove the <font color="..."> by hand,
353 # and give the highlight color
354 document.setHtml(QString("<font color=%1>%2</font>") \
355 .arg(palette.highlightedText().color().name())\
356 .arg(text.replace(rx, QString('')).
357 replace(QString('</font>'),QString(''))))
359 document.setHtml(text)
360 color = palette.highlight().color() \
361 if option.state & QStyle.State_Selected \
362 else palette.base().color()
364 # voodoo: if not highlighted, filling the rect
365 # with the base color makes no difference
366 painter.fillRect(option.rect, color)
367 painter.translate(option.rect.x(), option.rect.y())
368 document.drawContents(painter)