You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ledstrip-vm/src/ledstrip_controller.rs

85 lines
2.0 KiB
Rust

use std::io;
use std::io::Write;
use std::net::TcpStream;
const STATE_COMMAND_PREFIX: u8 = 0x71;
const PROGRAM_COMMAND_PREFIX: u8 = 0x61;
pub enum StateStripCommand {
On = 0x23,
Off = 0x24,
}
pub enum ProgramStripCommand {
SevenCrossFade = 0x25,
RedGradual = 0x26,
GreenGradual = 0x27,
BlueGradual = 0x28,
WhiteGradual = 0x2c,
RedGreenCross = 0x2d,
RedBlueCross = 0x2e,
GreenBlueCross = 0x2f,
SevenStrobe = 0x30,
RedStrobe = 0x31,
GreenStrobe = 0x32,
BlueStrobe = 0x33,
WhiteStrobe = 0x37,
SevenJumping = 0x38,
}
pub struct LedStripController {
stream: TcpStream,
pub r: u8,
pub g: u8,
pub b: u8,
}
impl LedStripController {
pub fn new(ip: &str, port: usize) -> io::Result<Self> {
let stream = TcpStream::connect(format!("{}:{}", ip, port))?;
Ok(Self {
stream,
r: 0,
g: 0,
b: 0,
})
}
/// Send an rgb color to the led strip
pub fn send_rgb_color(&mut self, r: u8, g: u8, b: u8) -> io::Result<()> {
self.r = r;
self.g = g;
self.b = b;
let message = create_message(&[0x31, r, g, b, 0xf0]);
self.stream.write(&message)?;
Ok(())
}
/// Sets the state of the strip to either on or off
pub fn set_state(&mut self, cmd: StateStripCommand) -> io::Result<()> {
self.stream
.write(&create_message(&[STATE_COMMAND_PREFIX, cmd as u8]))?;
Ok(())
}
/// Sends a strip command with a specified speed
/// that is one of 0x01 0x06, 0x10, 0x1c
pub fn send_command(&mut self, cmd: ProgramStripCommand, speed: u8) -> io::Result<()> {
self.stream
.write(&create_message(&[PROGRAM_COMMAND_PREFIX, cmd as u8, speed]))?;
Ok(())
}
}
/// Create a message for the led strip
fn create_message(data: &[u8]) -> Vec<u8> {
let mut data = data.clone().to_vec();
data.append(&mut vec![0x0f]);
data.push(data.iter().sum::<u8>() & 255);
data
}