leds/src/main.rs

57 lines
1.7 KiB
Rust
Raw Normal View History

2021-12-23 19:34:26 +00:00
pub mod util;
use util::lights::*;
2021-12-26 18:14:03 +00:00
use util::pattern::*;
2021-12-29 01:10:29 +00:00
use util::webserver::*;
2021-12-29 01:53:30 +00:00
use std::sync::{Arc, RwLock, Mutex};
2021-12-24 23:54:53 +00:00
use rs_ws281x::{RawColor, Controller, ControllerBuilder, ChannelBuilder, StripType};
use std::{thread, time};
2021-12-23 19:34:26 +00:00
pub const LED_SIZE: usize = 450; //450
2021-12-29 00:51:04 +00:00
pub const BRIGHTNESS: u8 = 100;
2021-12-28 22:37:02 +00:00
pub const LOOP_WAIT: u64 = 250;
2021-12-23 19:34:26 +00:00
fn main() {
2021-12-24 18:29:57 +00:00
let p: RawColor = [0, 0, 0, 0];
2021-12-24 16:39:02 +00:00
let lock = Arc::new(RwLock::new([p; LED_SIZE]));
2021-12-29 02:45:19 +00:00
let pattern: Arc<Mutex<Vec<Box<dyn Pattern>>>> = Arc::new(Mutex::new(Vec::new()));
2021-12-28 20:27:23 +00:00
let read = Arc::clone(&lock);
2021-12-28 22:40:46 +00:00
//light management
2021-12-01 01:27:28 +00:00
thread::spawn(move || {
let mut controller: Controller = ControllerBuilder::new()
.channel(0, ChannelBuilder::new()
2021-12-24 23:54:53 +00:00
.pin(18)
2021-12-01 01:27:28 +00:00
.count(i32::try_from(LED_SIZE).unwrap())
2021-12-24 23:54:53 +00:00
.strip_type(StripType::Ws2812)
.brightness(BRIGHTNESS)
.build()
)
2021-12-01 00:55:08 +00:00
.build()
.expect("Could not construct LED Controller.");
2021-12-01 01:27:28 +00:00
loop {
2021-12-28 20:27:23 +00:00
let lights = read.read().expect("Could not read array lock.");
run_lights(&mut controller, &lights).expect("Error running lights controller.");
2021-12-01 01:27:28 +00:00
}
2021-12-23 19:34:26 +00:00
});
2021-12-28 22:37:02 +00:00
2021-12-29 01:53:30 +00:00
let pat_clone = Arc::clone(&pattern);
2021-12-29 02:10:00 +00:00
thread::spawn(move || loop {
2021-12-29 01:53:30 +00:00
websocket(&pat_clone);
});
2021-12-28 22:37:02 +00:00
2021-12-28 22:40:46 +00:00
//pattern management
2021-12-29 02:45:00 +00:00
let mut ticks = 0;
loop {
2021-12-28 22:37:02 +00:00
thread::sleep(time::Duration::from_millis(LOOP_WAIT));
2021-12-29 00:51:04 +00:00
let c_lock = Arc::clone(&lock);
let mut lights = c_lock.write().unwrap();
2021-12-29 02:45:19 +00:00
let data = pattern.lock().unwrap();
for p in &*data {
2021-12-29 00:51:04 +00:00
//let c_lock = Arc::clone(&lock);
//let mut lights = c_lock.write().unwrap();
2021-12-29 02:45:00 +00:00
p.execute(&mut lights, ticks);
2021-12-28 22:25:59 +00:00
}
2021-12-29 02:46:48 +00:00
ticks+= 1;
2021-12-23 19:34:26 +00:00
}
}