From baf149ef8af6ebb0bb9d344b6c6368fe1901dc44 Mon Sep 17 00:00:00 2001 From: Lapin Raving Date: Wed, 23 Aug 2023 13:13:53 +0200 Subject: [PATCH] humm --- examples/populate_redis.rs | 12 +- examples/simple_client.rs | 35 ++-- src/conf.rs | 113 ++++++----- src/device.rs | 56 +++--- src/device/dummy.rs | 59 +++--- src/device/etherdream.rs | 250 +++++++++++------------ src/device/helios.rs | 213 ++++++++++++-------- src/errors.rs | 68 +++---- src/lib.rs | 4 +- src/main.rs | 279 ++++++++++++-------------- src/point.rs | 114 +++++++---- src/redis_ctrl.rs | 202 ++++++++++--------- src/transformer.rs | 22 +- src/transformer/angle_optimisation.rs | 86 ++++++++ src/transformer/flip_horizontal.rs | 23 ++- src/transformer/flip_vertical.rs | 23 ++- src/transformer/grid.rs | 66 +++--- src/transformer/homography.rs | 98 +++++---- src/transformer/replicate.rs | 37 ++-- src/transformer/rotate.rs | 33 ++- src/transformer/translate.rs | 25 +-- src/worldstate.rs | 66 +++--- tests/test_conf.rs | 2 +- 23 files changed, 1030 insertions(+), 856 deletions(-) create mode 100644 src/transformer/angle_optimisation.rs diff --git a/examples/populate_redis.rs b/examples/populate_redis.rs index cd95e77..a25d562 100644 --- a/examples/populate_redis.rs +++ b/examples/populate_redis.rs @@ -1,7 +1,6 @@ /// /// $ cargo run --example populate_redis /// - use redis::{ //RedisResult, Client, @@ -13,10 +12,13 @@ fn do_something() -> redis::RedisResult<()> { let client = Client::open("redis://127.0.0.1/")?; let mut con: Connection = client.get_connection()?; - let _ = con.set("/clientkey", "/pl/0/")?; - let _ = con.set("/EDH/0", "[[1.0, 0.0, 0.0],\n [ 0.0, 1.0, 0.0],\n [ 0.0, 0.0, 1.0]]")?; - let _ = con.set("/kpps/0", "5000")?; - let _ = con.set("/intensity/0", "255")?; + let _ = con.set("/clientkey", "/pl/0/")?; + let _ = con.set( + "/EDH/0", + "[[1.0, 0.0, 0.0],\n [ 0.0, 1.0, 0.0],\n [ 0.0, 0.0, 1.0]]", + )?; + let _ = con.set("/kpps/0", "5000")?; + let _ = con.set("/intensity/0", "255")?; Ok(()) } fn main() { diff --git a/examples/simple_client.rs b/examples/simple_client.rs index c5fc083..eb30062 100644 --- a/examples/simple_client.rs +++ b/examples/simple_client.rs @@ -1,7 +1,6 @@ /// /// $ cargo run --example simple_client /// - use redis::{ //RedisResult, Client, @@ -15,28 +14,24 @@ fn do_something() -> redis::RedisResult<()> { let client = Client::open("redis://127.0.0.1/")?; let mut con: Connection = client.get_connection()?; let start = Instant::now(); - + loop { - let elapsed = start.elapsed(); - let time = 60.0 * elapsed.as_millis() as f32 / 1000.0; + let elapsed = start.elapsed(); + let time = 60.0 * elapsed.as_millis() as f32 / 1000.0; - let mut v : Vec<(f32,f32,u32)> = vec![]; - - for i in 0..128 { - let a = (time + i as f32) / 128.0 * std::f32::consts::PI * 2.0; - let r = 1200.0 + (a*5.0).cos() * (500.0 * (time/5.0).cos()); + let mut v: Vec<(f32, f32, u32)> = vec![]; - let x = a.cos() * r; - let y = a.sin() * r; - let col = if i % 8 < 4 { - 0x000000ff - } else { - 0x00ff0000 - }; - v.push((x,y,col)); - } - // println!("{:?}", v); - let _ = con.set("/pl/0/0", format!("{:?}", v))?; + for i in 0..128 { + let a = (time + i as f32) / 128.0 * std::f32::consts::PI * 2.0; + let r = 1200.0 + (a * 5.0).cos() * (500.0 * (time / 5.0).cos()); + + let x = a.cos() * r; + let y = a.sin() * r; + let col = if i % 8 < 4 { 0x000000ff } else { 0x00ff0000 }; + v.push((x, y, col)); + } + // println!("{:?}", v); + let _ = con.set("/pl/0/0", format!("{:?}", v))?; } // Ok(()) } diff --git a/src/conf.rs b/src/conf.rs index 2996c7d..da74aff 100644 --- a/src/conf.rs +++ b/src/conf.rs @@ -1,7 +1,7 @@ -use config::Config; -use serde::{Serialize,Deserialize}; -use crate::errors::{LJError,LJResult}; +use crate::errors::{LJError, LJResult}; use crate::transformer; +use config::Config; +use serde::{Deserialize, Serialize}; #[derive(Serialize, Deserialize, Debug, Clone)] pub struct Conf { @@ -10,89 +10,100 @@ pub struct Conf { pub redis_url: String, pub dac: DacFamily, #[serde(default)] - pub transformers: Vec + pub transformers: Vec, } #[derive(Serialize, Deserialize, Debug, Clone)] pub enum DacFamily { - #[serde(rename = "helios")] + #[serde(rename = "helios")] Helios(HeliosConf), - #[serde(rename = "etherdream")] + #[serde(rename = "etherdream")] Etherdream(EtherDreamConf), - #[serde(rename = "dummy")] + #[serde(rename = "dummy")] Dummy, } #[derive(Serialize, Deserialize, Debug, Clone)] pub struct HeliosConf { - pub id: u8 + pub id: u8, } #[derive(Serialize, Deserialize, Debug, Clone)] pub struct EtherDreamConf { - pub ip: String + pub ip: String, } #[derive(Serialize, Deserialize, Debug, Clone)] pub enum TransformConf { - #[serde(rename = "translate")] + #[serde(rename = "translate")] Translate(transformer::Translate), - #[serde(rename = "replicate")] + #[serde(rename = "replicate")] Replicate(transformer::Replicate), - #[serde(rename = "rotate")] + #[serde(rename = "rotate")] Rotate(transformer::Rotate), - #[serde(rename = "flip_horizontal")] + #[serde(rename = "flip_horizontal")] FlipH(transformer::FlipHorizontal), - #[serde(rename = "flip_vertical")] - FlipV(transformer::FlipVertical), - #[serde(rename = "grid")] + #[serde(rename = "flip_vertical")] + FlipV(transformer::FlipVertical), + #[serde(rename = "grid")] Grid(transformer::Grid), - #[serde(rename = "homography")] - Homography(transformer::Homography) + #[serde(rename = "homography")] + Homography(transformer::Homography), + #[serde(rename = "angle_correction")] + AngleOptimisation(transformer::AngleOptimisation), } - impl Conf { pub fn new(path: &str) -> LJResult { - let settings = Config::builder() + let settings = Config::builder() .add_source(config::File::with_name(path)) - .build()?; - - let conf : Conf = settings.try_deserialize().map_err(LJError::Config)?; - Ok(conf) + .build()?; + + dbg!("path:", path); + //println!("plop"); + dbg!(settings.clone()); + let conf: Conf = settings.try_deserialize().map_err(LJError::Config)?; + + //println!("plum"); + Ok(conf) } + //println!("plop"); + pub fn get_transformers(&self) -> Vec> { - let mut v = vec![]; - for t in &self.transformers { - let t : Box = match t { - TransformConf::Translate(t) => Box::new(*t), - TransformConf::Replicate(r) => Box::new(*r), - TransformConf::Rotate(r) => Box::new(*r), - TransformConf::FlipH(r) => Box::new(*r), - TransformConf::FlipV(r) => Box::new(*r), - TransformConf::Grid(r) => Box::new(*r), - TransformConf::Homography(r) => Box::new(*r), - }; - v.push(t); - } - v + let mut v = vec![]; + for t in &self.transformers { + let t: Box = match t { + TransformConf::Translate(t) => Box::new(*t), + TransformConf::Replicate(r) => Box::new(*r), + TransformConf::Rotate(r) => Box::new(*r), + TransformConf::FlipH(r) => Box::new(*r), + TransformConf::FlipV(r) => Box::new(*r), + TransformConf::Grid(r) => Box::new(*r), + TransformConf::Homography(r) => Box::new(*r), + TransformConf::AngleOptimisation(r) => Box::new(*r), + }; + v.push(t); + } + v } #[allow(dead_code)] pub fn dump() { - let conf = Conf { - laser_id: 0, - debug: true, - redis_url: "redis://127.0.0.1:6379/".to_string(), - dac: DacFamily::Helios(HeliosConf { id: 0 }), - transformers: vec![ - TransformConf::Translate(transformer::Translate { x: 2000.0, - y: 2000.0 } ), - TransformConf::Replicate(transformer::Replicate::Until(48)) - ] - }; - let s = toml::to_string(&conf).unwrap(); - println!("{}", s); + let conf = Conf { + laser_id: 0, + debug: true, + redis_url: "redis://127.0.0.1:6379/".to_string(), + dac: DacFamily::Helios(HeliosConf { id: 0 }), + transformers: vec![ + TransformConf::Translate(transformer::Translate { + x: 2000.0, + y: 2000.0, + }), + TransformConf::Replicate(transformer::Replicate::Until(48)), + ], + }; + let s = toml::to_string(&conf).unwrap(); + println!("{}", s); } } diff --git a/src/device.rs b/src/device.rs index fe28f74..c5cdd9f 100644 --- a/src/device.rs +++ b/src/device.rs @@ -1,15 +1,15 @@ -mod helios; mod dummy; mod etherdream; +mod helios; -use std::fmt; use crate::conf::{Conf, DacFamily /*EtherDreamConf, HeliosConf*/}; -use crate::device::helios::HeliosDevice; use crate::device::dummy::DummyDevice; +use crate::device::etherdream::EtherdreamDevice; +use crate::device::helios::HeliosDevice; use crate::errors::LJResult; use crate::point::Point; use serde::Serialize; -use crate::device::etherdream::EtherdreamDevice; +use std::fmt; /* self.protocol_version, @@ -26,25 +26,25 @@ self.point_count #[repr(u8)] #[derive(Debug, PartialEq, Serialize, Copy, Clone)] pub enum PlaybackState { - IDLE = 0, - PREPARE = 1, - PLAYING = 2, - UNKNOWN = 99, + IDLE = 0, + PREPARE = 1, + PLAYING = 2, + UNKNOWN = 99, } impl fmt::Display for PlaybackState { - fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { - write!(f, "{:?}", self) - } + fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result { + write!(f, "{:?}", self) + } } #[derive(Debug)] pub struct Status { - pub last_traced_at: String, - pub properties: Vec, - pub playback_state: PlaybackState, - pub capacity: u16, - pub lack: String, + pub last_traced_at: String, + pub properties: Vec, + pub playback_state: PlaybackState, + pub capacity: u16, + pub lack: String, } // /lstt/lasernumber etherdream last_status.playback_state (0: idle 1: prepare 2: playing) @@ -52,21 +52,17 @@ pub struct Status { // /lack/lasernumber "a": ACK "F": Full "I": invalid. 64 or 35 for no connection. pub trait Device { - fn status(&mut self) -> Status; - fn draw( - &mut self, - frame: Vec, - speed: u32, - ) -> LJResult<()>; - fn stop(&mut self) -> LJResult<()>; - fn grid(&mut self) -> Vec; + fn status(&mut self) -> Status; + fn draw(&mut self, frame: Vec, speed: u32) -> LJResult<()>; + fn stop(&mut self) -> LJResult<()>; + fn grid(&mut self) -> Vec; } pub fn device_factory(config: &Conf) -> LJResult> { - let device: Box = match &config.dac { - DacFamily::Helios(conf) => Box::new(HeliosDevice::new(conf)?), - DacFamily::Etherdream( conf) => Box::new( EtherdreamDevice::new(conf)?), - DacFamily::Dummy => Box::new(DummyDevice::new()?) - }; - Ok(device) + let device: Box = match &config.dac { + DacFamily::Helios(conf) => Box::new(HeliosDevice::new(conf)?), + DacFamily::Etherdream(conf) => Box::new(EtherdreamDevice::new(conf)?), + DacFamily::Dummy => Box::new(DummyDevice::new()?), + }; + Ok(device) } diff --git a/src/device/dummy.rs b/src/device/dummy.rs index 724d132..bc2189b 100644 --- a/src/device/dummy.rs +++ b/src/device/dummy.rs @@ -1,43 +1,44 @@ -use crate::device::{Device, Status, PlaybackState}; +use crate::device::{Device, PlaybackState, Status}; use crate::errors::LJResult; use crate::point::{Color, Point}; use log::debug; pub struct DummyDevice { - state: PlaybackState + state: PlaybackState, } impl DummyDevice { - pub fn new() -> LJResult { - Ok(Self { state: PlaybackState::IDLE }) - } + pub fn new() -> LJResult { + Ok(Self { + state: PlaybackState::IDLE, + }) + } } impl Device for DummyDevice { - fn status(&mut self) -> Status { - Status { - last_traced_at: "never".to_string(), - properties: vec!["foo".to_string()], - playback_state: self.state, - capacity: 0, - lack: "lack".to_string() - } - } + fn status(&mut self) -> Status { + Status { + last_traced_at: "never".to_string(), + properties: vec!["foo".to_string()], + playback_state: self.state, + capacity: 0, + lack: "lack".to_string(), + } + } - fn draw(&mut self, - line: Vec, - speed: u32, - ) -> LJResult<()> { - debug!("Draw Line at speed {speed} : {:?}", line); - Ok(()) - } + fn draw(&mut self, line: Vec, speed: u32) -> LJResult<()> { + debug!("Draw Line at speed {speed} : {:?}", line); + Ok(()) + } - fn stop(&mut self) -> LJResult<()> { - Ok(()) - } - fn grid(&mut self) -> Vec { - vec!( - Point{ x: 0 as f32, y: 0 as f32, color:Color{ r: 0, g: 0, b: 0 }} - ) - } + fn stop(&mut self) -> LJResult<()> { + Ok(()) + } + fn grid(&mut self) -> Vec { + vec![Point { + x: 0 as f32, + y: 0 as f32, + color: Color { r: 0, g: 0, b: 0 }, + }] + } } diff --git a/src/device/etherdream.rs b/src/device/etherdream.rs index d1aaac6..12602f4 100644 --- a/src/device/etherdream.rs +++ b/src/device/etherdream.rs @@ -1,10 +1,10 @@ -use std::time; -use std::net::SocketAddr; use ether_dream::dac::stream::connect; use ether_dream::dac::Stream; +use std::net::SocketAddr; +use std::time; use crate::conf::EtherDreamConf; -use crate::device::{Device, Status, PlaybackState}; +use crate::device::{Device, PlaybackState, Status}; use crate::errors::{LJError, LJResult}; use crate::point::{Color, Point}; use ether_dream::protocol::{DacBroadcast, DacStatus}; @@ -12,142 +12,144 @@ use log::{info, warn}; #[warn(dead_code)] pub struct EtherdreamDevice { - pub conf: EtherDreamConf, - dac: DacBroadcast, - // source_address: SocketAddr, - stream: Stream, - // sent_points: u16, - lack: String, - last_traced_at: String, + pub conf: EtherDreamConf, + dac: DacBroadcast, + // source_address: SocketAddr, + stream: Stream, + // sent_points: u16, + lack: String, + last_traced_at: String, } impl EtherdreamDevice { - pub fn new(conf: &EtherDreamConf) -> LJResult { - let (dac, _source_address, stream) = EtherdreamDevice::get_dac(conf)?; - // let (dac, source_address) = EtherdreamDevice::get_dac(conf)?; - Ok(Self { - conf: (*conf).clone(), - dac, - // source_address, - stream, - // sent_points: 0, - lack: "".to_string(), - last_traced_at: "1985-04-12T23:20:50.52Z".to_string(), - }) - } - 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_timeout(Some(time::Duration::new(10, 0)))?; - info!("Attempting to get DAC broadcast..."); - let broadcast = dac_broadcast - .filter_map(|result| { - match result { - Err(err) => { - warn!( "Failed to find a valid DAC via broadcast. Error: {:?}", err); - info!( "Retrying..."); - None - }, - Ok((dac, source_addr)) => { - if source_addr.is_ipv6() { return None; } - if &source_addr.ip().to_string() != ip { return None; } - info!("Valid broadcast"); - Some(Ok((dac, source_addr))) - } - } - }) - .next() - .expect("Failed to receive broadcast."); - 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 new(conf: &EtherDreamConf) -> LJResult { + let (dac, _source_address, stream) = EtherdreamDevice::get_dac(conf)?; + // let (dac, source_address) = EtherdreamDevice::get_dac(conf)?; + Ok(Self { + conf: (*conf).clone(), + dac, + // source_address, + stream, + // sent_points: 0, + lack: "".to_string(), + last_traced_at: "1985-04-12T23:20:50.52Z".to_string(), + }) + } + 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_timeout(Some(time::Duration::new(10, 0)))?; + info!("Attempting to get DAC broadcast..."); + let broadcast = dac_broadcast + .filter_map(|result| match result { + Err(err) => { + warn!("Failed to find a valid DAC via broadcast. Error: {:?}", err); + info!("Retrying..."); + None + } + Ok((dac, source_addr)) => { + if source_addr.is_ipv6() { + return None; + } + if &source_addr.ip().to_string() != ip { + return None; + } + info!("Valid broadcast"); + Some(Ok((dac, source_addr))) + } + }) + .next() + .expect("Failed to receive broadcast."); + 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 { - // Establish the TCP connection. - let mut stream = connect(dac, source_address.ip())?; + pub fn get_tcp_stream(dac: &DacBroadcast, source_address: &SocketAddr) -> LJResult { + // Establish the TCP connection. + let mut stream = connect(dac, source_address.ip())?; - // Prepare stream - stream - .queue_commands() - .prepare_stream() - .submit() - .err() - .map(|err| { - eprintln!( - "err occurred when submitting PREPARE_STREAM \ + // Prepare stream + stream + .queue_commands() + .prepare_stream() + .submit() + .err() + .map(|err| { + eprintln!( + "err occurred when submitting PREPARE_STREAM \ command and listening for response: {}", - err - ); - }); + err + ); + }); - Ok(stream) - } + Ok(stream) + } + pub fn check_tcp_stream(&mut self) -> LJResult<()> { + // todo Reinit stream if needed + // self.stream = EtherdreamDevice::get_tcp_stream(&self.dac, &self.source_address)? + Ok(()) + } - pub fn check_tcp_stream(&mut self) -> LJResult<()> { - // todo Reinit stream if needed - // self.stream = EtherdreamDevice::get_tcp_stream(&self.dac, &self.source_address)? - Ok(()) - } - - // 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 - } + // 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 { - fn status(&mut self) -> Status { - let _ = self.check_tcp_stream(); + 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 - }; + // "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, + }; - Status { - last_traced_at: self.last_traced_at.clone(), - properties: vec!["foo".to_string()], - playback_state, - capacity: self.dac.dac_status.buffer_fullness, - lack: String::from(&self.lack), - } - } + Status { + last_traced_at: self.last_traced_at.clone(), + properties: vec!["foo".to_string()], + playback_state, + capacity: self.dac.dac_status.buffer_fullness, + lack: String::from(&self.lack), + } + } - fn draw(&mut self, - line: Vec, - _speed: u32, - ) -> LJResult<()> { - let n_points = self.points_to_generate(); - self.stream - .queue_commands() - .data(line.into_iter().map(|point| point.into()).take(n_points)) - .submit()?; - Ok(()) - } + fn draw(&mut self, line: Vec, _speed: u32) -> LJResult<()> { + let n_points = self.points_to_generate(); + self.stream + .queue_commands() + .data(line.into_iter().map(|point| point.into()).take(n_points)) + .submit()?; + Ok(()) + } - fn stop(&mut self) -> LJResult<()> { - self.stream - .queue_commands() - .stop() - .submit() - .expect("err occurred when submitting STOP command and listening for response"); - Ok(()) - } + fn stop(&mut self) -> LJResult<()> { + self.stream + .queue_commands() + .stop() + .submit() + .expect("err occurred when submitting STOP command and listening for response"); + Ok(()) + } - fn grid(&mut self) -> Vec { - vec!( - Point { x: 0.0, y: 0.0, color: Color { r: 255, g: 255, b: 255 } } - ) - } + fn grid(&mut self) -> Vec { + vec![Point { + x: 0.0, + y: 0.0, + color: Color { + r: 255, + g: 255, + b: 255, + }, + }] + } } diff --git a/src/device/helios.rs b/src/device/helios.rs index 19a346c..a26d5dd 100644 --- a/src/device/helios.rs +++ b/src/device/helios.rs @@ -1,103 +1,152 @@ +use crate::conf::HeliosConf; +use crate::device::{Device, PlaybackState, Status}; +use crate::errors::{LJError, LJResult}; +use crate::point::{Color, Point}; +use chrono::Utc; +use helios_dac::{ + // Coordinate, + // Color, + DeviceStatus, + Frame, + // Point as HeliosPoint, +}; /// /// Configure udev: /// https://github.com/Grix/helios_dac/blob/master/docs/udev_rules_for_linux.md /// use helios_dac::{NativeHeliosDac, NativeHeliosDacController}; -use helios_dac::{ - // Coordinate, - // Color, - DeviceStatus, - Frame, - // Point as HeliosPoint, -}; -use crate::conf::HeliosConf; -use crate::device::{Device, Status, PlaybackState}; -use crate::errors::{LJError, LJResult}; -use crate::point::{Color, Point}; -use chrono::Utc; pub struct HeliosDevice { - pub conf: HeliosConf, - dac: NativeHeliosDac, - sent_points: u16, - state: PlaybackState, - lack: String, - last_traced_at: String, + pub conf: HeliosConf, + dac: NativeHeliosDac, + sent_points: u16, + state: PlaybackState, + lack: String, + last_traced_at: String, } impl HeliosDevice { - pub fn new(conf: &HeliosConf) -> LJResult { - let id = conf.id; - let controller = NativeHeliosDacController::new()?; - let devices = controller.list_devices()?; - let Some(device) = devices.into_iter().nth(id as usize) else { + pub fn new(conf: &HeliosConf) -> LJResult { + let id = conf.id; + let controller = NativeHeliosDacController::new()?; + let devices = controller.list_devices()?; + let Some(device) = devices.into_iter().nth(id as usize) else { return Err(Box::new(LJError::HeliosDeviceMissing)); }; - let dac = device.open()?; - Ok(Self { - conf: (*conf).clone(), - dac, - sent_points: 0, - state: PlaybackState::PREPARE, - lack: "".to_string(), - last_traced_at: "1985-04-12T23:20:50.52Z".to_string(), - }) - } + let dac = device.open()?; + Ok(Self { + conf: (*conf).clone(), + dac, + sent_points: 0, + state: PlaybackState::PREPARE, + lack: "".to_string(), + last_traced_at: "1985-04-12T23:20:50.52Z".to_string(), + }) + } } impl Device for HeliosDevice { - fn status(&mut self) -> Status { - let lack = self.lack.clone(); - Status { - last_traced_at: self.last_traced_at.clone(), - properties: vec!["foo".to_string()], - playback_state: self.state, - capacity: self.sent_points, - lack, - } - } + fn status(&mut self) -> Status { + let lack = self.lack.clone(); + Status { + last_traced_at: self.last_traced_at.clone(), + properties: vec!["foo".to_string()], + playback_state: self.state, + capacity: self.sent_points, + lack, + } + } - fn draw(&mut self, - line: Vec, - speed: u32, - ) -> LJResult<()> { - self.state = PlaybackState::IDLE; - while let Ok(DeviceStatus::NotReady) = self.dac.status() {} - self.state = PlaybackState::PLAYING; + fn draw(&mut self, line: Vec, speed: u32) -> LJResult<()> { + self.state = PlaybackState::IDLE; + while let Ok(DeviceStatus::NotReady) = self.dac.status() {} + self.state = PlaybackState::PLAYING; - let points: Vec = line.into_iter().map(|p| p.into()).collect(); - 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(()) - } + let points: Vec = line.into_iter().map(|p| p.into()).collect(); + 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(()) + } - fn stop(&mut self) -> LJResult<()> { - self.dac.stop()?; - Ok(()) - } + fn stop(&mut self) -> LJResult<()> { + self.dac.stop()?; + Ok(()) + } - fn grid(&mut self) -> Vec { - let dim_min = 0 as f32; - let dim_mid = 2047 as f32; - let dim_max = 4095 as f32; - let col_min = Color { r: 0, g: 0, b: 0 }; - let col_max = Color { r: 255, g: 255, b: 255 }; + fn grid(&mut self) -> Vec { + let dim_min = 0 as f32; + let dim_mid = 2047 as f32; + let dim_max = 4095 as f32; + let col_min = Color { r: 0, g: 0, b: 0 }; + let col_max = Color { + r: 255, + g: 255, + b: 255, + }; - vec![ - Point { x: dim_min, y: dim_max, color: col_min }, - Point { x: dim_min, y: dim_max, color: col_max }, - Point { x: dim_max, y: dim_max, color: col_max }, - Point { x: dim_max, y: dim_min, color: col_max }, - Point { x: dim_min, y: dim_min, color: col_max }, - Point { x: dim_min, y: dim_min, color: col_min }, - Point { x: dim_min, y: dim_mid, color: col_min }, - Point { x: dim_min, y: dim_mid, color: col_max }, - Point { x: dim_mid, y: dim_mid, color: col_max }, - Point { x: dim_mid, y: dim_min, color: col_max }, - Point { x: dim_min, y: dim_min, color: col_max }, - Point { x: dim_min, y: dim_min, color: col_min }, - ] - } + vec![ + Point { + x: dim_min, + y: dim_max, + color: col_min, + }, + Point { + x: dim_min, + y: dim_max, + color: col_max, + }, + Point { + x: dim_max, + y: dim_max, + color: col_max, + }, + Point { + x: dim_max, + y: dim_min, + color: col_max, + }, + Point { + x: dim_min, + y: dim_min, + color: col_max, + }, + Point { + x: dim_min, + y: dim_min, + color: col_min, + }, + Point { + x: dim_min, + y: dim_mid, + color: col_min, + }, + Point { + x: dim_min, + y: dim_mid, + color: col_max, + }, + Point { + x: dim_mid, + y: dim_mid, + color: col_max, + }, + Point { + x: dim_mid, + y: dim_min, + color: col_max, + }, + Point { + x: dim_min, + y: dim_min, + color: col_max, + }, + Point { + x: dim_min, + y: dim_min, + color: col_min, + }, + ] + } } diff --git a/src/errors.rs b/src/errors.rs index 9640b11..9c376d5 100644 --- a/src/errors.rs +++ b/src/errors.rs @@ -1,50 +1,50 @@ +use config::ConfigError; +use redis::RedisError; use std::error::Error; use std::{fmt, io}; -use redis::RedisError; -use config::ConfigError; pub type LJResult = Result>; #[derive(Debug)] pub enum LJError { - Config(ConfigError), - RedisConnect(RedisError), - HeliosDeviceMissing, - BadEDH, - EtherdreamConnectError(io::Error), + Config(ConfigError), + RedisConnect(RedisError), + HeliosDeviceMissing, + BadEDH, + EtherdreamConnectError(io::Error), } impl fmt::Display for LJError { - fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { - use LJError::*; + fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { + use LJError::*; - match self { - Config(err) => { - write!(f, "unable to load config file: {err}") - } - RedisConnect(err) => { - write!(f, "unable to connect to redis server: {err}") - } - HeliosDeviceMissing => { - write!(f, "helios device not found") - } - BadEDH => { - write!(f, "EDH matrix is not a 3x3 matrix") - } - EtherdreamConnectError(err) => { - write!(f, "Failed to retrieve Etherdream device: {err}") - } - } - } + match self { + Config(err) => { + write!(f, "unable to load config file: {err}") + } + RedisConnect(err) => { + write!(f, "unable to connect to redis server: {err}") + } + HeliosDeviceMissing => { + write!(f, "helios device not found") + } + BadEDH => { + write!(f, "EDH matrix is not a 3x3 matrix") + } + EtherdreamConnectError(err) => { + write!(f, "Failed to retrieve Etherdream device: {err}") + } + } + } } impl Error for LJError { - fn source(&self) -> Option<&(dyn Error + 'static)> { - use LJError::*; + fn source(&self) -> Option<&(dyn Error + 'static)> { + use LJError::*; - match self { - RedisConnect(err) => Some(err), - _ => None - } - } + match self { + RedisConnect(err) => Some(err), + _ => None, + } + } } diff --git a/src/lib.rs b/src/lib.rs index 9048f40..be83429 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -1,7 +1,7 @@ -pub mod redis_ctrl; pub mod conf; -pub mod errors; pub mod device; +pub mod errors; pub mod point; +pub mod redis_ctrl; pub mod transformer; pub mod worldstate; diff --git a/src/main.rs b/src/main.rs index 501d8cd..6ebe17d 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,193 +1,178 @@ +mod conf; +mod device; +mod errors; +mod point; /// /// Configure udev: /// https://github.com/Grix/helios_dac/blob/master/docs/udev_rules_for_linux.md /// mod redis_ctrl; -mod conf; -mod errors; -mod point; mod transformer; -mod device; mod worldstate; +use conf::Conf; use device::device_factory; +use env_logger::Builder; +use errors::LJResult; +use log::{/* warn, */ error, info, LevelFilter}; +use point::{Color, Point}; +use redis_ctrl::{Order, RedisCtrl}; use std::sync::atomic::{AtomicBool, Ordering}; use std::sync::Arc; -use redis_ctrl::{RedisCtrl, Order}; -use conf::Conf; -use errors::LJResult; -use point::{Point, Color}; use transformer::Transformers; -use log::{LevelFilter, info, /* warn, */ error}; -use env_logger::Builder; use worldstate::WorldState; const DEFAULT_CONF_FILE: &str = "settings.toml"; pub fn main() { - match run_all() { - Ok(()) => {} - Err(err) => { - error!("Error: {}", err); - } - } + match run_all() { + Ok(()) => {} + Err(err) => { + error!("Error: {}", err); + } + } } - fn run_all() -> LJResult<()> { - // Setup configuration file and set up logs - let filename = std::env::args().nth(1).unwrap_or_else(|| { - DEFAULT_CONF_FILE.to_string() - }); + // Setup configuration file and set up logs + let filename = std::env::args() + .nth(1) + .unwrap_or_else(|| DEFAULT_CONF_FILE.to_string()); - let config = Conf::new(&filename); - init_logging(&config); - let config = config?; - info!("*** Starting up ***"); + let config = Conf::new(&filename); + init_logging(&config); + let config = config?; + info!("*** Starting up ***"); - info!("{:?}", config); + info!("{:?}", config); - // Setup Redis Service - let mut rs = RedisCtrl::new(&config.redis_url, &config.laser_id)?; + // Setup Redis Service + let mut rs = RedisCtrl::new(&config.redis_url, &config.laser_id)?; - let mut world_state = rs.init_world_state().unwrap(); - info!("WorldState: {:?}", world_state); + let mut world_state = rs.init_world_state().unwrap(); + info!("WorldState: {:?}", world_state); + dbg!("worldstate.kpps: {}", world_state.kpps); - // Setup handler for interrupt Signals - let running = Arc::new(AtomicBool::new(true)); - let r = running.clone(); - ctrlc::set_handler(move || { - r.store(false, Ordering::SeqCst); - })?; + // Setup handler for interrupt Signals + let running = Arc::new(AtomicBool::new(true)); + let r = running.clone(); + ctrlc::set_handler(move || { + r.store(false, Ordering::SeqCst); + })?; - // Setup Laser Device based on conf - let mut tracer = device_factory(&config)?; - world_state.grid = tracer.grid(); + // Setup Laser Device based on conf + let mut tracer = device_factory(&config)?; + world_state.grid = tracer.grid(); - // can't work, but we can add + Debug to Device to make it work... - //dbg!(tracer); + // can't work, but we can add + Debug to Device to make it work... + //dbg!(tracer); - // Setup geometry transformers on points lists - let transformers = config.get_transformers(); + // Setup geometry transformers on points lists + let transformers = config.get_transformers(); - // Dispatch based on redis requests - while running.load(Ordering::SeqCst) { - rs.set_status(tracer.status())?; + // Dispatch based on redis requests + while running.load(Ordering::SeqCst) { + rs.set_status(tracer.status())?; - let order = rs.get_order(config.laser_id)?; + let order = rs.get_order(config.laser_id)?; + match order { + Order::Draw | Order::Black | Order::Grid => { + // 0 : Draw Normal point list + // 2 : Draw BLACK point list + // 3 : Draw GRID point list + world_state.draw_black = order == Order::Black; + world_state.draw_grid = order == Order::Grid; + let frame = get_next_frame(&config, &transformers, &mut rs, &world_state)?; + // For now, draw all the time + tracer.draw(frame, world_state.kpps)?; + } + Order::Intensity => { + // 6 : Max Intensity Change = reread redis key /intensity + world_state.intensity = rs.get_int("intensity")?.try_into()?; + } + Order::Edh => { + // 1 : Get the new EDH = reread redis key /EDH/lasernumber + world_state.edh = rs.get_edh()?; + } + Order::Kpps => { + // 7 : kpps change = reread redis key /kpps + world_state.kpps = rs.get_int("kpps")?; + //dbg!(world_state.kpps); + } + Order::ClientKey => { + world_state.client_key = rs.get_client_key()?; + } + // Order::ColorBalance => {}, + _ => { + // 4 : Resampler Change (longs and shorts lsteps) + // 5 : Client Key Change = reread redis key /clientkey + // 8 : color balance change = reread redis keys /red /green /blue + // 9 : poweroff LJ + info!("Order: {:?}", order); + } + } + } - match order { - Order::Draw | Order::Black | Order::Grid => { - // 0 : Draw Normal point list - // 2 : Draw BLACK point list - // 3 : Draw GRID point list - world_state.draw_black = order == Order::Black; - world_state.draw_grid = order == Order::Grid; - let frame = get_next_frame( - &config, - &transformers, - &mut rs, - &world_state, - )?; - // For now, draw all the time - tracer.draw(frame, world_state.kpps)?; - } - Order::Intensity => { - // 6 : Max Intensity Change = reread redis key /intensity - world_state.intensity = rs.get_int("intensity")? - .try_into()?; - } - Order::Edh => { - // 1 : Get the new EDH = reread redis key /EDH/lasernumber - world_state.edh = rs.get_edh()?; - } - Order::Kpps => { - // 7 : kpps change = reread redis key /kpps - world_state.kpps = rs.get_int("kpps")?; - } - Order::ClientKey => { - world_state.client_key = rs.get_client_key()?; - }, - // Order::ColorBalance => {}, - _ => { - // 4 : Resampler Change (longs and shorts lsteps) - // 5 : Client Key Change = reread redis key /clientkey - // 8 : color balance change = reread redis keys /red /green /blue - // 9 : poweroff LJ - info!("Order: {:?}", order); - } - } - } - - info!("Exiting, stoping device."); - tracer.stop()?; - Ok(()) + info!("Exiting, stoping device."); + tracer.stop()?; + Ok(()) } fn init_logging(config: &LJResult) { - if let Ok(ref config) = config { - if config.debug { - let mut builder = Builder::from_default_env(); - builder - .filter(None, LevelFilter::Info) - .init(); - info!("Debug mode enabled from configuration file"); - return; - } - } - info!("Logging level inherited from env"); - env_logger::init(); + if let Ok(ref config) = config { + if config.debug { + let mut builder = Builder::from_default_env(); + builder.filter(None, LevelFilter::Info).init(); + info!("Debug mode enabled from configuration file"); + return; + } + } + info!("Logging level inherited from env"); + env_logger::init(); } fn get_next_frame( - config: &Conf, - transformers: &[Box], - rs: &mut RedisCtrl, - world_state: &WorldState, + config: &Conf, + transformers: &[Box], + rs: &mut RedisCtrl, + world_state: &WorldState, ) -> LJResult> { - let format_key = format!("{}{}", - world_state.client_key, - config.laser_id); + let format_key = format!("{}{}", world_state.client_key, config.laser_id); - // Handle the grid case + // Handle the grid case - let mut line: Vec = if world_state.draw_grid { - world_state.grid.clone() - } else { - let redis_line = rs.get_line(&format_key)?; - redis_line.into_iter() - .map(|tpl| tpl.into()) - .collect() - }; + let mut line: Vec = if world_state.draw_grid { + world_state.grid.clone() + } else { + let redis_line = rs.get_line(&format_key)?; + redis_line.into_iter().map(|tpl| tpl.into()).collect() + }; - for transformer in transformers { - line = transformer.apply(&line, world_state); - } + for transformer in transformers { + line = transformer.apply(&line, world_state); + } - info!("Draw Black -> {}", world_state.draw_black); - info!("Draw Grid -> {}", world_state.draw_grid); + info!("Draw Black -> {}", world_state.draw_black); + info!("Draw Grid -> {}", world_state.draw_grid); - // LIMITER and BLACK - line = line.into_iter() - .map(|p| { - let color = if world_state.draw_black { - Color { r: 0, g: 0, b: 0 } - } else { - Color { - r: p.color.r.min(world_state.intensity), - g: p.color.g.min(world_state.intensity), - b: p.color.b.min(world_state.intensity), - } - }; - Point { - color, - ..p - } - }) - .collect(); + // LIMITER and BLACK + line = line + .into_iter() + .map(|p| { + let color = if world_state.draw_black { + Color { r: 0, g: 0, b: 0 } + } else { + Color { + r: p.color.r.min(world_state.intensity), + g: p.color.g.min(world_state.intensity), + b: p.color.b.min(world_state.intensity), + } + }; + Point { color, ..p } + }) + .collect(); - //info!("Line: {:?}", line); - Ok(line) + //info!("Line: {:?}", line); + Ok(line) } - diff --git a/src/point.rs b/src/point.rs index b64a01d..73df5e4 100644 --- a/src/point.rs +++ b/src/point.rs @@ -2,63 +2,91 @@ use ether_dream::protocol::DacPoint; #[derive(Debug, Clone, Copy, Default, PartialEq)] pub struct Point { - pub x: f32, - pub y: f32, - pub color: Color, + pub x: f32, + pub y: f32, + pub color: Color, } #[derive(Debug, Clone, Copy, Default, PartialEq)] pub struct Color { - pub r: u8, - pub g: u8, - pub b: u8, + pub r: u8, + pub g: u8, + pub b: u8, } +impl Point { + pub fn diff(p1: &Self, p2: &Self) -> Self { + Point { + x: p1.x - p2.x, + y: p1.y - p2.y, + color: Color { r: 0, g: 0, b: 0 }, + } + } + + pub fn normalize(&mut self) -> Self { + let length = (self.x * self.x + self.y * self.y).sqrt(); + self.x = self.x / length; + self.y = self.y / length; + *self + } + + pub fn cross(&self, pt: &Self) -> f32 { + self.x * pt.x + self.y * pt.y + } +} + +// normalaize +// diff + impl From for u32 { - fn from(value: Color) -> Self { - let r = value.r as u32; - let g = value.g as u32; - let b = (value.b) as u32; - (r << 16) + (g << 8) + b - } + fn from(value: Color) -> Self { + let r = value.r as u32; + let g = value.g as u32; + let b = (value.b) as u32; + (r << 16) + (g << 8) + b + } } impl From<(f32, f32, u32)> for Point { - fn from((x, y, color): (f32, f32, u32)) -> Point { - let r = (color >> 16) as u8; - let g = ((color >> 8) & 255) as u8; - let b = (color & 255) as u8; - Point { x, y, color: Color { r, g, b } } - } + fn from((x, y, color): (f32, f32, u32)) -> Point { + let r = (color >> 16) as u8; + let g = ((color >> 8) & 255) as u8; + let b = (color & 255) as u8; + Point { + x, + y, + color: Color { r, g, b }, + } + } } impl From for helios_dac::Point { - fn from(pt: Point) -> helios_dac::Point { - let x = pt.x.clamp(0.0, 4095.0) as u16; - let y = pt.y.clamp(0.0, 4095.0) as u16; - helios_dac::Point { - coordinate: (x, y).into(), - color: helios_dac::Color::new(pt.color.r, pt.color.g, pt.color.b), - intensity: 0xFF, - } - } + fn from(pt: Point) -> helios_dac::Point { + let x = pt.x.clamp(0.0, 4095.0) as u16; + let y = pt.y.clamp(0.0, 4095.0) as u16; + helios_dac::Point { + coordinate: (x, y).into(), + color: helios_dac::Color::new(pt.color.r, pt.color.g, pt.color.b), + intensity: 0xFF, + } + } } impl From 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, - } - } + 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, + } + } } diff --git a/src/redis_ctrl.rs b/src/redis_ctrl.rs index 74ac5a0..fc891ad 100644 --- a/src/redis_ctrl.rs +++ b/src/redis_ctrl.rs @@ -1,132 +1,134 @@ -use redis::{Client, Commands, Connection}; -use ron::de::from_str; use crate::device::Status; use crate::errors::{LJError, LJResult}; -use crate::worldstate::{WorldState,EDH}; +use crate::worldstate::{WorldState, EDH}; +use redis::{Client, Commands, Connection}; +use ron::de::from_str; // use log::info; #[repr(u8)] #[derive(Debug, PartialEq)] pub enum Order { - Draw = 0, - Edh, - //homography - Black, - Grid, - Resampler, - ClientKey, - Intensity, - Kpps, - ColorBalance, + Draw = 0, + Edh, + //homography + Black, + Grid, + Resampler, + ClientKey, + Intensity, + Kpps, + ColorBalance, } impl TryFrom for Order { - type Error = String; + type Error = String; - fn try_from(value: u8) -> Result { - use Order::*; + fn try_from(value: u8) -> Result { + use Order::*; - if value > 8 { - return Err("order out of range".to_string()); - } + if value > 8 { + return Err("order out of range".to_string()); + } - Ok(match value { - 0 => Draw, - 1 => Edh, - 2 => Black, - 3 => Grid, - 4 => Resampler, - 5 => ClientKey, - 6 => Intensity, - 7 => Kpps, - 8 => ColorBalance, - _ => unreachable!() - }) - } + Ok(match value { + 0 => Draw, + 1 => Edh, + 2 => Black, + 3 => Grid, + 4 => Resampler, + 5 => ClientKey, + 6 => Intensity, + 7 => Kpps, + 8 => ColorBalance, + _ => unreachable!(), + }) + } } pub type Line = Vec<(f32, f32, u32)>; pub struct RedisCtrl { - pub client: Client, - pub connection: Connection, - laser_id: u8, + pub client: Client, + pub connection: Connection, + laser_id: u8, } impl RedisCtrl { - pub fn new(url: &str, laser_id: &u8) -> LJResult { - let client = Client::open(url) - .map_err(LJError::RedisConnect)?; - let connection = client.get_connection() - .map_err(LJError::RedisConnect)?; - Ok(RedisCtrl { client, connection, laser_id: *laser_id }) - } + pub fn new(url: &str, laser_id: &u8) -> LJResult { + let client = Client::open(url).map_err(LJError::RedisConnect)?; + let connection = client.get_connection().map_err(LJError::RedisConnect)?; + Ok(RedisCtrl { + client, + connection, + laser_id: *laser_id, + }) + } - pub fn get_line(&mut self, key: &str) -> LJResult { - let val: String = self.connection.get(key)?; - let line: Line = from_str(&val)?; - Ok(line) - } + pub fn get_line(&mut self, key: &str) -> LJResult { + let val: String = self.connection.get(key)?; + let line: Line = from_str(&val)?; + Ok(line) + } - pub fn set(&mut self, key: String, value: String) -> LJResult<()> { - self.connection.set(key, value)?; - Ok(()) - } + pub fn set(&mut self, key: String, value: String) -> LJResult<()> { + self.connection.set(key, value)?; + Ok(()) + } - pub fn get_order(&mut self, id: u8) -> LJResult { - let path = format!("/order/{id}"); - let val: u8 = self.connection.get(path.clone())?; + pub fn get_order(&mut self, id: u8) -> LJResult { + let path = format!("/order/{id}"); + let val: u8 = self.connection.get(path.clone())?; - if val == 1 || val >= 4 { - self.connection.set(path, 0)?; - } + if val == 1 || val >= 4 { + self.connection.set(path, 0)?; + } - Ok(val.try_into()?) - } + Ok(val.try_into()?) + } - /** - /lstt/lasernumber etherdream last_status.playback_state (0: idle 1: prepare 2: playing) - /cap/lasernumber number of empty points sent to fill etherdream buffer (up to 1799) - /lack/lasernumber "a": ACK "F": Full "I": invalid. 64 or 35 for no connection. - **/ - pub fn set_status(&mut self, status: Status) -> LJResult<()> { - let lstt_key = format!("/lstt/{}", self.laser_id); - let cap_key = format!("/cap/{}", self.laser_id); - let lack_key = format!("/lack/{}", self.laser_id); - self.set(lstt_key, status.playback_state.to_string())?; - self.set(cap_key, status.capacity.to_string())?; - self.set(lack_key, status.lack.to_string())?; - Ok(()) - } + /** + /lstt/lasernumber etherdream last_status.playback_state (0: idle 1: prepare 2: playing) + /cap/lasernumber number of empty points sent to fill etherdream buffer (up to 1799) + /lack/lasernumber "a": ACK "F": Full "I": invalid. 64 or 35 for no connection. + **/ + pub fn set_status(&mut self, status: Status) -> LJResult<()> { + let lstt_key = format!("/lstt/{}", self.laser_id); + let cap_key = format!("/cap/{}", self.laser_id); + let lack_key = format!("/lack/{}", self.laser_id); + self.set(lstt_key, status.playback_state.to_string())?; + self.set(cap_key, status.capacity.to_string())?; + self.set(lack_key, status.lack.to_string())?; + Ok(()) + } - pub fn init_world_state( &mut self) -> LJResult{ - Ok(WorldState { - client_key: self.get_client_key().unwrap(), - edh: self.get_edh().unwrap(), - kpps: self.get_int("kpps").unwrap().try_into().unwrap(), - intensity: self.get_int("intensity").unwrap().try_into().unwrap(), - ..WorldState::default() - }) - } + pub fn init_world_state(&mut self) -> LJResult { + Ok(WorldState { + client_key: self.get_client_key().unwrap(), + edh: self.get_edh().unwrap(), + kpps: self.get_int("kpps").unwrap().try_into().unwrap(), + intensity: self.get_int("intensity").unwrap().try_into().unwrap(), + ..WorldState::default() + }) + } - pub fn get_edh( &mut self ) -> LJResult { - // Get new EDH - let edh_key = format!("/EDH/{}", self.laser_id); - let edh : String = self.connection.get(edh_key)?; - let edh : Vec> = from_str(&edh)?; - let edh = EDH::new(edh)?; - Ok(edh) - } + pub fn get_edh(&mut self) -> LJResult { + // Get new EDH + let edh_key = format!("/EDH/{}", self.laser_id); + let edh: String = self.connection.get(edh_key)?; + let edh: Vec> = from_str(&edh)?; + let edh = EDH::new(edh)?; + Ok(edh) + } - pub fn get_client_key( &mut self ) -> LJResult { - let key : String = self.connection.get("/clientkey")?; - Ok(key) - } + pub fn get_client_key(&mut self) -> LJResult { + let key: String = self.connection.get("/clientkey")?; + Ok(key) + } - pub fn get_int(&mut self, key: &str ) -> LJResult { - // Get new Int - let fmt = format!("/{key}/{}", self.laser_id); - let val : u32 = self.connection.get(fmt)?; - Ok(val) - } + pub fn get_int(&mut self, key: &str) -> LJResult { + // Get new Int + let fmt = format!("/{key}/{}", self.laser_id); + let val: u32 = self.connection.get(fmt)?; + Ok(val) + } } diff --git a/src/transformer.rs b/src/transformer.rs index aa25291..336dfe4 100644 --- a/src/transformer.rs +++ b/src/transformer.rs @@ -1,28 +1,26 @@ //pub mod common; -mod translate; -mod replicate; -mod rotate; +mod angle_optimisation; mod flip_horizontal; mod flip_vertical; mod grid; mod homography; +mod replicate; +mod rotate; +mod translate; use crate::point::Point; use crate::worldstate::WorldState; // re-export transformers to be abe to use it directly from transformer:: -pub use translate::Translate; -pub use replicate::Replicate; -pub use rotate::Rotate; +pub use self::homography::Homography; +pub use angle_optimisation::AngleOptimisation; pub use flip_horizontal::FlipHorizontal; pub use flip_vertical::FlipVertical; pub use grid::Grid; -pub use self::homography::Homography; +pub use replicate::Replicate; +pub use rotate::Rotate; +pub use translate::Translate; pub trait Transformers { - fn apply( - &self, - point_list: &[Point], - world_state: &WorldState - ) -> Vec; + fn apply(&self, point_list: &[Point], world_state: &WorldState) -> Vec; } diff --git a/src/transformer/angle_optimisation.rs b/src/transformer/angle_optimisation.rs new file mode 100644 index 0000000..54d254c --- /dev/null +++ b/src/transformer/angle_optimisation.rs @@ -0,0 +1,86 @@ +use crate::point::{Color, Point}; +use crate::transformer::Transformers; +use crate::worldstate::WorldState; + +use serde::{Deserialize, Serialize}; +use std::f32::consts::PI; + +/// Angle Optimisation + +#[derive(Serialize, Deserialize, Debug, Clone, Copy)] +pub struct AngleOptimisation { + coef: f32, +} + +fn color_not_zero(p1: &Point, p2: &Point, p3: &Point) -> bool { + let c0 = Color { r: 0, g: 0, b: 0 }; + if p1.color == c0 || p2.color == c0 || p3.color == c0 { + false + } else { + true + } +} + +fn pos_different(p1: &Point, p2: &Point) -> bool { + if p1.x == p2.x && p1.y == p2.y { + false + } else { + true + } +} + +impl Transformers for AngleOptimisation { + fn apply(&self, point_list: &[Point], _ws: &WorldState) -> Vec { + let c0 = Color { r: 0, g: 0, b: 0 }; + let mut v = vec![]; + + //println!(); + //println!("===================="); + //println!("= next Frame ="); + //println!("===================="); + v.push(point_list[0]); + for i in 1..(point_list.len() - 1) { + let prev = point_list[i - 1]; + let node = point_list[i]; + let next = point_list[i + 1]; + + if color_not_zero(&prev, &node, &next) + && pos_different(&prev, &node) + && pos_different(&node, &next) + { + //println!("========="); + let mut d1 = Point::diff(&node, &prev); + let mut d2 = Point::diff(&next, &next); + //dbg!(d1); + //dbg!(d2); + d1.normalize(); + d2.normalize(); + + let cross = d1.cross(&d2); + let angle = d1.cross(&d2).acos(); + + let deg = angle * 180. / PI; + + //dbg!(prev); + //dbg!(node); + //dbg!(next); + //dbg!(d1); + //dbg!(d2); + //dbg!(cross); + //dbg!(angle); + //dbg!(deg); + + //println!("deg: {deg}"); + + let nb_add = (self.coef * (1. - angle / PI)) as usize; + + for _ in 0..nb_add {} + } + + v.push(node); // push node + } + v.push(point_list[point_list.len() - 1]); // push last + + v + } +} diff --git a/src/transformer/flip_horizontal.rs b/src/transformer/flip_horizontal.rs index ed9cec3..e2efe28 100644 --- a/src/transformer/flip_horizontal.rs +++ b/src/transformer/flip_horizontal.rs @@ -1,24 +1,27 @@ -use crate::transformer::Transformers; use crate::point::Point; +use crate::transformer::Transformers; use crate::worldstate::WorldState; -use serde::{Serialize,Deserialize}; +use serde::{Deserialize, Serialize}; /// Flip Horizontal -#[derive(Serialize,Deserialize,Debug,Clone,Copy)] +#[derive(Serialize, Deserialize, Debug, Clone, Copy)] pub struct FlipHorizontal { x: f32, } impl Transformers for FlipHorizontal { fn apply(&self, point_list: &[Point], _ws: &WorldState) -> Vec { - point_list.iter() - .map(| pt | { - let dx = pt.x - self.x; - Point { x: dx - 2. * dx, - ..*pt - } - }).collect() + point_list + .iter() + .map(|pt| { + let dx = pt.x - self.x; + Point { + x: dx - 2. * dx, + ..*pt + } + }) + .collect() } } diff --git a/src/transformer/flip_vertical.rs b/src/transformer/flip_vertical.rs index 19529e8..88ab729 100644 --- a/src/transformer/flip_vertical.rs +++ b/src/transformer/flip_vertical.rs @@ -1,24 +1,27 @@ -use crate::transformer::Transformers; use crate::point::Point; +use crate::transformer::Transformers; use crate::worldstate::WorldState; -use serde::{Serialize,Deserialize}; +use serde::{Deserialize, Serialize}; /// Flip Vertical -#[derive(Serialize,Deserialize,Debug,Clone,Copy)] +#[derive(Serialize, Deserialize, Debug, Clone, Copy)] pub struct FlipVertical { y: f32, } impl Transformers for FlipVertical { fn apply(&self, point_list: &[Point], _ws: &WorldState) -> Vec { - point_list.iter() - .map(| pt | { - let dy = pt.y - self.y; - Point { y: dy - 2. * dy, - ..*pt - } - }).collect() + point_list + .iter() + .map(|pt| { + let dy = pt.y - self.y; + Point { + y: dy - 2. * dy, + ..*pt + } + }) + .collect() } } diff --git a/src/transformer/grid.rs b/src/transformer/grid.rs index 7a9d37b..0e5eb52 100644 --- a/src/transformer/grid.rs +++ b/src/transformer/grid.rs @@ -1,58 +1,48 @@ -use crate::transformer::Transformers; use crate::point::Point; +use crate::transformer::Transformers; use crate::worldstate::WorldState; -use serde::{Serialize,Deserialize}; +use serde::{Deserialize, Serialize}; /// Translate -#[derive(Serialize,Deserialize,Debug,Clone,Copy)] +#[derive(Serialize, Deserialize, Debug, Clone, Copy)] pub struct Grid { width: f32, - height: f32 + height: f32, } fn square_box(size: f32, color: u32) -> Vec<(f32, f32, u32)> { vec![ - (-size, -size, 0), - (-size, -size, color), - - (0.0, -size, color), - (0.0, -size, color), - - (size, -size, color), - (size, -size, color), - - (size, 0.0, color), - (size, 0.0, color), - - (size, size, color), - (size, size, color), - - (0.0, size, color), - (0.0, size, color), - - (-size, size, color), - (-size, size, color), - - (-size, 0.0, color), - (-size, 0.0, color), - - (-size, -size, color), - (-size, -size, color), - - (-size, -size, 0) + (-size, -size, 0), + (-size, -size, color), + (0.0, -size, color), + (0.0, -size, color), + (size, -size, color), + (size, -size, color), + (size, 0.0, color), + (size, 0.0, color), + (size, size, color), + (size, size, color), + (0.0, size, color), + (0.0, size, color), + (-size, size, color), + (-size, size, color), + (-size, 0.0, color), + (-size, 0.0, color), + (-size, -size, color), + (-size, -size, color), + (-size, -size, 0), ] } impl Transformers for Grid { fn apply(&self, _point_list: &[Point], _ws: &WorldState) -> Vec { - let mut sq1 = square_box(1000.0, 255 << 8); - let mut line = square_box(2000.0, 255); - line.append(&mut sq1); + let mut sq1 = square_box(1000.0, 255 << 8); + let mut line = square_box(2000.0, 255); + line.append(&mut sq1); - let line: Vec = line.into_iter() - .map(|tpl| tpl.into()).collect(); - line + let line: Vec = line.into_iter().map(|tpl| tpl.into()).collect(); + line } } diff --git a/src/transformer/homography.rs b/src/transformer/homography.rs index 8f1091f..b711297 100644 --- a/src/transformer/homography.rs +++ b/src/transformer/homography.rs @@ -1,20 +1,18 @@ -use crate::transformer::Transformers; use crate::point::Point; -use crate::worldstate::{WorldState,EDH}; -use serde::{Serialize,Deserialize}; +use crate::transformer::Transformers; +use crate::worldstate::{WorldState, EDH}; +use serde::{Deserialize, Serialize}; /// Homography -#[derive(Serialize,Deserialize,Debug,Clone,Copy)] +#[derive(Serialize, Deserialize, Debug, Clone, Copy)] pub struct Homography {} impl Transformers for Homography { - fn apply(&self, point_list: &[Point], ws: &WorldState) -> Vec { - let edh : &EDH = &ws.edh; + fn apply(&self, point_list: &[Point], ws: &WorldState) -> Vec { + let edh: &EDH = &ws.edh; - point_list.iter() - .map(| point | edh.apply(point)) - .collect() + point_list.iter().map(|point| edh.apply(point)).collect() } } @@ -24,41 +22,67 @@ mod tests { #[test] fn identity_matrix_let_point_unchanged() { - let p0 = Point { x: -1500.0, y: 1500.0, ..Point::default() }; + let p0 = Point { + x: -1500.0, + y: 1500.0, + ..Point::default() + }; - let edh = EDH::new(vec![ - vec![ 1.0, 0.0, 0.0 ], - vec![ 0.0, 1.0, 0.0 ], - vec![ 0.0, 0.0, 1.0 ] - ]).unwrap(); + let edh = EDH::new(vec![ + vec![1.0, 0.0, 0.0], + vec![0.0, 1.0, 0.0], + vec![0.0, 0.0, 1.0], + ]) + .unwrap(); - let ws = WorldState { edh : edh, ..WorldState::default() }; - - let homography = Homography{}; - let result = homography.apply(&[p0], &ws); - - assert_eq!(result, vec![Point { x: -1500.0, - y: 1500.0, - ..Point::default() }]); + let ws = WorldState { + edh: edh, + ..WorldState::default() + }; + + let homography = Homography {}; + let result = homography.apply(&[p0], &ws); + + assert_eq!( + result, + vec![Point { + x: -1500.0, + y: 1500.0, + ..Point::default() + }] + ); } #[test] fn rotation_matrix_rotate_the_point() { - let p0 = Point { x: -1500.0, y: 1500.0, ..Point::default() }; + let p0 = Point { + x: -1500.0, + y: 1500.0, + ..Point::default() + }; - let edh = EDH::new(vec![ - vec![ 1.24107321e-03, 1.00500127e-03, 7.15439347e-01], - vec![-9.93223912e-04, 1.22652939e-03,-6.98671238e-01], - vec![ 1.06017142e-17,-4.69459541e-17, 3.32700590e-05] - ]).unwrap(); + let edh = EDH::new(vec![ + vec![1.24107321e-03, 1.00500127e-03, 7.15439347e-01], + vec![-9.93223912e-04, 1.22652939e-03, -6.98671238e-01], + vec![1.06017142e-17, -4.69459541e-17, 3.32700590e-05], + ]) + .unwrap(); - let ws = WorldState { edh : edh, ..WorldState::default() }; - - let homography = Homography{}; - let result = homography.apply(&[p0], &ws); - - assert_eq!(result, vec![Point { x: 10860.557, - y: 79078.87, - ..Point::default() }]); + let ws = WorldState { + edh: edh, + ..WorldState::default() + }; + + let homography = Homography {}; + let result = homography.apply(&[p0], &ws); + + assert_eq!( + result, + vec![Point { + x: 10860.557, + y: 79078.87, + ..Point::default() + }] + ); } } diff --git a/src/transformer/replicate.rs b/src/transformer/replicate.rs index a5e5ab2..590b82f 100644 --- a/src/transformer/replicate.rs +++ b/src/transformer/replicate.rs @@ -1,34 +1,33 @@ -use crate::transformer::Transformers; use crate::point::Point; +use crate::transformer::Transformers; use crate::worldstate::WorldState; -use serde::{Serialize,Deserialize}; +use serde::{Deserialize, Serialize}; /// Replicate #[allow(dead_code)] -#[derive(Serialize,Deserialize,Debug,Clone,Copy)] +#[derive(Serialize, Deserialize, Debug, Clone, Copy)] pub enum Replicate { Until(usize), - Times(usize) + Times(usize), } impl Transformers for Replicate { fn apply(&self, point_list: &[Point], _ws: &WorldState) -> Vec { - let mut point_list2 = vec![]; - match self { - Replicate::Until(n) => { - while point_list2.len() < *n { - point_list2.append(&mut point_list.to_vec()); - } - }, - Replicate::Times(n) => { - for _ in 0..*n { - point_list2.append(&mut point_list.to_vec()); - } - } - } - point_list2 + let mut point_list2 = vec![]; + match self { + Replicate::Until(n) => { + while point_list2.len() < *n { + point_list2.append(&mut point_list.to_vec()); + } + } + Replicate::Times(n) => { + for _ in 0..*n { + point_list2.append(&mut point_list.to_vec()); + } + } + } + point_list2 } } - diff --git a/src/transformer/rotate.rs b/src/transformer/rotate.rs index 0dac920..978b8b7 100644 --- a/src/transformer/rotate.rs +++ b/src/transformer/rotate.rs @@ -1,35 +1,32 @@ -use crate::transformer::Transformers; use crate::point::Point; +use crate::transformer::Transformers; use crate::worldstate::WorldState; -use serde::{Serialize,Deserialize}; +use serde::{Deserialize, Serialize}; //use std::f32::consts::PI; /// Rotate -#[derive(Serialize,Deserialize,Debug,Clone,Copy)] +#[derive(Serialize, Deserialize, Debug, Clone, Copy)] pub struct Rotate { cx: f32, cy: f32, angle: f32, - } impl Transformers for Rotate { fn apply(&self, point_list: &[Point], _ws: &WorldState) -> Vec { - point_list.iter() - .map(| pt | { - let dx = pt.x - self.cx; - let dy = pt.y - self.cy; - let cos = self.angle.cos(); - let sin = self.angle.sin(); - let x = (dx * cos - dy * sin) + self.cx; - let y = (dx * sin + dy * cos) + self.cy; - Point { x, - y, - ..*pt - } - }).collect() + point_list + .iter() + .map(|pt| { + let dx = pt.x - self.cx; + let dy = pt.y - self.cy; + let cos = self.angle.cos(); + let sin = self.angle.sin(); + let x = (dx * cos - dy * sin) + self.cx; + let y = (dx * sin + dy * cos) + self.cy; + Point { x, y, ..*pt } + }) + .collect() } } - diff --git a/src/transformer/translate.rs b/src/transformer/translate.rs index a0262a1..38b87a7 100644 --- a/src/transformer/translate.rs +++ b/src/transformer/translate.rs @@ -1,24 +1,25 @@ -use crate::transformer::Transformers; use crate::point::Point; +use crate::transformer::Transformers; use crate::worldstate::WorldState; -use serde::{Serialize,Deserialize}; +use serde::{Deserialize, Serialize}; /// Translate -#[derive(Serialize,Deserialize,Debug,Clone,Copy)] +#[derive(Serialize, Deserialize, Debug, Clone, Copy)] pub struct Translate { pub x: f32, - pub y: f32 + pub y: f32, } impl Transformers for Translate { - fn apply(&self, point_list: &[Point], _ws: &WorldState) -> Vec { - point_list.iter() - .map(| pt | { - Point { x: pt.x + self.x, - y: pt.y + self.y, - ..*pt - } - }).collect() + fn apply(&self, point_list: &[Point], _ws: &WorldState) -> Vec { + point_list + .iter() + .map(|pt| Point { + x: pt.x + self.x, + y: pt.y + self.y, + ..*pt + }) + .collect() } } diff --git a/src/worldstate.rs b/src/worldstate.rs index e03edb1..a6a7bff 100644 --- a/src/worldstate.rs +++ b/src/worldstate.rs @@ -1,52 +1,54 @@ -use crate::point::{Point, Color}; -use nalgebra::base::{Matrix3, Matrix1x3}; use crate::errors::{LJError, LJResult}; +use crate::point::{Color, Point}; use log::debug; +use nalgebra::base::{Matrix1x3, Matrix3}; #[derive(Debug, Default)] pub struct EDH { - pub matrix: Matrix3, + pub matrix: Matrix3, } impl EDH { - pub fn new(vec: Vec>) -> LJResult { - if vec.len() != 3 || - vec[0].len() != 3 || - vec[1].len() != 3 || - vec[2].len() != 3 { - return Err(Box::new(LJError::BadEDH)); - } + pub fn new(vec: Vec>) -> LJResult { + if vec.len() != 3 || vec[0].len() != 3 || vec[1].len() != 3 || vec[2].len() != 3 { + return Err(Box::new(LJError::BadEDH)); + } - // this is the matrix already transposed. - let matrix = Matrix3::new(vec[0][0], vec[1][0], vec[2][0], - vec[0][1], vec[1][1], vec[2][1], - vec[0][2], vec[1][2], vec[2][2]); + // this is the matrix already transposed. + let matrix = Matrix3::new( + vec[0][0], vec[1][0], vec[2][0], vec[0][1], vec[1][1], vec[2][1], vec[0][2], vec[1][2], + vec[2][2], + ); - Ok(EDH { matrix }) - } + Ok(EDH { matrix }) + } - pub fn apply(&self, point: &Point) -> Point { - let p = Matrix1x3::new(point.x, point.y, 1.0); - let p = p * self.matrix; - let new_p = Point { x: p[0] / p[2], y: p[1] / p[2], ..*point }; + pub fn apply(&self, point: &Point) -> Point { + let p = Matrix1x3::new(point.x, point.y, 1.0); + let p = p * self.matrix; + let new_p = Point { + x: p[0] / p[2], + y: p[1] / p[2], + ..*point + }; - debug!("{:?} => {:?}", point, new_p); + debug!("{:?} => {:?}", point, new_p); - new_p - } + new_p + } } #[derive(Debug, Default)] pub struct WorldState { - pub edh: EDH, - pub resampler: Vec, - pub client_key: String, - pub intensity: u8, - pub kpps: u32, - pub color: Color, - pub draw_black: bool, - pub draw_grid: bool, - pub grid: Vec, + pub edh: EDH, + pub resampler: Vec, + pub client_key: String, + pub intensity: u8, + pub kpps: u32, + pub color: Color, + pub draw_black: bool, + pub draw_grid: bool, + pub grid: Vec, } impl WorldState {} diff --git a/tests/test_conf.rs b/tests/test_conf.rs index 0dcb123..0be3a63 100644 --- a/tests/test_conf.rs +++ b/tests/test_conf.rs @@ -1,4 +1,4 @@ -use lj_rust::conf::{Conf, DacFamily, HeliosConf, EtherDreamConf}; +use lj_rust::conf::{Conf, DacFamily, EtherDreamConf, HeliosConf}; #[test] fn it_loads_a_valid_conf() {