use core::marker::PhantomData;
use core::ptr;
use crate::hal::serial;
use crate::stm32::{USART1, USART2, USART3};
use nb;
use void::Void;
use crate::gpio::gpioa::{PA10, PA2, PA3, PA9};
use crate::gpio::gpiob::{PB10, PB6, PB7};
#[cfg(any(
feature = "stm32f301",
feature = "stm32f318",
feature = "stm32f302",
feature = "stm32f303",
feature = "stm32f334",
feature = "stm32f328",
feature = "stm32f358",
feature = "stm32f398"
))]
use crate::gpio::gpiob::PB11;
use crate::gpio::gpioc::{PC10, PC11, PC4, PC5};
#[cfg(any(
feature = "stm32f302",
feature = "stm32f303",
feature = "stm32f373",
feature = "stm32f378",
feature = "stm32f334",
feature = "stm32f328",
feature = "stm32f358",
feature = "stm32f398"
))]
use crate::gpio::gpiod::{PD5, PD6, PD8, PD9};
#[cfg(any(
feature = "stm32f302",
feature = "stm32f303",
feature = "stm32f373",
feature = "stm32f378",
feature = "stm32f328",
feature = "stm32f358",
feature = "stm32f398"
))]
use crate::gpio::gpioe::{PE0, PE1, PE15};
use crate::gpio::AF7;
use crate::rcc::{Clocks, APB1, APB2};
use crate::time::Bps;
pub enum Event {
Rxne,
Txe,
}
#[derive(Debug)]
pub enum Error {
Framing,
Noise,
Overrun,
Parity,
#[doc(hidden)]
_Extensible,
}
pub unsafe trait TxPin<USART> {}
pub unsafe trait RxPin<USART> {}
unsafe impl TxPin<USART1> for PA9<AF7> {}
unsafe impl TxPin<USART1> for PB6<AF7> {}
unsafe impl TxPin<USART1> for PC4<AF7> {}
#[cfg(any(
feature = "stm32f302",
feature = "stm32f303",
feature = "stm32f373",
feature = "stm32f378",
feature = "stm32f328",
feature = "stm32f358",
feature = "stm32f398"
))]
unsafe impl TxPin<USART1> for PE0<AF7> {}
unsafe impl RxPin<USART1> for PA10<AF7> {}
unsafe impl RxPin<USART1> for PB7<AF7> {}
unsafe impl RxPin<USART1> for PC5<AF7> {}
#[cfg(any(
feature = "stm32f302",
feature = "stm32f303",
feature = "stm32f373",
feature = "stm32f378",
feature = "stm32f328",
feature = "stm32f358",
feature = "stm32f398"
))]
unsafe impl RxPin<USART1> for PE1<AF7> {}
unsafe impl TxPin<USART2> for PA2<AF7> {}
#[cfg(any(
feature = "stm32f302",
feature = "stm32f303",
feature = "stm32f373",
feature = "stm32f378",
feature = "stm32f334",
feature = "stm32f328",
feature = "stm32f358",
feature = "stm32f398"
))]
unsafe impl TxPin<USART2> for PD5<AF7> {}
unsafe impl RxPin<USART2> for PA3<AF7> {}
#[cfg(any(
feature = "stm32f302",
feature = "stm32f303",
feature = "stm32f373",
feature = "stm32f378",
feature = "stm32f334",
feature = "stm32f328",
feature = "stm32f358",
feature = "stm32f398"
))]
unsafe impl RxPin<USART2> for PD6<AF7> {}
unsafe impl TxPin<USART3> for PB10<AF7> {}
unsafe impl TxPin<USART3> for PC10<AF7> {}
#[cfg(any(
feature = "stm32f302",
feature = "stm32f303",
feature = "stm32f373",
feature = "stm32f378",
feature = "stm32f334",
feature = "stm32f328",
feature = "stm32f358",
feature = "stm32f398"
))]
unsafe impl TxPin<USART3> for PD8<AF7> {}
#[cfg(any(
feature = "stm32f301",
feature = "stm32f318",
feature = "stm32f302",
feature = "stm32f303",
feature = "stm32f334",
feature = "stm32f328",
feature = "stm32f358",
feature = "stm32f398"
))]
unsafe impl RxPin<USART3> for PB11<AF7> {}
unsafe impl RxPin<USART3> for PC11<AF7> {}
#[cfg(any(
feature = "stm32f302",
feature = "stm32f303",
feature = "stm32f373",
feature = "stm32f378",
feature = "stm32f334",
feature = "stm32f328",
feature = "stm32f358",
feature = "stm32f398"
))]
unsafe impl RxPin<USART3> for PD9<AF7> {}
#[cfg(any(
feature = "stm32f302",
feature = "stm32f303",
feature = "stm32f373",
feature = "stm32f378",
feature = "stm32f328",
feature = "stm32f358",
feature = "stm32f398"
))]
unsafe impl RxPin<USART3> for PE15<AF7> {}
pub struct Serial<USART, PINS> {
usart: USART,
pins: PINS,
}
pub struct Rx<USART> {
_usart: PhantomData<USART>,
}
pub struct Tx<USART> {
_usart: PhantomData<USART>,
}
macro_rules! hal {
($(
$USARTX:ident: ($usartX:ident, $APB:ident, $usartXen:ident, $usartXrst:ident, $pclkX:ident),
)+) => {
$(
impl<TX, RX> Serial<$USARTX, (TX, RX)> {
pub fn $usartX(
usart: $USARTX,
pins: (TX, RX),
baud_rate: Bps,
clocks: Clocks,
apb: &mut $APB,
) -> Self
where
TX: TxPin<$USARTX>,
RX: RxPin<$USARTX>,
{
apb.enr().modify(|_, w| w.$usartXen().set_bit());
apb.rstr().modify(|_, w| w.$usartXrst().set_bit());
apb.rstr().modify(|_, w| w.$usartXrst().clear_bit());
let brr = clocks.$pclkX().0 / baud_rate.0;
assert!(brr >= 16, "impossible baud rate");
usart.brr.write(|w| unsafe { w.bits(brr) });
usart
.cr1
.write(|w| w.ue().set_bit().re().set_bit().te().set_bit());
Serial { usart, pins }
}
pub fn listen(&mut self, event: Event) {
match event {
Event::Rxne => {
self.usart.cr1.modify(|_, w| w.rxneie().set_bit())
},
Event::Txe => {
self.usart.cr1.modify(|_, w| w.txeie().set_bit())
},
}
}
pub fn unlisten(&mut self, event: Event) {
match event {
Event::Rxne => {
self.usart.cr1.modify(|_, w| w.rxneie().clear_bit())
},
Event::Txe => {
self.usart.cr1.modify(|_, w| w.txeie().clear_bit())
},
}
}
pub fn split(self) -> (Tx<$USARTX>, Rx<$USARTX>) {
(
Tx {
_usart: PhantomData,
},
Rx {
_usart: PhantomData,
},
)
}
pub fn free(self) -> ($USARTX, (TX, RX)) {
(self.usart, self.pins)
}
}
impl serial::Read<u8> for Rx<$USARTX> {
type Error = Error;
fn read(&mut self) -> nb::Result<u8, Error> {
let isr = unsafe { (*$USARTX::ptr()).isr.read() };
Err(if isr.pe().bit_is_set() {
nb::Error::Other(Error::Parity)
} else if isr.fe().bit_is_set() {
nb::Error::Other(Error::Framing)
} else if isr.nf().bit_is_set() {
nb::Error::Other(Error::Noise)
} else if isr.ore().bit_is_set() {
nb::Error::Other(Error::Overrun)
} else if isr.rxne().bit_is_set() {
return Ok(unsafe {
ptr::read_volatile(&(*$USARTX::ptr()).rdr as *const _ as *const _)
});
} else {
nb::Error::WouldBlock
})
}
}
impl serial::Write<u8> for Tx<$USARTX> {
type Error = Void;
fn flush(&mut self) -> nb::Result<(), Void> {
let isr = unsafe { (*$USARTX::ptr()).isr.read() };
if isr.tc().bit_is_set() {
Ok(())
} else {
Err(nb::Error::WouldBlock)
}
}
fn write(&mut self, byte: u8) -> nb::Result<(), Void> {
let isr = unsafe { (*$USARTX::ptr()).isr.read() };
if isr.txe().bit_is_set() {
unsafe {
ptr::write_volatile(&(*$USARTX::ptr()).tdr as *const _ as *mut _, byte)
}
Ok(())
} else {
Err(nb::Error::WouldBlock)
}
}
}
)+
}
}
#[cfg(any(
feature = "stm32f301",
feature = "stm32f318",
feature = "stm32f303",
feature = "stm32f373",
feature = "stm32f378",
feature = "stm32f328",
feature = "stm32f358",
feature = "stm32f398"
))]
hal! {
USART1: (usart1, APB2, usart1en, usart1rst, pclk2),
USART2: (usart2, APB1, usart2en, usart2rst, pclk1),
USART3: (usart3, APB1, usart3en, usart3rst, pclk1),
}
#[cfg(any(feature = "stm32f302", feature = "stm32f334"))]
hal! {
USART1: (usart1, APB2, usart1en, usart1rst, pclk2),
USART2: (usart2, APB1, usart2en, usart2rst, pclk1),
}