rktk/
utils.rs

1//! Util types
2
3/// Print to the display
4#[macro_export]
5macro_rules! print {
6    ($literal:literal) => {{
7        use $crate::task::display::*;
8        let _ = DISPLAY_CONTROLLER.try_send(DisplayMessage::Message($literal));
9    }};
10    ($($arg:tt)*) => {{
11        use $crate::task::display::*;
12        use core::fmt::Write as _;
13
14        let mut str = $crate::reexports::heapless::String::<256>::new();
15        write!(str, $($arg)*).unwrap();
16        let _ = DISPLAY_DYNAMIC_MESSAGE_CONTROLLER.signal(str);
17    }};
18}
19
20/// Print to the display without formatting.
21///
22/// For string that does not require formatting, this macro is more efficient than `print!`.
23#[macro_export]
24macro_rules! print_str {
25    ($str:tt) => {{
26        use $crate::task::display::*;
27        let _ = DISPLAY_CONTROLLER.try_send(DisplayMessage::Message($str));
28    }};
29}
30
31macro_rules! display_state {
32    ($mes_type:ident,$val:expr) => {{
33        use $crate::task::display::*;
34        let _ = DISPLAY_CONTROLLER.try_send(DisplayMessage::$mes_type($val));
35    }};
36}
37
38pub(crate) use display_state;
39
40// Usually, we don't touch the interrupt code directly, so ThreadModeRawMutex is enough.
41#[cfg(target_arch = "arm")]
42pub type RawMutex = embassy_sync::blocking_mutex::raw::ThreadModeRawMutex;
43#[cfg(not(target_arch = "arm"))]
44pub type RawMutex = embassy_sync::blocking_mutex::raw::CriticalSectionRawMutex;
45
46pub type Mutex<T> = embassy_sync::mutex::Mutex<RawMutex, T>;
47pub type Channel<T, const N: usize> = embassy_sync::channel::Channel<RawMutex, T, N>;
48pub type Sender<'a, T, const N: usize> = embassy_sync::channel::Sender<'a, RawMutex, T, N>;
49pub type Receiver<'a, T, const N: usize> = embassy_sync::channel::Receiver<'a, RawMutex, T, N>;
50pub type Signal<T> = embassy_sync::signal::Signal<RawMutex, T>;
51
52/// sjoin or "spawn or join"
53pub(crate) mod sjoin {
54    macro_rules! join {
55        ($spawner:expr, $f1:expr, $($future:expr),* ) => {
56            $crate::utils::sjoin::join!(@alloc {$spawner}, {$f1}, $($future),* );
57            $crate::utils::sjoin::join!(@no_alloc {$f1}, $($future),* );
58        };
59        (@alloc $spawner:expr, $f1:expr, $($future:expr),* ) => {
60            #[cfg(feature = "alloc")]
61            {
62                {
63                    use ::alloc::boxed::Box;
64                    {
65                        $(
66                            let ts = Box::leak(Box::new(embassy_executor::raw::TaskStorage::new()));
67                            let st = ts.spawn(move || $future);
68                            $spawner.spawn(st).unwrap();
69                        )*
70                    }
71                }
72
73                $f1.await
74            };
75        };
76        (@no_alloc $f1:expr, $($future:expr),* ) => {
77            #[cfg(not(feature = "alloc"))]
78            futures::join!($f1, $($future),* );
79        };
80    }
81    pub(crate) use join;
82}
83
84#[macro_export]
85macro_rules! singleton {
86    ($val:expr, $type:ty) => {{
87        static STATIC_CELL: $crate::reexports::static_cell::StaticCell<$type> =
88            $crate::reexports::static_cell::StaticCell::new();
89        STATIC_CELL.init($val)
90    }};
91}
rktk/utils.rs

rktk/
utils.rs
