52 lines
1.7 KiB
Rust
52 lines
1.7 KiB
Rust
pub mod util;
|
|
|
|
use util::lights::*;
|
|
use util::pattern::*;
|
|
use std::sync::{Arc, RwLock};
|
|
use rs_ws281x::{RawColor, Controller, ControllerBuilder, ChannelBuilder, StripType};
|
|
use std::{thread, time};
|
|
|
|
pub const LED_SIZE: usize = 450; //450
|
|
pub const BRIGHTNESS: u8 = 150;
|
|
|
|
fn main() {
|
|
let p: RawColor = [0, 0, 0, 0];
|
|
let lock = Arc::new(RwLock::new([p; LED_SIZE]));
|
|
let pattern: Arc<RwLock<Vec<Box<dyn Pattern>>>> = Arc::new(RwLock::new(Vec::new()));
|
|
let read = Arc::clone(&lock);
|
|
thread::spawn(move || {
|
|
let mut controller: Controller = ControllerBuilder::new()
|
|
.channel(0, ChannelBuilder::new()
|
|
.pin(18)
|
|
.count(i32::try_from(LED_SIZE).unwrap())
|
|
.strip_type(StripType::Ws2812)
|
|
.brightness(BRIGHTNESS)
|
|
.build()
|
|
)
|
|
.build()
|
|
.expect("Could not construct LED Controller.");
|
|
loop {
|
|
let lights = read.read().expect("Could not read array lock.");
|
|
run_lights(&mut controller, &lights).expect("Error running lights controller.");
|
|
}
|
|
});
|
|
let mul_string = format_multiline("unit");
|
|
let res: Result<Vec<Box<dyn Pattern>>, ParseError> = mul_string.iter()
|
|
.map(|x: &Vec<String>| parse_line((*x).clone()))
|
|
.collect();
|
|
if res.is_ok() {
|
|
let mut p = pattern.write().unwrap();
|
|
*p = res.unwrap();
|
|
}
|
|
let lock_c = Arc::clone(&lock);
|
|
let mut lights = lock_c.write().unwrap();
|
|
loop {
|
|
thread::sleep(time::Duration::from_millis(250));
|
|
let pattern_c = Arc::clone(&pattern);
|
|
let v = pattern_c.read().expect("could not read pattern");
|
|
(*v).iter().for_each(move |x: &Box<dyn Pattern>| {
|
|
x.execute(&mut lights);
|
|
});
|
|
}
|
|
}
|