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 # sometimes it return a QString, sometimes a QVariant. not good.
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 obj.setProperty('nodeType', QString('element'))
45 obj.setProperty('content', ElemNameAndAtts)
47 elif node.nodeType() == QDomNode.AttributeNode:
49 elif node.nodeType() == QDomNode.TextNode:
51 obj.setProperty('nodeType', QString('text'))
52 obj.setProperty('content', node.nodeValue())
54 elif node.nodeType() == QDomNode.CDATASectionNode:
55 return QString('unsupported node type')
56 elif node.nodeType() == QDomNode.EntityReferenceNode:
57 return QString('unsupported node type')
58 elif node.nodeType() == QDomNode.EntityNode:
59 return QString('unsupported node type')
60 elif node.nodeType() == QDomNode.ProcessingInstructionNode:
62 elif node.nodeType() == QDomNode.CommentNode:
64 obj.setProperty('nodeType', QString('comment'))
65 obj.setProperty('content', node.nodeValue())
67 elif node.nodeType() == QDomNode.DocumentNode:
68 return QString('unsupported node type')
69 elif node.nodeType() == QDomNode.DocumentTypeNode:
70 return QString('unsupported node type')
71 elif node.nodeType() == QDomNode.DocumentFragmentNode:
72 return QString('unsupported node type')
73 elif node.nodeType() == QDomNode.NotationNode:
74 return QString('unsupported node type')
75 elif node.nodeType() == QDomNode.BaseNode:
76 return QString('unsupported node type')
77 elif node.nodeType() == QDomNode.CharacterDataNode:
78 return QString('unsupported node type')
84 def flags(self, index):
85 if not index.isValid():
86 return Qt.ItemIsEnabled
87 return Qt.ItemIsEnabled | Qt.ItemIsSelectable
89 def headerData(self, section, orientation, role):
92 def index(self, row, column, parent=None):
93 if not parent or not parent.isValid():
94 parentItem = self.rootItem
96 parentItem = parent.internalPointer()
98 childItem = parentItem.child(row)
99 # childItem would be None to say "false"?
101 return self.createIndex(row, column, childItem)
105 def parent(self, child):
106 if not child.isValid():
108 childItem = child.internalPointer()
109 parentItem = childItem.parent()
111 if not parentItem or parentItem == self.rootItem:
113 return self.createIndex(parentItem.row(), 0, parentItem)
115 def rowCount(self, parent=None):
116 if not parent or not parent.isValid():
117 parentItem = self.rootItem
119 parentItem = parent.internalPointer()
121 return parentItem.node().childNodes().count()
123 def columnCount(self, parent):
124 # just one column we'll print tag name (and attributes) or the
130 # wrapper around PyQt4.QtXml.QDomNode it keeps an hash of
131 # childrens for performance reasons
133 def __init__(self, node, row, parent = 0):
134 # node is of type PyQt4.QtXml.QDomNode
136 self.parentItem = parent
141 if i in self.childItems:
142 return self.childItems[i]
143 if i >= 0 and i < self.domNode.childNodes().count():
144 childNode = self.domNode.childNodes().item(i)
145 childItem = DomItem(childNode, i, self)
146 self.childItems[i] = childItem
151 return self.parentItem
157 return self.rowNumber
159 class XmlView(QTreeView):
160 def __init__(self, parent):
161 QTreeView.__init__(self, parent)
163 self.setAnimated(True)
164 self.setItemsExpandable(True)
165 self.setRootIsDecorated(True)
166 self.setHeaderHidden(True)
167 self.setAttribute(Qt.WA_MacShowFocusRect, 0)
168 self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
170 class XmlWindow(QDialog):
171 def __init__(self, parent=None, title='XML Window'):
172 QDialog.__init__(self, parent)
173 self.setWindowTitle(title)
179 self.view = XmlView(self)
180 self.delegate = XmlDelegate(self)
181 self.view.setItemDelegate(self.delegate)
182 self.delegate.insertNodeDelegate('element', ElemNodeDelegate())
183 self.delegate.insertNodeDelegate('text', TextNodeDelegate())
184 self.delegate.insertNodeDelegate('comment', CommentNodeDelegate())
185 layout = QVBoxLayout()
186 layout.addWidget(self.view)
187 self.setLayout(layout)
195 def updateView(self):
201 self.document = QDomDocument(self.title)
202 self.model = DomModel(self.document, self)
204 self.view.setModel(self.model)
205 self.view.expand(self.model.index(0, 0)) #expand first level only
207 #move the code below to rspec window
208 rspec_file = config.getSliceRSpecFile()
209 if not os.path.exists(rspec_file):
212 self.document.setContent(open(rspec_file,'r').read())
216 class XmlDelegate(QItemDelegate):
218 def __init__(self, parent=None):
219 QAbstractItemDelegate.__init__(self, parent)
222 def insertNodeDelegate(self, nodeType, delegate):
223 delegate.setParent(self)
224 self.delegates[nodeType] = delegate
226 def removeNodeDelegate(self, nodeType, delegate):
227 if nodeType in self.delegates:
228 del self.delegates[nodeType]
230 def paint(self, painter, option, index):
231 nodeType = index.model().data(index).property('nodeType')
232 delegate = self.delegates.get(str(nodeType.toString()))
233 #print "TYPE:", str(type(str(nodeType.toString())))
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 text = index.model().data(index).property('content').toString()
248 document = QTextDocument()
249 document.setDefaultFont(option.font)
250 document.setHtml(text)
251 # the +5 is for margin. The +4 is voodoo;
252 # fm.height just give it too small.
253 return QSize(document.idealWidth() + 5, fm.height() + 4)
255 class ElemNodeDelegate(QAbstractItemDelegate):
256 def paint(self, painter, option, index):
257 text = index.model().data(index)
258 palette = QApplication.palette()
259 document = QTextDocument()
260 document.setDefaultFont(option.font)
261 nonHighGlobPattern = '<<b><font color="#b42be2">%s</font></b>%s>'
262 nonHighAttPattern = ' <b>%s</b>="<font color="#1e90ff">%s</font>"'
263 highGlobPattern = '<<b>%s</b>%s>'
264 highAttPattern = ' <b>%s</b>="%s"'
265 def getHtmlText(plainText, globPattern, attPattern):
266 print "PLAIN TEXT:", plainText
267 tmp = plainText.split(' ', 1)
272 # many elems don't have atts...
273 attList = tmp[1].split()
277 attValue = tmp[1][1:-1]
278 AttListHtml += (nonHighAttPattern % (attName, attValue))
279 html = (globPattern % (elemName, AttListHtml))
281 def colorize(color, text):
282 return '<font color=' + color + '>' + text + '</font>'
283 text = str(index.model().data(index).property('content').toString())
285 if option.state & QStyle.State_Selected:
286 htmlText = colorize(palette.highlightedText().color().name(),
287 getHtmlText(text, highGlobPattern, highAttPattern))
288 document.setHtml(QString(htmlText))
290 htmlText = getHtmlText(text, nonHighGlobPattern, nonHighAttPattern)
291 document.setHtml(QString(htmlText))
292 color = palette.highlight().color() \
293 if option.state & QStyle.State_Selected \
294 else palette.base().color()
296 print "COLOR:", color.name()
297 # voodoo: if not highlighted, filling the rect
298 # with the base color makes no difference
299 painter.fillRect(option.rect, color)
300 painter.translate(option.rect.x(), option.rect.y())
301 document.drawContents(painter)
304 def sizeHint(self, option, index):
305 sizeHint(self, option, index)
307 class TextNodeDelegate(QAbstractItemDelegate):
308 def paint(self, painter, option, index):
309 #print "TEXT DELEG CALLED"
310 paint(self, painter, option, index)
312 def sizeHint(self, option, index):
313 sizeHint(self, option, index)
315 class CommentNodeDelegate(QAbstractItemDelegate):
316 def paint(self, painter, option, index):
317 #print "TEXT DELEG CALLED"
318 paint(self, painter, option, index)
320 def paint(self, painter, option, index):
321 text = index.model().data(index).property('content').toString()
322 palette = QApplication.palette()
323 document = QTextDocument()
324 document.setDefaultFont(option.font)
325 if option.state & QStyle.State_Selected:
326 rx = QRegExp(QString('<font .*>'))
328 # If selected, I remove the <font color="..."> by hand,
329 # and give the highlight color
330 document.setHtml(QString("<font color=%1>%2</font>") \
331 .arg(palette.highlightedText().color().name())\
332 .arg(text.replace(rx, QString('')).
333 replace(QString('</font>'),QString(''))))
335 document.setHtml(text)
336 color = palette.highlight().color() \
337 if option.state & QStyle.State_Selected \
338 else palette.base().color()
340 # voodoo: if not highlighted, filling the rect
341 # with the base color makes no difference
342 painter.fillRect(option.rect, color)
343 painter.translate(option.rect.x(), option.rect.y())
344 document.drawContents(painter)