Mercurial > lcfOS
annotate python/ppci/c3/visitor.py @ 306:b145f8e6050b
Start on c3 rewrite
author | Windel Bouwman |
---|---|
date | Mon, 09 Dec 2013 19:00:21 +0100 |
parents | 158068af716c |
children | e609d5296ee9 |
rev | line source |
---|---|
163 | 1 from .astnodes import * |
2 | |
288 | 3 |
163 | 4 class Visitor: |
228 | 5 """ |
215 | 6 Visitor that can visit all nodes in the AST |
7 and run pre and post functions. | |
8 """ | |
9 def visit(self, node, f_pre=None, f_post=None): | |
10 self.f_pre = f_pre | |
11 self.f_post = f_post | |
12 self.do(node) | |
13 | |
14 def do(self, node): | |
306 | 15 # Run pre function: |
228 | 16 if self.f_pre: |
215 | 17 self.f_pre(node) |
18 | |
228 | 19 # Descent into subnodes: |
20 if type(node) is Package: | |
215 | 21 for decl in node.declarations: |
22 self.do(decl) | |
228 | 23 elif type(node) is Function: |
215 | 24 for s in node.declarations: |
25 self.do(s) | |
306 | 26 self.do(node.typ) |
215 | 27 self.do(node.body) |
228 | 28 elif type(node) is CompoundStatement: |
230 | 29 for s in node.statements: |
30 self.do(s) | |
228 | 31 elif type(node) is IfStatement: |
32 self.do(node.condition) | |
33 self.do(node.truestatement) | |
306 | 34 self.do(node.falsestatement) |
228 | 35 elif type(node) is FunctionCall: |
36 for arg in node.args: | |
37 self.do(arg) | |
306 | 38 self.do(node.proc) |
228 | 39 elif type(node) is Assignment: |
40 self.do(node.lval) | |
41 self.do(node.rval) | |
42 elif type(node) is ReturnStatement: | |
43 self.do(node.expr) | |
44 elif type(node) is Binop: | |
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
45 self.do(node.a) |
228 | 46 self.do(node.b) |
47 elif type(node) is Unop: | |
48 self.do(node.a) | |
49 elif type(node) is ExpressionStatement: | |
222 | 50 self.do(node.ex) |
228 | 51 elif type(node) is TypeCast: |
222 | 52 self.do(node.a) |
306 | 53 elif type(node) is Member: |
225 | 54 self.do(node.base) |
228 | 55 elif type(node) is Deref: |
225 | 56 self.do(node.ptr) |
228 | 57 elif type(node) is Constant: |
230 | 58 self.do(node.value) |
306 | 59 elif type(node) is DefinedType: |
60 self.do(node.typ) | |
61 elif isinstance(node, Variable): | |
62 self.do(node.typ) | |
63 elif type(node) is PointerType: | |
64 self.do(node.ptype) | |
65 elif type(node) is StructureType: | |
66 for m in node.mems: | |
67 self.do(m.typ) | |
68 elif type(node) is FunctionType: | |
69 for pt in node.parametertypes: | |
70 self.do(pt) | |
71 self.do(node.returntype) | |
72 elif type(node) in [Identifier, Literal, EmptyStatement]: | |
230 | 73 # Those nodes do not have child nodes. |
74 pass | |
228 | 75 elif type(node) is WhileStatement: |
230 | 76 self.do(node.condition) |
77 self.do(node.statement) | |
228 | 78 else: |
230 | 79 raise Exception('Could not visit "{0}"'.format(node)) |
163 | 80 |
228 | 81 # run post function |
82 if self.f_post: | |
215 | 83 self.f_post(node) |
84 | |
288 | 85 |
86 class AstPrinter: | |
87 """ Prints an AST as text """ | |
88 def printAst(self, pkg): | |
89 self.indent = 0 | |
90 visitor = Visitor() | |
91 visitor.visit(pkg, self.print1, self.print2) | |
92 | |
93 def print1(self, node): | |
94 print(' ' * self.indent + str(node)) | |
95 self.indent += 2 | |
96 | |
97 def print2(self, node): | |
98 self.indent -= 2 |