gmt_m2_ctrl_fsm_piezo_135/
lib.rs

1#![allow(non_upper_case_globals)]
2#![allow(non_camel_case_types)]
3#![allow(non_snake_case)]
4#![allow(improper_ctypes)]
5
6include!(concat!(env!("OUT_DIR"), "/bindings.rs"));
7
8include!(concat!(env!("OUT_DIR"), "/controller.rs"));
9
10#[cfg(test)]
11mod tests {
12    use std::f64::EPSILON;
13
14    use super::*;
15    use matio_rs::MatFile;
16
17    #[test]
18    fn impulse() {
19        let path = std::path::Path::new(env!("CARGO_MANIFEST_DIR"))
20            .join("..")
21            .join("simulink-models")
22            .join("m2fsm_tests.mat");
23        let mat = MatFile::load(path.to_str().unwrap()).unwrap();
24        let pzt_135_Cfb_imp_t: Vec<f64> = mat.var("pzt_135_Cfb_imp_t").unwrap();
25        let pzt_135_Cfb_imp_y: Vec<f64> = mat.var("pzt_135_Cfb_imp_y").unwrap();
26
27        let n = pzt_135_Cfb_imp_t.len();
28
29        let sim_y: Vec<_> = pzt_135_Cfb_imp_y.chunks(n).take(1).flatten().collect();
30        dbg!(&sim_y[..10]);
31
32        let mut ctrl = FsmPiezo135::new();
33        let mut y = vec![];
34        for i in 0..n {
35            ctrl.inputs
36                .pzt_error
37                .iter_mut()
38                .take(1)
39                .for_each(|e| *e = if i == 0 { 1000f64 } else { 0f64 });
40            ctrl.step();
41            y.push(ctrl.outputs.pzt_control[0]);
42        }
43        dbg!(&y[..10]);
44
45        let n = sim_y.len();
46        let y_err = (sim_y
47            .into_iter()
48            .zip(&y)
49            .map(|(sim_y, y)| 1. - y / sim_y)
50            .map(|x| x * x)
51            .sum::<f64>()
52            / (n as f64))
53            .sqrt();
54        assert!(dbg!(y_err) < EPSILON * 1e3);
55    }
56}