peroxide/util/
api.rs

1//! Choose api - MATLAB, R, Python
2
3use crate::structure::matrix::*;
4
5pub trait MATLAB {
6    fn new(s: &str) -> Self;
7}
8
9pub trait PYTHON {
10    fn new<T>(v: Vec<Vec<T>>) -> Self
11    where
12        T: Into<f64> + Copy;
13}
14
15pub trait R {
16    fn new<T>(v: Vec<T>, x: usize, y: usize, shape: Shape) -> Self
17    where
18        T: Into<f64>;
19}
20
21impl MATLAB for Matrix {
22    fn new(s: &str) -> Self where {
23        let str_rows_temp: Vec<&str> = s.split(';').collect();
24        let str_rows = str_rows_temp
25            .into_iter()
26            .filter(|&t| t != "")
27            .collect::<Vec<&str>>();
28        let r = str_rows.len();
29        let str_data = str_rows
30            .into_iter()
31            .map(|x| x.trim().split(' ').collect::<Vec<&str>>())
32            .collect::<Vec<Vec<&str>>>();
33        let c = str_data[0].len();
34        let data = str_data
35            .into_iter()
36            .flat_map(|t| {
37                t.into_iter()
38                    .map(|x| x.parse::<f64>().unwrap())
39                    .collect::<Vec<f64>>()
40            })
41            .collect::<Vec<f64>>();
42        matrix(data, r, c, Row)
43    }
44}
45
46impl PYTHON for Matrix {
47    fn new<T>(v: Vec<Vec<T>>) -> Self
48    where
49        T: Into<f64> + Copy,
50    {
51        let r = v.len();
52        let c = v[0].len();
53        let mut data = vec![0f64; r * c];
54        for i in 0..r {
55            for j in 0..c {
56                let idx = i * c + j;
57                data[idx] = v[i][j].into();
58            }
59        }
60        matrix(data, r, c, Row)
61    }
62}
63
64impl R for Matrix {
65    fn new<T>(v: Vec<T>, x: usize, y: usize, shape: Shape) -> Self
66    where
67        T: Into<f64>,
68    {
69        matrix(
70            v.into_iter().map(|t| t.into()).collect::<Vec<f64>>(),
71            x,
72            y,
73            shape,
74        )
75    }
76}