aboutsummaryrefslogtreecommitdiff
path: root/rtic-macros/src/codegen
diff options
context:
space:
mode:
Diffstat (limited to 'rtic-macros/src/codegen')
-rw-r--r--rtic-macros/src/codegen/idle.rs24
-rw-r--r--rtic-macros/src/codegen/init.rs22
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;