     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

    use crate::drivers::chardev::{CharDevice, UART};
    use core::fmt::{self, Write};

    struct Stdout;

    impl Write for Stdout {
        fn write_str(&mut self, s: &str) -> fmt::Result {
            for c in s.chars() {
                UART.write(c as u8);
            }
            Ok(())
        }
    }

    pub fn print(args: fmt::Arguments) {
        Stdout.write_fmt(args).unwrap();
    }

    #[macro_export]
    macro_rules! print {
        ($fmt: literal $(, $($arg: tt)+)?) => {
            $crate::console::print(format_args!($fmt $(, $($arg)+)?))
        }
    }

    #[macro_export]
    macro_rules! println {
        ($fmt: literal $(, $($arg: tt)+)?) => {
            $crate::console::print(format_args!(concat!($fmt, "\n") $(, $($arg)+)?))
        }
    }

    use crate::sbi::console_putchar;
    struct Kstdout;

    impl Write for Kstdout {
        fn write_str(&mut self, s: &str) -> fmt::Result {
            for c in s.chars() {
                console_putchar(c as usize);
            }
            Ok(())
        }
    }

    pub fn kprint(args: fmt::Arguments) {
        Kstdout.write_fmt(args).unwrap();
    }

    #[macro_export]
    macro_rules! kprint {
        ($fmt: literal $(, $($arg: tt)+)?) => {
            $crate::console::kprint(format_args!($fmt $(, $($arg)+)?));
        }
    }

    #[macro_export]
    macro_rules! kprintln {
        ($fmt: literal $(, $($arg: tt)+)?) => {
            $crate::console::kprint(format_args!(concat!($fmt, "\n") $(, $($arg)+)?));
        }
    }