1 files changed, 31 insertions, 0 deletions
diff --git a/macros/src/codegen/post_init.rs b/macros/src/codegen/post_init.rs
new file mode 100644
index 00000000..c35c6976
--- /dev/null
+++ b/macros/src/codegen/post_init.rs
@@ -0,0 +1,31 @@
+use proc_macro2::TokenStream as TokenStream2;
+use quote::quote;
+use rtic_syntax::ast::App;
+
+use crate::analyze::Analysis;
+
+/// Generates code that runs after `#[init]` returns
+pub fn codegen(app: &App, analysis: &Analysis) -> Vec<TokenStream2> {
+    let mut stmts = vec![];
+
+    // Initialize late resources
+    if analysis.late_resources.len() > 0 {
+        // BTreeSet wrapped in a vector
+        for name in analysis.late_resources.first().unwrap() {
+            // If it's live
+            let cfgs = app.late_resources[name].cfgs.clone();
+            if analysis.locations.get(name).is_some() {
+                // Need to also include the cfgs
+                stmts.push(quote!(
+                #(#cfgs)*
+                #name.as_mut_ptr().write(late.#name);
+                ));
+            }
+        }
+    }
+
+    // Enable the interrupts -- this completes the `init`-ialization phase
+    stmts.push(quote!(rtic::export::interrupt::enable();));
+
+    stmts
+}