1use 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}