1
0
Fork 0
zynq-rs/libboard_zynq/src/stdio.rs

85 lines
2.2 KiB
Rust
Raw Normal View History

2019-11-21 00:00:57 +08:00
use core::ops::{Deref, DerefMut};
2020-08-18 01:38:41 +08:00
use libcortex_a9::{asm, mutex::{Mutex, MutexGuard}};
use crate::uart::Uart;
2019-06-20 06:30:18 +08:00
const UART_RATE: u32 = 115_200;
2019-11-21 00:00:57 +08:00
static mut UART: Mutex<LazyUart> = Mutex::new(LazyUart::Uninitialized);
2019-06-20 06:30:18 +08:00
#[doc(hidden)]
2019-11-21 00:00:57 +08:00
pub fn get_uart<'a>() -> MutexGuard<'a, LazyUart> {
unsafe { UART.lock() }
}
2020-08-18 01:38:41 +08:00
/// Deinitialize so that the Uart will be reinitialized on next
/// output.
///
/// Delays so that an outstanding transmission can finish.
pub fn drop_uart() {
2020-08-18 01:38:41 +08:00
for _ in 0..1_000_000 {
asm::nop();
}
unsafe { UART = Mutex::new(LazyUart::Uninitialized); }
}
2019-11-21 00:00:57 +08:00
/// Initializes the UART on first use through `.deref_mut()` for debug
/// output through the `print!` and `println!` macros.
pub enum LazyUart {
Uninitialized,
Initialized(Uart),
}
impl Deref for LazyUart {
type Target = Uart;
fn deref(&self) -> &Uart {
match self {
LazyUart::Uninitialized =>
panic!("stdio not initialized!"),
LazyUart::Initialized(uart) =>
uart,
}
}
}
impl DerefMut for LazyUart {
fn deref_mut(&mut self) -> &mut Uart {
match self {
LazyUart::Uninitialized => {
#[cfg(any(feature = "target_coraz7", feature = "target_redpitaya"))]
let uart = Uart::uart0(UART_RATE);
#[cfg(any(
feature = "target_zc706",
feature = "target_ebaz4205",
feature = "target_kasli_soc",
))]
let uart = Uart::uart1(UART_RATE);
2019-11-21 00:00:57 +08:00
*self = LazyUart::Initialized(uart);
self
2019-06-20 06:30:18 +08:00
}
2019-11-21 00:00:57 +08:00
LazyUart::Initialized(uart) =>
uart,
2019-06-20 06:30:18 +08:00
}
}
}
2019-06-21 07:18:24 +08:00
#[macro_export]
macro_rules! print {
($($arg:tt)*) => ({
use core::fmt::Write;
let mut uart = $crate::stdio::get_uart();
2019-08-11 06:55:27 +08:00
let _ = write!(uart, $($arg)*);
2019-06-21 07:18:24 +08:00
})
}
2019-06-20 06:30:18 +08:00
#[macro_export]
macro_rules! println {
($($arg:tt)*) => ({
use core::fmt::Write;
let mut uart = $crate::stdio::get_uart();
2019-08-11 06:55:27 +08:00
let _ = write!(uart, $($arg)*);
let _ = write!(uart, "\n");
// flush after the newline
while !uart.tx_idle() {}
2019-06-20 06:30:18 +08:00
})
}