annotate python/c3/visitor.py @ 254:bd26dc13f270

Added logger
author Windel Bouwman
date Wed, 31 Jul 2013 21:20:58 +0200
parents 88a1e0baef65
children e64bae57cda8
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
8104fc8b5e90 Added visitor to c3
Windel Bouwman
parents:
diff changeset
3 class Visitor:
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
4 """
215
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
5 Visitor that can visit all nodes in the AST
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
6 and run pre and post functions.
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
7 """
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
8 def visit(self, node, f_pre=None, f_post=None):
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
9 self.f_pre = f_pre
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
10 self.f_post = f_post
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
11 self.do(node)
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
12
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
13 def do(self, node):
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
14 # Run visitor:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
15 if self.f_pre:
215
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
16 self.f_pre(node)
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
17
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
18 # Descent into subnodes:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
19 if type(node) is Package:
215
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
20 for decl in node.declarations:
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
21 self.do(decl)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
22 elif type(node) is Function:
215
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
23 for s in node.declarations:
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
24 self.do(s)
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
25 self.do(node.body)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
26 elif type(node) is CompoundStatement:
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
27 for s in node.statements:
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
28 self.do(s)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
29 elif type(node) is IfStatement:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
30 self.do(node.condition)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
31 self.do(node.truestatement)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
32 if node.falsestatement:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
33 self.do(node.falsestatement)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
34 elif type(node) is FunctionCall:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
35 for arg in node.args:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
36 self.do(arg)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
37 elif type(node) is Assignment:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
38 self.do(node.lval)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
39 self.do(node.rval)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
40 elif type(node) is ReturnStatement:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
41 self.do(node.expr)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
42 elif type(node) is Binop:
220
3f6c30a5d234 Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents: 215
diff changeset
43 self.do(node.a)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
44 self.do(node.b)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
45 elif type(node) is Unop:
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
46 self.do(node.a)
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
47 elif type(node) is ExpressionStatement:
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 220
diff changeset
48 self.do(node.ex)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
49 elif type(node) is TypeCast:
222
c3f1ce8b638f Fixup of parser
Windel Bouwman
parents: 220
diff changeset
50 self.do(node.a)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
51 elif type(node) is FieldRef:
225
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
52 self.do(node.base)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
53 elif type(node) is Deref:
225
1c7364bd74c7 Fixed pointer deref
Windel Bouwman
parents: 222
diff changeset
54 self.do(node.ptr)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
55 elif type(node) is Constant:
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
56 self.do(node.value)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
57 elif type(node) in [VariableUse, Variable, Literal, FunctionType, DefinedType]:
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
58 # Those nodes do not have child nodes.
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
59 pass
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
60 elif type(node) is WhileStatement:
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
61 self.do(node.condition)
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
62 self.do(node.statement)
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
63 else:
230
88a1e0baef65 Added some tests for IR-code
Windel Bouwman
parents: 228
diff changeset
64 raise Exception('Could not visit "{0}"'.format(node))
163
8104fc8b5e90 Added visitor to c3
Windel Bouwman
parents:
diff changeset
65
228
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
66 # run post function
7f18ed9b6b7e Removal of emptystatement class
Windel Bouwman
parents: 225
diff changeset
67 if self.f_post:
215
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
68 self.f_post(node)
c1ccb1cb4cef Major changes in c3 frontend
Windel Bouwman
parents: 186
diff changeset
69