const_shader_layout 0.4.0

Compile-time checks to ensure structs conform to WGSL's memory layout rules
Documentation 
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

use core::cmp::Reverse;
use core::num::{NonZero, Saturating, Wrapping};

use crate::{
    ShaderLayout, ShaderLayoutArrayElement, ShaderLayoutCompat, ShaderLayoutCompatArrayElement,
    impl_shader_layout_array_element, impl_shader_layout_compat,
};

// Scalar
impl_shader_layout_compat!(
    i16,
    u16,
    NonZero<i16>,
    NonZero<u16>,
    f32,
    i32,
    u32,
    NonZero<i32>,
    NonZero<u32>,
);

// Array
impl_shader_layout_array_element!(
    i16,
    u16,
    NonZero<i16>,
    NonZero<u16>,
    f32,
    i32,
    u32,
    NonZero<i32>,
    NonZero<u32>,
);

impl<T: ShaderLayoutCompatArrayElement, const N: usize> ShaderLayoutCompat for [T; N] {}
impl<T: ShaderLayoutCompatArrayElement, const N: usize> ShaderLayoutCompatArrayElement for [T; N] {}

impl<T: ShaderLayoutArrayElement, const N: usize> ShaderLayout for [T; N] {}
impl<T: ShaderLayoutArrayElement, const N: usize> ShaderLayoutArrayElement for [T; N] {}

macro_rules! impl_transparent_generic_wrapper {
    ($($ty:ty),+$(,)?) => {
        $(
            impl<T: ShaderLayout> ShaderLayout for $ty {
                const ALIGN: NonZero<u64> = NonZero::new(align_of::<T>() as u64).unwrap();
            }
            impl<T: ShaderLayoutCompat> ShaderLayoutCompat for $ty {}

            impl<T: ShaderLayoutArrayElement> ShaderLayoutArrayElement for $ty {}
            impl<T: ShaderLayoutCompatArrayElement> ShaderLayoutCompatArrayElement
                for $ty
            {}
        )+
    };
}

// Derive the traits since them are `repr(transparent)`
impl_transparent_generic_wrapper!(Saturating<T>, Wrapping<T>, Reverse<T>);