110 lines
3.2 KiB
Rust
110 lines
3.2 KiB
Rust
use crate::errors::{LJError, LJResult};
|
|
use crate::transformer;
|
|
use config::Config;
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
#[derive(Serialize, Deserialize, Debug, Clone)]
|
|
pub struct Conf {
|
|
pub laser_id: u8,
|
|
pub debug: bool,
|
|
pub redis_url: String,
|
|
pub dac: DacFamily,
|
|
#[serde(default)]
|
|
pub transformers: Vec<TransformConf>,
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Debug, Clone)]
|
|
pub enum DacFamily {
|
|
#[serde(rename = "helios")]
|
|
Helios(HeliosConf),
|
|
#[serde(rename = "etherdream")]
|
|
Etherdream(EtherDreamConf),
|
|
#[serde(rename = "dummy")]
|
|
Dummy,
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Debug, Clone)]
|
|
pub struct HeliosConf {
|
|
pub id: u8,
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Debug, Clone)]
|
|
pub struct EtherDreamConf {
|
|
pub ip: String,
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, Debug, Clone)]
|
|
pub enum TransformConf {
|
|
#[serde(rename = "translate")]
|
|
Translate(transformer::Translate),
|
|
#[serde(rename = "replicate")]
|
|
Replicate(transformer::Replicate),
|
|
#[serde(rename = "rotate")]
|
|
Rotate(transformer::Rotate),
|
|
#[serde(rename = "flip_horizontal")]
|
|
FlipH(transformer::FlipHorizontal),
|
|
#[serde(rename = "flip_vertical")]
|
|
FlipV(transformer::FlipVertical),
|
|
#[serde(rename = "grid")]
|
|
Grid(transformer::Grid),
|
|
#[serde(rename = "homography")]
|
|
Homography(transformer::Homography),
|
|
#[serde(rename = "angle_correction")]
|
|
AngleOptimisation(transformer::AngleOptimisation),
|
|
}
|
|
|
|
impl Conf {
|
|
pub fn new(path: &str) -> LJResult<Conf> {
|
|
let settings = Config::builder()
|
|
.add_source(config::File::with_name(path))
|
|
.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<Box<dyn transformer::Transformers>> {
|
|
let mut v = vec![];
|
|
for t in &self.transformers {
|
|
let t: Box<dyn transformer::Transformers> = 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);
|
|
}
|
|
}
|