Compare commits

...

4 Commits

Author SHA1 Message Date
alban
e148a1ec5e fix: cleanup for errors 2023-07-20 21:34:23 +02:00
alban
f45b9e5748 fix: dac/etherdream should work 2023-07-20 21:12:34 +02:00
alban
810a3677de fix: populate redis general geometry 2023-07-20 21:12:17 +02:00
alban
71fabde385 fix: Resampler should have correct type 2023-07-20 21:11:57 +02:00
6 changed files with 203 additions and 147 deletions

View File

@ -3,148 +3,149 @@ extern crate ether_dream;
use ether_dream::dac; use ether_dream::dac;
fn main() { fn main() {
println!("Listening for an Ether Dream DAC..."); println!("Listening for an Ether Dream DAC...");
let (dac_broadcast, source_addr) = ether_dream::recv_dac_broadcasts() let (dac_broadcast, source_addr) = ether_dream::recv_dac_broadcasts()
.expect("failed to bind to UDP socket") .expect("failed to bind to UDP socket")
.filter_map(Result::ok) .filter_map(Result::ok)
.next() .next()
.unwrap(); .unwrap();
let mac_address = dac::MacAddress(dac_broadcast.mac_address); let mac_address = dac::MacAddress(dac_broadcast.mac_address);
println!( println!(
"Discovered DAC \"{}\" at \"{}\"! Connecting...", "Discovered DAC \"{}\" at \"{}\"! Connecting...",
mac_address, source_addr mac_address, source_addr
); );
// Establish the TCP connection. // Establish the TCP connection.
let mut stream = dac::stream::connect(&dac_broadcast, source_addr.ip().clone()).unwrap(); let mut stream = dac::stream::connect(&dac_broadcast, source_addr.ip().clone()).unwrap();
// If we want to create an animation (in our case a moving sine wave) we need a frame rate. // If we want to create an animation (in our case a moving sine wave) we need a frame rate.
let frames_per_second = 60.0; let frames_per_second = 60.0;
// Lets use the DAC at an eighth the maximum scan rate. // Lets use the DAC at an eighth the maximum scan rate.
let points_per_second = stream.dac().max_point_rate / 32; let points_per_second = stream.dac().max_point_rate / 32;
// Determine the number of points per frame given our target frame and point rates. // Determine the number of points per frame given our target frame and point rates.
let points_per_frame = (points_per_second as f32 / frames_per_second) as u16; let points_per_frame = (points_per_second as f32 / frames_per_second) as u16;
println!( println!(
"Preparing for playback:\n\tframe_hz: {}\n\tpoint_hz: {}\n\tpoints_per_frame: {}\n", "Preparing for playback:\n\tframe_hz: {}\n\tpoint_hz: {}\n\tpoints_per_frame: {}\n",
frames_per_second, points_per_second, points_per_frame frames_per_second, points_per_second, points_per_frame
); );
// Prepare the DAC's playback engine and await the repsonse. // Prepare the DAC's playback engine and await the repsonse.
stream stream
.queue_commands() .queue_commands()
.prepare_stream() .prepare_stream()
.submit() .submit()
.err() .err()
.map(|err| { .map(|err| {
eprintln!( eprintln!(
"err occurred when submitting PREPARE_STREAM \ "err occurred when submitting PREPARE_STREAM \
command and listening for response: {}", command and listening for response: {}",
err err
); );
}); });
println!("Beginning playback!"); println!("Beginning playback!");
// The sine wave used to generate points. // The sine wave used to generate points.
let mut sine_wave = SineWave { let mut sine_wave = SineWave {
point: 0, point: 0,
points_per_frame, points_per_frame,
frames_per_second, frames_per_second,
}; };
// Queue the initial frame and tell the DAC to begin producing output. // Queue the initial frame and tell the DAC to begin producing output.
let n_points = points_to_generate(stream.dac()); let n_points = points_to_generate(stream.dac());
stream stream
.queue_commands() .queue_commands()
.data(sine_wave.by_ref().take(n_points)) .data(sine_wave.by_ref().take(n_points))
.begin(0, points_per_second) .begin(0, points_per_second)
.submit() .submit()
.err() .err()
.map(|err| { .map(|err| {
eprintln!( eprintln!(
"err occurred when submitting initial DATA and BEGIN \ "err occurred when submitting initial DATA and BEGIN \
commands and listening for response: {}", commands and listening for response: {}",
err err
); );
}); });
eprintln!("Stream dac{:?}", stream.dac());
// Loop and continue to send points forever. // Loop and continue to send points forever.
loop { loop {
// Determine how many points the DAC can currently receive. // Determine how many points the DAC can currently receive.
let n_points = points_to_generate(stream.dac()); let n_points = points_to_generate(stream.dac());
if let Err(err) = stream if let Err(err) = stream
.queue_commands() .queue_commands()
.data(sine_wave.by_ref().take(n_points)) .data(sine_wave.by_ref().take(n_points))
.submit() .submit()
{ {
eprintln!( eprintln!(
"err occurred when submitting DATA command and listening \ "err occurred when submitting DATA command and listening \
for response: {}", for response: {}",
err err
); );
break; break;
} }
} }
// Tell the DAC to stop producing output and return to idle. Wait for the response. // Tell the DAC to stop producing output and return to idle. Wait for the response.
// //
// Note that the DAC is commanded to stop on `Drop` if this is not called and any errors // Note that the DAC is commanded to stop on `Drop` if this is not called and any errors
// produced are ignored. // produced are ignored.
stream stream
.queue_commands() .queue_commands()
.stop() .stop()
.submit() .submit()
.expect("err occurred when submitting STOP command and listening for response"); .expect("err occurred when submitting STOP command and listening for response");
} }
// Determine the number of points needed to fill the DAC. // Determine the number of points needed to fill the DAC.
fn points_to_generate(dac: &ether_dream::dac::Dac) -> usize { fn points_to_generate(dac: &ether_dream::dac::Dac) -> usize {
dac.buffer_capacity as usize - 1 - dac.status.buffer_fullness as usize dac.buffer_capacity as usize - 1 - dac.status.buffer_fullness as usize
} }
// An iterator that endlessly generates a sine wave of DAC points. // An iterator that endlessly generates a sine wave of DAC points.
// //
// The sine wave oscillates at a rate of once per second. // The sine wave oscillates at a rate of once per second.
struct SineWave { struct SineWave {
point: u32, point: u32,
points_per_frame: u16, points_per_frame: u16,
frames_per_second: f32, frames_per_second: f32,
} }
impl Iterator for SineWave { impl Iterator for SineWave {
type Item = ether_dream::protocol::DacPoint; type Item = ether_dream::protocol::DacPoint;
fn next(&mut self) -> Option<Self::Item> { fn next(&mut self) -> Option<Self::Item> {
let coloured_points_per_frame = self.points_per_frame - 1; let coloured_points_per_frame = self.points_per_frame - 1;
let i = (self.point % self.points_per_frame as u32) as u16; let i = (self.point % self.points_per_frame as u32) as u16;
let hz = 1.0; let hz = 1.0;
let fract = i as f32 / coloured_points_per_frame as f32; let fract = i as f32 / coloured_points_per_frame as f32;
let phase = (self.point as f32 / coloured_points_per_frame as f32) / self.frames_per_second; let phase = (self.point as f32 / coloured_points_per_frame as f32) / self.frames_per_second;
let amp = (hz * (fract + phase) * 2.0 * std::f32::consts::PI).sin(); let amp = (hz * (fract + phase) * 2.0 * std::f32::consts::PI).sin();
let (r, g, b) = match i { let (r, g, b) = match i {
i if i == coloured_points_per_frame || i < 13 => (0, 0, 0), i if i == coloured_points_per_frame || i < 13 => (0, 0, 0),
_ => (std::u16::MAX, std::u16::MAX, std::u16::MAX), _ => (std::u16::MAX, std::u16::MAX, std::u16::MAX),
}; };
let x_min = std::i16::MIN; let x_min = std::i16::MIN;
let x_max = std::i16::MAX; let x_max = std::i16::MAX;
let x = (x_min as f32 + fract * (x_max as f32 - x_min as f32)) as i16; let x = (x_min as f32 + fract * (x_max as f32 - x_min as f32)) as i16;
let y = (amp * x_max as f32) as i16; let y = (amp * x_max as f32) as i16;
let control = 0; let control = 0;
let (u1, u2) = (0, 0); let (u1, u2) = (0, 0);
let p = ether_dream::protocol::DacPoint { let p = ether_dream::protocol::DacPoint {
control, control,
x, x,
y, y,
i, i,
r, r,
g, g,
b, b,
u1, u1,
u2, u2,
}; };
self.point += 1; self.point += 1;
Some(p) Some(p)
} }
} }

View File

@ -22,7 +22,7 @@ fn do_something() -> redis::RedisResult<()> {
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("/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("/kpps/0", "5000")?;
let _ = con.set("/intensity/0", "255")?; let _ = con.set("/intensity/0", "255")?;
let _ = con.set("/pl/0/0", "[(-300, 300, 0), (-300, -300, 65280), (300, -300, 65280), (300, 300, 65280), (-300, 300, 65280)]")?; let _ = con.set("/pl/0/0", "[(1000, 2000, 0), (1000, 1000, 65535), (2000, 1000, 65535), (2000, 2000, 65535), (1000, 2000, 65535)]")?;
Ok(()) Ok(())
} }

View File

@ -9,7 +9,7 @@ use crate::conf::EtherDreamConf;
use crate::device::{Device, Status, PlaybackState}; use crate::device::{Device, Status, PlaybackState};
use crate::errors::{LJError, LJResult}; use crate::errors::{LJError, LJResult};
use crate::point::{Color, Point}; use crate::point::{Color, Point};
use ether_dream::protocol::{DacBroadcast, DacPoint, DacResponse}; use ether_dream::protocol::{DacBroadcast, DacResponse};
use log::{debug, info, warn}; use log::{debug, info, warn};
#[warn(dead_code)] #[warn(dead_code)]
@ -71,6 +71,7 @@ impl EtherdreamDevice {
Err(Box::new(LJError::EtherdreamConnectError(err))) Err(Box::new(LJError::EtherdreamConnectError(err)))
} }
Ok((dac, source_addr)) => { Ok((dac, source_addr)) => {
info!("Trying to open TCP stream...");
let stream = EtherdreamDevice::get_tcp_stream(&dac, &source_addr)?; let stream = EtherdreamDevice::get_tcp_stream(&dac, &source_addr)?;
info!("Finished configuring DAC and TCP stream."); info!("Finished configuring DAC and TCP stream.");
Ok((dac, source_addr, stream)) Ok((dac, source_addr, stream))
@ -87,18 +88,29 @@ impl EtherdreamDevice {
Err(err) => warn!("err occurred when submitting PREPARE_STREAM command and listening for response: {}",err), Err(err) => warn!("err occurred when submitting PREPARE_STREAM command and listening for response: {}",err),
Ok(_) => info!("Prepared Stream.") Ok(_) => info!("Prepared Stream.")
} }
let begin_list = vec![ // If we want to create an animation (in our case a moving sine wave) we need a frame rate.
DacPoint { control: 0, x: 0, y: 0, i: 255, r: 0, g: 0, b: 0, u1: 0, u2: 0 }, let frames_per_second = 60.0;
]; // Lets use the DAC at an eighth the maximum scan rate.
let points_per_second = stream.dac().max_point_rate / 32; let points_per_second = stream.dac().max_point_rate / 32;
// Determine the number of points per frame given our target frame and point rates.
let points_per_frame = (points_per_second as f32 / frames_per_second) as u16;
let mut sine_wave = SineWave {
point: 0,
points_per_frame,
frames_per_second,
};
match stream match stream
.queue_commands() .queue_commands()
.data(begin_list.into_iter().take(1 as usize)) .data(sine_wave.by_ref().take(400))
// .data(begin_list.into_iter().take(400 as usize))
.begin(0, points_per_second) .begin(0, points_per_second)
.submit() { .submit() {
Err(err) => warn!("err occurred when submitting first data: {}",err), Err(err) => warn!("err occurred when submitting first data: {}",err),
Ok(_) => info!("Sent first data to Etherdream.") Ok(_) => info!("Sent first data to Etherdream.")
} }
Ok(stream) Ok(stream)
} }
@ -130,9 +142,9 @@ impl Device for EtherdreamDevice {
capacity: self.points_capacity(), capacity: self.points_capacity(),
lack: self.dac_response.to_string(), lack: self.dac_response.to_string(),
}; };
// info!("Dac Status: {:?} ", status ); // debug!("Dac Status: {:?} ", status );
// info!("Etherdream Dac {:?} ", self.dac ); // debug!("Etherdream Dac {:?} ", self.dac );
// info!("Stream dac{:?}", self.stream.dac()); debug!("Stream dac{:?}", self.stream.dac());
status status
} }
@ -143,14 +155,17 @@ impl Device for EtherdreamDevice {
let n_points = self.points_capacity(); let n_points = self.points_capacity();
// let n_points = &line.len(); // let n_points = &line.len();
debug!("Etherdream::device draw Generating {:?} points", n_points); debug!("Etherdream::device draw Generating {:?} points", n_points);
match self.stream match self.stream
.queue_commands() .queue_commands()
.data( .data(
line.into_iter() line.into_iter()
.map(|point| point.into()) .map(|point| point.into())
// .take(line.len() as usize)
.take(n_points as usize) .take(n_points as usize)
) )
// .data(sine_wave.by_ref().take(n_points as usize))
.submit() { .submit() {
Err(err) => { Err(err) => {
// We should account for // We should account for
@ -170,7 +185,6 @@ impl Device for EtherdreamDevice {
err.response.response err.response.response
} }
}; };
} }
Ok(_) => { Ok(_) => {
self.dac_response = DacResponse::ACK; self.dac_response = DacResponse::ACK;
@ -220,3 +234,48 @@ impl Device for EtherdreamDevice {
] ]
} }
} }
// An iterator that endlessly generates a sine wave of DAC points.
//
// The sine wave oscillates at a rate of once per second.
struct SineWave {
point: u32,
points_per_frame: u16,
frames_per_second: f32,
}
impl Iterator for SineWave {
type Item = ether_dream::protocol::DacPoint;
fn next(&mut self) -> Option<Self::Item> {
let coloured_points_per_frame = self.points_per_frame - 1;
let i = (self.point % self.points_per_frame as u32) as u16;
let hz = 1.0;
let fract = i as f32 / coloured_points_per_frame as f32;
let phase = (self.point as f32 / coloured_points_per_frame as f32) / self.frames_per_second;
let amp = (hz * (fract + phase) * 2.0 * std::f32::consts::PI).sin();
let (r, g, b) = match i {
i if i == coloured_points_per_frame || i < 13 => (0, 0, 0),
_ => (std::u16::MAX, std::u16::MAX, std::u16::MAX),
};
let x_min = std::i16::MIN;
let x_max = std::i16::MAX;
let x = (x_min as f32 + fract * (x_max as f32 - x_min as f32)) as i16;
let y = (amp * x_max as f32) as i16;
let control = 0;
let (u1, u2) = (0, 0);
let p = ether_dream::protocol::DacPoint {
control,
x,
y,
i,
r,
g,
b,
u1,
u2,
};
debug!("{:?}",p);
self.point += 1;
Some(p)
}
}

View File

@ -1,3 +1,4 @@
use std::time::SystemTime;
/// ///
/// Configure udev: /// Configure udev:
/// https://github.com/Grix/helios_dac/blob/master/docs/udev_rules_for_linux.md /// https://github.com/Grix/helios_dac/blob/master/docs/udev_rules_for_linux.md
@ -14,7 +15,7 @@ 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::{Color, Point}; use crate::point::{Color, Point};
use chrono::Utc; use chrono::{DateTime, Utc};
pub struct HeliosDevice { pub struct HeliosDevice {
pub conf: HeliosConf, pub conf: HeliosConf,
@ -34,13 +35,17 @@ impl HeliosDevice {
return Err(Box::new(LJError::HeliosDeviceMissing)); return Err(Box::new(LJError::HeliosDeviceMissing));
}; };
let dac = device.open()?; let dac = device.open()?;
let now = SystemTime::now();
let now: DateTime<Utc> = now.into();
let last_traced_at = now.to_rfc3339();
Ok(Self { Ok(Self {
conf: (*conf).clone(), conf: (*conf).clone(),
dac, dac,
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,
}) })
} }
} }

View File

@ -1,15 +1,5 @@
use ether_dream::protocol::DacPoint; use ether_dream::protocol::DacPoint;
fn clamp(val: f32, min: f32, max: f32) -> f32 {
if val < min {
return min;
}
if val > max {
return max;
}
val
}
#[derive(Debug, Clone, Copy, Default, PartialEq)] #[derive(Debug, Clone, Copy, Default, PartialEq)]
pub struct Point { pub struct Point {
pub x: f32, pub x: f32,
@ -55,21 +45,20 @@ impl From<Point> for helios_dac::Point {
} }
impl From<Point> for DacPoint { impl From<Point> for DacPoint {
fn from(pt: Point) -> DacPoint { fn from(pt: Point) -> DacPoint {
let control = 0; let control = 0;
let (u1, u2) = (0, 0); let (u1, u2) = (0, 0);
let i = 255; let i = 255;
let x = clamp(pt.x, -32000 as f32, 32000 as f32); let x = pt.x.clamp(-32000.0, 32000.0);
let y = clamp(pt.y, -32000 as f32, 32000 as f32); let y = pt.y.clamp(-32000.0, 32000.0);
DacPoint { DacPoint {
control, control,
x: x as i16, x: x as i16,
y: y as i16, y: y as i16,
i, i,
r: pt.color.r.into(), r: (pt.color.r as u16) * 255,
g: pt.color.g.into(), g: (pt.color.g as u16) * 255,
b: pt.color.b.into(), b: (pt.color.b as u16) * 255,
u1, u1,
u2, u2,
} }

View File

@ -17,6 +17,7 @@ pub enum Order {
Intensity, Intensity,
Kpps, Kpps,
ColorBalance, ColorBalance,
PowerOff
} }
impl TryFrom<u8> for Order { impl TryFrom<u8> for Order {
@ -39,13 +40,14 @@ impl TryFrom<u8> for Order {
6 => Intensity, 6 => Intensity,
7 => Kpps, 7 => Kpps,
8 => ColorBalance, 8 => ColorBalance,
9 => PowerOff,
_ => unreachable!() _ => unreachable!()
}) })
} }
} }
pub type Line = Vec<(f32, f32, u32)>; pub type Line = Vec<(f32, f32, u32)>;
pub type Resampler = Vec<(f32,f32)>; pub type Resampler = Vec<Vec<(f32,f32)>>;
pub struct RedisCtrl { pub struct RedisCtrl {
pub client: Client, pub client: Client,