arr_rs/macros/
arange.rs

1/// Create new 2d array with ones on the diagonal and zeros elsewhere
2///
3/// # Examples
4///
5/// ```
6/// use arr_rs::prelude::*;
7///
8/// let arr = array_arange!(i32, 0, 4).unwrap();
9/// assert_eq!(array!(i32, [0, 1, 2, 3, 4]).unwrap(), arr);
10/// let arr = array_arange!(i32, 0, 7, 2).unwrap();
11/// assert_eq!(array!(i32, [0, 2, 4, 6]).unwrap(), arr);
12/// ```
13#[macro_export]
14macro_rules! array_arange {
15    (Tuple2<$t1:ty, $t2:ty>, $($tt:tt)*) => {{
16        compile_error!("only `Numeric` types are supported")
17    }};
18    (Tuple3<$t1:ty, $t2:ty, $t3:ty>, $($tt:tt)*) => {{
19        compile_error!("only `Numeric` types are supported")
20    }};
21    (List<$t1:ty>, $($tt:tt)*) => {{
22        compile_error!("only `Numeric` types are supported")
23    }};
24    (char, $($tt:tt)*) => {{
25        compile_error!("only `Numeric` types are supported")
26    }};
27    (String, $($tt:tt)*) => {{
28        compile_error!("only `Numeric` types are supported")
29    }};
30    ($tt:ty, $n:expr, $m:expr) => {
31        Array::<$tt>::arange($n, $m, None)
32    };
33    ($tt:ty, $n:expr, $m:expr, $k:expr) => {{
34        Array::<$tt>::arange($n, $m, Some($k))
35    }};
36}
arr_rs/macros/arange.rs

arr_rs/macros/
arange.rs
