fix: format using rustfmt
This commit is contained in:
parent
4ce35dfd0d
commit
ae14cb600d
36
src/conf.rs
36
src/conf.rs
@ -1,32 +1,28 @@
|
|||||||
|
|
||||||
use config::Config;
|
use config::Config;
|
||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
|
|
||||||
|
#[derive(Deserialize, Debug)]
|
||||||
#[derive(Deserialize,Debug)]
|
|
||||||
pub enum DacFamily {
|
pub enum DacFamily {
|
||||||
Helios,
|
Helios,
|
||||||
Etherdream
|
Etherdream,
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Deserialize,Debug)]
|
|
||||||
pub struct Conf {
|
|
||||||
laser_id : u8,
|
|
||||||
debug : bool,
|
|
||||||
redis_url : String,
|
|
||||||
dac_family: DacFamily,
|
|
||||||
dac_id : Option<u8>,
|
|
||||||
dac_url : Option<String>
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Deserialize, Debug)]
|
||||||
|
pub struct Conf {
|
||||||
|
laser_id: u8,
|
||||||
|
debug: bool,
|
||||||
|
redis_url: String,
|
||||||
|
dac_family: DacFamily,
|
||||||
|
dac_id: Option<u8>,
|
||||||
|
dac_url: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
pub fn load_config( path : &str )-> Result<Conf, Box<dyn std::error::Error>> {
|
pub fn load_config(path: &str) -> Result<Conf, Box<dyn std::error::Error>> {
|
||||||
let settings = Config::builder()
|
let settings = Config::builder()
|
||||||
.add_source(config::File::with_name(path))
|
.add_source(config::File::with_name(path))
|
||||||
.build()?;
|
.build()?;
|
||||||
|
|
||||||
let conf = settings
|
let conf = settings.try_deserialize::<Conf>()?;
|
||||||
.try_deserialize::<Conf>()?;
|
|
||||||
|
|
||||||
Ok(conf)
|
Ok(conf)
|
||||||
}
|
}
|
||||||
|
@ -1,2 +1,2 @@
|
|||||||
pub mod conf;
|
pub mod conf;
|
||||||
pub mod redis_ctrl;
|
pub mod redis_ctrl;
|
||||||
|
120
src/main.rs
120
src/main.rs
@ -1,81 +1,77 @@
|
|||||||
|
mod conf;
|
||||||
///
|
///
|
||||||
/// 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
|
||||||
///
|
///
|
||||||
|
|
||||||
mod redis_ctrl;
|
mod redis_ctrl;
|
||||||
mod conf;
|
|
||||||
|
|
||||||
use helios_dac::{Frame,
|
|
||||||
Point,
|
|
||||||
DeviceStatus,
|
|
||||||
// Coordinate,
|
|
||||||
Color};
|
|
||||||
use helios_dac::NativeHeliosDacController;
|
use helios_dac::NativeHeliosDacController;
|
||||||
|
use helios_dac::{
|
||||||
|
// Coordinate,
|
||||||
|
Color,
|
||||||
|
DeviceStatus,
|
||||||
|
Frame,
|
||||||
|
Point,
|
||||||
|
};
|
||||||
use std::sync::atomic::{AtomicBool, Ordering};
|
use std::sync::atomic::{AtomicBool, Ordering};
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
|
||||||
use redis_ctrl::{RedisCtrl,Order};
|
use conf::{load_config, Conf};
|
||||||
use conf::{load_config,Conf};
|
use redis_ctrl::{Order, RedisCtrl};
|
||||||
|
|
||||||
|
const CENTER: (u16, u16) = (2000, 2000);
|
||||||
const CENTER : (u16,u16) = (2000, 2000);
|
|
||||||
|
|
||||||
pub fn main() {
|
pub fn main() {
|
||||||
|
|
||||||
let config = match load_config("Settings") {
|
let config = match load_config("Settings") {
|
||||||
Ok(c) => c,
|
Ok(c) => c,
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
panic!("Unable to load config file: {:?}", err)
|
panic!("Unable to load config file: {:?}", err)
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
let rs = match RedisCtrl::new() {
|
let rs = match RedisCtrl::new() {
|
||||||
Ok(rs) => rs,
|
Ok(rs) => rs,
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
panic!("Unable to connect to redis: {:?}", err);
|
panic!("Unable to connect to redis: {:?}", err);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
match run_dac(config, rs) {
|
match run_dac(config, rs) {
|
||||||
Ok(()) => {},
|
Ok(()) => {}
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
panic!("Error: {:?}", err)
|
panic!("Error: {:?}", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn run_dac(
|
fn run_dac(config: Conf, mut rs: RedisCtrl) -> Result<(), Box<dyn std::error::Error>> {
|
||||||
config: Conf,
|
|
||||||
mut rs: RedisCtrl
|
|
||||||
) -> Result<(), Box<dyn std::error::Error>> {
|
|
||||||
let running = Arc::new(AtomicBool::new(true));
|
let running = Arc::new(AtomicBool::new(true));
|
||||||
let r = running.clone();
|
let r = running.clone();
|
||||||
|
|
||||||
ctrlc::set_handler(move || {
|
ctrlc::set_handler(move || {
|
||||||
r.store(false, Ordering::SeqCst);
|
r.store(false, Ordering::SeqCst);
|
||||||
}).expect("Error setting Ctrl-C handler");
|
})
|
||||||
|
.expect("Error setting Ctrl-C handler");
|
||||||
|
|
||||||
let controller = NativeHeliosDacController::new()?;
|
let controller = NativeHeliosDacController::new()?;
|
||||||
let devices = controller.list_devices()?;
|
let devices = controller.list_devices()?;
|
||||||
|
|
||||||
let Some(device) = devices.into_iter().next() else {
|
let Some(device) = devices.into_iter().next() else {
|
||||||
println!("Unable to find an helios device");
|
println!("Unable to find an helios device");
|
||||||
return Ok(());
|
return Ok(());
|
||||||
};
|
};
|
||||||
|
|
||||||
let mut device = device.open()?;
|
|
||||||
|
|
||||||
while running.load(Ordering::SeqCst) {
|
|
||||||
let order = rs.get_order(0)?;
|
|
||||||
if order != Order::Draw {
|
|
||||||
println!("{:?}", order);
|
|
||||||
}
|
|
||||||
|
|
||||||
let frame = get_next_frame(1000, &mut rs)?;
|
let mut device = device.open()?;
|
||||||
|
|
||||||
while let Ok(DeviceStatus::NotReady) = device.status() {
|
while running.load(Ordering::SeqCst) {
|
||||||
}
|
let order = rs.get_order(0)?;
|
||||||
|
if order != Order::Draw {
|
||||||
|
println!("{:?}", order);
|
||||||
|
}
|
||||||
|
|
||||||
|
let frame = get_next_frame(1000, &mut rs)?;
|
||||||
|
|
||||||
|
while let Ok(DeviceStatus::NotReady) = device.status() {}
|
||||||
device.write_frame(frame)?;
|
device.write_frame(frame)?;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -84,45 +80,39 @@ fn run_dac(
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
fn get_next_frame(
|
fn get_next_frame(speed: u32, rs: &mut RedisCtrl) -> Result<Frame, Box<dyn std::error::Error>> {
|
||||||
speed: u32,
|
|
||||||
rs: &mut RedisCtrl
|
|
||||||
) -> Result<Frame, Box<dyn std::error::Error>> {
|
|
||||||
|
|
||||||
let line = rs.get("/pl/0/0")?;
|
let line = rs.get("/pl/0/0")?;
|
||||||
let line : Vec<Point> = line.iter()
|
let line: Vec<Point> = line.iter().map(tuple_to_point).collect();
|
||||||
.map(tuple_to_point)
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
let mut line2 = vec![];
|
let mut line2 = vec![];
|
||||||
while line2.len() < 48 {
|
while line2.len() < 48 {
|
||||||
for p in &line {
|
for p in &line {
|
||||||
line2.push(*p);
|
line2.push(*p);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(Frame::new(speed, line2))
|
Ok(Frame::new(speed, line2))
|
||||||
}
|
}
|
||||||
|
|
||||||
fn tuple_to_point(tpl: &(f32,f32,u32)) -> Point {
|
fn tuple_to_point(tpl: &(f32, f32, u32)) -> Point {
|
||||||
let (x, y, col) = tpl;
|
let (x, y, col) = tpl;
|
||||||
|
|
||||||
let r = (col >> 16) as u8 ;
|
let r = (col >> 16) as u8;
|
||||||
let g = ((col >> 8) & 255) as u8 ;
|
let g = ((col >> 8) & 255) as u8;
|
||||||
let b = (col & 255) as u8 ;
|
let b = (col & 255) as u8;
|
||||||
|
|
||||||
let x = CENTER.0 + *x as u16;
|
let x = CENTER.0 + *x as u16;
|
||||||
let y = CENTER.1 + *y as u16;
|
let y = CENTER.1 + *y as u16;
|
||||||
|
|
||||||
if x >= 4096 || y >= 4096 {
|
if x >= 4096 || y >= 4096 {
|
||||||
println!("WARN: coordinate out of range: {} {}", x, y);
|
println!("WARN: coordinate out of range: {} {}", x, y);
|
||||||
}
|
}
|
||||||
let x = x.clamp(0, 4095);
|
let x = x.clamp(0, 4095);
|
||||||
let y = y.clamp(0, 4095);
|
let y = y.clamp(0, 4095);
|
||||||
|
|
||||||
Point {
|
Point {
|
||||||
coordinate: (x, y).into(),
|
coordinate: (x, y).into(),
|
||||||
color: Color::new(r, g, b),
|
color: Color::new(r, g, b),
|
||||||
intensity: 0xFF
|
intensity: 0xFF,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -1,12 +1,8 @@
|
|||||||
use redis::{
|
use redis::{Client, Commands, Connection};
|
||||||
Client,
|
|
||||||
Connection,
|
|
||||||
Commands
|
|
||||||
};
|
|
||||||
use ron::de::from_str;
|
use ron::de::from_str;
|
||||||
|
|
||||||
#[repr(u8)]
|
#[repr(u8)]
|
||||||
#[derive(Debug,PartialEq)]
|
#[derive(Debug, PartialEq)]
|
||||||
pub enum Order {
|
pub enum Order {
|
||||||
Draw = 0,
|
Draw = 0,
|
||||||
Edh, //homography
|
Edh, //homography
|
||||||
@ -16,71 +12,62 @@ pub enum Order {
|
|||||||
ClientKey,
|
ClientKey,
|
||||||
Intensity,
|
Intensity,
|
||||||
Kpps,
|
Kpps,
|
||||||
ColorBalance
|
ColorBalance,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl TryFrom<u8> for Order {
|
impl TryFrom<u8> for Order {
|
||||||
type Error = String;
|
type Error = String;
|
||||||
|
|
||||||
fn try_from(value: u8) -> Result<Self, Self::Error> {
|
fn try_from(value: u8) -> Result<Self, Self::Error> {
|
||||||
use Order::*;
|
use Order::*;
|
||||||
|
|
||||||
if value > 8 {
|
if value > 8 {
|
||||||
return Err("order out of range".to_string());
|
return Err("order out of range".to_string());
|
||||||
}
|
}
|
||||||
|
|
||||||
Ok(match value {
|
Ok(match value {
|
||||||
0 => Draw,
|
0 => Draw,
|
||||||
1 => Edh,
|
1 => Edh,
|
||||||
2 => Black,
|
2 => Black,
|
||||||
3 => Grid,
|
3 => Grid,
|
||||||
4 => Resampler,
|
4 => Resampler,
|
||||||
5 => ClientKey,
|
5 => ClientKey,
|
||||||
6 => Intensity,
|
6 => Intensity,
|
||||||
7 => Kpps,
|
7 => Kpps,
|
||||||
8 => ColorBalance,
|
8 => ColorBalance,
|
||||||
_ => panic!("Can't be there")
|
_ => panic!("Can't be there"),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub type Line = Vec<(f32,f32,u32)>;
|
pub type Line = Vec<(f32, f32, u32)>;
|
||||||
|
|
||||||
pub struct RedisCtrl {
|
pub struct RedisCtrl {
|
||||||
pub client: Client,
|
pub client: Client,
|
||||||
pub connection: Connection
|
pub connection: Connection,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl RedisCtrl {
|
impl RedisCtrl {
|
||||||
|
|
||||||
pub fn new() -> Result<Self, Box<dyn std::error::Error>> {
|
pub fn new() -> Result<Self, Box<dyn std::error::Error>> {
|
||||||
let client = Client::open("redis://127.0.0.1/")?;
|
let client = Client::open("redis://127.0.0.1/")?;
|
||||||
let connection = client.get_connection()?;
|
let connection = client.get_connection()?;
|
||||||
Ok(RedisCtrl { client, connection })
|
Ok(RedisCtrl { client, connection })
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn get(
|
pub fn get(&mut self, key: &str) -> Result<Line, Box<dyn std::error::Error>> {
|
||||||
&mut self,
|
let val: String = self.connection.get(key)?;
|
||||||
key: &str
|
let line: Line = from_str(&val)?;
|
||||||
) -> Result<Line, Box<dyn std::error::Error>> {
|
Ok(line)
|
||||||
let val : String = self.connection.get(key)?;
|
|
||||||
let line : Line = from_str(&val)?;
|
|
||||||
Ok(line)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn get_order(
|
|
||||||
&mut self,
|
|
||||||
id: u8
|
|
||||||
) -> Result<Order, Box<dyn std::error::Error>> {
|
|
||||||
let path = format!("/order/{id}");
|
|
||||||
let val : u8 = self.connection.get(path.clone())?;
|
|
||||||
|
|
||||||
if val == 1 || val >= 4 {
|
|
||||||
self.connection.set(path, 0)?;
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(val.try_into()?)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn get_order(&mut self, id: u8) -> Result<Order, Box<dyn std::error::Error>> {
|
||||||
|
let path = format!("/order/{id}");
|
||||||
|
let val: u8 = self.connection.get(path.clone())?;
|
||||||
|
|
||||||
|
if val == 1 || val >= 4 {
|
||||||
|
self.connection.set(path, 0)?;
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(val.try_into()?)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user