From 204baabfd21b9bf0c482bcfd7a2935cdfa447e8b Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Sat, 6 Nov 2021 14:14:53 +0800
Subject: [PATCH] nac3artiq: add compile_method_to_mem

---
 nac3artiq/demo/min_artiq.py |  2 +-
 nac3artiq/src/lib.rs        | 20 +++++++++++++++++---
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/nac3artiq/demo/min_artiq.py b/nac3artiq/demo/min_artiq.py
index 6bc4f114..06c3640d 100644
--- a/nac3artiq/demo/min_artiq.py
+++ b/nac3artiq/demo/min_artiq.py
@@ -110,7 +110,7 @@ class Core:
             obj = method
             name = ""
 
-        compiler.compile_method(obj, name, args)
+        compiler.compile_method_to_file(obj, name, args, "module.elf")
 
     @kernel
     def reset(self):
diff --git a/nac3artiq/src/lib.rs b/nac3artiq/src/lib.rs
index 8bd19109..b3936924 100644
--- a/nac3artiq/src/lib.rs
+++ b/nac3artiq/src/lib.rs
@@ -317,11 +317,12 @@ impl Nac3 {
         Ok(())
     }
 
-    fn compile_method(
+    fn compile_method_to_file(
         &mut self,
         obj: &PyAny,
-        method_name: String,
+        method_name: &str,
         args: Vec<&PyAny>,
+        filename: &str,
         py: Python,
     ) -> PyResult<()> {
         let id_fun = PyModule::import(py, "builtins")?.getattr("id")?;
@@ -464,7 +465,7 @@ impl Nac3 {
             "--eh-frame-hdr".to_string(),
             "-x".to_string(),
             "-o".to_string(),
-            "module.elf".to_string(),
+            filename.to_string(),
         ];
         if isa != Isa::Host {
             linker_args.push("-T".to_string() + self.working_directory.path().join("kernel.ld").to_str().unwrap());
@@ -484,6 +485,19 @@ impl Nac3 {
 
         Ok(())
     }
+
+    fn compile_method_to_mem(
+        &mut self,
+        obj: &PyAny,
+        method_name: &str,
+        args: Vec<&PyAny>,
+        py: Python,
+    ) -> PyResult<Vec<u8>> {
+        let filename_path = self.working_directory.path().join("module.elf");
+        let filename = filename_path.to_str().unwrap();
+        self.compile_method_to_file(obj, method_name, args, filename, py)?;
+        Ok(fs::read(filename).unwrap())
+    }
 }
 
 #[pymodule]