-
Notifications
You must be signed in to change notification settings - Fork 0
/
interpreter.py
142 lines (104 loc) · 4.01 KB
/
interpreter.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
from tbparser import Parser
from tokenizer import Tokenizer, Token
from collections import OrderedDict
class Interpreter:
def __init__(self):
self.lines = {}
self._program_counter = 0
self._running = False
self._stack = []
self._variables = {}
self._parser = Parser(self._variables)
def interactive(self):
while True:
line = input('>')
if len(line):
self.interpret_line(line)
def interpret_line(self, line):
tokenizer = Tokenizer()
tokenizer.parse(line)
first_token = tokenizer.getNextToken()
if first_token.type == Token.NUMBER:
self.lines[int(first_token.value)] = tokenizer.prog[tokenizer.pos:]
self.sort_lines()
else:
self.run_line(line)
def run_line(self, line):
tokenizer = Tokenizer()
tokenizer.parse(line)
self.execute_statement(tokenizer)
def execute_statement(self, tokenizer):
statement = self._parser.match_statement(tokenizer)
if statement == 'LET':
self.stat_let(tokenizer)
elif statement == 'PRINT':
self.stat_print(tokenizer)
elif statement == 'LIST':
self.stat_list()
elif statement == 'INPUT':
self.stat_input(tokenizer)
elif statement == 'IF':
self.stat_if(tokenizer)
elif statement == 'RUN':
self.run_program()
elif statement == 'END':
self.stat_end()
elif statement == 'GOTO':
self.stat_goto(tokenizer)
elif statement == 'GOSUB':
self.stat_gosub(tokenizer)
elif statement == 'RETURN':
self.stat_return()
else:
raise Exception('Unrecognised statement: ' + statement.value)
def run_program(self):
self._program_counter = 0
self._running = True
self.sort_lines()
statements = [x for x in self.lines.values()]
while self._program_counter < len(statements) and self._running:
statement = statements[self._program_counter]
self.run_line(statement)
self._program_counter += 1
def sort_lines(self):
self.lines = OrderedDict(sorted(self.lines.items(), key=lambda x: x[0]))
def stat_let(self, tokenizer):
variable = tokenizer.getNextToken()
if variable.type != Token.VARIABLE:
raise Exception("Expected a variable")
if tokenizer.getNextToken().type != Token.EQUALS:
raise Exception('Expected an equals')
self._parser._variables[variable.value] = self._parser.match_expression(tokenizer)
def stat_print(self, tokenizer):
list = self._parser.match_expression_list(tokenizer)
print(','.join([str(i) for i in list]))
def stat_list(self):
for no, line in iter(self.lines.items()):
print(no, line,)
def stat_input(self, tokenizer):
vars = self._parser.match_var_list(tokenizer)
for var in vars:
self._parser._variables[var] = input("?")
def stat_if(self, tokenizer):
result = self._parser.match_relop(tokenizer)
then = tokenizer.getNextToken()
if then.type != Token.COMMAND or then.value != 'THEN':
raise Exception('Expected then after relative operator')
if result:
self.execute_statement(tokenizer)
def stat_end(self):
self._running = False
def stat_goto(self, tokenizer):
line_number = self._parser.match_expression(tokenizer)
self.sort_lines()
line_numbers = [x for x in self.lines.keys()]
self._program_counter = line_numbers.index(line_number) - 1
def stat_gosub(self, tokenizer):
self._stack.append(self._program_counter)
self.stat_goto(tokenizer)
def stat_return(self):
self._program_counter = self._stack.pop()
if __name__ == '__main__':
print('Tiny Basic in Python')
interp = Interpreter()
interp.interactive()