-
Notifications
You must be signed in to change notification settings - Fork 0
/
basil.py
83 lines (59 loc) · 2.41 KB
/
basil.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
import antlr4
from BasilLexer import BasilLexer
from BasilListener import BasilListener
from BasilParser import BasilParser
import io
class BasilCompiler(BasilListener):
def __init__(self, asm: io.FileIO):
self.asm = asm
self.arithmetic_operators = []
self.variables = {}
def enterLine(self, ctx: BasilParser.LineContext):
print(ctx.getText())
def enterArithmetic_expression(self, ctx: BasilParser.Arithmetic_expressionContext):
for item in ctx.getChildren():
# print(item)
if str(item).isnumeric():
self.asm.write("PUSH,{},".format(item))
elif str(item) in ["+", "-", "*", "/"]:
self.arithmetic_operators.append(item)
def exitArithmetic_expression(self, ctx: BasilParser.Arithmetic_expressionContext):
for item in self.arithmetic_operators:
if str(item) == "+":
self.asm.write("ADD,")
elif str(item) == "-":
self.asm.write("SUB,")
elif str(item) == "*":
self.asm.write("MUL,")
elif str(item) == "/":
self.asm.write("DIV,")
self.arithmetic_operators = []
def enterStatement(self, ctx: BasilParser.StatementContext):
text = ctx.getText().strip("0123456789 = qwertyuiopasdfghjklzxcvbnm")
other = ctx.getText().strip(text)
another = other.strip("1234567890 =")
other = other.strip("qwertyuiopasdfghjklzxcvbnm =")
variable = ctx.getText().split("PRINT")[-1]
# print("Strips:", text, other, another)
if text == "END":
self.asm.write("HALT")
elif text == "PRINT":
if variable in self.variables:
print_ = self.variables[variable]
else:
print_ = other
self.asm.write("PRINT,{},".format(print_))
elif text == "INPUT":
self.asm.write("IN,")
elif text == "LET":
self.asm.write("MOVE,{},{},".format(other, len(self.variables)))
self.variables[another] = other
if __name__ == "__main__":
lexer = BasilLexer(antlr4.FileStream("example.bas"))
stream = antlr4.CommonTokenStream(lexer)
parser = BasilParser(stream)
tree = parser.code()
with open("temp.sasm", "w") as asm:
compile_ = BasilCompiler(asm)
walker = antlr4.ParseTreeWalker()
walker.walk(compile_, tree)