feat: adds Etherdream device

This commit is contained in:
alban 2023-07-16 20:28:06 +02:00
parent ffba2efeed
commit 86e32e08b1
7 changed files with 190 additions and 123 deletions

View File

@ -1,4 +1,3 @@
mod helios; mod helios;
mod dummy; mod dummy;
mod etherdream; mod etherdream;
@ -10,6 +9,7 @@ use crate::device::dummy::DummyDevice;
use crate::errors::LJResult; use crate::errors::LJResult;
use crate::point::Point; use crate::point::Point;
use serde::Serialize; use serde::Serialize;
use crate::device::etherdream::EtherdreamDevice;
/* /*
self.protocol_version, self.protocol_version,
@ -27,22 +27,24 @@ self.point_count
#[derive(Debug, PartialEq, Serialize, Copy, Clone)] #[derive(Debug, PartialEq, Serialize, Copy, Clone)]
pub enum PlaybackState { pub enum PlaybackState {
IDLE = 0, IDLE = 0,
PREPARE, PREPARE = 1,
PLAYING, PLAYING = 2,
UNKNOWN = 99,
} }
impl fmt::Display for PlaybackState { impl fmt::Display for PlaybackState {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "{:?}", self) write!(f, "{:?}", self)
} }
} }
#[derive(Debug, Default )] #[derive(Debug)]
pub struct Status { pub struct Status {
pub last_traced_at: String, pub last_traced_at: String,
pub properties: Vec<String>, pub properties: Vec<String>,
pub playback_state: PlaybackState, pub playback_state: PlaybackState,
pub capacity: u16, pub capacity: u16,
pub lack: String pub lack: String,
} }
// /lstt/lasernumber etherdream last_status.playback_state (0: idle 1: prepare 2: playing) // /lstt/lasernumber etherdream last_status.playback_state (0: idle 1: prepare 2: playing)
@ -50,7 +52,7 @@ pub struct Status {
// /lack/lasernumber "a": ACK "F": Full "I": invalid. 64 or 35 for no connection. // /lack/lasernumber "a": ACK "F": Full "I": invalid. 64 or 35 for no connection.
pub trait Device { pub trait Device {
fn status(&self) -> Status; fn status(&mut self) -> Status;
fn draw( fn draw(
&mut self, &mut self,
frame: Vec<Point>, frame: Vec<Point>,
@ -63,7 +65,7 @@ pub trait Device {
pub fn device_factory(config: &Conf) -> LJResult<Box<dyn Device>> { pub fn device_factory(config: &Conf) -> LJResult<Box<dyn Device>> {
let device: Box<dyn Device> = match &config.dac { let device: Box<dyn Device> = match &config.dac {
DacFamily::Helios(conf) => Box::new(HeliosDevice::new(conf)?), DacFamily::Helios(conf) => Box::new(HeliosDevice::new(conf)?),
DacFamily::Etherdream(_conf) => todo!(), DacFamily::Etherdream( conf) => Box::new( EtherdreamDevice::new(conf)?),
DacFamily::Dummy => Box::new(DummyDevice::new()?) DacFamily::Dummy => Box::new(DummyDevice::new()?)
}; };
Ok(device) Ok(device)

View File

@ -14,7 +14,7 @@ impl DummyDevice {
} }
impl Device for DummyDevice { impl Device for DummyDevice {
fn status(&self) -> Status { fn status(&mut self) -> Status {
Status { Status {
last_traced_at: "never".to_string(), last_traced_at: "never".to_string(),
properties: vec!["foo".to_string()], properties: vec!["foo".to_string()],

View File

@ -1,51 +1,73 @@
use ether_dream::{dac}; use std::time;
use std::net::SocketAddr;
use ether_dream::dac::stream::connect;
use ether_dream::dac::Stream;
use crate::conf::EtherDreamConf; use crate::conf::EtherDreamConf;
use crate::device::{Device, Status, PlaybackState}; use crate::device::{Device, Status, PlaybackState};
use crate::errors::{LJError, LJResult}; use crate::errors::{LJError, LJResult};
use crate::point::Point; use crate::point::{Color, Point};
use chrono::Utc; use ether_dream::protocol::{DacBroadcast, DacStatus};
use log::info;
#[warn(dead_code)]
pub struct EtherdreamDevice { pub struct EtherdreamDevice {
pub conf: EtherDreamConf, pub conf: EtherDreamConf,
dac: dac, dac: DacBroadcast,
sent_points: u16, // source_address: SocketAddr,
state: PlaybackState, stream: Stream,
// sent_points: u16,
lack: String, lack: String,
last_traced_at : String last_traced_at: String,
} }
impl EtherdreamDevice { impl EtherdreamDevice {
pub fn new(conf: &EtherDreamConf) -> LJResult<Self> { pub fn new(conf: &EtherDreamConf) -> LJResult<Self> {
let id = conf.id; let (dac, _source_address, stream) = EtherdreamDevice::get_dac(conf)?;
// let (dac, source_address) = EtherdreamDevice::get_dac(conf)?;
// // Todo : use the config Ok(Self {
// let (dac_broadcast, source_addr) = ether_dream::recv_dac_broadcasts() conf: (*conf).clone(),
// .expect("failed to bind to UDP socket") dac,
// .filter_map(Result::ok) // source_address,
// .next() stream,
// .unwrap(); // sent_points: 0,
// let mac_address = dac::MacAddress(dac_broadcast.mac_address); lack: "".to_string(),
// last_traced_at: "1985-04-12T23:20:50.52Z".to_string(),
// println!( })
// "Discovered DAC \"{}\" at \"{}\"! Connecting...", }
// mac_address, source_addr pub fn get_dac(conf: &EtherDreamConf) -> LJResult<(DacBroadcast, SocketAddr, Stream)> {
// ); let ip = &conf.ip;
let dac_broadcast = ether_dream::recv_dac_broadcasts()?;
dac_broadcast.set_nonblocking(true)?;
dac_broadcast.set_timeout(Some(time::Duration::new(60, 30)))?;
let broadcast = dac_broadcast
.filter_map(|result| {
info!("Received new broadcast {:?}", result);
if result.is_err() { return None; }
let (dac, source_addr) = result.unwrap();
if source_addr.is_ipv6() { return None; }
if source_addr.ip().to_string() != ip.clone() { return None; }
info!("Valid broadcast");
Some(Ok((dac, source_addr)))
})
.next()
.ok_or(LJError::EtherdreamError("Failed to receive broadcast".to_string()))?;
match broadcast {
Err(err) => {
Err(Box::new(LJError::EtherdreamConnectError(err)))
}
Ok((dac, source_addr)) => {
let stream = EtherdreamDevice::get_tcp_stream(&dac, &source_addr)?;
Ok((dac, source_addr, stream))
}
}
}
pub fn get_tcp_stream(dac: &DacBroadcast, source_address: &SocketAddr) -> LJResult<Stream> {
// Establish the TCP connection. // Establish the TCP connection.
let mut stream = dac::stream::connect(&dac_broadcast, source_addr.ip().clone()).unwrap(); let mut stream = connect(dac, source_address.ip())?;
// If we want to create an animation (in our case a moving sine wave) we need a frame rate. // Prepare stream
let frames_per_second = 60.0;
// Lets use the DAC at an eighth the maximum scan rate.
let points_per_second = stream.dac().max_point_rate / 32;
// Determine the number of points per frame given our target frame and point rates.
let points_per_frame = (points_per_second as f32 / frames_per_second) as u16;
// Prepare the DAC's playback engine and await the repsonse.
stream stream
.queue_commands() .queue_commands()
.prepare_stream() .prepare_stream()
@ -59,50 +81,67 @@ impl EtherdreamDevice {
); );
}); });
println!("Ready for playback!"); Ok(stream)
}
Ok(Self {
conf: (*conf).clone(), pub fn check_tcp_stream(&mut self) -> LJResult<()> {
dac, // todo Reinit stream if needed
sent_points: 0, // self.stream = EtherdreamDevice::get_tcp_stream(&self.dac, &self.source_address)?
state: PlaybackState::PREPARE, Ok(())
lack: "".to_string(), }
last_traced_at: "1985-04-12T23:20:50.52Z".to_string()
}) // Determine the number of points needed to fill the DAC.
fn points_to_generate(&self) -> usize {
self.dac.buffer_capacity as usize - 1 - self.dac.dac_status.buffer_fullness as usize
} }
} }
impl Device for EtherdreamDevice { impl Device for EtherdreamDevice {
fn status(&self) -> Status { fn status(&mut self) -> Status {
let _ = self.check_tcp_stream();
// "a": ACK "F": Full "I": invalid. 64 or 35 for no connection.
let playback_state = match self.dac.dac_status.playback_state {
DacStatus::PLAYBACK_IDLE => PlaybackState::IDLE,
DacStatus::PLAYBACK_PREPARED => PlaybackState::PREPARE,
DacStatus::PLAYBACK_PLAYING => PlaybackState::PLAYING,
_ => PlaybackState::UNKNOWN
};
let lack = self.lack.clone();
Status { Status {
last_traced_at: self.last_traced_at.clone(), last_traced_at: self.last_traced_at.clone(),
properties: vec!["foo".to_string()], properties: vec!["foo".to_string()],
playback_state: self.state, playback_state,
capacity: self.sent_points, capacity: self.dac.dac_status.buffer_fullness,
lack, lack: String::from(&self.lack),
} }
} }
fn draw(&mut self, fn draw(&mut self,
line: Vec<Point>, line: Vec<Point>,
speed: u32, _speed: u32,
) -> LJResult<()> { ) -> LJResult<()> {
self.state = PlaybackState::IDLE; let n_points = self.points_to_generate();
while let Ok(DeviceStatus::NotReady) = self.dac.status() {} self.stream
self.state = PlaybackState::PLAYING; .queue_commands()
.data(line.into_iter().map(|point| point.into()).take(n_points))
let points: Vec<helios_dac::Point> = line.into_iter().map(|p| p.into()).collect(); .submit()?;
let frame = Frame::new(speed, points.clone());
self.dac.write_frame(frame.clone())?;
self.sent_points = points.len() as u16;
self.last_traced_at = Utc::now().to_rfc3339();
Ok(()) Ok(())
} }
fn stop(&mut self) -> LJResult<()> { fn stop(&mut self) -> LJResult<()> {
self.dac.stop()?; self.stream
.queue_commands()
.stop()
.submit()
.expect("err occurred when submitting STOP command and listening for response");
Ok(()) Ok(())
} }
fn grid(&mut self) -> Vec<Point> {
vec!(
Point { x: 0.0, y: 0.0, color: Color { r: 255, g: 255, b: 255 } }
)
}
} }

View File

@ -46,7 +46,7 @@ impl HeliosDevice {
} }
impl Device for HeliosDevice { impl Device for HeliosDevice {
fn status(&self) -> Status { fn status(&mut self) -> Status {
let lack = self.lack.clone(); let lack = self.lack.clone();
Status { Status {
last_traced_at: self.last_traced_at.clone(), last_traced_at: self.last_traced_at.clone(),

View File

@ -1,5 +1,5 @@
use std::error::Error; use std::error::Error;
use std::fmt; use std::{fmt, io};
use redis::RedisError; use redis::RedisError;
use config::ConfigError; use config::ConfigError;
@ -10,7 +10,9 @@ pub enum LJError {
Config(ConfigError), Config(ConfigError),
RedisConnect(RedisError), RedisConnect(RedisError),
HeliosDeviceMissing, HeliosDeviceMissing,
BadEDH BadEDH,
EtherdreamError(String),
EtherdreamConnectError(io::Error),
} }
impl fmt::Display for LJError { impl fmt::Display for LJError {
@ -20,17 +22,22 @@ impl fmt::Display for LJError {
match self { match self {
Config(err) => { Config(err) => {
write!(f, "unable to load config file: {err}") write!(f, "unable to load config file: {err}")
}, }
RedisConnect(err) => { RedisConnect(err) => {
write!(f, "unable to connect to redis server: {err}") write!(f, "unable to connect to redis server: {err}")
}, }
HeliosDeviceMissing => { HeliosDeviceMissing => {
write!(f, "helios device not found") write!(f, "helios device not found")
}, }
BadEDH => { BadEDH => {
write!(f, "EDH matrix is not a 3x3 matrix") write!(f, "EDH matrix is not a 3x3 matrix")
} }
EtherdreamError(msg) => {
write!(f, "Unexpected Etherdream device error: {msg}")
}
EtherdreamConnectError(err) => {
write!(f, "Failed to retrieve Etherdream device: {err}")
}
} }
} }
} }

View File

@ -114,7 +114,7 @@ fn run_all() -> LJResult<()> {
// 4 : Resampler Change (longs and shorts lsteps) // 4 : Resampler Change (longs and shorts lsteps)
// 5 : Client Key Change = reread redis key /clientkey // 5 : Client Key Change = reread redis key /clientkey
// 8 : color balance change = reread redis keys /red /green /blue // 8 : color balance change = reread redis keys /red /green /blue
// 9 : poweroff LJ world_state.intensity = rs.get_int("intensity")? // 9 : poweroff LJ
info!("Order: {:?}", order); info!("Order: {:?}", order);
} }
} }
@ -152,17 +152,15 @@ fn get_next_frame(
// Handle the grid case // Handle the grid case
let mut line: Vec<Point>; let mut line: Vec<Point> = if world_state.draw_grid {
if world_state.draw_grid { world_state.grid.clone()
line = world_state.grid.clone();
} else { } else {
let redis_line = rs.get_line(&format_key)?; let redis_line = rs.get_line(&format_key)?;
line = redis_line.into_iter() redis_line.into_iter()
.map(|tpl| tpl.into()) .map(|tpl| tpl.into())
.collect(); .collect()
}; };
for transformer in transformers { for transformer in transformers {
line = transformer.apply(&line, world_state); line = transformer.apply(&line, world_state);
} }

View File

@ -1,3 +1,5 @@
use ether_dream::protocol::DacPoint;
#[derive(Debug, Clone, Copy, Default, PartialEq)] #[derive(Debug, Clone, Copy, Default, PartialEq)]
pub struct Point { pub struct Point {
pub x: f32, pub x: f32,
@ -41,3 +43,22 @@ impl From<Point> for helios_dac::Point {
} }
} }
} }
impl From<Point> for DacPoint {
fn from(pt: Point) -> DacPoint {
let control = 0;
let (u1, u2) = (0, 0);
let i = 255;
DacPoint {
control,
x: pt.x as i16,
y: pt.y as i16,
i,
r: pt.color.r.into(),
g: pt.color.g.into(),
b: pt.color.b.into(),
u1,
u2,
}
}
}