rtic_macros/codegen/
software_tasks.rs

Help
1use crate::syntax::{ast::App, Context};
2use crate::{
3    analyze::Analysis,
4    codegen::{local_resources_struct, module, shared_resources_struct},
5};
6use proc_macro2::TokenStream as TokenStream2;
7use quote::quote;
8
9pub fn codegen(app: &App, analysis: &Analysis) -> TokenStream2 {
10    let mut mod_app = vec![];
11    let mut root = vec![];
12    let mut user_tasks = vec![];
13
14    // Any task
15    for (name, task) in app.software_tasks.iter() {
16        if !task.args.local_resources.is_empty() {
17            let (item, constructor) =
18                local_resources_struct::codegen(Context::SoftwareTask(name), app);
19
20            root.push(item);
21
22            mod_app.push(constructor);
23        }
24
25        if !task.args.shared_resources.is_empty() {
26            let (item, constructor) =
27                shared_resources_struct::codegen(Context::SoftwareTask(name), app);
28
29            root.push(item);
30
31            mod_app.push(constructor);
32        }
33
34        if !task.is_extern {
35            let context = &task.context;
36            let attrs = &task.attrs;
37            let cfgs = &task.cfgs;
38            let stmts = &task.stmts;
39            let inputs = &task.inputs;
40            let lifetime = if task.is_bottom { quote!('static) } else { quote!('a) };
41            let generics = if task.is_bottom { quote!() } else { quote!(<'a>) };
42
43            user_tasks.push(quote!(
44                #(#attrs)*
45                #(#cfgs)*
46                #[allow(non_snake_case)]
47                async fn #name #generics(#context: #name::Context<#lifetime> #(,#inputs)*) {
48                    use rtic::Mutex as _;
49                    use rtic::mutex::prelude::*;
50
51                    #(#stmts)*
52                }
53            ));
54        }
55
56        root.push(module::codegen(Context::SoftwareTask(name), app, analysis));
57    }
58
59    quote!(
60        #(#mod_app)*
61
62        #(#root)*
63
64        #(#user_tasks)*
65    )
66}
rtic_macros/codegen/software_tasks.rs

rtic_macros/codegen/
software_tasks.rs
