use core::{fmt, mem, ptr, slice};
use aligned::{Aligned, A4};
#[cfg(armv8m_base)]
use crate::peripheral::itm::Stim;
#[cfg(not(armv8m_base))]
use cortex_m_0_7::peripheral::itm::Stim;
#[allow(clippy::missing_inline_in_public_items)]
unsafe fn write_words(stim: &mut Stim, bytes: &[u32]) {
let mut p = bytes.as_ptr();
for _ in 0..bytes.len() {
while !stim.is_fifo_ready() {}
stim.write_u32(ptr::read(p));
p = p.offset(1);
}
}
struct Port<'p>(&'p mut Stim);
impl<'p> fmt::Write for Port<'p> {
#[inline]
fn write_str(&mut self, s: &str) -> fmt::Result {
write_all(self.0, s.as_bytes());
Ok(())
}
}
#[allow(clippy::cast_ptr_alignment)]
#[allow(clippy::missing_inline_in_public_items)]
#[allow(clippy::transmute_ptr_to_ptr)]
pub fn write_all(port: &mut Stim, buffer: &[u8]) {
unsafe {
let mut len = buffer.len();
let mut ptr = buffer.as_ptr();
if len == 0 {
return;
}
if ptr as usize % 2 == 1 {
while !port.is_fifo_ready() {}
port.write_u8(*ptr);
ptr = ptr.offset(1);
len -= 1;
}
if ptr as usize % 4 == 2 {
if len > 1 {
while !port.is_fifo_ready() {}
port.write_u16(ptr::read(ptr as *const u16));
ptr = ptr.offset(2);
len -= 2;
} else {
if len == 1 {
while !port.is_fifo_ready() {}
port.write_u8(*ptr);
}
return;
}
}
write_aligned(port, mem::transmute(slice::from_raw_parts(ptr, len)));
}
}
#[allow(clippy::cast_ptr_alignment)]
#[allow(clippy::missing_inline_in_public_items)]
#[allow(clippy::transmute_ptr_to_ptr)]
pub fn write_aligned(port: &mut Stim, buffer: &Aligned<A4, [u8]>) {
unsafe {
let len = buffer.len();
if len == 0 {
return;
}
let split = len & !0b11;
write_words(
port,
slice::from_raw_parts(buffer.as_ptr() as *const u32, split >> 2),
);
let mut left = len & 0b11;
let mut ptr = buffer.as_ptr().add(split);
if left > 1 {
while !port.is_fifo_ready() {}
port.write_u16(ptr::read(ptr as *const u16));
ptr = ptr.offset(2);
left -= 2;
}
if left == 1 {
while !port.is_fifo_ready() {}
port.write_u8(*ptr);
}
}
}
#[inline]
pub fn write_fmt(port: &mut Stim, args: fmt::Arguments) {
use core::fmt::Write;
Port(port).write_fmt(args).ok();
}
#[inline]
pub fn write_str(port: &mut Stim, string: &str) {
write_all(port, string.as_bytes())
}