1
|
1 from PyQt4.QtCore import *
|
|
2 from PyQt4.QtGui import *
|
168
|
3 from c3 import Visitor, astnodes
|
1
|
4
|
165
|
5 class AstModelBuilder:
|
|
6 def __init__(self):
|
168
|
7 self.functionIco = QIcon(QPixmap('functionicon.png').scaled(32, 32))
|
|
8 self.variableIco = QIcon(QPixmap('variableicon.png').scaled(32, 32))
|
165
|
9 self.visitor = Visitor(self.p1, self.p2)
|
168
|
10 self.model = QStandardItemModel()
|
165
|
11 def build(self, pkg):
|
168
|
12 self.model.clear()
|
|
13 self.model.setHorizontalHeaderLabels(['Object', 'Type'])
|
|
14 self.curItem = self.model.invisibleRootItem()
|
165
|
15 self.visitor.visit(pkg)
|
|
16 def p1(self, node):
|
168
|
17 if type(node) is astnodes.Variable:
|
|
18 i = QStandardItem(self.variableIco, str(node))
|
|
19 elif type(node) is astnodes.Function:
|
|
20 i = QStandardItem(self.functionIco, str(node))
|
|
21 elif type(node) is astnodes.Package:
|
|
22 i = QStandardItem(str(node))
|
|
23 else:
|
|
24 return
|
165
|
25 typ = str(node.typ) if hasattr(node, 'typ') else ''
|
|
26 ti = QStandardItem(str(typ))
|
|
27 ti.setData(node)
|
|
28 i.setData(node)
|
|
29 self.curItem.appendRow([i, ti])
|
|
30 self.curItem = i
|
|
31 def p2(self, node):
|
168
|
32 if type(node) in [astnodes.Variable, astnodes.Function, astnodes.Package]:
|
|
33 self.curItem = self.curItem.parent()
|
1
|
34
|
|
35 # The actual widget:
|
|
36 class AstViewer(QTreeView):
|
4
|
37 sigNodeSelected = pyqtSignal(object)
|
1
|
38 def __init__(self, parent=None):
|
|
39 super(AstViewer, self).__init__(parent)
|
4
|
40 self.clicked.connect(self.selectHandler)
|
165
|
41 self.modelBuilder = AstModelBuilder()
|
168
|
42 self.setModel(self.modelBuilder.model)
|
1
|
43
|
|
44 def setAst(self, ast):
|
|
45 """ Create a new model and add all ast elements to it """
|
168
|
46 self.modelBuilder.build(ast)
|
|
47 self.expandAll()
|
1
|
48
|
4
|
49 def selectHandler(self, index):
|
1
|
50 if not index.isValid():
|
|
51 return
|
4
|
52 model = self.model()
|
|
53 item = model.itemFromIndex(index)
|
|
54 node = item.data()
|
|
55 self.sigNodeSelected.emit(node)
|
1
|
56
|
|
57
|