aboutsummaryrefslogtreecommitdiff
path: root/rtic-macros/src/codegen/local_resources.rs
diff options
context:
space:
mode:
Diffstat (limited to 'rtic-macros/src/codegen/local_resources.rs')
-rw-r--r--rtic-macros/src/codegen/local_resources.rs65
1 files changed, 65 insertions, 0 deletions
diff --git a/rtic-macros/src/codegen/local_resources.rs b/rtic-macros/src/codegen/local_resources.rs
new file mode 100644
index 00000000..e6d15533
--- /dev/null
+++ b/rtic-macros/src/codegen/local_resources.rs
@@ -0,0 +1,65 @@
+use crate::syntax::ast::App;
+use crate::{analyze::Analysis, codegen::util};
+use proc_macro2::TokenStream as TokenStream2;
+use quote::quote;
+
+/// Generates `local` variables and local resource proxies
+///
+/// I.e. the `static` variables and theirs proxies.
+pub fn codegen(app: &App, _analysis: &Analysis) -> TokenStream2 {
+ let mut mod_app = vec![];
+
+ // All local resources declared in the `#[local]' struct
+ for (name, res) in &app.local_resources {
+ let cfgs = &res.cfgs;
+ let ty = &res.ty;
+ let mangled_name = util::static_local_resource_ident(name);
+
+ let attrs = &res.attrs;
+
+ // late resources in `util::link_section_uninit`
+ // unless user specifies custom link section
+ let section = if attrs.iter().any(|attr| attr.path.is_ident("link_section")) {
+ None
+ } else {
+ Some(util::link_section_uninit())
+ };
+
+ // For future use
+ // let doc = format!(" RTIC internal: {}:{}", file!(), line!());
+ mod_app.push(quote!(
+ #[allow(non_camel_case_types)]
+ #[allow(non_upper_case_globals)]
+ // #[doc = #doc]
+ #[doc(hidden)]
+ #(#attrs)*
+ #(#cfgs)*
+ #section
+ static #mangled_name: rtic::RacyCell<core::mem::MaybeUninit<#ty>> = rtic::RacyCell::new(core::mem::MaybeUninit::uninit());
+ ));
+ }
+
+ // All declared `local = [NAME: TY = EXPR]` local resources
+ for (task_name, resource_name, task_local) in app.declared_local_resources() {
+ let cfgs = &task_local.cfgs;
+ let ty = &task_local.ty;
+ let expr = &task_local.expr;
+ let attrs = &task_local.attrs;
+
+ let mangled_name = util::declared_static_local_resource_ident(resource_name, task_name);
+
+ // For future use
+ // let doc = format!(" RTIC internal: {}:{}", file!(), line!());
+ mod_app.push(quote!(
+ #[allow(non_camel_case_types)]
+ #[allow(non_upper_case_globals)]
+ // #[doc = #doc]
+ #[doc(hidden)]
+ #(#attrs)*
+ #(#cfgs)*
+ static #mangled_name: rtic::RacyCell<#ty> = rtic::RacyCell::new(#expr);
+ ));
+ }
+
+ quote!(#(#mod_app)*)
+}