mirror of
https://github.com/italicsjenga/rp-hal-boards.git
synced 2024-10-17 14:31:31 +11:00
42e929d7e1
* Use rp2040-hal in all example (possibly through their bsp) Some of the examples were using the cortex_m_rt::entry method which misses the device specific spinlock re-initialisation. This commits makes the usage more consistent by using rp2040_hal exported macro as the only `entry` method used across examples.
108 lines
2.8 KiB
Rust
108 lines
2.8 KiB
Rust
//! # Rainbow Example for the Thing Plus RP2040
|
|
//!
|
|
//! Runs a rainbow-effect colour wheel on the on-board LED.
|
|
//!
|
|
//! Uses the `ws2812_pio` driver to control the LED, which in turns uses the
|
|
//! RP2040's PIO block.
|
|
|
|
#![no_std]
|
|
#![no_main]
|
|
|
|
use core::iter::once;
|
|
use embedded_hal::timer::CountDown;
|
|
use embedded_time::duration::Extensions;
|
|
use panic_halt as _;
|
|
|
|
use smart_leds::{brightness, SmartLedsWrite, RGB8};
|
|
use sparkfun_thing_plus_rp2040::{
|
|
hal::{
|
|
clocks::{init_clocks_and_plls, Clock},
|
|
pac,
|
|
pio::PIOExt,
|
|
timer::Timer,
|
|
watchdog::Watchdog,
|
|
Sio,
|
|
},
|
|
XOSC_CRYSTAL_FREQ,
|
|
};
|
|
use ws2812_pio::Ws2812;
|
|
|
|
/// Entry point to our bare-metal application.
|
|
///
|
|
/// The `#[sparkfun_thing_plus_rp2040::entry]` macro ensures the Cortex-M start-up code calls this
|
|
/// function as soon as all global variables are initialised.
|
|
///
|
|
/// The function configures the RP2040 peripherals, then the LED, then runs
|
|
/// the colour wheel in an infinite loop.
|
|
#[sparkfun_thing_plus_rp2040::entry]
|
|
fn main() -> ! {
|
|
// Configure the RP2040 peripherals
|
|
|
|
let mut pac = pac::Peripherals::take().unwrap();
|
|
let mut watchdog = Watchdog::new(pac.WATCHDOG);
|
|
|
|
let clocks = init_clocks_and_plls(
|
|
XOSC_CRYSTAL_FREQ,
|
|
pac.XOSC,
|
|
pac.CLOCKS,
|
|
pac.PLL_SYS,
|
|
pac.PLL_USB,
|
|
&mut pac.RESETS,
|
|
&mut watchdog,
|
|
)
|
|
.ok()
|
|
.unwrap();
|
|
|
|
let sio = Sio::new(pac.SIO);
|
|
|
|
let pins = sparkfun_thing_plus_rp2040::Pins::new(
|
|
pac.IO_BANK0,
|
|
pac.PADS_BANK0,
|
|
sio.gpio_bank0,
|
|
&mut pac.RESETS,
|
|
);
|
|
|
|
let timer = Timer::new(pac.TIMER, &mut pac.RESETS);
|
|
let mut delay = timer.count_down();
|
|
|
|
// Configure the addressable LED
|
|
let (mut pio, sm0, _, _, _) = pac.PIO0.split(&mut pac.RESETS);
|
|
let mut ws = Ws2812::new(
|
|
pins.ws2812.into_mode(),
|
|
&mut pio,
|
|
sm0,
|
|
clocks.peripheral_clock.freq(),
|
|
timer.count_down(),
|
|
);
|
|
|
|
// Infinite colour wheel loop
|
|
|
|
let mut n: u8 = 128;
|
|
loop {
|
|
ws.write(brightness(once(wheel(n)), 32)).unwrap();
|
|
n = n.wrapping_add(1);
|
|
|
|
delay.start(25.milliseconds());
|
|
let _ = nb::block!(delay.wait());
|
|
}
|
|
}
|
|
|
|
/// Convert a number from `0..=255` to an RGB color triplet.
|
|
///
|
|
/// The colours are a transition from red, to green, to blue and back to red.
|
|
fn wheel(mut wheel_pos: u8) -> RGB8 {
|
|
wheel_pos = 255 - wheel_pos;
|
|
if wheel_pos < 85 {
|
|
// No green in this sector - red and blue only
|
|
(255 - (wheel_pos * 3), 0, wheel_pos * 3).into()
|
|
} else if wheel_pos < 170 {
|
|
// No red in this sector - green and blue only
|
|
wheel_pos -= 85;
|
|
(0, wheel_pos * 3, 255 - (wheel_pos * 3)).into()
|
|
} else {
|
|
// No blue in this sector - red and green only
|
|
wheel_pos -= 170;
|
|
(wheel_pos * 3, 255 - (wheel_pos * 3), 0).into()
|
|
}
|
|
}
|