esp-hal/esp32s2-hal/examples/embassy_serial.rs
Scott Mabin eeb2d1db6f
Fix UART to handle CPU/APB clock changes (#808)
* Ensure that uart is configured to account for clock changes, not just boot defaults

* fix examples

* changelog

---------

Co-authored-by: Jesse Braham <jessebraham@users.noreply.github.com>
2023-09-21 09:06:56 -07:00

99 lines
2.7 KiB
Rust

//! embassy serial
//!
//! This is an example of running the embassy executor and asynchronously
//! writing to and reading from uart
#![no_std]
#![no_main]
#![feature(type_alias_impl_trait)]
use esp32s2_hal::{
clock::ClockControl,
embassy::{self, executor::Executor},
interrupt,
peripherals::{Interrupt, Peripherals, UART0},
prelude::*,
Uart,
};
use esp_backtrace as _;
use esp_hal_common::uart::{config::AtCmdConfig, UartRx, UartTx};
use heapless::Vec;
use static_cell::make_static;
// rx_fifo_full_threshold
const READ_BUF_SIZE: usize = 64;
// EOT (CTRL-D)
const AT_CMD: u8 = 0x04;
#[embassy_executor::task]
async fn writer(mut tx: UartTx<'static, UART0>) {
esp_println::println!("writing...");
embedded_io_async::Write::write(
&mut tx,
b"Hello async serial. Enter something ended with EOT (CTRL-D).\r\n",
)
.await
.unwrap();
embedded_io_async::Write::flush(&mut tx).await.unwrap();
}
#[embassy_executor::task]
async fn reader(mut rx: UartRx<'static, UART0>) {
esp_println::println!("reading...");
// max message size to receive
// leave some extra space for AT-CMD characters
const MAX_BUFFER_SIZE: usize = 10 * READ_BUF_SIZE + 16;
let mut rbuf: Vec<u8, MAX_BUFFER_SIZE> = Vec::new();
let mut offset = 0;
while let Ok(len) = embedded_io_async::Read::read(&mut rx, &mut rbuf[offset..]).await {
offset += len;
if offset == 0 {
rbuf.truncate(0);
break;
}
// if set_at_cmd is used than stop reading
if len < READ_BUF_SIZE {
rbuf.truncate(offset);
break;
}
}
}
#[entry]
fn main() -> ! {
esp_println::println!("Init!");
let peripherals = Peripherals::take();
let mut system = peripherals.SYSTEM.split();
let clocks = ClockControl::boot_defaults(system.clock_control).freeze();
#[cfg(feature = "embassy-time-timg0")]
{
let timer_group0 = esp32s2_hal::timer::TimerGroup::new(
peripherals.TIMG0,
&clocks,
&mut system.peripheral_clock_control,
);
embassy::init(&clocks, timer_group0.timer0);
}
let mut uart0 = Uart::new(
peripherals.UART0,
&clocks,
&mut system.peripheral_clock_control,
);
uart0.set_at_cmd(AtCmdConfig::new(None, None, None, AT_CMD, None));
uart0
.set_rx_fifo_full_threshold(READ_BUF_SIZE as u16)
.unwrap();
let (tx, rx) = uart0.split();
interrupt::enable(Interrupt::UART0, interrupt::Priority::Priority1).unwrap();
let executor = make_static!(Executor::new());
executor.run(|spawner| {
spawner.spawn(reader(rx)).ok();
spawner.spawn(writer(tx)).ok();
});
}