rtic_macros/codegen/
post_init.rs

1use crate::{analyze::Analysis, codegen::util, syntax::ast::App};
2use proc_macro2::TokenStream as TokenStream2;
3use quote::quote;
4
5/// Generates code that runs after `#[init]` returns
6pub fn codegen(app: &App, analysis: &Analysis) -> Vec<TokenStream2> {
7    let mut stmts = vec![];
8
9    // Initialize shared resources
10    for (name, res) in &app.shared_resources {
11        let mangled_name = util::static_shared_resource_ident(name);
12        // If it's live
13        let cfgs = res.cfgs.clone();
14        if analysis.shared_resources.get(name).is_some() {
15            stmts.push(quote!(
16                // We include the cfgs
17                #(#cfgs)*
18                // Resource is a RacyCell<MaybeUninit<T>>
19                // - `get_mut` to obtain a raw pointer to `MaybeUninit<T>`
20                // - `write` the defined value for the late resource T
21                #mangled_name.get_mut().write(core::mem::MaybeUninit::new(shared_resources.#name));
22            ));
23        }
24    }
25
26    // Initialize local resources
27    for (name, res) in &app.local_resources {
28        let mangled_name = util::static_local_resource_ident(name);
29        // If it's live
30        let cfgs = res.cfgs.clone();
31        if analysis.local_resources.get(name).is_some() {
32            stmts.push(quote!(
33                // We include the cfgs
34                #(#cfgs)*
35                // Resource is a RacyCell<MaybeUninit<T>>
36                // - `get_mut` to obtain a raw pointer to `MaybeUninit<T>`
37                // - `write` the defined value for the late resource T
38                #mangled_name.get_mut().write(core::mem::MaybeUninit::new(local_resources.#name));
39            ));
40        }
41    }
42
43    // Enable the interrupts -- this completes the `init`-ialization phase
44    stmts.push(quote!(rtic::export::interrupt::enable();));
45
46    stmts
47}