use crate::{ArrayLength, GenericArray, GenericArrayImplEven, GenericArrayImplOdd};
use const_default::ConstDefault;
impl<T, U: ConstDefault> ConstDefault for GenericArrayImplEven<T, U> {
const DEFAULT: Self = Self {
parents: [U::DEFAULT; 2],
_marker: core::marker::PhantomData,
};
}
impl<T: ConstDefault, U: ConstDefault> ConstDefault for GenericArrayImplOdd<T, U> {
const DEFAULT: Self = Self {
parents: [U::DEFAULT; 2],
data: T::DEFAULT,
};
}
impl<T, U: ArrayLength> ConstDefault for GenericArray<T, U>
where
U::ArrayType<T>: ConstDefault,
{
const DEFAULT: Self = Self {
data: ConstDefault::DEFAULT,
};
}
impl<T: ConstDefault, U: ArrayLength> GenericArray<T, U>
where
Self: ConstDefault,
{
#[inline(always)]
pub const fn const_default() -> Self {
Self::DEFAULT
}
}
#[cfg(test)]
mod tests {
use crate::{arr, typenum::U4, GenericArray};
use const_default::ConstDefault;
#[test]
fn const_default_works() {
let a: GenericArray<i32, U4> = <GenericArray<i32, U4> as ConstDefault>::DEFAULT;
assert_eq!(a, arr![0, 0, 0, 0]);
let b: GenericArray<i32, U4> = GenericArray::const_default();
assert_eq!(b, arr![0, 0, 0, 0]);
}
}