konst_macro_rules/
array_macros.rs

1use core::mem::MaybeUninit;
2
3#[macro_export]
4macro_rules! array_map {
5    ($array:expr, |$param:tt $(: $type:ty)?| $mapper:expr $(,)? ) => (
6        match $array {
7            ref array => {
8                let array = $crate::__::assert_array(array);
9                let len = array.len();
10                let mut out = $crate::__::uninit_array(array);
11
12                let mut i = 0;
13                while i < len {
14                    let $param $(: $type)? = array[i];
15                    out[i] = $crate::__::MaybeUninit::new($mapper);
16                    i += 1;
17                }
18
19                unsafe{
20                    $crate::__::AssumInitCopyArray{uninit: out}.init
21                }
22            }
23        }
24    );
25    ($array:expr, | $($anything:tt)* ) => {
26        compile_error!("expected the closure to take an argument")
27    };
28    ($array:expr, $function:expr $(,)?) => {
29        $crate::array_map!($array, |x| $function(x))
30    };
31}
32
33#[inline(always)]
34pub const fn assert_array<T, const N: usize>(array: &[T; N]) -> &[T; N] {
35    array
36}
37
38struct UNINIT<T>(T);
39
40impl<T> UNINIT<T> {
41    pub const V: MaybeUninit<T> = MaybeUninit::uninit();
42}
43
44#[repr(C)]
45pub union AssumInitCopyArray<T: Copy, const N: usize> {
46    pub uninit: [MaybeUninit<T>; N],
47    pub init: [T; N],
48}
49
50#[inline(always)]
51pub const fn uninit_array<T, U, const N: usize>(_input: &[T; N]) -> [MaybeUninit<U>; N] {
52    [UNINIT::V; N]
53}
konst_macro_rules/array_macros.rs

konst_macro_rules/
array_macros.rs
