|
| 1 | +from .Node import NodeVisitor |
| 2 | +from .Parser import Parser |
| 3 | +from .Token import RESERVED_KEYWORDS, Token |
| 4 | + |
| 5 | + |
| 6 | +class Interpreter(NodeVisitor): |
| 7 | + |
| 8 | + GLOBAL_SCOPE = {} |
| 9 | + |
| 10 | + def __init__(self, text): |
| 11 | + self.parser = Parser(text) |
| 12 | + Interpreter.GLOBAL_SCOPE = {} |
| 13 | + |
| 14 | + def visit_binop(self, node): |
| 15 | + if node.op.type == Token.PLUS: |
| 16 | + return self.visit(node.left) + self.visit(node.right) |
| 17 | + elif node.op.type == Token.MINUS: |
| 18 | + return self.visit(node.left) - self.visit(node.right) |
| 19 | + elif node.op.type == Token.MUL: |
| 20 | + return self.visit(node.left) * self.visit(node.right) |
| 21 | + elif node.op.type == RESERVED_KEYWORDS['DIV'].type: |
| 22 | + return self.visit(node.left) // self.visit(node.right) |
| 23 | + else: |
| 24 | + raise Exception('Unknown binary operator') |
| 25 | + |
| 26 | + def visit_num(self, node): |
| 27 | + return node.value |
| 28 | + |
| 29 | + def visit_unaryop(self, node): |
| 30 | + op = node.op.type |
| 31 | + if op == Token.PLUS: |
| 32 | + return +self.visit(node.expr) |
| 33 | + elif op == Token.MINUS: |
| 34 | + return -self.visit(node.expr) |
| 35 | + else: |
| 36 | + raise Exception('Unknown unary operator') |
| 37 | + |
| 38 | + def visit_compound(self, node): |
| 39 | + for child in node.children: |
| 40 | + self.visit(child) |
| 41 | + |
| 42 | + def visit_assign(self, node): |
| 43 | + var_name = node.left.value |
| 44 | + self.GLOBAL_SCOPE[var_name] = self.visit(node.right) |
| 45 | + |
| 46 | + def visit_var(self, node): |
| 47 | + var_name = node.value |
| 48 | + val = self.GLOBAL_SCOPE.get(var_name) |
| 49 | + if val is None: |
| 50 | + raise Exception(f'NameError: name \'{var_name}\' is not defined') |
| 51 | + else: |
| 52 | + return val |
| 53 | + |
| 54 | + def visit_noop(self, node): |
| 55 | + pass |
| 56 | + |
| 57 | + def interpret(self): |
| 58 | + tree = self.parser.parse() |
| 59 | + if tree is None: |
| 60 | + return '' |
| 61 | + |
| 62 | + return self.visit(tree) |
0 commit comments