-
Notifications
You must be signed in to change notification settings - Fork 0
/
tpl.ml
20 lines (18 loc) · 826 Bytes
/
tpl.ml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
(* Top-level of the TPL compiler: scan & parse the input,
check the resulting AST, generate LLVM IR, and dump the module *)
type action = Ast | LLVM_IR | Compile
let _ =
let action = if Array.length Sys.argv > 1 then
List.assoc Sys.argv.(1) [ ("-a", Ast); (* Print the AST only *)
("-l", LLVM_IR); (* Generate LLVM, don't check *)
("-c", Compile) ] (* Generate, check LLVM IR *)
else Compile in
let lexbuf = Lexing.from_channel stdin in
let ast = Parser.program Scanner.token lexbuf in
Semant.check ast;
match action with
Ast -> print_string (Ast.string_of_program ast)
| LLVM_IR -> print_string (Llvm.string_of_llmodule (Codegen.translate ast))
| Compile -> let m = Codegen.translate ast in
Llvm_analysis.assert_valid_module m;
print_string (Llvm.string_of_llmodule m)