linux-rtic-macros 0.1.1

Procedural macros of the linux-rtic crate
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

use proc_macro2::TokenStream;
use quote::quote;
use rtic_syntax::{analyze::Analysis, ast::App, Context};

use crate::codegen::{local_resources_struct, module, shared_resources_struct};

/// Generates support code for `#[idle]` functions
pub fn codegen(
    app: &App,
    analysis: &Analysis,
) -> (
    // all generated idle definitions
    Vec<TokenStream>,
    // call_idle
    TokenStream,
) {
    if let Some(idle) = &app.idle {
        let mut shared_needs_lt = false;
        let mut local_needs_lt = false;
        let mut defs = vec![];

        let name = &idle.name;

        if !idle.args.shared_resources.is_empty() {
            let item = shared_resources_struct::codegen(Context::Idle, &mut shared_needs_lt, app);
            defs.push(item);
        }

        if !idle.args.local_resources.is_empty() {
            let item = local_resources_struct::codegen(Context::Idle, &mut local_needs_lt, app);
            defs.push(item);
        }

        defs.push(module::codegen(
            Context::Idle,
            shared_needs_lt,
            local_needs_lt,
            app,
            analysis,
        ));

        let attrs = &idle.attrs;
        let context = &idle.context;
        let stmts = &idle.stmts;
        defs.push(quote!(
            #(#attrs)*
            #[allow(non_snake_case)]
            fn #name(#context: #name::Context) -> ! {
                use rtic::Mutex as _;
                use rtic::mutex_prelude::*;

                #(#stmts)*
            }
        ));

        let call_idle = quote!(#name(
            #name::Context::new(&core::marker::PhantomData)
        ));

        (defs, call_idle)
    } else {
        (
            vec![],
            quote!(
                let thread = std::thread::current();
                rtic::ctrlc::set_handler(move || {
                    thread.unpark();
                }).expect("Failed to set ctrl-c handler");

                std::thread::park();

                // Fixes newline in terminal
                println!("");
            ),
        )
    }
}