diff options
Diffstat (limited to 'rtic-macros/src/codegen')
| -rw-r--r-- | rtic-macros/src/codegen/idle.rs | 24 | ||||
| -rw-r--r-- | rtic-macros/src/codegen/init.rs | 22 |
2 files changed, 28 insertions, 18 deletions
diff --git a/rtic-macros/src/codegen/idle.rs b/rtic-macros/src/codegen/idle.rs index 0c833ef..9e608cb 100644 --- a/rtic-macros/src/codegen/idle.rs +++ b/rtic-macros/src/codegen/idle.rs @@ -34,16 +34,20 @@ pub fn codegen(app: &App, analysis: &Analysis) -> TokenStream2 { let attrs = &idle.attrs; let context = &idle.context; let stmts = &idle.stmts; - let user_idle = Some(quote!( - #(#attrs)* - #[allow(non_snake_case)] - fn #name(#context: #name::Context) -> ! { - use rtic::Mutex as _; - use rtic::mutex::prelude::*; - - #(#stmts)* - } - )); + let user_idle = if !idle.is_extern { + Some(quote!( + #(#attrs)* + #[allow(non_snake_case)] + fn #name(#context: #name::Context) -> ! { + use rtic::Mutex as _; + use rtic::mutex::prelude::*; + + #(#stmts)* + } + )) + } else { + None + }; quote!( #(#mod_app)* diff --git a/rtic-macros/src/codegen/init.rs b/rtic-macros/src/codegen/init.rs index b667ae0..f6bb43d 100644 --- a/rtic-macros/src/codegen/init.rs +++ b/rtic-macros/src/codegen/init.rs @@ -54,10 +54,12 @@ pub fn codegen(app: &App, analysis: &Analysis) -> TokenStream2 { .collect(); root_init.push(quote! { + #[doc = r"Shared resources"] #shared_vis struct #shared { #(#shared_resources)* } + #[doc = r"Local resources"] #local_vis struct #local { #(#local_resources)* } @@ -67,14 +69,18 @@ pub fn codegen(app: &App, analysis: &Analysis) -> TokenStream2 { let user_init_return = quote! {#shared, #local}; - let user_init = quote!( - #(#attrs)* - #[inline(always)] - #[allow(non_snake_case)] - fn #name(#context: #name::Context) -> (#user_init_return) { - #(#stmts)* - } - ); + let user_init = if !init.is_extern { + Some(quote!( + #(#attrs)* + #[inline(always)] + #[allow(non_snake_case)] + fn #name(#context: #name::Context) -> (#user_init_return) { + #(#stmts)* + } + )) + } else { + None + }; let mut mod_app = None; |
