simpleos 0.6.3

A Simple OS use for MCU, that implements single-threaded asynchronous runtime.
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

use crate::{driver::Driver, println};

pub struct LazyInit<T, F = fn() -> T>
where
    T: Driver,
    F: FnOnce() -> T,
{
    value: Option<T>,
    init_func: Option<F>,
}

impl<T, F> LazyInit<T, F>
where
    T: Driver,
    F: FnOnce() -> T,
{
    pub const fn new(f: F) -> Self {
        Self {
            value: None,
            init_func: Some(f),
        }
    }

    pub fn get_or_init(&mut self) -> &mut T {
        if self.value.is_none() {
            if let Some(init_func) = self.init_func.take() {
                self.value = Some(init_func());
                if let Err(e) = self.value.as_mut().unwrap().driver_init() {
                    println!("{} INIT ERROR: {:?}", self.value.as_ref().unwrap().driver_dev_name(), e);
                }else {
                    println!("{} INIT OK.", self.value.as_ref().unwrap().driver_dev_name());
                }
            }
        }
        self.value.as_mut().unwrap()
    }

    pub fn get(&mut self) -> Option<&mut T> {
        self.value.as_mut()
    }
}

impl<T, F> Drop for LazyInit<T, F>
where
    T: Driver,
    F: FnOnce() -> T,
{
    fn drop(&mut self) {
        if let Some(v) = &mut self.value {
            let _ = v.driver_deinit();
        }
    }
}

#[macro_export]
macro_rules! lazy_init {
    ($init:expr) => {
        $crate::driver::lazy_init::LazyInit::new(|| {$init} )
    };
}