1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

/// Create an array of ones
///
/// # Examples
///
/// ```
/// use arr_rs::prelude::*;
///
/// let arr = array_ones!(i32, 8);
/// assert_eq!(array!(i32, [1, 1, 1, 1, 1, 1, 1, 1]), arr);
/// let arr = array_ones!(i32, 2, 2, 2);
/// assert_eq!(array!(i32, [[[1, 1], [1, 1]], [[1, 1], [1, 1]]]), arr);
/// ```
#[macro_export]
macro_rules! array_ones {
    ($([$($nested:expr),*]),* $(,)*) => {{
        compile_error!("array_ones! only accepts a flat list of elements");
    }};
    ($tt:ty, $([$($nested:expr),*]),* $(,)*) => {{
        compile_error!("array_ones! only accepts a flat list of elements");
    }};
    ($tt:ty, $($x:expr),* $(,)*) => {{
        Array::<$tt>::ones(vec![$($x,)*])
    }};
}