annotate python/ppci/c3/visitor.py @ 315:084cccaa5deb

Added console and screen
author Windel Bouwman
date Sat, 21 Dec 2013 10:03:01 +0100
parents 04cf4d26a3bc
children 5477e499b039
rev   line source
163
8104fc8b5e90 Added visitor to c3
Windel Bouwman
parents:
diff changeset
1 from .astnodes import *
8104fc8b5e90 Added visitor to c3
Windel Bouwman
parents:
diff changeset
2
288
a747a45dcd78 Various styling work
Windel Bouwman
parents: 272
diff changeset
3
163
8104fc8b5e90 Added visitor to c3
Windel Bouwman
parents:
diff changeset
4 class Visitor:
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
5 """
215
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
6 Visitor that can visit all nodes in the AST
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
7 and run pre and post functions.
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
8 """
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
9 def visit(self, node, f_pre=None, f_post=None):
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
10 self.f_pre = f_pre
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
11 self.f_post = f_post
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
12 self.do(node)
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
13
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
14 def do(self, node):
306
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
15 # Run pre function:
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
16 if self.f_pre:
215
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
17 self.f_pre(node)
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
18
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
19 # Descent into subnodes:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
20 if type(node) is Package:
215
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
21 for decl in node.declarations:
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
22 self.do(decl)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
23 elif type(node) is Function:
215
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
24 for s in node.declarations:
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
25 self.do(s)
306
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
26 self.do(node.typ)
215
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
27 self.do(node.body)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
28 elif type(node) is Compound:
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
29 for s in node.statements:
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
30 self.do(s)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
31 elif type(node) is If:
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
32 self.do(node.condition)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
33 self.do(node.truestatement)
306
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
34 self.do(node.falsestatement)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
35 elif type(node) is While:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
36 self.do(node.condition)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
37 self.do(node.statement)
315
084cccaa5deb Added console and screen
Windel Bouwman
parents: 313
diff changeset
38 elif type(node) is For:
084cccaa5deb Added console and screen
Windel Bouwman
parents: 313
diff changeset
39 self.do(node.init)
084cccaa5deb Added console and screen
Windel Bouwman
parents: 313
diff changeset
40 self.do(node.condition)
084cccaa5deb Added console and screen
Windel Bouwman
parents: 313
diff changeset
41 self.do(node.final)
084cccaa5deb Added console and screen
Windel Bouwman
parents: 313
diff changeset
42 self.do(node.statement)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
43 elif type(node) is Assignment:
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
44 self.do(node.lval)
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
45 self.do(node.rval)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
46 elif type(node) is FunctionCall:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
47 for arg in node.args:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
48 self.do(arg)
306
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
49 self.do(node.proc)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
50 elif type(node) is Return:
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
51 self.do(node.expr)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
52 elif type(node) is Binop:
220
3f6c30a5d234 Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents: 215
diff changeset
53 self.do(node.a)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
54 self.do(node.b)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
55 elif type(node) is Unop:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
56 self.do(node.a)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
57 elif type(node) is ExpressionStatement:
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 220
diff changeset
58 self.do(node.ex)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
59 elif type(node) is TypeCast:
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 220
diff changeset
60 self.do(node.a)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
61 self.do(node.to_type)
306
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
62 elif type(node) is Member:
225
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
63 self.do(node.base)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
64 elif type(node) is Deref:
225
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
65 self.do(node.ptr)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
66 elif type(node) is Constant:
313
04cf4d26a3bc Added constant function
Windel Bouwman
parents: 307
diff changeset
67 self.do(node.typ)
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
68 self.do(node.value)
306
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
69 elif type(node) is DefinedType:
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
70 self.do(node.typ)
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
71 elif isinstance(node, Variable):
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
72 self.do(node.typ)
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
73 elif type(node) is PointerType:
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
74 self.do(node.ptype)
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
75 elif type(node) is StructureType:
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
76 for m in node.mems:
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
77 self.do(m.typ)
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
78 elif type(node) is FunctionType:
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
79 for pt in node.parametertypes:
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
80 self.do(pt)
b145f8e6050b Start on c3 rewrite
Windel Bouwman
parents: 300
diff changeset
81 self.do(node.returntype)
307
e609d5296ee9 Massive rewrite of codegenerator
Windel Bouwman
parents: 306
diff changeset
82 elif type(node) in [Identifier, Literal, Empty]:
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
83 # Those nodes do not have child nodes.
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
84 pass
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
85 else:
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
86 raise Exception('Could not visit "{0}"'.format(node))
163
8104fc8b5e90 Added visitor to c3
Windel Bouwman
parents:
diff changeset
87
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
88 # run post function
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
89 if self.f_post:
215
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
90 self.f_post(node)
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
91
288
a747a45dcd78 Various styling work
Windel Bouwman
parents: 272
diff changeset
92
a747a45dcd78 Various styling work
Windel Bouwman
parents: 272
diff changeset
93 class AstPrinter:
a747a45dcd78 Various styling work
Windel Bouwman
parents: 272
diff changeset
94 """ Prints an AST as text """
313
04cf4d26a3bc Added constant function
Windel Bouwman
parents: 307
diff changeset
95 def printAst(self, pkg, f):
04cf4d26a3bc Added constant function
Windel Bouwman
parents: 307
diff changeset
96 self.indent = 2
04cf4d26a3bc Added constant function
Windel Bouwman
parents: 307
diff changeset
97 self.f = f
288
a747a45dcd78 Various styling work
Windel Bouwman
parents: 272
diff changeset
98 visitor = Visitor()
a747a45dcd78 Various styling work
Windel Bouwman
parents: 272
diff changeset
99 visitor.visit(pkg, self.print1, self.print2)
a747a45dcd78 Various styling work
Windel Bouwman
parents: 272
diff changeset
100
a747a45dcd78 Various styling work
Windel Bouwman
parents: 272
diff changeset
101 def print1(self, node):
313
04cf4d26a3bc Added constant function
Windel Bouwman
parents: 307
diff changeset
102 print(' ' * self.indent + str(node), file=self.f)
288
a747a45dcd78 Various styling work
Windel Bouwman
parents: 272
diff changeset
103 self.indent += 2
a747a45dcd78 Various styling work
Windel Bouwman
parents: 272
diff changeset
104
a747a45dcd78 Various styling work
Windel Bouwman
parents: 272
diff changeset
105 def print2(self, node):
a747a45dcd78 Various styling work
Windel Bouwman
parents: 272
diff changeset
106 self.indent -= 2