Library-izing brilift

brilift/Cargo.toml

@@ -4,7 +4,7 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
-bril-rs = { path = "../bril-rs", features = ["float", "memory"] }
+bril-rs = { path = "../bril-rs", features = ["memory", "float", "ssa", "speculate", "position", "import", "char"] }
 cranelift-codegen = { version = "0.103.0", features = ["all-arch"] }
 cranelift-frontend = "0.103.0"
 cranelift-object = "0.103.0"

brilift/src/lib.rs

@@ -0,0 +1,73 @@
+mod rt;
+pub mod translator;
+
+use crate::translator::{find_func, Translator};
+use bril_rs as bril;
+use bril_rs::Program;
+use cranelift_jit::JITModule;
+use cranelift_object::ObjectModule;
+
+/// Compile a program ahead of time to an object file.
+///
+/// * `program` - the Bril program to compile
+/// * `target` - the target triple, or None to target the host
+/// * `output` - the filename where we should write the object file
+/// * `opt_level` - a Cranelift optimization level
+/// * `dump_ir` - optionally emit the Cranelift IR to stdout
+pub fn compile(
+    program: &Program,
+    target: Option<String>,
+    output: &str,
+    opt_level: &str,
+    dump_ir: bool,
+) {
+    // Compile.
+    let mut trans = Translator::<ObjectModule>::new(target, opt_level);
+    trans.compile_prog(program, dump_ir);
+
+    // Add a C-style `main` wrapper.
+    let main = find_func(&program.functions, "main");
+    trans.add_c_main(&main.args, dump_ir);
+
+    // Write object file.
+    trans.emit(output);
+}
+
+/// Just-in-time compile and execute a Bril program.
+///
+/// * `program` - the Bril program to compile
+/// * `args` - the arguments to pass to the `@main` function
+/// * `dump_ir` - optionally emit the Cranelift IR to stdout
+pub fn jit_run(program: &Program, args: Vec<String>, dump_ir: bool) {
+    // Compile.
+    let mut trans = Translator::<JITModule>::new();
+    trans.compile_prog(program, dump_ir);
+
+    // Add a JIT wrapper for `main`.
+    let main = find_func(&program.functions, "main");
+    let entry_id = trans.add_mem_wrapper("main", &main.args, dump_ir);
+
+    // Parse CLI arguments.
+    if main.args.len() != args.len() {
+        panic!(
+            "@main expects {} arguments; got {}",
+            main.args.len(),
+            args.len()
+        );
+    }
+    let main_args: Vec<bril::Literal> = main
+        .args
+        .iter()
+        .zip(args)
+        .map(|(arg, val_str)| match arg.arg_type {
+            bril::Type::Int => bril::Literal::Int(val_str.parse().unwrap()),
+            bril::Type::Bool => bril::Literal::Bool(val_str == "true"),
+            bril::Type::Float => bril::Literal::Float(val_str.parse().unwrap()),
+            bril::Type::Char => bril::Literal::Char(val_str.parse().unwrap()),
+            bril::Type::Pointer(_) => unimplemented!("pointers not supported as main args"),
+        })
+        .collect();
+
+    // Invoke the main function.
+    unsafe { trans.run(entry_id, &main_args) };
+}