Compare commits
10 Commits
69b619dbf5
...
4a87821c22
Author | SHA1 | Date | |
---|---|---|---|
|
4a87821c22 | ||
|
fd9ad62278 | ||
|
60121a01c2 | ||
|
afaa397a52 | ||
|
86e32e08b1 | ||
|
ffba2efeed | ||
|
ee4d0e049f | ||
|
6fa8846997 | ||
|
e4580e0e53 | ||
|
76cc1c1275 |
@ -10,6 +10,7 @@ chrono = "0.4.26"
|
|||||||
config = "0.13.3"
|
config = "0.13.3"
|
||||||
ctrlc = "3.4.0"
|
ctrlc = "3.4.0"
|
||||||
env_logger = "0.10.0"
|
env_logger = "0.10.0"
|
||||||
|
ether-dream = "0.2.5"
|
||||||
helios-dac = { version = "0.1", default-features = false, features = ["native"] }
|
helios-dac = { version = "0.1", default-features = false, features = ["native"] }
|
||||||
|
|
||||||
log = "0.4.18"
|
log = "0.4.18"
|
||||||
|
16
README.md
Normal file
16
README.md
Normal file
@ -0,0 +1,16 @@
|
|||||||
|
# LJ rust
|
||||||
|
|
||||||
|
## Crashcourse
|
||||||
|
|
||||||
|
```shell
|
||||||
|
|
||||||
|
# Copy and edit the settings file
|
||||||
|
$ cp copyme.settings.toml settings.toml
|
||||||
|
|
||||||
|
# Populate the redis database
|
||||||
|
$ cargo run --example populate_redis
|
||||||
|
|
||||||
|
# Run
|
||||||
|
$ cargo run --release
|
||||||
|
|
||||||
|
```
|
@ -1,4 +1,4 @@
|
|||||||
# file: Settings.toml
|
# file: settings.toml
|
||||||
# Rename me !
|
# Rename me !
|
||||||
|
|
||||||
# The main key of your laser in LJ
|
# The main key of your laser in LJ
|
||||||
@ -12,20 +12,26 @@ redis_url = "redis://127.0.0.1:6379/"
|
|||||||
|
|
||||||
# Either Helios or Etherdream
|
# Either Helios or Etherdream
|
||||||
# For Helios. USB Device Id of the DAC
|
# For Helios. USB Device Id of the DAC
|
||||||
[dac.helios]
|
#[dac.helios]
|
||||||
id = 0
|
#id = 0
|
||||||
|
|
||||||
# For dummy dac:
|
# For dummy dac:
|
||||||
# [dac.dummy]
|
[dac.dummy]
|
||||||
|
|
||||||
# For Etherdream. IP of the DAC
|
# For Etherdream. IP of the DAC
|
||||||
# [dac.etherdream]
|
# [dac.etherdream]
|
||||||
# url = "192.168.1.68"
|
# ip = "192.168.1.68"
|
||||||
|
|
||||||
[[transformers]]
|
[[transformers]]
|
||||||
[transformers.translate]
|
[transformers.translate]
|
||||||
x = 2000
|
x = 2000
|
||||||
y = 2000
|
y = 2000
|
||||||
|
|
||||||
[[transformers]]
|
[[transformers]]
|
||||||
[transformers.replicate]
|
[transformers.replicate]
|
||||||
Until = 48
|
Until = 48
|
||||||
|
|
||||||
|
# Never remove this : this is mandatory
|
||||||
|
[[transformers]]
|
||||||
|
[transformers.intensity]
|
||||||
|
|
||||||
|
24
examples/populate_redis.rs
Normal file
24
examples/populate_redis.rs
Normal file
@ -0,0 +1,24 @@
|
|||||||
|
///
|
||||||
|
/// $ cargo run --example populate_redis
|
||||||
|
///
|
||||||
|
|
||||||
|
use redis::{
|
||||||
|
//RedisResult,
|
||||||
|
Client,
|
||||||
|
Commands,
|
||||||
|
Connection,
|
||||||
|
};
|
||||||
|
|
||||||
|
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")?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
fn main() {
|
||||||
|
_ = do_something();
|
||||||
|
}
|
@ -30,7 +30,7 @@ pub struct HeliosConf {
|
|||||||
|
|
||||||
#[derive(Serialize, Deserialize, Debug, Clone)]
|
#[derive(Serialize, Deserialize, Debug, Clone)]
|
||||||
pub struct EtherDreamConf {
|
pub struct EtherDreamConf {
|
||||||
pub url: String
|
pub ip: String
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Serialize, Deserialize, Debug, Clone)]
|
#[derive(Serialize, Deserialize, Debug, Clone)]
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
|
|
||||||
mod helios;
|
mod helios;
|
||||||
mod dummy;
|
mod dummy;
|
||||||
|
mod etherdream;
|
||||||
|
|
||||||
use std::fmt;
|
use std::fmt;
|
||||||
use crate::conf::{Conf, DacFamily /*EtherDreamConf, HeliosConf*/};
|
use crate::conf::{Conf, DacFamily /*EtherDreamConf, HeliosConf*/};
|
||||||
@ -9,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,
|
||||||
@ -26,13 +27,15 @@ 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)]
|
#[derive(Debug)]
|
||||||
@ -41,7 +44,7 @@ pub struct Status {
|
|||||||
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)
|
||||||
@ -49,20 +52,21 @@ 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>,
|
||||||
speed: u32,
|
speed: u32,
|
||||||
) -> LJResult<()>;
|
) -> LJResult<()>;
|
||||||
fn stop(&mut self) -> LJResult<()>;
|
fn stop(&mut self) -> LJResult<()>;
|
||||||
|
fn grid(&mut self) -> Vec<Point>;
|
||||||
}
|
}
|
||||||
|
|
||||||
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)
|
||||||
}
|
}
|
||||||
|
@ -1,31 +0,0 @@
|
|||||||
|
|
||||||
/*
|
|
||||||
self.protocol_version,
|
|
||||||
self.le_state,
|
|
||||||
self.playback_state,
|
|
||||||
self.source,
|
|
||||||
self.le_flags,
|
|
||||||
self.playback_flags,
|
|
||||||
self.source_flags,
|
|
||||||
self.fullness,
|
|
||||||
self.point_rate,
|
|
||||||
self.point_count
|
|
||||||
*/
|
|
||||||
|
|
||||||
pub struct Status {
|
|
||||||
pub active: bool,
|
|
||||||
pub last_traced_at: String,
|
|
||||||
pub properties: Vec<String>
|
|
||||||
}
|
|
||||||
|
|
||||||
pub trait Device {
|
|
||||||
/**
|
|
||||||
fn intersect(&self, orig : &Vec3, dir : &Vec3) -> Option<Float>;
|
|
||||||
fn get_surface(&self, v : &Vec3) -> Vec3;
|
|
||||||
fn get_normal(&self, v : &Vec3) -> Vec3;
|
|
||||||
fn get_material(&self) -> &dyn Material;
|
|
||||||
**/
|
|
||||||
|
|
||||||
fn status( &self ) -> Status;
|
|
||||||
|
|
||||||
}
|
|
@ -1,6 +1,6 @@
|
|||||||
use crate::device::{Device, Status, PlaybackState};
|
use crate::device::{Device, Status, PlaybackState};
|
||||||
use crate::errors::LJResult;
|
use crate::errors::LJResult;
|
||||||
use crate::point::Point;
|
use crate::point::{Color, Point};
|
||||||
use log::debug;
|
use log::debug;
|
||||||
|
|
||||||
pub struct DummyDevice {
|
pub struct DummyDevice {
|
||||||
@ -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()],
|
||||||
@ -35,4 +35,9 @@ impl Device for DummyDevice {
|
|||||||
fn stop(&mut self) -> LJResult<()> {
|
fn stop(&mut self) -> LJResult<()> {
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
fn grid(&mut self) -> Vec<Point> {
|
||||||
|
vec!(
|
||||||
|
Point{ x: 0 as f32, y: 0 as f32, color:Color{ r: 0, g: 0, b: 0 }}
|
||||||
|
)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
153
src/device/etherdream.rs
Normal file
153
src/device/etherdream.rs
Normal file
@ -0,0 +1,153 @@
|
|||||||
|
use std::time;
|
||||||
|
use std::net::SocketAddr;
|
||||||
|
use ether_dream::dac::stream::connect;
|
||||||
|
use ether_dream::dac::Stream;
|
||||||
|
|
||||||
|
use crate::conf::EtherDreamConf;
|
||||||
|
use crate::device::{Device, Status, PlaybackState};
|
||||||
|
use crate::errors::{LJError, LJResult};
|
||||||
|
use crate::point::{Color, Point};
|
||||||
|
use ether_dream::protocol::{DacBroadcast, DacStatus};
|
||||||
|
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,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl EtherdreamDevice {
|
||||||
|
pub fn new(conf: &EtherDreamConf) -> LJResult<Self> {
|
||||||
|
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<Stream> {
|
||||||
|
// 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 \
|
||||||
|
command and listening for response: {}",
|
||||||
|
err
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
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(())
|
||||||
|
}
|
||||||
|
|
||||||
|
// 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();
|
||||||
|
|
||||||
|
// "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),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn draw(&mut self,
|
||||||
|
line: Vec<Point>,
|
||||||
|
_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 grid(&mut self) -> Vec<Point> {
|
||||||
|
vec!(
|
||||||
|
Point { x: 0.0, y: 0.0, color: Color { r: 255, g: 255, b: 255 } }
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
@ -13,7 +13,7 @@ use helios_dac::{
|
|||||||
use crate::conf::HeliosConf;
|
use crate::conf::HeliosConf;
|
||||||
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 chrono::Utc;
|
||||||
|
|
||||||
pub struct HeliosDevice {
|
pub struct HeliosDevice {
|
||||||
@ -22,7 +22,7 @@ pub struct HeliosDevice {
|
|||||||
sent_points: u16,
|
sent_points: u16,
|
||||||
state: PlaybackState,
|
state: PlaybackState,
|
||||||
lack: String,
|
lack: String,
|
||||||
last_traced_at : String
|
last_traced_at: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl HeliosDevice {
|
impl HeliosDevice {
|
||||||
@ -40,17 +40,16 @@ impl HeliosDevice {
|
|||||||
sent_points: 0,
|
sent_points: 0,
|
||||||
state: PlaybackState::PREPARE,
|
state: PlaybackState::PREPARE,
|
||||||
lack: "".to_string(),
|
lack: "".to_string(),
|
||||||
last_traced_at: "1985-04-12T23:20:50.52Z".to_string()
|
last_traced_at: "1985-04-12T23:20:50.52Z".to_string(),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
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(),
|
||||||
properties: vec!["foo".to_string()],
|
properties: vec!["foo".to_string()],
|
||||||
playback_state: self.state,
|
playback_state: self.state,
|
||||||
capacity: self.sent_points,
|
capacity: self.sent_points,
|
||||||
@ -78,4 +77,27 @@ impl Device for HeliosDevice {
|
|||||||
self.dac.stop()?;
|
self.dac.stop()?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn grid(&mut self) -> Vec<Point> {
|
||||||
|
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 },
|
||||||
|
]
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
@ -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;
|
||||||
|
|
||||||
@ -7,41 +7,44 @@ pub type LJResult<T> = Result<T, Box<dyn std::error::Error>>;
|
|||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub enum LJError {
|
pub enum LJError {
|
||||||
Config(ConfigError),
|
Config(ConfigError),
|
||||||
RedisConnect(RedisError),
|
RedisConnect(RedisError),
|
||||||
HeliosDeviceMissing,
|
HeliosDeviceMissing,
|
||||||
BadEDH
|
BadEDH,
|
||||||
|
EtherdreamConnectError(io::Error),
|
||||||
}
|
}
|
||||||
|
|
||||||
impl fmt::Display for LJError {
|
impl fmt::Display for LJError {
|
||||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||||
use LJError::*;
|
use LJError::*;
|
||||||
|
|
||||||
match self {
|
|
||||||
Config(err) => {
|
|
||||||
write!(f, "unable to load config file: {err}")
|
|
||||||
},
|
|
||||||
|
|
||||||
RedisConnect(err) => {
|
match self {
|
||||||
write!(f, "unable to connect to redis server: {err}")
|
Config(err) => {
|
||||||
},
|
write!(f, "unable to load config file: {err}")
|
||||||
HeliosDeviceMissing => {
|
}
|
||||||
write!(f, "helios device not found")
|
RedisConnect(err) => {
|
||||||
},
|
write!(f, "unable to connect to redis server: {err}")
|
||||||
BadEDH => {
|
}
|
||||||
write!(f, "EDH matrix is not a 3x3 matrix")
|
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 {
|
impl Error for LJError {
|
||||||
fn source(&self) -> Option<&(dyn Error + 'static)> {
|
fn source(&self) -> Option<&(dyn Error + 'static)> {
|
||||||
use LJError::*;
|
use LJError::*;
|
||||||
|
|
||||||
match self {
|
match self {
|
||||||
RedisConnect(err) => Some(err),
|
RedisConnect(err) => Some(err),
|
||||||
_ => None
|
_ => None
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
135
src/main.rs
135
src/main.rs
@ -16,7 +16,7 @@ use std::sync::Arc;
|
|||||||
use redis_ctrl::{RedisCtrl, Order};
|
use redis_ctrl::{RedisCtrl, Order};
|
||||||
use conf::Conf;
|
use conf::Conf;
|
||||||
use errors::LJResult;
|
use errors::LJResult;
|
||||||
use point::{Point,Color};
|
use point::{Point, Color};
|
||||||
use transformer::Transformers;
|
use transformer::Transformers;
|
||||||
use log::{LevelFilter, info, /* warn, */ error};
|
use log::{LevelFilter, info, /* warn, */ error};
|
||||||
use env_logger::Builder;
|
use env_logger::Builder;
|
||||||
@ -50,9 +50,9 @@ fn run_all() -> LJResult<()> {
|
|||||||
// Setup Redis Service
|
// Setup Redis Service
|
||||||
let mut rs = RedisCtrl::new(&config.redis_url, &config.laser_id)?;
|
let mut rs = RedisCtrl::new(&config.redis_url, &config.laser_id)?;
|
||||||
|
|
||||||
let mut world_state = rs.init_world_state()?;
|
let mut world_state = rs.init_world_state().unwrap();
|
||||||
info!("WorldState: {:?}", world_state);
|
info!("WorldState: {:?}", world_state);
|
||||||
|
|
||||||
// Setup handler for interrupt Signals
|
// Setup handler for interrupt Signals
|
||||||
let running = Arc::new(AtomicBool::new(true));
|
let running = Arc::new(AtomicBool::new(true));
|
||||||
let r = running.clone();
|
let r = running.clone();
|
||||||
@ -62,6 +62,7 @@ fn run_all() -> LJResult<()> {
|
|||||||
|
|
||||||
// Setup Laser Device based on conf
|
// Setup Laser Device based on conf
|
||||||
let mut tracer = device_factory(&config)?;
|
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...
|
// can't work, but we can add + Debug to Device to make it work...
|
||||||
//dbg!(tracer);
|
//dbg!(tracer);
|
||||||
@ -74,46 +75,46 @@ fn run_all() -> LJResult<()> {
|
|||||||
rs.set_status(tracer.status())?;
|
rs.set_status(tracer.status())?;
|
||||||
|
|
||||||
let order = rs.get_order(config.laser_id)?;
|
let order = rs.get_order(config.laser_id)?;
|
||||||
// 0 : Draw Normal point list
|
|
||||||
// 2 : Draw BLACK point list
|
|
||||||
// 3 : Draw GRID point list
|
|
||||||
|
|
||||||
// /worldstate.rs
|
|
||||||
// /edh.rs
|
|
||||||
|
|
||||||
// 1 : Get the new EDH = reread redis key /EDH/lasernumber
|
|
||||||
// 4 : Resampler Change (longs and shorts lsteps)
|
|
||||||
// 5 : Client Key Change = reread redis key /clientkey
|
|
||||||
// 6 : Max Intensity Change = reread redis key /intensity
|
|
||||||
// 7 : kpps change = reread redis key /kpps
|
|
||||||
// 8 : color balance change = reread redis keys /red /green /blue
|
|
||||||
match order {
|
match order {
|
||||||
Order::Draw | Order::Black => {
|
Order::Draw | Order::Black | Order::Grid => {
|
||||||
world_state.black = order == Order::Black;
|
// 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(
|
let frame = get_next_frame(
|
||||||
&config,
|
&config,
|
||||||
&transformers,
|
&transformers,
|
||||||
&mut rs,
|
&mut rs,
|
||||||
&world_state
|
&world_state,
|
||||||
)?;
|
)?;
|
||||||
// For now, draw all the time
|
// For now, draw all the time
|
||||||
tracer.draw(frame, world_state.kpps)?;
|
tracer.draw(frame, world_state.kpps)?;
|
||||||
},
|
}
|
||||||
Order::Intensity => {
|
Order::Intensity => {
|
||||||
world_state.intensity = rs.get_int("intensity")?
|
// 6 : Max Intensity Change = reread redis key /intensity
|
||||||
.try_into()?;
|
world_state.intensity = rs.get_int("intensity")?
|
||||||
},
|
.try_into()?;
|
||||||
|
}
|
||||||
Order::Edh => {
|
Order::Edh => {
|
||||||
|
// 1 : Get the new EDH = reread redis key /EDH/lasernumber
|
||||||
world_state.edh = rs.get_edh()?;
|
world_state.edh = rs.get_edh()?;
|
||||||
},
|
}
|
||||||
Order::Kpps => {
|
Order::Kpps => {
|
||||||
|
// 7 : kpps change = reread redis key /kpps
|
||||||
world_state.kpps = rs.get_int("kpps")?;
|
world_state.kpps = rs.get_int("kpps")?;
|
||||||
|
}
|
||||||
|
Order::ClientKey => {
|
||||||
|
world_state.client_key = rs.get_client_key()?;
|
||||||
},
|
},
|
||||||
|
|
||||||
// Order::ClientKey => rs.client_key(),
|
|
||||||
// Order::ColorBalance => {},
|
// 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!("Order: {:?}", order);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -143,42 +144,50 @@ fn get_next_frame(
|
|||||||
config: &Conf,
|
config: &Conf,
|
||||||
transformers: &[Box<dyn Transformers>],
|
transformers: &[Box<dyn Transformers>],
|
||||||
rs: &mut RedisCtrl,
|
rs: &mut RedisCtrl,
|
||||||
world_state : &WorldState
|
world_state: &WorldState,
|
||||||
) -> LJResult<Vec<Point>> {
|
) -> LJResult<Vec<Point>> {
|
||||||
let format_key = format!("{}{}",
|
let format_key = format!("{}{}",
|
||||||
world_state.client_key,
|
world_state.client_key,
|
||||||
config.laser_id);
|
config.laser_id);
|
||||||
let line = rs.get_line(&format_key)?;
|
|
||||||
let mut line: Vec<Point> = line.into_iter()
|
|
||||||
.map(|tpl| tpl.into())
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
for transformer in transformers {
|
|
||||||
line = transformer.apply(&line, world_state);
|
|
||||||
}
|
|
||||||
|
|
||||||
info!("-> {}", world_state.black);
|
// Handle the grid case
|
||||||
|
|
||||||
// LIMITER and BLACK
|
let mut line: Vec<Point> = if world_state.draw_grid {
|
||||||
line = line.into_iter()
|
world_state.grid.clone()
|
||||||
.map(| p | {
|
} else {
|
||||||
|
let redis_line = rs.get_line(&format_key)?;
|
||||||
let color = if world_state.black {
|
redis_line.into_iter()
|
||||||
Color { r: 0, g: 0, b: 0 }
|
.map(|tpl| tpl.into())
|
||||||
} else {
|
.collect()
|
||||||
Color {
|
};
|
||||||
r: p.color.r.min(world_state.intensity),
|
|
||||||
g: p.color.g.min(world_state.intensity),
|
for transformer in transformers {
|
||||||
b: p.color.b.min(world_state.intensity)
|
line = transformer.apply(&line, world_state);
|
||||||
}
|
}
|
||||||
};
|
|
||||||
Point { color,
|
info!("Draw Black -> {}", world_state.draw_black);
|
||||||
..p
|
info!("Draw Grid -> {}", world_state.draw_grid);
|
||||||
}
|
|
||||||
})
|
// LIMITER and BLACK
|
||||||
.collect();
|
line = line.into_iter()
|
||||||
|
.map(|p| {
|
||||||
//info!("Line: {:?}", line);
|
let color = if world_state.draw_black {
|
||||||
Ok(line)
|
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)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
76
src/point.rs
76
src/point.rs
@ -1,34 +1,64 @@
|
|||||||
#[derive(Debug,Clone,Copy,Default,PartialEq)]
|
use ether_dream::protocol::DacPoint;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Copy, Default, PartialEq)]
|
||||||
pub struct Point {
|
pub struct Point {
|
||||||
pub x: f32,
|
pub x: f32,
|
||||||
pub y: f32,
|
pub y: f32,
|
||||||
pub color: Color
|
pub color: Color,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug,Clone,Copy,Default,PartialEq)]
|
#[derive(Debug, Clone, Copy, Default, PartialEq)]
|
||||||
pub struct Color {
|
pub struct Color {
|
||||||
pub r: u8,
|
pub r: u8,
|
||||||
pub g: u8,
|
pub g: u8,
|
||||||
pub b: u8
|
pub b: u8,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl From<(f32,f32,u32)> for Point {
|
impl From<Color> for u32 {
|
||||||
fn from((x, y, color) : (f32, f32, u32)) -> Point {
|
fn from(value: Color) -> Self {
|
||||||
let r = (color >> 16) as u8 ;
|
let r = value.r as u32;
|
||||||
let g = ((color >> 8) & 255) as u8 ;
|
let g = value.g as u32;
|
||||||
let b = (color & 255) as u8 ;
|
let b = (value.b) as u32;
|
||||||
Point { x, y, color: Color { r, g, b } }
|
(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 } }
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl From<Point> for helios_dac::Point {
|
impl From<Point> for helios_dac::Point {
|
||||||
fn from(pt: Point) -> helios_dac::Point {
|
fn from(pt: Point) -> helios_dac::Point {
|
||||||
let x = pt.x.clamp(0.0, 4095.0) as u16;
|
let x = pt.x.clamp(0.0, 4095.0) as u16;
|
||||||
let y = pt.y.clamp(0.0, 4095.0) as u16;
|
let y = pt.y.clamp(0.0, 4095.0) as u16;
|
||||||
helios_dac::Point {
|
helios_dac::Point {
|
||||||
coordinate: (x, y).into(),
|
coordinate: (x, y).into(),
|
||||||
color: helios_dac::Color::new(pt.color.r, pt.color.g, pt.color.b),
|
color: helios_dac::Color::new(pt.color.r, pt.color.g, pt.color.b),
|
||||||
intensity: 0xFF
|
intensity: 0xFF,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
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,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
@ -101,10 +101,10 @@ impl RedisCtrl {
|
|||||||
|
|
||||||
pub fn init_world_state( &mut self) -> LJResult<WorldState>{
|
pub fn init_world_state( &mut self) -> LJResult<WorldState>{
|
||||||
Ok(WorldState {
|
Ok(WorldState {
|
||||||
client_key: self.connection.get("/clientkey")?,
|
client_key: self.get_client_key().unwrap(),
|
||||||
edh: self.get_edh()?,
|
edh: self.get_edh().unwrap(),
|
||||||
kpps: self.get_int("kpps")?.try_into()?,
|
kpps: self.get_int("kpps").unwrap().try_into().unwrap(),
|
||||||
intensity: self.get_int("intensity")?.try_into()?,
|
intensity: self.get_int("intensity").unwrap().try_into().unwrap(),
|
||||||
..WorldState::default()
|
..WorldState::default()
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
@ -118,6 +118,11 @@ impl RedisCtrl {
|
|||||||
Ok(edh)
|
Ok(edh)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn get_client_key( &mut self ) -> LJResult<String> {
|
||||||
|
let key : String = self.connection.get("/clientkey")?;
|
||||||
|
Ok(key)
|
||||||
|
}
|
||||||
|
|
||||||
pub fn get_int(&mut self, key: &str ) -> LJResult<u32> {
|
pub fn get_int(&mut self, key: &str ) -> LJResult<u32> {
|
||||||
// Get new Int
|
// Get new Int
|
||||||
let fmt = format!("/{key}/{}", self.laser_id);
|
let fmt = format!("/{key}/{}", self.laser_id);
|
||||||
|
@ -1,52 +1,52 @@
|
|||||||
use crate::point::{Point,Color};
|
use crate::point::{Point, Color};
|
||||||
use nalgebra::base::{Matrix3,Matrix1x3};
|
use nalgebra::base::{Matrix3, Matrix1x3};
|
||||||
use crate::errors::{LJError,LJResult};
|
use crate::errors::{LJError, LJResult};
|
||||||
use log::debug;
|
use log::debug;
|
||||||
|
|
||||||
#[derive(Debug, Default)]
|
#[derive(Debug, Default)]
|
||||||
pub struct EDH {
|
pub struct EDH {
|
||||||
pub matrix: Matrix3<f32>
|
pub matrix: Matrix3<f32>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl EDH {
|
impl EDH {
|
||||||
pub fn new(vec: Vec<Vec<f32>>) -> LJResult<EDH> {
|
pub fn new(vec: Vec<Vec<f32>>) -> LJResult<EDH> {
|
||||||
if vec.len() != 3 ||
|
if vec.len() != 3 ||
|
||||||
vec[0].len() != 3 ||
|
vec[0].len() != 3 ||
|
||||||
vec[1].len() != 3 ||
|
vec[1].len() != 3 ||
|
||||||
vec[2].len() != 3 {
|
vec[2].len() != 3 {
|
||||||
return Err(Box::new(LJError::BadEDH));
|
return Err(Box::new(LJError::BadEDH));
|
||||||
}
|
}
|
||||||
|
|
||||||
// this is the matrix already transposed.
|
// this is the matrix already transposed.
|
||||||
let matrix = Matrix3::new(vec[0][0], vec[1][0], vec[2][0],
|
let matrix = Matrix3::new(vec[0][0], vec[1][0], vec[2][0],
|
||||||
vec[0][1], vec[1][1], vec[2][1],
|
vec[0][1], vec[1][1], vec[2][1],
|
||||||
vec[0][2], vec[1][2], vec[2][2]);
|
vec[0][2], vec[1][2], vec[2][2]);
|
||||||
|
|
||||||
Ok(EDH { matrix })
|
Ok(EDH { matrix })
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn apply(&self, point: &Point) -> Point {
|
pub fn apply(&self, point: &Point) -> Point {
|
||||||
let p = Matrix1x3::new(point.x, point.y, 1.0);
|
let p = Matrix1x3::new(point.x, point.y, 1.0);
|
||||||
let p = p * self.matrix;
|
let p = p * self.matrix;
|
||||||
let new_p = Point { x: p[0] / p[2], y: p[1] / p[2], ..*point };
|
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)]
|
#[derive(Debug, Default)]
|
||||||
pub struct WorldState {
|
pub struct WorldState {
|
||||||
pub edh: EDH,
|
pub edh: EDH,
|
||||||
pub resampler: Vec<f32>,
|
pub resampler: Vec<f32>,
|
||||||
pub client_key: String, //u8,
|
pub client_key: String,
|
||||||
pub intensity: u8,
|
pub intensity: u8,
|
||||||
pub kpps: u32,
|
pub kpps: u32,
|
||||||
pub color: Color,
|
pub color: Color,
|
||||||
pub black: bool
|
pub draw_black: bool,
|
||||||
|
pub draw_grid: bool,
|
||||||
|
pub grid: Vec<Point>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl WorldState {
|
impl WorldState {}
|
||||||
|
|
||||||
}
|
|
||||||
|
@ -12,4 +12,4 @@ redis_url = "redis://127.0.0.1:6379/"
|
|||||||
|
|
||||||
# For Etherdream. IP of the DAC
|
# For Etherdream. IP of the DAC
|
||||||
[dac.etherdream]
|
[dac.etherdream]
|
||||||
url = "192.168.1.68"
|
ip = "192.168.1.68"
|
||||||
|
@ -28,7 +28,7 @@ fn it_finds_struct_fields() {
|
|||||||
fn it_finds_etherdream_fields() {
|
fn it_finds_etherdream_fields() {
|
||||||
let config = Conf::new("tests/settings/valid_etherdream").unwrap();
|
let config = Conf::new("tests/settings/valid_etherdream").unwrap();
|
||||||
assert!(match config.dac {
|
assert!(match config.dac {
|
||||||
DacFamily::Etherdream(EtherDreamConf { url }) if url == "192.168.1.68" => true,
|
DacFamily::Etherdream(EtherDreamConf { ip }) if ip == "192.168.1.68" => true,
|
||||||
_ => false,
|
_ => false,
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user