Mercurial > lcfOS
annotate python/ppci/c3/visitor.py @ 393:6ae782a085e0
Added init program
| author | Windel Bouwman |
|---|---|
| date | Sat, 17 May 2014 21:17:40 +0200 |
| parents | c05ab629976a |
| children |
| 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) |
| 362 | 27 if node.body: |
| 28 self.do(node.body) | |
| 307 | 29 elif type(node) is Compound: |
| 230 | 30 for s in node.statements: |
| 31 self.do(s) | |
| 307 | 32 elif type(node) is If: |
| 228 | 33 self.do(node.condition) |
| 34 self.do(node.truestatement) | |
| 306 | 35 self.do(node.falsestatement) |
| 307 | 36 elif type(node) is While: |
| 37 self.do(node.condition) | |
| 38 self.do(node.statement) | |
| 315 | 39 elif type(node) is For: |
| 40 self.do(node.init) | |
| 41 self.do(node.condition) | |
| 42 self.do(node.final) | |
| 43 self.do(node.statement) | |
| 307 | 44 elif type(node) is Assignment: |
| 45 self.do(node.lval) | |
| 46 self.do(node.rval) | |
| 228 | 47 elif type(node) is FunctionCall: |
| 48 for arg in node.args: | |
| 49 self.do(arg) | |
| 306 | 50 self.do(node.proc) |
| 307 | 51 elif type(node) is Return: |
| 228 | 52 self.do(node.expr) |
| 53 elif type(node) is Binop: | |
|
220
3f6c30a5d234
Major change in expression parsing to enable pointers and structs
Windel Bouwman
parents:
215
diff
changeset
|
54 self.do(node.a) |
| 228 | 55 self.do(node.b) |
| 56 elif type(node) is Unop: | |
| 57 self.do(node.a) | |
| 58 elif type(node) is ExpressionStatement: | |
| 222 | 59 self.do(node.ex) |
| 228 | 60 elif type(node) is TypeCast: |
| 222 | 61 self.do(node.a) |
| 307 | 62 self.do(node.to_type) |
| 393 | 63 elif type(node) is Sizeof: |
| 64 self.do(node.query_typ) | |
| 306 | 65 elif type(node) is Member: |
| 225 | 66 self.do(node.base) |
| 354 | 67 elif type(node) is Index: |
| 68 self.do(node.base) | |
| 69 self.do(node.i) | |
| 228 | 70 elif type(node) is Deref: |
| 225 | 71 self.do(node.ptr) |
| 228 | 72 elif type(node) is Constant: |
| 313 | 73 self.do(node.typ) |
| 230 | 74 self.do(node.value) |
| 306 | 75 elif type(node) is DefinedType: |
| 76 self.do(node.typ) | |
| 77 elif isinstance(node, Variable): | |
| 78 self.do(node.typ) | |
| 79 elif type(node) is PointerType: | |
| 80 self.do(node.ptype) | |
| 81 elif type(node) is StructureType: | |
| 82 for m in node.mems: | |
| 83 self.do(m.typ) | |
| 354 | 84 elif type(node) is ArrayType: |
| 85 self.do(node.element_type) | |
| 86 self.do(node.size) | |
| 306 | 87 elif type(node) is FunctionType: |
| 88 for pt in node.parametertypes: | |
| 89 self.do(pt) | |
| 90 self.do(node.returntype) | |
| 307 | 91 elif type(node) in [Identifier, Literal, Empty]: |
| 230 | 92 # Those nodes do not have child nodes. |
| 93 pass | |
| 228 | 94 else: |
| 230 | 95 raise Exception('Could not visit "{0}"'.format(node)) |
| 163 | 96 |
| 228 | 97 # run post function |
| 98 if self.f_post: | |
| 215 | 99 self.f_post(node) |
| 100 | |
| 288 | 101 |
| 102 class AstPrinter: | |
| 103 """ Prints an AST as text """ | |
| 313 | 104 def printAst(self, pkg, f): |
| 105 self.indent = 2 | |
| 106 self.f = f | |
| 288 | 107 visitor = Visitor() |
| 108 visitor.visit(pkg, self.print1, self.print2) | |
| 109 | |
| 110 def print1(self, node): | |
| 313 | 111 print(' ' * self.indent + str(node), file=self.f) |
| 288 | 112 self.indent += 2 |
| 113 | |
| 114 def print2(self, node): | |
| 115 self.indent -= 2 |