57 lines
1.7 KiB
Rust
57 lines
1.7 KiB
Rust
pub mod util;
|
|
|
|
use util::lights::*;
|
|
use util::pattern::*;
|
|
use util::webserver::*;
|
|
use std::sync::{Arc, RwLock, Mutex};
|
|
use rs_ws281x::{RawColor, Controller, ControllerBuilder, ChannelBuilder, StripType};
|
|
use std::{thread, time};
|
|
|
|
pub const LED_SIZE: usize = 450; //450
|
|
pub const BRIGHTNESS: u8 = 100;
|
|
pub const LOOP_WAIT: u64 = 250;
|
|
|
|
fn main() {
|
|
let p: RawColor = [0, 0, 0, 0];
|
|
let lock = Arc::new(RwLock::new([p; LED_SIZE]));
|
|
let pattern: Arc<Mutex<Vec<Box<dyn Pattern>>>> = Arc::new(Mutex::new(Vec::new()));
|
|
let read = Arc::clone(&lock);
|
|
//light management
|
|
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 pat_clone = Arc::clone(&pattern);
|
|
thread::spawn(move || loop {
|
|
websocket(&pat_clone);
|
|
});
|
|
|
|
//pattern management
|
|
let mut ticks = 0;
|
|
loop {
|
|
thread::sleep(time::Duration::from_millis(LOOP_WAIT));
|
|
let c_lock = Arc::clone(&lock);
|
|
let mut lights = c_lock.write().unwrap();
|
|
let data = pattern.lock().unwrap();
|
|
for p in &*data {
|
|
//let c_lock = Arc::clone(&lock);
|
|
//let mut lights = c_lock.write().unwrap();
|
|
p.execute(&mut lights, ticks);
|
|
}
|
|
ticks+= 1;
|
|
}
|
|
}
|