Mercurial > lcfOS
annotate python/c3/visitor.py @ 169:ee0d30533dae
Added more tests and improved the diagnostic update
| author | Windel Bouwman |
|---|---|
| date | Sat, 23 Mar 2013 18:34:41 +0100 |
| parents | 0b5b2ee6b435 |
| children | 5fd02aa38b42 |
| rev | line source |
|---|---|
| 163 | 1 from .astnodes import * |
| 2 | |
| 3 class Visitor: | |
| 4 """ Visitor that visits all nodes in the ast and runs the function 'f' """ | |
| 5 def __init__(self, f1, f2): | |
| 6 self.f1 = f1 | |
| 7 self.f2 = f2 | |
| 8 def visit(self, node): | |
| 9 # Run visitor: | |
| 10 self.f1(node) | |
| 11 # Descent into subnodes: | |
| 12 if type(node) is Package: | |
| 13 for s in node.scope: | |
| 14 self.visit(s) | |
| 15 elif type(node) is Function: | |
| 16 for s in node.scope: | |
| 17 self.visit(s) | |
| 18 self.visit(node.typ) | |
| 19 self.visit(node.body) | |
| 20 elif type(node) is CompoundStatement: | |
| 21 for s in node.statements: | |
| 22 self.visit(s) | |
| 23 elif type(node) is IfStatement: | |
| 24 self.visit(node.condition) | |
| 25 self.visit(node.truestatement) | |
| 26 self.visit(node.falsestatement) | |
| 167 | 27 elif type(node) is FunctionCall: |
| 28 for arg in node.args: | |
| 29 self.visit(arg) | |
| 163 | 30 elif type(node) is Assignment: |
| 31 self.visit(node.lval) | |
| 32 self.visit(node.rval) | |
| 33 elif type(node) is ReturnStatement: | |
| 34 self.visit(node.expr) | |
| 35 elif type(node) is Binop: | |
| 36 self.visit(node.a) | |
| 37 self.visit(node.b) | |
| 166 | 38 elif type(node) is Constant: |
| 39 self.visit(node.value) | |
| 40 elif type(node) in [EmptyStatement, VariableUse, Variable, Literal, FunctionType]: | |
| 163 | 41 # Those nodes do not have child nodes. |
| 42 pass | |
| 43 else: | |
|
169
ee0d30533dae
Added more tests and improved the diagnostic update
Windel Bouwman
parents:
167
diff
changeset
|
44 print('UNK visit "{0}"'.format(node)) |
| 163 | 45 self.f2(node) |
| 46 | |
| 47 |