1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

mod sealed {
    pub trait Sealed {}
}
use sealed::Sealed;

/// 将 SIMD vector 中的 lanes 的数量指定为类型。
pub struct LaneCount<const LANES: usize>;

impl<const LANES: usize> LaneCount<LANES> {
    /// 具有这么多 lanes 的位掩码中的字节数。
    pub const BITMASK_LEN: usize = (LANES + 7) / 8;
}

/// 静态保证 lane 计数被标记为受支持。
///
/// 这个 trait 是*密封的*: 下面的实现者列表是全部的。
/// 用户无法将其他 `LaneCount<N>` 值标记为受支持。
/// 只有支持 lane 计数的 SIMD vectors 是可构造的。
pub trait SupportedLaneCount: Sealed {
    #[doc(hidden)]
    type BitMask: Copy + Default + AsRef<[u8]> + AsMut<[u8]>;
}

impl<const LANES: usize> Sealed for LaneCount<LANES> {}

macro_rules! supported_lane_count {
    ($($lanes:literal),+) => {
        $(
            impl SupportedLaneCount for LaneCount<$lanes> {
                type BitMask = [u8; ($lanes + 7) / 8];
            }
        )+
    };
}

supported_lane_count!(1, 2, 4, 8, 16, 32, 64);
#[cfg(feature = "all_lane_counts")]
supported_lane_count!(
    3, 5, 6, 7, 9, 10, 11, 12, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30,
    31, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55,
    56, 57, 58, 59, 60, 61, 62, 63
);