embassy/embassy-nrf-examples/src/bin/qspi.rs

#![no_std]
#![no_main]
#![feature(min_type_alias_impl_trait)]
#![feature(impl_trait_in_bindings)]
#![feature(type_alias_impl_trait)]

#[path = "../example_common.rs"]
mod example_common;
use example_common::*;

use cortex_m_rt::entry;
use defmt::{assert_eq, panic};
use futures::pin_mut;
use nrf52840_hal::gpio;

use embassy::executor::{task, Executor};
use embassy::traits::flash::Flash;
use embassy::util::Forever;
use embassy_nrf::{interrupt, qspi};

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]);

#[task]
async fn run() {
    let p = unwrap!(embassy_nrf::pac::Peripherals::take());

    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,
        deep_power_down: None,
    };

    let irq = interrupt::take!(QSPI);
    let q = qspi::Qspi::new(p.QSPI, irq, config);
    pin_mut!(q);

    let mut id = [1; 3];
    q.as_mut()
        .custom_instruction(0x9F, &[], &mut id)
        .await
        .unwrap();
    info!("id: {}", id);

    // Read status register
    let mut status = [0; 1];
    q.as_mut()
        .custom_instruction(0x05, &[], &mut status)
        .await
        .unwrap();

    info!("status: {:?}", status[0]);

    if status[0] & 0x40 == 0 {
        status[0] |= 0x40;

        q.as_mut()
            .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.as_mut().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.as_mut().write(i * PAGE_SIZE, &buf.0).await.unwrap();
    }

    for i in 0..8 {
        info!("page {:?}: reading... ", i);
        q.as_mut().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!")
}

static EXECUTOR: Forever<Executor> = Forever::new();

#[entry]
fn main() -> ! {
    info!("Hello World!");

    let executor = EXECUTOR.put(Executor::new());
    executor.run(|spawner| {
        unwrap!(spawner.spawn(run()));
    });
}