First commit

3 files changed, 263 insertions, 0 deletions

diff --git a/examples/src/bin/qspi.rs b/examples/src/bin/qspi.rs
new file mode 100644
index 000000000..395422e7f
--- /dev/null
+++ b/examples/src/bin/qspi.rs
@@ -0,0 +1,123 @@
+#![no_std]
+#![no_main]
+#![feature(type_alias_impl_trait)]
+
+#[path = "../example_common.rs"]
+mod example_common;
+use example_common::*;
+
+use cortex_m_rt::entry;
+use embassy::flash::Flash;
+use embassy_nrf::qspi;
+use nrf52840_hal::gpio;
+
+const PAGE_SIZE: usize = 4096;
+
+// Workaround for alignment requirements.
+// Nicer API will probably come in the future.
+#[repr(C, align(4))]
+struct AlignedBuf([u8; 4096]);
+
+#[static_executor::task]
+async fn run() {
+    let p = embassy_nrf::pac::Peripherals::take().dewrap();
+
+    let port0 = gpio::p0::Parts::new(p.P0);
+
+    let pins = qspi::Pins {
+        csn: port0
+            .p0_17
+            .into_push_pull_output(gpio::Level::High)
+            .degrade(),
+        sck: port0
+            .p0_19
+            .into_push_pull_output(gpio::Level::High)
+            .degrade(),
+        io0: port0
+            .p0_20
+            .into_push_pull_output(gpio::Level::High)
+            .degrade(),
+        io1: port0
+            .p0_21
+            .into_push_pull_output(gpio::Level::High)
+            .degrade(),
+        io2: Some(
+            port0
+                .p0_22
+                .into_push_pull_output(gpio::Level::High)
+                .degrade(),
+        ),
+        io3: Some(
+            port0
+                .p0_23
+                .into_push_pull_output(gpio::Level::High)
+                .degrade(),
+        ),
+    };
+
+    let config = qspi::Config {
+        pins,
+        read_opcode: qspi::ReadOpcode::READ4IO,
+        write_opcode: qspi::WriteOpcode::PP4IO,
+        xip_offset: 0,
+        write_page_size: qspi::WritePageSize::_256BYTES,
+    };
+
+    let mut q = qspi::Qspi::new(p.QSPI, config);
+
+    let mut id = [1; 3];
+    q.custom_instruction(0x9F, &[], &mut id).await.unwrap();
+    info!("id: {:[u8]}", id);
+
+    // Read status register
+    let mut status = [0; 1];
+    q.custom_instruction(0x05, &[], &mut status).await.unwrap();
+
+    info!("status: {:?}", status[0]);
+
+    if status[0] & 0x40 == 0 {
+        status[0] |= 0x40;
+
+        q.custom_instruction(0x01, &status, &mut []).await.unwrap();
+
+        info!("enabled quad in status");
+    }
+
+    let mut buf = AlignedBuf([0u8; PAGE_SIZE]);
+
+    let pattern = |a: u32| (a ^ (a >> 8) ^ (a >> 16) ^ (a >> 24)) as u8;
+
+    for i in 0..8 {
+        info!("page {:?}: erasing... ", i);
+        q.erase(i * PAGE_SIZE).await.unwrap();
+
+        for j in 0..PAGE_SIZE {
+            buf.0[j] = pattern((j + i * PAGE_SIZE) as u32);
+        }
+
+        info!("programming...");
+        q.write(i * PAGE_SIZE, &buf.0).await.unwrap();
+    }
+
+    for i in 0..8 {
+        info!("page {:?}: reading... ", i);
+        q.read(i * PAGE_SIZE, &mut buf.0).await.unwrap();
+
+        info!("verifying...");
+        for j in 0..PAGE_SIZE {
+            assert_eq!(buf.0[j], pattern((j + i * PAGE_SIZE) as u32));
+        }
+    }
+
+    info!("done!")
+}
+
+#[entry]
+fn main() -> ! {
+    info!("Hello World!");
+
+    unsafe {
+        run.spawn().dewrap();
+        static_executor::run();
+    }
+}
diff --git a/examples/src/bin/uart.rs b/examples/src/bin/uart.rs
new file mode 100644
index 000000000..21e26e3ad
--- /dev/null
+++ b/examples/src/bin/uart.rs
@@ -0,0 +1,72 @@
+#![no_std]
+#![no_main]
+#![feature(type_alias_impl_trait)]
+
+#[path = "../example_common.rs"]
+mod example_common;
+use example_common::*;
+
+use cortex_m_rt::entry;
+use embassy::io::{AsyncBufRead, AsyncBufReadExt, AsyncWrite, AsyncWriteExt};
+use embassy_nrf::uarte;
+use futures::pin_mut;
+use nrf52840_hal::gpio;
+
+#[static_executor::task]
+async fn run() {
+    let p = embassy_nrf::pac::Peripherals::take().dewrap();
+
+    let port0 = gpio::p0::Parts::new(p.P0);
+
+    let pins = uarte::Pins {
+        rxd: port0.p0_08.into_floating_input().degrade(),
+        txd: port0
+            .p0_06
+            .into_push_pull_output(gpio::Level::Low)
+            .degrade(),
+        cts: None,
+        rts: None,
+    };
+
+    let u = uarte::Uarte::new(
+        p.UARTE0,
+        pins,
+        uarte::Parity::EXCLUDED,
+        uarte::Baudrate::BAUD115200,
+    );
+    pin_mut!(u);
+
+    info!("uarte initialized!");
+
+    u.write_all(b"Hello!\r\n").await.dewrap();
+    info!("wrote hello in uart!");
+
+    // Simple demo, reading 8-char chunks and echoing them back reversed.
+    loop {
+        info!("reading...");
+        let mut buf = [0u8; 8];
+        u.read_exact(&mut buf).await.dewrap();
+        info!("read done, got {:[u8]}", buf);
+
+        // Reverse buf
+        for i in 0..4 {
+            let tmp = buf[i];
+            buf[i] = buf[7 - i];
+            buf[7 - i] = tmp;
+        }
+
+        info!("writing...");
+        u.write_all(&buf).await.dewrap();
+        info!("write done");
+    }
+}
+
+#[entry]
+fn main() -> ! {
+    info!("Hello World!");
+
+    unsafe {
+        run.spawn().dewrap();
+        static_executor::run();
+    }
+}
diff --git a/examples/src/example_common.rs b/examples/src/example_common.rs
new file mode 100644
index 000000000..e9919153c
--- /dev/null
+++ b/examples/src/example_common.rs
@@ -0,0 +1,68 @@
+#![macro_use]
+
+use defmt_rtt as _; // global logger
+use nrf52840_hal as _;
+use panic_probe as _;
+use static_executor_cortex_m as _;
+
+pub use defmt::{info, intern};
+
+use core::sync::atomic::{AtomicUsize, Ordering};
+
+#[defmt::timestamp]
+fn timestamp() -> u64 {
+    static COUNT: AtomicUsize = AtomicUsize::new(0);
+    // NOTE(no-CAS) `timestamps` runs with interrupts disabled
+    let n = COUNT.load(Ordering::Relaxed);
+    COUNT.store(n + 1, Ordering::Relaxed);
+    n as u64
+}
+
+macro_rules! depanic {
+    ($( $i:expr ),*) => {
+        {
+            defmt::error!($( $i ),*);
+            panic!();
+        }
+    }
+}
+
+pub trait Dewrap<T> {
+    /// dewrap = defmt unwrap
+    fn dewrap(self) -> T;
+
+    /// dexpect = defmt expect
+    fn dexpect<M: defmt::Format>(self, msg: M) -> T;
+}
+
+impl<T> Dewrap<T> for Option<T> {
+    fn dewrap(self) -> T {
+        match self {
+            Some(t) => t,
+            None => depanic!("Dewrap failed: enum is none"),
+        }
+    }
+
+    fn dexpect<M: defmt::Format>(self, msg: M) -> T {
+        match self {
+            Some(t) => t,
+            None => depanic!("Unexpected None: {:?}", msg),
+        }
+    }
+}
+
+impl<T, E: defmt::Format> Dewrap<T> for Result<T, E> {
+    fn dewrap(self) -> T {
+        match self {
+            Ok(t) => t,
+            Err(e) => depanic!("Dewrap failed: {:?}", e),
+        }
+    }
+
+    fn dexpect<M: defmt::Format>(self, msg: M) -> T {
+        match self {
+            Ok(t) => t,
+            Err(e) => depanic!("Unexpected error: {:?}: {:?}", msg, e),
+        }
+    }
+}