82 lines
1.8 KiB
Python
82 lines
1.8 KiB
Python
from lexer import Lexer
|
|
from plasma_parser import Parser
|
|
from compiler import Compiler
|
|
from AST import Program
|
|
import json
|
|
import time
|
|
|
|
from llvmlite import ir
|
|
import llvmlite.binding as llvm
|
|
from ctypes import CFUNCTYPE, c_int, c_float
|
|
|
|
LEXER_DEBUG: bool = False
|
|
PARSER_DEBUG: bool = True
|
|
COMPILER_DEBUG: bool = False
|
|
RUN_CODE: bool = False
|
|
|
|
|
|
if __name__ == "__main__":
|
|
with open("tests/test.pla") as f:
|
|
code: str = f.read()
|
|
|
|
if LEXER_DEBUG:
|
|
debug_lex: Lexer = Lexer(source=code)
|
|
while debug_lex.current_char is not None:
|
|
print(debug_lex.next_token())
|
|
|
|
l: Lexer = Lexer(source=code)
|
|
p: Parser = Parser(lexer=l)
|
|
|
|
program: Program = p.parse_program()
|
|
if len(p.errors) > 0:
|
|
for err in p.errors:
|
|
print(err)
|
|
exit(1)
|
|
|
|
if PARSER_DEBUG:
|
|
print("===== PARSER DEBUG =====")
|
|
#program: Program = p.parse_program()
|
|
|
|
with open("debug/ast.json", "w") as f:
|
|
json.dump(program.json(), f, indent=4)
|
|
|
|
print("Wrote AST to debug/ast.json successfully.")
|
|
|
|
c: Compiler = Compiler()
|
|
c.compile(program)
|
|
|
|
module: ir.Module = c.module
|
|
module.triple = llvm.get_default_triple()
|
|
|
|
if COMPILER_DEBUG:
|
|
with open("debug/ir.ll", "w") as f:
|
|
f.write(str(module))
|
|
|
|
if RUN_CODE:
|
|
#llvm.initialize()
|
|
llvm.initialize_native_target()
|
|
llvm.initialize_native_asmparser()
|
|
llvm.initialize_native_asmprinter()
|
|
|
|
try:
|
|
llvm_ir_parsed = llvm.parse_assembly(str(module))
|
|
llvm_ir_parsed.verify()
|
|
except Exception as e:
|
|
print(e)
|
|
raise
|
|
|
|
target_machine = llvm.Target.from_default_triple().create_target_machine()
|
|
|
|
engine = llvm.create_mcjit_compiler(llvm_ir_parsed, target_machine)
|
|
engine.finalize_object()
|
|
|
|
entry = engine.get_function_address("main")
|
|
cfunc = CFUNCTYPE(c_int)(entry)
|
|
|
|
st = time.time()
|
|
|
|
result = cfunc()
|
|
|
|
et = time.time()
|
|
|
|
print(f"\n\nProgram returned: {result}\n=== Executed in {round((et - st) * 1000, 6)} ms. ===") |